Set umask for temp file generation (#2637)

Set umask and recover after generate temporary file to avoid security issue

Signed-off-by: Hyeongseok Oh <hseok82.oh@samsung.com>

diff --git a/runtimes/tests/neural_networks_test/TestValidation.cpp b/runtimes/tests/neural_networks_test/TestValidation.cpp
index 3061f96..e4f287a 100644 (file)
--- a/runtimes/tests/neural_networks_test/TestValidation.cpp
+++ b/runtimes/tests/neural_networks_test/TestValidation.cpp
@@ -33,7 +33,12 @@ static int shmem_create_region(size_t size)
 {
     char temp[PATH_MAX];
     snprintf(temp, sizeof(temp), "/tmp/nn-shmem-%d-%d-XXXXXXXXX", getpid(), shmem_num++);
+
+    // Set umask and recover after generate temporary file to avoid security issue
+    mode_t umaskPrev = umask(S_IRUSR|S_IWUSR);
     int fd = mkstemp(temp);
+    umask(umaskPrev);
+
     if (fd == -1) return -1;
 
     unlink(temp);