From ffe03c3d061bde3abe13be28f51574b9454cfc85 Mon Sep 17 00:00:00 2001 From: xuyongfei Date: Sat, 19 Sep 2020 09:56:13 +0800 Subject: [PATCH] o serving: codex --- serving/CMakeLists.txt | 2 +- serving/core/http_process.cc | 10 +++++++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/serving/CMakeLists.txt b/serving/CMakeLists.txt index dd9421fe4b..0b9433bcd9 100644 --- a/serving/CMakeLists.txt +++ b/serving/CMakeLists.txt @@ -116,7 +116,7 @@ if (ENABLE_ACL) ${ACL_LIB_DIR}/lib64/libacl_retr.so ${ACL_LIB_DIR}/lib64/libacl_cblas.so ${ACL_LIB_DIR}/lib64/libacl_dvpp.so) target_link_libraries(ms_serving ${ALC_LIB_SO}) - target_link_libraries(ms_serving jpeg_turbo::jpeg) + target_link_libraries(ms_serving jpeg_turbo::jpeg securec) else () target_link_libraries(ms_serving inference mindspore_gvar) endif () diff --git a/serving/core/http_process.cc b/serving/core/http_process.cc index c53dc40808..f8a082d2b2 100644 --- a/serving/core/http_process.cc +++ b/serving/core/http_process.cc @@ -52,7 +52,15 @@ Status GetPostMessage(struct evhttp_request *const req, std::string *const buf) return status; } else { buf->resize(post_size); - memcpy_s(buf->data(), post_size, evbuffer_pullup(req->input_buffer, -1), post_size); + auto src_data = evbuffer_pullup(req->input_buffer, -1); + if (src_data == nullptr) { + ERROR_INFER_STATUS(status, FAILED, "get http message failed."); + return status; + } + if (memcpy_s(buf->data(), post_size, src_data, post_size) != EOK) { + ERROR_INFER_STATUS(status, FAILED, "copy http message failed."); + return status; + } return status; } }