From: Guus Sliepen Date: Sun, 26 Sep 2021 22:25:01 +0000 (+0200) Subject: Add a metering test. X-Git-Url: https://git.meshlink.io/?a=commitdiff_plain;ds=inline;p=meshlink-tiny Add a metering test. Currently it only measures traffic between tiny and full nodes during the initial connection and while idle. --- diff --git a/test/Makefile.am b/test/Makefile.am index 3090ae9..06cb07c 100644 --- a/test/Makefile.am +++ b/test/Makefile.am @@ -4,6 +4,7 @@ TESTS = \ ephemeral \ import-export \ invite-join \ + metering \ sign-verify \ storage-policy @@ -18,6 +19,7 @@ check_PROGRAMS = \ ephemeral \ import-export \ invite-join \ + metering \ sign-verify \ storage-policy @@ -37,5 +39,6 @@ ephemeral_SOURCES = ephemeral.c $(default_SOURCES) get_all_nodes_SOURCES = get-all-nodes.c $(default_SOURCES) import_export_SOURCES = import-export.c $(default_SOURCES) invite_join_SOURCES = invite-join.c $(default_SOURCES) +metering_SOURCES = metering.c netns_utils.c $(default_SOURCES) sign_verify_SOURCES = sign-verify.c $(default_SOURCES) storage_policy_SOURCES = storage-policy.c $(default_SOURCES) diff --git a/test/metering.c b/test/metering.c new file mode 100644 index 0000000..fdccd5e --- /dev/null +++ b/test/metering.c @@ -0,0 +1,80 @@ +#ifdef NDEBUG +#undef NDEBUG +#endif + +#include +#include +#include +#include +#include + +#include "meshlink-tiny.h" +#include "devtools.h" +#include "netns_utils.h" +#include "utils.h" + +#include "full.h" + +static void print_counters(peer_config_t *peers, const char *description) { + printf("%s:\n", description); + printf(" %9s %9s %9s %9s %9s %9s\n", + "in data", + "forward", + "meta", + "out data", + "forward", + "meta"); + + assert(peers[0].full); + + for(int i = 0; i < 3; i++) { + meshlink_node_t *node = full_meshlink_get_node(peers[0].mesh, peers[i].name); + assert(node); + struct devtool_node_status status; + full_devtool_reset_node_counters(peers[0].mesh, node, &status); + printf(" %5s: %9" PRIu64 " %9" PRIu64 " %9" PRIu64 " %9" PRIu64 " %9" PRIu64 " %9" PRIu64 "\n", + node->name, + status.in_data, + status.in_forward, + status.in_meta, + status.out_data, + status.out_forward, + status.out_meta); + } +} + + +int main(void) { + init_full(); + + meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_cb); + + // Set up relay, peer and NUT + peer_config_t *peers = setup_relay_peer_nut("metering"); + + // Ensure DEV_CLASS_STATIONARY uses a 5 minute ping time + for(int i = 0; i < 3; i++) { + peers[i].full + ? full_meshlink_set_dev_class_timeouts(peers[i].mesh, DEV_CLASS_STATIONARY, 300, 5) + : meshlink_set_dev_class_timeouts(peers[i].mesh, DEV_CLASS_BACKBONE, 300, 5); + } + + for(int i = 0; i < 3; i++) { + assert(peers[i].full + ? full_meshlink_start(peers[i].mesh) + : meshlink_start(peers[i].mesh) + ); + } + + // Measure traffic after 1 minute of PMTU probing + sleep(60); + print_counters(peers, "PMTU probing (1 min)"); + + // Measure traffic after 1 minute of idle + for(int i = 0; i < 10; i++) { + sleep(60); + print_counters(peers, "Idle (1 min)"); + } + + close_relay_peer_nut(peers); +} diff --git a/test/netns_utils.c b/test/netns_utils.c index 5e0ed21..476adbc 100644 --- a/test/netns_utils.c +++ b/test/netns_utils.c @@ -1,6 +1,6 @@ #define _GNU_SOURCE 1 -#ifndef NDEBUG +#ifdef NDEBUG #undef NDEBUG #endif @@ -10,10 +10,12 @@ #include #include -#include "../src/meshlink-tiny.h" +#include "meshlink-tiny.h" #include "netns_utils.h" #include "utils.h" +#include "full.h" + static int ip = 1; /// Create meshlink instances and network namespaces for a list of peers @@ -24,6 +26,8 @@ static void create_peers(peer_config_t *peers, int npeers, const char *prefix) { } for(int i = 0; i < npeers; i++) { + bool full = peers[i].full; + assert(asprintf(&peers[i].netns_name, "%s%d", prefix, i) > 0); char *command = NULL; assert(asprintf(&command, @@ -45,16 +49,25 @@ static void create_peers(peer_config_t *peers, int npeers, const char *prefix) { char *conf_path = NULL; assert(asprintf(&conf_path, "%s_conf.%d", prefix, i + 1) > 0); assert(conf_path); - assert(meshlink_destroy(conf_path)); + assert(full ? full_meshlink_destroy(conf_path) : meshlink_destroy(conf_path)); - meshlink_open_params_t *params = meshlink_open_params_init(conf_path, peers[i].name, prefix, peers[i].devclass); + meshlink_open_params_t *params = full + ? full_meshlink_open_params_init(conf_path, peers[i].name, prefix, peers[i].devclass) + : meshlink_open_params_init(conf_path, peers[i].name, prefix, peers[i].devclass); assert(params); - assert(meshlink_open_params_set_netns(params, peers[i].netns)); + assert(full + ? full_meshlink_open_params_set_netns(params, peers[i].netns) + : meshlink_open_params_set_netns(params, peers[i].netns) + ); - peers[i].mesh = meshlink_open_ex(params); + peers[i].mesh = full ? full_meshlink_open_ex(params) : meshlink_open_ex(params); assert(peers[i].mesh); free(params); free(conf_path); + + if(full) { + full_meshlink_enable_discovery(peers[i].mesh, false); + } } } @@ -90,6 +103,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; @@ -104,23 +137,25 @@ void change_peer_ip(peer_config_t *peer) { } /// Let the first peer in a list invite all the subsequent peers -static void link_peers(peer_config_t *peers, int npeers) { +static void invite_peers(peer_config_t *peers, int npeers) { + assert(peers[0].full); + assert(full_meshlink_start(peers[0].mesh)); + for(int i = 1; i < npeers; i++) { - char *export_relay = meshlink_export(peers[0].mesh); - char *export_peer = meshlink_export(peers[i].mesh); - assert(export_relay); - assert(export_peer); - assert(meshlink_import(peers[0].mesh, export_peer)); - assert(meshlink_import(peers[i].mesh, export_relay)); - free(export_relay); - free(export_peer); + char *invitation = full_meshlink_invite_ex(peers[0].mesh, NULL, peers[i].name, MESHLINK_INVITE_LOCAL | MESHLINK_INVITE_NUMERIC); + assert(invitation); + printf("%s\n", invitation); + assert(peers[i].full ? full_meshlink_join(peers[i].mesh, invitation) : meshlink_join(peers[i].mesh, invitation)); + free(invitation); } + + full_meshlink_stop(peers[0].mesh); } /// Close meshlink instances and clean up static void close_peers(peer_config_t *peers, int npeers) { for(int i = 0; i < npeers; i++) { - meshlink_close(peers[i].mesh); + peers[i].full ? full_meshlink_close(peers[i].mesh) : meshlink_close(peers[i].mesh); close(peers[i].netns); free(peers[i].netns_name); } @@ -129,18 +164,49 @@ 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, true, NULL}, + {"peer", DEV_CLASS_STATIONARY, NULL, 0, false, NULL}, + {"nut", DEV_CLASS_STATIONARY, NULL, 0, false, NULL}, }; create_peers(peers, 3, prefix); setup_lan_topology(peers, 3); - link_peers(peers, 3); + invite_peers(peers, 3); + + 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, true, NULL}, + {"peer", DEV_CLASS_STATIONARY, NULL, 0, false, NULL}, + {"nut", DEV_CLASS_STATIONARY, NULL, 0, false, NULL}, + }; + + create_peers(peers, 3, prefix); + setup_indirect_topology(peers, 3); + assert(full_meshlink_add_invitation_address(peers[0].mesh, "192.168.1.1", NULL)); + assert(full_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); } diff --git a/test/netns_utils.h b/test/netns_utils.h index 7bbe51a..80a5e67 100644 --- a/test/netns_utils.h +++ b/test/netns_utils.h @@ -7,11 +7,14 @@ typedef struct peer_config { char *netns_name; int netns; + bool full; meshlink_handle_t *mesh; } peer_config_t; extern void change_peer_ip(peer_config_t *peer); extern peer_config_t *setup_relay_peer_nut(const char *prefix); +extern peer_config_t *setup_relay_peer_nut_indirect(const char *prefix); +extern void set_peers_tcponly(peer_config_t *peers, int npeers); extern void close_relay_peer_nut(peer_config_t *peers); #endif