From fb3c6c5121df975cb40dd861abf824e8df5ed138 Mon Sep 17 00:00:00 2001 From: "jay.ho.park" Date: Wed, 2 Dec 2020 11:25:51 +0900 Subject: [PATCH] Free allocated memory while using 'nltk_get_string_from_element' function Change-Id: Ic4e1ed18923369918110443620a55da7ebe10d31 --- tests/src/nlp_service_unittests.cpp | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/tests/src/nlp_service_unittests.cpp b/tests/src/nlp_service_unittests.cpp index 71377db..74e619c 100644 --- a/tests/src/nlp_service_unittests.cpp +++ b/tests/src/nlp_service_unittests.cpp @@ -84,6 +84,11 @@ TEST_F(NlpServiceTest, utc_nlp_service_language_detect_p) char *tmp_str = nltk_get_string_from_element(ld_result); string detected_language = string(tmp_str); + if (tmp_str) { + free(tmp_str); + tmp_str = NULL; + } + Py_DECREF(ld_result); cout << "Detected language: " << detected_language.c_str() << endl; @@ -110,11 +115,21 @@ TEST_F(NlpServiceTest, utc_nlp_service_pos_tag_p) tmp_str = nltk_get_string_from_element(nltk_get_element_from_tuple_by_index(pt_elm_tuple, 1)); tags.push_back(string(tmp_str)); + if (tmp_str) { + free(tmp_str); + tmp_str = NULL; + } + cout << "tag : " << tmp_str << endl; tmp_str = nltk_get_string_from_element(nltk_get_element_from_tuple_by_index(pt_elm_tuple, 0)); tokens.push_back(string(tmp_str)); + if (tmp_str) { + free(tmp_str); + tmp_str = NULL; + } + Py_DECREF(pt_elm_tuple); } -- 2.7.4