use std::process;
use netlink_packet_audit::{AuditMessage, StatusMessage};
use netlink_packet_core::{
NetlinkBuffer, NetlinkMessage, NLM_F_ACK, NLM_F_REQUEST,
};
use netlink_sys::{
protocols::NETLINK_AUDIT, AsyncSocket, AsyncSocketExt, SmolSocket,
SocketAddr,
};
const AUDIT_STATUS_ENABLED: u32 = 1;
const AUDIT_STATUS_PID: u32 = 4;
#[async_std::main]
async fn main() {
let kernel_unicast: SocketAddr = SocketAddr::new(0, 0);
let socket = SmolSocket::new(NETLINK_AUDIT).unwrap();
let mut status = StatusMessage::new();
status.enabled = 1;
status.pid = process::id();
status.mask = AUDIT_STATUS_ENABLED | AUDIT_STATUS_PID;
let payload = AuditMessage::SetStatus(status);
let mut nl_msg = NetlinkMessage::from(payload);
nl_msg.header.flags = NLM_F_REQUEST | NLM_F_ACK;
nl_msg.finalize();
let mut buf = vec![0; 1024 * 8];
nl_msg.serialize(&mut buf[..nl_msg.buffer_len()]);
println!(">>> {:?}", nl_msg);
socket
.send_to(&buf[..nl_msg.buffer_len()], &kernel_unicast)
.await
.unwrap();
let mut buf = bytes::BytesMut::with_capacity(1024 * 8);
loop {
buf.clear();
let _addr = socket.recv_from(&mut buf).await.unwrap();
{
let n = buf.len();
let mut nl_buf = NetlinkBuffer::new(&mut buf);
if n != nl_buf.length() as usize {
nl_buf.set_length(n as u32);
}
}
let parsed = NetlinkMessage::<AuditMessage>::deserialize(&buf).unwrap();
println!("<<< {:?}", parsed);
}
}