Fix #780, restrict permissions on file create

nasa · Feb 22, 2021 · 9e64376 · 9e64376
1 parent ff4f523
commit 9e64376
Showing 1 changed file with 16 additions and 2 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"
@@ -56,9 +57,16 @@ typedef struct
 bool UtMem2BinFile(const void *Memory, const char *Filename, uint32 Length)
 {
     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);
@@ -98,10 +106,16 @@ bool UtMem2HexFile(const void *Memory, const char *Filename, uint32 Length)
     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)
         {
             fprintf(fp, "   %06lX: ", (unsigned long)i);