CHANNEL(callid): Give dialplan access to the callid.
[asterisk/asterisk.git] / funcs / func_cdr.c
index 3f24816..83942db 100644 (file)
@@ -31,8 +31,6 @@
 
 #include "asterisk.h"
 
-ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
-
 #include "asterisk/module.h"
 #include "asterisk/channel.h"
 #include "asterisk/pbx.h"
@@ -40,6 +38,7 @@ ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
 #include "asterisk/app.h"
 #include "asterisk/cdr.h"
 #include "asterisk/stasis.h"
+#include "asterisk/stasis_message_router.h"
 
 /*** DOCUMENTATION
        <function name="CDR" language="en_US">
@@ -179,7 +178,8 @@ ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
                                                <para>Write-Only</para>
                                        </enum>
                                        <enum name="disable">
-                                               <para>Disable CDRs for this channel.</para>
+                                               <para>Setting to 1 will disable CDRs for this channel.
+                                               Setting to 0 will enable CDRs for this channel.</para>
                                                <para>Write-Only</para>
                                        </enum>
                                </enumlist>
@@ -207,6 +207,7 @@ struct cdr_func_payload {
        const char *cmd;
        const char *arguments;
        const char *value;
+       void *data;
 };
 
 struct cdr_func_data {
@@ -218,10 +219,36 @@ STASIS_MESSAGE_TYPE_DEFN_LOCAL(cdr_read_message_type);
 STASIS_MESSAGE_TYPE_DEFN_LOCAL(cdr_write_message_type);
 STASIS_MESSAGE_TYPE_DEFN_LOCAL(cdr_prop_write_message_type);
 
+static struct timeval cdr_retrieve_time(struct ast_channel *chan, const char *time_name)
+{
+       struct timeval time = { 0 };
+       char *value = NULL;
+       char tempbuf[128];
+       long int tv_sec;
+       long int tv_usec;
+
+       if (ast_strlen_zero(ast_channel_name(chan))) {
+               /* Format request on a dummy channel */
+               ast_cdr_format_var(ast_channel_cdr(chan), time_name, &value, tempbuf, sizeof(tempbuf), 1);
+       } else {
+               ast_cdr_getvar(ast_channel_name(chan), time_name, tempbuf, sizeof(tempbuf));
+       }
+
+       /* time.tv_usec is suseconds_t, which could be int or long */
+       if (sscanf(tempbuf, "%ld.%ld", &tv_sec, &tv_usec) == 2) {
+               time.tv_sec = tv_sec;
+               time.tv_usec = tv_usec;
+       } else {
+               ast_log(AST_LOG_WARNING, "Failed to fully extract '%s' from CDR\n", time_name);
+       }
+
+       return time;
+}
+
 static void cdr_read_callback(void *data, struct stasis_subscription *sub, struct stasis_message *message)
 {
-       struct cdr_func_data *output = data;
        struct cdr_func_payload *payload = stasis_message_data(message);
+       struct cdr_func_data *output;
        char *info;
        char *value = NULL;
        struct ast_flags flags = { 0 };
@@ -235,9 +262,9 @@ static void cdr_read_callback(void *data, struct stasis_subscription *sub, struc
                return;
        }
 
-       if (!payload || !output) {
-               return;
-       }
+       ast_assert(payload != NULL);
+       output = payload->data;
+       ast_assert(output != NULL);
 
        if (ast_strlen_zero(payload->arguments)) {
                ast_log(AST_LOG_WARNING, "%s requires a variable (%s(variable[,option]))\n)",
@@ -265,16 +292,21 @@ static void cdr_read_callback(void *data, struct stasis_subscription *sub, struc
 
        if (ast_test_flag(&flags, OPT_FLOAT)
                && (!strcasecmp("billsec", args.variable) || !strcasecmp("duration", args.variable))) {
-               long ms;
-               double dtime;
+               struct timeval start = cdr_retrieve_time(payload->chan, !strcasecmp("billsec", args.variable) ? "answer" : "start");
+               struct timeval finish = cdr_retrieve_time(payload->chan, "end");
+               double delta;
 
-               if (sscanf(tempbuf, "%30ld", &ms) != 1) {
-                       ast_log(AST_LOG_WARNING, "Unable to parse %s (%s) from the CDR for channel %s\n",
-                               args.variable, tempbuf, ast_channel_name(payload->chan));
-                       return;
+               if (ast_tvzero(finish)) {
+                       finish = ast_tvnow();
+               }
+
+               if (ast_tvzero(start)) {
+                       delta = 0.0;
+               } else {
+                       delta = (double)(ast_tvdiff_us(finish, start) / 1000000.0);
                }
-               dtime = (double)(ms / 1000.0);
-               snprintf(tempbuf, sizeof(tempbuf), "%lf", dtime);
+               snprintf(tempbuf, sizeof(tempbuf), "%lf", delta);
+
        } else if (!ast_test_flag(&flags, OPT_UNPARSED)) {
                if (!strcasecmp("start", args.variable)
                        || !strcasecmp("end", args.variable)
@@ -282,16 +314,22 @@ static void cdr_read_callback(void *data, struct stasis_subscription *sub, struc
                        struct timeval fmt_time;
                        struct ast_tm tm;
                        /* tv_usec is suseconds_t, which could be int or long */
+                       long int tv_sec;
                        long int tv_usec;
 
-                       if (sscanf(tempbuf, "%ld.%ld", &fmt_time.tv_sec, &tv_usec) != 2) {
+                       if (sscanf(tempbuf, "%ld.%ld", &tv_sec, &tv_usec) != 2) {
                                ast_log(AST_LOG_WARNING, "Unable to parse %s (%s) from the CDR for channel %s\n",
                                        args.variable, tempbuf, ast_channel_name(payload->chan));
                                return;
                        }
-                       fmt_time.tv_usec = tv_usec;
-                       ast_localtime(&fmt_time, &tm, NULL);
-                       ast_strftime(tempbuf, sizeof(*tempbuf), "%Y-%m-%d %T", &tm);
+                       if (tv_sec) {
+                               fmt_time.tv_sec = tv_sec;
+                               fmt_time.tv_usec = tv_usec;
+                               ast_localtime(&fmt_time, &tm, NULL);
+                               ast_strftime(tempbuf, sizeof(tempbuf), "%Y-%m-%d %T", &tm);
+                       } else {
+                               tempbuf[0] = '\0';
+                       }
                } else if (!strcasecmp("disposition", args.variable)) {
                        int disposition;
 
@@ -431,16 +469,28 @@ static int cdr_read(struct ast_channel *chan, const char *cmd, char *parse,
                    char *buf, size_t len)
 {
        RAII_VAR(struct stasis_message *, message, NULL, ao2_cleanup);
-       RAII_VAR(struct cdr_func_payload *, payload,
-               ao2_alloc(sizeof(*payload), NULL), ao2_cleanup);
+       RAII_VAR(struct cdr_func_payload *, payload, NULL, ao2_cleanup);
        struct cdr_func_data output = { 0, };
 
+       if (!chan) {
+               ast_log(LOG_WARNING, "No channel was provided to %s function.\n", cmd);
+               return -1;
+       }
+
+       if (!cdr_read_message_type()) {
+               ast_log(AST_LOG_WARNING, "Failed to manipulate CDR for channel %s: message type not available\n",
+                       ast_channel_name(chan));
+               return -1;
+       }
+
+       payload = ao2_alloc(sizeof(*payload), NULL);
        if (!payload) {
                return -1;
        }
        payload->chan = chan;
        payload->cmd = cmd;
        payload->arguments = parse;
+       payload->data = &output;
 
        buf[0] = '\0';/* Ensure the buffer is initialized. */
        output.buf = buf;
@@ -460,18 +510,14 @@ static int cdr_read(struct ast_channel *chan, const char *cmd, char *parse,
        if (ast_strlen_zero(ast_channel_name(chan))) {
                cdr_read_callback(NULL, NULL, message);
        } else {
-               RAII_VAR(struct stasis_subscription *, subscription, NULL, ao2_cleanup);
+               RAII_VAR(struct stasis_message_router *, router, ast_cdr_message_router(), ao2_cleanup);
 
-               subscription = stasis_subscribe(ast_channel_topic(chan), cdr_read_callback, &output);
-               if (!subscription) {
-                       ast_log(AST_LOG_WARNING, "Failed to manipulate CDR for channel %s: unable to create subscription\n",
+               if (!router) {
+                       ast_log(AST_LOG_WARNING, "Failed to manipulate CDR for channel %s: no message router\n",
                                ast_channel_name(chan));
                        return -1;
                }
-
-               stasis_publish(ast_channel_topic(chan), message);
-
-               subscription = stasis_unsubscribe_and_join(subscription);
+               stasis_message_router_publish_sync(router, message);
        }
 
        return 0;
@@ -481,10 +527,28 @@ static int cdr_write(struct ast_channel *chan, const char *cmd, char *parse,
                     const char *value)
 {
        RAII_VAR(struct stasis_message *, message, NULL, ao2_cleanup);
-       RAII_VAR(struct cdr_func_payload *, payload,
-               ao2_alloc(sizeof(*payload), NULL), ao2_cleanup);
-       RAII_VAR(struct stasis_subscription *, subscription, NULL, ao2_cleanup);
+       RAII_VAR(struct cdr_func_payload *, payload, NULL, ao2_cleanup);
+       RAII_VAR(struct stasis_message_router *, router,
+                    ast_cdr_message_router(), ao2_cleanup);
 
+       if (!chan) {
+               ast_log(LOG_WARNING, "No channel was provided to %s function.\n", cmd);
+               return -1;
+       }
+
+       if (!router) {
+               ast_log(AST_LOG_WARNING, "Failed to manipulate CDR for channel %s: no message router\n",
+                       ast_channel_name(chan));
+               return -1;
+       }
+
+       if (!cdr_write_message_type()) {
+               ast_log(AST_LOG_WARNING, "Failed to manipulate CDR for channel %s: message type not available\n",
+                       ast_channel_name(chan));
+               return -1;
+       }
+
+       payload = ao2_alloc(sizeof(*payload), NULL);
        if (!payload) {
                return -1;
        }
@@ -499,17 +563,7 @@ static int cdr_write(struct ast_channel *chan, const char *cmd, char *parse,
                        ast_channel_name(chan));
                return -1;
        }
-
-       subscription = stasis_subscribe(ast_channel_topic(chan), cdr_write_callback, NULL);
-       if (!subscription) {
-               ast_log(AST_LOG_WARNING, "Failed to manipulate CDR for channel %s: unable to create subscription\n",
-                       ast_channel_name(chan));
-               return -1;
-       }
-
-       stasis_publish(ast_channel_topic(chan), message);
-
-       subscription = stasis_unsubscribe_and_join(subscription);
+       stasis_message_router_publish_sync(router, message);
 
        return 0;
 }
@@ -518,10 +572,27 @@ static int cdr_prop_write(struct ast_channel *chan, const char *cmd, char *parse
                     const char *value)
 {
        RAII_VAR(struct stasis_message *, message, NULL, ao2_cleanup);
-       RAII_VAR(struct cdr_func_payload *, payload,
-               ao2_alloc(sizeof(*payload), NULL), ao2_cleanup);
-       RAII_VAR(struct stasis_subscription *, subscription, NULL, ao2_cleanup);
+       RAII_VAR(struct cdr_func_payload *, payload, NULL, ao2_cleanup);
+       RAII_VAR(struct stasis_message_router *, router, ast_cdr_message_router(), ao2_cleanup);
 
+       if (!chan) {
+               ast_log(LOG_WARNING, "No channel was provided to %s function.\n", cmd);
+               return -1;
+       }
+
+       if (!router) {
+               ast_log(AST_LOG_WARNING, "Failed to manipulate CDR for channel %s: no message router\n",
+                       ast_channel_name(chan));
+               return -1;
+       }
+
+       if (!cdr_write_message_type()) {
+               ast_log(AST_LOG_WARNING, "Failed to manipulate CDR for channel %s: message type not available\n",
+                       ast_channel_name(chan));
+               return -1;
+       }
+
+       payload = ao2_alloc(sizeof(*payload), NULL);
        if (!payload) {
                return -1;
        }
@@ -536,17 +607,7 @@ static int cdr_prop_write(struct ast_channel *chan, const char *cmd, char *parse
                        ast_channel_name(chan));
                return -1;
        }
-
-       subscription = stasis_subscribe(ast_channel_topic(chan), cdr_prop_write_callback, NULL);
-       if (!subscription) {
-               ast_log(AST_LOG_WARNING, "Failed to manipulate CDR for channel %s: unable to create subscription\n",
-                       ast_channel_name(chan));
-               return -1;
-       }
-
-       stasis_publish(ast_channel_topic(chan), message);
-
-       subscription = stasis_unsubscribe_and_join(subscription);
+       stasis_message_router_publish_sync(router, message);
 
        return 0;
 }
@@ -565,8 +626,14 @@ static struct ast_custom_function cdr_prop_function = {
 
 static int unload_module(void)
 {
+       RAII_VAR(struct stasis_message_router *, router, ast_cdr_message_router(), ao2_cleanup);
        int res = 0;
 
+       if (router) {
+               stasis_message_router_remove(router, cdr_prop_write_message_type());
+               stasis_message_router_remove(router, cdr_write_message_type());
+               stasis_message_router_remove(router, cdr_read_message_type());
+       }
        STASIS_MESSAGE_TYPE_CLEANUP(cdr_read_message_type);
        STASIS_MESSAGE_TYPE_CLEANUP(cdr_write_message_type);
        STASIS_MESSAGE_TYPE_CLEANUP(cdr_prop_write_message_type);
@@ -578,15 +645,29 @@ static int unload_module(void)
 
 static int load_module(void)
 {
+       RAII_VAR(struct stasis_message_router *, router, ast_cdr_message_router(), ao2_cleanup);
        int res = 0;
 
+       if (!router) {
+               return AST_MODULE_LOAD_FAILURE;
+       }
+
        res |= STASIS_MESSAGE_TYPE_INIT(cdr_read_message_type);
        res |= STASIS_MESSAGE_TYPE_INIT(cdr_write_message_type);
        res |= STASIS_MESSAGE_TYPE_INIT(cdr_prop_write_message_type);
        res |= ast_custom_function_register(&cdr_function);
        res |= ast_custom_function_register(&cdr_prop_function);
-
-       return res;
+       res |= stasis_message_router_add(router, cdr_prop_write_message_type(),
+                                        cdr_prop_write_callback, NULL);
+       res |= stasis_message_router_add(router, cdr_write_message_type(),
+                                        cdr_write_callback, NULL);
+       res |= stasis_message_router_add(router, cdr_read_message_type(),
+                                        cdr_read_callback, NULL);
+
+       if (res) {
+               return AST_MODULE_LOAD_FAILURE;
+       }
+       return AST_MODULE_LOAD_SUCCESS;
 }
 
 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Call Detail Record (CDR) dialplan functions");