git.meshlink.io Git - meshlink/blob - test/blackbox/common/common_handlers.c

   1 /*
   2     common_handlers.c -- Implementation of common callback handling and signal handling
   3                             functions for black box tests
   4     Copyright (C) 2018  Guus Sliepen <guus@meshlink.io>
   5
   6     This program is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     This program is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License along
  17     with this program; if not, write to the Free Software Foundation, Inc.,
  18     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19 */
  20 #include <stdio.h>
  21 #include <signal.h>
  22 #include <stdlib.h>
  23 #include <assert.h>
  24 #include <string.h>
  25 #include <errno.h>
  26 #include <sys/ioctl.h>
  27 #include <net/if.h>
  28 #include <arpa/inet.h>
  29 #include <sys/socket.h>
  30 #include <netdb.h>
  31 #include <ifaddrs.h>
  32 #include <ctype.h>
  33 #include "test_step.h"
  34 #include "common_handlers.h"
  35
  36 char *lxc_bridge = NULL;
  37 black_box_state_t *state_ptr = NULL;
  38
  39 bool meta_conn_status[10];
  40
  41 bool test_running;
  42
  43 static int meshlink_get_node_in_container(const char *name) {
  44         int i;
  45
  46         for(i = 0; i < state_ptr->num_nodes; i++) {
  47                 if(!strcasecmp(state_ptr->node_names[i], name)) {
  48                         return i;
  49                         break;
  50                 }
  51         }
  52
  53         return -1;
  54 }
  55
  56 void mesh_close_signal_handler(int a) {
  57         test_running = false;
  58
  59         exit(EXIT_SUCCESS);
  60 }
  61
  62 void setup_signals(void) {
  63         test_running = true;
  64         signal(SIGTERM, mesh_close_signal_handler);
  65 }
  66
  67 /* Return the IP Address of the Interface 'if_name'
  68     The caller is responsible for freeing the dynamically allocated string that is returned */
  69 char *get_ip(const char *if_name) {
  70         struct ifaddrs *ifaddr, *ifa;
  71         char *ip;
  72         int family;
  73
  74         ip = malloc(NI_MAXHOST);
  75         assert(ip);
  76         assert(getifaddrs(&ifaddr) != -1);
  77
  78         for(ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) {
  79                 if(ifa->ifa_addr == NULL) {
  80                         continue;
  81                 }
  82
  83                 family = ifa->ifa_addr->sa_family;
  84
  85                 if(family == AF_INET && !strcmp(ifa->ifa_name, if_name)) {
  86                         assert(!getnameinfo(ifa->ifa_addr, (family == AF_INET) ? sizeof(struct sockaddr_in) : sizeof(struct sockaddr_in6), ip, NI_MAXHOST, NULL, 0, NI_NUMERICHOST));
  87                         break;
  88                 }
  89         }
  90
  91         return ip;
  92 }
  93
  94 /* Return the IP Address of the Interface 'if_name'
  95     The caller is responsible for freeing the dynamically allocated string that is returned */
  96 char *get_netmask(const char *if_name) {
  97         struct ifaddrs *ifaddr, *ifa;
  98         char *ip;
  99         int family;
 100
 101         ip = malloc(NI_MAXHOST);
 102         assert(ip);
 103         assert(getifaddrs(&ifaddr) != -1);
 104
 105         for(ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) {
 106                 if(ifa->ifa_addr == NULL) {
 107                         continue;
 108                 }
 109
 110                 family = ifa->ifa_addr->sa_family;
 111
 112                 if(family == AF_INET && !strcmp(ifa->ifa_name, if_name)) {
 113                         assert(!getnameinfo(ifa->ifa_netmask, (family == AF_INET) ? sizeof(struct sockaddr_in) : sizeof(struct sockaddr_in6), ip, NI_MAXHOST, NULL, 0, NI_NUMERICHOST));
 114                         break;
 115                 }
 116         }
 117
 118         return ip;
 119 }
 120
 121 /* Change the IP Address of an interface */
 122 void set_ip(const char *if_name, const char *new_ip) {
 123         char set_ip_cmd[100];
 124         assert(snprintf(set_ip_cmd, sizeof(set_ip_cmd), "ifconfig %s %s", if_name, new_ip) >= 0);
 125         assert(system(set_ip_cmd) == 0);
 126 }
 127
 128 /* Change the Netmask of an interface */
 129 void set_netmask(const char *if_name, const char *new_netmask) {
 130         char set_mask_cmd[100];
 131         assert(snprintf(set_mask_cmd, sizeof(set_mask_cmd), "ifconfig %s netmask %s", if_name, new_netmask) >= 0);
 132         assert(system(set_mask_cmd) == 0);
 133 }
 134
 135 /* Bring a network interface down (before making changes such as the IP Address) */
 136 void stop_nw_intf(const char *if_name) {
 137         char nw_down_cmd[100];
 138         assert(snprintf(nw_down_cmd, sizeof(nw_down_cmd), "ifconfig %s down", if_name) >= 0);
 139         assert(system(nw_down_cmd) == 0);
 140 }
 141
 142 /* Bring a network interface up (after bringing it down and making changes such as
 143     the IP Address) */
 144 void start_nw_intf(const char *if_name) {
 145         char nw_up_cmd[100];
 146         assert(snprintf(nw_up_cmd, sizeof(nw_up_cmd), "ifconfig %s up", if_name) >= 0);
 147         assert(system(nw_up_cmd) == 0);
 148 }
 149
 150 void meshlink_callback_node_status(meshlink_handle_t *mesh, meshlink_node_t *node,
 151                                    bool reachable) {
 152         (void)mesh;
 153         fprintf(stderr, "Node %s became %s\n", node->name, (reachable) ? "reachable" : "unreachable");
 154 }
 155
 156 void meshlink_callback_logger(meshlink_handle_t *mesh, meshlink_log_level_t level,
 157                               const char *text) {
 158         (void)mesh;
 159         (void)level;
 160
 161         fprintf(stderr, "meshlink>> %s\n", text);
 162
 163         if(state_ptr) {
 164                 bool status;
 165                 char name[100];
 166
 167                 if(sscanf(text, "Connection with %s activated", name) == 1) {
 168                         status = true;
 169                 } else if(sscanf(text, "Already connected to %s", name) == 1) {
 170                         status = true;
 171                 } else if(sscanf(text, "Connection closed by %s", name) == 1) {
 172                         status = false;
 173                 } else if(sscanf(text, "Closing connection with %s", name) == 1) {
 174                         status = false;
 175                 } else {
 176                         return;
 177                 }
 178
 179                 int i = meshlink_get_node_in_container(name);
 180                 assert(i != -1);
 181                 meta_conn_status[i] = status;
 182         }
 183 }