2707e171f5793955a93c17f0e196f52df7bc073d
[l2tpns.git] / cli.c
1 // L2TPNS Command Line Interface
2 // vim: sw=4 ts=8
3
4 char const *cvs_name = "$Name: $";
5 char const *cvs_id_cli = "$Id: cli.c,v 1.9 2004-07-08 16:54:35 bodea Exp $";
6
7 #include <stdio.h>
8 #include <stdarg.h>
9 #include <sys/file.h>
10 #include <sys/stat.h>
11 #include <syslog.h>
12 #include <malloc.h>
13 #include <sched.h>
14 #include <string.h>
15 #include <ctype.h>
16 #include <stdlib.h>
17 #include <time.h>
18 #include <arpa/inet.h>
19 #include <errno.h>
20 #include <sys/socket.h>
21 #include <sys/types.h>
22 #include <signal.h>
23 #include <unistd.h>
24 #include <dlfcn.h>
25 #include <libcli.h>
26 #include "l2tpns.h"
27 #include "util.h"
28 #include "cluster.h"
29 #include "tbf.h"
30 #include "ll.h"
31 #ifdef BGP
32 #include "bgp.h"
33 #endif
34
35 extern tunnelt *tunnel;
36 extern sessiont *session;
37 extern radiust *radius;
38 extern ippoolt *ip_address_pool;
39 extern struct Tstats *_statistics;
40 struct cli_def *cli = NULL;
41 int cli_quit = 0;
42 extern int clifd, udpfd, tunfd, snoopfd, ifrfd, cluster_sockfd;
43 extern int *radfds;
44 extern struct configt *config;
45 extern struct config_descriptt config_values[];
46 #ifdef RINGBUFFER
47 extern struct Tringbuffer *ringbuffer;
48 #endif
49 extern struct cli_session_actions *cli_session_actions;
50 extern struct cli_tunnel_actions *cli_tunnel_actions;
51
52 char *debug_levels[] = {
53 "CRIT",
54 "ERROR",
55 "WARN",
56 "INFO",
57 "CALL",
58 "DATA",
59 };
60
61 struct
62 {
63 char critical;
64 char error;
65 char warning;
66 char info;
67 char calls;
68 char data;
69 } debug_flags;
70
71 int debug_session;
72 int debug_tunnel;
73 int debug_rb_tail;
74 FILE *save_config_fh;
75
76 int cmd_show_session(struct cli_def *cli, char *command, char **argv, int argc);
77 int cmd_show_tunnels(struct cli_def *cli, char *command, char **argv, int argc);
78 int cmd_show_users(struct cli_def *cli, char *command, char **argv, int argc);
79 int cmd_show_radius(struct cli_def *cli, char *command, char **argv, int argc);
80 int cmd_show_counters(struct cli_def *cli, char *command, char **argv, int argc);
81 int cmd_show_version(struct cli_def *cli, char *command, char **argv, int argc);
82 int cmd_show_pool(struct cli_def *cli, char *command, char **argv, int argc);
83 int cmd_show_run(struct cli_def *cli, char *command, char **argv, int argc);
84 int cmd_show_banana(struct cli_def *cli, char *command, char **argv, int argc);
85 int cmd_show_plugins(struct cli_def *cli, char *command, char **argv, int argc);
86 int cmd_show_throttle(struct cli_def *cli, char *command, char **argv, int argc);
87 int cmd_show_cluster(struct cli_def *cli, char *command, char **argv, int argc);
88 int cmd_write_memory(struct cli_def *cli, char *command, char **argv, int argc);
89 int cmd_clear_counters(struct cli_def *cli, char *command, char **argv, int argc);
90 int cmd_drop_user(struct cli_def *cli, char *command, char **argv, int argc);
91 int cmd_drop_tunnel(struct cli_def *cli, char *command, char **argv, int argc);
92 int cmd_drop_session(struct cli_def *cli, char *command, char **argv, int argc);
93 int cmd_snoop(struct cli_def *cli, char *command, char **argv, int argc);
94 int cmd_no_snoop(struct cli_def *cli, char *command, char **argv, int argc);
95 int cmd_throttle(struct cli_def *cli, char *command, char **argv, int argc);
96 int cmd_no_throttle(struct cli_def *cli, char *command, char **argv, int argc);
97 int cmd_debug(struct cli_def *cli, char *command, char **argv, int argc);
98 int cmd_no_debug(struct cli_def *cli, char *command, char **argv, int argc);
99 int cmd_set(struct cli_def *cli, char *command, char **argv, int argc);
100 int cmd_load_plugin(struct cli_def *cli, char *command, char **argv, int argc);
101 int cmd_remove_plugin(struct cli_def *cli, char *command, char **argv, int argc);
102 int cmd_uptime(struct cli_def *cli, char *command, char **argv, int argc);
103 int regular_stuff(struct cli_def *cli);
104
105 void init_cli(char *hostname)
106 {
107 FILE *f;
108 char buf[4096];
109 struct cli_command *c;
110 struct cli_command *c2;
111 int on = 1;
112 struct sockaddr_in addr;
113
114 cli = cli_init();
115 if (hostname && *hostname)
116 cli_set_hostname(cli, hostname);
117 else
118 cli_set_hostname(cli, "l2tpns");
119
120 c = cli_register_command(cli, NULL, "show", NULL, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, NULL);
121 cli_register_command(cli, c, "banana", cmd_show_banana, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Show a banana");
122 #ifdef BGP
123 cli_register_command(cli, c, "bgp", cmd_show_bgp, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Show BGP status");
124 #endif /* BGP */
125 cli_register_command(cli, c, "cluster", cmd_show_cluster, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Show cluster information");
126 cli_register_command(cli, c, "ipcache", cmd_show_ipcache, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Show contents of the IP cache");
127 cli_register_command(cli, c, "plugins", cmd_show_plugins, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "List all installed plugins");
128 cli_register_command(cli, c, "pool", cmd_show_pool, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Show the IP address allocation pool");
129 cli_register_command(cli, c, "radius", cmd_show_radius, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Show active radius queries");
130 cli_register_command(cli, c, "running-config", cmd_show_run, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Show the currently running configuration");
131 cli_register_command(cli, c, "session", cmd_show_session, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Show a list of sessions or details for a single session");
132 cli_register_command(cli, c, "tbf", cmd_show_tbf, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "List all token bucket filters in use");
133 cli_register_command(cli, c, "throttle", cmd_show_throttle, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "List all throttled sessions and associated TBFs");
134 cli_register_command(cli, c, "tunnels", cmd_show_tunnels, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Show a list of tunnels or details for a single tunnel");
135 cli_register_command(cli, c, "users", cmd_show_users, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Show a list of all connected users or details of selected user");
136 cli_register_command(cli, c, "version", cmd_show_version, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Show currently running software version");
137
138 c2 = cli_register_command(cli, c, "histogram", NULL, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, NULL);
139 cli_register_command(cli, c2, "idle", cmd_show_hist_idle, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Show histogram of session idle times");
140 cli_register_command(cli, c2, "open", cmd_show_hist_open, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Show histogram of session durations");
141
142 #ifdef STATISTICS
143 cli_register_command(cli, c, "counters", cmd_show_counters, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Display all the internal counters and running totals");
144
145 c = cli_register_command(cli, NULL, "clear", NULL, PRIVILEGE_PRIVILEGED, MODE_EXEC, NULL);
146 cli_register_command(cli, c, "counters", cmd_clear_counters, PRIVILEGE_PRIVILEGED, MODE_EXEC, "Clear internal counters");
147 #endif
148
149 cli_register_command(cli, NULL, "uptime", cmd_uptime, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Show uptime and bandwidth utilisation");
150
151 c = cli_register_command(cli, NULL, "write", NULL, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, NULL);
152 cli_register_command(cli, c, "memory", cmd_write_memory, PRIVILEGE_PRIVILEGED, MODE_EXEC, "Save the running config to flash");
153 cli_register_command(cli, c, "terminal", cmd_show_run, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Show the running config");
154
155 cli_register_command(cli, NULL, "snoop", cmd_snoop, PRIVILEGE_PRIVILEGED, MODE_EXEC, "Temporarily enable interception for a user");
156 cli_register_command(cli, NULL, "throttle", cmd_throttle, PRIVILEGE_PRIVILEGED, MODE_EXEC, "Temporarily enable throttling for a user");
157 cli_register_command(cli, NULL, "debug", cmd_debug, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Set the level of logging that is shown on the console");
158
159 c = cli_register_command(cli, NULL, "suspend", NULL, PRIVILEGE_PRIVILEGED, MODE_EXEC, NULL);
160 cli_register_command(cli, c, "bgp", cmd_suspend_bgp, PRIVILEGE_PRIVILEGED, MODE_EXEC, "Withdraw routes from BGP peer");
161
162 c = cli_register_command(cli, NULL, "no", NULL, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, NULL);
163 cli_register_command(cli, c, "snoop", cmd_no_snoop, PRIVILEGE_PRIVILEGED, MODE_EXEC, "Temporarily disable interception for a user");
164 cli_register_command(cli, c, "throttle", cmd_no_throttle, PRIVILEGE_PRIVILEGED, MODE_EXEC, "Temporarily disable throttling for a user");
165 cli_register_command(cli, c, "debug", cmd_no_debug, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, "Turn off logging of a certain level of debugging");
166 c2 = cli_register_command(cli, c, "suspend", NULL, PRIVILEGE_PRIVILEGED, MODE_EXEC, NULL);
167 cli_register_command(cli, c2, "bgp", cmd_no_suspend_bgp, PRIVILEGE_PRIVILEGED, MODE_EXEC, "Advertise routes to BGP peer");
168
169 c = cli_register_command(cli, NULL, "drop", NULL, PRIVILEGE_PRIVILEGED, MODE_EXEC, NULL);
170 cli_register_command(cli, c, "user", cmd_drop_user, PRIVILEGE_PRIVILEGED, MODE_EXEC, "Disconnect a user");
171 cli_register_command(cli, c, "tunnel", cmd_drop_tunnel, PRIVILEGE_PRIVILEGED, MODE_EXEC, "Disconnect a tunnel and all sessions on that tunnel");
172 cli_register_command(cli, c, "session", cmd_drop_session, PRIVILEGE_PRIVILEGED, MODE_EXEC, "Disconnect a session");
173
174 c = cli_register_command(cli, NULL, "restart", NULL, PRIVILEGE_PRIVILEGED, MODE_EXEC, NULL);
175 cli_register_command(cli, c, "bgp", cmd_restart_bgp, PRIVILEGE_PRIVILEGED, MODE_EXEC, "Restart BGP");
176
177 c = cli_register_command(cli, NULL, "load", NULL, PRIVILEGE_PRIVILEGED, MODE_CONFIG, NULL);
178 cli_register_command(cli, c, "plugin", cmd_load_plugin, PRIVILEGE_PRIVILEGED, MODE_CONFIG, "Load a plugin");
179
180 c = cli_register_command(cli, NULL, "remove", NULL, PRIVILEGE_PRIVILEGED, MODE_CONFIG, NULL);
181 cli_register_command(cli, c, "plugin", cmd_remove_plugin, PRIVILEGE_PRIVILEGED, MODE_CONFIG, "Remove a plugin");
182
183 cli_register_command(cli, NULL, "set", cmd_set, PRIVILEGE_PRIVILEGED, MODE_CONFIG, "Set a configuration variable");
184
185 // Enable regular processing
186 cli_regular(cli, regular_stuff);
187
188 if (!(f = fopen(CLIUSERS, "r")))
189 {
190 log(0, 0, 0, 0, "WARNING! No users specified. Command-line access is open to all\n");
191 }
192 else
193 {
194 while (fgets(buf, 4096, f))
195 {
196 char *p;
197 if (*buf == '#') continue;
198 if ((p = strchr(buf, '\r'))) *p = 0;
199 if ((p = strchr(buf, '\n'))) *p = 0;
200 if (!*buf) continue;
201 if (!(p = strchr((char *)buf, ':'))) continue;
202 *p++ = 0;
203 if (!strcmp(buf, "enable"))
204 {
205 cli_allow_enable(cli, p);
206 log(3, 0, 0, 0, "Setting enable password\n");
207 }
208 else
209 {
210 cli_allow_user(cli, buf, p);
211 log(3, 0, 0, 0, "Allowing user %s to connect to the CLI\n", buf);
212 }
213 }
214 fclose(f);
215 }
216
217 memset(&addr, 0, sizeof(addr));
218 clifd = socket(PF_INET, SOCK_STREAM, 6);
219 setsockopt(clifd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
220 {
221 int flags;
222 // Set cli fd as non-blocking
223 flags = fcntl(clifd, F_GETFL, 0);
224 fcntl(clifd, F_SETFL, flags | O_NONBLOCK);
225 }
226 addr.sin_family = AF_INET;
227 addr.sin_port = htons(23);
228 if (bind(clifd, (void *) &addr, sizeof(addr)) < 0)
229 {
230 log(0, 0, 0, 0, "Error listening on cli port 23: %s\n", strerror(errno));
231 return;
232 }
233 listen(clifd, 10);
234 }
235
236 void cli_do(int sockfd)
237 {
238 int i;
239
240 if (fork()) return;
241 if (config->scheduler_fifo)
242 {
243 int ret;
244 struct sched_param params = {0};
245 params.sched_priority = 0;
246 if ((ret = sched_setscheduler(0, SCHED_OTHER, &params)) == 0)
247 {
248 log(3, 0, 0, 0, "Dropped FIFO scheduler\n");
249 }
250 else
251 {
252 log(0, 0, 0, 0, "Error setting scheduler to OTHER: %s\n", strerror(errno));
253 log(0, 0, 0, 0, "This is probably really really bad.\n");
254 }
255 }
256
257 signal(SIGPIPE, SIG_DFL);
258 signal(SIGCHLD, SIG_DFL);
259 signal(SIGHUP, SIG_DFL);
260 signal(SIGUSR1, SIG_DFL);
261 signal(SIGQUIT, SIG_DFL);
262 signal(SIGKILL, SIG_DFL);
263 signal(SIGALRM, SIG_DFL);
264 signal(SIGTERM, SIG_DFL);
265
266 // Close sockets
267 if (udpfd) close(udpfd); udpfd = 0;
268 if (tunfd) close(tunfd); tunfd = 0;
269 if (snoopfd) close(snoopfd); snoopfd = 0;
270 for (i = 0; i < config->num_radfds; i++)
271 if (radfds[i]) close(radfds[i]);
272 if (ifrfd) close(ifrfd); ifrfd = 0;
273 if (cluster_sockfd) close(cluster_sockfd); cluster_sockfd = 0;
274 if (clifd) close(clifd); clifd = 0;
275 #ifdef BGP
276 for (i = 0; i < BGP_NUM_PEERS; i++)
277 if (bgp_peers[i].sock != -1)
278 close(bgp_peers[i].sock);
279 #endif /* BGP */
280
281 {
282 int require_auth = 1;
283 struct sockaddr_in addr;
284 int l = sizeof(addr);
285 if (getpeername(sockfd, (struct sockaddr *)&addr, &l) == 0)
286 {
287 log(3, 0, 0, 0, "Accepted connection to CLI from %s\n", inet_toa(addr.sin_addr.s_addr));
288 require_auth = addr.sin_addr.s_addr != inet_addr("127.0.0.1");
289 }
290 else
291 log(0, 0, 0, 0, "getpeername() failed on cli socket. Requiring authentication: %s\n", strerror(errno));
292
293 if (require_auth)
294 {
295 log(3, 0, 0, 0, "CLI is remote, requiring authentication\n");
296 if (!cli->users) /* paranoia */
297 {
298 log(0, 0, 0, 0, "No users for remote authentication! Exiting CLI\n");
299 exit(0);
300 }
301 }
302 else
303 {
304 /* no username/pass required */
305 cli->users = 0;
306 }
307 }
308
309 debug_session = 0;
310 debug_tunnel = 0;
311 #ifdef RINGBUFFER
312 debug_rb_tail = ringbuffer->tail;
313 #endif
314 memset(&debug_flags, 0, sizeof(debug_flags));
315 debug_flags.critical = 1;
316
317 {
318 char prompt[1005];
319 snprintf(prompt, 1005, "l2tpns> ");
320 cli_loop(cli, sockfd, prompt);
321 }
322
323 close(sockfd);
324 log(3, 0, 0, 0, "Closed CLI connection\n");
325 exit(0);
326 }
327
328 void cli_print_log(struct cli_def *cli, char *string)
329 {
330 log(3, 0, 0, 0, "%s\n", string);
331 }
332
333 void cli_do_file(FILE *fh)
334 {
335 log(3, 0, 0, 0, "Reading configuration file\n");
336 cli_print_callback(cli, cli_print_log);
337 cli_file(cli, fh, PRIVILEGE_PRIVILEGED, MODE_CONFIG);
338 cli_print_callback(cli, NULL);
339 }
340
341 int cli_arg_help(struct cli_def *cli, int cr_ok, char *entry, ...)
342 {
343 va_list ap;
344 char *desc;
345 char buf[16];
346 char *p;
347
348 va_start(ap, entry);
349 while (entry)
350 {
351 /* allow one %d */
352 if ((p = strchr(entry, '%')) && !strchr(p+1, '%') && p[1] == 'd')
353 {
354 int v = va_arg(ap, int);
355 snprintf(buf, sizeof(buf), entry, v);
356 p = buf;
357 }
358 else
359 p = entry;
360
361 desc = va_arg(ap, char *);
362 if (desc && *desc)
363 cli_print(cli, " %-20s %s", p, desc);
364 else
365 cli_print(cli, " %s", p);
366
367 entry = desc ? va_arg(ap, char *) : 0;
368 }
369
370 va_end(ap);
371 if (cr_ok)
372 cli_print(cli, " <cr>");
373
374 return CLI_OK;
375 }
376
377 int cmd_show_session(struct cli_def *cli, char *command, char **argv, int argc)
378 {
379 int i;
380
381 if (CLI_HELP_REQUESTED)
382 return cli_arg_help(cli, 1,
383 "<1-%d>", MAXSESSION-1, "Show specific session by id",
384 NULL);
385
386 time(&time_now);
387 if (argc > 0)
388 {
389 // Show individual session
390 for (i = 0; i < argc; i++)
391 {
392 unsigned int s;
393 s = atoi(argv[i]);
394 if (s <= 0 || s >= MAXSESSION)
395 {
396 cli_print(cli, "Invalid session id \"%s\"", argv[i]);
397 continue;
398 }
399 cli_print(cli, "\r\nSession %d:", s);
400 cli_print(cli, " User: %s", session[s].user[0] ? session[s].user : "none");
401 cli_print(cli, " Calling Num: %s", session[s].calling);
402 cli_print(cli, " Called Num: %s", session[s].called);
403 cli_print(cli, " Tunnel ID: %d", session[s].tunnel);
404 cli_print(cli, " IP address: %s", inet_toa(htonl(session[s].ip)));
405 cli_print(cli, " HSD sid: %lu", session[s].sid);
406 cli_print(cli, " Idle time: %u seconds", abs(time_now - session[s].last_packet));
407 cli_print(cli, " Next Recv: %u", session[s].nr);
408 cli_print(cli, " Next Send: %u", session[s].ns);
409 cli_print(cli, " Bytes In/Out: %lu/%lu", (unsigned long)session[s].total_cout, (unsigned long)session[s].total_cin);
410 cli_print(cli, " Pkts In/Out: %lu/%lu", (unsigned long)session[s].pout, (unsigned long)session[s].pin);
411 cli_print(cli, " MRU: %d", session[s].mru);
412 cli_print(cli, " Radius Session: %u", session[s].radius);
413 cli_print(cli, " Rx Speed: %lu", session[s].rx_connect_speed);
414 cli_print(cli, " Tx Speed: %lu", session[s].tx_connect_speed);
415 if (session[s].snoop_ip && session[s].snoop_port)
416 cli_print(cli, " Intercepted: %s:%d", inet_toa(session[s].snoop_ip), session[s] .snoop_port);
417 else
418 cli_print(cli, " Intercepted: no");
419 cli_print(cli, " Throttled: %s", session[s].throttle ? "YES" : "no");
420 cli_print(cli, " Walled Garden: %s", session[s].walled_garden ? "YES" : "no");
421 cli_print(cli, " Filter BucketI: %d", session[s].tbf_in);
422 cli_print(cli, " Filter BucketO: %d", session[s].tbf_out);
423 }
424 return CLI_OK;
425 }
426
427 // Show Summary
428 cli_print(cli, " %s %4s %-32s %-15s %s %s %s %10s %10s %10s %4s %-15s %s",
429 "SID",
430 "TID",
431 "Username",
432 "IP",
433 "I",
434 "T",
435 "G",
436 "opened",
437 "downloaded",
438 "uploaded",
439 "idle",
440 "LAC",
441 "CLI");
442 for (i = 1; i < MAXSESSION; i++)
443 {
444 char *userip, *tunnelip;
445 if (!session[i].opened) continue;
446 userip = strdup(inet_toa(htonl(session[i].ip)));
447 tunnelip = strdup(inet_toa(htonl(tunnel[ session[i].tunnel ].ip)));
448 cli_print(cli, "%5d %4d %-32s %-15s %s %s %s %10u %10lu %10lu %4u %-15s %s",
449 i,
450 session[i].tunnel,
451 session[i].user[0] ? session[i].user : "*",
452 userip,
453 (session[i].snoop_ip && session[i].snoop_port) ? "Y" : "N",
454 (session[i].throttle) ? "Y" : "N",
455 (session[i].walled_garden) ? "Y" : "N",
456 abs(time_now - (unsigned long)session[i].opened),
457 (unsigned long)session[i].total_cout,
458 (unsigned long)session[i].total_cin,
459 abs(time_now - (session[i].last_packet ? session[i].last_packet : time_now)),
460 tunnelip,
461 session[i].calling[0] ? session[i].calling : "*");
462 if (userip) free(userip);
463 if (tunnelip) free(tunnelip);
464 }
465 return CLI_OK;
466 }
467
468 int cmd_show_tunnels(struct cli_def *cli, char *command, char **argv, int argc)
469 {
470 int i, x, show_all = 0;
471 char *states[] = {
472 "Free",
473 "Open",
474 "Closing",
475 "Opening",
476 };
477
478 if (CLI_HELP_REQUESTED)
479 {
480 if (argc > 1)
481 return cli_arg_help(cli, 1,
482 "<1-%d>", MAXTUNNEL-1, "Show specific tunnel by id",
483 NULL);
484
485 return cli_arg_help(cli, 1,
486 "all", "Show all tunnels, including unused",
487 "<1-%d>", MAXTUNNEL-1, "Show specific tunnel by id",
488 NULL);
489 }
490
491 time(&time_now);
492 if (argc > 0)
493 {
494 if (strcmp(argv[0], "all") == 0)
495 {
496 show_all = 1;
497 }
498 else
499 {
500 // Show individual tunnel
501 for (i = 0; i < argc; i++)
502 {
503 char s[65535] = {0};
504 unsigned int t;
505 t = atoi(argv[i]);
506 if (t <= 0 || t >= MAXTUNNEL)
507 {
508 cli_print(cli, "Invalid tunnel id \"%s\"", argv[i]);
509 continue;
510 }
511 cli_print(cli, "\r\nTunnel %d:", t);
512 cli_print(cli, " State: %s", states[tunnel[t].state]);
513 cli_print(cli, " Hostname: %s", tunnel[t].hostname[0] ? tunnel[t].hostname : "(none)");
514 cli_print(cli, " Remote IP: %s", inet_toa(htonl(tunnel[t].ip)));
515 cli_print(cli, " Remote Port: %d", tunnel[t].port);
516 cli_print(cli, " Rx Window: %u", tunnel[t].window);
517 cli_print(cli, " Next Recv: %u", tunnel[t].nr);
518 cli_print(cli, " Next Send: %u", tunnel[t].ns);
519 cli_print(cli, " Queue Len: %u", tunnel[t].controlc);
520 cli_print(cli, " Last Packet Age:%u", (unsigned)(time_now - tunnel[t].last));
521
522 for (x = 0; x < MAXSESSION; x++)
523 if (session[x].tunnel == t && session[x].opened && !session[x].die)
524 sprintf(s, "%s%u ", s, x);
525 cli_print(cli, " Sessions: %s", s);
526 }
527 return CLI_OK;
528 }
529 }
530
531 // Show tunnel summary
532 cli_print(cli, "%4s %20s %20s %6s %s",
533 "TID",
534 "Hostname",
535 "IP",
536 "State",
537 "Sessions");
538 for (i = 1; i < MAXTUNNEL; i++)
539 {
540 int sessions = 0;
541 if (!show_all && (!tunnel[i].ip || tunnel[i].die || !tunnel[i].hostname[0])) continue;
542
543 for (x = 0; x < MAXSESSION; x++) if (session[x].tunnel == i && session[x].opened && !session[x].die) sessions++;
544 cli_print(cli, "%4d %20s %20s %6s %6d",
545 i,
546 *tunnel[i].hostname ? tunnel[i].hostname : "(null)",
547 inet_toa(htonl(tunnel[i].ip)),
548 states[tunnel[i].state],
549 sessions);
550 }
551 return CLI_OK;
552 }
553
554 int cmd_show_users(struct cli_def *cli, char *command, char **argv, int argc)
555 {
556 char sid[32][8];
557 char *sargv[32];
558 int sargc = 0;
559 int i;
560
561 if (CLI_HELP_REQUESTED)
562 return cli_arg_help(cli, 1,
563 "USER", "Show details for specific username",
564 NULL);
565
566 for (i = 0; i < MAXSESSION; i++)
567 {
568 if (!session[i].opened) continue;
569 if (!session[i].user[0]) continue;
570 if (argc > 0)
571 {
572 int j;
573 for (j = 0; j < argc && sargc < 32; j++)
574 {
575 if (strcmp(argv[j], session[i].user) == 0)
576 {
577 snprintf(sid[sargc], sizeof(sid[0]), "%d", i);
578 sargv[sargc] = sid[sargc];
579 sargc++;
580 }
581 }
582
583 continue;
584 }
585
586 cli_print(cli, "%s", session[i].user);
587 }
588
589 if (sargc > 0)
590 return cmd_show_session(cli, "users", sargv, sargc);
591
592 return CLI_OK;
593 }
594
595 int cmd_show_counters(struct cli_def *cli, char *command, char **argv, int argc)
596 {
597 if (CLI_HELP_REQUESTED)
598 return CLI_HELP_NO_ARGS;
599
600 cli_print(cli, "%-10s %-8s %-10s %-8s", "Ethernet", "Bytes", "Packets", "Errors");
601 cli_print(cli, "%-10s %8lu %8lu %8lu", "RX",
602 GET_STAT(tun_rx_bytes),
603 GET_STAT(tun_rx_packets),
604 GET_STAT(tun_rx_errors));
605 cli_print(cli, "%-10s %8lu %8lu %8lu", "TX",
606 GET_STAT(tun_tx_bytes),
607 GET_STAT(tun_tx_packets),
608 GET_STAT(tun_tx_errors));
609 cli_print(cli, "");
610
611 cli_print(cli, "%-10s %-8s %-10s %-8s %-8s", "Tunnel", "Bytes", "Packets", "Errors", "Retries");
612 cli_print(cli, "%-10s %8lu %8lu %8lu %8lu", "RX",
613 GET_STAT(tunnel_rx_bytes),
614 GET_STAT(tunnel_rx_packets),
615 GET_STAT(tunnel_rx_errors),
616 0L);
617 cli_print(cli, "%-10s %8lu %8lu %8lu %8lu", "TX",
618 GET_STAT(tunnel_tx_bytes),
619 GET_STAT(tunnel_tx_packets),
620 GET_STAT(tunnel_tx_errors),
621 GET_STAT(tunnel_retries));
622 cli_print(cli, "");
623
624 cli_print(cli, "%-30s%-10s", "Counter", "Value");
625 cli_print(cli, "-----------------------------------------");
626 cli_print(cli, "%-30s%lu", "radius_retries", GET_STAT(radius_retries));
627 cli_print(cli, "%-30s%lu", "arp_sent", GET_STAT(arp_sent));
628 cli_print(cli, "%-30s%lu", "packets_snooped", GET_STAT(packets_snooped));
629 cli_print(cli, "%-30s%lu", "tunnel_created", GET_STAT(tunnel_created));
630 cli_print(cli, "%-30s%lu", "session_created", GET_STAT(session_created));
631 cli_print(cli, "%-30s%lu", "tunnel_timeout", GET_STAT(tunnel_timeout));
632 cli_print(cli, "%-30s%lu", "session_timeout", GET_STAT(session_timeout));
633 cli_print(cli, "%-30s%lu", "radius_timeout", GET_STAT(radius_timeout));
634 cli_print(cli, "%-30s%lu", "radius_overflow", GET_STAT(radius_overflow));
635 cli_print(cli, "%-30s%lu", "tunnel_overflow", GET_STAT(tunnel_overflow));
636 cli_print(cli, "%-30s%lu", "session_overflow", GET_STAT(session_overflow));
637 cli_print(cli, "%-30s%lu", "ip_allocated", GET_STAT(ip_allocated));
638 cli_print(cli, "%-30s%lu", "ip_freed", GET_STAT(ip_freed));
639 cli_print(cli, "%-30s%lu", "cluster_forwarded", GET_STAT(c_forwarded));
640 cli_print(cli, "%-30s%lu", "recv_forward", GET_STAT(recv_forward));
641
642
643 #ifdef STATISTICS
644 cli_print(cli, "\n%-30s%-10s", "Counter", "Value");
645 cli_print(cli, "-----------------------------------------");
646 cli_print(cli, "%-30s%lu", "call_processtun", GET_STAT(call_processtun));
647 cli_print(cli, "%-30s%lu", "call_processipout", GET_STAT(call_processipout));
648 cli_print(cli, "%-30s%lu", "call_processudp", GET_STAT(call_processudp));
649 cli_print(cli, "%-30s%lu", "call_processpap", GET_STAT(call_processpap));
650 cli_print(cli, "%-30s%lu", "call_processchap", GET_STAT(call_processchap));
651 cli_print(cli, "%-30s%lu", "call_processlcp", GET_STAT(call_processlcp));
652 cli_print(cli, "%-30s%lu", "call_processipcp", GET_STAT(call_processipcp));
653 cli_print(cli, "%-30s%lu", "call_processipin", GET_STAT(call_processipin));
654 cli_print(cli, "%-30s%lu", "call_processccp", GET_STAT(call_processccp));
655 cli_print(cli, "%-30s%lu", "call_processrad", GET_STAT(call_processrad));
656 cli_print(cli, "%-30s%lu", "call_sendarp", GET_STAT(call_sendarp));
657 cli_print(cli, "%-30s%lu", "call_sendipcp", GET_STAT(call_sendipcp));
658 cli_print(cli, "%-30s%lu", "call_sendchap", GET_STAT(call_sendchap));
659 cli_print(cli, "%-30s%lu", "call_sessionbyip", GET_STAT(call_sessionbyip));
660 cli_print(cli, "%-30s%lu", "call_sessionbyuser", GET_STAT(call_sessionbyuser));
661 cli_print(cli, "%-30s%lu", "call_tunnelsend", GET_STAT(call_tunnelsend));
662 cli_print(cli, "%-30s%lu", "call_tunnelkill", GET_STAT(call_tunnelkill));
663 cli_print(cli, "%-30s%lu", "call_tunnelshutdown", GET_STAT(call_tunnelshutdown));
664 cli_print(cli, "%-30s%lu", "call_sessionkill", GET_STAT(call_sessionkill));
665 cli_print(cli, "%-30s%lu", "call_sessionshutdown", GET_STAT(call_sessionshutdown));
666 cli_print(cli, "%-30s%lu", "call_sessionsetup", GET_STAT(call_sessionsetup));
667 cli_print(cli, "%-30s%lu", "call_assign_ip_address",GET_STAT(call_assign_ip_address));
668 cli_print(cli, "%-30s%lu", "call_free_ip_address", GET_STAT(call_free_ip_address));
669 cli_print(cli, "%-30s%lu", "call_dump_acct_info", GET_STAT(call_dump_acct_info));
670 cli_print(cli, "%-30s%lu", "call_radiussend", GET_STAT(call_radiussend));
671 cli_print(cli, "%-30s%lu", "call_radiusretry", GET_STAT(call_radiusretry));
672 #endif
673 return CLI_OK;
674 }
675
676 int cmd_show_version(struct cli_def *cli, char *command, char **argv, int argc)
677 {
678 int tag = 0;
679 int file = 0;
680 int i = 0;
681
682 if (CLI_HELP_REQUESTED)
683 return cli_arg_help(cli, 1,
684 "tag", "Include CVS release tag",
685 "file", "Include file versions",
686 NULL);
687
688 for (i = 0; i < argc; i++)
689 if (!strcmp(argv[i], "tag"))
690 tag++;
691 else if (!strcmp(argv[i], "file"))
692 file++;
693
694 cli_print(cli, "L2TPNS %s", VERSION);
695 if (tag)
696 {
697 char const *p = strchr(cvs_name, ':');
698 char const *e;
699 if (p)
700 {
701 p++;
702 while (isspace(*p))
703 p++;
704 }
705
706 if (!p || *p == '$')
707 p = "HEAD";
708
709 e = strpbrk(p, " \t$");
710 cli_print(cli, "Tag: %.*s", e ? e - p + 1 : strlen(p), p);
711 }
712
713 if (file)
714 {
715 extern linked_list *loaded_plugins;
716 void *p;
717
718 cli_print(cli, "Files:");
719 cli_print(cli, " %s", cvs_id_arp);
720 #ifdef BGP
721 cli_print(cli, " %s", cvs_id_bgp);
722 #endif /* BGP */
723 cli_print(cli, " %s", cvs_id_cli);
724 cli_print(cli, " %s", cvs_id_cluster);
725 cli_print(cli, " %s", cvs_id_constants);
726 cli_print(cli, " %s", cvs_id_control);
727 cli_print(cli, " %s", cvs_id_icmp);
728 cli_print(cli, " %s", cvs_id_l2tpns);
729 cli_print(cli, " %s", cvs_id_ll);
730 cli_print(cli, " %s", cvs_id_md5);
731 cli_print(cli, " %s", cvs_id_ppp);
732 cli_print(cli, " %s", cvs_id_radius);
733 cli_print(cli, " %s", cvs_id_tbf);
734 cli_print(cli, " %s", cvs_id_util);
735
736 ll_reset(loaded_plugins);
737 while ((p = ll_next(loaded_plugins)))
738 {
739 char const **id = dlsym(p, "cvs_id");
740 if (id)
741 cli_print(cli, " %s", *id);
742 }
743 }
744
745 return CLI_OK;
746 }
747
748 int cmd_show_pool(struct cli_def *cli, char *command, char **argv, int argc)
749 {
750 int i;
751 int used = 0, free = 0, show_all = 0;
752
753 if (!config->cluster_iam_master)
754 {
755 cli_print(cli, "Can't do this on a slave. Do it on %s", inet_toa(config->cluster_master_address));
756 return CLI_OK;
757 }
758
759 if (CLI_HELP_REQUESTED)
760 {
761 if (argc > 1)
762 return cli_arg_help(cli, 1, NULL);
763
764 return cli_arg_help(cli, 1,
765 "all", "Show all pool addresses, including unused",
766 NULL);
767 }
768
769 if (argc > 0 && strcmp(argv[0], "all") == 0)
770 show_all = 1;
771
772 time(&time_now);
773 cli_print(cli, "%-15s %4s %8s %s", "IP Address", "Used", "Session", "User");
774 for (i = 0; i < MAXIPPOOL; i++)
775 {
776 if (!ip_address_pool[i].address) continue;
777 if (ip_address_pool[i].assigned)
778 {
779 cli_print(cli, "%-15s Y %8d %s",
780 inet_toa(htonl(ip_address_pool[i].address)), ip_address_pool[i].session, session[ip_address_pool[i].session].user);
781
782 used++;
783 }
784 else
785 {
786 if (ip_address_pool[i].last)
787 cli_print(cli, "%-15s N %8s [%s] %ds",
788 inet_toa(htonl(ip_address_pool[i].address)), "",
789 ip_address_pool[i].user, time_now - ip_address_pool[i].last);
790 else if (show_all)
791 cli_print(cli, "%-15s N", inet_toa(htonl(ip_address_pool[i].address)));
792
793 free++;
794 }
795 }
796
797 if (!show_all)
798 cli_print(cli, "(Not displaying unused addresses)");
799
800 cli_print(cli, "\r\nFree: %d\r\nUsed: %d", free, used);
801 return CLI_OK;
802 }
803
804 void print_save_config(struct cli_def *cli, char *string)
805 {
806 if (save_config_fh)
807 fprintf(save_config_fh, "%s\n", string);
808 }
809
810 int cmd_write_memory(struct cli_def *cli, char *command, char **argv, int argc)
811 {
812 if (CLI_HELP_REQUESTED)
813 return CLI_HELP_NO_ARGS;
814
815 if ((save_config_fh = fopen(config->config_file, "w")))
816 {
817 cli_print(cli, "Writing configuration");
818 cli_print_callback(cli, print_save_config);
819 cmd_show_run(cli, command, argv, argc);
820 cli_print_callback(cli, NULL);
821 fclose(save_config_fh);
822 }
823 else
824 {
825 cli_print(cli, "Error writing configuration: %s", strerror(errno));
826 }
827 return CLI_OK;
828 }
829
830 int cmd_show_run(struct cli_def *cli, char *command, char **argv, int argc)
831 {
832 int i;
833
834 if (CLI_HELP_REQUESTED)
835 return CLI_HELP_NO_ARGS;
836
837 cli_print(cli, "# Current configuration:");
838
839 for (i = 0; config_values[i].key; i++)
840 {
841 void *value = ((void *)config) + config_values[i].offset;
842 if (config_values[i].type == STRING)
843 cli_print(cli, "set %s \"%.*s\"", config_values[i].key, config_values[i].size, (char *)value);
844 else if (config_values[i].type == IP)
845 cli_print(cli, "set %s %s", config_values[i].key, inet_toa(*(unsigned *)value));
846 else if (config_values[i].type == SHORT)
847 cli_print(cli, "set %s %hu", config_values[i].key, *(short *)value);
848 else if (config_values[i].type == BOOL)
849 cli_print(cli, "set %s %s", config_values[i].key, (*(int *)value) ? "yes" : "no");
850 else if (config_values[i].type == INT)
851 cli_print(cli, "set %s %d", config_values[i].key, *(int *)value);
852 else if (config_values[i].type == UNSIGNED_LONG)
853 cli_print(cli, "set %s %lu", config_values[i].key, *(unsigned long *)value);
854 }
855
856 cli_print(cli, "# Plugins");
857 for (i = 0; i < MAXPLUGINS; i++)
858 {
859 if (*config->plugins[i])
860 {
861 cli_print(cli, "load plugin \"%s\"", config->plugins[i]);
862 }
863 }
864
865 cli_print(cli, "# end");
866 return CLI_OK;
867 }
868
869 int cmd_show_radius(struct cli_def *cli, char *command, char **argv, int argc)
870 {
871 int i, free = 0, used = 0, show_all = 0;
872 char *states[] = {
873 "NULL",
874 "CHAP",
875 "AUTH",
876 "IPCP",
877 "START",
878 "STOP",
879 "WAIT",
880 };
881
882 if (CLI_HELP_REQUESTED)
883 {
884 if (argc > 1)
885 return cli_arg_help(cli, 1, NULL);
886
887 return cli_arg_help(cli, 1,
888 "all", "Show all RADIUS sessions, including unused",
889 NULL);
890 }
891
892 cli_print(cli, "%6s%5s%6s%9s%9s%4s", "Radius", "Sock", "State", "Session", "Retry", "Try");
893
894 time(&time_now);
895
896 if (argc > 0 && strcmp(argv[0], "all") == 0)
897 show_all = 1;
898
899 for (i = 1; i < MAXRADIUS; i++)
900 {
901 if (radius[i].state == RADIUSNULL)
902 free++;
903 else
904 used++;
905
906 if (!show_all && radius[i].state == RADIUSNULL) continue;
907
908 cli_print(cli, "%6d%5d%6s%9d%9u%4d",
909 i >> RADIUS_SHIFT,
910 i & RADIUS_MASK,
911 states[radius[i].state],
912 radius[i].session,
913 radius[i].retry,
914 radius[i].try);
915 }
916
917 cli_print(cli, "\r\nFree: %d\r\nUsed: %d", free, used);
918
919 return CLI_OK;
920 }
921
922 int cmd_show_plugins(struct cli_def *cli, char *command, char **argv, int argc)
923 {
924 int i;
925
926 if (CLI_HELP_REQUESTED)
927 return CLI_HELP_NO_ARGS;
928
929 cli_print(cli, "Plugins currently loaded:");
930 for (i = 0; i < MAXPLUGINS; i++)
931 {
932 if (*config->plugins[i])
933 {
934 cli_print(cli, " %s", config->plugins[i]);
935 }
936 }
937 return CLI_OK;
938 }
939
940 int cmd_show_throttle(struct cli_def *cli, char *command, char **argv, int argc)
941 {
942 int i;
943
944 if (CLI_HELP_REQUESTED)
945 return CLI_HELP_NO_ARGS;
946
947 cli_print(cli, "Token bucket filters:");
948 cli_print(cli, "%-6s %8s %-4s", "ID", "Handle", "Used");
949 for (i = 0; i < MAXSESSION; i++)
950 {
951 if (!session[i].throttle)
952 continue;
953
954 cli_print(cli, "%-6d %8d %8d",
955 i,
956 session[i].tbf_in,
957 session[i].tbf_out);
958 }
959 return CLI_OK;
960 }
961
962 int cmd_show_banana(struct cli_def *cli, char *command, char **argv, int argc)
963 {
964 if (CLI_HELP_REQUESTED)
965 return CLI_HELP_NO_ARGS;
966
967 cli_print(cli, " _\n"
968 "//\\\n"
969 "V \\\n"
970 " \\ \\_\n"
971 " \\,'.`-.\n"
972 " |\\ `. `.\n"
973 " ( \\ `. `-. _,.-:\\\n"
974 " \\ \\ `. `-._ __..--' ,-';/\n"
975 " \\ `. `-. `-..___..---' _.--' ,'/\n"
976 " `. `. `-._ __..--' ,' /\n"
977 " `. `-_ ``--..'' _.-' ,'\n"
978 " `-_ `-.___ __,--' ,'\n"
979 " `-.__ `----\"\"\" __.-'\n"
980 "hh `--..____..--'");
981
982 return CLI_OK;
983 }
984
985 int cmd_clear_counters(struct cli_def *cli, char *command, char **argv, int argc)
986 {
987 if (CLI_HELP_REQUESTED)
988 return CLI_HELP_NO_ARGS;
989
990 cli_print(cli, "Counters cleared");
991 SET_STAT(last_reset, time(NULL));
992 return CLI_OK;
993 }
994
995 int cmd_drop_user(struct cli_def *cli, char *command, char **argv, int argc)
996 {
997 int i;
998 sessionidt s;
999
1000 if (CLI_HELP_REQUESTED)
1001 return cli_arg_help(cli, argc > 1,
1002 "USER", "Username of session to drop", NULL);
1003
1004 if (!config->cluster_iam_master)
1005 {
1006 cli_print(cli, "Can't do this on a slave. Do it on %s", inet_toa(config->cluster_master_address));
1007 return CLI_OK;
1008 }
1009
1010 if (!argc)
1011 {
1012 cli_print(cli, "Specify a user to drop");
1013 return CLI_OK;
1014 }
1015
1016 for (i = 0; i < argc; i++)
1017 {
1018 if (!(s = sessionbyuser(argv[i])))
1019 {
1020 cli_print(cli, "User %s is not connected", argv[i]);
1021 continue;
1022 }
1023
1024 if (session[s].ip && session[s].opened && !session[s].die)
1025 {
1026 cli_print(cli, "Dropping user %s", session[s].user);
1027 cli_session_actions[s].action |= CLI_SESS_KILL;
1028 }
1029 }
1030
1031 return CLI_OK;
1032 }
1033
1034 int cmd_drop_tunnel(struct cli_def *cli, char *command, char **argv, int argc)
1035 {
1036 int i;
1037 tunnelidt t;
1038
1039 if (CLI_HELP_REQUESTED)
1040 return cli_arg_help(cli, argc > 1,
1041 "<1-%d>", MAXTUNNEL-1, "Tunnel id to drop", NULL);
1042
1043 if (!config->cluster_iam_master)
1044 {
1045 cli_print(cli, "Can't do this on a slave. Do it on %s", inet_toa(config->cluster_master_address));
1046 return CLI_OK;
1047 }
1048
1049 if (!argc)
1050 {
1051 cli_print(cli, "Specify a tunnel to drop");
1052 return CLI_OK;
1053 }
1054
1055 for (i = 0; i < argc; i++)
1056 {
1057 if ((t = atol(argv[i])) <= 0 || (t >= MAXTUNNEL))
1058 {
1059 cli_print(cli, "Invalid tunnel ID (1-%d)", MAXTUNNEL-1);
1060 continue;
1061 }
1062
1063 if (!tunnel[t].ip)
1064 {
1065 cli_print(cli, "Tunnel %d is not connected", t);
1066 continue;
1067 }
1068
1069 if (tunnel[t].die)
1070 {
1071 cli_print(cli, "Tunnel %d is already being shut down", t);
1072 continue;
1073 }
1074
1075 cli_print(cli, "Tunnel %d shut down (%s)", t, tunnel[t].hostname);
1076 cli_tunnel_actions[t].action |= CLI_TUN_KILL;
1077 }
1078
1079 return CLI_OK;
1080 }
1081
1082 int cmd_drop_session(struct cli_def *cli, char *command, char **argv, int argc)
1083 {
1084 int i;
1085 sessionidt s;
1086
1087 if (CLI_HELP_REQUESTED)
1088 return cli_arg_help(cli, argc > 1,
1089 "<1-%d>", MAXSESSION-1, "Session id to drop", NULL);
1090
1091 if (!config->cluster_iam_master)
1092 {
1093 cli_print(cli, "Can't do this on a slave. Do it on %s", inet_toa(config->cluster_master_address));
1094 return CLI_OK;
1095 }
1096
1097 if (!argc)
1098 {
1099 cli_print(cli, "Specify a session id to drop");
1100 return CLI_OK;
1101 }
1102
1103 for (i = 0; i < argc; i++)
1104 {
1105 if ((s = atol(argv[i])) <= 0 || (s > MAXSESSION))
1106 {
1107 cli_print(cli, "Invalid session ID (1-%d)", MAXSESSION-1);
1108 continue;
1109 }
1110
1111 if (session[s].ip && session[s].opened && !session[s].die)
1112 {
1113 cli_print(cli, "Dropping session %d", s);
1114 cli_session_actions[s].action |= CLI_SESS_KILL;
1115 }
1116 else
1117 {
1118 cli_print(cli, "Session %d is not active.", s);
1119 }
1120 }
1121
1122 return CLI_OK;
1123 }
1124
1125 int cmd_snoop(struct cli_def *cli, char *command, char **argv, int argc)
1126 {
1127 ipt ip;
1128 u16 port;
1129 sessionidt s;
1130
1131 if (CLI_HELP_REQUESTED)
1132 {
1133 switch (argc)
1134 {
1135 case 1:
1136 return cli_arg_help(cli, 0,
1137 "USER", "Username of session to snoop", NULL);
1138
1139 case 2:
1140 return cli_arg_help(cli, 0,
1141 "A.B.C.D", "IP address of snoop destination", NULL);
1142
1143 case 3:
1144 return cli_arg_help(cli, 0,
1145 "N", "Port of snoop destination", NULL);
1146
1147 case 4:
1148 if (!argv[3][1])
1149 return cli_arg_help(cli, 1, NULL);
1150
1151 default:
1152 return CLI_OK;
1153 }
1154 }
1155
1156 if (!config->cluster_iam_master)
1157 {
1158 cli_print(cli, "Can't do this on a slave. Do it on %s", inet_toa(config->cluster_master_address));
1159 return CLI_OK;
1160 }
1161
1162 if (argc < 3)
1163 {
1164 cli_print(cli, "Specify username ip port");
1165 return CLI_OK;
1166 }
1167
1168 if (!(s = sessionbyuser(argv[0])))
1169 {
1170 cli_print(cli, "User %s is not connected", argv[0]);
1171 return CLI_OK;
1172 }
1173
1174 ip = inet_addr(argv[1]);
1175 if (!ip || ip == INADDR_NONE)
1176 {
1177 cli_print(cli, "Cannot parse IP \"%s\"", argv[1]);
1178 return CLI_OK;
1179 }
1180
1181 port = atoi(argv[2]);
1182 if (!port)
1183 {
1184 cli_print(cli, "Invalid port %s", argv[2]);
1185 return CLI_OK;
1186 }
1187
1188 cli_print(cli, "Snooping user %s to %s:%d", argv[0], inet_toa(session[s].snoop_ip), session[s].snoop_port);
1189 cli_session_actions[s].snoop_ip = ip;
1190 cli_session_actions[s].snoop_port = port;
1191 cli_session_actions[s].action |= CLI_SESS_SNOOP;
1192
1193 return CLI_OK;
1194 }
1195
1196 int cmd_no_snoop(struct cli_def *cli, char *command, char **argv, int argc)
1197 {
1198 int i;
1199 sessionidt s;
1200
1201 if (CLI_HELP_REQUESTED)
1202 return cli_arg_help(cli, argc > 1,
1203 "USER", "Username of session to un-snoop", NULL);
1204
1205 if (!config->cluster_iam_master)
1206 {
1207 cli_print(cli, "Can't do this on a slave. Do it on %s", inet_toa(config->cluster_master_address));
1208 return CLI_OK;
1209 }
1210
1211 if (!argc)
1212 {
1213 cli_print(cli, "Specify a user");
1214 return CLI_OK;
1215 }
1216
1217 for (i = 0; i < argc; i++)
1218 {
1219 if (!(s = sessionbyuser(argv[i])))
1220 {
1221 cli_print(cli, "User %s is not connected", argv[i]);
1222 continue;
1223 }
1224
1225 cli_print(cli, "Not snooping user %s", argv[i]);
1226 cli_session_actions[s].action |= CLI_SESS_NOSNOOP;
1227 }
1228 return CLI_OK;
1229 }
1230
1231 int cmd_throttle(struct cli_def *cli, char *command, char **argv, int argc)
1232 {
1233 int i;
1234 sessionidt s;
1235
1236 if (CLI_HELP_REQUESTED)
1237 return cli_arg_help(cli, argc > 1,
1238 "USER", "Username of session to throttle", NULL);
1239
1240 if (!config->cluster_iam_master)
1241 {
1242 cli_print(cli, "Can't do this on a slave. Do it on %s", inet_toa(config->cluster_master_address));
1243 return CLI_OK;
1244 }
1245
1246 if (!argc)
1247 {
1248 cli_print(cli, "Specify a user");
1249 return CLI_OK;
1250 }
1251
1252 for (i = 0; i < argc; i++)
1253 {
1254 if (!(s = sessionbyuser(argv[i])))
1255 {
1256 cli_print(cli, "User %s is not connected", argv[i]);
1257 continue;
1258 }
1259
1260 if (session[s].throttle)
1261 {
1262 cli_print(cli, "User %s already throttled", argv[i]);
1263 continue;
1264 }
1265
1266 cli_print(cli, "Throttling user %s", argv[i]);
1267 cli_session_actions[s].throttle = config->rl_rate; // could be configurable at some stage
1268 cli_session_actions[s].action |= CLI_SESS_THROTTLE;
1269 }
1270
1271 return CLI_OK;
1272 }
1273
1274 int cmd_no_throttle(struct cli_def *cli, char *command, char **argv, int argc)
1275 {
1276 int i;
1277 sessionidt s;
1278
1279 if (CLI_HELP_REQUESTED)
1280 return cli_arg_help(cli, argc > 1,
1281 "USER", "Username of session to un-throttle", NULL);
1282
1283 if (!config->cluster_iam_master)
1284 {
1285 cli_print(cli, "Can't do this on a slave. Do it on %s", inet_toa(config->cluster_master_address));
1286 return CLI_OK;
1287 }
1288
1289 if (!argc)
1290 {
1291 cli_print(cli, "Specify a user");
1292 return CLI_OK;
1293 }
1294
1295 for (i = 0; i < argc; i++)
1296 {
1297 if (!(s = sessionbyuser(argv[i])))
1298 {
1299 cli_print(cli, "User %s is not connected", argv[i]);
1300 continue;
1301 }
1302
1303 if (!session[s].throttle)
1304 {
1305 cli_print(cli, "User %s not throttled", argv[i]);
1306 continue;
1307 }
1308
1309 cli_print(cli, "Unthrottling user %s", argv[i]);
1310 cli_session_actions[s].action |= CLI_SESS_NOTHROTTLE;
1311 }
1312
1313 return CLI_OK;
1314 }
1315
1316 int cmd_debug(struct cli_def *cli, char *command, char **argv, int argc)
1317 {
1318 int i;
1319
1320 if (CLI_HELP_REQUESTED)
1321 return cli_arg_help(cli, 1,
1322 "all", "Enable debugging for all except \"data\"",
1323 "critical", "", // FIXME: add descriptions
1324 "error", "",
1325 "warning", "",
1326 "info", "",
1327 "calls", "",
1328 "data", "",
1329 NULL);
1330
1331 if (!argc)
1332 {
1333 char *p = (char *) &debug_flags;
1334 for (i = 0; i < sizeof(debug_flags); i++)
1335 {
1336 if (p[i])
1337 {
1338 cli_print(cli, "Currently debugging:%s%s%s%s%s%s",
1339 (debug_flags.critical) ? " critical" : "",
1340 (debug_flags.error) ? " error" : "",
1341 (debug_flags.warning) ? " warning" : "",
1342 (debug_flags.info) ? " info" : "",
1343 (debug_flags.calls) ? " calls" : "",
1344 (debug_flags.data) ? " data" : "");
1345
1346 return CLI_OK;
1347 }
1348 }
1349
1350 cli_print(cli, "Debugging off");
1351 return CLI_OK;
1352 }
1353
1354 for (i = 0; i < argc; i++)
1355 {
1356 int len = strlen(argv[i]);
1357
1358 if (argv[i][0] == 'c' && len < 2)
1359 len = 2; /* distinguish [cr]itical from [ca]lls */
1360
1361 if (!strncasecmp(argv[i], "critical", len)) { debug_flags.critical = 1; continue; }
1362 if (!strncasecmp(argv[i], "error", len)) { debug_flags.error = 1; continue; }
1363 if (!strncasecmp(argv[i], "warning", len)) { debug_flags.warning = 1; continue; }
1364 if (!strncasecmp(argv[i], "info", len)) { debug_flags.info = 1; continue; }
1365 if (!strncasecmp(argv[i], "calls", len)) { debug_flags.calls = 1; continue; }
1366 if (!strncasecmp(argv[i], "data", len)) { debug_flags.data = 1; continue; }
1367 if (!strncasecmp(argv[i], "all", len))
1368 {
1369 memset(&debug_flags, 1, sizeof(debug_flags));
1370 debug_flags.data = 0;
1371 continue;
1372 }
1373
1374 cli_print(cli, "Invalid debugging flag \"%s\"", argv[i]);
1375 }
1376
1377 return CLI_OK;
1378 }
1379
1380 int cmd_no_debug(struct cli_def *cli, char *command, char **argv, int argc)
1381 {
1382 int i;
1383
1384 if (CLI_HELP_REQUESTED)
1385 return cli_arg_help(cli, 1,
1386 "all", "Disable all debugging",
1387 "critical", "", // FIXME: add descriptions
1388 "error", "",
1389 "warning", "",
1390 "info", "",
1391 "calls", "",
1392 "data", "",
1393 NULL);
1394
1395 if (!argc)
1396 {
1397 memset(&debug_flags, 0, sizeof(debug_flags));
1398 return CLI_OK;
1399 }
1400
1401 for (i = 0; i < argc; i++)
1402 {
1403 int len = strlen(argv[i]);
1404
1405 if (argv[i][0] == 'c' && len < 2)
1406 len = 2; /* distinguish [cr]itical from [ca]lls */
1407
1408 if (!strncasecmp(argv[i], "critical", len)) { debug_flags.critical = 0; continue; }
1409 if (!strncasecmp(argv[i], "error", len)) { debug_flags.error = 0; continue; }
1410 if (!strncasecmp(argv[i], "warning", len)) { debug_flags.warning = 0; continue; }
1411 if (!strncasecmp(argv[i], "info", len)) { debug_flags.info = 0; continue; }
1412 if (!strncasecmp(argv[i], "calls", len)) { debug_flags.calls = 0; continue; }
1413 if (!strncasecmp(argv[i], "data", len)) { debug_flags.data = 0; continue; }
1414 if (!strncasecmp(argv[i], "all", len))
1415 {
1416 memset(&debug_flags, 0, sizeof(debug_flags));
1417 continue;
1418 }
1419
1420 cli_print(cli, "Invalid debugging flag \"%s\"", argv[i]);
1421 }
1422
1423 return CLI_OK;
1424 }
1425
1426 int cmd_load_plugin(struct cli_def *cli, char *command, char **argv, int argc)
1427 {
1428 int i, firstfree = 0;
1429
1430 if (CLI_HELP_REQUESTED)
1431 return cli_arg_help(cli, argc > 1,
1432 "PLUGIN", "Name of plugin to load", NULL);
1433
1434 if (argc != 1)
1435 {
1436 cli_print(cli, "Specify a plugin to load");
1437 return CLI_OK;
1438 }
1439
1440 for (i = 0; i < MAXPLUGINS; i++)
1441 {
1442 if (!*config->plugins[i] && !firstfree)
1443 firstfree = i;
1444 if (strcmp(config->plugins[i], argv[0]) == 0)
1445 {
1446 cli_print(cli, "Plugin is already loaded");
1447 return CLI_OK;
1448 }
1449 }
1450
1451 if (firstfree)
1452 {
1453 strncpy(config->plugins[firstfree], argv[0], sizeof(config->plugins[firstfree]) - 1);
1454 config->reload_config = 1;
1455 cli_print(cli, "Loading plugin %s", argv[0]);
1456 }
1457
1458 return CLI_OK;
1459 }
1460
1461 int cmd_remove_plugin(struct cli_def *cli, char *command, char **argv, int argc)
1462 {
1463 int i;
1464
1465 if (CLI_HELP_REQUESTED)
1466 return cli_arg_help(cli, argc > 1,
1467 "PLUGIN", "Name of plugin to unload", NULL);
1468
1469 if (argc != 1)
1470 {
1471 cli_print(cli, "Specify a plugin to remove");
1472 return CLI_OK;
1473 }
1474
1475 for (i = 0; i < MAXPLUGINS; i++)
1476 {
1477 if (strcmp(config->plugins[i], argv[0]) == 0)
1478 {
1479 config->reload_config = 1;
1480 memset(config->plugins[i], 0, sizeof(config->plugins[i]));
1481 return CLI_OK;
1482 }
1483 }
1484
1485 cli_print(cli, "Plugin is not loaded");
1486 return CLI_OK;
1487 }
1488
1489 char *duration(time_t secs)
1490 {
1491 static char *buf = NULL;
1492 int p = 0;
1493
1494 if (!buf) buf = calloc(64, 1);
1495
1496 if (secs >= 86400)
1497 {
1498 int days = secs / 86400;
1499 p = sprintf(buf, "%d day%s, ", days, days > 1 ? "s" : "");
1500 secs %= 86400;
1501 }
1502
1503 if (secs >= 3600)
1504 {
1505 int mins = secs / 60;
1506 int hrs = mins / 60;
1507
1508 mins %= 60;
1509 sprintf(buf + p, "%d:%02d", hrs, mins);
1510 }
1511 else if (secs >= 60)
1512 {
1513 int mins = secs / 60;
1514 sprintf(buf + p, "%d min%s", mins, mins > 1 ? "s" : "");
1515 }
1516 else
1517 sprintf(buf, "%ld sec%s", secs, secs > 1 ? "s" : "");
1518
1519 return buf;
1520 }
1521
1522 int cmd_uptime(struct cli_def *cli, char *command, char **argv, int argc)
1523 {
1524 FILE *fh;
1525 char buf[100], *p = buf, *loads[3];
1526 int i, num_sessions = 0;
1527
1528 if (CLI_HELP_REQUESTED)
1529 return CLI_HELP_NO_ARGS;
1530
1531 fh = fopen("/proc/loadavg", "r");
1532 fgets(buf, 100, fh);
1533 fclose(fh);
1534
1535 for (i = 0; i < 3; i++)
1536 loads[i] = strdup(strsep(&p, " "));
1537
1538 time(&time_now);
1539 strftime(buf, 99, "%H:%M:%S", localtime(&time_now));
1540
1541 for (i = 1; i < MAXSESSION; i++)
1542 if (session[i].opened) num_sessions++;
1543
1544 cli_print(cli, "%s up %s, %d users, load average: %s, %s, %s",
1545 buf,
1546 duration(time_now - config->start_time),
1547 num_sessions,
1548 loads[0], loads[1], loads[2]
1549 );
1550 for (i = 0; i < 3; i++)
1551 if (loads[i]) free(loads[i]);
1552
1553 cli_print(cli, "Bandwidth: %s", config->bandwidth);
1554
1555 return CLI_OK;
1556 }
1557
1558 int cmd_set(struct cli_def *cli, char *command, char **argv, int argc)
1559 {
1560 int i;
1561
1562 if (CLI_HELP_REQUESTED)
1563 {
1564 switch (argc)
1565 {
1566 case 1:
1567 {
1568 int len = strlen(argv[0])-1;
1569 for (i = 0; config_values[i].key; i++)
1570 if (!len || !strncmp(argv[0], config_values[i].key, len))
1571 cli_print(cli, " %s", config_values[i].key);
1572 }
1573
1574 return CLI_OK;
1575
1576 case 2:
1577 return cli_arg_help(cli, 0,
1578 "VALUE", "Value for variable", NULL);
1579
1580 case 3:
1581 if (!argv[2][1])
1582 return cli_arg_help(cli, 1, NULL);
1583
1584 default:
1585 return CLI_OK;
1586 }
1587 }
1588
1589 if (argc != 2)
1590 {
1591 cli_print(cli, "Specify variable and value");
1592 return CLI_OK;
1593 }
1594
1595 for (i = 0; config_values[i].key; i++)
1596 {
1597 void *value = ((void *)config) + config_values[i].offset;
1598 if (strcmp(config_values[i].key, argv[0]) == 0)
1599 {
1600 // Found a value to set
1601 cli_print(cli, "Setting \"%s\" to \"%s\"", argv[0], argv[1]);
1602 switch (config_values[i].type)
1603 {
1604 case STRING:
1605 strncpy((char *)value, argv[1], config_values[i].size - 1);
1606 break;
1607 case INT:
1608 *(int *)value = atoi(argv[1]);
1609 break;
1610 case UNSIGNED_LONG:
1611 *(unsigned long *)value = atol(argv[1]);
1612 break;
1613 case SHORT:
1614 *(short *)value = atoi(argv[1]);
1615 break;
1616 case IP:
1617 *(unsigned *)value = inet_addr(argv[1]);
1618 break;
1619 case BOOL:
1620 if (strcasecmp(argv[1], "yes") == 0 || strcasecmp(argv[1], "true") == 0 || strcasecmp(argv[1], "1") == 0)
1621 *(int *)value = 1;
1622 else
1623 *(int *)value = 0;
1624 break;
1625 default:
1626 cli_print(cli, "Unknown variable type");
1627 break;
1628 }
1629 config->reload_config = 1;
1630 return CLI_OK;
1631 }
1632 }
1633
1634 cli_print(cli, "Unknown variable \"%s\"", argv[0]);
1635 return CLI_OK;
1636 }
1637
1638 int regular_stuff(struct cli_def *cli)
1639 {
1640 int i = debug_rb_tail;
1641 int reprompt = 0;
1642
1643 #ifdef RINGBUFFER
1644 while (i != ringbuffer->tail)
1645 {
1646 int show_message = 0;
1647
1648 if (*ringbuffer->buffer[i].message)
1649 {
1650 // Always show messages if we are doing general debug
1651 if (ringbuffer->buffer[i].level == 0 && debug_flags.critical) show_message = 1;
1652 if (ringbuffer->buffer[i].level == 1 && debug_flags.error) show_message = 1;
1653 if (ringbuffer->buffer[i].level == 2 && debug_flags.warning) show_message = 1;
1654 if (ringbuffer->buffer[i].level == 3 && debug_flags.info) show_message = 1;
1655 if (ringbuffer->buffer[i].level == 4 && debug_flags.calls) show_message = 1;
1656 if (ringbuffer->buffer[i].level == 5 && debug_flags.data) show_message = 1;
1657 }
1658
1659 if (show_message)
1660 {
1661 ipt address = htonl(ringbuffer->buffer[i].address);
1662 char *ipaddr;
1663 struct in_addr addr;
1664
1665 memcpy(&addr, &address, sizeof(ringbuffer->buffer[i].address));
1666 ipaddr = inet_ntoa(addr);
1667
1668 cli_print(cli, "\r%s-%s-%u-%u %s",
1669 debug_levels[(int)ringbuffer->buffer[i].level],
1670 ipaddr,
1671 ringbuffer->buffer[i].tunnel,
1672 ringbuffer->buffer[i].session,
1673 ringbuffer->buffer[i].message);
1674
1675 reprompt = 1;
1676 }
1677
1678 if (++i == ringbuffer->tail) break;
1679 if (i == RINGBUFFER_SIZE) i = 0;
1680 }
1681
1682 debug_rb_tail = ringbuffer->tail;
1683 if (reprompt)
1684 cli_reprompt(cli);
1685 #endif
1686 return CLI_OK;
1687 }