Add missing check on PyUnicode_AsUTF8AndSize() possible error

python-rapidjson · Jan 7, 2018 · f5fc756 · f5fc756
1 parent 54c716c
commit f5fc756
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/rapidjson.cpp b/rapidjson.cpp
@@ -276,8 +276,12 @@ class PyReadStreamWrapper {
                 len = PyBytes_GET_SIZE(chunk);
                 buffer = PyBytes_AS_STRING(chunk);
             }
-            else
+            else {
                 buffer = PyUnicode_AsUTF8AndSize(chunk, &len);
+                if (buffer == NULL) {
+                    len = 0;
+                }
+            }
 
             if (len == 0) {
                 eof = true;