Fix #780, restrict permissions on file create

nasa · Feb 22, 2021 · 32a5f5f · 32a5f5f
1 parent ff4f523
commit 32a5f5f
Showing 1 changed file with 18 additions and 4 deletions.
diff --git a/ut_assert/src/uttools.c b/ut_assert/src/uttools.c
@@ -33,6 +33,7 @@
 #include <errno.h>
 #include <string.h>
 #include <ctype.h>
+#include <sys/stat.h>
 
 #include "common_types.h"
 #include "utassert.h"
@@ -55,10 +56,17 @@ typedef struct
 
 bool UtMem2BinFile(const void *Memory, const char *Filename, uint32 Length)
 {
-    FILE *fp;
+    FILE *      fp;
+    struct stat dststat;
 
     if ((fp = fopen(Filename, "w")))
     {
+        if (stat(Filename, &dststat) == 0)
+        {
+            chmod(Filename, dststat.st_mode & ~(S_IRGRP | S_IWGRP | S_IXGRP | S_IROTH | S_IWOTH | S_IXOTH));
+            stat(Filename, &dststat);
+        }
+
         fwrite(Memory, Length, 1, fp);
         fclose(fp);
         return (true);
@@ -95,12 +103,18 @@ bool UtBinFile2Mem(void *Memory, const char *Filename, uint32 Length)
 
 bool UtMem2HexFile(const void *Memory, const char *Filename, uint32 Length)
 {
-    FILE * fp;
-    uint32 i;
-    uint32 j;
+    FILE *      fp;
+    uint32      i;
+    uint32      j;
+    struct stat dststat;
 
     if ((fp = fopen(Filename, "w")))
     {
+        if (stat(Filename, &dststat) == 0)
+        {
+            chmod(Filename, dststat.st_mode & ~(S_IRGRP | S_IWGRP | S_IXGRP | S_IROTH | S_IWOTH | S_IXOTH));
+            stat(Filename, &dststat);
+        }
 
         for (i = 0; i < Length; i += 16)
         {