update allow traffic

topohub_to_gns3.py

@@ -51,14 +51,21 @@ DIR = "/home/anh/Documents/kernel_preparation/gns3_unikernel_testbed"
 #
 
 def send_para_server(A_raw, f, num_host, p_NAT, P_values):
+    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as server_socket:
+        server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+        server_socket.bind(('0.0.0.0', 12348))
+        server_socket.listen()
+        print("Server started on 12348")
+
+        while True:
+            client_socket, client_address = server_socket.accept()
+            with client_socket:  # Ensures cleanup
                 data = {'A_raw': A_raw, 'f': f, 'num_host': num_host, 'p_NAT': p_NAT, 'P_values': P_values}
                 data_js = json.dumps(data)
-    HOST = "10.1.1.1"
-    PORT = 12348
-    client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-    client.connect((HOST, PORT))
-    client.sendall(data_js.encode('utf-8'))  # Send JSON data
-    client.close()
+                client_socket.sendall(data_js.encode('utf-8'))  # Send JSON data
+                break
+        client_socket.close()
+    server_socket.close()
 
 
 def run_vc_server(map_connection_ubuntu_base):
@@ -297,9 +304,6 @@ def generate_topohub_GNS3_topology():
             sys.exit()
         GNS3Manager.start_node(ubuntu_server.id)
 
-        if input("Start sending parameter?") != "y":
-            sys.exit()
-
         send_para_server(g.graph['demands'], f, NUM_HOST, p_NAT, P_values)
 
         # Capture packets from communications/links

ubuntu_base/if_utils.py

@@ -144,9 +144,9 @@ def drop_internet_and_allow_VC_server_rule(inf_to_internet, server_port):
     :param server_port: port of the VC server running on the main host
     :param inf_to_internet: the interface that connect to the main host (internet)
     """
-    commands = [["sudo", "iptables", "-A", "FORWARD", "-o", inf_to_internet, "-j", "DROP"],
+    commands = [["sudo", "iptables", "-A", "FORWARD", "-o", inf_to_internet, "-p", "tcp", "--dport", str(12348), "-j", "ACCEPT"],
+                ["sudo", "iptables", "-A", "FORWARD", "-o", inf_to_internet, "-j", "DROP"],
                 ["sudo", "iptables", "-A", "OUTPUT", "-o", inf_to_internet, "-p", "tcp", "--dport", str(server_port), "-j", "ACCEPT"],
-                ["sudo", "iptables", "-A", "INPUT", "-i", inf_to_internet, "-m", "state", "--state", "ESTABLISHED,RELATED", "-j", "ACCEPT"],
                 ["sudo", "iptables", "-A", "OUTPUT", "-o", inf_to_internet, "-j", "DROP"]]
 
     for command in commands:

window/tcp_crash_listener.py

@@ -114,18 +114,13 @@ def read_and_write_points():
 
 
 def get_parameters():
-    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as server_socket:
-        server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
-        server_socket.bind(('0.0.0.0', 12348))
-        server_socket.listen()
-        print("Server started on 12348")
-
-        while True:
-            client_socket, client_address = server_socket.accept()
-            with client_socket:  # Ensures cleanup
+    HOST = "192.168.122.1"
+    PORT = 12348
+    client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    client.connect((HOST, PORT))
     data = b""
     while True:  # Read until all data is received
-                    chunk = client_socket.recv(1024)
+        chunk = client.recv(1024)
         if not chunk:
             break
         data += chunk
@@ -138,12 +133,11 @@ def get_parameters():
         p_NAT_raw = data_js['p_NAT']
         P_values_raw = data_js['P_values']
         print(f"Received: {data_js}")  # Debugging
-                except json.JSONDecodeError:
-                    print("Invalid JSON received")
 
-            client_socket.close()
-            server_socket.close()
+        client.close()
         return A_raw, f_raw, num_machine_raw, p_NAT_raw, P_values_raw
+    except json.JSONDecodeError:
+        print("Invalid JSON received")
 
 
 def main():