Allow the user to specify their own socket

This is useful when dealing with multiple network/user namespaces.

Requires rust-netlink/netlink-proto#25

Cargo.toml

@@ -36,3 +36,6 @@ ipnetwork = "0.18.0"
 tokio = { version = "1.0.1", features = ["macros", "rt", "rt-multi-thread"] }
 async-std = { version = "1.9.0", features = ["attributes"]}
 macaddr = "1.0"
+
+[patch.crates-io]
+netlink-proto = {path = "../netlink-proto"}

src/connection.rs

@@ -33,3 +33,19 @@ where
         netlink_proto::new_connection_with_socket(NETLINK_ROUTE)?;
     Ok((conn, Handle::new(handle), messages))
 }
+
+#[allow(clippy::type_complexity)]
+pub fn from_socket<S>(
+    socket: S,
+) -> (
+    Connection<RouteNetlinkMessage, S>,
+    Handle,
+    UnboundedReceiver<(NetlinkMessage<RouteNetlinkMessage>, SocketAddr)>,
+)
+where
+    S: AsyncSocket,
+{
+    let (conn, handle, messages) =
+        netlink_proto::from_socket_with_codec(socket);
+    (conn, Handle::new(handle), messages)
+}

src/lib.rs

@@ -29,6 +29,7 @@ mod traffic_control;
 pub use crate::addr::{
     AddressAddRequest, AddressDelRequest, AddressGetRequest, AddressHandle,
 };
+pub use crate::connection::from_socket;
 #[cfg(feature = "tokio_socket")]
 pub use crate::connection::new_connection;
 pub use crate::connection::new_connection_with_socket;