linux网络编程二十六：多进程编程之消息队列

最新推荐文章于 2023-12-07 07:00:00 发布

jasonliuvip

最新推荐文章于 2023-12-07 07:00:00 发布

阅读量1.2k

点赞数

分类专栏： linux网络编程文章标签： linux 网络编程多进程消息队列服务器

本文链接：https://blog.csdn.net/jasonliuvip/article/details/25623057

版权

linux网络编程专栏收录该内容

29 篇文章 19 订阅

订阅专栏

1、把子进程中打开的文件描述符传递给父进程。

#include <sys/socket.h>
#include <fcntl.h>
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <assert.h>
#include <string.h>

//CMSG_LEN()宏接受我们希望放置在附属数据缓冲区中的对象尺寸作为输入参数。
//这个宏会计算cmsghdr头结构加上所需要的填充字符的字节长度。
//这个值用来设置cmsghdr对象的cmsg_len成员。
static const int CONTROL_LEN = CMSG_LEN(sizeof(int));

void send_fd(int fd, int fd_to_send);
int recv_fd(int fd);


int main()
{
    int pipefd[2];
    int fd_to_pass = 0;

    int ret = socketpair(PF_UNIX, SOCK_DGRAM, 0, pipefd);
    assert(ret != -1);

    pid_t pid = fork();
    assert(pid >=0);

    if (pid == 0) {
        close(pipefd[0]);
        fd_to_pass = open("./test.txt", O_RDWR, 0666);

        send_fd(pipefd[1], (fd_to_pass > 0) ? fd_to_pass : 0);

        close(fd_to_pass);
        exit(0);
    }

    close(pipefd[1]);
    fd_to_pass = recv_fd(pipefd[0]);

    char buf[1024];
    memset(buf, '\0', sizeof(buf));

    read(fd_to_pass, buf, 1024);

    printf("I got fd %d and data %s\n", fd_to_pass, buf);

    close(fd_to_pass);


    return 0;
}

//CMSG_DATA()宏接受一个指向cmsghdr结构的指针。
//返回的指针值指向跟随在头部以及填充字节之后的附属数据的第一个字节(如果存在),
//指向一个描述文件描述符的可用的附属数据信息头部
void send_fd(int fd, int fd_to_send)
{
    struct iovec iov[1];
    struct msghdr msg;
    char buf[0];

    iov[0].iov_base = buf;
    iov[0].iov_len = 1;

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

    cmsghdr cm;
    cm.cmsg_len = CONTROL_LEN;
    cm.cmsg_level = SOL_SOCKET;
    cm.cmsg_type = SCM_RIGHTS;

    *(int*)CMSG_DATA(&cm) = fd_to_send;

    msg.msg_control = &cm;
    msg.msg_controllen = CONTROL_LEN;

    sendmsg(fd, &msg, 0);
}

int recv_fd(int fd)
{
    struct iovec iov[1];
    struct msghdr msg;
    char buf[0];

    iov[0].iov_base = buf;
    iov[0].iov_len = 1;

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

    cmsghdr cm;
    msg.msg_control = &cm;
    msg.msg_controllen = CONTROL_LEN;

    recvmsg(fd, &msg, 0);

    int fd_to_read = *(int*)CMSG_DATA(&cm);

    return fd_to_read;
}

参考：《linux高性能服务器编程》