1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/xpcom/tests/RegFactory.cpp Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,130 @@
1.4 +/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
1.5 +/* This Source Code Form is subject to the terms of the Mozilla Public
1.6 + * License, v. 2.0. If a copy of the MPL was not distributed with this
1.7 + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
1.8 +
1.9 +#include <iostream.h>
1.10 +#include "plstr.h"
1.11 +#include "prlink.h"
1.12 +#include "nsIComponentRegistrar.h"
1.13 +#include "nsIComponentManager.h"
1.14 +#include "nsIServiceManager.h"
1.15 +#include "nsIFile.h"
1.16 +#include "nsCOMPtr.h"
1.17 +#include "nsString.h"
1.18 +
1.19 +static bool gUnreg = false;
1.20 +
1.21 +void print_err(nsresult err)
1.22 +{
1.23 + switch (err) {
1.24 + case NS_ERROR_FACTORY_NOT_LOADED:
1.25 + cerr << "Factory not loaded";
1.26 + break;
1.27 + case NS_NOINTERFACE:
1.28 + cerr << "No Interface";
1.29 + break;
1.30 + case NS_ERROR_NULL_POINTER:
1.31 + cerr << "Null pointer";
1.32 + break;
1.33 + case NS_ERROR_OUT_OF_MEMORY:
1.34 + cerr << "Out of memory";
1.35 + break;
1.36 + default:
1.37 + cerr << hex << err << dec;
1.38 + }
1.39 +}
1.40 +
1.41 +nsresult Register(nsIComponentRegistrar* registrar, const char *path)
1.42 +{
1.43 + nsCOMPtr<nsIFile> file;
1.44 + nsresult rv =
1.45 + NS_NewLocalFile(
1.46 + NS_ConvertUTF8toUTF16(path),
1.47 + true,
1.48 + getter_AddRefs(file));
1.49 + if (NS_FAILED(rv)) return rv;
1.50 + rv = registrar->AutoRegister(file);
1.51 + return rv;
1.52 +}
1.53 +
1.54 +nsresult Unregister(const char *path)
1.55 +{
1.56 + /* NEEDS IMPLEMENTATION */
1.57 +#if 0
1.58 + nsresult res = nsComponentManager::AutoUnregisterComponent(path);
1.59 + return res;
1.60 +#else
1.61 + return NS_ERROR_FAILURE;
1.62 +#endif
1.63 +}
1.64 +
1.65 +int ProcessArgs(nsIComponentRegistrar* registrar, int argc, char *argv[])
1.66 +{
1.67 + int i = 1;
1.68 + nsresult res;
1.69 +
1.70 + while (i < argc) {
1.71 + if (argv[i][0] == '-') {
1.72 + int j;
1.73 + for (j = 1; argv[i][j] != '\0'; j++) {
1.74 + switch (argv[i][j]) {
1.75 + case 'u':
1.76 + gUnreg = true;
1.77 + break;
1.78 + default:
1.79 + cerr << "Unknown option '" << argv[i][j] << "'\n";
1.80 + }
1.81 + }
1.82 + i++;
1.83 + } else {
1.84 + if (gUnreg) {
1.85 + res = Unregister(argv[i]);
1.86 + if (NS_SUCCEEDED(res)) {
1.87 + cout << "Successfully unregistered: " << argv[i] << "\n";
1.88 + } else {
1.89 + cerr << "Unregister failed (";
1.90 + print_err(res);
1.91 + cerr << "): " << argv[i] << "\n";
1.92 + }
1.93 + } else {
1.94 + res = Register(registrar, argv[i]);
1.95 + if (NS_SUCCEEDED(res)) {
1.96 + cout << "Successfully registered: " << argv[i] << "\n";
1.97 + } else {
1.98 + cerr << "Register failed (";
1.99 + print_err(res);
1.100 + cerr << "): " << argv[i] << "\n";
1.101 + }
1.102 + }
1.103 + i++;
1.104 + }
1.105 + }
1.106 + return 0;
1.107 +}
1.108 +
1.109 +int main(int argc, char *argv[])
1.110 +{
1.111 + int ret = 0;
1.112 + nsresult rv;
1.113 + {
1.114 + nsCOMPtr<nsIServiceManager> servMan;
1.115 + rv = NS_InitXPCOM2(getter_AddRefs(servMan), nullptr, nullptr);
1.116 + if (NS_FAILED(rv)) return -1;
1.117 + nsCOMPtr<nsIComponentRegistrar> registrar = do_QueryInterface(servMan);
1.118 + NS_ASSERTION(registrar, "Null nsIComponentRegistrar");
1.119 +
1.120 + /* With no arguments, RegFactory will autoregister */
1.121 + if (argc <= 1)
1.122 + {
1.123 + rv = registrar->AutoRegister(nullptr);
1.124 + ret = (NS_FAILED(rv)) ? -1 : 0;
1.125 + }
1.126 + else
1.127 + ret = ProcessArgs(registrar, argc, argv);
1.128 + } // this scopes the nsCOMPtrs
1.129 + // no nsCOMPtrs are allowed to be alive when you call NS_ShutdownXPCOM
1.130 + rv = NS_ShutdownXPCOM(nullptr);
1.131 + NS_ASSERTION(NS_SUCCEEDED(rv), "NS_ShutdownXPCOM failed");
1.132 + return ret;
1.133 +}
