netstat 0.7.0

//! Cross-platform library to retrieve network sockets information.
//! Tries to be optimal by using low-level OS APIs instead of command line utilities.
//! Provides unified interface and returns data structures which may have additional fields depending on platform.

#![allow(non_camel_case_types)]

#[macro_use]
extern crate bitflags;
extern crate libc;

mod integrations;
mod types;
mod utils;

pub use integrations::*;
pub use types::*;

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn result_is_ok_for_any_flags() {
        let af_flags_combs = (0..AddressFamilyFlags::all().bits() + 1)
            .filter_map(|x| AddressFamilyFlags::from_bits(x))
            .collect::<Vec<AddressFamilyFlags>>();
        let proto_flags_combs = (0..ProtocolFlags::all().bits() + 1)
            .filter_map(|x| ProtocolFlags::from_bits(x))
            .collect::<Vec<ProtocolFlags>>();
        for af_flags in af_flags_combs.iter() {
            for proto_flags in proto_flags_combs.iter() {
                assert!(get_sockets_info(*af_flags, *proto_flags).is_ok());
            }
        }
    }

    #[test]
    fn result_is_empty_for_empty_flags() {
        let sockets_info =
            get_sockets_info(AddressFamilyFlags::empty(), ProtocolFlags::empty()).unwrap();
        assert!(sockets_info.len() == 0);
    }

    #[test]
    fn api_is_consistent() {
        let af_flags = AddressFamilyFlags::all();
        let proto_flags = ProtocolFlags::all();
        assert_eq!(
            Result_(get_sockets_info(af_flags, proto_flags)),
            Result_(flatten_result(
                iterate_sockets_info(af_flags, proto_flags).map(|r| r.collect())
            ))
        );

        #[derive(Debug)]
        struct Result_(Result<Vec<SocketInfo>, Error>);

        impl PartialEq<Result_> for Result_ {
            fn eq(&self, other: &Result_) -> bool {
                match (&self.0, &other.0) {
                    (Ok(a), Ok(b)) => a.len() == b.len(),
                    _ => false,
                }
            }
        }

        fn flatten_result<T, E>(r: Result<Result<T, E>, E>) -> Result<T, E> {
            match r {
                Ok(Ok(x)) => Ok(x),
                Ok(Err(e)) => Err(e),
                Err(e) => Err(e),
            }
        }
    }
}