chiark / gitweb /
2be121c0228e02d1acb9978bbfbea10baa6125b9
[chiark-utils.git] / cprogs / readbuffer.c
1 /*
2  * readbuffer.c
3  *
4  * A program for reading input from devices which don't like constant
5  * stopping and starting, such as tape drives.  readbuffer is:
6  *  Copyright (C) 1997-1998,2000-2001 Ian Jackson <ian@chiark.greenend.org.uk>
7  *
8  * readbuffer is part of chiark backup, a system for backing up GNU/Linux and
9  * 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 "rwbuffer.h"
31
32 const char *progname= "readbuffer";
33
34 static size_t waitempty;
35
36 int main(int argc, const char *const *argv) {
37   int r,reading;
38
39   startup(argv);
40   waitempty= (buffersize*1)/4;
41   reading=1;
42   
43   while (!seeneof || used) {
44     
45     FD_ZERO(&readfds);
46     if (reading) {
47       if (used<buffersize-1) {
48         FD_SET(0,&readfds);
49       } else {
50         reading=0;
51       }
52     }
53     FD_ZERO(&writefds); if (used) FD_SET(1,&writefds);
54
55     callselect();
56
57     if (FD_ISSET(0,&readfds)) {
58       r= read(0,rp,min(buffersize-1-used,buf+buffersize-rp));
59       if (!r) {
60         seeneof=1; reading=0;
61       } else if (r<0) {
62         if (!(errno == EAGAIN || errno == EINTR)) { perror("read"); exit(1); }
63       } else {
64         used+= r;
65         rp+= r;
66         if (rp == buf+buffersize) rp=buf;
67       }
68     }
69
70     if (FD_ISSET(1,&writefds)) {
71       assert(used);
72       r= write(1,wp,min(used,buf+buffersize-wp));
73       if (r<=0) {
74         if (!(errno == EAGAIN || errno == EINTR)) { perror("write"); exit(1); }
75       } else {
76         used-= r;
77         wp+= r;
78         if (wp == buf+buffersize) wp=buf;
79       }
80       if (used < waitempty && !seeneof) {
81         reading=1;
82       }
83     }
84   }
85   exit(0);
86 }