Add option to hide console connect messages
[asterisk/asterisk.git] / res / res_config_curl.c
index 622b863..1d259c8 100644 (file)
@@ -61,7 +61,7 @@ static struct ast_variable *realtime_curl(const char *url, const char *unused, v
        char *stringp, *pair, *key;
        int i;
        struct ast_variable *var=NULL, *prev=NULL;
-       const int EncodeSpecialChars = 1;
+       const int EncodeSpecialChars = 1, bufsize = 64000;
        char *buffer;
 
        if (!ast_custom_function_find("CURL")) {
@@ -72,12 +72,12 @@ static struct ast_variable *realtime_curl(const char *url, const char *unused, v
        if (!(query = ast_str_create(1000)))
                return NULL;
 
-       if (!(buffer = ast_malloc(64000))) {
+       if (!(buffer = ast_malloc(bufsize))) {
                ast_free(query);
                return NULL;
        }
 
-       ast_str_set(&query, 0, "${CURL(%s,", url);
+       ast_str_set(&query, 0, "${CURL(%s/single,", url);
 
        for (i = 0; (newparam = va_arg(ap, const char *)); i++) {
                newval = va_arg(ap, const char *);
@@ -88,7 +88,7 @@ static struct ast_variable *realtime_curl(const char *url, const char *unused, v
        va_end(ap);
 
        ast_str_append(&query, 0, ")}");
-       pbx_substitute_variables_helper(NULL, query->str, buffer, sizeof(buffer));
+       pbx_substitute_variables_helper(NULL, ast_str_buffer(query), buffer, bufsize);
 
        /* Remove any trailing newline characters */
        if ((stringp = strchr(buffer, '\r')) || (stringp = strchr(buffer, '\n')))
@@ -131,7 +131,8 @@ static struct ast_config *realtime_multi_curl(const char *url, const char *unuse
        char buf1[200], buf2[200];
        const char *newparam, *newval;
        char *stringp, *line, *pair, *key, *initfield = NULL;
-       int i, EncodeSpecialChars = 1;
+       int i;
+       const int EncodeSpecialChars = 1, bufsize = 256000;
        struct ast_variable *var=NULL;
        struct ast_config *cfg=NULL;
        struct ast_category *cat=NULL;
@@ -145,7 +146,7 @@ static struct ast_config *realtime_multi_curl(const char *url, const char *unuse
        if (!(query = ast_str_create(1000)))
                return NULL;
 
-       if (!(buffer = ast_malloc(256000))) {
+       if (!(buffer = ast_malloc(bufsize))) {
                ast_free(query);
                return NULL;
        }
@@ -154,8 +155,12 @@ static struct ast_config *realtime_multi_curl(const char *url, const char *unuse
 
        for (i = 0; (newparam = va_arg(ap, const char *)); i++) {
                newval = va_arg(ap, const char *);
-               if (i == 0)
+               if (i == 0) {
+                       char *op;
                        initfield = ast_strdupa(newparam);
+                       if ((op = strchr(initfield, ' ')))
+                               *op = '\0';
+               }
                ast_uri_encode(newparam, buf1, sizeof(buf1), EncodeSpecialChars);
                ast_uri_encode(newval, buf2, sizeof(buf2), EncodeSpecialChars);
                ast_str_append(&query, 0, "%s%s=%s", i > 0 ? "&" : "", buf1, buf2);
@@ -165,7 +170,7 @@ static struct ast_config *realtime_multi_curl(const char *url, const char *unuse
        ast_str_append(&query, 0, ")}");
 
        /* Do the CURL query */
-       pbx_substitute_variables_helper(NULL, query->str, buffer, sizeof(buffer));
+       pbx_substitute_variables_helper(NULL, ast_str_buffer(query), buffer, bufsize);
 
        if (!(cfg = ast_config_new()))
                goto exit_multi;
@@ -224,7 +229,7 @@ static int update_curl(const char *url, const char *unused, const char *keyfield
        const char *newparam, *newval;
        char *stringp;
        int i, rowcount = -1;
-       const int EncodeSpecialChars = 1;
+       const int EncodeSpecialChars = 1, bufsize = 100;
        char *buffer;
 
        if (!ast_custom_function_find("CURL")) {
@@ -235,7 +240,7 @@ static int update_curl(const char *url, const char *unused, const char *keyfield
        if (!(query = ast_str_create(1000)))
                return -1;
 
-       if (!(buffer = ast_malloc(100))) {
+       if (!(buffer = ast_malloc(bufsize))) {
                ast_free(query);
                return -1;
        }
@@ -253,7 +258,70 @@ static int update_curl(const char *url, const char *unused, const char *keyfield
        va_end(ap);
 
        ast_str_append(&query, 0, ")}");
-       pbx_substitute_variables_helper(NULL, query->str, buffer, sizeof(buffer));
+       pbx_substitute_variables_helper(NULL, ast_str_buffer(query), buffer, bufsize);
+
+       /* Line oriented output */
+       stringp = buffer;
+       while (*stringp <= ' ')
+               stringp++;
+       sscanf(stringp, "%d", &rowcount);
+
+       ast_free(buffer);
+       ast_free(query);
+
+       if (rowcount >= 0)
+               return (int)rowcount;
+
+       return -1;
+}
+
+static int update2_curl(const char *url, const char *unused, va_list ap)
+{
+       struct ast_str *query;
+       char buf1[200], buf2[200];
+       const char *newparam, *newval;
+       char *stringp;
+       int rowcount = -1, lookup = 1, first = 1;
+       const int EncodeSpecialChars = 1, bufsize = 100;
+       char *buffer;
+
+       if (!ast_custom_function_find("CURL")) {
+               ast_log(LOG_ERROR, "func_curl.so must be loaded in order to use res_config_curl.so!!\n");
+               return -1;
+       }
+
+       if (!(query = ast_str_create(1000)))
+               return -1;
+
+       if (!(buffer = ast_malloc(bufsize))) {
+               ast_free(query);
+               return -1;
+       }
+
+       ast_str_set(&query, 0, "${CURL(%s/update?", url);
+
+       for (;;) {
+               if ((newparam = va_arg(ap, const char *)) == SENTINEL) {
+                       if (lookup) {
+                               lookup = 0;
+                               ast_str_append(&query, 0, ",");
+                               /* Back to the first parameter; we don't need a starting '&' */
+                               first = 1;
+                               continue;
+                       } else {
+                               break;
+                       }
+               }
+               newval = va_arg(ap, const char *);
+               ast_uri_encode(newparam, buf1, sizeof(buf1), EncodeSpecialChars);
+               ast_uri_encode(newval, buf2, sizeof(buf2), EncodeSpecialChars);
+               ast_str_append(&query, 0, "%s%s=%s", first ? "" : "&", buf1, buf2);
+       }
+       va_end(ap);
+
+       ast_str_append(&query, 0, ")}");
+       /* TODO: Make proxies work */
+       pbx_substitute_variables_helper(NULL, ast_str_buffer(query), buffer, bufsize);
 
        /* Line oriented output */
        stringp = buffer;
@@ -272,8 +340,8 @@ static int update_curl(const char *url, const char *unused, const char *keyfield
 
 /*!
  * \brief Execute an INSERT query
- * \param database
- * \param table
+ * \param url
+ * \param unused
  * \param ap list containing one or more field/value set(s)
  *
  * Insert a new record into database table, prepare the sql statement.
@@ -290,7 +358,7 @@ static int store_curl(const char *url, const char *unused, va_list ap)
        const char *newparam, *newval;
        char *stringp;
        int i, rowcount = -1;
-       const int EncodeSpecialChars = 1;
+       const int EncodeSpecialChars = 1, bufsize = 100;
        char *buffer;
 
        if (!ast_custom_function_find("CURL")) {
@@ -301,7 +369,7 @@ static int store_curl(const char *url, const char *unused, va_list ap)
        if (!(query = ast_str_create(1000)))
                return -1;
 
-       if (!(buffer = ast_malloc(100))) {
+       if (!(buffer = ast_malloc(bufsize))) {
                ast_free(query);
                return -1;
        }
@@ -317,7 +385,7 @@ static int store_curl(const char *url, const char *unused, va_list ap)
        va_end(ap);
 
        ast_str_append(&query, 0, ")}");
-       pbx_substitute_variables_helper(NULL, query->str, buffer, sizeof(buffer));
+       pbx_substitute_variables_helper(NULL, ast_str_buffer(query), buffer, bufsize);
 
        stringp = buffer;
        while (*stringp <= ' ')
@@ -355,7 +423,7 @@ static int destroy_curl(const char *url, const char *unused, const char *keyfiel
        const char *newparam, *newval;
        char *stringp;
        int i, rowcount = -1;
-       const int EncodeSpecialChars = 1;
+       const int EncodeSpecialChars = 1, bufsize = 100;
        char *buffer;
 
        if (!ast_custom_function_find("CURL")) {
@@ -366,7 +434,7 @@ static int destroy_curl(const char *url, const char *unused, const char *keyfiel
        if (!(query = ast_str_create(1000)))
                return -1;
 
-       if (!(buffer = ast_malloc(100))) {
+       if (!(buffer = ast_malloc(bufsize))) {
                ast_free(query);
                return -1;
        }
@@ -384,7 +452,7 @@ static int destroy_curl(const char *url, const char *unused, const char *keyfiel
        va_end(ap);
 
        ast_str_append(&query, 0, ")}");
-       pbx_substitute_variables_helper(NULL, query->str, buffer, sizeof(buffer));
+       pbx_substitute_variables_helper(NULL, ast_str_buffer(query), buffer, bufsize);
 
        /* Line oriented output */
        stringp = buffer;
@@ -401,13 +469,59 @@ static int destroy_curl(const char *url, const char *unused, const char *keyfiel
        return -1;
 }
 
+static int require_curl(const char *url, const char *unused, va_list ap)
+{
+       struct ast_str *query;
+       char *elm, field[256], buffer[128];
+       int type, size;
+       const int EncodeSpecialChars = 1;
+
+       if (!ast_custom_function_find("CURL")) {
+               ast_log(LOG_ERROR, "func_curl.so must be loaded in order to use res_config_curl.so!!\n");
+               return -1;
+       }
+
+       if (!(query = ast_str_create(100))) {
+               return -1;
+       }
+
+       ast_str_set(&query, 0, "${CURL(%s/require,", url);
+
+       while ((elm = va_arg(ap, char *))) {
+               type = va_arg(ap, require_type);
+               size = va_arg(ap, int);
+               ast_uri_encode(elm, field, sizeof(field), EncodeSpecialChars);
+               ast_str_append(&query, 0, "%s=%s%%3A%d", field,
+                       type == RQ_CHAR ? "char" :
+                       type == RQ_INTEGER1 ? "integer1" :
+                       type == RQ_UINTEGER1 ? "uinteger1" :
+                       type == RQ_INTEGER2 ? "integer2" :
+                       type == RQ_UINTEGER2 ? "uinteger2" :
+                       type == RQ_INTEGER3 ? "integer3" :
+                       type == RQ_UINTEGER3 ? "uinteger3" :
+                       type == RQ_INTEGER4 ? "integer4" :
+                       type == RQ_UINTEGER4 ? "uinteger4" :
+                       type == RQ_INTEGER8 ? "integer8" :
+                       type == RQ_UINTEGER8 ? "uinteger8" :
+                       type == RQ_DATE ? "date" :
+                       type == RQ_DATETIME ? "datetime" :
+                       type == RQ_FLOAT ? "float" :
+                       "unknown", size);
+       }
+       va_end(ap);
 
-static struct ast_config *config_curl(const char *url, const char *unused, const char *file, struct ast_config *cfg, struct ast_flags flags, const char *sugg_incl)
+       ast_str_append(&query, 0, ")}");
+       pbx_substitute_variables_helper(NULL, ast_str_buffer(query), buffer, sizeof(buffer));
+       return atoi(buffer);
+}
+
+static struct ast_config *config_curl(const char *url, const char *unused, const char *file, struct ast_config *cfg, struct ast_flags flags, const char *sugg_incl, const char *who_asked)
 {
        struct ast_str *query;
        char buf1[200];
        char *stringp, *line, *pair, *key;
-       int EncodeSpecialChars = 1, last_cat_metric = -1, cat_metric = -1;
+       const int EncodeSpecialChars = 1, bufsize = 256000;
+       int last_cat_metric = -1, cat_metric = -1;
        struct ast_category *cat=NULL;
        char *buffer, *cur_cat = "";
        char *category = "", *var_name = "", *var_val = "";
@@ -421,7 +535,7 @@ static struct ast_config *config_curl(const char *url, const char *unused, const
        if (!(query = ast_str_create(1000)))
                return NULL;
 
-       if (!(buffer = ast_malloc(256000))) {
+       if (!(buffer = ast_malloc(bufsize))) {
                ast_free(query);
                return NULL;
        }
@@ -430,7 +544,7 @@ static struct ast_config *config_curl(const char *url, const char *unused, const
        ast_str_set(&query, 0, "${CURL(%s/static?file=%s)}", url, buf1);
 
        /* Do the CURL query */
-       pbx_substitute_variables_helper(NULL, query->str, buffer, sizeof(buffer));
+       pbx_substitute_variables_helper(NULL, ast_str_buffer(query), buffer, bufsize);
 
        /* Line oriented output */
        stringp = buffer;
@@ -457,7 +571,7 @@ static struct ast_config *config_curl(const char *url, const char *unused, const
                }
 
                if (!strcmp(var_name, "#include")) {
-                       if (!ast_config_internal_load(var_val, cfg, loader_flags, ""))
+                       if (!ast_config_internal_load(var_val, cfg, loader_flags, "", who_asked))
                                return NULL;
                }
 
@@ -483,18 +597,27 @@ static struct ast_config_engine curl_engine = {
        .realtime_multi_func = realtime_multi_curl,
        .store_func = store_curl,
        .destroy_func = destroy_curl,
-       .update_func = update_curl
+       .update_func = update_curl,
+       .update2_func = update2_curl,
+       .require_func = require_curl,
 };
 
-static int unload_module (void)
+static int unload_module(void)
 {
        ast_config_engine_deregister(&curl_engine);
        ast_verb(1, "res_config_curl unloaded.\n");
        return 0;
 }
 
-static int load_module (void)
+static int load_module(void)
 {
+       if (!ast_module_check("res_curl.so")) {
+               if (ast_load_resource("res_curl.so") != AST_MODULE_LOAD_SUCCESS) {
+                       ast_log(LOG_ERROR, "Cannot load res_curl, so res_config_curl cannot be loaded\n");
+                       return AST_MODULE_LOAD_DECLINE;
+               }
+       }
+
        ast_config_engine_register(&curl_engine);
        ast_verb(1, "res_config_curl loaded.\n");
        return 0;