学习win32网络编程

结构体列表：

• WSADATA
• SOCKET
• SOCKADDR_IN

函数列表：

• WSAStartup
• socket
• htonl
• htons
• bind
• listen
• accept
• send
• recv
• closesocket

代码：

#pragma comment(lib, "wsock32.lib")
#include <WinSock2.h>
#include <stdlib.h>
#include <windows.h>

/***
* @author:zanzan101
*/

static HANDLE event_server = CreateEvent(0, 0, 0, "server_event");
static HANDLE event_client = CreateEvent(0, 0, 0, "client_event");
static HANDLE event_end = CreateEvent(0, 0, 0, "end_event");

class Thread
{
private:
    HANDLE thread;
public:
    Thread(int(*func)(int* para), void* para)    // 故意简化了参数，但这并不是一个规范的做法
    {
        thread = NULL;

        // 创建一个挂起的进程
        thread = CreateThread(0, 0, reinterpret_cast<DWORD(CALLBACK*)(void*)>(func), para, CREATE_SUSPENDED, 0);
    }
    ~Thread()
    {
        CloseHandle(thread);
    }
    void resume()
    {
        ResumeThread(thread);
    }
    void suspend()
    {
        SuspendThread(thread);
    }
    void terminate()
    {
        TerminateThread(thread, 0);
    }
    
};

int server_func(int* para)
{
    int wsa_res = 0;
    // 打开
    WSADATA wsa_data;
    if(!(wsa_res = WSAStartup(0x0101, &wsa_data)))
        printf("server_%d	打开成功!
", *para);
    else
    {
        printf("server_%d	ERROR:%d
", *para, wsa_res - WSABASEERR);
        return -1;
    }

    // 创建套接字
    SOCKET socket_server = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
    SOCKADDR_IN addr_server;
    SOCKADDR_IN addr_client;

    addr_server.sin_addr.S_un.S_addr=htonl(INADDR_ANY);
    addr_server.sin_family=AF_INET;
    addr_server.sin_port=htons(6666);

    // 将套接字与端口号绑定
    bind(socket_server, (SOCKADDR*)&addr_server, sizeof(SOCKADDR));

    // 监听端口
    listen(socket_server, SOMAXCONN);


    int len = sizeof(SOCKADDR_IN);

    printf("server 准备完毕...
");
    // 设置信号，通知client
    SetEvent(event_server);

    printf("server在等待client...
");


    while(WaitForSingleObject(event_client, 0) != 0)
    {
        char recv_buf[100];
        char send_buf[100];

        //////////////////////////////////////////////////////////////////////////

        SOCKET socket_connect=accept(socket_server,(SOCKADDR*)&addr_client,&len);
        int bytes = recv(socket_connect, recv_buf, 100, 0);
        recv_buf[bytes] = 0;
        int num = atoi(recv_buf);
        if(bytes > 0 && -1 != num)
            printf("server收到：%d
",num);
        else if(-1 != num)
            printf("server什么也没收到
");
        if(-1 != num)
        {
            sprintf(send_buf,"%d的平方是：%d
",num, num*num);
            send(socket_connect,send_buf,strlen(send_buf)+1,0);
        }
        closesocket(socket_connect);

        //////////////////////////////////////////////////////////////////////////
    }

    if(!(wsa_res = WSACleanup()))
        printf("server_%d	关闭成功!
", *para);
    else
    {
        printf("server_%d	ERROR:%d
", *para, wsa_res - WSABASEERR);
        return -1;
    }
    SetEvent(event_end);
    return 0;
}

int client_func(int* para)
{
    printf("client在等待server运行...
");
    // 等待server运行
    while(WaitForSingleObject(event_server, 0) != 0)
        Sleep(100);// 100ms的响应时间

    int wsa_res = 0;
    WSADATA wsa_data;
    if(!(wsa_res = WSAStartup(0x0101, &wsa_data)))
        printf("client_%d	打开成功!
", *para);
    else
    {
        printf("client_%d	ERROR:%d
", *para, wsa_res - WSABASEERR);
        SetEvent(event_client);
        return -1;
    }
    printf("client正在运行...
");

    while(true)
    {
        int num;
        scanf("%d", &num);

        if(-1 == num)
        {
            printf("client结束...
");
            SetEvent(event_client);
        }

        char recv_buf[100];
        char send_buf[100];

        itoa(num, send_buf, 10);

        //////////////////////////////////////////////////////////////////////////

        SOCKET socket_client=socket(AF_INET,SOCK_STREAM,0);
        SOCKADDR_IN addr_server;
        addr_server.sin_addr.S_un.S_addr=inet_addr("127.0.0.1");
        addr_server.sin_family=AF_INET;
        addr_server.sin_port=htons(6666);
        connect(socket_client,(SOCKADDR*)&addr_server,sizeof(SOCKADDR));
        send(socket_client, send_buf,strlen(send_buf), 0);
        if(0 < recv(socket_client,recv_buf,100,0) && -1 != num)
            printf("client收到：%s
",recv_buf);
        else if( -1 != num)
            printf("client什么也没有收到。。。
");

        closesocket(socket_client);

        if(-1 == num)
            break;

        //////////////////////////////////////////////////////////////////////////
    }
    if(!(wsa_res = WSACleanup()))
        printf("client_%d	关闭成功!
", *para);
    else
    {
        printf("client_%d	ERROR:%d
", *para, wsa_res - WSABASEERR);        
        SetEvent(event_client);

        return -1;
    }        

    return 0;
}

int main()
{
    int i = 0;
    int j = 1;
    
    // 重置信号，默认状态下不需要重置
    //ResetEvent(event_server);
    //ResetEvent(event_client);
    //ResetEvent(event_end);

    Thread server_thread(server_func, &i);
    Thread client_thread(client_func, &j);
    server_thread.resume();    
    client_thread.resume();
    
    // 等待server结束
    while(WaitForSingleObject(event_end, 0) != 0)
        Sleep(100);// 100ms的响应时间
    printf("全部结束！
");

    server_thread.terminate();
    client_thread.terminate();
    
    system("pause");
    return 0;
}

输出结果：

client在等待server运行...
server_0        打开成功!
server 准备完毕...
server在等待client...
client_1        打开成功!
client正在运行...
2
server收到：2
client收到：2的平方是：4

32
server收到：32
client收到：32的平方是：1024

-1
client结束...
server_0        关闭成功!
client_1        关闭成功!
全部结束！
请按任意键继续. . .