初始化磁盘_c++怎么初始化

2022-09-29 11:13:16 浏览数 (2)

大家好,又见面了,我是你们的朋友全栈君。

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

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

大致思路同博客:C 实现磁盘联机

先获取磁盘的id, 然后执行无参数方法Initialize

核心模块代码如下:

代码语言:javascript复制
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代码如下:

代码语言:javascript复制
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结构的初始化发生了变化:

代码语言:javascript复制
CREATE_DISK disk;
disk.PartitionStyle = PARTITION_STYLE_GPT;
memcpy(&(disk.Gpt.DiskId),guid,sizeof(GUID));
disk.Gpt.MaxPartitionCount = 128;
GUIDToString(diskguid,*guid);

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/193370.html原文链接:https://javaforall.cn

0 人点赞