// MainFrm.cpp : implementation of the CMainFrame class
//

#include "stdafx.h"
#include "FirewallApp.h"

#include "MainFrm.h"
#include "RuleDlg.h"
#include "FirewallAppDoc.h"
#include "FirewallAppView.h"
#include "rules.h"
#include "SelectDlg.h"
#include "ruleindlg.h"

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

/////////////////////////////////////////////////////////////////////////////
// CMainFrame

IMPLEMENT_DYNCREATE(CMainFrame, CFrameWnd)

BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
	//{{AFX_MSG_MAP(CMainFrame)
	ON_WM_CREATE()
	ON_COMMAND(ID_BUTTONSTART, OnButtonstart)
	ON_COMMAND(ID_BUTTONADD, OnButtonadd)
	ON_COMMAND(ID_BUTTONDEL, OnButtondel)
	ON_COMMAND(ID_BUTTONDESINSTALL, OnButtondesinstall)
	ON_COMMAND(ID_BUTTONINSTALL, OnButtoninstall)
	ON_COMMAND(ID_BUTTONSTOP, OnButtonstop)
	ON_UPDATE_COMMAND_UI(ID_BUTTONSTART, OnUpdateButtonstart)
	ON_UPDATE_COMMAND_UI(ID_BUTTONSTOP, OnUpdateButtonstop)
	ON_COMMAND(IDMENU_ADDRULE, OnMenuAddrule)
	ON_COMMAND(IDMENU_DELRULE, OnMenuDelrule)
	ON_COMMAND(IDMENU_INSTALLRULES, OnMenuInstallrules)
	ON_COMMAND(IDMENU_UNINSTALLRULES, OnMenuUninstallrules)
	ON_COMMAND(ID_MENUSTART, OnMenustart)
	ON_UPDATE_COMMAND_UI(ID_MENUSTART, OnUpdateMenustart)
	ON_COMMAND(ID_MENUSTOP, OnMenustop)
	ON_UPDATE_COMMAND_UI(ID_MENUSTOP, OnUpdateMenustop)
	ON_COMMAND(ID_APP_EXIT, OnAppExit)
	ON_COMMAND(IDMENU_LOADRULES, OnLoadRules)
	ON_COMMAND(IDMENU_SAVERULES, OnSaveRules)
	ON_UPDATE_COMMAND_UI(IDMENU_INSTALLRULES, OnUpdateInstallrules)
	ON_UPDATE_COMMAND_UI(IDMENU_UNINSTALLRULES, OnUpdateUninstallrules)
	ON_UPDATE_COMMAND_UI(ID_BUTTONDESINSTALL, OnUpdateButtondesinstall)
	ON_UPDATE_COMMAND_UI(ID_BUTTONINSTALL, OnUpdateButtoninstall)
	ON_UPDATE_COMMAND_UI(ID_BUTTONDEL, OnUpdateButtondel)
	ON_UPDATE_COMMAND_UI(ID_BUTTONADD, OnUpdateButtonadd)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

static UINT indicators[] =
{
	ID_SEPARATOR,           // status line indicator
	ID_INDICATOR_CAPS,
	ID_INDICATOR_NUM,
	ID_INDICATOR_SCRL,
};

/////////////////////////////////////////////////////////////////////////////
// CMainFrame construction/destruction

CMainFrame::CMainFrame()
{
	started = FALSE;
	installed = FALSE;
}

CMainFrame::~CMainFrame()
{
	
}

int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
	if (CFrameWnd::OnCreate(lpCreateStruct) == -1)
		return -1;
	
	if (!m_wndToolBar.CreateEx(this, TBSTYLE_FLAT, WS_CHILD | WS_VISIBLE | CBRS_TOP
		| CBRS_GRIPPER | CBRS_TOOLTIPS | CBRS_FLYBY | CBRS_SIZE_DYNAMIC) ||
		!m_wndToolBar.LoadToolBar(IDR_MAINFRAME))
	{
		TRACE0("無法創建狀態欄\n");
		return -1;      // fail to create
	}

	SetWindowText("防火墻");

	return 0;
}

BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
{
	if( !CFrameWnd::PreCreateWindow(cs) )
		return FALSE;


	//para quitar el untitled del titulo de la aplicacion
	cs.style &= ~ FWS_ADDTOTITLE;

	return TRUE;
}

/////////////////////////////////////////////////////////////////////////////
// CMainFrame diagnostics

#ifdef _DEBUG
void CMainFrame::AssertValid() const
{
	CFrameWnd::AssertValid();
}

void CMainFrame::Dump(CDumpContext& dc) const
{
	CFrameWnd::Dump(dc);
}

#endif //_DEBUG

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

/**
 *	啟動過濾功能
 *
 */
void CMainFrame::OnButtonstart() 
{
	if(cIpFilter.Filtering())
	{
		started = TRUE;
	}
}

/**
 *	停止過濾功能
 *
 */
void CMainFrame::OnButtonstop() 
{
	if(cIpFilter.UnFiltering())
	{
		started = FALSE;
	}
}

/**
 *	增加過濾規則
 *
 */
void CMainFrame::OnButtonadd() 
{
	CFirewallAppDoc *doc = (CFirewallAppDoc *)GetActiveDocument();
	CSelectDlg dlg;
	
	INT_PTR ret = dlg.DoModal();
	/*如果規則數目還沒有達到最大值,可繼續增加規則*/
	if(doc->nRules < MAX_RULES)
	{
		/*增加規則*/
		if( ret != IDCANCEL){
			/*增加流入數據包的過濾規則*/
			if(ret == IDC_IN){
				OnIn();
			}
			/*增加流出數據的過濾規則*/
			else if(ret == IDC_OUT){
				OnOut();
			}
			/*確定存在增加的過濾規則時的處理,*/
			if(bAdded){
				if(!doc->AddRule(rule))
					AfxMessageBox("添加過濾規則出錯");
				else
				{
					CFirewallAppView *view = (CFirewallAppView *)GetActiveView();
					view->UpdateList();
					if(installed) cIpFilter.AddRule(rule);
				}
			}
		}
	}

	else
		AfxMessageBox("過濾規則數目已經達到最大值");
}

/**
 *	刪除規則列表中被選中的規則
 *
 */
void CMainFrame::OnButtondel() 
{
	CFirewallAppView *view = (CFirewallAppView *)GetActiveView();

	POSITION pos = view->m_rules.GetFirstSelectedItemPosition();
	if (pos == NULL)
	{
		AfxMessageBox("請選擇一條過濾規則");
		return;
	}
	
	/*取得要刪除的規則的下標*/
	int position;
	position = view->m_rules.GetNextSelectedItem(pos);

	CFirewallAppDoc *doc = (CFirewallAppDoc *)GetActiveDocument();
	
	/*如果規則已經被應用在過濾器上,則在過濾器中刪除該規則,同時更新規則的內部表示及界面*/
	if(installed)
	{
		if(cIpFilter.DeleteRule(doc->rules[position]))
		{
			doc->DeleteRule(position);	
			view->UpdateList();
		}
	}
	else 
	{
		doc->DeleteRule(position);	
		view->UpdateList();
	}

}


/**
 *	把已經定義好的規則應用到過濾器上
 *
 */
void CMainFrame::OnButtoninstall() 
{
	CFirewallAppDoc *doc = (CFirewallAppDoc *)GetActiveDocument();

	unsigned int i;

	if(doc->nRules > 0){
		for(i=0;i<doc->nRules;i++)
		{
			/*把規則逐條應用到過濾器上*/
			if (!cIpFilter.AddRule(doc->rules[i])){
				AfxMessageBox("無法應用過濾規則");
				return ;
			}
		}	
		installed = TRUE;//最后標識規則已經應用到過濾器上
	}
}

/**
 *	將應用在過濾器上的規則不發生作用
 *
 */
void CMainFrame::OnButtondesinstall() 
{
	CFirewallAppDoc *doc = (CFirewallAppDoc *)GetActiveDocument();
	for(unsigned int i = 0; i < doc -> nRules; i ++){
		/*把規則逐條從與過濾器相關的Interface上刪除*/
		if(!cIpFilter.DeleteRule(doc -> rules[i])){
			AfxMessageBox("無法停用過濾規則");
			return ;		
		}
	}
	installed = FALSE;	//最后標識規則沒有被應用在過濾器上
}

/**
 *
 *
 */
void CMainFrame::OnUpdateButtonstart(CCmdUI* pCmdUI) 
{
	if(started)
		pCmdUI->Enable(FALSE);
	else
		pCmdUI->Enable(TRUE);	
}

/**
 *
 *
 */
void CMainFrame::OnUpdateButtonstop(CCmdUI* pCmdUI) 
{
	if(started)
		pCmdUI->Enable(TRUE);

	else
		pCmdUI->Enable(FALSE);
}

/**
 *
 *
 */
void CMainFrame::OnMenuAddrule() 
{
	OnButtonadd();	
}

/**
 *
 *
 */
void CMainFrame::OnMenuDelrule() 
{
	OnButtondel();	
}

/**
 *
 *
 */
void CMainFrame::OnMenuInstallrules() 
{
	OnButtoninstall();
}

/**
 *
 *
 */
void CMainFrame::OnMenuUninstallrules() 
{
	OnButtondesinstall();
	
}

/**
 *
 *
 */
void CMainFrame::OnMenustart() 
{
	OnButtonstart();
}

/**
 *
 *
 */
void CMainFrame::OnUpdateMenustart(CCmdUI* pCmdUI) 
{
	if(started)
		pCmdUI->Enable(FALSE);

	else
		pCmdUI->Enable(TRUE);	
	
}

/**
 *
 *
 */
void CMainFrame::OnMenustop() 
{
	OnButtonstop();
}

/**
 *
 *
 */
void CMainFrame::OnUpdateMenustop(CCmdUI* pCmdUI) 
{
	if(started)
		pCmdUI->Enable(TRUE);

	else
		pCmdUI->Enable(FALSE);
	
}

/**
 *
 *
 */
void CMainFrame::OnAppExit() 
{

}

/**
 *
 *
 */
void CMainFrame::OnLoadRules() 
{
	CFile file;
	CFileException e;
	DWORD nRead;

	CFirewallAppDoc *doc = (CFirewallAppDoc *)GetActiveDocument();

	CFileDialog dg(TRUE,NULL, NULL, OFN_HIDEREADONLY | OFN_CREATEPROMPT,"Rule Files(*.rul)|*.rul|all(*.*)|*.*||", NULL);
	if(dg.DoModal()==IDCANCEL)
		return;
	
	CString nf=dg.GetPathName();

	if(nf.GetLength() == 0)
	{
		AfxMessageBox("文件名無效");

		return;
	}

	if( !file.Open(nf, CFile::modeRead, &e ) )
	{
		AfxMessageBox("打開文件出錯");

		return;
	}

	doc->ResetRules();
	RuleInfo rule;
	
    do
    {
 		nRead = file.Read(&rule, sizeof(RuleInfo));
		
		if(nRead == 0)
			break;

		if(!doc->AddRule(rule))
		{
			AfxMessageBox("添加過濾規則出錯");
			break;
		}

    }while (1);
		
	CFirewallAppView *view = (CFirewallAppView *)GetActiveView();

	view->UpdateList();
	OnButtoninstall();
	OnButtonstart();
}

/**
 *
 *
 */
void CMainFrame::OnSaveRules() 
{
	CFirewallAppDoc *doc = (CFirewallAppDoc *)GetActiveDocument();

	if(doc->nRules == 0)
	{
		AfxMessageBox("無過濾規則");

		return;
	}


	CFileDialog dg(FALSE, NULL, NULL, OFN_HIDEREADONLY | OFN_CREATEPROMPT,"Rule Files(*.rul)|*.rul|all(*.*)|*.*||", NULL);
	if(dg.DoModal()==IDCANCEL)
		return;
	
	CString nf=dg.GetPathName();

	if(nf.GetLength() == 0)
	{
		AfxMessageBox("文件名無效");

		return;
	}

	CFile file;
	CFileException e;
	nf +=".rul";
	
	if( !file.Open( nf, CFile::modeCreate | CFile::modeWrite, &e ) )
	{
		AfxMessageBox("文件打開出錯");

		return;
	}


	unsigned int i;

	for(i=0;i<doc->nRules;i++)
	{
		file.Write(&doc->rules[i], sizeof(RuleInfo));
	}
	
	file.Close();
}

/**
 *
 *
 */
void CMainFrame::OnUpdateInstallrules(CCmdUI* pCmdUI) 
{
	if(installed) 
		pCmdUI ->Enable(FALSE);
	else 
		pCmdUI ->Enable(TRUE);
}

/**
 *
 *
 */
void CMainFrame::OnUpdateUninstallrules(CCmdUI* pCmdUI) 
{
	if(installed)
		pCmdUI ->Enable(TRUE);
	else 
		pCmdUI ->Enable(FALSE);
}

/**
 *
 *
 */
void CMainFrame::OnUpdateButtondesinstall(CCmdUI* pCmdUI) 
{
	if(installed)
		pCmdUI ->Enable(TRUE);
	else 
		pCmdUI ->Enable(FALSE);
	
}

/**
 *
 *
 */
void CMainFrame::OnUpdateButtoninstall(CCmdUI* pCmdUI) 
{
	if(installed) 
		pCmdUI ->Enable(FALSE);
	else 
		pCmdUI ->Enable(TRUE);
	
}

/**
 *
 *
 */
void CMainFrame::OnUpdateButtondel(CCmdUI* pCmdUI) 
{
	CFirewallAppDoc *doc = (CFirewallAppDoc *)GetActiveDocument();
	if(doc ->nRules > 0)
		pCmdUI ->Enable(TRUE);
	else 
		pCmdUI -> Enable(FALSE);
}

/**
 *
 *
 */
void CMainFrame::OnUpdateButtonadd(CCmdUI* pCmdUI) 
{
	CFirewallAppDoc *doc = (CFirewallAppDoc *)GetActiveDocument();
	if(doc ->nRules == MAX_RULES)
		pCmdUI ->Enable(FALSE);
	else pCmdUI ->Enable(TRUE);
}

/**
 *
 *
 */
void CMainFrame::OnIn() 
{
	CRuleInDlg dlg;
	if(dlg.DoModal() == IDOK)
	{		
		rule = dlg.rule;
		rule.destinationIp = cIpFilter.localInAddr;
		rule.destinationMask = ALL_MASK;
		rule.destinationPort = rule.sourcePort;
		bAdded = true;
	} else bAdded = false;
}

/**
 *
 *
 */
void CMainFrame::OnOut() 
{
	CRuleDlg dlg;	
	if(dlg.DoModal() == IDOK)
	{
		rule = dlg.rule;
		rule.sourceIp = cIpFilter.localInAddr;
		rule.sourceMask = ALL_MASK;
		rule.sourcePort = rule.destinationPort;
		bAdded = true;
	} else bAdded = false;	
}