mainfrm.cpp

关于远程网络监视程序的源码

// MainFrm.cpp : CMainFrame 类的实现
//

#include "stdafx.h"
#include "PortMon.h"

#include "PortMonDoc.h"
#include "PortMonView.h"
#include "ModuleView.h"
#include "dlgoptions.h"
#include "comboipaddr.h"
#include ".\mainfrm.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// CMainFrame

IMPLEMENT_DYNCREATE(CMainFrame, CFrameWnd)

BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
	ON_REGISTERED_MESSAGE(BCGM_RESETTOOLBAR, OnToolbarReset)
	ON_WM_CREATE()
	ON_WM_SIZE()
	ON_WM_TIMER()
	ON_WM_SYSCOMMAND()
	ON_COMMAND(ID_SHOW, OnShow)
	ON_COMMAND(ID_HIDE, OnHide)
	ON_MESSAGE(WM_MYICONNOTIFY, OnMyIconNotify)	//托盘事件函数
	ON_COMMAND(ID_HOMEPAGE, OnHomePage)

	ON_COMMAND(ID_TOOL_CUSTOMIZE, OnToolCustomize)
	ON_COMMAND(ID_VIEW_NETPROC, OnViewNetProc)
	ON_UPDATE_COMMAND_UI(ID_VIEW_NETPROC, OnUpdateViewNetProc)
	ON_COMMAND(ID_VIEW_REFRESH, OnViewRefresh)
	ON_COMMAND(ID_TOOL_OPTIONS, OnToolOptions)
	ON_WM_DESTROY()
END_MESSAGE_MAP()

static UINT indicators[] =
{
	ID_SEPARATOR,           // 状态行指示器
	ID_INDICATOR_CAPS,
	ID_INDICATOR_NUM,
	ID_INDICATOR_SCRL,
};

// CMainFrame 构造/析构

CMainFrame::CMainFrame()
	: m_bPanelCreated(FALSE)
	, m_obTrayIcon(ID_TRAYICON)
{
	// TODO: 在此添加成员初始化代码
}

CMainFrame::~CMainFrame()
{
}

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

	CBCGPVisualManager::SetDefaultManager (RUNTIME_CLASS(CBCGPWinXPVisualManager));
	CBCGPToolBar::EnableQuickCustomization ();

	//---------------------------------
	// Set toolbar and menu image size:
	//---------------------------------
	CBCGPToolBar::SetSizes (CSize (32, 30), CSize (23, 23));
	CBCGPToolBar::SetMenuSizes (CSize (22, 22), CSize (16, 16));

	// create menu bar
	if (!m_wndMenuBar.CreateEx(this, TBSTYLE_TRANSPARENT))
	{
		TRACE0("Failed to create menubar\n");
		return -1;      // fail to create
	}

	m_wndMenuBar.SetBarStyle( m_wndMenuBar.GetBarStyle() | CBRS_SIZE_DYNAMIC | CBRS_TOOLTIPS | CBRS_FLYBY );
	m_wndMenuBar.SetBarStyle( m_wndMenuBar.GetBarStyle() &
		~(CBRS_GRIPPER | CBRS_BORDER_TOP | CBRS_BORDER_BOTTOM | CBRS_BORDER_LEFT | CBRS_BORDER_RIGHT) );

	// Create the animation control
	if (!m_wndAnimator.Create( WS_CHILD | WS_VISIBLE, CRect(0, 0, 48, 24), this, 0) ||
		!m_wndAnimator.Open(IDR_FLAG))
	{
		TRACE0("Failed to create animation\n");
		return -1;      // fail to create
	}

	// create a colorful toolbar with true color bitmaps
	if ( !m_wndToolBar.CreateEx(this, TBSTYLE_TRANSPARENT) ||
		!m_wndToolBar.LoadToolBar(IDR_MAINFRAME, 0, 0, FALSE, 0, 0, IDB_HTOOLBAR) )
	{
		TRACE0("未能创建工具栏\n");
		return -1;      // 未能创建
	}

	// set tool bar style
	m_wndToolBar.SetBarStyle( m_wndToolBar.GetBarStyle() | CBRS_SIZE_DYNAMIC | CBRS_TOOLTIPS | CBRS_FLYBY );
	m_wndToolBar.SetBarStyle( m_wndToolBar.GetBarStyle() &
		~(CBRS_GRIPPER | CBRS_BORDER_TOP | CBRS_BORDER_BOTTOM | CBRS_BORDER_LEFT | CBRS_BORDER_RIGHT) );

	m_wndToolBar.EnableCustomizeButton (TRUE, ID_TOOL_CUSTOMIZE, _T("自定义..."));

	// Create rebar, get rid of gripper
	if ( !m_wndReBar.Create(this) ||
		!m_wndReBar.AddBar(&m_wndMenuBar, NULL, NULL, RBBS_NOGRIPPER | RBBS_FIXEDBMP) ||
		!m_wndReBar.AddBar(&m_wndToolBar, NULL, NULL, RBBS_NOGRIPPER | RBBS_FIXEDBMP | RBBS_BREAK) ||
		!m_wndReBar.AddBar(&m_wndAnimator, NULL, NULL, RBBS_FIXEDSIZE | RBBS_FIXEDBMP) )
	{
		TRACE0("Failed to create rebar\n");
		return -1;      // fail to create
	}

	m_wndMenuBar.AdjustLayout();
	m_wndToolBar.AdjustLayout();

	// create status bar
	if (!m_wndStatusBar.Create(this) ||
		!m_wndStatusBar.SetIndicators(indicators,
		  sizeof(indicators)/sizeof(UINT)))
	{
		TRACE0("未能创建状态栏\n");
		return -1;      // 未能创建
	}
	m_wndStatusBar.SetPaneStyle(0, SBPS_STRETCH);

	// TODO: 如果不需要工具栏可停靠，则删除这三行
	EnableDocking(CBRS_ALIGN_ANY);

	m_wndReBar.EnableDocking(CBRS_TOP);
	DockControlBar( &m_wndReBar );

	//在任务栏状态区添加图标
	CPortMonApp *pApp = (CPortMonApp *)AfxGetApp();
	if( pApp->m_bTrayMode )
	{
		m_obTrayIcon.SetNotificationWnd(this, WM_MYICONNOTIFY);
		if( !m_obTrayIcon.SetIcon(IDR_MAINFRAME, IDR_TRAYMENU) )
		{
			return -1;
		}
	}

	// start timer and animation
	SetTimer(ID_TIMER_REFRESH, pApp->m_nIntervals*1000, NULL);
	CAnimateCtrl &aniCtrl = GetAnimateCtrl();
	if( aniCtrl.GetSafeHwnd() != NULL )
	{
		aniCtrl.Play(0, -1, -1);
	}

	return 0;
}

BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
{
	if( !CFrameWnd::PreCreateWindow(cs) )
		return FALSE;
	// TODO: 在此处通过修改 CREATESTRUCT cs 来修改窗口类或
	// 样式

	return TRUE;
}


// CMainFrame 诊断

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

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

#endif //_DEBUG


// CMainFrame 消息处理程序


BOOL CMainFrame::OnCreateClient(LPCREATESTRUCT lpcs, CCreateContext* pContext)
{
	// create a splitter with 1 row, 2 columns
	if (!m_wndSplitter.CreateStatic(this, 2, 1))
	{
		TRACE0("Failed to Create First Splitter\n");
		return FALSE;
	}

	// add the first splitter pane - the default view in column 0
	if (!m_wndSplitter.CreateView(0, 0,
		pContext->m_pNewViewClass, CSize(0, 400), pContext))
	{
		TRACE0("Failed to create first pane\n");
		return FALSE;
	}

	// now create the two views inside the nested splitter
	if (!m_wndSplitter.CreateView(1, 0,
		RUNTIME_CLASS(CModuleView), CSize(0, 0), pContext))
	{
		TRACE0("Failed to create third pane\n");
		return FALSE;
	}

	m_bPanelCreated = TRUE;

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

void CMainFrame::OnSize(UINT nType, int cx, int cy)
{
	CFrameWnd::OnSize(nType, cx, cy);

	// TODO: 在此处添加消息处理程序代码
	CRect rect;
	GetWindowRect(&rect);

	if(m_bPanelCreated)
	{
		m_wndSplitter.SetRowInfo(0, max(rect.Height()-300, 300), 10);
		m_wndSplitter.RecalcLayout();
	}
}

LRESULT CMainFrame::OnToolbarReset(WPARAM wp, LPARAM)
{
	UINT uiToolBarId = (UINT) wp;
	if (uiToolBarId != IDR_MAINFRAME)
	{
		return 0;
	}

	// "上一进程" and "网络进程" button has a text label:
	m_wndToolBar.SetToolBarBtnText (m_wndToolBar.CommandToIndex(ID_OPER_PREV),
		_T("上移"));
	m_wndToolBar.SetToolBarBtnText (m_wndToolBar.CommandToIndex(ID_VIEW_NETPROC),
		_T("网络进程"));

	// IP address combobox button
	m_wndToolBar.ReplaceButton( ID_TOOL_IPADDR, CComboIpAddr() );

	return 0;
}

void CMainFrame::SetStatusBarInfo(LPCTSTR szInfo)
{
	m_wndStatusBar.SetPaneText(ID_SEPARATOR, szInfo);
}

CAnimateCtrl& CMainFrame::GetAnimateCtrl(void)
{
	return m_wndAnimator;
}

void CMainFrame::OnSysCommand(UINT nID, LPARAM lParam)
{
	// TODO: 在此添加消息处理程序代码和/或调用默认值
	if( nID == SC_MINIMIZE && ((CPortMonApp *)AfxGetApp())->m_bTrayMode )
	{
		OnHide();
	}

	CBCGPFrameWnd::OnSysCommand(nID, lParam);
}


/////////////////
//托盘图标事件
LRESULT CMainFrame::OnMyIconNotify(WPARAM wParam/*uID*/, LPARAM lParam/*消息标识值*/)
{
	switch( lParam)
	{
	case WM_LBUTTONUP:	//点击图标
		if (IsWindowVisible())	//如果窗口已打开，隐藏窗口
		{
			//利用这两个动作就可实现动画了
			OnHide();		//隐藏动画
		}
		else					//否则，显示窗口
		{
			//利用这两个动作就可实现动画了	
			OnShow();	//显示动画
		}
		break;

	case WM_RBUTTONUP:	//右击图标
		{
			//右击是为了弹出菜单
			CPoint point;
			CMenu menu;

			GetCursorPos( &point);
			VERIFY( menu.LoadMenu( IDR_TRAYMENU ) );
			CMenu * pMenu = menu.GetSubMenu(0);

			//要显示窗口还是隐藏
			if( IsWindowVisible() )
			{
				pMenu->DeleteMenu( ID_SHOW, MF_BYCOMMAND);
			}
			else
			{
				pMenu->DeleteMenu( ID_HIDE, MF_BYCOMMAND);
			}
		
			SetForegroundWindow();
			pMenu->TrackPopupMenu( TPM_LEFTALIGN | TPM_LEFTBUTTON, point.x, point.y, this );
		}
		break;

	default:
		m_obTrayIcon.OnTrayNotification(wParam, lParam);
		break;
	}

	return 0;
}

void CMainFrame::OnShow() 
{
	if( !IsWindowVisible() )	//如果窗口已打开，隐藏窗口
	{
		//利用这两个动作就可实现动画了	
		ShowWindow( SW_MINIMIZE );
		ShowWindow( SW_SHOWNORMAL );	//显示动画
	}

	SetTimer( ID_TIMER_REFRESH, ((CPortMonApp *)AfxGetApp())->m_nIntervals*1000, NULL );
	CAnimateCtrl &aniCtrl = GetAnimateCtrl();
	if( aniCtrl.GetSafeHwnd() != NULL )
	{
		aniCtrl.Play(0, -1, -1);
	}
}

void CMainFrame::OnHide() 
{
	StopMon();

	if ( IsWindowVisible() )	//如果窗口已打开，隐藏窗口
	{
		//利用这两个动作就可实现动画了
		ShowWindow( SW_MINIMIZE );
		ShowWindow( SW_HIDE );		//隐藏动画
	}
}

void CMainFrame::OnHomePage()
{
	::ShellExecute(
			GetSafeHwnd(),
			"open",
			"http://www.6663.net",
			"",
			"",
			SW_SHOWNORMAL);
}

CPortMonView *CMainFrame::GetPortView(void)
{
	CPortMonView *pView = (CPortMonView *)m_wndSplitter.GetPane(0, 0);

	return pView;
}

void CMainFrame::OnViewNetProc()
{
	CPortMonApp *pApp = (CPortMonApp *)AfxGetApp();
	ASSERT(pApp);

	pApp->m_bShowNetProc = !pApp->m_bShowNetProc;
	OnViewRefresh();

	CPortMonDoc *pDoc = (CPortMonDoc *)GetActiveDocument();
	pDoc->SetCurProcess( 0 );
}

void CMainFrame::OnUpdateViewNetProc(CCmdUI *pCmdUI)
{
	pCmdUI->SetCheck( ((CPortMonApp *)AfxGetApp())->m_bShowNetProc );
}

void CMainFrame::OnTimer(UINT nIDEvent)
{
	// TODO: 在此添加消息处理程序代码和/或调用默认值
	if( nIDEvent == ID_TIMER_REFRESH )
	{
		KillTimer(ID_TIMER_REFRESH);

		CPortMonView *pView = GetPortView();
		CPortMonApp *pApp = (CPortMonApp *)AfxGetApp();
		if( pView && pApp->m_bShowNetProc )
		{
			pView->UpdateNetProc();
		}

		SetTimer(ID_TIMER_REFRESH, pApp->m_nIntervals*1000, NULL);
	}
	else
	{
		CFrameWnd::OnTimer(nIDEvent);
	}
}

void CMainFrame::OnToolCustomize()
{
	//------------------------------------
	// Create a customize toolbars dialog:
	//------------------------------------
	CBCGPToolbarCustomize* pDlgCust = new CBCGPToolbarCustomize (this,
		TRUE /* Automatic menus scaning */,
		BCGCUSTOMIZE_MENU_SHADOWS | BCGCUSTOMIZE_TEXT_LABELS | 
		BCGCUSTOMIZE_LOOK_2000 | BCGCUSTOMIZE_MENU_ANIMATIONS /*|
		BCGCUSTOMIZE_SELECT_SKINS*/);

	//pDlgCust->EnableUserDefinedToolbars();
	pDlgCust->ReplaceButton( ID_TOOL_IPADDR, CComboIpAddr() );

	pDlgCust->Create ();
}

void CMainFrame::OnToolOptions()
{
	// TODO: 在此添加命令处理程序代码
	CPortMonApp *pApp = (CPortMonApp *)AfxGetApp();
	ASSERT(pApp);

	CDlgOptions dlg(pApp->m_nIntervals, pApp->m_bAnimation, pApp->m_bTrayMode); // set current value
	if( dlg.DoModal() == IDOK )
	{
		pApp->m_nIntervals = dlg.m_nIntervals;
		pApp->m_bTrayMode = dlg.m_bTrayMode;
		pApp->m_bAnimation = dlg.m_bAnimation;

		// set timer intervals
		KillTimer(ID_TIMER_REFRESH);
		SetTimer( ID_TIMER_REFRESH, pApp->m_nIntervals*1000, NULL );

		// set animation
		CAnimateCtrl &aniCtrl = GetAnimateCtrl();
		if( aniCtrl.GetSafeHwnd() != NULL )
		{
			if( pApp->m_bAnimation )
			{
				aniCtrl.Play(0, -1, -1);
			}
			else
			{
				aniCtrl.Stop();
				aniCtrl.Seek(0);
			}
		}

		// set tray mode
		if( pApp->m_bTrayMode )
		{
			m_obTrayIcon.SetNotificationWnd(this, WM_MYICONNOTIFY);
			m_obTrayIcon.SetIcon(IDR_MAINFRAME, IDR_TRAYMENU);
		}
		else
		{
			m_obTrayIcon.SetIcon(0, 0);
		}
	}
}

void CMainFrame::OnViewRefresh()
{
	CPortMonView *pView = GetPortView();
	if( ((CPortMonApp *)AfxGetApp())->m_bShowNetProc )
	{
		pView->AddAllNetProc();
	}
	else
	{
		pView->AddAllProc();
	}
}

void CMainFrame::OnDestroy()
{
	CBCGPFrameWnd::OnDestroy();

	// TODO: 在此处添加消息处理程序代码
	StopMon();
}

void CMainFrame::StopMon(void)
{
	KillTimer( ID_TIMER_REFRESH );

	CAnimateCtrl &aniCtrl = GetAnimateCtrl();
	if( aniCtrl.GetSafeHwnd() != NULL )
	{
		aniCtrl.Stop();
		aniCtrl.Seek(0);
	}
}

BOOL CMainFrame::OnCommand(WPARAM wParam, LPARAM lParam)
{
	//----------------------------------------------------
	// If control located on toolbar has focus, we should 
	// redirect clipboard commands to this control:
	//----------------------------------------------------
	if (CComboIpAddr::HasFocus () &&
		HIWORD (wParam) == 1)	// Command from accelerator
	{
		UINT uiCmd = LOWORD (wParam);
		switch (uiCmd)
		{
		case ID_EDIT_PASTE:
			::SendMessage (::GetFocus (), WM_PASTE, 0, 0);
			return TRUE;

		case ID_EDIT_COPY:
			::SendMessage (::GetFocus (), WM_COPY, 0, 0);
			return TRUE;

		case ID_EDIT_CUT:
			::SendMessage (::GetFocus (), WM_CUT, 0, 0);
			return TRUE;
		}
	}

	return CBCGPFrameWnd::OnCommand(wParam, lParam);
}

CBCGPToolbarComboBoxButton* CMainFrame::GetComboIpAddr(void)
{
	CBCGPToolbarComboBoxButton* pIpAddrCombo = NULL;

	CObList listButtons;
	if (CBCGPToolBar::GetCommandButtons (ID_TOOL_IPADDR, listButtons) > 0)
	{
		POSITION posCombo = listButtons.GetHeadPosition ();
		while( posCombo != NULL )
		{
			CBCGPToolbarComboBoxButton* pCombo = 
				DYNAMIC_DOWNCAST( CBCGPToolbarComboBoxButton, listButtons.GetNext(posCombo) );

			if (pCombo != NULL)
			{
				pIpAddrCombo = pCombo;
				break;
			}
		}
	}

	return pIpAddrCombo;
}