525 lines
12 KiB
C++
525 lines
12 KiB
C++
// SelectSystemsDlg.cpp : implementation file
|
|
//
|
|
|
|
#include "stdafx.h"
|
|
#include "snapin.h"
|
|
#include "SelectSystemsDlg.h"
|
|
#include <commctrl.h>
|
|
#include <lmerr.h>
|
|
#include <lmcons.h>
|
|
#include <lmserver.h>
|
|
#include <lmapibuf.h>
|
|
#include <lmaccess.h>
|
|
#include <lmwksta.h>
|
|
#include <mmc.h>
|
|
|
|
#ifdef _DEBUG
|
|
#define new DEBUG_NEW
|
|
#undef THIS_FILE
|
|
static char THIS_FILE[] = __FILE__;
|
|
#endif
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CSelectSystemsDlg dialog
|
|
|
|
|
|
CSelectSystemsDlg::CSelectSystemsDlg(CWnd* pParent /*=NULL*/)
|
|
: CResizeableDialog(CSelectSystemsDlg::IDD, pParent)
|
|
{
|
|
//{{AFX_DATA_INIT(CSelectSystemsDlg)
|
|
m_sDomain = _T("");
|
|
m_sSystems = _T("");
|
|
//}}AFX_DATA_INIT
|
|
}
|
|
|
|
|
|
void CSelectSystemsDlg::DoDataExchange(CDataExchange* pDX)
|
|
{
|
|
CResizeableDialog::DoDataExchange(pDX);
|
|
//{{AFX_DATA_MAP(CSelectSystemsDlg)
|
|
DDX_Control(pDX, IDC_EDIT_SYSTEMS, m_SystemsEditBox);
|
|
DDX_Control(pDX, IDC_LIST_SYSTEMS, m_SystemsList);
|
|
DDX_Control(pDX, IDC_COMBO_DOMAIN, m_Domains);
|
|
DDX_CBString(pDX, IDC_COMBO_DOMAIN, m_sDomain);
|
|
DDX_Text(pDX, IDC_EDIT_SYSTEMS, m_sSystems);
|
|
//}}AFX_DATA_MAP
|
|
|
|
CWnd* pCheckNamesBtn = NULL;
|
|
if( (pCheckNamesBtn = GetDlgItem(IDC_BUTTON_CHECK_NAMES)) != NULL )
|
|
{
|
|
pCheckNamesBtn->EnableWindow( ! m_sSystems.IsEmpty() );
|
|
}
|
|
}
|
|
|
|
|
|
UINT CSelectSystemsDlg::AddDomains(LPVOID pParam)
|
|
{
|
|
CWaitCursor wait;
|
|
|
|
HWND* pHwnd = (HWND*)pParam;
|
|
|
|
if( ! pHwnd )
|
|
{
|
|
return 1L;
|
|
}
|
|
|
|
if( ! ::IsWindow(*pHwnd) )
|
|
{
|
|
return 1L;
|
|
}
|
|
|
|
HWND hDomainCombobox = ::GetDlgItem(*pHwnd,IDC_COMBO_DOMAIN);
|
|
::EnableWindow(hDomainCombobox,FALSE);
|
|
|
|
CString sWindowTitle;
|
|
sWindowTitle.LoadString(IDS_STRING_WORKING);
|
|
::SetWindowText(hDomainCombobox,sWindowTitle);
|
|
|
|
// enumerate all domains
|
|
NET_API_STATUS status = NERR_Success;
|
|
LPSERVER_INFO_100 lpserverinfo = NULL;
|
|
DWORD dwTotalEntries = 0L;
|
|
DWORD dwEntriesRead = 0L;
|
|
DWORD dwResumeHandle = 0L;
|
|
DWORD dwSize = MAX_PREFERRED_LENGTH;
|
|
|
|
status = NetServerEnum(NULL,100,(LPBYTE*)&lpserverinfo,dwSize,&dwEntriesRead,&dwTotalEntries,
|
|
SV_TYPE_DOMAIN_ENUM,NULL,&dwResumeHandle);
|
|
|
|
if( status != NERR_Success && status == ERROR_MORE_DATA )
|
|
{
|
|
lpserverinfo = NULL;
|
|
dwSize = sizeof(SERVER_INFO_100)*dwTotalEntries*_MAX_PATH;
|
|
dwTotalEntries = 0L;
|
|
dwEntriesRead = 0L;
|
|
dwResumeHandle = 0L;
|
|
|
|
status = NetServerEnum(NULL,100,(LPBYTE*)&lpserverinfo,dwSize,&dwEntriesRead,&dwTotalEntries,
|
|
SV_TYPE_DOMAIN_ENUM,NULL,&dwResumeHandle);
|
|
}
|
|
|
|
::SendDlgItemMessage(*pHwnd,IDC_COMBO_DOMAIN,CB_RESETCONTENT,0,0);
|
|
|
|
for( DWORD dw = 0L; dw < dwTotalEntries; dw++ )
|
|
{
|
|
CString sDomainName = lpserverinfo[dw].sv100_name;
|
|
// add domains to the combobox
|
|
::SendMessage(hDomainCombobox,CB_ADDSTRING,0,(LPARAM)(LPCTSTR)sDomainName);
|
|
}
|
|
|
|
NetApiBufferFree(lpserverinfo);
|
|
|
|
::SetWindowText(hDomainCombobox,_T(""));
|
|
::EnableWindow(hDomainCombobox,TRUE);
|
|
|
|
delete pHwnd;
|
|
|
|
return 0L;
|
|
|
|
}
|
|
|
|
UINT CSelectSystemsDlg::AddSystems(LPVOID pParam)
|
|
{
|
|
CWaitCursor wait;
|
|
|
|
HWND* pHwnd = (HWND*)pParam;
|
|
|
|
if( ! pHwnd )
|
|
{
|
|
return 1L;
|
|
}
|
|
|
|
if( ! ::IsWindow(*pHwnd) )
|
|
{
|
|
return 1L;
|
|
}
|
|
|
|
HWND hSystemListCtrl = ::GetDlgItem(*pHwnd,IDC_LIST_SYSTEMS);
|
|
HWND hDomainCombobox = ::GetDlgItem(*pHwnd,IDC_COMBO_DOMAIN);
|
|
|
|
::EnableWindow(hSystemListCtrl,FALSE);
|
|
::EnableWindow(hDomainCombobox,FALSE);
|
|
|
|
ListView_DeleteAllItems(hSystemListCtrl);
|
|
|
|
// add the "Working..." item
|
|
|
|
CString sText;
|
|
sText.LoadString(IDS_STRING_WORKING);
|
|
LVITEM lvi;
|
|
ZeroMemory(&lvi,sizeof(LVITEM));
|
|
lvi.mask = LVIF_TEXT;
|
|
lvi.pszText = (LPTSTR)(LPCTSTR)sText;
|
|
|
|
ListView_InsertItem(hSystemListCtrl,&lvi);
|
|
ListView_SetColumnWidth(hSystemListCtrl,0,LVSCW_AUTOSIZE_USEHEADER);
|
|
ListView_SetColumnWidth(hSystemListCtrl,1,LVSCW_AUTOSIZE_USEHEADER);
|
|
|
|
// get the domain name from the combobox
|
|
CString sDomainName;
|
|
int iCurSel = (int)::SendDlgItemMessage(*pHwnd,IDC_COMBO_DOMAIN,CB_GETCURSEL,0,0);
|
|
if( iCurSel != -1 )
|
|
{
|
|
int iTextLen = (int)::SendDlgItemMessage(*pHwnd,IDC_COMBO_DOMAIN,CB_GETLBTEXTLEN,iCurSel,0);
|
|
::SendDlgItemMessage(*pHwnd,IDC_COMBO_DOMAIN,CB_GETLBTEXT,iCurSel,(LPARAM)sDomainName.GetBuffer(iTextLen + 1));
|
|
sDomainName.ReleaseBuffer();
|
|
}
|
|
else
|
|
{
|
|
int iTextLen = ::GetWindowTextLength(hDomainCombobox);
|
|
::GetWindowText(hDomainCombobox,sDomainName.GetBuffer(iTextLen+1),iTextLen+1);
|
|
sDomainName.ReleaseBuffer();
|
|
}
|
|
|
|
// enumerate for all the systems in the domain
|
|
NET_API_STATUS status = NERR_Success;
|
|
LPSERVER_INFO_101 lpserverinfo = NULL;
|
|
DWORD dwTotalEntries = 0L;
|
|
DWORD dwEntriesRead = 0L;
|
|
DWORD dwResumeHandle = 0L;
|
|
DWORD dwSize = MAX_PREFERRED_LENGTH;
|
|
|
|
status = NetServerEnum(NULL,101,(LPBYTE*)&lpserverinfo,dwSize,&dwEntriesRead,&dwTotalEntries,
|
|
SV_TYPE_NT,sDomainName,&dwResumeHandle);
|
|
|
|
if( status != NERR_Success && status == ERROR_MORE_DATA )
|
|
{
|
|
lpserverinfo = NULL;
|
|
dwSize = sizeof(SERVER_INFO_101)*dwTotalEntries*_MAX_PATH;
|
|
dwTotalEntries = 0L;
|
|
dwEntriesRead = 0L;
|
|
dwResumeHandle = 0L;
|
|
|
|
status = NetServerEnum(NULL,101,(LPBYTE*)&lpserverinfo,dwSize,&dwEntriesRead,&dwTotalEntries,
|
|
SV_TYPE_NT,sDomainName,&dwResumeHandle);
|
|
}
|
|
|
|
ListView_DeleteAllItems(hSystemListCtrl);
|
|
|
|
bool bCommentFound = false;
|
|
|
|
for( DWORD dw = 0L; dw < dwTotalEntries; dw++ )
|
|
{
|
|
CString sSystemName = lpserverinfo[dw].sv101_name;
|
|
|
|
// add system to the list control
|
|
LVITEM lvi;
|
|
ZeroMemory(&lvi,sizeof(LVITEM));
|
|
lvi.mask = LVIF_TEXT;
|
|
lvi.pszText = (LPTSTR)(LPCTSTR)sSystemName;
|
|
|
|
int iItemIndex = ListView_InsertItem(hSystemListCtrl,&lvi);
|
|
|
|
lvi.iItem = iItemIndex;
|
|
lvi.iSubItem = 1;
|
|
lvi.pszText = lpserverinfo[dw].sv101_comment;
|
|
if( _tcslen(lvi.pszText) > 0 )
|
|
{
|
|
bCommentFound = true;
|
|
ListView_SetItem(hSystemListCtrl,&lvi);
|
|
}
|
|
}
|
|
|
|
if( dwTotalEntries <= 0L )
|
|
{
|
|
ListView_SetColumnWidth(hSystemListCtrl,0,LVSCW_AUTOSIZE_USEHEADER);
|
|
}
|
|
else
|
|
{
|
|
ListView_SetColumnWidth(hSystemListCtrl,0,LVSCW_AUTOSIZE);
|
|
}
|
|
|
|
if( bCommentFound )
|
|
{
|
|
ListView_SetColumnWidth(hSystemListCtrl,1,LVSCW_AUTOSIZE);
|
|
}
|
|
else
|
|
{
|
|
ListView_SetColumnWidth(hSystemListCtrl,1,LVSCW_AUTOSIZE_USEHEADER);
|
|
}
|
|
|
|
NetApiBufferFree(lpserverinfo);
|
|
|
|
::EnableWindow(hSystemListCtrl,TRUE);
|
|
::EnableWindow(hDomainCombobox,TRUE);
|
|
|
|
delete pHwnd;
|
|
|
|
return 0L;
|
|
}
|
|
|
|
void CSelectSystemsDlg::CompileArrayOfSystems()
|
|
{
|
|
m_saSystems.RemoveAll();
|
|
|
|
LPTSTR lpszSystems = new TCHAR[m_sSystems.GetLength()+1];
|
|
_tcscpy(lpszSystems,m_sSystems);
|
|
|
|
LPTSTR lpszToken = _tcstok(lpszSystems,_T(";"));
|
|
|
|
while( lpszToken )
|
|
{
|
|
CString sSystem = lpszToken;
|
|
sSystem.TrimLeft(_T(" "));
|
|
sSystem.TrimRight(_T(" "));
|
|
m_saSystems.Add(sSystem);
|
|
lpszToken = _tcstok(NULL,_T(";"));
|
|
}
|
|
|
|
delete[] lpszSystems;
|
|
}
|
|
|
|
bool CSelectSystemsDlg::IsInSystemArray(const CString& sSystem)
|
|
{
|
|
for( int i = 0; i < m_saSystems.GetSize(); i++ )
|
|
{
|
|
if( m_saSystems[i] == sSystem )
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool CSelectSystemsDlg::CheckSystemNames()
|
|
{
|
|
CWaitCursor wait;
|
|
|
|
for( int i = (int)m_saSystems.GetSize()-1; i >= 0; i-- )
|
|
{
|
|
IWbemServices* pServices = NULL;
|
|
BOOL bAvail = FALSE;
|
|
|
|
if( ! m_saSystems[i].IsEmpty() )
|
|
{
|
|
HRESULT hr = CnxGetConnection(m_saSystems[i],pServices,bAvail);
|
|
if( hr == E_FAIL )
|
|
{
|
|
MessageBeep(MB_ICONEXCLAMATION);
|
|
int iIndex = m_sSystems.Find(m_saSystems[i]);
|
|
m_SystemsEditBox.SetFocus();
|
|
m_SystemsEditBox.SetSel(iIndex,iIndex+m_saSystems[i].GetLength(),TRUE);
|
|
CnxDisplayErrorMsgBox(hr,m_saSystems[i]);
|
|
return false;
|
|
}
|
|
|
|
if( pServices )
|
|
{
|
|
pServices->Release();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
m_saSystems.RemoveAt(i);
|
|
}
|
|
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
BEGIN_MESSAGE_MAP(CSelectSystemsDlg, CResizeableDialog)
|
|
//{{AFX_MSG_MAP(CSelectSystemsDlg)
|
|
ON_NOTIFY(NM_DBLCLK, IDC_LIST_SYSTEMS, OnDblclkListSystems)
|
|
ON_BN_CLICKED(IDC_BUTTON_HELP, OnButtonHelp)
|
|
ON_CBN_DROPDOWN(IDC_COMBO_DOMAIN, OnDropdownComboDomain)
|
|
ON_CBN_CLOSEUP(IDC_COMBO_DOMAIN, OnCloseupComboDomain)
|
|
ON_CBN_SELENDOK(IDC_COMBO_DOMAIN, OnSelendokComboDomain)
|
|
ON_EN_CHANGE(IDC_EDIT_SYSTEMS, OnChangeEditSystems)
|
|
ON_BN_CLICKED(IDC_BUTTON_CHECK_NAMES, OnButtonCheckNames)
|
|
ON_NOTIFY(NM_SETFOCUS, IDC_LIST_SYSTEMS, OnSetfocusListSystems)
|
|
ON_NOTIFY(NM_KILLFOCUS, IDC_LIST_SYSTEMS, OnKillfocusListSystems)
|
|
ON_BN_CLICKED(IDC_BUTTON_ADD, OnButtonAdd)
|
|
ON_NOTIFY(NM_CLICK, IDC_LIST_SYSTEMS, OnClickListSystems)
|
|
//}}AFX_MSG_MAP
|
|
END_MESSAGE_MAP()
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CSelectSystemsDlg message handlers
|
|
|
|
BOOL CSelectSystemsDlg::OnInitDialog()
|
|
{
|
|
CResizeableDialog::OnInitDialog();
|
|
|
|
SetControlInfo(IDC_STATIC_TITLE, ANCHOR_LEFT | ANCHOR_TOP );
|
|
SetControlInfo(IDC_STATIC_BAR, ANCHOR_LEFT | ANCHOR_TOP | RESIZE_HOR);
|
|
SetControlInfo(IDC_STATIC_LOOK_IN, ANCHOR_LEFT | ANCHOR_TOP );
|
|
SetControlInfo(IDC_LIST_SYSTEMS, ANCHOR_LEFT | ANCHOR_TOP | RESIZE_HOR | RESIZE_VER);
|
|
SetControlInfo(IDC_COMBO_DOMAIN, ANCHOR_LEFT | ANCHOR_TOP | RESIZE_HOR);
|
|
SetControlInfo(IDC_BUTTON_ADD, ANCHOR_BOTTOM | ANCHOR_LEFT );
|
|
SetControlInfo(IDC_BUTTON_CHECK_NAMES, ANCHOR_BOTTOM | ANCHOR_LEFT );
|
|
SetControlInfo(IDC_EDIT_SYSTEMS, ANCHOR_BOTTOM | ANCHOR_LEFT | RESIZE_HOR );
|
|
SetControlInfo(IDOK, ANCHOR_BOTTOM | ANCHOR_LEFT );
|
|
SetControlInfo(IDCANCEL, ANCHOR_BOTTOM | ANCHOR_LEFT );
|
|
SetControlInfo(IDC_BUTTON_HELP, ANCHOR_BOTTOM | ANCHOR_LEFT );
|
|
|
|
|
|
NET_API_STATUS status = NERR_Success;
|
|
LPWKSTA_INFO_100 pinfo = NULL;
|
|
|
|
status = NetWkstaGetInfo(NULL,100,(LPBYTE*)&pinfo);
|
|
|
|
if( status == NERR_Success )
|
|
{
|
|
m_Domains.AddString(pinfo->wki100_langroup);
|
|
m_Domains.SetCurSel(0);
|
|
OnSelendokComboDomain();
|
|
}
|
|
|
|
NetApiBufferFree(pinfo);
|
|
|
|
CString sTitle;
|
|
sTitle.LoadString(IDS_STRING_NAME);
|
|
m_SystemsList.InsertColumn(0,sTitle,LVCFMT_LEFT,LVSCW_AUTOSIZE_USEHEADER);
|
|
sTitle.LoadString(IDS_STRING_COMMENT);
|
|
m_SystemsList.InsertColumn(1,sTitle,LVCFMT_LEFT,LVSCW_AUTOSIZE_USEHEADER);
|
|
|
|
m_SystemsList.SetColumnWidth(0,LVSCW_AUTOSIZE_USEHEADER);
|
|
m_SystemsList.SetColumnWidth(1,LVSCW_AUTOSIZE_USEHEADER);
|
|
|
|
return TRUE; // return TRUE unless you set the focus to a control
|
|
// EXCEPTION: OCX Property Pages should return FALSE
|
|
}
|
|
|
|
void CSelectSystemsDlg::OnButtonAdd()
|
|
{
|
|
CompileArrayOfSystems();
|
|
|
|
if( ! m_sSystems.IsEmpty() )
|
|
{
|
|
m_sSystems += _T(" ; ");
|
|
}
|
|
|
|
POSITION pos = m_SystemsList.GetFirstSelectedItemPosition();
|
|
int iIndex = 0;
|
|
while( pos && iIndex >= 0 )
|
|
{
|
|
iIndex = m_SystemsList.GetNextSelectedItem(pos);
|
|
CString sSystem = m_SystemsList.GetItemText(iIndex,0);
|
|
if( ! IsInSystemArray(sSystem) )
|
|
{
|
|
m_sSystems += sSystem + _T(" ; ");
|
|
}
|
|
}
|
|
|
|
m_sSystems.TrimRight(_T(" ; "));
|
|
|
|
UpdateData(FALSE);
|
|
|
|
GetDlgItem(IDC_BUTTON_ADD)->EnableWindow(FALSE);
|
|
}
|
|
|
|
void CSelectSystemsDlg::OnButtonCheckNames()
|
|
{
|
|
GetDlgItem(IDC_BUTTON_ADD)->EnableWindow(FALSE);
|
|
|
|
CompileArrayOfSystems();
|
|
|
|
CheckSystemNames();
|
|
}
|
|
|
|
void CSelectSystemsDlg::OnClickListSystems(NMHDR* pNMHDR, LRESULT* pResult)
|
|
{
|
|
if( m_SystemsList.GetItemCount() )
|
|
{
|
|
GetDlgItem(IDC_BUTTON_ADD)->EnableWindow(TRUE);
|
|
}
|
|
|
|
|
|
*pResult = 0;
|
|
}
|
|
|
|
void CSelectSystemsDlg::OnDblclkListSystems(NMHDR* pNMHDR, LRESULT* pResult)
|
|
{
|
|
OnButtonAdd();
|
|
|
|
*pResult = 0;
|
|
}
|
|
|
|
void CSelectSystemsDlg::OnOK()
|
|
{
|
|
CWaitCursor wait;
|
|
|
|
CompileArrayOfSystems();
|
|
|
|
if( m_saSystems.GetSize() == 0 )
|
|
{
|
|
return;
|
|
}
|
|
|
|
if( ! CheckSystemNames() )
|
|
{
|
|
return;
|
|
}
|
|
|
|
CResizeableDialog::OnOK();
|
|
}
|
|
|
|
void CSelectSystemsDlg::OnCancel()
|
|
{
|
|
m_saSystems.RemoveAll();
|
|
|
|
CResizeableDialog::OnCancel();
|
|
}
|
|
|
|
void CSelectSystemsDlg::OnButtonHelp()
|
|
{
|
|
MMCPropertyHelp(_T("HMon21.chm::/dnewsys.htm")); // 62212
|
|
|
|
}
|
|
|
|
void CSelectSystemsDlg::OnDropdownComboDomain()
|
|
{
|
|
if( m_Domains.GetCount() <= 1 )
|
|
{
|
|
AfxBeginThread((AFX_THREADPROC)AddDomains,new HWND(GetSafeHwnd()));
|
|
}
|
|
|
|
GetDlgItem(IDC_BUTTON_ADD)->EnableWindow(FALSE);
|
|
}
|
|
|
|
void CSelectSystemsDlg::OnCloseupComboDomain()
|
|
{
|
|
// TODO: Add your control notification handler code here
|
|
|
|
}
|
|
|
|
void CSelectSystemsDlg::OnSelendokComboDomain()
|
|
{
|
|
if( m_SystemsList.IsWindowEnabled() )
|
|
{
|
|
AfxBeginThread((AFX_THREADPROC)AddSystems,new HWND(GetSafeHwnd()));
|
|
GetDlgItem(IDC_BUTTON_ADD)->EnableWindow(FALSE);
|
|
}
|
|
}
|
|
|
|
void CSelectSystemsDlg::OnChangeEditSystems()
|
|
{
|
|
// TODO: If this is a RICHEDIT control, the control will not
|
|
// send this notification unless you override the CResizeableDialog::OnInitDialog()
|
|
// function and call CRichEditCtrl().SetEventMask()
|
|
// with the ENM_CHANGE flag ORed into the mask.
|
|
GetDlgItem(IDC_BUTTON_ADD)->EnableWindow(FALSE);
|
|
UpdateData();
|
|
}
|
|
|
|
|
|
void CSelectSystemsDlg::OnSetfocusListSystems(NMHDR* pNMHDR, LRESULT* pResult)
|
|
{
|
|
if( m_SystemsList.GetItemCount() )
|
|
{
|
|
GetDlgItem(IDC_BUTTON_ADD)->EnableWindow(TRUE);
|
|
}
|
|
*pResult = 0;
|
|
}
|
|
|
|
void CSelectSystemsDlg::OnKillfocusListSystems(NMHDR* pNMHDR, LRESULT* pResult)
|
|
{
|
|
|
|
*pResult = 0;
|
|
}
|
|
|
|
|