reactos操作系统实现(90)

下面来分析函数IoAllocateDriverObjectExtension的实现，这个函数主要实现创建驱动程序扩展内存。

#001  NTSTATUS

#002  NTAPI

#003  IoAllocateDriverObjectExtension(IN PDRIVER_OBJECT DriverObject,

#004                                  IN PVOID ClientIdentificationAddress,

#005                                  IN ULONG DriverObjectExtensionSize,

#006                                  OUT PVOID *DriverObjectExtension)

#007  {

#008      KIRQL OldIrql;

#009      PIO_CLIENT_EXTENSION DriverExtensions, NewDriverExtension;

#010      BOOLEAN Inserted = FALSE;

#011 

 

首先假定分配扩展驱动程序对象失败。

#012      /* Assume failure */

#013      *DriverObjectExtension = NULL;

#014 

 

为扩展驱动程序对象分配新的内存。

#015      /* Allocate the extension */

#016      NewDriverExtension = ExAllocatePoolWithTag(NonPagedPool,

#017                                                 sizeof(IO_CLIENT_EXTENSION) +

#018                                                 DriverObjectExtensionSize,

#019                                                 TAG_DRIVER_EXTENSION);

 

如果分配内存失败，就直接返回。

#020      if (!NewDriverExtension) return STATUS_INSUFFICIENT_RESOURCES;

#021 

 

清空扩展区对象的内存空间。

#022      /* Clear the extension for teh caller */

#023      RtlZeroMemory(NewDriverExtension,

#024                    sizeof(IO_CLIENT_EXTENSION) + DriverObjectExtensionSize);

#025 

 

获取DPC级别锁，以便操作扩展对象。

#026      /* Acqure lock */

#027      OldIrql = KeRaiseIrqlToDpcLevel();

#028 

 

填写扩展对象。

#029      /* Fill out the extension */

#030      NewDriverExtension->ClientIdentificationAddress = ClientIdentificationAddress;

#031 

 

查找到当前扩展对象，并判断是否有冲突。

#032      /* Loop the current extensions */

#033      DriverExtensions = IoGetDrvObjExtension(DriverObject)->

#034                         ClientDriverExtension;

#035      while (DriverExtensions)

#036      {

#037          /* Check if the identifier matches */

#038          if (DriverExtensions->ClientIdentificationAddress ==

#039              ClientIdentificationAddress)

#040          {

 

这里发现有冲突的ID，就跳出循环，返回失败。

#041              /* We have a collision, break out */

#042              break;

#043          }

#044 

#045          /* Go to the next one */

#046          DriverExtensions = DriverExtensions->NextExtension;

#047      }

#048 

 

如果没有冲突的驱动程序扩展，就创建一个新的扩展。

#049      /* Check if we didn't collide */

#050      if (!DriverExtensions)

#051      {

#052          /* Link this one in */

#053          NewDriverExtension->NextExtension =

#054              IoGetDrvObjExtension(DriverObject)->ClientDriverExtension;

#055          IoGetDrvObjExtension(DriverObject)->ClientDriverExtension =

#056              NewDriverExtension;

 

标记已经插入到扩展。

#057          Inserted = TRUE;

#058      }

#059 

#060      /* Release the lock */

#061      KeLowerIrql(OldIrql);

#062 

 

如果没有插入到驱动程序扩展，说明有冲突，因此删除之前分配的内存。

#063      /* Check if insertion failed */

#064      if (!Inserted)

#065      {

#066          /* Free the entry and fail */

#067          ExFreePool(NewDriverExtension);

#068          return STATUS_OBJECT_NAME_COLLISION;

#069      }

#070 

 

返回驱动程序扩展的开始位置。

#071      /* Otherwise, return the pointer */

#072      *DriverObjectExtension = NewDriverExtension + 1;

#073      return STATUS_SUCCESS;

#074  }

#075 

有了驱动程序的扩展，就可以填写用户编写驱动程序保存数据的地方了。比如驱动程序锁、列表、还有驱动程序要访问的IO资源等等。

 

转载于:https://www.cnblogs.com/ajuanabc/archive/2009/09/19/2463709.html