Do the (expensive) LOCAL_USER_ADD only after we are sure
[asterisk/asterisk.git] / funcs / func_enum.c
index 2dd982e..87f321e 100644 (file)
@@ -50,8 +50,6 @@ ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
 
  static char *synopsis = "Syntax: ENUMLOOKUP(number[|Method-type[|options[|record#[|zone-suffix]]]])\n";
 
-STANDARD_LOCAL_USER;
-
 LOCAL_USER_DECL;
 
 static int function_enum(struct ast_channel *chan, char *cmd, char *data,
@@ -66,7 +64,7 @@ static int function_enum(struct ast_channel *chan, char *cmd, char *data,
        );
        int res = 0;
        char tech[80];
-       char dest[80] = "";
+       char dest[256] = "", tmp[2] = "", num[AST_MAX_EXTENSION] = "";
        struct localuser *u;
        char *s, *p;
 
@@ -84,31 +82,36 @@ static int function_enum(struct ast_channel *chan, char *cmd, char *data,
                return -1;
        }
 
+       LOCAL_USER_ADD(u);
+
        ast_copy_string(tech, args.tech ? args.tech : "sip", sizeof(tech));
 
+       if (!args.zone)
+               args.zone = "e164.arpa";
+
        if (!args.options)
                args.options = "1";
 
        /* strip any '-' signs from number */
-       for (p = args.number, s = p; *s; *s++) {
-               if (*s != '-')
-                       *p++ = *s;
-       }
-       *p = '\0';
+       for (s = p = args.number; *s; s++) {
+               if (*s != '-') {
+                       snprintf(tmp, sizeof(tmp), "%c", *s);
+                       strncat(num, tmp, sizeof(num));
+               }
 
-       LOCAL_USER_ADD(u);
+       }
 
-       res = ast_get_enum(chan, p, dest, sizeof(dest), tech, sizeof(tech), args.zone,
+       res = ast_get_enum(chan, num, dest, sizeof(dest), tech, sizeof(tech), args.zone,
                           args.options);
 
-       LOCAL_USER_REMOVE(u);
-
        p = strchr(dest, ':');
        if (p && strcasecmp(tech, "ALL"))
                ast_copy_string(buf, p + 1, len);
        else
                ast_copy_string(buf, dest, len);
 
+       LOCAL_USER_REMOVE(u);
+
        return 0;
 }
 
@@ -123,7 +126,7 @@ static struct ast_custom_function enum_function = {
                "Option 'c' returns an integer count of the number of NAPTRs of a certain RR type.\n"
                "Combination of 'c' and Method-type of 'ALL' will return a count of all NAPTRs for the record.\n"
                "Defaults are: Method-type=sip, no options, record=1, zone-suffix=e164.arpa\n\n"
-               "For more information, see README.enum",
+               "For more information, see doc/enum.txt",
        .read = function_enum,
 };
 
@@ -138,14 +141,14 @@ static int function_txtcidname(struct ast_channel *chan, char *cmd,
 
        buf[0] = '\0';
 
-       LOCAL_USER_ADD(u);
 
        if (ast_strlen_zero(data)) {
                ast_log(LOG_WARNING, "TXTCIDNAME requires an argument (number)\n");
-               LOCAL_USER_REMOVE(u);
                return -1;
        }
 
+       LOCAL_USER_ADD(u);
+
        res = ast_get_txt(chan, data, dest, sizeof(dest), tech, sizeof(tech), txt,
                          sizeof(txt));