Function test_unix_peercred_implicit

tests/net/unix.rs:692–745

()

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

    use rustix::io::{IoSlice, IoSliceMut};
    use rustix::net::{
        recvmsg, sendmsg, sockopt, RecvAncillaryBuffer, RecvAncillaryMessage, RecvFlags,
        ReturnFlags, SendAncillaryBuffer, 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 mut cmsg_buffer = SendAncillaryBuffer::default();

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

    let mut cmsg_space = [MaybeUninit::uninit(); rustix::cmsg_space!(ScmCredentials(1))];
    let mut cmsg_buffer = RecvAncillaryBuffer::new(cmsg_space.as_mut_slice());

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

    assert_eq!(result.flags, ReturnFlags::empty());

    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.