git.meshlink.io Git - meshlink/blob - src/devtools.c

   1
   2 #include "system.h"
   3
   4 #include "logger.h"
   5 #include "meshlink_internal.h"
   6 #include "node.h"
   7 #include "splay_tree.h"
   8 #include "netutl.h"
   9 #include "xalloc.h"
  10
  11 #include "devtools.h"
  12
  13 static bool fstrwrite(const char *str, FILE *stream) {
  14         size_t len = strlen(str);
  15
  16         if(fwrite((void *)str, 1, len, stream) != len)
  17                 return false;
  18
  19         return true;
  20 }
  21
  22 static const char *__itoa(int value) {
  23         static char buffer[sizeof(int) * 8 + 1];        // not thread safe
  24
  25         if(snprintf(buffer, sizeof(buffer), "%d", value) == -1)
  26                 return "";
  27
  28         return buffer;
  29 }
  30
  31 bool devtool_export_json_all_edges_state(meshlink_handle_t *mesh, FILE *stream) {
  32         bool result = true;
  33
  34         pthread_mutex_lock(&(mesh->mesh_mutex));
  35
  36         // export edges and nodes
  37         size_t node_count = 0;
  38         size_t edge_count = 0;
  39
  40         meshlink_node_t **nodes = meshlink_get_all_nodes(mesh, NULL, &node_count);
  41         meshlink_edge_t **edges = meshlink_get_all_edges_state(mesh, NULL, &edge_count);
  42
  43         if((!nodes && node_count != 0) || (!edges && edge_count != 0))
  44                 goto fail;
  45
  46         // export begin
  47         if(!fstrwrite("{\n", stream))
  48                 goto fail;
  49
  50         // export nodes
  51         if(!fstrwrite("\t\"nodes\": {\n", stream))
  52                 goto fail;
  53
  54         for(size_t i = 0; i < node_count; ++i) {
  55                 if(!fstrwrite("\t\t\"", stream) || !fstrwrite(((node_t *)nodes[i])->name, stream) || !fstrwrite("\": {\n", stream))
  56                         goto fail;
  57
  58                 if(!fstrwrite("\t\t\t\"name\": \"", stream) || !fstrwrite(((node_t *)nodes[i])->name, stream) || !fstrwrite("\",\n", stream))
  59                         goto fail;
  60
  61                 if(!fstrwrite("\t\t\t\"options\": ", stream) || !fstrwrite(__itoa(((node_t *)nodes[i])->options), stream) || !fstrwrite(",\n", stream))
  62                         goto fail;
  63
  64                 if(!fstrwrite("\t\t\t\"devclass\": ", stream) || !fstrwrite(__itoa(((node_t *)nodes[i])->devclass), stream) || !fstrwrite("\n", stream))
  65                         goto fail;
  66
  67                 if(!fstrwrite((i + 1) != node_count ? "\t\t},\n" : "\t\t}\n", stream))
  68                         goto fail;
  69         }
  70
  71         if(!fstrwrite("\t},\n", stream))
  72                 goto fail;
  73
  74         // export edges
  75
  76         if(!fstrwrite("\t\"edges\": {\n", stream))
  77                 goto fail;
  78
  79         for(size_t i = 0; i < edge_count; ++i) {
  80                 if(!fstrwrite("\t\t\"", stream) || !fstrwrite(edges[i]->from->name, stream) || !fstrwrite("_to_", stream) || !fstrwrite(edges[i]->to->name, stream) || !fstrwrite("\": {\n", stream))
  81                         goto fail;
  82
  83                 if(!fstrwrite("\t\t\t\"from\": \"", stream) || !fstrwrite(edges[i]->from->name, stream) || !fstrwrite("\",\n", stream))
  84                         goto fail;
  85
  86                 if(!fstrwrite("\t\t\t\"to\": \"", stream) || !fstrwrite(edges[i]->to->name, stream) || !fstrwrite("\",\n", stream))
  87                         goto fail;
  88
  89                 char *host = NULL, *port = NULL, *address = NULL;
  90                 sockaddr2str((const sockaddr_t *) & (edges[i]->address), &host, &port);
  91
  92                 if(host && port)
  93                         xasprintf(&address, "{ \"host\": \"%s\", \"port\": %s }", host, port);
  94
  95                 free(host);
  96                 free(port);
  97
  98                 if(!fstrwrite("\t\t\t\"address\": ", stream) || !fstrwrite(address ? address : "null", stream) || !fstrwrite(",\n", stream)) {
  99                         free(address);
 100                         goto fail;
 101                 }
 102
 103                 free(address);
 104
 105                 if(!fstrwrite("\t\t\t\"options\": ", stream) || !fstrwrite(__itoa(edges[i]->options), stream) || !fstrwrite(",\n", stream))
 106                         goto fail;
 107
 108                 if(!fstrwrite("\t\t\t\"weight\": ", stream) || !fstrwrite(__itoa(edges[i]->weight), stream) || !fstrwrite("\n", stream))
 109                         goto fail;
 110
 111                 if(!fstrwrite((i + 1) != edge_count ? "\t\t},\n" : "\t\t}\n", stream))
 112                         goto fail;
 113         }
 114
 115         if(!fstrwrite("\t}\n", stream))
 116                 goto fail;
 117
 118         // DONE!
 119
 120         if(!fstrwrite("}", stream))
 121                 goto fail;
 122
 123         goto done;
 124
 125 fail:
 126         result = false;
 127
 128 done:
 129
 130         if(nodes)
 131                 free(nodes);
 132
 133         for(size_t i = 0; edges && i < edge_count; ++i)
 134                 free(edges[i]);
 135
 136         if(nodes)
 137                 free(edges);
 138
 139         pthread_mutex_unlock(&(mesh->mesh_mutex));
 140
 141         return result;
 142 }