dlls/wer/main.c

   1 /*
   2  * Copyright 2010 Louis Lenders
   3  * Copyright 2010 Detlef Riekenberg
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  18  */
  19
  20 #include "config.h"
  21
  22 #include <stdarg.h>
  23
  24 #include "windef.h"
  25 #include "winbase.h"
  26 #include "winreg.h"
  27 #include "werapi.h"
  28 #include "wine/list.h"
  29 #include "wine/unicode.h"
  30 #include "wine/debug.h"
  31
  32 WINE_DEFAULT_DEBUG_CHANNEL(wer);
  33
  34 typedef struct {
  35     struct list entry;
  36     WER_REPORT_INFORMATION info;
  37     WER_REPORT_TYPE reporttype;
  38     WCHAR eventtype[1];
  39 } report_t;
  40
  41
  42 static CRITICAL_SECTION report_table_cs;
  43 static CRITICAL_SECTION_DEBUG report_table_cs_debug =
  44 {
  45     0, 0, &report_table_cs,
  46     { &report_table_cs_debug.ProcessLocksList, &report_table_cs_debug.ProcessLocksList },
  47       0, 0, { (DWORD_PTR)(__FILE__ ": report_table_cs") }
  48 };
  49 static CRITICAL_SECTION report_table_cs = { &report_table_cs_debug, -1, 0, 0, 0, 0 };
  50
  51 static struct list report_table = LIST_INIT(report_table);
  52
  53 static WCHAR regpath_exclude[] = {'S','o','f','t','w','a','r','e','\\',
  54                                   'M','i','c','r','o','s','o','f','t','\\',
  55                                   'W','i','n','d','o','w','s',' ','E','r','r','o','r',' ','R','e','p','o','r','t','i','n','g','\\',
  56                                   'E','x','c','l','u','d','e','d','A','p','p','l','i','c','a','t','i','o','n','s',0};
  57
  58 /***********************************************************************
  59  * Memory alloccation helper
  60  */
  61
  62 static inline void * __WINE_ALLOC_SIZE(1) heap_alloc_zero(size_t len)
  63 {
  64     return HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len);
  65 }
  66
  67 static inline BOOL heap_free(void *mem)
  68 {
  69     return HeapFree(GetProcessHeap(), 0, mem);
  70 }
  71
  72 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
  73 {
  74     TRACE("(0x%p, %d, %p)\n", hinstDLL, fdwReason, lpvReserved);
  75
  76     switch (fdwReason)
  77     {
  78         case DLL_WINE_PREATTACH:
  79             return FALSE;    /* prefer native version */
  80         case DLL_PROCESS_ATTACH:
  81             DisableThreadLibraryCalls(hinstDLL);
  82             break;
  83         case DLL_PROCESS_DETACH:
  84             break;
  85     }
  86
  87     return TRUE;
  88 }
  89
  90 /***********************************************************************
  91  * WerAddExcludedApplication (wer.@)
  92  *
  93  * Add an application to the user specific or the system wide exclusion list
  94  *
  95  * PARAMS
  96  *  exeName  [i] The application name
  97  *  allUsers [i] for all users (TRUE) or for the current user (FALSE)
  98  *
  99  * RETURNS
 100  *  Success: S_OK
 101  *  Faulure: A HRESULT error code
 102  *
 103  */
 104 HRESULT WINAPI WerAddExcludedApplication(PCWSTR exeName, BOOL allUsers)
 105 {
 106     HKEY hkey;
 107     DWORD value = 1;
 108     LPWSTR bs;
 109
 110     TRACE("(%s, %d)\n",debugstr_w(exeName), allUsers);
 111     if (!exeName || !exeName[0])
 112         return E_INVALIDARG;
 113
 114     bs = strrchrW(exeName, '\\');
 115     if (bs) {
 116         bs++;   /* skip the backslash */
 117         if (!bs[0]) {
 118             return E_INVALIDARG;
 119         }
 120     } else
 121         bs = (LPWSTR) exeName;
 122
 123     if (!RegCreateKeyW(allUsers ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER, regpath_exclude, &hkey)) {
 124         RegSetValueExW(hkey, bs, 0, REG_DWORD, (LPBYTE)&value, sizeof(DWORD));
 125         RegCloseKey(hkey);
 126         return S_OK;
 127     }
 128     return E_ACCESSDENIED;
 129 }
 130
 131 /***********************************************************************
 132  * WerRemoveExcludedApplication (wer.@)
 133  *
 134  * remove an application from the exclusion list
 135  *
 136  * PARAMS
 137  *  exeName  [i] The application name
 138  *  allUsers [i] for all users (TRUE) or for the current user (FALSE)
 139  *
 140  * RETURNS
 141  *  Success: S_OK
 142  *  Faulure: A HRESULT error code
 143  *
 144  */
 145 HRESULT WINAPI WerRemoveExcludedApplication(PCWSTR exeName, BOOL allUsers)
 146 {
 147     HKEY hkey;
 148     LPWSTR bs;
 149     LONG lres;
 150
 151     TRACE("(%s, %d)\n",debugstr_w(exeName), allUsers);
 152     if (!exeName || !exeName[0])
 153         return E_INVALIDARG;
 154
 155     bs = strrchrW(exeName, '\\');
 156     if (bs) {
 157         bs++;   /* skip the backslash */
 158         if (!bs[0]) {
 159             return E_INVALIDARG;
 160         }
 161     } else
 162         bs = (LPWSTR) exeName;
 163
 164     if (!RegCreateKeyW(allUsers ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER, regpath_exclude, &hkey)) {
 165         lres = RegDeleteValueW(hkey, bs);
 166         RegCloseKey(hkey);
 167         return lres ? __HRESULT_FROM_WIN32(ERROR_ENVVAR_NOT_FOUND) : S_OK;
 168     }
 169     return E_ACCESSDENIED;
 170 }
 171
 172 /***********************************************************************
 173  * WerReportCloseHandle (wer.@)
 174  *
 175  * Close an error reporting handle and free associated resources
 176  *
 177  * PARAMS
 178  *  hreport [i] error reporting handle to close
 179  *
 180  * RETURNS
 181  *  Success: S_OK
 182  *  Failure: A HRESULT error code
 183  *
 184  */
 185 HRESULT WINAPI WerReportCloseHandle(HREPORT hreport)
 186 {
 187     report_t * report = (report_t *) hreport;
 188     report_t * cursor;
 189     BOOL found = FALSE;
 190
 191     TRACE("(%p)\n", hreport);
 192     EnterCriticalSection(&report_table_cs);
 193     if (report) {
 194         LIST_FOR_EACH_ENTRY(cursor, &report_table, report_t, entry)
 195         {
 196             if (cursor == report) {
 197                 found = TRUE;
 198                 list_remove(&report->entry);
 199                 break;
 200             }
 201         }
 202     }
 203     LeaveCriticalSection(&report_table_cs);
 204     if (!found)
 205         return E_INVALIDARG;
 206
 207     heap_free(report);
 208
 209     return S_OK;
 210 }
 211
 212 /***********************************************************************
 213  * WerReportCreate (wer.@)
 214  *
 215  * Create an error report in memory and return a related HANDLE
 216  *
 217  * PARAMS
 218  *  eventtype  [i] a name for the event type
 219  *  reporttype [i] what type of report should be created
 220  *  reportinfo [i] NULL or a ptr to a struct with some detailed information
 221  *  phandle    [o] ptr, where the resulting handle should be saved
 222  *
 223  * RETURNS
 224  *  Success: S_OK
 225  *  Failure: A HRESULT error code
 226  *
 227  * NOTES
 228  *  The event type must be registered at microsoft. Predefined types are
 229  *  "APPCRASH" as the default on Windows, "Crash32" and "Crash64"
 230  *
 231  */
 232 HRESULT WINAPI WerReportCreate(PCWSTR eventtype, WER_REPORT_TYPE reporttype, PWER_REPORT_INFORMATION reportinfo, HREPORT *phandle)
 233 {
 234     report_t *report;
 235     DWORD len;
 236
 237     TRACE("(%s, %d, %p, %p)\n", debugstr_w(eventtype), reporttype, reportinfo, phandle);
 238     if (reportinfo) {
 239         TRACE(".wzFriendlyEventName: %s\n", debugstr_w(reportinfo->wzFriendlyEventName));
 240         TRACE(".wzApplicationName: %s\n", debugstr_w(reportinfo->wzApplicationName));
 241     }
 242
 243     if (phandle)  *phandle = NULL;
 244     if (!eventtype || !eventtype[0] || !phandle) {
 245         return E_INVALIDARG;
 246     }
 247
 248     len = lstrlenW(eventtype) + 1;
 249
 250     report = heap_alloc_zero(len * sizeof(WCHAR) + sizeof(report_t));
 251     if (!report)
 252         return __HRESULT_FROM_WIN32(ERROR_OUTOFMEMORY);
 253
 254     lstrcpyW(report->eventtype, eventtype);
 255     report->reporttype = reporttype;
 256
 257     if (reportinfo) {
 258         report->info = *reportinfo;
 259     } else {
 260         FIXME("build report information from scratch for %p\n", report);
 261     }
 262
 263     EnterCriticalSection(&report_table_cs);
 264     list_add_head(&report_table, &report->entry);
 265     LeaveCriticalSection(&report_table_cs);
 266
 267     *phandle = report;
 268     TRACE("=> %p\n", report);
 269     return S_OK;
 270 }
 271
 272 /***********************************************************************
 273  * WerReportSetParameter (wer.@)
 274  *
 275  * Set one of 10 parameter / value pairs for a report handle
 276  *
 277  * PARAMS
 278  *  hreport [i] error reporting handle to add the parameter
 279  *  id      [i] parameter to set (WER_P0 up to WER_P9)
 280  *  name    [i] optional name of the parameter
 281  *  value   [i] value of the parameter
 282  *
 283  * RETURNS
 284  *  Success: S_OK
 285  *  Failure: A HRESULT error code
 286  *
 287  */
 288 HRESULT WINAPI WerReportSetParameter(HREPORT hreport, DWORD id, PCWSTR name, PCWSTR value)
 289 {
 290     FIXME("(%p, %d, %s, %s) :stub\n", hreport, id, debugstr_w(name), debugstr_w(value));
 291
 292     return E_NOTIMPL;
 293 }
 294
 295 /***********************************************************************
 296  * WerReportSubmit (wer.@)
 297  *
 298  * Ask the user for permission and send the error report
 299  * then kill or restart the application, when requested
 300  *
 301  * PARAMS
 302  *  hreport [i] error reporting handle to send
 303  *  consent [i] current transmit permission
 304  *  flags   [i] flag to select dialog, transmission snd restart options
 305  *  presult [o] ptr, where the transmission result should be saved
 306  *
 307  * RETURNS
 308  *  Success: S_OK
 309  *  Failure: A HRESULT error code
 310  *
 311  */
 312 HRESULT WINAPI WerReportSubmit(HREPORT hreport, WER_CONSENT consent, DWORD flags, PWER_SUBMIT_RESULT presult)
 313 {
 314     FIXME("(%p, %d, 0x%x, %p) :stub\n", hreport, consent, flags, presult);
 315
 316     if(!presult)
 317         return E_INVALIDARG;
 318
 319     *presult = WerDisabled;
 320     return E_NOTIMPL;
 321 }