123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131 |
- #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(RFMEM|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 */
- }
|