2002-04-30 01:47:39 +02:00
|
|
|
/*
|
2002-07-20 20:54:03 +02:00
|
|
|
* PURPOSE: Register OLE components in the registry
|
2002-04-30 01:47:39 +02:00
|
|
|
*
|
|
|
|
* Copyright 2001 ReactOS project
|
|
|
|
* Copyright 2001 Jurgen Van Gael [jurgen.vangael@student.kuleuven.ac.be]
|
|
|
|
* Copyright 2002 Andriy Palamarchuk
|
2015-06-26 12:04:22 +02:00
|
|
|
* Copyright 2014, 2015 Hugh McMaster
|
2002-04-30 01:47:39 +02:00
|
|
|
*
|
|
|
|
* This library is free software; you can redistribute it and/or
|
|
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
|
|
* License as published by the Free Software Foundation; either
|
|
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This library is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
|
|
* Lesser General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
|
|
* License along with this library; if not, write to the Free Software
|
2006-05-18 14:49:52 +02:00
|
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
2002-07-20 20:54:03 +02:00
|
|
|
*
|
2002-08-29 01:43:43 +02:00
|
|
|
* This version deliberately differs in error handling compared to the
|
2015-06-26 12:04:22 +02:00
|
|
|
* Windows version.
|
2002-04-30 01:47:39 +02:00
|
|
|
*/
|
|
|
|
|
2006-01-18 14:23:11 +01:00
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
|
|
|
2002-06-13 21:11:53 +02:00
|
|
|
#include "config.h"
|
|
|
|
#include "wine/port.h"
|
|
|
|
|
2002-04-30 01:47:39 +02:00
|
|
|
#include <windows.h>
|
2006-02-09 17:48:12 +01:00
|
|
|
#include <ole2.h>
|
2014-03-19 01:54:57 +01:00
|
|
|
#include "regsvr32.h"
|
2015-06-23 13:27:22 +02:00
|
|
|
#include "wine/unicode.h"
|
2014-03-19 01:54:57 +01:00
|
|
|
#include "wine/debug.h"
|
|
|
|
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(regsvr32);
|
2002-04-30 01:47:39 +02:00
|
|
|
|
2015-05-28 12:00:57 +02:00
|
|
|
typedef HRESULT (WINAPI *DLLREGISTER) (void);
|
|
|
|
typedef HRESULT (WINAPI *DLLUNREGISTER) (void);
|
|
|
|
typedef HRESULT (WINAPI *DLLINSTALL) (BOOL,LPCWSTR);
|
2002-04-30 01:47:39 +02:00
|
|
|
|
2013-11-05 01:06:04 +01:00
|
|
|
static BOOL Silent = FALSE;
|
2002-04-30 01:47:39 +02:00
|
|
|
|
2014-03-19 01:54:57 +01:00
|
|
|
static void __cdecl output_write(UINT id, ...)
|
2002-04-30 01:47:39 +02:00
|
|
|
{
|
2015-06-23 13:27:22 +02:00
|
|
|
WCHAR fmt[1024];
|
2014-03-19 01:54:57 +01:00
|
|
|
__ms_va_list va_args;
|
2015-06-23 13:27:22 +02:00
|
|
|
WCHAR *str;
|
2014-03-19 01:54:57 +01:00
|
|
|
DWORD len, nOut, ret;
|
|
|
|
|
2014-03-19 02:03:49 +01:00
|
|
|
if (Silent) return;
|
|
|
|
|
2015-06-23 13:27:22 +02:00
|
|
|
if (!LoadStringW(GetModuleHandleW(NULL), id, fmt, sizeof(fmt)/sizeof(fmt[0])))
|
2014-03-19 01:54:57 +01:00
|
|
|
{
|
|
|
|
WINE_FIXME("LoadString failed with %d\n", GetLastError());
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
__ms_va_start(va_args, id);
|
|
|
|
SetLastError(NO_ERROR);
|
2015-06-23 13:27:22 +02:00
|
|
|
len = FormatMessageW(FORMAT_MESSAGE_FROM_STRING|FORMAT_MESSAGE_ALLOCATE_BUFFER,
|
|
|
|
fmt, 0, 0, (LPWSTR)&str, 0, &va_args);
|
2014-03-19 01:54:57 +01:00
|
|
|
__ms_va_end(va_args);
|
|
|
|
if (len == 0 && GetLastError() != NO_ERROR)
|
|
|
|
{
|
2015-06-23 13:27:22 +02:00
|
|
|
WINE_FIXME("Could not format string: le=%u, fmt=%s\n", GetLastError(), wine_dbgstr_w(fmt));
|
2014-03-19 01:54:57 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2015-06-23 13:27:22 +02:00
|
|
|
ret = WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE), str, len, &nOut, NULL);
|
2014-03-19 01:54:57 +01:00
|
|
|
|
2015-06-23 13:27:23 +02:00
|
|
|
/* WriteConsole fails if its output is redirected to a file.
|
|
|
|
* If this occurs, we should use an OEM codepage and call WriteFile.
|
|
|
|
*/
|
2014-03-19 01:54:57 +01:00
|
|
|
if (!ret)
|
2015-06-23 13:27:23 +02:00
|
|
|
{
|
|
|
|
DWORD lenA;
|
|
|
|
char *strA;
|
|
|
|
|
|
|
|
lenA = WideCharToMultiByte(GetConsoleOutputCP(), 0, str, len, NULL, 0, NULL, NULL);
|
|
|
|
strA = HeapAlloc(GetProcessHeap(), 0, lenA);
|
|
|
|
if (strA)
|
|
|
|
{
|
|
|
|
WideCharToMultiByte(GetConsoleOutputCP(), 0, str, len, strA, lenA, NULL, NULL);
|
|
|
|
WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), strA, lenA, &nOut, FALSE);
|
|
|
|
HeapFree(GetProcessHeap(), 0, strA);
|
|
|
|
}
|
|
|
|
}
|
2014-03-19 01:54:57 +01:00
|
|
|
LocalFree(str);
|
2002-04-30 01:47:39 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Loads procedure.
|
|
|
|
*
|
|
|
|
* Parameters:
|
|
|
|
* strDll - name of the dll.
|
2015-06-26 12:04:22 +02:00
|
|
|
* procName - name of the procedure to load from the dll.
|
|
|
|
* DllHandle - a variable that receives the handle of the loaded dll.
|
2002-04-30 01:47:39 +02:00
|
|
|
*/
|
2015-06-23 13:27:22 +02:00
|
|
|
static VOID *LoadProc(const WCHAR* strDll, const char* procName, HMODULE* DllHandle)
|
2002-04-30 01:47:39 +02:00
|
|
|
{
|
|
|
|
VOID* (*proc)(void);
|
|
|
|
|
2015-06-23 13:27:22 +02:00
|
|
|
*DllHandle = LoadLibraryExW(strDll, 0, LOAD_WITH_ALTERED_SEARCH_PATH);
|
2002-04-30 01:47:39 +02:00
|
|
|
if(!*DllHandle)
|
|
|
|
{
|
2014-03-19 02:03:49 +01:00
|
|
|
output_write(STRING_DLL_LOAD_FAILED, strDll);
|
2015-09-02 13:12:27 +02:00
|
|
|
ExitProcess(LOADLIBRARY_FAILED);
|
2002-04-30 01:47:39 +02:00
|
|
|
}
|
|
|
|
proc = (VOID *) GetProcAddress(*DllHandle, procName);
|
|
|
|
if(!proc)
|
|
|
|
{
|
2014-03-19 02:03:49 +01:00
|
|
|
output_write(STRING_PROC_NOT_IMPLEMENTED, procName, strDll);
|
2002-04-30 01:47:39 +02:00
|
|
|
FreeLibrary(*DllHandle);
|
2010-05-02 08:24:40 +02:00
|
|
|
return NULL;
|
2002-04-30 01:47:39 +02:00
|
|
|
}
|
|
|
|
return proc;
|
|
|
|
}
|
|
|
|
|
2015-06-23 13:27:22 +02:00
|
|
|
static int RegisterDll(const WCHAR* strDll)
|
2002-04-30 01:47:39 +02:00
|
|
|
{
|
|
|
|
HRESULT hr;
|
|
|
|
DLLREGISTER pfRegister;
|
|
|
|
HMODULE DllHandle = NULL;
|
|
|
|
|
|
|
|
pfRegister = LoadProc(strDll, "DllRegisterServer", &DllHandle);
|
2010-05-02 08:24:40 +02:00
|
|
|
if (!pfRegister)
|
2015-09-02 13:12:26 +02:00
|
|
|
return GETPROCADDRESS_FAILED;
|
2002-04-30 01:47:39 +02:00
|
|
|
|
|
|
|
hr = pfRegister();
|
|
|
|
if(FAILED(hr))
|
|
|
|
{
|
2014-03-19 02:03:49 +01:00
|
|
|
output_write(STRING_REGISTER_FAILED, strDll);
|
2002-04-30 01:47:39 +02:00
|
|
|
return -1;
|
|
|
|
}
|
2014-03-19 02:03:49 +01:00
|
|
|
output_write(STRING_REGISTER_SUCCESSFUL, strDll);
|
2002-04-30 01:47:39 +02:00
|
|
|
|
|
|
|
if(DllHandle)
|
|
|
|
FreeLibrary(DllHandle);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2015-06-23 13:27:22 +02:00
|
|
|
static int UnregisterDll(const WCHAR* strDll)
|
2002-04-30 01:47:39 +02:00
|
|
|
{
|
|
|
|
HRESULT hr;
|
|
|
|
DLLUNREGISTER pfUnregister;
|
|
|
|
HMODULE DllHandle = NULL;
|
|
|
|
|
|
|
|
pfUnregister = LoadProc(strDll, "DllUnregisterServer", &DllHandle);
|
2010-05-02 08:24:40 +02:00
|
|
|
if (!pfUnregister)
|
2015-09-02 13:12:26 +02:00
|
|
|
return GETPROCADDRESS_FAILED;
|
2010-05-02 08:24:40 +02:00
|
|
|
|
2002-04-30 01:47:39 +02:00
|
|
|
hr = pfUnregister();
|
|
|
|
if(FAILED(hr))
|
|
|
|
{
|
2014-03-19 02:03:49 +01:00
|
|
|
output_write(STRING_UNREGISTER_FAILED, strDll);
|
2002-04-30 01:47:39 +02:00
|
|
|
return -1;
|
|
|
|
}
|
2014-03-19 02:03:49 +01:00
|
|
|
output_write(STRING_UNREGISTER_SUCCESSFUL, strDll);
|
2002-04-30 01:47:39 +02:00
|
|
|
|
|
|
|
if(DllHandle)
|
|
|
|
FreeLibrary(DllHandle);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2015-06-23 13:27:22 +02:00
|
|
|
static int InstallDll(BOOL install, const WCHAR *strDll, const WCHAR *command_line)
|
2002-04-30 01:47:39 +02:00
|
|
|
{
|
|
|
|
HRESULT hr;
|
|
|
|
DLLINSTALL pfInstall;
|
|
|
|
HMODULE DllHandle = NULL;
|
|
|
|
|
|
|
|
pfInstall = LoadProc(strDll, "DllInstall", &DllHandle);
|
2010-05-02 08:24:40 +02:00
|
|
|
if (!pfInstall)
|
2015-09-02 13:12:26 +02:00
|
|
|
return GETPROCADDRESS_FAILED;
|
2010-05-02 08:24:40 +02:00
|
|
|
|
2002-04-30 01:47:39 +02:00
|
|
|
hr = pfInstall(install, command_line);
|
|
|
|
if(FAILED(hr))
|
2014-03-19 01:54:57 +01:00
|
|
|
{
|
|
|
|
if (install)
|
2014-03-19 02:03:49 +01:00
|
|
|
output_write(STRING_INSTALL_FAILED, strDll);
|
2014-03-19 01:54:57 +01:00
|
|
|
else
|
2014-03-19 02:03:49 +01:00
|
|
|
output_write(STRING_UNINSTALL_FAILED, strDll);
|
|
|
|
return -1;
|
2014-03-19 01:54:57 +01:00
|
|
|
}
|
2014-03-19 02:03:49 +01:00
|
|
|
if (install)
|
|
|
|
output_write(STRING_INSTALL_SUCCESSFUL, strDll);
|
|
|
|
else
|
|
|
|
output_write(STRING_UNINSTALL_SUCCESSFUL, strDll);
|
2002-04-30 01:47:39 +02:00
|
|
|
|
|
|
|
if(DllHandle)
|
|
|
|
FreeLibrary(DllHandle);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2015-06-23 13:27:22 +02:00
|
|
|
static WCHAR *parse_command_line(WCHAR *command_line)
|
2015-06-15 13:46:06 +02:00
|
|
|
{
|
|
|
|
if (command_line[0] == ':' && command_line[1])
|
|
|
|
{
|
2015-06-23 13:27:22 +02:00
|
|
|
int len = strlenW(command_line);
|
2015-06-15 13:46:06 +02:00
|
|
|
|
|
|
|
command_line++;
|
|
|
|
len--;
|
|
|
|
/* remove double quotes */
|
|
|
|
if (command_line[0] == '"')
|
|
|
|
{
|
|
|
|
command_line++;
|
|
|
|
len--;
|
|
|
|
if (command_line[0])
|
|
|
|
{
|
|
|
|
len--;
|
|
|
|
command_line[len] = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (command_line[0])
|
2015-06-23 13:27:22 +02:00
|
|
|
return command_line;
|
2015-06-15 13:46:06 +02:00
|
|
|
}
|
2015-06-23 13:27:22 +02:00
|
|
|
return NULL;
|
2015-06-15 13:46:06 +02:00
|
|
|
}
|
|
|
|
|
2015-06-23 13:27:22 +02:00
|
|
|
int wmain(int argc, WCHAR* argv[])
|
2002-04-30 01:47:39 +02:00
|
|
|
{
|
|
|
|
int i;
|
|
|
|
BOOL CallRegister = TRUE;
|
|
|
|
BOOL CallInstall = FALSE;
|
|
|
|
BOOL Unregister = FALSE;
|
|
|
|
BOOL DllFound = FALSE;
|
|
|
|
WCHAR* wsCommandLine = NULL;
|
|
|
|
WCHAR EmptyLine[1] = {0};
|
2002-06-01 01:06:46 +02:00
|
|
|
|
2006-02-09 17:48:12 +01:00
|
|
|
OleInitialize(NULL);
|
2002-04-30 01:47:39 +02:00
|
|
|
|
2015-08-31 11:51:44 +02:00
|
|
|
/* We mirror the Microsoft version by processing all of the flags before
|
2014-03-19 01:54:57 +01:00
|
|
|
* the files (e.g. regsvr32 file1 /s file2 is silent even for file1).
|
2015-06-26 12:04:22 +02:00
|
|
|
*
|
|
|
|
* Note the complication that this version may be passed Unix format filenames
|
|
|
|
* which could be mistaken for flags. The Windows version conveniently
|
|
|
|
* requires each flag to be separate (e.g. no /su), so we will simply
|
|
|
|
* assume that anything longer than /. is a filename.
|
2002-07-20 20:54:03 +02:00
|
|
|
*/
|
2002-04-30 01:47:39 +02:00
|
|
|
for(i = 1; i < argc; i++)
|
|
|
|
{
|
2015-06-15 14:12:45 +02:00
|
|
|
if ((argv[i][0] == '/' || argv[i][0] == '-') && (!argv[i][2] || argv[i][2] == ':'))
|
|
|
|
{
|
2015-06-23 13:27:22 +02:00
|
|
|
switch (tolowerW(argv[i][1]))
|
2015-06-15 14:12:45 +02:00
|
|
|
{
|
|
|
|
case 'u':
|
2002-04-30 01:47:39 +02:00
|
|
|
Unregister = TRUE;
|
2015-06-15 14:12:45 +02:00
|
|
|
break;
|
|
|
|
case 's':
|
2013-11-05 01:06:04 +01:00
|
|
|
Silent = TRUE;
|
2015-06-15 14:12:45 +02:00
|
|
|
break;
|
|
|
|
case 'i':
|
|
|
|
CallInstall = TRUE;
|
2015-06-23 13:27:22 +02:00
|
|
|
wsCommandLine = parse_command_line(argv[i] + 2); /* argv[i] + strlen("/i") */
|
2015-06-15 14:12:45 +02:00
|
|
|
if (!wsCommandLine)
|
|
|
|
wsCommandLine = EmptyLine;
|
|
|
|
break;
|
|
|
|
case 'n':
|
|
|
|
CallRegister = FALSE;
|
|
|
|
break;
|
|
|
|
case 'c':
|
|
|
|
/* console output */;
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
output_write(STRING_UNRECOGNIZED_SWITCH, argv[i]);
|
|
|
|
output_write(STRING_USAGE);
|
|
|
|
return 1;
|
|
|
|
}
|
2015-08-31 11:51:44 +02:00
|
|
|
argv[i] = NULL;
|
2014-03-19 01:54:57 +01:00
|
|
|
}
|
2015-08-31 11:51:44 +02:00
|
|
|
}
|
|
|
|
|
2015-09-02 07:32:07 +02:00
|
|
|
if (!CallInstall && !CallRegister) /* flags: /n or /u /n */
|
|
|
|
return 1;
|
|
|
|
|
2015-08-31 11:51:44 +02:00
|
|
|
for (i = 1; i < argc; i++)
|
|
|
|
{
|
|
|
|
if (argv[i])
|
2002-04-30 01:47:39 +02:00
|
|
|
{
|
2015-06-23 13:27:22 +02:00
|
|
|
WCHAR *DllName = argv[i];
|
2002-04-30 01:47:39 +02:00
|
|
|
int res = 0;
|
|
|
|
|
|
|
|
DllFound = TRUE;
|
|
|
|
if (!CallInstall || (CallInstall && CallRegister))
|
|
|
|
{
|
|
|
|
if(Unregister)
|
|
|
|
res = UnregisterDll(DllName);
|
|
|
|
else
|
|
|
|
res = RegisterDll(DllName);
|
|
|
|
}
|
2002-06-01 01:06:46 +02:00
|
|
|
|
2002-04-30 01:47:39 +02:00
|
|
|
if (res)
|
|
|
|
return res;
|
2015-06-26 12:04:22 +02:00
|
|
|
/* Confirmed. The Windows version stops on the first error. */
|
2002-04-30 01:47:39 +02:00
|
|
|
|
|
|
|
if (CallInstall)
|
|
|
|
{
|
|
|
|
res = InstallDll(!Unregister, DllName, wsCommandLine);
|
|
|
|
}
|
2002-06-01 01:06:46 +02:00
|
|
|
|
2002-07-20 20:54:03 +02:00
|
|
|
if (res)
|
|
|
|
return res;
|
2002-04-30 01:47:39 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!DllFound)
|
|
|
|
{
|
2014-03-19 02:03:49 +01:00
|
|
|
output_write(STRING_HEADER);
|
|
|
|
output_write(STRING_USAGE);
|
2014-03-19 01:54:57 +01:00
|
|
|
return 1;
|
2002-04-30 01:47:39 +02:00
|
|
|
}
|
|
|
|
|
2006-02-09 17:48:12 +01:00
|
|
|
OleUninitialize();
|
|
|
|
|
2002-04-30 01:47:39 +02:00
|
|
|
return 0;
|
|
|
|
}
|