7 #include <linux/limits.h>
9 #include "../src/meshlink.h"
12 static meshlink_handle_t **mesh;
14 static meshlink_node_t **nodes;
17 // Make all nodes know about each other by importing each others public keys and addresses.
18 static void linkmesh() {
19 for(int i = 0; i < n; i++) {
20 char *datai = meshlink_export(mesh[i]);
22 for(int j = i + 1; j < n; j++) {
23 char *dataj = meshlink_export(mesh[j]);
24 meshlink_import(mesh[i], dataj);
25 meshlink_import(mesh[j], datai);
33 static void parse_command(char *buf) {
34 char *arg = strchr(buf, ' ');
38 if(!strcasecmp(buf, "invite")) {
42 fprintf(stderr, "/invite requires an argument!\n");
46 invitation = meshlink_invite(mesh[0], arg);
48 fprintf(stderr, "Could not invite '%s': %s\n", arg, meshlink_strerror(meshlink_errno));
52 printf("Invitation for %s: %s\n", arg, invitation);
54 } else if(!strcasecmp(buf, "join")) {
56 fprintf(stderr, "/join requires an argument!\n");
60 if(!meshlink_join(mesh[0], arg))
61 fprintf(stderr, "Could not join using invitation: %s\n", meshlink_strerror(meshlink_errno));
63 fprintf(stderr, "Invitation accepted!\n");
64 } else if(!strcasecmp(buf, "kick")) {
66 fprintf(stderr, "/kick requires an argument!\n");
70 meshlink_node_t *node = meshlink_get_node(mesh[0], arg);
72 fprintf(stderr, "Unknown node '%s'\n", arg);
76 meshlink_blacklist(mesh[0], node);
78 printf("Node '%s' blacklisted.\n", arg);
79 } else if(!strcasecmp(buf, "who")) {
81 nodes = meshlink_get_all_nodes(mesh[0], nodes, &nnodes);
83 fprintf(stderr, "Could not get list of nodes: %s\n", meshlink_strerror(meshlink_errno));
85 printf("%zu known nodes:", nnodes);
86 for(int i = 0; i < nnodes; i++)
87 printf(" %s", nodes[i]->name);
91 meshlink_node_t *node = meshlink_get_node(mesh[0], arg);
93 fprintf(stderr, "Unknown node '%s'\n", arg);
95 printf("Node %s found\n", arg);
98 } else if(!strcasecmp(buf, "link")) {
100 } else if(!strcasecmp(buf, "quit")) {
103 } else if(!strcasecmp(buf, "help")) {
105 "<name>: <message> Send a message to the given node.\n"
106 " Subsequent messages don't need the <name>: prefix.\n"
107 "/invite <name> Create an invitation for a new node.\n"
108 "/join <invitation> Join an existing mesh using an invitation.\n"
109 "/kick <name> Blacklist the given node.\n"
110 "/who [<name>] List all nodes or show information about the given node.\n"
111 "/link Link all nodes together.\n"
112 "/quit Exit this program.\n"
115 fprintf(stderr, "Unknown command '/%s'\n", buf);
119 static void parse_input(char *buf) {
120 static meshlink_node_t *destination;
130 if(len && buf[len - 1] == '\n')
133 if(len && buf[len - 1] == '\r')
136 // Ignore empty lines.
141 // Commands start with '/'
144 return parse_command(buf + 1);
146 // Lines in the form "name: message..." set the destination node.
149 char *colon = strchr(buf, ':');
157 destination = meshlink_get_node(mesh[0], buf);
159 fprintf(stderr, "Unknown node '%s'\n", buf);
165 fprintf(stderr, "Who are you talking to? Write 'name: message...'\n");
169 if(!meshlink_send(mesh[0], destination, msg, strlen(msg) + 1)) {
170 fprintf(stderr, "Could not send message to '%s': %s\n", destination->name, meshlink_strerror(meshlink_errno));
174 printf("Message sent to '%s'.\n", destination->name);
177 int main(int argc, char *argv[]) {
178 const char *basebase = ".manynodes";
185 fprintf(stderr, "Usage: %s [number of local nodes] [confbase]\n", argv[0]);
192 mesh = calloc(n, sizeof *mesh);
194 mkdir(basebase, 0750);
196 char filename[PATH_MAX];
198 for(int i = 0; i < n; i++) {
199 snprintf(nodename, sizeof nodename, "node%d", i);
200 snprintf(filename, sizeof filename, "%s/%s", basebase, nodename);
201 bool itsnew = access(filename, R_OK);
202 mesh[i] = meshlink_open(filename, nodename, "manynodes");
204 meshlink_add_address(mesh[i], "localhost");
206 fprintf(stderr, "errno is: %d\n", meshlink_errno);
207 fprintf(stderr, "Could not open %s: %s\n", filename, meshlink_strerror(meshlink_errno));
214 for(int i = 0; i < n; i++) {
215 if(!meshlink_start(mesh[i]))
216 fprintf(stderr, "Could not start node %d: %s\n", i, meshlink_strerror(meshlink_errno));
222 fprintf(stderr, "Could not start any node!\n");
226 printf("%d nodes started.\nType /help for a list of commands.\n", started);
228 while(fgets(buf, sizeof buf, stdin))
231 printf("Nodes stopping.\n");
233 for(int i = 0; i < n; i++)
234 meshlink_close(mesh[i]);