Test incoming connection on localhost

Author: raoulstrackx

fortanix-vme/aws-nitro-enclaves/tests/incoming_connection/src/main.rs

@@ -3,9 +3,13 @@ use std::io::{ErrorKind, Read, Write};
 use std::net::{IpAddr, Ipv4Addr, Shutdown, SocketAddr, TcpListener, TcpStream};
 use std::os::unix::io::{AsRawFd, FromRawFd};
 
-fn server_run() {
+fn server_run(run: u32) {
     println!("Bind TCP socket to port 3400");
-    let listener = TcpListener::bind("127.0.0.1:3400").expect("Bind failed");
+    let listener = if run == 1 {
+        TcpListener::bind("127.0.0.1:3400").expect("Bind failed")
+    } else {
+        TcpListener::bind("localhost:3400").expect("Bind failed")
+    };
     assert_eq!(listener.local_addr().unwrap(), SocketAddr::new(IpAddr::V4(Ipv4Addr::new(127, 0, 0, 1)), 3400));
 
     let fd = listener.as_raw_fd();
@@ -49,7 +53,7 @@ fn server_run() {
 fn main() {
     for run in 1..=2 {
         println!("Server run #{}", run);
-        server_run()
+        server_run(run)
     }
     println!("Bye bye");
 }