reactos操作系统实现(121)

CreatePartitionDeviceObjects函数是通过分析MBR扇区的数据，然后来创建所有分区对象。具体实现代码如下：

#001 NTSTATUS

#002 NTAPI

#003 CreatePartitionDeviceObjects(

#004 IN PDEVICE_OBJECT PhysicalDeviceObject,

#005 IN PUNICODE_STRING RegistryPath

#006 )

#007 {

#008 CCHAR ntNameBuffer[MAXIMUM_FILENAME_LENGTH];

#009 ULONG partitionNumber = 0;

#010 NTSTATUS status;

#011 PDEVICE_OBJECT deviceObject = NULL;

#012 PDISK_GEOMETRY diskGeometry = NULL;

#013 PDRIVE_LAYOUT_INFORMATION partitionList = NULL;

#014 PDEVICE_EXTENSION deviceExtension;

#015 PDEVICE_EXTENSION physicalDeviceExtension;

#016 PCLASS_INIT_DATA initData = NULL;

#017 PDISK_DATA diskData;

#018 PDISK_DATA physicalDiskData;

#019 ULONG bytesPerSector;

#020 UCHAR sectorShift;

#021 ULONG srbFlags;

#022 ULONG dmByteSkew = 0;

#023 PULONG dmSkew;

#024 BOOLEAN dmActive = FALSE;

#025 ULONG numberListElements = 0;

#026

#027

#028 //

#029 // Get physical device geometry information for partition table reads.

#030 //

#031

通过磁盘描述信息来获取磁盘组成结构，比如每扇区多少个字节。

#032 physicalDeviceExtension = PhysicalDeviceObject->DeviceExtension;

#033 diskGeometry = physicalDeviceExtension->DiskGeometry;

#034 bytesPerSector = diskGeometry->BytesPerSector;

#035

#036 //

#037 // Make sure sector size is not zero.

#038 //

#039

确保每个扇区的字节数不为0，如果为0的时候，就让它缺省为512个字节。

#040 if (bytesPerSector == 0) {

#041

#042 //

#043 // Default sector size for disk is 512.

#044 //

#045

#046 bytesPerSector = diskGeometry->BytesPerSector = 512;

#047 }

#048

#049 sectorShift = physicalDeviceExtension->SectorShift;

#050

#051 //

#052 // Set pointer to disk data area that follows device extension.

#053 //

#054

设置指向磁盘结构数据指针。

#055 diskData = (PDISK_DATA)(physicalDeviceExtension + 1);

设置磁盘分区表格正在初始化。

#056 diskData->PartitionListState = Initializing;

#057

#058 //

#059 // Determine is DM Driver is loaded on an IDE drive that is

#060 // under control of Atapi - this could be either a crashdump or

#061 // an Atapi device is sharing the controller with an IDE disk.

#062 //

#063

调用函数HalExamineMBR来读取指定类型的MBR扇区数据。其实这个函数调用IO管理器后，生成一个IRP调用ATAPI驱动程序去读取磁盘0扇区数据。

#064 HalExamineMBR(PhysicalDeviceObject,

#065 physicalDeviceExtension->DiskGeometry->BytesPerSector,

#066 (ULONG)0x54,

#067 (PVOID)&dmSkew);

#068

判断是否有DM驱动程序，如果有就需要调整相关的磁盘信息。

#069 if (dmSkew) {

#070

#071 //

#072 // Update the device extension,so that the call to IoReadPartitionTable

#073 // will get the correct information. Any I/O to this disk will have

#074 // to be skewed by *dmSkew sectors aka DMByteSkew.

#075 //

#076

#077 physicalDeviceExtension->DMSkew = *dmSkew;

#078 physicalDeviceExtension->DMActive = TRUE;

#079 physicalDeviceExtension->DMByteSkew = physicalDeviceExtension->DMSkew * bytesPerSector;

#080

#081 //

#082 // Save away the infomation that we need,since this deviceExtension will soon be

#083 // blown away.

#084 //

#085

#086 dmActive = TRUE;

#087 dmByteSkew = physicalDeviceExtension->DMByteSkew;

#088

#089 }

#090

#091 //

#092 // Create objects for all the partitions on the device.

#093 //

#094

为这个磁盘设备的所有分区创建分区对象。

#095 status = IoReadPartitionTable(PhysicalDeviceObject,

#096 physicalDeviceExtension->DiskGeometry->BytesPerSector,

#097 TRUE,

#098 (PVOID)&partitionList);

#099

#100 //

#101 // If the I/O read partition table failed and this is a removable device,

#102 // then fix up the partition list to make it look like there is one

#103 // zero length partition.

#104 //

#105 DPRINT("IoReadPartitionTable() status: 0x%08X/n",status);

#106 if ((!NT_SUCCESS(status) || partitionList->PartitionCount == 0) &&

#107 PhysicalDeviceObject->Characteristics & FILE_REMOVABLE_MEDIA) {

#108

如果读分区表出错，就设置磁盘没有准备好。

#109 if (!NT_SUCCESS(status)) {

#110

#111 //

#112 // Remember this disk is not ready.

#113 //

#114

#115 diskData->DriveNotReady = TRUE;

#116

#117 } else {

#118

#119 //

#120 // Free the partition list allocated by IoReadPartitionTable.

#121 //

#122

#123 ExFreePool(partitionList);

#124 }

#125

#126 //

#127 // Allocate and zero a partition list.

#128 //

#129

分配分区列表。

#130 partitionList = ExAllocatePool(NonPagedPool,sizeof(*partitionList ));

#131

#132

#133 if (partitionList != NULL) {

#134

#135 RtlZeroMemory( partitionList,sizeof( *partitionList ));

#136

#137 //

#138 // Set the partition count to one and the status to success

#139 // so one device object will be created. Set the partition type

#140 // to a bogus value.

#141 //

#142

#143 partitionList->PartitionCount = 1;

#144

#145 status = STATUS_SUCCESS;

#146 }

#147 }

#148

#149 if (NT_SUCCESS(status)) {

#150

#151 //

#152 // Record disk signature.

#153 //

#154

保存磁盘的标志。

#155 diskData->Signature = partitionList->Signature;

#156

#157 //

#158 // If disk signature is zero,then calculate the MBR checksum.

#159 //

#160

如果磁盘的标志为0，那么就计算MBR的校验码是否正确。

#161 if (!diskData->Signature) {

#162

#163 if (!CalculateMbrCheckSum(physicalDeviceExtension,

#164 &diskData->MbrCheckSum)) {

#165

#166 DebugPrint((1,

#167 "SCSIDISK: Can't calculate MBR checksum for disk %x/n",

#168 physicalDeviceExtension->DeviceNumber));

#169 } else {

#170

#171 DebugPrint((2,

#172 "SCSIDISK: MBR checksum for disk %x is %x/n",

#173 physicalDeviceExtension->DeviceNumber,

#174 diskData->MbrCheckSum));

#175 }

#176 }

#177

#178 //

#179 // Check the registry and determine if the BIOS knew about this drive. If

#180 // it did then update the geometry with the BIOS information.

#181 //

#182

查询注册表，这个磁盘是否在BIOS里可以读取的。

#183 UpdateGeometry(physicalDeviceExtension);

#184

#185 srbFlags = physicalDeviceExtension->SrbFlags;

#186

创建磁盘的操作函数。

#187 initData = ExAllocatePool(NonPagedPool,sizeof(CLASS_INIT_DATA));

#188 if (!initData)

#189 {

#190 DebugPrint((1,

#191 "Disk.CreatePartionDeviceObjects - Allocation of initData failed/n"));

#192

#193 status = STATUS_INSUFFICIENT_RESOURCES;

#194 goto CreatePartitionDeviceObjectsExit;

#195 }

#196

#197 RtlZeroMemory(initData,sizeof(CLASS_INIT_DATA));

#198

#199 initData->InitializationDataSize = sizeof(CLASS_INIT_DATA);

#200 initData->DeviceExtensionSize = DEVICE_EXTENSION_SIZE;

#201 initData->DeviceType = FILE_DEVICE_DISK;

#202 initData->DeviceCharacteristics = PhysicalDeviceObject->Characteristics;

#203 initData->ClassError = physicalDeviceExtension->ClassError;

#204 initData->ClassReadWriteVerification = physicalDeviceExtension->ClassReadWriteVerification;

#205 initData->ClassFindDevices = physicalDeviceExtension->ClassFindDevices;

#206 initData->ClassDeviceControl = physicalDeviceExtension->ClassDeviceControl;

#207 initData->ClassShutdownFlush = physicalDeviceExtension->ClassShutdownFlush;

#208 initData->ClassCreateClose = physicalDeviceExtension->ClassCreateClose;

#209 initData->ClassStartIo = physicalDeviceExtension->ClassStartIo;

#210

#211 //

#212 // Create device objects for the device partitions (if any).

#213 // PartitionCount includes physical device partition 0,

#214 // so only one partition means no objects to create.

#215 //

#216

#217 DebugPrint((2,

#218 "CreateDiskDeviceObjects: Number of partitions is %d/n",

#219 partitionList->PartitionCount));

#220

为所有磁盘分区创建分区对象。

#221 for (partitionNumber = 0; partitionNumber <

#222 partitionList->PartitionCount; partitionNumber++) {

#223

#224 //

#225 // Create partition object and set up partition parameters.

#226 //

#227

#228 sprintf(ntNameBuffer,

#229 "//Device//Harddisk%lu//Partition%lu",

#230 physicalDeviceExtension->DeviceNumber,

#231 partitionNumber + 1);

#232

#233 DebugPrint((2,

#234 "CreateDiskDeviceObjects: Create device object %s/n",

#235 ntNameBuffer));

#236

#237 status = ScsiClassCreateDeviceObject(PhysicalDeviceObject->DriverObject,

#238 ntNameBuffer,

#239 PhysicalDeviceObject,

#240 &deviceObject,

#241 initData);

#242

#243 if (!NT_SUCCESS(status)) {

#244

#245 DebugPrint((1,"CreateDiskDeviceObjects: Can't create device object for %s/n",ntNameBuffer));

#246

#247 break;

#248 }

#249

#250 //

#251 // Set up device object fields.

#252 //

#253

设置设备是直接通过IO访问。

#254 deviceObject->Flags |= DO_DIRECT_IO;

#255

#256 //

#257 // Check if this is during initialization. If not indicate that

#258 // system initialization already took place and this disk is ready

#259 // to be accessed.

#260 //

#261

#262 if (!RegistryPath) {

#263 deviceObject->Flags &= ~DO_DEVICE_INITIALIZING;

#264 }

#265

设置设备栈。

#266 deviceObject->StackSize = (CCHAR)physicalDeviceExtension->PortDeviceObject->StackSize + 1;

#267

#268 //

#269 // Set up device extension fields.

#270 //

#271

设置设备扩展结构。

#272 deviceExtension = deviceObject->DeviceExtension;

#273

#274 if (dmActive) {

#275

#276 //

#277 // Restore any saved DM values.

#278 //

#279

#280 deviceExtension->DMByteSkew = dmByteSkew;

#281 deviceExtension->DMSkew = *dmSkew;

#282 deviceExtension->DMActive = TRUE;

#283

#284 }

#285

#286 //

#287 // Link new device extension to previous disk data

#288 // to support dynamic partitioning.

#289 //

#290

设置设备连接下一个分区。

#291 diskData->NextPartition = deviceExtension;

#292

#293 //

#294 // Get pointer to new disk data.

#295 //

#296

#297 diskData = (PDISK_DATA)(deviceExtension + 1);

#298

#299 //

#300 // Set next partition pointer to NULL in case this is the

#301 // last partition.

#302 //

#303

#304 diskData->NextPartition = NULL;

#305

#306 //

#307 // Allocate spinlock for zoning for split-request completion.

#308 //

#309

#310 KeInitializeSpinLock(&deviceExtension->SplitRequestSpinLock);

#311

#312 //

#313 // Copy port device object pointer to device extension.

#314 //

#315

设置设备指向端口驱动程序。

#316 deviceExtension->PortDeviceObject = physicalDeviceExtension->PortDeviceObject;

#317

#318 //

#319 // Set the alignment requirements for the device based on the

#320 // host adapter requirements

#321 //

#322

#323 if (physicalDeviceExtension->PortDeviceObject->AlignmentRequirement > deviceObject->AlignmentRequirement) {

#324 deviceObject->AlignmentRequirement = physicalDeviceExtension->PortDeviceObject->AlignmentRequirement;

#325 }

#326

#327

#328 if (srbFlags & SRB_FLAGS_QUEUE_ACTION_ENABLE) {

#329 numberListElements = 30;

#330 } else {

#331 numberListElements = 8;

#332 }

#333

#334 //

#335 // Build the lookaside list for srb's for this partition based on

#336 // whether the adapter and disk can do tagged queueing.

#337 //

#338

设置设备扩展的后备缓冲列表。

#339 ScsiClassInitializeSrbLookasideList(deviceExtension,

#340 numberListElements);

#341

#342 deviceExtension->SrbFlags = srbFlags;

#343

#344 //

#345 // Set the sense-data pointer in the device extension.

#346 //

#347

#348 deviceExtension->SenseData = physicalDeviceExtension->SenseData;

#349 deviceExtension->PortCapabilities = physicalDeviceExtension->PortCapabilities;

#350 deviceExtension->DiskGeometry = diskGeometry;

#351 diskData->PartitionOrdinal = diskData->PartitionNumber = partitionNumber + 1;

#352 diskData->PartitionType = partitionList->PartitionEntry[partitionNumber].PartitionType;

#353 diskData->BootIndicator = partitionList->PartitionEntry[partitionNumber].BootIndicator;

#354

#355 DebugPrint((2,"CreateDiskDeviceObjects: Partition type is %x/n",

#356 diskData->PartitionType));

#357

#358 deviceExtension->StartingOffset = partitionList->PartitionEntry[partitionNumber].StartingOffset;

#359 deviceExtension->PartitionLength = partitionList->PartitionEntry[partitionNumber].PartitionLength;

#360 diskData->HiddenSectors = partitionList->PartitionEntry[partitionNumber].HiddenSectors;

#361 deviceExtension->PortNumber = physicalDeviceExtension->PortNumber;

#362 deviceExtension->PathId = physicalDeviceExtension->PathId;

#363 deviceExtension->TargetId = physicalDeviceExtension->TargetId;

#364 deviceExtension->Lun = physicalDeviceExtension->Lun;

#365

#366 //

#367 // Check for removable media support.

#368 //

#369

检查可移动磁盘的支持。

#370 if (PhysicalDeviceObject->Characteristics & FILE_REMOVABLE_MEDIA) {

#371 deviceObject->Characteristics |= FILE_REMOVABLE_MEDIA;

#372 }

#373

#374 //

#375 // Set timeout value in seconds.

#376 //

#377

设置设备超时时间。

#378 deviceExtension->TimeOutValue = physicalDeviceExtension->TimeOutValue;

#379 deviceExtension->DiskGeometry->BytesPerSector = bytesPerSector;

#380 deviceExtension->SectorShift = sectorShift;

#381 deviceExtension->DeviceObject = deviceObject;

#382 deviceExtension->DeviceFlags |= physicalDeviceExtension->DeviceFlags;

#383

#384 } // end for (partitionNumber) ...

#385

#386 //

#387 // Free the buffer allocated by reading the

#388 // partition table.

#389 //

#390

#391 ExFreePool(partitionList);

#392

#393 } else {

#394

#395 CreatePartitionDeviceObjectsExit:

#396

#397 if (partitionList) {

#398 ExFreePool(partitionList);

#399 }

#400 if (initData) {

#401 ExFreePool(initData);

#402 }

#403

#404 return status;

#405

#406 } // end if...else

#407

#408

#409 physicalDiskData = (PDISK_DATA)(physicalDeviceExtension + 1);

#410 physicalDiskData->PartitionListState = Initialized;

#411

#412 return(STATUS_SUCCESS);

#413

#414

#415 } // end CreatePartitionDeviceObjects()

#416