Skip to content
Snippets Groups Projects
asterisk.c 39.2 KiB
Newer Older
Mark Spencer's avatar
Mark Spencer committed
/*
 * Asterisk -- A telephony toolkit for Linux.
 *
 * Top level source file for asterisk
 * 
Mark Spencer's avatar
Mark Spencer committed
 * Copyright (C) 1999, Mark Spencer
Mark Spencer's avatar
Mark Spencer committed
 *
 * Mark Spencer <markster@linux-support.net>
 *
 * This program is free software, distributed under the terms of
 * the GNU General Public License
 */

#include <unistd.h>
#include <stdlib.h>
#include <asterisk/logger.h>
#include <asterisk/options.h>
Mark Spencer's avatar
Mark Spencer committed
#include <asterisk/cli.h>
Mark Spencer's avatar
Mark Spencer committed
#include <asterisk/channel.h>
Mark Spencer's avatar
Mark Spencer committed
#include <asterisk/ulaw.h>
Mark Spencer's avatar
Mark Spencer committed
#include <asterisk/alaw.h>
Mark Spencer's avatar
Mark Spencer committed
#include <asterisk/callerid.h>
Mark Spencer's avatar
Mark Spencer committed
#include <asterisk/module.h>
Mark Spencer's avatar
Mark Spencer committed
#include <asterisk/image.h>
Mark Spencer's avatar
Mark Spencer committed
#include <asterisk/tdd.h>
Mark Spencer's avatar
Mark Spencer committed
#include <asterisk/term.h>
Mark Spencer's avatar
Mark Spencer committed
#include <asterisk/manager.h>
Mark Spencer's avatar
Mark Spencer committed
#include <asterisk/pbx.h>
#include <asterisk/enum.h>
#include <asterisk/rtp.h>
Mark Spencer's avatar
Mark Spencer committed
#include <sys/resource.h>
Mark Spencer's avatar
Mark Spencer committed
#include <fcntl.h>
Mark Spencer's avatar
Mark Spencer committed
#include <stdio.h>
#include <signal.h>
Mark Spencer's avatar
Mark Spencer committed
#include <sched.h>
Mark Spencer's avatar
Mark Spencer committed
#include <asterisk/io.h>
Mark Spencer's avatar
Mark Spencer committed
#include <pthread.h>
Mark Spencer's avatar
Mark Spencer committed
#include <sys/socket.h>
#include <sys/un.h>
#include <sys/select.h>
Mark Spencer's avatar
Mark Spencer committed
#include <string.h>
#include <errno.h>
Mark Spencer's avatar
Mark Spencer committed
#include <ctype.h>
#include "editline/histedit.h"
Mark Spencer's avatar
Mark Spencer committed
#include "asterisk.h"
Mark Spencer's avatar
Mark Spencer committed
#include <asterisk/config.h>
Mark Spencer's avatar
Mark Spencer committed
#define AST_MAX_CONNECTS 128
#define NUM_MSGS 64

Mark Spencer's avatar
Mark Spencer committed
int option_verbose=0;
int option_debug=0;
int option_nofork=0;
int option_quiet=0;
Mark Spencer's avatar
Mark Spencer committed
int option_console=0;
Mark Spencer's avatar
Mark Spencer committed
int option_highpriority=0;
Mark Spencer's avatar
Mark Spencer committed
int option_remote=0;
int option_exec=0;
Mark Spencer's avatar
Mark Spencer committed
int option_initcrypto=0;
Mark Spencer's avatar
Mark Spencer committed
int option_nocolor;
Mark Spencer's avatar
Mark Spencer committed
int option_dumpcore = 0;
int option_overrideconfig = 0;
Mark Spencer's avatar
Mark Spencer committed
int fully_booted = 0;
Mark Spencer's avatar
Mark Spencer committed
static int ast_socket = -1;		/* UNIX Socket for allowing remote control */
static int ast_consock = -1;		/* UNIX Socket for controlling another asterisk */
static int mainpid;
struct console {
	int fd;					/* File descriptor */
	int p[2];				/* Pipe */
	pthread_t t;			/* Thread of handler */
};

static struct ast_atexit {
	void (*func)(void);
	struct ast_atexit *next;
} *atexits = NULL;
static ast_mutex_t atexitslock = AST_MUTEX_INITIALIZER;
Mark Spencer's avatar
Mark Spencer committed
time_t ast_startuptime;
time_t ast_lastreloadtime;

Mark Spencer's avatar
Mark Spencer committed
static History *el_hist = NULL;
static EditLine *el = NULL;
static char *remotehostname;

Mark Spencer's avatar
Mark Spencer committed
struct console consoles[AST_MAX_CONNECTS];

Mark Spencer's avatar
Mark Spencer committed
char defaultlanguage[MAX_LANGUAGE] = DEFAULT_LANGUAGE;

Mark Spencer's avatar
Mark Spencer committed
static int ast_el_add_history(char *);
static int ast_el_read_history(char *);
static int ast_el_write_history(char *);

char ast_config_AST_CONFIG_DIR[AST_CONFIG_MAX_PATH];
char ast_config_AST_CONFIG_FILE[AST_CONFIG_MAX_PATH];
char ast_config_AST_MODULE_DIR[AST_CONFIG_MAX_PATH];
char ast_config_AST_SPOOL_DIR[AST_CONFIG_MAX_PATH];
char ast_config_AST_VAR_DIR[AST_CONFIG_MAX_PATH];
char ast_config_AST_LOG_DIR[AST_CONFIG_MAX_PATH];
char ast_config_AST_AGI_DIR[AST_CONFIG_MAX_PATH];
char ast_config_AST_DB[AST_CONFIG_MAX_PATH];
char ast_config_AST_KEY_DIR[AST_CONFIG_MAX_PATH];
char ast_config_AST_PID[AST_CONFIG_MAX_PATH];
char ast_config_AST_SOCKET[AST_CONFIG_MAX_PATH];
char ast_config_AST_RUN_DIR[AST_CONFIG_MAX_PATH];

static char *_argv[256];
static int shuttingdown = 0;
static int restartnow = 0;
static pthread_t consolethread = (pthread_t) -1;

int ast_register_atexit(void (*func)(void))
{
	int res = -1;
	struct ast_atexit *ae;
	ast_unregister_atexit(func);
	ae = malloc(sizeof(struct ast_atexit));
	if (ae) {
		memset(ae, 0, sizeof(struct ast_atexit));
		ae->next = atexits;
		ae->func = func;
		atexits = ae;
		res = 0;
	}
	return res;
}

void ast_unregister_atexit(void (*func)(void))
{
	struct ast_atexit *ae, *prev = NULL;
	ae = atexits;
	while(ae) {
		if (ae->func == func) {
			if (prev)
				prev->next = ae->next;
			else
				atexits = ae->next;
			break;
		}
		prev = ae;
		ae = ae->next;
	}
static int fdprint(int fd, const char *s)
Mark Spencer's avatar
Mark Spencer committed
{
	return write(fd, s, strlen(s) + 1);
}

/*
 * write the string to all attached console clients
 */
static void ast_network_puts(const char *string)
{
    int x;
    for (x=0;x<AST_MAX_CONNECTS; x++) {
	if (consoles[x].fd > -1) 
	    fdprint(consoles[x].p[1], string);
    }
}


/*
 * write the string to the console, and all attached
 * console clients
 */
void ast_console_puts(const char *string)
{
    fputs(string, stdout);
    fflush(stdout);
    ast_network_puts(string);
}

static void network_verboser(const char *s, int pos, int replace, int complete)
Mark Spencer's avatar
Mark Spencer committed
{
    ast_network_puts(s);
Mark Spencer's avatar
Mark Spencer committed
}

static pthread_t lthread;

static void *netconsole(void *vconsole)
{
	struct console *con = vconsole;
	char hostname[256];
	char tmp[512];
	int res;
	int max;
	fd_set rfds;
	
	if (gethostname(hostname, sizeof(hostname)))
Mark Spencer's avatar
Mark Spencer committed
		strncpy(hostname, "<Unknown>", sizeof(hostname)-1);
Mark Spencer's avatar
Mark Spencer committed
	snprintf(tmp, sizeof(tmp), "%s/%d/%s\n", hostname, mainpid, ASTERISK_VERSION);
	fdprint(con->fd, tmp);
	for(;;) {
		FD_ZERO(&rfds);	
		FD_SET(con->fd, &rfds);
		FD_SET(con->p[0], &rfds);
		max = con->fd;
		if (con->p[0] > max)
			max = con->p[0];
		res = ast_select(max + 1, &rfds, NULL, NULL, NULL);
Mark Spencer's avatar
Mark Spencer committed
		if (res < 0) {
			ast_log(LOG_WARNING, "select returned < 0: %s\n", strerror(errno));
			continue;
		}
		if (FD_ISSET(con->fd, &rfds)) {
			res = read(con->fd, tmp, sizeof(tmp));
Mark Spencer's avatar
Mark Spencer committed
			if (res < 1) {
Mark Spencer's avatar
Mark Spencer committed
				break;
Mark Spencer's avatar
Mark Spencer committed
			}
Mark Spencer's avatar
Mark Spencer committed
			tmp[res] = 0;
			ast_cli_command(con->fd, tmp);
		}
		if (FD_ISSET(con->p[0], &rfds)) {
			res = read(con->p[0], tmp, sizeof(tmp));
			if (res < 1) {
				ast_log(LOG_ERROR, "read returned %d\n", res);
				break;
			}
			res = write(con->fd, tmp, res);
			if (res < 1)
				break;
		}
	}
	if (option_verbose > 2) 
		ast_verbose(VERBOSE_PREFIX_3 "Remote UNIX connection disconnected\n");
	close(con->fd);
	close(con->p[0]);
	close(con->p[1]);
	con->fd = -1;
	
	return NULL;
}

static void *listener(void *unused)
{
	struct sockaddr_un sun;
Mark Spencer's avatar
Mark Spencer committed
	int s;
	int len;
	int x;
Mark Spencer's avatar
Mark Spencer committed
	int flags;
Mark Spencer's avatar
Mark Spencer committed
	pthread_attr_t attr;
	pthread_attr_init(&attr);
	pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
	for(;;) {
Mark Spencer's avatar
Mark Spencer committed
		if (ast_socket < 0)
			return NULL;
		FD_ZERO(&fds);
		FD_SET(ast_socket, &fds);
		s = ast_select(ast_socket + 1, &fds, NULL, NULL, NULL);
		if (s < 0) {
			ast_log(LOG_WARNING, "Select retured error: %s\n", strerror(errno));
			continue;
		}
Mark Spencer's avatar
Mark Spencer committed
		len = sizeof(sun);
		s = accept(ast_socket, (struct sockaddr *)&sun, &len);
		if (s < 0) {
			if (errno != EINTR)
				ast_log(LOG_WARNING, "Accept retured %d: %s\n", s, strerror(errno));
Mark Spencer's avatar
Mark Spencer committed
		} else {
			for (x=0;x<AST_MAX_CONNECTS;x++) {
				if (consoles[x].fd < 0) {
Mark Spencer's avatar
Mark Spencer committed
					if (socketpair(AF_LOCAL, SOCK_STREAM, 0, consoles[x].p)) {
Mark Spencer's avatar
Mark Spencer committed
						ast_log(LOG_ERROR, "Unable to create pipe: %s\n", strerror(errno));
						consoles[x].fd = -1;
						fdprint(s, "Server failed to create pipe\n");
						close(s);
						break;
					}
Mark Spencer's avatar
Mark Spencer committed
					flags = fcntl(consoles[x].p[1], F_GETFL);
					fcntl(consoles[x].p[1], F_SETFL, flags | O_NONBLOCK);
Mark Spencer's avatar
Mark Spencer committed
					consoles[x].fd = s;
					if (pthread_create(&consoles[x].t, &attr, netconsole, &consoles[x])) {
						ast_log(LOG_ERROR, "Unable to spawn thread to handle connection\n");
						consoles[x].fd = -1;
						fdprint(s, "Server failed to spawn thread\n");
						close(s);
					}
					break;
				}
			}
			if (x >= AST_MAX_CONNECTS) {
				fdprint(s, "No more connections allowed\n");
				ast_log(LOG_WARNING, "No more connections allowed\n");
				close(s);
			} else if (consoles[x].fd > -1) {
				if (option_verbose > 2) 
					ast_verbose(VERBOSE_PREFIX_3 "Remote UNIX connection\n");
			}
		}
	}
	return NULL;
}

static int ast_makesocket(void)
{
	struct sockaddr_un sun;
	int res;
	int x;
	for (x=0;x<AST_MAX_CONNECTS;x++)	
		consoles[x].fd = -1;
Mark Spencer's avatar
Mark Spencer committed
	unlink((char *)ast_config_AST_SOCKET);
Mark Spencer's avatar
Mark Spencer committed
	ast_socket = socket(PF_LOCAL, SOCK_STREAM, 0);
	if (ast_socket < 0) {
		ast_log(LOG_WARNING, "Unable to create control socket: %s\n", strerror(errno));
		return -1;
	}		
	memset(&sun, 0, sizeof(sun));
	sun.sun_family = AF_LOCAL;
Mark Spencer's avatar
Mark Spencer committed
	strncpy(sun.sun_path, (char *)ast_config_AST_SOCKET, sizeof(sun.sun_path)-1);
Mark Spencer's avatar
Mark Spencer committed
	res = bind(ast_socket, (struct sockaddr *)&sun, sizeof(sun));
	if (res) {
Mark Spencer's avatar
Mark Spencer committed
		ast_log(LOG_WARNING, "Unable to bind socket to %s: %s\n", (char *)ast_config_AST_SOCKET, strerror(errno));
Mark Spencer's avatar
Mark Spencer committed
		close(ast_socket);
		ast_socket = -1;
		return -1;
	}
	res = listen(ast_socket, 2);
	if (res < 0) {
Mark Spencer's avatar
Mark Spencer committed
		ast_log(LOG_WARNING, "Unable to listen on socket %s: %s\n", (char *)ast_config_AST_SOCKET, strerror(errno));
Mark Spencer's avatar
Mark Spencer committed
		close(ast_socket);
		ast_socket = -1;
		return -1;
	}
	ast_register_verbose(network_verboser);
	pthread_create(&lthread, NULL, listener, NULL);
	return 0;
}

static int ast_tryconnect(void)
{
	struct sockaddr_un sun;
	int res;
	ast_consock = socket(PF_LOCAL, SOCK_STREAM, 0);
	if (ast_consock < 0) {
		ast_log(LOG_WARNING, "Unable to create socket: %s\n", strerror(errno));
		return 0;
	}
	memset(&sun, 0, sizeof(sun));
	sun.sun_family = AF_LOCAL;
Mark Spencer's avatar
Mark Spencer committed
	strncpy(sun.sun_path, (char *)ast_config_AST_SOCKET, sizeof(sun.sun_path)-1);
Mark Spencer's avatar
Mark Spencer committed
	res = connect(ast_consock, (struct sockaddr *)&sun, sizeof(sun));
	if (res) {
		close(ast_consock);
		ast_consock = -1;
		return 0;
	} else
		return 1;
}

Mark Spencer's avatar
Mark Spencer committed
static void urg_handler(int num)
{
	/* Called by soft_hangup to interrupt the select, read, or other
	   system call.  We don't actually need to do anything though.  */
	/* Cannot EVER ast_log from within a signal handler */
Mark Spencer's avatar
Mark Spencer committed
	if (option_debug) 
Mark Spencer's avatar
Mark Spencer committed
	signal(num, urg_handler);
Mark Spencer's avatar
Mark Spencer committed
	return;
}

Mark Spencer's avatar
Mark Spencer committed
static void hup_handler(int num)
{
	if (option_verbose > 1) 
		printf("Received HUP signal -- Reloading configs\n");
	if (restartnow)
		execvp(_argv[0], _argv);
	/* XXX This could deadlock XXX */
Mark Spencer's avatar
Mark Spencer committed
	ast_module_reload();
}

static void child_handler(int sig)
Mark Spencer's avatar
Mark Spencer committed
{
	/* Must not ever ast_log or ast_verbose within signal handler */
	int n, status;

	/*
	 * Reap all dead children -- not just one
	 */
	for (n = 0; wait4(-1, &status, WNOHANG, NULL) > 0; n++)
		;
	if (n == 0 && option_debug)	
		printf("Huh?  Child handler, but nobody there?\n");
Mark Spencer's avatar
Mark Spencer committed
}
Mark Spencer's avatar
Mark Spencer committed
static void set_title(char *text)
{
	/* Set an X-term or screen title */
	if (getenv("TERM") && strstr(getenv("TERM"), "xterm"))
		fprintf(stdout, "\033]2;%s\007", text);
}

static void set_icon(char *text)
{
	if (getenv("TERM") && strstr(getenv("TERM"), "xterm"))
		fprintf(stdout, "\033]1;%s\007", text);
}

Mark Spencer's avatar
Mark Spencer committed
static int set_priority(int pri)
{
	struct sched_param sched;
Mark Spencer's avatar
Mark Spencer committed
	memset(&sched, 0, sizeof(sched));
Mark Spencer's avatar
Mark Spencer committed
	/* We set ourselves to a high priority, that we might pre-empt everything
	   else.  If your PBX has heavy activity on it, this is a good thing.  */
Mark Spencer's avatar
Mark Spencer committed
#ifdef __linux__
Mark Spencer's avatar
Mark Spencer committed
	if (pri) {  
Mark Spencer's avatar
Mark Spencer committed
		sched.sched_priority = 10;
		if (sched_setscheduler(0, SCHED_RR, &sched)) {
Mark Spencer's avatar
Mark Spencer committed
			ast_log(LOG_WARNING, "Unable to set high priority\n");
			return -1;
Mark Spencer's avatar
Mark Spencer committed
		} else
			if (option_verbose)
				ast_verbose("Set to realtime thread\n");
Mark Spencer's avatar
Mark Spencer committed
	} else {
		sched.sched_priority = 0;
		if (sched_setscheduler(0, SCHED_OTHER, &sched)) {
			ast_log(LOG_WARNING, "Unable to set normal priority\n");
			return -1;
		}
	}
Mark Spencer's avatar
Mark Spencer committed
#else
	if (pri) {
		if (setpriority(PRIO_PROCESS, 0, -10) == -1) {
			ast_log(LOG_WARNING, "Unable to set high priority\n");
			return -1;
		} else
			if (option_verbose)
				ast_verbose("Set to high priority\n");
	} else {
		if (setpriority(PRIO_PROCESS, 0, 0) == -1) {
			ast_log(LOG_WARNING, "Unable to set normal priority\n");
			return -1;
		}
	}
#endif
Mark Spencer's avatar
Mark Spencer committed
	return 0;
}

static void ast_run_atexits(void)
{
	struct ast_atexit *ae;
	ae = atexits;
	while(ae) {
		if (ae->func) 
			ae->func();
		ae = ae->next;
	}
Mark Spencer's avatar
Mark Spencer committed
static void quit_handler(int num, int nice, int safeshutdown, int restart)
Mark Spencer's avatar
Mark Spencer committed
{
Mark Spencer's avatar
Mark Spencer committed
	char filename[80] = "";
Mark Spencer's avatar
Mark Spencer committed
	time_t s,e;
	int x;
	if (safeshutdown) {
		shuttingdown = 1;
		if (!nice) {
			/* Begin shutdown routine, hanging up active channels */
			ast_begin_shutdown(1);
			if (option_verbose && option_console)
				ast_verbose("Beginning asterisk %s....\n", restart ? "restart" : "shutdown");
			time(&s);
			for(;;) {
				time(&e);
				/* Wait up to 15 seconds for all channels to go away */
				if ((e - s) > 15)
					break;
				if (!ast_active_channels())
					break;
				if (!shuttingdown)
					break;
				/* Sleep 1/10 of a second */
				usleep(100000);
			}
		} else {
			if (nice < 2)
				ast_begin_shutdown(0);
			if (option_verbose && option_console)
				ast_verbose("Waiting for inactivity to perform %s...\n", restart ? "restart" : "halt");
			for(;;) {
				if (!ast_active_channels())
					break;
				if (!shuttingdown)
					break;
				sleep(1);
			}
		}

		if (!shuttingdown) {
			if (option_verbose && option_console)
				ast_verbose("Asterisk %s cancelled.\n", restart ? "restart" : "shutdown");
			return;
		}
	}
Mark Spencer's avatar
Mark Spencer committed
	if (option_console || option_remote) {
		if (getenv("HOME")) 
			snprintf(filename, sizeof(filename), "%s/.asterisk_history", getenv("HOME"));
		if (strlen(filename))
Mark Spencer's avatar
Mark Spencer committed
			ast_el_write_history(filename);
		if (el != NULL)
			el_end(el);
		if (el_hist != NULL)
			history_end(el_hist);
Mark Spencer's avatar
Mark Spencer committed
	}
	if (option_verbose)
		ast_verbose("Executing last minute cleanups\n");
	ast_run_atexits();
Mark Spencer's avatar
Mark Spencer committed
	/* Called on exit */
Mark Spencer's avatar
Mark Spencer committed
	if (option_verbose && option_console)
Mark Spencer's avatar
Mark Spencer committed
		ast_verbose("Asterisk %s ending (%d).\n", ast_active_channels() ? "uncleanly" : "cleanly", num);
Mark Spencer's avatar
Mark Spencer committed
	else if (option_debug)
		ast_log(LOG_DEBUG, "Asterisk ending (%d).\n", num);
	manager_event(EVENT_FLAG_SYSTEM, "Shutdown", "Shutdown: %s\r\nRestart: %s\r\n", ast_active_channels() ? "Uncleanly" : "Cleanly", restart ? "True" : "False");
Mark Spencer's avatar
Mark Spencer committed
	if (ast_socket > -1) {
Mark Spencer's avatar
Mark Spencer committed
		close(ast_socket);
Mark Spencer's avatar
Mark Spencer committed
		ast_socket = -1;
	}
Mark Spencer's avatar
Mark Spencer committed
	if (ast_consock > -1)
		close(ast_consock);
	if (ast_socket > -1)
Mark Spencer's avatar
Mark Spencer committed
		unlink((char *)ast_config_AST_SOCKET);
	if (!option_remote) unlink((char *)ast_config_AST_PID);
Mark Spencer's avatar
Mark Spencer committed
	printf(term_quit());
	if (restart) {
		if (option_verbose || option_console)
			ast_verbose("Preparing for Asterisk restart...\n");
		/* Mark all FD's for closing on exec */
		for (x=3;x<32768;x++) {
			fcntl(x, F_SETFD, FD_CLOEXEC);
		}
		if (option_verbose || option_console)
			ast_verbose("Restarting Asterisk NOW...\n");
		restartnow = 1;
		/* If there is a consolethread running send it a SIGHUP 
		   so it can execvp, otherwise we can do it ourselves */
		if (consolethread != (pthread_t) -1)
			pthread_kill(consolethread, SIGHUP);
		else
			execvp(_argv[0], _argv);
	
Mark Spencer's avatar
Mark Spencer committed
}

static void __quit_handler(int num)
{
	quit_handler(num, 0, 1, 0);
Mark Spencer's avatar
Mark Spencer committed
static const char *fix_header(char *outbuf, int maxout, const char *s, char *cmp)
Mark Spencer's avatar
Mark Spencer committed
{
Mark Spencer's avatar
Mark Spencer committed
	const char *c;
	if (!strncmp(s, cmp, strlen(cmp))) {
		c = s + strlen(cmp);
Mark Spencer's avatar
Mark Spencer committed
		term_color(outbuf, cmp, COLOR_GRAY, 0, maxout);
Mark Spencer's avatar
Mark Spencer committed
		return c;
Mark Spencer's avatar
Mark Spencer committed
	}
Mark Spencer's avatar
Mark Spencer committed
	return NULL;
static void console_verboser(const char *s, int pos, int replace, int complete)
Mark Spencer's avatar
Mark Spencer committed
{
Mark Spencer's avatar
Mark Spencer committed
	char tmp[80];
Mark Spencer's avatar
Mark Spencer committed
	const char *c=NULL;
Mark Spencer's avatar
Mark Spencer committed
	/* Return to the beginning of the line */
Mark Spencer's avatar
Mark Spencer committed
	if (!pos) {
Mark Spencer's avatar
Mark Spencer committed
		fprintf(stdout, "\r");
Mark Spencer's avatar
Mark Spencer committed
		if ((c = fix_header(tmp, sizeof(tmp), s, VERBOSE_PREFIX_4)) ||
			(c = fix_header(tmp, sizeof(tmp), s, VERBOSE_PREFIX_3)) ||
			(c = fix_header(tmp, sizeof(tmp), s, VERBOSE_PREFIX_2)) ||
			(c = fix_header(tmp, sizeof(tmp), s, VERBOSE_PREFIX_1)))
Mark Spencer's avatar
Mark Spencer committed
			fputs(tmp, stdout);
	}
Mark Spencer's avatar
Mark Spencer committed
	if (c)
		fputs(c + pos,stdout);
	else
		fputs(s + pos,stdout);
Mark Spencer's avatar
Mark Spencer committed
	fflush(stdout);
Mark Spencer's avatar
Mark Spencer committed
	if (complete)
	/* Wake up a select()ing console */
		if (option_console && consolethread != (pthread_t) -1)
Mark Spencer's avatar
Mark Spencer committed
			pthread_kill(consolethread, SIGURG);
Mark Spencer's avatar
Mark Spencer committed
}

static void consolehandler(char *s)
{
Mark Spencer's avatar
Mark Spencer committed
	printf(term_end());
	fflush(stdout);
Mark Spencer's avatar
Mark Spencer committed
	/* Called when readline data is available */
	if (s && strlen(s))
Mark Spencer's avatar
Mark Spencer committed
		ast_el_add_history(s);
Mark Spencer's avatar
Mark Spencer committed
	/* Give the console access to the shell */
	if (s) {
Mark Spencer's avatar
Mark Spencer committed
		/* The real handler for bang */
Mark Spencer's avatar
Mark Spencer committed
		if (s[0] == '!') {
			if (s[1])
				system(s+1);
			else
				system(getenv("SHELL") ? getenv("SHELL") : "/bin/sh");
		} else 
Mark Spencer's avatar
Mark Spencer committed
		ast_cli_command(STDOUT_FILENO, s);
Mark Spencer's avatar
Mark Spencer committed
	} else
		fprintf(stdout, "\nUse \"quit\" to exit\n");
Mark Spencer's avatar
Mark Spencer committed
static int remoteconsolehandler(char *s)
Mark Spencer's avatar
Mark Spencer committed
{
Mark Spencer's avatar
Mark Spencer committed
	int ret = 0;
Mark Spencer's avatar
Mark Spencer committed
	/* Called when readline data is available */
	if (s && strlen(s))
Mark Spencer's avatar
Mark Spencer committed
		ast_el_add_history(s);
Mark Spencer's avatar
Mark Spencer committed
	/* Give the console access to the shell */
	if (s) {
Mark Spencer's avatar
Mark Spencer committed
		/* The real handler for bang */
Mark Spencer's avatar
Mark Spencer committed
		if (s[0] == '!') {
			if (s[1])
				system(s+1);
			else
				system(getenv("SHELL") ? getenv("SHELL") : "/bin/sh");
Mark Spencer's avatar
Mark Spencer committed
			ret = 1;
		}
		if ((strncasecmp(s, "quit", 4) == 0 || strncasecmp(s, "exit", 4) == 0) &&
		    (s[4] == '\0' || isspace(s[4]))) {
Mark Spencer's avatar
Mark Spencer committed
			quit_handler(0, 0, 0, 0);
Mark Spencer's avatar
Mark Spencer committed
			ret = 1;
		}
Mark Spencer's avatar
Mark Spencer committed
	} else
		fprintf(stdout, "\nUse \"quit\" to exit\n");
Mark Spencer's avatar
Mark Spencer committed

	return ret;
Mark Spencer's avatar
Mark Spencer committed
static char quit_help[] = 
"Usage: quit\n"
"       Exits Asterisk.\n";

Mark Spencer's avatar
Mark Spencer committed
static char abort_halt_help[] = 
"Usage: abort shutdown\n"
"       Causes Asterisk to abort an executing shutdown or restart, and resume normal\n"
"       call operations.\n";

static char shutdown_now_help[] = 
Mark Spencer's avatar
Mark Spencer committed
"Usage: stop now\n"
Mark Spencer's avatar
Mark Spencer committed
"       Shuts down a running Asterisk immediately, hanging up all active calls .\n";

static char shutdown_gracefully_help[] = 
Mark Spencer's avatar
Mark Spencer committed
"Usage: stop gracefully\n"
Mark Spencer's avatar
Mark Spencer committed
"       Causes Asterisk to not accept new calls, and exit when all\n"
"       active calls have terminated normally.\n";

Mark Spencer's avatar
Mark Spencer committed
static char shutdown_when_convenient_help[] = 
"Usage: stop when convenient\n"
"       Causes Asterisk to perform a shutdown when all active calls have ended.\n";

Mark Spencer's avatar
Mark Spencer committed
static char restart_now_help[] = 
"Usage: restart now\n"
"       Causes Asterisk to hangup all calls and exec() itself performing a cold.\n"
"       restart.\n";

static char restart_gracefully_help[] = 
"Usage: restart gracefully\n"
"       Causes Asterisk to stop accepting new calls and exec() itself performing a cold.\n"
"       restart when all active calls have ended.\n";

static char restart_when_convenient_help[] = 
"Usage: restart when convenient\n"
"       Causes Asterisk to perform a cold restart when all active calls have ended.\n";
Mark Spencer's avatar
Mark Spencer committed
static char bang_help[] =
"Usage: !<command>\n"
"       Executes a given shell command\n";

Mark Spencer's avatar
Mark Spencer committed
static int handle_quit(int fd, int argc, char *argv[])
{
	if (argc != 1)
		return RESULT_SHOWUSAGE;
Mark Spencer's avatar
Mark Spencer committed
	quit_handler(0, 0, 1, 0);
	return RESULT_SUCCESS;
}
Mark Spencer's avatar
Mark Spencer committed
static int no_more_quit(int fd, int argc, char *argv[])
{
	if (argc != 1)
		return RESULT_SHOWUSAGE;
	ast_cli(fd, "The QUIT and EXIT commands may no longer be used to shutdown the PBX.\n"
	            "Please use STOP NOW instead, if you wish to shutdown the PBX.\n");
	return RESULT_SUCCESS;
}

Mark Spencer's avatar
Mark Spencer committed
static int handle_shutdown_now(int fd, int argc, char *argv[])
{
	if (argc != 2)
		return RESULT_SHOWUSAGE;
	quit_handler(0, 0 /* Not nice */, 1 /* safely */, 0 /* not restart */);
	return RESULT_SUCCESS;
}

static int handle_shutdown_gracefully(int fd, int argc, char *argv[])
{
	if (argc != 2)
		return RESULT_SHOWUSAGE;
	quit_handler(0, 1 /* nicely */, 1 /* safely */, 0 /* no restart */);
	return RESULT_SUCCESS;
}

Mark Spencer's avatar
Mark Spencer committed
static int handle_shutdown_when_convenient(int fd, int argc, char *argv[])
{
	if (argc != 3)
		return RESULT_SHOWUSAGE;
	quit_handler(0, 2 /* really nicely */, 1 /* safely */, 0 /* don't restart */);
	return RESULT_SUCCESS;
}

Mark Spencer's avatar
Mark Spencer committed
static int handle_restart_now(int fd, int argc, char *argv[])
{
	if (argc != 2)
		return RESULT_SHOWUSAGE;
	quit_handler(0, 0 /* not nicely */, 1 /* safely */, 1 /* restart */);
	return RESULT_SUCCESS;
}

static int handle_restart_gracefully(int fd, int argc, char *argv[])
{
	if (argc != 2)
		return RESULT_SHOWUSAGE;
	quit_handler(0, 1 /* nicely */, 1 /* safely */, 1 /* restart */);
	return RESULT_SUCCESS;
}

static int handle_restart_when_convenient(int fd, int argc, char *argv[])
{
	if (argc != 3)
		return RESULT_SHOWUSAGE;
	quit_handler(0, 2 /* really nicely */, 1 /* safely */, 1 /* restart */);
	return RESULT_SUCCESS;
}

static int handle_abort_halt(int fd, int argc, char *argv[])
{
	if (argc != 2)
		return RESULT_SHOWUSAGE;
	ast_cancel_shutdown();
	shuttingdown = 0;
Mark Spencer's avatar
Mark Spencer committed
	return RESULT_SUCCESS;
}

Mark Spencer's avatar
Mark Spencer committed
static int handle_bang(int fd, int argc, char *argv[])
{
	return RESULT_SUCCESS;
}

Mark Spencer's avatar
Mark Spencer committed
#define ASTERISK_PROMPT "*CLI> "

Mark Spencer's avatar
Mark Spencer committed
#define ASTERISK_PROMPT2 "%s*CLI> "

Mark Spencer's avatar
Mark Spencer committed
static struct ast_cli_entry aborthalt = { { "abort", "halt", NULL }, handle_abort_halt, "Cancel a running halt", abort_halt_help };

Mark Spencer's avatar
Mark Spencer committed
static struct ast_cli_entry quit = 	{ { "quit", NULL }, no_more_quit, "Exit Asterisk", quit_help };
static struct ast_cli_entry astexit = 	{ { "exit", NULL }, no_more_quit, "Exit Asterisk", quit_help };
Mark Spencer's avatar
Mark Spencer committed
static struct ast_cli_entry astshutdownnow = 	{ { "stop", "now", NULL }, handle_shutdown_now, "Shut down Asterisk immediately", shutdown_now_help };
Mark Spencer's avatar
Mark Spencer committed
static struct ast_cli_entry astshutdowngracefully = 	{ { "stop", "gracefully", NULL }, handle_shutdown_gracefully, "Gracefully shut down Asterisk", shutdown_gracefully_help };
Mark Spencer's avatar
Mark Spencer committed
static struct ast_cli_entry astshutdownwhenconvenient = 	{ { "stop", "when","convenient", NULL }, handle_shutdown_when_convenient, "Shut down Asterisk at empty call volume", shutdown_when_convenient_help };
Mark Spencer's avatar
Mark Spencer committed
static struct ast_cli_entry astrestartnow = 	{ { "restart", "now", NULL }, handle_restart_now, "Restart Asterisk immediately", restart_now_help };
static struct ast_cli_entry astrestartgracefully = 	{ { "restart", "gracefully", NULL }, handle_restart_gracefully, "Restart Asterisk gracefully", restart_gracefully_help };
static struct ast_cli_entry astrestartwhenconvenient= 	{ { "restart", "when", "convenient", NULL }, handle_restart_when_convenient, "Restart Asterisk at empty call volume", restart_when_convenient_help };
Mark Spencer's avatar
Mark Spencer committed
static struct ast_cli_entry astbang = { { "!", NULL }, handle_bang, "Execute a shell command", bang_help };
Mark Spencer's avatar
Mark Spencer committed
static int ast_el_read_char(EditLine *el, char *cp)
{
        int num_read=0;
	int lastpos=0;
	fd_set rfds;
	int res;
	int max;
	char buf[512];

	for (;;) {
		FD_ZERO(&rfds);
		FD_SET(ast_consock, &rfds);
		max = ast_consock;
Mark Spencer's avatar
Mark Spencer committed
		if (!option_exec) {
			FD_SET(STDIN_FILENO, &rfds);
			if (STDIN_FILENO > max)
				max = STDIN_FILENO;
		}
Mark Spencer's avatar
Mark Spencer committed
		res = ast_select(max+1, &rfds, NULL, NULL, NULL);
Mark Spencer's avatar
Mark Spencer committed
		if (res < 0) {
			if (errno == EINTR)
				continue;
			ast_log(LOG_ERROR, "select failed: %s\n", strerror(errno));
			break;
		}

		if (FD_ISSET(STDIN_FILENO, &rfds)) {
			num_read = read(STDIN_FILENO, cp, 1);
			if (num_read < 1) {
				break;
			} else 
				return (num_read);
		}
		if (FD_ISSET(ast_consock, &rfds)) {
			res = read(ast_consock, buf, sizeof(buf) - 1);
			/* if the remote side disappears exit */
			if (res < 1) {
				fprintf(stderr, "\nDisconnected from Asterisk server\n");
				quit_handler(0, 0, 0, 0);
			}

			buf[res] = '\0';

Mark Spencer's avatar
Mark Spencer committed
			if (!option_exec && !lastpos)
Mark Spencer's avatar
Mark Spencer committed
				write(STDOUT_FILENO, "\r", 1);
			write(STDOUT_FILENO, buf, res);
Mark Spencer's avatar
Mark Spencer committed
			if ((buf[res-1] == '\n') || (buf[res-2] == '\n')) {
				*cp = CC_REFRESH;
				return(1);
Mark Spencer's avatar
Mark Spencer committed
			} else {
				lastpos = 1;
			}
		}
	}

	*cp = '\0';
	return (0);
}

static char *cli_prompt(EditLine *el)
Mark Spencer's avatar
Mark Spencer committed
{
	static char prompt[80];
Mark Spencer's avatar
Mark Spencer committed

	if (remotehostname)
		snprintf(prompt, sizeof(prompt), ASTERISK_PROMPT2, remotehostname);
	else
		snprintf(prompt, sizeof(prompt), ASTERISK_PROMPT);

	return(prompt);	
Mark Spencer's avatar
Mark Spencer committed
}

static char **ast_el_strtoarr(char *buf)
{
	char **match_list = NULL, *retstr;
        size_t match_list_len;
	int matches = 0;

        match_list_len = 1;
	while ( (retstr = strsep(&buf, " ")) != NULL) {

                if (matches + 1 >= match_list_len) {
                        match_list_len <<= 1;
                        match_list = realloc(match_list, match_list_len * sizeof(char *));
		}

		match_list[matches++] = retstr;
	}

        if (!match_list)
                return (char **) NULL;

	if (matches>= match_list_len)
		match_list = realloc(match_list, (match_list_len + 1) * sizeof(char *));

	match_list[matches] = (char *) NULL;

	return match_list;
}

static int ast_el_sort_compare(const void *i1, const void *i2)
{
	char *s1, *s2;

	s1 = ((char **)i1)[0];
	s2 = ((char **)i2)[0];

	return strcasecmp(s1, s2);
}

static int ast_cli_display_match_list(char **matches, int len, int max)
{
	int i, idx, limit, count;
	int screenwidth = 0;
	int numoutput = 0, numoutputline = 0;

	screenwidth = ast_get_termcols(STDOUT_FILENO);

	/* find out how many entries can be put on one line, with two spaces between strings */
	limit = screenwidth / (max + 2);
	if (limit == 0)
		limit = 1;

	/* how many lines of output */
	count = len / limit;
	if (count * limit < len)
		count++;

	idx = 1;

	qsort(&matches[0], (size_t)(len + 1), sizeof(char *), ast_el_sort_compare);

	for (; count > 0; count--) {
		numoutputline = 0;
		for (i=0; i < limit && matches[idx]; i++, idx++) {

			/* Don't print dupes */
			if ( (matches[idx+1] != NULL && strcmp(matches[idx], matches[idx+1]) == 0 ) ) {
				i--;
				continue;
			}

			numoutput++;  numoutputline++;
			fprintf(stdout, "%-*s  ", max, matches[idx]);
		}
		if (numoutputline > 0)
			fprintf(stdout, "\n");
	}

	return numoutput;
Mark Spencer's avatar
Mark Spencer committed

static char *cli_complete(EditLine *el, int ch)
Mark Spencer's avatar
Mark Spencer committed
{
Mark Spencer's avatar
Mark Spencer committed
	int len=0;
	char *ptr;
	int nummatches = 0;
	char **matches;
	int retval = CC_ERROR;
Mark Spencer's avatar
Mark Spencer committed
	char buf[1024];
	int res;
Mark Spencer's avatar
Mark Spencer committed

	LineInfo *lf = (LineInfo *)el_line(el);

	*(char *)lf->cursor = '\0';
	ptr = (char *)lf->cursor;
Mark Spencer's avatar
Mark Spencer committed
	if (ptr) {
		while (ptr > lf->buffer) {
			if (isspace(*ptr)) {
				ptr++;
				break;
			}
			ptr--;
		}
	}

	len = lf->cursor - ptr;

	if (option_remote) {
		snprintf(buf, sizeof(buf),"_COMMAND NUMMATCHES \"%s\" \"%s\"", lf->buffer, ptr); 
		fdprint(ast_consock, buf);
		res = read(ast_consock, buf, sizeof(buf));
		buf[res] = '\0';
		nummatches = atoi(buf);

		if (nummatches > 0) {
			snprintf(buf, sizeof(buf),"_COMMAND MATCHESARRAY \"%s\" \"%s\"", lf->buffer, ptr); 
			fdprint(ast_consock, buf);
			res = read(ast_consock, buf, sizeof(buf));
			buf[res] = '\0';

			matches = ast_el_strtoarr(buf);
		} else
			matches = (char **) NULL;


	}  else {

		nummatches = ast_cli_generatornummatches((char *)lf->buffer,ptr);
		matches = ast_cli_completion_matches((char *)lf->buffer,ptr);
	}

	if (matches) {
		int i;
		int matches_num, maxlen, match_len;

		if (matches[0][0] != '\0') {
			el_deletestr(el, (int) len);
			el_insertstr(el, matches[0]);
			retval = CC_REFRESH;
		}

		if (nummatches == 1) {
			/* Found an exact match */
			el_insertstr(el, " ");
Mark Spencer's avatar
Mark Spencer committed
			retval = CC_REFRESH;
		} else {
			/* Must be more than one match */
			for (i=1, maxlen=0; matches[i]; i++) {
				match_len = strlen(matches[i]);