OS_Develop/Windows-Server-2003
OS_Develop/Windows-Server-2003
1
0
Fork 0
Code Pull Requests Activity
7507d3c0e1
Windows-Server-2003/net/mmc/ipsecmon/srchmflt.cpp

525 lines
13 KiB
C++
Raw Normal View History

Initiall commit
2024-08-04 01:28:15 +02:00
/**********************************************************************/
/** Microsoft Windows/NT **/
/** Copyright(c) Microsoft Corporation, 1997 - 2002 **/
/**********************************************************************/
/*
edituser.h
Edit user dialog implementation file
FILE HISTORY:
*/
#include "stdafx.h"
#include "mdlsdlg.h"
#include "SrchMFlt.h"
#include "spdutil.h"
#include "ncglobal.h" // network console global defines
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
/////////////////////////////////////////////////////////////////////////////
// CSearchMMFilters dialog
CSearchMMFilters::CSearchMMFilters(ISpdInfo * pSpdInfo)
: CModelessDlg()
{
//{{AFX_DATA_INIT(CSearchMMFilters)
// NOTE: the ClassWizard will add member initialization here
//}}AFX_DATA_INIT
m_spSpdInfo.Set(pSpdInfo);
}
void CSearchMMFilters::DoDataExchange(CDataExchange* pDX)
{
CBaseDialog::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CSearchMMFilters)
DDX_Control(pDX, IDC_MM_SRCH_LIST, m_listResult);
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CSearchMMFilters, CBaseDialog)
//{{AFX_MSG_MAP(CSearchMMFilters)
ON_BN_CLICKED(IDC_MM_SEARCH, OnButtonSearch)
ON_BN_CLICKED(IDC_MM_SRCH_SRC_ANY, OnSrcOptionClicked)
ON_BN_CLICKED(IDC_MM_SRCH_SRC_SPEC, OnSrcOptionClicked)
ON_BN_CLICKED(IDC_MM_SRCH_DEST_ANY, OnDestOptionClicked)
ON_BN_CLICKED(IDC_MM_SRCH_SRC_ME, OnSrcMeOptionClicked)
ON_BN_CLICKED(IDC_MM_SRCH_DST_ME, OnDstMeOptionClicked)
ON_BN_CLICKED(IDC_MM_SRCH_DEST_SPEC, OnDestOptionClicked)
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
//To manually create the IP control and disable mirroring if the parent dialog is mirrored
//
//Arguments:
// uID [IN] the control that the new IP control should overwrite
// uIDIpCtr [IN] the ID of the IP control to create
//
//Note: $REVIEW (nsun) this should be removed after the snapin is themed since IP controls
// in comctl v6 will handle the mirroring by itself
//
HWND CSearchMMFilters::CreateIPControl(UINT uID, UINT uIDIpCtl)
{
HWND hwndIPControl = NULL;
RECT rcClient; // client area of parent window
CWnd* pWnd = GetDlgItem(uID);
if (pWnd)
{
// get pos info from our template static and then make sure it is hidden
pWnd->GetWindowRect(&rcClient);
pWnd->ShowWindow (SW_HIDE);
ScreenToClient (&rcClient);
//$REVIEW WS_EX_NOINHERITLAYOUT is to fix the mirroring problem of IP control
//See WinXP bug 261926. We should remove that we switch the comctl32 v6
LONG lExStyles = 0;
LONG lExStyles0 = 0;
if (m_hWnd)
{
lExStyles0 = lExStyles = GetWindowLong(m_hWnd, GWL_EXSTYLE);
if (lExStyles & WS_EX_LAYOUTRTL)
{
lExStyles |= WS_EX_NOINHERITLAYOUT;
SetWindowLong(m_hWnd, GWL_EXSTYLE, lExStyles);
}
}
// create the new edit control
hwndIPControl = ::CreateWindowEx(WS_EX_NOINHERITLAYOUT, WC_IPADDRESS, NULL, WS_CHILD | WS_VISIBLE | WS_TABSTOP | WS_BORDER,
rcClient.left,
rcClient.top,
rcClient.right - rcClient.left,
rcClient.bottom - rcClient.top,
GetSafeHwnd(),
(HMENU) IntToPtr(uIDIpCtl),
AfxGetInstanceHandle (), //g_hinst,
NULL);
if (lExStyles0 != lExStyles && m_hWnd)
{
SetWindowLong(m_hWnd, GWL_EXSTYLE, lExStyles0);
}
// move the control directly behind the pWnd in the Z order
if (hwndIPControl)
{
::SetWindowPos (hwndIPControl, pWnd->GetSafeHwnd(), 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE);
}
}
return hwndIPControl;
}
/////////////////////////////////////////////////////////////////////////////
// CSearchMMFilters message handlers
BOOL CSearchMMFilters::OnInitDialog()
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
HWND hIpCtrl = CreateIPControl(IDC_MM_SRCH_SRC_IP_TEXT, IDC_MM_SRCH_SRC_IP);
m_ipSrc.Create(hIpCtrl);
hIpCtrl = CreateIPControl(IDC_MM_SRCH_DEST_IP_TEXT, IDC_MM_SRCH_DEST_IP);
m_ipDest.Create(hIpCtrl);
m_ipSrc.SetFieldRange(0, 1, 223);
m_ipDest.SetFieldRange(0, 1, 223);
CBaseDialog::OnInitDialog();
CString st;
CheckDlgButton(IDC_MM_SRCH_SRC_ANY, BST_CHECKED);
OnSrcOptionClicked();
CheckDlgButton(IDC_MM_SRCH_DEST_ANY, BST_CHECKED);
OnDestOptionClicked();
AddIpAddrsToCombo();
//disable the two combo boxes
((CWnd*)GetDlgItem(IDC_MM_SRCH_SRC_ME_COMBO))->EnableWindow(FALSE);
((CWnd*)GetDlgItem(IDC_MM_SRCH_DST_ME_COMBO))->EnableWindow(FALSE);
CheckDlgButton(IDC_MM_SRCH_INBOUND, BST_CHECKED);
CheckDlgButton(IDC_MM_SRCH_OUTBOUND, BST_CHECKED);
CheckDlgButton(IDC_MM_SRCH_RADIO_BEST, BST_CHECKED);
int nWidth;
nWidth = m_listResult.GetStringWidth(_T("555.555.555.555 - "));
st.LoadString(IDS_COL_FLTR_SRC);
m_listResult.InsertColumn(0, st, LVCFMT_LEFT, nWidth);
nWidth = m_listResult.GetStringWidth(_T("555.555.555.555 - "));
st.LoadString(IDS_COL_FLTR_DEST);
m_listResult.InsertColumn(1, st, LVCFMT_LEFT, nWidth);
st.LoadString(IDS_FLTR_DIR_OUT);
nWidth = m_listResult.GetStringWidth((LPCTSTR)st) + 20;
st.LoadString(IDS_FILTER_PP_COL_DIRECTION);
m_listResult.InsertColumn(2, st, LVCFMT_LEFT, nWidth);
st.LoadString(IDS_FILTER_PP_COL_WEIGHT);
nWidth = m_listResult.GetStringWidth((LPCTSTR)st) + 20;
m_listResult.InsertColumn(3, st, LVCFMT_LEFT, nWidth);
st.LoadString(IDS_FLTER_PP_COL_IKE_POL);
nWidth = m_listResult.GetStringWidth((LPCTSTR)st) + 20;
m_listResult.InsertColumn(4, st, LVCFMT_LEFT, nWidth);
st.LoadString(IDS_FLTER_PP_COL_MM_AUTH);
nWidth = m_listResult.GetStringWidth((LPCTSTR)st) + 20;
m_listResult.InsertColumn(5, st, LVCFMT_LEFT, nWidth);
return TRUE; // return TRUE unless you set the focus to a control
// EXCEPTION: OCX Property Pages should return FALSE
}
void CSearchMMFilters::AddIpAddrsToCombo()
{
PMIB_IPADDRTABLE pIpTable;
ULONG dwSize=0;
ULONG index;
DWORD dwRet;
CString strIP;
dwRet = GetIpAddrTable(
NULL, // buffer for mapping table
&dwSize, // size of buffer
FALSE // sort the table
);
if( ERROR_INSUFFICIENT_BUFFER != dwRet && ERROR_SUCCESS != dwRet ) {
return;
} else {
pIpTable = (PMIB_IPADDRTABLE) LocalAlloc(LMEM_ZEROINIT,dwSize);
}
dwRet = GetIpAddrTable(
pIpTable, // buffer for mapping table
&dwSize, // size of buffer
FALSE // sort the table
);
if(ERROR_SUCCESS != dwRet) {
if(pIpTable)
LocalFree(pIpTable);
return;
}
CComboBox* pComboSrc = (CComboBox*) GetDlgItem(IDC_MM_SRCH_SRC_ME_COMBO);
CComboBox* pComboDst = (CComboBox*) GetDlgItem(IDC_MM_SRCH_DST_ME_COMBO);
for(index=0; index<pIpTable->dwNumEntries; index++) {
strIP.Format(_T("%d.%d.%d.%d"),GET_SOCKADDR(pIpTable->table[index].dwAddr));
if(lstrcmp(strIP, _T("127.0.0.1"))) {
pComboSrc->AddString(strIP);
pComboDst->AddString(strIP);
}
}
if(pIpTable)
LocalFree(pIpTable);
}
void CSearchMMFilters::OnButtonSearch()
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
CMmFilterInfo fltr;
CMmFilterInfoArray arrMatchFltrs;
if (!LoadConditionInfoFromControls(&fltr))
return;
DWORD dwNum = 1000; //TODO BUGBUG, should change to 0 to mean search all matches
if(IsDlgButtonChecked(IDC_MM_SRCH_RADIO_BEST))
{
dwNum = 1;
}
m_spSpdInfo->GetMatchMMFilters(&fltr, dwNum, &arrMatchFltrs);
PopulateFilterListToControl(&arrMatchFltrs);
FreeItemsAndEmptyArray(arrMatchFltrs);
}
BOOL CSearchMMFilters::LoadConditionInfoFromControls(CMmFilterInfo * pFltr)
{
CString st;
if (IsDlgButtonChecked(IDC_MM_SRCH_SRC_ANY))
{
pFltr->m_SrcAddr.AddrType = IP_ADDR_SUBNET;
pFltr->m_SrcAddr.uIpAddr = 0;
pFltr->m_SrcAddr.uSubNetMask = 0;
}
else if (IsDlgButtonChecked(IDC_MM_SRCH_SRC_ME))
{
USES_CONVERSION;
pFltr->m_SrcAddr.AddrType = IP_ADDR_UNIQUE;
CComboBox* pCombo = (CComboBox*) GetDlgItem(IDC_MM_SRCH_SRC_ME_COMBO);
INT nSelected = pCombo->GetCurSel();
if( CB_ERR != nSelected)
{
pCombo->GetLBText(nSelected, st);
try
{
pFltr->m_SrcAddr.uIpAddr = inet_addr(T2A((LPCTSTR)st));
}
catch(...)
{
AfxMessageBox(IDS_ERR_OUTOFMEMORY);
return FALSE;
}
pFltr->m_SrcAddr.uSubNetMask = 0xFFFFFFFF;
}
}
else
{
USES_CONVERSION;
pFltr->m_SrcAddr.AddrType = IP_ADDR_UNIQUE;
m_ipSrc.GetAddress(st);
try
{
pFltr->m_SrcAddr.uIpAddr = inet_addr(T2A((LPCTSTR)st));
}
catch(...)
{
AfxMessageBox(IDS_ERR_OUTOFMEMORY);
return FALSE;
}
pFltr->m_SrcAddr.uSubNetMask = 0xFFFFFFFF;
}
if (IsDlgButtonChecked(IDC_MM_SRCH_DEST_ANY))
{
pFltr->m_DesAddr.AddrType = IP_ADDR_SUBNET;
pFltr->m_DesAddr.uIpAddr = 0;
pFltr->m_DesAddr.uSubNetMask = 0;
}
else if (IsDlgButtonChecked(IDC_MM_SRCH_DST_ME))
{
USES_CONVERSION;
pFltr->m_DesAddr.AddrType = IP_ADDR_UNIQUE;
CComboBox* pCombo = (CComboBox*) GetDlgItem(IDC_MM_SRCH_DST_ME_COMBO);
INT nSelected = pCombo->GetCurSel();
if( CB_ERR != nSelected)
{
pCombo->GetLBText(nSelected, st);
try
{
pFltr->m_DesAddr.uIpAddr = inet_addr(T2A((LPCTSTR)st));
}
catch(...)
{
AfxMessageBox(IDS_ERR_OUTOFMEMORY);
return FALSE;
}
pFltr->m_DesAddr.uSubNetMask = 0xFFFFFFFF;
}
}
else
{
USES_CONVERSION;
pFltr->m_DesAddr.AddrType = IP_ADDR_UNIQUE;
m_ipDest.GetAddress(st);
try
{
pFltr->m_DesAddr.uIpAddr = inet_addr(T2A((LPCTSTR)st));
}
catch(...)
{
AfxMessageBox(IDS_ERR_OUTOFMEMORY);
return FALSE;
}
pFltr->m_DesAddr.uSubNetMask = 0xFFFFFFFF;
}
if (IsDlgButtonChecked(IDC_MM_SRCH_INBOUND))
{
//if both inbound and outbound are chosen, then
//set the driection valude as 0
if (IsDlgButtonChecked(IDC_MM_SRCH_OUTBOUND))
{
pFltr->m_dwDirection = 0;
}
else
{
pFltr->m_dwDirection = FILTER_DIRECTION_INBOUND;
}
}
else if (IsDlgButtonChecked(IDC_MM_SRCH_OUTBOUND))
{
pFltr->m_dwDirection = FILTER_DIRECTION_OUTBOUND;
}
else
{
::AfxMessageBox(IDS_ERR_NO_DIRECTION);
return FALSE;
}
return TRUE;
}
void CSearchMMFilters::PopulateFilterListToControl(CMmFilterInfoArray * parrFltrs)
{
CString st;
m_listResult.DeleteAllItems();
int nRows = -1;
for (int i = 0; i < parrFltrs->GetSize(); i++)
{
nRows++;
nRows = m_listResult.InsertItem(nRows, _T(""));
if (-1 != nRows)
{
AddressToString((*parrFltrs)[i]->m_SrcAddr, &st);
m_listResult.SetItemText(nRows, 0, st);
AddressToString((*parrFltrs)[i]->m_DesAddr, &st);
m_listResult.SetItemText(nRows, 1, st);
DirectionToString((*parrFltrs)[i]->m_dwDirection, &st);
m_listResult.SetItemText(nRows, 2, st);
st.Format(_T("%d"), (*parrFltrs)[i]->m_dwWeight);
m_listResult.SetItemText(nRows, 3, st);
st = (*parrFltrs)[i]->m_stPolicyName;
if( (*parrFltrs)[i]->m_dwFlags & IPSEC_MM_POLICY_DEFAULT_POLICY )
{
AfxFormatString1(st, IDS_POL_DEFAULT, (LPCTSTR) (*parrFltrs)[i]->m_stPolicyName);
}
m_listResult.SetItemText(nRows, 4, st);
m_listResult.SetItemText(nRows, 5, (*parrFltrs)[i]->m_stAuthDescription);
m_listResult.SetItemData(nRows, i);
}
}
if ( 0 == parrFltrs->GetSize() )
{
AfxMessageBox(IDS_ERROR_NOMATCH_FILTER);
}
}
void CSearchMMFilters::OnSrcOptionClicked()
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
BOOL fAny = IsDlgButtonChecked(IDC_MM_SRCH_SRC_ANY);
((CWnd*)GetDlgItem(IDC_MM_SRCH_SRC_ME_COMBO))->EnableWindow(FALSE);
if (fAny)
{
m_ipSrc.ClearAddress();
}
if (m_ipSrc.m_hIPaddr)
{
::EnableWindow(m_ipSrc.m_hIPaddr, !fAny);
}
CComboBox* pCombo = (CComboBox*) GetDlgItem(IDC_MM_SRCH_SRC_ME_COMBO);
pCombo->SetCurSel(-1);
}
void CSearchMMFilters::OnDestOptionClicked()
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
BOOL fAny = IsDlgButtonChecked(IDC_MM_SRCH_DEST_ANY);
((CWnd*)GetDlgItem(IDC_MM_SRCH_DST_ME_COMBO))->EnableWindow(FALSE);
if (fAny)
{
m_ipDest.ClearAddress();
}
if (m_ipDest.m_hIPaddr)
{
::EnableWindow(m_ipDest.m_hIPaddr, !fAny);
}
CComboBox* pCombo = (CComboBox*) GetDlgItem(IDC_MM_SRCH_DST_ME_COMBO);
pCombo->SetCurSel(-1);
}
void CSearchMMFilters::OnSrcMeOptionClicked()
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
((CWnd*)GetDlgItem(IDC_MM_SRCH_SRC_ME_COMBO))->EnableWindow(TRUE);
::EnableWindow(m_ipSrc.m_hIPaddr, FALSE);
CComboBox* pCombo = (CComboBox*) GetDlgItem(IDC_MM_SRCH_SRC_ME_COMBO);
int nCount = pCombo->GetCount();
if (nCount > 0) {
pCombo->SetCurSel(0);
}
}
void CSearchMMFilters::OnDstMeOptionClicked()
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
((CWnd*)GetDlgItem(IDC_MM_SRCH_DST_ME_COMBO))->EnableWindow(TRUE);
::EnableWindow(m_ipDest.m_hIPaddr, FALSE);
CComboBox* pCombo = (CComboBox*) GetDlgItem(IDC_MM_SRCH_DST_ME_COMBO);
int nCount = pCombo->GetCount();
if (nCount > 0) {
pCombo->SetCurSel(0);
}
}
void CSearchMMFilters::OnOK()
{
//Since this is a modelless dialog, need to call DestroyWindow
DestroyWindow();
}
Copy Permalink