From 573c2bcd93403d5f7fd1797d67418780cce4f9ef Mon Sep 17 00:00:00 2001
From: Gemfield <gemfield@civilnet.cn>
Date: Thu, 21 Mar 2019 16:16:37 +0800
Subject: [PATCH] Fix crash issue during load_model (#848)

* Fix crash issue during load_model

* Fix crash issue during load_model 2nd part
---
 src/net.cpp | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/net.cpp b/src/net.cpp
index b68334463..db6fb6f92 100644
--- a/src/net.cpp
+++ b/src/net.cpp
@@ -633,6 +633,13 @@ int Net::load_model(FILE* fp)
     for (size_t i=0; i<layers.size(); i++)
     {
         Layer* layer = layers[i];
+        
+        //Here we found inconsistent content in the parameter file.
+        if (!layer){
+            fprintf(stderr, "load_model error at layer %d, parameter file has inconsistent content.\n", (int)i);
+            ret = -1;
+            break;
+        }
 
         int lret = layer->load_model(mb);
         if (lret != 0)
@@ -823,6 +830,12 @@ int Net::load_model(const unsigned char* _mem)
     {
         Layer* layer = layers[i];
 
+        //Here we found inconsistent content in the parameter file.
+        if (!layer){
+            fprintf(stderr, "load_model error at layer %d, parameter file has inconsistent content.\n", (int)i);
+            return -1;
+        }
+
         int lret = layer->load_model(mb);
         if (lret != 0)
         {