rpcss_main.c File Reference

#include <stdio.h>
#include <stdarg.h>
#include <limits.h>
#include <assert.h>
#include "windef.h"
#include "winbase.h"
#include "winnt.h"
#include "winsvc.h"
#include "irot.h"
#include "epm.h"
#include "irpcss.h"
#include "wine/debug.h"
#include "wine/list.h"

Include dependency graph for rpcss_main.c:

Go to the source code of this file.

Classes
struct	registered_class

Macros
#define	COBJMACROS

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (ole)
HRESULT __cdecl	irpcss_server_register (handle_t h, const GUID *clsid, unsigned int flags, PMInterfacePointer object, unsigned int *cookie)
static void	scm_revoke_class (struct registered_class *_class)
HRESULT __cdecl	irpcss_server_revoke (handle_t h, unsigned int cookie)
HRESULT __cdecl	irpcss_get_class_object (handle_t h, const GUID *clsid, PMInterfacePointer *object)
HRESULT __cdecl	irpcss_get_thread_seq_id (handle_t h, DWORD *id)
static RPC_STATUS	RPCSS_Initialize (void)
static DWORD WINAPI	service_handler (DWORD ctrl, DWORD event_type, LPVOID event_data, LPVOID context)
static void WINAPI	ServiceMain (DWORD argc, LPWSTR *argv)
int __cdecl	wmain (int argc, WCHAR *argv[])

Variables
static WCHAR	rpcssW [] = L"RpcSs"
static HANDLE	exit_event
static SERVICE_STATUS_HANDLE	service_handle
static CRITICAL_SECTION	registered_classes_cs = { NULL, -1, 0, 0, 0, 0 }
static struct list	registered_classes = LIST_INIT(registered_classes)

Macro Definition Documentation

COBJMACROS

#define COBJMACROS

Definition at line 24 of file rpcss_main.c.

Function Documentation

irpcss_get_class_object()

HRESULT __cdecl irpcss_get_class_object	(	handle_t	h,
		const GUID *	clsid,
		PMInterfacePointer *	object
	)

Definition at line 112 of file rpcss_main.c.

{
    struct registered_class *cur;
 
    *object = NULL;
 
    EnterCriticalSection(&registered_classes_cs);
 
    LIST_FOR_EACH_ENTRY(cur, &registered_classes, struct registered_class, entry)
    {
        if (!memcmp(clsid, &cur->clsid, sizeof(*clsid)))
        {
            *object = MIDL_user_allocate(FIELD_OFFSET(MInterfacePointer, abData[cur->object->ulCntData]));
            if (*object)
            {
                (*object)->ulCntData = cur->object->ulCntData;
                memcpy((*object)->abData, cur->object->abData, cur->object->ulCntData);
            }
 
            if (cur->single_use)
                scm_revoke_class(cur);
 
            break;
        }
    }
 
    LeaveCriticalSection(&registered_classes_cs);
 
    return *object ? S_OK : E_NOINTERFACE;
}

Referenced by rpcss_get_class_object().

irpcss_get_thread_seq_id()

HRESULT __cdecl irpcss_get_thread_seq_id	(	handle_t	h,
		DWORD *	id
	)

Definition at line 144 of file rpcss_main.c.

{
    static LONG thread_seq_id;
    *id = InterlockedIncrement(&thread_seq_id);
    return S_OK;
}

Referenced by rpcss_get_next_seqid().

irpcss_server_register()

HRESULT __cdecl irpcss_server_register	(	handle_t	h,
		const GUID *	clsid,
		unsigned int	flags,
		PMInterfacePointer	object,
		unsigned int *	cookie
	)

Definition at line 58 of file rpcss_main.c.

{
    struct registered_class *entry;
    static LONG next_cookie;
 
    if (!(entry = calloc(1, sizeof(*entry))))
        return E_OUTOFMEMORY;
 
    entry->clsid = *clsid;
    entry->single_use = !(flags & (REGCLS_MULTIPLEUSE | REGCLS_MULTI_SEPARATE));
    if (!(entry->object = malloc(FIELD_OFFSET(MInterfacePointer, abData[object->ulCntData]))))
    {
        free(entry);
        return E_OUTOFMEMORY;
    }
    entry->object->ulCntData = object->ulCntData;
    memcpy(&entry->object->abData, object->abData, object->ulCntData);
    *cookie = entry->cookie = InterlockedIncrement(&next_cookie);
 
    EnterCriticalSection(&registered_classes_cs);
    list_add_tail(&registered_classes, &entry->entry);
    LeaveCriticalSection(&registered_classes_cs);
 
    return S_OK;
}

Referenced by rpcss_server_register().

irpcss_server_revoke()

HRESULT __cdecl irpcss_server_revoke	(	handle_t	h,
		unsigned int	cookie
	)

Definition at line 92 of file rpcss_main.c.

{
    struct registered_class *cur;
 
    EnterCriticalSection(&registered_classes_cs);
 
    LIST_FOR_EACH_ENTRY(cur, &registered_classes, struct registered_class, entry)
    {
        if (cur->cookie == cookie)
        {
            scm_revoke_class(cur);
            break;
        }
    }
 
    LeaveCriticalSection(&registered_classes_cs);
 
    return S_OK;
}

Referenced by rpc_revoke_local_server().

RPCSS_Initialize()

static RPC_STATUS RPCSS_Initialize ( void  )

static

Definition at line 151 of file rpcss_main.c.

{
    static unsigned short irot_protseq[] = IROT_PROTSEQ;
    static unsigned short irot_endpoint[] = IROT_ENDPOINT;
    static unsigned short epm_protseq[] = L"ncacn_np";
    static unsigned short epm_endpoint[] = L"\\pipe\\epmapper";
    static unsigned short epm_protseq_lrpc[] = L"ncalrpc";
    static unsigned short epm_endpoint_lrpc[] = L"epmapper";
    static unsigned short irpcss_protseq[] = IRPCSS_PROTSEQ;
    static unsigned short irpcss_endpoint[] = IRPCSS_ENDPOINT;
    static const struct protseq_map
    {
        unsigned short *protseq;
        unsigned short *endpoint;
    } protseqs[] =
    {
        { epm_protseq, epm_endpoint },
        { epm_protseq_lrpc, epm_endpoint_lrpc },
        { irot_protseq, irot_endpoint },
        { irpcss_protseq, irpcss_endpoint },
    };
    RPC_IF_HANDLE ifspecs[] =
    {
        epm_v3_0_s_ifspec,
        Irot_v0_2_s_ifspec,
        Irpcss_v0_0_s_ifspec,
    };
    RPC_STATUS status;
    int i, j;
 
    WINE_TRACE("\n");
 
    for (i = 0, j = 0; i < ARRAY_SIZE(ifspecs); ++i, j = i)
    {
        status = RpcServerRegisterIf(ifspecs[i], NULL, NULL);
        if (status != RPC_S_OK)
            goto fail;
    }
 
    for (i = 0; i < ARRAY_SIZE(protseqs); ++i)
    {
        status = RpcServerUseProtseqEpW(protseqs[i].protseq, RPC_C_PROTSEQ_MAX_REQS_DEFAULT,
                protseqs[i].endpoint, NULL);
        if (status != RPC_S_OK)
            goto fail;
    }
 
    status = RpcServerListen(1, RPC_C_LISTEN_MAX_CALLS_DEFAULT, TRUE);
    if (status != RPC_S_OK)
        goto fail;
 
    return RPC_S_OK;
 
fail:
    for (i = 0; i < j; ++i)
        RpcServerUnregisterIf(ifspecs[i], NULL, FALSE);
 
    return status;
}

Referenced by ServiceMain().

scm_revoke_class()

static void scm_revoke_class ( struct registered_class *  _class )

static

Definition at line 85 of file rpcss_main.c.

{
    list_remove(&_class->entry);
    free(_class->object);
    free(_class);
}

Referenced by irpcss_get_class_object(), and irpcss_server_revoke().

service_handler()

static DWORD WINAPI service_handler ( DWORD  ctrl, DWORD  event_type, LPVOID  event_data, LPVOID  context  )

static

Definition at line 211 of file rpcss_main.c.

{
    SERVICE_STATUS status;
 
    status.dwServiceType             = SERVICE_WIN32;
#ifdef __REACTOS__
    status.dwControlsAccepted        = 0;
#else
    status.dwControlsAccepted        = SERVICE_ACCEPT_STOP;
#endif
    status.dwWin32ExitCode           = 0;
    status.dwServiceSpecificExitCode = 0;
    status.dwCheckPoint              = 0;
    status.dwWaitHint                = 0;
 
    switch (ctrl)
    {
    case SERVICE_CONTROL_STOP:
    case SERVICE_CONTROL_SHUTDOWN:
        TRACE( "shutting down\n" );
        RpcMgmtStopServerListening( NULL );
        RpcServerUnregisterIf( epm_v3_0_s_ifspec, NULL, TRUE );
        RpcServerUnregisterIf( Irot_v0_2_s_ifspec, NULL, TRUE );
        status.dwCurrentState = SERVICE_STOP_PENDING;
        status.dwControlsAccepted = 0;
        SetServiceStatus( service_handle, &status );
        SetEvent( exit_event );
        return NO_ERROR;
    default:
        FIXME( "got service ctrl %lx\n", ctrl );
        status.dwCurrentState = SERVICE_RUNNING;
        SetServiceStatus( service_handle, &status );
        return NO_ERROR;
    }
}

Referenced by ServiceMain().

ServiceMain()

static void WINAPI ServiceMain ( DWORD  argc, LPWSTR *  argv  )

static

Definition at line 251 of file rpcss_main.c.

{
    SERVICE_STATUS status;
    RPC_STATUS ret;
 
    TRACE( "starting service\n" );
 
    if ((ret = RPCSS_Initialize()))
    {
        WARN("Failed to initialize rpc interfaces, status %ld.\n", ret);
        return;
    }
 
    exit_event = CreateEventW( NULL, TRUE, FALSE, NULL );
 
    service_handle = RegisterServiceCtrlHandlerExW( rpcssW, service_handler, NULL );
    if (!service_handle) return;
 
    status.dwServiceType             = SERVICE_WIN32;
    status.dwCurrentState            = SERVICE_RUNNING;
#ifdef __REACTOS__
    status.dwControlsAccepted        = 0;
#else
    status.dwControlsAccepted        = SERVICE_ACCEPT_STOP;
#endif
    status.dwWin32ExitCode           = 0;
    status.dwServiceSpecificExitCode = 0;
    status.dwCheckPoint              = 0;
#ifdef __REACTOS__
    status.dwWaitHint                = 0;
#else
    status.dwWaitHint                = 10000;
#endif
    SetServiceStatus( service_handle, &status );
 
#ifdef __REACTOS__
    DoRpcSsSetupConfiguration();
#endif
 
    WaitForSingleObject( exit_event, INFINITE );
 
    status.dwCurrentState     = SERVICE_STOPPED;
    status.dwControlsAccepted = 0;
    SetServiceStatus( service_handle, &status );
    TRACE( "service stopped\n" );
}

Referenced by wmain().

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

ole

)

wmain()

int __cdecl wmain	(	int	argc,
		WCHAR *	argv[]
	)

Definition at line 298 of file rpcss_main.c.

{
    static const SERVICE_TABLE_ENTRYW service_table[] =
    {
        { rpcssW, ServiceMain },
        { NULL, NULL }
    };
 
    StartServiceCtrlDispatcherW( service_table );
    return 0;
}

Variable Documentation

exit_event

HANDLE exit_event

static

Definition at line 43 of file rpcss_main.c.

Referenced by at_exit_func1(), service_handler(), ServiceMain(), test_exit(), and test_quick_exit().

registered_classes

struct list registered_classes = LIST_INIT(registered_classes)

static

Definition at line 56 of file rpcss_main.c.

Referenced by irpcss_get_class_object(), irpcss_server_register(), and irpcss_server_revoke().

registered_classes_cs

CRITICAL_SECTION registered_classes_cs = { NULL, -1, 0, 0, 0, 0 }

static

Definition at line 55 of file rpcss_main.c.

Referenced by irpcss_get_class_object(), irpcss_server_register(), and irpcss_server_revoke().

rpcssW

WCHAR rpcssW[] = L"RpcSs"

static

Definition at line 42 of file rpcss_main.c.

Referenced by ServiceMain(), and wmain().

service_handle

SERVICE_STATUS_HANDLE service_handle

static

Definition at line 44 of file rpcss_main.c.

Referenced by RequestAddition(), RunApphelpCacheControlTests(), RunTest(), service_handler(), ServiceMain(), and WaitService().