Add safe pton and ntop rules

Author: lucic71

rules/arpa_inet/tgt_refcount.rs

@@ -0,0 +1,60 @@
+// Copyright (c) 2022-present INESC-ID.
+// Distributed under the MIT license that can be found in the LICENSE file.
+
+use libcc2rs::*;
+
+fn f5(a0: i32, a1: Ptr<u8>, a2: AnyPtr) -> i32 {
+    if a0 == libc::AF_INET {
+        match a1.to_rust_string().parse::<std::net::Ipv4Addr>() {
+            Ok(__ip) => {
+                let __octets = __ip.octets();
+                for __i in 0..4 {
+                    a2.reinterpret_cast::<u8>().offset(__i).write(__octets[__i]);
+                }
+                1
+            }
+            Err(_) => 0,
+        }
+    } else if a0 == libc::AF_INET6 {
+        match a1.to_rust_string().parse::<std::net::Ipv6Addr>() {
+            Ok(__ip) => {
+                let __octets = __ip.octets();
+                for __i in 0..16 {
+                    a2.reinterpret_cast::<u8>().offset(__i).write(__octets[__i]);
+                }
+                1
+            }
+            Err(_) => 0,
+        }
+    } else {
+        -1
+    }
+}
+
+fn f6(a0: i32, a1: AnyPtr, a2: Ptr<u8>, a3: u32) -> Ptr<u8> {
+    let __text = if a0 == libc::AF_INET {
+        let mut __b = [0u8; 4];
+        for __i in 0..4 {
+            __b[__i] = a1.reinterpret_cast::<u8>().offset(__i).read();
+        }
+        Some(std::net::Ipv4Addr::from(__b).to_string())
+    } else if a0 == libc::AF_INET6 {
+        let mut __b = [0u8; 16];
+        for __i in 0..16 {
+            __b[__i] = a1.reinterpret_cast::<u8>().offset(__i).read();
+        }
+        Some(std::net::Ipv6Addr::from(__b).to_string())
+    } else {
+        None
+    };
+    match __text {
+        Some(__s) if (__s.len() as u32) < a3 => {
+            for __i in 0..__s.len() {
+                a2.offset(__i).write(__s.as_bytes()[__i]);
+            }
+            a2.offset(__s.len()).write(0);
+            a2.clone()
+        }
+        _ => Ptr::null(),
+    }
+}

rules/src/modules.rs

@@ -2,6 +2,8 @@
 pub mod algorithm_tgt_refcount;
 #[path = r#"../algorithm/tgt_unsafe.rs"#]
 pub mod algorithm_tgt_unsafe;
+#[path = r#"../arpa_inet/tgt_refcount.rs"#]
+pub mod arpa_inet_tgt_refcount;
 #[path = r#"../arpa_inet/tgt_unsafe.rs"#]
 pub mod arpa_inet_tgt_unsafe;
 #[path = r#"../array/tgt_refcount.rs"#]

tests/unit/inet_pton_ntop.c

@@ -0,0 +1,24 @@
+#include <arpa/inet.h>
+#include <assert.h>
+#include <string.h>
+
+int main(void) {
+  unsigned char buf[16];
+  assert(inet_pton(AF_INET, "1.2.3.4", buf) == 1);
+  assert(buf[0] == 1 && buf[1] == 2 && buf[2] == 3 && buf[3] == 4);
+  assert(inet_pton(AF_INET, "999.1.1.1", buf) == 0);
+  assert(inet_pton(AF_INET, "not an ip", buf) == 0);
+  assert(inet_pton(AF_INET6, "::1", buf) == 1);
+  assert(buf[0] == 0 && buf[15] == 1);
+  assert(inet_pton(AF_INET6, "2001:db8::5", buf) == 1);
+  assert(buf[0] == 0x20 && buf[1] == 0x01 && buf[15] == 5);
+
+  char text[64];
+  unsigned char four[4] = {10, 0, 0, 1};
+  assert(strcmp(inet_ntop(AF_INET, four, text, sizeof(text)), "10.0.0.1") == 0);
+  unsigned char sixteen[16] = {0};
+  sixteen[15] = 1;
+  assert(strcmp(inet_ntop(AF_INET6, sixteen, text, sizeof(text)), "::1") == 0);
+  assert(inet_ntop(AF_INET, four, text, 4) == 0);
+  return 0;
+}