//+--------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1994 - 1998.
//
// File: browser.cpp
//
// Contents: implementation of the general GPO browser pane
//
// Classes: CBrowserPP
//
// Functions:
//
// History: 04-30-1998 stevebl Created
//
// Notes: This is the pane that behaves much like the standard file
// open dialog. The class is used for all panes that have this
// format since they share so much functionality. The
// dwPageType parameter passed to CBrowserPP::Initialize is
// used to distinguish between the different flavors.
//
//---------------------------------------------------------------------------
#include "main.h"
#include "browser.h"
#include "commctrl.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
//
// Help ids
//
DWORD aBrowserDomainHelpIds[] =
{
IDC_COMBO1, IDH_BROWSER_LOOKIN,
IDC_LIST1, IDH_BROWSER_DOMAINGPO,
IDC_DESCRIPTION, IDH_NOCONTEXTHELP,
0, 0
};
DWORD aBrowserSiteHelpIds[] =
{
IDC_COMBO1, IDH_BROWSER_SITELIST,
IDC_LIST1, IDH_BROWSER_GPOLIST,
IDC_DESCRIPTION, IDH_NOCONTEXTHELP,
0, 0
};
DWORD aBrowserAllHelpIds[] =
{
IDC_COMBO1, IDH_BROWSER_DOMAINLIST,
IDC_LIST1, IDH_BROWSER_FULLGPOLIST,
IDC_DESCRIPTION, IDH_NOCONTEXTHELP,
0, 0
};
CBrowserPP::CBrowserPP()
{
m_ppActive = NULL;
m_pGBI = NULL;
m_pPrevSel = NULL;
m_szServerName = NULL;
m_szDomainName = NULL;
}
//+--------------------------------------------------------------------------
//
// Function: CopyAsFriendlyName
//
// Synopsis: Copies a LDAP path converting it to a friendly name by
// removing the "LDAP://" and "XX=" and converting "," to "."
// and removing a server name (if any)
//
// Arguments: [lpDest] - destination buffer
// [lpSrc] - source buffer
//
// Returns: nothing
//
// History: 5-07-1998 stevebl Created
//
// Notes: The destination buffer should be as large as the source
// buffer to ensure safe completion. lpDest and lpSrc may both
// point to the same buffer.
//
// As an example, this routine would convert the following path:
// LDAP://DC=foo,DC=bar
// into this:
// foo.bar
//
//---------------------------------------------------------------------------
void CopyAsFriendlyName(WCHAR * lpDest, WCHAR * lpSrc)
{
LPOLESTR lpProvider = L"LDAP://";
DWORD dwStrLen = wcslen(lpProvider);
// lpStopChecking marks the last spot where we can safely
// look ahead 2 spaces for an '=' character. Anything past
// this and we are looking in memory we don't own.
OLECHAR * lpStopChecking = (wcslen(lpSrc) - 2) + lpSrc;
//
// Skip the LDAP:// if found
//
if (CompareString (LOCALE_USER_DEFAULT, NORM_IGNORECASE | NORM_STOP_ON_NULL,
lpProvider, dwStrLen, lpSrc, dwStrLen) == CSTR_EQUAL)
{
lpSrc += dwStrLen;
}
//
// Remove server name (if any)
//
if (lpSrc < lpStopChecking)
{
if (*(lpSrc+2) != L'=')
{
// look for a '/' character marking the end of a server name
while (*lpSrc)
{
if (*lpSrc == L'/')
{
lpSrc++;
break;
}
lpSrc++;
}
}
}
//
// Parse through the name replacing all the XX= with .
//
while (*lpSrc)
{
if (lpSrc < lpStopChecking)
{
if (*(lpSrc+2) == L'=')
{
lpSrc += 3;
}
}
while (*lpSrc && (*lpSrc != L','))
{
// remove escape sequences
if (*lpSrc == L'\\')
{
lpSrc++;
// special cases
// make sure that '\\x' becomes '\x'
if (*lpSrc == L'\\')
{
*lpDest++ = *lpSrc++;
}
// make sure that '\0D' becomes '\r'
else if (*lpSrc == L'0' && *(lpSrc+1) == L'D')
{
*lpDest++ = L'\r';
lpSrc += 2;
}
// make sure that '\0A' becomes '\n'
else if (*lpSrc == L'0' && *(lpSrc+1) == L'A')
{
*lpDest++ = L'\n';
lpSrc += 2;
}
}
else
{
*lpDest++ = *lpSrc++;
}
}
if (*lpSrc == L',')
{
*lpDest++ = L'.';
lpSrc++;
}
}
*lpDest = L'\0';
}
//+--------------------------------------------------------------------------
//
// Member: CBrowserPP::Initialize
//
// Synopsis: Initializes the property page.
//
// Arguments: [dwPageType] - used to identify which page this is. (See
// notes.)
// [pGBI] - pointer to the browse info structure passed
// by caller
// [ppActive] - pointer to a common variable that remembers
// which object was last given the focus.
// Needed because only the page with the focus
// is allowed to return data to the caller when
// the property sheet is dismissed.
//
// Returns: Handle to the newly created property page.
//
// Modifies:
//
// Derivation:
//
// History: 04-30-1998 stevebl Created
//
// Notes: This class implements the following property pages:
// PAGETYPE_DOMAINS - GPO's linked to domains
// PAGETYPE_SITES - GPO's linked to sites
// PAGETYPE_ALL - All GPO's in a selected
//
// PAGETYPE_COMPUTERS is implemented by CCompsPP since it
// behaves so differently.
//
//---------------------------------------------------------------------------
HPROPSHEETPAGE CBrowserPP::Initialize(DWORD dwPageType, LPGPOBROWSEINFO pGBI, void * * ppActive)
{
m_ppActive = ppActive;
m_dwPageType = dwPageType;
m_pGBI = pGBI;
if (m_pGBI->lpInitialOU)
{
//
// Get the server name
//
m_szServerName = ExtractServerName(m_pGBI->lpInitialOU);
DebugMsg((DM_VERBOSE, TEXT("CBrowserPP::Initialize extracted server name: %s"), m_szServerName));
//
// Get the friendly domain name
//
LPOLESTR pszDomain = GetDomainFromLDAPPath(m_pGBI->lpInitialOU);
//
// Convert LDAP to dot (DN) style
//
if (pszDomain)
{
ConvertToDotStyle (pszDomain, &m_szDomainName);
DebugMsg((DM_VERBOSE, TEXT("CBrowserPP::Initialize extracted domain name: %s"), m_szDomainName));
delete [] pszDomain;
}
}
DWORD dwTitle;
switch (dwPageType)
{
case PAGETYPE_DOMAINS:
dwTitle = IDS_DOMAINS;
break;
case PAGETYPE_SITES:
dwTitle = IDS_SITES;
break;
case PAGETYPE_ALL:
default:
dwTitle = IDS_ALL;
break;
}
LoadString(g_hInstance, dwTitle, m_szTitle, sizeof(m_szTitle) / sizeof(WCHAR));
PROPSHEETPAGE psp;
memset(&psp, 0, sizeof(psp));
psp.dwSize = sizeof(psp);
psp.dwFlags = PSP_USETITLE;
psp.pszTitle = m_szTitle;
psp.hInstance = g_hInstance;
psp.pszTemplate = MAKEINTRESOURCE(IDD_PROPPAGE_GPOBROWSER);
return CreatePropertySheetPage(&psp);
}
CBrowserPP::~CBrowserPP()
{
if (m_szServerName)
{
LocalFree(m_szServerName);
}
if (m_szDomainName)
{
LocalFree(m_szDomainName);
}
}
/////////////////////////////////////////////////////////////////////////////
// CBrowserPP message handlers
INT CBrowserPP::AddElement(MYLISTEL * pel, INT index)
{
LV_ITEM item;
memset(&item, 0, sizeof(item));
item.mask = LVIF_TEXT | LVIF_IMAGE | LVIF_PARAM;
if (-1 == index)
{
index = ListView_GetItemCount(m_hList);
}
item.iItem = index;
item.pszText = pel->szName;
if (pel->nType == ITEMTYPE_FOREST)
{
item.iImage = 10;
}
else if (pel->nType == ITEMTYPE_SITE)
{
item.iImage = 6;
}
else if (pel->nType == ITEMTYPE_DOMAIN)
{
item.iImage = 7;
}
else if (pel->nType == ITEMTYPE_OU)
{
item.iImage = 0;
}
else
{
if (pel->bDisabled)
{
item.iImage = 3;
}
else
{
item.iImage = 2;
}
}
item.lParam = (LPARAM)pel;
index = ListView_InsertItem(m_hList, &item);
if (index != -1 && pel->nType == ITEMTYPE_GPO)
{
// check to see if we need to add the domain name
LPOLESTR szObject = GetCurrentObject();
LPOLESTR szDomain = GetDomainFromLDAPPath(pel->szData);
if (szDomain && szObject)
{
// ignore potential differences in server name when we compare
// the domain paths
LPOLESTR szBuffer1 = NULL;
LPOLESTR szBuffer2 = NULL;
szBuffer1 = new OLECHAR[wcslen(szObject) + 1];
szBuffer2 = new OLECHAR[wcslen(szDomain) + 1];
if (NULL != szBuffer1 && NULL != szBuffer1)
{
CopyAsFriendlyName(szBuffer1, szObject);
CopyAsFriendlyName(szBuffer2, szDomain);
if (0 != wcscmp(szBuffer1, szBuffer2))
{
// Need to add the domain name since the domain is different
// from the focus object.
// Need to convert the domain to a friendly name.
// Let's just do it in place so I don't have to allocate any
// more memory. :)
// We can get away with this because the string can only get smaller.
CopyAsFriendlyName(szDomain, szDomain);
memset(&item, 0, sizeof(item));
item.mask = LVIF_TEXT;
item.iItem = index;
item.iSubItem = 1;
item.pszText = szDomain;
ListView_SetItem(m_hList, &item);
}
}
if (szBuffer1)
{
delete [] szBuffer1;
}
if (szBuffer2)
{
delete [] szBuffer2;
}
}
if (szDomain)
delete [] szDomain;
if (szObject)
delete [] szObject;
}
return (index);
}
#include "ntdsapi.h"
//+--------------------------------------------------------------------------
//
// Member: CBrowserPP::FillSitesList
//
// Synopsis: Fills the combobox with the trusted sites information.
// The szData member of the combobox element structure is the
// containing domain.
//
// Returns: TRUE - successful
// FALSE - error
//
// History: 05-04-1998 stevebl created
// 05-27-1999 stevebl now initializes to site in lpInitialOU
//
//---------------------------------------------------------------------------
BOOL CBrowserPP::FillSitesList ()
{
HCURSOR hcur = SetCursor(LoadCursor(NULL, IDC_WAIT));
PDS_NAME_RESULTW pSites;
int iInitialSite = 0;
int iIndex = -1;
HANDLE hDs;
DWORD dw = DsBindW(NULL, NULL, &hDs);
if (ERROR_SUCCESS == dw)
{
dw = DsListSitesW(hDs, &pSites);
if (ERROR_SUCCESS == dw)
{
DWORD n = 0;
for (n = 0; n < pSites->cItems; n++)
{
//
// Add the site name (if it has a name)
//
if (pSites->rItems[n].pName)
{
LPTSTR lpFullPath, lpTempPath;
LOOKDATA * pdata;
pdata = new LOOKDATA;
if (pdata)
{
pdata->szName = new WCHAR[wcslen(pSites->rItems[n].pName)+1];
if (pdata->szName)
{
wcscpy(pdata->szName, pSites->rItems[n].pName);
}
pdata->szData = NULL;
lpTempPath = (LPTSTR) LocalAlloc (LPTR, (lstrlen(pSites->rItems[n].pName) + 10) * sizeof(TCHAR));
if (lpTempPath)
{
lstrcpy (lpTempPath, TEXT("LDAP://"));
lstrcat (lpTempPath, pSites->rItems[n].pName);
lpFullPath = GetFullPath (lpTempPath, m_hwndDlg);
if (lpFullPath)
{
pdata->szData = new WCHAR[wcslen(lpFullPath)+1];
if (pdata->szData)
{
wcscpy(pdata->szData, lpFullPath);
}
LocalFree (lpFullPath);
}
LocalFree (lpTempPath);
}
if (!pdata->szData)
{
if (pdata->szName)
{
delete [] pdata->szName;
}
delete pdata;
continue;
}
// try and use a friendlier name for the site
{
IADs * pADs = NULL;
// Get the friendly display name
HRESULT hr = OpenDSObject(pdata->szData, IID_IADs,
(void **)&pADs);
if (SUCCEEDED(hr))
{
VARIANT varName;
BSTR bstrNameProp;
VariantInit(&varName);
bstrNameProp = SysAllocString(SITE_NAME_PROPERTY);
if (bstrNameProp)
{
hr = pADs->Get(bstrNameProp, &varName);
if (SUCCEEDED(hr))
{
LPOLESTR sz = new OLECHAR[wcslen(varName.bstrVal) + 1];
if (sz)
{
wcscpy(sz, varName.bstrVal);
if (pdata->szName)
delete [] pdata->szName;
pdata->szName = sz;
}
}
SysFreeString(bstrNameProp);
}
VariantClear(&varName);
pADs->Release();
}
}
pdata->nIndent = 0;
pdata->nType = ITEMTYPE_SITE;
iIndex = (int)SendMessage(m_hCombo, CB_INSERTSTRING, (WPARAM) -1, (LPARAM) (LPCTSTR) pdata);
if (CB_ERR == iIndex)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::AddSitesList: Failed to alloc memory with %d"), GetLastError()));
}
if (NULL != pdata->szData && NULL != m_pGBI->lpInitialOU)
{
if (0 == wcscmp(pdata->szData, m_pGBI->lpInitialOU))
{
iInitialSite = iIndex;
}
}
}
}
}
DsFreeNameResultW(pSites);
}
DsUnBindW(&hDs);
}
else
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::AddSitesList: DsBindW failed with 0x%x"), dw));
ReportError(m_hwndDlg, dw, IDS_DSBINDFAILED);
}
if (iIndex >= 0)
{
SendMessage (m_hCombo, CB_SETCURSEL, iInitialSite, 0);
}
SetCursor(hcur);
return TRUE;
}
PDS_DOMAIN_TRUSTS Domains;
int __cdecl CompareDomainInfo(const void * arg1, const void * arg2)
{
WCHAR * sz1, *sz2;
sz1 = Domains[*(ULONG *)arg1].DnsDomainName;
sz2 = Domains[*(ULONG *)arg2].DnsDomainName;
if (!sz1)
{
sz1 = Domains[*(ULONG *)arg1].NetbiosDomainName;
}
if (!sz2)
{
sz2 = Domains[*(ULONG *)arg2].NetbiosDomainName;
}
return _wcsicmp(sz1,sz2);
}
typedef struct tag_WORKING_LIST_EL
{
ULONG index;
struct tag_WORKING_LIST_EL * pNext;
} WORKING_LIST_EL;
//+--------------------------------------------------------------------------
//
// Function: BuildDomainList
//
// Synopsis: Builds a tree containing all domains that have a trust
// relationship with the server.
//
// Siblings within the tree are alphabetized.
//
// Arguments: [szServerName] - (NULL for local)
//
// Returns: pointer to the root node of the tree (NULL on error)
//
// History: 10-16-1998 stevebl Created
//
// Notes: Tree nodes must be freed by the caller (using delete).
//
//---------------------------------------------------------------------------
LOOKDATA * BuildDomainList(WCHAR * szServerName)
{
ULONG DomainCount;
OLECHAR szBuffer[128];
#if FGPO_SUPPORT
LOOKDATA * pDomainList = new LOOKDATA;
if (!pDomainList)
{
// failed to even create the Forest node!
return NULL;
}
pDomainList->szData = GetPathToForest(szServerName);
if (!pDomainList->szData)
{
delete pDomainList;
return NULL;
}
// load the name for the forest from resources
if (0 == LoadStringW(g_hInstance, IDS_FOREST, szBuffer, sizeof(szBuffer) / sizeof(szBuffer[0])))
{
// failed to get the resource name
delete pDomainList;
return NULL;
}
pDomainList->szName = new OLECHAR [lstrlen(szBuffer) + 1];
if (NULL == pDomainList->szName)
{
// not enough memory to create name of the forest node
delete pDomainList;
return NULL;
}
lstrcpy(pDomainList->szName, szBuffer);
pDomainList->nIndent = 0;
pDomainList->nType = ITEMTYPE_FOREST;
pDomainList->pParent = NULL;
pDomainList->pSibling = NULL;
pDomainList->pChild = NULL;
#else
LOOKDATA * pDomainList = NULL;
#endif
long l = DsEnumerateDomainTrusts(szServerName,
DS_DOMAIN_IN_FOREST | DS_DOMAIN_NATIVE_MODE |
DS_DOMAIN_PRIMARY | DS_DOMAIN_TREE_ROOT,
&Domains,
&DomainCount);
//
// Some of the below code might be unnecessary since DsEnumerateTrusts will no
// longer return domains from other forests. Shouldn't do any harm though..
//
if ((0 == l) && (DomainCount > 0))
{
// sort the list of domains alphabetically
ULONG * rgSorted = new ULONG[DomainCount];
if (rgSorted)
{
ULONG n = DomainCount;
while (n--)
{
rgSorted[n] = n;
}
qsort(rgSorted, DomainCount, sizeof (ULONG), CompareDomainInfo);
// Build a working list of the domains, sorted alphabetically in
// INVERTED order.
WORKING_LIST_EL * pWorkList = NULL;
LOOKDATA ** rgDataMap = new LOOKDATA * [DomainCount];
if (rgDataMap)
{
n = 0;
while (n < DomainCount)
{
WORKING_LIST_EL * pNew = new WORKING_LIST_EL;
if (pNew)
{
pNew->index = rgSorted[n];
pNew->pNext = pWorkList;
pWorkList = pNew;
}
rgDataMap[n] = NULL;
n++;
}
// Build the ordered tree of domains by removing domains from the
// working list and inserting them into the new tree until there are
// none left in the working list.
// NOTE - if this routine runs out of memory it will begin
// to drop nodes rather than AV.
WORKING_LIST_EL ** ppWorker;
BOOL fContinue = TRUE;
while (pWorkList && fContinue)
{
fContinue = FALSE;
ppWorker = &pWorkList;
while (*ppWorker)
{
if (NULL == Domains[(*ppWorker)->index].DnsDomainName)
{
//
// For now, if it doesn't have a
// DnsDomainName then we're going to
// skip it.
// Eventually we'll want to make sure it doesn't
// have a DC by calling DsGetDcName with
// DS_DIRECTORY_SERVICE_PREFERRED.
// remove it from the worker list
WORKING_LIST_EL * pNext = (*ppWorker)->pNext;
delete *ppWorker;
*ppWorker = pNext;
}
else
{
// Does this node have a parent?
ULONG flags = Domains[(*ppWorker)->index].Flags;
if ((0 != (flags & DS_DOMAIN_IN_FOREST)) && (0 == (flags & DS_DOMAIN_TREE_ROOT)))
{
// it has a parent has its parent been added?
LOOKDATA * pParent = rgDataMap[Domains[(*ppWorker)->index].ParentIndex];
if (pParent != NULL)
{
// its parent has been added
// insert this one in its parent's child list
LOOKDATA * pData = new LOOKDATA;
if (pData)
{
WCHAR * szName = Domains[(*ppWorker)->index].DnsDomainName;
if (!szName)
{
szName = Domains[(*ppWorker)->index].NetbiosDomainName;
}
pData->szName = new WCHAR[wcslen(szName) + 1];
if (pData->szName)
{
int cch = 0;
int n=0;
// count the dots in szName;
while (szName[n])
{
if (L'.' == szName[n])
{
cch++;
}
n++;
}
cch *= 3; // multiply the number of dots by 3;
cch += 11; // add 10 + 1 (for the null)
cch += n; // add the string size;
pData->szData = new WCHAR[cch];
if (pData->szData)
{
NameToPath(pData->szData, szName, cch);
wcscpy(pData->szName, szName);
pData->nIndent = pParent->nIndent+1;
pData->nType = ITEMTYPE_DOMAIN;
pData->pParent = pParent;
pData->pSibling = pParent->pChild;
pData->pChild = NULL;
pParent->pChild = pData;
rgDataMap[(*ppWorker)->index] = pData;
// make sure we remember
// that we added something
// to the master list (helps
// us avoid infinite loops
// in case of an error)
fContinue = TRUE;
}
else
{
delete [] pData->szName;
delete pData;
}
}
else
{
delete pData;
}
}
// and remove it from the worker list
WORKING_LIST_EL * pNext = (*ppWorker)->pNext;
delete *ppWorker;
*ppWorker = pNext;
}
else
{
// skip it for now
ppWorker = &((*ppWorker)->pNext);
}
}
else
{
// it doesn't have a parent add it just under the forest
// level of the list
LOOKDATA * pData = new LOOKDATA;
if (pData)
{
WCHAR * szName = Domains[(*ppWorker)->index].DnsDomainName;
if (!szName)
{
szName = Domains[(*ppWorker)->index].NetbiosDomainName;
}
pData->szName = new WCHAR[wcslen(szName) + 1];
if (pData->szName)
{
int cch = 0;
int n=0;
// count the dots in szName;
while (szName[n])
{
if (L'.' == szName[n])
{
cch++;
}
n++;
}
cch *= 3; // multiply the number of dots by 3;
cch += 11; // add 10 + 1 for the null
cch += n; // add the string size;
pData->szData = new WCHAR[cch];
if (pData->szData)
{
NameToPath(pData->szData, szName, cch);
wcscpy(pData->szName, szName);
#if FGPO_SUPPORT
pData->nIndent = 1;
pData->nType = ITEMTYPE_DOMAIN;
pData->pParent = pDomainList;
pData->pSibling = pDomainList->pChild;
pData->pChild = NULL;
pDomainList->pChild = pData;
#else
pData->nIndent = 0;
pData->nType = ITEMTYPE_DOMAIN;
pData->pParent = NULL;
pData->pSibling = pDomainList;
pData->pChild = NULL;
pDomainList = pData;
#endif
rgDataMap[(*ppWorker)->index] = pData;
// make sure we remember
// that we added something
// to the master list (helps
// us avoid infinite loops
// in case of an error)
fContinue = TRUE;
}
else
{
delete [] pData->szName;
delete pData;
}
}
else
{
delete pData;
}
}
// and remove it from the worker list
WORKING_LIST_EL * pNext = (*ppWorker)->pNext;
delete *ppWorker;
*ppWorker = pNext;
}
}
}
}
delete [] rgDataMap;
}
delete [] rgSorted;
}
NetApiBufferFree(Domains);
}
else
{
if (0 != l)
{
DebugMsg((DM_WARNING, TEXT("DsEnumerateDomainTrustsW failed with %u"), l));
}
}
return pDomainList;
}
VOID FreeDomainInfo (LOOKDATA * pEntry)
{
if (!pEntry)
{
return;
}
if (pEntry->pChild)
{
FreeDomainInfo (pEntry->pChild);
}
if (pEntry->pSibling)
{
FreeDomainInfo (pEntry->pSibling);
}
delete [] pEntry->szName;
delete pEntry;
}
//+--------------------------------------------------------------------------
//
// Member: CBrowserPP::FillDomainList
//
// Synopsis: Fills the combobox with the trusted domain information.
// The szData member of the combobox element structure is the
// LDAP domain name.
//
// Returns: TRUE - successful
// FALSE - error
//
// History: 04-30-1998 stevebl Modified from original version
// written by EricFlo
// 10-20-1998 stevebl Heavily modified to support domains
// "outside the forest" and to fix a
// whole passle o' bugs.
//
// Note: This routine also sets the focus to the domain of the object
// passed in via the lpInitialOU member of the GPOBROWSEINFO
// structure.
//
//---------------------------------------------------------------------------
BOOL CBrowserPP::FillDomainList ()
{
BOOL bResult = TRUE;
HRESULT hr;
DWORD dwIndex;
BOOL fEnableBackbutton = FALSE;
HCURSOR hcur = SetCursor(LoadCursor(NULL, IDC_WAIT));
WCHAR * szBuffer1 = NULL;
if (m_pGBI->lpInitialOU)
{
if (IsForest(m_pGBI->lpInitialOU))
{
szBuffer1 = new TCHAR[128];
LoadStringW(g_hInstance, IDS_FOREST, szBuffer1, 128);
}
else
{
WCHAR * sz = GetDomainFromLDAPPath(m_pGBI->lpInitialOU);
if (sz)
{
szBuffer1 = new WCHAR[wcslen(sz) + 1];
if (szBuffer1)
{
CopyAsFriendlyName(szBuffer1, sz);
}
delete [] sz;
}
}
}
LOOKDATA * pDomainList = BuildDomainList(m_szServerName);
if (!pDomainList)
{
ReportError(m_hwndDlg, GetLastError(), IDS_DOMAINLIST);
}
// Walk the ordered tree of domains, inserting each one into the
// dialog box
DWORD dwInitialDomain = -1;
// start at the head
while (pDomainList)
{
WCHAR * szBuffer2 = NULL;
// add this node
dwIndex = (DWORD)SendMessage(m_hCombo, CB_INSERTSTRING, (WPARAM) -1, (LPARAM)(LPCTSTR) pDomainList);
szBuffer2 = new WCHAR[wcslen(pDomainList->szData) + 1];
if (szBuffer2)
{
CopyAsFriendlyName(szBuffer2, pDomainList->szData);
}
if (NULL != szBuffer1 && NULL !=szBuffer2 && 0 ==_wcsicmp(szBuffer1, szBuffer2))
{
// replace the domain path with the path provided by the caller
// (because it contains the server)
WCHAR * sz = GetDomainFromLDAPPath(m_pGBI->lpInitialOU);
if (sz)
{
DebugMsg((DM_VERBOSE, TEXT("CBrowserPP::FillDomainList: Resetting domain path to user specified path: %s"), sz));
delete [] pDomainList->szData;
pDomainList->szData = sz;
}
dwInitialDomain = dwIndex;
if (pDomainList->nIndent > 0)
fEnableBackbutton = TRUE;
}
if (szBuffer2)
{
delete [] szBuffer2;
}
if (pDomainList->pChild)
{
// go to its child
pDomainList = pDomainList->pChild;
}
else
{
if (pDomainList->pSibling)
{
// go to its sibling if there are no children
pDomainList = pDomainList->pSibling;
}
else
{
// there are no children and no siblings
// back up until we find a parent with a sibling
// or there are no more parents (we're done)
do
{
pDomainList = pDomainList->pParent;
if (pDomainList)
{
if (pDomainList->pSibling)
{
pDomainList = pDomainList->pSibling;
break;
}
}
else
{
break;
}
} while (TRUE);
}
}
}
if (szBuffer1)
{
delete [] szBuffer1;
}
if (-1 == dwInitialDomain)
{
// didn't find the initial domain anywhere in that list
// Set the first entry by default
dwInitialDomain = 0;
}
SendMessage (m_hCombo, CB_SETCURSEL, dwInitialDomain, 0);
SendMessage (m_toolbar, TB_ENABLEBUTTON, (WPARAM) ID_BACKBUTTON, (LPARAM) MAKELONG(fEnableBackbutton, 0));
SetCursor(hcur);
return bResult;
}
//+--------------------------------------------------------------------------
//
// Member: CBrowserPP::SetInitialOU
//
// Synopsis: Adds nodes to the combobox until the initial OU specified by
// the caller via the lpInitalOU member of the GPOBROWSEINFO
// structure is present and gives it the focus.
//
// Returns: TRUE - success
//
// History: 10-20-1998 stevebl Created
//
// Notes: This routine assumes that FillDomainList() was just called.
// It will not work properly otherwise.
//
//---------------------------------------------------------------------------
BOOL CBrowserPP::SetInitialOU()
{
if (!m_pGBI->lpInitialOU)
{
// nothing requested so nothing required
return TRUE;
}
int iIndex = (int)SendMessage (m_hCombo, CB_GETCURSEL, 0, 0);
if (iIndex == CB_ERR)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::SetInitialOU: No object selected.")));
return FALSE;
}
// get the current object to see what's selected
LOOKDATA * pdataSelected = (LOOKDATA *) SendMessage (m_hCombo, CB_GETITEMDATA, iIndex, 0);
if (pdataSelected)
{
// is it the same as the requested object?
WCHAR * szSelected = NULL;
WCHAR * szRequested = NULL;
szSelected = new WCHAR[wcslen(pdataSelected->szData) + 1];
if (szSelected)
{
CopyAsFriendlyName(szSelected, pdataSelected->szData);
}
szRequested = new WCHAR[wcslen(m_pGBI->lpInitialOU + 1)];
if (NULL != szSelected && NULL != szRequested && 0 != wcscmp(szSelected, szRequested))
{
// it's not the same
// try and bind to the requested object
IADs * pADs = NULL;
HRESULT hr = OpenDSObject(m_pGBI->lpInitialOU,
IID_IADs, (void **)&pADs);
if (SUCCEEDED(hr))
{
// the requested object exists and we have access permission
// now make sure that it's a domain or OU
BOOL fDomainOrOU = FALSE;
VARIANT var;
VariantInit(&var);
BSTR bstrProperty = SysAllocString(L"objectClass");
if (bstrProperty)
{
hr = pADs->Get(bstrProperty, &var);
if (SUCCEEDED(hr))
{
int cElements = var.parray->rgsabound[0].cElements;
VARIANT * rgData = (VARIANT *)var.parray->pvData;
while (cElements--)
{
if (0 == _wcsicmp(L"domain", rgData[cElements].bstrVal))
{
fDomainOrOU = TRUE;
}
if (0 == _wcsicmp(L"organizationalUnit", rgData[cElements].bstrVal))
{
fDomainOrOU = TRUE;
}
}
}
SysFreeString(bstrProperty);
}
VariantClear(&var);
pADs->Release();
if (fDomainOrOU)
{
LOOKDATA * pLast = NULL;
LOOKDATA * pNew = NULL;
// build a list of nodes
// repeat removing leaf nodes until we're down to the domain
// (which will be the same as the selected object)
IADsPathname * pADsPathname = NULL;
BSTR bstr;
hr = CoCreateInstance(CLSID_Pathname,
NULL,
CLSCTX_INPROC_SERVER,
IID_IADsPathname,
(LPVOID*)&pADsPathname);
if (SUCCEEDED(hr))
{
hr = pADsPathname->Set(m_pGBI->lpInitialOU, ADS_SETTYPE_FULL);
if (SUCCEEDED(hr))
{
while (TRUE)
{
// add this node to the list
hr = pADsPathname->Retrieve(ADS_FORMAT_X500, &bstr);
if (FAILED(hr))
{
break;
}
if (szRequested)
{
delete [] szRequested;
}
szRequested = new WCHAR[wcslen(bstr) + 1];
if (szRequested)
{
CopyAsFriendlyName(szRequested, bstr);
}
if (NULL != szRequested && 0 == wcscmp(szSelected, szRequested))
{
// we're back to the first node
SysFreeString(bstr);
break;
}
pNew = new LOOKDATA;
if (!pNew)
{
// ran out of memory
SysFreeString(bstr);
break;
}
pNew->szName = new WCHAR[wcslen(szRequested) + 1];
if (!pNew->szName)
{
// ran out of memory
delete pNew;
SysFreeString(bstr);
break;
}
pNew->szData = new WCHAR[wcslen(bstr) + 1];
if (!pNew->szData)
{
// ran out of memory
delete [] pNew->szName;
delete pNew;
SysFreeString(bstr);
break;
}
wcscpy(pNew->szData, bstr);
wcscpy(pNew->szName, szRequested);
SysFreeString(bstr);
pNew->nIndent = 0;
pNew->nType = ITEMTYPE_OU;
pNew->pParent = NULL;
pNew->pSibling = NULL;
pNew->pChild = pLast;
if (pLast)
{
pLast->pParent = pNew;
}
pLast = pNew;
// strip off a leaf node and go again
hr = pADsPathname->RemoveLeafElement();
if (FAILED(hr))
{
break;
}
}
}
pADsPathname->Release();
}
// At this point I should have a list of LOOKDATA nodes
// (in pLast).
// The only things left to do are to link them into the
// tree, set their nIndent members, add them to the combo
// box and set the combo box's focus to the last one.
if (pLast)
{
// link in the list
pLast->pSibling = pdataSelected->pChild;
pLast->pParent = pdataSelected;
pLast->nIndent = pdataSelected->nIndent+1;
pdataSelected->pChild = pLast;
// now walk the tree, adding entries to the combo box
// and updating the nIndent members
while (pLast)
{
iIndex = (int)SendMessage(m_hCombo, CB_INSERTSTRING, iIndex+1, (LPARAM)(LPCTSTR) pLast);
if (pLast->pChild)
{
pLast->pChild->nIndent = pLast->nIndent+1;
}
pLast = pLast->pChild;
}
if (iIndex != CB_ERR)
{
SendMessage(m_hCombo, CB_SETCURSEL, iIndex, 0);
SendMessage(m_toolbar, TB_ENABLEBUTTON, (WPARAM) ID_BACKBUTTON, (LPARAM) MAKELONG(TRUE, 0));
}
}
}
}
}
if (szSelected)
{
delete [] szSelected;
}
if (szRequested)
{
delete [] szRequested;
}
}
return TRUE;
}
//+--------------------------------------------------------------------------
//
// Member: CBrowserPP::GetCurrentObject
//
// Synopsis: returns the LDAP path to the currently selected object
//
// Arguments: [] -
//
// Returns: NULL if no ojbect is selected else the LDAP path of the object
//
// Modifies:
//
// Derivation:
//
// History: 5-05-1998 stevebl Created
// 06-23-1999 stevebl Added logic to give DCs names
//
// Notes:
// Checks to see if a domain has a named server. If it doesn't
// then it calls GetDCName to get it one.
//
//---------------------------------------------------------------------------
LPOLESTR CBrowserPP::GetCurrentObject()
{
int iIndex = (int)SendMessage (m_hCombo, CB_GETCURSEL, 0, 0);
if (iIndex == CB_ERR)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::GetCurrentObject: No object selected.")));
return NULL;
}
LPOLESTR sz=NULL;
LOOKDATA * pdata = (LOOKDATA *) SendMessage (m_hCombo, CB_GETITEMDATA, iIndex, 0);
if (pdata)
{
if (pdata->szData)
{
if (ITEMTYPE_DOMAIN == pdata->nType)
{
// make sure that domains are resolved to a server
LPTSTR szServer = ExtractServerName(pdata->szData);
if (NULL == szServer)
{
LPWSTR szTemp = GetDCName(pdata->szName, NULL, NULL, TRUE, 0);
if (szTemp)
{
LPWSTR szFullPath = MakeFullPath(pdata->szData, szTemp);
if (szFullPath)
{
LPWSTR sz = new WCHAR[wcslen(szFullPath)+1];
if (sz)
{
wcscpy(sz, szFullPath);
delete [] pdata->szData;
pdata->szData = sz;
}
LocalFree(szFullPath);
}
LocalFree(szTemp);
}
else
{
return NULL;
}
}
else
{
LocalFree(szServer);
}
}
sz = new OLECHAR[wcslen(pdata->szData) + 1];
if (sz)
{
wcscpy(sz, pdata->szData);
}
}
}
return sz;
}
//+--------------------------------------------------------------------------
//
// Member: CBrowserPP::IsCurrentObjectAForest
//
// Synopsis: tests to see if the currently selected object is a forest
//
// Arguments: [] -
//
// Returns: TRUE - if it is a forest
// FALSE - otherwise
//
// History: 03-31-2000 stevebl Created
//
//---------------------------------------------------------------------------
BOOL CBrowserPP::IsCurrentObjectAForest()
{
int iIndex = (int)SendMessage (m_hCombo, CB_GETCURSEL, 0, 0);
if (iIndex == CB_ERR)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::IsCurrentObjectAForest: No object selected.")));
return FALSE;
}
LOOKDATA * pdata = (LOOKDATA *) SendMessage (m_hCombo, CB_GETITEMDATA, iIndex, 0);
return (ITEMTYPE_FOREST == pdata->nType);
}
//+--------------------------------------------------------------------------
//
// Member: CBrowserPP::GetCurrentDomain
//
// Synopsis: returns the domain of the currently selecte object (if the
// currently currently selected object is the domain then they
// are one and the same)
//
// Arguments: [] -
//
// Returns: NULL - if no object is selected else returns LDAP path of
// domain
//
// History: 05-04-1998 stevebl Created
// 06-23-1999 stevebl Added logic to give DCs names
//
// Notes: Checks to see if a domain has a named server. If it doesn't
// then it calls GetDCName to get it one.
//
//---------------------------------------------------------------------------
LPOLESTR CBrowserPP::GetCurrentDomain()
{
int iIndex = (int)SendMessage (m_hCombo, CB_GETCURSEL, 0, 0);
if (iIndex == CB_ERR)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::GetCurrentDomain: No object selected.")));
return NULL;
}
LOOKDATA * pdata = (LOOKDATA *) SendMessage (m_hCombo, CB_GETITEMDATA, iIndex, 0);
switch (pdata->nType)
{
case ITEMTYPE_DOMAIN:
{
if (pdata->szData)
{
// make sure the domain has a server
LPTSTR szServer = ExtractServerName(pdata->szData);
if (NULL == szServer)
{
LPWSTR szTemp = GetDCName(pdata->szName, NULL, NULL, TRUE, 0);
if (szTemp)
{
LPWSTR szFullPath = MakeFullPath(pdata->szData, szTemp);
if (szFullPath)
{
LPWSTR sz = new WCHAR[wcslen(szFullPath)+1];
if (sz)
{
wcscpy(sz, szFullPath);
delete [] pdata->szData;
pdata->szData = sz;
}
LocalFree(szFullPath);
}
LocalFree(szTemp);
}
else
{
return NULL;
}
}
else
{
LocalFree(szServer);
}
LPOLESTR sz = new OLECHAR[wcslen(pdata->szData)+1];
if (sz)
{
wcscpy(sz, pdata->szData);
}
return sz;
}
return NULL;
}
case ITEMTYPE_FOREST:
case ITEMTYPE_SITE:
case ITEMTYPE_OU:
{
return GetDomainFromLDAPPath(pdata->szData);
}
break;
default:
break;
}
return NULL;
}
BOOL CBrowserPP::AddGPOsLinkedToObject()
{
HCURSOR hcur = SetCursor(LoadCursor(NULL, IDC_WAIT));
LPOLESTR lpObject;
HRESULT hr;
IADs * pADs = NULL;
IADs * pADsGPO;
VARIANT var;
BSTR bstrProperty;
BOOL fResult = FALSE;
int index = ListView_GetItemCount(m_hList);
//
// Get the current object name
//
lpObject = GetCurrentObject();
if (NULL == lpObject)
{
return FALSE;
}
DebugMsg((DM_VERBOSE, TEXT("CBrowserPP::AddGPOsLinkedToObject: Reading gPLink property from %s"), lpObject));
hr = OpenDSObject(lpObject, IID_IADs, (void **)&pADs);
delete [] lpObject;
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::AddGPOsLinkedToObject: OpenDSObject failed with 0x%x"), hr));
ReportError(m_hwndDlg, hr, IDS_FAILEDGPLINK);
goto Exit;
}
VariantInit(&var);
bstrProperty = SysAllocString(GPM_LINK_PROPERTY);
if (bstrProperty)
{
hr = pADs->Get(bstrProperty, &var);
if (SUCCEEDED(hr))
{
LPOLESTR szGPOList = var.bstrVal;
OLECHAR * pchTemp;
OLECHAR * pchGPO;
VARIANT varName;
BSTR bstrNameProp;
if (szGPOList)
{
OLECHAR * szGPO = new WCHAR[wcslen(szGPOList) + 1];
if (szGPO)
{
pchTemp = szGPOList;
while (TRUE)
{
// Look for the [
while (*pchTemp && (*pchTemp != L'['))
pchTemp++;
if (!(*pchTemp))
break;
pchTemp++;
// Copy the GPO name
pchGPO = szGPO;
while (*pchTemp && (*pchTemp != L';'))
*pchGPO++ = *pchTemp++;
*pchGPO = L'\0';
// Add the object to the list view
MYLISTEL * pel = new MYLISTEL;
if (pel)
{
pel->szData = NULL;
pel->bDisabled = FALSE;
LPTSTR szFullGPOPath = GetFullPath(szGPO, m_hwndDlg);
if (szFullGPOPath)
{
pel->szData = new WCHAR[wcslen(szFullGPOPath) + 1];
if (pel->szData)
{
wcscpy(pel->szData, szFullGPOPath);
}
else
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::AddGPOsLinkedToObject: Failed to allocate memory for new full gpo path")));
LocalFree(szFullGPOPath);
delete pel;
continue;
}
LocalFree(szFullGPOPath);
}
else
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::AddGPOsLinkedToObject: Failed to get full gpo path")));
delete pel;
continue;
}
VariantInit(&varName);
// get the friendly display name
hr = OpenDSObject(pel->szData, IID_IADs,
(void **)&pADsGPO);
if (hr == HRESULT_FROM_WIN32(ERROR_DS_NO_SUCH_OBJECT))
{
delete pel;
continue;
}
if (SUCCEEDED(hr))
{
bstrNameProp = SysAllocString(GPO_NAME_PROPERTY);
if (bstrNameProp)
{
hr = pADsGPO->Get(bstrNameProp, &varName);
SysFreeString(bstrNameProp);
}
else
{
hr = HRESULT_FROM_WIN32(ERROR_NOT_ENOUGH_MEMORY);
}
pADsGPO->Release();
}
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::AddGPOsLinkedToObject: Couldn't get display name for %s with 0x%x"), pel->szData, hr));
pel->szName = new WCHAR[200];
if (pel->szName)
{
LoadString(g_hInstance, IDS_GPM_NOGPONAME, pel->szName, 200);
}
pel->bDisabled = TRUE;
}
else
{
pel->szName = new WCHAR[wcslen(varName.bstrVal) + 1];
if (pel->szName)
{
wcscpy(pel->szName, varName.bstrVal);
}
}
VariantClear(&varName);
pel->nType = ITEMTYPE_GPO;
AddElement(pel, index);
}
}
delete [] szGPO;
}
}
}
SysFreeString(bstrProperty);
}
VariantClear(&var);
fResult = TRUE;
Exit:
if (pADs)
{
pADs->Release();
}
SetCursor(hcur);
return fResult;
}
//+--------------------------------------------------------------------------
//
// Member: CBrowserPP::AddGPOsForDomain
//
// Synopsis: Adds all the GPOs in the specified domain to the list view
// control. The szData member of the list element structure
// contains the LDAP path of the GPO.
//
// The domain is indicated by the currently selected combobox
// element.
//
// Returns: TRUE - successful
// FALSE - error
//
// History: 04-30-1998 stevebl Modified from original routine
// written by EricFlo.
//
//---------------------------------------------------------------------------
BOOL CBrowserPP::AddGPOsForDomain()
{
LPTSTR lpDomain;
LPTSTR lpGPO;
INT iIndex;
VARIANT var;
VARIANT varGPO;
ULONG ulResult;
HRESULT hr = E_FAIL;
IADsPathname * pADsPathname = NULL;
IADs * pADs = NULL;
IADsContainer * pADsContainer = NULL;
IDispatch * pDispatch = NULL;
IEnumVARIANT *pVar = NULL;
BSTR bstrContainer = NULL;
BSTR bstrCommonName = NULL;
BSTR bstrDisplayName = NULL;
BSTR bstrGPO = NULL;
TCHAR szDisplayName[512];
TCHAR szCommonName[50];
MYLISTEL * pel;
//
// Test to see if we're focused on a forest
//
BOOL fForest = IsCurrentObjectAForest();
//
// Get the current domain name
//
lpDomain = GetCurrentDomain();
if (!lpDomain)
{
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: NULL domain name.")));
return FALSE;
}
HCURSOR hcur = SetCursor(LoadCursor(NULL, IDC_WAIT));
//
// Create a pathname object we can work with
//
hr = CoCreateInstance(CLSID_Pathname, NULL, CLSCTX_INPROC_SERVER,
IID_IADsPathname, (LPVOID*)&pADsPathname);
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: Failed to create adspathname instance with 0x%x"), hr));
goto Exit;
}
//
// Add the domain name
//
hr = pADsPathname->Set (lpDomain, ADS_SETTYPE_FULL);
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: Failed to set pathname with 0x%x"), hr));
goto Exit;
}
if (fForest)
{
//
// Add the configuration folder to the path
//
hr = pADsPathname->AddLeafElement (TEXT("CN=Configuration"));
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: Failed to add system folder with 0x%x"), hr));
goto Exit;
}
}
else
{
//
// Add the system folder to the path
//
hr = pADsPathname->AddLeafElement (TEXT("CN=System"));
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: Failed to add system folder with 0x%x"), hr));
goto Exit;
}
}
//
// Add the policies container to the path
//
hr = pADsPathname->AddLeafElement (TEXT("CN=Policies"));
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: Failed to add policies folder with 0x%x"), hr));
goto Exit;
}
//
// Retreive the container path - this is the path to the policies folder
//
hr = pADsPathname->Retrieve (ADS_FORMAT_X500, &bstrContainer);
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: Failed to retreive container path with 0x%x"), hr));
goto Exit;
}
//
// Release the pathname object
//
pADsPathname->Release();
pADsPathname = NULL;
//
// Build an enumerator
//
hr = OpenDSObject(bstrContainer, IID_IADsContainer, (void **)&pADsContainer);
if (FAILED(hr))
{
if (hr != HRESULT_FROM_WIN32(ERROR_DS_NO_SUCH_OBJECT))
{
DebugMsg((DM_VERBOSE, TEXT("AddGPOsForDomain: Failed to get gpo container interface with 0x%x for object %s"),
hr, bstrContainer));
ReportError(m_hwndDlg, hr, IDS_FAILEDGPLINK);
}
goto Exit;
}
hr = ADsBuildEnumerator (pADsContainer, &pVar);
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: Failed to get enumerator with 0x%x"), hr));
goto Exit;
}
bstrCommonName = SysAllocString (L"cn");
if (!bstrCommonName)
{
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: Failed to allocate memory with %d"), GetLastError()));
goto Exit;
}
bstrDisplayName = SysAllocString (GPO_NAME_PROPERTY);
if (!bstrDisplayName)
{
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: Failed to allocate memory with %d"), GetLastError()));
goto Exit;
}
//
// Enumerate
//
while (TRUE)
{
BOOL fNeedDisplayName = FALSE;
VariantInit(&var);
hr = ADsEnumerateNext(pVar, 1, &var, &ulResult);
if (FAILED(hr))
{
DebugMsg((DM_VERBOSE, TEXT("AddGPOsForDomain: Failed to enumerator with 0x%x"), hr));
VariantClear (&var);
break;
}
if (S_FALSE == hr)
{
VariantClear (&var);
break;
}
//
// If var.vt isn't VT_DISPATCH, we're finished.
//
if (var.vt != VT_DISPATCH)
{
VariantClear (&var);
break;
}
//
// We found something, get the IDispatch interface
//
pDispatch = var.pdispVal;
if (!pDispatch)
{
DebugMsg((DM_VERBOSE, TEXT("AddGPOsForDomain: Failed to get IDispatch interface")));
goto LoopAgain;
}
//
// Now query for the IADs interface so we can get some
// properties from this GPO.
//
hr = pDispatch->QueryInterface(IID_IADs, (LPVOID *)&pADs);
if (FAILED(hr)) {
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: QI for IADs failed with 0x%x"), hr));
goto LoopAgain;
}
//
// Get the display name
//
VariantInit(&varGPO);
hr = pADs->Get(bstrDisplayName, &varGPO);
if (FAILED(hr))
{
DebugMsg((DM_VERBOSE, TEXT("AddGPOsForDomain: Failed to get display name with 0x%x"),hr));
fNeedDisplayName = TRUE;
}
else
{
wcsncpy (szDisplayName, varGPO.bstrVal, (sizeof(szDisplayName) / sizeof(szDisplayName[0])) - 1);
}
VariantClear (&varGPO);
//
// Get the common name
//
VariantInit(&varGPO);
hr = pADs->Get(bstrCommonName, &varGPO);
if (FAILED(hr))
{
DebugMsg((DM_VERBOSE, TEXT("AddGPOsForDomain: Failed to get common name with 0x%x"),hr));
VariantClear (&varGPO);
pADs->Release();
goto LoopAgain;
}
lstrcpy (szCommonName, TEXT("CN="));
lstrcat (szCommonName, varGPO.bstrVal);
VariantClear (&varGPO);
//
// Clean up
//
pADs->Release();
//
// Create a pathname object so we can tack the common name
// onto the end of the LDAP path
//
hr = CoCreateInstance(CLSID_Pathname, NULL, CLSCTX_INPROC_SERVER,
IID_IADsPathname, (LPVOID*)&pADsPathname);
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: Failed to create adspathname instance with 0x%x"), hr));
goto LoopAgain;
}
//
// Add the LDAP path
//
hr = pADsPathname->Set (bstrContainer, ADS_SETTYPE_FULL);
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: Failed to set the ldap path with 0x%x"), hr));
goto LoopAgain;
}
//
// Add the GPO's common name
//
hr = pADsPathname->AddLeafElement (szCommonName);
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: Failed to add the common name with 0x%x"), hr));
goto LoopAgain;
}
//
// Retreive the gpo path
//
hr = pADsPathname->Retrieve (ADS_FORMAT_X500, &bstrGPO);
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: Failed to retreive gpo path with 0x%x"), hr));
goto LoopAgain;
}
//
// Make a copy of it
//
lpGPO = new WCHAR[wcslen(bstrGPO) + 1];
if (!lpGPO)
{
DebugMsg((DM_WARNING, TEXT("AddGPOsForDomain: Failed to alloc memory for gpo path with 0x%x"),
GetLastError()));
goto LoopAgain;
}
wcscpy (lpGPO, bstrGPO);
pel = new MYLISTEL;
if (pel)
{
if (fNeedDisplayName)
{
pel->szName = new WCHAR[wcslen(lpGPO) + 1];
if (pel->szName)
{
CopyAsFriendlyName(pel->szName, lpGPO);
}
}
else
{
pel->szName = new WCHAR[wcslen(szDisplayName) + 1];
if (pel->szName)
{
wcscpy(pel->szName, szDisplayName);
}
}
pel->szData = lpGPO;
pel->nType = ITEMTYPE_GPO;
pel->bDisabled = FALSE;
AddElement(pel, -1);
}
LoopAgain:
if (pADsPathname)
{
pADsPathname->Release();
pADsPathname = NULL;
}
if (bstrGPO)
{
SysFreeString (bstrGPO);
bstrGPO = NULL;
}
VariantClear (&var);
}
SendMessage (m_hList, LB_SETCURSEL, 0, 0);
Exit:
if (pVar)
{
ADsFreeEnumerator (pVar);
}
if (pADsPathname)
{
pADsPathname->Release();
}
if (pADsContainer)
{
pADsContainer->Release();
}
if (bstrContainer)
{
SysFreeString (bstrContainer);
}
if (bstrCommonName)
{
SysFreeString (bstrCommonName);
}
if (bstrDisplayName)
{
SysFreeString (bstrDisplayName);
}
if (lpDomain)
{
delete [] lpDomain;
}
SetCursor(hcur);
return TRUE;
}
//+--------------------------------------------------------------------------
//
// Member: CBrowserPP::AddChildContainers
//
// Synopsis: Adds the child domains and OUs for the currently selected object
//
// History: 05-006-1998 stevebl Created
//
//---------------------------------------------------------------------------
BOOL CBrowserPP::AddChildContainers()
{
LPOLESTR szObject = NULL;
int iIndex = (int)SendMessage (m_hCombo, CB_GETCURSEL, 0, 0);
if (iIndex == CB_ERR)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::AddChildContainers: No object selected.")));
return FALSE;
}
LOOKDATA * pdata = (LOOKDATA *) SendMessage (m_hCombo, CB_GETITEMDATA, iIndex, 0);
if (pdata)
{
if (ITEMTYPE_DOMAIN == pdata->nType)
{
// make sure that domains are resolved to a server
LPTSTR szServer = ExtractServerName(pdata->szData);
if (NULL == szServer)
{
BOOL bDCFound = FALSE;
LPWSTR szTemp = GetDCName(pdata->szName, NULL, NULL, TRUE, 0);
if (szTemp)
{
LPWSTR szFullPath = MakeFullPath(pdata->szData, szTemp);
if (szFullPath)
{
LPWSTR sz = new WCHAR[wcslen(szFullPath)+1];
if (sz)
{
wcscpy(sz, szFullPath);
delete [] pdata->szData;
pdata->szData = sz;
bDCFound = TRUE;
}
LocalFree(szFullPath);
}
LocalFree(szTemp);
}
if (!bDCFound)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::AddChildContainers: Failed to get a DC name for %s"),
pdata->szName));
return FALSE;
}
}
else
{
LocalFree(szServer);
}
}
LOOKDATA * pChild = pdata->pChild;
while (pChild)
{
// Add child domains this way since ADsEnumerateNext doesn't
// seem to be giving them to us.
if (ITEMTYPE_DOMAIN == pChild->nType)
{
// got something we can work with
MYLISTEL * pel = new MYLISTEL;
if (pel)
{
memset(pel, 0, sizeof(MYLISTEL));
pel->szData = new OLECHAR[wcslen(pChild->szData) + 1];
if (pel->szData)
{
wcscpy(pel->szData, pChild->szData);
}
pel->szName = new OLECHAR[wcslen(pChild->szName) + 1];
if (pel->szName)
{
wcscpy(pel->szName, pChild->szName);
}
pel->bDisabled = FALSE;
pel->nType = ITEMTYPE_DOMAIN;
INT index = -1;
AddElement(pel, -1);
}
pChild = pChild->pSibling;
}
}
szObject = pdata->szData;
m_pPrevSel = pdata;
} else {
m_pPrevSel = NULL;
}
if ( ! szObject )
{
return FALSE;
}
HRESULT hr;
IADsContainer * pADsContainer;
hr = OpenDSObject(szObject, IID_IADsContainer, (void **)&pADsContainer);
if (SUCCEEDED(hr))
{
IEnumVARIANT *pVar;
hr = ADsBuildEnumerator(pADsContainer, &pVar);
if (SUCCEEDED(hr))
{
VARIANT var;
VariantInit(&var);
ULONG ulResult;
while (SUCCEEDED(ADsEnumerateNext(pVar, 1, &var, &ulResult)))
{
if (0 == ulResult)
{
break;
}
if (var.vt == VT_DISPATCH)
{
// query for the IADs interface so we can get its properties
IADs * pDSObject;
hr = var.pdispVal->QueryInterface(IID_IADs, (LPVOID *)&pDSObject);
if (SUCCEEDED(hr))
{
BSTR bstr;
DWORD dwType = -1;
hr = pDSObject->get_Class(&bstr);
if (SUCCEEDED(hr))
{
if (0 == wcscmp(bstr, CLASSNAME_OU))
{
dwType = ITEMTYPE_OU;
}
else if (0 == wcscmp(bstr, CLASSNAME_DOMAIN))
{
dwType = ITEMTYPE_DOMAIN;
}
SysFreeString(bstr);
}
if (ITEMTYPE_DOMAIN == dwType || ITEMTYPE_OU == dwType)
{
// got something we can work with
MYLISTEL * pel = new MYLISTEL;
if (pel)
{
memset(pel, 0, sizeof(MYLISTEL));
hr = pDSObject->get_ADsPath(&bstr);
if (SUCCEEDED(hr))
{
pel->szData = new OLECHAR[wcslen(bstr) + 1];
if (pel->szData)
{
wcscpy(pel->szData, bstr);
}
pel->szName = new OLECHAR[wcslen(bstr) + 1];
if (pel->szName)
{
// Need to convert to a friendly name.
CopyAsFriendlyName(pel->szName, bstr);
}
SysFreeString(bstr);
}
pel->nType = dwType;
INT index = -1;
AddElement(pel, -1);
}
}
pDSObject->Release();
}
}
VariantClear(&var);
}
ADsFreeEnumerator(pVar);
}
pADsContainer->Release();
}
return TRUE;
}
//+--------------------------------------------------------------------------
//
// Member: CBrowserPP::RefreshDomains
//
// Synopsis: refreshes the listview for the "domains" page
//
// History: 04-30-1998 stevebl Created
//
//---------------------------------------------------------------------------
void CBrowserPP::RefreshDomains()
{
LONG lStyle;
ListView_DeleteAllItems(m_hList);
lStyle = GetWindowLong (m_hList, GWL_STYLE);
lStyle &= ~LVS_SORTASCENDING;
SetWindowLong (m_hList, GWL_STYLE, lStyle);
if (AddChildContainers())
{
AddGPOsLinkedToObject();
EnableWindow (m_hList, TRUE);
if (!(m_pGBI->dwFlags & GPO_BROWSE_DISABLENEW)) {
SendMessage (m_toolbar, TB_ENABLEBUTTON, (WPARAM) ID_NEWFOLDER, (LPARAM) MAKELONG(1, 0));
}
}
else
{
EnableWindow (m_hList, FALSE);
SendMessage (m_toolbar, TB_ENABLEBUTTON, (WPARAM) ID_NEWFOLDER, (LPARAM) MAKELONG(0, 0));
}
}
//+--------------------------------------------------------------------------
//
// Member: CBrowserPP::RefreshSites
//
// Synopsis: refreshes the listview for the "sites" page
//
// History: 04-30-1998 stevebl Created
//
//---------------------------------------------------------------------------
void CBrowserPP::RefreshSites()
{
LONG lStyle;
ListView_DeleteAllItems(m_hList);
lStyle = GetWindowLong (m_hList, GWL_STYLE);
lStyle &= ~LVS_SORTASCENDING;
SetWindowLong (m_hList, GWL_STYLE, lStyle);
AddGPOsLinkedToObject();
}
//+--------------------------------------------------------------------------
//
// Member: CBrowserPP::RefreshAll
//
// Synopsis: refreshes the listview for the "all" page
//
// History: 04-30-1998 stevebl Created
//
//---------------------------------------------------------------------------
void CBrowserPP::RefreshAll()
{
LONG lStyle;
ListView_DeleteAllItems(m_hList);
lStyle = GetWindowLong (m_hList, GWL_STYLE);
lStyle |= LVS_SORTASCENDING;
SetWindowLong (m_hList, GWL_STYLE, lStyle);
if (AddGPOsForDomain())
{
EnableWindow (m_hList, TRUE);
SendMessage (m_toolbar, TB_ENABLEBUTTON, (WPARAM) ID_NEWFOLDER, (LPARAM) MAKELONG(1, 0));
}
else
{
EnableWindow (m_hList, FALSE);
SendMessage (m_toolbar, TB_ENABLEBUTTON, (WPARAM) ID_NEWFOLDER, (LPARAM) MAKELONG(0, 0));
}
}
void CBrowserPP::SetButtonState()
{
if (ListView_GetNextItem (m_hList, -1, LVNI_ALL | LVNI_SELECTED) != -1)
{
EnableWindow (GetDlgItem(GetParent(m_hwndDlg), IDOK), TRUE);
}
else
{
EnableWindow (GetDlgItem(GetParent(m_hwndDlg), IDOK), FALSE);
}
}
BOOL CBrowserPP::OnInitDialog()
{
DWORD dwDescription;
switch (m_dwPageType)
{
case PAGETYPE_DOMAINS:
dwDescription = IDS_DOMAINDESCRIPTION;
break;
case PAGETYPE_SITES:
dwDescription = IDS_SITEDESCRIPTION;
break;
case PAGETYPE_ALL:
default:
dwDescription = IDS_ALLDESCRIPTION;
break;
}
WCHAR szDescription[MAX_PATH]; // this is a resource - size doesn't need to be dynamic
LoadString(g_hInstance, dwDescription, szDescription, MAX_PATH);
SetDlgItemText(m_hwndDlg, IDC_DESCRIPTION, szDescription);
m_hList = GetDlgItem(m_hwndDlg, IDC_LIST1);
m_ilSmall = ImageList_LoadBitmap(g_hInstance, MAKEINTRESOURCE(IDB_16x16), SMALLICONSIZE, 0, RGB(255,0,255));
m_ilLarge = ImageList_LoadBitmap(g_hInstance, MAKEINTRESOURCE(IDB_32x32), LARGEICONSIZE, 0, RGB(255, 0 ,255));
m_hCombo = GetDlgItem(m_hwndDlg, IDC_COMBO1);
RECT rect;
GetClientRect(m_hList, &rect);
WCHAR szText[32];
int dxScrollBar = GetSystemMetrics(SM_CXVSCROLL);
if (PAGETYPE_ALL == m_dwPageType)
{
LV_COLUMN lvcol;
memset(&lvcol, 0, sizeof(lvcol));
lvcol.mask = LVCF_FMT | LVCF_TEXT | LVCF_WIDTH;
lvcol.fmt = LVCFMT_LEFT;
lvcol.cx = (rect.right - rect.left) - dxScrollBar;
LoadString(g_hInstance, IDS_NAMECOLUMN, szText, 32);
lvcol.pszText = szText;
ListView_InsertColumn(m_hList, 0, &lvcol);
}
else
{
LV_COLUMN lvcol;
memset(&lvcol, 0, sizeof(lvcol));
lvcol.mask = LVCF_FMT | LVCF_TEXT | LVCF_WIDTH;
lvcol.fmt = LVCFMT_LEFT;
int cx = ((rect.right - rect.left) - dxScrollBar)*2/3;
lvcol.cx = cx;
LoadString(g_hInstance, IDS_NAMECOLUMN, szText, 32);
lvcol.pszText = szText;
ListView_InsertColumn(m_hList, 0, &lvcol);
memset(&lvcol, 0, sizeof(lvcol));
lvcol.mask = LVCF_FMT | LVCF_TEXT | LVCF_WIDTH;
lvcol.fmt = LVCFMT_LEFT;
lvcol.cx = ((rect.right - rect.left) - dxScrollBar) - cx;
LoadString(g_hInstance, IDS_DOMAINCOLUMN, szText, 32);
lvcol.pszText = szText;
ListView_InsertColumn(m_hList, 1, &lvcol);
}
ListView_SetImageList(m_hList, m_ilSmall, LVSIL_SMALL);
ListView_SetImageList(m_hList, m_ilLarge, LVSIL_NORMAL);
SendMessage(m_hList, LVM_SETEXTENDEDLISTVIEWSTYLE, 0, LVS_EX_LABELTIP);
GetWindowRect(GetDlgItem(m_hwndDlg, IDC_STATIC1), &rect);
MapWindowPoints(NULL , m_hwndDlg, (LPPOINT) &rect , 2);
TBBUTTON rgButtons[3];
rgButtons[0].iBitmap = 0;
rgButtons[0].idCommand = ID_BACKBUTTON;
rgButtons[0].fsState = 0; // this button will be disabled by
// default and only enabled when there
// is something to back up to
//rgButtons[0].fsState = PAGETYPE_ALL == m_dwPageType ? 0 : TBSTATE_ENABLED;
rgButtons[0].fsStyle = TBSTYLE_BUTTON;
rgButtons[0].dwData = 0;
rgButtons[0].iString = 0;
rgButtons[1].iBitmap = 1;
rgButtons[1].idCommand = ID_NEWFOLDER;
rgButtons[1].fsStyle = TBSTYLE_BUTTON;
rgButtons[1].dwData = 0;
rgButtons[1].iString = 0;
if (PAGETYPE_ALL != m_dwPageType)
{
if (m_pGBI->dwFlags & GPO_BROWSE_DISABLENEW)
{
rgButtons[1].fsState = 0;
}
else
{
rgButtons[1].fsState = TBSTATE_ENABLED;
}
}
else
{
rgButtons[1].fsState =TBSTATE_ENABLED;
}
rgButtons[2].iBitmap = 2;
rgButtons[2].idCommand = ID_ROTATEVIEW;
rgButtons[2].fsState = TBSTATE_ENABLED ;
rgButtons[2].fsStyle = TBSTYLE_DROPDOWN;
rgButtons[2].dwData = 0;
rgButtons[2].iString = 0;
m_toolbar = CreateToolbarEx(m_hwndDlg,
WS_CHILD | WS_VISIBLE | CCS_NODIVIDER | CCS_NORESIZE | TBSTYLE_FLAT | TBSTYLE_TOOLTIPS,
IDR_TOOLBAR1,
4,
g_hInstance,
IDR_TOOLBAR1,
rgButtons,
3,
BUTTONSIZE,
BUTTONSIZE,
BUTTONSIZE,
BUTTONSIZE,
sizeof(TBBUTTON));
SendMessage(m_toolbar, TB_SETEXTENDEDSTYLE, TBSTYLE_EX_DRAWDDARROWS, TBSTYLE_EX_DRAWDDARROWS);
MoveWindow(m_toolbar, rect.left, rect.top, rect.right-rect.left, rect.bottom-rect.top, FALSE);
// Don't need to call Refresh in any of these because we're calling it in
// OnComboChange().
switch (m_dwPageType)
{
case PAGETYPE_DOMAINS:
FillDomainList();
SetInitialOU();
// RefreshDomains();
break;
case PAGETYPE_SITES:
SendMessage(m_hCombo, CB_RESETCONTENT, (WPARAM) 0, (LPARAM) 0);
FillSitesList();
// RefreshSites();
break;
default:
case PAGETYPE_ALL:
SendMessage(m_hCombo, CB_RESETCONTENT, (WPARAM) 0, (LPARAM) 0);
FillDomainList();
// RefreshAll();
break;
}
SetButtonState();
return TRUE; // return TRUE unless you set the focus to a control
// EXCEPTION: OCX Property Pages should return FALSE
}
BOOL CBrowserPP::DoBackButton()
{
int iIndex = (int)SendMessage (m_hCombo, CB_GETCURSEL, 0, 0);
if (iIndex == CB_ERR)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::DoBackButton: No object selected.")));
return FALSE;
}
LOOKDATA * pdata = (LOOKDATA *) SendMessage (m_hCombo, CB_GETITEMDATA, iIndex, 0);
if (pdata)
{
if (pdata->pParent)
{
// if this item has a parent then select it
SendMessage(m_hCombo, CB_SELECTSTRING, (WPARAM)-1, (LPARAM) (LPCTSTR) pdata->pParent);
// force everything to refresh
OnComboChange();
}
}
return FALSE;
}
BOOL CBrowserPP::DeleteGPO()
{
BOOL fSucceeded = FALSE;
BOOL fRemoveListEntry = FALSE;
int index = ListView_GetNextItem(m_hList, -1, LVNI_SELECTED);
if (-1 == index)
{
return FALSE;
}
LVITEM item;
memset(&item, 0, sizeof(item));
item.mask = LVIF_PARAM;
item.iItem = index;
ListView_GetItem(m_hList, &item);
MYLISTEL * pel = (MYLISTEL *)item.lParam;
LPGROUPPOLICYOBJECT pGPO = NULL;
HRESULT hr;
WCHAR szBuffer[100];
WCHAR szConfirm[MAX_FRIENDLYNAME + 100];
WCHAR szTitle[100];
if (pel->nType != ITEMTYPE_GPO)
{
goto CleanUp;
}
LoadString(g_hInstance, IDS_CONFIRMTITLE, szTitle, 100);
LoadString(g_hInstance, IDS_DELETECONFIRM, szBuffer, 100);
wsprintf (szConfirm, szBuffer, pel->szName);
if (IDNO == MessageBox(m_hwndDlg, szConfirm, szTitle, MB_YESNO | MB_ICONEXCLAMATION))
{
goto CleanUp;
}
// If we're on any page other than the "All" page then we need to break
// the association before we can delete the object.
if (m_dwPageType != PAGETYPE_ALL)
{
// break the association
LPOLESTR szContainer = GetCurrentObject();
if (szContainer)
{
DeleteLink(pel->szData, szContainer);
delete [] szContainer;
}
}
hr = CoCreateInstance(CLSID_GroupPolicyObject, NULL,
CLSCTX_SERVER, IID_IGroupPolicyObject,
(void **)&pGPO);
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("CoCreateInstance failed with 0x%x\r\n"), hr));
goto Done;
}
// open GPO object without opening registry data
hr = pGPO->OpenDSGPO(pel->szData, 0);
if (FAILED(hr))
{
ReportError(m_hwndDlg, hr, IDS_FAILEDDS);
DebugMsg((DM_WARNING, TEXT("OpenDSGPO failed with 0x%x\r\n"), hr));
goto Done;
}
// delete it
hr = pGPO->Delete();
if (FAILED(hr))
{
ReportError(m_hwndDlg, hr, IDS_FAILEDDELETE);
DebugMsg((DM_WARNING, TEXT("Delete failed with 0x%x\r\n"), hr));
goto Done;
}
fRemoveListEntry = TRUE;
Done:
if (pGPO)
{
pGPO->Release();
}
// remove the list entry
if (fRemoveListEntry)
fSucceeded = ListView_DeleteItem(m_hList, index);
CleanUp:
return fSucceeded;
}
BOOL CBrowserPP::DoNewGPO()
{
BOOL fSucceeded = FALSE;
HRESULT hr;
LPGROUPPOLICYOBJECT pGPO = NULL;
BOOL fEdit = FALSE;
MYLISTEL * pel = NULL;
LPOLESTR szObject = GetCurrentObject();
LPOLESTR szDomain = GetCurrentDomain();
INT index = -1;
int cch = 0;
LPTSTR szFullPath = NULL;
LPTSTR szServerName = NULL;
DWORD dwOptions = 0;
if (NULL == szDomain)
{
goto Done;
}
if (NULL == szObject)
{
goto Done;
}
pel = new MYLISTEL;
if (NULL == pel)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::DoNewGPO failed to allocate memory for GPO name")));
goto Done;
}
pel->bDisabled = FALSE;
pel->szData = NULL;
pel->szName = new OLECHAR[MAX_FRIENDLYNAME];
if (NULL == pel->szName)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::DoNewGPO failed to allocate memory for GPO name")));
goto Done;
}
GetNewGPODisplayName (pel->szName, MAX_FRIENDLYNAME);
pel->nType = ITEMTYPE_GPO;
// Create a new GPO named "New Group Policy Object"
//
// Create a new GPO object to work with
//
hr = CoCreateInstance (CLSID_GroupPolicyObject, NULL,
CLSCTX_SERVER, IID_IGroupPolicyObject,
(void**)&pGPO);
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("CoCreateInstance failed with 0x%x\r\n"), hr));
goto Done;
}
//
// Open the requested object without mounting the registry
//
#if FGPO_SUPPORT
if (IsCurrentObjectAForest())
{
dwOptions = GPO_OPEN_FOREST;
}
#endif
hr = pGPO->New(szDomain, pel->szName, dwOptions);
if (FAILED(hr))
{
ReportError(m_hwndDlg, hr, IDS_FAILEDNEW);
DebugMsg((DM_WARNING, TEXT("Failed to create GPO object with 0x%x\r\n"), hr));
goto Done;
}
// continue to try to allocate memory until either a big enough buffer is
// created to load the GPO path or we run out of memory
pel->szData = NULL;
do
{
if (pel->szData)
{
delete [] pel->szData;
}
cch += MAX_PATH;
pel->szData = new OLECHAR[cch];
if (NULL == pel->szData)
{
}
hr = pGPO->GetPath(pel->szData, cch);
} while (hr == E_OUTOFMEMORY);
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("Failed to get GPO object path with 0x%x\r\n"), hr));
goto Done;
}
szServerName = ExtractServerName(szDomain);
szFullPath = MakeFullPath(pel->szData, szServerName);
if (szFullPath)
{
delete [] pel->szData;
pel->szData = new OLECHAR[wcslen(szFullPath) + 1];
if (NULL == pel->szData)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::DoNewGPO failed to allocate memory for GPO path")));
goto Done;
}
wcscpy(pel->szData, szFullPath);
}
if (m_dwPageType != PAGETYPE_ALL)
{
// If we're not on the "All" page then we need to create a link.
CreateLink(pel->szData, szObject);
}
// Add the entry to the list view
index = AddElement(pel, -1);
fSucceeded = index != -1;
// It's been added so now we need to make sure we don't delete it below
pel = NULL;
// Record that we got this far
fEdit = TRUE;
Done:
if (pel)
{
if (pel->szData)
{
delete [] pel->szData;
}
if (pel->szName)
{
delete [] pel->szName;
}
delete pel;
}
if (pGPO)
pGPO->Release();
if (fEdit)
{
// Now trigger an edit of the entry
SetFocus(m_hList);
ListView_EditLabel(m_hList, index);
}
if (szServerName)
LocalFree(szServerName);
if (szFullPath)
LocalFree(szFullPath);
if (szDomain)
delete [] szDomain;
if (szObject)
delete [] szObject;
return fSucceeded;
}
BOOL CBrowserPP::CreateLink(LPOLESTR szObject, LPOLESTR szContainer)
{
HRESULT hr = CreateGPOLink(szObject, szContainer, FALSE);
if (SUCCEEDED(hr))
{
return TRUE;
}
ReportError(m_hwndDlg, hr, IDS_FAILEDLINK);
return FALSE;
}
BOOL CBrowserPP::DeleteLink(LPOLESTR szObject, LPOLESTR szContainer)
{
HRESULT hr = DeleteGPOLink(szObject, szContainer);
if (SUCCEEDED(hr))
{
return TRUE;
}
ReportError(m_hwndDlg, hr, IDS_FAILEDUNLINK);
return FALSE;
}
BOOL CBrowserPP::DoRotateView()
{
DWORD dwStyle = GetWindowLong(m_hList, GWL_STYLE);
DWORD dw = dwStyle & LVS_TYPEMASK;
switch (dw)
{
case LVS_ICON:
dw = LVS_SMALLICON;
break;
case LVS_SMALLICON:
dw = LVS_LIST;
break;
case LVS_REPORT:
dw = LVS_ICON;
break;
case LVS_LIST:
default:
dw = LVS_REPORT;
break;
}
dwStyle -= dwStyle & LVS_TYPEMASK;
dwStyle += dw;
SetWindowLong(m_hList, GWL_STYLE, dwStyle);
return TRUE;
}
void CBrowserPP::OnDetails()
{
DWORD dwStyle = GetWindowLong(m_hList, GWL_STYLE);
dwStyle -= dwStyle & LVS_TYPEMASK;
SetWindowLong(m_hList, GWL_STYLE, dwStyle + LVS_REPORT);
}
void CBrowserPP::OnList()
{
DWORD dwStyle = GetWindowLong(m_hList, GWL_STYLE);
dwStyle -= dwStyle & LVS_TYPEMASK;
SetWindowLong(m_hList, GWL_STYLE, dwStyle + LVS_LIST);
}
void CBrowserPP::OnLargeicons()
{
DWORD dwStyle = GetWindowLong(m_hList, GWL_STYLE);
dwStyle -= dwStyle & LVS_TYPEMASK;
SetWindowLong(m_hList, GWL_STYLE, dwStyle + LVS_ICON);
}
void CBrowserPP::OnSmallicons()
{
DWORD dwStyle = GetWindowLong(m_hList, GWL_STYLE);
dwStyle -= dwStyle & LVS_TYPEMASK;
SetWindowLong(m_hList, GWL_STYLE, dwStyle + LVS_SMALLICON);
}
void CBrowserPP::OnContextMenu(LPARAM lParam)
{
int i = ListView_GetNextItem(m_hList, -1, LVNI_SELECTED);
RECT rc;
POINT pt;
pt.x = ((int)(short)LOWORD(lParam));
pt.y = ((int)(short)HIWORD(lParam));
GetWindowRect (GetDlgItem (m_hwndDlg, IDC_LIST1), &rc);
if (!PtInRect (&rc, pt))
{
if ((lParam == (LPARAM) -1) && (i >= 0))
{
rc.left = LVIR_SELECTBOUNDS;
SendMessage (m_hList, LVM_GETITEMRECT, i, (LPARAM) &rc);
pt.x = rc.left + 8;
pt.y = rc.top + ((rc.bottom - rc.top) / 2);
ClientToScreen (m_hList, &pt);
}
else
{
pt.x = rc.left + ((rc.right - rc.left) / 2);
pt.y = rc.top + ((rc.bottom - rc.top) / 2);
}
}
// get the popup menu
HMENU hPopup;
hPopup = LoadMenu(g_hInstance, MAKEINTRESOURCE(IDR_LISTMENU));
HMENU hSubMenu = GetSubMenu(hPopup, 0);
if (i >= 0)
{
// item selected
// figure out what type it is
LVITEM item;
memset(&item, 0, sizeof(item));
item.mask = LVIF_PARAM;
item.iItem = i;
ListView_GetItem(m_hList, &item);
MYLISTEL * pel = (MYLISTEL *)item.lParam;
// get rid of the view menu and separator
RemoveMenu(hSubMenu, 0, MF_BYPOSITION);
RemoveMenu(hSubMenu, 0, MF_BYPOSITION);
// get rid of the arrange and line-up items
RemoveMenu(hSubMenu, 0, MF_BYPOSITION);
RemoveMenu(hSubMenu, 0, MF_BYPOSITION);
RemoveMenu(hSubMenu, 0, MF_BYPOSITION);
// get rid of the "new" menu item
RemoveMenu(hSubMenu, ID_NEW, MF_BYCOMMAND);
switch (pel->nType)
{
case ITEMTYPE_GPO:
if (pel->bDisabled)
{
// disable edit, rename, delete
EnableMenuItem(hSubMenu, ID_EDIT, MF_BYCOMMAND | MF_DISABLED | MF_GRAYED);
EnableMenuItem(hSubMenu, ID_RENAME, MF_BYCOMMAND | MF_DISABLED | MF_GRAYED);
EnableMenuItem(hSubMenu, ID_DELETE, MF_BYCOMMAND | MF_DISABLED | MF_GRAYED);
}
break;
default:
case ITEMTYPE_FOREST:
case ITEMTYPE_SITE:
case ITEMTYPE_DOMAIN:
case ITEMTYPE_OU:
// remove the edit menu item and the separator
RemoveMenu(hSubMenu, ID_EDIT, MF_BYCOMMAND);
RemoveMenu(hSubMenu, 0, MF_BYPOSITION);
// disable rename, delete and properties
EnableMenuItem(hSubMenu, ID_RENAME, MF_BYCOMMAND | MF_DISABLED | MF_GRAYED);
EnableMenuItem(hSubMenu, ID_DELETE, MF_BYCOMMAND | MF_DISABLED | MF_GRAYED);
EnableMenuItem(hSubMenu, ID_PROPERTIES, MF_BYCOMMAND | MF_DISABLED | MF_GRAYED);
break;
}
}
else
{
// no item selected
// get rid of the edit menu item
RemoveMenu(hSubMenu, ID_EDIT, MF_BYCOMMAND);
// get rid of the delete and rename items
RemoveMenu(hSubMenu, ID_DELETE, MF_BYCOMMAND);
RemoveMenu(hSubMenu, ID_RENAME, MF_BYCOMMAND);
if (PAGETYPE_ALL != m_dwPageType)
{
if (m_pGBI->dwFlags & GPO_BROWSE_DISABLENEW)
{
// get rid of the "new" menu item
RemoveMenu(hSubMenu, ID_NEW, MF_BYCOMMAND);
RemoveMenu(hSubMenu, 4, MF_BYPOSITION);
}
}
RemoveMenu(hSubMenu, (GetMenuItemCount(hSubMenu) - 1), MF_BYPOSITION);
RemoveMenu(hSubMenu, (GetMenuItemCount(hSubMenu) - 1), MF_BYPOSITION);
// set view radio button
UINT ui = ID_LIST;
DWORD dw = GetWindowLong(m_hList, GWL_STYLE) & LVS_TYPEMASK;
if (dw == LVS_ICON || dw == LVS_SMALLICON)
{
// Auto-Arrange means something in these views so we need to enable it
EnableMenuItem(hSubMenu, ID_ARRANGE_AUTO, MF_BYCOMMAND | MF_ENABLED);
// also need to make sure it's set correctly
if (LVS_AUTOARRANGE == (GetWindowLong(m_hList, GWL_STYLE) & LVS_AUTOARRANGE))
CheckMenuItem(hSubMenu, ID_ARRANGE_AUTO, MF_BYCOMMAND | MF_CHECKED);
}
switch (dw)
{
case LVS_ICON:
ui = ID_LARGEICONS;
break;
case LVS_SMALLICON:
ui = ID_SMALLICONS;
break;
case LVS_REPORT:
ui = ID_DETAILS;
break;
case LVS_LIST:
default:
ui = ID_LIST;
break;
}
CheckMenuRadioItem(hSubMenu, ui, ui, ui, MF_BYCOMMAND);
}
TrackPopupMenu(hSubMenu,
TPM_LEFTALIGN,
pt.x, pt.y,
0,
m_hwndDlg,
NULL);
DestroyMenu(hPopup);
}
void CBrowserPP::OnArrangeAuto()
{
DWORD dwStyle = GetWindowLong(m_hList, GWL_STYLE);
if (LVS_AUTOARRANGE == (dwStyle & LVS_AUTOARRANGE))
SetWindowLong(m_hList, GWL_STYLE, dwStyle - LVS_AUTOARRANGE);
else
SetWindowLong(m_hList, GWL_STYLE, dwStyle + LVS_AUTOARRANGE);
}
int CALLBACK CompareName(LPARAM lParam1, LPARAM lParam2, LPARAM lParamsort)
{
MYLISTEL * pel1 = (MYLISTEL *)lParam1;
MYLISTEL * pel2 = (MYLISTEL *)lParam2;
return _wcsicmp(pel1->szName, pel2->szName);
}
int CALLBACK CompareType(LPARAM lParam1, LPARAM lParam2, LPARAM lParamsort)
{
MYLISTEL * pel1 = (MYLISTEL *)lParam1;
MYLISTEL * pel2 = (MYLISTEL *)lParam2;
return pel1->nType - pel2->nType;
}
void CBrowserPP::OnArrangeByname()
{
ListView_SortItems(m_hList, CompareName, 0);
}
void CBrowserPP::OnArrangeBytype()
{
ListView_SortItems(m_hList, CompareType, 0);
}
void CBrowserPP::OnDelete()
{
DeleteGPO();
}
void CBrowserPP::OnEdit()
{
INT i;
HRESULT hr;
LVITEM item;
MYLISTEL * pel;
LPTSTR lpDomainName;
LPOLESTR pszDomain;
i = ListView_GetNextItem(m_hList, -1, LVNI_SELECTED);
if (i >= 0)
{
memset(&item, 0, sizeof(item));
item.mask = LVIF_PARAM;
item.iItem = i;
ListView_GetItem(m_hList, &item);
pel = (MYLISTEL *)item.lParam;
if (pel->nType == ITEMTYPE_GPO)
{
//
// Get the friendly domain name
//
pszDomain = GetDomainFromLDAPPath(pel->szData);
if (!pszDomain)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::OnEdit: Failed to get domain name")));
return;
}
//
// Convert LDAP to dot (DN) style
//
hr = ConvertToDotStyle (pszDomain, &lpDomainName);
delete [] pszDomain;
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("CGroupPolicyObject::CreatePropertyPages: Failed to convert domain name with 0x%x"), hr));
return;
}
//
// Check if the GPO is in the same domain as GPM is focused on
//
if (!lstrcmpi(lpDomainName, m_szDomainName))
{
SpawnGPE (pel->szData, GPHintUnknown, m_szServerName, m_hwndDlg);
}
else
{
SpawnGPE (pel->szData, GPHintUnknown, NULL, m_hwndDlg);
}
LocalFree (lpDomainName);
}
}
}
void CBrowserPP::OnNew()
{
DoNewGPO();
}
void CBrowserPP::OnProperties()
{
INT iIndex;
LVITEM item;
HRESULT hr;
LPGROUPPOLICYOBJECT pGPO;
HPROPSHEETPAGE *hPages;
UINT i, uPageCount;
PROPSHEETHEADER psh;
iIndex = ListView_GetNextItem(m_hList, -1, LVNI_ALL | LVNI_SELECTED);
if (iIndex >= 0)
{
memset(&item, 0, sizeof(item));
item.mask = LVIF_PARAM;
item.iItem = iIndex;
ListView_GetItem(m_hList, &item);
MYLISTEL * pel = (MYLISTEL *)item.lParam;
if (pel && pel->nType == ITEMTYPE_GPO)
{
hr = CoCreateInstance (CLSID_GroupPolicyObject, NULL,
CLSCTX_SERVER, IID_IGroupPolicyObject,
(void**)&pGPO);
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::OnProperties: CoCreateInstance failed with 0x%x\r\n"), hr));
return;
}
//
// Open the requested object without mounting the registry
//
hr = pGPO->OpenDSGPO(pel->szData, 0);
if (hr == HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED))
{
hr = pGPO->OpenDSGPO(pel->szData, GPO_OPEN_READ_ONLY);
}
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::OnProperties: Failed to open GPO object with 0x%x\r\n"), hr));
ReportError(m_hwndDlg, hr, IDS_FAILEDDS);
return;
}
//
// Ask the GPO for the property sheet pages
//
hr = pGPO->GetPropertySheetPages (&hPages, &uPageCount);
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::OnProperties: Failed to query property sheet pages with 0x%x."), hr));
pGPO->Release();
return;
}
//
// Display the property sheet
//
ZeroMemory (&psh, sizeof(psh));
psh.dwSize = sizeof(psh);
psh.dwFlags = PSH_PROPTITLE;
psh.hwndParent = m_hwndDlg;
psh.hInstance = g_hInstance;
psh.pszCaption = pel->szName;
psh.nPages = uPageCount;
psh.phpage = hPages;
PropertySheet (&psh);
LocalFree (hPages);
pGPO->Release();
}
}
}
void CBrowserPP::OnRefresh()
{
switch (m_dwPageType)
{
case PAGETYPE_DOMAINS:
RefreshDomains();
break;
case PAGETYPE_SITES:
RefreshSites();
break;
default:
case PAGETYPE_ALL:
RefreshAll();
break;
}
SetButtonState();
}
void CBrowserPP::OnRename()
{
//
// alow the rename only if it is possible to rename
//
int i = ListView_GetNextItem(m_hList, -1, LVNI_SELECTED);
if (i >= 0)
{
// item selected
// figure out what type it is
LVITEM item;
memset(&item, 0, sizeof(item));
item.mask = LVIF_PARAM;
item.iItem = i;
ListView_GetItem(m_hList, &item);
MYLISTEL * pel = (MYLISTEL *)item.lParam;
if ((pel) && (pel->nType == ITEMTYPE_GPO)
&& (!(pel->bDisabled))) {
ListView_EditLabel(m_hList, ListView_GetNextItem(m_hList, -1, LVNI_SELECTED));
}
}
}
void CBrowserPP::OnTopLineupicons()
{
ListView_Arrange(m_hList, LVA_SNAPTOGRID);
}
void CBrowserPP::OnBeginlabeleditList(NMHDR* pNMHDR, LRESULT* pResult)
{
LV_DISPINFO* pDispInfo = (LV_DISPINFO*)pNMHDR;
// Return FALSE to enable editing, TRUE to disable it
MYLISTEL * pel = (MYLISTEL *)pDispInfo->item.lParam;
*pResult = (pel->nType == ITEMTYPE_GPO) ? FALSE : TRUE;
}
void CBrowserPP::OnEndlabeleditList(NMHDR* pNMHDR, LRESULT* pResult)
{
*pResult = FALSE;
LPGROUPPOLICYOBJECT pGPO = NULL;
HRESULT hr;
LV_DISPINFO* pDispInfo = (LV_DISPINFO*)pNMHDR;
if (NULL == pDispInfo->item.pszText)
{
// user cancelled edit
return;
}
if (TEXT('\0') == (*pDispInfo->item.pszText))
{
// user entered an empty string
return;
}
MYLISTEL * pel = (MYLISTEL *)pDispInfo->item.lParam;
if (0 ==wcscmp(pDispInfo->item.pszText, pel->szName))
{
// user didn't change anything
return;
}
LPWSTR sz = new WCHAR[wcslen(pDispInfo->item.pszText)+1];
if (NULL == sz)
{
*pResult = FALSE;
goto Done;
return;
}
hr = CoCreateInstance(CLSID_GroupPolicyObject, NULL,
CLSCTX_SERVER, IID_IGroupPolicyObject,
(void **)&pGPO);
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("CoCreateInstance failed with 0x%x\r\n"), hr));
goto Done;
}
// open GPO object without opening registry data
hr = pGPO->OpenDSGPO(pel->szData, 0);
if (FAILED(hr))
{
ReportError(m_hwndDlg, hr, IDS_FAILEDDS);
DebugMsg((DM_WARNING, TEXT("OpenDSGPO failed with 0x%x\r\n"), hr));
goto Done;
}
// rename it
hr = pGPO->SetDisplayName(pDispInfo->item.pszText);
if (FAILED(hr))
{
ReportError(m_hwndDlg, hr, IDS_FAILEDSETNAME);
DebugMsg((DM_WARNING, TEXT("SetDisplayName failed with 0x%x\r\n"), hr));
goto Done;
}
// requery for the name
hr = pGPO->GetDisplayName(sz, (wcslen(pDispInfo->item.pszText)+1));
if (FAILED(hr))
{
ReportError(m_hwndDlg, hr, IDS_FAILEDSETNAME);
DebugMsg((DM_WARNING, TEXT("GetDisplayName failed with 0x%x\r\n"), hr));
goto Done;
}
delete [] pel->szName;
pel->szName = sz;
sz = NULL;
// return TRUE to accept the rename, FALSE to reject it
*pResult = TRUE;
PostMessage (m_hwndDlg, WM_REFRESHDISPLAY, (WPARAM) pDispInfo->item.iItem, 0);
Done:
if (sz)
{
delete [] sz;
}
if (pGPO)
{
pGPO->Release();
}
}
void CBrowserPP::OnBegindragList(NMHDR* pNMHDR, LRESULT* pResult)
{
NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
*pResult = 0;
}
void CBrowserPP::OnDeleteitemList(NMHDR* pNMHDR, LRESULT* pResult)
{
NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
MYLISTEL * pel = (MYLISTEL *)pNMListView->lParam;
if (pel)
{
if (pel->szName)
{
delete [] pel->szName;
}
if (pel->szData)
{
delete [] pel->szData;
}
delete pel;
}
*pResult = 0;
}
void CBrowserPP::OnDoubleclickList(NMHDR* pNMHDR, LRESULT* pResult)
{
NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
if (pNMListView->iItem >= 0)
{
// item selected
PropSheet_PressButton(GetParent(m_hwndDlg), PSBTN_OK);
}
*pResult = 0;
}
void CBrowserPP::OnColumnclickList(NMHDR* pNMHDR, LRESULT* pResult)
{
NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
switch (pNMListView->iSubItem)
{
case 0:
ListView_SortItems(m_hList, CompareName, 0);
break;
case 1:
default:
ListView_SortItems(m_hList, CompareType, 0);
break;
}
*pResult = 0;
}
void CBrowserPP::OnKeyDownList(NMHDR * pNMHDR, LRESULT * pResult)
{
LV_KEYDOWN * pnkd = (LV_KEYDOWN *)pNMHDR;
switch (pnkd->wVKey)
{
case VK_F5:
OnRefresh();
break;
case VK_F2:
OnRename();
break;
case VK_DELETE:
OnDelete();
break;
case VK_BACK:
DoBackButton();
break;
case VK_RETURN:
OnProperties();
break;
default:
break;
}
}
void CBrowserPP::OnItemChanged(NMHDR * pNMHDR, LRESULT * pResult)
{
SetButtonState();
}
void CBrowserPP::TrimComboBox()
{
LOOKDATA * pdataSelected = NULL;
int iCount;
// first check to see if something is selected
int iIndex = (int)SendMessage (m_hCombo, CB_GETCURSEL, 0, 0);
if (iIndex != CB_ERR)
{
// something's selected, get a pointer to it's data
pdataSelected = (LOOKDATA *) SendMessage (m_hCombo, CB_GETITEMDATA, iIndex, 0);
// check if the user selected the same thing again
if (m_pPrevSel && (m_pPrevSel == pdataSelected))
{
return;
}
// if it has a parent then enable the back button
SendMessage(m_toolbar, TB_ENABLEBUTTON, (WPARAM) ID_BACKBUTTON, (LPARAM)MAKELONG(NULL != pdataSelected->pParent, 0));
}
// If the child of the selected object is an OU then delete all of it's children
// otherwise delete ALL OUs from the list.
if (pdataSelected)
{
if (pdataSelected->pChild)
{
if (ITEMTYPE_OU == pdataSelected->pChild->nType)
{
// delete all of its children
goto DeleteChildren;
}
}
}
iCount = (int)SendMessage(m_hCombo, CB_GETCOUNT, 0, 0);
iIndex = 0;
while (iIndex < iCount)
{
// find the first entry that has an OU for a child.
pdataSelected = (LOOKDATA *) SendMessage (m_hCombo, CB_GETITEMDATA, iIndex, 0);
if (pdataSelected)
{
if (pdataSelected->pChild)
{
if (ITEMTYPE_OU == pdataSelected->pChild->nType)
{
DeleteChildren:
LOOKDATA * pChild = pdataSelected->pChild;
pdataSelected->pChild = pChild->pSibling;
while (pChild)
{
iIndex = (int)SendMessage(m_hCombo, CB_FINDSTRING, iIndex, (LPARAM)(LPCTSTR*)pChild);
if (iIndex != CB_ERR)
{
pChild = pChild->pChild;
SendMessage(m_hCombo, CB_DELETESTRING, iIndex, 0);
}
else
{
pChild = NULL;
}
}
return;
}
}
}
iIndex++;
}
}
void CBrowserPP::OnComboChange()
{
switch (m_dwPageType)
{
case PAGETYPE_DOMAINS:
{
TrimComboBox();
}
// fall through to refresh the list view
case PAGETYPE_SITES:
case PAGETYPE_ALL:
default:
OnRefresh();
break;
}
}
BOOL CBrowserPP::OnSetActive()
{
*m_ppActive = this;
OnRefresh();
return TRUE;
}
BOOL CBrowserPP::OnApply()
{
if (*m_ppActive == (void *) this)
{
// perform the proper task on the selected item
int i = ListView_GetNextItem(m_hList, -1, LVNI_SELECTED);
if (i >= 0)
{
LVITEM item;
memset(&item, 0, sizeof(item));
item.mask = LVIF_PARAM;
item.iItem = i;
ListView_GetItem(m_hList, &item);
MYLISTEL * pel = (MYLISTEL *)item.lParam;
switch (pel->nType)
{
case ITEMTYPE_GPO:
m_pGBI->gpoType = GPOTypeDS;
wcsncpy(m_pGBI->lpDSPath, pel->szData, m_pGBI->dwDSPathSize);
if (m_pGBI->lpName)
{
wcsncpy(m_pGBI->lpName, pel->szName, m_pGBI->dwNameSize);
}
m_pGBI->gpoHint = GPHintUnknown;
break;
default:
case ITEMTYPE_FOREST:
case ITEMTYPE_SITE:
case ITEMTYPE_DOMAIN:
// change the focus
{
LOOKDATA * pdataSelected = NULL;
// first make sure something is selected
int iIndex = (int)SendMessage (m_hCombo, CB_GETCURSEL, 0, 0);
if (iIndex != CB_ERR)
{
// something's selected, get a pointer to it's data
pdataSelected = (LOOKDATA *) SendMessage (m_hCombo, CB_GETITEMDATA, iIndex, 0);
if (pdataSelected)
{
// Now walk its children until we find a match
pdataSelected = pdataSelected->pChild;
while (pdataSelected)
{
if (0 == wcscmp(pdataSelected->szData, pel->szData))
{
iIndex = (int)SendMessage(m_hCombo, CB_FINDSTRING, iIndex, (LPARAM) (LPCTSTR)pdataSelected);
if (iIndex != CB_ERR)
{
SendMessage(m_hCombo, CB_SETCURSEL, iIndex, 0);
// Enable the back-button
SendMessage(m_toolbar, TB_ENABLEBUTTON, (WPARAM) ID_BACKBUTTON, (LPARAM) MAKELONG(TRUE, 0));
}
break;
}
pdataSelected = pdataSelected->pSibling;
}
}
}
}
OnRefresh();
return FALSE; // don't allow propsheet to close
case ITEMTYPE_OU:
// Add the new object to combobox and change the focus.
{
LOOKDATA * pdataSelected = NULL;
// first make sure something is selected
int iIndex = (int)SendMessage (m_hCombo, CB_GETCURSEL, 0, 0);
if (iIndex != CB_ERR)
{
// something's selected, get a pointer to it's data
pdataSelected = (LOOKDATA *) SendMessage (m_hCombo, CB_GETITEMDATA, iIndex, 0);
if (pdataSelected)
{
LOOKDATA * pNew = new LOOKDATA;
if (pNew)
{
pNew->szName = new WCHAR[wcslen(pel->szName)+1];
if (pNew->szName)
{
pNew->szData = new WCHAR[wcslen(pel->szData)+1];
if (pNew->szData)
{
wcscpy(pNew->szName, pel->szName);
wcscpy(pNew->szData, pel->szData);
pNew->nIndent = pdataSelected->nIndent + 1;
pNew->nType = ITEMTYPE_OU;
pNew->pParent = pdataSelected;
pNew->pSibling = pdataSelected->pChild;
pNew->pChild = NULL;
pdataSelected ->pChild = pNew;
SendMessage(m_hCombo, CB_INSERTSTRING, (WPARAM) iIndex + 1, (LPARAM) (LPCTSTR) pNew);
SendMessage(m_hCombo, CB_SETCURSEL, iIndex + 1, 0);
// Enable the back-button
SendMessage(m_toolbar, TB_ENABLEBUTTON, (WPARAM) ID_BACKBUTTON, (LPARAM) MAKELONG(TRUE, 0));
}
else
{
delete [] pNew->szName;
delete pNew;
}
}
else
{
delete pNew;
}
}
}
}
}
OnRefresh();
return FALSE; // don't allow propsheet to close
}
return TRUE;
}
else
return FALSE; // don't allow propsheet to close
}
return TRUE;
}
BOOL CBrowserPP::DialogProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
BOOL fReturn = FALSE;
m_hwndDlg = hwndDlg;
switch (uMsg)
{
case WM_INITDIALOG:
{
return OnInitDialog();
}
break;
case WM_NOTIFY:
{
LPNMHDR pnmh = (LPNMHDR) lParam;
LRESULT lResult = 0;
switch (pnmh->code)
{
case NM_KEYDOWN:
{
LPNMKEY pnkd = (LPNMKEY)pnmh;
if (VK_F5 == pnkd->nVKey)
{
OnRefresh();
}
}
break;
case PSN_SETACTIVE:
OnSetActive();
break;
case PSN_APPLY:
lResult = OnApply() ? PSNRET_NOERROR : PSNRET_INVALID_NOCHANGEPAGE;
fReturn = TRUE;
break;
case NM_DBLCLK:
if (IDC_LIST1 == wParam)
{
OnDoubleclickList(pnmh, &lResult);
fReturn = TRUE;
}
break;
case LVN_BEGINLABELEDIT:
OnBeginlabeleditList(pnmh, &lResult);
fReturn = TRUE;
break;
case LVN_ENDLABELEDIT:
OnEndlabeleditList(pnmh, &lResult);
fReturn = TRUE;
break;
case LVN_BEGINDRAG:
OnBegindragList(pnmh, &lResult);
fReturn = TRUE;
break;
case LVN_DELETEITEM:
OnDeleteitemList(pnmh, &lResult);
fReturn = TRUE;
break;
case LVN_COLUMNCLICK:
OnColumnclickList(pnmh, &lResult);
fReturn = TRUE;
break;
case LVN_KEYDOWN:
OnKeyDownList(pnmh, &lResult);
break;
case LVN_ITEMCHANGED:
OnItemChanged(pnmh, &lResult);
break;
case TBN_DROPDOWN:
{
RECT r;
SendMessage(m_toolbar, TB_GETRECT, ((TBNOTIFY *)lParam)->iItem, (LPARAM)&r);
MapWindowPoints(m_toolbar, NULL, (POINT *)&r, 2);
HMENU hPopup;
hPopup = LoadMenu(g_hInstance, MAKEINTRESOURCE(IDR_LISTMENU));
if ( ! hPopup )
{
break;
}
UINT ui = ID_LIST;
DWORD dw = GetWindowLong(m_hList, GWL_STYLE) & LVS_TYPEMASK;
switch (dw)
{
case LVS_ICON:
ui = ID_LARGEICONS;
break;
case LVS_SMALLICON:
ui = ID_SMALLICONS;
break;
case LVS_REPORT:
ui = ID_DETAILS;
break;
case LVS_LIST:
default:
ui = ID_LIST;
break;
}
HMENU hSubMenu = GetSubMenu(GetSubMenu(hPopup, 0), 0);
CheckMenuRadioItem(hSubMenu, ui, ui, ui, MF_BYCOMMAND);
TrackPopupMenu(hSubMenu,
TPM_LEFTALIGN,
r.left, r.bottom,
0,
m_hwndDlg,
&r);
fReturn = TRUE;
DestroyMenu(hPopup);
break;
}
break;
case TTN_GETDISPINFO:
{
LPNMTTDISPINFO pDI = (LPNMTTDISPINFO) lParam;
UINT id = 0;
if (pDI->hdr.idFrom == ID_BACKBUTTON)
id = IDS_TOOLTIP_BACK;
else if (pDI->hdr.idFrom == ID_NEWFOLDER)
id = IDS_TOOLTIP_NEW;
else if (pDI->hdr.idFrom == ID_ROTATEVIEW)
id = IDS_TOOLTIP_ROTATE;
if (id)
LoadString (g_hInstance, id, pDI->szText, 80);
else
pDI->szText[0] = TEXT('\0');
fReturn = TRUE;
}
break;
default:
break;
}
SetWindowLongPtr(m_hwndDlg, DWLP_MSGRESULT, lResult);
}
break;
case WM_COMMAND:
switch (LOWORD(wParam))
{
case IDC_COMBO1:
if (CBN_SELCHANGE == HIWORD(wParam))
{
OnComboChange();
}
break;
case ID_BACKBUTTON:
return DoBackButton();
case ID_NEWFOLDER:
return DoNewGPO();
case ID_ROTATEVIEW:
return DoRotateView();
case ID_DETAILS:
OnDetails();
fReturn = TRUE;
break;
case ID_LIST:
OnList();
fReturn = TRUE;
break;
case ID_LARGEICONS:
OnLargeicons();
fReturn = TRUE;
break;
case ID_SMALLICONS:
OnSmallicons();
fReturn = TRUE;
break;
case ID_ARRANGE_AUTO:
OnArrangeAuto();
fReturn = TRUE;
break;
case ID_ARRANGE_BYNAME:
OnArrangeByname();
fReturn = TRUE;
break;
case ID_ARRANGE_BYTYPE:
OnArrangeBytype();
fReturn = TRUE;
break;
case ID_DELETE:
OnDelete();
fReturn = TRUE;
break;
case ID_EDIT:
OnEdit();
fReturn = TRUE;
break;
case ID_NEW:
OnNew();
fReturn = TRUE;
break;
case ID_PROPERTIES:
OnProperties();
fReturn = TRUE;
break;
case ID_REFRESH:
OnRefresh();
fReturn = TRUE;
break;
case ID_RENAME:
OnRename();
fReturn = TRUE;
break;
case ID_TOP_LINEUPICONS:
OnTopLineupicons();
fReturn = TRUE;
break;
default:
break;
}
break;
case WM_CONTEXTMENU:
fReturn = TRUE;
if ((HWND)wParam != m_toolbar)
{
if (GetDlgItem(hwndDlg, IDC_LIST1) == (HWND)wParam)
{
OnContextMenu(lParam);
}
else
{
// right mouse click
switch (m_dwPageType)
{
case PAGETYPE_DOMAINS:
WinHelp((HWND) wParam, HELP_FILE, HELP_CONTEXTMENU,
(ULONG_PTR) (LPSTR) aBrowserDomainHelpIds);
break;
case PAGETYPE_SITES:
WinHelp((HWND) wParam, HELP_FILE, HELP_CONTEXTMENU,
(ULONG_PTR) (LPSTR) aBrowserSiteHelpIds);
break;
case PAGETYPE_ALL:
WinHelp((HWND) wParam, HELP_FILE, HELP_CONTEXTMENU,
(ULONG_PTR) (LPSTR) aBrowserAllHelpIds);
break;
}
}
}
break;
case WM_HELP:
// F1 help
if (((LPHELPINFO) lParam)->iCtrlId != IDR_TOOLBAR1)
{
switch (m_dwPageType)
{
case PAGETYPE_DOMAINS:
WinHelp((HWND)((LPHELPINFO) lParam)->hItemHandle, HELP_FILE, HELP_WM_HELP,
(ULONG_PTR) (LPSTR) aBrowserDomainHelpIds);
break;
case PAGETYPE_SITES:
WinHelp((HWND)((LPHELPINFO) lParam)->hItemHandle, HELP_FILE, HELP_WM_HELP,
(ULONG_PTR) (LPSTR) aBrowserSiteHelpIds);
break;
case PAGETYPE_ALL:
WinHelp((HWND)((LPHELPINFO) lParam)->hItemHandle, HELP_FILE, HELP_WM_HELP,
(ULONG_PTR) (LPSTR) aBrowserAllHelpIds);
break;
}
}
fReturn = TRUE;
break;
case WM_DRAWITEM:
if (IDC_COMBO1 == wParam)
{
DrawItem((LPDRAWITEMSTRUCT)lParam);
fReturn = TRUE;
}
break;
case WM_MEASUREITEM:
if (IDC_COMBO1 == wParam)
{
MeasureItem((LPMEASUREITEMSTRUCT)lParam);
fReturn = TRUE;
}
break;
case WM_COMPAREITEM:
if (IDC_COMBO1 == wParam)
{
int iReturn = CompareItem((LPCOMPAREITEMSTRUCT)lParam);
SetWindowLongPtr(m_hwndDlg, DWLP_MSGRESULT, iReturn);
fReturn = TRUE;
}
break;
case WM_DELETEITEM:
if (IDC_COMBO1 == wParam)
{
DeleteItem((LPDELETEITEMSTRUCT)lParam);
fReturn = TRUE;
}
break;
case WM_REFRESHDISPLAY:
{
MYLISTEL * pel;
LVITEM item;
ZeroMemory (&item, sizeof(item));
item.mask = LVIF_PARAM;
item.iItem = (INT) wParam;
if (ListView_GetItem(m_hList, &item))
{
pel = (MYLISTEL *)item.lParam;
ListView_SetItemText(m_hList, (INT)wParam, 0, pel->szName);
}
}
break;
default:
break;
}
return fReturn;
}
void CBrowserPP::DrawItem(LPDRAWITEMSTRUCT lpDrawItemStruct)
{
// DRAWITEMSTRUCT:
// UINT CtlType // type of the control
// UINT CtlID; // ID of the control
// UINT itemID; // index of the item
// UINT itemAction;
// UINT itemState;
// HWND hwndItem;
// HDC hDC;
// RECT rcItem;
// DWORD itemData; // user-defined data
if (-1 != lpDrawItemStruct->itemID)
{
LOOKDATA * pdata = (LOOKDATA *)lpDrawItemStruct->itemData;
POINT pt;
INT iIndex;
if (pdata->nType == ITEMTYPE_FOREST)
{
iIndex = 10;
}
else if (pdata->nType == ITEMTYPE_SITE)
{
iIndex = 6;
}
else if (pdata->nType == ITEMTYPE_DOMAIN)
{
iIndex = 7;
}
else
{
iIndex = 0;
}
pt.x = lpDrawItemStruct->rcItem.left;
BOOL fSelected = ODS_SELECTED == (ODS_SELECTED & lpDrawItemStruct->itemState);
BOOL fComboBoxEdit = ODS_COMBOBOXEDIT != (ODS_COMBOBOXEDIT & lpDrawItemStruct->itemState);
if (fComboBoxEdit)
pt.x += (INDENT * pdata->nIndent);
pt.y = lpDrawItemStruct->rcItem.top;
ImageList_Draw(m_ilSmall, iIndex, lpDrawItemStruct->hDC, pt.x, pt.y, fSelected ? ILD_SELECTED : ILD_NORMAL);
SIZE size;
GetTextExtentPoint32(lpDrawItemStruct->hDC, pdata->szName, wcslen(pdata->szName), &size);
COLORREF crBk;
COLORREF crText;
if (fSelected)
{
crBk = GetBkColor(lpDrawItemStruct->hDC);
crText = GetTextColor(lpDrawItemStruct->hDC);
SetBkColor(lpDrawItemStruct->hDC, GetSysColor(COLOR_HIGHLIGHT));
SetTextColor(lpDrawItemStruct->hDC, GetSysColor(COLOR_HIGHLIGHTTEXT));
}
// NOTE, SMALLICONSIZE + 1 is used here to ensure it rounds UP
// instead of down when centering the text. (It looks better this
// way.)
// Adding 18 to the x coord spaces us past the icon.
ExtTextOut(lpDrawItemStruct->hDC, pt.x + (SMALLICONSIZE + 2), pt.y + (((SMALLICONSIZE + 1) - size.cy) / 2), ETO_CLIPPED, &lpDrawItemStruct->rcItem, pdata->szName, wcslen(pdata->szName), NULL);
if (fSelected)
{
SetBkColor(lpDrawItemStruct->hDC, crBk);
SetTextColor(lpDrawItemStruct->hDC, crText);
}
}
}
void CBrowserPP::MeasureItem(LPMEASUREITEMSTRUCT lpMeasureItemStruct)
{
// MEASUREITEMSTRUCT:
// UINT CtlType // type of the control
// UINT CtlID; // ID of the control
// UINT itemID; // index of the item
// UINT itemWidth; // width of item in pixels
// UINT itemHeight; // height of item in pixels
// DWORD itemData; // user-defined data
lpMeasureItemStruct->itemHeight = SMALLICONSIZE;
}
int CBrowserPP::CompareItem(LPCOMPAREITEMSTRUCT lpCompareItemStruct)
{
// COMPAREITEMSTRUCT:
// UINT CtlType // type of the control
// UINT CtlID; // ID of the control
// HWND hwndItem; // handle of the control
// UINT itemID; // index of the item
// DWORD itemData1; // user-defined data
// UINT itemID2; // index of the second item
// DWORD itemData2; // user-defined data
// I'm not doing any sorting.
return 0;
}
void CBrowserPP::DeleteItem(LPDELETEITEMSTRUCT lpDeleteItemStruct)
{
LOOKDATA * pdata = (LOOKDATA *)lpDeleteItemStruct->itemData;
if (NULL != pdata)
{
if (NULL != pdata->szName)
{
delete [] pdata->szName;
}
if (NULL != pdata->szData)
{
delete [] pdata->szData;
}
delete pdata;
}
}
LPTSTR CBrowserPP::GetFullPath (LPTSTR lpPath, HWND hParent)
{
LPTSTR lpFullPath = NULL, lpDomainName = NULL;
LPTSTR lpGPDCName;
LPOLESTR pszDomain;
HRESULT hr;
//
// Get the friendly domain name
//
pszDomain = GetDomainFromLDAPPath(lpPath);
if (!pszDomain)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::GetFullPath: Failed to get domain name")));
return NULL;
}
//
// Convert LDAP to dot (DN) style
//
hr = ConvertToDotStyle (pszDomain, &lpDomainName);
delete [] pszDomain;
if (FAILED(hr))
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::GetFullPath: Failed to convert domain name with 0x%x"), hr));
return NULL;
}
if (!lstrcmpi(lpDomainName, m_szDomainName))
{
//
// Make the full path
//
lpFullPath = MakeFullPath (lpPath, m_szServerName);
if (!lpFullPath)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::GetFullPath: Failed to build new DS object path")));
goto Exit;
}
}
else
{
//
// Get the GPO DC for this domain
//
lpGPDCName = GetDCName (lpDomainName, NULL, hParent, TRUE, 0);
if (!lpGPDCName)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::GetFullPath: Failed to get DC name for %s"),
lpDomainName));
goto Exit;
}
//
// Make the full path
//
lpFullPath = MakeFullPath (lpPath, lpGPDCName);
LocalFree (lpGPDCName);
if (!lpFullPath)
{
DebugMsg((DM_WARNING, TEXT("CBrowserPP::GetFullPath: Failed to build new DS object path")));
goto Exit;
}
}
Exit:
if (lpDomainName)
{
LocalFree (lpDomainName);
}
return lpFullPath;
}