Skip to content
Snippets Groups Projects
asterisk.c 38.4 KiB
Newer Older
  • Learn to ignore specific revisions
  • 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];
    
    
    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);
    }
    
    
    static void network_verboser(const char *s, int pos, int replace, int complete)
    
    Mark Spencer's avatar
    Mark Spencer committed
    {
    	int x;
    	for (x=0;x<AST_MAX_CONNECTS; x++) {
    		if (consoles[x].fd > -1) 
    			fdprint(consoles[x].p[1], s);
    	}
    }
    
    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];
    
    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) {
    			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) {
    			ast_log(LOG_WARNING, "Accept retured %d: %s\n", s, strerror(errno));
    		} 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.  */
    
    Mark Spencer's avatar
    Mark Spencer committed
    	if (option_debug) 
    
    Mark Spencer's avatar
    Mark Spencer committed
    		ast_log(LOG_DEBUG, "Urgent handler\n");
    
    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) 
    		ast_verbose(VERBOSE_PREFIX_2 "Received HUP signal -- Reloading configs\n");
    	ast_module_reload();
    }
    
    
    static void child_handler(int sig)
    
    Mark Spencer's avatar
    Mark Spencer committed
    {
    
    	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)	
    		ast_log(LOG_DEBUG, "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;
    }
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    static char *_argv[256];
    
    static int shuttingdown = 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");
    		execvp(_argv[0], _argv);
    
    Mark Spencer's avatar
    Mark Spencer committed
    }
    
    static void __quit_handler(int num)
    {
    	quit_handler(num, 0, 1, 0);
    
    static pthread_t consolethread = (pthread_t) -1;
    
    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 (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 imediately", shutdown_now_help };
    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);
    			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]);
    				if (match_len > maxlen)
    					maxlen = match_len;
    			}
    			matches_num = i - 1;
    			if (matches_num >1) {
    				fprintf(stdout, "\n");
    				ast_cli_display_match_list(matches, nummatches, maxlen);
    				retval = CC_REDISPLAY;
    			} else { 
    
    				el_insertstr(el," ");
    
    Mark Spencer's avatar
    Mark Spencer committed
    				retval = CC_REFRESH;
    			}
    		}
    
    	free(matches);
    
    Mark Spencer's avatar
    Mark Spencer committed
    	}
    
    	return (char *)retval;
    }
    
    static int ast_el_initialize(void)
    {
    	HistEvent ev;
    
    	if (el != NULL)
    		el_end(el);
    	if (el_hist != NULL)
    		history_end(el_hist);
    
    	el = el_init("asterisk", stdin, stdout, stderr);
    	el_set(el, EL_PROMPT, cli_prompt);
    
    	el_set(el, EL_EDITMODE, 1);		
    	el_set(el, EL_EDITOR, "emacs");