networkdlg.cpp

Visual C++_ 600 编程学习捷径

// NetworkDlg.cpp : implementation file
//

#include "stdafx.h"
#include "Network.h"
#include "NetworkDlg.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()

/////////////////////////////////////////////////////////////////////////////
// CNetworkDlg dialog

CNetworkDlg::CNetworkDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CNetworkDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CNetworkDlg)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	wVersionRequested = 0x0101;
	sock_listen = 0;
	sock_udp    = 0;
	sock_connect = 0;
	sock_client = NULL;

	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CNetworkDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CNetworkDlg)
	DDX_Control(pDX, IDC_TCP_IP, m_TcpIp);
	DDX_Control(pDX, IDC_UDP_IP, m_UdpIp);
	DDX_Control(pDX, IDC_RECF_INFO, m_RecvInfo);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CNetworkDlg, CDialog)
	//{{AFX_MSG_MAP(CNetworkDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_MESSAGE(US_SOCK, OnListenSockMessage)
	ON_MESSAGE(US_UDP_SOCK, OnUdpSockMessage)
	ON_MESSAGE(US_CONNECT_SOCK, OnConnectSockMessage)
	ON_BN_CLICKED(IDC_BROADCAST, OnBroadcast)
	ON_BN_CLICKED(IDC_UDP_SEND, OnUdpSend)
	ON_BN_CLICKED(IDC_TCP_LINK, OnTcpLink)
	ON_BN_CLICKED(IDC_TCP_DISCONNECT, OnTcpDisconnect)
	ON_BN_CLICKED(IDC_TCP_SEND, OnTcpSend)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CNetworkDlg message handlers

BOOL CNetworkDlg::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
	
	// TODO: Add extra initialization here
	//Initialize the socket
	InitSocket();

	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CNetworkDlg::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 CNetworkDlg::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 CNetworkDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

BOOL CNetworkDlg::InitSocket(void)
{
	struct sockaddr_in server;
	WSADATA wsa;
	WSAStartup(wVersionRequested, &wsa);

	/* intialize listen socket */
	if(sock_listen != 0)
	{
	//	WSAAsyncSelect(sock_listen, hWnd, 0, 0);
		closesocket(sock_listen);
		sock_listen = 0;
	}
	sock_listen = socket(AF_INET, SOCK_STREAM, 0);
	if(sock_listen == INVALID_SOCKET)
	{
		sock_listen = 0;
		MessageBox("can not create listen socket", "ERROR", MB_OK);
		WSACleanup();
		return FALSE;
	}
	server.sin_family = AF_INET;
	server.sin_port = htons((u_short)5000);
	server.sin_addr.s_addr = INADDR_ANY;

	if(bind(sock_listen, (struct sockaddr *)&server, sizeof(server)) == SOCKET_ERROR)
	{
		MessageBox("listen socket bind error!", "ERROR", MB_OK);
		closesocket(sock_listen);
		sock_listen = 0;
		WSACleanup();
		return FALSE;
	}  
	if(WSAAsyncSelect(sock_listen, m_hWnd, US_SOCK, FD_ACCEPT | FD_CONNECT | FD_READ | FD_WRITE | FD_CLOSE ) == SOCKET_ERROR)
	{
		MessageBox("select listen socket event error!", "ERROR", MB_OK);
		closesocket(sock_listen);
		WSACleanup();
		return FALSE;
	}
	int nRet = listen(sock_listen, SOMAXCONN);
    if (nRet == SOCKET_ERROR)
	{
		MessageBox("listen error!", "ERROR", MB_OK);
		closesocket(sock_listen);
		WSACleanup();
		return FALSE;
	}

	/* intialize udp DGRAM socket */
	if(sock_udp != 0)
	{
	//	WSAAsyncSelect(sock_udp, hWnd, 0, 0);
		closesocket(sock_udp);
		sock_udp = 0;
	}
	sock_udp = socket(AF_INET, SOCK_DGRAM, 0);
	if(sock_udp == INVALID_SOCKET)
	{
		MessageBox("can not create udp socket", "ERROR", MB_OK);
		sock_udp = 0;
		WSACleanup();
		return FALSE;
	}
	server.sin_family = AF_INET;
	server.sin_port = htons((u_short)6000);
	server.sin_addr.s_addr = INADDR_ANY;

	BOOL optval = TRUE;
//	setsockopt(sock_udp, SOL_SOCKET, SO_REUSEADDR, (char FAR *)&optval, sizeof(optval));

	if(bind(sock_udp, (struct sockaddr *)&server, sizeof(server)) == SOCKET_ERROR)
	{
		closesocket(sock_udp);
		MessageBox("bind udp socket error ", "ERROR", MB_OK);
		sock_udp = 0;
		WSACleanup();
		return FALSE;
	}  
	if(setsockopt(sock_udp, SOL_SOCKET, SO_BROADCAST, (char FAR *)&optval, sizeof(optval)) == SOCKET_ERROR)
	{
		MessageBox("set broadcast mode error!", "ERROR", MB_OK);
		return -1;
	}
	if(WSAAsyncSelect(sock_udp, m_hWnd, US_UDP_SOCK, FD_READ | FD_WRITE | FD_CLOSE) == SOCKET_ERROR)
	{
		closesocket(sock_udp);
		MessageBox("select udp socket  event error ", "ERROR", MB_OK);
		sock_udp = 0;
		WSACleanup();
		return FALSE;
	}

	return TRUE;
}
void CNetworkDlg::OnListenSockMessage(WPARAM wParam, LPARAM lParam)
{
	BYTE buf[2048];
	int len;
	SOCKLOG * psock;

	switch(WSAGETSELECTEVENT(lParam))
	{
		case FD_READ:
		{
			if((len = recv(wParam, (char *)buf, 2048, 0)) == SOCKET_ERROR)
				break;
			psock = sock_client;
			while(psock)
			{
				if(psock->socket == wParam)
				{
					DealWithNetComm(5, psock->ip, buf, len);
					break;
				}
				else
				{
					psock = psock->next;
				}
			}
		}
			break;
		case FD_WRITE:
			break;
		case FD_ACCEPT:
			{
				struct sockaddr_in from;
				int length = sizeof(from);
				SOCKET client = accept(sock_listen, (struct sockaddr *)&from, &length);
				psock = new SOCKLOG;
				psock->socket = client;
				psock->next = sock_client;
				sock_client = psock;
				struct hostent * host = gethostbyaddr((const char *)&from.sin_addr, sizeof(in_addr), AF_INET);
				memset(psock->ip, 0, 20);
				strcpy(psock->ip, inet_ntoa(from.sin_addr));
				if(host)
					strcpy(psock->name, host->h_name);
				else
					strcpy(psock->name, "Unknown Host");
				DealWithNetComm(1, psock->ip, buf, 0);//call back
			}
			break;
		case FD_CONNECT:
			break;
		case FD_CLOSE:
			{
				SOCKLOG * p, * find = NULL;
				psock = sock_client;
				while(psock)
				{
					if(psock->socket == wParam)
					{
						find = psock;
						if(psock == sock_client)
						{
							sock_client = psock->next;
							break;
						}
						else
						{
							p->next = psock->next;
							break;
						}
					}
					else
					{
						p = psock;
						psock = psock->next;
					}
				}
				if(find)
				{
					closesocket(find->socket);
					DealWithNetComm(2, find->ip, buf, 0);
					delete find;
				}
			}				
			break;
		default:
			break;
	}
}

void CNetworkDlg::OnUdpSockMessage(WPARAM wParam, LPARAM lParam)
{
	BYTE buf[2048];
	int len;
	struct sockaddr_in from;

	if(wParam != sock_udp) return;
	
	switch(WSAGETSELECTEVENT(lParam))
	{
		case FD_READ:
		{
			len = sizeof(from);
			if((len = recvfrom(sock_udp, (char *)buf, 2048, 0, (struct sockaddr *)&from, &len)) == SOCKET_ERROR)
				break;
			char name[50], ip[50];
			strcpy(ip, inet_ntoa(from.sin_addr));
			gethostname(name, 50);
			struct hostent * host = gethostbyname(name);
			if(from.sin_addr.s_addr == *((unsigned long *)host->h_addr)) break;
			DealWithNetComm(4, ip, buf, len);
		}
			break;
		case FD_WRITE:
			break;
		case FD_CLOSE:
			break;
		default:
			break;
	}
}

void CNetworkDlg::OnConnectSockMessage(WPARAM wParam, LPARAM lParam)
{
	BYTE buf[2048];
	int len;

	if(wParam != sock_connect) return;
	switch(WSAGETSELECTEVENT(lParam))
	{