match numeric protocol valuesstun

1 files changed, 6 insertions, 2 deletions

diff --git a/src/fw.rs b/src/fw.rs
index abb6e20..a3e6dec 100644
--- a/src/fw.rs
+++ b/src/fw.rs
@@ -62,10 +62,12 @@ pub fn get_opened_ports(ipt: &iptables::IPTables) -> Result<messages::PublicExpo
                     let proto = String::from(raw_proto.as_str());
                     let number = String::from(raw_port.as_str()).parse::<u16>()?;
 
-                    if proto == "tcp" {
+                    if proto == "tcp" || proto == "6" {
                         ports.tcp_ports.insert(number);
-                    } else {
+                    } else if proto == "udp" || proto == "17" {
                         ports.udp_ports.insert(number);
+                    } else {
+                        error!("Unexpected protocol in iptables rule: {}", proto);
                     }
                 } else {
                     error!("Unexpected rule found in DIPLONAT chain")
@@ -77,6 +79,8 @@ pub fn get_opened_ports(ipt: &iptables::IPTables) -> Result<messages::PublicExpo
         }
     }
 
+    debug!("{} ports already openned: {:?}", ipt.cmd, ports);
+
     Ok(ports)
 }