testsocket2dlg.cpp

简单的聊天工具,适合初学者,网络编程,VC++实现

// testsocket2Dlg.cpp : implementation file
//

#include "stdafx.h"
#include "testsocket2.h"
#include "testsocket2Dlg.h"

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

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	//{{AFX_DATA_INIT(CAboutDlg)
	//}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CAboutDlg)
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CTestsocket2Dlg dialog

CTestsocket2Dlg::CTestsocket2Dlg(CWnd* pParent /*=NULL*/)
	: CDialog(CTestsocket2Dlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CTestsocket2Dlg)
	m_Port = 0;
	m_Input = _T("");
	m_Output = _T("");
	m_IPAddress = _T("");
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CTestsocket2Dlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CTestsocket2Dlg)
	DDX_Text(pDX, IDC_PORT, m_Port);
	DDX_Text(pDX, IDC_RECIVER, m_Input);
	DDX_Text(pDX, IDC_SEND, m_Output);
	DDX_Text(pDX, IDC_SERVERIP, m_IPAddress);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CTestsocket2Dlg, CDialog)
	//{{AFX_MSG_MAP(CTestsocket2Dlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDSEND, OnSend)
	ON_BN_CLICKED(IDDICONNECT, OnDiconnect)
	ON_BN_CLICKED(IDCONNECT, OnConnect)
	ON_EN_CHANGE(IDC_SEND, OnChangeSend)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CTestsocket2Dlg message handlers

BOOL CTestsocket2Dlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon
	m_ServerSocket=NULL;
    m_ClientSocket=NULL;
	m_arIn=NULL;
	m_arOut=NULL;
	m_file=NULL;
	m_Port=5000;
	m_IPAddress="";//"192.168.0.20";
	// TODO: Add extra initialization here
	
	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CTestsocket2Dlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CTestsocket2Dlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CTestsocket2Dlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

void CTestsocket2Dlg::OnSend() 
{
     UpdateData(TRUE);
	  *m_arOut<<m_Output;
	  m_arOut->Flush(); 
}




void CTestsocket2Dlg::OnDiconnect() 
{
  if (m_arOut!=NULL)
  {
    SocketReset();
    m_Input="断开连接!";
	UpdateData(FALSE);
  }
	
}

void CTestsocket2Dlg::OnConnect() 
{
   
    UpdateData(TRUE); 
    if(!AfxSocketInit())
	{
			MessageBox("WindowsSocket initial failed!","Send",MB_ICONSTOP); 
			return; 
	} 
	
    
	m_ClientSocket=new TestSock; 
	m_ClientSocket->GetDlg(this); 
	
	if(!m_ClientSocket->Create())
	{ 
			MessageBox("ReceiveSocket create failed!","Receive",MB_ICONSTOP); 
			return; 
	}
	else
	{       
			if (!m_ClientSocket->Connect((m_IPAddress),m_Port))
			{
						
					CString str="connect ";
					SocketReset();
					m_Input=str;
					m_Input+="Error!";
					UpdateData(FALSE);
			} 
			else 
			{ 
                    m_Input="Connect to the Server...\n";
	                UpdateData(FALSE);
					m_Input+="OK!";
					m_file=new CSocketFile(m_ClientSocket); 
					m_arIn=new CArchive(m_file, CArchive::load); 
					m_arOut=new CArchive(m_file, CArchive::store); 

			}
					UpdateData(FALSE);
		} 
}
void CTestsocket2Dlg::SocketReset()
{ 
  if(m_arIn!=NULL)
  {
	  delete m_arIn;
	  m_arIn=NULL;
  }
  if (m_arOut!=NULL) 
  {
	  delete m_arOut;
	  m_arOut=NULL;
  } 
  if (m_file!=NULL) 
  {
	  delete m_file;
	  m_file=NULL;
  }
  if (m_ClientSocket!=NULL) 
  {
	  delete m_ClientSocket;
	  m_ClientSocket=NULL;
  }
  UpdateData(FALSE);
}
void CTestsocket2Dlg::OnClose()
{
  
  m_Input+="client has disconnected!"; 
  UpdateData(FALSE);
}
void CTestsocket2Dlg::OnReceive()
{
	*m_arIn>>m_Input; 
	UpdateData(FALSE); 

}

void CTestsocket2Dlg::OnAccept()
{
  m_Input+="OK!";
  UpdateData(FALSE);
  m_ClientSocket=new TestSock;
  m_ClientSocket->GetDlg(this);
  m_ServerSocket->Accept(*m_ClientSocket);
  m_file=new CSocketFile(m_ClientSocket); 
  m_arIn=new CArchive(m_file, CArchive::load); 
  m_arOut=new CArchive(m_file, CArchive::store);
}

void CTestsocket2Dlg::OnChangeSend() 
{
	// TODO: If this is a RICHEDIT control, the control will not
	// send this notification unless you override the CDialog::OnInitDialog()
	// function and call CRichEditCtrl().SetEventMask()
	// with the ENM_CHANGE flag ORed into the mask.
	
	// TODO: Add your control notification handler code here
	
}