zero out after realloc

krunal · Sep 17, 2012 · a5e4253 · a5e4253
1 parent 7817974
commit a5e4253
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ext/rblineprof.c b/ext/rblineprof.c
@@ -1,6 +1,7 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <stdbool.h>
+#include <string.h>
 
 #include <ruby.h>
 #include <node.h>
@@ -91,8 +92,10 @@ profiler_hook(rb_event_t event, NODE *node, VALUE self, ID mid, VALUE klass)
 
       /* grow the per-line array if necessary */
       if (line >= sourcefile->nlines) {
+        long prev_nlines = sourcefile->nlines;
         sourcefile->nlines = line + 100;
         sourcefile->lines = realloc(sourcefile->lines, sizeof(uint64_t) * sourcefile->nlines);
+        memset(sourcefile->lines + prev_nlines, 0, sizeof(uint64_t) * (sourcefile->nlines - prev_nlines));
       }
 
       /* record the sample */