1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
/*
* $Header$
*
* Copyright 2008 Massachusetts Institute of Technology.
* All Rights Reserved.
*
* Export of this software from the United States of America may
* require a specific license from the United States Government.
* It is the responsibility of any person or organization contemplating
* export to obtain such a license before exporting.
*
* WITHIN THAT CONSTRAINT, permission to use, copy, modify, and
* distribute this software and its documentation for any purpose and
* without fee is hereby granted, provided that the above copyright
* notice appear in all copies and that both that copyright notice and
* this permission notice appear in supporting documentation, and that
* the name of M.I.T. not be used in advertising or publicity pertaining
* to distribution of the software without specific, written prior
* permission. Furthermore if you modify this software you must label
* your software as modified software and not distribute it in such a
* fashion that it might be confused with the original M.I.T. software.
* M.I.T. makes no representations about the suitability of
* this software for any purpose. It is provided "as is" without express
* or implied warranty.
*/
#pragma once
#include "autolock.hxx"
#include <rpc.h>
typedef RPC_STATUS (RPC_ENTRY *FP_RpcBindingSetAuthInfoExA)(
IN RPC_BINDING_HANDLE Binding,
IN unsigned char __RPC_FAR * ServerPrincName,
IN unsigned long AuthnLevel,
IN unsigned long AuthnSvc,
IN RPC_AUTH_IDENTITY_HANDLE AuthIdentity, OPTIONAL
IN unsigned long AuthzSvc,
IN RPC_SECURITY_QOS *SecurityQos OPTIONAL
);
typedef RPC_STATUS (RPC_ENTRY *FP_RpcBindingSetAuthInfoExW)(
IN RPC_BINDING_HANDLE Binding,
IN unsigned short __RPC_FAR * ServerPrincName,
IN unsigned long AuthnLevel,
IN unsigned long AuthnSvc,
IN RPC_AUTH_IDENTITY_HANDLE AuthIdentity, OPTIONAL
IN unsigned long AuthzSvc, OPTIONAL
IN RPC_SECURITY_QOS *SecurityQOS
);
typedef RPC_STATUS (RPC_ENTRY *FP_RpcServerRegisterIfEx)(
IN RPC_IF_HANDLE IfSpec,
IN UUID __RPC_FAR * MgrTypeUuid,
IN RPC_MGR_EPV __RPC_FAR * MgrEpv,
IN unsigned int Flags,
IN unsigned int MaxCalls,
IN RPC_IF_CALLBACK_FN __RPC_FAR *IfCallback
);
#ifdef UNICODE
#define FP_RpcBindingSetAuthInfoEx FP_RpcBindingSetAuthInfoExW
#define FN_RpcBindingSetAuthInfoEx "RpcBindingSetAuthInfoExW"
#else
#define FP_RpcBindingSetAuthInfoEx FP_RpcBindingSetAuthInfoExA
#define FN_RpcBindingSetAuthInfoEx "RpcBindingSetAuthInfoExA"
#endif
#define FN_RpcServerRegisterIfEx "RpcServerRegisterIfEx"
class Init
{
public:
struct InitInfo {
BOOL isNT;
FP_RpcBindingSetAuthInfoEx fRpcBindingSetAuthInfoEx;
FP_RpcServerRegisterIfEx fRpcServerRegisterIfEx;
};
static DWORD Initialize();
static DWORD Cleanup();
static DWORD Info(InitInfo& info);
static bool Initialized() { return s_init; }
private:
static CcOsLock s_lock;
static DWORD s_refcount;
static DWORD s_error;
static bool s_init;
static InitInfo s_info;
static HINSTANCE s_hRpcDll;
};
#define INIT_INIT_EX(trap, error) \
do \
{ \
if (!Init::Initialized()) \
{ \
DWORD rc = Init::Initialize(); \
if (rc) return (trap) ? (error) : rc; \
} \
} while(0)
|