aboutsummaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/args.c32
-rw-r--r--lib/functions.c8
-rw-r--r--lib/lib.h4
3 files changed, 42 insertions, 2 deletions
diff --git a/lib/args.c b/lib/args.c
index a6a3b229..be62c7f7 100644
--- a/lib/args.c
+++ b/lib/args.c
@@ -289,3 +289,35 @@ notflag:
if (optarg<minargs) error_exit("Need %d arguments", minargs);
if (optarg>maxargs) error_exit("Max %d arguments", maxargs);
}
+
+// Loop through files listed on the command line
+
+static int dofileargs(char ***files, int fd, int iswrite)
+{
+ char *filename = *((*files)++);
+ static int flags[] = {O_RDONLY, O_CREAT|O_TRUNC, O_RDWR};
+
+ if (fd != -1) close(fd);
+
+ for (;;) {
+
+ // Are there no more files?
+ if (!*filename)
+ return (fd == -1) ? iswrite : -1;
+
+ // A filename of "-" means stdin.
+ if (*filename == '-' && !filename[1]) return 0;
+
+ fd = xcreate(filename, flags[iswrite], 0777);
+ }
+}
+
+int readfileargs(char ***files, int fd)
+{
+ return dofileargs(files, fd, 0);
+}
+
+int writefileargs(char ***files, int fd)
+{
+ return dofileargs(files, fd, 1);
+}
diff --git a/lib/functions.c b/lib/functions.c
index b42f62d3..990c34fa 100644
--- a/lib/functions.c
+++ b/lib/functions.c
@@ -141,13 +141,19 @@ void xexec(char **argv)
}
// Die unless we can open/create a file, returning file descriptor.
-int xopen(char *path, int flags, int mode)
+int xcreate(char *path, int flags, int mode)
{
int fd = open(path, flags, mode);
if (fd == -1) error_exit("No file %s\n", path);
return fd;
}
+// Die unless we can open a file, returning file descriptor.
+int xopen(char *path, int flags)
+{
+ return xcreate(path, flags, 0);
+}
+
// Die unless we can open/create a file, returning FILE *.
FILE *xfopen(char *path, char *mode)
{
diff --git a/lib/lib.h b/lib/lib.h
index 6a398c1a..cb140275 100644
--- a/lib/lib.h
+++ b/lib/lib.h
@@ -38,7 +38,8 @@ void *xstrndup(char *s, size_t n);
void *xstrdup(char *s);
char *xmsprintf(char *format, ...);
void xexec(char **argv);
-int xopen(char *path, int flags, int mode);
+int xcreate(char *path, int flags, int mode);
+int xopen(char *path, int flags);
FILE *xfopen(char *path, char *mode);
ssize_t reread(int fd, void *buf, size_t count);
ssize_t readall(int fd, void *buf, size_t count);
@@ -63,3 +64,4 @@ struct mtab_list {
struct mtab_list *getmountlist(int die);
+char *bunzipStream(int src_fd, int dst_fd);