基于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;
}