시스템에서 드라이브 추가 혹은 제거되었을 때를 자동으로 감시하는 코드

출처 : http://www.codeproject.com/useritems/Drive_Detect.asp

시스템에서 플로피/하드/CD-ROM/이동 드라이브가 추가 혹은 제거되었을 때를 자동으로 감시하는 코드입니다. 우선 헤더 파일을 프로젝트에 추가하여 줍니다.

#include <dbt.h>

아래의 메시지 핸들러를 추가합니다. 클래스 위저드를 통해서 추가할 수 없으니 수동으로 추가하여 주세요.

bool inline IsBitSet (DWORD64 dwMask, UINT nTHBit)
{
DWORD64 dwBit = 1;
dwBit <<= nTHBit; dwMask &= dwBit; return dwMask ? true : false; } BOOL CT1Dlg::OnDeviceChange( UINT nEventType, DWORD dwData ) { BOOL bReturn = CWnd::OnDeviceChange (nEventType, dwData); DEV_BROADCAST_VOLUME *volume = (DEV_BROADCAST_VOLUME *)dwData; CString log; if (nEventType == DBT_DEVICEARRIVAL) { if (volume->dbcv_devicetype == DBT_DEVTYP_VOLUME)
{
for (int n = 0; n < 32; n++) { if (IsBitSet (volume->dbcv_unitmask, n))
{
log.Format ("Drive %c: Inserted\n", n + 'A');
AfxMessageBox(log);
}
}
}
}

if (nEventType == DBT_DEVICEREMOVECOMPLETE)
{
if (volume->dbcv_devicetype == DBT_DEVTYP_VOLUME)
{
for (int n = 0; n < 32; n++) { if (IsBitSet (volume->dbcv_unitmask, n))
{
log.Format ("Drive %c: Removed\n", n + 'A');
AfxMessageBox(log);
}
}
}
}

return bReturn;
}