Add Null check for PyUnicode_AsUTF8 69/248669/1
authorjay.ho.park <jay.ho.park@samsung.com>
Wed, 2 Dec 2020 02:41:30 +0000 (11:41 +0900)
committerjay.ho.park <jay.ho.park@samsung.com>
Wed, 2 Dec 2020 02:41:30 +0000 (11:41 +0900)
Change-Id: I444781da9ec89f4b6b04c1e74e0a1a0d0df8395c

service/src/service.c

index 45102b5..743df37 100755 (executable)
@@ -640,7 +640,15 @@ char* nltk_get_string_from_element(PyObject* elm)
         return ch;
     }
     memset(ch, 0, BUF_LEN_256);
-    strncpy(ch, PyUnicode_AsUTF8(elm), BUF_LEN_256-1);
+
+    const char *tmp_str = PyUnicode_AsUTF8(elm);
+    if (tmp_str == NULL) {
+        PERR("failed to get char from PyObject");
+        free(ch);
+        return NULL;
+    }
+
+    strncpy(ch, tmp_str, BUF_LEN_256-1);
     return ch;
 }