磁盘在联机后要初始化,同样也有两种方法,一种是调用IOCTL_DISK_CREATE_DISK,还有一种是调用WMI的Initialize方法。
1.首先说说简单的WMI的方法:
大致思路同博客:C++ 实现磁盘联机
先获取磁盘的id, 然后执行无参数方法Initialize
核心模块代码如下:
wchar_t msftDiskObjectID[256]; GetWMIMSFTDiskObjectId(msftDiskObjectID, index); BSTR MethodName = SysAllocString(L"Initialize"); CComPtr< IWbemClassObject > pOutParams = NULL; hr = service->ExecMethod(msftDiskObjectID, MethodName, 0, NULL, NULL, &pOutParams, NULL); if (FAILED(hr)) { SysFreeString(MethodName); return -1; } SysFreeString(MethodName);
2. 使用IOCTL_DISK_CREATE_DISK方法的话,需要分成MBR和GPT两种情况处理,因为MBR使用的是磁盘signature,而GPT磁盘使用的是磁盘GUID.
MBR代码如下:
int InitalizeDiskMBR(int index, unsigned int signature) { int retcode = 0; wchar_t name[MAX_PATH]; swprintf_s(name,MAX_PATH,L"\\\\.\\PhysicalDrive%d",index); HANDLE hFile = CreateFileW ( name, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL); if (INVALID_HANDLE_VALUE == hFile) { cout << "Initalize MBR format failed when open disk, error code:" << GetLastError(); return 1; } //判断是否可写 for (int i = 0; i < 10; ++i) { DWORD writablelen; BOOL writable = DeviceIoControl(hFile, IOCTL_DISK_IS_WRITABLE, NULL, 0, NULL, 0, &writablelen, NULL); if (!writable) cout << "Disk is not writable,error code:" << GetLastError(); else break; Sleep(1000); } CREATE_DISK disk; disk.PartitionStyle=PARTITION_STYLE_MBR; disk.Mbr.Signature=signature; DWORD returnedLength; BOOL ret=DeviceIoControl(hFile, IOCTL_DISK_CREATE_DISK, &disk, sizeof(CREATE_DISK), NULL, 0, &returnedLength, NULL); if (!ret) { retcode = 1; cout << "Initalize MBR format failed,error code:" << GetLastError(); } CloseHandle(hFile); return retcode; }
GPT磁盘类似,不同的地方在于传入的参数变成了GUID
核心代码在于CREATE_DISK结构的初始化发生了变化:
CREATE_DISK disk; disk.PartitionStyle = PARTITION_STYLE_GPT; memcpy(&(disk.Gpt.DiskId),guid,sizeof(GUID)); disk.Gpt.MaxPartitionCount = 128; GUIDToString(diskguid,*guid);