mainfrm.cpp

一个网络监视的程序

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

#include "stdafx.h"
#include "CCAMS.h"

#include "MainFrm.h"
#include "CCAMSView.h"
#include "LeftView.h"

#include "MulThreadSever.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_START_LISTENTHREAD, OnStartListenthread)
	ON_UPDATE_COMMAND_UI(ID_START_LISTENTHREAD, OnUpdateStartListenthread)
	ON_COMMAND(ID_END_LISTENTHREAD, OnEndListenthread)
	ON_UPDATE_COMMAND_UI(ID_END_LISTENTHREAD, OnUpdateEndListenthread)
	ON_COMMAND(ID_HELP, OnHelp)
	//}}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()
{
	// TODO: add member initialization code here
	m_bStart = 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("Failed to create toolbar\n");
		return -1;      // fail to create
	}

	if (!m_wndStatusBar.Create(this) ||
		!m_wndStatusBar.SetIndicators(indicators,
		  sizeof(indicators)/sizeof(UINT)))
	{
		TRACE0("Failed to create status bar\n");
		return -1;      // fail to create
	}

	// TODO: Delete these three lines if you don't want the toolbar to
	//  be dockable
//	m_wndToolBar.EnableDocking(CBRS_ALIGN_ANY);
//	EnableDocking(CBRS_ALIGN_ANY);
//	DockControlBar(&m_wndToolBar);


	// CreateComboBox;

	//找到指定的工具项
    int index = 0;
	while (m_wndToolBar.GetItemID(index) != ID_TOOL_CATEGORY)
		index++;

    //设置指定工具项的宽度并获取新的区域  80是宽度
    m_wndToolBar.SetButtonInfo(index, ID_TOOL_CATEGORY, TBBS_SEPARATOR, 120);

	RECT rect;
    m_wndToolBar.GetItemRect(index, &rect);

    //设置位置
    rect.top += 2;
    rect.bottom += 200;

    // 创建并显示
    if ( !m_wndCategory.Create(
		WS_CHILD | WS_VISIBLE | CBS_AUTOHSCROLL | CBS_DROPDOWNLIST | CBS_HASSTRINGS,
		rect, &m_wndToolBar, ID_TOOL_CATEGORY) )
    {
       TRACE0("Failed to create combo-box\n");
       return FALSE;
    }
	m_wndCategory.ShowWindow(SW_SHOW);

	// 
    m_wndCategory.AddString("CPU");
	m_wndCategory.AddString("FILE");
    m_wndCategory.AddString("MAIL");
	m_wndCategory.AddString("NET");
    m_wndCategory.AddString("PROCESS");
	m_wndCategory.AddString("All ACTIVITY");
	m_wndCategory.SetCurSel( 0 );
/*
	// 创建按钮: Refresh
    index = 0;
    while (m_wndToolBar.GetItemID(index) != ID_BUTTON_REFRESH)
		index++;

    //设置指定工具项的宽度并获取新的区域  80是宽度
	m_wndToolBar.SetButtonInfo(index, ID_BUTTON_REFRESH, TBBS_SEPARATOR, 100);

	m_wndToolBar.GetItemRect(index, &rect);
    rect.top += 2;
    rect.bottom -= 2;

    if ( !m_wndRef.Create( "Refresh", WS_CHILD | WS_VISIBLE | WS_OVERLAPPED,
		rect, &m_wndToolBar, ID_TOOL_CATEGORY) )
    {
       TRACE0("Failed to create combo-box\n");
       return FALSE;
    }
    m_wndRef.ShowWindow(SW_SHOW);
*/
	this->CenterWindow();

	return 0;
}

BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
{

//	cs.style &= ~FWS_PREFIXTITLE;
//	cs.style |= FWS_ADDTOTITLE;

	if( !CFrameWnd::PreCreateWindow(cs) )
		return FALSE;
	// TODO: Modify the Window class or styles here by modifying
	//  the CREATESTRUCT cs
//	cs.style = WS_OVERLAPPED | WS_CAPTION | FWS_PREFIXTITLE
//		| WS_SYSMENU | WS_MINIMIZEBOX | WS_MAXIMIZEBOX | WS_THICKFRAME;
	cs.style = WS_OVERLAPPEDWINDOW;

	// 窗口标题
	cs.lpszName = "Central Computer Activity Monitor System";

	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


BOOL CMainFrame::OnCreateClient(LPCREATESTRUCT lpcs, CCreateContext* pContext) 
{
	// TODO: Add your specialized code here and/or call the base class
	if (!m_wndSplitter.CreateStatic(this, 1, 2, WS_CHILD | WS_VISIBLE))
		return FALSE;

//	m_pDoc = (CCCAMSDoc*)(pContext->m_pCurrentDoc);

	if (!m_wndSplitter.CreateView(0, 0, RUNTIME_CLASS(CLeftView), CSize(200, 0), pContext) ||
		!m_wndSplitter.CreateView(0, 1, RUNTIME_CLASS(CCCAMSView), CSize(0, 0), pContext))
	{
		m_wndSplitter.DestroyWindow();
		return FALSE;
	}
	return true;

//	return CFrameWnd::OnCreateClient(lpcs, pContext);
}


void CMainFrame::OnStartListenthread() 
{
	StartListenThread();
}

void CMainFrame::StartListenThread()
{
	InitSocket();

	extern bool bNewAccess;
	bNewAccess = true;	// 创建主监听线程前, 不存在多线程问题
	extern HANDLE gdwListenThread;	//侦听线程句柄
	extern SOCKET listenSocket;	//侦听套接字
	extern HANDLE ghExit;	//结束事件句柄
	unsigned long ThreadID;
	gdwListenThread = CreateThread(	
		NULL, 
		0,
		(LPTHREAD_START_ROUTINE)ListenThread,
		&ghExit, 
		0,	
		&ThreadID );

	//创建主侦听线程
	if ( !gdwListenThread )
	{
		MessageBox( " CreateThread for ListenThread failed! " );
		closesocket( listenSocket );
		return;
	}

//	MessageBox( " CreateThread for ListenThread Succeed! " );
	m_bStart = !m_bStart;
}

void CMainFrame::OnUpdateStartListenthread(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
	pCmdUI->Enable( !m_bStart );
}

void CMainFrame::OnEndListenthread() 
{
	CString prompt = "Are You Sure to Disconnect \nthe All connections?";
	if (AfxMessageBox(prompt,MB_OKCANCEL) != IDOK)
		return;

//	extern HANDLE gdwListenThread;	//侦听线程句柄
//	TerminateThread(gdwListenThread, 0);

//	MessageBox( " TerminateThread for ListenThread Succeed! " );

	StopAllConnect();
	m_bStart = !m_bStart;

	extern CRITICAL_SECTION g_criActiveClientList;
	extern ActivClient theActiveClientList;
	extern CRITICAL_SECTION g_criNewAccess;
	extern bool bNewAccess;

	EnterCriticalSection( &g_criActiveClientList );
	EnterCriticalSection( &g_criNewAccess );

	theActiveClientList.RemoveAll();
	bNewAccess = true;

	LeaveCriticalSection( &g_criNewAccess );
	LeaveCriticalSection( &g_criActiveClientList );

}

void CMainFrame::OnUpdateEndListenthread(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
	pCmdUI->Enable( m_bStart );
}



/*
void CMainFrame::DisDiagram()
{
	CCreateContext Context;	
	Context.m_pNewViewClass = RUNTIME_CLASS(CDiagView);
	Context.m_pCurrentDoc = m_pDoc;
	Context.m_pCurrentFrame = this;
	Context.m_pNewDocTemplate = Context.m_pCurrentDoc->GetDocTemplate();
	Context.m_pLastView = (CView*)m_wndSplitter.GetPane(0, 0);

	m_wndSplitter.DeleteView(0, 1);
	m_wndSplitter.CreateView(0, 1, RUNTIME_CLASS(CDiagView), CSize(0, 0), &Context);

	CDiagView *pView = (CDiagView*)m_wndSplitter.GetPane(0, 1);
	pView->GetParentFrame()->RecalcLayout();
	m_wndSplitter.RecalcLayout();
	pView->OnInitialUpdate();
	m_wndSplitter.SetActivePane(0, 1);
}
*/

void CMainFrame::OnHelp()
{
	// TODO: Add your command handler code here
    HWND handle=0; 
	ShellExecute(handle, "open", "ccamshelp.chm", NULL, NULL, SW_SHOWNORMAL); 
}

BOOL CMainFrame::DestroyWindow() 
{
	// TODO: Add your specialized code here and/or call the base class
	if ( m_bStart )
		StopAllConnect();
	
	return CFrameWnd::DestroyWindow();
}