source: trunk/src/ole32/regsvr32.cpp@ 9437

/* $Id: regsvr32.cpp,v 1.8 2002-05-28 07:35:07 sandervl Exp $ */
/* 
 * RegSvr32 for OS/2
 * 
 * 1/9/99
 * 
 * Copyright 1999 David J. Raison
 * 
 * 
 * Project Odin Software License can be found in LICENSE.TXT
 * 
 */

#include <windows.h>
#include <odinlx.h>
#include <stdarg.h>
#include <stdio.h>
#include <string.h>
#include <ctype.h>
#include <heapstring.h>

// ======================================================================
// Local Data
// ======================================================================
static  int     fRegister = 1;
static  int     fSilent = 0;
static  char    szFile[1024];
static  HANDLE  hdlObject;

// ======================================================================
// Prototypes.
// ======================================================================

static  HRESULT parse_command(int argc, char * argv[]);
static  HRESULT load_object(void);
static  HRESULT register_object(void);
static  HRESULT deregister_object(void);
static  HRESULT unload_object(void);
static  HRESULT sign_on(void);
static  HRESULT sign_off(void);
static  HRESULT report_msg(char * fmt, ...);
static  HRESULT report_err(HRESULT hr, char * fmt, ...);
static  HRESULT usage(char * fmt, ...);
static int WIN32API EntryPoint(HINSTANCE hInstance, HINSTANCE hPrevInstance,
                    LPSTR lpCmdLine, int nCmdShow);

// ======================================================================
// Public API's
// ======================================================================

// ----------------------------------------------------------------------
// main
// ----------------------------------------------------------------------
int main(int argc, char *argv[])
{
    HRESULT     hr;

    if((hr = parse_command(argc, argv)) != S_OK)
        return hr;

    RegisterLxExe(EntryPoint, NULL);

    return 0;
}

// ======================================================================
// Private functions.
// ======================================================================

// ----------------------------------------------------------------------
// EntryPoint
// ----------------------------------------------------------------------
static int WIN32API EntryPoint
   (HINSTANCE   hInstance,
    HINSTANCE   hPrevInstance,
    LPSTR       lpCmdLine,
    int         nCmdShow)
{
    HRESULT     hr;

    if((hr = sign_on()) != S_OK)
        return hr;

    if((hr = load_object()) != S_OK)
        return hr;

    if (fRegister)
        hr = register_object();
    else
        hr = deregister_object();

    if (hr != S_OK)
        return hr;

    if((hr = unload_object()) != S_OK)
        return hr;

    return sign_off();
}

// ----------------------------------------------------------------------
// sign_on
// ----------------------------------------------------------------------
static  HRESULT sign_on(void)
{
    HRESULT     hr;

    hr = OleInitialize(NULL);
    if (hr != S_OK)
        return report_err(hr, "Error initialising OLE");

    return S_OK;
}

// ----------------------------------------------------------------------
// sign_off
// ----------------------------------------------------------------------
static  HRESULT sign_off(void)
{
    OleUninitialize();
    return report_msg("%s initialised OK", szFile);
}

// ----------------------------------------------------------------------
// parse_command
// ----------------------------------------------------------------------
static  HRESULT parse_command(int argc, char * argv[])
{
    int         ii;
    char *      pArg;

    // Initial conditions.
    fRegister = 1;
    fSilent = 0;
    szFile[0] = 0;

    // Loop through args...
    for (ii = 1; ii < argc; ii++)
    {
        pArg = argv[ii];

        if ((pArg[0] == '-') || (pArg[0] == '/'))
        {
            switch(toupper(pArg[1]))
            {
                case 'S':
                {
                    fSilent = 1;
                    break;
                }

                case 'U':
                {
                    fRegister = 0;
                    break;
                }

                default:
                {
                    return usage("Unknown switch '%c'", pArg[1] );
                }
            }
        }
        else if (strlen(szFile) == 0) 
        {
            strcpy(szFile, pArg);
        }
        else
        {
            return usage("Too many files specified");
        }
    }
    if (strlen(szFile) == 0) 
        return usage("No file specified");

    return S_OK;
}

// ----------------------------------------------------------------------
// load_object
// ----------------------------------------------------------------------
static  HRESULT load_object(void)
{
//SvL: Use LoadLibrary here, not CoLoadLibrary
//     Some dlls (divx codec) call CoFreeUnusedLibraries which unloads our dll 
//     if we use CoLoadLibrary
//     NT's regsvr32 doesn't use CoLoadLibrary either. 
#if 1
    hdlObject = LoadLibraryA(szFile);
#else
    LPOLESTR libnameW = (LPOLESTR)HEAP_strdupAtoW(GetProcessHeap(), 0, szFile);
    hdlObject = LoadLibraryW(libnameW);
    HeapFree(GetProcessHeap(), 0, libnameW);
#endif
    if (hdlObject == 0)
        return report_err(S_FALSE, "Failed to load '%s'", szFile);

    return S_OK;
}

// ----------------------------------------------------------------------
// register_object
// ----------------------------------------------------------------------
static  HRESULT register_object(void)
{
    typedef HRESULT(*LPFNREGISTER)(void);

    LPFNREGISTER        pfnRegister;
    HRESULT             hr;

    pfnRegister = (LPFNREGISTER)GetProcAddress(hdlObject, "DllRegisterServer");

    if (pfnRegister == NULL)
        return report_err(S_FALSE, "Unable to locate registration entry point in '%s'", szFile);

    hr = pfnRegister();
    if (hr != S_OK)
        return report_err(hr, "Registration for '%s' returned error 0x%08lx", szFile, hr);

    return S_OK;
}

// ----------------------------------------------------------------------
// deregister_object
// ----------------------------------------------------------------------
static  HRESULT deregister_object(void)
{
    typedef HRESULT(*LPFNUNREGISTER)(void);

    LPFNUNREGISTER      pfnUnregister;
    HRESULT             hr;

    pfnUnregister = (LPFNUNREGISTER)GetProcAddress(hdlObject, "DllUnregisterServer");

    if (pfnUnregister == NULL)
        return report_err(S_FALSE, "Unable to locate unregistration entry point in '%s'", szFile);

    hr = pfnUnregister();
    if (hr != S_OK)
        return report_err(hr, "Unregistration for '%s' returned error 0x%08lx", szFile, hr);

    return S_OK;
}

// ----------------------------------------------------------------------
// unload_object
// ----------------------------------------------------------------------
static  HRESULT unload_object(void)
{
#if 1
    FreeLibrary(hdlObject);
#else
    CoFreeLibrary(hdlObject);
#endif
    return S_OK;
}

// ----------------------------------------------------------------------
// report_msg
// ----------------------------------------------------------------------
static  HRESULT report_msg(char * fmt, ...)
{
    if (!fSilent)
    {
        va_list va_ptr;
        HWND    hwnd = GetDesktopWindow();
        char    text[1024];

        va_start(va_ptr, fmt);
        vsprintf(text, fmt, va_ptr);
        va_end(va_ptr);

        MessageBoxA(hwnd, text, "RegSvr32", MB_OK | MB_ICONINFORMATION);
    }
    return S_OK;
}

// ----------------------------------------------------------------------
// report_err
// ----------------------------------------------------------------------
static  HRESULT report_err(HRESULT hr, char * fmt, ...)
{
    va_list     va_ptr;
    HWND        hwnd = GetDesktopWindow();
    char        text[1024];

    va_start(va_ptr, fmt);
    vsprintf(text, fmt, va_ptr);
    va_end(va_ptr);

    MessageBoxA(hwnd, text, "RegSvr32", MB_OK | MB_ICONERROR);

    return hr;
}

// ----------------------------------------------------------------------
// usage
// ----------------------------------------------------------------------
static  HRESULT usage(char * fmt, ...)
{
    va_list     va_ptr;
    HWND        hwnd = GetDesktopWindow();
    char        text[256];
    char        msg[1024];

    va_start(va_ptr, fmt);
    vsprintf(text, fmt, va_ptr);
    va_end(va_ptr);
    sprintf(msg, "Error - %s\n\n"
                 "Usage:\n"
                 "regsvr32 [/s] [/u] filename\n"
                 "where\n"
                 "/s - silent completion\n"
                 "/u - unregister", text);

    MessageBoxA(hwnd, msg, "RegSvr32", MB_OK | MB_ICONWARNING);

    return S_FALSE;
}