Function test_unix_peercred_explicit

tests/net/unix.rs:634–685

()

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

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

    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();
    let msg = SendAncillaryMessage::ScmCredentials(ucred);
    let mut space = [MaybeUninit::uninit(); rustix::cmsg_space!(ScmCredentials(1))];
    let mut cmsg_buffer = SendAncillaryBuffer::new(space.as_mut_slice());
    assert!(cmsg_buffer.push(msg));

    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_peercred_explicit`, but relies on the fact that
/// `set_socket_passcred` enables passing of the credentials implicitly