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

diff --git a/test/netns_utils.c b/test/netns_utils.c
index 70fa3130..58c01ff5 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);
 }