archive
/
windows-xp
mirror of https://github.com/tongzx/nt5src
2
0
Fork 0
Code Issues Projects Releases Wiki Activity
Source code of Windows XP (NT5)
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
8 Commits
1 Branch
0 Tags
2.0 GiB
 
 
 
 
 
 
Tree: 744f0b4006
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '744f0b4006'
${ noResults }
windows-xp/Source/XPSP1/NT/ds/security/gina/snapins/fde/rsoputil.cpp

475 lines
10 KiB
Raw Blame History

//+--------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1994 - 1997.
//
// File: rsoputil.cpp
//
// Contents: helper functions for working with the RSOP databases
//
// History: 10-18-1999 stevebl Created
//
//---------------------------------------------------------------------------
#include "precomp.hxx"
#include <wbemcli.h>
#include "rsoputil.h"
// Something in the build environment is causing this
// to be #defined to a version before 0x0500 and
// that causes the ConvertStringSecurityDescriptor... function
// to be undefined.
#undef _WIN32_WINNT
#define _WIN32_WINNT 0x0500
#include "sddl.h"
//+--------------------------------------------------------------------------
//
// Function: SetParameter
//
// Synopsis: sets a paramter's value in a WMI parameter list
//
// Arguments: [pInst] - instance on which to set the value
// [szParam] - the name of the parameter
// [xData] - the data
//
// History: 10-08-1999 stevebl Created
//
// Notes: There may be several flavors of this procedure, one for
// each data type.
//
//---------------------------------------------------------------------------
HRESULT SetParameter(IWbemClassObject * pInst, TCHAR * szParam, TCHAR * szData)
{
VARIANT var;
HRESULT hr = S_OK;
var.vt = VT_BSTR;
var.bstrVal = SysAllocString(szData);
hr = pInst->Put(szParam, 0, &var, 0);
SysFreeString(var.bstrVal);
return hr;
}
//+--------------------------------------------------------------------------
//
// Function: GetParameter
//
// Synopsis: retrieves a parameter value from a WMI paramter list
//
// Arguments: [pInst] - instance to get the paramter value from
// [szParam] - the name of the paramter
// [xData] - [out] data
//
// History: 10-08-1999 stevebl Created
//
// Notes: There are several flavors of this procedure, one for each
// data type.
//
//---------------------------------------------------------------------------
HRESULT GetParameter(IWbemClassObject * pInst, TCHAR * szParam, TCHAR * &szData)
{
VARIANT var;
HRESULT hr = S_OK;
VariantInit(&var);
if (szData)
{
delete szData;
szData = NULL;
}
hr = pInst->Get(szParam, 0, &var, 0, 0);
if (SUCCEEDED(hr) && var.vt != VT_NULL)
{
if (var.bstrVal)
{
szData = (TCHAR *) OLEALLOC(sizeof(TCHAR) * (_tcslen(var.bstrVal) + 1));
if (szData)
{
_tcscpy(szData, var.bstrVal);
}
else
{
hr = E_OUTOFMEMORY;
}
}
}
VariantClear(&var);
return hr;
}
HRESULT GetParameter(IWbemClassObject * pInst, TCHAR * szParam, CString &szData)
{
VARIANT var;
HRESULT hr = S_OK;
VariantInit(&var);
if (szData)
{
szData = "";
}
hr = pInst->Get(szParam, 0, &var, 0, 0);
if (SUCCEEDED(hr) && var.vt != VT_NULL)
{
if (var.bstrVal)
{
szData = var.bstrVal;
}
}
VariantClear(&var);
return hr;
}
HRESULT GetParameterBSTR(IWbemClassObject * pInst, TCHAR * szParam, BSTR &bstrData)
{
VARIANT var;
HRESULT hr = S_OK;
VariantInit(&var);
if (bstrData)
{
SysFreeString(bstrData);
}
hr = pInst->Get(szParam, 0, &var, 0, 0);
if (SUCCEEDED(hr) && var.vt != VT_NULL)
{
bstrData = SysAllocStringLen(var.bstrVal, SysStringLen(var.bstrVal));
if (NULL == bstrData)
{
hr = E_OUTOFMEMORY;
}
}
VariantClear(&var);
return hr;
}
HRESULT GetParameter(IWbemClassObject * pInst, TCHAR * szParam, BOOL &fData)
{
VARIANT var;
HRESULT hr = S_OK;
VariantInit(&var);
hr = pInst->Get(szParam, 0, &var, 0, 0);
if (SUCCEEDED(hr) && var.vt != VT_NULL)
{
fData = var.bVal;
}
VariantClear(&var);
return hr;
}
HRESULT GetParameter(IWbemClassObject * pInst, TCHAR * szParam, HRESULT &hrData)
{
VARIANT var;
HRESULT hr = S_OK;
VariantInit(&var);
hr = pInst->Get(szParam, 0, &var, 0, 0);
if (SUCCEEDED(hr) && var.vt != VT_NULL)
{
hrData = (HRESULT) var.lVal;
}
VariantClear(&var);
return hr;
}
HRESULT GetParameter(IWbemClassObject * pInst, TCHAR * szParam, ULONG &ulData)
{
VARIANT var;
HRESULT hr = S_OK;
VariantInit(&var);
hr = pInst->Get(szParam, 0, &var, 0, 0);
if (SUCCEEDED(hr) && var.vt != VT_NULL)
{
ulData = var.ulVal;
}
VariantClear(&var);
return hr;
}
HRESULT GetParameter(IWbemClassObject * pInst, TCHAR * szParam, GUID &guid)
{
TCHAR * sz = NULL;
memset(&guid, 0, sizeof(GUID));
HRESULT hr = GetParameter(pInst, szParam, sz);
if (SUCCEEDED(hr))
{
hr = CLSIDFromString(sz, &guid);
}
if (sz)
{
OLESAFE_DELETE(sz);
}
return hr;
}
HRESULT GetParameter(IWbemClassObject * pInst, TCHAR * szParam, unsigned int &ui)
{
VARIANT var;
HRESULT hr = S_OK;
VariantInit(&var);
ui = 0;
hr = pInst->Get(szParam, 0, &var, 0, 0);
if (SUCCEEDED(hr) && var.vt != VT_NULL)
{
ui = (HRESULT) var.uiVal;
}
VariantClear(&var);
return hr;
}
// array variation - gets an array of guids and a count
HRESULT GetParameter(IWbemClassObject * pInst, TCHAR *szParam, UINT &uiCount, GUID * &rgGuid)
{
VARIANT var;
HRESULT hr = S_OK;
VariantInit(&var);
hr = pInst->Get(szParam, 0, &var, 0, 0);
if (SUCCEEDED(hr) && var.vt == (VT_ARRAY | VT_BSTR))
{
// build the array
SAFEARRAY * parray = var.parray;
uiCount = parray->rgsabound[0].cElements;
if (uiCount > 0)
{
rgGuid = (GUID *)OLEALLOC(sizeof(GUID) * uiCount);
if (rgGuid)
{
BSTR * rgData = (BSTR *)parray->pvData;
UINT ui = uiCount;
while (ui--)
{
hr = CLSIDFromString(rgData[ui], &rgGuid[ui]);
if (FAILED(hr))
{
return hr;
}
}
}
else
{
uiCount = 0;
hr = E_OUTOFMEMORY;
}
}
}
VariantClear(&var);
return hr;
}
// array variation - gets an array of strings and a count
HRESULT GetParameter(IWbemClassObject * pInst, TCHAR *szParam, UINT &uiCount, TCHAR ** &rgszData)
{
VARIANT var;
HRESULT hr = S_OK;
VariantInit(&var);
hr = pInst->Get(szParam, 0, &var, 0, 0);
if (SUCCEEDED(hr) && var.vt == (VT_ARRAY | VT_BSTR))
{
// build the array
SAFEARRAY * parray = var.parray;
uiCount = parray->rgsabound[0].cElements;
if (uiCount > 0)
{
rgszData = (TCHAR **)OLEALLOC(sizeof(TCHAR *) * uiCount);
if (rgszData)
{
BSTR * rgData = (BSTR *)parray->pvData;
UINT ui = uiCount;
while (ui--)
{
OLESAFE_COPYSTRING(rgszData[ui], rgData[ui]);
}
}
else
{
uiCount = 0;
hr = E_OUTOFMEMORY;
}
}
}
VariantClear(&var);
return hr;
}
HRESULT GetParameter(IWbemClassObject * pInst, TCHAR * szParam, PSECURITY_DESCRIPTOR &psd)
{
VARIANT var;
HRESULT hr = S_OK;
if (psd)
{
LocalFree(psd);
psd = NULL;
}
VariantInit(&var);
hr = pInst->Get(szParam, 0, &var, 0, 0);
if (SUCCEEDED(hr) && var.vt != VT_NULL)
{
PSECURITY_DESCRIPTOR psdTemp;
BOOL f = ConvertStringSecurityDescriptorToSecurityDescriptor(
var.bstrVal,
//(LPCTSTR)var.parray->pvData,
SDDL_REVISION_1,
&psd,
NULL);
if (!f)
{
// it failed
// could call GetLastError here to figure out why
// but at the moment I don't really care
}
if (!IsValidSecurityDescriptor(psd))
{
LocalFree(psd);
psd = NULL;
}
}
VariantClear(&var);
return hr;
}
HRESULT GetGPOFriendlyName(IWbemServices *pIWbemServices,
LPTSTR lpGPOID, BSTR pLanguage,
LPTSTR *pGPOName)
{
BSTR pQuery = NULL, pName = NULL;
LPTSTR lpQuery = NULL;
IEnumWbemClassObject * pEnum = NULL;
IWbemClassObject *pObjects[2];
HRESULT hr;
ULONG ulRet;
VARIANT varGPOName;
//
// Set the default
//
*pGPOName = NULL;
//
// Build the query
//
lpQuery = (LPTSTR) LocalAlloc (LPTR, ((lstrlen(lpGPOID) + 50) * sizeof(TCHAR)));
if (!lpQuery)
{
hr = HRESULT_FROM_WIN32(ERROR_OUTOFMEMORY);
goto Exit;
}
wsprintf (lpQuery, TEXT("SELECT name, id FROM RSOP_GPO where id=\"%s\""), lpGPOID);
pQuery = SysAllocString (lpQuery);
if (!pQuery)
{
hr = HRESULT_FROM_WIN32(ERROR_OUTOFMEMORY);
goto Exit;
}
//
// Allocate BSTRs for the property names we want to retreive
//
pName = SysAllocString (TEXT("name"));
if (!pName)
{
hr = HRESULT_FROM_WIN32(ERROR_OUTOFMEMORY);
goto Exit;
}
//
// Execute the query
//
hr = pIWbemServices->ExecQuery (pLanguage, pQuery,
WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY,
NULL, &pEnum);
if (FAILED(hr))
{
goto Exit;
}
//
// Loop through the results
//
hr = pEnum->Next(WBEM_INFINITE, 1, pObjects, &ulRet);
if (FAILED(hr))
{
goto Exit;
}
//
// Check for the "data not available case"
//
if (ulRet == 0)
{
hr = S_OK;
goto Exit;
}
//
// Get the name
//
hr = pObjects[0]->Get (pName, 0, &varGPOName, NULL, NULL);
if (FAILED(hr))
{
goto Exit;
}
//
// Save the name
//
*pGPOName = (LPTSTR) LocalAlloc (LPTR, (lstrlen(varGPOName.bstrVal) + 1) * sizeof(TCHAR));
if (!(*pGPOName))
{
goto Exit;
}
lstrcpy (*pGPOName, varGPOName.bstrVal);
VariantClear (&varGPOName);
hr = S_OK;
Exit:
if (pEnum)
{
pEnum->Release();
}
if (pQuery)
{
SysFreeString (pQuery);
}
if (lpQuery)
{
LocalFree (lpQuery);
}
if (pName)
{
SysFreeString (pName);
}
return hr;
}