use nix::sys::socket::InetAddr;
use std::{
    io::{self, Read, Write},
    net::{TcpListener, TcpStream},
    os::unix::io::{AsRawFd, FromRawFd},
};
use iou::sqe::SockAddr;

const MESSAGE: &'static [u8] = b"Hello World";

#[test]
#[ignore] //  kernel 5.5 needed for accept
fn accept() -> io::Result<()> {
    let mut ring = iou::IoUring::new(1)?;

    let listener = TcpListener::bind(("0.0.0.0", 0))?;
    listener.set_nonblocking(true)?;

    let mut stream = TcpStream::connect(listener.local_addr()?)?;
    stream.write_all(MESSAGE)?;

    let fd = listener.as_raw_fd();
    let mut sq = ring.sq();
    let mut sqe = sq.prepare_sqe().expect("failed to get sqe");
    unsafe {
        sqe.prep_accept(fd, None, iou::sqe::SockFlag::empty());
        sq.submit()?;
    }
    let cqe = ring.wait_for_cqe()?;
    let accept_fd = cqe.result()?;
    let mut accept_buf = [0; MESSAGE.len()];
    let mut stream = unsafe { TcpStream::from_raw_fd(accept_fd as _) };
    stream.read_exact(&mut accept_buf)?;
    assert_eq!(accept_buf, MESSAGE);
    Ok(())
}

#[test]
#[ignore] // kernel 5.5 needed for accept
fn accept_with_params() -> io::Result<()> {
    let mut ring = iou::IoUring::new(1)?;

    let listener = TcpListener::bind(("0.0.0.0", 0))?;
    listener.set_nonblocking(true)?;

    let mut connection_stream = TcpStream::connect(listener.local_addr()?)?;
    connection_stream.write_all(MESSAGE)?;

    let fd = listener.as_raw_fd();
    let mut sq = ring.sq();
    let mut sqe = sq.prepare_sqe().expect("failed to get sqe");
    let mut accept_params = iou::sqe::SockAddrStorage::uninit();
    unsafe {
        sqe.prep_accept(fd, Some(&mut accept_params), iou::sqe::SockFlag::empty());
        sq.submit()?;
    }
    let cqe = ring.wait_for_cqe()?;
    let accept_fd = cqe.result()?;
    let mut accept_buf = [0; MESSAGE.len()];
    let mut accepted_stream = unsafe { TcpStream::from_raw_fd(accept_fd as _) };
    accepted_stream.read_exact(&mut accept_buf)?;
    assert_eq!(accept_buf, MESSAGE);

    let addr = unsafe { accept_params.as_socket_addr()? };
    let connection_addr = SockAddr::Inet(InetAddr::from_std(&connection_stream.local_addr()?));
    assert_eq!(addr, connection_addr);
    Ok(())
}