Memset improvement inspired by Michael Rayment <mtr@mun.ca>

[infodrom/cgilib] / cgi.c

diff --git a/cgi.c b/cgi.c
index e390b37..ad555c2 100644 (file)
--- a/cgi.c
+++ b/cgi.c
@@ -54,7 +54,7 @@ int cgiSetHeader (const char *name, const char *value)
        return 0;
 
     if (cgiHeaderString) {
-       len = (strlen (cgiHeaderString) + cp-name + vp-value + 4) * sizeof (char);
+       len = (strlen (cgiHeaderString) + cp-name + vp-value + 5) * sizeof (char);
        if ((pivot = (char *)realloc (cgiHeaderString,len)) == NULL)
            return 0;
        cgiHeaderString = pivot;
@@ -66,7 +66,7 @@ int cgiSetHeader (const char *name, const char *value)
        pivot = cgiHeaderString;
        *pivot = '\0';
     }
-    memset(pivot+1, 0, (cp-name + vp-value + 4));
+    memset(pivot, 0, (cp-name + vp-value + 5) * sizeof (char));
 
     strncpy (pivot, name, cp-name);
     strncat (pivot, ": ", 2);