VC 隐藏驱动器 -win7下测试可用

BOOL ReadRegistryDword(LPCTSTR lpSubKey, LPCTSTR lpName, DWORD *dwValue){    HKEY hKey = 0;    DWORD dwSize = sizeof(*dwValue);    DWORD dwType;    LONG lStatus;    SECURITY_ATTRIBUTES sa;    sa.nLength = sizeof(SECURITY_ATTRIBUTES);    sa.bInheritHandle = TRUE;    sa.lpSecurityDescriptor = NULL;    lStatus = RegOpenKeyEx(HKEY_CURRENT_USER, lpSubKey, 0, KEY_READ, &hKey);    if (lStatus != ERROR_SUCCESS)    {        SetLastError(lStatus);        return FALSE;    }    lStatus = RegQueryValueEx(hKey, lpName, NULL, &dwType, (BYTE *)dwValue, &dwSize);    if (lStatus != ERROR_SUCCESS)    {        RegCloseKey(hKey);        SetLastError(lStatus);        return FALSE;    }    RegCloseKey(hKey);    return dwType == REG_DWORD;}BOOL WriteRegistryDword(LPCTSTR lpSubKey, LPCTSTR lpName, DWORD dwValue){    HKEY hKey = 0;    DWORD dwDisp;    LONG lStatus;    SECURITY_ATTRIBUTES sa;    sa.nLength = sizeof(SECURITY_ATTRIBUTES);    sa.bInheritHandle = TRUE;    sa.lpSecurityDescriptor = NULL;    lStatus = RegCreateKeyEx(HKEY_CURRENT_USER, lpSubKey, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, &sa, &hKey, &dwDisp);    if (lStatus != ERROR_SUCCESS)    {        SetLastError(lStatus);        return FALSE;    }    lStatus = RegSetValueEx(hKey, lpName, 0, REG_DWORD, (BYTE *)&dwValue, sizeof dwValue);    if (lStatus != ERROR_SUCCESS)    {        RegCloseKey(hKey);        SetLastError(lStatus);        return FALSE;    }    RegCloseKey(hKey);    return TRUE;}void HideVolume( int nDriveNo){    DWORD dwResult;    DEV_BROADCAST_VOLUME* p01 = new DEV_BROADCAST_VOLUME;    p01->dbcv_devicetype = 2;    p01->dbcv_flags = 0;    p01->dbcv_size = 20;    p01->dbcv_unitmask = (1 << nDriveNo);    char szRoot[] = { (char)(nDriveNo + 'A'), ':', '\\', 0 };    SHChangeNotify(SHCNE_DRIVEREMOVED, SHCNF_PATH, szRoot, NULL);    DWORD dwRecipients = BSM_APPLICATIONS;    BroadcastSystemMessage(BSF_IGNORECURRENTTASK | BSF_POSTMESSAGE | BSF_FORCEIFHUNG, &dwRecipients,        WM_DEVICECHANGE, DBT_DEVNODES_CHANGED, NULL);    dwRecipients = BSM_APPLICATIONS;    BroadcastSystemMessage(BSF_IGNORECURRENTTASK | BSF_POSTMESSAGE | BSF_FORCEIFHUNG, &dwRecipients,        WM_DEVICECHANGE, DBT_DEVICEREMOVECOMPLETE, (LPARAM)p01);    DWORD value = 0;    LPCTSTR skey = _T("Software\\Microsoft\\Windows\\CurrentVersion\\Policies\\Explorer");    if (!ReadRegistryDword(skey, _T("NoDrives"), &value))    {        value = 0;    }    value |= (1 << nDriveNo);    WriteRegistryDword(skey, _T("NoDrives"), value);    SHChangeNotify(SHCNE_DRIVEREMOVED, SHCNF_PATH, szRoot, NULL);    //DWORD dwRecipients = BSM_APPLICATIONS;    BroadcastSystemMessage(BSF_IGNORECURRENTTASK | BSF_POSTMESSAGE | BSF_FORCEIFHUNG, &dwRecipients,        WM_DEVICECHANGE, DBT_DEVNODES_CHANGED, NULL);    dwRecipients = BSM_APPLICATIONS;    BroadcastSystemMessage(BSF_IGNORECURRENTTASK | BSF_POSTMESSAGE | BSF_FORCEIFHUNG, &dwRecipients,        WM_DEVICECHANGE, DBT_DEVICEREMOVECOMPLETE, (LPARAM)p01);    delete[] p01;    SendMessageTimeout(HWND_BROADCAST, WM_SETTINGCHANGE, 0, LPARAM("Policy"), SMTO_ABORTIFHUNG, 500, &dwResult);}void ShowVolume(int nDriveNo){    DWORD dwResult;    DEV_BROADCAST_VOLUME* p01 = new DEV_BROADCAST_VOLUME;    p01->dbcv_devicetype = 2;    p01->dbcv_flags = 0;    p01->dbcv_size = 20;    p01->dbcv_unitmask = (1 << nDriveNo);//H    char szRoot[] = { (char)(nDriveNo + 'A'), ':', '\\', 0 };    SHChangeNotify(SHCNE_DRIVEADD, SHCNF_PATH, szRoot, NULL);    SHChangeNotify(SHCNE_UPDATEDIR, SHCNF_PATH, szRoot, NULL);    DWORD dwRecipients = BSM_APPLICATIONS;    BroadcastSystemMessage(BSF_IGNORECURRENTTASK | BSF_POSTMESSAGE | BSF_FORCEIFHUNG, &dwRecipients,        WM_DEVICECHANGE, DBT_DEVNODES_CHANGED, NULL);    dwRecipients = BSM_APPLICATIONS;    BroadcastSystemMessage(BSF_IGNORECURRENTTASK | BSF_POSTMESSAGE | BSF_FORCEIFHUNG, &dwRecipients,        WM_DEVICECHANGE, DBT_DEVICEARRIVAL, (LPARAM)p01);    dwRecipients = BSM_APPLICATIONS;    BroadcastSystemMessage(BSF_IGNORECURRENTTASK | BSF_POSTMESSAGE | BSF_FORCEIFHUNG, &dwRecipients,        WM_DEVICECHANGE, DBT_DEVNODES_CHANGED, NULL);    DWORD value = 0;    LPCTSTR skey = _T("Software\\Microsoft\\Windows\\CurrentVersion\\Policies\\Explorer");    if (!ReadRegistryDword(skey, _T("NoDrives"), &value))    {        value = 0;    }    value &= ~(1 << nDriveNo);    WriteRegistryDword(skey, _T("NoDrives"), value);    SHChangeNotify(SHCNE_DRIVEADD, SHCNF_PATH, szRoot, NULL);    SHChangeNotify(SHCNE_UPDATEDIR, SHCNF_PATH, szRoot, NULL);    // DWORD dwRecipients = BSM_APPLICATIONS;    BroadcastSystemMessage(BSF_IGNORECURRENTTASK | BSF_POSTMESSAGE | BSF_FORCEIFHUNG, &dwRecipients,        WM_DEVICECHANGE, DBT_DEVNODES_CHANGED, NULL);    dwRecipients = BSM_APPLICATIONS;    BroadcastSystemMessage(BSF_IGNORECURRENTTASK | BSF_POSTMESSAGE | BSF_FORCEIFHUNG, &dwRecipients,        WM_DEVICECHANGE, DBT_DEVICEARRIVAL, (LPARAM)p01);    dwRecipients = BSM_APPLICATIONS;    BroadcastSystemMessage(BSF_IGNORECURRENTTASK | BSF_POSTMESSAGE | BSF_FORCEIFHUNG, &dwRecipients,        WM_DEVICECHANGE, DBT_DEVNODES_CHANGED, NULL);    delete[] p01;    SendMessageTimeout(HWND_BROADCAST, WM_SETTINGCHANGE, 0, LPARAM("Policy"), SMTO_ABORTIFHUNG, 500, &dwResult);}