基于Win32的多线程客户/服务器通信

客户端：


// Client.cpp : Defines the entry point for the application.


//




#include "stdafx.h"


#include "resource.h"


#include <winsock.h>




#pragma warning(disable:4700)




#define MAX_LOADSTRING 100




// Global Variables:


HINSTANCE hInst; // current instance


TCHAR szTitle[MAX_LOADSTRING]; // The title bar text


TCHAR szWindowClass[MAX_LOADSTRING]; // The title bar text




// Foward declarations of functions included in this code module:


ATOM MyRegisterClass(HINSTANCE hInstance);


BOOL InitInstance(HINSTANCE, int);


LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);


LRESULT CALLBACK About(HWND, UINT, WPARAM, LPARAM);




int APIENTRY WinMain(HINSTANCE hInstance,


HINSTANCE hPrevInstance,


LPSTR lpCmdLine,


int nCmdShow)






{


// TODO: Place code here.


MSG msg;


HACCEL hAccelTable;




// Initialize global strings


LoadString(hInstance, IDS_APP_TITLE, szTitle, MAX_LOADSTRING);


LoadString(hInstance, IDC_CLIENT, szWindowClass, MAX_LOADSTRING);


MyRegisterClass(hInstance);




// Perform application initialization:


if (!InitInstance (hInstance, nCmdShow))






{


return FALSE;


}




hAccelTable = LoadAccelerators(hInstance, (LPCTSTR)IDC_CLIENT);




// Main message loop:


while (GetMessage(&msg, NULL, 0, 0))






{


if (!TranslateAccelerator(msg.hwnd, hAccelTable, &msg))






{


TranslateMessage(&msg);


DispatchMessage(&msg);


}


}




return msg.wParam;


}




LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)






{


int wmId, wmEvent;


PAINTSTRUCT ps;


HDC hdc;


TCHAR szHello[MAX_LOADSTRING];


LoadString(hInst, IDS_HELLO, szHello, MAX_LOADSTRING);




WORD version; //used to store the socket version


WSADATA wsaData; //used to store info about socket


SOCKET theSocket; //used to create client socket


SOCKADDR_IN serverInfo; //used to specify a local or remote endpoint address to which to connect a socket


int rVal; //used to store return value


LPHOSTENT hostEntry; //Windows Sockets allocates this structure


char *buf = "Data On Socket"; //Data to be send on socket




char data[5];


switch (message)






{


case WM_COMMAND:


wmId = LOWORD(wParam);


wmEvent = HIWORD(wParam);


// Parse the menu selections:


switch (wmId)






{


case IDM_ABOUT:


DialogBox(hInst, (LPCTSTR)IDD_ABOUTBOX, hWnd, (DLGPROC)About);


break;


case IDM_EXIT:


DestroyWindow(hWnd);


break;


case IDM_COMMUNICATE:


//get the version of socket


version = MAKEWORD(1,1);


//The Windows Sockets WSAStartup function initiates use of WS2_32.DLL by a process


rVal = WSAStartup(version,(LPWSADATA)&wsaData);


//store information about the server


//Here we are suppose to pass the server information to the client,


//so that the client knows where is the server


//I am using the machine name on wich my server.exe is running


hostEntry = gethostbyname("seclore3");


if(!hostEntry)






{


MessageBox(hWnd,"Failed in gethostbyname API","Gethostbyname Error",MB_OK);


WSACleanup();


break;


}


//here we are creating the socket


theSocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);


if(theSocket == SOCKET_ERROR)






{


MessageBox(hWnd,"Failed in socket API","Socket Error",MB_OK);


}


//Fill in the sockaddr_in struct


serverInfo.sin_family = PF_INET;


serverInfo.sin_addr = *((LPIN_ADDR)*hostEntry->h_addr_list);


serverInfo.sin_port = htons(8888);


//Connect to the socket we just created


rVal=connect(theSocket,(LPSOCKADDR)&serverInfo, sizeof(serverInfo));


if(rVal==SOCKET_ERROR)






{


MessageBox(hWnd,"Failed in connect API","Connect Error",MB_OK);


break;


}


//Send the data to the server


rVal = send(theSocket, buf, strlen(buf), 0);


if(rVal == SOCKET_ERROR)






{


MessageBox(hWnd,"Failed send","Send Error",MB_OK);


}


//Get/Recieve the data sent by the server


rVal = recv(theSocket,data,5,0);


if(rVal)






{


MessageBox(hWnd,data,"Data from server",MB_OK);


}


break;


default:


return DefWindowProc(hWnd, message, wParam, lParam);


}


break;


case WM_PAINT:


hdc = BeginPaint(hWnd, &ps);


// TODO: Add any drawing code here



RECT rt;


GetClientRect(hWnd, &rt);


DrawText(hdc, szHello, strlen(szHello), &rt, DT_CENTER);


EndPaint(hWnd, &ps);


break;


case WM_DESTROY:


//Close the socket


closesocket(theSocket);


//The WSACleanup function initiates no action


WSACleanup();


PostQuitMessage(0);


break;


default:


return DefWindowProc(hWnd, message, wParam, lParam);


}


return 0;


}



服务器端：


// Server.cpp : Defines the entry point for the application.


//




#include "stdafx.h"


#include "resource.h"


#include <winsock.h>




//function declaration


DWORD WINAPI ValidateData(LPVOID Parameter);




int APIENTRY WinMain(HINSTANCE hInstance,


HINSTANCE hPrevInstance,


LPSTR lpCmdLine,


int nCmdShow)






{


WORD sockVersion; //Used to store socket version information


WSADATA wsaData; //used to store info about socket


SOCKET s,client; //used to create client and server socket


SOCKADDR_IN sin; //used to specify a local or remote endpoint address to which to connect a socket


int rVal; //used to store return value




HANDLE hThread; //Handle to thread


DWORD ThreadId; //used to store the thread id




//Get the current socket version


sockVersion = MAKEWORD(1,1);


//初始化Socket库


//The Windows Sockets WSAStartup function initiates use of WS2_32.DLL by a process


WSAStartup(sockVersion, &wsaData);


//here we are creating the socket


s = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);


if(s == INVALID_SOCKET)






{


MessageBox(0,"Invalid socket","Socket Error",MB_OK);


WSACleanup();


}


//fill in sockaddr_in struct


sin.sin_family = PF_INET;


sin.sin_port = htons(8888);//监听端口为


sin.sin_addr.s_addr = INADDR_ANY;




//bind to the socket


rVal = bind(s, (LPSOCKADDR)&sin, sizeof(sin));


if(rVal == SOCKET_ERROR)






{


MessageBox(0,"Failed in bind API","Bind Error",MB_OK);


WSACleanup();


}




//Listen to the desire port on which client suppose to connect


rVal = listen(s, 2);//最大客户数目为


if(rVal == SOCKET_ERROR)






{


MessageBox(0,"Failed in listen API","Listen Error",MB_OK);


WSACleanup();


}


//infinite loop to serve all the clients which want service


while(1)






{


//Accept the data from the client


client = accept(s, NULL, NULL);//接受来自客户端的连接




//Once the new client come up create a new thread t serve the client


if(client)






{//创建线程去处理此请求,将此连接作为参数传给处理线程


hThread = CreateThread(NULL,


0,


ValidateData,


(LPVOID)client,


0,


&ThreadId);


}


else


return 0;


}


return 0;


}




DWORD WINAPI ValidateData(LPVOID Parameter)






{


//Get the information about client entity


SOCKET client = (SOCKET)Parameter;


int rVal; //Return val


//数据缓冲区


char buf[11]; //used to send the validated data to client




//Get the data form client


//接收数据


rVal = recv(client,buf,11,0);


//here we are performing simple check, the data came form client


//is valid or not


//at this point you can check your own data also, which needs some modification


//回传数据给客户端


if(strcmp(buf,"Data On Socket"))






{


//Send back the data to the client


rVal = send(client,"YES",3,0);


}


else






{


//Send back the data to the client


rVal = send(client,"NO",2,0);


}


return 0;


}