Windows 环境下的 Socket 编程 4 - 基于 UDP 的服务器/客户端

在 TCP 中，套接字之间应该是一对一的关系。若向 10 个客户端提供服务，则除了守门的服务器套接字外，还需要 10 个服务器端套接字。但在 UDP 中，不管是服务器端还是客户端都只需要 1 个套接字。

基于 UDP 的数据 I/O 函数

UDP 套接字不会保持连接状态，因此每次传输数据都要添加目标地址信息。

1. sendto 函数

//发送数据函数，成功返回传输的字节数，失败时返回 -1
int sendto(SOCKET s,const char *buf,int len,int flags,const struct sockaddr *to,int tolen);

• s：套接字
• buf：指向待传输数据的缓冲区
• len：待传输数据的长度
• flags：可选参数，若没有则传递 0
• to：指向包含目的地址信息的 sockaddr 结构体变量
• tolen：变量 to 的长度

2. recvfrom 函数

//接收数据函数，成功时返回接收的字节数，失败时返回 -1
int WSAAPI recvfrom(SOCKET s,char *buf,int len,int flags,struct sockaddr *from,int *fromlen);

• s：套接字
• buff：指向保存接收数据的缓冲区
• len：可接收的最大字节数
• flags：可选参数，若没有则传入 0
• from：指向包含发送端地址信息的 sockaddr 结构体变量
• fromlen：指向保存变量 from 长度的变量

什么时候调用 recvfrom ，当操作系统接收到 UDP 数据， 会自动通知（回调）应用程序吗？
如果 len 很小，对方又发送大数据包，如何知道接收缓冲区的数据大小，以方便多次读取？
如果 len 很大，recvfrom 函数会一直等待吗？

基于 UDP 的回声服务器端/客户端

UDP 不同于 TCP，不存在请求连接和受理过程，因此在某种意义上无法明确区分服务器端和客户端。只是因其提供服务而称为服务器端。
服务器端实现：

#include <stdio.h>
#include <stdlib.h>
#include <winsock2.h>

void ErrorHandler(char *message);

#define BUF_SIZE 1024
int main(int argc, char *argv[])
{
    WSADATA wsaData;
    SOCKET hServSock, hClntSock;
    struct sockaddr_in servAddr, clntAddr;
    int szClntAddr;
    char message[BUF_SIZE];
    int strLen, i;
    if(argc != 2)
    {
        printf ("Usage: %s <port>\n", argv[0]);
        exit(1);
    }

    if(WSAStartup(MAKEWORD(2,2), &wsaData) != 0)
        ErrorHandler("WSAStartup failed");

    hServSock = socket(PF_INET, SOCK_DGRAM, 0);
    if(hServSock == INVALID_SOCKET)
        ErrorHandler("socket error");
    
    memset(&servAddr, 0, sizeof(servAddr));
    servAddr.sin_family = AF_INET;
    servAddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servAddr.sin_port = htons(atoi(argv[1]));

    if(bind(hServSock, (struct sockaddr *)&servAddr, sizeof(servAddr)) == SOCKET_ERROR)
        ErrorHandler("bind socket error");
    
    while(1)
    {
        strLen = recvfrom(hServSock, message, BUF_SIZE, 0, (struct sockaddr *)&clntAddr, &szClntAddr);
        sendto(hServSock, message, strLen, 0, (struct sockaddr *)&clntAddr, sizeof(clntAddr));
    }
    closesocket(hServSock);
    WSACleanup();
    return 0;
}

void ErrorHandler(char *message)
{
    fputs(message, stderr);
    fputc('\n', stderr);
    exit(1);
}

客户端实现：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <winsock2.h>

#define BUF_SIZE    1024
void ErrorHandler(char *message);

int main(int argc, char *argv[])
{
    WSADATA wsaData;
    SOCKET hClntSock;
    struct sockaddr_in servAddr,fromAddr;
    char message[BUF_SIZE];
    int strLen;
    int adr_sz;
    if(argc != 3)
    {
        printf ("Usage: %s <IP> <port>\n", argv[0]);
        exit(1);
    }

    if(WSAStartup(MAKEWORD(2,2), &wsaData) != 0)
        ErrorHandler("WSAStartup failed");

    hClntSock = socket(PF_INET, SOCK_DGRAM, 0);
    if(hClntSock == INVALID_SOCKET)
        ErrorHandler("socket error");
    
    memset(&servAddr, 0, sizeof(servAddr));
    servAddr.sin_family = AF_INET;
    servAddr.sin_addr.s_addr = inet_addr(argv[1]);
    servAddr.sin_port = htons(atoi(argv[2]));


    while(1)
    {
        fputs("Input message(Q to quit): ",stdout);
        fgets(message, BUF_SIZE, stdin);

        if(!strcmp(message, "q\n") ||!strcmp(message, "Q\n"))
            break;

        sendto(hClntSock, message, strlen(message), 0, (struct sockaddr *)&servAddr, sizeof(servAddr));
        strLen = recvfrom(hClntSock, message, BUF_SIZE, 0, (struct sockaddr *)&fromAddr, &adr_sz);
        message[strLen] = 0;
        printf("Message from server: %s", message);
    }
    
    closesocket(hClntSock);
    WSACleanup();
    return 0;
}

void ErrorHandler(char *message)
{
    fputs(message, stderr);
    fputc('\n', stderr);
    exit(1);
}

UDP 套接字存在数据边界

TCP 数据传输不存在边界，这表示数据传输过程中调用 I/O 函数的次数不具有任何意义。但是，UDP 是具有数据边界的协议，传输中调用 I/O 函数的次数非常重要。调用 3 次 sendto 发送的函数要调用 3 次 recvfrom 函数才能接收完。

已连接的 UDP 套接字

使用 sendto 函数，每次都要传入目标地址信息，这在某些场合会低效且繁琐，已连接的 UDP 套接字可以解决这个问题。

1. connect 函数：创建已连接的 UDP 套接字

/*成功返回 0 ，失败返回 SOCKET_ERROR */
int connect(SOCKET s, const struct sockaddr *name, int namelen);

这个函数参数与 TCP 相同，但是结果不同，TCP 调用这个函数会进行 3 次握手，UDP 调用这个函数只是注册目标 IP 和端口信息。
一旦调用 connect 函数后，就不必使用 sendto 和 recvfrom 函数发送和接收数据，取而代之的是和 TCP 相同的 API 函数：send 和 recv 。