140 lines
3.4 KiB
C
140 lines
3.4 KiB
C
|
#include <precomp.h>
|
||
|
|
|
||
|
|
LPWSTR pszStrBindingOptions = L"Security=Impersonation Dynamic False";
|
||
|
|
|
||
|
|
handle_t
|
||
|
|
STRING_HANDLE_bind(
|
||
|
|
STRING_HANDLE lpStr)
|
||
|
|
{
|
||
|
|
RPC_STATUS RpcStatus = RPC_S_OK;
|
||
|
|
LPWSTR strSrv;
|
||
|
|
LPWSTR pszStringBinding = NULL;
|
||
|
|
handle_t hBinding = NULL;
|
||
|
|
LPWSTR pszServerPrincipalName = NULL;
|
||
|
|
BOOL bBound = FALSE;
|
||
|
|
|
||
|
|
strSrv = (!lpStr || !*lpStr || !lstrcmp(L"127.0.0.1", lpStr)) ? NULL : lpStr;
|
||
|
|
RpcStatus = RpcStringBindingComposeW(
|
||
|
|
0,
|
||
|
|
(strSrv==NULL) ? L"ncalrpc" : L"ncacn_np",
|
||
|
|
strSrv,
|
||
|
|
(strSrv==NULL) ? L"wzcsvc" : L"\\pipe\\wzcsvc",
|
||
|
|
pszStrBindingOptions,
|
||
|
|
&pszStringBinding);
|
||
|
|
|
||
|
|
if (RpcStatus == RPC_S_OK)
|
||
|
|
{
|
||
|
|
RpcStatus = RpcBindingFromStringBindingW(
|
||
|
|
pszStringBinding,
|
||
|
|
&hBinding);
|
||
|
|
RpcStringFreeW(&pszStringBinding);
|
||
|
|
bBound = (RpcStatus == RPC_S_OK);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (RpcStatus == RPC_S_OK)
|
||
|
|
{
|
||
|
|
RpcStatus = RpcBindingSetOption(
|
||
|
|
hBinding,
|
||
|
|
RPC_C_OPT_UNIQUE_BINDING,
|
||
|
|
(ULONG_PTR) 1);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (RpcStatus == RPC_S_OK)
|
||
|
|
{
|
||
|
|
RpcStatus = RpcMgmtInqServerPrincName(
|
||
|
|
hBinding,
|
||
|
|
RPC_C_AUTHN_WINNT,
|
||
|
|
&pszServerPrincipalName);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (RpcStatus == RPC_S_OK)
|
||
|
|
{
|
||
|
|
RpcStatus = RpcBindingSetAuthInfoW(
|
||
|
|
hBinding,
|
||
|
|
pszServerPrincipalName,
|
||
|
|
RPC_C_PROTECT_LEVEL_PKT_PRIVACY,
|
||
|
|
(strSrv==NULL) ? RPC_C_AUTHN_WINNT : RPC_C_AUTHN_GSS_NEGOTIATE,
|
||
|
|
NULL,
|
||
|
|
RPC_C_AUTHZ_NONE);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (pszServerPrincipalName)
|
||
|
|
RpcStringFree(&pszServerPrincipalName);
|
||
|
|
|
||
|
|
if ( bBound && RpcStatus != RPC_S_OK)
|
||
|
|
{
|
||
|
|
STRING_HANDLE_unbind(lpStr, hBinding);
|
||
|
|
hBinding = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
return (hBinding);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
VOID
|
||
|
|
STRING_HANDLE_unbind(
|
||
|
|
STRING_HANDLE lpStr,
|
||
|
|
handle_t hBinding)
|
||
|
|
{
|
||
|
|
RPC_STATUS RpcStatus = 0;
|
||
|
|
|
||
|
|
RpcStatus = RpcBindingFree(&hBinding);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
handle_t
|
||
|
|
WZC_DBLOG_SESSION_HANDLE_bind(
|
||
|
|
WZC_DBLOG_SESSION_HANDLE hSession,
|
||
|
|
STRING_HANDLE lpStr
|
||
|
|
)
|
||
|
|
{
|
||
|
|
RPC_STATUS RpcStatus = 0;
|
||
|
|
LPWSTR strSrv = NULL;
|
||
|
|
LPWSTR pszStringBinding = NULL;
|
||
|
|
handle_t hBinding = NULL;
|
||
|
|
|
||
|
|
|
||
|
|
strSrv = (!lpStr || !*lpStr || !lstrcmp(L"127.0.0.1", lpStr)) ? NULL : lpStr;
|
||
|
|
RpcStatus = RpcStringBindingComposeW(
|
||
|
|
0,
|
||
|
|
(strSrv==NULL) ? L"ncalrpc" : L"ncacn_np",
|
||
|
|
strSrv,
|
||
|
|
(strSrv==NULL) ? L"wzcsvc" : L"\\pipe\\wzcsvc",
|
||
|
|
pszStrBindingOptions,
|
||
|
|
&pszStringBinding
|
||
|
|
);
|
||
|
|
|
||
|
|
if (RpcStatus != RPC_S_OK) {
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
RpcStatus = RpcBindingFromStringBindingW(
|
||
|
|
pszStringBinding,
|
||
|
|
&hBinding
|
||
|
|
);
|
||
|
|
RpcStringFreeW(&pszStringBinding);
|
||
|
|
|
||
|
|
if (RpcStatus != RPC_S_OK) {
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
return (hBinding);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
VOID
|
||
|
|
WZC_DBLOG_SESSION_HANDLE_unbind(
|
||
|
|
WZC_DBLOG_SESSION_HANDLE hSession,
|
||
|
|
handle_t hBinding
|
||
|
|
)
|
||
|
|
{
|
||
|
|
RPC_STATUS RpcStatus = 0;
|
||
|
|
|
||
|
|
RpcStatus = RpcBindingFree(&hBinding);
|
||
|
|
|
||
|
|
ASSERT(RpcStatus == RPC_S_OK);
|
||
|
|
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|