able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/admin/wmi/wbem/adapters/odbc/conndlg.cpp
CryptoAlgo Inc b3cac27891 Add source files
2020-09-26 16:20:57 +08:00

1876 lines
48 KiB
C++
Raw Blame History

// conndlg.cpp : implementation file
//
// Commenting #define out - causing compiler error - not sure if needed, compiles
// okay without it.
//#define WINVER 0x0400
#include "precomp.h"
#include "afxtempl.h"
#include <comdef.h> //for _bstr_t
#include "resource.h"
#include "wbemidl.h"
#include <comdef.h>
//smart pointer
_COM_SMARTPTR_TYPEDEF(IWbemServices, IID_IWbemServices);
_COM_SMARTPTR_TYPEDEF(IEnumWbemClassObject, IID_IEnumWbemClassObject);
//_COM_SMARTPTR_TYPEDEF(IWbemContext, IID_IWbemContext );
_COM_SMARTPTR_TYPEDEF(IWbemLocator, IID_IWbemLocator);
#include "drdbdr.h"
#include "Browse.h"
#include "dlgcback.h"
#include "cominit.h" //for Dcom blanket
#include "htmlhelp.h" //for HELP
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
/////////////////////////////////////////////////////////////////////////////
// CConnectionDialog dialog
HTREEITEM CConnectionDialog :: InsertItem (CTreeCtrl &treeCtrl,
HTREEITEM hParentTreeItem,
const char *namespaceName)
{
TV_INSERTSTRUCT newItem;
newItem.item.mask = TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_PARAM;
newItem.hParent = hParentTreeItem;
char *name = new char [strlen (namespaceName) + 1];
name[0] = 0;
lstrcpy (name, namespaceName);
newItem.item.pszText = name;
// newItem.item.cchTextMax = strlen (namespaceName) + 1;
//get parent name and prepend to name to generate absolute name
char *absName = NULL;
if (hParentTreeItem == TVI_ROOT)
{
absName = new char [strlen (namespaceName) + 1];
absName[0] = 0;
lstrcpy (absName, namespaceName);
}
else
{
ISAMTreeItemData* parentData =
(ISAMTreeItemData*)treeCtrl.GetItemData (hParentTreeItem);
absName = new char [strlen (namespaceName) + strlen (parentData->absName) + 2 + 1];
absName[0] = 0;
strcpy (absName, parentData->absName);
strcat (absName, "\\");
strcat (absName, namespaceName);
}
newItem.item.iImage = newItem.item.iSelectedImage = m_idxMode1Image;
ISAMTreeItemData *itemData = new ISAMTreeItemData;
itemData->absName = absName;
itemData->pszText = name; //this copy will be deleted
itemData->included = FALSE;
itemData->childInclude = 0;
itemData->childChildInclude = 0;
//Add to list for quick cleanup
if (pCurrentSelectionList)
{
pCurrentItem->pNext = itemData;
pCurrentItem = itemData;
}
else
{
pCurrentSelectionList = itemData;
pCurrentItem = itemData;
}
newItem.item.lParam = (DWORD)itemData;
return treeCtrl.InsertItem (&newItem);
}
int CConnectionDialog :: FindAbsName (char *name,
HTREEITEM hStartAt,
HTREEITEM& hFoundItem)
{
if (!name)
return FALSE;
{
ISAMTreeItemData *pItemData = (ISAMTreeItemData*)m_tree1.GetItemData (hStartAt);
if (pItemData && pItemData->absName &&
!_stricmp (name, pItemData->absName))
{
hFoundItem = hStartAt;
return TRUE;
}
else
{
HTREEITEM firstChild = NULL;
if (firstChild = m_tree1.GetChildItem (hStartAt))
{
if (FindAbsName (name, firstChild, hFoundItem))
return TRUE;
HTREEITEM nextSibling = firstChild;
while (nextSibling = m_tree1.GetNextItem (nextSibling, TVGN_NEXT))
{
if (FindAbsName (name, nextSibling, hFoundItem))
return TRUE;
}
}
return FALSE;
}
}
}
int CConnectionDialog :: CreateNamespace (char *name,
HTREEITEM hStartAt,
HTREEITEM& hFoundItem)
{
hFoundItem = NULL;
//If we are to create the namespace in the tree control
//we need to do it relative to its parent namespace
//The first step therefore is to get a handle to its parent
//We work out the name of the parent namespace
//Start at the end of the string and work your way
//back until you reach a backslash character or
//you run out of string
long len = lstrlen(name);
//Note: a namespace cannot end with a backslash
//so we check for this error condition
long index = len - 1;
if (name[index] == '\\')
return FALSE;
while (index && (name[index] != '\\'))
index--;
//We have either run out of string or found a backslash
if (!index)
{
//We have run out of string
return FALSE;
}
else
{
name[index] = 0;
char* parent = name;
char* child = parent + index + 1;
//Get Parent
HTREEITEM hParent;
if (! FindAbsName(parent, hStartAt, hParent))
{
// Could not get parent, so create it
CreateNamespace (parent, hStartAt, hParent);
}
if (hParent)
{
//Found parent, so now create child
hFoundItem = InsertItem(m_tree1,hParent, (const char*)child);
}
}
return TRUE;
}
int CConnectionDialog :: UnincludedChild (HTREEITEM hStartAt, int checkSelf)
{
if (checkSelf && !(((ISAMTreeItemData*)m_tree1.GetItemData (hStartAt))->included))
{
return TRUE;
}
else
{
HTREEITEM firstChild = NULL;
if (firstChild = m_tree1.GetChildItem (hStartAt))
{
if (UnincludedChild (firstChild, TRUE))
return TRUE;
HTREEITEM nextSibling = firstChild;
while (nextSibling = m_tree1.GetNextItem (nextSibling, TVGN_NEXT))
{
if (UnincludedChild (nextSibling, TRUE))
return TRUE;;
}
}
return FALSE;
}
}
void CConnectionDialog :: GenerateOutMap (HTREEITEM hStartAt)
{
ISAMTreeItemData *itemData = (ISAMTreeItemData*)m_tree1.GetItemData (hStartAt);
if (itemData->included)
{
// if there's an unincluded child add this individually and recurse
// else add deep and no need to recurse
if (UnincludedChild (hStartAt, FALSE))
{
// get name and add it shallow and recurse
pMapStringToObOut->SetAt (itemData->absName,
new CNamespace (itemData->absName));
HTREEITEM firstChild = NULL;
if (firstChild = m_tree1.GetChildItem (hStartAt))
{
GenerateOutMap (firstChild);
HTREEITEM nextSibling = firstChild;
while (nextSibling = m_tree1.GetNextItem (nextSibling, TVGN_NEXT))
{
GenerateOutMap (nextSibling);
}
}
}
else
{
// get name and add with deep flag (no recurse)
// for now do same as above ie don't use deep flag
pMapStringToObOut->SetAt (itemData->absName,
new CNamespace (itemData->absName));
HTREEITEM firstChild = NULL;
if (firstChild = m_tree1.GetChildItem (hStartAt))
{
GenerateOutMap (firstChild);
HTREEITEM nextSibling = firstChild;
while (nextSibling = m_tree1.GetNextItem (nextSibling, TVGN_NEXT))
{
GenerateOutMap (nextSibling);
}
}
}
}
else
{
// just recurse
HTREEITEM firstChild = NULL;
if (firstChild = m_tree1.GetChildItem (hStartAt))
{
GenerateOutMap (firstChild);
HTREEITEM nextSibling = firstChild;
while (nextSibling = m_tree1.GetNextItem (nextSibling, TVGN_NEXT))
{
GenerateOutMap (nextSibling);
}
}
}
}
void CConnectionDialog :: GenerateOutString (HTREEITEM hStartAt)
{
ISAMTreeItemData *itemData = (ISAMTreeItemData*)m_tree1.GetItemData (hStartAt);
if (itemData->included)
{
// if there's an unincluded child add this individually and recurse
// else add deep and no need to recurse
if (UnincludedChild (hStartAt, FALSE))
{
// get name and add it shallow and recurse
char *temp = *lpszNamespacesOut;
if (temp)
{
*lpszNamespacesOut = new char [strlen (temp) + 1 + 4 /* for punctuation */
+ strlen (itemData->absName) +
strlen ("shallow")];
(*lpszNamespacesOut)[0] = 0;
strcpy (*lpszNamespacesOut, temp);
strcat (*lpszNamespacesOut, ",");
strcat (*lpszNamespacesOut, "{");
strcat (*lpszNamespacesOut, itemData->absName);
strcat (*lpszNamespacesOut, ",shallow}");
delete temp;
}
else
{
*lpszNamespacesOut = new char [strlen (itemData->absName) + 1 + 4 +
strlen ("shallow")];
(*lpszNamespacesOut)[0] = 0;
strcpy (*lpszNamespacesOut, "{");
strcat (*lpszNamespacesOut, itemData->absName);
strcat (*lpszNamespacesOut, ",shallow}");
}
HTREEITEM firstChild = NULL;
if (firstChild = m_tree1.GetChildItem (hStartAt))
{
GenerateOutString (firstChild);
HTREEITEM nextSibling = firstChild;
while (nextSibling = m_tree1.GetNextItem (nextSibling, TVGN_NEXT))
{
GenerateOutString (nextSibling);
}
}
}
else
{
// get name and add with deep flag (no recurse)
// for now do same as above ie don't use deep flag
char *temp = *lpszNamespacesOut;
if (temp)
{
*lpszNamespacesOut = new char [strlen (temp) + 1 + 4 /* for punctuation */
+ strlen (itemData->absName) +
strlen ("deep")];
(*lpszNamespacesOut)[0] = 0;
strcpy (*lpszNamespacesOut, temp);
strcat (*lpszNamespacesOut, ",");
strcat (*lpszNamespacesOut, "{");
strcat (*lpszNamespacesOut, itemData->absName);
strcat (*lpszNamespacesOut, ",deep}");
delete temp;
}
else
{
*lpszNamespacesOut = new char [strlen (itemData->absName) + 1 + 4
+ strlen("deep")];
(*lpszNamespacesOut)[0] = 0;
strcpy (*lpszNamespacesOut, "{");
strcat (*lpszNamespacesOut, itemData->absName);
strcat (*lpszNamespacesOut, ",deep}");
}
}
}
else
{
// just recurse
HTREEITEM firstChild = NULL;
if (firstChild = m_tree1.GetChildItem (hStartAt))
{
GenerateOutString (firstChild);
HTREEITEM nextSibling = firstChild;
while (nextSibling = m_tree1.GetNextItem (nextSibling, TVGN_NEXT))
{
GenerateOutString (nextSibling);
}
}
}
}
void CConnectionDialog :: UpdateChildChildInclude (HTREEITEM hNode, BOOL fIncrement)
{
//Get the parent node, so that you can flag that a non-immediate child
//has been selected
HTREEITEM hParentNode = m_tree1.GetParentItem(hNode);
if (hParentNode)
{
TV_ITEM tvItem;
tvItem.hItem = hParentNode;
tvItem.mask = TVIF_PARAM | TVIF_IMAGE | TVIF_SELECTEDIMAGE;
if (m_tree1.GetItem (&tvItem))
{
ISAMTreeItemData* temp = (ISAMTreeItemData*)tvItem.lParam;
if (fIncrement)
{
++(temp->childChildInclude);
//update state of parent if no immediate children
if (0 == temp->childInclude)
{
//I can find out the state via the iImage field
if ( tvItem.iImage == m_idxMode1Image)
{
//In state 1, move to state 2
tvItem.iSelectedImage = tvItem.iImage = m_idxMode2Image;
m_tree1.SetItem (&tvItem);
}
else if ( tvItem.iImage == m_idxMode3Image)
{
//In state 3, move to state 4
tvItem.iSelectedImage = tvItem.iImage = m_idxMode4Image;
m_tree1.SetItem (&tvItem);
}
}
}
else
{
--(temp->childChildInclude);
//update state of parent if no immediate children
if ((0 == temp->childInclude) && (0 == temp->childChildInclude))
{
//I can find out the state via the iImage field
if ( tvItem.iImage == m_idxMode2Image)
{
//In state 2, move to state 1
tvItem.iSelectedImage = tvItem.iImage = m_idxMode1Image;
m_tree1.SetItem (&tvItem);
}
else if ( tvItem.iImage == m_idxMode4Image)
{
//In state 4, move to state 3
tvItem.iSelectedImage = tvItem.iImage = m_idxMode3Image;
m_tree1.SetItem (&tvItem);
}
}
}
}
//Recursive call up the chain of parents
UpdateChildChildInclude(hParentNode, fIncrement);
}
}
void CConnectionDialog :: AddNamespaces (HTREEITEM hParent,
int deep)
{
TV_ITEM tvItem;
tvItem.hItem = hParent;
tvItem.mask = TVIF_PARAM | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_CHILDREN;
if (m_tree1.GetItem (&tvItem))
{
//If cel not already selected, then select
if (! ((ISAMTreeItemData*)tvItem.lParam)->included )
{
cSelectedCels++;
//Update count in parent node
HTREEITEM hParentNode = m_tree1.GetParentItem(hParent);
if (hParentNode)
{
TV_ITEM tvItem2;
tvItem2.hItem = hParentNode;
tvItem2.mask = TVIF_PARAM | TVIF_IMAGE | TVIF_SELECTEDIMAGE;
if (m_tree1.GetItem (&tvItem2))
{
++(((ISAMTreeItemData*)tvItem2.lParam)->childInclude);
//Parent node must be in either state 1, 2 or 3
//I can find out the state via the iImage field
if ( tvItem2.iImage == m_idxMode1Image)
{
//In state 1, move to state 2
//tvItem2.iSelectedImage = tvItem2.iImage = m_idxMode2Image;
m_tree1.SetItemImage (hParentNode, m_idxMode2Image, m_idxMode2Image);
}
else if ( tvItem2.iImage == m_idxMode3Image)
{
//In state 3, move to state 4
//tvItem2.iSelectedImage = tvItem2.iImage = m_idxMode4Image;
m_tree1.SetItemImage (hParentNode, m_idxMode4Image, m_idxMode4Image);
}
//Now indicate to parents parent that a non-immediate child
//has been selected
UpdateChildChildInclude(hParentNode, TRUE);
}
}
}
if (deep) //recurse
{
HTREEITEM firstChild = NULL;
if (firstChild = m_tree1.GetChildItem (hParent))
{
AddNamespaces (firstChild, deep);
HTREEITEM nextSibling = firstChild;
while (nextSibling = m_tree1.GetNextItem (nextSibling, TVGN_NEXT))
AddNamespaces (nextSibling, deep);
}
}
m_tree1.GetItem (&tvItem);
((ISAMTreeItemData*)tvItem.lParam)->included = TRUE;
//Check if this node has 'included' children
int children = tvItem.cChildren;
ISAMTreeItemData* temp = (ISAMTreeItemData*)tvItem.lParam;
if (children && (temp->childInclude || temp->childChildInclude))
{
// tvItem.iSelectedImage = tvItem.iImage = m_idxMode4Image;
m_tree1.SetItemImage (hParent, m_idxMode4Image, m_idxMode4Image);
}
else
{
// tvItem.iSelectedImage = tvItem.iImage = m_idxMode3Image;
m_tree1.SetItemImage (hParent, m_idxMode3Image, m_idxMode3Image);
}
// m_tree1.SetItem (&tvItem);
}
//Update state of OK and Cancel pushbutton depending on
//number of cels selected
m_okButton.EnableWindow(cSelectedCels ? TRUE : FALSE);
}
void CConnectionDialog :: RemoveNamespaces (HTREEITEM hParent,
int deep)
{
TV_ITEM tvItem;
tvItem.hItem = hParent;
tvItem.mask = TVIF_PARAM | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_CHILDREN;
if (m_tree1.GetItem (&tvItem))
{
//If cel not unselected, then unselect
if ( ((ISAMTreeItemData*)tvItem.lParam)->included )
{
cSelectedCels--;
//update state of parent
HTREEITEM hParentNode = m_tree1.GetParentItem(hParent);
if (hParentNode)
{
TV_ITEM tvItem2;
tvItem2.hItem = hParentNode;
tvItem2.mask = TVIF_PARAM | TVIF_IMAGE | TVIF_SELECTEDIMAGE;
if (m_tree1.GetItem (&tvItem2))
{
--(((ISAMTreeItemData*)tvItem2.lParam)->childInclude);
//Parent node must be in either state 2 or 4
//I can find out the state via the iImage field
if ( tvItem2.iImage == m_idxMode4Image)
{
//In state 4
if (!((ISAMTreeItemData*)tvItem2.lParam)->childInclude)
{
//Check non-immediate children
if (!((ISAMTreeItemData*)tvItem2.lParam)->childChildInclude)
{
//If no more children move to state 3
// tvItem2.iSelectedImage = tvItem2.iImage = m_idxMode3Image;
//// m_tree1.SetItem (&tvItem2);
m_tree1.SetItemImage (hParentNode, m_idxMode3Image, m_idxMode3Image);
}
}
}
else
{
//In state 2
if (!((ISAMTreeItemData*)tvItem2.lParam)->childInclude)
{
//Check non-immediate children
if (!((ISAMTreeItemData*)tvItem2.lParam)->childChildInclude)
{
//If no more children move to state 1
// tvItem2.iSelectedImage = tvItem2.iImage = m_idxMode1Image;
//// m_tree1.SetItem (&tvItem2);
m_tree1.SetItemImage (hParentNode, m_idxMode1Image, m_idxMode1Image);
}
}
}
//Now indicate to parents parent that a non-immediate child
//has been de-selected
UpdateChildChildInclude(hParentNode, FALSE);
}
}
}
if (deep) //recurse
{
HTREEITEM firstChild = NULL;
if (firstChild = m_tree1.GetChildItem (hParent))
{
RemoveNamespaces (firstChild, deep);
HTREEITEM nextSibling = firstChild;
while (nextSibling = m_tree1.GetNextItem (nextSibling, TVGN_NEXT))
RemoveNamespaces (nextSibling, deep);
}
}
m_tree1.GetItem (&tvItem);
((ISAMTreeItemData*)tvItem.lParam)->included = FALSE;
//Check if this node has 'included' children
int children = tvItem.cChildren;
ISAMTreeItemData* temp = (ISAMTreeItemData*)tvItem.lParam;
if (children && (temp->childInclude || temp->childChildInclude))
{
// tvItem.iSelectedImage = tvItem.iImage = m_idxMode2Image;
m_tree1.SetItemImage (hParent, m_idxMode2Image, m_idxMode2Image);
}
else
{
// tvItem.iSelectedImage = tvItem.iImage = m_idxMode1Image;
m_tree1.SetItemImage (hParent, m_idxMode1Image, m_idxMode1Image);
}
//// m_tree1.SetItem (&tvItem);
}
//Update state of OK and Cancel pushbutton depending on
//number of cels selected
m_okButton.EnableWindow(cSelectedCels ? TRUE : FALSE);
}
CConnectionDialog::CConnectionDialog(CWnd* pParent, char *pszServer,
// WBEM_LOGIN_AUTHENTICATION loginMethod,
char *pszUsername, char *pszPassword,
char **pszLocale, char** pszAuthority,
BOOL FAR* fSysProp, BOOL fConnParmSpec,
CMapStringToOb *pNamespaceMap,
CMapStringToOb *pNamespaceMapOut,
BOOL enableDefaultDatabase, char **connectionString, BOOL FAR* fImpersonate, BOOL FAR* fPassthrghOnly, BOOL FAR* fItprtEmptPwdAsBlk)
: CDialog(CConnectionDialog::IDD, pParent)
{
ODBCTRACE(_T("\nWBEM ODBC Driver : CConnectionDialog::CConnectionDialog\n"));
//{{AFX_DATA_INIT(CConnectionDialog)
//}}AFX_DATA_INIT
pMapStringToObOut = pNamespaceMapOut;
pMapStringToObIn = pNamespaceMap;
// pServerIn = NULL;
// pUsernameIn = NULL;
// pPasswordIn = NULL;
lpszNamespacesOut = connectionString;
fDoubleClicked = FALSE;
fConnParmSpecified = fConnParmSpec;
//Make copy of Server, UserName and Password memory addresses
lpszServer = pszServer;
lpszUserName = pszUsername;
lpszPassword = pszPassword;
lpszAuthority = NULL;
lpszLocale = NULL;
lpszAuthorityOut = pszAuthority;
lpszLocaleOut = pszLocale;
if (pszLocale && *pszLocale)
{
int localeLen = lstrlen(*pszLocale);
lpszLocale = new char [localeLen + 1];
lpszLocale[0] = 0;
lstrcpy(lpszLocale, *pszLocale);
}
if (pszAuthority && *pszAuthority)
{
int authLen = lstrlen(*pszAuthority);
lpszAuthority = new char [authLen + 1];
lpszAuthority[0] = 0;
lstrcpy(lpszAuthority, *pszAuthority);
ODBCTRACE(_T("\nWBEM ODBC Driver : Authority :"));
ODBCTRACE(_T(lpszAuthority));
ODBCTRACE(_T("\n"));
}
// m_loginMethod = loginMethod;
//A counter to indicated number of selected namespace cels
cSelectedCels = 0;
*connectionString = NULL;
//Setup System Properties
fSystemProperties = fSysProp;
// if (pszUsername)
// {
// pUsernameIn = new char [strlen (pszUsername) + 1];
// pUsernameIn[0] = 0;
// strcpy (pUsernameIn, pszUsername);
// }
// if (pszPassword)
// {
// pPasswordIn = new char [strlen (pszPassword) + 1];
// pPasswordIn[0] = 0;
// strcpy (pPasswordIn, pszPassword);
// }
fImpersonation = fImpersonate;
fPassthroughOnly = fPassthrghOnly;
fIntpretEmptPwdAsBlank = fItprtEmptPwdAsBlk;
impersonateMgr = NULL;
if (*fImpersonation)
{
//Now check if impersonation is necessary
//only if connecting locally
if (IsLocalServer(lpszServer))
{
impersonateMgr = new ImpersonationManager(lpszUserName, lpszPassword, lpszAuthority);
}
else
{
ODBCTRACE("\nWBEM ODBC Driver : Server not detected as local, not impersonating\n");
}
}
}
CConnectionDialog :: ~CConnectionDialog()
{
ODBCTRACE ("\nWBEM ODBC Driver : CConnectionDialog :: ~CConnectionDialog\n");
//Tidy Up
// if (pUsernameIn)
// delete pUsernameIn;
// if (pPasswordIn)
// delete pPasswordIn;
delete impersonateMgr;
ODBCTRACE ("\nWBEM ODBC Driver : CConnectionDialog :: ~CConnectionDialog exit point\n");
}
void CConnectionDialog::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CConnectionDialog)
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CConnectionDialog, CDialog)
//{{AFX_MSG_MAP(CConnectionDialog)
ON_NOTIFY(NM_CLICK, IDC_TREE1, OnClickTree1)
ON_NOTIFY(TVN_KEYDOWN, IDC_TREE1, OnKeyDown)
ON_NOTIFY(TVN_ITEMEXPANDING, IDC_TREE1, OnTreeExpand)
ON_NOTIFY(TVN_DELETEITEM, IDC_TREE1, OnDeleteitemTree1)
ON_BN_CLICKED(IDC_BUTTON_BROWSE, OnButtonBrowse)
ON_BN_CLICKED(IDC_CHECK1, OnButtonSysProp)
ON_BN_CLICKED(IDC_CHECK_IMPERSONATE, OnButtonImpersonation)
ON_BN_CLICKED(IDC_CHECK_PASSTHROUGHONLY, OnButtonPassthroughOnly)
ON_NOTIFY(NM_DBLCLK, IDC_TREE1, OnDblclkTree1)
ON_EN_KILLFOCUS(IDC_BROWSE_EDIT, OnKillfocusBrowseEdit)
ON_EN_CHANGE(IDC_EDIT_USER_NAME, OnUserNameChange)
ON_EN_CHANGE(IDC_EDIT_PSWD, OnPasswordChange)
ON_EN_CHANGE(IDC_EDIT_AUTH, OnAuthorityChange)
ON_EN_CHANGE(IDC_EDIT_LOCALE, OnLocaleChange)
ON_EN_CHANGE(IDC_BROWSE_EDIT, OnServerChange)
ON_BN_CLICKED(IDC_REFRESH_BUTTON, OnButtonRefresh)
ON_BN_CLICKED(IDC_HELP_BUTTON, OnHelp)
ON_BN_CLICKED(IDC_RADIO_BLANK, OnButtonInterpretEmpty)
ON_BN_CLICKED(IDC_RADIO_NULL, OnButtonInterpretEmpty)
ON_WM_NCDESTROY()
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CConnectionDialog message handlers
void CConnectionDialog::OnKeyDown(NMHDR* pNMHDR, LRESULT* pResult)
{
TV_KEYDOWN* lParam = (TV_KEYDOWN*)pNMHDR;
// TODO: Add your control notification handler code here
*pResult = 0;
//Check if SPACE BAR is pressed
if (lParam->wVKey == VK_SPACE)
{
HTREEITEM item = m_tree1.GetSelectedItem();
//Grab the WM_CHAR message from the message queue
MSG myMessage;
BOOL status = GetMessage(&myMessage, lParam->hdr.hwndFrom, WM_CHAR, WM_CHAR);
if (item)
{
//Get the item
TV_ITEM tvItem;
tvItem.hItem = item;
tvItem.mask = TVIF_STATE | TVIF_PARAM;
if (m_tree1.GetItem (&tvItem))
{
//Check if item is expanded, if not we add/remove in 'deep' mode
BOOL fDeepMode = (tvItem.state & TVIS_EXPANDED) ? FALSE : TRUE;
//If the item is 'included' we remove
//If the item is not 'included' we add
if ( ((ISAMTreeItemData*)tvItem.lParam)->included )
{
RemoveNamespaces(item, fDeepMode);
}
else
{
BeginWaitCursor();
BOOL itemSelected = ((ISAMTreeItemData*)tvItem.lParam)->included;
BOOL fChildrenChecked = FALSE;
//Get immediate child items and expand them to 1 level only
//Note: Remember to check if it has already been expanded
HTREEITEM firstChild = NULL;
if (firstChild = m_tree1.GetChildItem (item))
{
//Get absolute name for this item
TV_ITEM tvItem2;
tvItem2.hItem = firstChild;
tvItem2.mask = TVIF_PARAM;
if (m_tree1.GetItem (&tvItem2))
{
//Check if this node has already been checked for children
fChildrenChecked = ((ISAMTreeItemData*)tvItem2.lParam)->fExpanded;
if (!fChildrenChecked)
{
char* txt = ((ISAMTreeItemData*)tvItem2.lParam)->absName;
ISAMBuildTreeChildren (firstChild,
txt,
m_tree1,
*this,
lpszServer,
// m_loginMethod,
lpszUserName,
lpszPassword,
*fIntpretEmptPwdAsBlank,
lpszLocale,
lpszAuthority,
! itemSelected);
}
}
HTREEITEM nextSibling = firstChild;
while (nextSibling = m_tree1.GetNextItem (nextSibling, TVGN_NEXT))
{
TV_ITEM tvItem3;
tvItem3.hItem = nextSibling;
tvItem3.mask = TVIF_PARAM;
if (m_tree1.GetItem (&tvItem3))
{
//Check if this node has already been checked for children
fChildrenChecked = ((ISAMTreeItemData*)tvItem3.lParam)->fExpanded;
if (!fChildrenChecked)
{
char* txt = ((ISAMTreeItemData*)tvItem3.lParam)->absName;
ISAMBuildTreeChildren (nextSibling,
txt,
m_tree1,
*this,
lpszServer,
// m_loginMethod,
lpszUserName,
lpszPassword,
*fIntpretEmptPwdAsBlank,
lpszLocale,
lpszAuthority,
! itemSelected);
}
}
}
}
AddNamespaces (item, fDeepMode);
EndWaitCursor();
}
}
}
}
}
void CConnectionDialog::OnTreeExpand(NMHDR* pNMHDR, LRESULT* pResult)
{
NM_TREEVIEW* lParam = (NM_TREEVIEW*)pNMHDR;
// TODO: Add your control notification handler code here
*pResult = 0;
TV_ITEM tvItem = lParam->itemNew;
UINT action = lParam->action;
HTREEITEM item = tvItem.hItem;
//Check if item is expanded, if not we add/remove in 'deep' mode
BOOL fExpanding = (action == TVE_EXPAND) ? TRUE : FALSE;
if (!fExpanding)
return;
BeginWaitCursor();
char* txt = ((ISAMTreeItemData*)tvItem.lParam)->pszText;
BOOL fChildrenChecked = FALSE;
//Get immediate child items and expand them to 1 level only
//Note: Remember to check if it has already been expanded
HTREEITEM firstChild = NULL;
if (firstChild = m_tree1.GetChildItem (item))
{
//Get absolute name for this item
TV_ITEM tvItem2;
tvItem2.hItem = firstChild;
tvItem2.mask = TVIF_PARAM;
if (m_tree1.GetItem (&tvItem2))
{
//Check if this node has already been checked for children
fChildrenChecked = ((ISAMTreeItemData*)tvItem2.lParam)->fExpanded;
if (!fChildrenChecked)
{
txt = ((ISAMTreeItemData*)tvItem2.lParam)->absName;
ISAMBuildTreeChildren (firstChild,
txt,
m_tree1,
*this,
lpszServer,
// m_loginMethod,
lpszUserName,
lpszPassword,
*fIntpretEmptPwdAsBlank,
lpszLocale,
lpszAuthority,
FALSE);
}
}
HTREEITEM nextSibling = firstChild;
while (nextSibling = m_tree1.GetNextItem (nextSibling, TVGN_NEXT))
{
TV_ITEM tvItem3;
tvItem3.hItem = nextSibling;
tvItem3.mask = TVIF_PARAM;
if (m_tree1.GetItem (&tvItem3))
{
//Check if this node has already been checked for children
fChildrenChecked = ((ISAMTreeItemData*)tvItem3.lParam)->fExpanded;
if (!fChildrenChecked)
{
txt = ((ISAMTreeItemData*)tvItem3.lParam)->absName;
ISAMBuildTreeChildren (nextSibling,
txt,
m_tree1,
*this,
lpszServer,
// m_loginMethod,
lpszUserName,
lpszPassword,
*fIntpretEmptPwdAsBlank,
lpszLocale,
lpszAuthority,
FALSE);
}
}
}
}
EndWaitCursor();
}
void CConnectionDialog::OnClickTree1(NMHDR* pNMHDR, LRESULT* pResult)
{
// TODO: Add your control notification handler code here
*pResult = 0;
//Check if you double-clicked
// if (fDoubleClicked)
// {
// fDoubleClicked = FALSE;
// return;
// }
fDoubleClicked = FALSE;
//Get current position of mouse cursor and perform
//a hit test on the tree control
POINT cursorPos;
GetCursorPos(&cursorPos);
m_tree1.ScreenToClient(&cursorPos);
UINT fFlags = 0;
HTREEITEM item = m_tree1.HitTest(cursorPos, &fFlags);
if (fFlags & TVHT_ONITEMICON)
{
//Support for mouse operation to 'check' the node checkbox
if (item)
{
//Get the item
TV_ITEM tvItem;
tvItem.hItem = item;
tvItem.mask = TVIF_STATE | TVIF_PARAM;
if (m_tree1.GetItem (&tvItem))
{
//Check if item is expanded, if not we add/remove in 'deep' mode
BOOL fDeepMode = (tvItem.state & TVIS_EXPANDED) ? FALSE : TRUE;
//If the item is 'included' we remove
//If the item is not 'included' we add
if ( ((ISAMTreeItemData*)tvItem.lParam)->included )
{
RemoveNamespaces(item, fDeepMode);
}
else
{
BeginWaitCursor();
BOOL itemSelected = ((ISAMTreeItemData*)tvItem.lParam)->included;
BOOL fChildrenChecked = FALSE;
//Get immediate child items and expand them to 1 level only
//Note: Remember to check if it has already been expanded
HTREEITEM firstChild = NULL;
if (firstChild = m_tree1.GetChildItem (item))
{
//Get absolute name for this item
TV_ITEM tvItem2;
tvItem2.hItem = firstChild;
tvItem2.mask = TVIF_PARAM;
if (m_tree1.GetItem (&tvItem2))
{
//Check if this node has already been checked for children
fChildrenChecked = ((ISAMTreeItemData*)tvItem2.lParam)->fExpanded;
if (!fChildrenChecked)
{
char* txt = ((ISAMTreeItemData*)tvItem2.lParam)->absName;
ISAMBuildTreeChildren (firstChild,
txt,
m_tree1,
*this,
lpszServer,
// m_loginMethod,
lpszUserName,
lpszPassword,
*fIntpretEmptPwdAsBlank,
lpszLocale,
lpszAuthority,
! itemSelected);
}
}
HTREEITEM nextSibling = firstChild;
while (nextSibling = m_tree1.GetNextItem (nextSibling, TVGN_NEXT))
{
TV_ITEM tvItem3;
tvItem3.hItem = nextSibling;
tvItem3.mask = TVIF_PARAM;
if (m_tree1.GetItem (&tvItem3))
{
//Check if this node has already been checked for children
fChildrenChecked = ((ISAMTreeItemData*)tvItem3.lParam)->fExpanded;
if (!fChildrenChecked)
{
char* txt = ((ISAMTreeItemData*)tvItem3.lParam)->absName;
ISAMBuildTreeChildren (nextSibling,
txt,
m_tree1,
*this,
lpszServer,
// m_loginMethod,
lpszUserName,
lpszPassword,
*fIntpretEmptPwdAsBlank,
lpszLocale,
lpszAuthority,
! itemSelected);
}
}
}
}
AddNamespaces (item, fDeepMode);
EndWaitCursor();
}
}
}
}
}
void CConnectionDialog::OnDeleteitemTree1(NMHDR* pNMHDR, LRESULT* pResult)
{
NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
// TODO: Add your control notification handler code here
*pResult = 0;
}
// Window Procedure to circumvent bug(?) in mfc
CMap<SHORT, SHORT, CWindowInfo*, CWindowInfo*> windowMap;
LRESULT CALLBACK
MySubClassProc (HWND hWnd, UINT nMsg, WPARAM wParam, LPARAM lParam)
{
CWindowInfo *pwindowInfo = NULL;
VERIFY (windowMap.Lookup ((SHORT) ((DWORD)hWnd & 0xffff), pwindowInfo) && pwindowInfo);
return :: CallWindowProc (pwindowInfo->m_oldWindowProc,
pwindowInfo->m_hWnd,
nMsg, wParam, lParam);
}
BOOL CConnectionDialog::OnInitDialog()
{
ODBCTRACE ("\nWBEM ODBC Driver : CConnectionDialog::OnInitDialog\n");
CDialog::OnInitDialog();
// subclass the window to circumvent a bug (?) in mfc
WNDPROC oldWindowProc = (WNDPROC):: SetWindowLong (m_hWnd, GWL_WNDPROC, (DWORD) MySubClassProc);
CWindowInfo *pwindowInfo = new CWindowInfo (m_hWnd, oldWindowProc);
windowMap.SetAt ((SHORT)((DWORD)m_hWnd & 0xffff), pwindowInfo);
// hook up the controls
m_browseEdit.Attach (::GetDlgItem (m_hWnd, IDC_BROWSE_EDIT));
m_browse.Attach (::GetDlgItem (m_hWnd, IDC_BUTTON_BROWSE));
m_cancelButton.Attach(::GetDlgItem (m_hWnd, IDCANCEL));
m_okButton.Attach (::GetDlgItem (m_hWnd, IDOK));
m_UserName.Attach(::GetDlgItem (m_hWnd, IDC_EDIT_USER_NAME));
m_Password.Attach(::GetDlgItem (m_hWnd, IDC_EDIT_PSWD));
m_Authority.Attach(::GetDlgItem (m_hWnd, IDC_EDIT_AUTH));
m_Locale.Attach(::GetDlgItem (m_hWnd, IDC_EDIT_LOCALE));
m_tree1.Attach(::GetDlgItem (m_hWnd, IDC_TREE1));
m_sysPropCheck.Attach(::GetDlgItem (m_hWnd, IDC_CHECK1));
m_impersonateCheck.Attach(::GetDlgItem (m_hWnd, IDC_CHECK_IMPERSONATE));
m_PassthroughOnlyCheck.Attach(::GetDlgItem (m_hWnd, IDC_CHECK_PASSTHROUGHONLY));
m_messageEdit.Attach (::GetDlgItem (m_hWnd, IDC_MESSAGE));
m_RefreshButton.Attach (::GetDlgItem (m_hWnd, IDC_REFRESH_BUTTON));
m_PwdAsNull.Attach (::GetDlgItem (m_hWnd, IDC_RADIO_NULL));
m_PwdAsBlank.Attach (::GetDlgItem (m_hWnd, IDC_RADIO_BLANK));
// TODO: Add extra initialization here
pCurrentSelectionList = NULL;
pCurrentItem = NULL;
m_browseEdit.LimitText(MAX_SERVER_NAME_LENGTH);
m_UserName.LimitText(MAX_USER_NAME_LENGTH);
m_Password.LimitText(MAX_PASSWORD_LENGTH);
m_Password.SetPasswordChar('*');
m_browseEdit.SetWindowText((lpszServer ? lpszServer : ""));
m_browseEdit.SetModify(FALSE);
m_UserName.SetWindowText((lpszUserName ? lpszUserName : ""));
m_UserName.SetModify(FALSE);
m_Password.SetWindowText((lpszPassword ? lpszPassword : ""));
m_Password.SetModify(FALSE);
//Set Authority
m_Authority.SetWindowText((lpszAuthority ? lpszAuthority : ""));
m_Authority.SetModify(FALSE);
//Set Locale
m_Locale.SetWindowText((lpszLocale ? lpszLocale : ""));
m_Locale.SetModify(FALSE);
//Set System Properties checkbox
m_sysPropCheck.SetCheck(*fSystemProperties ? 1 : 0);
//Set Impersonation checkbox
m_impersonateCheck.SetCheck(impersonateMgr ? 1 : 0);
//Set Passthrough only checkbox
m_PassthroughOnlyCheck.SetCheck(*fPassthroughOnly ? 1 : 0);
//Set the Interpret an empty password as
m_PwdAsNull.SetCheck(*fIntpretEmptPwdAsBlank ? 0 : 1);
m_PwdAsBlank.SetCheck(*fIntpretEmptPwdAsBlank ? 1 : 0);
m_messageEdit.FmtLines(TRUE);
m_messageEdit.ShowWindow(SW_HIDE);
m_tree1.EnableWindow(TRUE);
m_tree1.ShowWindow(SW_SHOWNA);
char buffer [129];
buffer[0] = 0;
LoadString(s_hModule, STR_CONNECT, (char*)buffer, 128);
m_RefreshButton.SetWindowText(buffer);
m_RefreshButton.EnableWindow(FALSE);
// Fill in the first tree with all the namespace information
// The second tree is empty for now. In future this will depend on the
// connection string
//Each node in the CTreeCtrl will be in one of the following 4 states
// 1) node not selected, child nodes not selected
// 2) node not selected, some or all child nodes selected
// 3) node selected, no child nodes selected
// 4) node selected, some or all child nodes selected
m_imageList.Create (32, 16, TRUE, 25, 0);
bmap1.LoadBitmap(IDB_BITMAP1);
bmap2.LoadBitmap(IDB_BITMAP2);
bmap3.LoadBitmap(IDB_BITMAP3);
bmap4.LoadBitmap(IDB_BITMAP4);
bmask.LoadBitmap(IDB_MASK);
m_idxMode1Image = m_imageList.Add (&bmap1, &bmask);
m_idxMode2Image = m_imageList.Add (&bmap2, &bmask);
m_idxMode3Image = m_imageList.Add (&bmap3, &bmask);
m_idxMode4Image = m_imageList.Add (&bmap4, &bmask);
m_tree1.SetImageList (&m_imageList, TVSIL_NORMAL);
//Tidy Up
bmap1.DeleteObject();
bmap2.DeleteObject();
bmap3.DeleteObject();
bmap4.DeleteObject();
bmask.DeleteObject();
LONG oldStyle = :: GetWindowLong(m_tree1.m_hWnd, GWL_STYLE);
:: SetWindowLong (m_tree1.m_hWnd,
GWL_STYLE,
oldStyle | TVS_HASLINES | TVS_HASBUTTONS | TVS_LINESATROOT);
BOOL fStatusOK = TRUE;
//We may or may not be using impersonation
//but if we do, has it worked ?
if (*fImpersonation && impersonateMgr)
{
//Check if impersonation request has worked
if (! impersonateMgr->CanWeImpersonate() )
{
fStatusOK = FALSE;
}
}
if (fStatusOK)
{
BeginWaitCursor();
HTREEITEM hDummy;
SWORD treeStatus = ISAMBuildTree (TVI_ROOT, "root", /*"root\\default",*/
m_tree1, *this,
lpszServer ? lpszServer : NULL, // server
// m_loginMethod,
lpszUserName ? lpszUserName : NULL, // username
lpszPassword ? lpszPassword : NULL, // password
*fIntpretEmptPwdAsBlank,
lpszLocale,
lpszAuthority,
TRUE, FALSE, hDummy);
EndWaitCursor();
// enumerate elements in list and set the namespaces passed in in the
// connection string to be included
POSITION pos;
CString key;
CNamespace *pNamespace;
for (pos = pMapStringToObIn->GetStartPosition ();pos != NULL;)
{
pMapStringToObIn->GetNextAssoc (pos, key, (CObject*&)pNamespace);
HTREEITEM hFoundItem;
int len = (pNamespace->GetName ()).GetLength ();
LPTSTR str = (pNamespace->GetName ()).GetBuffer (len);
if (FindAbsName (str, m_tree1.GetRootItem (), hFoundItem))
{
//include it
AddNamespaces (hFoundItem, FALSE /* shallow */);
}
else
{
// attempt to create namespace
if (CreateNamespace (str, m_tree1.GetRootItem (), hFoundItem))
{
//include it
AddNamespaces (hFoundItem, FALSE /* shallow */);
}
}
}
//Update state of OK pushbutton depending on
//number of cels selected
m_okButton.EnableWindow(cSelectedCels ? TRUE : FALSE);
}
else
{
//Impersonation failed
m_tree1.EnableWindow(FALSE);
m_tree1.ShowWindow(SW_HIDE);
char buffer [128 + 1];
buffer[0] = 0;
LoadString(s_hModule, STR_FAILED_WBEM_CONN,
(char*)buffer, 128);
m_messageEdit.SetWindowText(buffer);
m_messageEdit.ShowWindow(SW_SHOWNA);
}
//Final check to see if proper connection parameters were entered
if (!fConnParmSpecified)
{
ConnectionParameterChange();
//Give the user name edit box focus
m_UserName.SetFocus();
}
return TRUE; // return TRUE unless you set the focus to a control
// EXCEPTION: OCX Property Pages should return FALSE
}
void CConnectionDialog::OnNcDestroy()
{
CWindowInfo *pwindowInfo = NULL;
BOOL found = windowMap.Lookup ((SHORT) ((DWORD)m_hWnd & 0xffff), pwindowInfo);
ASSERT (found);
if (found)
{
// unsubclass the window
:: SetWindowLong (m_hWnd, GWL_WNDPROC, (DWORD) pwindowInfo->m_oldWindowProc);
delete pwindowInfo;
windowMap.RemoveKey ((SHORT) ((DWORD)m_hWnd & 0xffff));
}
m_browseEdit.Detach ();
m_browse.Detach ();
m_cancelButton.Detach();
m_okButton.Detach ();
m_UserName.Detach();
m_Password.Detach();
m_Authority.Detach();
m_Locale.Detach();
m_tree1.Detach();
m_sysPropCheck.Detach();
m_impersonateCheck.Detach();
m_PassthroughOnlyCheck.Detach();
m_messageEdit.Detach();
m_RefreshButton.Detach();
m_PwdAsNull.Detach();
m_PwdAsBlank.Detach();
CDialog :: OnNcDestroy ();
}
void CConnectionDialog::OnOK()
{
// TODO: Add extra validation here
GenerateOutMap (m_tree1.GetRootItem ());
GenerateOutString (m_tree1.GetRootItem ());
//Copy back new Username and Password
lpszUserName[0] = 0;
lpszPassword[0] = 0;
m_UserName.GetWindowText(lpszUserName, MAX_USER_NAME_LENGTH);
m_Password.GetWindowText(lpszPassword, MAX_PASSWORD_LENGTH);
//Copy Authority & Locale
if (lpszAuthority)
delete lpszAuthority;
*lpszAuthorityOut = NULL;
int AuthLen = m_Authority.GetWindowTextLength();
if (AuthLen)
{
*lpszAuthorityOut = new char [AuthLen + 1];
(*lpszAuthorityOut)[0] = 0;
m_Authority.GetWindowText(*lpszAuthorityOut, AuthLen + 1);
(*lpszAuthorityOut)[AuthLen] = 0;
}
if (lpszLocale)
delete lpszLocale;
*lpszLocaleOut = NULL;
int LocaleLen = m_Locale.GetWindowTextLength();
if (LocaleLen)
{
*lpszLocaleOut = new char [LocaleLen + 1];
(*lpszLocaleOut)[0] = 0;
m_Locale.GetWindowText(*lpszLocaleOut, LocaleLen + 1);
(*lpszLocaleOut)[LocaleLen] = 0;
}
HTREEITEM hTreeItem = m_tree1.GetRootItem();
CleanUpTreeCtrl(hTreeItem);
CDialog::OnOK();
}
void CConnectionDialog :: CleanUpTreeCtrl(HTREEITEM& hTreeItem)
{
if (pCurrentSelectionList)
{
ISAMTreeItemData* theNext = NULL;
do
{
theNext = pCurrentSelectionList->pNext;
delete pCurrentSelectionList;
pCurrentSelectionList = theNext;
} while (pCurrentSelectionList);
}
pCurrentSelectionList = NULL;
pCurrentItem = NULL;
}
void CConnectionDialog::OnButtonSysProp()
{
//Clicked on the System Properties checkbox
*fSystemProperties = m_sysPropCheck.GetCheck() ? TRUE : FALSE;
}
void CConnectionDialog::OnButtonInterpretEmpty()
{
//Clicked on one of the Interpret empty password as radio buttons
*fIntpretEmptPwdAsBlank = m_PwdAsBlank.GetCheck() ? TRUE : FALSE;
ConnectionParameterChange();
}
void CConnectionDialog::OnButtonImpersonation()
{
//Clicked on the Impersonation checkbox
*fImpersonation = m_impersonateCheck.GetCheck() ? TRUE : FALSE;
ConnectionParameterChange();
}
void CConnectionDialog::OnButtonPassthroughOnly()
{
//Clicked on the Passthrough only checkbox
*fPassthroughOnly = m_PassthroughOnlyCheck.GetCheck() ? TRUE : FALSE;
}
void CConnectionDialog::OnButtonBrowse()
{
// TODO: Add your control notification handler code here
m_browse.EnableWindow(FALSE);
BeginWaitCursor();
//Show dialog box
CBrowseDialog d;
int result = d.DoModal();
m_browse.EnableWindow(TRUE);
EndWaitCursor();
//abort if OK pushbutton was not pressed
if (result != IDOK)
return;
m_browseEdit.SetWindowText( d.GetServerName() );
m_browseEdit.SetModify(FALSE);
//update tree
RefreshTree();
}
void CConnectionDialog::OnCancel()
{
// TODO: Add extra cleanup here
HTREEITEM hTreeItem = m_tree1.GetRootItem();
CleanUpTreeCtrl(hTreeItem);
CDialog::OnCancel();
}
BOOL GetWbemDirectory(_bstr_t& wbemDir)
{
HKEY keyHandle = (HKEY)1;
TCHAR buff [1001];
DWORD sizebuff = 1000;
long fStatus = 0;
_bstr_t location("Software\\ODBC\\ODBCINST.INI\\WBEM ODBC Driver");
buff[0] = 0;
fStatus = RegOpenKeyEx(HKEY_LOCAL_MACHINE,
(LPCTSTR)location, 0, KEY_READ, &keyHandle);
if (fStatus == ERROR_SUCCESS)
{
DWORD typeValue;
_bstr_t subkey("Help");
fStatus = RegQueryValueEx(keyHandle, (LPCTSTR)subkey, NULL,
&typeValue, (LPBYTE)buff, &sizebuff);
if (fStatus == ERROR_SUCCESS)
{
wbemDir = buff;
return TRUE;
}
}
return FALSE;
}
void CConnectionDialog::OnHelp()
{
_bstr_t sName = ">Main";
//Get the WBEM directory
_bstr_t wbemDir;
if ( !GetWbemDirectory(wbemDir) )
return;
_bstr_t sPrefix = "mk:@MSITStore:" + wbemDir + "\\WBEMDR32.CHM";
_bstr_t sIndex = sPrefix + "::" + "/_hmm_connecting_to_an_odbc_data_source.htm";
HtmlHelp
(
NULL, // HWND - this is where WM_TCARD messages will be sent
sName, // Use "Main" window name
HH_DISPLAY_TOPIC, // We want to display a topic please
(DWORD)(TCHAR*)sIndex // Full path of topic to display
);
}
void CConnectionDialog::OnDblclkTree1(NMHDR* pNMHDR, LRESULT* pResult)
{
// TODO: Add your control notification handler code here
fDoubleClicked = TRUE;
*pResult = 0;
}
void CConnectionDialog::ConnectionParameterChange()
{
m_tree1.EnableWindow(FALSE);
m_tree1.ShowWindow(SW_HIDE);
char buffer [128 + 1];
buffer[0] = 0;
LoadString(s_hModule, STR_PLEASE_REFRESH,
(char*)buffer, 128);
m_messageEdit.SetWindowText(buffer);
m_messageEdit.ShowWindow(SW_SHOWNA);
LoadString(s_hModule, STR_CONNECT, (char*)buffer, 128);
m_RefreshButton.SetWindowText(buffer);
m_RefreshButton.EnableWindow(TRUE);
m_okButton.EnableWindow(FALSE);
}
void CConnectionDialog::OnLocaleChange()
{
ConnectionParameterChange();
}
void CConnectionDialog::OnAuthorityChange()
{
ConnectionParameterChange();
}
void CConnectionDialog::OnUserNameChange()
{
ConnectionParameterChange();
}
void CConnectionDialog::OnPasswordChange()
{
ConnectionParameterChange();
}
void CConnectionDialog::OnServerChange()
{
ConnectionParameterChange();
}
void CConnectionDialog::OnButtonRefresh()
{
char buffer [129];
buffer[0] = 0;
m_tree1.EnableWindow(TRUE);
m_tree1.ShowWindow(SW_SHOWNA);
m_messageEdit.ShowWindow(SW_HIDE);
//Disable button and rename to CONNECT
LoadString(s_hModule, STR_CONNECT, (char*)buffer, 128);
m_RefreshButton.SetWindowText(buffer);
m_RefreshButton.EnableWindow(FALSE);
BOOL status = RefreshTree();
if (status)
{
//If refresh was successful
//enable button and rename to REFRESH
LoadString(s_hModule, STR_REFRESH, (char*)buffer, 128);
m_RefreshButton.SetWindowText(buffer);
m_RefreshButton.EnableWindow(TRUE);
}
//Give the tree control or message text focus
if ( m_tree1.IsWindowEnabled() )
m_tree1.SetFocus();
else
m_messageEdit.SetFocus();
}
void CConnectionDialog::OnKillfocusBrowseEdit()
{
/*
// TODO: Add your control notification handler code here
//Check if server, user name or password have been changed
//If so, refresh tree
if ( m_browseEdit.GetModify() || m_UserName.GetModify() || m_Password.GetModify() )
{
//reset flags
m_browseEdit.SetModify(FALSE);
m_UserName.SetModify(FALSE);
m_Password.SetModify(FALSE);
//refresh tree view
RefreshTree();
}
*/
}
BOOL CConnectionDialog::RefreshTree()
{
ODBCTRACE ("\nWBEM ODBC Driver : CConnectionDialog::RefreshTree\n");
BOOL status = FALSE;
//Update tree view with namespaces associated with new server
HTREEITEM hTreeItem = m_tree1.GetRootItem();
CleanUpTreeCtrl(hTreeItem);
m_tree1.DeleteAllItems();
cSelectedCels = 0;
//Refresh Window
UpdateWindow();
lpszServer[0] = 0;
lpszUserName[0] = 0;
lpszPassword[0] = 0;
m_browseEdit.GetWindowText(lpszServer, MAX_SERVER_NAME_LENGTH);
m_UserName.GetWindowText(lpszUserName, MAX_USER_NAME_LENGTH);
m_Password.GetWindowText(lpszPassword, MAX_PASSWORD_LENGTH);
//Copy Authority & Locale
if (lpszAuthority)
delete lpszAuthority;
lpszAuthority = NULL;
int AuthLen = m_Authority.GetWindowTextLength();
if (AuthLen)
{
lpszAuthority = new char [AuthLen + 1];
lpszAuthority[0] = 0;
m_Authority.GetWindowText(lpszAuthority, AuthLen + 1);
lpszAuthority[AuthLen] = 0;
ODBCTRACE(_T("\nWBEM ODBC Driver : CConnectionDialog::RefreshTree\n"));
ODBCTRACE(_T(" Authority :"));
ODBCTRACE(_T(lpszAuthority));
ODBCTRACE(_T("\n"));
}
if (lpszLocale)
delete lpszLocale;
lpszLocale = NULL;
int LocaleLen = m_Locale.GetWindowTextLength();
if (LocaleLen)
{
lpszLocale = new char [LocaleLen + 1];
lpszLocale[0] = 0;
m_Locale.GetWindowText(lpszLocale, LocaleLen + 1);
lpszLocale[LocaleLen] = 0;
}
//Update impersonation if applicable
BOOL fImpersonateStatusOK = TRUE;
if (*fImpersonation)
{
ODBCTRACE ("\nWBEM ODBC Driver : Impersonation checkbox ON\n");
delete impersonateMgr;
impersonateMgr = NULL;
//Now check if impersonation is necessary
//only if connecting locally
if (IsLocalServer(lpszServer))
{
impersonateMgr = new ImpersonationManager(lpszUserName, lpszPassword, lpszAuthority);
//Check if impersonation request has worked
if (! impersonateMgr->CanWeImpersonate() )
{
ODBCTRACE ("\nWBEM ODBC Driver : We cannot impersonate\n");
fImpersonateStatusOK = FALSE;
}
else
{
ODBCTRACE ("\nWBEM ODBC Driver : We can impersonate\n");
//Do the actual impersonation so cloaking will work
if ( ! impersonateMgr->ImpersonatingNow() )
impersonateMgr->Impersonate("CConnectionDialog");
}
}
else
{
ODBCTRACE("\nWBEM ODBC Driver : Server not detected as local, not impersonating\n");
}
}
else
{
ODBCTRACE ("\nWBEM ODBC Driver : Impersonation checkbox OFF\n");
}
BeginWaitCursor();
//Disable OK pushbutton when updating
m_okButton.EnableWindow(FALSE);
//Check if you can connect using specified authentication information
char* lpRootNamespace = "root";
DWORD dwAuthLevel = 0;
DWORD dwImpLevel = 0;
COAUTHIDENTITY* pAuthIdent = NULL;
IWbemServicesPtr pGateway = NULL;
ISAMGetGatewayServer (pGateway,
lpszServer ? (LPUSTR)lpszServer : NULL, //server
// m_loginMethod,
(LPUSTR)lpRootNamespace,
lpszUserName ? (LPUSTR)lpszUserName : NULL, // username
lpszPassword ? (LPUSTR)lpszPassword : NULL, // password
(LPUSTR)lpszLocale, //locale
(LPUSTR)lpszAuthority,//authority
dwAuthLevel,
dwImpLevel,
*fIntpretEmptPwdAsBlank,
&pAuthIdent
);
if ( pAuthIdent )
{
WbemFreeAuthIdentity( pAuthIdent );
pAuthIdent = NULL;
}
if (fImpersonateStatusOK && (pGateway != NULL))
{
ODBCTRACE ("\nWBEM ODBC Driver : fImpersonateStatusOK && pGateway\n");
BOOL fIsLocalConnection = IsLocalServer((LPSTR) lpszServer);
//cloaking
if ( fIsLocalConnection && IsW2KOrMore() )
{
WbemSetDynamicCloaking(pGateway, dwAuthLevel, dwImpLevel);
}
status = TRUE;
HTREEITEM hDummy;
ISAMBuildTree (TVI_ROOT, "root", /*"root\\default",*/
m_tree1, *this,
lpszServer ? lpszServer : NULL, //server
// m_loginMethod,
lpszUserName ? lpszUserName : NULL, // username
lpszPassword ? lpszPassword : NULL, // password
*fIntpretEmptPwdAsBlank,
lpszLocale,
lpszAuthority,
TRUE, FALSE, hDummy);
//finished impersonating
// if ( (*fImpersonation) && (impersonateMgr) )
// impersonateMgr->RevertToYourself("ConnectionDialog");
}
else
{
ODBCTRACE ("\nWBEM ODBC Driver : Failed Connection\n");
status = FALSE;
m_tree1.EnableWindow(FALSE);
m_tree1.ShowWindow(SW_HIDE);
char buffer [128 + 1];
buffer[0] = 0;
LoadString(s_hModule, STR_FAILED_WBEM_CONN,
(char*)buffer, 128);
m_messageEdit.SetWindowText(buffer);
m_messageEdit.ShowWindow(SW_SHOWNA);
}
EndWaitCursor();
if (status)
ODBCTRACE ("\nWBEM ODBC Driver : CConnectionDialog::RefreshTree returns TRUE\n");
else
ODBCTRACE ("\nWBEM ODBC Driver : CConnectionDialog::RefreshTree returns FALSE\n");
return status;
}
Reference in a new issue View git blame Copy permalink