dlls/ntoskrnl.exe/tests/ntoskrnl.c

   1 /*
   2  * ntoskrnl.exe testing framework
   3  *
   4  * Copyright 2015 Sebastian Lackner
   5  * Copyright 2015 Michael Müller
   6  * Copyright 2015 Christian Costa
   7  * Copyright 2020-2021 Zebediah Figura for CodeWeavers
   8  *
   9  * This library is free software; you can redistribute it and/or
  10  * modify it under the terms of the GNU Lesser General Public
  11  * License as published by the Free Software Foundation; either
  12  * version 2.1 of the License, or (at your option) any later version.
  13  *
  14  * This library is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  17  * Lesser General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Lesser General Public
  20  * License along with this library; if not, write to the Free Software
  21  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  22  */
  23
  24 #include <stdio.h>
  25 #include "ntstatus.h"
  26 #define WIN32_NO_STATUS
  27 #include "windows.h"
  28 #include "winsvc.h"
  29 #include "winioctl.h"
  30 #include "winternl.h"
  31 #include "winsock2.h"
  32 #include "wincrypt.h"
  33 #include "ntsecapi.h"
  34 #include "mscat.h"
  35 #include "mssip.h"
  36 #include "setupapi.h"
  37 #include "cfgmgr32.h"
  38 #include "newdev.h"
  39 #include "regstr.h"
  40 #include "dbt.h"
  41 #include "initguid.h"
  42 #include "devguid.h"
  43 #include "ddk/hidclass.h"
  44 #include "ddk/hidsdi.h"
  45 #include "ddk/hidpi.h"
  46 #include "wine/test.h"
  47 #include "wine/mssign.h"
  48
  49 #include "driver.h"
  50
  51 static const GUID GUID_NULL;
  52
  53 static HANDLE device;
  54
  55 static struct test_data *test_data;
  56
  57 static BOOL (WINAPI *pRtlDosPathNameToNtPathName_U)(const WCHAR *, UNICODE_STRING *, WCHAR **, CURDIR *);
  58 static BOOL (WINAPI *pRtlFreeUnicodeString)(UNICODE_STRING *);
  59 static BOOL (WINAPI *pCancelIoEx)(HANDLE, OVERLAPPED *);
  60 static BOOL (WINAPI *pIsWow64Process)(HANDLE, BOOL *);
  61 static BOOL (WINAPI *pSetFileCompletionNotificationModes)(HANDLE, UCHAR);
  62 static HRESULT (WINAPI *pSignerSign)(SIGNER_SUBJECT_INFO *subject, SIGNER_CERT *cert,
  63         SIGNER_SIGNATURE_INFO *signature, SIGNER_PROVIDER_INFO *provider,
  64         const WCHAR *timestamp, CRYPT_ATTRIBUTES *attr, void *sip_data);
  65
  66 static void load_resource(const WCHAR *name, WCHAR *filename)
  67 {
  68     static WCHAR path[MAX_PATH];
  69     DWORD written;
  70     HANDLE file;
  71     HRSRC res;
  72     void *ptr;
  73
  74     GetTempPathW(ARRAY_SIZE(path), path);
  75     GetTempFileNameW(path, name, 0, filename);
  76
  77     file = CreateFileW(filename, GENERIC_READ|GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0);
  78     ok(file != INVALID_HANDLE_VALUE, "failed to create %s, error %lu\n", debugstr_w(filename), GetLastError());
  79
  80     res = FindResourceW(NULL, name, L"TESTDLL");
  81     ok( res != 0, "couldn't find resource\n" );
  82     ptr = LockResource( LoadResource( GetModuleHandleA(NULL), res ));
  83     WriteFile( file, ptr, SizeofResource( GetModuleHandleA(NULL), res ), &written, NULL );
  84     ok( written == SizeofResource( GetModuleHandleA(NULL), res ), "couldn't write resource\n" );
  85     CloseHandle( file );
  86 }
  87
  88 struct testsign_context
  89 {
  90     HCRYPTPROV provider;
  91     const CERT_CONTEXT *cert, *root_cert, *publisher_cert;
  92     HCERTSTORE root_store, publisher_store;
  93 };
  94
  95 static BOOL testsign_create_cert(struct testsign_context *ctx)
  96 {
  97     BYTE encoded_name[100], encoded_key_id[200], public_key_info_buffer[1000];
  98     WCHAR container_name[26];
  99     BYTE hash_buffer[16], cert_buffer[1000], provider_nameA[100], serial[16];
 100     CERT_PUBLIC_KEY_INFO *public_key_info = (CERT_PUBLIC_KEY_INFO *)public_key_info_buffer;
 101     CRYPT_KEY_PROV_INFO provider_info = {0};
 102     CRYPT_ALGORITHM_IDENTIFIER algid = {0};
 103     CERT_AUTHORITY_KEY_ID_INFO key_info;
 104     CERT_INFO cert_info = {0};
 105     WCHAR provider_nameW[100];
 106     CERT_EXTENSION extension;
 107     HCRYPTKEY key;
 108     DWORD size;
 109     BOOL ret;
 110
 111     memset(ctx, 0, sizeof(*ctx));
 112
 113     srand(time(NULL));
 114     swprintf(container_name, ARRAY_SIZE(container_name), L"wine_testsign%u", rand());
 115
 116     ret = CryptAcquireContextW(&ctx->provider, container_name, NULL, PROV_RSA_FULL, CRYPT_NEWKEYSET);
 117     ok(ret, "Failed to create container, error %#lx\n", GetLastError());
 118
 119     ret = CryptGenKey(ctx->provider, AT_SIGNATURE, CRYPT_EXPORTABLE, &key);
 120     ok(ret, "Failed to create key, error %#lx\n", GetLastError());
 121     ret = CryptDestroyKey(key);
 122     ok(ret, "Failed to destroy key, error %#lx\n", GetLastError());
 123     ret = CryptGetUserKey(ctx->provider, AT_SIGNATURE, &key);
 124     ok(ret, "Failed to get user key, error %#lx\n", GetLastError());
 125     ret = CryptDestroyKey(key);
 126     ok(ret, "Failed to destroy key, error %#lx\n", GetLastError());
 127
 128     size = sizeof(encoded_name);
 129     ret = CertStrToNameA(X509_ASN_ENCODING, "CN=winetest_cert", CERT_X500_NAME_STR, NULL, encoded_name, &size, NULL);
 130     ok(ret, "Failed to convert name, error %#lx\n", GetLastError());
 131     key_info.CertIssuer.cbData = size;
 132     key_info.CertIssuer.pbData = encoded_name;
 133
 134     size = sizeof(public_key_info_buffer);
 135     ret = CryptExportPublicKeyInfo(ctx->provider, AT_SIGNATURE, X509_ASN_ENCODING, public_key_info, &size);
 136     ok(ret, "Failed to export public key, error %#lx\n", GetLastError());
 137     cert_info.SubjectPublicKeyInfo = *public_key_info;
 138
 139     size = sizeof(hash_buffer);
 140     ret = CryptHashPublicKeyInfo(ctx->provider, CALG_MD5, 0, X509_ASN_ENCODING, public_key_info, hash_buffer, &size);
 141     ok(ret, "Failed to hash public key, error %#lx\n", GetLastError());
 142
 143     key_info.KeyId.cbData = size;
 144     key_info.KeyId.pbData = hash_buffer;
 145
 146     RtlGenRandom(serial, sizeof(serial));
 147     key_info.CertSerialNumber.cbData = sizeof(serial);
 148     key_info.CertSerialNumber.pbData = serial;
 149
 150     size = sizeof(encoded_key_id);
 151     ret = CryptEncodeObject(X509_ASN_ENCODING, X509_AUTHORITY_KEY_ID, &key_info, encoded_key_id, &size);
 152     ok(ret, "Failed to convert name, error %#lx\n", GetLastError());
 153
 154     extension.pszObjId = (char *)szOID_AUTHORITY_KEY_IDENTIFIER;
 155     extension.fCritical = TRUE;
 156     extension.Value.cbData = size;
 157     extension.Value.pbData = encoded_key_id;
 158
 159     cert_info.dwVersion = CERT_V3;
 160     cert_info.SerialNumber = key_info.CertSerialNumber;
 161     cert_info.SignatureAlgorithm.pszObjId = (char *)szOID_RSA_SHA1RSA;
 162     cert_info.Issuer = key_info.CertIssuer;
 163     GetSystemTimeAsFileTime(&cert_info.NotBefore);
 164     GetSystemTimeAsFileTime(&cert_info.NotAfter);
 165     cert_info.NotAfter.dwHighDateTime += 1;
 166     cert_info.Subject = key_info.CertIssuer;
 167     cert_info.cExtension = 1;
 168     cert_info.rgExtension = &extension;
 169     algid.pszObjId = (char *)szOID_RSA_SHA1RSA;
 170     size = sizeof(cert_buffer);
 171     ret = CryptSignAndEncodeCertificate(ctx->provider, AT_SIGNATURE, X509_ASN_ENCODING,
 172             X509_CERT_TO_BE_SIGNED, &cert_info, &algid, NULL, cert_buffer, &size);
 173     ok(ret, "Failed to create certificate, error %#lx\n", GetLastError());
 174
 175     ctx->cert = CertCreateCertificateContext(X509_ASN_ENCODING, cert_buffer, size);
 176     ok(!!ctx->cert, "Failed to create context, error %#lx\n", GetLastError());
 177
 178     size = sizeof(provider_nameA);
 179     ret = CryptGetProvParam(ctx->provider, PP_NAME, provider_nameA, &size, 0);
 180     ok(ret, "Failed to get prov param, error %#lx\n", GetLastError());
 181     MultiByteToWideChar(CP_ACP, 0, (char *)provider_nameA, -1, provider_nameW, ARRAY_SIZE(provider_nameW));
 182
 183     provider_info.pwszContainerName = (WCHAR *)container_name;
 184     provider_info.pwszProvName = provider_nameW;
 185     provider_info.dwProvType = PROV_RSA_FULL;
 186     provider_info.dwKeySpec = AT_SIGNATURE;
 187     ret = CertSetCertificateContextProperty(ctx->cert, CERT_KEY_PROV_INFO_PROP_ID, 0, &provider_info);
 188     ok(ret, "Failed to set provider info, error %#lx\n", GetLastError());
 189
 190     ctx->root_store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY_A, 0, 0, CERT_SYSTEM_STORE_LOCAL_MACHINE, "root");
 191     if (!ctx->root_store && GetLastError() == ERROR_ACCESS_DENIED)
 192     {
 193         skip("Failed to open root store.\n");
 194
 195         ret = CertFreeCertificateContext(ctx->cert);
 196         ok(ret, "Failed to free certificate, error %lu\n", GetLastError());
 197         ret = CryptReleaseContext(ctx->provider, 0);
 198         ok(ret, "failed to release context, error %lu\n", GetLastError());
 199
 200         return FALSE;
 201     }
 202     ok(!!ctx->root_store, "Failed to open store, error %lu\n", GetLastError());
 203     ret = CertAddCertificateContextToStore(ctx->root_store, ctx->cert, CERT_STORE_ADD_ALWAYS, &ctx->root_cert);
 204     if (!ret && GetLastError() == ERROR_ACCESS_DENIED)
 205     {
 206         skip("Failed to add self-signed certificate to store.\n");
 207
 208         ret = CertFreeCertificateContext(ctx->cert);
 209         ok(ret, "Failed to free certificate, error %lu\n", GetLastError());
 210         ret = CertCloseStore(ctx->root_store, CERT_CLOSE_STORE_CHECK_FLAG);
 211         ok(ret, "Failed to close store, error %lu\n", GetLastError());
 212         ret = CryptReleaseContext(ctx->provider, 0);
 213         ok(ret, "failed to release context, error %lu\n", GetLastError());
 214
 215         return FALSE;
 216     }
 217     ok(ret, "Failed to add certificate, error %lu\n", GetLastError());
 218
 219     ctx->publisher_store = CertOpenStore(CERT_STORE_PROV_SYSTEM_REGISTRY_A, 0, 0,
 220             CERT_SYSTEM_STORE_LOCAL_MACHINE, "trustedpublisher");
 221     ok(!!ctx->publisher_store, "Failed to open store, error %lu\n", GetLastError());
 222     ret = CertAddCertificateContextToStore(ctx->publisher_store, ctx->cert,
 223             CERT_STORE_ADD_ALWAYS, &ctx->publisher_cert);
 224     ok(ret, "Failed to add certificate, error %lu\n", GetLastError());
 225
 226     return TRUE;
 227 }
 228
 229 static void testsign_cleanup(struct testsign_context *ctx)
 230 {
 231     BOOL ret;
 232
 233     ret = CertFreeCertificateContext(ctx->cert);
 234     ok(ret, "Failed to free certificate, error %lu\n", GetLastError());
 235
 236     ret = CertFreeCertificateContext(ctx->root_cert);
 237     ok(ret, "Failed to free certificate context, error %lu\n", GetLastError());
 238     ret = CertCloseStore(ctx->root_store, CERT_CLOSE_STORE_CHECK_FLAG);
 239     ok(ret, "Failed to close store, error %lu\n", GetLastError());
 240
 241     ret = CertFreeCertificateContext(ctx->publisher_cert);
 242     ok(ret, "Failed to free certificate context, error %lu\n", GetLastError());
 243     ret = CertCloseStore(ctx->publisher_store, CERT_CLOSE_STORE_CHECK_FLAG);
 244     ok(ret, "Failed to close store, error %lu\n", GetLastError());
 245
 246     ret = CryptReleaseContext(ctx->provider, 0);
 247     ok(ret, "failed to release context, error %lu\n", GetLastError());
 248 }
 249
 250 static void testsign_sign(struct testsign_context *ctx, const WCHAR *filename)
 251 {
 252     SIGNER_ATTR_AUTHCODE authcode = {sizeof(authcode)};
 253     SIGNER_SIGNATURE_INFO signature = {sizeof(signature)};
 254     SIGNER_SUBJECT_INFO subject = {sizeof(subject)};
 255     SIGNER_CERT_STORE_INFO store = {sizeof(store)};
 256     SIGNER_CERT cert_info = {sizeof(cert_info)};
 257     SIGNER_FILE_INFO file = {sizeof(file)};
 258     DWORD index = 0;
 259     HRESULT hr;
 260
 261     subject.dwSubjectChoice = 1;
 262     subject.pdwIndex = &index;
 263     subject.pSignerFileInfo = &file;
 264     file.pwszFileName = (WCHAR *)filename;
 265     cert_info.dwCertChoice = 2;
 266     cert_info.pCertStoreInfo = &store;
 267     store.pSigningCert = ctx->cert;
 268     store.dwCertPolicy = 0;
 269     signature.algidHash = CALG_SHA_256;
 270     signature.dwAttrChoice = SIGNER_AUTHCODE_ATTR;
 271     signature.pAttrAuthcode = &authcode;
 272     authcode.pwszName = L"";
 273     authcode.pwszInfo = L"";
 274     hr = pSignerSign(&subject, &cert_info, &signature, NULL, NULL, NULL, NULL);
 275     todo_wine ok(hr == S_OK || broken(hr == NTE_BAD_ALGID) /* < 7 */, "Failed to sign, hr %#lx\n", hr);
 276 }
 277
 278 static void unload_driver(SC_HANDLE service)
 279 {
 280     SERVICE_STATUS status;
 281
 282     ControlService(service, SERVICE_CONTROL_STOP, &status);
 283     while (status.dwCurrentState == SERVICE_STOP_PENDING)
 284     {
 285         BOOL ret;
 286         Sleep(100);
 287         ret = QueryServiceStatus(service, &status);
 288         ok(ret, "QueryServiceStatus failed: %lu\n", GetLastError());
 289     }
 290     ok(status.dwCurrentState == SERVICE_STOPPED,
 291        "expected SERVICE_STOPPED, got %ld\n", status.dwCurrentState);
 292
 293     DeleteService(service);
 294     CloseServiceHandle(service);
 295 }
 296
 297 static SC_HANDLE load_driver(struct testsign_context *ctx, WCHAR *filename,
 298         const WCHAR *resname, const WCHAR *driver_name)
 299 {
 300     SC_HANDLE manager, service;
 301
 302     manager = OpenSCManagerA(NULL, NULL, SC_MANAGER_ALL_ACCESS);
 303     if (!manager && GetLastError() == ERROR_ACCESS_DENIED)
 304     {
 305         skip("Failed to open SC manager, not enough permissions\n");
 306         return FALSE;
 307     }
 308     ok(!!manager, "OpenSCManager failed\n");
 309
 310     /* stop any old drivers running under this name */
 311     service = OpenServiceW(manager, driver_name, SERVICE_ALL_ACCESS);
 312     if (service) unload_driver(service);
 313
 314     load_resource(resname, filename);
 315     testsign_sign(ctx, filename);
 316     trace("Trying to load driver %s\n", debugstr_w(filename));
 317
 318     service = CreateServiceW(manager, driver_name, driver_name,
 319                              SERVICE_ALL_ACCESS, SERVICE_KERNEL_DRIVER,
 320                              SERVICE_DEMAND_START, SERVICE_ERROR_NORMAL,
 321                              filename, NULL, NULL, NULL, NULL, NULL);
 322     ok(!!service, "CreateService failed: %lu\n", GetLastError());
 323
 324     CloseServiceHandle(manager);
 325     return service;
 326 }
 327
 328 static BOOL start_driver(HANDLE service, BOOL vista_plus)
 329 {
 330     SERVICE_STATUS status;
 331     BOOL ret;
 332
 333     SetLastError(0xdeadbeef);
 334     ret = StartServiceA(service, 0, NULL);
 335     if (!ret && (GetLastError() == ERROR_DRIVER_BLOCKED || GetLastError() == ERROR_INVALID_IMAGE_HASH
 336             || (vista_plus && GetLastError() == ERROR_FILE_NOT_FOUND)))
 337     {
 338         if (vista_plus && GetLastError() == ERROR_FILE_NOT_FOUND)
 339         {
 340             skip("Windows Vista or newer is required to run this service.\n");
 341         }
 342         else
 343         {
 344             /* If Secure Boot is enabled or the machine is 64-bit, it will reject an unsigned driver. */
 345             skip("Failed to start service; probably your machine doesn't accept unsigned drivers.\n");
 346         }
 347         DeleteService(service);
 348         CloseServiceHandle(service);
 349         return FALSE;
 350     }
 351     ok(ret, "StartService failed: %lu\n", GetLastError());
 352
 353     /* wait for the service to start up properly */
 354     ret = QueryServiceStatus(service, &status);
 355     ok(ret, "QueryServiceStatus failed: %lu\n", GetLastError());
 356     while (status.dwCurrentState == SERVICE_START_PENDING)
 357     {
 358         Sleep(100);
 359         ret = QueryServiceStatus(service, &status);
 360         ok(ret, "QueryServiceStatus failed: %lu\n", GetLastError());
 361     }
 362     ok(status.dwCurrentState == SERVICE_RUNNING,
 363        "expected SERVICE_RUNNING, got %ld\n", status.dwCurrentState);
 364     ok(status.dwServiceType == SERVICE_KERNEL_DRIVER,
 365        "expected SERVICE_KERNEL_DRIVER, got %#lx\n", status.dwServiceType);
 366
 367     return TRUE;
 368 }
 369
 370 static HANDLE okfile;
 371
 372 static void cat_okfile(void)
 373 {
 374     char buffer[512];
 375     DWORD size;
 376
 377     SetFilePointer(okfile, 0, NULL, FILE_BEGIN);
 378
 379     do
 380     {
 381         ReadFile(okfile, buffer, sizeof(buffer), &size, NULL);
 382         printf("%.*s", (int)size, buffer);
 383     } while (size == sizeof(buffer));
 384
 385     SetFilePointer(okfile, 0, NULL, FILE_BEGIN);
 386     SetEndOfFile(okfile);
 387
 388     InterlockedAdd(&winetest_successes, InterlockedExchange(&test_data->successes, 0));
 389     winetest_add_failures(InterlockedExchange(&test_data->failures, 0));
 390     InterlockedAdd(&winetest_todo_successes, InterlockedExchange(&test_data->todo_successes, 0));
 391     winetest_add_failures(InterlockedExchange(&test_data->todo_failures, 0));
 392     InterlockedAdd(&winetest_skipped, InterlockedExchange(&test_data->skipped, 0));
 393 }
 394
 395 static ULONG64 modified_value;
 396
 397 static void main_test(void)
 398 {
 399     struct main_test_input test_input;
 400     DWORD size;
 401     BOOL res;
 402
 403     test_input.process_id = GetCurrentProcessId();
 404     test_input.teststr_offset = (SIZE_T)((BYTE *)&teststr - (BYTE *)NtCurrentTeb()->Peb->ImageBaseAddress);
 405     test_input.modified_value = &modified_value;
 406     modified_value = 0;
 407
 408     res = DeviceIoControl(device, IOCTL_WINETEST_MAIN_TEST, &test_input, sizeof(test_input), NULL, 0, &size, NULL);
 409     ok(res, "DeviceIoControl failed: %lu\n", GetLastError());
 410     ok(!size, "got size %lu\n", size);
 411 }
 412
 413 static void test_basic_ioctl(void)
 414 {
 415     char inbuf[64], buf[32];
 416     DWORD written;
 417     BOOL res;
 418
 419     res = DeviceIoControl(device, IOCTL_WINETEST_BASIC_IOCTL, NULL, 0, buf,
 420                           sizeof(buf), &written, NULL);
 421     ok(res, "DeviceIoControl failed: %lu\n", GetLastError());
 422     ok(written == sizeof(teststr), "got size %ld\n", written);
 423     ok(!strcmp(buf, teststr), "got '%s'\n", buf);
 424
 425     memset(buf, 0, sizeof(buf));
 426     res = DeviceIoControl(device, IOCTL_WINETEST_BASIC_IOCTL, inbuf,
 427             sizeof(inbuf), buf, 10, &written, NULL);
 428     ok(res, "DeviceIoControl failed: %lu\n", GetLastError());
 429     ok(written == 10, "got size %ld\n", written);
 430     ok(!strcmp(buf, "Wine is no"), "got '%s'\n", buf);
 431 }
 432
 433 static void test_overlapped(void)
 434 {
 435     OVERLAPPED overlapped, overlapped2, *o;
 436     DWORD cancel_cnt, size;
 437     HANDLE file, port;
 438     ULONG_PTR key;
 439     BOOL res;
 440
 441     overlapped.hEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
 442     overlapped2.hEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
 443
 444     file = CreateFileA("\\\\.\\WineTestDriver", FILE_READ_ATTRIBUTES | FILE_WRITE_ATTRIBUTES,
 445                        0, NULL, OPEN_EXISTING, FILE_FLAG_OVERLAPPED, NULL);
 446     ok(file != INVALID_HANDLE_VALUE, "failed to open device: %lu\n", GetLastError());
 447
 448     /* test cancelling all device requests */
 449     res = DeviceIoControl(file, IOCTL_WINETEST_RESET_CANCEL, NULL, 0, NULL, 0, NULL, &overlapped);
 450     ok(res, "DeviceIoControl failed: %lu\n", GetLastError());
 451
 452     res = DeviceIoControl(file, IOCTL_WINETEST_TEST_CANCEL, NULL, 0, NULL, 0, NULL, &overlapped);
 453     ok(!res && GetLastError() == ERROR_IO_PENDING, "DeviceIoControl failed: %lu\n", GetLastError());
 454
 455     res = DeviceIoControl(file, IOCTL_WINETEST_TEST_CANCEL, NULL, 0, NULL, 0, NULL, &overlapped2);
 456     ok(!res && GetLastError() == ERROR_IO_PENDING, "DeviceIoControl failed: %lu\n", GetLastError());
 457
 458     cancel_cnt = 0xdeadbeef;
 459     res = DeviceIoControl(file, IOCTL_WINETEST_GET_CANCEL_COUNT, NULL, 0, &cancel_cnt, sizeof(cancel_cnt), NULL, &overlapped);
 460     ok(res, "DeviceIoControl failed: %lu\n", GetLastError());
 461     ok(cancel_cnt == 0, "cancel_cnt = %lu\n", cancel_cnt);
 462
 463     CancelIo(file);
 464
 465     cancel_cnt = 0xdeadbeef;
 466     res = DeviceIoControl(file, IOCTL_WINETEST_GET_CANCEL_COUNT, NULL, 0, &cancel_cnt, sizeof(cancel_cnt), NULL, &overlapped);
 467     ok(res, "DeviceIoControl failed: %lu\n", GetLastError());
 468     ok(cancel_cnt == 2, "cancel_cnt = %lu\n", cancel_cnt);
 469
 470     /* test cancelling selected overlapped event */
 471     if (pCancelIoEx)
 472     {
 473         res = DeviceIoControl(file, IOCTL_WINETEST_RESET_CANCEL, NULL, 0, NULL, 0, NULL, &overlapped);
 474         ok(res, "DeviceIoControl failed: %lu\n", GetLastError());
 475
 476         res = DeviceIoControl(file, IOCTL_WINETEST_TEST_CANCEL, NULL, 0, NULL, 0, NULL, &overlapped);
 477         ok(!res && GetLastError() == ERROR_IO_PENDING, "DeviceIoControl failed: %lu\n", GetLastError());
 478
 479         res = DeviceIoControl(file, IOCTL_WINETEST_TEST_CANCEL, NULL, 0, NULL, 0, NULL, &overlapped2);
 480         ok(!res && GetLastError() == ERROR_IO_PENDING, "DeviceIoControl failed: %lu\n", GetLastError());
 481
 482         pCancelIoEx(file, &overlapped);
 483
 484         cancel_cnt = 0xdeadbeef;
 485         res = DeviceIoControl(file, IOCTL_WINETEST_GET_CANCEL_COUNT, NULL, 0, &cancel_cnt, sizeof(cancel_cnt), NULL, &overlapped);
 486         ok(res, "DeviceIoControl failed: %lu\n", GetLastError());
 487         ok(cancel_cnt == 1, "cancel_cnt = %lu\n", cancel_cnt);
 488
 489         pCancelIoEx(file, &overlapped2);
 490
 491         cancel_cnt = 0xdeadbeef;
 492         res = DeviceIoControl(file, IOCTL_WINETEST_GET_CANCEL_COUNT, NULL, 0, &cancel_cnt, sizeof(cancel_cnt), NULL, &overlapped);
 493         ok(res, "DeviceIoControl failed: %lu\n", GetLastError());
 494         ok(cancel_cnt == 2, "cancel_cnt = %lu\n", cancel_cnt);
 495     }
 496
 497     port = CreateIoCompletionPort(file, NULL, 0xdeadbeef, 0);
 498     ok(port != NULL, "CreateIoCompletionPort failed, error %lu\n", GetLastError());
 499     res = GetQueuedCompletionStatus(port, &size, &key, &o, 0);
 500     ok(!res && GetLastError() == WAIT_TIMEOUT, "GetQueuedCompletionStatus returned %x(%lu)\n", res, GetLastError());
 501
 502     res = DeviceIoControl(file, IOCTL_WINETEST_RESET_CANCEL, NULL, 0, NULL, 0, NULL, &overlapped);
 503     ok(res, "DeviceIoControl failed: %lu\n", GetLastError());
 504     res = GetQueuedCompletionStatus(port, &size, &key, &o, 0);
 505     ok(res, "GetQueuedCompletionStatus failed: %lu\n", GetLastError());
 506     ok(o == &overlapped, "o != overlapped\n");
 507
 508     if (pSetFileCompletionNotificationModes)
 509     {
 510         res = pSetFileCompletionNotificationModes(file, FILE_SKIP_COMPLETION_PORT_ON_SUCCESS);
 511         ok(res, "SetFileCompletionNotificationModes failed: %lu\n", GetLastError());
 512
 513         res = DeviceIoControl(file, IOCTL_WINETEST_RESET_CANCEL, NULL, 0, NULL, 0, NULL, &overlapped);
 514         ok(res, "DeviceIoControl failed: %lu\n", GetLastError());
 515         res = GetQueuedCompletionStatus(port, &size, &key, &o, 0);
 516         ok(!res && GetLastError() == WAIT_TIMEOUT, "GetQueuedCompletionStatus returned %x(%lu)\n", res, GetLastError());
 517     }
 518
 519     CloseHandle(port);
 520     CloseHandle(overlapped.hEvent);
 521     CloseHandle(overlapped2.hEvent);
 522     CloseHandle(file);
 523 }
 524
 525 static void test_load_driver(SC_HANDLE service)
 526 {
 527     SERVICE_STATUS status;
 528     BOOL load, res;
 529     DWORD sz;
 530
 531     res = QueryServiceStatus(service, &status);
 532     ok(res, "QueryServiceStatusEx failed: %lu\n", GetLastError());
 533     ok(status.dwCurrentState == SERVICE_STOPPED, "got state %#lx\n", status.dwCurrentState);
 534
 535     load = TRUE;
 536     res = DeviceIoControl(device, IOCTL_WINETEST_LOAD_DRIVER, &load, sizeof(load), NULL, 0, &sz, NULL);
 537     ok(res, "DeviceIoControl failed: %lu\n", GetLastError());
 538
 539     res = QueryServiceStatus(service, &status);
 540     ok(res, "QueryServiceStatusEx failed: %lu\n", GetLastError());
 541     ok(status.dwCurrentState == SERVICE_RUNNING, "got state %#lx\n", status.dwCurrentState);
 542
 543     load = FALSE;
 544     res = DeviceIoControl(device, IOCTL_WINETEST_LOAD_DRIVER, &load, sizeof(load), NULL, 0, &sz, NULL);
 545     ok(res, "DeviceIoControl failed: %lu\n", GetLastError());
 546
 547     res = QueryServiceStatus(service, &status);
 548     ok(res, "QueryServiceStatusEx failed: %lu\n", GetLastError());
 549     ok(status.dwCurrentState == SERVICE_STOPPED, "got state %#lx\n", status.dwCurrentState);
 550 }
 551
 552 static void test_file_handles(void)
 553 {
 554     DWORD count, ret_size;
 555     HANDLE file, dup, file2;
 556     BOOL ret;
 557
 558     ret = DeviceIoControl(device, IOCTL_WINETEST_GET_CREATE_COUNT, NULL, 0, &count, sizeof(count), &ret_size, NULL);
 559     ok(ret, "ioctl failed: %lu\n", GetLastError());
 560     ok(count == 2, "got %lu\n", count);
 561
 562     ret = DeviceIoControl(device, IOCTL_WINETEST_GET_CLOSE_COUNT, NULL, 0, &count, sizeof(count), &ret_size, NULL);
 563     ok(ret, "ioctl failed: %lu\n", GetLastError());
 564     ok(count == 1, "got %lu\n", count);
 565
 566     file = CreateFileA("\\\\.\\WineTestDriver", 0, 0, NULL, OPEN_EXISTING, 0, NULL);
 567     ok(file != INVALID_HANDLE_VALUE, "failed to open device: %lu\n", GetLastError());
 568
 569     ret = DeviceIoControl(device, IOCTL_WINETEST_GET_CREATE_COUNT, NULL, 0, &count, sizeof(count), &ret_size, NULL);
 570     ok(ret, "ioctl failed: %lu\n", GetLastError());
 571     ok(count == 3, "got %lu\n", count);
 572
 573     file2 = CreateFileA("\\\\.\\WineTestDriver", 0, 0, NULL, OPEN_EXISTING, 0, NULL);
 574     ok(file2 != INVALID_HANDLE_VALUE, "failed to open device: %lu\n", GetLastError());
 575
 576     ret = DeviceIoControl(device, IOCTL_WINETEST_GET_CREATE_COUNT, NULL, 0, &count, sizeof(count), &ret_size, NULL);
 577     ok(ret, "ioctl failed: %lu\n", GetLastError());
 578     ok(count == 4, "got %lu\n", count);
 579
 580     ret = DuplicateHandle(GetCurrentProcess(), file, GetCurrentProcess(), &dup, 0, FALSE, DUPLICATE_SAME_ACCESS);
 581     ok(ret, "failed to duplicate handle: %lu\n", GetLastError());
 582
 583     ret = DeviceIoControl(device, IOCTL_WINETEST_GET_CREATE_COUNT, NULL, 0, &count, sizeof(count), &ret_size, NULL);
 584     ok(ret, "ioctl failed: %lu\n", GetLastError());
 585     ok(count == 4, "got %lu\n", count);
 586
 587     ret = DeviceIoControl(device, IOCTL_WINETEST_GET_FSCONTEXT, NULL, 0, &count, sizeof(count), &ret_size, NULL);
 588     ok(ret, "ioctl failed: %lu\n", GetLastError());
 589     ok(count == 1, "got %lu\n", count);
 590
 591     ret = DeviceIoControl(file, IOCTL_WINETEST_GET_FSCONTEXT, NULL, 0, &count, sizeof(count), &ret_size, NULL);
 592     ok(ret, "ioctl failed: %lu\n", GetLastError());
 593     ok(count == 3, "got %lu\n", count);
 594
 595     ret = DeviceIoControl(file2, IOCTL_WINETEST_GET_FSCONTEXT, NULL, 0, &count, sizeof(count), &ret_size, NULL);
 596     ok(ret, "ioctl failed: %lu\n", GetLastError());
 597     ok(count == 4, "got %lu\n", count);
 598
 599     ret = DeviceIoControl(dup, IOCTL_WINETEST_GET_FSCONTEXT, NULL, 0, &count, sizeof(count), &ret_size, NULL);
 600     ok(ret, "ioctl failed: %lu\n", GetLastError());
 601     ok(count == 3, "got %lu\n", count);
 602
 603     CloseHandle(dup);
 604
 605     ret = DeviceIoControl(device, IOCTL_WINETEST_GET_CLOSE_COUNT, NULL, 0, &count, sizeof(count), &ret_size, NULL);
 606     ok(ret, "ioctl failed: %lu\n", GetLastError());
 607     ok(count == 1, "got %lu\n", count);
 608
 609     CloseHandle(file2);
 610
 611     ret = DeviceIoControl(device, IOCTL_WINETEST_GET_CLOSE_COUNT, NULL, 0, &count, sizeof(count), &ret_size, NULL);
 612     ok(ret, "ioctl failed: %lu\n", GetLastError());
 613     ok(count == 2, "got %lu\n", count);
 614
 615     CloseHandle(file);
 616
 617     ret = DeviceIoControl(device, IOCTL_WINETEST_GET_CLOSE_COUNT, NULL, 0, &count, sizeof(count), &ret_size, NULL);
 618     ok(ret, "ioctl failed: %lu\n", GetLastError());
 619     ok(count == 3, "got %lu\n", count);
 620 }
 621
 622 static unsigned int got_return_status_apc;
 623
 624 static void WINAPI return_status_apc(void *apc_user, IO_STATUS_BLOCK *io, ULONG reserved)
 625 {
 626     ++got_return_status_apc;
 627     ok(apc_user == (void *)456, "got %p\n", apc_user);
 628     ok(!reserved, "got reserved %#lx\n", reserved);
 629 }
 630
 631 static void do_return_status(ULONG ioctl, struct return_status_params *params)
 632 {
 633     const char *expect_buffer;
 634     LARGE_INTEGER zero = {{0}};
 635     HANDLE file, port, event;
 636     NTSTATUS expect_status;
 637     ULONG_PTR key, value;
 638     IO_STATUS_BLOCK io;
 639     char buffer[7];
 640     DWORD size;
 641     BOOL ret;
 642
 643     if (params->ret_status == STATUS_PENDING && !params->pending)
 644     {
 645         /* this causes kernel hangs under certain conditions */
 646         return;
 647     }
 648
 649     event = CreateEventW(NULL, TRUE, FALSE, NULL);
 650
 651     if (ioctl != IOCTL_WINETEST_RETURN_STATUS_BUFFERED)
 652         expect_buffer = "ghijkl";
 653     else if (NT_ERROR(params->iosb_status))
 654         expect_buffer = "abcdef";
 655     else
 656         expect_buffer = "ghidef";
 657
 658     /* Test the non-overlapped case. */
 659
 660     expect_status = (params->ret_status == STATUS_PENDING ? params->iosb_status : params->ret_status);
 661
 662     strcpy(buffer, "abcdef");
 663     size = 0xdeadf00d;
 664     SetLastError(0xdeadf00d);
 665     ret = DeviceIoControl(device, ioctl, params, sizeof(*params), buffer, sizeof(buffer), &size, NULL);
 666     todo_wine_if (params->ret_status == STATUS_PENDING && params->iosb_status == STATUS_PENDING)
 667         ok(ret == NT_SUCCESS(expect_status), "got %d\n", ret);
 668     if (NT_SUCCESS(expect_status))
 669     {
 670         todo_wine_if (params->ret_status == STATUS_PENDING && params->iosb_status == STATUS_PENDING)
 671             ok(GetLastError() == 0xdeadf00d, "got error %lu\n", GetLastError());
 672     }
 673     else
 674     {
 675         ok(GetLastError() == RtlNtStatusToDosErrorNoTeb(expect_status), "got error %lu\n", GetLastError());
 676     }
 677     if (NT_ERROR(expect_status))
 678         ok(size == 0xdeadf00d, "got size %lu\n", size);
 679     else if (!NT_ERROR(params->iosb_status))
 680         ok(size == 3, "got size %lu\n", size);
 681     /* size is garbage if !NT_ERROR(expect_status) && NT_ERROR(iosb_status) */
 682     ok(!strcmp(buffer, expect_buffer), "got buffer %s\n", buffer);
 683
 684     strcpy(buffer, "abcdef");
 685     io.Status = 0xdeadf00d;
 686     io.Information = 0xdeadf00d;
 687     ret = NtDeviceIoControlFile(device, NULL, NULL, NULL, &io,
 688             ioctl, params, sizeof(*params), buffer, sizeof(buffer));
 689     ok(ret == expect_status, "got %#x\n", ret);
 690     if (NT_ERROR(params->iosb_status))
 691     {
 692         ok(io.Status == 0xdeadf00d, "got %#lx\n", io.Status);
 693         ok(io.Information == 0xdeadf00d, "got size %Iu\n", io.Information);
 694     }
 695     else
 696     {
 697         ok(io.Status == params->iosb_status, "got %#lx\n", io.Status);
 698         ok(io.Information == 3, "got size %Iu\n", io.Information);
 699     }
 700     ok(!strcmp(buffer, expect_buffer), "got buffer %s\n", buffer);
 701
 702     /* Test the overlapped case. */
 703
 704     file = CreateFileA("\\\\.\\WineTestDriver", FILE_ALL_ACCESS,
 705             0, NULL, OPEN_EXISTING, FILE_FLAG_OVERLAPPED, NULL);
 706     ok(file != INVALID_HANDLE_VALUE, "failed to open device, error %lu\n", GetLastError());
 707     port = CreateIoCompletionPort(file, NULL, 123, 0);
 708     ok(port != NULL, "failed to create port, error %lu\n", GetLastError());
 709
 710     ret = WaitForSingleObject(file, 0);
 711     ok(!ret, "got %d\n", ret);
 712
 713     ResetEvent(event);
 714     strcpy(buffer, "abcdef");
 715     io.Status = 0xdeadf00d;
 716     io.Information = 0xdeadf00d;
 717     ret = NtDeviceIoControlFile(file, event, NULL, (void *)456, &io,
 718             ioctl, params, sizeof(*params), buffer, sizeof(buffer));
 719     ok(ret == params->ret_status
 720             || broken(NT_WARNING(params->ret_status) && ret == STATUS_PENDING), /* win10 */
 721             "got %#x\n", ret);
 722     if (!params->pending && NT_ERROR(params->iosb_status))
 723     {
 724         ok(io.Status == 0xdeadf00d, "got %#lx\n", io.Status);
 725         ok(io.Information == 0xdeadf00d, "got size %Iu\n", io.Information);
 726         ret = WaitForSingleObject(event, 0);
 727         ok(ret == WAIT_TIMEOUT, "got %d\n", ret);
 728     }
 729     else
 730     {
 731         ok(io.Status == params->iosb_status, "got %#lx\n", io.Status);
 732         ok(io.Information == 3, "got size %Iu\n", io.Information);
 733         ret = WaitForSingleObject(event, 0);
 734         ok(!ret, "got %d\n", ret);
 735     }
 736     ok(!strcmp(buffer, expect_buffer), "got buffer %s\n", buffer);
 737
 738     ret = WaitForSingleObject(file, 0);
 739     ok(ret == WAIT_TIMEOUT, "got %d\n", ret);
 740
 741     key = 0xdeadf00d;
 742     value = 0xdeadf00d;
 743     memset(&io, 0xcc, sizeof(io));
 744     ret = NtRemoveIoCompletion(port, &key, &value, &io, &zero);
 745     if (!params->pending && NT_ERROR(params->iosb_status))
 746     {
 747         ok(ret == STATUS_TIMEOUT, "got %#x\n", ret);
 748     }
 749     else
 750     {
 751         ok(!ret, "got %#x\n", ret);
 752         ok(key == 123, "got key %Iu\n", key);
 753         ok(value == 456, "got value %Iu\n", value);
 754         ok(io.Status == params->iosb_status, "got iosb status %#lx\n", io.Status);
 755         ok(io.Information == 3, "got information %Iu\n", io.Information);
 756     }
 757
 758     /* As above, but set the event first, to show that the event is always
 759      * reset. */
 760     ResetEvent(event);
 761     strcpy(buffer, "abcdef");
 762     io.Status = 0xdeadf00d;
 763     io.Information = 0xdeadf00d;
 764     ret = NtDeviceIoControlFile(file, event, NULL, NULL, &io,
 765             ioctl, params, sizeof(*params), buffer, sizeof(buffer));
 766     ok(ret == params->ret_status
 767             || broken(NT_WARNING(params->ret_status) && ret == STATUS_PENDING), /* win10 */
 768             "got %#x\n", ret);
 769     if (!params->pending && NT_ERROR(params->iosb_status))
 770     {
 771         ok(io.Status == 0xdeadf00d, "got %#lx\n", io.Status);
 772         ok(io.Information == 0xdeadf00d, "got size %Iu\n", io.Information);
 773         ret = WaitForSingleObject(event, 0);
 774         ok(ret == WAIT_TIMEOUT, "got %d\n", ret);
 775     }
 776     else
 777     {
 778         ok(io.Status == params->iosb_status, "got %#lx\n", io.Status);
 779         ok(io.Information == 3, "got size %Iu\n", io.Information);
 780         ret = WaitForSingleObject(event, 0);
 781         ok(!ret, "got %d\n", ret);
 782     }
 783     ok(!strcmp(buffer, expect_buffer), "got buffer %s\n", buffer);
 784
 785     /* As above, but use the file handle instead of an event. */
 786     ret = WaitForSingleObject(file, 0);
 787     ok(ret == WAIT_TIMEOUT, "got %d\n", ret);
 788
 789     strcpy(buffer, "abcdef");
 790     io.Status = 0xdeadf00d;
 791     io.Information = 0xdeadf00d;
 792     ret = NtDeviceIoControlFile(file, NULL, NULL, NULL, &io,
 793             ioctl, params, sizeof(*params), buffer, sizeof(buffer));
 794     ok(ret == params->ret_status
 795             || broken(NT_WARNING(params->ret_status) && ret == STATUS_PENDING), /* win10 */
 796             "got %#x\n", ret);
 797     if (!params->pending && NT_ERROR(params->iosb_status))
 798     {
 799         ok(io.Status == 0xdeadf00d, "got %#lx\n", io.Status);
 800         ok(io.Information == 0xdeadf00d, "got size %Iu\n", io.Information);
 801         ret = WaitForSingleObject(file, 0);
 802         ok(ret == WAIT_TIMEOUT, "got %d\n", ret);
 803     }
 804     else
 805     {
 806         ok(io.Status == params->iosb_status, "got %#lx\n", io.Status);
 807         ok(io.Information == 3, "got size %Iu\n", io.Information);
 808         ret = WaitForSingleObject(file, 0);
 809         ok(!ret, "got %d\n", ret);
 810     }
 811     ok(!strcmp(buffer, expect_buffer), "got buffer %s\n", buffer);
 812
 813     /* Test FILE_SKIP_COMPLETION_PORT_ON_SUCCESS. */
 814
 815     if (pSetFileCompletionNotificationModes)
 816     {
 817         ret = pSetFileCompletionNotificationModes(file, FILE_SKIP_COMPLETION_PORT_ON_SUCCESS);
 818         ok(ret, "got error %lu\n", GetLastError());
 819
 820         SetEvent(event);
 821         strcpy(buffer, "abcdef");
 822         io.Status = 0xdeadf00d;
 823         io.Information = 0xdeadf00d;
 824         ret = NtDeviceIoControlFile(file, event, NULL, (void *)456, &io,
 825                 ioctl, params, sizeof(*params), buffer, sizeof(buffer));
 826         ok(ret == params->ret_status
 827                 || broken(NT_WARNING(params->ret_status) && ret == STATUS_PENDING), /* win10 */
 828                 "got %#x\n", ret);
 829         if (!params->pending && NT_ERROR(params->iosb_status))
 830         {
 831             ok(io.Status == 0xdeadf00d, "got %#lx\n", io.Status);
 832             ok(io.Information == 0xdeadf00d, "got size %Iu\n", io.Information);
 833             ret = WaitForSingleObject(event, 0);
 834             ok(ret == WAIT_TIMEOUT, "got %d\n", ret);
 835         }
 836         else
 837         {
 838             ok(io.Status == params->iosb_status, "got %#lx\n", io.Status);
 839             ok(io.Information == 3, "got size %Iu\n", io.Information);
 840             ret = WaitForSingleObject(event, 0);
 841             ok(!ret, "got %d\n", ret);
 842         }
 843         ok(!strcmp(buffer, expect_buffer), "got buffer %s\n", buffer);
 844
 845         key = 0xdeadf00d;
 846         value = 0xdeadf00d;
 847         memset(&io, 0xcc, sizeof(io));
 848         ret = NtRemoveIoCompletion(port, &key, &value, &io, &zero);
 849         if (!params->pending)
 850         {
 851             /* Completion is skipped on non-pending NT_ERROR regardless of file
 852              * options. Windows < 8 interprets
 853              * FILE_SKIP_COMPLETION_PORT_ON_SUCCESS to mean that !NT_ERROR
 854              * should also be skipped. Windows >= 8 restricts this to
 855              * NT_SUCCESS, which has the weird effect that non-pending
 856              * NT_WARNING does *not* skip completion. It's not clear whether
 857              * this is a bug or not—it looks like one, but on the other hand it
 858              * arguably follows the letter of the documentation more closely. */
 859             ok(ret == STATUS_TIMEOUT || (NT_WARNING(params->iosb_status) && !ret), "got %#x\n", ret);
 860         }
 861         else
 862         {
 863             ok(!ret, "got %#x\n", ret);
 864         }
 865         if (!ret)
 866         {
 867             ok(key == 123, "got key %Iu\n", key);
 868             ok(value == 456, "got value %Iu\n", value);
 869             ok(io.Status == params->iosb_status, "got iosb status %#lx\n", io.Status);
 870             ok(io.Information == 3, "got information %Iu\n", io.Information);
 871         }
 872     }
 873
 874     ret = CloseHandle(file);
 875     ok(ret, "failed to close file, error %lu\n", GetLastError());
 876     ret = CloseHandle(port);
 877     ok(ret, "failed to close port, error %lu\n", GetLastError());
 878
 879     /* Test with an APC. */
 880
 881     got_return_status_apc = 0;
 882
 883     file = CreateFileA("\\\\.\\WineTestDriver", FILE_ALL_ACCESS,
 884             0, NULL, OPEN_EXISTING, FILE_FLAG_OVERLAPPED, NULL);
 885     ok(file != INVALID_HANDLE_VALUE, "failed to open device, error %lu\n", GetLastError());
 886
 887     strcpy(buffer, "abcdef");
 888     io.Status = 0xdeadf00d;
 889     io.Information = 0xdeadf00d;
 890     ret = NtDeviceIoControlFile(file, NULL, return_status_apc, (void *)456, &io,
 891             ioctl, params, sizeof(*params), buffer, sizeof(buffer));
 892     ok(ret == params->ret_status, "got %#x\n", ret);
 893     if (!params->pending && NT_ERROR(params->iosb_status))
 894     {
 895         ok(io.Status == 0xdeadf00d, "got %#lx\n", io.Status);
 896         ok(io.Information == 0xdeadf00d, "got size %Iu\n", io.Information);
 897     }
 898     else
 899     {
 900         ok(io.Status == params->iosb_status, "got %#lx\n", io.Status);
 901         ok(io.Information == 3, "got size %Iu\n", io.Information);
 902     }
 903     ok(!strcmp(buffer, expect_buffer), "got buffer %s\n", buffer);
 904
 905     ret = SleepEx(0, TRUE);
 906     if (!params->pending && NT_ERROR(params->iosb_status))
 907     {
 908         ok(!ret, "got %d\n", ret);
 909         ok(!got_return_status_apc, "got %u APC calls\n", got_return_status_apc);
 910     }
 911     else
 912     {
 913         ok(ret == WAIT_IO_COMPLETION, "got %d\n", ret);
 914         ok(got_return_status_apc == 1, "got %u APC calls\n", got_return_status_apc);
 915     }
 916
 917     ret = CloseHandle(file);
 918     ok(ret, "failed to close file, error %lu\n", GetLastError());
 919
 920     CloseHandle(event);
 921 }
 922
 923 static void test_return_status(void)
 924 {
 925     struct return_status_params params;
 926     unsigned int i, j, k;
 927
 928     static const ULONG method_tests[] =
 929     {
 930         IOCTL_WINETEST_RETURN_STATUS_BUFFERED,
 931         IOCTL_WINETEST_RETURN_STATUS_DIRECT,
 932         IOCTL_WINETEST_RETURN_STATUS_NEITHER,
 933     };
 934
 935     static const NTSTATUS status_tests[] =
 936     {
 937         STATUS_SUCCESS,
 938         STATUS_PENDING,
 939         STATUS_TIMEOUT,
 940         0x0eadbeef,
 941         0x4eadbeef,
 942         STATUS_BUFFER_OVERFLOW,
 943         0x8eadbeef,
 944         STATUS_NOT_IMPLEMENTED,
 945         0xceadbeef,
 946     };
 947
 948     for (i = 0; i < ARRAY_SIZE(status_tests); ++i)
 949     {
 950         for (j = 0; j < ARRAY_SIZE(status_tests); ++j)
 951         {
 952             for (params.pending = 0; params.pending <= 1; ++params.pending)
 953             {
 954                 for (k = 0; k < ARRAY_SIZE(method_tests); ++k)
 955                 {
 956                     params.ret_status = status_tests[i];
 957                     params.iosb_status = status_tests[j];
 958
 959                     winetest_push_context("return 0x%08lx, iosb 0x%08lx, pending %d, method %lu",
 960                             params.ret_status, params.iosb_status, params.pending, method_tests[k] & 3);
 961
 962                     do_return_status(method_tests[k], &params);
 963
 964                     winetest_pop_context();
 965                 }
 966             }
 967         }
 968     }
 969 }
 970
 971 static BOOL compare_unicode_string(const WCHAR *buffer, ULONG len, const WCHAR *expect)
 972 {
 973     return len == wcslen(expect) * sizeof(WCHAR) && !memcmp(buffer, expect, len);
 974 }
 975
 976 static void test_object_info(void)
 977 {
 978     char buffer[200];
 979     OBJECT_NAME_INFORMATION *name_info = (OBJECT_NAME_INFORMATION *)buffer;
 980     OBJECT_TYPE_INFORMATION *type_info = (OBJECT_TYPE_INFORMATION *)buffer;
 981     FILE_FS_VOLUME_INFORMATION *volume_info = (FILE_FS_VOLUME_INFORMATION *)buffer;
 982     FILE_NAME_INFORMATION *file_info = (FILE_NAME_INFORMATION *)buffer;
 983     HANDLE file;
 984     NTSTATUS status;
 985     IO_STATUS_BLOCK io;
 986     ULONG size;
 987
 988     status = NtQueryObject(device, ObjectNameInformation, buffer, sizeof(buffer), NULL);
 989     ok(!status, "got %#lx\n", status);
 990     ok(compare_unicode_string(name_info->Name.Buffer, name_info->Name.Length, L"\\Device\\WineTestDriver"),
 991             "wrong name %s\n", debugstr_w(name_info->Name.Buffer));
 992
 993     status = NtQueryObject(device, ObjectTypeInformation, buffer, sizeof(buffer), NULL);
 994     ok(!status, "got %#lx\n", status);
 995     ok(compare_unicode_string(type_info->TypeName.Buffer, type_info->TypeName.Length, L"File"),
 996             "wrong name %s\n", debugstr_wn(type_info->TypeName.Buffer, type_info->TypeName.Length / sizeof(WCHAR)));
 997
 998     status = NtQueryInformationFile(device, &io, buffer, sizeof(buffer), FileNameInformation);
 999     todo_wine ok(status == STATUS_INVALID_DEVICE_REQUEST, "got %#lx\n", status);
1000
1001     status = NtQueryVolumeInformationFile(device, &io, buffer, sizeof(buffer), FileFsVolumeInformation);
1002     todo_wine ok(status == STATUS_INVALID_DEVICE_REQUEST, "got %#lx\n", status);
1003
1004     file = CreateFileA("\\\\.\\WineTestDriver\\subfile", 0, 0, NULL, OPEN_EXISTING, 0, NULL);
1005     todo_wine ok(file != INVALID_HANDLE_VALUE, "got error %lu\n", GetLastError());
1006     if (file == INVALID_HANDLE_VALUE) return;
1007
1008     memset(buffer, 0xcc, sizeof(buffer));
1009     status = NtQueryObject(file, ObjectNameInformation, buffer, sizeof(buffer), &size);
1010     ok(!status, "got %#lx\n", status);
1011     ok(size == sizeof(*name_info) + sizeof(L"\\Device\\WineTestDriver\\subfile"), "wrong size %lu\n", size);
1012     ok(compare_unicode_string(name_info->Name.Buffer, name_info->Name.Length, L"\\Device\\WineTestDriver\\subfile"),
1013             "wrong name %s\n", debugstr_w(name_info->Name.Buffer));
1014
1015     memset(buffer, 0xcc, sizeof(buffer));
1016     status = NtQueryObject(file, ObjectNameInformation, buffer, size - 2, &size);
1017     ok(status == STATUS_BUFFER_OVERFLOW, "got %#lx\n", status);
1018     ok(size == sizeof(*name_info) + sizeof(L"\\Device\\WineTestDriver\\subfile"), "wrong size %lu\n", size);
1019     ok(compare_unicode_string(name_info->Name.Buffer, name_info->Name.Length, L"\\Device\\WineTestDriver\\subfil"),
1020             "wrong name %s\n", debugstr_w(name_info->Name.Buffer));
1021
1022     memset(buffer, 0xcc, sizeof(buffer));
1023     status = NtQueryObject(file, ObjectNameInformation, buffer, sizeof(*name_info), &size);
1024     ok(status == STATUS_BUFFER_OVERFLOW, "got %#lx\n", status);
1025     ok(size == sizeof(*name_info) + sizeof(L"\\Device\\WineTestDriver\\subfile"), "wrong size %lu\n", size);
1026
1027     status = NtQueryObject(file, ObjectTypeInformation, buffer, sizeof(buffer), NULL);
1028     ok(!status, "got %#lx\n", status);
1029     ok(compare_unicode_string(type_info->TypeName.Buffer, type_info->TypeName.Length, L"File"),
1030             "wrong name %s\n", debugstr_wn(type_info->TypeName.Buffer, type_info->TypeName.Length / sizeof(WCHAR)));
1031
1032     status = NtQueryInformationFile(file, &io, buffer, sizeof(buffer), FileNameInformation);
1033     ok(!status, "got %#lx\n", status);
1034     ok(compare_unicode_string(file_info->FileName, file_info->FileNameLength, L"\\subfile"),
1035             "wrong name %s\n", debugstr_wn(file_info->FileName, file_info->FileNameLength / sizeof(WCHAR)));
1036
1037     io.Status = 0xdeadf00d;
1038     io.Information = 0xdeadf00d;
1039     status = NtQueryVolumeInformationFile(file, &io, buffer, sizeof(buffer), FileFsVolumeInformation);
1040     ok(!status, "got %#lx\n", status);
1041     ok(!io.Status, "got status %#lx\n", io.Status);
1042     size = offsetof(FILE_FS_VOLUME_INFORMATION, VolumeLabel) + volume_info->VolumeLabelLength;
1043     ok(io.Information == size, "expected information %lu, got %Iu\n", size, io.Information);
1044     ok(volume_info->VolumeSerialNumber == 0xdeadbeef,
1045             "wrong serial number 0x%08lx\n", volume_info->VolumeSerialNumber);
1046     ok(compare_unicode_string(volume_info->VolumeLabel, volume_info->VolumeLabelLength, L"WineTestDriver"),
1047             "wrong name %s\n", debugstr_wn(volume_info->VolumeLabel, volume_info->VolumeLabelLength / sizeof(WCHAR)));
1048
1049     io.Status = 0xdeadf00d;
1050     io.Information = 0xdeadf00d;
1051     status = NtQueryVolumeInformationFile(file, &io, buffer, sizeof(buffer), FileFsAttributeInformation);
1052     ok(status == STATUS_NOT_IMPLEMENTED, "got %#lx\n", status);
1053     ok(io.Status == 0xdeadf00d, "got status %#lx\n", io.Status);
1054     ok(io.Information == 0xdeadf00d, "got information %Iu\n", io.Information);
1055
1056     CloseHandle(file);
1057
1058     file = CreateFileA("\\\\.\\WineTestDriver\\notimpl", 0, 0, NULL, OPEN_EXISTING, 0, NULL);
1059     ok(file != INVALID_HANDLE_VALUE, "got error %lu\n", GetLastError());
1060
1061     status = NtQueryObject(file, ObjectNameInformation, buffer, sizeof(buffer), NULL);
1062     ok(!status, "got %#lx\n", status);
1063     ok(compare_unicode_string(name_info->Name.Buffer, name_info->Name.Length, L"\\Device\\WineTestDriver"),
1064             "wrong name %s\n", debugstr_w(name_info->Name.Buffer));
1065
1066     status = NtQueryInformationFile(file, &io, buffer, sizeof(buffer), FileNameInformation);
1067     ok(status == STATUS_NOT_IMPLEMENTED, "got %#lx\n", status);
1068
1069     CloseHandle(file);
1070
1071     file = CreateFileA("\\\\.\\WineTestDriver\\badparam", 0, 0, NULL, OPEN_EXISTING, 0, NULL);
1072     ok(file != INVALID_HANDLE_VALUE, "got error %lu\n", GetLastError());
1073
1074     status = NtQueryObject(file, ObjectNameInformation, buffer, sizeof(buffer), NULL);
1075     ok(!status, "got %#lx\n", status);
1076     ok(compare_unicode_string(name_info->Name.Buffer, name_info->Name.Length, L"\\Device\\WineTestDriver"),
1077             "wrong name %s\n", debugstr_w(name_info->Name.Buffer));
1078
1079     status = NtQueryInformationFile(file, &io, buffer, sizeof(buffer), FileNameInformation);
1080     ok(status == STATUS_INVALID_PARAMETER, "got %#lx\n", status);
1081
1082     CloseHandle(file);
1083
1084     file = CreateFileA("\\\\.\\WineTestDriver\\genfail", 0, 0, NULL, OPEN_EXISTING, 0, NULL);
1085     ok(file != INVALID_HANDLE_VALUE, "got error %lu\n", GetLastError());
1086
1087     status = NtQueryObject(file, ObjectNameInformation, buffer, sizeof(buffer), NULL);
1088     ok(status == STATUS_UNSUCCESSFUL, "got %#lx\n", status);
1089
1090     status = NtQueryInformationFile(file, &io, buffer, sizeof(buffer), FileNameInformation);
1091     ok(status == STATUS_UNSUCCESSFUL, "got %#lx\n", status);
1092
1093     CloseHandle(file);
1094
1095     file = CreateFileA("\\\\.\\WineTestDriver\\badtype", 0, 0, NULL, OPEN_EXISTING, 0, NULL);
1096     ok(file != INVALID_HANDLE_VALUE, "got error %lu\n", GetLastError());
1097
1098     status = NtQueryObject(file, ObjectNameInformation, buffer, sizeof(buffer), NULL);
1099     ok(status == STATUS_OBJECT_TYPE_MISMATCH, "got %#lx\n", status);
1100
1101     status = NtQueryInformationFile(file, &io, buffer, sizeof(buffer), FileNameInformation);
1102     ok(status == STATUS_OBJECT_TYPE_MISMATCH, "got %#lx\n", status);
1103
1104     CloseHandle(file);
1105 }
1106
1107 static void test_blocking_irp(void)
1108 {
1109     char buffer[40];
1110     IO_STATUS_BLOCK io;
1111     NTSTATUS status;
1112     HANDLE file;
1113
1114     file = CreateFileA("\\\\.\\WineTestDriver\\", FILE_ALL_ACCESS, 0, NULL, OPEN_EXISTING, 0, NULL);
1115     ok(file != INVALID_HANDLE_VALUE, "failed to open device: %lu\n", GetLastError());
1116
1117     memset(&io, 0xcc, sizeof(io));
1118     status = NtQueryVolumeInformationFile(file, &io, buffer, sizeof(buffer), FileFsSizeInformation);
1119     ok(!status, "got %#lx\n", status);
1120     ok(!io.Status, "got iosb status %#lx\n", io.Status);
1121     ok(!io.Information, "got information %#Ix\n", io.Information);
1122
1123     io.Status = 0xdeadf00d;
1124     io.Information = 0xdeadf00d;
1125     status = NtQueryVolumeInformationFile(file, &io, buffer, sizeof(buffer), FileFsFullSizeInformation);
1126     ok(status == STATUS_DEVICE_NOT_READY, "got %#lx\n", status);
1127     ok(io.Status == 0xdeadf00d, "got iosb status %#lx\n", io.Status);
1128     ok(io.Information == 0xdeadf00d, "got information %#Ix\n", io.Information);
1129
1130     CloseHandle(file);
1131
1132     file = CreateFileA("\\\\.\\WineTestDriver\\", FILE_ALL_ACCESS, 0, NULL, OPEN_EXISTING, FILE_FLAG_OVERLAPPED, NULL);
1133     ok(file != INVALID_HANDLE_VALUE, "failed to open device: %lu\n", GetLastError());
1134
1135     memset(&io, 0xcc, sizeof(io));
1136     status = NtQueryVolumeInformationFile(file, &io, buffer, sizeof(buffer), FileFsSizeInformation);
1137     ok(!status, "got %#lx\n", status);
1138     ok(!io.Status, "got iosb status %#lx\n", io.Status);
1139     ok(!io.Information, "got information %#Ix\n", io.Information);
1140
1141     memset(&io, 0xcc, sizeof(io));
1142     status = NtQueryVolumeInformationFile(file, &io, buffer, sizeof(buffer), FileFsFullSizeInformation);
1143     ok(status == STATUS_DEVICE_NOT_READY, "got %#lx\n", status);
1144     ok(io.Status == STATUS_DEVICE_NOT_READY, "got iosb status %#lx\n", io.Status);
1145     ok(!io.Information, "got information %#Ix\n", io.Information);
1146
1147     CloseHandle(file);
1148 }
1149
1150 static void test_driver3(struct testsign_context *ctx)
1151 {
1152     WCHAR filename[MAX_PATH];
1153     SC_HANDLE service;
1154     BOOL ret;
1155
1156     service = load_driver(ctx, filename, L"driver3.dll", L"WineTestDriver3");
1157     ok(service != NULL, "driver3 failed to load\n");
1158
1159     ret = StartServiceA(service, 0, NULL);
1160     ok(!ret, "driver3 should fail to start\n");
1161     ok(GetLastError() == ERROR_CALL_NOT_IMPLEMENTED ||
1162        GetLastError() == ERROR_INVALID_FUNCTION ||
1163        GetLastError() == ERROR_PROC_NOT_FOUND /* XP */ ||
1164        GetLastError() == ERROR_FILE_NOT_FOUND /* Win7 */, "got %lu\n", GetLastError());
1165
1166     DeleteService(service);
1167     CloseServiceHandle(service);
1168     DeleteFileW(filename);
1169 }
1170
1171 static DWORD WINAPI wsk_test_thread(void *parameter)
1172 {
1173     static const char test_send_string[] = "Client test string 1.";
1174     static const WORD version = MAKEWORD(2, 2);
1175     SOCKET s_listen, s_accept, s_connect;
1176     struct sockaddr_in addr;
1177     char buffer[256];
1178     int ret, err;
1179     WSADATA data;
1180     int opt_val;
1181
1182     ret = WSAStartup(version, &data);
1183     ok(!ret, "WSAStartup() failed, ret %u.\n", ret);
1184
1185     s_connect = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
1186     ok(s_connect != INVALID_SOCKET, "Error creating socket, WSAGetLastError() %u.\n", WSAGetLastError());
1187
1188     s_listen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
1189     ok(s_listen != INVALID_SOCKET, "Error creating socket, WSAGetLastError() %u.\n", WSAGetLastError());
1190
1191     opt_val = 1;
1192     setsockopt(s_listen, SOL_SOCKET, SO_REUSEADDR, (const char *)&opt_val, sizeof(opt_val));
1193
1194     memset(&addr, 0, sizeof(addr));
1195     addr.sin_family = AF_INET;
1196     addr.sin_port = htons(CLIENT_LISTEN_PORT);
1197     addr.sin_addr.s_addr = htonl(0x7f000001);
1198     ret = bind(s_listen, (struct sockaddr *)&addr, sizeof(addr));
1199     ok(!ret, "Got unexpected ret %d, WSAGetLastError() %u.\n", ret, WSAGetLastError());
1200
1201     ret = listen(s_listen, SOMAXCONN);
1202     ok(!ret, "Got unexpected ret %d, WSAGetLastError() %u.\n", ret, WSAGetLastError());
1203
1204     addr.sin_port = htons(SERVER_LISTEN_PORT);
1205
1206     ret = connect(s_connect, (struct sockaddr *)&addr, sizeof(addr));
1207     while (ret && ((err = WSAGetLastError()) == WSAECONNREFUSED || err == WSAECONNABORTED))
1208     {
1209         SwitchToThread();
1210         ret = connect(s_connect, (struct sockaddr *)&addr, sizeof(addr));
1211     }
1212     ok(!ret, "Error connecting, WSAGetLastError() %u.\n", WSAGetLastError());
1213
1214     ret = send(s_connect, test_send_string, sizeof(test_send_string), 0);
1215     ok(ret == sizeof(test_send_string), "Got unexpected ret %d.\n", ret);
1216
1217     ret = recv(s_connect, buffer, sizeof(buffer), 0);
1218     ok(ret == sizeof(buffer), "Got unexpected ret %d.\n", ret);
1219     ok(!strcmp(buffer, "Server test string 1."), "Received unexpected data.\n");
1220
1221     s_accept = accept(s_listen, NULL, NULL);
1222     ok(s_accept != INVALID_SOCKET, "Error creating socket, WSAGetLastError() %u.\n", WSAGetLastError());
1223
1224     closesocket(s_accept);
1225     closesocket(s_connect);
1226     closesocket(s_listen);
1227     return TRUE;
1228 }
1229
1230 static void test_driver_netio(struct testsign_context *ctx)
1231 {
1232     WCHAR filename[MAX_PATH];
1233     SC_HANDLE service;
1234     HANDLE hthread;
1235     BOOL ret;
1236
1237     if (!(service = load_driver(ctx, filename, L"driver_netio.dll", L"winetest_netio")))
1238         return;
1239
1240     if (!start_driver(service, TRUE))
1241     {
1242         DeleteFileW(filename);
1243         return;
1244     }
1245
1246     device = CreateFileA("\\\\.\\winetest_netio", 0, 0, NULL, OPEN_EXISTING, 0, NULL);
1247     ok(device != INVALID_HANDLE_VALUE, "failed to open device: %lu\n", GetLastError());
1248
1249     hthread = CreateThread(NULL, 0, wsk_test_thread, NULL, 0, NULL);
1250     main_test();
1251     WaitForSingleObject(hthread, INFINITE);
1252
1253     CloseHandle(device);
1254
1255     unload_driver(service);
1256     ret = DeleteFileW(filename);
1257     ok(ret, "DeleteFile failed: %lu\n", GetLastError());
1258
1259     cat_okfile();
1260 }
1261
1262 #ifdef __i386__
1263 #define EXT "x86"
1264 #elif defined(__x86_64__)
1265 #define EXT "amd64"
1266 #elif defined(__arm__)
1267 #define EXT "arm"
1268 #elif defined(__aarch64__)
1269 #define EXT "arm64"
1270 #else
1271 #define EXT
1272 #endif
1273
1274 static const char inf_text[] =
1275     "[Version]\n"
1276     "Signature=$Chicago$\n"
1277     "ClassGuid={4d36e97d-e325-11ce-bfc1-08002be10318}\n"
1278     "CatalogFile=winetest.cat\n"
1279     "DriverVer=09/21/2006,6.0.5736.1\n"
1280
1281     "[Manufacturer]\n"
1282     "Wine=mfg_section,NT" EXT "\n"
1283
1284     "[mfg_section.NT" EXT "]\n"
1285     "Wine test root driver=device_section,test_hardware_id\n"
1286
1287     "[device_section.NT" EXT "]\n"
1288     "CopyFiles=file_section\n"
1289
1290     "[device_section.NT" EXT ".Services]\n"
1291     "AddService=winetest,0x2,svc_section\n"
1292
1293     "[file_section]\n"
1294     "winetest.sys\n"
1295
1296     "[SourceDisksFiles]\n"
1297     "winetest.sys=1\n"
1298
1299     "[SourceDisksNames]\n"
1300     "1=,winetest.sys\n"
1301
1302     "[DestinationDirs]\n"
1303     "DefaultDestDir=12\n"
1304
1305     "[svc_section]\n"
1306     "ServiceBinary=%12%\\winetest.sys\n"
1307     "ServiceType=1\n"
1308     "StartType=3\n"
1309     "ErrorControl=1\n"
1310     "LoadOrderGroup=Extended Base\n"
1311     "DisplayName=\"winetest bus driver\"\n"
1312     "; they don't sleep anymore, on the beach\n";
1313
1314 static void add_file_to_catalog(HANDLE catalog, const WCHAR *file)
1315 {
1316     SIP_SUBJECTINFO subject_info = {sizeof(SIP_SUBJECTINFO)};
1317     SIP_INDIRECT_DATA *indirect_data;
1318     const WCHAR *filepart = file;
1319     CRYPTCATMEMBER *member;
1320     WCHAR hash_buffer[100];
1321     GUID subject_guid;
1322     unsigned int i;
1323     DWORD size;
1324     BOOL ret;
1325
1326     ret = CryptSIPRetrieveSubjectGuidForCatalogFile(file, NULL, &subject_guid);
1327     todo_wine ok(ret, "Failed to get subject guid, error %lu\n", GetLastError());
1328
1329     size = 0;
1330     subject_info.pgSubjectType = &subject_guid;
1331     subject_info.pwsFileName = file;
1332     subject_info.DigestAlgorithm.pszObjId = (char *)szOID_OIWSEC_sha1;
1333     subject_info.dwFlags = SPC_INC_PE_RESOURCES_FLAG | SPC_INC_PE_IMPORT_ADDR_TABLE_FLAG | SPC_EXC_PE_PAGE_HASHES_FLAG | 0x10000;
1334     ret = CryptSIPCreateIndirectData(&subject_info, &size, NULL);
1335     todo_wine ok(ret, "Failed to get indirect data size, error %lu\n", GetLastError());
1336
1337     indirect_data = malloc(size);
1338     ret = CryptSIPCreateIndirectData(&subject_info, &size, indirect_data);
1339     todo_wine ok(ret, "Failed to get indirect data, error %lu\n", GetLastError());
1340     if (ret)
1341     {
1342         memset(hash_buffer, 0, sizeof(hash_buffer));
1343         for (i = 0; i < indirect_data->Digest.cbData; ++i)
1344             swprintf(&hash_buffer[i * 2], 2, L"%02X", indirect_data->Digest.pbData[i]);
1345
1346         member = CryptCATPutMemberInfo(catalog, (WCHAR *)file,
1347                 hash_buffer, &subject_guid, 0, size, (BYTE *)indirect_data);
1348         ok(!!member, "Failed to write member, error %lu\n", GetLastError());
1349
1350         if (wcsrchr(file, '\\'))
1351             filepart = wcsrchr(file, '\\') + 1;
1352
1353         ret = !!CryptCATPutAttrInfo(catalog, member, (WCHAR *)L"File",
1354                 CRYPTCAT_ATTR_NAMEASCII | CRYPTCAT_ATTR_DATAASCII | CRYPTCAT_ATTR_AUTHENTICATED,
1355                 (wcslen(filepart) + 1) * 2, (BYTE *)filepart);
1356         ok(ret, "Failed to write attr, error %lu\n", GetLastError());
1357
1358         ret = !!CryptCATPutAttrInfo(catalog, member, (WCHAR *)L"OSAttr",
1359                 CRYPTCAT_ATTR_NAMEASCII | CRYPTCAT_ATTR_DATAASCII | CRYPTCAT_ATTR_AUTHENTICATED,
1360                 sizeof(L"2:6.0"), (BYTE *)L"2:6.0");
1361         ok(ret, "Failed to write attr, error %lu\n", GetLastError());
1362     }
1363
1364     free(indirect_data);
1365 }
1366
1367 static const GUID bus_class     = {0xdeadbeef, 0x29ef, 0x4538, {0xa5, 0xfd, 0xb6, 0x95, 0x73, 0xa3, 0x62, 0xc1}};
1368 static const GUID child_class   = {0xdeadbeef, 0x29ef, 0x4538, {0xa5, 0xfd, 0xb6, 0x95, 0x73, 0xa3, 0x62, 0xc2}};
1369
1370 static unsigned int got_bus_arrival, got_bus_removal, got_child_arrival, got_child_removal;
1371
1372 static LRESULT WINAPI device_notify_proc(HWND window, UINT message, WPARAM wparam, LPARAM lparam)
1373 {
1374     if (message != WM_DEVICECHANGE)
1375         return DefWindowProcA(window, message, wparam, lparam);
1376
1377     switch (wparam)
1378     {
1379         case DBT_DEVNODES_CHANGED:
1380             if (winetest_debug > 1) trace("device nodes changed\n");
1381
1382             ok(InSendMessageEx(NULL) == ISMEX_NOTIFY, "got message flags %#lx\n", InSendMessageEx(NULL));
1383             ok(!lparam, "got lparam %#Ix\n", lparam);
1384             break;
1385
1386         case DBT_DEVICEARRIVAL:
1387         {
1388             const DEV_BROADCAST_DEVICEINTERFACE_A *iface = (const DEV_BROADCAST_DEVICEINTERFACE_A *)lparam;
1389             DWORD expect_size = offsetof(DEV_BROADCAST_DEVICEINTERFACE_A, dbcc_name[strlen(iface->dbcc_name)]);
1390
1391             if (winetest_debug > 1) trace("device arrival %s\n", iface->dbcc_name);
1392
1393             ok(InSendMessageEx(NULL) == ISMEX_SEND, "got message flags %#lx\n", InSendMessageEx(NULL));
1394
1395             ok(iface->dbcc_devicetype == DBT_DEVTYP_DEVICEINTERFACE,
1396                     "got unexpected notification type %#lx\n", iface->dbcc_devicetype);
1397             ok(iface->dbcc_size >= expect_size, "expected size at least %lu, got %lu\n", expect_size, iface->dbcc_size);
1398             ok(!iface->dbcc_reserved, "got reserved %#lx\n", iface->dbcc_reserved);
1399             if (IsEqualGUID(&iface->dbcc_classguid, &bus_class))
1400             {
1401                 ++got_bus_arrival;
1402                 ok(!strcmp(iface->dbcc_name, "\\\\?\\ROOT#WINETEST#0#{deadbeef-29ef-4538-a5fd-b69573a362c1}"),
1403                         "got name %s\n", debugstr_a(iface->dbcc_name));
1404             }
1405             else if (IsEqualGUID(&iface->dbcc_classguid, &child_class))
1406             {
1407                 ++got_child_arrival;
1408                 ok(!strcmp(iface->dbcc_name, "\\\\?\\Wine#Test#1#{deadbeef-29ef-4538-a5fd-b69573a362c2}"),
1409                         "got name %s\n", debugstr_a(iface->dbcc_name));
1410             }
1411             break;
1412         }
1413
1414         case DBT_DEVICEREMOVECOMPLETE:
1415         {
1416             const DEV_BROADCAST_DEVICEINTERFACE_A *iface = (const DEV_BROADCAST_DEVICEINTERFACE_A *)lparam;
1417             DWORD expect_size = offsetof(DEV_BROADCAST_DEVICEINTERFACE_A, dbcc_name[strlen(iface->dbcc_name)]);
1418
1419             if (winetest_debug > 1) trace("device removal %s\n", iface->dbcc_name);
1420
1421             ok(InSendMessageEx(NULL) == ISMEX_SEND, "got message flags %#lx\n", InSendMessageEx(NULL));
1422
1423             ok(iface->dbcc_devicetype == DBT_DEVTYP_DEVICEINTERFACE,
1424                     "got unexpected notification type %#lx\n", iface->dbcc_devicetype);
1425             ok(iface->dbcc_size >= expect_size, "expected size at least %lu, got %lu\n", expect_size, iface->dbcc_size);
1426             ok(!iface->dbcc_reserved, "got reserved %#lx\n", iface->dbcc_reserved);
1427             if (IsEqualGUID(&iface->dbcc_classguid, &bus_class))
1428             {
1429                 ++got_bus_removal;
1430                 ok(!strcmp(iface->dbcc_name, "\\\\?\\ROOT#WINETEST#0#{deadbeef-29ef-4538-a5fd-b69573a362c1}"),
1431                         "got name %s\n", debugstr_a(iface->dbcc_name));
1432             }
1433             else if (IsEqualGUID(&iface->dbcc_classguid, &child_class))
1434             {
1435                 ++got_child_removal;
1436                 ok(!strcmp(iface->dbcc_name, "\\\\?\\Wine#Test#1#{deadbeef-29ef-4538-a5fd-b69573a362c2}"),
1437                         "got name %s\n", debugstr_a(iface->dbcc_name));
1438             }
1439             break;
1440         }
1441     }
1442     return DefWindowProcA(window, message, wparam, lparam);
1443 }
1444
1445 static void pump_messages(void)
1446 {
1447     MSG msg;
1448
1449     if (!MsgWaitForMultipleObjects(0, NULL, FALSE, 200, QS_ALLINPUT))
1450     {
1451         while (PeekMessageA(&msg, NULL, 0, 0, PM_REMOVE))
1452         {
1453             TranslateMessage(&msg);
1454             DispatchMessageA(&msg);
1455         }
1456     }
1457 }
1458
1459 static void test_pnp_devices(void)
1460 {
1461     static const char expect_hardware_id[] = "winetest_hardware\0winetest_hardware_1\0";
1462     static const char expect_compat_id[] = "winetest_compat\0winetest_compat_1\0";
1463     static const WCHAR expect_container_id_w[] = L"{12345678-1234-1234-1234-123456789123}";
1464
1465     char buffer[200];
1466     WCHAR buffer_w[200];
1467     SP_DEVICE_INTERFACE_DETAIL_DATA_A *iface_detail = (void *)buffer;
1468     SP_DEVICE_INTERFACE_DATA iface = {sizeof(iface)};
1469     SP_DEVINFO_DATA device = {sizeof(device)};
1470     DEV_BROADCAST_DEVICEINTERFACE_A filter =
1471     {
1472         .dbcc_size = sizeof(filter),
1473         .dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE,
1474     };
1475     static const WNDCLASSA class =
1476     {
1477         .lpszClassName = "ntoskrnl_test_wc",
1478         .lpfnWndProc = device_notify_proc,
1479     };
1480     HDEVNOTIFY notify_handle;
1481     DWORD size, type, dword;
1482     HANDLE bus, child, tmp;
1483     OBJECT_ATTRIBUTES attr;
1484     UNICODE_STRING string;
1485     OVERLAPPED ovl = {0};
1486     IO_STATUS_BLOCK io;
1487     HDEVINFO set;
1488     HWND window;
1489     BOOL ret;
1490     int id;
1491
1492     ret = RegisterClassA(&class);
1493     ok(ret, "failed to register class\n");
1494     window = CreateWindowA("ntoskrnl_test_wc", NULL, 0, 0, 0, 0, 0, HWND_MESSAGE, NULL, NULL, NULL);
1495     ok(!!window, "failed to create window\n");
1496     notify_handle = RegisterDeviceNotificationA(window, &filter, DEVICE_NOTIFY_ALL_INTERFACE_CLASSES);
1497     ok(!!notify_handle, "failed to register window, error %lu\n", GetLastError());
1498
1499     set = SetupDiGetClassDevsA(&control_class, NULL, NULL, DIGCF_PRESENT | DIGCF_DEVICEINTERFACE);
1500     ok(set != INVALID_HANDLE_VALUE, "failed to get device list, error %#lx\n", GetLastError());
1501
1502     ret = SetupDiEnumDeviceInfo(set, 0, &device);
1503     ok(ret, "failed to get device, error %#lx\n", GetLastError());
1504     ok(IsEqualGUID(&device.ClassGuid, &GUID_DEVCLASS_SYSTEM), "wrong class %s\n", debugstr_guid(&device.ClassGuid));
1505
1506     ret = SetupDiGetDeviceInstanceIdA(set, &device, buffer, sizeof(buffer), NULL);
1507     ok(ret, "failed to get device ID, error %#lx\n", GetLastError());
1508     ok(!strcmp(buffer, "ROOT\\WINETEST\\0"), "got ID %s\n", debugstr_a(buffer));
1509
1510     ret = SetupDiEnumDeviceInterfaces(set, NULL, &control_class, 0, &iface);
1511     ok(ret, "failed to get interface, error %#lx\n", GetLastError());
1512     ok(IsEqualGUID(&iface.InterfaceClassGuid, &control_class),
1513             "wrong class %s\n", debugstr_guid(&iface.InterfaceClassGuid));
1514     ok(iface.Flags == SPINT_ACTIVE, "got flags %#lx\n", iface.Flags);
1515
1516     iface_detail->cbSize = sizeof(*iface_detail);
1517     ret = SetupDiGetDeviceInterfaceDetailA(set, &iface, iface_detail, sizeof(buffer), NULL, NULL);
1518     ok(ret, "failed to get interface path, error %#lx\n", GetLastError());
1519     ok(!strcmp(iface_detail->DevicePath, "\\\\?\\root#winetest#0#{deadbeef-29ef-4538-a5fd-b69573a362c0}"),
1520             "wrong path %s\n", debugstr_a(iface_detail->DevicePath));
1521
1522     SetupDiDestroyDeviceInfoList(set);
1523
1524     bus = CreateFileA(iface_detail->DevicePath, 0, 0, NULL, OPEN_EXISTING, 0, NULL);
1525     ok(bus != INVALID_HANDLE_VALUE, "got error %lu\n", GetLastError());
1526
1527     ret = DeviceIoControl(bus, IOCTL_WINETEST_BUS_MAIN, NULL, 0, NULL, 0, &size, NULL);
1528     ok(ret, "got error %lu\n", GetLastError());
1529
1530     /* Test IoRegisterDeviceInterface() and IoSetDeviceInterfaceState(). */
1531
1532     set = SetupDiGetClassDevsA(&bus_class, NULL, NULL, DIGCF_DEVICEINTERFACE);
1533     ok(set != INVALID_HANDLE_VALUE, "failed to get device list, error %#lx\n", GetLastError());
1534     ret = SetupDiEnumDeviceInterfaces(set, NULL, &bus_class, 0, &iface);
1535     ok(!ret, "expected failure\n");
1536     ok(GetLastError() == ERROR_NO_MORE_ITEMS, "got error %#lx\n", GetLastError());
1537     SetupDiDestroyDeviceInfoList(set);
1538
1539     ret = DeviceIoControl(bus, IOCTL_WINETEST_BUS_REGISTER_IFACE, NULL, 0, NULL, 0, &size, NULL);
1540     ok(ret, "got error %lu\n", GetLastError());
1541
1542     set = SetupDiGetClassDevsA(&bus_class, NULL, NULL, DIGCF_DEVICEINTERFACE);
1543     ok(set != INVALID_HANDLE_VALUE, "failed to get device list, error %#lx\n", GetLastError());
1544     ret = SetupDiEnumDeviceInterfaces(set, NULL, &bus_class, 0, &iface);
1545     ok(ret, "failed to get interface, error %#lx\n", GetLastError());
1546     ok(IsEqualGUID(&iface.InterfaceClassGuid, &bus_class),
1547             "wrong class %s\n", debugstr_guid(&iface.InterfaceClassGuid));
1548     ok(!iface.Flags, "got flags %#lx\n", iface.Flags);
1549     SetupDiDestroyDeviceInfoList(set);
1550
1551     set = SetupDiGetClassDevsA(&bus_class, NULL, NULL, DIGCF_DEVICEINTERFACE | DIGCF_PRESENT);
1552     ok(set != INVALID_HANDLE_VALUE, "failed to get device list, error %#lx\n", GetLastError());
1553     ret = SetupDiEnumDeviceInterfaces(set, NULL, &bus_class, 0, &iface);
1554     ok(!ret, "expected failure\n");
1555     ok(GetLastError() == ERROR_NO_MORE_ITEMS, "got error %#lx\n", GetLastError());
1556     SetupDiDestroyDeviceInfoList(set);
1557
1558     ret = DeviceIoControl(bus, IOCTL_WINETEST_BUS_ENABLE_IFACE, NULL, 0, NULL, 0, &size, NULL);
1559     ok(ret, "got error %lu\n", GetLastError());
1560
1561     pump_messages();
1562     ok(got_bus_arrival == 1, "got %u bus arrival messages\n", got_bus_arrival);
1563     ok(!got_bus_removal, "got %u bus removal messages\n", got_bus_removal);
1564
1565     set = SetupDiGetClassDevsA(&bus_class, NULL, NULL, DIGCF_DEVICEINTERFACE | DIGCF_PRESENT);
1566     ok(set != INVALID_HANDLE_VALUE, "failed to get device list, error %#lx\n", GetLastError());
1567     ret = SetupDiEnumDeviceInterfaces(set, NULL, &bus_class, 0, &iface);
1568     ok(ret, "failed to get interface, error %#lx\n", GetLastError());
1569     ok(IsEqualGUID(&iface.InterfaceClassGuid, &bus_class),
1570             "wrong class %s\n", debugstr_guid(&iface.InterfaceClassGuid));
1571     ok(iface.Flags == SPINT_ACTIVE, "got flags %#lx\n", iface.Flags);
1572     SetupDiDestroyDeviceInfoList(set);
1573
1574     ret = DeviceIoControl(bus, IOCTL_WINETEST_BUS_DISABLE_IFACE, NULL, 0, NULL, 0, &size, NULL);
1575     ok(ret, "got error %lu\n", GetLastError());
1576
1577     pump_messages();
1578     ok(got_bus_arrival == 1, "got %u bus arrival messages\n", got_bus_arrival);
1579     ok(got_bus_removal == 1, "got %u bus removal messages\n", got_bus_removal);
1580
1581     set = SetupDiGetClassDevsA(&bus_class, NULL, NULL, DIGCF_DEVICEINTERFACE);
1582     ok(set != INVALID_HANDLE_VALUE, "failed to get device list, error %#lx\n", GetLastError());
1583     ret = SetupDiEnumDeviceInterfaces(set, NULL, &bus_class, 0, &iface);
1584     ok(ret, "failed to get interface, error %#lx\n", GetLastError());
1585     ok(IsEqualGUID(&iface.InterfaceClassGuid, &bus_class),
1586             "wrong class %s\n", debugstr_guid(&iface.InterfaceClassGuid));
1587     ok(!iface.Flags, "got flags %#lx\n", iface.Flags);
1588     SetupDiDestroyDeviceInfoList(set);
1589
1590     set = SetupDiGetClassDevsA(&bus_class, NULL, NULL, DIGCF_DEVICEINTERFACE | DIGCF_PRESENT);
1591     ok(set != INVALID_HANDLE_VALUE, "failed to get device list, error %#lx\n", GetLastError());
1592     ret = SetupDiEnumDeviceInterfaces(set, NULL, &bus_class, 0, &iface);
1593     ok(!ret, "expected failure\n");
1594     ok(GetLastError() == ERROR_NO_MORE_ITEMS, "got error %#lx\n", GetLastError());
1595     SetupDiDestroyDeviceInfoList(set);
1596
1597     /* Test exposing a child device. */
1598
1599     id = 1;
1600     ret = DeviceIoControl(bus, IOCTL_WINETEST_BUS_ADD_CHILD, &id, sizeof(id), NULL, 0, &size, NULL);
1601     ok(ret, "got error %lu\n", GetLastError());
1602
1603     pump_messages();
1604     ok(got_child_arrival == 1, "got %u child arrival messages\n", got_child_arrival);
1605     ok(!got_child_removal, "got %u child removal messages\n", got_child_removal);
1606
1607     set = SetupDiGetClassDevsA(&child_class, NULL, NULL, DIGCF_DEVICEINTERFACE | DIGCF_PRESENT);
1608     ok(set != INVALID_HANDLE_VALUE, "failed to get device list, error %#lx\n", GetLastError());
1609
1610     ret = SetupDiEnumDeviceInfo(set, 0, &device);
1611     ok(ret, "failed to get device, error %#lx\n", GetLastError());
1612     ok(IsEqualGUID(&device.ClassGuid, &GUID_NULL), "wrong class %s\n", debugstr_guid(&device.ClassGuid));
1613
1614     ret = SetupDiGetDeviceInstanceIdA(set, &device, buffer, sizeof(buffer), NULL);
1615     ok(ret, "failed to get device ID, error %#lx\n", GetLastError());
1616     ok(!strcmp(buffer, "WINE\\TEST\\1"), "got ID %s\n", debugstr_a(buffer));
1617
1618     ret = SetupDiGetDeviceRegistryPropertyA(set, &device, SPDRP_CAPABILITIES,
1619             &type, (BYTE *)&dword, sizeof(dword), NULL);
1620     todo_wine ok(ret, "got error %#lx\n", GetLastError());
1621     if (ret)
1622     {
1623         ok(dword == (CM_DEVCAP_EJECTSUPPORTED | CM_DEVCAP_UNIQUEID
1624                 | CM_DEVCAP_RAWDEVICEOK | CM_DEVCAP_SURPRISEREMOVALOK), "got flags %#lx\n", dword);
1625         ok(type == REG_DWORD, "got type %lu\n", type);
1626     }
1627
1628     ret = SetupDiGetDeviceRegistryPropertyA(set, &device, SPDRP_CLASSGUID,
1629             &type, (BYTE *)buffer, sizeof(buffer), NULL);
1630     todo_wine ok(!ret, "expected failure\n");
1631     if (ret)
1632         ok(GetLastError() == ERROR_INVALID_DATA, "got error %#lx\n", GetLastError());
1633
1634     ret = SetupDiGetDeviceRegistryPropertyA(set, &device, SPDRP_CONFIGFLAGS,
1635             &type, (BYTE *)&dword, sizeof(dword), NULL);
1636     ok(ret, "got error %#lx\n", GetLastError());
1637     /* windows 7 sets CONFIGFLAG_FINISH_INSTALL; it's not clear what this means */
1638     ok(!(dword & ~CONFIGFLAG_FINISH_INSTALL), "got flags %#lx\n", dword);
1639     ok(type == REG_DWORD, "got type %lu\n", type);
1640
1641     ret = SetupDiGetDeviceRegistryPropertyA(set, &device, SPDRP_DEVTYPE,
1642             &type, (BYTE *)&dword, sizeof(dword), NULL);
1643     ok(!ret, "expected failure\n");
1644     ok(GetLastError() == ERROR_INVALID_DATA, "got error %#lx\n", GetLastError());
1645
1646     ret = SetupDiGetDeviceRegistryPropertyA(set, &device, SPDRP_DRIVER,
1647             &type, (BYTE *)buffer, sizeof(buffer), NULL);
1648     ok(!ret, "expected failure\n");
1649     ok(GetLastError() == ERROR_INVALID_DATA, "got error %#lx\n", GetLastError());
1650
1651     ret = SetupDiGetDeviceRegistryPropertyA(set, &device, SPDRP_HARDWAREID,
1652             &type, (BYTE *)buffer, sizeof(buffer), &size);
1653     ok(ret, "got error %#lx\n", GetLastError());
1654     ok(type == REG_MULTI_SZ, "got type %lu\n", type);
1655     ok(size == sizeof(expect_hardware_id), "got size %lu\n", size);
1656     ok(!memcmp(buffer, expect_hardware_id, size), "got hardware IDs %s\n", debugstr_an(buffer, size));
1657
1658     /* Using the WCHAR variant because Windows returns a WCHAR for this property even when using SetupDiGetDeviceRegistryPropertyA */
1659     ret = SetupDiGetDeviceRegistryPropertyW(set, &device, SPDRP_BASE_CONTAINERID,
1660             &type, (BYTE *)buffer_w, sizeof(buffer_w), &size);
1661     ok(ret, "got error %#lx\n", GetLastError());
1662     ok(type == REG_SZ, "got type %lu\n", type);
1663     ok(size == sizeof(expect_container_id_w), "got size %lu\n", size);
1664     ok(!memcmp(buffer_w, expect_container_id_w, size), "got container ID %s\n", debugstr_w(buffer_w));
1665
1666     ret = SetupDiGetDeviceRegistryPropertyA(set, &device, SPDRP_COMPATIBLEIDS,
1667             &type, (BYTE *)buffer, sizeof(buffer), &size);
1668     ok(ret, "got error %#lx\n", GetLastError());
1669     ok(type == REG_MULTI_SZ, "got type %lu\n", type);
1670     ok(size == sizeof(expect_compat_id), "got size %lu\n", size);
1671     ok(!memcmp(buffer, expect_compat_id, size), "got compatible IDs %s\n", debugstr_an(buffer, size));
1672
1673     ret = SetupDiGetDeviceRegistryPropertyA(set, &device, SPDRP_PHYSICAL_DEVICE_OBJECT_NAME,
1674             &type, (BYTE *)buffer, sizeof(buffer), NULL);
1675     todo_wine ok(ret, "got error %#lx\n", GetLastError());
1676     if (ret)
1677     {
1678         ok(type == REG_SZ, "got type %lu\n", type);
1679         ok(!strcmp(buffer, "\\Device\\winetest_pnp_1"), "got PDO name %s\n", debugstr_a(buffer));
1680     }
1681
1682     ret = SetupDiEnumDeviceInterfaces(set, NULL, &child_class, 0, &iface);
1683     ok(ret, "failed to get interface, error %#lx\n", GetLastError());
1684     ok(IsEqualGUID(&iface.InterfaceClassGuid, &child_class),
1685             "wrong class %s\n", debugstr_guid(&iface.InterfaceClassGuid));
1686     ok(iface.Flags == SPINT_ACTIVE, "got flags %#lx\n", iface.Flags);
1687
1688     iface_detail->cbSize = sizeof(*iface_detail);
1689     ret = SetupDiGetDeviceInterfaceDetailA(set, &iface, iface_detail, sizeof(buffer), NULL, NULL);
1690     ok(ret, "failed to get interface path, error %#lx\n", GetLastError());
1691     ok(!strcmp(iface_detail->DevicePath, "\\\\?\\wine#test#1#{deadbeef-29ef-4538-a5fd-b69573a362c2}"),
1692             "wrong path %s\n", debugstr_a(iface_detail->DevicePath));
1693
1694     SetupDiDestroyDeviceInfoList(set);
1695
1696     RtlInitUnicodeString(&string, L"\\Device\\winetest_pnp_1");
1697     InitializeObjectAttributes(&attr, &string, OBJ_CASE_INSENSITIVE, NULL, NULL);
1698     ret = NtOpenFile(&child, SYNCHRONIZE, &attr, &io, 0, FILE_SYNCHRONOUS_IO_NONALERT);
1699     ok(!ret, "failed to open child: %#x\n", ret);
1700
1701     id = 0xdeadbeef;
1702     ret = DeviceIoControl(child, IOCTL_WINETEST_CHILD_GET_ID, NULL, 0, &id, sizeof(id), &size, NULL);
1703     ok(ret, "got error %lu\n", GetLastError());
1704     ok(id == 1, "got id %d\n", id);
1705     ok(size == sizeof(id), "got size %lu\n", size);
1706
1707     CloseHandle(child);
1708
1709     ret = NtOpenFile(&child, SYNCHRONIZE, &attr, &io, 0, 0);
1710     ok(!ret, "failed to open child: %#x\n", ret);
1711
1712     ret = DeviceIoControl(child, IOCTL_WINETEST_CHILD_MARK_PENDING, NULL, 0, NULL, 0, &size, &ovl);
1713     ok(!ret, "DeviceIoControl succeeded\n");
1714     ok(GetLastError() == ERROR_IO_PENDING, "got error %lu\n", GetLastError());
1715     ok(size == 0, "got size %lu\n", size);
1716
1717     id = 1;
1718     ret = DeviceIoControl(bus, IOCTL_WINETEST_BUS_REMOVE_CHILD, &id, sizeof(id), NULL, 0, &size, NULL);
1719     ok(ret, "got error %lu\n", GetLastError());
1720
1721     pump_messages();
1722     ok(got_child_arrival == 1, "got %u child arrival messages\n", got_child_arrival);
1723     ok(got_child_removal == 1, "got %u child removal messages\n", got_child_removal);
1724
1725     ret = DeviceIoControl(child, IOCTL_WINETEST_CHILD_CHECK_REMOVED, NULL, 0, NULL, 0, &size, NULL);
1726     todo_wine ok(ret, "got error %lu\n", GetLastError());
1727
1728     ret = NtOpenFile(&tmp, SYNCHRONIZE, &attr, &io, 0, FILE_SYNCHRONOUS_IO_NONALERT);
1729     todo_wine ok(ret == STATUS_NO_SUCH_DEVICE, "got %#x\n", ret);
1730
1731     ret = GetOverlappedResult(child, &ovl, &size, TRUE);
1732     ok(!ret, "unexpected success.\n");
1733     ok(GetLastError() == ERROR_ACCESS_DENIED, "got error %lu\n", GetLastError());
1734     ok(size == 0, "got size %lu\n", size);
1735
1736     CloseHandle(child);
1737
1738     pump_messages();
1739     ok(got_child_arrival == 1, "got %u child arrival messages\n", got_child_arrival);
1740     ok(got_child_removal == 1, "got %u child removal messages\n", got_child_removal);
1741
1742     ret = NtOpenFile(&tmp, SYNCHRONIZE, &attr, &io, 0, FILE_SYNCHRONOUS_IO_NONALERT);
1743     ok(ret == STATUS_OBJECT_NAME_NOT_FOUND, "got %#x\n", ret);
1744
1745     CloseHandle(bus);
1746
1747     UnregisterDeviceNotification(notify_handle);
1748     DestroyWindow(window);
1749     UnregisterClassA("ntoskrnl_test_wc", GetModuleHandleA(NULL));
1750 }
1751
1752 static void test_pnp_driver(struct testsign_context *ctx)
1753 {
1754     static const char hardware_id[] = "test_hardware_id\0";
1755     char path[MAX_PATH], dest[MAX_PATH], *filepart;
1756     SP_DEVINFO_DATA device = {sizeof(device)};
1757     char cwd[MAX_PATH], tempdir[MAX_PATH];
1758     WCHAR driver_filename[MAX_PATH];
1759     SC_HANDLE manager, service;
1760     BOOL ret, need_reboot;
1761     HANDLE catalog, file;
1762     DWORD dword, type;
1763     unsigned int i;
1764     HDEVINFO set;
1765     FILE *f;
1766
1767     GetCurrentDirectoryA(ARRAY_SIZE(cwd), cwd);
1768     GetTempPathA(ARRAY_SIZE(tempdir), tempdir);
1769     SetCurrentDirectoryA(tempdir);
1770
1771     load_resource(L"driver_pnp.dll", driver_filename);
1772     ret = MoveFileExW(driver_filename, L"winetest.sys", MOVEFILE_COPY_ALLOWED | MOVEFILE_REPLACE_EXISTING);
1773     ok(ret, "failed to move file, error %lu\n", GetLastError());
1774
1775     f = fopen("winetest.inf", "w");
1776     ok(!!f, "failed to open winetest.inf: %s\n", strerror(errno));
1777     fputs(inf_text, f);
1778     fclose(f);
1779
1780     /* Create the catalog file. */
1781
1782     catalog = CryptCATOpen((WCHAR *)L"winetest.cat", CRYPTCAT_OPEN_CREATENEW, 0, CRYPTCAT_VERSION_1, 0);
1783     ok(catalog != INVALID_HANDLE_VALUE, "Failed to create catalog, error %#lx\n", GetLastError());
1784
1785     ret = !!CryptCATPutCatAttrInfo(catalog, (WCHAR *)L"HWID1",
1786             CRYPTCAT_ATTR_NAMEASCII | CRYPTCAT_ATTR_DATAASCII | CRYPTCAT_ATTR_AUTHENTICATED,
1787             sizeof(L"test_hardware_id"), (BYTE *)L"test_hardware_id");
1788     todo_wine ok(ret, "failed to add attribute, error %#lx\n", GetLastError());
1789
1790     ret = !!CryptCATPutCatAttrInfo(catalog, (WCHAR *)L"OS",
1791             CRYPTCAT_ATTR_NAMEASCII | CRYPTCAT_ATTR_DATAASCII | CRYPTCAT_ATTR_AUTHENTICATED,
1792             sizeof(L"VistaX64"), (BYTE *)L"VistaX64");
1793     todo_wine ok(ret, "failed to add attribute, error %#lx\n", GetLastError());
1794
1795     add_file_to_catalog(catalog, L"winetest.sys");
1796     add_file_to_catalog(catalog, L"winetest.inf");
1797
1798     ret = CryptCATPersistStore(catalog);
1799     todo_wine ok(ret, "Failed to write catalog, error %lu\n", GetLastError());
1800
1801     ret = CryptCATClose(catalog);
1802     ok(ret, "Failed to close catalog, error %lu\n", GetLastError());
1803
1804     testsign_sign(ctx, L"winetest.cat");
1805
1806     /* Install the driver. */
1807
1808     set = SetupDiCreateDeviceInfoList(NULL, NULL);
1809     ok(set != INVALID_HANDLE_VALUE, "failed to create device list, error %#lx\n", GetLastError());
1810
1811     ret = SetupDiCreateDeviceInfoA(set, "root\\winetest\\0", &GUID_NULL, NULL, NULL, 0, &device);
1812     ok(ret, "failed to create device, error %#lx\n", GetLastError());
1813
1814     ret = SetupDiSetDeviceRegistryPropertyA( set, &device, SPDRP_HARDWAREID,
1815             (const BYTE *)hardware_id, sizeof(hardware_id) );
1816     ok(ret, "failed to create set hardware ID, error %#lx\n", GetLastError());
1817
1818     ret = SetupDiCallClassInstaller(DIF_REGISTERDEVICE, set, &device);
1819     ok(ret, "failed to register device, error %#lx\n", GetLastError());
1820
1821     ret = SetupDiGetDeviceRegistryPropertyA(set, &device, SPDRP_CONFIGFLAGS,
1822             &type, (BYTE *)&dword, sizeof(dword), NULL);
1823     ok(!ret, "expected failure\n");
1824     ok(GetLastError() == ERROR_INVALID_DATA, "got error %#lx\n", GetLastError());
1825
1826     GetFullPathNameA("winetest.inf", sizeof(path), path, NULL);
1827     ret = UpdateDriverForPlugAndPlayDevicesA(NULL, hardware_id, path, INSTALLFLAG_FORCE, &need_reboot);
1828     ok(ret, "failed to install device, error %#lx\n", GetLastError());
1829     ok(!need_reboot, "expected no reboot necessary\n");
1830
1831     ret = SetupDiGetDeviceRegistryPropertyA(set, &device, SPDRP_CONFIGFLAGS,
1832             &type, (BYTE *)&dword, sizeof(dword), NULL);
1833     ok(ret, "got error %#lx\n", GetLastError());
1834     ok(!dword, "got flags %#lx\n", dword);
1835     ok(type == REG_DWORD, "got type %lu\n", type);
1836
1837     /* Tests. */
1838
1839     test_pnp_devices();
1840
1841     /* Clean up. */
1842
1843     ret = SetupDiCallClassInstaller(DIF_REMOVE, set, &device);
1844     ok(ret, "failed to remove device, error %#lx\n", GetLastError());
1845
1846     file = CreateFileA("\\\\?\\root#winetest#0#{deadbeef-29ef-4538-a5fd-b69573a362c0}", 0, 0, NULL, OPEN_EXISTING, 0, NULL);
1847     ok(file == INVALID_HANDLE_VALUE, "expected failure\n");
1848     ok(GetLastError() == ERROR_FILE_NOT_FOUND, "got error %lu\n", GetLastError());
1849
1850     ret = SetupDiDestroyDeviceInfoList(set);
1851     ok(ret, "failed to destroy set, error %#lx\n", GetLastError());
1852
1853     set = SetupDiGetClassDevsA(NULL, "wine", NULL, DIGCF_ALLCLASSES);
1854     ok(set != INVALID_HANDLE_VALUE, "failed to get device list, error %#lx\n", GetLastError());
1855
1856     for (i = 0; SetupDiEnumDeviceInfo(set, i, &device); ++i)
1857     {
1858         ret = SetupDiCallClassInstaller(DIF_REMOVE, set, &device);
1859         ok(ret, "failed to remove device, error %#lx\n", GetLastError());
1860     }
1861
1862     SetupDiDestroyDeviceInfoList(set);
1863
1864     /* Windows stops the service but does not delete it. */
1865     manager = OpenSCManagerA(NULL, NULL, SC_MANAGER_CONNECT);
1866     ok(!!manager, "failed to open service manager, error %lu\n", GetLastError());
1867     service = OpenServiceA(manager, "winetest", SERVICE_STOP | DELETE);
1868     ok(!!service, "failed to open service, error %lu\n", GetLastError());
1869     unload_driver(service);
1870     CloseServiceHandle(manager);
1871
1872     cat_okfile();
1873
1874     GetFullPathNameA("winetest.inf", sizeof(path), path, NULL);
1875     ret = SetupCopyOEMInfA(path, NULL, 0, 0, dest, sizeof(dest), NULL, &filepart);
1876     ok(ret, "Failed to copy INF, error %#lx\n", GetLastError());
1877     ret = SetupUninstallOEMInfA(filepart, SUOI_FORCEDELETE, NULL);
1878     ok(ret, "Failed to uninstall INF, error %lu\n", GetLastError());
1879
1880     ret = DeleteFileA("winetest.cat");
1881     ok(ret, "Failed to delete file, error %lu\n", GetLastError());
1882     ret = DeleteFileA("winetest.inf");
1883     ok(ret, "Failed to delete file, error %lu\n", GetLastError());
1884     ret = DeleteFileA("winetest.sys");
1885     ok(ret, "Failed to delete file, error %lu\n", GetLastError());
1886     /* Windows 10 apparently deletes the image in SetupUninstallOEMInf(). */
1887     ret = DeleteFileA("C:/windows/system32/drivers/winetest.sys");
1888     ok(ret || GetLastError() == ERROR_FILE_NOT_FOUND, "Failed to delete file, error %lu\n", GetLastError());
1889
1890     SetCurrentDirectoryA(cwd);
1891 }
1892
1893 START_TEST(ntoskrnl)
1894 {
1895     WCHAR filename[MAX_PATH], filename2[MAX_PATH];
1896     struct testsign_context ctx;
1897     SC_HANDLE service, service2;
1898     BOOL ret, is_wow64;
1899     HANDLE mapping;
1900     DWORD written;
1901
1902     pRtlDosPathNameToNtPathName_U = (void *)GetProcAddress(GetModuleHandleA("ntdll"), "RtlDosPathNameToNtPathName_U");
1903     pRtlFreeUnicodeString = (void *)GetProcAddress(GetModuleHandleA("ntdll"), "RtlFreeUnicodeString");
1904     pCancelIoEx = (void *)GetProcAddress(GetModuleHandleA("kernel32.dll"), "CancelIoEx");
1905     pIsWow64Process = (void *)GetProcAddress(GetModuleHandleA("kernel32.dll"), "IsWow64Process");
1906     pSetFileCompletionNotificationModes = (void *)GetProcAddress(GetModuleHandleA("kernel32.dll"),
1907                                                                  "SetFileCompletionNotificationModes");
1908     pSignerSign = (void *)GetProcAddress(LoadLibraryA("mssign32"), "SignerSign");
1909
1910     if (IsWow64Process(GetCurrentProcess(), &is_wow64) && is_wow64)
1911     {
1912         skip("Running in WoW64.\n");
1913         return;
1914     }
1915
1916     if (!testsign_create_cert(&ctx))
1917         return;
1918
1919     mapping = CreateFileMappingA(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE,
1920             0, sizeof(*test_data), "Global\\winetest_ntoskrnl_section");
1921     ok(!!mapping, "got error %lu\n", GetLastError());
1922     test_data = MapViewOfFile(mapping, FILE_MAP_READ | FILE_MAP_WRITE, 0, 0, 1024);
1923     test_data->running_under_wine = !strcmp(winetest_platform, "wine");
1924     test_data->winetest_report_success = winetest_report_success;
1925     test_data->winetest_debug = winetest_debug;
1926
1927     okfile = CreateFileA("C:\\windows\\winetest_ntoskrnl_okfile", GENERIC_READ | GENERIC_WRITE,
1928             FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, 0, NULL);
1929     ok(okfile != INVALID_HANDLE_VALUE, "failed to create file, error %lu\n", GetLastError());
1930
1931     subtest("driver");
1932     if (!(service = load_driver(&ctx, filename, L"driver.dll", L"WineTestDriver")))
1933         goto out;
1934
1935     if (!start_driver(service, FALSE))
1936     {
1937         DeleteFileW(filename);
1938         goto out;
1939     }
1940     service2 = load_driver(&ctx, filename2, L"driver2.dll", L"WineTestDriver2");
1941
1942     device = CreateFileA("\\\\.\\WineTestDriver", 0, 0, NULL, OPEN_EXISTING, 0, NULL);
1943     ok(device != INVALID_HANDLE_VALUE, "failed to open device: %lu\n", GetLastError());
1944
1945     test_basic_ioctl();
1946
1947     main_test();
1948     todo_wine ok(modified_value == 0xdeadbeeffeedcafe, "Got unexpected value %#I64x.\n", modified_value);
1949
1950     test_overlapped();
1951     test_load_driver(service2);
1952     test_file_handles();
1953     test_return_status();
1954     test_object_info();
1955     test_blocking_irp();
1956
1957     /* We need a separate ioctl to call IoDetachDevice(); calling it in the
1958      * driver unload routine causes a live-lock. */
1959     ret = DeviceIoControl(device, IOCTL_WINETEST_DETACH, NULL, 0, NULL, 0, &written, NULL);
1960     ok(ret, "DeviceIoControl failed: %lu\n", GetLastError());
1961
1962     CloseHandle(device);
1963
1964     unload_driver(service2);
1965     unload_driver(service);
1966     ret = DeleteFileW(filename);
1967     ok(ret, "DeleteFile failed: %lu\n", GetLastError());
1968     ret = DeleteFileW(filename2);
1969     ok(ret, "DeleteFile failed: %lu\n", GetLastError());
1970
1971     cat_okfile();
1972
1973     test_driver3(&ctx);
1974     subtest("driver_netio");
1975     test_driver_netio(&ctx);
1976
1977     subtest("driver_pnp");
1978     test_pnp_driver(&ctx);
1979
1980 out:
1981     testsign_cleanup(&ctx);
1982     UnmapViewOfFile(test_data);
1983     CloseHandle(mapping);
1984     CloseHandle(okfile);
1985     DeleteFileA("C:\\windows\\winetest_ntoskrnl_okfile");
1986 }