123 lines
4.1 KiB
C++
123 lines
4.1 KiB
C++
|
// Copyright (c) 1999 Microsoft Corporation. All Rights Reserved.
|
||
|
|
#include <stdafx.h>
|
||
|
|
|
||
|
|
#include <winuser.h>
|
||
|
|
#include <windows.h>
|
||
|
|
|
||
|
|
#include "mswebdvd.h"
|
||
|
|
#include "msdvd.h"
|
||
|
|
#include "MediaHndlr.h"
|
||
|
|
#include <dbt.h> // device broadcast structure
|
||
|
|
|
||
|
|
//
|
||
|
|
// Ejection event handler
|
||
|
|
//
|
||
|
|
|
||
|
|
CMediaHandler::CMediaHandler()
|
||
|
|
: m_driveMask( 0 )
|
||
|
|
, m_ejected( false )
|
||
|
|
, m_inserted( false )
|
||
|
|
{
|
||
|
|
m_pDVD = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
CMediaHandler::~CMediaHandler()
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
void CMediaHandler::ClearFlags()
|
||
|
|
{
|
||
|
|
m_ejected = false;
|
||
|
|
m_inserted = false;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool CMediaHandler::SetDrive( TCHAR tcDriveLetter )
|
||
|
|
{
|
||
|
|
if( TEXT('A') <= tcDriveLetter && tcDriveLetter <= TEXT('Z') ) {
|
||
|
|
m_driveMask = 1<< (tcDriveLetter-TEXT('A'));
|
||
|
|
} else if( TEXT('a') <= tcDriveLetter && tcDriveLetter <= TEXT('z')) {
|
||
|
|
m_driveMask = 1<< (tcDriveLetter-TEXT('a'));
|
||
|
|
} else {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
ClearFlags();
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
LRESULT CMediaHandler::WndProc( UINT uMsg, WPARAM wParam, LPARAM lParam )
|
||
|
|
{
|
||
|
|
if (m_pDVD == NULL)
|
||
|
|
return ParentClass::WndProc( uMsg, wParam, lParam);
|
||
|
|
|
||
|
|
switch (uMsg) {
|
||
|
|
case WM_DEVICECHANGE:
|
||
|
|
{
|
||
|
|
switch( wParam ) {
|
||
|
|
case DBT_DEVICEREMOVECOMPLETE:
|
||
|
|
case DBT_DEVICEARRIVAL:
|
||
|
|
case DBT_DEVICEQUERYREMOVE:
|
||
|
|
{
|
||
|
|
PDEV_BROADCAST_HDR pdbch = (PDEV_BROADCAST_HDR) lParam;
|
||
|
|
if(pdbch->dbch_devicetype == DBT_DEVTYP_VOLUME ) {
|
||
|
|
PDEV_BROADCAST_VOLUME pdbcv = (PDEV_BROADCAST_VOLUME) pdbch;
|
||
|
|
if (pdbcv->dbcv_flags == DBTF_MEDIA || wParam == DBT_DEVICEQUERYREMOVE)
|
||
|
|
{
|
||
|
|
// pdbcv->dbcv_unitmask identifies which logical drive
|
||
|
|
switch( wParam ) {
|
||
|
|
case DBT_DEVICEARRIVAL:
|
||
|
|
if( pdbcv->dbcv_unitmask & m_driveMask ) {
|
||
|
|
// send change event
|
||
|
|
if( !m_inserted ) {
|
||
|
|
CComVariant var = 0;
|
||
|
|
m_pDVD->Fire_DVDNotify(EC_DVD_DISC_INSERTED, var, var);
|
||
|
|
}
|
||
|
|
m_inserted = true;
|
||
|
|
m_ejected = false;
|
||
|
|
m_pDVD->SetDiscEjected(false);
|
||
|
|
}
|
||
|
|
// return TRUE .. see the MSDN docs for DBT_DEVICEQUERYREMOVE
|
||
|
|
return TRUE;
|
||
|
|
|
||
|
|
case DBT_DEVICEQUERYREMOVE:
|
||
|
|
if( pdbcv->dbcv_unitmask & m_driveMask ) {
|
||
|
|
if( !m_ejected ) {
|
||
|
|
CComVariant var = 0;
|
||
|
|
m_pDVD->Fire_DVDNotify(EC_DVD_DISC_EJECTED, var, var);
|
||
|
|
}
|
||
|
|
m_ejected = true;
|
||
|
|
m_inserted = false;
|
||
|
|
m_pDVD->SetDiscEjected(true);
|
||
|
|
|
||
|
|
}
|
||
|
|
return TRUE; // grant permission
|
||
|
|
|
||
|
|
case DBT_DEVICEREMOVECOMPLETE:
|
||
|
|
if( pdbcv->dbcv_unitmask & m_driveMask ) {
|
||
|
|
if( !m_ejected ) {
|
||
|
|
CComVariant var = 0;
|
||
|
|
m_pDVD->Fire_DVDNotify(EC_DVD_DISC_EJECTED, var, var);
|
||
|
|
}
|
||
|
|
m_ejected = true;
|
||
|
|
m_inserted = false;
|
||
|
|
m_pDVD->SetDiscEjected(true);
|
||
|
|
}
|
||
|
|
return TRUE; // grant permission
|
||
|
|
|
||
|
|
default:
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
default:
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
default:
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
return ParentClass::WndProc( uMsg, wParam, lParam);
|
||
|
|
}
|