Address PR feedback

Author: buggmagnet

ios/MullvadRustRuntime/include/mullvad_rust_runtime.h

@@ -5,6 +5,9 @@
 #include <stdint.h>
 #include <stdlib.h>
 
+/**
+ * SAFETY: `TunnelObfuscatorProtocol` values must either be `0` or `1`
+ */
 enum TunnelObfuscatorProtocol {
   UdpOverTcp = 0,
   Shadowsocks,

ios/MullvadRustRuntimeTests/UnsafeListener.swift

@@ -8,6 +8,7 @@
 
 import Network
 
+/// > Warning: Do not use this implementation in production code. See the warning in `start()`.
 class UnsafeListener<T: Connection> {
     private let dispatchQueue = DispatchQueue(label: "com.test.unsafeListener")
     private let listener: NWListener

mullvad-ios/src/lib.rs

@@ -2,7 +2,7 @@
 mod encrypted_dns_proxy;
 mod ephemeral_peer_proxy;
 mod shadowsocks_proxy;
-mod tunnel_obfuscator_proxy;
+pub mod tunnel_obfuscator_proxy;
 
 #[repr(C)]
 pub struct ProxyHandle {

mullvad-ios/src/tunnel_obfuscator_proxy/ffi.rs

@@ -7,7 +7,7 @@ use std::{
 
 static INIT_LOGGING: Once = Once::new();
 
-#[allow(dead_code)]
+/// SAFETY: `TunnelObfuscatorProtocol` values must either be `0` or `1`
 #[repr(u8)]
 pub enum TunnelObfuscatorProtocol {
     UdpOverTcp = 0,
@@ -35,8 +35,7 @@ pub unsafe extern "C" fn start_tunnel_obfuscator_proxy(
             return -1;
         };
 
-    let result = TunnelObfuscatorRuntime::new(peer_sock_addr, obfuscation_protocol)
-        .and_then(|runtime| runtime.run());
+    let result = TunnelObfuscatorRuntime::new(peer_sock_addr, obfuscation_protocol).run();
 
     match result {
         Ok((local_endpoint, obfuscator_handle)) => {

mullvad-ios/src/tunnel_obfuscator_proxy/mod.rs

@@ -17,10 +17,7 @@ pub struct TunnelObfuscatorRuntime {
 }
 
 impl TunnelObfuscatorRuntime {
-    pub fn new(
-        peer: SocketAddr,
-        obfuscation_protocol: TunnelObfuscatorProtocol,
-    ) -> io::Result<Self> {
+    pub fn new(peer: SocketAddr, obfuscation_protocol: TunnelObfuscatorProtocol) -> Self {
         let settings: ObfuscationSettings = match obfuscation_protocol {
             TunnelObfuscatorProtocol::UdpOverTcp => {
                 ObfuscationSettings::Udp2Tcp(udp2tcp::Settings { peer })
@@ -33,7 +30,7 @@ impl TunnelObfuscatorRuntime {
             }
         };
 
-        Ok(Self { settings })
+        Self { settings }
     }
 
     pub fn run(self) -> io::Result<(SocketAddr, TunnelObfuscatorHandle)> {