register.c File Reference

#include "config.h"
#include <stdarg.h>
#include "windef.h"
#include "winbase.h"
#include "winnls.h"
#include "ole2.h"
#include "rpcproxy.h"
#include "atliface.h"

Include dependency graph for register.c:

Go to the source code of this file.

Classes
struct	reg_info

Macros
#define	COBJMACROS
#define	ATL_INITGUID

Functions
static void *	image_base (void)
static	HRESULT (WINAPI *pAtlCreateRegistrar)(IRegistrar **)
static IRegistrar *	create_registrar (HMODULE inst, struct reg_info *info)
static BOOL CALLBACK	register_resource (HMODULE module, LPCWSTR type, LPWSTR name, LONG_PTR arg)
HRESULT	__wine_register_resources (HMODULE module)
HRESULT	__wine_unregister_resources (HMODULE module)
HRESULT __cdecl	__wine_register_resources_new (void)
HRESULT __cdecl	__wine_unregister_resources_new (void)

Variables
static const WCHAR	atl100W [] = {'a','t','l','1','0','0','.','d','l','l',0}
static const WCHAR	regtypeW [] = {'W','I','N','E','_','R','E','G','I','S','T','R','Y',0}
static const WCHAR	moduleW [] = {'M','O','D','U','L','E',0}
static const WCHAR	systemrootW [] = {'S','y','s','t','e','m','R','o','o','t',0}
static HMODULE	atl100

Macro Definition Documentation

ATL_INITGUID

#define ATL_INITGUID

Definition at line 25 of file register.c.

COBJMACROS

#define COBJMACROS

Definition at line 24 of file register.c.

Function Documentation

__wine_register_resources()

HRESULT __wine_register_resources

(

HMODULE

module

)

Definition at line 112 of file register.c.

{
    struct reg_info info;
 
    info.registrar = NULL;
    info.do_register = TRUE;
    info.result = S_OK;
    EnumResourceNamesW( module, regtypeW, register_resource, (LONG_PTR)&info );
    if (info.registrar) IRegistrar_Release( info.registrar );
    return info.result;
}

Referenced by DllRegisterServer(), and RegisterTypeLibraries().

__wine_register_resources_new()

HRESULT __cdecl __wine_register_resources_new

(

void

)

Definition at line 138 of file register.c.

{
    struct reg_info info;
 
    info.registrar = NULL;
    info.do_register = TRUE;
    info.result = S_OK;
    EnumResourceNamesW( image_base(), regtypeW, register_resource, (LONG_PTR)&info );
    if (info.registrar) IRegistrar_Release( info.registrar );
    return info.result;
}

__wine_unregister_resources()

HRESULT __wine_unregister_resources

(

HMODULE

module

)

Definition at line 124 of file register.c.

{
    struct reg_info info;
 
    info.registrar = NULL;
    info.do_register = FALSE;
    info.result = S_OK;
    EnumResourceNamesW( module, regtypeW, register_resource, (LONG_PTR)&info );
    if (info.registrar) IRegistrar_Release( info.registrar );
    return info.result;
}

Referenced by DllUnregisterServer().

__wine_unregister_resources_new()

HRESULT __cdecl __wine_unregister_resources_new

(

void

)

Definition at line 150 of file register.c.

{
    struct reg_info info;
 
    info.registrar = NULL;
    info.do_register = FALSE;
    info.result = S_OK;
    EnumResourceNamesW( image_base(), regtypeW, register_resource, (LONG_PTR)&info );
    if (info.registrar) IRegistrar_Release( info.registrar );
    return info.result;
}

create_registrar()

static IRegistrar * create_registrar ( HMODULE  inst, struct reg_info *  info  )

static

Definition at line 59 of file register.c.

{
    if (!pAtlCreateRegistrar)
    {
        if (!(atl100 = LoadLibraryW( atl100W )) ||
            !(pAtlCreateRegistrar = (void *)GetProcAddress( atl100, "AtlCreateRegistrar" )))
        {
            info->result = E_NOINTERFACE;
            return NULL;
        }
    }
 
    info->result = pAtlCreateRegistrar( &info->registrar );
    if (SUCCEEDED( info->result ))
    {
        WCHAR str[MAX_PATH];
 
        GetModuleFileNameW( inst, str, MAX_PATH );
        IRegistrar_AddReplacement( info->registrar, moduleW, str );
        GetEnvironmentVariableW( systemrootW, str, MAX_PATH );
        IRegistrar_AddReplacement( info->registrar, systemrootW, str );
    }
    return info->registrar;
}

Referenced by register_resource().

HRESULT()

static HRESULT ( WINAPI *  pAtlCreateRegistrar )

static

image_base()

static void * image_base ( void  )

inlinestatic

Definition at line 33 of file register.c.

{
#if defined(__MINGW32__) || defined(_MSC_VER)
    extern IMAGE_DOS_HEADER __ImageBase;
    return (void *)&__ImageBase;
#else
    extern IMAGE_NT_HEADERS __wine_spec_nt_header;
    return (void *)((__wine_spec_nt_header.OptionalHeader.ImageBase + 0xffff) & ~0xffff);
#endif
}

Referenced by __wine_register_resources_new(), and __wine_unregister_resources_new().

register_resource()

static BOOL CALLBACK register_resource ( HMODULE  module, LPCWSTR  type, LPWSTR  name, LONG_PTR  arg  )

static

Definition at line 84 of file register.c.

{
    struct reg_info *info = (struct reg_info *)arg;
    WCHAR *buffer;
    HRSRC rsrc = FindResourceW( module, name, type );
    char *str = LoadResource( module, rsrc );
    DWORD lenW, lenA = SizeofResource( module, rsrc );
 
    if (!str) return FALSE;
    if (!info->registrar && !create_registrar( module, info )) return FALSE;
    lenW = MultiByteToWideChar( CP_UTF8, 0, str, lenA, NULL, 0 ) + 1;
    if (!(buffer = HeapAlloc( GetProcessHeap(), 0, lenW * sizeof(WCHAR) )))
    {
        info->result = E_OUTOFMEMORY;
        return FALSE;
    }
    MultiByteToWideChar( CP_UTF8, 0, str, lenA, buffer, lenW );
    buffer[lenW - 1] = 0;
 
    if (info->do_register)
        info->result = IRegistrar_StringRegister( info->registrar, buffer );
    else
        info->result = IRegistrar_StringUnregister( info->registrar, buffer );
 
    HeapFree( GetProcessHeap(), 0, buffer );
    return SUCCEEDED(info->result);
}

Referenced by __wine_register_resources(), __wine_register_resources_new(), __wine_unregister_resources(), and __wine_unregister_resources_new().

Variable Documentation

atl100

HMODULE atl100

static

Definition at line 56 of file register.c.

Referenced by create_registrar().

atl100W

const WCHAR atl100W[] = {'a','t','l','1','0','0','.','d','l','l',0}

static

Definition at line 44 of file register.c.

Referenced by create_registrar().

moduleW

const WCHAR moduleW[] = {'M','O','D','U','L','E',0}

static

Definition at line 46 of file register.c.

Referenced by AtlUpdateRegistryFromResourceD(), and create_registrar().

regtypeW

const WCHAR regtypeW[] = {'W','I','N','E','_','R','E','G','I','S','T','R','Y',0}

static

Definition at line 45 of file register.c.

Referenced by __wine_register_resources(), __wine_register_resources_new(), __wine_unregister_resources(), and __wine_unregister_resources_new().

systemrootW

const WCHAR systemrootW[] = {'S','y','s','t','e','m','R','o','o','t',0}

static

Definition at line 47 of file register.c.

Referenced by create_registrar().