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

diff --git a/test/netns_utils.c b/test/netns_utils.c
index 70fa3130..ec7aee4a 100644
--- a/test/netns_utils.c
+++ b/test/netns_utils.c
@@ -181,6 +181,20 @@ peer_config_t *setup_relay_peer_nut_indirect(const char *prefix) {
 	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);
 }