programs/regsvr32/regsvr32.c

   1 /*
   2  * PURPOSE: Register OLE components in the registry
   3  *
   4  * Copyright 2001 ReactOS project
   5  * Copyright 2001 Jurgen Van Gael [jurgen.vangael@student.kuleuven.ac.be]
   6  * Copyright 2002 Andriy Palamarchuk
   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  * This version deliberately differs in error handling compared to the
  23  * windows version.
  24  */
  25
  26 /*
  27  *
  28  *  regsvr32 [/u] [/s] [/n] [/i[:cmdline]] dllname ...
  29  *  [/u]    unregister server
  30  *  [/s]    silent (no message boxes)
  31  *  [/i]    Call DllInstall passing it an optional [cmdline];
  32  *          when used with /u calls dll uninstall.
  33  *  [/n]    Do not call DllRegisterServer; this option must be used with [/i]
  34  *  [/c]    Console output (seems to be deprecated and ignored)
  35  *
  36  *  Note the complication that this version may be passed unix format file names
  37  *  which might be mistaken for flags.  Conveniently the Windows version
  38  *  requires each flag to be separate (e.g. no /su ) and so we will simply
  39  *  assume that anything longer than /. is a filename.
  40  */
  41
  42 /**
  43  * FIXME - currently receives command-line parameters in ASCII only and later
  44  * converts to Unicode. Ideally the function should have wWinMain entry point
  45  * and then work in Unicode only, but it seems Wine does not have necessary
  46  * support.
  47  */
  48
  49 #define WIN32_LEAN_AND_MEAN
  50
  51 #include "config.h"
  52 #include "wine/port.h"
  53
  54 #include <stdio.h>
  55 #include <string.h>
  56 #include <windows.h>
  57 #include <ole2.h>
  58
  59 typedef HRESULT (*DLLREGISTER)          (void);
  60 typedef HRESULT (*DLLUNREGISTER)        (void);
  61 typedef HRESULT (*DLLINSTALL)           (BOOL,LPCWSTR);
  62
  63 int Silent = 0;
  64
  65 static int Usage(void)
  66 {
  67     printf("regsvr32 [/u] [/s] [/n] [/i[:cmdline]] dllname ...\n");
  68     printf("\t[/u]  unregister server\n");
  69     printf("\t[/s]  silent (no message boxes)\n");
  70     printf("\t[/i]  Call DllInstall passing it an optional [cmdline];\n");
  71     printf("\t      when used with /u calls dll uninstall\n");
  72     printf("\t[/n]  Do not call DllRegisterServer; this option "
  73            "must be used with [/i]\n");
  74     return 0;
  75 }
  76
  77 /**
  78  * Loads procedure.
  79  *
  80  * Parameters:
  81  * strDll - name of the dll.
  82  * procName - name of the procedure to load from dll
  83  * pDllHanlde - output variable receives handle of the loaded dll.
  84  */
  85 static VOID *LoadProc(const char* strDll, const char* procName, HMODULE* DllHandle)
  86 {
  87     VOID* (*proc)(void);
  88
  89     *DllHandle = LoadLibraryExA(strDll, 0, LOAD_WITH_ALTERED_SEARCH_PATH);
  90     if(!*DllHandle)
  91     {
  92         if(!Silent)
  93             printf("Failed to load DLL %s\n", strDll);
  94
  95         ExitProcess(1);
  96     }
  97     proc = (VOID *) GetProcAddress(*DllHandle, procName);
  98     if(!proc)
  99     {
 100         if(!Silent)
 101             printf("%s not implemented in DLL %s\n", procName, strDll);
 102         FreeLibrary(*DllHandle);
 103         ExitProcess(1);
 104     }
 105     return proc;
 106 }
 107
 108 static int RegisterDll(const char* strDll)
 109 {
 110     HRESULT hr;
 111     DLLREGISTER pfRegister;
 112     HMODULE DllHandle = NULL;
 113
 114     pfRegister = LoadProc(strDll, "DllRegisterServer", &DllHandle);
 115
 116     hr = pfRegister();
 117     if(FAILED(hr))
 118     {
 119         if(!Silent)
 120             printf("Failed to register DLL %s\n", strDll);
 121
 122         return -1;
 123     }
 124     if(!Silent)
 125         printf("Successfully registered DLL %s\n", strDll);
 126
 127     if(DllHandle)
 128         FreeLibrary(DllHandle);
 129     return 0;
 130 }
 131
 132 static int UnregisterDll(char* strDll)
 133 {
 134     HRESULT hr;
 135     DLLUNREGISTER pfUnregister;
 136     HMODULE DllHandle = NULL;
 137
 138     pfUnregister = LoadProc(strDll, "DllUnregisterServer", &DllHandle);
 139     hr = pfUnregister();
 140     if(FAILED(hr))
 141     {
 142         if(!Silent)
 143             printf("Failed to unregister DLL %s\n", strDll);
 144
 145         return -1;
 146     }
 147     if(!Silent)
 148         printf("Successfully unregistered DLL %s\n", strDll);
 149
 150     if(DllHandle)
 151         FreeLibrary(DllHandle);
 152     return 0;
 153 }
 154
 155 static int InstallDll(BOOL install, char *strDll, WCHAR *command_line)
 156 {
 157     HRESULT hr;
 158     DLLINSTALL pfInstall;
 159     HMODULE DllHandle = NULL;
 160
 161     pfInstall = LoadProc(strDll, "DllInstall", &DllHandle);
 162     hr = pfInstall(install, command_line);
 163     if(FAILED(hr))
 164     {
 165         if(!Silent)
 166             printf("Failed to %s DLL %s\n", install ? "install" : "uninstall",
 167                    strDll);
 168         return -1;
 169     }
 170     if(!Silent)
 171         printf("Successfully %s DLL %s\n",  install ? "installed" : "uninstalled",
 172                strDll);
 173
 174     if(DllHandle)
 175         FreeLibrary(DllHandle);
 176     return 0;
 177 }
 178
 179 int main(int argc, char* argv[])
 180 {
 181     int             i;
 182     BOOL            CallRegister = TRUE;
 183     BOOL            CallInstall = FALSE;
 184     BOOL            Unregister = FALSE;
 185     BOOL            DllFound = FALSE;
 186     WCHAR*          wsCommandLine = NULL;
 187     WCHAR           EmptyLine[1] = {0};
 188
 189     OleInitialize(NULL);
 190
 191     /* Strictly, the Microsoft version processes all the flags before
 192      * the files (e.g. regsvr32 file1 /s file2 is silent even for file1.
 193      * For ease, we will not replicate that and will process the arguments
 194      * in order.
 195      */
 196     for(i = 1; i < argc; i++)
 197     {
 198         if ((!strcasecmp(argv[i], "/u")) ||(!strcasecmp(argv[i], "-u")))
 199                 Unregister = TRUE;
 200         else if ((!strcasecmp(argv[i], "/s"))||(!strcasecmp(argv[i], "-s")))
 201                 Silent = 1;
 202         else if ((!strncasecmp(argv[i], "/i", strlen("/i")))||(!strncasecmp(argv[i], "-i", strlen("-i"))))
 203         {
 204             CHAR* command_line = argv[i] + strlen("/i");
 205
 206             CallInstall = TRUE;
 207             if (command_line[0] == ':' && command_line[1])
 208             {
 209                 int len = strlen(command_line);
 210
 211                 command_line++;
 212                 len--;
 213                 /* remove double quotes */
 214                 if (command_line[0] == '"')
 215                 {
 216                     command_line++;
 217                     len--;
 218                     if (command_line[0])
 219                     {
 220                         len--;
 221                         command_line[len] = 0;
 222                     }
 223                 }
 224                 if (command_line[0])
 225                 {
 226                     len = MultiByteToWideChar(CP_ACP, 0, command_line, -1,
 227                                               NULL, 0);
 228                     wsCommandLine = HeapAlloc(GetProcessHeap(), 0,
 229                                               len * sizeof(WCHAR));
 230                     if (wsCommandLine)
 231                         MultiByteToWideChar(CP_ACP, 0, command_line, -1,
 232                                             wsCommandLine, len);
 233                 }
 234                 else
 235                 {
 236                     wsCommandLine = EmptyLine;
 237                 }
 238             }
 239             else
 240             {
 241                 wsCommandLine = EmptyLine;
 242             }
 243         }
 244         else if((!strcasecmp(argv[i], "/n"))||(!strcasecmp(argv[i], "-n")))
 245             CallRegister = FALSE;
 246         else if((!strcasecmp(argv[i], "/c"))||(!strcasecmp(argv[i], "-c")))
 247             /* console output */;
 248         else if (argv[i][0] == '/' && (!argv[i][2] || argv[i][2] == ':'))
 249             printf("Unrecognized switch %s\n", argv[i]);
 250         else
 251         {
 252             char *DllName = argv[i];
 253             int res = 0;
 254
 255             DllFound = TRUE;
 256             if (!CallInstall || (CallInstall && CallRegister))
 257             {
 258                 if(Unregister)
 259                     res = UnregisterDll(DllName);
 260                 else
 261                     res = RegisterDll(DllName);
 262             }
 263
 264             if (res)
 265                 return res;
 266             /* Confirmed.  The windows version does stop on the first error.*/
 267
 268             if (CallInstall)
 269             {
 270                 res = InstallDll(!Unregister, DllName, wsCommandLine);
 271             }
 272
 273             if (res)
 274                 return res;
 275         }
 276     }
 277
 278     if (!DllFound)
 279     {
 280         if(!Silent)
 281             return Usage();
 282         else
 283             return -1;
 284     }
 285
 286     OleUninitialize();
 287
 288     return 0;
 289 }