[PATCH] vmsplice2: another test program

diff --git a/Makefile b/Makefile
index 5690727acd0062f9312ca9e091d66c1154d66a0f..ea113fcba251c24c8f17836c77f88aaeca2f5402 100644 (file)
--- a/Makefile
+++ b/Makefile
@@ -1,6 +1,6 @@
 CC     = gcc
 CFLAGS = -Wall -O2 -g -D_GNU_SOURCE -D_LARGEFILE_SOURCE -D_FILE_OFFSET_BITS=64
-PROGS  = ktee ktee-net splice-cp splice-in splice-out splice-net splice-test4c splice-test4s vmsplice splice-bench
+PROGS  = ktee ktee-net splice-cp splice-in splice-out splice-net splice-test4c splice-test4s vmsplice splice-bench vmsplice2
 MANS   = splice.2 tee.2 vmsplice.2
 
 all: depend $(PROGS)

diff --git a/vmsplice2.c b/vmsplice2.c
new file mode 100644 (file)
index 0000000..189846e
--- /dev/null
+++ b/vmsplice2.c
@@ -0,0 +1,78 @@
+/*
+ * Use vmsplice to fill some user memory into a pipe. vmsplice writes
+ * to stdout, so that must be a pipe.
+ */
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <limits.h>
+#include <string.h>
+#include <getopt.h>
+#include <sys/poll.h>
+#include <sys/types.h>
+
+#include "splice.h"
+
+int do_vmsplice(int fd, struct iovec *iov, unsigned long nr_vecs)
+{
+       struct pollfd pfd = { .fd = fd, .events = POLLOUT, };
+       long written;
+
+       while (nr_vecs) {
+               /*
+                * in a real app you'd be more clever with poll of course,
+                * here we are basically just blocking on output room and
+                * not using the free time for anything interesting.
+                */
+               if (poll(&pfd, 1, -1) < 0)
+                       return error("poll");
+
+               written = vmsplice(fd, iov, nr_vecs, 0);
+
+               if (written <= 0)
+                       return error("vmsplice");
+
+               while (written) {
+                       int this_len = iov->iov_len;
+
+                       if (this_len > written)
+                               this_len = written;
+
+                       iov->iov_len -= this_len;
+                       if (!iov->iov_len) {
+                               nr_vecs--;
+                               iov++;
+                       }
+
+                       written -= this_len;
+               }
+       }
+
+       return 0;
+}
+
+static int usage(char *name)
+{
+       fprintf(stderr, "%s | ...\n", name);
+       return 1;
+}
+
+int main(int argc, char *argv[])
+{
+       char h[] = "header header header header header header header header";
+       char b[] = "body body body body body body body body body body body";
+       char f[] = "footer footer footer footer footer footer footer footer";
+       struct iovec vecs[3];
+
+       vecs[0].iov_base = h;
+       vecs[0].iov_len = strlen(h);
+       vecs[1].iov_base = b;
+       vecs[1].iov_len = strlen(b);
+       vecs[2].iov_base = f;
+       vecs[2].iov_len = strlen(f);
+               
+       if (check_output_pipe())
+               return usage(argv[0]);
+
+       return do_vmsplice(STDOUT_FILENO, vecs, 3);
+}