move own_address to util module

1 files changed, 7 insertions, 0 deletions

diff --git a/src/util.rs b/src/util.rs
index d2143f8..81e4fce 100644
--- a/src/util.rs
+++ b/src/util.rs
@@ -1,4 +1,6 @@
 use crate::error::{Error, Result};
+use socket2::Socket;
+use std::net::Ipv4Addr;
 
 pub fn format_client_id(client_id: &[u8]) -> Result<String> {
     client_id
@@ -7,3 +9,8 @@ pub fn format_client_id(client_id: &[u8]) -> Result<String> {
         .reduce(|acc, octet| acc + ":" + &octet)
         .ok_or(Error::EmptyClientId)
 }
+
+pub fn local_ip(sock: &Socket) -> Ipv4Addr {
+    let local_addr = sock.local_addr().unwrap().as_socket_ipv4().unwrap();
+    *local_addr.ip()
+}