网络编程

开发平台：
Visual C++

MainFrm.cpp：源码内容
							// MainFrm.cpp : implementation of the CMainFrame class
//
#include "stdafx.h"
#include "MainFrm.h"
#include "MyMonitor.h"
#include "Splash.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
#define LOG(x) { if((m_LogFile.m_uLogLevel) > 0) m_LogFile.WriteLog(m_csModuleName+x); }
#define INFO(x) { if((m_LogFile.m_uLogLevel) > 1) m_LogFile.WriteLog(m_csModuleName+x); }
/////////////////////////////////////////////////////////////////////////////
// CMainFrame
IMPLEMENT_DYNCREATE(CMainFrame, CFrameWnd)
BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
	// When register to USHA successful
	ON_MESSAGE(WM_MYUDP_CALLBACK, TreatSocketThreadMessage)
	//{{AFX_MSG_MAP(CMainFrame)
	ON_WM_CREATE()
	ON_COMMAND(ID_APP_ABOUT, OnAppAbout)
	ON_WM_TIMER()
	ON_WM_CLOSE()
	ON_WM_ENDSESSION()
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CMainFrame construction/destruction
CMainFrame::CMainFrame()
{
	// TODO: add member initialization code here
	m_csModuleName = LOG_STR_MAIN_NAME;
	// Set Log File Path
	m_LogFile.CreateLogPath(_T(DEFAULT_LOG_PATH));
	m_LogFile.m_uLogLevel = LOG_LEVEL;// record both LOG and INFO
	LOG("I Start Here");
	// Show message boxes
	m_bNotify = TRUE;
}
CMainFrame::~CMainFrame()
{
}
BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
{
	if( !CFrameWnd::PreCreateWindow(cs) )
		return FALSE;
	// TODO: Modify the Window class or styles here by modifying
	//  the CREATESTRUCT cs
	return TRUE;
}
/////////////////////////////////////////////////////////////////////////////
// CMainFrame diagnostics
#ifdef _DEBUG
void CMainFrame::AssertValid() const
{
	CFrameWnd::AssertValid();
}
void CMainFrame::Dump(CDumpContext& dc) const
{
	CFrameWnd::Dump(dc);
}
#endif //_DEBUG
/////////////////////////////////////////////////////////////////////////////
// CMainFrame functions
void CMainFrame::MyInfoBoxEx(CString strInfo, BOOL* pbMsgShowing)
{
	// ignore suspend mode
	MessageInfo MsgInfo;
	MsgInfo.hwndShowMsg = NULL;
	//MsgInfo.hwndShowMsg = m_hWnd;
	MsgInfo.pbMsgShowing = pbMsgShowing;
	MsgInfo.strMsgText = strInfo;
	MsgInfo.strMsgTitle = "UDP test MessageBox";
	MsgInfo.uMsgStyle = MB_ICONINFORMATION | MB_TOPMOST | MB_SETFOREGROUND;
	// Initialize event to avoid race between message threads
	// auto reset, initially reset
	MsgInfo.hEvent_GetParametersOK = CreateEvent(NULL, FALSE, FALSE, NULL);
	AfxBeginThread(ThreadMessageBox, &MsgInfo);
	// Give this thread time to transfer parameters to local MsgInfo
	// wait for GetParametersOK signal of this thread
	WaitForSingleObject(MsgInfo.hEvent_GetParametersOK, INFINITE);
}
void CMainFrame::MyInfoBox(CString strInfo, BOOL* pbMsgShowing)
{
	// if in suspend mode, don't show message
	if(!m_bNotify) return;
	MessageInfo MsgInfo;
	MsgInfo.hwndShowMsg = NULL;
	//MsgInfo.hwndShowMsg = m_hWnd;
	MsgInfo.pbMsgShowing = pbMsgShowing;
	MsgInfo.strMsgText = strInfo;
	MsgInfo.strMsgTitle = "UDP test MessageBox";
	MsgInfo.uMsgStyle = MB_ICONINFORMATION | MB_TOPMOST | MB_SETFOREGROUND;
	// Initialize event to avoid race between message threads
	// auto reset, initially reset
	MsgInfo.hEvent_GetParametersOK = CreateEvent(NULL, FALSE, FALSE, NULL);
	AfxBeginThread(ThreadMessageBox, &MsgInfo);
	// Give this thread time to transfer parameters to local MsgInfo
	// wait for GetParametersOK signal of this thread
	WaitForSingleObject(MsgInfo.hEvent_GetParametersOK, INFINITE);
}
CString CMainFrame::FormatString(DWORD dwState, CString strAddress)
{
	CString		_szDesc;
	_szDesc = "Some info to display...";
	return _szDesc;
}
/***********************************************************
 *
 * Event handlers for SocketThread.
 *
 ***********************************************************/
void CMainFrame::OnEvtConnected(WPARAM wParam, LPARAM lParam)
{
	if(((CMyUdpInfo*)lParam)->m_dwMyUdpState & MY_UDP_STATUS_CONNECTED) return;
	LOG("UDP Connected - " + ((CMyUdpInfo*)lParam)->m_csAddress);
	((CMyUdpInfo*)lParam)->m_dwMyUdpState |= MY_UDP_STATUS_CONNECTED;
	MyInfoBox(FormatString(MY_UDP_STATUS_CONNECTED,((CMyUdpInfo*)lParam)->m_csAddress), 
		&(((CMyUdpInfo*)lParam)->m_baMessageShowing[2]));
}
LRESULT CMainFrame::TreatSocketThreadMessage(WPARAM wParam, LPARAM lParam)
{
	//INFO("SocketThread Message in.");
	switch (LOWORD(wParam))
	{
		case EVENT_UDP_CONNECTED :
			OnEvtConnected(wParam,lParam); 
			break;
		default :
			LOG("Unknown SocketThread Message.");
			break;
	}
	return 0;
}
/////////////////////////////////////////////////////////////////////////////
// CMainFrame message handlers
BOOL CMainFrame::OnCmdMsg(UINT nID, int nCode, void* pExtra, AFX_CMDHANDLERINFO* pHandlerInfo)
{
	// let the view have first crack at the command
	if (m_wndView.OnCmdMsg(nID, nCode, pExtra, pHandlerInfo))
		return TRUE;
	// otherwise, do default handling
	return CFrameWnd::OnCmdMsg(nID, nCode, pExtra, pHandlerInfo);
}
int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct) 
{
	if (CFrameWnd::OnCreate(lpCreateStruct) == -1)
		return -1;
	
	// TODO: Add your specialized creation code here
	// For SystemTray Icon
    HICON hIcon = ::LoadIcon(AfxGetResourceHandle(), MAKEINTRESOURCE(IDR_MAINFRAME));  // Icon to use
	if (!m_TrayIcon.Create(
                        NULL,                            // Let icon deal with its own messages
                        WM_ICON_NOTIFY,                  // Icon notify message to use
                        _T("This is a Tray Icon - Right click on me!"),  // tooltip
                        hIcon,
                        IDR_POPUP_MENU,                  // ID of tray icon popup menu
                        FALSE,
                        _T("Here's a cool new Win2K balloon!"), // balloon tip
                        _T("Look at me!"),               // balloon title
                        NIIF_WARNING,                    // balloon icon
                        20 ))                            // balloon timeout
    {
		return -1;
    }
	// CG: The following line was added by the Splash Screen component.
	CSplashWnd::ShowSplashScreen(this);
	return 0;
}
int CMainFrame::BeginMyUdpThread(CMyUdpInfo* pMyUdpInfo)
{
	// Set the info of a MyUdp thread and start it
	pMyUdpInfo->m_hwndNotify = m_hWnd;
	pMyUdpInfo->m_pLogFile = &m_LogFile;
	pMyUdpInfo->m_csClientName = DEFAULT_CLIENTNAME;
	pMyUdpInfo->m_dwMyUdpState = 0;
	pMyUdpInfo->m_csAddress = DEFAULT_IP;
	ResetEvent(pMyUdpInfo->m_hEvent_GetParametersDone);
	ResetEvent(pMyUdpInfo->m_hEvent_Kill);
	SetEvent(pMyUdpInfo->m_hEvent_Killed);
	AfxBeginThread(MyUdpThreadProc, pMyUdpInfo);
	// Give this thread time to transfer parameters to local variable
	// wait for GetParametersOK signal of this thread
	WaitForSingleObject(pMyUdpInfo->m_hEvent_GetParametersDone, INFINITE);
	return 0;
}
void CMainFrame::OnTimer(UINT nIDEvent)
{
	// TODO: Add your message handler code here and/or call default
	switch (nIDEvent)
	{
		case TIMER_ID_MAINFRAME :
			KillTimer(TIMER_ID_MAINFRAME);
			break;
		default :
			break;
	}
	// Splash screen has ended
	// start communication now
}
// App command to run the About dialog
void CMainFrame::OnAppAbout()
{
	// TODO: Add your command handler code here
	static BOOL bShowing;
	if(bShowing) return;
	bShowing = TRUE;
	MyInfoBoxEx(ABOUT_VERSION, &bShowing);
	bShowing = FALSE;
}
void CMainFrame::OnClose() 
{
	// TODO: Add your message handler code here and/or call default
	//Kill all the threads
	for(int i=0;i<UDP_THREAD_NUMBER;i++)
	{
		SetEvent(m_MyUdpArray[i].m_hEvent_Kill);
	}
	for(i=0;i<UDP_THREAD_NUMBER;i++)
	{
		WaitForSingleObject(m_MyUdpArray[i].m_hEvent_Killed, INFINITE);
	}
	
	CFrameWnd::OnClose();
}
void CMainFrame::OnEndSession(BOOL bEnding) 
{
	CFrameWnd::OnEndSession(bEnding);
	
	// TODO: Add your message handler code here
	if(bEnding)
	{
		PostMessage(WM_CLOSE, 0, 0);
	}
	
}

						