chiark / gitweb /
No need for Makefile-etc now - we hope.
[chiark-utils.git] / backup / writebuffer.c
1 /*
2  * writebuffer.c
3  *
4  * A program for writing output to devices which don't like constant
5  * stopping and starting, such as tape drives.  writebuffer is:
6  *  Copyright (C) 1997-1998,2000-2001 Ian Jackson <ian@chiark.greenend.org.uk>
7  *
8  * writebuffer is part of chiark backup, a system for backing up GNU/Linux
9  * and other UN*X-compatible machines, as used on chiark.greenend.org.uk.
10  * chiark backup is:
11  *  Copyright (C) 1997-1998,2000-2001 Ian Jackson <ian@chiark.greenend.org.uk>
12  *  Copyright (C) 1999 Peter Maydell <pmaydell@chiark.greenend.org.uk>
13  *
14  * This is free software; you can redistribute it and/or modify
15  * it under the terms of the GNU General Public License as
16  * published by the Free Software Foundation; either version 2,
17  * or (at your option) any later version.
18  *
19  * This is distributed in the hope that it will be useful, but
20  * WITHOUT ANY WARRANTY; without even the implied warranty of
21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
22  * GNU General Public License for more details.
23  *
24  * You should have received a copy of the GNU General Public
25  * License along with this file; if not, write to the Free Software
26  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
27  *
28  */
29
30 #include <sys/time.h>
31 #include <sys/types.h>
32 #include <fcntl.h>
33 #include <stdio.h>
34 #include <errno.h>
35 #include <unistd.h>
36
37 #define BUFFER 16*1024*1024
38 #define WAITFILL ((BUFFER*3)/4)
39
40 static inline int min(int a, int b) { return a<=b ? a : b; }
41
42 static void nonblock(int fd) {
43   int r;
44   r= fcntl(fd,F_GETFL,0); if (r == -1) { perror("fcntl getfl"); exit(1); }
45   r |= O_NDELAY;
46   if (fcntl(fd,F_SETFL,r) == -1) { perror("fcntl setfl"); exit(1); }
47 }
48
49 int main(int argc, const char *const *argv) {
50   static unsigned char buf[BUFFER];
51   
52   if (argv[1]) {
53     fputs("readbuffer: no arguments allowed\n", stderr);
54     exit(-1);
55   }
56
57   unsigned char *wp, *rp;
58   int used,r,writing,seeneof;
59   fd_set readfds;
60   fd_set writefds;
61
62   used=0; wp=rp=buf; writing=0; seeneof=0;
63   nonblock(0); nonblock(1);
64   while (!seeneof || used) {
65     FD_ZERO(&readfds); if (!seeneof && used+1<BUFFER) FD_SET(0,&readfds);
66     FD_ZERO(&writefds); if (writing) FD_SET(1,&writefds);
67 /*fprintf(stderr,"used %6d writing %d seeneof %d wp %8lx rp %8lx read %d write %d\n",
68         used,writing,seeneof,(unsigned long)(wp-buf),(unsigned long)(rp-buf),
69         FD_ISSET(0,&readfds),FD_ISSET(1,&writefds));*/
70     r= select(2,&readfds,&writefds,0,0);
71 /*fprintf(stderr,"\t readable %d writeable %d\n",
72         FD_ISSET(0,&readfds),FD_ISSET(1,&writefds));*/
73     if (r == -1) {
74       if (errno == EINTR) continue;
75       perror("select"); exit(1);
76     }
77     if (FD_ISSET(1,&writefds) && !FD_ISSET(0,&readfds) && !used) {
78       writing= 0;
79       FD_CLR(1,&writefds);
80 /*fprintf(stderr,"\t buffers empty - stopping\n");*/
81     }
82     if (FD_ISSET(0,&readfds)) {
83       r= read(0,rp,min(BUFFER-1-used,buf+BUFFER-rp));
84       if (!r) {
85 /*fprintf(stderr,"\t eof detected\n");*/
86         seeneof=1; writing=1;
87       } else if (r<0) {
88         if (!(errno == EAGAIN || errno == EINTR)) { perror("read"); exit(1); }
89 fprintf(stderr,"\t read transient error\n");
90       } else {
91 /*fprintf(stderr,"\t read %d\n",r);*/
92         used+= r;
93         rp+= r;
94         if (rp == buf+BUFFER) rp=buf;
95 /*fprintf(stderr,"\t starting writes\n");*/
96       }
97       if (used > WAITFILL) writing=1;
98     }
99     if (FD_ISSET(1,&writefds) && used) {
100       r= write(1,wp,min(used,buf+BUFFER-wp));
101       if (r<=0) {
102         if (!(errno == EAGAIN || errno == EINTR)) { perror("write"); exit(1); }
103 /*fprintf(stderr,"\t write transient error\n");*/
104       } else {
105 /*fprintf(stderr,"\t wrote %d\n",r);*/
106         used-= r;
107         wp+= r;
108         if (wp == buf+BUFFER) wp=buf;
109       }
110     }
111   }
112   exit(0);
113 }