Changed script for gredler coding style
[olsrd.git] / src / olsr_switch / ohs_cmd.c
1
2 /*
3  * The olsr.org Optimized Link-State Routing daemon(olsrd)
4  * Copyright (c) 2005, Andreas Tonnesen(andreto@olsr.org)
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  *
11  * * Redistributions of source code must retain the above copyright
12  *   notice, this list of conditions and the following disclaimer.
13  * * Redistributions in binary form must reproduce the above copyright
14  *   notice, this list of conditions and the following disclaimer in
15  *   the documentation and/or other materials provided with the
16  *   distribution.
17  * * Neither the name of olsr.org, olsrd nor the names of its
18  *   contributors may be used to endorse or promote products derived
19  *   from this software without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
24  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
25  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
26  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
27  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
28  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
29  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
31  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
32  * POSSIBILITY OF SUCH DAMAGE.
33  *
34  * Visit http://www.olsr.org for more information.
35  *
36  * If you find this software useful feel free to make a donation
37  * to the project. For more information see the website or contact
38  * the copyright holders.
39  *
40  */
41
42 #include "olsr_host_switch.h"
43 #include "olsr_types.h"
44 #include "commands.h"
45 #include "link_rules.h"
46 #include "ipcalc.h"
47
48 #include <string.h>
49 #include <stdlib.h>
50 #include <sys/socket.h>
51 #include <netinet/in.h>
52 #include <arpa/inet.h>
53 #include <sys/types.h>
54 #include <sys/stat.h>
55 #include <unistd.h>
56 #include <errno.h>
57
58 #define TOK_BUF_SIZE 500
59 static char tok_buf[TOK_BUF_SIZE];
60
61 #define MAX_OLSRD_ARGS 10
62 static char olsrd_path[FILENAME_MAX];
63
64 static int
65 get_next_token(const char *src, char *dst, size_t buflen)
66 {
67   int i = 0, j = 0;
68
69   dst[0] = 0;
70   /* Skip leading spaces */
71   while (src[j] == ' ' && src[j] != 0) {
72     j++;
73   }
74
75   src += j;
76   i = 0;
77   while ((src[i] != ' ') && (src[i] != 0) && (i < ((int)buflen - 1))) {
78     dst[i] = src[i];
79     i++;
80   }
81   dst[i] = 0;
82
83   //if(strlen(dst))
84   //printf("Extracted token: %s\n", dst);
85   return i + j;
86 }
87
88 int
89 ohs_set_olsrd_path(const char *path)
90 {
91   strscpy(olsrd_path, path, sizeof(olsrd_path));
92   return 0;
93 }
94
95 #ifdef WIN32
96 int
97 ohs_cmd_olsrd(char *args __attribute__ ((unused)))
98 {
99   printf("olsrd command not available in windows version\nStart instances manually\n");
100   return 0;
101 }
102 #else
103 int
104 ohs_cmd_olsrd(const char *args)
105 {
106   const char *olsrd_args[MAX_OLSRD_ARGS];
107   struct in_addr iaddr;
108
109   args += get_next_token(args, tok_buf, TOK_BUF_SIZE);
110
111   if (!strlen(tok_buf))
112     goto print_usage;
113
114   /* Start olsrd instance */
115   if (!strncmp(tok_buf, "start", strlen("start"))) {
116     int argc = 0, i = 0;
117
118     args += get_next_token(args, tok_buf, TOK_BUF_SIZE);
119
120     if (!strlen(tok_buf))
121       goto print_usage;
122
123     if (!inet_aton(tok_buf, &iaddr)) {
124       printf("Invalid IP %s\n", tok_buf);
125       goto print_usage;
126     }
127
128     olsrd_args[argc++] = olsrd_path;
129
130     if (1) {                    /* config file is set */
131       olsrd_args[argc++] = "-f";
132       olsrd_args[argc++] = "/etc/olsrd-emu.conf";
133     }
134     olsrd_args[argc++] = "-hemu";
135     olsrd_args[argc++] = tok_buf;
136
137     olsrd_args[argc++] = "-d";
138     olsrd_args[argc++] = "0";
139     olsrd_args[argc++] = "-nofork";
140     olsrd_args[argc] = NULL;
141
142     printf("Executing: %s", olsrd_path);
143     for (i = 0; i < argc; i++)
144       printf(" %s", olsrd_args[i]);
145     printf("\n");
146
147     if (fork())
148       return 1;
149
150     if (execve(olsrd_path, (char *const *)olsrd_args, NULL) < 0) {
151       printf("Error executing olsrd: %s\n", strerror(errno));
152       exit(1);
153     }
154   }
155   /* Stop olsrd instance */
156   else if (!strncmp(tok_buf, "stop", strlen("stop"))) {
157     struct ohs_connection *oc;
158
159     args += get_next_token(args, tok_buf, TOK_BUF_SIZE);
160
161     if (!strlen(tok_buf))
162       goto print_usage;
163
164     if (!inet_aton(tok_buf, &iaddr)) {
165       printf("Invalid IP %s\n", tok_buf);
166       goto print_usage;
167     }
168
169     oc = get_client_by_addr((union olsr_ip_addr *)&iaddr.s_addr);
170
171     if (!oc) {
172       printf("No such client: %s\n", tok_buf);
173       return -1;
174     }
175     ohs_delete_connection(oc);
176
177     return 1;
178   }
179   /* Set olsrd binary path */
180   else if (!strncmp(tok_buf, "setb", strlen("setb"))) {
181     struct stat sbuf;
182
183     args += get_next_token(args, tok_buf, TOK_BUF_SIZE);
184
185     if (!strlen(tok_buf))
186       goto print_usage;
187
188     if (stat(tok_buf, &sbuf) < 0) {
189       printf("Error setting binary \"%s\": %s\n", tok_buf, strerror(errno));
190       return -1;
191     }
192
193     if ((sbuf.st_mode & S_IFDIR) || !(sbuf.st_mode & S_IXUSR)) {
194       printf("Error setting binary \"%s\": Not a regular execuatble file!\n", tok_buf);
195       return -1;
196     }
197
198     printf("New olsrd binary path:\"%s\"\n", tok_buf);
199     ohs_set_olsrd_path(tok_buf);
200
201     return 1;
202
203   }
204   /* Set arguments */
205   else if (!strncmp(tok_buf, "seta", strlen("seta"))) {
206     printf("Error - NOT IMPLEMENTED YET\n");
207     return 1;
208   }
209   /* Show settings */
210   else if (!strncmp(tok_buf, "show", strlen("show"))) {
211     printf("olsrd command settings:\n\tBinary path: %s\n\tArguments  : \n", olsrd_path);
212     return 1;
213   }
214
215 print_usage:
216   printf("Usage: olsrd [start|stop|show|setb|seta] [IP|path|args]\n");
217   return 0;
218 }
219 #endif
220
221 int
222 ohs_cmd_link(const char *args)
223 {
224   olsr_u8_t bi = 0, wildc_src = 0, wildc_dst = 0;
225   struct ohs_connection *src, *dst;
226   struct in_addr iaddr;
227   int qual;
228   struct ohs_ip_link *link, *inv_link;
229
230   args += get_next_token(args, tok_buf, TOK_BUF_SIZE);
231
232   if (!strlen(tok_buf)) {
233     goto print_usage;
234   }
235   if (!strncmp(tok_buf, "bi", strlen("bi"))) {
236     bi = 1;
237     args += get_next_token(args, tok_buf, TOK_BUF_SIZE);
238
239     if (!strlen(tok_buf)) {
240       goto print_usage;
241     }
242   }
243
244   if (tok_buf[0] == '*') {
245     wildc_src = 1;
246     src = ohs_conns;
247   } else {
248     if (!inet_aton(tok_buf, &iaddr)) {
249       printf("Invalid src IP %s\n", tok_buf);
250       return -1;
251     }
252
253     src = get_client_by_addr((union olsr_ip_addr *)&iaddr.s_addr);
254
255     if (!src) {
256       printf("No such client: %s!\n", tok_buf);
257       return -1;
258     }
259   }
260
261   args += get_next_token(args, tok_buf, TOK_BUF_SIZE);
262
263   if (!strlen(tok_buf)) {
264     goto print_usage;
265   }
266
267   if (tok_buf[0] == '*') {
268     wildc_dst = 1;
269     dst = ohs_conns;
270   } else {
271     if (!inet_aton(tok_buf, &iaddr)) {
272       printf("Invalid src IP %s\n", tok_buf);
273       return -1;
274     }
275
276     dst = get_client_by_addr((union olsr_ip_addr *)&iaddr.s_addr);
277     if (!dst) {
278       printf("No such client: %s!\n", tok_buf);
279       return -1;
280     }
281   }
282
283   args += get_next_token(args, tok_buf, TOK_BUF_SIZE);
284
285   if (!strlen(tok_buf)) {
286     goto print_usage;
287   }
288
289   /* No use for bi if both src and dst are widcards */
290   if (wildc_src && wildc_dst) {
291     bi = 0;
292   }
293
294   qual = atoi(tok_buf);
295
296   if (qual < 0 || qual > 100) {
297     printf("Link quality out of range(0-100)\n");
298     return -1;
299   }
300
301   while (src) {
302     while (dst) {
303       struct ipaddr_str srcaddrstr, dstaddrstr;
304
305       if (src != dst) {
306         link = get_link(src, &dst->ip_addr);
307         inv_link = bi ? get_link(dst, &src->ip_addr) : NULL;
308         if (qual == 100) {
309           /* Remove link entry */
310           if (link) {
311             remove_link(src, link);
312           }
313           if (inv_link) {
314             remove_link(dst, inv_link);
315           }
316         } else {
317           if (!link) {
318             /* Create new link */
319             link = add_link(src, dst);
320           }
321
322           link->quality = qual;
323
324           if (bi) {
325             if (!inv_link) {
326               /* Create new link */
327               inv_link = add_link(dst, src);
328             }
329             inv_link->quality = qual;
330           }
331         }
332         printf("%s %sdirectional link(s) %s %c=> %s quality %d\n",
333                (qual == 100) ? "Removing" : "Setting",
334                bi ? "bi" : "uni", olsr_ip_to_string(&srcaddrstr,
335                                                     &src->ip_addr),
336                bi ? '<' : '=', olsr_ip_to_string(&dstaddrstr, &dst->ip_addr), qual);
337       }
338       if (wildc_dst) {
339         dst = dst->next;
340       } else {
341         break;
342       }
343     }
344     dst = wildc_dst ? ohs_conns : dst;
345     src = wildc_src ? src->next : NULL;
346   }
347
348   return 1;
349 print_usage:
350   printf("link <bi> srcIP dstIP [0-100]");
351   return -1;
352 }
353
354 int
355 ohs_cmd_list(const char *args)
356 {
357   struct ohs_connection *oc = ohs_conns;
358
359   args += get_next_token(args, tok_buf, TOK_BUF_SIZE);
360
361   if (!strlen(tok_buf) || !strncmp(tok_buf, "clients", strlen("clients"))) {
362     printf("All connected clients:\n");
363
364     while (oc) {
365       struct ipaddr_str addrstr;
366       printf("\t%s - Rx: %d Tx: %d LinkCnt: %d\n", olsr_ip_to_string(&addrstr, &oc->ip_addr), oc->rx, oc->tx, oc->linkcnt);
367       oc = oc->next;
368     }
369   } else if (!strncmp(tok_buf, "links", strlen("links"))) {
370     printf("All configured links:\n");
371     while (oc) {
372       struct ohs_ip_link *links = oc->links;
373       while (links) {
374         struct ipaddr_str addrstr, dststr;
375         printf("\t%s => %s Quality: %d\n",
376                olsr_ip_to_string(&addrstr, &oc->ip_addr), olsr_ip_to_string(&dststr, &links->dst), links->quality);
377
378         links = links->next;
379       }
380       oc = oc->next;
381     }
382   } else {
383     printf("list [clients|links]");
384     return -1;
385   }
386   return 1;
387 }
388
389 int
390 ohs_cmd_help(const char *args)
391 {
392   int i;
393
394   args += get_next_token(args, tok_buf, TOK_BUF_SIZE);
395
396   if (!strlen(tok_buf)) {
397     printf("Olsrd host switch version %s\n", OHS_VERSION);
398     printf("Available commands:\n");
399
400     for (i = 0; ohs_commands[i].cmd; i++) {
401       if (ohs_commands[i].helptext_brief)
402         printf("\t%s - %s\n", ohs_commands[i].cmd, ohs_commands[i].helptext_brief);
403     }
404     printf("\nType 'help cmd' for help on a specific command\n");
405   } else {
406     for (i = 0; ohs_commands[i].cmd; i++) {
407       if (!strncmp(tok_buf, ohs_commands[i].cmd, strlen(ohs_commands[i].cmd))) {
408         printf("Usage: %s\nDescription:\n%s\n", ohs_commands[i].syntax, ohs_commands[i].helptext_long);
409         return 1;
410       }
411     }
412
413     printf("Usage: help <command>\n");
414   }
415
416   return i;
417 }
418
419 int
420 ohs_cmd_log(const char *args)
421 {
422   olsr_u8_t set = 0;
423
424   args += get_next_token(args, tok_buf, TOK_BUF_SIZE);
425
426   if (strlen(tok_buf)
427       && ((set = !strncmp(tok_buf, "set", strlen("set")))
428           || !strncmp(tok_buf, "unset", strlen("unset")))) {
429     olsr_u32_t new_bit = 0;
430
431     args += get_next_token(args, tok_buf, TOK_BUF_SIZE);
432
433     if (!strlen(tok_buf))
434       goto print_usage;
435
436     if (!strncmp(tok_buf, "CON", strlen("CON")))
437       new_bit = LOG_CONNECT;
438     else if (!strncmp(tok_buf, "FOR", strlen("FOR")))
439       new_bit = LOG_FORWARD;
440     else if (!strncmp(tok_buf, "LIN", strlen("LIN")))
441       new_bit = LOG_LINK;
442
443     if (!new_bit)
444       goto print_usage;
445
446     if (set)
447       logbits |= new_bit;
448     else
449       logbits &= ~new_bit;
450
451     printf("%s log bit: 0x%08x, new log: 0x%08x\n", set ? "Setting" : "Removing", new_bit, logbits);
452
453   } else {
454     if (strlen(tok_buf))
455       goto print_usage;
456
457     printf("Log: (0x%08x) ", logbits);
458     if (logbits & LOG_CONNECT)
459       printf("CONNECT ");
460     if (logbits & LOG_FORWARD)
461       printf("FORWARD ");
462     if (logbits & LOG_LINK)
463       printf("LINK ");
464
465     printf("\n");
466   }
467   return 1;
468
469 print_usage:
470   printf("Usage: log <[set|unset] [CONNECT|FORWARD|LINK]>\n");
471   return 0;
472
473 }
474
475 int
476 ohs_cmd_exit(const char *args __attribute__ ((unused)))
477 {
478
479   printf("Exitting... bye-bye!\n");
480
481   ohs_close(0);
482 }
483
484 void
485 ohs_parse_command(void)
486 {
487   static char cmd_line[500];
488   static int cmd_len = 0;
489   char *args;
490   char cmd_token[20];
491   int i;
492 #if defined WIN32
493   char c;
494   unsigned long Read;
495   INPUT_RECORD InRec;
496   KEY_EVENT_RECORD *KeyEventRec;
497 #endif
498
499 #if defined WIN32
500   if (!ReadConsoleInput(GetStdHandle(STD_INPUT_HANDLE), &InRec, sizeof(InRec), &Read)) {
501     fprintf(stderr, "ReadConsoleInput failed: %s\n", strerror(GetLastError()));
502     return;
503   }
504
505   if (InRec.EventType != KEY_EVENT)
506     return;
507
508   KeyEventRec = &InRec.Event.KeyEvent;
509
510   if (!KeyEventRec->bKeyDown)
511     return;
512
513   c = KeyEventRec->uChar.AsciiChar;
514
515   if (c == 8) {
516     if (cmd_len == 0)
517       return;
518
519     cmd_len--;
520
521     fputc(8, stdout);
522     fputc(32, stdout);
523     fputc(8, stdout);
524
525     fflush(stdout);
526
527     return;
528   }
529
530   fputc(c, stdout);
531   fflush(stdout);
532
533   if (c != '\n' && c != '\r' && cmd_len < (int)sizeof(cmd_line) - 1)
534     cmd_line[cmd_len++] = (char)c;
535
536   else
537 #else
538   if (fgets(cmd_line, sizeof(cmd_line), stdin) == NULL) {
539     ohs_cmd_exit(NULL);
540   }
541
542   for (cmd_len = 0; cmd_line[cmd_len] != 0 && cmd_line[cmd_len] != '\n'; cmd_len++);
543 #endif
544
545   {
546     cmd_line[cmd_len] = 0;
547     cmd_len = 0;
548
549     args = cmd_line + get_next_token(cmd_line, cmd_token, sizeof(cmd_token));
550
551     for (i = 0; ohs_commands[i].cmd != NULL; i++) {
552       if (strcmp(cmd_token, ohs_commands[i].cmd) == 0) {
553         if (ohs_commands[i].cmd_cb != NULL)
554           ohs_commands[i].cmd_cb(args);
555
556         else
557           printf("No action registered on cmd %s!\n", cmd_token);
558
559         break;
560       }
561     }
562
563     if (ohs_commands[i].cmd == NULL)
564       printf("%s: no such cmd!\n", cmd_token);
565
566     printf("\n> ");
567     fflush(stdout);
568   }
569 }
570
571 /*
572  * Local Variables:
573  * c-basic-offset: 2
574  * indent-tabs-mode: nil
575  * End:
576  */