C++ 实现磁盘初始化

磁盘在联机后要初始化，同样也有两种方法，一种是调用IOCTL_DISK_CREATE_DISK，还有一种是调用WMI的Initialize方法。

1.首先说说简单的WMI的方法：

先获取磁盘的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);

C++ 实现磁盘初始化

猜你喜欢