]> git.meshlink.io Git - meshlink/blob - examples/chat.c
Add support for encrypted storage.
[meshlink] / examples / chat.c
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <string.h>
4 #include <strings.h>
5 #include "../src/meshlink.h"
6
7 static void log_message(meshlink_handle_t *mesh, meshlink_log_level_t level, const char *text) {
8         (void)mesh;
9
10         static const char *levelstr[] = {
11                 [MESHLINK_DEBUG] = "\x1b[34mDEBUG",
12                 [MESHLINK_INFO] = "\x1b[32mINFO",
13                 [MESHLINK_WARNING] = "\x1b[33mWARNING",
14                 [MESHLINK_ERROR] = "\x1b[31mERROR",
15                 [MESHLINK_CRITICAL] = "\x1b[31mCRITICAL",
16         };
17
18         fprintf(stderr, "%s:\x1b[0m %s\n", levelstr[level], text);
19 }
20
21 static void receive(meshlink_handle_t *mesh, meshlink_node_t *source, const void *data, size_t len) {
22         (void)mesh;
23
24         const char *msg = data;
25
26         if(!len || msg[len - 1]) {
27                 fprintf(stderr, "Received invalid data from %s\n", source->name);
28                 return;
29         }
30
31         printf("%s says: %s\n", source->name, msg);
32 }
33
34 static void node_status(meshlink_handle_t *mesh, meshlink_node_t *node, bool reachable) {
35         (void)mesh;
36
37         if(reachable) {
38                 printf("%s joined.\n", node->name);
39         } else {
40                 printf("%s left.\n", node->name);
41         }
42 }
43
44 static meshlink_node_t **nodes;
45 static size_t nnodes;
46
47 static void parse_command(meshlink_handle_t *mesh, char *buf) {
48         char *arg = strchr(buf, ' ');
49
50         if(arg) {
51                 *arg++ = 0;
52         }
53
54         if(!strcasecmp(buf, "invite")) {
55                 char *invitation;
56
57                 if(!arg) {
58                         fprintf(stderr, "/invite requires an argument!\n");
59                         return;
60                 }
61
62                 invitation = meshlink_invite(mesh, NULL, arg);
63
64                 if(!invitation) {
65                         fprintf(stderr, "Could not invite '%s': %s\n", arg, meshlink_strerror(meshlink_errno));
66                         return;
67                 }
68
69                 printf("Invitation for %s: %s\n", arg, invitation);
70                 free(invitation);
71         } else if(!strcasecmp(buf, "join")) {
72                 if(!arg) {
73                         fprintf(stderr, "/join requires an argument!\n");
74                         return;
75                 }
76
77                 meshlink_stop(mesh);
78
79                 if(!meshlink_join(mesh, arg)) {
80                         fprintf(stderr, "Could not join using invitation: %s\n", meshlink_strerror(meshlink_errno));
81                 } else {
82                         fprintf(stderr, "Invitation accepted!\n");
83                 }
84
85                 if(!meshlink_start(mesh)) {
86                         fprintf(stderr, "Could not restart MeshLink: %s\n", meshlink_strerror(meshlink_errno));
87                         exit(1);
88                 }
89         } else if(!strcasecmp(buf, "kick")) {
90                 if(!arg) {
91                         fprintf(stderr, "/kick requires an argument!\n");
92                         return;
93                 }
94
95                 meshlink_node_t *node = meshlink_get_node(mesh, arg);
96
97                 if(!node) {
98                         fprintf(stderr, "Error looking up '%s': %s\n", arg, meshlink_strerror(meshlink_errno));
99                         return;
100                 }
101
102                 meshlink_blacklist(mesh, node);
103
104                 printf("Node '%s' blacklisted.\n", arg);
105         } else if(!strcasecmp(buf, "who")) {
106                 if(!arg) {
107                         nodes = meshlink_get_all_nodes(mesh, nodes, &nnodes);
108
109                         if(!nnodes) {
110                                 fprintf(stderr, "Could not get list of nodes: %s\n", meshlink_strerror(meshlink_errno));
111                         } else {
112                                 printf("%zu known nodes:", nnodes);
113
114                                 for(size_t i = 0; i < nnodes; i++) {
115                                         printf(" %s", nodes[i]->name);
116                                 }
117
118                                 printf("\n");
119                         }
120                 } else {
121                         meshlink_node_t *node = meshlink_get_node(mesh, arg);
122
123                         if(!node) {
124                                 fprintf(stderr, "Error looking up '%s': %s\n", arg, meshlink_strerror(meshlink_errno));
125                         } else {
126                                 printf("Node %s found\n", arg);
127                         }
128                 }
129         } else if(!strcasecmp(buf, "quit")) {
130                 printf("Bye!\n");
131                 fclose(stdin);
132         } else if(!strcasecmp(buf, "help")) {
133                 printf(
134                         "<name>: <message>     Send a message to the given node.\n"
135                         "                      Subsequent messages don't need the <name>: prefix.\n"
136                         "/invite <name>        Create an invitation for a new node.\n"
137                         "/join <invitation>    Join an existing mesh using an invitation.\n"
138                         "/kick <name>          Blacklist the given node.\n"
139                         "/who [<name>]         List all nodes or show information about the given node.\n"
140                         "/quit                 Exit this program.\n"
141                 );
142         } else {
143                 fprintf(stderr, "Unknown command '/%s'\n", buf);
144         }
145 }
146
147 static void parse_input(meshlink_handle_t *mesh, char *buf) {
148         static meshlink_node_t *destination;
149         size_t len;
150
151         if(!buf) {
152                 return;
153         }
154
155         // Remove newline.
156
157         len = strlen(buf);
158
159         if(len && buf[len - 1] == '\n') {
160                 buf[--len] = 0;
161         }
162
163         if(len && buf[len - 1] == '\r') {
164                 buf[--len] = 0;
165         }
166
167         // Ignore empty lines.
168
169         if(!len) {
170                 return;
171         }
172
173         // Commands start with '/'
174
175         if(*buf == '/') {
176                 parse_command(mesh, buf + 1);
177                 return;
178         }
179
180         // Lines in the form "name: message..." set the destination node.
181
182         char *msg = buf;
183         char *colon = strchr(buf, ':');
184
185         if(colon) {
186                 *colon = 0;
187                 msg = colon + 1;
188
189                 if(*msg == ' ') {
190                         msg++;
191                 }
192
193                 destination = meshlink_get_node(mesh, buf);
194
195                 if(!destination) {
196                         fprintf(stderr, "Error looking up '%s': %s\n", buf, meshlink_strerror(meshlink_errno));
197                         return;
198                 }
199         }
200
201         if(!destination) {
202                 fprintf(stderr, "Who are you talking to? Write 'name: message...'\n");
203                 return;
204         }
205
206         if(!meshlink_send(mesh, destination, msg, strlen(msg) + 1)) {
207                 fprintf(stderr, "Could not send message to '%s': %s\n", destination->name, meshlink_strerror(meshlink_errno));
208                 return;
209         }
210
211         printf("Message sent to '%s'.\n", destination->name);
212 }
213
214 int main(int argc, char *argv[]) {
215         const char *confbase = ".chat";
216         const char *nick = NULL;
217         char buf[1024];
218
219         if(argc > 1) {
220                 confbase = argv[1];
221         }
222
223         if(argc > 2) {
224                 nick = argv[2];
225         }
226
227         meshlink_set_log_cb(NULL, MESHLINK_INFO, log_message);
228
229         meshlink_handle_t *mesh = meshlink_open(confbase, nick, "chat", DEV_CLASS_STATIONARY);
230
231         if(!mesh) {
232                 fprintf(stderr, "Could not open MeshLink: %s\n", meshlink_strerror(meshlink_errno));
233                 return 1;
234         }
235
236         meshlink_set_receive_cb(mesh, receive);
237         meshlink_set_node_status_cb(mesh, node_status);
238         meshlink_set_log_cb(mesh, MESHLINK_INFO, log_message);
239
240         if(!meshlink_start(mesh)) {
241                 fprintf(stderr, "Could not start MeshLink: %s\n", meshlink_strerror(meshlink_errno));
242                 return 1;
243         }
244
245         printf("Chat started.\nType /help for a list of commands.\n");
246
247         while(fgets(buf, sizeof(buf), stdin)) {
248                 parse_input(mesh, buf);
249         }
250
251         printf("Chat stopping.\n");
252
253         meshlink_stop(mesh);
254         meshlink_close(mesh);
255
256         return 0;
257 }