This patch adds the ability for regapi to call the DllRegisterServer or
DllUnregisterServer for a list of DLLs.
This commit is contained in:
parent
6edc9517a4
commit
5b5d04dc77
|
@ -20,7 +20,7 @@
|
|||
* Defines and consts
|
||||
*/
|
||||
#define IDENTICAL 0
|
||||
#define COMMAND_COUNT 5
|
||||
#define COMMAND_COUNT 7
|
||||
|
||||
#define KEY_MAX_LEN 1024
|
||||
#define STDIN_MAX_LEN 2048
|
||||
|
@ -97,6 +97,8 @@ static void doDeleteValue(LPSTR lpsLine);
|
|||
static void doCreateKey(LPSTR lpsLine);
|
||||
static void doDeleteKey(LPSTR lpsLine);
|
||||
static void doQueryValue(LPSTR lpsLine);
|
||||
static void doRegisterDLL(LPSTR lpsLine);
|
||||
static void doUnregisterDLL(LPSTR lpsLine);
|
||||
|
||||
/*
|
||||
* current supported api
|
||||
|
@ -106,7 +108,9 @@ static const char* commandNames[COMMAND_COUNT] = {
|
|||
"deleteValue",
|
||||
"createKey",
|
||||
"deleteKey",
|
||||
"queryValue"
|
||||
"queryValue",
|
||||
"registerDLL",
|
||||
"unregisterDLL"
|
||||
};
|
||||
|
||||
/*
|
||||
|
@ -117,7 +121,9 @@ static const commandAPI commandAPIs[COMMAND_COUNT] = {
|
|||
doDeleteValue,
|
||||
doCreateKey,
|
||||
doDeleteKey,
|
||||
doQueryValue
|
||||
doQueryValue,
|
||||
doRegisterDLL,
|
||||
doUnregisterDLL
|
||||
};
|
||||
|
||||
/*
|
||||
|
@ -128,7 +134,9 @@ static const BOOL commandSaveRegistry[COMMAND_COUNT] = {
|
|||
TRUE,
|
||||
TRUE,
|
||||
TRUE,
|
||||
FALSE
|
||||
FALSE,
|
||||
TRUE,
|
||||
TRUE
|
||||
};
|
||||
|
||||
/*
|
||||
|
@ -212,6 +220,11 @@ static char helpText[] =
|
|||
" \"Value2\"\n"
|
||||
" \"Valuen\"\n"
|
||||
" ...\n"
|
||||
" registerDLL\n"
|
||||
" The input file format is a list of DLLs to register\n"
|
||||
"\n"
|
||||
" unregisterDLL\n"
|
||||
" The input file format is a list of DLLs to unregister\n"
|
||||
" February 1999.\n"
|
||||
;
|
||||
|
||||
|
@ -918,6 +931,72 @@ static void doCreateKey(LPSTR line) {
|
|||
printf ("regapi: createKey not yet implemented\n");
|
||||
}
|
||||
|
||||
/******************************************************************************
|
||||
* This funtion is the main entry point to the registerDLL action. It
|
||||
* receives the currently read line, then loads and registers the requested DLLs
|
||||
*/
|
||||
static void doRegisterDLL(LPSTR stdInput) {
|
||||
HMODULE theLib = 0;
|
||||
UINT retVal = 0;
|
||||
|
||||
/* Check for valid input */
|
||||
if (stdInput == NULL)
|
||||
return;
|
||||
|
||||
/* Load and register the library, then free it */
|
||||
theLib = LoadLibrary(stdInput);
|
||||
if (theLib)
|
||||
{
|
||||
FARPROC lpfnDLLRegProc = GetProcAddress(theLib, "DllRegisterServer");
|
||||
if (lpfnDLLRegProc)
|
||||
retVal = (*lpfnDLLRegProc)();
|
||||
else
|
||||
printf("regapi: Couldn't find DllRegisterServer proc in '%s'.\n", stdInput);
|
||||
|
||||
if (retVal != S_OK)
|
||||
printf("regapi: DLLRegisterServer error 0x%x in '%s'.\n", retVal, stdInput);
|
||||
|
||||
FreeLibrary(theLib);
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("regapi: Could not load DLL '%s'.\n", stdInput);
|
||||
}
|
||||
}
|
||||
|
||||
/******************************************************************************
|
||||
* This funtion is the main entry point to the unregisterDLL action. It
|
||||
* receives the currently read line, then loads and unregisters the requested DLLs
|
||||
*/
|
||||
static void doUnregisterDLL(LPSTR stdInput) {
|
||||
HMODULE theLib = 0;
|
||||
UINT retVal = 0;
|
||||
|
||||
/* Check for valid input */
|
||||
if (stdInput == NULL)
|
||||
return;
|
||||
|
||||
/* Load and unregister the library, then free it */
|
||||
theLib = LoadLibrary(stdInput);
|
||||
if (theLib)
|
||||
{
|
||||
FARPROC lpfnDLLRegProc = GetProcAddress(theLib, "DllUnregisterServer");
|
||||
if (lpfnDLLRegProc)
|
||||
retVal = (*lpfnDLLRegProc)();
|
||||
else
|
||||
printf("regapi: Couldn't find DllUnregisterServer proc in '%s'.\n", stdInput);
|
||||
|
||||
if (retVal != S_OK)
|
||||
printf("regapi: DLLUnregisterServer error 0x%x in '%s'.\n", retVal, stdInput);
|
||||
|
||||
FreeLibrary(theLib);
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("regapi: Could not load DLL '%s'.\n", stdInput);
|
||||
}
|
||||
}
|
||||
|
||||
/******************************************************************************
|
||||
* MAIN - The main simply validate the first parameter (command to perform)
|
||||
* It then read the STDIN lines by lines forwarding their processing
|
||||
|
|
Loading…
Reference in New Issue