1 files changed, 131 insertions, 0 deletions
diff --git a/src/cmd/page/util.c b/src/cmd/page/util.c
new file mode 100644
index 00000000..22832baa
--- /dev/null
+++ b/src/cmd/page/util.c
@@ -0,0 +1,131 @@
+#include <u.h>
+#include <libc.h>
+#include <draw.h>
+#include <event.h>
+#include <bio.h>
+#include "page.h"
+
+void*
+emalloc(int sz)
+{
+	void *v;
+	v = malloc(sz);
+	if(v == nil) {
+		fprint(2, "out of memory allocating %d\n", sz);
+		wexits("mem");
+	}
+	memset(v, 0, sz);
+	return v;
+}
+
+void*
+erealloc(void *v, int sz)
+{
+	v = realloc(v, sz);
+	if(v == nil) {
+		fprint(2, "out of memory allocating %d\n", sz);
+		wexits("mem");
+	}
+	return v;
+}
+
+char*
+estrdup(char *s)
+{
+	char *t;
+	if((t = strdup(s)) == nil) {
+		fprint(2, "out of memory in strdup(%.10s)\n", s);
+		wexits("mem");
+	}
+	return t;
+}
+
+int
+opentemp(char *template)
+{
+	int fd, i;
+	char *p;
+
+	p = estrdup(template);
+	fd = -1;
+	for(i=0; i<10; i++){
+		mktemp(p);
+		if(access(p, 0) < 0 && (fd=create(p, ORDWR|ORCLOSE, 0400)) >= 0)
+			break;
+		strcpy(p, template);
+	}
+	if(fd < 0){
+		fprint(2, "couldn't make temporary file\n");
+		wexits("Ecreat");
+	}
+	strcpy(template, p);
+	free(p);
+
+	return fd;
+}
+
+/*
+ * spool standard input to /tmp.
+ * we've already read the initial in bytes into ibuf.
+ */
+int
+spooltodisk(uchar *ibuf, int in, char **name)
+{
+	uchar buf[8192];
+	int fd, n;
+	char temp[40];
+
+	strcpy(temp, "/tmp/pagespoolXXXXXXXXX");
+	fd = opentemp(temp);
+	if(name)
+		*name = estrdup(temp);
+
+	if(write(fd, ibuf, in) != in){
+		fprint(2, "error writing temporary file\n");
+		wexits("write temp");
+	}
+
+	while((n = read(stdinfd, buf, sizeof buf)) > 0){
+		if(write(fd, buf, n) != n){
+			fprint(2, "error writing temporary file\n");
+			wexits("write temp0");
+		}
+	}
+	seek(fd, 0, 0);
+	return fd;
+}
+
+/*
+ * spool standard input into a pipe.
+ * we've already ready the first in bytes into ibuf
+ */
+int
+stdinpipe(uchar *ibuf, int in)
+{
+	uchar buf[8192];
+	int n;
+	int p[2];
+	if(pipe(p) < 0){
+		fprint(2, "pipe fails: %r\n");	
+		wexits("pipe");
+	}
+
+	switch(rfork(RFPROC|RFFDG)){
+	case -1:
+		fprint(2, "fork fails: %r\n");
+		wexits("fork");
+	default:
+		close(p[1]);
+		return p[0];
+	case 0:
+		break;
+	}
+
+	close(p[0]);
+	write(p[1], ibuf, in);
+	while((n = read(stdinfd, buf, sizeof buf)) > 0)
+		write(p[1], buf, n);
+
+	_exits(0);
+	return -1;	/* not reached */
+}