|
|
// DhcpRouteArrayEditor.cpp : implementation file
//
#include "stdafx.h"
#include "RouteArrayEditor.h"
#include "optcfg.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;#endif
/////////////////////////////////////////////////////////////////////////////
// CDhcpRouteArrayEditor dialog
CDhcpRouteArrayEditor::CDhcpRouteArrayEditor( CDhcpOption * pdhcType, DHCP_OPTION_SCOPE_TYPE dhcScopeType, CWnd *pParent ) : CBaseDialog(CDhcpRouteArrayEditor::IDD, pParent), m_p_type( pdhcType ), m_option_type( dhcScopeType ){ //{{AFX_DATA_INIT(CDhcpRouteArrayEditor)
//}}AFX_DATA_INIT
}
void CDhcpRouteArrayEditor::DoDataExchange(CDataExchange* pDX){ CDialog::DoDataExchange(pDX); //{{AFX_DATA_MAP(CDhcpRouteArrayEditor)
DDX_Control(pDX, IDC_STATIC_OPTION_NAME, m_st_option); DDX_Control(pDX, IDC_STATIC_APPLICATION, m_st_application); DDX_Control(pDX, IDC_LIST_OF_ROUTES, m_lc_routes); DDX_Control(pDX, IDC_BUTN_ROUTE_DELETE, m_butn_route_delete); DDX_Control(pDX, IDC_BUTN_ROUTE_ADD, m_butn_route_add); //}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CDhcpRouteArrayEditor, CBaseDialog) //{{AFX_MSG_MAP(CDhcpRouteArrayEditor)
ON_BN_CLICKED(IDC_BUTN_ROUTE_ADD, OnButnRouteAdd) ON_BN_CLICKED(IDC_BUTN_ROUTE_DELETE, OnButnRouteDelete) //}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CDhcpRouteArrayEditor message handlers
void CDhcpRouteArrayEditor::OnButnRouteAdd() { CAddRoute NewRoute( NULL ); NewRoute.DoModal();
if ( NewRoute.m_bChange ) { const int IP_ADDR_LEN = 20; WCHAR strDest[ IP_ADDR_LEN ]; WCHAR strMask[ IP_ADDR_LEN ]; WCHAR strRouter[ IP_ADDR_LEN ]; // obtain the three strings..
::UtilCvtIpAddrToWstr( NewRoute.Dest, strDest, IP_ADDR_LEN ); ::UtilCvtIpAddrToWstr( NewRoute.Mask, strMask, IP_ADDR_LEN ); ::UtilCvtIpAddrToWstr( NewRoute.Router, strRouter, IP_ADDR_LEN ); LV_ITEM lvi; lvi.mask = LVIF_TEXT; lvi.iItem = m_lc_routes.GetItemCount(); lvi.iSubItem = 0; lvi.pszText = ( LPTSTR )( LPCTSTR ) strDest; lvi.iImage = 0; lvi.stateMask = 0;
int nItem = m_lc_routes.InsertItem( &lvi ); m_lc_routes.SetItemText( nItem, 1, strMask ); m_lc_routes.SetItemText( nItem, 2, strRouter ); // unselect others
for ( int i = 0; i < m_lc_routes.GetItemCount(); i++ ) { m_lc_routes.SetItemState( i, 0, LVIS_SELECTED ); } // for
// Make this route as selected item
m_lc_routes.SetItemState( nItem, LVIS_SELECTED, LVIS_SELECTED ); HandleActivation(); } // if route added
} // CDhcpRouteArrayEditor::OnButnRouteAdd()
void CDhcpRouteArrayEditor::OnButnRouteDelete() { int nItem = m_lc_routes.GetNextItem( -1, LVNI_SELECTED ); int nDelItem = 0;
while ( nItem != -1 ) { m_lc_routes.DeleteItem( nItem ); nDelItem = nItem;
// Make the next one or the last one as selected
nItem = m_lc_routes.GetNextItem( -1, LVNI_SELECTED ); } // while
// Select an item
int items = m_lc_routes.GetItemCount(); if ( items > 0 ) { if ( nDelItem >= items ) { nDelItem = items - 1; } m_lc_routes.SetItemState( nDelItem, LVIS_SELECTED, LVIS_SELECTED ); } // if
HandleActivation();
} // CDhcpRouteArrayEditor::OnButnRouteDelete()
void CDhcpRouteArrayEditor::OnCancel() { CBaseDialog::OnCancel();}
void CDhcpRouteArrayEditor::OnOK() { DWORD err = 0; int nItems, BufSize; CListCtrl *pList; LPBYTE Buffer;
// The following code is borrowed from
// CDhcpOptCfgPropPage::HandleActivationRouteArray()
nItems = m_lc_routes.GetItemCount(); Buffer = new BYTE[ sizeof( DWORD ) * 4 * nItems ]; if ( NULL == Buffer ) { return; }
BufSize = 0; for ( int i = 0; i < nItems; i++ ) { DHCP_IP_ADDRESS Dest, Mask, Router;
Dest = UtilCvtWstrToIpAddr( m_lc_routes.GetItemText( i, 0 )); Mask = UtilCvtWstrToIpAddr( m_lc_routes.GetItemText( i, 1 )); Router = UtilCvtWstrToIpAddr( m_lc_routes.GetItemText( i, 2 ));
Dest = htonl( Dest ); Router = htonl( Router );
int nBitsInMask = 0; while ( Mask != 0 ) { nBitsInMask++; Mask <<= 1; }
// first add destination descriptor
// first byte contains # of bits in mask
// next few bytes contain the dest address for only
// the significant octets
Buffer[ BufSize++ ] = ( BYTE ) nBitsInMask; memcpy( &Buffer[ BufSize ], &Dest, ( nBitsInMask + 7 ) / 8 ); BufSize += ( nBitsInMask + 7 ) / 8; // now just copy the router address
memcpy(& Buffer[ BufSize ], &Router, sizeof( Router )); BufSize += sizeof( Router ); } // for
// Now write back the option value
DHCP_OPTION_DATA_ELEMENT DataElement = { DhcpBinaryDataOption }; DHCP_OPTION_DATA Data = { 1, &DataElement }; DataElement.Element.BinaryDataOption.DataLength = BufSize; DataElement.Element.BinaryDataOption.Data = Buffer;
err = m_p_type->QueryValue().SetData( &Data );
delete[] Buffer;
m_p_type->SetDirty();
if ( err ) { ::DhcpMessageBox( err ); OnCancel(); } else { CBaseDialog::OnOK(); }
} // CDhcpRouteArrayEditor::OnOK()
//
// strings and widths used for the list control
//
const int ROUTE_LIST_COL_HEADERS[3] = { IDS_ROUTE_LIST_COL_DEST, IDS_ROUTE_LIST_COL_MASK, IDS_ROUTE_LIST_COL_ROUTER};
const int ROUTE_COLS = sizeof( ROUTE_LIST_COL_HEADERS ) / sizeof( ROUTE_LIST_COL_HEADERS[ 0 ]);
BOOL CDhcpRouteArrayEditor::OnInitDialog() { CString strColHeader; RECT rect; LONG width;
CDhcpOptionValue &optValue = m_p_type->QueryValue(); // make sure the option type is set as Binary data type
ASSERT( DhcpBinaryDataOption == optValue.QueryDataType());
CBaseDialog::OnInitDialog(); DWORD err = 0;
int cStrId = ( m_option_type == DhcpDefaultOptions) ? IDS_INFO_TITLE_DEFAULT_OPTIONS : (( m_option_type == DhcpGlobalOptions ) ? IDS_INFO_TITLE_GLOBAL_OPTIONS : IDS_INFO_TITLE_SCOPE_OPTIONS );
//
// setup the columns in the list control
//
m_lc_routes.GetClientRect( &rect ); width = ( rect.right - rect.left ) / 3; for ( int i = 0; i < ROUTE_COLS; i++ ) { strColHeader.LoadString( ROUTE_LIST_COL_HEADERS[ i ] ); m_lc_routes.InsertColumn( i, strColHeader, LVCFMT_LEFT, width, -1 ); } // for
// Select a full row
m_lc_routes.SetExtendedStyle( m_lc_routes.GetExtendedStyle() | LVS_EX_FULLROWSELECT );
const CByteArray *pbaData = optValue.QueryBinaryArray(); ASSERT( pbaData != NULL );
int nDataSize = ( int ) pbaData->GetSize(); LPBYTE pData = ( LPBYTE ) pbaData->GetData();
//
// The following loop is copied from optcfg.cpp,
// COptionsCfgPropPage::FillDataEntry()
//
while ( nDataSize > sizeof( DWORD )) { // first 1 byte contains the # of bits in subnetmask
nDataSize --; BYTE nBitsMask = *pData ++; DWORD Mask = (~0); if( nBitsMask < 32 ) Mask <<= (32-nBitsMask); // based on the # of bits, the next few bytes contain
// the subnet address for the 1-bits of subnet mask
int nBytesDest = (nBitsMask+7)/8; if( nBytesDest > 4 ) nBytesDest = 4; DWORD Dest = 0; memcpy( &Dest, pData, nBytesDest ); pData += nBytesDest; nDataSize -= nBytesDest; // subnet address is obviously in network order.
Dest = ntohl(Dest); // now the four bytes would be the router address
DWORD Router = 0; if( nDataSize < sizeof(DWORD) ) { Assert( FALSE ); break; } memcpy(&Router, pData, sizeof(DWORD)); Router = ntohl( Router ); pData += sizeof(DWORD); nDataSize -= sizeof(DWORD); // now fill the list box..
const int IP_ADDR_LEN = 20; WCHAR strDest[ IP_ADDR_LEN ]; WCHAR strMask[ IP_ADDR_LEN ]; WCHAR strRouter[ IP_ADDR_LEN ]; ::UtilCvtIpAddrToWstr( Dest, strDest, IP_ADDR_LEN ); ::UtilCvtIpAddrToWstr( Mask, strMask, IP_ADDR_LEN ); ::UtilCvtIpAddrToWstr( Router, strRouter, IP_ADDR_LEN ); LV_ITEM lvi; lvi.mask = LVIF_TEXT; lvi.iItem = m_lc_routes.GetItemCount(); lvi.iSubItem = 0; lvi.pszText = ( LPTSTR )( LPCTSTR ) strDest; lvi.iImage = 0; lvi.stateMask = 0; int nItem = m_lc_routes.InsertItem( &lvi ); m_lc_routes.SetItemText( nItem, 1, strMask ); m_lc_routes.SetItemText( nItem, 2, strRouter ); } // while
// set the first item as selected if any is added.
if ( m_lc_routes.GetItemCount() > 0 ) { m_lc_routes.SetItemState( 0, LVIS_SELECTED, LVIS_SELECTED ); }
CATCH_MEM_EXCEPTION { CString str; m_st_option.SetWindowText( m_p_type->QueryName()); str.LoadString( cStrId ); m_st_application.SetWindowText( str );
// Set proper button states.
HandleActivation(); } END_MEM_EXCEPTION( err ); if ( err ) { ::DhcpMessageBox( err ); EndDialog( -1 ); }
return TRUE; // return TRUE unless you set the focus to a control
// EXCEPTION: OCX Property Pages should return FALSE
} // CDhcpRouteArrayEditor::OnInitDialog()
void CDhcpRouteArrayEditor::HandleActivation(){
int cItems = m_lc_routes.GetItemCount();
// set the focus to Add button
m_butn_route_add.SetFocus();
m_butn_route_delete.EnableWindow( 0 != cItems );
UpdateData( FALSE );} // CDhcpRouteArrayEditor::HandleActivation()
|
