config: fix possible access of non-allocated memory

[gregoa/bti.git] / config.c

diff --git a/config.c b/config.c
index 7315c6962fe9db53cbaeebfe706e538bd1df1ac8..42a038422e0e7231d6bf42e9345610a18bf428d9 100644 (file)
--- a/config.c
+++ b/config.c
@@ -140,7 +140,7 @@ static int get_key(struct session *session, char *line, char **key, char **value
                 * the value.
                 */
        }
                 * the value.
                 */
        }

-       printf("%s = %s\n", *key, *value);
+       /* printf("%s = %s\n", *key, *value); */

        return 0;
 }
 
        return 0;
 }
 


@@ -309,7 +309,10 @@ static void process_line(struct session *session, char *key, char *value)
                        break;
 
                if (strncasecmp(item->key, key, strlen(item->key)) == 0) {
                        break;
 
                if (strncasecmp(item->key, key, strlen(item->key)) == 0) {

-                       printf("calling %p, for key = '%s' and value = '%s'\n", item->callback, key, value);
+                       /*
+                        * printf("calling %p, for key = '%s' and value = * '%s'\n",
+                        *        item->callback, key, value);
+                        */

                        result = item->callback(session, value);
                        if (!result)
                                return;
                        result = item->callback(session, value);
                        if (!result)
                                return;


@@ -324,6 +327,7 @@ void bti_parse_configfile(struct session *session)
        char *line = NULL;
        char *key = NULL;
        char *value = NULL;
        char *line = NULL;
        char *key = NULL;
        char *value = NULL;

+       char *hashmarker;

        size_t len = 0;
        ssize_t n;
        char *c;
        size_t len = 0;
        ssize_t n;
        char *c;


@@ -341,8 +345,20 @@ void bti_parse_configfile(struct session *session)
                if (line[n - 1] == '\n')
                        line[n - 1] = '\0';
 
                if (line[n - 1] == '\n')
                        line[n - 1] = '\0';
 

-               /* '#' is comment markers, like bash style */
-               *strchrnul(line, '#') = '\0';
+               /*
+                * '#' is comment markers, like bash style but it is a valid
+                * character in some fields, so only treat it as a comment
+                * marker if it occurs at the beginning of the line, or after
+                * whitespace
+                */
+               hashmarker = strchrnul(line, '#');
+               if (line == hashmarker)
+                       line[0] = '\0';
+               else {
+                       --hashmarker;
+                       if (isblank(hashmarker[0]))
+                               hashmarker[0] = '\0';
+               }

                c = line;
                while (isspace(*c))
                        c++;
                c = line;
                while (isspace(*c))
                        c++;