programs/uninstaller/main.c

   1 /*
   2  * Uninstaller
   3  *
   4  * Copyright 2000 Andreas Mohr
   5  * Copyright 2004 Hannu Valtonen
   6  * Copyright 2005 Jonathan Ernst
   7  *
   8  * This library is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2.1 of the License, or (at your option) any later version.
  12  *
  13  * This library is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public
  19  * License along with this library; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  21  *
  22  */
  23
  24 #include <locale.h>
  25 #include <string.h>
  26 #include <windows.h>
  27 #include <commctrl.h>
  28 #include <shlwapi.h>
  29 #include "resource.h"
  30 #include "regstr.h"
  31 #include "wine/debug.h"
  32
  33 WINE_DEFAULT_DEBUG_CHANNEL(uninstaller);
  34
  35 extern void WINAPI Control_RunDLL(HWND hWnd, HINSTANCE hInst, LPCSTR cmd, DWORD nCmdShow);
  36
  37 typedef struct {
  38     HKEY  root;
  39     WCHAR *key;
  40     WCHAR *descr;
  41     WCHAR *command;
  42     int active;
  43 } uninst_entry;
  44 static uninst_entry *entries = NULL;
  45 static unsigned int numentries = 0;
  46 static int oldsel = -1;
  47 static WCHAR *sFilter;
  48 static BOOL silent;
  49
  50 static int FetchUninstallInformation(void);
  51 static void UninstallProgram(void);
  52 static const WCHAR PathUninstallW[] = L"Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall";
  53
  54
  55 static void WINAPIV output_message(BOOL with_usage, unsigned int id, ...)
  56 {
  57     WCHAR fmt[2048];
  58     LCID current_lcid;
  59
  60     current_lcid = GetThreadLocale();
  61     if (silent) /* force en-US not to have localized strings */
  62         SetThreadLocale(MAKELCID(MAKELANGID(LANG_ENGLISH, SUBLANG_ENGLISH_US), SORT_DEFAULT));
  63
  64     if (LoadStringW(GetModuleHandleW(NULL), id, fmt, ARRAY_SIZE(fmt)) &&
  65         (!with_usage || LoadStringW(GetModuleHandleW(NULL), STRING_USAGE, fmt + wcslen(fmt), ARRAY_SIZE(fmt) - wcslen(fmt))))
  66     {
  67         va_list va_args;
  68         WCHAR *str;
  69         DWORD len;
  70
  71         va_start(va_args, id);
  72         len = FormatMessageW(FORMAT_MESSAGE_FROM_STRING|FORMAT_MESSAGE_ALLOCATE_BUFFER,
  73                              fmt, 0, 0, (LPWSTR)&str, 0, &va_args);
  74         if (len > 0 || GetLastError() == ERROR_NO_WORK_DONE)
  75         {
  76             if (silent)
  77                 MESSAGE("%ls", str);
  78             else
  79                 MessageBoxW(NULL, str, MAKEINTRESOURCEW(IDS_APPNAME), MB_OK | MB_ICONSTOP);
  80             LocalFree(str);
  81         }
  82         else
  83             WINE_FIXME("Could not format string: le=%lu, fmt=%s\n", GetLastError(), wine_dbgstr_w(fmt));
  84         va_end(va_args);
  85     }
  86     else
  87         WINE_FIXME("LoadString failed with %ld\n", GetLastError());
  88
  89     SetThreadLocale(current_lcid);
  90 }
  91
  92 /**
  93  * Used to output program list when used with --list
  94  */
  95 static void ListUninstallPrograms(void)
  96 {
  97     unsigned int i;
  98
  99     FetchUninstallInformation();
 100
 101     setlocale(LC_ALL, "en-US");
 102     for (i=0; i < numentries; i++)
 103         MESSAGE("%ls|||%ls\n", entries[i].key, entries[i].descr);
 104 }
 105
 106
 107 static void RemoveSpecificProgram(WCHAR *nameW)
 108 {
 109     unsigned int i;
 110
 111     FetchUninstallInformation();
 112
 113     for (i=0; i < numentries; i++)
 114     {
 115         if (CompareStringW(GetThreadLocale(), NORM_IGNORECASE, entries[i].key, -1, nameW, -1) == CSTR_EQUAL)
 116         {
 117             entries[i].active++;
 118             break;
 119         }
 120     }
 121
 122     if (i < numentries)
 123         UninstallProgram();
 124     else
 125         output_message(FALSE, STRING_NO_APP_MATCH, nameW);
 126 }
 127
 128
 129 int __cdecl wmain(int argc, WCHAR *argv[])
 130 {
 131     LPCWSTR token = NULL;
 132     int i = 1;
 133     BOOL is_wow64;
 134
 135     if (IsWow64Process( GetCurrentProcess(), &is_wow64 ) && is_wow64)
 136     {
 137         STARTUPINFOW si;
 138         PROCESS_INFORMATION pi;
 139         WCHAR filename[MAX_PATH];
 140         void *redir;
 141         DWORD exit_code;
 142
 143         memset( &si, 0, sizeof(si) );
 144         si.cb = sizeof(si);
 145         GetSystemDirectoryW( filename, MAX_PATH );
 146         wcscat( filename, L"\\uninstaller.exe" );
 147
 148         Wow64DisableWow64FsRedirection( &redir );
 149         if (CreateProcessW( filename, GetCommandLineW(), NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi ))
 150         {
 151             WINE_TRACE( "restarting %s\n", wine_dbgstr_w(filename) );
 152             WaitForSingleObject( pi.hProcess, INFINITE );
 153             GetExitCodeProcess( pi.hProcess, &exit_code );
 154             ExitProcess( exit_code );
 155         }
 156         else WINE_ERR( "failed to restart 64-bit %s, err %ld\n", wine_dbgstr_w(filename), GetLastError() );
 157         Wow64RevertWow64FsRedirection( redir );
 158     }
 159
 160     InitCommonControls();
 161
 162     while( i<argc )
 163     {
 164         token = argv[i++];
 165
 166         if( !lstrcmpW( token, L"--help" ) )
 167         {
 168             output_message(TRUE, STRING_HEADER);
 169             return 0;
 170         }
 171         else if( !lstrcmpW( token, L"--silent" ) )
 172         {
 173             silent = TRUE;
 174         }
 175         else if( !lstrcmpW( token, L"--list" ) )
 176         {
 177             ListUninstallPrograms();
 178             return 0;
 179         }
 180         else if( !lstrcmpW( token, L"--remove" ) )
 181         {
 182             if( i >= argc )
 183             {
 184                 output_message(FALSE, STRING_PARAMETER_REQUIRED);
 185                 return 1;
 186             }
 187
 188             RemoveSpecificProgram( argv[i++] );
 189             return 0;
 190         }
 191         else
 192         {
 193             output_message(FALSE, STRING_INVALID_OPTION, token);
 194             return 1;
 195         }
 196     }
 197
 198     /* Start the GUI control panel */
 199     Control_RunDLL(GetDesktopWindow(), 0, "appwiz.cpl", SW_SHOW);
 200     return 1;
 201 }
 202
 203
 204 /**
 205  * Used to sort entries by name.
 206  */
 207 static int __cdecl cmp_by_name(const void *a, const void *b)
 208 {
 209     return lstrcmpiW(((const uninst_entry *)a)->descr, ((const uninst_entry *)b)->descr);
 210 }
 211
 212
 213 /**
 214  * Fetch information from the uninstall key.
 215  */
 216 static int FetchFromRootKey(HKEY root)
 217 {
 218     HKEY hkeyApp;
 219     int i;
 220     DWORD sizeOfSubKeyName, displen, uninstlen, value, type, size;
 221     WCHAR subKeyName[256];
 222
 223     sizeOfSubKeyName = 255;
 224     for (i=0; RegEnumKeyExW( root, i, subKeyName, &sizeOfSubKeyName, NULL, NULL, NULL, NULL ) != ERROR_NO_MORE_ITEMS; ++i)
 225     {
 226         RegOpenKeyExW(root, subKeyName, 0, KEY_READ, &hkeyApp);
 227         size = sizeof(value);
 228         if (!RegQueryValueExW(hkeyApp, L"SystemComponent", NULL, &type, (BYTE *)&value, &size) &&
 229             type == REG_DWORD && value == 1)
 230         {
 231             RegCloseKey(hkeyApp);
 232             sizeOfSubKeyName = 255;
 233             continue;
 234         }
 235         if (!RegQueryValueExW(hkeyApp, L"DisplayName", NULL, NULL, NULL, &displen))
 236         {
 237             WCHAR *command;
 238
 239             size = sizeof(value);
 240             if (!RegQueryValueExW(hkeyApp, L"WindowsInstaller", NULL, &type, (BYTE *)&value, &size) &&
 241                 type == REG_DWORD && value == 1)
 242             {
 243                 command = malloc(sizeof(L"msiexec /x") + wcslen(subKeyName) * sizeof(WCHAR));
 244                 wsprintfW(command, L"msiexec /x%s", subKeyName);
 245             }
 246             else if (!RegQueryValueExW(hkeyApp, L"UninstallString", NULL, NULL, NULL, &uninstlen))
 247             {
 248                 command = malloc(uninstlen);
 249                 RegQueryValueExW(hkeyApp, L"UninstallString", 0, 0, (BYTE *)command, &uninstlen);
 250             }
 251             else
 252             {
 253                 RegCloseKey(hkeyApp);
 254                 sizeOfSubKeyName = 255;
 255                 continue;
 256             }
 257             numentries++;
 258             entries = realloc(entries, numentries * sizeof(uninst_entry));
 259             entries[numentries-1].root = root;
 260             entries[numentries-1].key = wcsdup(subKeyName);
 261             entries[numentries-1].descr = malloc(displen);
 262             RegQueryValueExW(hkeyApp, L"DisplayName", 0, 0, (BYTE *)entries[numentries-1].descr, &displen);
 263             entries[numentries-1].command = command;
 264             entries[numentries-1].active = 0;
 265             WINE_TRACE("allocated entry #%d: %s (%s), %s\n",
 266             numentries, wine_dbgstr_w(entries[numentries-1].key), wine_dbgstr_w(entries[numentries-1].descr), wine_dbgstr_w(entries[numentries-1].command));
 267             if(sFilter != NULL && StrStrIW(entries[numentries-1].descr,sFilter)==NULL)
 268                 numentries--;
 269         }
 270         RegCloseKey(hkeyApp);
 271         sizeOfSubKeyName = 255;
 272     }
 273     return 1;
 274
 275 }
 276
 277 static int FetchUninstallInformation(void)
 278 {
 279     static const BOOL is_64bit = sizeof(void *) > sizeof(int);
 280     int rc = 0;
 281     HKEY root;
 282
 283     numentries = 0;
 284     oldsel = -1;
 285     if (!entries)
 286         entries = malloc(sizeof(uninst_entry));
 287
 288     if (!RegOpenKeyExW(HKEY_LOCAL_MACHINE, PathUninstallW, 0, KEY_READ, &root))
 289     {
 290         rc |= FetchFromRootKey(root);
 291         RegCloseKey(root);
 292     }
 293     if (is_64bit &&
 294         !RegOpenKeyExW(HKEY_LOCAL_MACHINE, PathUninstallW, 0, KEY_READ|KEY_WOW64_32KEY, &root))
 295     {
 296         rc |= FetchFromRootKey(root);
 297         RegCloseKey(root);
 298     }
 299     if (!RegOpenKeyExW(HKEY_CURRENT_USER, PathUninstallW, 0, KEY_READ, &root))
 300     {
 301         rc |= FetchFromRootKey(root);
 302         RegCloseKey(root);
 303     }
 304
 305     qsort(entries, numentries, sizeof(uninst_entry), cmp_by_name);
 306     return rc;
 307 }
 308
 309 static void UninstallProgram(void)
 310 {
 311     unsigned int i;
 312     WCHAR errormsg[1024];
 313     BOOL res;
 314     STARTUPINFOW si;
 315     PROCESS_INFORMATION info;
 316     DWORD exit_code;
 317     HKEY hkey;
 318     for (i=0; i < numentries; i++)
 319     {
 320         if (!(entries[i].active)) /* don't uninstall this one */
 321             continue;
 322         WINE_TRACE("uninstalling %s\n", wine_dbgstr_w(entries[i].descr));
 323         memset(&si, 0, sizeof(STARTUPINFOW));
 324         si.cb = sizeof(STARTUPINFOW);
 325         si.wShowWindow = SW_NORMAL;
 326         res = CreateProcessW(NULL, entries[i].command, NULL, NULL, FALSE, 0, NULL, NULL, &si, &info);
 327         if (res)
 328         {   /* wait for the process to exit */
 329             WaitForSingleObject(info.hProcess, INFINITE);
 330             res = GetExitCodeProcess(info.hProcess, &exit_code);
 331             WINE_TRACE("%d: %08lx\n", res, exit_code);
 332         }
 333         else
 334         {
 335             WCHAR sAppName[MAX_STRING_LEN];
 336             WCHAR sUninstallFailed[MAX_STRING_LEN];
 337             HINSTANCE hInst = GetModuleHandleW(0);
 338
 339             LoadStringW(hInst, IDS_APPNAME, sAppName, ARRAY_SIZE(sAppName));
 340             LoadStringW(hInst, IDS_UNINSTALLFAILED, sUninstallFailed, ARRAY_SIZE(sUninstallFailed));
 341             wsprintfW(errormsg, sUninstallFailed, entries[i].command);
 342             if(MessageBoxW(0, errormsg, sAppName, MB_YESNO | MB_ICONQUESTION)==IDYES)
 343             {
 344                 /* delete the application's uninstall entry */
 345                 RegOpenKeyExW(entries[i].root, PathUninstallW, 0, KEY_READ, &hkey);
 346                 RegDeleteKeyW(hkey, entries[i].key);
 347                 RegCloseKey(hkey);
 348             }
 349         }
 350     }
 351     WINE_TRACE("finished uninstall phase.\n");
 352 }