mainframe.cpp

著名的防火墙源代码

// MainFrame.cpp : implementation file
//

#include "stdafx.h"
#include "..\Property.h"
#include "MainFrame.h"
#include "..\MainSheet.h"
#include "..\..\common\debug.h"
#include "..\GuiRes.h"
#include "..\MainSheet.h"
#include "..\PacketMonitor.h"
#include "..\LogQuery.h"
#include "..\About.h"
#include "..\Acl.h"
#include "..\SystemSet.h"
#include "..\..\common\xlogfile.h"
#include "..\Splash.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

//=============================================================================================
// CMainFrame

extern CPropertyApp theApp;
extern CAclFile		m_AclFile;
CMainSheet			*dlg = NULL;
CSystemTray			m_TrayIcon;
BOOL				m_IsSplash		= FALSE;
int					m_MessageIndex	= 0;

IMPLEMENT_DYNCREATE(CMainFrame, CFrameWnd)

CMainFrame::CMainFrame()
{
}

CMainFrame::~CMainFrame()
{
}

BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
	//{{AFX_MSG_MAP(CMainFrame)
	ON_WM_CREATE()
	ON_WM_TIMER()
	//}}AFX_MSG_MAP
	ON_MESSAGE(WM_ICON_NOTIFY,     OnTrayNotification)
	ON_MESSAGE(WM_SESSION_NOTIFY,  OnSessionNotification)
	ON_MESSAGE(WM_QUERY_ACL_NOTIFY,  OnAclNotification)
	ON_MESSAGE(WM_NET_MESSAGE,  OnShowNetMessage)
END_MESSAGE_MAP()

//=============================================================================================
// CMainFrame message handlers

int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct) 
{
	if (CFrameWnd::OnCreate(lpCreateStruct) == -1)
		return -1;

	if (!m_TrayIcon.Create(this, WM_ICON_NOTIFY, GUI_APP_CAPTION, theApp.LoadIcon(IDI_SMALL_PASS), IDR_MAINFRAME))
		return -1;

	theApp.SetTrayIcon();

	if(!m_TrayIcon.SetMenuDefaultItem(2,TRUE))
		return -1;
	
	DWORD dwFlags;
	if(InternetGetConnectedState(&dwFlags, 0))
		theApp.m_DownloadCommand.InitNetCommand();
	else
		SetTimer(1, 600000, NULL);

	if(!theApp.m_RegisterRequest.m_IsUploaded)
		SetTimer(2, 300000, NULL);

	// CG: The following line was added by the Splash Screen component.
	CSplashWnd::ShowSplashScreen(this);

	return 0;
}

LRESULT CMainFrame::OnTrayNotification(WPARAM wParam, LPARAM lParam)
{
	if(wParam == IDR_MAINFRAME && lParam == WM_LBUTTONDOWN)
	{
		if(m_MessageIndex != 0)
		{
			int tmpIndex = m_MessageIndex;
			m_MessageIndex = 0;
			AfxMessageBox(theApp.m_sMessage[tmpIndex], MB_ICONINFORMATION);
			return 0;
		}

		lParam = WM_LBUTTONDBLCLK;
	}

    return m_TrayIcon.OnTrayNotification(wParam, lParam);
}

void CMainFrame::InitMainSheet(int iActivePage)
{
	theApp.SetTrayIcon();

	if(dlg != NULL)
	{
		if(iActivePage >= 0)
			dlg->SetActivePage(iActivePage);

		if(!dlg->IsWindowVisible())
			dlg->ShowWindow(SW_SHOW);
		
		dlg->SetForegroundWindow();
		return;
	}

	CLogQuery		mLogQuery;
	CPacketMonitor	mPacketMonitor;
	CAcl			mAcl;
	CSystemSet		mSystemSet;
	CAbout			mAbout;

	dlg = new CMainSheet(GUI_APP_CAPTION);
	dlg->AddPage(&mPacketMonitor);
	dlg->AddPage(&mLogQuery);
	dlg->AddPage(&mAcl);
	dlg->AddPage(&mSystemSet);
	dlg->AddPage(&mAbout);
	dlg->SetLogoText(GUI_APP_LOGO);

	if(iActivePage >= 0)
		dlg->SetActivePage(iActivePage);

	dlg->DoModal();
	delete dlg;
	dlg = NULL;

	ODS(_T("XFILTER: release dlg = null"))
}

LRESULT CMainFrame::OnSessionNotification(UINT wParam, LONG lParam)
{
	static CXLogFile m_LogFile;

	if(theApp.m_XfIoControl == NULL)
		return XERR_SUCCESS;

	XFILTER_IO_CONTROL	ioControl;
	ioControl.isession	= wParam;
	ioControl.session	= new SESSION;

	theApp.m_XfIoControl(IO_CONTROL_GET_SESSION, &ioControl);

	if(ioControl.session->s == 0)
		return XERR_SUCCESS;

	CPacketMonitor::InsertOneList(
		CPacketMonitor::m_pListPacketMonitor, 
		ioControl.session,
		CPacketMonitor::IsMonitor,
		CPacketMonitor::IsScroll
		);
	
	if(CAclFile::GetBit(m_AclFile.mAclHeader.bSet, 0))
		m_LogFile.AddLog(ioControl.session, m_AclFile.mAclHeader.uiLogSize);

	if(ioControl.session->bAction == ACL_ACTION_DENY)
	{	
		if(CAclFile::GetBit(m_AclFile.mAclHeader.bSet, 2))
			MessageBeep(0xFFFFFFFF);

		if(!m_IsSplash && CAclFile::GetBit(m_AclFile.mAclHeader.bSet, 3))
		{
			DWORD		dwThreadId;
			::CreateThread(NULL, 0, SplashIcon, 0, 0, &dwThreadId);
		}
	}

	free(ioControl.session);

	return XERR_SUCCESS;
}

LRESULT CMainFrame::OnAclNotification(UINT wParam, LONG lParam)
{
	if(wParam < 0 || wParam > MAX_QUERY_SESSION)
		return XERR_INVALID_PARAMETER;

	XFILTER_IO_CONTROL ioControl;

	ioControl.isession = wParam;

	theApp.m_XfIoControl(IO_CONTROL_GET_QUERY_SESSION, &ioControl);

	CAcl::AddAclQuery(ioControl.sPathName, FALSE);
	
	theApp.m_XfIoControl(IO_CONTROL_SET_QUERY_SESSION, &ioControl);

	return XERR_SUCCESS;
}

DWORD WINAPI SplashIcon(LPVOID pVoid)
{
	m_IsSplash = TRUE;

	for(int i = 0; i < 3; i++)
	{
		m_TrayIcon.SetIcon(IDI_SMALL_ALERT);
		if(i == 2)
			continue;
		Sleep(180);
		m_TrayIcon.SetIcon(IDI_SMALL_NULL);
		Sleep(400);
	}
	
	m_IsSplash = FALSE;

	return 0;
}

void CMainFrame::OnTimer(UINT nIDEvent) 
{
	if(nIDEvent == 1)
	{
		DWORD dwFlags;
		if(InternetGetConnectedState(&dwFlags, 0))
		{
			theApp.m_DownloadCommand.InitNetCommand();
			KillTimer(nIDEvent);
		}
	}
	else if(nIDEvent == 2)
	{
		if(theApp.m_DownloadCommand.PreUpload())
			KillTimer(nIDEvent);
	}
	
	CFrameWnd::OnTimer(nIDEvent);
}

LRESULT CMainFrame::OnShowNetMessage(UINT wParam, LONG lParam)
{
	if(wParam <= 0 || wParam > MAX_NET_COMMAND)
		return XERR_INVALID_PARAMETER;

 	ODS(theApp.m_sMessage[wParam]);
	
	static DWORD	dwThreadId = NULL;
	if(dwThreadId == NULL)
		::CreateThread(NULL, 0, SplashMessage, 0, 0, &dwThreadId);

	return XERR_SUCCESS;
}

DWORD WINAPI SplashMessage(LPVOID pVoid)
{
	for(int i = 0; i < MAX_NET_COMMAND; i++)
	{
		if(theApp.m_sMessage[i][0] != '\0')
		{
			m_MessageIndex = i;

			m_IsSplash = TRUE;
			while(m_MessageIndex)
			{
				m_TrayIcon.SetIcon(IDI_SMALL_MESSAGE);
				Sleep(180);
				m_TrayIcon.SetIcon(IDI_SMALL_NULL);
				Sleep(400);
			}
			theApp.SetTrayIcon();
			m_IsSplash = FALSE;
		}
	}
	
	ODS("Exit Thread SplashMessage...");
	return 0;
}