根据盘符安全移除U盘

2. //////////////////////
3. //http://www.codeproject.com/KB/system/RemoveDriveByLetter.aspx
4. //根据盘符停止U盘
5. ////////////////////////
7. BOOL  CPubFunction::StopDeviceByLetter(LPCTSTR lpDisk)  
8. {   
9.     TCHAR szRootPath[] = _T("X://");   // "X:/"  -> for GetDriveType
10.     szRootPath[0] = lpDisk[0];
12.     TCHAR szDevicePath[] = _T("X:");   // "X:"   -> for QueryDosDevice
13.     szDevicePath[0] = lpDisk[0];
15.     CString strLogicDisk;
16.     strLogicDisk.Format(_T("////.//%c:"), lpDisk[0]);   //   "//./X:" 
17.     DWORD DiskNumber   =   -1;   
19.     HANDLE   hVolume = CreateFile(strLogicDisk, 0, FILE_SHARE_READ| FILE_SHARE_WRITE, NULL, OPEN_EXISTING, NULL, NULL);   
20.     if(hVolume   ==   INVALID_HANDLE_VALUE)   
21.     {   
22.         return 0;   
23.     }   
25.     STORAGE_DEVICE_NUMBER  sdn;   
26.     DWORD  dwBytesReturned = 0;   
27.     BOOL res = DeviceIoControl(hVolume, IOCTL_STORAGE_GET_DEVICE_NUMBER, NULL, 0, &sdn, sizeof(sdn), &dwBytesReturned, NULL);   
28.     if(res)   
29.     {   
30.         DiskNumber = sdn.DeviceNumber;   
31.     }   
32.     CloseHandle(hVolume);   
34.     if(DiskNumber==-1   )   
35.     {   
36.         return   0;   
37.     }   
38.     
39.     // get the drive type which is required to match the device numbers correctely
40.     UINT DriveType = GetDriveType(lpDisk);
42.     // get the dos device name (like /device/floppy0) to decide if it's a floppy or not - who knows a better way?
43.     TCHAR szDosDeviceName[MAX_PATH];
44.     res = QueryDosDevice(szDevicePath, szDosDeviceName, MAX_PATH);
45.     if ( !res ) {
46.         return 0;
47.     }
49.     // get the device instance handle of the storage volume by means of a SetupDi enum and matching the device number
50.     DWORD DevInst = GetDrivesDevInstByDeviceNumber(DiskNumber, DriveType, szDosDeviceName);
52.     if ( DevInst == 0 ) {
53.         return 0;
54.     }
56.     PNP_VETO_TYPE VetoType = PNP_VetoTypeUnknown; 
57.     WCHAR VetoNameW[MAX_PATH];
58.     VetoNameW[0] = 0;
59.     BOOL bSuccess = FALSE;
61.     // get drives's parent, e.g. the USB bridge, the SATA port, an IDE channel with two drives!
62.     DWORD DevInstParent = 0;
63.     res = CM_Get_Parent(&DevInstParent, DevInst, 0); 
68.     for ( int tries=1; tries<=3; tries++ ) { // sometimes we need some tries...
70.         VetoNameW[0] = 0;
72.         // CM_Query_And_Remove_SubTree doesn't work for restricted users
73.         //res = CM_Query_And_Remove_SubTreeW(DevInstParent, &VetoType, VetoNameW, MAX_PATH, CM_REMOVE_NO_RESTART); // CM_Query_And_Remove_SubTreeA is not implemented under W2K!
74.         //res = CM_Query_And_Remove_SubTreeW(DevInstParent, NULL, NULL, 0, CM_REMOVE_NO_RESTART);  // with messagebox (W2K, Vista) or balloon (XP)
75.         
76.         //res = CM_Request_Device_EjectW(DevInstParent, &VetoType, VetoNameW, MAX_PATH, 0);
77.         res = CM_Request_Device_EjectW(DevInstParent, NULL, NULL, 0, 0); // with messagebox (W2K, Vista) or balloon (XP)
79.         bSuccess = (res==CR_SUCCESS && VetoType==PNP_VetoTypeUnknown);
80.         if ( bSuccess )  { 
81.             break;
82.         }
84.         Sleep(500); // required to give the next tries a chance!
85.     }
87.     if ( bSuccess ) {
88.     //  printf("Success/n/n");
89.         return 1;
90.     }
92.     //printf("failed/n");
93.     
94.     //printf("Result=0x%2X/n", res);
96.     //if ( VetoNameW[0] ) {
97.     //  printf("VetoName=%ws)/n/n", VetoNameW);
98.     //} 
99.     return 0;
101. }   
102. //-----------------------------------------------------------   
107. //http://www.codeproject.com/KB/system/RemoveDriveByLetter.aspx
108. //----------------------------------------------------------------------
109. // returns the device instance handle of a storage volume or 0 on error
110. //----------------------------------------------------------------------
111. DEVINST CPubFunction::GetDrivesDevInstByDeviceNumber(DWORD DeviceNumber, UINT DriveType, TCHAR* szDosDeviceName)
112. {
113. //  bool IsFloppy = (strstr(szDosDeviceName, "//Floppy") != NULL); // who knows a better way?
115.     GUID* guid;
117.     switch (DriveType) {
118.     case DRIVE_REMOVABLE:
119. //      if ( IsFloppy ) {
120. //          guid = (GUID*)&GUID_DEVINTERFACE_FLOPPY;
121. //      } else {
122.             guid = (GUID*)&GUID_DEVINTERFACE_DISK;
123. //      }
124.         break;
125.     case DRIVE_FIXED:
126.         guid = (GUID*)&GUID_DEVINTERFACE_DISK;
127.         break;
128.     case DRIVE_CDROM:
129.         guid = (GUID*)&GUID_DEVINTERFACE_CDROM;
130.         break;
131.     default:
132.         return 0;
133.     }
135.     // Get device interface info set handle for all devices attached to system
136.     HDEVINFO hDevInfo = SetupDiGetClassDevs(guid, NULL, NULL, DIGCF_PRESENT | DIGCF_DEVICEINTERFACE);
138.     if (hDevInfo == INVALID_HANDLE_VALUE)   {
139.         return 0;
140.     }
142.     // Retrieve a context structure for a device interface of a device information set
143.     DWORD dwIndex = 0;
144.     int res;
146.     BYTE Buf[1024];
147.     PSP_DEVICE_INTERFACE_DETAIL_DATA pspdidd = (PSP_DEVICE_INTERFACE_DETAIL_DATA)Buf;
148.     SP_DEVICE_INTERFACE_DATA         spdid;
149.     SP_DEVINFO_DATA                  spdd;
150.     DWORD                            dwSize;
151.     
152.     spdid.cbSize = sizeof(spdid);
154.     while ( true )  {
155.         res = SetupDiEnumDeviceInterfaces(hDevInfo, NULL, guid, dwIndex, &spdid);
156.         if ( !res ) {
157.             break;
158.         }
160.         dwSize = 0;
161.         SetupDiGetDeviceInterfaceDetail(hDevInfo, &spdid, NULL, 0, &dwSize, NULL); // check the buffer size
163.         if ( dwSize!=0 && dwSize<=sizeof(Buf) ) {
165.             pspdidd->cbSize = sizeof(*pspdidd); // 5 Bytes!
167.             ZeroMemory(&spdd, sizeof(spdd));
168.             spdd.cbSize = sizeof(spdd);
170.             int res = SetupDiGetDeviceInterfaceDetail(hDevInfo, &spdid, pspdidd, dwSize, &dwSize, &spdd);
171.             if ( res ) {
173.                 // in case you are interested in the USB serial number:
174.                 // the device id string contains the serial number if the device has one,
175.                 // otherwise a generated id that contains the '&' char...
176.                 /*
177.                 DEVINST DevInstParent = 0;
178.                 CM_Get_Parent(&DevInstParent, spdd.DevInst, 0); 
179.                 char szDeviceIdString[MAX_PATH];
180.                 CM_Get_Device_ID(DevInstParent, szDeviceIdString, MAX_PATH, 0);
181.                 printf("DeviceId=%s/n", szDeviceIdString);
182.                 */
184.                 // open the disk or cdrom or floppy
185.                 HANDLE hDrive = CreateFile(pspdidd->DevicePath, 0, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL);
186.                 if ( hDrive != INVALID_HANDLE_VALUE ) {
187.                     // get its device number
188.                     STORAGE_DEVICE_NUMBER sdn;
189.                     DWORD dwBytesReturned = 0;
190.                     res = DeviceIoControl(hDrive, IOCTL_STORAGE_GET_DEVICE_NUMBER, NULL, 0, &sdn, sizeof(sdn), &dwBytesReturned, NULL);
191.                     if ( res ) {
192.                         if ( DeviceNumber == (DWORD)sdn.DeviceNumber ) {  // match the given device number with the one of the current device
193.                             CloseHandle(hDrive);
194.                             SetupDiDestroyDeviceInfoList(hDevInfo);
195.                             return spdd.DevInst;
196.                         }
197.                     }
198.                     CloseHandle(hDrive);
199.                 }
200.             }
201.         }
202.         dwIndex++;
203.     }
205.     SetupDiDestroyDeviceInfoList(hDevInfo);
207.     return 0;
208. }
209. //-----------------------------------------------------------

 

这个方法可以用来确定多盘符U盘的多个逻辑盘符所对应的物理U盘。根据盘符可以得到这个设备接口的句柄DevInst， 通过CM_Get_Parent函数可以得到这个句柄对应的父设备的句柄，这个父设备的句柄就是两个设备间的usb pridge,  SATA port, IDE channel 。而同一个物理U盘的各个逻辑盘符的父设备的句柄则相同。