Merge slepp's sub-second expiration timer although hopefully it will never have to...
[asterisk/asterisk.git] / asterisk.c
index 5ee2a4f..d1b0dad 100755 (executable)
@@ -13,6 +13,7 @@
 
 #include <unistd.h>
 #include <stdlib.h>
+#include <sys/poll.h>
 #include <asterisk/logger.h>
 #include <asterisk/options.h>
 #include <asterisk/cli.h>
 #include <asterisk/term.h>
 #include <asterisk/manager.h>
 #include <asterisk/pbx.h>
+#include <asterisk/enum.h>
+#include <asterisk/rtp.h>
+#include <asterisk/app.h>
+#include <asterisk/lock.h>
 #include <sys/resource.h>
 #include <fcntl.h>
 #include <stdio.h>
 #include <signal.h>
 #include <sched.h>
 #include <asterisk/io.h>
-#include <pthread.h>
+#include <asterisk/lock.h>
 #include <sys/socket.h>
 #include <sys/un.h>
-#include <sys/select.h>
+#include <sys/wait.h>
 #include <string.h>
 #include <errno.h>
 #include <ctype.h>
 #include "asterisk.h"
 #include <asterisk/config.h>
 
+#if  defined(__FreeBSD__)
+#include <netdb.h>
+#endif
+
 #define AST_MAX_CONNECTS 128
 #define NUM_MSGS 64
 
@@ -62,13 +71,22 @@ int fully_booted = 0;
 
 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;
+int ast_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;
+
+time_t ast_startuptime;
+time_t ast_lastreloadtime;
+
 static History *el_hist = NULL;
 static EditLine *el = NULL;
 static char *remotehostname;
@@ -94,18 +112,117 @@ 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));
+       ast_mutex_lock(&atexitslock);
+       if (ae) {
+               memset(ae, 0, sizeof(struct ast_atexit));
+               ae->next = atexits;
+               ae->func = func;
+               atexits = ae;
+               res = 0;
+       }
+       ast_mutex_unlock(&atexitslock);
+       return res;
+}
+
+void ast_unregister_atexit(void (*func)(void))
+{
+       struct ast_atexit *ae, *prev = NULL;
+       ast_mutex_lock(&atexitslock);
+       ae = atexits;
+       while(ae) {
+               if (ae->func == func) {
+                       if (prev)
+                               prev->next = ae->next;
+                       else
+                               atexits = ae->next;
+                       break;
+               }
+               prev = ae;
+               ae = ae->next;
+       }
+       ast_mutex_unlock(&atexitslock);
+}
+
 static int fdprint(int fd, const char *s)
 {
        return write(fd, s, strlen(s) + 1);
 }
 
-static void network_verboser(const char *s, int pos, int replace, int complete)
+int ast_safe_system(const char *s)
 {
+       /* XXX This function needs some optimization work XXX */
+       pid_t pid;
        int x;
-       for (x=0;x<AST_MAX_CONNECTS; x++) {
-               if (consoles[x].fd > -1) 
-                       fdprint(consoles[x].p[1], s);
+       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);
+               }
+               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)
+     /* ARGUSED */
+{
+    ast_network_puts(s);
 }
 
 static pthread_t lthread;
@@ -116,26 +233,24 @@ static void *netconsole(void *vconsole)
        char hostname[256];
        char tmp[512];
        int res;
-       int max;
-       fd_set rfds;
+       struct pollfd fds[2];
        
        if (gethostname(hostname, sizeof(hostname)))
                strncpy(hostname, "<Unknown>", sizeof(hostname)-1);
-       snprintf(tmp, sizeof(tmp), "%s/%d/%s\n", hostname, mainpid, ASTERISK_VERSION);
+       snprintf(tmp, sizeof(tmp), "%s/%d/%s\n", hostname, ast_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);
+               fds[0].fd = con->fd;
+               fds[0].events = POLLIN;
+               fds[1].fd = con->p[0];
+               fds[1].events = POLLIN;
+
+               res = poll(fds, 2, -1);
                if (res < 0) {
-                       ast_log(LOG_WARNING, "select returned < 0: %s\n", strerror(errno));
+                       ast_log(LOG_WARNING, "poll returned < 0: %s\n", strerror(errno));
                        continue;
                }
-               if (FD_ISSET(con->fd, &rfds)) {
+               if (fds[0].revents) {
                        res = read(con->fd, tmp, sizeof(tmp));
                        if (res < 1) {
                                break;
@@ -143,7 +258,7 @@ static void *netconsole(void *vconsole)
                        tmp[res] = 0;
                        ast_cli_command(con->fd, tmp);
                }
-               if (FD_ISSET(con->p[0], &rfds)) {
+               if (fds[1].revents) {
                        res = read(con->p[0], tmp, sizeof(tmp));
                        if (res < 1) {
                                ast_log(LOG_ERROR, "read returned %d\n", res);
@@ -171,16 +286,26 @@ static void *listener(void *unused)
        int len;
        int x;
        int flags;
+       struct pollfd fds[1];
        pthread_attr_t attr;
        pthread_attr_init(&attr);
        pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
        for(;;) {
                if (ast_socket < 0)
                        return NULL;
+               fds[0].fd = ast_socket;
+               fds[0].events= POLLIN;
+               s = poll(fds, 1, -1);
+               if (s < 0) {
+                       if (errno != EINTR)
+                               ast_log(LOG_WARNING, "poll returned error: %s\n", strerror(errno));
+                       continue;
+               }
                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));
+                       if (errno != EINTR)
+                               ast_log(LOG_WARNING, "Accept returned %d: %s\n", s, strerror(errno));
                } else {
                        for (x=0;x<AST_MAX_CONNECTS;x++) {
                                if (consoles[x].fd < 0) {
@@ -274,10 +399,11 @@ static int ast_tryconnect(void)
 
 static void urg_handler(int num)
 {
-       /* Called by soft_hangup to interrupt the select, read, or other
+       /* Called by soft_hangup to interrupt the poll, read, or other
           system call.  We don't actually need to do anything though.  */
+       /* Cannot EVER ast_log from within a signal handler */
        if (option_debug) 
-               ast_log(LOG_DEBUG, "Urgent handler\n");
+               printf("Urgent handler\n");
        signal(num, urg_handler);
        return;
 }
@@ -285,15 +411,27 @@ static void urg_handler(int num)
 static void hup_handler(int num)
 {
        if (option_verbose > 1) 
-               ast_verbose(VERBOSE_PREFIX_2 "Received HUP signal -- Reloading configs\n");
+               printf("Received HUP signal -- Reloading configs\n");
+       if (restartnow)
+               execvp(_argv[0], _argv);
+       /* XXX This could deadlock XXX */
        ast_module_reload();
 }
 
-
-static void pipe_handler(int num)
+static void child_handler(int sig)
 {
-       /* Ignore sigpipe */
+       /* 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");
 }
+
 static void set_title(char *text)
 {
        /* Set an X-term or screen title */
@@ -347,9 +485,18 @@ static int set_priority(int pri)
        return 0;
 }
 
-static char *_argv[256];
-
-static int shuttingdown = 0;
+static void ast_run_atexits(void)
+{
+       struct ast_atexit *ae;
+       ast_mutex_lock(&atexitslock);
+       ae = atexits;
+       while(ae) {
+               if (ae->func) 
+                       ae->func();
+               ae = ae->next;
+       }
+       ast_mutex_unlock(&atexitslock);
+}
 
 static void quit_handler(int num, int nice, int safeshutdown, int restart)
 {
@@ -406,6 +553,9 @@ static void quit_handler(int num, int nice, int safeshutdown, int restart)
                if (el_hist != NULL)
                        history_end(el_hist);
        }
+       if (option_verbose)
+               ast_verbose("Executing last minute cleanups\n");
+       ast_run_atexits();
        /* Called on exit */
        if (option_verbose && option_console)
                ast_verbose("Asterisk %s ending (%d).\n", ast_active_channels() ? "uncleanly" : "cleanly", num);
@@ -420,7 +570,7 @@ static void quit_handler(int num, int nice, int safeshutdown, int restart)
                close(ast_consock);
        if (ast_socket > -1)
                unlink((char *)ast_config_AST_SOCKET);
-       unlink((char *)ast_config_AST_PID);
+       if (!option_remote) unlink((char *)ast_config_AST_PID);
        printf(term_quit());
        if (restart) {
                if (option_verbose || option_console)
@@ -431,9 +581,18 @@ static void quit_handler(int num, int nice, int safeshutdown, int restart)
                }
                if (option_verbose || option_console)
                        ast_verbose("Restarting Asterisk NOW...\n");
-               execvp(_argv[0], _argv);
-       } else
-               exit(0);
+               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);
+                       /* Give the signal handler some time to complete */
+                       sleep(2);
+               } else
+                       execvp(_argv[0], _argv);
+       
+       }
+       exit(0);
 }
 
 static void __quit_handler(int num)
@@ -441,35 +600,38 @@ static void __quit_handler(int num)
        quit_handler(num, 0, 1, 0);
 }
 
-static pthread_t consolethread = -1;
-
-static int fix_header(char *outbuf, int maxout, char **s, char *cmp)
+static const char *fix_header(char *outbuf, int maxout, const char *s, char *cmp)
 {
-       if (!strncmp(*s, cmp, strlen(cmp))) {
-               *s += strlen(cmp);
+       const char *c;
+       if (!strncmp(s, cmp, strlen(cmp))) {
+               c = s + strlen(cmp);
                term_color(outbuf, cmp, COLOR_GRAY, 0, maxout);
-               return 1;
+               return c;
        }
-       return 0;
+       return NULL;
 }
 
 static void console_verboser(const char *s, int pos, int replace, int complete)
 {
        char tmp[80];
+       const char *c=NULL;
        /* Return to the beginning of the line */
        if (!pos) {
                fprintf(stdout, "\r");
-               if (fix_header(tmp, sizeof(tmp), &s, VERBOSE_PREFIX_4) ||
-                       fix_header(tmp, sizeof(tmp), &s, VERBOSE_PREFIX_3) ||
-                       fix_header(tmp, sizeof(tmp), &s, VERBOSE_PREFIX_2) ||
-                       fix_header(tmp, sizeof(tmp), &s, VERBOSE_PREFIX_1))
+               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)))
                        fputs(tmp, stdout);
        }
-       fputs(s + pos,stdout);
+       if (c)
+               fputs(c + pos,stdout);
+       else
+               fputs(s + pos,stdout);
        fflush(stdout);
        if (complete)
-       /* Wake up a select()ing console */
-               if (consolethread > -1)
+       /* Wake up a poll()ing console */
+               if (option_console && consolethread != AST_PTHREADT_NULL)
                        pthread_kill(consolethread, SIGURG);
 }
 
@@ -482,15 +644,14 @@ static void consolehandler(char *s)
                ast_el_add_history(s);
        /* Give the console access to the shell */
        if (s) {
+               /* The real handler for bang */
                if (s[0] == '!') {
                        if (s[1])
-                               system(s+1);
+                               ast_safe_system(s+1);
                        else
-                               system(getenv("SHELL") ? getenv("SHELL") : "/bin/sh");
+                               ast_safe_system(getenv("SHELL") ? getenv("SHELL") : "/bin/sh");
                } else 
                ast_cli_command(STDOUT_FILENO, s);
-               if (!strcasecmp(s, "help"))
-                       fprintf(stdout, "          !<command>   Executes a given shell command\n");
        } else
                fprintf(stdout, "\nUse \"quit\" to exit\n");
 }
@@ -503,22 +664,16 @@ static int remoteconsolehandler(char *s)
                ast_el_add_history(s);
        /* Give the console access to the shell */
        if (s) {
+               /* The real handler for bang */
                if (s[0] == '!') {
                        if (s[1])
-                               system(s+1);
+                               ast_safe_system(s+1);
                        else
-                               system(getenv("SHELL") ? getenv("SHELL") : "/bin/sh");
-                       ret = 1;
-               }
-               if (!strcasecmp(s, "help")) {
-                       fprintf(stdout, "          !<command>   Executes a given shell command\n");
-                       ret = 0;
-               }
-               if (!strcasecmp(s, "quit")) {
-                       quit_handler(0, 0, 0, 0);
+                               ast_safe_system(getenv("SHELL") ? getenv("SHELL") : "/bin/sh");
                        ret = 1;
                }
-               if (!strcasecmp(s, "exit")) {
+               if ((strncasecmp(s, "quit", 4) == 0 || strncasecmp(s, "exit", 4) == 0) &&
+                   (s[4] == '\0' || isspace(s[4]))) {
                        quit_handler(0, 0, 0, 0);
                        ret = 1;
                }
@@ -564,6 +719,10 @@ 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";
 
+static char bang_help[] =
+"Usage: !<command>\n"
+"       Executes a given shell command\n";
+
 #if 0
 static int handle_quit(int fd, int argc, char *argv[])
 {
@@ -640,6 +799,11 @@ static int handle_abort_halt(int fd, int argc, char *argv[])
        return RESULT_SUCCESS;
 }
 
+static int handle_bang(int fd, int argc, char *argv[])
+{
+       return RESULT_SUCCESS;
+}
+
 #define ASTERISK_PROMPT "*CLI> "
 
 #define ASTERISK_PROMPT2 "%s*CLI> "
@@ -649,45 +813,48 @@ static struct ast_cli_entry aborthalt = { { "abort", "halt", NULL }, handle_abor
 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 };
 
-static struct ast_cli_entry astshutdownnow =   { { "stop", "now", NULL }, handle_shutdown_now, "Shut down Asterisk imediately", shutdown_now_help };
+static struct ast_cli_entry astshutdownnow =   { { "stop", "now", NULL }, handle_shutdown_now, "Shut down Asterisk immediately", shutdown_now_help };
 static struct ast_cli_entry astshutdowngracefully =    { { "stop", "gracefully", NULL }, handle_shutdown_gracefully, "Gracefully shut down Asterisk", shutdown_gracefully_help };
 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 };
 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 };
+static struct ast_cli_entry astbang = { { "!", NULL }, handle_bang, "Execute a shell command", bang_help };
 
 static int ast_el_read_char(EditLine *el, char *cp)
 {
-        int num_read=0;
+    int num_read=0;
        int lastpos=0;
-       fd_set rfds;
+       struct pollfd fds[2];
        int res;
        int max;
        char buf[512];
 
        for (;;) {
-               FD_ZERO(&rfds);
-               FD_SET(ast_consock, &rfds);
-               FD_SET(STDIN_FILENO, &rfds);
-               max = ast_consock;
-               if (STDIN_FILENO > max)
-                       max = STDIN_FILENO;
-               res = ast_select(max+1, &rfds, NULL, NULL, NULL);
+               max = 1;
+               fds[0].fd = ast_consock;
+               fds[0].events = POLLIN;
+               if (!option_exec) {
+                       fds[1].fd = STDIN_FILENO;
+                       fds[1].events = POLLIN;
+                       max++;
+               }
+               res = poll(fds, max, -1);
                if (res < 0) {
                        if (errno == EINTR)
                                continue;
-                       ast_log(LOG_ERROR, "select failed: %s\n", strerror(errno));
+                       ast_log(LOG_ERROR, "poll failed: %s\n", strerror(errno));
                        break;
                }
 
-               if (FD_ISSET(STDIN_FILENO, &rfds)) {
+               if (!option_exec && fds[1].revents) {
                        num_read = read(STDIN_FILENO, cp, 1);
                        if (num_read < 1) {
                                break;
                        } else 
                                return (num_read);
                }
-               if (FD_ISSET(ast_consock, &rfds)) {
+               if (fds[0].revents) {
                        res = read(ast_consock, buf, sizeof(buf) - 1);
                        /* if the remote side disappears exit */
                        if (res < 1) {
@@ -697,7 +864,7 @@ static int ast_el_read_char(EditLine *el, char *cp)
 
                        buf[res] = '\0';
 
-                       if (!lastpos)
+                       if (!option_exec && !lastpos)
                                write(STDOUT_FILENO, "\r", 1);
                        write(STDOUT_FILENO, buf, res);
                        if ((buf[res-1] == '\n') || (buf[res-2] == '\n')) {
@@ -715,9 +882,140 @@ static int ast_el_read_char(EditLine *el, char *cp)
 
 static char *cli_prompt(EditLine *el)
 {
-       static char prompt[80];
-
-       if (remotehostname)
+       static char prompt[200];
+       char *pfmt;
+       int color_used=0;
+       char term_code[20];
+
+       if ((pfmt = getenv("ASTERISK_PROMPT"))) {
+               char *t = pfmt, *p = prompt;
+               memset(prompt, 0, sizeof(prompt));
+               while (*t != '\0' && *p < sizeof(prompt)) {
+                       if (*t == '%') {
+                               char hostname[256];
+                               int i;
+                               struct timeval tv;
+                               struct tm tm;
+                               FILE *LOADAVG;
+                               int fgcolor = COLOR_WHITE, bgcolor = COLOR_BLACK;
+
+                               t++;
+                               switch (*t) {
+                                       case 'C': /* color */
+                                               t++;
+                                               if (sscanf(t, "%d;%d%n", &fgcolor, &bgcolor, &i) == 2) {
+                                                       strncat(p, term_color_code(term_code, fgcolor, bgcolor, sizeof(term_code)),sizeof(prompt) - strlen(prompt));
+                                                       t += i - 1;
+                                               } else if (sscanf(t, "%d%n", &fgcolor, &i) == 1) {
+                                                       strncat(p, term_color_code(term_code, fgcolor, 0, sizeof(term_code)),sizeof(prompt) - strlen(prompt));
+                                                       t += i - 1;
+                                               }
+
+                                               /* If the color has been reset correctly, then there's no need to reset it later */
+                                               if ((fgcolor == COLOR_WHITE) && (bgcolor == COLOR_BLACK)) {
+                                                       color_used = 0;
+                                               } else {
+                                                       color_used = 1;
+                                               }
+                                               break;
+                                       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;
+                                       case 'H': /* short hostname */
+                                               if (!gethostname(hostname, sizeof(hostname) - 1)) {
+                                                       for (i=0;i<sizeof(hostname);i++) {
+                                                               if (hostname[i] == '.') {
+                                                                       hostname[i] = '\0';
+                                                                       break;
+                                                               }
+                                                       }
+                                                       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++;
+                       }
+               }
+               if (color_used) {
+                       /* Force colors back to normal at end */
+                       term_color_code(term_code, COLOR_WHITE, COLOR_BLACK, sizeof(term_code));
+                       if (strlen(term_code) > sizeof(prompt) - strlen(prompt)) {
+                               strncat(prompt + sizeof(prompt) - strlen(term_code) - 1, term_code, strlen(term_code));
+                       } else {
+                               strncat(p, term_code, sizeof(term_code));
+                       }
+               }
+       } else if (remotehostname)
                snprintf(prompt, sizeof(prompt), ASTERISK_PROMPT2, remotehostname);
        else
                snprintf(prompt, sizeof(prompt), ASTERISK_PROMPT);
@@ -734,12 +1032,14 @@ static char **ast_el_strtoarr(char *buf)
         match_list_len = 1;
        while ( (retstr = strsep(&buf, " ")) != NULL) {
 
+               if (!strcmp(retstr, AST_CLI_COMPLETE_EOF))
+                       break;
                 if (matches + 1 >= match_list_len) {
                         match_list_len <<= 1;
                         match_list = realloc(match_list, match_list_len * sizeof(char *));
                }
 
-               match_list[matches++] = retstr;
+               match_list[matches++] = strdup(retstr);
        }
 
         if (!match_list)
@@ -813,12 +1113,12 @@ static char *cli_complete(EditLine *el, int ch)
        int nummatches = 0;
        char **matches;
        int retval = CC_ERROR;
-       char buf[1024];
+       char buf[2048];
        int res;
 
        LineInfo *lf = (LineInfo *)el_line(el);
 
-       *lf->cursor = '\0';
+       *(char *)lf->cursor = '\0';
        ptr = (char *)lf->cursor;
        if (ptr) {
                while (ptr > lf->buffer) {
@@ -840,12 +1140,33 @@ static char *cli_complete(EditLine *el, int ch)
                nummatches = atoi(buf);
 
                if (nummatches > 0) {
+                       char *mbuf;
+                       int mlen = 0, maxmbuf = 2048;
+                       /* Start with a 2048 byte buffer */
+                       mbuf = malloc(maxmbuf);
+                       if (!mbuf)
+                               return (char *)(CC_ERROR);
                        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';
+                       res = 0;
+                       mbuf[0] = '\0';
+                       while (!strstr(mbuf, AST_CLI_COMPLETE_EOF) && res != -1) {
+                               if (mlen + 1024 > maxmbuf) {
+                                       /* Every step increment buffer 1024 bytes */
+                                       maxmbuf += 1024;
+                                       mbuf = realloc(mbuf, maxmbuf);
+                                       if (!mbuf)
+                                               return (char *)(CC_ERROR);
+                               }
+                               /* Only read 1024 bytes at a time */
+                               res = read(ast_consock, mbuf + mlen, 1024);
+                               if (res > 0)
+                                       mlen += res;
+                       }
+                       mbuf[mlen] = '\0';
 
-                       matches = ast_el_strtoarr(buf);
+                       matches = ast_el_strtoarr(mbuf);
+                       free(mbuf);
                } else
                        matches = (char **) NULL;
 
@@ -921,6 +1242,8 @@ static int ast_el_initialize(void)
        el_set(el, EL_BIND, "^I", "ed-complete", NULL);
        /* Bind ? to command completion */
        el_set(el, EL_BIND, "?", "ed-complete", NULL);
+       /* Bind ^D to redisplay */
+       el_set(el, EL_BIND, "^D", "ed-redisplay", NULL);
 
        return 0;
 }
@@ -931,7 +1254,8 @@ static int ast_el_add_history(char *buf)
 
        if (el_hist == NULL || el == NULL)
                ast_el_initialize();
-
+       if (strlen(buf) > 256)
+               return 0;
        return (history(el_hist, &ev, H_ENTER, buf));
 }
 
@@ -990,7 +1314,7 @@ static void ast_remotecontrol(char * data)
        stringp=buf;
        hostname = strsep(&stringp, "/");
        cpid = strsep(&stringp, "/");
-       version = strsep(&stringp, "/");
+       version = strsep(&stringp, "\n");
        if (!version)
                version = "<Version Unknown>";
        stringp=hostname;
@@ -1018,12 +1342,14 @@ static void ast_remotecontrol(char * data)
 #if 0
        ast_cli_register(&astshutdown);
 #endif 
+       if (option_exec && data) {  /* hack to print output then exit if asterisk -rx is used */
+               char tempchar;
+               ast_el_read_char(el, &tempchar);
+               return;
+       }
        for(;;) {
                ebuf = (char *)el_gets(el, &num);
 
-               if (data)       /* hack to print output then exit if asterisk -rx is used */
-                       ebuf = strdup("quit");
-
                if (ebuf && strlen(ebuf)) {
                        if (ebuf[strlen(ebuf)-1] == '\n')
                                ebuf[strlen(ebuf)-1] = '\0';
@@ -1114,7 +1440,7 @@ static void ast_readconfig(void) {
 
 int main(int argc, char *argv[])
 {
-       char c;
+       int c;
        char filename[80] = "";
        char hostname[256];
        char tmp[80];
@@ -1134,9 +1460,14 @@ int main(int argc, char *argv[])
                _argv[x] = argv[x];
        _argv[x] = NULL;
 
+       /* if the progname is rasterisk consider it a remote console */
+       if ( argv[0] && (strstr(argv[0], "rasterisk")) != NULL)  {
+               option_remote++;
+               option_nofork++;
+       }
        if (gethostname(hostname, sizeof(hostname)))
                strncpy(hostname, "<Unknown>", sizeof(hostname)-1);
-       mainpid = getpid();
+       ast_mainpid = getpid();
        ast_ulaw_init();
        ast_alaw_init();
        callerid_init();
@@ -1151,7 +1482,7 @@ int main(int argc, char *argv[])
        }
        */
        /* Check for options */
-       while((c=getopt(argc, argv, "hfdvqprgcinx:C:")) != EOF) {
+       while((c=getopt(argc, argv, "hfdvqprgcinx:C:")) != -1) {
                switch(c) {
                case 'd':
                        option_debug++;
@@ -1238,8 +1569,8 @@ int main(int argc, char *argv[])
                        }
                        printf(term_quit());
                        ast_register_verbose(console_verboser);
-                       ast_verbose( "Asterisk " ASTERISK_VERSION ", Copyright (C) 1999-2001 Linux Support Services, Inc.\n");
-                       ast_verbose( "Written by Mark Spencer <markster@linux-support.net>\n");
+                       ast_verbose( "Asterisk " ASTERISK_VERSION ", Copyright (C) 1999-2004 Digium.\n");
+                       ast_verbose( "Written by Mark Spencer <markster@digium.com>\n");
                        ast_verbose( "=========================================================================\n");
                        ast_remotecontrol(NULL);
                        quit_handler(0, 0, 0, 0);
@@ -1264,18 +1595,15 @@ int main(int argc, char *argv[])
                ast_log(LOG_WARNING, "Unable to open pid file '%s': %s\n", (char *)ast_config_AST_PID, strerror(errno));
 
        if (!option_verbose && !option_debug && !option_nofork && !option_console) {
-#if 1
                daemon(0,0);
-#else  
-               pid = fork();
-               if (pid < 0) {
-                       ast_log(LOG_ERROR, "Unable to fork(): %s\n", strerror(errno));
-                       printf(term_quit());
-                       exit(1);
-               }
-               if (pid) 
-                       exit(0);
-#endif                 
+               /* Blindly re-write pid file since we are forking */
+               unlink((char *)ast_config_AST_PID);
+               f = fopen((char *)ast_config_AST_PID, "w");
+               if (f) {
+                       fprintf(f, "%d\n", getpid());
+                       fclose(f);
+               } else
+                       ast_log(LOG_WARNING, "Unable to open pid file '%s': %s\n", (char *)ast_config_AST_PID, strerror(errno));
        }
 
        ast_makesocket();
@@ -1296,11 +1624,14 @@ int main(int argc, char *argv[])
        }
        if (option_console && !option_verbose) 
                ast_verbose("[ Booting...");
+
        signal(SIGURG, urg_handler);
        signal(SIGINT, __quit_handler);
        signal(SIGTERM, __quit_handler);
        signal(SIGHUP, hup_handler);
-       signal(SIGPIPE, pipe_handler);
+       signal(SIGCHLD, child_handler);
+       signal(SIGPIPE, SIG_IGN);
+
        if (set_priority(option_highpriority)) {
                printf(term_quit());
                exit(1);
@@ -1313,6 +1644,7 @@ int main(int argc, char *argv[])
                printf(term_quit());
                exit(1);
        }
+       ast_rtp_init();
        if (ast_image_init()) {
                printf(term_quit());
                exit(1);
@@ -1333,14 +1665,24 @@ int main(int argc, char *argv[])
                printf(term_quit());
                exit(1);
        }
+       if (ast_enum_init()) {
+               printf(term_quit());
+               exit(1);
+       }
        /* We might have the option of showing a console, but for now just
           do nothing... */
        if (option_console && !option_verbose)
                ast_verbose(" ]\n");
        if (option_verbose || option_console)
                ast_verbose(term_color(tmp, "Asterisk Ready.\n", COLOR_BRWHITE, COLOR_BLACK, sizeof(tmp)));
+       if (option_nofork)
+               consolethread = pthread_self();
        fully_booted = 1;
        pthread_sigmask(SIG_UNBLOCK, &sigs, NULL);
+#ifdef __AST_DEBUG_MALLOC
+       __ast_mm_init();
+#endif 
+       time(&ast_startuptime);
        ast_cli_register(&astshutdownnow);
        ast_cli_register(&astshutdowngracefully);
        ast_cli_register(&astrestartnow);
@@ -1348,28 +1690,166 @@ int main(int argc, char *argv[])
        ast_cli_register(&astrestartwhenconvenient);
        ast_cli_register(&astshutdownwhenconvenient);
        ast_cli_register(&aborthalt);
+       ast_cli_register(&astbang);
        if (option_console) {
                /* Console stuff now... */
                /* Register our quit function */
                char title[256];
                set_icon("Asterisk");
-               snprintf(title, sizeof(title), "Asterisk Console on '%s' (pid %d)", hostname, mainpid);
+               snprintf(title, sizeof(title), "Asterisk Console on '%s' (pid %d)", hostname, ast_mainpid);
                set_title(title);
            ast_cli_register(&quit);
            ast_cli_register(&astexit);
-               consolethread = pthread_self();
-
-               while ( (buf = (char *)el_gets(el, &num) ) != NULL && num != 0) {
 
-                       if (buf[strlen(buf)-1] == '\n')
-                               buf[strlen(buf)-1] = '\0';
+               for (;;) {
+                       buf = (char *)el_gets(el, &num);
+                       if (buf) {
+                               if (buf[strlen(buf)-1] == '\n')
+                                       buf[strlen(buf)-1] = '\0';
 
-                       consolehandler((char *)buf);
+                               consolehandler((char *)buf);
+                       } else {
+                               if (option_remote)
+                                       ast_cli(STDOUT_FILENO, "\nUse EXIT or QUIT to exit the asterisk console\n");
+                       }
                }
 
        } else {
                /* Do nothing */
-               ast_select(0,NULL,NULL,NULL,NULL);
+               for(;;) 
+                       poll(NULL,0, -1);
        }
        return 0;
 }
+
+
+#if  defined(__FreeBSD__)
+
+/* duh? ERANGE value copied from web... */
+#define ERANGE 34
+#undef gethostbyname
+
+int gethostbyname_r (const char *name,
+                    struct hostent *ret,
+                    char *buf,
+                    size_t buflen,
+                    struct hostent **result,
+                    int *h_errnop);
+
+int gethostbyname_r (const char *name,
+                    struct hostent *ret,
+                    char *buf,
+                    size_t buflen,
+                    struct hostent **result,
+                    int *h_errnop) {
+
+  int hsave;
+  struct hostent *ph;
+  static ast_mutex_t __mutex = AST_MUTEX_INITIALIZER;
+  ast_mutex_lock(&__mutex); /* begin critical area */
+  hsave = h_errno;
+
+  ph = gethostbyname(name);
+  *h_errnop = h_errno; /* copy h_errno to *h_herrnop */
+  if (ph == NULL) {
+    *result = NULL;
+  } else {
+    char **p, **q;
+    char *pbuf;
+    int nbytes=0;
+    int naddr=0, naliases=0;
+    /* determine if we have enough space in buf */
+
+    /* count how many addresses */
+    for (p = ph->h_addr_list; *p != 0; p++) {
+      nbytes += ph->h_length; /* addresses */
+      nbytes += sizeof(*p); /* pointers */
+      naddr++;
+    }
+    nbytes += sizeof(*p); /* one more for the terminating NULL */
+
+    /* count how many aliases, and total length of strings */
+
+    for (p = ph->h_aliases; *p != 0; p++) {
+      nbytes += (strlen(*p)+1); /* aliases */
+      nbytes += sizeof(*p);  /* pointers */
+      naliases++;
+    }
+    nbytes += sizeof(*p); /* one more for the terminating NULL */
+
+    /* here nbytes is the number of bytes required in buffer */
+    /* as a terminator must be there, the minimum value is ph->h_length */
+    if(nbytes > buflen) {
+      *result = NULL;
+      pthread_mutex_unlock(&__mutex); /* end critical area */
+      return ERANGE; /* not enough space in buf!! */
+    }
+
+    /* There is enough space. Now we need to do a deep copy! */
+    /* Allocation in buffer:
+       from [0] to [(naddr-1) * sizeof(*p)]:
+       pointers to addresses
+       at [naddr * sizeof(*p)]:
+       NULL
+       from [(naddr+1) * sizeof(*p)] to [(naddr+naliases) * sizeof(*p)] :
+       pointers to aliases
+       at [(naddr+naliases+1) * sizeof(*p)]:
+       NULL
+       then naddr addresses (fixed length), and naliases aliases (asciiz).
+    */
+
+    *ret = *ph;   /* copy whole structure (not its address!) */
+
+    /* copy addresses */
+    q = (char **)buf; /* pointer to pointers area (type: char **) */
+    ret->h_addr_list = q; /* update pointer to address list */
+    pbuf = buf + ((naddr+naliases+2)*sizeof(*p)); /* skip that area */
+    for (p = ph->h_addr_list; *p != 0; p++) {
+      memcpy(pbuf, *p, ph->h_length); /* copy address bytes */
+      *q++ = pbuf; /* the pointer is the one inside buf... */
+      pbuf += ph->h_length; /* advance pbuf */
+    }
+    *q++ = NULL; /* address list terminator */
+
+    /* copy aliases */
+
+    ret->h_aliases = q; /* update pointer to aliases list */
+    for (p = ph->h_aliases; *p != 0; p++) {
+      strcpy(pbuf, *p); /* copy alias strings */
+      *q++ = pbuf; /* the pointer is the one inside buf... */
+      pbuf += strlen(*p); /* advance pbuf */
+      *pbuf++ = 0; /* string terminator */
+    }
+    *q++ = NULL; /* terminator */
+
+    strcpy(pbuf, ph->h_name); /* copy alias strings */
+    ret->h_name = pbuf;
+    pbuf += strlen(ph->h_name); /* advance pbuf */
+    *pbuf++ = 0; /* string terminator */
+
+    *result = ret;  /* and let *result point to structure */
+
+  }
+  h_errno = hsave;  /* restore h_errno */
+
+  ast_mutex_unlock(&__mutex); /* end critical area */
+
+  return (*result != NULL);
+
+}
+
+
+#endif
+
+struct hostent *ast_gethostbyname(const char *host, struct ast_hostent *hp)
+{
+       int res;
+       int herrno;
+       struct hostent *result = NULL;
+
+       res = gethostbyname_r(host, &hp->hp, hp->buf, sizeof(hp->buf), &result, &herrno);
+
+       if (res)
+               return NULL;
+       return &hp->hp;
+}