mainfrm.cpp

用socket编写聊天程序

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

#include "stdafx.h"
#include "CSocket.h"

#include "MainFrm.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_CONNECT_OPEN, OnConnectOpen)
	ON_COMMAND(ID_CONNECT_CLOSE, OnConnectClose)
	ON_COMMAND(ID_INFO_SEND, OnInfoSend)
	//}}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_pSocket=NULL;
	m_ServerName="123";
	m_PortAddress=700;
}

CMainFrame::~CMainFrame()
{
		if(m_pSocket)
	delete m_pSocket;
}

void CMainFrame::FetchMsg(CRequestSocket *pRequestSocket)
{
	char pMsg[10000],tempMsg[1000];
	int ByteCount;
	int EndFlag=0;
	CStringArray *temp=new CStringArray;
	strcpy(pMsg,"");
	do
	{
		strcpy(tempMsg,"");
		ByteCount=pRequestSocket->Receive(tempMsg,1000);
		if(ByteCount>1000||ByteCount<=0)
		{
			MessageBox("receive error","network message",MB_OK);
			return;
		}
		else if(ByteCount<1000 && ByteCount>0)
		{
			EndFlag=1;
		}
		tempMsg[ByteCount]=0;
		printf(tempMsg);
		strcat(pMsg,tempMsg);
	}while(EndFlag==0);
	MessageBox(pMsg,"network message",MB_OK);
}

void CMainFrame::SendMsg(CRequestSocket *pRequestSocket,char *pMsg)
{
	pRequestSocket->Send(pMsg,strlen(pMsg));
}

CRequestSocket *CMainFrame::ConnectServer()
{
	CRequestSocket *pSocket=new CRequestSocket(this);
	if(!(pSocket->Create()))
	{
		delete pSocket;
		MessageBox("create socket fail","network msg",MB_OK);
		return NULL;
	}
	if(!pSocket->Connect(m_ServerName,m_PortAddress))
	{
		delete pSocket;
		MessageBox("request connecting server fail","network msg",MB_OK);
		return NULL;
	}
	return pSocket;
}

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);

	return 0;
}

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 message handlers

void CMainFrame::OnConnectOpen() 
{
	// TODO: Add your command handler code here
	if(m_pSocket)
	{
		MessageBox("OK","warning",MB_OK);
		return;
	}
	CConnectDlg m_Dlg1;

	if(m_Dlg1.DoModal()==IDOK)
	{
	m_ServerName=m_Dlg1.m_ServerName;
	m_PortAddress=m_Dlg1.m_PortAddress;
	m_pSocket=new CRequestSocket(this);
	if(!(m_pSocket=ConnectServer()))
	{
		m_pSocket=NULL;
		return;
	}
	}
}




void CMainFrame::OnConnectClose() 
{
	// TODO: Add your command handler code here
	if(m_pSocket)
	{
		delete m_pSocket;
		m_pSocket=NULL;
	}
	else
		MessageBox("fail to connect to server","warning",MB_OK);
}





void CMainFrame::OnInfoSend() 
{
	// TODO: Add your command handler code here
		if(!m_pSocket)
	{
		MessageBox("fail to connect to server","warning",MB_OK);
		return;
	}

	CSendinfoDlg m_Dlg2;

	if(m_Dlg2.DoModal()==IDOK)
	{
		if(m_pSocket)
		{
			char pMsg[10000];
 			sprintf(pMsg,"%s",m_Dlg2.m_sendmsg);
			SendMsg(m_pSocket,pMsg);
		}
	}
}