Test connecting to 'localhost' on parent

Author: raoulstrackx

fortanix-vme/aws-nitro-enclaves/tests/outgoing_connection/out.expected

@@ -1 +1,3 @@
 Connected to Google successfully!
+Connected to Google successfully!
+Connected to Google successfully!

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

@@ -1,13 +1,13 @@
-use std::net::{Ipv4Addr, TcpStream};
+use std::net::TcpStream;
 use std::io::{Read, Write};
 
-fn main() {
+fn connect(host: &str, port: u16) {
+    let remote = format!("{}:{}", host, port);
     println!("# Running outgoing connection test");
-    let mut socket = TcpStream::connect(format!("google.com:80")).unwrap();
+    let mut socket = TcpStream::connect(remote).unwrap();
     // `socket.local_addr()` may return the actual local IP address, not 127.0.0.1
     assert!(socket.local_addr().unwrap().port() != 80);
-    assert!(socket.peer_addr().unwrap().ip() != Ipv4Addr::new(127, 0, 0, 1));
-    assert_eq!(socket.peer_addr().unwrap().port(), 80);
+    assert_eq!(socket.peer_addr().unwrap().port(), port);
     socket.write(b"GET / HTTP/1.1\n\n").unwrap();
     socket.flush().unwrap();
     let mut page = [0; 4192];
@@ -20,3 +20,9 @@ fn main() {
         println!("Failed to read from connection, got: {}", page);
     }
 }
+
+fn main() {
+    connect("www.google.com", 80);
+    connect("127.0.0.1", 3080);
+    connect("localhost", 3080);
+}

fortanix-vme/aws-nitro-enclaves/tests/outgoing_connection/test_interaction.sh

@@ -0,0 +1,10 @@
+#!/bin/bash -ex
+
+function cleanup {
+    killall socat
+}
+
+trap cleanup err
+trap cleanup exit
+
+socat -v TCP-LISTEN:3080,fork TCP:www.google.com:80

fortanix-vme/ci-common.sh

@@ -126,7 +126,7 @@ function cargo_test {
     fi
 
     if [ -f ./test_interaction.sh ]; then
-        kill ${test_interaction}
+        kill ${test_interaction} || true
     fi
 
     popd