X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=test%2Fnetns_utils.c;h=ec7aee4a8e65bda041788cd890bfd8e3e3d7f18f;hb=bc2e55e940e45748a49f1a7eb44b156917fca469;hp=d957d5b63386640e622821ef673ec45cb1b3e41b;hpb=e38737be28c742b241a6b0ffdbad541c3c0b13b2;p=meshlink

diff --git a/test/netns_utils.c b/test/netns_utils.c
index d957d5b6..ec7aee4a 100644
--- a/test/netns_utils.c
+++ b/test/netns_utils.c
@@ -92,6 +92,26 @@ static void setup_lan_topology(peer_config_t *peers, int npeers) {
 	}
 }
 
+/// Set up an indirect topology where all peers can only access the relay
+static void setup_indirect_topology(peer_config_t *peers, int npeers) {
+	// Add an interface to each peer that is connected to the relay
+	for(int i = 1; i < npeers; i++) {
+		char *command = NULL;
+		assert(asprintf(&command,
+		                "/bin/ip netns exec %1$s /bin/ip link add eth0 type veth peer eth%3$d netns %2$s;"
+		                "/bin/ip netns exec %1$s ip addr flush dev eth0;"
+		                "/bin/ip netns exec %1$s ip addr add 192.168.%3$d.2/24 dev eth0;"
+		                "/bin/ip netns exec %1$s /bin/ip link set dev eth0 up;"
+		                "/bin/ip netns exec %2$s ip addr flush dev eth%3$d;"
+		                "/bin/ip netns exec %2$s ip addr add 192.168.%3$d.1/24 dev eth%3$d;"
+		                "/bin/ip netns exec %2$s /bin/ip link set dev eth%3$d up;",
+		                peers[i].netns_name, peers[0].netns_name, i));
+		assert(command);
+		assert(system(command) == 0);
+		free(command);
+	}
+}
+
 /// Give a peer a unique IP address
 void change_peer_ip(peer_config_t *peer) {
 	char *command = NULL;
@@ -112,6 +132,7 @@ static void invite_peers(peer_config_t *peers, int npeers) {
 	for(int i = 1; i < npeers; i++) {
 		char *invitation = meshlink_invite_ex(peers[0].mesh, NULL, peers[i].name, MESHLINK_INVITE_LOCAL | MESHLINK_INVITE_NUMERIC);
 		assert(invitation);
+		printf("%s\n", invitation);
 		assert(meshlink_join(peers[i].mesh, invitation));
 		free(invitation);
 	}
@@ -131,9 +152,9 @@ static void close_peers(peer_config_t *peers, int npeers) {
 /// Set up relay, peer and NUT that are directly connected
 peer_config_t *setup_relay_peer_nut(const char *prefix) {
 	static peer_config_t peers[] = {
-		{"relay", DEV_CLASS_BACKBONE},
-		{"peer", DEV_CLASS_STATIONARY},
-		{"nut", DEV_CLASS_STATIONARY},
+		{"relay", DEV_CLASS_BACKBONE, NULL, 0, NULL},
+		{"peer", DEV_CLASS_STATIONARY, NULL, 0, NULL},
+		{"nut", DEV_CLASS_STATIONARY, NULL, 0, NULL},
 	};
 
 	create_peers(peers, 3, prefix);
@@ -143,6 +164,37 @@ peer_config_t *setup_relay_peer_nut(const char *prefix) {
 	return peers;
 }
 
+/// Set up relay, peer and NUT that are directly connected
+peer_config_t *setup_relay_peer_nut_indirect(const char *prefix) {
+	static peer_config_t peers[] = {
+		{"relay", DEV_CLASS_BACKBONE, NULL, 0, NULL},
+		{"peer", DEV_CLASS_STATIONARY, NULL, 0, NULL},
+		{"nut", DEV_CLASS_STATIONARY, NULL, 0, NULL},
+	};
+
+	create_peers(peers, 3, prefix);
+	setup_indirect_topology(peers, 3);
+	assert(meshlink_add_invitation_address(peers[0].mesh, "192.168.1.1", NULL));
+	assert(meshlink_add_invitation_address(peers[0].mesh, "192.168.2.1", NULL));
+	invite_peers(peers, 3);
+
+	return peers;
+}
+
+/// Make all nodes only be able to communicate via TCP
+void set_peers_tcponly(peer_config_t *peers, int npeers) {
+	for(int i = 0; i < npeers; i++) {
+		char *command = NULL;
+		assert(asprintf(&command,
+		                "/bin/ip netns exec %1$s iptables -A INPUT -p udp -j DROP;"
+		                "/bin/ip netns exec %1$s iptables -A OUTPUT -p udp -j DROP;",
+		                peers[i].netns_name));
+		assert(command);
+		assert(system(command) == 0);
+		free(command);
+	}
+}
+
 void close_relay_peer_nut(peer_config_t *peers) {
 	close_peers(peers, 3);
 }