mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-11-08 04:27:37 +00:00
144 lines
3.6 KiB
C++
144 lines
3.6 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/*
|
|
* The contents of this file are subject to the Netscape Public License
|
|
* Version 1.0 (the "NPL"); you may not use this file except in
|
|
* compliance with the NPL. You may obtain a copy of the NPL at
|
|
* http://www.mozilla.org/NPL/
|
|
*
|
|
* Software distributed under the NPL is distributed on an "AS IS" basis,
|
|
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the NPL
|
|
* for the specific language governing rights and limitations under the
|
|
* NPL.
|
|
*
|
|
* The Initial Developer of this code under the NPL is Netscape
|
|
* Communications Corporation. Portions created by Netscape are
|
|
* Copyright (C) 1998 Netscape Communications Corporation. All Rights
|
|
* Reserved.
|
|
*/
|
|
|
|
#include <iostream.h>
|
|
#include "plstr.h"
|
|
#include "prlink.h"
|
|
#include "nsIComponentManager.h"
|
|
#include "nsIServiceManager.h"
|
|
|
|
static PRBool gUnreg = PR_FALSE;
|
|
|
|
void print_err(nsresult err)
|
|
{
|
|
switch (err) {
|
|
case NS_ERROR_FACTORY_NOT_LOADED:
|
|
cerr << "Factory not loaded";
|
|
break;
|
|
case NS_NOINTERFACE:
|
|
cerr << "No Interface";
|
|
break;
|
|
case NS_ERROR_NULL_POINTER:
|
|
cerr << "Null pointer";
|
|
break;
|
|
case NS_ERROR_OUT_OF_MEMORY:
|
|
cerr << "Out of memory";
|
|
break;
|
|
default:
|
|
cerr << hex << err << dec;
|
|
}
|
|
}
|
|
PRLibrary *GetLib(const char *path) {
|
|
return PR_LoadLibrary(path);
|
|
}
|
|
|
|
nsresult Register(const char *path)
|
|
{
|
|
nsresult res = NS_OK;
|
|
PRLibrary *instance = GetLib(path);
|
|
if (instance != NULL) {
|
|
nsRegisterProc proc = (nsRegisterProc) PR_FindSymbol(instance,
|
|
"NSRegisterSelf");
|
|
if (proc != NULL) {
|
|
nsIServiceManager* servMgr = NULL;
|
|
res = nsServiceManager::GetGlobalServiceManager(&servMgr);
|
|
if (res == NS_OK)
|
|
{
|
|
res = proc(servMgr, path);
|
|
}
|
|
}
|
|
PR_UnloadLibrary(instance);
|
|
} else {
|
|
res = NS_ERROR_FACTORY_NOT_LOADED;
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
nsresult Unregister(const char *path)
|
|
{
|
|
nsresult res = NS_OK;
|
|
PRLibrary *instance = GetLib(path);
|
|
if (instance != NULL) {
|
|
nsUnregisterProc proc = (nsUnregisterProc) PR_FindSymbol(instance,
|
|
"NSUnregisterSelf");
|
|
if (proc != NULL) {
|
|
nsIServiceManager* servMgr = NULL;
|
|
res = nsServiceManager::GetGlobalServiceManager(&servMgr);
|
|
if (res == NS_OK)
|
|
{
|
|
res = proc(servMgr, path);
|
|
}
|
|
}
|
|
PR_UnloadLibrary(instance);
|
|
} else {
|
|
res = NS_ERROR_FACTORY_NOT_LOADED;
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
int ProcessArgs(int argc, char *argv[])
|
|
{
|
|
int i = 1;
|
|
nsresult res;
|
|
|
|
while (i < argc) {
|
|
if (argv[i][0] == '-') {
|
|
int j;
|
|
for (j = 1; argv[i][j] != '\0'; j++) {
|
|
switch (argv[i][j]) {
|
|
case 'u':
|
|
gUnreg = PR_TRUE;
|
|
break;
|
|
default:
|
|
cerr << "Unknown option '" << argv[i][j] << "'\n";
|
|
}
|
|
}
|
|
i++;
|
|
} else {
|
|
if (gUnreg == PR_TRUE) {
|
|
res = Unregister(argv[i]);
|
|
if (NS_SUCCEEDED(res)) {
|
|
cout << "Successfully unregistered: " << argv[i] << "\n";
|
|
} else {
|
|
cerr << "Unregister failed (";
|
|
print_err(res);
|
|
cerr << "): " << argv[i] << "\n";
|
|
}
|
|
} else {
|
|
res = Register(argv[i]);
|
|
if (NS_SUCCEEDED(res)) {
|
|
cout << "Successfully registered: " << argv[i] << "\n";
|
|
} else {
|
|
cerr << "Register failed (";
|
|
print_err(res);
|
|
cerr << "): " << argv[i] << "\n";
|
|
}
|
|
}
|
|
i++;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
return ProcessArgs(argc, argv);
|
|
}
|