Version 0.1.1 from FTP
[asterisk/asterisk.git] / asterisk.c
index 1bc7be1..a794740 100755 (executable)
@@ -3,7 +3,7 @@
  *
  * Top level source file for asterisk
  * 
- * Copyright (C) 1999, Adtran Inc. and Linux Support Services, LLC
+ * Copyright (C) 1999, Mark Spencer
  *
  * Mark Spencer <markster@linux-support.net>
  *
 #include <stdlib.h>
 #include <asterisk/logger.h>
 #include <asterisk/options.h>
+#include <asterisk/cli.h>
 #include <stdio.h>
 #include <signal.h>
+#include <sched.h>
+#include <pthread.h>
+#include <readline/readline.h>
+#include <readline/history.h>
 #include "asterisk.h"
 
 int option_verbose=0;
 int option_debug=0;
 int option_nofork=0;
 int option_quiet=0;
+int option_highpriority=0;
+
+#define HIGH_PRIORITY 1
+#define HIGH_PRIORITY_SCHED SCHED_RR
 
 static void urg_handler(int num)
 {
@@ -33,31 +42,110 @@ static void urg_handler(int num)
        return;
 }
 
+static int set_priority(int pri)
+{
+       struct sched_param sched;
+       /* 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.  */
+       if (pri) {  
+               sched.sched_priority = HIGH_PRIORITY;
+               if (sched_setscheduler(0, HIGH_PRIORITY_SCHED, &sched)) {
+                       ast_log(LOG_WARNING, "Unable to set high priority\n");
+                       return -1;
+               }
+       } else {
+               sched.sched_priority = 0;
+               if (sched_setscheduler(0, SCHED_OTHER, &sched)) {
+                       ast_log(LOG_WARNING, "Unable to set normal priority\n");
+                       return -1;
+               }
+       }
+       return 0;
+}
+
 static void quit_handler(int num)
 {
+       static pthread_mutex_t quitlock = PTHREAD_MUTEX_INITIALIZER;
+       char filename[80] = "";
+       if (getenv("HOME")) 
+               snprintf(filename, sizeof(filename), "%s/.asterisk_history", getenv("HOME"));
+       /* Quit only once */
+       pthread_mutex_lock(&quitlock);
        /* Called on exit */
        if (option_verbose)
                ast_verbose("Asterisk ending (%d).\n", num);
        else if (option_debug)
                ast_log(LOG_DEBUG, "Asterisk ending (%d).\n", num);
+       if (strlen(filename))
+               write_history(filename);
        exit(0);
 }
 
+static pthread_t consolethread = -1;
+
+static void console_verboser(char *s, int pos, int replace, int complete)
+{
+       /* Return to the beginning of the line */
+       if (!pos)
+               fprintf(stdout, "\r");
+       fprintf(stdout, s + pos);
+       if (complete)
+       /* Wake up a select()ing console */
+               pthread_kill(consolethread, SIGURG);
+}
+
+static void consolehandler(char *s)
+{
+       /* Called when readline data is available */
+       if (s && strlen(s))
+               add_history(s);
+       if (s)
+               ast_cli_command(STDOUT_FILENO, s);
+}
+
+static char quit_help[] = 
+"Usage: quit\n"
+"       Exits Asterisk.\n";
+
+static int handle_quit(int fd, int argc, char *argv[])
+{
+       if (argc != 1)
+               return RESULT_SHOWUSAGE;
+       quit_handler(0);
+       return RESULT_SUCCESS;
+}
+
+#define ASTERISK_PROMPT "*CLI> "
+
+static struct ast_cli_entry quit =     { { "quit", NULL }, handle_quit, "Exit Asterisk", quit_help };
+
+static char *cli_generator(char *text, int state)
+{
+       return ast_cli_generator(rl_line_buffer, text, state);
+}
+
 int main(int argc, char *argv[])
 {
        char c;
+       fd_set rfds;
+       int res;
+       char filename[80] = "";
+       if (getenv("HOME")) 
+               snprintf(filename, sizeof(filename), "%s/.asterisk_history", getenv("HOME"));
        /* Check if we're root */
        if (geteuid()) {
                ast_log(LOG_ERROR, "Must be run as root\n");
                exit(1);
        }
        /* Check for options */
-       while((c=getopt(argc, argv, "dvq")) != EOF) {
+       while((c=getopt(argc, argv, "dvqp")) != EOF) {
                switch(c) {
                case 'd':
                        option_debug++;
                        option_nofork++;
-                       option_verbose++;
+                       break;
+               case 'p':
+                       option_highpriority++;
                        break;
                case 'v':
                        option_verbose++;
@@ -71,7 +159,7 @@ int main(int argc, char *argv[])
        }
        /* Print a welcome message if desired */
        if (option_verbose) {
-               ast_verbose( "Asterisk, Copyright (C) 1999 Adtran, Inc. and Linux Support Services, LLC\n");
+               ast_verbose( "Asterisk, Copyright (C) 1999 Mark Spencer\n");
                ast_verbose( "Written by Mark Spencer <markster@linux-support.net>\n");
                ast_verbose( "=========================================================================\n");
        }
@@ -85,10 +173,32 @@ int main(int argc, char *argv[])
                exit(1);
        if (load_modules())
                exit(1);
+       if (set_priority(option_highpriority))
+               exit(1);
        /* We might have the option of showing a console, but for now just
           do nothing... */
+
+       /* Console stuff now... */
+       /* Register our quit function */
+    ast_cli_register(&quit);
+       consolethread = pthread_self();
+       ast_register_verbose(console_verboser);
        if (option_verbose)
                ast_verbose( "Asterisk Ready.\n");
-       select(0,NULL,NULL,NULL,NULL);
+       if (strlen(filename))
+               read_history(filename);
+       rl_callback_handler_install(ASTERISK_PROMPT, consolehandler);
+       rl_completion_entry_function = (Function *)cli_generator;
+       for(;;) {
+               FD_ZERO(&rfds);
+               FD_SET(STDIN_FILENO, &rfds);
+               res = select(STDIN_FILENO + 1, &rfds, NULL, NULL, NULL);
+               if (res > 0) {
+                       rl_callback_read_char();
+               } else if (res < 1) {
+                       rl_forced_update_display();
+               }
+
+       }       
        return 0;
 }