Security-57031.1.35.tar.gz

[apple/security.git] / SecurityTests / cspxutils / genErrorStrings / fileIo.c

diff --git a/SecurityTests/cspxutils/genErrorStrings/fileIo.c b/SecurityTests/cspxutils/genErrorStrings/fileIo.c
new file mode 100644 (file)
index 0000000..3f666ac
--- /dev/null
+++ b/SecurityTests/cspxutils/genErrorStrings/fileIo.c
@@ -0,0 +1,89 @@
+#include <unistd.h>
+#include <fcntl.h>
+#include <errno.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include "fileIo.h"
+
+int writeFile(
+       const char                      *fileName,
+       const unsigned char     *bytes,
+       unsigned                        numBytes)
+{
+       int             rtn;
+       int     fd;
+       
+       fd = open(fileName, O_RDWR | O_CREAT | O_TRUNC, 0600);
+       if(fd <= 0) {
+               return errno;
+       }
+       rtn = lseek(fd, 0, SEEK_SET);
+       if(rtn < 0) {
+               return errno;
+       }
+       rtn = write(fd, bytes, (size_t)numBytes);
+       if(rtn != (int)numBytes) {
+               if(rtn >= 0) {
+                       printf("writeFile: short write\n");
+               }
+               rtn = EIO;
+       }
+       else {
+               rtn = 0;
+       }
+       close(fd);
+       return rtn;
+}
+       
+/*
+ * Read entire file. 
+ */
+int readFile(
+       const char              *fileName,
+       unsigned char   **bytes,                // mallocd and returned
+       unsigned                *numBytes)              // returned
+{
+       int rtn;
+       int fd;
+       char *buf;
+       struct stat     sb;
+       unsigned size;
+       
+       *numBytes = 0;
+       *bytes = NULL;
+       fd = open(fileName, O_RDONLY, 0);
+       if(fd <= 0) {
+               return errno;
+       }
+       rtn = fstat(fd, &sb);
+       if(rtn) {
+               goto errOut;
+       }
+       size = sb.st_size;
+       buf = (char *)malloc(size);
+       if(buf == NULL) {
+               rtn = ENOMEM;
+               goto errOut;
+       }
+       rtn = lseek(fd, 0, SEEK_SET);
+       if(rtn < 0) {
+               goto errOut;
+       }
+       rtn = read(fd, buf, (size_t)size);
+       if(rtn != (int)size) {
+               if(rtn >= 0) {
+                       printf("readFile: short read\n");
+               }
+               rtn = EIO;
+       }
+       else {
+               rtn = 0;
+               *bytes = (unsigned char *)buf;
+               *numBytes = size;
+       }
+errOut:
+       close(fd);
+       return rtn;
+}