// L2TPNS Command Line Interface
// $Id: cli.c,v 1.1 2003/12/16 07:07:39 fred_nerk Exp $
// vim: sw=4 ts=8

#include <stdio.h>
#include <sys/file.h>
#include <sys/stat.h>
#include <malloc.h>
#include <string.h>
#include <stdlib.h>
#include <time.h>
#include <arpa/inet.h>
#include <errno.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <signal.h>
#include <unistd.h>
#include <libcli.h>
#include "l2tpns.h"
#include "util.h"
#include "config.h"

#ifdef HAVE_LIBCLI

extern tunnelt *tunnel;
extern sessiont *session;
extern radiust *radius;
extern ippoolt *ip_address_pool;
extern struct Tstats *_statistics;
extern int cli_pid;
struct cli_def *cli = NULL;
int cli_quit = 0;
extern int clifd, udpfd, tapfd, snoopfd, radfd, ifrfd, cluster_sockfd;
extern sessionidt *cli_session_kill;
extern tunnelidt *cli_tunnel_kill;
extern tbft *filter_buckets;
#ifdef RINGBUFFER
extern struct Tringbuffer *ringbuffer;
#endif

char *rcs_id = "$Id: cli.c,v 1.1 2003/12/16 07:07:39 fred_nerk Exp $";

char *debug_levels[] = {
    "CRIT",
    "ERROR",
    "WARN",
    "INFO",
    "CALL",
    "DATA",
};

struct
{
    char critical;
    char error;
    char warning;
    char info;
    char calls;
    char data;
} debug_flags;

int debug_session;
int debug_tunnel;
int debug_rb_tail;

int cmd_show_session(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_show_tunnels(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_show_users(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_show_counters(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_show_version(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_show_pool(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_show_banana(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_clear_counters(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_drop_user(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_drop_tunnel(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_drop_session(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_snoop(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_no_snoop(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_throttle(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_no_throttle(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_debug(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_no_debug(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_watch_session(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int cmd_watch_tunnel(struct cli_def *cli, FILE *w, char *command, char **argv, int argc);
int regular_stuff(struct cli_def *cli, FILE *w);

void init_cli()
{
    FILE *f;
    char buf[4096];
    struct cli_command *c;
    int on = 1;
    struct sockaddr_in addr;

    cli = cli_init();

    c = cli_register_command(cli, NULL, "show", NULL, NULL);
    cli_register_command(cli, c, "session", cmd_show_session, "Show a list of sessions");
    cli_register_command(cli, c, "tunnels", cmd_show_tunnels, NULL);
    cli_register_command(cli, c, "users", cmd_show_users, NULL);
    cli_register_command(cli, c, "version", cmd_show_version, NULL);
    cli_register_command(cli, c, "banana", cmd_show_banana, "Show a banana");
    cli_register_command(cli, c, "pool", cmd_show_pool, NULL);

#ifdef STATISTICS
    cli_register_command(cli, c, "counters", cmd_show_counters, NULL);

    c = cli_register_command(cli, NULL, "clear", NULL, NULL);
    cli_register_command(cli, c, "counters", cmd_clear_counters, NULL);
#endif

    cli_register_command(cli, NULL, "snoop", cmd_snoop, NULL);
    cli_register_command(cli, NULL, "throttle", cmd_throttle, NULL);

    c = cli_register_command(cli, NULL, "no", NULL, NULL);
    cli_register_command(cli, c, "snoop", cmd_no_snoop, NULL);
    cli_register_command(cli, c, "throttle", cmd_no_throttle, NULL);
    cli_register_command(cli, c, "debug", cmd_no_debug, NULL);

    c = cli_register_command(cli, NULL, "drop", NULL, NULL);
    cli_register_command(cli, c, "user", cmd_drop_user, NULL);
    cli_register_command(cli, c, "tunnel", cmd_drop_tunnel, NULL);
    cli_register_command(cli, c, "session", cmd_drop_session, NULL);

    cli_register_command(cli, NULL, "debug", cmd_debug, "Specify a debugging level");

    c = cli_register_command(cli, NULL, "watch", NULL, NULL);
    cli_register_command(cli, c, "session", cmd_watch_session, "Dump logs for a tunnel");
    cli_register_command(cli, c, "tunnel", cmd_watch_tunnel, "Dump logs for a tunnel");

    // Enable regular processing
    cli_regular(cli, regular_stuff);

    if (!(f = fopen(CLIUSERS, "r")))
    {
	log(0, 0, 0, 0, "WARNING! No users specified. Command-line access is open to all\n");
    }
    else
    {
	while (fgets(buf, 4096, f))
	{
	    char *p;
	    if (*buf == '#') continue;
	    if ((p = strchr(buf, '\r'))) *p = 0;
	    if ((p = strchr(buf, '\n'))) *p = 0;
	    if (!*buf) continue;
	    if (!(p = strchr((char *)buf, ':'))) continue;
	    *p++ = 0;
	    cli_allow_user(cli, buf, p);
	    log(3, 0, 0, 0, "Allowing user %s to connect to the CLI\n", buf);
	}
	fclose(f);
    }

    memset(&addr, 0, sizeof(addr));
    clifd = socket(PF_INET, SOCK_STREAM, 6);
    setsockopt(clifd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
    {
	int flags;
	// Set cli fd as non-blocking
	flags = fcntl(clifd, F_GETFL, 0);
	fcntl(clifd, F_SETFL, flags | O_NONBLOCK);
    }
    addr.sin_family = AF_INET;
    addr.sin_port = htons(23);
    if (bind(clifd, (void *) &addr, sizeof(addr)) < 0)
    {
	    log(0, 0, 0, 0, "Error listening on cli port 23: %s\n", strerror(errno));
	    return;
    }
    listen(clifd, 10);
}

void cli_do(int sockfd)
{
    if ((cli_pid = fork())) return;

    // Close sockets
    if (udpfd) close(udpfd); udpfd = 0;
    if (tapfd) close(tapfd); tapfd = 0;
    if (snoopfd) close(snoopfd); snoopfd = 0;
    if (radfd) close(radfd); radfd = 0;
    if (ifrfd) close(ifrfd); ifrfd = 0;
    if (cluster_sockfd) close(cluster_sockfd); cluster_sockfd = 0;
    if (clifd) close(clifd); clifd = 0;

    signal(SIGPIPE, SIG_DFL);
    signal(SIGCHLD, SIG_DFL);
    signal(SIGHUP, SIG_DFL);
    signal(SIGUSR1, SIG_DFL);
    signal(SIGQUIT, SIG_DFL);
    signal(SIGKILL, SIG_DFL);
    signal(SIGALRM, SIG_DFL);

    log(3, 0, 0, 0, "Accepted connection to CLI\n");

    debug_session = 0;
    debug_tunnel = 0;
#ifdef RINGBUFFER
    debug_rb_tail = ringbuffer->tail;
#endif
    memset(&debug_flags, 0, sizeof(debug_flags));
    debug_flags.critical = 1;

    cli_loop(cli, sockfd, "l2tpns> ");

    close(sockfd);
    log(3, 0, 0, 0, "Closed CLI connection\n");
    exit(0);
}

int cmd_show_session(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    int i;
    time_t time_now;

    time(&time_now);
    if (argc > 0)
    {
	// Show individual session
	for (i = 0; i < argc; i++)
	{
	    unsigned int s;
	    s = atoi(argv[i]);
	    if (!s || s > MAXSESSION)
	    {
		fprintf(w, "Invalid session id \"%s\"\r\n", argv[i]);
		continue;
	    }
	    fprintf(w, "\r\nSession %d:\r\n", s);
	    fprintf(w, "	User:		%s\r\n", session[s].user[0] ? session[s].user : "none");
	    fprintf(w, "	Calling Num:	%s\r\n", session[s].calling);
	    fprintf(w, "	Called Num:	%s\r\n", session[s].called);
	    fprintf(w, "	Tunnel ID:	%d\r\n", session[s].tunnel);
	    fprintf(w, "	IP address:	%s\r\n", inet_toa(htonl(session[s].ip)));
	    fprintf(w, "	HSD sid:	%lu\r\n", session[s].sid);
	    fprintf(w, "	Idle time:	%u seconds\r\n", abs(time_now - session[s].last_packet));
	    fprintf(w, "	Next Recv:	%u\r\n", session[s].nr);
	    fprintf(w, "	Next Send:	%u\r\n", session[s].ns);
	    fprintf(w, "	Bytes In/Out:	%lu/%lu\r\n", (unsigned long)session[s].cin, (unsigned long)session[s].cout);
	    fprintf(w, "	Pkts In/Out:	%lu/%lu\r\n", (unsigned long)session[s].pin, (unsigned long)session[s].pout);
	    fprintf(w, "	Radius Session:	%u\r\n", session[s].radius);
	    fprintf(w, "	Rx Speed:	%lu\r\n", session[s].rx_connect_speed);
	    fprintf(w, "	Tx Speed:	%lu\r\n", session[s].tx_connect_speed);
	    fprintf(w, "	Intercepted:	%s\r\n", session[s].snoop ? "YES" : "no");
	    fprintf(w, "	Throttled:	%s\r\n", session[s].throttle ? "YES" : "no");
	    fprintf(w, "	Walled Garden:	%s\r\n", session[s].walled_garden ? "YES" : "no");
	    fprintf(w, "	Filter Bucket:	%s\r\n", session[s].tbf ? filter_buckets[session[s].tbf].handle : "none");
	}
	return CLI_OK;
    }

    // Show Summary
    fprintf(w, "  %s %4s %-32s %-15s %s %s %s %10s %10s %10s %4s %-15s %s\r\n",
	    "SID",
	    "TID",
	    "Username",
	    "IP",
	    "I",
	    "T",
	    "S",
	    "opened",
	    "downloaded",
	    "uploaded",
	    "idle",
	    "LAC",
	    "CLI");
    for (i = 0; i < MAXSESSION; i++)
    {
	char *userip, *tunnelip;
	if (!session[i].opened) continue;
	userip = strdup(inet_toa(htonl(session[i].ip)));
	tunnelip = strdup(inet_toa(htonl(tunnel[ session[i].tunnel ].ip)));
	fprintf(w, "%5d %4d %-32s %-15s %s %s %s %10lu %10lu %10lu %4u %-15s %s\r\n",
		i,
		session[i].tunnel,
		session[i].user[0] ? session[i].user : "*",
		userip,
		(session[i].snoop) ? "Y" : "N",
		(session[i].throttle) ? "Y" : "N",
		(session[i].walled_garden) ? "Y" : "N",
		(unsigned long)session[i].opened,
		(unsigned long)session[i].cout,
		(unsigned long)session[i].cin,
		abs(time_now - (session[i].last_packet ? session[i].last_packet : time_now)),
                tunnelip,
		session[i].calling[0] ? session[i].calling : "*");
	if (userip) free(userip);
	if (tunnelip) free(tunnelip);
    }
    return CLI_OK;
}

int cmd_show_tunnels(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    int i, x;
    time_t time_now;

    time(&time_now);
    if (argc > 0)
    {
	// Show individual tunnel
	for (i = 0; i < argc; i++)
	{
	    char s[65535] = {0};
	    unsigned int t;
	    t = atoi(argv[i]);
	    if (!t || t > MAXTUNNEL)
	    {
		fprintf(w, "Invalid tunnel id \"%s\"\r\n", argv[i]);
		continue;
	    }
	    fprintf(w, "\r\nTunnel %d:\r\n", t);
	    fprintf(w, "	Hostname:	%s\r\n", tunnel[t].hostname[0] ? tunnel[t].hostname : "(none)");
	    fprintf(w, "	Remote IP:	%s\r\n", inet_toa(htonl(tunnel[t].ip)));
	    fprintf(w, "	Remote Port:	%d\r\n", tunnel[t].port);
	    fprintf(w, "	Rx Window:	%u\r\n", tunnel[t].window);
	    fprintf(w, "	Next Recv:	%u\r\n", tunnel[t].nr);
	    fprintf(w, "	Next Send:	%u\r\n", tunnel[t].ns);
	    fprintf(w, "	Queue Len:	%u\r\n", tunnel[t].controlc);
	    fprintf(w, "	Last Packet Age:%u\r\n", (unsigned)(time_now - tunnel[t].last));

	    for (x = 0; x < MAXSESSION; x++)
		    if (session[x].tunnel == t && session[x].opened && !session[x].die)
			    sprintf(s, "%s%u ", s, x);
	    fprintf(w, "	Sessions:	%s\r\n", s);
	}
	return CLI_OK;
    }

    // Show tunnel summary
    fprintf(w, "%s %s %s %s\r\n",
	    "TID",
	    "Hostname",
	    "IP",
	    "Sessions");
    for (i = 0; i < MAXTUNNEL; i++)
    {
	int sessions = 0;
	if (!tunnel[i].ip || tunnel[i].die || !tunnel[i].hostname[0]) continue;

	for (x = 0; x < MAXSESSION; x++) if (session[x].tunnel == i && session[x].opened && !session[x].die) sessions++;
	fprintf(w, "%d %s %s %d\r\n",
		i,
		tunnel[i].hostname,
		inet_toa(htonl(tunnel[i].ip)),
		sessions);
    }
    return CLI_OK;
}

int cmd_show_users(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    int i;
    for (i = 0; i < MAXSESSION; i++)
    {
	if (!session[i].opened) continue;
	if (!session[i].user[0]) continue;
	fprintf(w, "%s\r\n",
		session[i].user);
    }
    return CLI_OK;
}

int cmd_show_counters(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    fprintf(w, "%-10s %-8s %-10s %-8s\r\n", "Ethernet", "Bytes", "Packets", "Errors");
    fprintf(w, "%-10s %8lu %8lu %8lu\r\n", "RX",
	    GET_STAT(tap_rx_bytes),
	    GET_STAT(tap_rx_packets),
	    GET_STAT(tap_rx_errors));
    fprintf(w, "%-10s %8lu %8lu %8lu\r\n", "TX",
	    GET_STAT(tap_tx_bytes),
	    GET_STAT(tap_tx_packets),
	    GET_STAT(tap_tx_errors));
    fprintf(w, "\r\n");

    fprintf(w, "%-10s %-8s %-10s %-8s %-8s\r\n", "Tunnel", "Bytes", "Packets", "Errors", "Retries");
    fprintf(w, "%-10s %8lu %8lu %8lu %8lu\r\n", "RX",
	    GET_STAT(tunnel_rx_bytes),
	    GET_STAT(tunnel_rx_packets),
	    GET_STAT(tunnel_rx_errors),
	    0L);
    fprintf(w, "%-10s %8lu %8lu %8lu %8lu\r\n", "TX",
	    GET_STAT(tunnel_tx_bytes),
	    GET_STAT(tunnel_tx_packets),
	    GET_STAT(tunnel_rx_errors),
	    GET_STAT(tunnel_retries));
    fprintf(w, "\r\n");

    fprintf(w, "%-30s%-10s\r\n", "Counter", "Value");
    fprintf(w, "-----------------------------------------\r\n");
    fprintf(w, "%-30s%lu\r\n", "radius_retries",	GET_STAT(radius_retries));
    fprintf(w, "%-30s%lu\r\n", "arp_errors",		GET_STAT(arp_errors));
    fprintf(w, "%-30s%lu\r\n", "arp_replies",		GET_STAT(arp_replies));
    fprintf(w, "%-30s%lu\r\n", "arp_discarded",		GET_STAT(arp_discarded));
    fprintf(w, "%-30s%lu\r\n", "arp_sent",		GET_STAT(arp_sent));
    fprintf(w, "%-30s%lu\r\n", "arp_recv",		GET_STAT(arp_recv));
    fprintf(w, "%-30s%lu\r\n", "packets_snooped",	GET_STAT(packets_snooped));
    fprintf(w, "%-30s%lu\r\n", "tunnel_created",	GET_STAT(tunnel_created));
    fprintf(w, "%-30s%lu\r\n", "session_created",	GET_STAT(session_created));
    fprintf(w, "%-30s%lu\r\n", "tunnel_timeout",	GET_STAT(tunnel_timeout));
    fprintf(w, "%-30s%lu\r\n", "session_timeout",	GET_STAT(session_timeout));
    fprintf(w, "%-30s%lu\r\n", "radius_timeout",	GET_STAT(radius_timeout));
    fprintf(w, "%-30s%lu\r\n", "radius_overflow",	GET_STAT(radius_overflow));
    fprintf(w, "%-30s%lu\r\n", "tunnel_overflow",	GET_STAT(tunnel_overflow));
    fprintf(w, "%-30s%lu\r\n", "session_overflow",	GET_STAT(session_overflow));
    fprintf(w, "%-30s%lu\r\n", "ip_allocated",		GET_STAT(ip_allocated));
    fprintf(w, "%-30s%lu\r\n", "ip_freed",		GET_STAT(ip_freed));

#ifdef STAT_CALLS
    fprintf(w, "\n%-30s%-10s\r\n", "Counter", "Value");
    fprintf(w, "-----------------------------------------\r\n");
    fprintf(w, "%-30s%lu\r\n", "call_processtap",	GET_STAT(call_processtap));
    fprintf(w, "%-30s%lu\r\n", "call_processarp",	GET_STAT(call_processarp));
    fprintf(w, "%-30s%lu\r\n", "call_processipout",	GET_STAT(call_processipout));
    fprintf(w, "%-30s%lu\r\n", "call_processudp",	GET_STAT(call_processudp));
    fprintf(w, "%-30s%lu\r\n", "call_processpap",	GET_STAT(call_processpap));
    fprintf(w, "%-30s%lu\r\n", "call_processchap",	GET_STAT(call_processchap));
    fprintf(w, "%-30s%lu\r\n", "call_processlcp",	GET_STAT(call_processlcp));
    fprintf(w, "%-30s%lu\r\n", "call_processipcp",	GET_STAT(call_processipcp));
    fprintf(w, "%-30s%lu\r\n", "call_processipin",	GET_STAT(call_processipin));
    fprintf(w, "%-30s%lu\r\n", "call_processccp",	GET_STAT(call_processccp));
    fprintf(w, "%-30s%lu\r\n", "call_processrad",	GET_STAT(call_processrad));
    fprintf(w, "%-30s%lu\r\n", "call_sendarp",		GET_STAT(call_sendarp));
    fprintf(w, "%-30s%lu\r\n", "call_sendipcp",		GET_STAT(call_sendipcp));
    fprintf(w, "%-30s%lu\r\n", "call_sendchap",		GET_STAT(call_sendchap));
    fprintf(w, "%-30s%lu\r\n", "call_sessionbyip",	GET_STAT(call_sessionbyip));
    fprintf(w, "%-30s%lu\r\n", "call_sessionbyuser",	GET_STAT(call_sessionbyuser));
    fprintf(w, "%-30s%lu\r\n", "call_tunnelsend",	GET_STAT(call_tunnelsend));
    fprintf(w, "%-30s%lu\r\n", "call_tunnelkill",	GET_STAT(call_tunnelkill));
    fprintf(w, "%-30s%lu\r\n", "call_tunnelshutdown",	GET_STAT(call_tunnelshutdown));
    fprintf(w, "%-30s%lu\r\n", "call_sessionkill",	GET_STAT(call_sessionkill));
    fprintf(w, "%-30s%lu\r\n", "call_sessionshutdown",	GET_STAT(call_sessionshutdown));
    fprintf(w, "%-30s%lu\r\n", "call_sessionsetup",	GET_STAT(call_sessionsetup));
    fprintf(w, "%-30s%lu\r\n", "call_assign_ip_address",GET_STAT(call_assign_ip_address));
    fprintf(w, "%-30s%lu\r\n", "call_free_ip_address",	GET_STAT(call_free_ip_address));
    fprintf(w, "%-30s%lu\r\n", "call_dump_acct_info",	GET_STAT(call_dump_acct_info));
    fprintf(w, "%-30s%lu\r\n", "call_radiussend",	GET_STAT(call_radiussend));
    fprintf(w, "%-30s%lu\r\n", "call_radiusretry",	GET_STAT(call_radiusretry));
#endif
    return CLI_OK;
}

int cmd_show_version(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    fprintf(w, "L2TPNS %s\r\n", VERSION);
    fprintf(w, "ID: %s\r\n", rcs_id);
    return CLI_OK;
}
int cmd_show_pool(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    int i;
    int used = 0, free = 0;

    fprintf(w, "%-15s  %4s   %8s %s\r\n", "IP Address", "Used", "Session", "User");
    for (i = 0; i < MAXIPPOOL; i++)
    {
	sessionidt s = 0;

	if (!ip_address_pool[i].address) continue;
	if (ip_address_pool[i].assigned)
	{
	    used++;
	    s = sessionbyip(ip_address_pool[i].address);
	}
	else
	{
	    free++;
	}

	fprintf(w, "%-15s %4s %8d %s\r\n",
		inet_toa(ip_address_pool[i].address),
		(s) ? "Y" : "N",
		s,
		session[s].user);
    }
    fprintf(w, "\r\nFree: %d\r\nUsed: %d\r\n", free, used);
    return CLI_OK;
}
int cmd_show_banana(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    fputs(" _\r\n"
          "//\\\r\n"
          "V  \\\r\n"
          " \\  \\_\r\n"
          "  \\,'.`-.\r\n"
          "   |\\ `. `.\r\n"
          "   ( \\  `. `-.                        _,.-:\\\r\n"
          "    \\ \\   `.  `-._             __..--' ,-';/\r\n"
          "     \\ `.   `-.   `-..___..---'   _.--' ,'/\r\n"
          "      `. `.    `-._        __..--'    ,' /\r\n"
          "        `. `-_     ``--..''       _.-' ,'\r\n"
          "          `-_ `-.___        __,--'   ,'\r\n"
          "             `-.__  `----\"\"\"    __.-'\r\n"
          "hh                `--..____..--'\r\n", w);

    return CLI_OK;
}

int cmd_clear_counters(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    fprintf(w, "Counters cleared\r\n");
    SET_STAT(last_reset, time(NULL));
    return CLI_OK;
}

int cmd_drop_user(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    int i;
    sessionidt s;

    if (!argc)
    {
	fprintf(w, "Specify a user to drop\r\n");
	return CLI_OK;
    }
    for (i = 0; i < argc; i++)
    {
	if (strchr(argv[i], '?'))
	{
	    fprintf(w, "username ...");
	    return CLI_OK;
	}
    }

    for (i = 0; i < argc; i++)
    {
	if (!(s = sessionbyuser(argv[i])))
	{
	    fprintf(w, "User %s is not connected\r\n", argv[i]);
	    continue;
	}

	if (session[s].ip && session[s].opened && !session[s].die)
	{
	    int x;

	    fprintf(w, "Dropping user %s\r\n", session[s].user);
	    for (x = 0; x < MAXSESSION; x++)
	    {
		if (!cli_session_kill[x])
		{
		    cli_session_kill[x] = s;
		    break;
		}
	    }
	}
    }

    return CLI_OK;
}

int cmd_drop_tunnel(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    int i;
    tunnelidt tid;

    if (!argc)
    {
	fprintf(w, "Specify a tunnel to drop\r\n");
	return CLI_OK;
    }
    for (i = 0; i < argc; i++)
    {
	if (strchr(argv[i], '?'))
	{
	    fprintf(w, "tunnel_id ...");
	    return CLI_OK;
	}
    }

    for (i = 0; i < argc; i++)
    {
	int x;

	if ((tid = atol(argv[i])) <= 0 || (tid > MAXTUNNEL))
	{
	    fprintf(w, "Invalid tunnel ID (%d - %d)\r\n", 0, MAXTUNNEL);
	    continue;
	}

	if (!tunnel[tid].ip)
	{
	    fprintf(w, "Tunnel %d is not connected\r\n", tid);
	    continue;
	}

	if (tunnel[tid].die)
	{
	    fprintf(w, "Tunnel %d is already being shut down\r\n", tid);
	    continue;
	}

	for (x = 0; x < MAXTUNNEL; x++)
	{
	    if (!cli_tunnel_kill[x])
	    {
		cli_tunnel_kill[x] = tid;
		fprintf(w, "Tunnel %d shut down (%s)\r\n", tid, tunnel[tid].hostname);
		break;
	    }
	}
    }

    return CLI_OK;
}

int cmd_drop_session(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    int i;
    sessionidt s;

    if (!argc)
    {
	fprintf(w, "Specify a session id to drop\r\n");
	return CLI_OK;
    }
    for (i = 0; i < argc; i++)
    {
	if (strchr(argv[i], '?'))
	{
	    fprintf(w, "session_id ...");
	    return CLI_OK;
	}
    }

    for (i = 0; i < argc; i++)
    {
	if ((s = atol(argv[i])) <= 0 || (s > MAXSESSION))
	{
	    fprintf(w, "Invalid session ID (%d - %d)\r\n", 0, MAXSESSION);
	    continue;
	}

	if (session[s].ip && session[s].opened && !session[s].die)
	{
	    int x;
	    for (x = 0; x < MAXSESSION; x++)
	    {
		if (!cli_session_kill[x])
		{
		    cli_session_kill[x] = s;
		    break;
		}
	    }
	    fprintf(w, "Dropping session %d\r\n", s);
	}
	else
	{
	    fprintf(w, "Session %d is not active.\r\n", s);
	}
    }

    return CLI_OK;
}

int cmd_snoop(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    int i;
    sessionidt s;

    if (!argc)
    {
	fprintf(w, "Specify a user\r\n");
	return CLI_OK;
    }
    for (i = 0; i < argc; i++)
    {
	if (strchr(argv[i], '?'))
	{
	    fprintf(w, "username ...");
	    return CLI_OK;
	}
    }

    for (i = 0; i < argc; i++)
    {
	if (!(s = sessionbyuser(argv[i])))
	{
	    fprintf(w, "User %s is not connected\r\n", argv[i]);
	    continue;
	}
	session[s].snoop = 1;

	fprintf(w, "Snooping user %s\r\n", argv[i]);
    }
    return CLI_OK;
}

int cmd_no_snoop(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    int i;
    sessionidt s;

    if (!argc)
    {
	fprintf(w, "Specify a user\r\n");
	return CLI_OK;
    }
    for (i = 0; i < argc; i++)
    {
	if (strchr(argv[i], '?'))
	{
	    fprintf(w, "username ...");
	    return CLI_OK;
	}
    }

    for (i = 0; i < argc; i++)
    {
	if (!(s = sessionbyuser(argv[i])))
	{
	    fprintf(w, "User %s is not connected\r\n", argv[i]);
	    continue;
	}
	session[s].snoop = 0;

	fprintf(w, "Not snooping user %s\r\n", argv[i]);
    }
    return CLI_OK;
}

int cmd_throttle(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    int i;
    sessionidt s;

    if (!argc)
    {
	fprintf(w, "Specify a user\r\n");
	return CLI_OK;
    }
    for (i = 0; i < argc; i++)
    {
	if (strchr(argv[i], '?'))
	{
	    fprintf(w, "username ...");
	    return CLI_OK;
	}
    }

    for (i = 0; i < argc; i++)
    {
	if (!(s = sessionbyuser(argv[i])))
	{
	    fprintf(w, "User %s is not connected\r\n", argv[i]);
	    continue;
	}
	throttle_session(s, 1);

	fprintf(w, "throttling user %s\r\n", argv[i]);
    }
    return CLI_OK;
}

int cmd_no_throttle(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    int i;
    sessionidt s;

    if (!argc)
    {
	fprintf(w, "Specify a user\r\n");
	return CLI_OK;
    }
    for (i = 0; i < argc; i++)
    {
	if (strchr(argv[i], '?'))
	{
	    fprintf(w, "username ...");
	    return CLI_OK;
	}
    }

    for (i = 0; i < argc; i++)
	{
	if (!(s = sessionbyuser(argv[i])))
	{
	    fprintf(w, "User %s is not connected\r\n", argv[i]);
	    continue;
	}
	throttle_session(s, 0);

	fprintf(w, "unthrottling user %s\r\n", argv[i]);
    }
    return CLI_OK;
}

int cmd_debug(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    int i;

    if (!argc)
    {
	fprintf(w, "Currently debugging: ");
	if (debug_flags.critical) fprintf(w, "critical ");
	if (debug_flags.error) fprintf(w, "error ");
	if (debug_flags.warning) fprintf(w, "warning ");
	if (debug_flags.info) fprintf(w, "info ");
	if (debug_flags.calls) fprintf(w, "calls ");
	if (debug_flags.data) fprintf(w, "data ");
	fprintf(w, "\r\n");
	return CLI_OK;
    }

    for (i = 0; i < argc; i++)
    {
	if (*argv[i] == '?')
	{
	    fprintf(w, "Possible debugging states are:\r\n");
	    fprintf(w, "	critical\r\n");
	    fprintf(w, "	error\r\n");
	    fprintf(w, "	warning\r\n");
	    fprintf(w, "	info\r\n");
	    fprintf(w, "	calls\r\n");
	    fprintf(w, "	data\r\n");
	    return CLI_OK;
	}
    }

    for (i = 0; i < argc; i++)
    {
	if (strcasecmp(argv[i], "critical") == 0) debug_flags.critical = 1;
	if (strcasecmp(argv[i], "error") == 0) debug_flags.error = 1;
	if (strcasecmp(argv[i], "warning") == 0) debug_flags.warning = 1;
	if (strcasecmp(argv[i], "info") == 0) debug_flags.info = 1;
	if (strcasecmp(argv[i], "calls") == 0) debug_flags.calls = 1;
	if (strcasecmp(argv[i], "data") == 0) debug_flags.data = 1;
	if (strcasecmp(argv[i], "all") == 0)
	{
	    memset(&debug_flags, 1, sizeof(debug_flags));
	    debug_flags.data = 0;
	}
    }

    return CLI_OK;
}

int cmd_no_debug(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    int i;

    for (i = 0; i < argc; i++)
    {
	if (strcasecmp(argv[i], "critical") == 0) debug_flags.critical = 0;
	if (strcasecmp(argv[i], "error") == 0) debug_flags.error = 0;
	if (strcasecmp(argv[i], "warning") == 0) debug_flags.warning = 0;
	if (strcasecmp(argv[i], "info") == 0) debug_flags.info = 0;
	if (strcasecmp(argv[i], "calls") == 0) debug_flags.calls = 0;
	if (strcasecmp(argv[i], "data") == 0) debug_flags.data = 0;
	if (strcasecmp(argv[i], "all") == 0) memset(&debug_flags, 0, sizeof(debug_flags));
    }

    return CLI_OK;
}

int cmd_watch_session(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    sessionidt s;

    if (argc != 1)
    {
	fprintf(w, "Specify a single session to debug (0 to disable)\r\n");
	return CLI_OK;
    }
    s = atoi(argv[0]);

    if (debug_session)
	fprintf(w, "No longer debugging session %d\r\n", debug_session);

    if (s) fprintf(w, "Debugging session %d.\r\n", s);
    debug_session = s;

    return CLI_OK;
}

int cmd_watch_tunnel(struct cli_def *cli, FILE *w, char *command, char **argv, int argc)
{
    tunnelidt s;

    if (argc != 1)
    {
	fprintf(w, "Specify a single tunnel to debug (0 to disable)\r\n");
	return CLI_OK;
    }
    s = atoi(argv[0]);

    if (debug_tunnel)
	fprintf(w, "No longer debugging tunnel %d\r\n", debug_tunnel);

    if (s) fprintf(w, "Debugging tunnel %d.\r\n", s);
    debug_tunnel = s;

    return CLI_OK;
}

int regular_stuff(struct cli_def *cli, FILE *w)
{
    int i = debug_rb_tail;

#ifdef RINGBUFFER
    while (i != ringbuffer->tail)
    {
	int show_message = 0;

	if (*ringbuffer->buffer[i].message)
	{
	    // Always show messages if we are doing general debug
	    if (ringbuffer->buffer[i].level == 0 && debug_flags.critical) show_message = 1;
	    if (ringbuffer->buffer[i].level == 1 && debug_flags.error) show_message = 1;
	    if (ringbuffer->buffer[i].level == 2 && debug_flags.warning) show_message = 1;
	    if (ringbuffer->buffer[i].level == 3 && debug_flags.info) show_message = 1;
	    if (ringbuffer->buffer[i].level == 4 && debug_flags.calls) show_message = 1;
	    if (ringbuffer->buffer[i].level == 5 && debug_flags.data) show_message = 1;
	}

	if (show_message)
	{
	    ipt address = ntohl(ringbuffer->buffer[i].address);
	    char *ipaddr;
	    struct in_addr addr;

	    memcpy(&addr, &address, sizeof(ringbuffer->buffer[i].address));
	    ipaddr = inet_ntoa(addr);

	    fprintf(w, "%s-%s-%u-%u %s\r",
		    debug_levels[(int)ringbuffer->buffer[i].level],
		    ipaddr,
		    ringbuffer->buffer[i].tunnel,
		    ringbuffer->buffer[i].session,
		    ringbuffer->buffer[i].message);
	}

	if (++i == ringbuffer->tail) break;
	if (i == RINGBUFFER_SIZE) i = 0;
    }

    debug_rb_tail = ringbuffer->tail;
#endif
    return CLI_OK;
}

#endif