Check NULL pointer before unref

2 files changed, 9 insertions, 0 deletions

diff --git a/l4array.c b/l4array.c
index 99e0aa4..a4a5243 100644
--- a/l4array.c
+++ b/l4array.c
@@ -103,6 +103,9 @@ void* lbs_array_ref_generic (void* array_generic) {
 }
 
 void lbs_array_unref_generic (void* array_generic) {
+    if (array_generic == NULL) {
+        return;
+    }
     LbsArray* array = LBS_ARRAY (array_generic);
     array->ref_count--;
     if (array->ref_count <= 0) {
@@ -111,6 +114,9 @@ void lbs_array_unref_generic (void* array_generic) {
 }
 
 void lbs_array_free_generic (void* array_generic) {
+    if (array_generic == NULL) {
+        return;
+    }
     LbsArray* array = LBS_ARRAY (array_generic);
     if (array->free_func != NULL) {
         int i = 0;
diff --git a/l4array2.c b/l4array2.c
index d1f1b99..f066cf4 100644
--- a/l4array2.c
+++ b/l4array2.c
@@ -37,6 +37,9 @@ void* lbs_array2_ref_generic (void* array2_generic) {
 }
 
 void lbs_array2_unref_generic (void* array2_generic) {
+    if (array2_generic == NULL) {
+        return;
+    }
     LbsArray2* array2 = LBS_ARRAY2 (array2_generic);
     array2->ref_count--;
     if (array2->ref_count <= 0) {