dlls/faultrep/tests/faultrep.c

   1 /*
   2  * Unit test suite for fault reporting in XP and above
   3  *
   4  * Copyright 2010 Detlef Riekenberg
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  *
  20  */
  21
  22 #include <stdarg.h>
  23 #include <stdio.h>
  24
  25 #include "windef.h"
  26 #include "winbase.h"
  27 #include "winerror.h"
  28 #include "winreg.h"
  29
  30 #include "errorrep.h"
  31 #include "wine/test.h"
  32
  33 static const char regpath_root[] = "Software\\Microsoft\\PCHealth\\ErrorReporting";
  34 static const char regpath_exclude[] = "ExclusionList";
  35
  36
  37 static BOOL is_process_limited(void)
  38 {
  39     static BOOL (WINAPI *pCheckTokenMembership)(HANDLE,PSID,PBOOL) = NULL;
  40     static BOOL (WINAPI *pOpenProcessToken)(HANDLE, DWORD, PHANDLE) = NULL;
  41     SID_IDENTIFIER_AUTHORITY NtAuthority = {SECURITY_NT_AUTHORITY};
  42     PSID Group;
  43     BOOL IsInGroup;
  44     HANDLE token;
  45
  46     if (!pOpenProcessToken)
  47     {
  48         HMODULE hadvapi32 = GetModuleHandleA("advapi32.dll");
  49         pOpenProcessToken = (void*)GetProcAddress(hadvapi32, "OpenProcessToken");
  50         pCheckTokenMembership = (void*)GetProcAddress(hadvapi32, "CheckTokenMembership");
  51         if (!pCheckTokenMembership || !pOpenProcessToken)
  52         {
  53             /* Win9x (power to the masses) or NT4 (no way to know) */
  54             trace("missing pOpenProcessToken or CheckTokenMembership\n");
  55             return FALSE;
  56         }
  57     }
  58
  59     if (!AllocateAndInitializeSid(&NtAuthority, 2, SECURITY_BUILTIN_DOMAIN_RID,
  60                                   DOMAIN_ALIAS_RID_ADMINS,
  61                                   0, 0, 0, 0, 0, 0, &Group) ||
  62         !pCheckTokenMembership(NULL, Group, &IsInGroup))
  63     {
  64         trace("Could not check if the current user is an administrator\n");
  65         return FALSE;
  66     }
  67     FreeSid(Group);
  68     if (!IsInGroup)
  69     {
  70         if (!AllocateAndInitializeSid(&NtAuthority, 2,
  71                                       SECURITY_BUILTIN_DOMAIN_RID,
  72                                       DOMAIN_ALIAS_RID_POWER_USERS,
  73                                       0, 0, 0, 0, 0, 0, &Group) ||
  74             !pCheckTokenMembership(NULL, Group, &IsInGroup))
  75         {
  76             trace("Could not check if the current user is a power user\n");
  77             return FALSE;
  78         }
  79         FreeSid(Group);
  80         if (!IsInGroup)
  81         {
  82             /* Only administrators and power users can be powerful */
  83             return TRUE;
  84         }
  85     }
  86
  87     if (pOpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &token))
  88     {
  89         BOOL ret;
  90         TOKEN_ELEVATION_TYPE type = TokenElevationTypeDefault;
  91         DWORD size;
  92
  93         ret = GetTokenInformation(token, TokenElevationType, &type, sizeof(type), &size);
  94         CloseHandle(token);
  95         return (ret && type == TokenElevationTypeLimited);
  96     }
  97     return FALSE;
  98 }
  99
 100
 101 /* ###### */
 102
 103 static void test_AddERExcludedApplicationA(void)
 104 {
 105     BOOL res;
 106     LONG lres;
 107     HKEY hroot;
 108     HKEY hexclude = 0;
 109
 110     /* clean state */
 111     lres = RegCreateKeyA(HKEY_LOCAL_MACHINE, regpath_root, &hroot);
 112     if (lres == ERROR_ACCESS_DENIED)
 113     {
 114         skip("Not enough access rights\n");
 115         return;
 116     }
 117
 118     if (!lres)
 119         lres = RegOpenKeyA(hroot, regpath_exclude, &hexclude);
 120
 121     if (!lres)
 122         RegDeleteValueA(hexclude, "winetest_faultrep.exe");
 123
 124
 125     SetLastError(0xdeadbeef);
 126     res = AddERExcludedApplicationA(NULL);
 127     ok(!res, "got %d and 0x%x (expected FALSE)\n", res, GetLastError());
 128
 129     SetLastError(0xdeadbeef);
 130     res = AddERExcludedApplicationA("");
 131     ok(!res, "got %d and 0x%x (expected FALSE)\n", res, GetLastError());
 132
 133     SetLastError(0xdeadbeef);
 134     /* existence of the path doesn't matter this function succeeded */
 135     res = AddERExcludedApplicationA("winetest_faultrep.exe");
 136     if (is_process_limited())
 137     {
 138         /* LastError is not set! */
 139         ok(!res, "AddERExcludedApplicationA should have failed got %d\n", res);
 140     }
 141     else
 142     {
 143         ok(res, "AddERExcludedApplicationA failed (le=0x%x)\n", GetLastError());
 144
 145         /* add, when already present */
 146         SetLastError(0xdeadbeef);
 147         res = AddERExcludedApplicationA("winetest_faultrep.exe");
 148         ok(res, "AddERExcludedApplicationA failed (le=0x%x)\n", GetLastError());
 149     }
 150
 151     /* cleanup */
 152     RegDeleteValueA(hexclude, "winetest_faultrep.exe");
 153
 154     RegCloseKey(hexclude);
 155     RegCloseKey(hroot);
 156 }
 157
 158 /* ########################### */
 159
 160 START_TEST(faultrep)
 161 {
 162     test_AddERExcludedApplicationA();
 163 }