inline BOOL CreateDirectory(const WTL::CString& strDir)
{
BOOL retval = FALSE;
WTL::CString strTemp = strDir;
WTL::CString strBase;
int nFind;
if (GetFileAttributes(strDir) == FILE_ATTRIBUTE_DIRECTORY)
{
retval = TRUE;
goto clean0;
}
if (strTemp[strTemp.GetLength() - 1] != _T('\\'))
strTemp += _T("\\");
nFind = strTemp.Find(_T("\\"));
while (nFind != -1)
{
strBase += strTemp.Left(nFind + 1);
strTemp.Delete(0, nFind + 1);
if (GetFileAttributes(strBase) == INVALID_FILE_ATTRIBUTES)
{
if (!CreateDirectory(strBase, NULL))
goto clean0;
}
nFind = strTemp.Find(_T("\\"));
}
retval = TRUE;
clean0:
return retval;
}
LRESULT CMainFrame::OnCreate(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/)
{
#ifndef BTNS_SHOWTEXT
#define BTNS_SHOWTEXT 0x0040
#endif
// set title
WTL::CString strTitle;
strTitle.LoadString( IDS_APPLICATION );
SetWindowText(strTitle);
CreateSimpleStatusBar();
HWND hWndToolBar =
CreateSimpleToolBarCtrl(
m_hWnd, IDR_MAINFRAME, FALSE,
ATL_SIMPLE_TOOLBAR_STYLE | BTNS_SHOWTEXT | TBSTYLE_LIST | TBSTYLE_FLAT | TBSTYLE_FLAT);
TBBUTTON tbButton = { 0 };
TBBUTTONINFO tbButtonInfo = { 0 };
TBREPLACEBITMAP replaceBitmap = { 0 };
m_wndToolBar.Attach( hWndToolBar );
m_wndToolBar.SetExtendedStyle( TBSTYLE_EX_DRAWDDARROWS );
// Replace toolbar bitmap to display true color image
replaceBitmap.hInstNew = _Module.GetResourceInstance();
replaceBitmap.hInstOld = _Module.GetResourceInstance();
replaceBitmap.nIDOld = IDR_MAINFRAME;
replaceBitmap.nIDNew = IDB_MAINFRAME;
replaceBitmap.nButtons = 4;
m_wndToolBar.ReplaceBitmap( &replaceBitmap );
// Add strings to the toolbar
m_wndToolBar.SetButtonStructSize(sizeof(TBBUTTON));
for ( int i=0; i < m_wndToolBar.GetButtonCount(); i++ )
{
WTL::CString strCommand;
m_wndToolBar.GetButton( i, &tbButton );
tbButtonInfo.cbSize = sizeof(TBBUTTONINFO);
tbButtonInfo.dwMask = TBIF_STYLE;
m_wndToolBar.GetButtonInfo( tbButton.idCommand, &tbButtonInfo );
tbButtonInfo.dwMask = TBIF_TEXT | TBIF_STYLE;
strCommand.LoadString( tbButton.idCommand );
strCommand = strCommand.Right(
strCommand.GetLength() - strCommand.Find('\n') - 1
);
tbButtonInfo.pszText =
const_cast<LPTSTR>(static_cast<LPCTSTR>(strCommand));
tbButtonInfo.cchText = strCommand.GetLength();
tbButtonInfo.fsState |= BTNS_SHOWTEXT;
m_wndToolBar.AddString( tbButton.idCommand );
m_wndToolBar.SetButtonInfo( tbButton.idCommand, &tbButtonInfo );
}
// Modify mirror button as dropdown button
m_wndToolBar.GetButton(
m_wndToolBar.CommandToIndex(IDM_AGGR_MIRROR),
&tbButton
);
tbButtonInfo.cbSize = sizeof(TBBUTTONINFO);
tbButtonInfo.dwMask = TBIF_STYLE;
tbButtonInfo.fsStyle = tbButton.fsStyle | TBSTYLE_DROPDOWN;
m_wndToolBar.SetButtonInfo( IDM_AGGR_MIRROR, &tbButtonInfo );
CreateSimpleReBar(ATL_SIMPLE_REBAR_NOBORDER_STYLE);
AddSimpleReBarBand(m_wndToolBar);
UIAddToolBar(m_wndToolBar);
m_hWndClient = m_view.Create(
m_hWnd,
rcDefault,
NULL,
WS_CHILD | WS_VISIBLE | WS_CLIPSIBLINGS | WS_CLIPCHILDREN
| TVS_HASLINES | /* TVS_LINESATROOT | */ TVS_SHOWSELALWAYS,
WS_EX_CLIENTEDGE
);
UISetCheck(ID_VIEW_TOOLBAR, 1);
UISetCheck(ID_VIEW_STATUS_BAR, 1);
// TODO : It will be better if we display splash window while
// the treeview is initialized
m_bRefreshing = FALSE;
::InitializeCriticalSection(&m_csThreadRefreshStatus);
StartRefreshStatus();
m_hEventCallback =
::NdasRegisterEventCallback(pNdasEventProc,m_hWnd);
// register object for message filtering and idle updates
CMessageLoop* pLoop = _Module.GetMessageLoop();
ATLASSERT(pLoop != NULL);
pLoop->AddMessageFilter(this);
pLoop->AddIdleHandler(this);
// FIXME : We need to remember the window size
CRect rectResize;
GetClientRect( rectResize );
rectResize = CRect( rectResize.TopLeft(), CSize(500, 500) );
ClientToScreen( rectResize );
MoveWindow( rectResize );
CenterWindow();
return 0;
}
/*
void SetPaneWidths(int* arrWidths, int nPanes)
{
// find the size of the borders
int arrBorders[3];
m_status.GetBorders(arrBorders);
// calculate right edge of default pane (0)
arrWidths[0] += arrBorders[2];
for (int i = 1; i < nPanes; i++)
arrWidths[0] += arrWidths[i];
// calculate right edge of remaining panes (1 thru nPanes-1)
for (int j = 1; j < nPanes; j++)
arrWidths[j] += arrBorders[2] + arrWidths[j - 1];
// set the pane widths
m_status.SetParts(m_status.m_nPanes, arrWidths);
}
*/
LRESULT CMainFrame::OnCreate(LPCREATESTRUCT /*lParam*/)
{
//
// create command bar window
//
HWND hWndCmdBar = m_CmdBar.Create(
m_hWnd,
rcDefault,
NULL,
ATL_SIMPLE_CMDBAR_PANE_STYLE);
// attach menu
m_CmdBar.AttachMenu(GetMenu());
// load command bar images
m_CmdBar.SetImageSize(CSize(9,9));
// m_CmdBar.LoadImages(IDR_MAINFRAME);
// remove old menu
SetMenu(NULL);
// set title
WTL::CString strTitle;
strTitle.LoadString( IDS_APPLICATION );
SetWindowText(strTitle);
//
// setting up a tool bar
//
HWND hWndToolBar = CreateSimpleToolBarCtrl(
m_hWnd,
IDR_MAINFRAME,
FALSE,
ATL_SIMPLE_TOOLBAR_PANE_STYLE | TBSTYLE_LIST);
m_wndToolBar.Attach( hWndToolBar );
m_wndToolBar.SetExtendedStyle( TBSTYLE_EX_MIXEDBUTTONS | TBSTYLE_EX_DRAWDDARROWS );
//
// patria:
//
// Some bitmaps are distorted when used with TB_ADDBITMAP
// which is sent from CreateSimpleToolBarCtrl when the bitmap is not true color.
// This is the case with IO-DATA's tool bar image.
// As an workaround, we can directly create a image list directly
// and replace the image list of the tool bar, which corrects such misbehaviors.
//
{
CImageList imageList;
WORD wWidth = 32; // we are using 32 x 32 buttons
imageList.CreateFromImage(
IDR_MAINFRAME,
wWidth,
1,
CLR_DEFAULT,
IMAGE_BITMAP,
LR_CREATEDIBSECTION | LR_DEFAULTSIZE);
m_wndToolBar.SetImageList(imageList);
}
TBBUTTON tbButton = { 0 };
TBBUTTONINFO tbButtonInfo = { 0 };
TBREPLACEBITMAP replaceBitmap = { 0 };
// Add strings to the tool bar
m_wndToolBar.SetButtonStructSize(sizeof(TBBUTTON));
for ( int i=0; i < m_wndToolBar.GetButtonCount(); i++ )
{
WTL::CString strCommand;
m_wndToolBar.GetButton( i, &tbButton );
tbButtonInfo.cbSize = sizeof(TBBUTTONINFO);
tbButtonInfo.dwMask = TBIF_STYLE;
m_wndToolBar.GetButtonInfo( tbButton.idCommand, &tbButtonInfo );
tbButtonInfo.dwMask = TBIF_TEXT | TBIF_STYLE;
strCommand.LoadString( tbButton.idCommand );
strCommand = strCommand.Right(
strCommand.GetLength() - strCommand.Find('\n') - 1
);
tbButtonInfo.pszText =
const_cast<LPTSTR>(static_cast<LPCTSTR>(strCommand));
tbButtonInfo.cchText = strCommand.GetLength();
tbButtonInfo.fsStyle |= BTNS_SHOWTEXT | BTNS_AUTOSIZE;
m_wndToolBar.AddString( tbButton.idCommand );
m_wndToolBar.SetButtonInfo( tbButton.idCommand, &tbButtonInfo );
}
#define ATL_CUSTOM_REBAR_STYLE \
((ATL_SIMPLE_REBAR_STYLE & ~RBS_AUTOSIZE) | CCS_NODIVIDER)
//
// patria: reason to use ATL_CUSTOM_REBAR_STYLE
//
// ATL_SIMPLE_REBAR_STYLE (not a NO_BRODER style) has a problem
// with theme-enabled Windows XP,
// rendering some transparent lines above the rebar.
//
CreateSimpleReBar(ATL_CUSTOM_REBAR_STYLE);
AddSimpleReBarBand(hWndCmdBar);
AddSimpleReBarBand(m_wndToolBar.m_hWnd, NULL, TRUE);
CReBarCtrl reBar = m_hWndToolBar;
DWORD cBands = reBar.GetBandCount();
for (DWORD i = 0; i < cBands; ++i)
{
REBARBANDINFO rbi = {0};
rbi.cbSize = sizeof(REBARBANDINFO);
rbi.fMask = RBBIM_STYLE;
reBar.GetBandInfo(i, &rbi);
rbi.fStyle |= RBBS_NOGRIPPER;
reBar.SetBandInfo(i, &rbi);
}
// work on status bar, progress bar
CreateSimpleStatusBar();
RECT rectRefreshProgress;
::GetClientRect(m_hWndStatusBar, &rectRefreshProgress);
rectRefreshProgress.right = 300;
m_wndRefreshProgress.Create(m_hWndStatusBar, &rectRefreshProgress, NULL, WS_CHILD | WS_VISIBLE);
m_wndRefreshProgress.SetRange32(0, 100);
m_wndRefreshProgress.SetPos(50);
m_wndRefreshProgress.ShowWindow(SW_HIDE);
m_viewTreeList.Create(
*this, rcDefault, NULL,
WS_BORDER | WS_CHILD | WS_VISIBLE | WS_CLIPCHILDREN | WS_CLIPSIBLINGS);
m_viewTreeList.Initialize();
m_hWndClient = m_viewTreeList;
UIAddToolBar(m_wndToolBar);
UISetCheck(ID_VIEW_TOOLBAR, 1);
UISetCheck(ID_VIEW_STATUS_BAR, 1);
// TODO : It will be better if we display splash window while
// the treeview is initialized
PostMessage(WM_COMMAND, IDM_TOOL_REFRESH, 0);
m_hEventCallback =
::NdasRegisterEventCallback(pNdasEventProc,m_hWnd);
// register object for message filtering and idle updates
CMessageLoop* pLoop = _Module.GetMessageLoop();
ATLASSERT(pLoop != NULL);
pLoop->AddMessageFilter(this);
pLoop->AddIdleHandler(this);
// FIXME : We need to remember the window size
CRect rectResize;
GetClientRect( rectResize );
rectResize = CRect( rectResize.TopLeft(), CSize(500, 500) );
ClientToScreen( rectResize );
MoveWindow( rectResize );
CenterWindow();
return 0;
}
DWORD WINAPI CMainDlg::DetectVersion(LPVOID lpParameter)
{
CMainDlg *pThis = static_cast<CMainDlg *> (lpParameter);
WTL::CString result;
const wchar_t *p_versionstr;
const wchar_t *p_versiondat;
const wchar_t *p_term;
DWORD res;
int pos;
pThis->m_edit.AppendText(L"Detecting scan executable\r\n");
pThis->m_process = new Process(L"scan.exe /?");
res = pThis->m_process->Exec(result);
delete pThis->m_process;
pThis->m_process = NULL;
if (res) return -1;
pos = result.Find(L"Scan engine v");
if (pos == -1) /* V2 */
{
p_versionstr = L"McAfee VirusScan Command Line for Win32 Version: ";
p_versiondat = L"Dat set version: ";
p_term = L"\r";
pThis->m_process = new Process(L"scan.exe /version");
res = pThis->m_process->Exec(result);
delete pThis->m_process;
pThis->m_process = NULL;
if (res) return -1;
pos = result.Find(p_versionstr);
if (pos == -1)
{
pThis->m_edit.AppendText(L"Unknown Version\r\n");
return -1;
}
pThis->m_toolversion = 2;
}
else /* V1 */
{
p_versionstr = L"Scan engine v";
p_versiondat = L"Virus data file v";
p_term = L" ";
pThis->m_toolversion = 1;
}
pos += wcslen(p_versionstr);
WTL::CString verstr = result.Mid(pos);
pos = verstr.Find(p_term);
pThis->m_version = verstr.Left(pos);
pos = result.Find(p_versiondat);
if (pos == -1)
{
pThis->m_edit.AppendText(L"Cannot parse dat version\r\n");
return -1;
}
pos += wcslen(p_versiondat);
result = result.Mid(pos);
pThis->m_datversion = _wtoi(result.GetBuffer(0));
WTL::CString message;
message.Format(L"Tool version %s - Dat Version %d\r\n", pThis->m_version, pThis->m_datversion);
pThis->m_edit.AppendText(message);
return 0;
}