www.pudn.com > ServerClient_Socket.rar > ServerSocketDlg.cpp

// ServerSocketDlg.cpp : implementation file 
// 
 
#include "stdafx.h" 
#include  
#include "ServerSocket.h" 
#include "ServerSocketDlg.h" 
 
#ifdef _DEBUG 
#define new DEBUG_NEW 
#undef THIS_FILE 
static char THIS_FILE[] = __FILE__; 
#endif 
 
const int SOCK_TCP	= 0; 
const int SOCK_UDP  = 1; 
 
///////////////////////////////////////////////////////////////////////////// 
// 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() 
 
///////////////////////////////////////////////////////////////////////////// 
// CServerSocketDlg dialog 
 
CServerSocketDlg::CServerSocketDlg(CWnd* pParent /*=NULL*/) 
	: CDialog(CServerSocketDlg::IDD, pParent) 
{ 
	//{{AFX_DATA_INIT(CServerSocketDlg) 
	m_strPort = _T("2000"); 
	m_nSockType = SOCK_TCP;	// default TCP 
	//}}AFX_DATA_INIT 
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32 
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME); 
} 
 
void CServerSocketDlg::DoDataExchange(CDataExchange* pDX) 
{ 
	CDialog::DoDataExchange(pDX); 
	//{{AFX_DATA_MAP(CServerSocketDlg) 
	DDX_Control(pDX, IDC_TXT_MESSAGE, m_ctlMessage); 
	DDX_Control(pDX, IDC_MESSAGE_LIST, m_ctlMsgList); 
	DDX_Control(pDX, IDC_SVR_PORTINC, m_ctlPortInc); 
	DDX_Text(pDX, IDC_SVR_PORT, m_strPort); 
	DDX_Radio(pDX, IDC_TCP, m_nSockType); 
	//}}AFX_DATA_MAP 
} 
 
BEGIN_MESSAGE_MAP(CServerSocketDlg, CDialog) 
	//{{AFX_MSG_MAP(CServerSocketDlg) 
	ON_WM_SYSCOMMAND() 
	ON_WM_PAINT() 
	ON_WM_QUERYDRAGICON() 
	ON_BN_CLICKED(IDC_BTN_START, OnBtnStart) 
	ON_BN_CLICKED(IDC_BTN_STOP, OnBtnStop) 
	ON_WM_DESTROY() 
	ON_BN_CLICKED(IDC_BTN_SEND, OnBtnSend) 
	//}}AFX_MSG_MAP 
	ON_MESSAGE(WM_UPDATE_CONNECTION, OnUpdateConnection) 
END_MESSAGE_MAP() 
 
///////////////////////////////////////////////////////////////////////////// 
// CServerSocketDlg message handlers 
 
BOOL CServerSocketDlg::PreTranslateMessage(MSG* pMsg)  
{ 
	if (pMsg->message == WM_KEYDOWN) 
	{ 
		int nVirtKey = (int) pMsg->wParam; 
		if (nVirtKey == VK_ESCAPE) 
			return TRUE; 
		if (nVirtKey == VK_RETURN && (GetFocus()->m_hWnd  == m_ctlMessage.m_hWnd)) 
		{ 
			if (m_pCurServer->IsOpen()) 
				OnBtnSend(); 
			return TRUE; 
		} 
	} 
	 
	return CDialog::PreTranslateMessage(pMsg); 
} 
 
/////////////////////////////////////////////////////////////////////////////// 
// PickNextAvailable : this is useful only for TCP socket 
void CServerSocketDlg::PickNextAvailable() 
{ 
	m_pCurServer = NULL; 
	for(int i=0; iSetSmartAddressing( false ); 
			bSuccess = m_pCurServer->CreateSocket( m_strPort, AF_INET, SOCK_STREAM, 0); // TCP 
		} 
		else 
		{ 
			m_pCurServer->SetSmartAddressing( true ); 
			bSuccess = m_pCurServer->CreateSocket( m_strPort, AF_INET, SOCK_DGRAM, SO_BROADCAST); // UDP 
		} 
 
		if (bSuccess && m_pCurServer->WatchComm()) 
		{ 
			GetDlgItem(IDC_BTN_SEND)->EnableWindow( TRUE ); 
			GetDlgItem(IDC_BTN_STOP)->EnableWindow( TRUE ); 
			NextDlgCtrl(); 
			GetDlgItem(IDC_BTN_START)->EnableWindow( FALSE ); 
			GetDlgItem(IDC_TCP)->EnableWindow( FALSE ); 
			GetDlgItem(IDC_UDP)->EnableWindow( FALSE ); 
			CString strServer, strAddr; 
			m_pCurServer->GetLocalName( strServer.GetBuffer(256), 256); 
			strServer.ReleaseBuffer(); 
			m_pCurServer->GetLocalAddress( strAddr.GetBuffer(256), 256); 
			strAddr.ReleaseBuffer(); 
			CString strMsg = _T("Server: ") + strServer; 
					strMsg += _T(", @Address: ") + strAddr; 
					strMsg += _T(" is running on port ") + m_strPort + CString("\r\n"); 
			m_pCurServer->AppendMessage( strMsg ); 
		} 
	} 
 
	return bSuccess; 
} 
 
BOOL CServerSocketDlg::OnInitDialog() 
{ 
	ASSERT( GetDlgItem(IDC_BTN_SEND) != NULL ); 
	ASSERT( GetDlgItem(IDC_BTN_START) != NULL ); 
	ASSERT( GetDlgItem(IDC_BTN_STOP) != NULL ); 
 
	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 
	 
	// TODO: Add extra initialization here 
	m_ctlPortInc.SetRange32( 2000, 4500); 
	GetDlgItem(IDC_BTN_SEND)->EnableWindow( FALSE ); 
	GetDlgItem(IDC_BTN_STOP)->EnableWindow( FALSE ); 
 
	for(int i=0; i( lParam ); 
 
	// We need to do this only for TCP socket 
	if (m_nSockType != SOCK_TCP) 
		return 0L; 
 
	if ( pManager != NULL) 
	{ 
		// Server socket is now connected, we need to pick a new one 
		if (uEvent == EVT_CONSUCCESS) 
		{ 
			PickNextAvailable(); 
			StartServer(); 
		} 
		else if (uEvent == EVT_CONFAILURE || uEvent == EVT_CONDROP) 
		{ 
			pManager->StopComm(); 
			if (m_pCurServer == NULL) 
			{ 
				PickNextAvailable(); 
				StartServer(); 
			} 
		} 
	} 
 
	return 1L; 
} 
 
// The system calls this to obtain the cursor to display while the user drags 
//  the minimized window. 
HCURSOR CServerSocketDlg::OnQueryDragIcon() 
{ 
	return (HCURSOR) m_hIcon; 
} 
 
void CServerSocketDlg::OnBtnStart()  
{ 
	UpdateData(); 
 
	StartServer(); 
} 
 
void CServerSocketDlg::OnBtnStop()  
{ 
	// Disconnect all clients 
	for(int i=0; iIsOpen()) 
	{ 
		GetDlgItem(IDC_BTN_START)->EnableWindow( TRUE ); 
		PrevDlgCtrl(); 
		GetDlgItem(IDC_BTN_STOP)->EnableWindow( FALSE ); 
		GetDlgItem(IDC_TCP)->EnableWindow( TRUE ); 
		GetDlgItem(IDC_UDP)->EnableWindow( TRUE ); 
	} 
} 
 
 
void CServerSocketDlg::OnBtnSend()  
{ 
	CString strText; 
	m_ctlMessage.GetWindowText( strText ); 
	int nLen = strText.GetLength(); 
	stMessageProxy msgProxy; 
 
	if (nLen > 0) 
	{ 
		USES_CONVERSION; 
		strText += _T("\r\n"); 
		nLen = strText.GetLength(); 
		if (m_nSockType == SOCK_UDP) 
		{ 
			// send broadcast... 
			msgProxy.address.CreateFrom(_T("255.255.255.255"), m_strPort); 
			memcpy(msgProxy.byData, T2CA(strText), nLen); 
			nLen += msgProxy.address.Size(); 
		} 
		else 
		{ 
			nLen = __min(sizeof(msgProxy.byData)-1, nLen+1); 
			memcpy(msgProxy.byData, T2CA(strText), nLen); 
		} 
 
		// Send data to peer... 
		if (m_nSockType == SOCK_UDP) 
			m_pCurServer->WriteComm((const LPBYTE)&msgProxy, nLen, INFINITE); 
		else 
		{ 
			// Send to all clients 
			for(int i=0; i