Merge pull request #5 from chengyuhui/linux-uid

Add UID to Linux and Android targets, along with some convenience functions

Author: ohadravid

src/integrations/linux/netlink_iterator.rs

@@ -153,6 +153,7 @@ unsafe fn parse_diag_msg(
             }),
             associated_pids: Vec::with_capacity(0),
             inode: diag_msg.inode,
+            uid: diag_msg.uid,
         },
         IPPROTO_UDP => SocketInfo {
             protocol_socket_info: ProtocolSocketInfo::Udp(UdpSocketInfo {
@@ -161,6 +162,7 @@ unsafe fn parse_diag_msg(
             }),
             associated_pids: Vec::with_capacity(0),
             inode: diag_msg.inode,
+            uid: diag_msg.uid,
         },
         _ => return Err(Error::UnknownProtocol(protocol)),
     };

src/types/socket_info.rs

@@ -9,7 +9,11 @@ pub struct SocketInfo {
     /// Identifiers of processes associated with this socket.
     pub associated_pids: Vec<u32>,
     #[cfg(any(target_os = "linux", target_os = "android"))]
+    /// Inode number of this socket.
     pub inode: u32,
+    #[cfg(any(target_os = "linux", target_os = "android"))]
+    /// Owner UID of this socket.
+    pub uid: u32,
 }
 
 /// Protocol-specific socket information.
@@ -37,3 +41,21 @@ pub struct UdpSocketInfo {
     pub local_addr: IpAddr,
     pub local_port: u16,
 }
+
+impl SocketInfo {
+    /// Local address of this socket.
+    pub fn local_addr(&self) -> IpAddr {
+        match &self.protocol_socket_info {
+            ProtocolSocketInfo::Tcp(s) => s.local_addr,
+            ProtocolSocketInfo::Udp(s) => s.local_addr,
+        }
+    }
+
+    /// Local port of this socket.
+    pub fn local_port(&self) -> u16 {
+        match &self.protocol_socket_info {
+            ProtocolSocketInfo::Tcp(s) => s.local_port,
+            ProtocolSocketInfo::Udp(s) => s.local_port,
+        }
+    }
+}