Changeset 141 for pjproject/trunk/pjsip/src/pjsua/main.c

Timestamp:

Feb 7, 2006 6:48:01 PM (18 years ago)

Author:

bennylp

Message:

Tested initial implementation: basic UAC, client registration, authentication, etc

File:

• pjproject/trunk/pjsip/src/pjsua/main.c (modified) (12 diffs)

pjproject/trunk/pjsip/src/pjsua/main.c

@@ -18 +18 @@
  */
 #include "pjsua.h"
+#include "getopt.h"
+
 
 /* For debugging, disable threading. */
@@ -33 +35 @@
  * Notify UI when invite state has changed.
  */
-void ui_inv_on_state_changed(pjsip_inv_session *inv, pjsip_event *e)
+void pjsua_ui_inv_on_state_changed(pjsip_inv_session *inv, pjsip_event *e)
 {
     const char *state_names[] =
@@ -51 +53 @@
     PJ_LOG(3,(THIS_FILE, "INVITE session state changed to %s", state_names[inv->state]));
 
-    if (inv->state == PJSIP_INV_STATE_DISCONNECTED ||
-        inv->state == PJSIP_INV_STATE_TERMINATED)
-    {
+    if (inv->state == PJSIP_INV_STATE_DISCONNECTED) {
         if (inv == inv_session)
             inv_session = NULL;
@@ -104 +104 @@
             }
 
-#if 0
+#if 1
             printf("Enter URL to call: ");
             fgets(buf, sizeof(buf), stdin);
@@ -122 +122 @@
 
             pjsua_invite(buf, &inv);
+
+#else
+
+            pjsua_invite("sip:localhost:5061", &inv);
 #endif
-
-            pjsua_invite("sip:localhost:5061", &inv);
             break;
 
@@ -163 +165 @@
 }
 
+
+/*****************************************************************************
+ * This is a very simple PJSIP module, whose sole purpose is to display
+ * incoming and outgoing messages to log. This module will have priority
+ * higher than transport layer, which means:
+ *
+ *  - incoming messages will come to this module first before reaching
+ *    transaction layer.
+ *
+ *  - outgoing messages will come to this module last, after the message
+ *    has been 'printed' to contiguous buffer by transport layer and
+ *    appropriate transport instance has been decided for this message.
+ *
+ */
+
+/* Notification on incoming messages */
 static pj_bool_t console_on_rx_msg(pjsip_rx_data *rdata)
 {
@@ -174 +192 @@
                          rdata->msg_info.msg_buf));
     
-    /* Must return false for logger! */
+    /* Always return false, otherwise messages will not get processed! */
     return PJ_FALSE;
 }
 
+/* Notification on outgoing messages */
 static pj_status_t console_on_tx_msg(pjsip_tx_data *tdata)
 {
+    
+    /* Important note:
+     *  tp_info field is only valid after outgoing messages has passed
+     *  transport layer. So don't try to access tp_info when the module
+     *  has lower priority than transport layer.
+     */
+
     PJ_LOG(4,(THIS_FILE, "TX %d bytes %s to %s:%d:\n"
                          "%s\n"
@@ -189 +215 @@
                          tdata->buf.start));
 
+    /* Always return success, otherwise message will not get sent! */
     return PJ_SUCCESS;
 }
 
+/* The module instance. */
 static pjsip_module console_msg_logger = 
 {
@@ -212 +240 @@
 
 
-int main()
+
+/*****************************************************************************
+ * Console application custom logging:
+ */
+
+
+static FILE *log_file;
+
+
+static void app_log_writer(int level, const char *buffer, int len)
+{
+    /* Write to both stdout and file. */
+
+    if (level <= pjsua.app_log_level)
+        pj_log_write(level, buffer, len);
+
+    if (log_file) {
+        fwrite(buffer, len, 1, log_file);
+        fflush(log_file);
+    }
+}
+
+
+void app_logging_init(void)
+{
+    /* Redirect log function to ours */
+
+    pj_log_set_log_func( &app_log_writer );
+
+    /* If output log file is desired, create the file: */
+
+    if (pjsua.log_filename)
+        log_file = fopen(pjsua.log_filename, "wt");
+}
+
+
+void app_logging_shutdown(void)
+{
+    /* Close logging file, if any: */
+
+    if (log_file) {
+        fclose(log_file);
+        log_file = NULL;
+    }
+}
+
+/*****************************************************************************
+ * Command line argument processing:
+ */
+
+
+/* Show usage */
+static void usage(void)
+{
+    puts("Usage:");
+    puts("  pjsua [options] [sip-url]");
+    puts("");
+    puts("  [sip-url]   Default URL to invite.");
+    puts("");
+    puts("General options:");
+    puts("  --config-file=file  Read the config/arguments from file.");
+    puts("  --log-file=fname    Log to filename (default stderr)");
+    puts("  --log-level=N       Set log max level to N (0(none) to 6(trace))");
+    puts("  --app-log-level=N   Set log max level for stdout display to N");
+    puts("  --help              Display this help screen");
+    puts("  --version           Display version info");
+    puts("");
+    puts("Media options:");
+    puts("  --null-audio        Use NULL audio device");
+    puts("");
+    puts("User Agent options:");
+    puts("  --auto-answer=sec   Auto-answer all incoming calls after sec seconds.");
+    puts("  --auto-hangup=sec   Auto-hangup all calls after sec seconds.");
+    puts("");
+    puts("SIP options:");
+    puts("  --local-port=port   Set TCP/UDP port");
+    puts("  --id=url            Set the URL of local ID (used in From header)");
+    puts("  --contact=url       Override the Contact information");
+    puts("  --proxy=url         Set the URL of proxy server");
+    puts("  --outbound=url      Set the URL of outbound proxy server");
+    puts("  --registrar=url     Set the URL of registrar server");
+    puts("  --reg-timeout=secs  Set registration interval to secs (default 3600)");
+    puts("");
+    puts("Authentication options:");
+    puts("  --realm=string      Set realm");
+    puts("  --username=string   Set authentication username");
+    puts("  --password=string   Set authentication password");
+    puts("");
+    puts("STUN options (all must be specified):");
+    puts("  --use-stun1=host[:port]");
+    puts("  --use-stun2=host[:port]  Use STUN and set host name and port of STUN servers");
+    puts("");
+    puts("SIMPLE options (may be specified more than once):");
+    puts("  --add-buddy url     Add the specified URL to the buddy list.");
+    puts("  --offer-x-ms-msg    Offer \"x-ms-message\" in outgoing INVITE");
+    puts("  --no-presence       Do not subscribe presence of buddies");
+    puts("");
+    fflush(stdout);
+}
+
+
+/*
+ * Verify that valid SIP url is given.
+ */
+static pj_status_t verify_sip_url(char *url)
+{
+    pjsip_uri *p;
+    pj_pool_t *pool;
+    int len = (url ? strlen(url) : 0);
+
+    if (!len) return -1;
+
+    pool = pj_pool_create(&pjsua.cp.factory, "check%p", 1024, 0, NULL);
+    if (!pool) return -1;
+
+    p = pjsip_parse_uri(pool, url, len, 0);
+    if (!p || pj_stricmp2(pjsip_uri_get_scheme(p), "sip") != 0)
+        p = NULL;
+
+    pj_pool_release(pool);
+    return p ? 0 : -1;
+}
+
+
+/*
+ * Read command arguments from config file.
+ */
+static int read_config_file(pj_pool_t *pool, const char *filename, 
+                            int *app_argc, char ***app_argv)
+{
+    int i;
+    FILE *fhnd;
+    char line[200];
+    int argc = 0;
+    char **argv;
+    enum { MAX_ARGS = 64 };
+
+    /* Allocate MAX_ARGS+1 (argv needs to be terminated with NULL argument) */
+    argv = pj_pool_calloc(pool, MAX_ARGS+1, sizeof(char*));
+    argv[argc++] = *app_argv[0];
+
+    /* Open config file. */
+    fhnd = fopen(filename, "rt");
+    if (!fhnd) {
+        printf("Unable to open config file %s\n", filename);
+        return -1;
+    }
+
+    /* Scan tokens in the file. */
+    while (argc < MAX_ARGS && !feof(fhnd)) {
+        char *token, *p = line;
+
+        if (fgets(line, sizeof(line), fhnd) == NULL) break;
+
+        for (token = strtok(p, " \t\r\n"); argc < MAX_ARGS; 
+             token = strtok(NULL, " \t\r\n"))
+        {
+            int token_len;
+            
+            if (!token) break;
+            if (*token == '#') break;
+
+            token_len = strlen(token);
+            if (!token_len)
+                continue;
+            argv[argc] = pj_pool_alloc(pool, token_len+1);
+            pj_memcpy(argv[argc], token, token_len+1);
+            ++argc;
+        }
+    }
+
+    /* Copy arguments from command line */
+    for (i=1; i<*app_argc && argc < MAX_ARGS; ++i)
+        argv[argc++] = (*app_argv)[i];
+
+    if (argc == MAX_ARGS && (i!=*app_argc || !feof(fhnd))) {
+        printf("Too many arguments specified in cmd line/config file\n");
+        fclose(fhnd);
+        return -1;
+    }
+
+    fclose(fhnd);
+
+    /* Assign the new command line back to the original command line. */
+    *app_argc = argc;
+    *app_argv = argv;
+    return 0;
+
+}
+
+
+/* Parse arguments. */
+static pj_status_t parse_args(int argc, char *argv[])
+{
+    int c;
+    int option_index;
+    enum { OPT_CONFIG_FILE, OPT_LOG_FILE, OPT_LOG_LEVEL, OPT_APP_LOG_LEVEL, 
+           OPT_HELP, OPT_VERSION, OPT_NULL_AUDIO,
+           OPT_LOCAL_PORT, OPT_PROXY, OPT_OUTBOUND_PROXY, OPT_REGISTRAR,
+           OPT_REG_TIMEOUT, OPT_ID, OPT_CONTACT, 
+           OPT_REALM, OPT_USERNAME, OPT_PASSWORD,
+           OPT_USE_STUN1, OPT_USE_STUN2, 
+           OPT_ADD_BUDDY, OPT_OFFER_X_MS_MSG, OPT_NO_PRESENCE,
+           OPT_AUTO_ANSWER, OPT_AUTO_HANGUP};
+    struct option long_options[] = {
+        { "config-file",1, 0, OPT_CONFIG_FILE},
+        { "log-file",   1, 0, OPT_LOG_FILE},
+        { "log-level",  1, 0, OPT_LOG_LEVEL},
+        { "app-log-level",1,0,OPT_APP_LOG_LEVEL},
+        { "help",       0, 0, OPT_HELP},
+        { "version",    0, 0, OPT_VERSION},
+        { "null-audio", 0, 0, OPT_NULL_AUDIO},
+        { "local-port", 1, 0, OPT_LOCAL_PORT},
+        { "proxy",      1, 0, OPT_PROXY},
+        { "outbound",   1, 0, OPT_OUTBOUND_PROXY},
+        { "registrar",  1, 0, OPT_REGISTRAR},
+        { "reg-timeout",1, 0, OPT_REG_TIMEOUT},
+        { "id",         1, 0, OPT_ID},
+        { "contact",    1, 0, OPT_CONTACT},
+        { "realm",      1, 0, OPT_REALM},
+        { "username",   1, 0, OPT_USERNAME},
+        { "password",   1, 0, OPT_PASSWORD},
+        { "use-stun1",  1, 0, OPT_USE_STUN1},
+        { "use-stun2",  1, 0, OPT_USE_STUN2},
+        { "add-buddy",  1, 0, OPT_ADD_BUDDY},
+        { "offer-x-ms-msg",0,0,OPT_OFFER_X_MS_MSG},
+        { "no-presence", 0, 0, OPT_NO_PRESENCE},
+        { "auto-answer",1, 0, OPT_AUTO_ANSWER},
+        { "auto-hangup",1, 0, OPT_AUTO_HANGUP},
+        { NULL, 0, 0, 0}
+    };
+    pj_status_t status;
+    char *config_file = NULL;
+
+    /* Run getopt once to see if user specifies config file to read. */
+    while ((c=getopt_long(argc, argv, "", long_options, &option_index)) != -1) {
+        switch (c) {
+        case OPT_CONFIG_FILE:
+            config_file = optarg;
+            break;
+        }
+        if (config_file)
+            break;
+    }
+
+    if (config_file) {
+        status = read_config_file(pjsua.pool, config_file, &argc, &argv);
+        if (status != 0)
+            return status;
+    }
+
+
+    /* Reinitialize and re-run getopt again, possibly with new arguments
+     * read from config file.
+     */
+    optind = 0;
+    while ((c=getopt_long(argc, argv, "", long_options, &option_index)) != -1) {
+        char *p;
+        pj_str_t tmp;
+        long lval;
+
+        switch (c) {
+
+        case OPT_LOG_FILE:
+            pjsua.log_filename = optarg;
+            break;
+
+        case OPT_LOG_LEVEL:
+            c = pj_strtoul(pj_cstr(&tmp, optarg));
+            if (c < 0 || c > 6) {
+                printf("Error: expecting integer value 0-6 for --log-level\n");
+                return PJ_EINVAL;
+            }
+            pj_log_set_level( c );
+            break;
+
+        case OPT_APP_LOG_LEVEL:
+            pjsua.app_log_level = pj_strtoul(pj_cstr(&tmp, optarg));
+            if (pjsua.app_log_level < 0 || pjsua.app_log_level > 6) {
+                printf("Error: expecting integer value 0-6 for --app-log-level\n");
+                return PJ_EINVAL;
+            }
+            break;
+
+        case OPT_HELP:
+            usage();
+            return PJ_EINVAL;
+
+        case OPT_VERSION:   /* version */
+            pj_dump_config();
+            return PJ_EINVAL;
+
+        case OPT_NULL_AUDIO:
+            pjsua.null_audio = 1;
+            break;
+
+        case OPT_LOCAL_PORT:   /* local-port */
+            lval = pj_strtoul(pj_cstr(&tmp, optarg));
+            if (lval < 1 || lval > 65535) {
+                printf("Error: expecting integer value for --local-port\n");
+                return PJ_EINVAL;
+            }
+            pjsua.sip_port = (pj_uint16_t)lval;
+            break;
+
+        case OPT_PROXY:   /* proxy */
+            if (verify_sip_url(optarg) != 0) {
+                printf("Error: invalid SIP URL '%s' in proxy argument\n", optarg);
+                return PJ_EINVAL;
+            }
+            pjsua.proxy = pj_str(optarg);
+            break;
+
+        case OPT_OUTBOUND_PROXY:   /* outbound proxy */
+            if (verify_sip_url(optarg) != 0) {
+                printf("Error: invalid SIP URL '%s' in outbound proxy argument\n", optarg);
+                return PJ_EINVAL;
+            }
+            pjsua.outbound_proxy = pj_str(optarg);
+            break;
+
+        case OPT_REGISTRAR:   /* registrar */
+            if (verify_sip_url(optarg) != 0) {
+                printf("Error: invalid SIP URL '%s' in registrar argument\n", optarg);
+                return PJ_EINVAL;
+            }
+            pjsua.registrar_uri = pj_str(optarg);
+            break;
+
+        case OPT_REG_TIMEOUT:   /* reg-timeout */
+            pjsua.reg_timeout = pj_strtoul(pj_cstr(&tmp,optarg));
+            if (pjsua.reg_timeout < 1 || pjsua.reg_timeout > 3600) {
+                printf("Error: invalid value for --reg-timeout (expecting 1-3600)\n");
+                return PJ_EINVAL;
+            }
+            break;
+
+        case OPT_ID:   /* id */
+            if (verify_sip_url(optarg) != 0) {
+                printf("Error: invalid SIP URL '%s' in local id argument\n", optarg);
+                return PJ_EINVAL;
+            }
+            pjsua.local_uri = pj_str(optarg);
+            break;
+
+        case OPT_CONTACT:   /* contact */
+            if (verify_sip_url(optarg) != 0) {
+                printf("Error: invalid SIP URL '%s' in contact argument\n", optarg);
+                return PJ_EINVAL;
+            }
+            pjsua.contact_uri = pj_str(optarg);
+            break;
+
+        case OPT_USERNAME:   /* Default authentication user */
+            if (!pjsua.cred_count) pjsua.cred_count = 1;
+            pjsua.cred_info[0].username = pj_str(optarg);
+            break;
+
+        case OPT_REALM:     /* Default authentication realm. */
+            if (!pjsua.cred_count) pjsua.cred_count = 1;
+            pjsua.cred_info[0].realm = pj_str(optarg);
+            break;
+
+        case OPT_PASSWORD:   /* authentication password */
+            if (!pjsua.cred_count) pjsua.cred_count = 1;
+            pjsua.cred_info[0].data_type = 0;
+            pjsua.cred_info[0].data = pj_str(optarg);
+            break;
+
+        case OPT_USE_STUN1:   /* STUN server 1 */
+            p = pj_native_strchr(optarg, ':');
+            if (p) {
+                *p = '\0';
+                pjsua.stun_srv1 = pj_str(optarg);
+                pjsua.stun_port1 = pj_strtoul(pj_cstr(&tmp, p+1));
+                if (pjsua.stun_port1 < 1 || pjsua.stun_port1 > 65535) {
+                    printf("Error: expecting port number with option --use-stun1\n");
+                    return PJ_EINVAL;
+                }
+            } else {
+                pjsua.stun_port1 = 3478;
+                pjsua.stun_srv1 = pj_str(optarg);
+            }
+            break;
+
+        case OPT_USE_STUN2:   /* STUN server 2 */
+            p = pj_native_strchr(optarg, ':');
+            if (p) {
+                *p = '\0';
+                pjsua.stun_srv2 = pj_str(optarg);
+                pjsua.stun_port2 = pj_strtoul(pj_cstr(&tmp,p+1));
+                if (pjsua.stun_port2 < 1 || pjsua.stun_port2 > 65535) {
+                    printf("Error: expecting port number with option --use-stun2\n");
+                    return PJ_EINVAL;
+                }
+            } else {
+                pjsua.stun_port2 = 3478;
+                pjsua.stun_srv2 = pj_str(optarg);
+            }
+            break;
+        }
+    }
+
+    if (optind != argc) {
+        printf("Error: unknown options %s\n", argv[optind]);
+        return PJ_EINVAL;
+    }
+
+    if (pjsua.reg_timeout == 0)
+        pjsua.reg_timeout = 3600;
+
+
+    return PJ_SUCCESS;
+}
+
+
+
+/*****************************************************************************
+ * main():
+ */
+int main(int argc, char *argv[])
 {
     /* Init default settings. */
@@ -224 +672 @@
 
 
-    /* Initialize pjsua.
-     * This will start worker thread, client registration, etc.
+    /* Initialize pjsua (to create pool etc).
      */
 
     if (pjsua_init() != PJ_SUCCESS)
         return 1;
+
+
+    /* Parse command line arguments: */
+
+    if (parse_args(argc, argv) != PJ_SUCCESS)
+        return 1;
+
+
+    /* Init logging: */
+
+    app_logging_init();
+
 
     /* Register message logger to print incoming and outgoing
@@ -238 +697 @@
 
 
+    /* Start pjsua! */
+
+    if (pjsua_start() != PJ_SUCCESS) {
+
+        pjsua_destroy();
+        return 1;
+    }
+
+
     /* Sleep for a while, let any messages get printed to console: */
 
@@ -252 +720 @@
     pjsua_destroy();
 
+
+    /* Close logging: */
+
+    app_logging_shutdown();
+
+
     /* Exit... */