provide an alternate getloadavg implementation and a fallback for systems that don...
[asterisk/asterisk.git] / utils.c
diff --git a/utils.c b/utils.c
index d33242f..054258c 100755 (executable)
--- a/utils.c
+++ b/utils.c
  * at the top of the source tree.
  */
 
-/*
+/*! \file
  *
- * Utility functions
+ * \brief Utility functions
  *
+ * \note These are important for portability and security,
+ * so please use them in favour of other routines.
+ * Please consult the CODING GUIDELINES for more information.
  */
 
 #include <ctype.h>
@@ -31,7 +34,6 @@
 #include <sys/socket.h>
 #include <netinet/in.h>
 #include <arpa/inet.h>
-#include <stdarg.h>
 
 #include "asterisk.h"
 
@@ -164,13 +166,14 @@ static int gethostbyname_r (const char *name, struct hostent *ret, char *buf,
 
 #endif
 
-/* Re-entrant (thread safe) version of gethostbyname that replaces the 
+/*! \brief Re-entrant (thread safe) version of gethostbyname that replaces the 
    standard gethostbyname (which is not thread safe)
 */
 struct hostent *ast_gethostbyname(const char *host, struct ast_hostent *hp)
 {
        int res;
        int herrno;
+       int dots=0;
        const char *s;
        struct hostent *result = NULL;
        /* Although it is perfectly legitimate to lookup a pure integer, for
@@ -180,12 +183,24 @@ struct hostent *ast_gethostbyname(const char *host, struct ast_hostent *hp)
        s = host;
        res = 0;
        while(s && *s) {
-               if (!isdigit(*s))
+               if (*s == '.')
+                       dots++;
+               else if (!isdigit(*s))
                        break;
                s++;
        }
-       if (!s || !*s)
+       if (!s || !*s) {
+               /* Forge a reply for IP's to avoid octal IP's being interpreted as octal */
+               if (dots != 3)
+                       return NULL;
+               memset(hp, 0, sizeof(struct ast_hostent));
+               hp->hp.h_addr_list = (void *) hp->buf;
+               hp->hp.h_addr = hp->buf + sizeof(void *);
+               if (inet_pton(AF_INET, host, hp->hp.h_addr) > 0)
+                       return &hp->hp;
                return NULL;
+               
+       }
 #ifdef SOLARIS
        result = gethostbyname_r(host, &hp->hp, hp->buf, sizeof(hp->buf), &herrno);
 
@@ -201,9 +216,6 @@ struct hostent *ast_gethostbyname(const char *host, struct ast_hostent *hp)
 }
 
 
-/* This is a regression test for recursive mutexes.
-   test_for_thread_safety() will return 0 if recursive mutex locks are
-   working properly, and non-zero if they are not working properly. */
 
 AST_MUTEX_DEFINE_STATIC(test_lock);
 AST_MUTEX_DEFINE_STATIC(test_lock2);
@@ -211,6 +223,9 @@ static pthread_t test_thread;
 static int lock_count = 0;
 static int test_errors = 0;
 
+/*! \brief This is a regression test for recursive mutexes.
+   test_for_thread_safety() will return 0 if recursive mutex locks are
+   working properly, and non-zero if they are not working properly. */
 static void *test_thread_body(void *data) 
 { 
        ast_mutex_lock(&test_lock);
@@ -262,7 +277,7 @@ int test_for_thread_safety(void)
        return(test_errors);          /* return 0 on success. */
 }
 
-/*--- ast_md5_hash: Produce 16 char MD5 hash of value. ---*/
+/*! \brief ast_md5_hash: Produce 16 char MD5 hash of value. ---*/
 void ast_md5_hash(char *output, char *input)
 {
        struct MD5Context md5;
@@ -278,7 +293,7 @@ void ast_md5_hash(char *output, char *input)
                ptr += sprintf(ptr, "%2.2x", digest[x]);
 }
 
-int ast_base64decode(unsigned char *dst, char *src, int max)
+int ast_base64decode(unsigned char *dst, const char *src, int max)
 {
        int cnt = 0;
        unsigned int byte = 0;
@@ -316,7 +331,7 @@ int ast_base64decode(unsigned char *dst, char *src, int max)
        return cnt;
 }
 
-int ast_base64encode(char *dst, unsigned char *src, int srclen, int max)
+int ast_base64encode(char *dst, const unsigned char *src, int srclen, int max)
 {
        int cnt = 0;
        unsigned int byte = 0;
@@ -393,7 +408,7 @@ static void base64_init(void)
 #endif
 }
 
-/*--- ast_uri_encode: Turn text string to URI-encoded %XX version ---*/
+/*! \brief  ast_uri_encode: Turn text string to URI-encoded %XX version ---*/
 /*     At this point, we're converting from ISO-8859-x (8-bit), not UTF8
        as in the SIP protocol spec 
        If doreserved == 1 we will convert reserved characters also.
@@ -435,7 +450,7 @@ char *ast_uri_encode(char *string, char *outbuf, int buflen, int doreserved)
        return outbuf;
 }
 
-/*--- ast_uri_decode: Decode SIP URI, URN, URL (overwrite the string)  ---*/
+/*! \brief  ast_uri_decode: Decode SIP URI, URN, URL (overwrite the string)  ---*/
 void ast_uri_decode(char *s) 
 {
        char *o;
@@ -452,7 +467,7 @@ void ast_uri_decode(char *s)
        *o = '\0';
 }
 
-/*--- ast_inet_ntoa: Recursive thread safe replacement of inet_ntoa */
+/*! \brief  ast_inet_ntoa: Recursive thread safe replacement of inet_ntoa */
 const char *ast_inet_ntoa(char *buf, int bufsiz, struct in_addr ia)
 {
        return inet_ntop(AF_INET, &ia, buf, bufsiz);
@@ -466,7 +481,7 @@ int ast_utils_init(void)
 
 #ifndef __linux__
 #undef pthread_create /* For ast_pthread_create function only */
-#endif /* ! LINUX */
+#endif /* !__linux__ */
 
 int ast_pthread_create_stack(pthread_t *thread, pthread_attr_t *attr, void *(*start_routine)(void *), void *data, size_t stacksize)
 {
@@ -475,6 +490,19 @@ int ast_pthread_create_stack(pthread_t *thread, pthread_attr_t *attr, void *(*st
                pthread_attr_init(&lattr);
                attr = &lattr;
        }
+#ifdef __linux__
+       /* On Linux, pthread_attr_init() defaults to PTHREAD_EXPLICIT_SCHED,
+          which is kind of useless. Change this here to
+          PTHREAD_INHERIT_SCHED; that way the -p option to set realtime
+          priority will propagate down to new threads by default.
+          This does mean that callers cannot set a different priority using
+          PTHREAD_EXPLICIT_SCHED in the attr argument; instead they must set
+          the priority afterwards with pthread_setschedparam(). */
+       errno = pthread_attr_setinheritsched(attr, PTHREAD_INHERIT_SCHED);
+       if (errno)
+               ast_log(LOG_WARNING, "pthread_attr_setinheritsched returned non-zero: %s\n", strerror(errno));
+#endif
+
        if (!stacksize)
                stacksize = AST_STACKSIZE;
        errno = pthread_attr_setstacksize(attr, stacksize);
@@ -509,17 +537,14 @@ char *ast_strip_quoted(char *s, const char *beg_quotes, const char *end_quotes)
        return s;
 }
 
-int ast_build_string(char **buffer, size_t *space, const char *fmt, ...)
+int ast_build_string_va(char **buffer, size_t *space, const char *fmt, va_list ap)
 {
-       va_list ap;
        int result;
 
        if (!buffer || !*buffer || !space || !*space)
                return -1;
 
-       va_start(ap, fmt);
        result = vsnprintf(*buffer, *space, fmt, ap);
-       va_end(ap);
 
        if (result < 0)
                return -1;
@@ -531,9 +556,21 @@ int ast_build_string(char **buffer, size_t *space, const char *fmt, ...)
        return 0;
 }
 
+int ast_build_string(char **buffer, size_t *space, const char *fmt, ...)
+{
+       va_list ap;
+       int result;
+
+       va_start(ap, fmt);
+       result = ast_build_string_va(buffer, space, fmt, ap);
+       va_end(ap);
+
+       return result;
+}
+
 int ast_true(const char *s)
 {
-       if (!s || ast_strlen_zero(s))
+       if (ast_strlen_zero(s))
                return 0;
 
        /* Determine if this is a true value */
@@ -550,7 +587,7 @@ int ast_true(const char *s)
 
 int ast_false(const char *s)
 {
-       if (!s || ast_strlen_zero(s))
+       if (ast_strlen_zero(s))
                return 0;
 
        /* Determine if this is a false value */
@@ -654,7 +691,7 @@ char *strcasestr(const char *haystack, const char *needle)
                return NULL;
        }
 }
-#endif
+#endif /* !HAVE_STRCASESTR */
 
 #ifndef HAVE_STRNLEN
 size_t strnlen(const char *s, size_t n)
@@ -667,7 +704,7 @@ size_t strnlen(const char *s, size_t n)
 
        return len;
 }
-#endif
+#endif /* !HAVE_STRNLEN */
 
 #if !defined(HAVE_STRNDUP) && !defined(__AST_DEBUG_MALLOC)
 char *strndup(const char *s, size_t n)
@@ -681,7 +718,7 @@ char *strndup(const char *s, size_t n)
        new[len] = '\0';
        return memcpy(new, s, len);
 }
-#endif
+#endif /* !defined(HAVE_STRNDUP) && !defined(__AST_DEBUG_MALLOC) */
 
 #if !defined(HAVE_VASPRINTF) && !defined(__AST_DEBUG_MALLOC)
 int vasprintf(char **strp, const char *fmt, va_list ap)
@@ -701,7 +738,7 @@ int vasprintf(char **strp, const char *fmt, va_list ap)
 
        return size;
 }
-#endif
+#endif /* !defined(HAVE_VASPRINTF) && !defined(__AST_DEBUG_MALLOC) */
 
 #ifndef HAVE_STRTOQ
 #define LONG_MIN        (-9223372036854775807L-1L)
@@ -798,4 +835,66 @@ uint64_t strtoq(const char *nptr, char **endptr, int base)
                 *((const char **)endptr) = any ? s - 1 : nptr;
         return acc;
 }
-#endif
+#endif /* !HAVE_STRTOQ */
+
+#if (!defined(getloadavg))
+#ifdef linux
+/* Alternative method of getting load avg on Linux only */
+int getloadavg(double *list, int nelem)
+{
+       FILE *LOADAVG;
+       double avg[3] = { 0.0, 0.0, 0.0 };
+       int i, res = -1;
+
+       if ((LOADAVG = fopen("/proc/loadavg", "r"))) {
+               fscanf(LOADAVG, "%lf %lf %lf", &avg[0], &avg[1], &avg[2]);
+               res = 0;
+               fclose(LOADAVG);
+       }
+
+       for (i = 0; (i < nelem) && (i < 3); i++) {
+               list[i] = avg[i];
+       }
+
+       return res;
+}
+#else /* !linux */
+/* Return something that won't cancel the call, but still return -1, in case
+ * we correct the implementation to check return value */
+int getloadavg(double *list, int nelem)
+{
+       int i;
+
+       for (i = 0; i < nelem; i++) {
+               list[i] = 0.1;
+       }
+       return -1;
+}
+#endif /* linux */
+#endif /* !defined(getloadavg) */
+
+char *ast_process_quotes_and_slashes(char *start, char find, char replace_with)
+{
+       char *dataPut = start;
+       int inEscape = 0;
+       int inQuotes = 0;
+
+       for (; *start; start++) {
+               if (inEscape) {
+                       *dataPut++ = *start;       /* Always goes verbatim */
+                       inEscape = 0;
+               } else {
+                       if (*start == '\\') {
+                               inEscape = 1;      /* Do not copy \ into the data */
+                       } else if (*start == '\'') {
+                               inQuotes = 1-inQuotes;   /* Do not copy ' into the data */
+                       } else {
+                               /* Replace , with |, unless in quotes */
+                               *dataPut++ = inQuotes ? *start : ((*start==find) ? replace_with : *start);
+                       }
+               }
+       }
+       if (start != dataPut)
+               *dataPut = 0;
+       return dataPut;
+}