深入了解recvmsg函數

一、recvmsg函數概述

recvmsg函數是操作系統提供的用於接收數據的系統調用之一，用於從一個打開的套接字上讀取數據報文。與recv函數不同的是，它可以接收完整的數據報，而不是像recv函數那樣需要多次調用獲得完整數據。

二、recvmsg函數的參數

recvmsg函數的參數與recv函數相比略有不同。下面是recvmsg函數的參數列表：

ssize_t recvmsg(int sockfd, struct msghdr *msg, int flags);

其中：

• sockfd：代表要讀取數據的套接字描述符
• *msg：代表一個指向msghdr結構的指針，用於存儲接收到的數據和有關數據的信息
• flags：可以是0或者MSG_DONTWAIT（非阻塞模式）

三、msghdr結構體

msghdr結構體由以下成員組成：

struct msghdr {
    void         *msg_name;       //協議地址
    socklen_t    msg_namelen;     //地址長度
    struct iovec *msg_iov;        //散列表指針
    size_t       msg_iovlen;      //散列表長度
    void         *msg_control;    //輔助數據
    size_t       msg_controllen;  //輔助數據長度
    int          msg_flags;       //msg_control中的控制信息類型
};

其中，重要的成員包括：

• msg_name和msg_namelen：指向存有協議地址的緩存和地址長度的變量
• msg_iov和msg_iovlen：指向一個iovec結構列表和列表長度的變量，用於存儲接收到的數據
• msg_control和msg_controllen：指向一個控制信息的緩存和緩存長度的變量，用於存儲附加數據
• msg_flags：用於存儲與msg_control相關的信息類型

四、recvmsg函數的返回值

與recv函數一樣，recvmsg函數的返回值為已讀取字節數。如果發生錯誤則返回-1。

五、示例代碼

下面是一個使用recvmsg函數從套接字中讀取數據並輸出到標準輸出的示例代碼：

#include 
#include 
#include 
#include 

int main(int argc, char *argv[]) {
    int sockfd;
    struct sockaddr_in servaddr;

    // 創建一個套接字
    sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (sockfd == -1) {
        perror("socket");
        return 1;
    }

    memset(&servaddr, 0, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port = htons(8080);

    // 將該套接字綁定到一個 IP 地址和端口號上    
    if (bind(sockfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) != 0) {
        perror("bind");
        return 1;
    }

    // 接收數據
    char buffer[1024];
    struct msghdr msg = {0};
    struct iovec iov[1];
    ssize_t len;

    iov[0].iov_base = buffer;
    iov[0].iov_len = sizeof(buffer);

    msg.msg_name = NULL;
    msg.msg_namelen = 0;
    msg.msg_iov = iov;
    msg.msg_iovlen = 1;
    msg.msg_control = NULL;
    msg.msg_controllen = 0;
    msg.msg_flags = 0;

    len = recvmsg(sockfd, &msg, 0);
    if (len == -1) {
        perror("recvmsg");
        return 1;
    }

    printf("%s", buffer);
    close(sockfd);

    return 0;
}