chatcsocketdlg.h

个完整的代码

// ChatCSocketDlg.h : header file
//

#if !defined(AFX_CHATCSOCKETDLG_H__8F6A80A2_3695_44D9_9F65_681684D0ABD5__INCLUDED_)
#define AFX_CHATCSOCKETDLG_H__8F6A80A2_3695_44D9_9F65_681684D0ABD5__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

/////////////////////////////////////////////////////////////////////////////
// CChatCSocketDlg dialog

#include <AFXTEMPL.H>
#include <AFXSOCK.H>
#include "DataSocket.h"

#define WM_CONNECTION_COMEIN WM_USER+100

class CChatCSocketDlg : public CDialog
{
// Construction
public:
	CChatCSocketDlg(CWnd* pParent = NULL);	// standard constructor
	void OnSocketReceive(CDataSocket *pSocket, CMessg & msg);
	void OnSocketClose(CDataSocket *pSocket);
	void ExitChat();
	void GetErrorReason(int nErrorCode);
	void RefreshScreen();
	UINT m_ListenPort;
	CString m_ChatWords;
	BOOL m_bActive;

private:
	void OnServerReceive(CDataSocket *pSocket,CMessg & msg);
	void OnServerClose(CDataSocket *pSocket);
	void OnClientReceive(CMessg & msg);
	void OnClientClose();
	BOOL m_bClient;
	CDataSocket m_ClientSocket;
	CArray<CDataSocket*, CDataSocket*> m_DataSockets;
	
	void OnReceiveData(const CString & buffer);
	void ExitChat_Client();
	void ExitChat_Server();
	void SendMessageToPeer_Client();
	void SendMessageToPeer_Server();
	HANDLE m_ListenThreadHandle;
// Dialog Data
	//{{AFX_DATA(CChatCSocketDlg)
	enum { IDD = IDD_CHATCSOCKET_DIALOG };
		// NOTE: the ClassWizard will add data members here
	//}}AFX_DATA

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

// Implementation
protected:
	HICON m_hIcon;

	// Generated message map functions
	//{{AFX_MSG(CChatCSocketDlg)
	virtual BOOL OnInitDialog();
	afx_msg void OnSysCommand(UINT nID, LPARAM lParam);
	afx_msg void OnPaint();
	afx_msg HCURSOR OnQueryDragIcon();
	afx_msg void OnConnect();
	afx_msg void OnListen();
	afx_msg	void SendMessageToPeer();
	afx_msg void WelcomeNewClient(WPARAM w, LPARAM l);
	virtual void OnOK();
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

//{{AFX_INSERT_LOCATION}}
// Microsoft Visual C++ will insert additional declarations immediately before the previous line.

#endif // !defined(AFX_CHATCSOCKETDLG_H__8F6A80A2_3695_44D9_9F65_681684D0ABD5__INCLUDED_)