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 }
290
This page was automatically generated by the
LXR engine.
Visit the LXR main site for more
information.