From 902443579f3a165120b2ffa0462e2c8389cec63f Mon Sep 17 00:00:00 2001
From: Alexander Morozov <amorozov@etersoft.ru>
Date: Tue, 16 Dec 2008 12:19:08 +0300
Subject: [PATCH] ntoskrnl.exe: Add stubs for
 Io{Allocate,Get}DriverObjectExtension.

---
 dlls/ntoskrnl.exe/ntoskrnl.c        | 25 +++++++++++++++++++++++++
 dlls/ntoskrnl.exe/ntoskrnl.exe.spec |  4 ++--
 include/ddk/wdm.h                   |  2 ++
 3 files changed, 29 insertions(+), 2 deletions(-)

diff --git a/dlls/ntoskrnl.exe/ntoskrnl.c b/dlls/ntoskrnl.exe/ntoskrnl.c
index 652f5ffbb91..a669d3b21eb 100644
--- a/dlls/ntoskrnl.exe/ntoskrnl.c
+++ b/dlls/ntoskrnl.exe/ntoskrnl.c
@@ -259,6 +259,31 @@ NTSTATUS wine_ntoskrnl_main_loop( HANDLE stop_event )
 }
 
 
+/***********************************************************************
+ *           IoAllocateDriverObjectExtension  (NTOSKRNL.EXE.@)
+ */
+NTSTATUS WINAPI IoAllocateDriverObjectExtension( PDRIVER_OBJECT DriverObject,
+                                                 PVOID ClientIdentificationAddress,
+                                                 ULONG DriverObjectExtensionSize,
+                                                 PVOID *DriverObjectExtension )
+{
+    FIXME( "%p, %p, %u, %p\n", DriverObject, ClientIdentificationAddress,
+            DriverObjectExtensionSize, DriverObjectExtension );
+    return STATUS_NOT_IMPLEMENTED;
+}
+
+
+/***********************************************************************
+ *           IoGetDriverObjectExtension  (NTOSKRNL.EXE.@)
+ */
+PVOID WINAPI IoGetDriverObjectExtension( PDRIVER_OBJECT DriverObject,
+                                         PVOID ClientIdentificationAddress )
+{
+    FIXME( "%p, %p\n", DriverObject, ClientIdentificationAddress );
+    return NULL;
+}
+
+
 /***********************************************************************
  *           IoInitializeIrp  (NTOSKRNL.EXE.@)
  */
diff --git a/dlls/ntoskrnl.exe/ntoskrnl.exe.spec b/dlls/ntoskrnl.exe/ntoskrnl.exe.spec
index cf4c86e5368..946249d82fb 100644
--- a/dlls/ntoskrnl.exe/ntoskrnl.exe.spec
+++ b/dlls/ntoskrnl.exe/ntoskrnl.exe.spec
@@ -309,7 +309,7 @@
 @ stub IoAdapterObjectType
 @ stub IoAllocateAdapterChannel
 @ stub IoAllocateController
-@ stub IoAllocateDriverObjectExtension
+@ stdcall IoAllocateDriverObjectExtension(ptr ptr long ptr)
 @ stub IoAllocateErrorLogEntry
 @ stdcall IoAllocateIrp(long long)
 @ stdcall IoAllocateMdl(ptr long long long ptr)
@@ -387,7 +387,7 @@
 @ stub IoGetDeviceToVerify
 @ stub IoGetDiskDeviceObject
 @ stub IoGetDmaAdapter
-@ stub IoGetDriverObjectExtension
+@ stdcall IoGetDriverObjectExtension(ptr ptr)
 @ stub IoGetFileObjectGenericMapping
 @ stub IoGetInitialStack
 @ stub IoGetLowerDeviceObject
diff --git a/include/ddk/wdm.h b/include/ddk/wdm.h
index efc3f6aacb6..0f241b6a708 100644
--- a/include/ddk/wdm.h
+++ b/include/ddk/wdm.h
@@ -1026,6 +1026,7 @@ PVOID     WINAPI ExAllocatePoolWithQuotaTag(POOL_TYPE,SIZE_T,ULONG);
 void      WINAPI ExFreePool(PVOID);
 void      WINAPI ExFreePoolWithTag(PVOID,ULONG);
 
+NTSTATUS  WINAPI IoAllocateDriverObjectExtension(PDRIVER_OBJECT,PVOID,ULONG,PVOID*);
 PIRP      WINAPI IoAllocateIrp(CCHAR,BOOLEAN);
 NTSTATUS  WINAPI IoCreateDevice(DRIVER_OBJECT*,ULONG,UNICODE_STRING*,DEVICE_TYPE,ULONG,BOOLEAN,DEVICE_OBJECT**);
 NTSTATUS  WINAPI IoCreateDriver(UNICODE_STRING*,PDRIVER_INITIALIZE);
@@ -1036,6 +1037,7 @@ NTSTATUS  WINAPI IoDeleteSymbolicLink(UNICODE_STRING*);
 void      WINAPI IoFreeIrp(IRP*);
 PEPROCESS WINAPI IoGetCurrentProcess(void);
 NTSTATUS  WINAPI IoGetDeviceObjectPointer(UNICODE_STRING*,ACCESS_MASK,PFILE_OBJECT*,PDEVICE_OBJECT*);
+PVOID     WINAPI IoGetDriverObjectExtension(PDRIVER_OBJECT,PVOID);
 PDEVICE_OBJECT WINAPI IoGetRelatedDeviceObject(PFILE_OBJECT);
 void      WINAPI IoInitializeIrp(IRP*,USHORT,CCHAR);