初始化多个磁盘时出现脱机现象

lwei2

如题，在MFC中对多个磁盘进行初始化，代码如下：

1. DWORD CreateDisk(DWORD disk, WORD partNum)
   
2. {
   
3.     HANDLE hDevice;               // handle to the drive to be examined
   
4.     BOOL result;                  // results flag
   
5.     DWORD readed;                 // discard results
   
6.     DWORD ret;
   
7.     WORD i;
   
8.     CHAR diskPath[DISK_PATH_LEN];
   
9.     DISK_GEOMETRY pdg;
   
10.     DWORD sectorSize;
    
11.     DWORD signature;
    
12.     LARGE_INTEGER diskSize;
    
13.     LARGE_INTEGER partSize;
    
14.     BYTE actualPartNum;
    
15.    
    
16.     DWORD layoutStructSize;
    
17.     DRIVE_LAYOUT_INFORMATION_EX *dl;
    
18.     CREATE_DISK newDisk;
    
19. 
    
20.     sprintf(diskPath, "\\\\.\\PhysicalDrive%d", disk);
    
21. 
    
22.     actualPartNum = 4;
    
23.     if (partNum > actualPartNum)
    
24.     {
    
25.         return (WORD)-1;
    
26.     }
    
27.    
    
28.     hDevice = CreateFile(
    
29.                 diskPath,
    
30.                 GENERIC_READ|GENERIC_WRITE,
    
31.                 FILE_SHARE_READ|FILE_SHARE_WRITE,
    
32.                 NULL,           //default security attributes  
    
33.                 OPEN_EXISTING, // disposition  
    
34.                 0,              // file attributes  
    
35.                 NULL
    
36.                 );
    
37.     if (hDevice == INVALID_HANDLE_VALUE) // cannot open the drive
    
38.     {
    
39.         fprintf(stderr, "CreateFile() Error: %ld\n", GetLastError());
    
40.         return DWORD(-1);
    
41.     }
    
42.    
    
43.     // Create primary partition MBR
    
44.     newDisk.PartitionStyle = PARTITION_STYLE_MBR;
    
45.     signature = (DWORD)time(NULL);     //get signature from current time
    
46.     newDisk.Mbr.Signature = signature;
    
47.    
    
48.     result = DeviceIoControl(
    
49.                 hDevice,
    
50.                 IOCTL_DISK_CREATE_DISK,
    
51.                 &newDisk,
    
52.                 sizeof(CREATE_DISK),
    
53.                 NULL,
    
54.                 0,
    
55.                 &readed,
    
56.                 NULL
    
57.                 );
    
58.     if (!result)
    
59.     {
    
60.         fprintf(stderr, "IOCTL_DISK_CREATE_DISK Error: %ld\n", GetLastError());
    
61.         (void)CloseHandle(hDevice);
    
62.         return DWORD(-1);
    
63.     }
    
64. 
    
65.     //fresh the partition table
    
66.     result = DeviceIoControl(
    
67.                 hDevice,
    
68.                 IOCTL_DISK_UPDATE_PROPERTIES,
    
69.                 NULL,
    
70.                 0,
    
71.                 NULL,
    
72.                 0,
    
73.                 &readed,
    
74.                 NULL
    
75.                 );
    
76.     if (!result)
    
77.     {
    
78.         fprintf(stderr, "IOCTL_DISK_UPDATE_PROPERTIES Error: %ld\n", GetLastError());
    
79.         (void)CloseHandle(hDevice);
    
80.         return DWORD(-1);
    
81.     }
    
82. 
    
83.     //Now create the partitions
    
84.     ret = GetDriveGeometry(diskPath, &pdg);
    
85.     if ((DWORD)-1 == ret)
    
86.     {
    
87.         return ret;
    
88.     }
    
89.     sectorSize = pdg.BytesPerSector;
    
90.     diskSize.QuadPart = pdg.Cylinders.QuadPart * pdg.TracksPerCylinder *
    
91.                         pdg.SectorsPerTrack * pdg.BytesPerSector;       //calculate the disk size;
    
92.     partSize.QuadPart = diskSize.QuadPart / partNum;
    
93. 
    
94.     layoutStructSize = sizeof(DRIVE_LAYOUT_INFORMATION_EX) + (actualPartNum - 1) * sizeof(PARTITION_INFORMATION_EX);
    
95.     dl = (DRIVE_LAYOUT_INFORMATION_EX*)malloc(layoutStructSize);
    
96.     if (NULL == dl)
    
97.     {
    
98.         (void)CloseHandle(hDevice);
    
99.         return (WORD)-1;
    
100.     }
     
101. 
     
102.     dl->PartitionStyle = (DWORD)PARTITION_STYLE_MBR;
     
103.     dl->PartitionCount = actualPartNum;
     
104.     dl->Mbr.Signature = signature;
     
105. 
     
106.     //clear the unused partitions
     
107.     for (i = 0; i < actualPartNum; i++){
     
108.         dl->PartitionEntry[i].RewritePartition = 1;
     
109.         dl->PartitionEntry[i].Mbr.PartitionType = PARTITION_ENTRY_UNUSED;
     
110.     }
     
111.     //set the profile of the partitions
     
112.     for (i = 0; i < partNum; i++){
     
113.         dl->PartitionEntry[i].PartitionStyle = PARTITION_STYLE_MBR;
     
114.         dl->PartitionEntry[i].StartingOffset.QuadPart =
     
115.             (partSize.QuadPart * i) + ((LONGLONG)(pdg.SectorsPerTrack) * (LONGLONG)(pdg.BytesPerSector));   //32256
     
116.         dl->PartitionEntry[i].PartitionLength.QuadPart = partSize.QuadPart;
     
117.         dl->PartitionEntry[i].PartitionNumber = i + 1;
     
118.         dl->PartitionEntry[i].RewritePartition = TRUE;
     
119.         dl->PartitionEntry[i].Mbr.PartitionType = PARTITION_IFS;
     
120.         dl->PartitionEntry[i].Mbr.BootIndicator = FALSE;
     
121.         dl->PartitionEntry[i].Mbr.RecognizedPartition = TRUE;
     
122.         dl->PartitionEntry[i].Mbr.HiddenSectors =
     
123.             pdg.SectorsPerTrack + (DWORD)((partSize.QuadPart / sectorSize) * i);
     
124.     }
     
125.     //execute the layout  
     
126.     result = DeviceIoControl(
     
127.                 hDevice,
     
128.                 IOCTL_DISK_SET_DRIVE_LAYOUT_EX,
     
129.                 dl,
     
130.                 layoutStructSize,
     
131.                 NULL,
     
132.                 0,
     
133.                 &readed,
     
134.                 NULL
     
135.                 );
     
136.     if (!result)
     
137.     {
     
138.         fprintf(stderr, "IOCTL_DISK_SET_DRIVE_LAYOUT_EX Error: %ld\n", GetLastError());
     
139.         free(dl);
     
140.         (void)CloseHandle(hDevice);
     
141.         return DWORD(-1);
     
142.     }
     
143. 
     
144.     //fresh the partition table
     
145.     result = DeviceIoControl(
     
146.                 hDevice,
     
147.                 IOCTL_DISK_UPDATE_PROPERTIES,
     
148.                 NULL,
     
149.                 0,
     
150.                 NULL,
     
151.                 0,
     
152.                 &readed,
     
153.                 NULL
     
154.                 );
     
155.     if (!result)
     
156.     {
     
157.         fprintf(stderr, "IOCTL_DISK_UPDATE_PROPERTIES Error: %ld\n", GetLastError());
     
158.         free(dl);
     
159.         (void)CloseHandle(hDevice);
     
160.         return DWORD(-1);
     
161.     }
     
162. 
     
163.     free(dl);
     
164.     (void)CloseHandle(hDevice);
     
165.     Sleep(3000);            //wait the operations take effect
     
166.     return 0;
     
167. }

复制代码

以上代码是放进多线程里，然后，在多线程里运行时，对多个磁盘进行初始化，会出现某个磁盘脱机的现象。请问各位大佬，有没有办法解决这个问题？或者以上代码，怎么修改，才能满足初始化多个磁盘并且不会出现脱机的现象呢？劳烦各位大佬们看看，给点意见，小弟在此多谢了！