Initial for executing filedescriptor

1 files changed, 12 insertions, 1 deletions

diff --git a/newbatch.c b/newbatch.c
index 6b27234..0bac7ee 100644
--- a/newbatch.c
+++ b/newbatch.c
@@ -18,6 +18,7 @@ main(int argc, char **argv)
 	char *timestr = NULL, *queuedir = NULL, *tmpfile = NULL, *newfile = NULL;
 	char buf[1024], *tmp;
 	int fd, l, m;
+	int perm = 0640;
 
 	mypid = getpid();
 	gettimeofday(&tv, NULL);
@@ -35,12 +36,19 @@ main(int argc, char **argv)
 	free(timestr);
 	free(queuedir);
 
-	fd = open(tmpfile, O_WRONLY | O_CREAT | O_EXCL, 0660);
+	if (getenv("NEWBATCH_EXEC") != NULL) {
+		perm = 0750;
+	}
+
+	fd = open(tmpfile, O_WRONLY | O_CREAT | O_EXCL, perm);
 	if (fd == -1)
 		err(1, "open()");
 
 	while ((l = read(STDIN_FILENO, buf, sizeof(buf))) != 0) {
 		if (l == -1) {
+			if (errno == EINTR)
+				continue;
+
 			unlink(tmpfile);
 			err(1, NULL);
 		}
@@ -49,6 +57,9 @@ main(int argc, char **argv)
 		while (l > 0) {
 			m = write(fd, tmp, l);
 			if (m == -1) {
+				if (errno == EINTR)
+					continue;
+
 				unlink(tmpfile);
 				err(1, NULL);
 			}