+ model->size was not set.

+ Partially incorporated Pull Request #24.
chokkan · Jan 24, 2016 · 2d5d3a7 · 2d5d3a7
1 parent 5554ee4
commit 2d5d3a7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/lib/crf/src/crf1d_model.c b/lib/crf/src/crf1d_model.c
@@ -713,7 +713,12 @@ static crf1dm_t* crf1dm_new_impl(uint8_t* buffer_orig, const uint8_t* buffer, ui
 
     model->buffer_orig = buffer_orig;
     model->buffer = buffer;
+    model->size = size;
 
+    if (model->size <= sizeof(header_t)) {
+      goto error_exit;
+    }
+
     header = (header_t*)calloc(1, sizeof(header_t));
     if (header == NULL) {
         goto error_exit;