Function test_unix_peercred

tests/net/unix_alloc.rs:631–685

()

#[cfg(all(feature = "process", linux_kernel))]
#[test]
fn test_unix_peercred() {
    crate::init();

    use rustix::io::{IoSlice, IoSliceMut};
    use rustix::net::{
        recvmsg, sendmsg, sockopt, RecvAncillaryBuffer, RecvAncillaryMessage, RecvFlags,
        SendAncillaryBuffer, SendAncillaryMessage, SendFlags, SocketFlags,
    };
    use rustix::process::{getgid, getpid, getuid};

    let (send_sock, recv_sock) = rustix::net::socketpair(
        AddressFamily::UNIX,
        SocketType::STREAM,
        SocketFlags::CLOEXEC,
        None,
    )
    .unwrap();

    sockopt::set_socket_passcred(&recv_sock, true).unwrap();

    let ucred = sockopt::socket_peercred(&send_sock).unwrap();
    assert_eq!(ucred.pid, getpid());
    assert_eq!(ucred.uid, getuid());
    assert_eq!(ucred.gid, getgid());

    let msg = SendAncillaryMessage::ScmCredentials(ucred);
    let mut space = Vec::with_capacity(msg.size());
    let mut cmsg_buffer = SendAncillaryBuffer::new(space.spare_capacity_mut());
    assert!(cmsg_buffer.push(msg));

    sendmsg(
        &send_sock,
        &[IoSlice::new(b"cred")],
        &mut cmsg_buffer,
        SendFlags::empty(),
    )
    .unwrap();

    let mut cmsg_space = Vec::with_capacity(rustix::cmsg_space!(ScmCredentials(1)));
    let mut cmsg_buffer = RecvAncillaryBuffer::new(cmsg_space.spare_capacity_mut());

    let mut buffer = vec![0; BUFFER_SIZE];
    recvmsg(
        &recv_sock,
        &mut [IoSliceMut::new(&mut buffer)],
        &mut cmsg_buffer,
        RecvFlags::empty(),
    )
    .unwrap();

    match cmsg_buffer.drain().next().unwrap() {
        RecvAncillaryMessage::ScmCredentials(ucred2) => assert_eq!(ucred2, ucred),
        _ => panic!("Unexpected ancillary message"),
    };
}

/// Like `test_unix_msg_with_scm_rights`, but with multiple file descriptors
/// over multiple control messages.