windows-nt/Source/XPSP1/NT/admin/wmi/wbem/winmgmt/wbemcomn/callsec.cpp
2020-09-26 16:20:57 +08:00

1003 lines
24 KiB
C++

/*++
Copyright (C) 1998-2001 Microsoft Corporation
Module Name:
CALLSEC.CPP
Abstract:
IWbemCallSecurity, IServerSecurity implementation for
provider impersonation.
History:
raymcc 29-Jul-98 First draft.
--*/
#include "precomp.h"
#include <stdio.h>
#include <initguid.h>
#include <winntsec.h>
#include <callsec.h>
#include <cominit.h>
#include <arrtempl.h>
#include <cominit.h>
#include <genutils.h>
//***************************************************************************
//
// CWbemCallSecurity
//
// This object is used to supply client impersonation to providers.
//
// Usage:
// (1) When client first makes call, call CreateInst() and get a new
// empty object (ref count of 1). Constructors/Destructors are private.
//
//***************************************************************************
// ok
CWbemCallSecurity::CWbemCallSecurity()
{
OSVERSIONINFO os;
os.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
GetVersionEx(&os);
if (os.dwPlatformId != VER_PLATFORM_WIN32_NT)
m_bWin9x = TRUE;
else
m_bWin9x = FALSE;
m_lRef = 1; // Ref count
m_hThreadToken = 0; // Handle to thread imp token
m_dwPotentialImpLevel = 0; // Potential
m_dwActiveImpLevel = 0; // Active impersonation
m_dwAuthnSvc = 0;
m_dwAuthzSvc = 0;
m_dwAuthnLevel = 0;
m_pServerPrincNam = 0;
m_pIdentity = 0;
}
//***************************************************************************
//
// ~CWbemCallSecurity
//
// Destructor. Closes any open handles, deallocates any non-NULL
// strings.
//
//***************************************************************************
// ok
CWbemCallSecurity::~CWbemCallSecurity()
{
if (m_hThreadToken)
CloseHandle(m_hThreadToken);
if (m_pServerPrincNam)
CoTaskMemFree(m_pServerPrincNam);
if (m_pIdentity)
CoTaskMemFree(m_pIdentity);
}
void
CWbemCallSecurity::operator=(const CWbemCallSecurity& Other)
{
if(m_hThreadToken)
{
CloseHandle(m_hThreadToken);
m_hThreadToken=NULL ;
}
if ( Other.m_hThreadToken )
{
DuplicateHandle(
GetCurrentProcess(),
Other.m_hThreadToken,
GetCurrentProcess(),
&m_hThreadToken,
0,
TRUE,
DUPLICATE_SAME_ACCESS);
}
m_dwPotentialImpLevel = Other.m_dwPotentialImpLevel;
m_dwActiveImpLevel = 0;
m_dwAuthnSvc = Other.m_dwAuthnSvc;
m_dwAuthzSvc = Other.m_dwAuthzSvc;
m_dwAuthnLevel = Other.m_dwAuthnLevel;
if(m_pServerPrincNam)
{
CoTaskMemFree(m_pServerPrincNam);
m_pServerPrincNam = NULL;
}
if (Other.m_pServerPrincNam)
{
m_pServerPrincNam = (LPWSTR)CoTaskMemAlloc(
(wcslen(Other.m_pServerPrincNam) + 1) * 2);
if(m_pServerPrincNam)
wcscpy(m_pServerPrincNam, Other.m_pServerPrincNam);
}
if(m_pIdentity)
{
CoTaskMemFree(m_pIdentity);
m_pIdentity = NULL;
}
if (Other.m_pIdentity)
{
m_pIdentity = (LPWSTR)CoTaskMemAlloc(
(wcslen(Other.m_pIdentity) + 1) * 2);
if(m_pIdentity)
wcscpy(m_pIdentity, Other.m_pIdentity);
}
}
//***************************************************************************
//
// CWbemCallSecurity::AddRef
//
//***************************************************************************
// ok
ULONG CWbemCallSecurity::AddRef()
{
return InterlockedIncrement(&m_lRef);
}
//***************************************************************************
//
// CWbemCallSecurity::Release
//
//***************************************************************************
// ok
ULONG CWbemCallSecurity::Release()
{
long lRef = InterlockedDecrement(&m_lRef);
if(lRef == 0)
delete this;
return lRef;
}
//***************************************************************************
//
// CWbemCallSecurity::QueryInterface
//
//***************************************************************************
// ok
HRESULT CWbemCallSecurity::QueryInterface(REFIID riid, void** ppv)
{
if(riid == IID_IUnknown)
{
*ppv = (IUnknown *) this;
AddRef();
return S_OK;
}
else if (riid == IID_IServerSecurity)
{
*ppv = (IServerSecurity *) this;
AddRef();
return S_OK;
}
else if (riid == IID_IWbemCallSecurity)
{
*ppv = (IWbemCallSecurity *) this;
AddRef();
return S_OK;
}
else return E_NOINTERFACE;
}
//***************************************************************************
//
// CWbemCallSecurity:QueryBlanket
//
//***************************************************************************
// ok
HRESULT STDMETHODCALLTYPE CWbemCallSecurity::QueryBlanket(
/* [out] */ DWORD __RPC_FAR *pAuthnSvc,
/* [out] */ DWORD __RPC_FAR *pAuthzSvc,
/* [out] */ OLECHAR __RPC_FAR *__RPC_FAR *pServerPrincName,
/* [out] */ DWORD __RPC_FAR *pAuthnLevel,
/* [out] */ DWORD __RPC_FAR *pImpLevel,
/* [out] */ void __RPC_FAR *__RPC_FAR *pPrivs,
/* [out] */ DWORD __RPC_FAR *pCapabilities
)
{
if (m_bWin9x)
return E_NOTIMPL;
if (m_dwPotentialImpLevel == 0 )
return E_FAIL;
// Return DWORD parameters, after checking.
// ========================================
if (pAuthnSvc)
*pAuthnSvc = m_dwAuthnSvc;
if (pAuthzSvc)
*pAuthzSvc = m_dwAuthzSvc ;
if (pImpLevel)
*pImpLevel = m_dwActiveImpLevel ;
if (pAuthnLevel)
*pAuthnLevel = m_dwAuthnLevel;
if (pServerPrincName)
{
*pServerPrincName = 0;
if (m_pServerPrincNam)
{
*pServerPrincName = (LPWSTR) CoTaskMemAlloc((wcslen(m_pServerPrincNam) + 1) * 2);
wcscpy(*pServerPrincName, m_pServerPrincNam);
}
}
if (pPrivs)
{
*pPrivs = m_pIdentity; // Documented to point to an internal!!
}
return S_OK;
}
//***************************************************************************
//
// CWbemCallSecurity::ImpersonateClient
//
//***************************************************************************
// ok
HRESULT STDMETHODCALLTYPE CWbemCallSecurity::ImpersonateClient(void)
{
if (m_bWin9x)
return E_NOTIMPL;
if (m_dwActiveImpLevel != 0) // Already impersonating
return S_OK;
if(m_hThreadToken == NULL)
{
return WBEM_E_INVALID_CONTEXT;
}
if (m_dwPotentialImpLevel == 0)
return (ERROR_CANT_OPEN_ANONYMOUS | 0x80070000);
BOOL bRes;
bRes = SetThreadToken(NULL, m_hThreadToken);
if (bRes)
{
m_dwActiveImpLevel = m_dwPotentialImpLevel;
return S_OK;
}
return E_FAIL;
}
//***************************************************************************
//
// CWbemCallSecurity::RevertToSelf
//
// Returns S_OK or E_FAIL.
// Returns E_NOTIMPL on Win9x platforms.
//
//***************************************************************************
// ok
HRESULT STDMETHODCALLTYPE CWbemCallSecurity::RevertToSelf( void)
{
if (m_bWin9x)
return E_NOTIMPL;
if (m_dwActiveImpLevel == 0)
return S_OK;
if (m_dwPotentialImpLevel == 0)
return (ERROR_CANT_OPEN_ANONYMOUS | 0x80070000);
// If here,we are impersonating and can definitely revert.
// =======================================================
BOOL bRes = SetThreadToken(NULL, NULL);
if (bRes == FALSE)
return E_FAIL;
m_dwActiveImpLevel = 0; // No longer actively impersonating
return S_OK;
}
//***************************************************************************
//
// CWbemCallSecurity::IsImpersonating
//
//***************************************************************************
BOOL STDMETHODCALLTYPE CWbemCallSecurity::IsImpersonating( void)
{
if (m_hThreadToken && m_dwActiveImpLevel != 0)
return TRUE;
return FALSE;
}
//***************************************************************************
//
// CWbemCallSecurity::CreateInst
//
// Creates a new instance
//***************************************************************************
// ok
IWbemCallSecurity * CWbemCallSecurity::CreateInst()
{
return (IWbemCallSecurity *) new CWbemCallSecurity; // Constructed with ref count of 1
}
//***************************************************************************
//
// CWbemCallSecurity::GetPotentialImpersonation
//
// Returns 0 if no impersonation is currently possible or the
// level which would be active during impersonation:
//
// RPC_C_IMP_LEVEL_ANONYMOUS
// RPC_C_IMP_LEVEL_IDENTIFY
// RPC_C_IMP_LEVEL_IMPERSONATE
// RPC_C_IMP_LEVEL_DELEGATE
//
//***************************************************************************
// ok
HRESULT CWbemCallSecurity::GetPotentialImpersonation()
{
return m_dwPotentialImpLevel;
}
//***************************************************************************
//
// CWbemCallSecurity::GetActiveImpersonation
//
// Returns 0 if no impersonation is currently active or the
// currently active level:
//
// RPC_C_IMP_LEVEL_ANONYMOUS
// RPC_C_IMP_LEVEL_IDENTIFY
// RPC_C_IMP_LEVEL_IMPERSONATE
// RPC_C_IMP_LEVEL_DELEGATE
//
//***************************************************************************
// ok
HRESULT CWbemCallSecurity::GetActiveImpersonation()
{
return m_dwActiveImpLevel;
}
//***************************************************************************
//
// CWbemCallSecurity::CloneThreadContext
//
// Call this on a thread to retrieve the potential impersonation info for
// that thread and set the current object to be able to duplicate it later.
//
// Return codes:
//
// S_OK
// E_FAIL
// E_NOTIMPL on Win9x
// E_ABORT if the calling thread is already impersonating a client.
//
//***************************************************************************
HRESULT CWbemCallSecurity::CloneThreadContext(BOOL bInternallyIssued)
{
// If on Win9x, don't bother.
// ==========================
if (m_bWin9x)
return E_NOTIMPL;
if (m_hThreadToken) // Already called this
return E_ABORT;
// Get the current context.
// ========================
IServerSecurity *pSec = 0;
HRESULT hRes = WbemCoGetCallContext(IID_IServerSecurity, (LPVOID *) &pSec);
CReleaseMe rmSec(pSec);
if (hRes != S_OK)
{
// There is no call context --- this must be an in-proc object calling
// us from its own thread. Initialize from current thread token
// ===================================================================
return CloneThreadToken();
}
// Figure out if the call context is ours or RPCs
// ==============================================
IWbemCallSecurity* pInternal = NULL;
if(SUCCEEDED(pSec->QueryInterface(IID_IWbemCallSecurity,
(void**)&pInternal)))
{
CReleaseMe rmInt(pInternal);
// This is our own call context --- this must be ab in-proc object
// calling us from our thread. Behave depending on the flags
// ===============================================================
if(bInternallyIssued)
{
// Internal requests always propagate context. Therefore, we just
// copy the context that we have got
// ==============================================================
*this = *(CWbemCallSecurity*)pInternal;
return S_OK;
}
else
{
// Provider request --- Initialize from the current thread token
// =============================================================
return CloneThreadToken();
}
}
// If here, we are not impersonating and we want to gather info
// about the client's call.
// ============================================================
RPC_AUTHZ_HANDLE hAuth;
// Ensures auto release of the mutex if we crash
CAutoSecurityMutex autosm;
DWORD t_ImpLevel = 0 ;
hRes = pSec->QueryBlanket(
&m_dwAuthnSvc,
&m_dwAuthzSvc,
&m_pServerPrincNam,
&m_dwAuthnLevel,
&t_ImpLevel,
&hAuth, // RPC_AUTHZ_HANDLE
NULL // Capabilities; not used
);
if(FAILED(hRes))
{
// In some cases, we cant get the name, but the rest is ok. In particular
// the temporary SMS accounts have that property. Or nt 4 after IPCONFIG /RELEASE
hRes = pSec->QueryBlanket(
&m_dwAuthnSvc,
&m_dwAuthzSvc,
&m_pServerPrincNam,
&m_dwAuthnLevel,
&t_ImpLevel,
NULL, // RPC_AUTHZ_HANDLE
NULL // Capabilities; not used
);
hAuth = NULL;
}
// We don't need this anymore.
autosm.Release();
if(FAILED(hRes))
{
// THIS IS A WORKAROUND FOR COM BUG:
// This failure is indicative of an anonymous-level client.
// ========================================================
m_dwPotentialImpLevel = 0;
return S_OK;
}
if (hAuth)
{
m_pIdentity = LPWSTR(CoTaskMemAlloc((wcslen(LPWSTR(hAuth)) + 1) * 2));
if(m_pIdentity)
wcscpy(m_pIdentity, LPWSTR(hAuth));
}
// Impersonate the client long enough to clone the thread token.
// =============================================================
BOOL bImp = pSec->IsImpersonating();
if(!bImp)
hRes = pSec->ImpersonateClient();
if (FAILED(hRes))
{
if(!bImp)
pSec->RevertToSelf();
return E_FAIL;
}
HRESULT hres = CloneThreadToken();
if(!bImp)
pSec->RevertToSelf();
return hres;
}
HRESULT CWbemCallSecurity::CloneThreadToken()
{
HANDLE hPrimary = 0 ;
HANDLE hToken = 0;
BOOL bRes = OpenThreadToken(
GetCurrentThread(),
TOKEN_QUERY | TOKEN_DUPLICATE | TOKEN_IMPERSONATE,
TRUE,
&hToken
);
if (bRes == FALSE)
{
m_hThreadToken = NULL;
m_dwAuthnSvc = RPC_C_AUTHN_WINNT;
m_dwAuthzSvc = RPC_C_AUTHZ_NONE;
m_dwAuthnLevel = RPC_C_AUTHN_LEVEL_PKT_PRIVACY;
m_pServerPrincNam = NULL;
m_pIdentity = NULL;
long lRes = GetLastError();
if(lRes == ERROR_NO_IMPERSONATION_TOKEN || lRes == ERROR_NO_TOKEN)
{
// This is the basic process thread.
// =================================
bRes = OpenProcessToken(
GetCurrentProcess(),
TOKEN_QUERY | TOKEN_DUPLICATE | TOKEN_IMPERSONATE,
&hPrimary
);
if (bRes==FALSE)
{
// Unknown error
// =============
m_dwPotentialImpLevel = 0;
return E_FAIL;
}
}
else if(lRes == ERROR_CANT_OPEN_ANONYMOUS)
{
// Anonymous call
// ==============
m_dwPotentialImpLevel = 0;
return S_OK;
}
else
{
// Unknown error
// =============
m_dwPotentialImpLevel = 0;
return E_FAIL;
}
}
// Find out token info.
// =====================
SECURITY_IMPERSONATION_LEVEL t_Level = SecurityImpersonation ;
if ( hToken )
{
DWORD dwBytesReturned = 0;
bRes = GetTokenInformation (
hToken,
TokenImpersonationLevel,
( void * ) & t_Level,
sizeof ( t_Level ),
&dwBytesReturned
);
if (bRes == FALSE)
{
CloseHandle(hToken);
return E_FAIL;
}
}
switch (t_Level)
{
case SecurityAnonymous:
m_dwPotentialImpLevel = RPC_C_IMP_LEVEL_ANONYMOUS;
break;
case SecurityIdentification:
m_dwPotentialImpLevel = RPC_C_IMP_LEVEL_IDENTIFY;
break;
case SecurityImpersonation:
m_dwPotentialImpLevel = RPC_C_IMP_LEVEL_IMPERSONATE;
break;
case SecurityDelegation:
m_dwPotentialImpLevel = RPC_C_IMP_LEVEL_DELEGATE;
break;
default:
m_dwPotentialImpLevel = 0;
break;
}
// Duplicate the handle.
// ============================
bRes = DuplicateToken (
hToken ? hToken : hPrimary ,
(SECURITY_IMPERSONATION_LEVEL)t_Level,
&m_hThreadToken
);
if ( hToken )
{
CloseHandle(hToken);
}
else
{
CloseHandle(hPrimary);
}
if (bRes == FALSE)
return E_FAIL;
return S_OK;
}
RELEASE_ME CWbemCallSecurity* CWbemCallSecurity::MakeInternalCopyOfThread()
{
IServerSecurity* pSec;
HRESULT hres = WbemCoGetCallContext(IID_IServerSecurity, (void**)&pSec);
if(FAILED(hres))
return NULL;
CReleaseMe rm1(pSec);
IServerSecurity* pIntSec;
hres = pSec->QueryInterface(IID_IWbemCallSecurity, (void**)&pIntSec);
if(FAILED(hres))
return NULL;
CWbemCallSecurity* pCopy = new CWbemCallSecurity;
if (pCopy)
*pCopy = *(CWbemCallSecurity*)pIntSec;
pIntSec->Release();
return pCopy;
}
DWORD CWbemCallSecurity::GetAuthenticationId(LUID& rluid)
{
if(m_hThreadToken == NULL)
return ERROR_INVALID_HANDLE;
TOKEN_STATISTICS stat;
DWORD dwRet;
if(!GetTokenInformation(m_hThreadToken, TokenStatistics,
(void*)&stat, sizeof(stat), &dwRet))
{
return GetLastError();
}
rluid = stat.AuthenticationId;
return 0;
}
HANDLE CWbemCallSecurity::GetToken()
{
return m_hThreadToken;
}
//***************************************************************************
//
//***************************************************************************
void CWbemCallSecurity::DumpCurrentContext()
{
if (m_bWin9x)
return;
HANDLE hThreadToken;
BOOL bRes = OpenThreadToken(
GetCurrentThread(),
TOKEN_QUERY | TOKEN_DUPLICATE | TOKEN_IMPERSONATE,
TRUE,
&hThreadToken
);
if (!bRes)
return;
const DWORD dwBytes = 512;
BYTE Buf[dwBytes], Name[dwBytes], Domain[dwBytes];
DWORD dwBytesReturned = 0, dwDomainSize = dwBytes, dwNameSize = dwBytes;
GetTokenInformation(hThreadToken, TokenUser, Buf, dwBytes, &dwBytesReturned);
SID_NAME_USE snu;
SetThreadToken(NULL, NULL);
bRes = LookupAccountSidW(NULL, ((PTOKEN_USER) Buf)->User.Sid, LPWSTR(Name), &dwNameSize,
LPWSTR(Domain), &dwDomainSize, &snu);
SetThreadToken(NULL, hThreadToken);
printf("User = %S\\%S\n", Domain, Name);
CloseHandle(hThreadToken);
}
CDerivedObjectSecurity::CDerivedObjectSecurity()
{
m_bValid = FALSE;
m_bEnabled = TRUE;
if(!IsNT() || !IsDcomEnabled())
{
m_bEnabled = FALSE;
m_bValid = TRUE;
}
HRESULT hres = RetrieveSidFromCall(&m_sidUser);
if(FAILED(hres))
return;
// Revert to self while remebering the thread token
// ================================================
BOOL bRes;
HANDLE hThreadToken = NULL;
bRes = OpenThreadToken(GetCurrentThread(), TOKEN_READ | TOKEN_IMPERSONATE,
TRUE, &hThreadToken);
if(bRes)
SetThreadToken(NULL, NULL);
// Open process token
// ==================
HANDLE hProcessToken;
bRes = OpenProcessToken(GetCurrentProcess(), TOKEN_READ,
&hProcessToken);
if(!bRes)
{
if(hThreadToken)
CloseHandle(hThreadToken);
return;
}
// Retrieve our process SID
// ========================
hres = RetrieveSidFromToken(hProcessToken, &m_sidSystem);
CloseHandle(hProcessToken);
// Re-impersonate the original thread token
// ========================================
if(hThreadToken)
{
SetThreadToken(NULL, hThreadToken);
CloseHandle(hThreadToken);
}
if(FAILED(hres))
return;
m_bValid = TRUE;
}
HRESULT CDerivedObjectSecurity::RetrieveSidFromCall(CNtSid* psid)
{
HANDLE hToken;
HRESULT hres;
BOOL bRes;
// Check if we are on an impersonated thread
// =========================================
bRes = OpenThreadToken(GetCurrentThread(), TOKEN_READ, TRUE, &hToken);
if(bRes)
{
// We are --- just use this token for authentication
// =================================================
hres = RetrieveSidFromToken(hToken, psid);
CloseHandle(hToken);
return hres;
}
// Construct CWbemCallSecurity that will determine (according to our
// non-trivial provider handling rules) the security context of this
// call
// =================================================================
IWbemCallSecurity* pServerSec = CWbemCallSecurity::CreateInst();
if(pServerSec == NULL)
return WBEM_E_OUT_OF_MEMORY;
CReleaseMe rm1(pServerSec);
hres = pServerSec->CloneThreadContext(FALSE);
if(FAILED(hres))
return hres;
// Impersonate client
// ==================
hres = pServerSec->ImpersonateClient();
if(FAILED(hres))
return hres;
// Open impersonated token
// =======================
bRes = OpenThreadToken(GetCurrentThread(), TOKEN_READ, TRUE, &hToken);
if(!bRes)
{
long lRes = GetLastError();
if(lRes == ERROR_NO_IMPERSONATION_TOKEN || lRes == ERROR_NO_TOKEN)
{
// Not impersonating --- get the process token instead
// ===================================================
bRes = OpenProcessToken(GetCurrentProcess(), TOKEN_READ, &hToken);
if(!bRes)
{
pServerSec->RevertToSelf();
return WBEM_E_ACCESS_DENIED;
}
}
else
{
// Real problems
// =============
pServerSec->RevertToSelf();
return WBEM_E_ACCESS_DENIED;
}
}
hres = RetrieveSidFromToken(hToken, psid);
CloseHandle(hToken);
pServerSec->RevertToSelf();
return hres;
}
HRESULT CDerivedObjectSecurity::RetrieveSidFromToken(HANDLE hToken,
CNtSid* psid)
{
// Retrieve the length of the user sid structure
// =============================================
BOOL bRes;
DWORD dwLen = 0;
TOKEN_USER tu;
memset(&tu,0,sizeof(TOKEN_USER));
bRes = GetTokenInformation(hToken, TokenUser, &tu, sizeof(TOKEN_USER), &dwLen);
DWORD dwLast = GetLastError();
if(bRes)
{
// 1-length sid? I don't think so
return WBEM_E_CRITICAL_ERROR;
}
// Allocate buffer to hold user sid
// ================================
BYTE* pBuffer = new BYTE[dwLen];
if(pBuffer == NULL)
return WBEM_E_OUT_OF_MEMORY;
CVectorDeleteMe<BYTE> dm1(pBuffer);
// Retrieve the user sid structure
// ===============================
bRes = GetTokenInformation(hToken, TokenUser, pBuffer, dwLen, &dwLen);
if(!bRes)
return WBEM_E_ACCESS_DENIED;
TOKEN_USER* pUser = (TOKEN_USER*)pBuffer;
// Set our sid to the returned one
// ===============================
*psid = CNtSid(pUser->User.Sid);
return WBEM_S_NO_ERROR;
}
CDerivedObjectSecurity::~CDerivedObjectSecurity()
{
}
BOOL CDerivedObjectSecurity::AccessCheck()
{
if(!m_bValid)
return FALSE;
if(!m_bEnabled)
return TRUE;
// Find out who is calling
// =======================
CNtSid sidCaller;
HRESULT hres = RetrieveSidFromCall(&sidCaller);
if(FAILED(hres))
return FALSE;
// Compare the caller to the issuing user and ourselves
// ====================================================
if(sidCaller == m_sidUser || sidCaller == m_sidSystem)
return TRUE;
else
return FALSE;
}