Advanced Programming in UNIX Environment Episode 110

最新推荐文章于 2024-09-30 07:25:58 发布

duxingzhe103

最新推荐文章于 2024-09-30 07:25:58 发布

阅读量168

点赞数

分类专栏： Advanced Programming in the Unix Environment 文章标签： unix

本文链接：https://blog.csdn.net/myfather103/article/details/80495170

版权

Advanced Programming in the Unix Environment 专栏收录该内容

128 篇文章 7 订阅

订阅专栏

#include "aupe.h"
#include <sys/socket.h>

#if defined(SCM_CREDS)
#define CREDSTRUCT cmsgcred
#define SCM_CREDTYPE SCM_CREDS
#elif defined(SCM_CREDENTIALS)
#define CREDSTRUCT ucred
#define SCM_CREDTYPE SCM_CREDENTIALS
#else
#error passing credentials is unsupported!
#endif

#define RIGHTSLEN CMSG_LEN(sizeof(int))
#define CREDSLEN CMSG_LEN(sizeof(struct CREDSTRUCT))
#define CONTROLLEN (RIGHTSLEN+CREDSLEN)

static struct cmsghdr *cmptr=NULL;

int send_fd(int fd, int fd_to_send)
{
    struct CREDSTRUCT *credp;
    struct cmsghdr *cmp;
    struct iovec iov[1];
    struct msghdr msg;
    char buf[2];

    iov[0].iov_base=buf;
    iov[0].iov_len=2;
    msg.msg_iov=iov;
    msg.msg_iovlen=1;
    msg.msg_name=NULL;
    msg.msg_namelen=0;
    msg.msg_flags=0;
    if(fd_to_send<0)
    {
        msg.msg_control=NULL;
        msg.msg_controllen=0;
        buf[1]=-fd_to_send;
        if(buf[1]==0)
            buf[1]=1;
    }
    else
    {
        if(cmptr==NULL&&(cmptr=malloc(CONTROLLEN))==NULL)
            return -1;
        msg.msg_control=cmptr;
        msg.msg_controllen=CONTROLLEN;
        cmp=cmptr;
        cmp->cmsg_level=SOL_SOCKET;
        cmp->cmsg_type=SCM_RIGHTS;
        cmp->cmsg_len=RIGHTLEN;
        *(int *)CMSG_DATA(cmp)=fd_to_send;
        cmp->cmsg_level=SOL_SCOKET;
        cmp->cmsg_type=SCM_CREDTYPE;
        cmp->cmsg_len=CREDSLEN;
        credp=(struct CREDSTRUCT *)CMSG_DATA(cmp);
#if defined(SCM_CREDENTIALS)
        credp->uid=geteuid();
        credp->gid=getegid();
        credp->pid=getpid();
#endif
        buf[1]=0;
    }
    buf[0]=0;
    if(sendmsg(fd, &msg,0)!=2)
        return -1;
    return 0;
}

Sending credentials over UNIX domain sockets