From 56acc656199eb74160d892510d71d946427281db Mon Sep 17 00:00:00 2001
From: mayang <mayang20@huawei.com>
Date: Thu, 10 Sep 2020 20:14:21 +0800
Subject: [PATCH] [MD] Replace the unsafe func 'snprintf' with a safe one
 'snprintf_s' in soft_dp_log.h

---
 .../kernels/image/soft_dvpp/utils/soft_dp_log.h     | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/mindspore/ccsrc/minddata/dataset/kernels/image/soft_dvpp/utils/soft_dp_log.h b/mindspore/ccsrc/minddata/dataset/kernels/image/soft_dvpp/utils/soft_dp_log.h
index c28b1539d2..b40d9f5e54 100644
--- a/mindspore/ccsrc/minddata/dataset/kernels/image/soft_dvpp/utils/soft_dp_log.h
+++ b/mindspore/ccsrc/minddata/dataset/kernels/image/soft_dvpp/utils/soft_dp_log.h
@@ -65,19 +65,18 @@
 
 #elif defined(USE_GLOG)
 
+#include <securec.h>
 #include <cstdio>
 #include "glog/logging.h"
 
 template <typename... Args>
 inline std::string GetFormatString(const char *format, Args... args) {
   char buf[BUFSIZ];
-  int new_len = snprintf(&buf[0], BUFSIZ, format, args...);
-  new_len++;
-  if (new_len > BUFSIZ) {
-    std::vector<char> buf2(new_len);
-    snprintf(buf2.data(), new_len, format, args...);
-    return std::string(buf2.data());
-  }
+#ifdef _WIN32
+  _snprintf_s(&buf[0], BUFSIZ, BUFSIZ - 1, format, args...);
+#else
+  snprintf_s(&buf[0], BUFSIZ, BUFSIZ - 1, format, args...);
+#endif
   return buf;
 }