ntoskrnl.exe: Avoid loading a driver multiple times in ZwLoadDriver().

Wine-Bug: https://bugs.winehq.org/show_bug.cgi?id=45805Signed-off-by: Zebediah Figura <z.figura12@gmail.com> Signed-off-by: Alexandre Julliard <julliard@winehq.org>

ntoskrnl.exe: Avoid loading a driver multiple times in ZwLoadDriver().
e3ad1cff · Zebediah Figura · Alexandre Julliard · d209055e · e3ad1cff · e3ad1cff
Commit e3ad1cff authored Dec 21, 2018 by Zebediah Figura Committed by Alexandre Julliard Dec 24, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 0 deletions

ntoskrnl.c dlls/ntoskrnl.exe/ntoskrnl.c +7 -0

driver.c dlls/ntoskrnl.exe/tests/driver.c +3 -0

No files found.
--- a/dlls/ntoskrnl.exe/ntoskrnl.c
+++ b/dlls/ntoskrnl.exe/ntoskrnl.c
@@ -3519,6 +3519,13 @@ NTSTATUS WINAPI ZwLoadDriver( const UNICODE_STRING *service_name )
        return STATUS_NO_MEMORY;
    }

+    if (wine_rb_get( &wine_drivers, &drv_name ))
+    {
+        TRACE( "driver %s already loaded\n", debugstr_us(&drv_name) );
+        RtlFreeUnicodeString( &drv_name );
+        return STATUS_IMAGE_ALREADY_LOADED;
+    }
+
    set_service_status( service_handle, SERVICE_START_PENDING, 0 );

    status = IoCreateDriver( &drv_name, init_driver );

--- a/dlls/ntoskrnl.exe/tests/driver.c
+++ b/dlls/ntoskrnl.exe/tests/driver.c
@@ -206,6 +206,9 @@ static void test_load_driver(void)
    ret = ZwLoadDriver(&name);
    ok(!ret, "got %#x\n", ret);

+    ret = ZwLoadDriver(&name);
+    ok(ret == STATUS_IMAGE_ALREADY_LOADED, "got %#x\n", ret);
+
    ret = ZwUnloadDriver(&name);
    ok(!ret, "got %#x\n", ret);
 }