Aric Stewart : ntoskrnl: Build a more intelligent and correct RegistryPath if possible.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 2ffaaee1cf286072eea76fa5392eeabf3b060f26
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=2ffaaee1cf286072eea76fa539...

Author: Aric Stewart <aric(a)codeweavers.com>
Date:   Fri Jul 22 13:45:21 2016 -0500

ntoskrnl: Build a more intelligent and correct RegistryPath if possible.

Signed-off-by: Aric Stewart <aric(a)codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard(a)winehq.org>

---

 dlls/ntoskrnl.exe/ntoskrnl.c | 33 +++++++++++++++++++++++++++++++--
 1 file changed, 31 insertions(+), 2 deletions(-)

diff --git a/dlls/ntoskrnl.exe/ntoskrnl.c b/dlls/ntoskrnl.exe/ntoskrnl.c
index 8003ba1..dc19dee 100644
--- a/dlls/ntoskrnl.exe/ntoskrnl.c
+++ b/dlls/ntoskrnl.exe/ntoskrnl.c
@@ -817,6 +817,34 @@ PIRP WINAPI IoBuildSynchronousFsdRequest(ULONG majorfunc, PDEVICE_OBJECT device,
     return irp;
 }
 
+static void build_driver_keypath( const WCHAR *name, UNICODE_STRING *keypath )
+{
+    static const WCHAR driverW[] = {'\\','D','r','i','v','e','r','\\',0};
+    static const WCHAR servicesW[] = {'\\','R','e','g','i','s','t','r','y',
+                                      '\\','M','a','c','h','i','n','e',
+                                      '\\','S','y','s','t','e','m',
+                                      '\\','C','u','r','r','e','n','t','C','o','n','t','r','o','l','S','e','t',
+                                      '\\','S','e','r','v','i','c','e','s',
+                                      '\\',0};
+    WCHAR *str;
+
+    /* Check what prefix is present */
+    if (strncmpW( name, servicesW, strlenW(servicesW) ) == 0)
+    {
+        FIXME( "Driver name %s is malformed as the keypath\n", debugstr_w(name) );
+        RtlCreateUnicodeString( keypath, name );
+        return;
+    }
+    if (strncmpW( name, driverW, strlenW(driverW) ) == 0)
+        name += strlenW(driverW);
+    else
+        FIXME( "Driver name %s does not properly begin with \\Driver\\", debugstr_w(name) );
+
+    str = HeapAlloc( GetProcessHeap(), 0, sizeof(servicesW) + strlenW(name)*sizeof(WCHAR));
+    lstrcpyW( str, servicesW );
+    lstrcatW( str, name );
+    RtlInitUnicodeString( keypath, str );
+}
 
 /***********************************************************************
  *           IoCreateDriver   (NTOSKRNL.EXE.@)
@@ -844,9 +872,9 @@ NTSTATUS WINAPI IoCreateDriver( UNICODE_STRING *name, PDRIVER_INITIALIZE init )
     driver->DriverInit      = init;
     driver->DriverExtension = extension;
     extension->DriverObject   = driver;
-    extension->ServiceKeyName = driver->DriverName;
+    build_driver_keypath( driver->DriverName.Buffer, &extension->ServiceKeyName );
 
-    status = driver->DriverInit( driver, name );
+    status = driver->DriverInit( driver, &extension->ServiceKeyName );
 
     if (status)
     {
@@ -865,6 +893,7 @@ void WINAPI IoDeleteDriver( DRIVER_OBJECT *driver )
     TRACE("(%p)\n", driver);
 
     RtlFreeUnicodeString( &driver->DriverName );
+    RtlFreeUnicodeString( &driver->DriverExtension->ServiceKeyName );
     RtlFreeHeap( GetProcessHeap(), 0, driver );
 }