Skip to content
Snippets Groups Projects
asterisk.c 43.7 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 <asterisk/lock.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 <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 */
    
    Mark Spencer's avatar
    Mark Spencer committed
    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 = AST_PTHREADT_NULL;
    
    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);
    }
    
    
    int ast_safe_system(const char *s)
    {
    	/* XXX This function needs some optimization work XXX */
    	pid_t pid;
    	int x;
    	int res;
    	struct rusage rusage;
    	int status;
    	pid = fork();
    	if (pid == 0) {
    		/* Close file descriptors and launch system command */
    		for (x=STDERR_FILENO + 1; x<4096;x++) {
    			close(x);
    		}
    
    Mark Spencer's avatar
    Mark Spencer committed
    		res = execl("/bin/sh", "/bin/sh", "-c", s, NULL);
    		exit(1);
    
    	} else if (pid > 0) {
    		for(;;) {
    			res = wait4(pid, &status, 0, &rusage);
    			if (res > -1) {
    				if (WIFEXITED(status))
    					res = WEXITSTATUS(status);
    				else
    					res = -1;
    			} else {
    				if (errno != EINTR) 
    					break;
    			}
    		}
    	} else {
    		ast_log(LOG_WARNING, "Fork failed: %s\n", strerror(errno));
    		res = -1;
    	}
    	return res;
    }
    
    
    /*
     * 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;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	
    	if (gethostname(hostname, sizeof(hostname)))
    
    Mark Spencer's avatar
    Mark Spencer committed
    		strncpy(hostname, "<Unknown>", sizeof(hostname)-1);
    
    	snprintf(tmp, sizeof(tmp), "%s/%d/%s\n", hostname, ast_mainpid, ASTERISK_VERSION);
    
    Mark Spencer's avatar
    Mark Spencer committed
    	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));
    
    Mark Spencer's avatar
    Mark Spencer committed
    			continue;
    		}
    
    		if (FD_ISSET(con->fd, &rfds)) {
    
    Mark Spencer's avatar
    Mark Spencer committed
    			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)) {
    
    Mark Spencer's avatar
    Mark Spencer committed
    			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;
    
    	fd_set fds;
    
    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 (errno != EINTR)
    
    				ast_log(LOG_WARNING, "Select returned error: %s\n", strerror(errno));
    
    Mark Spencer's avatar
    Mark Spencer committed
    		len = sizeof(sun);
    		s = accept(ast_socket, (struct sockaddr *)&sun, &len);
    		if (s < 0) {
    
    			if (errno != EINTR)
    
    James Golovich's avatar
    James Golovich committed
    				ast_log(LOG_WARNING, "Accept returned %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
    
    Mark Spencer's avatar
    Mark Spencer committed
    	   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 != AST_PTHREADT_NULL) {
    
    			pthread_kill(consolethread, SIGHUP);
    
    James Golovich's avatar
    James Golovich committed
    			/* Give the signal handler some time to complete */
    			sleep(2);
    		} else
    
    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 != AST_PTHREADT_NULL)
    
    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])
    
    Mark Spencer's avatar
    Mark Spencer committed
    			else
    
    				ast_safe_system(getenv("SHELL") ? getenv("SHELL") : "/bin/sh");
    
    Mark Spencer's avatar
    Mark Spencer committed
    		} 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])
    
    Mark Spencer's avatar
    Mark Spencer committed
    			else
    
    				ast_safe_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;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	int lastpos=0;
    
    	fd_set rfds;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	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));
    
    		if (FD_ISSET(STDIN_FILENO, &rfds)) {
    
    Mark Spencer's avatar
    Mark Spencer committed
    			num_read = read(STDIN_FILENO, cp, 1);
    			if (num_read < 1) {
    				break;
    			} else 
    				return (num_read);
    		}
    
    		if (FD_ISSET(ast_consock, &rfds)) {
    
    Mark Spencer's avatar
    Mark Spencer committed
    			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[200];
    	char *pfmt;
    
    	if ((pfmt = getenv("ASTERISK_PROMPT"))) {
    		char *t = pfmt, *p = prompt;
    		memset(prompt, 0, sizeof(prompt));
    		while (*t != '\0' && *p < sizeof(prompt)) {
    			if (*t == '%') {
    				t++;
    				switch (*t) {
    					char hostname[256];
    					struct timeval tv;
    					struct tm tm;
    					FILE *LOADAVG;
    					case 'd': /* date */
    						memset(&tm, 0, sizeof(struct tm));
    						gettimeofday(&tv, NULL);
    						if (localtime_r(&(tv.tv_sec), &tm)) {
    							strftime(p, sizeof(prompt) - strlen(prompt), "%Y-%m-%d", &tm);
    						}
    						break;
    					case 'h': /* hostname */
    						if (!gethostname(hostname, sizeof(hostname) - 1)) {
    							strncat(p, hostname, sizeof(prompt) - strlen(prompt));
    						} else {
    							strncat(p, "localhost", sizeof(prompt) - strlen(prompt));
    						}
    						break;
    #ifdef linux
    					case 'l': /* load avg */
    						t++;
    						if ((LOADAVG = fopen("/proc/loadavg", "r"))) {
    							float avg1, avg2, avg3;
    							int actproc, totproc, npid, which;
    							fscanf(LOADAVG, "%f %f %f %d/%d %d",
    								&avg1, &avg2, &avg3, &actproc, &totproc, &npid);
    							if (sscanf(t, "%d", &which) == 1) {
    								switch (which) {
    									case 1:
    										snprintf(p, sizeof(prompt) - strlen(prompt), "%.2f", avg1);
    										break;
    									case 2:
    										snprintf(p, sizeof(prompt) - strlen(prompt), "%.2f", avg2);
    										break;
    									case 3:
    										snprintf(p, sizeof(prompt) - strlen(prompt), "%.2f", avg3);
    										break;
    									case 4:
    										snprintf(p, sizeof(prompt) - strlen(prompt), "%d/%d", actproc, totproc);
    										break;
    									case 5:
    										snprintf(p, sizeof(prompt) - strlen(prompt), "%d", npid);
    										break;
    								}
    							}
    						}
    						break;
    #endif
    					case 't': /* time */
    						memset(&tm, 0, sizeof(struct tm));
    						gettimeofday(&tv, NULL);
    						if (localtime_r(&(tv.tv_sec), &tm)) {
    							strftime(p, sizeof(prompt) - strlen(prompt), "%H:%M:%S", &tm);
    						}
    						break;
    					case '#': /* process console or remote? */
    						if (! option_remote) {
    							strncat(p, "#", sizeof(prompt) - strlen(prompt));
    						} else {
    							strncat(p, ">", sizeof(prompt) - strlen(prompt));
    						}
    						break;
    					case '%': /* literal % */
    						strncat(p, "%", sizeof(prompt) - strlen(prompt));
    						break;
    					case '\0': /* % is last character - prevent bug */
    						t--;
    						break;
    				}
    				while (*p != '\0') {
    					p++;
    				}
    				t++;
    			} else {
    				*p = *t;
    				p++;
    				t++;
    			}
    		}
    	} else if (remotehostname)
    
    Mark Spencer's avatar
    Mark Spencer committed
    		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 (!strcmp(retstr, AST_CLI_COMPLETE_EOF))
    			break;
    
    Mark Spencer's avatar
    Mark Spencer committed
                    if (matches + 1 >= match_list_len) {
                            match_list_len <<= 1;
                            match_list = realloc(match_list, match_list_len * sizeof(char *));
    		}
    
    
    		match_list[matches++] = strdup(retstr);
    
    Mark Spencer's avatar
    Mark Spencer committed
    	}
    
            if (!match_list)
                    return (char **) NULL;