Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

122 lines
4.2 KiB

// 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);
}