C++ 實作磁盤初始化

磁盤在聯機後要初始化，同樣也有兩種方法，一種是調用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);