windows-nt/Source/XPSP1/NT/ds/adsi/winnt/cggi.cxx
2020-09-26 16:20:57 +08:00

706 lines
15 KiB
C++

//---------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1992 - 1995
//
// File: cggi.cxx
//
// Contents: This file contains the Group Object's
// IADsGroup and IADsGroupOperation methods
//
// History: 11-1-95 krishnag Created.
//
//----------------------------------------------------------------------------
#include "winnt.hxx"
#pragma hdrstop
BOOL
VerifyIfMember(
BSTR bstrMember,
VARIANT * VariantArray,
ULONG cElementFetched
);
BOOL
IsStringSID(LPWSTR pszStringSID);
// Class CWinNTGroup
STDMETHODIMP CWinNTGroup::get_Description(THIS_ BSTR FAR* retval)
{
GET_PROPERTY_BSTR((IADsGroup *)this,Description);
}
STDMETHODIMP CWinNTGroup::put_Description(THIS_ BSTR bstrDescription)
{
PUT_PROPERTY_BSTR((IADsGroup *)this,Description);
}
STDMETHODIMP
CWinNTGroup::Members(
THIS_ IADsMembers FAR* FAR* ppMembers
)
{
HRESULT hr;
WCHAR szHostServerName[MAX_PATH];
NET_API_STATUS nasStatus = 0;
if (_ParentType == WINNT_DOMAIN_ID) {
hr = WinNTGetCachedDCName(
_DomainName,
szHostServerName,
_Credentials.GetFlags()
);
BAIL_ON_FAILURE(hr);
}
if (_GroupType == WINNT_GROUP_GLOBAL) {
hr = CWinNTGroupCollection::CreateGroupCollection(
_Parent,
_ParentType,
_DomainName,
_ParentType == WINNT_DOMAIN_ID ?
(szHostServerName + 2) :
_ServerName,
_Name,
_GroupType,
IID_IADsMembers,
_Credentials,
(void **)ppMembers
);
} else {
hr = CWinNTLocalGroupCollection::CreateGroupCollection(
_Parent,
_ParentType,
_DomainName,
_ParentType == WINNT_DOMAIN_ID ?
(szHostServerName + 2) :
_ServerName,
_Name,
_GroupType,
IID_IADsMembers,
_Credentials,
(void **)ppMembers
);
}
error:
RRETURN_EXP_IF_ERR(hr);
}
STDMETHODIMP
CWinNTGroup::IsMember(
THIS_ BSTR bstrMember,
VARIANT_BOOL FAR* bMember
)
{
IADsMembers FAR * pMembers = NULL;
IUnknown FAR * pUnknown = NULL;
IEnumVARIANT FAR * pEnumVar = NULL;
DWORD i = 0;
HRESULT hr = S_OK;
VARIANT_BOOL fMember = FALSE;
VARIANT VariantArray[10];
BOOL fContinue = TRUE;
ULONG cElementFetched = 0;
hr = Members(
&pMembers
);
BAIL_ON_FAILURE(hr);
hr = pMembers->get__NewEnum(
&pUnknown
);
BAIL_ON_FAILURE(hr);
hr = pUnknown->QueryInterface(
IID_IEnumVARIANT,
(void **)&pEnumVar
);
BAIL_ON_FAILURE(hr);
while (fContinue) {
IADs *pObject ;
hr = pEnumVar->Next(
10,
VariantArray,
&cElementFetched
);
if (hr == S_FALSE) {
fContinue = FALSE;
//
// Reset hr to S_OK, we want to return success
//
hr = S_OK;
}
fMember = (VARIANT_BOOL)VerifyIfMember(
bstrMember,
VariantArray,
cElementFetched
);
if (fMember) {
fContinue = FALSE;
}
for (i = 0; i < cElementFetched; i++ ) {
IDispatch *pDispatch = NULL;
pDispatch = VariantArray[i].pdispVal;
pDispatch->Release();
}
memset(VariantArray, 0, sizeof(VARIANT)*10);
}
error:
*bMember = fMember? VARIANT_TRUE : VARIANT_FALSE;
if (pEnumVar) {
pEnumVar->Release();
}
if (pUnknown) {
pUnknown->Release();
}
if (pMembers) {
pMembers->Release();
}
RRETURN_EXP_IF_ERR(hr);
}
STDMETHODIMP
CWinNTGroup::Add(THIS_ BSTR bstrNewItem)
{
HRESULT hr;
NET_API_STATUS nasStatus;
LOCALGROUP_MEMBERS_INFO_3 Member;
LPLOCALGROUP_MEMBERS_INFO_3 pMember = &Member;
POBJECTINFO pObjectInfo = NULL;
WCHAR szDomName[MAX_PATH];
WCHAR szHostServerName[MAX_PATH];
int iLastIndx = 0;
BOOL fStringSID = FALSE;
hr = BuildObjectInfo(
bstrNewItem,
&pObjectInfo
);
BAIL_ON_FAILURE(hr);
iLastIndx = pObjectInfo->NumComponents - 1;
//
// If there is only one component, it has to be in the SID form
// or it has to be a Special SID like everyone.
//
if (pObjectInfo->NumComponents == 1) {
//
// Check to see if this is S-12-11
//
fStringSID = IsStringSID(pObjectInfo->ComponentArray[0]);
}
memset(pMember, 0, sizeof(LOCALGROUP_MEMBERS_INFO_3));
if (_ParentType == WINNT_COMPUTER_ID) {
hr = MakeUncName(
_ServerName,
szHostServerName
);
BAIL_ON_FAILURE(hr);
}else if (_ParentType == WINNT_DOMAIN_ID){
hr = WinNTGetCachedDCName(
_DomainName,
szHostServerName,
_Credentials.GetFlags()
);
BAIL_ON_FAILURE(hr);
}
if (_GroupType == WINNT_GROUP_GLOBAL) {
#ifdef WIN95
if (_wcsicmp(pObjectInfo->ComponentArray[0], _DomainName)) {
#else
if (CompareStringW(
LOCALE_SYSTEM_DEFAULT,
NORM_IGNORECASE,
pObjectInfo->ComponentArray[0],
-1,
_DomainName,
-1
) != CSTR_EQUAL ) {
#endif
hr = E_ADS_INVALID_USER_OBJECT;
BAIL_ON_FAILURE(hr);
}
nasStatus = NetGroupAddUser(
szHostServerName,
_Name,
pObjectInfo->ComponentArray[(
pObjectInfo->NumComponents - 1)]
);
hr = HRESULT_FROM_WIN32(nasStatus);
BAIL_ON_FAILURE(hr);
}else if (_GroupType == WINNT_GROUP_LOCAL){
if (fStringSID) {
hr = AddBySID(
pObjectInfo->ComponentArray[0],
szHostServerName
);
goto error;
}
//
// 0 implies special sid name.
//
if (iLastIndx != 0) {
hr = MakeWinNTAccountName(pObjectInfo, szDomName, FALSE);
BAIL_ON_FAILURE(hr);
pMember->lgrmi3_domainandname = szDomName;
}
else {
pMember->lgrmi3_domainandname =
pObjectInfo->ComponentArray[iLastIndx];
}
//
// For performance reasos we will first assume that the
// use has domain name
//
nasStatus = NetLocalGroupAddMembers(
szHostServerName,
_Name,
3,
(LPBYTE)pMember,
1
);
if (nasStatus == ERROR_NO_SUCH_MEMBER) {
//
// Try with true to see if that makes a difference
//
hr = MakeWinNTAccountName(pObjectInfo, szDomName, TRUE);
if (SUCCEEDED(hr)) {
//
// Try again with this value
//
pMember->lgrmi3_domainandname = szDomName;
nasStatus = NetLocalGroupAddMembers(
szHostServerName,
_Name,
3,
(LPBYTE)pMember,
1
);
}
}
//
// Either way nasStatus will have the correct value
//
hr = HRESULT_FROM_WIN32(nasStatus);
BAIL_ON_FAILURE(hr);
}
error:
if (pObjectInfo) {
FreeObjectInfo(pObjectInfo);
}
RRETURN_EXP_IF_ERR(hr);
}
STDMETHODIMP
CWinNTGroup::Remove(THIS_ BSTR bstrItemToBeRemoved)
{
HRESULT hr;
NET_API_STATUS nasStatus;
LOCALGROUP_MEMBERS_INFO_3 Member;
LPLOCALGROUP_MEMBERS_INFO_3 pMember = &Member;
POBJECTINFO pObjectInfo = NULL;
WCHAR szDomName[MAX_PATH];
WCHAR szHostServerName[MAX_PATH];
BOOL fSpecialName = FALSE;
BOOL fStringSID = FALSE;
hr = BuildObjectInfo(
bstrItemToBeRemoved,
&pObjectInfo
);
BAIL_ON_FAILURE(hr);
//
// If there is only one component, it has to be in the SID form
// or it has to be a Special SID like everyone.
//
if (pObjectInfo->NumComponents == 1) {
//
// Check to see if this is S-12-11
//
fStringSID = IsStringSID(pObjectInfo->ComponentArray[0]);
if (!fStringSID) {
fSpecialName = TRUE;
}
}
memset(pMember, 0, sizeof(LOCALGROUP_MEMBERS_INFO_3));
if (_ParentType == WINNT_COMPUTER_ID) {
hr = MakeUncName(
_ServerName,
szHostServerName
);
BAIL_ON_FAILURE(hr);
}else if (_ParentType == WINNT_DOMAIN_ID){
hr = WinNTGetCachedDCName(
_DomainName,
szHostServerName,
_Credentials.GetFlags()
);
BAIL_ON_FAILURE(hr);
}
if (fStringSID) {
hr = DeleteBySID(
pObjectInfo->ComponentArray[0],
szHostServerName
);
goto error;
}
if (_GroupType == WINNT_GROUP_GLOBAL) {
#ifdef WIN95
if (_wcsicmp(pObjectInfo->ComponentArray[0], _DomainName)) {
#else
if (CompareStringW(
LOCALE_SYSTEM_DEFAULT,
NORM_IGNORECASE,
pObjectInfo->ComponentArray[0],
-1,
_DomainName,
-1
) != CSTR_EQUAL ) {
#endif
hr = E_ADS_INVALID_USER_OBJECT;
BAIL_ON_FAILURE(hr);
}
nasStatus = NetGroupDelUser(
szHostServerName,
_Name,
pObjectInfo->ComponentArray[(
pObjectInfo->NumComponents) - 1]
);
hr = HRESULT_FROM_WIN32(nasStatus);
BAIL_ON_FAILURE(hr);
}else {
if (!fSpecialName) {
hr = MakeWinNTAccountName(pObjectInfo, szDomName, FALSE);
BAIL_ON_FAILURE(hr);
pMember->lgrmi3_domainandname = szDomName;
}
else {
pMember->lgrmi3_domainandname =
pObjectInfo->ComponentArray[0];
}
nasStatus = NetLocalGroupDelMembers(
szHostServerName,
_Name,
3,
(LPBYTE)pMember,
1
);
if (nasStatus == ERROR_NO_SUCH_MEMBER) {
hr = MakeWinNTAccountName(pObjectInfo, szDomName, TRUE);
if (SUCCEEDED(hr)) {
pMember->lgrmi3_domainandname = szDomName;
nasStatus = NetLocalGroupDelMembers(
szHostServerName,
_Name,
3,
(LPBYTE)pMember,
1
);
}
}
hr = HRESULT_FROM_WIN32(nasStatus);
BAIL_ON_FAILURE(hr);
}
error:
if (pObjectInfo) {
FreeObjectInfo(pObjectInfo);
}
RRETURN_EXP_IF_ERR(hr);
}
BOOL
VerifyIfMember(
BSTR bstrMember,
VARIANT * VariantArray,
ULONG cElementFetched
)
{
DWORD i = 0;
HRESULT hr = S_OK;
IADs FAR * pObject = NULL;
IDispatch FAR * pDispatch = NULL;
for (i = 0; i < cElementFetched; i++ ) {
IDispatch *pDispatch = NULL;
BSTR bstrName = NULL;
pDispatch = VariantArray[i].pdispVal;
hr = pDispatch->QueryInterface(
IID_IADs,
(VOID **) &pObject
);
BAIL_ON_FAILURE(hr);
hr = pObject->get_ADsPath(&bstrName);
BAIL_ON_FAILURE(hr);
#ifdef WIN95
if (!_wcsicmp(bstrName, bstrMember)) {
#else
if (CompareStringW(
LOCALE_SYSTEM_DEFAULT,
NORM_IGNORECASE,
bstrName,
-1,
bstrMember,
-1
) == CSTR_EQUAL ) {
#endif
SysFreeString(bstrName);
bstrName = NULL;
pObject->Release();
return(TRUE);
}
SysFreeString(bstrName);
bstrName = NULL;
pObject->Release();
}
error:
return(FALSE);
}
HRESULT
CWinNTGroup::DeleteBySID(
LPWSTR pszStringSID,
LPWSTR pszServerName
)
{
HRESULT hr = S_OK;
BOOL fRet = FALSE;
PSID pSid = NULL;
NET_API_STATUS nasStatus;
LOCALGROUP_MEMBERS_INFO_0 member;
//
// SDDL.H is currently available only for Win2k
//
#if !defined(WIN95)
if (!pszStringSID) {
BAIL_ON_FAILURE(hr = E_FAIL);
}
fRet = ConvertStringSidToSidWrapper(
pszStringSID,
&pSid
);
if (!pSid) {
BAIL_ON_FAILURE(hr = HRESULT_FROM_WIN32(GetLastError()));
}
member.lgrmi0_sid = pSid;
nasStatus = NetLocalGroupDelMembers(
pszServerName,
_Name,
0,
(LPBYTE) &member,
1
);
BAIL_ON_FAILURE(hr = HRESULT_FROM_WIN32(nasStatus));
#else
BAIL_ON_FAILURE(hr = E_FAIL);
#endif
error:
if (pSid) {
LocalFree(pSid);
}
RRETURN(hr);
}
//
// Same as delete only this time to add by SID.
//
HRESULT
CWinNTGroup::AddBySID(
LPWSTR pszStringSID,
LPWSTR pszServerName
)
{
HRESULT hr = S_OK;
BOOL fRet = FALSE;
PSID pSid = NULL;
NET_API_STATUS nasStatus;
LOCALGROUP_MEMBERS_INFO_0 member;
//
// SDDL.H is currently available only for Win2k
//
#if !defined(WIN95)
if (!pszStringSID) {
BAIL_ON_FAILURE(hr = E_FAIL);
}
fRet = ConvertStringSidToSidWrapper(
pszStringSID,
&pSid
);
if (!pSid) {
BAIL_ON_FAILURE(hr = HRESULT_FROM_WIN32(GetLastError()));
}
member.lgrmi0_sid = pSid;
nasStatus = NetLocalGroupAddMembers(
pszServerName,
_Name,
0,
(LPBYTE) &member,
1
);
BAIL_ON_FAILURE(hr = HRESULT_FROM_WIN32(nasStatus));
#else
BAIL_ON_FAILURE(hr = E_FAIL);
#endif
error:
if (pSid) {
LocalFree(pSid);
}
RRETURN(hr);
}
//
// Helper routine that checks if a string is a sid or not.
//
BOOL
IsStringSID(LPWSTR pszStringSID)
{
BOOL fRet = FALSE;
if (!pszStringSID || (wcslen(pszStringSID) < 4)) {
return FALSE;
}
if (((*pszStringSID != L'S') && (*pszStringSID != L's'))
|| (*(pszStringSID + 1) != L'-')
) {
return FALSE;
}
return TRUE;
}