chiark / gitweb /
remove debugging and cruft; size window more sensibly
[chiark-utils.git] / cprogs / rwbuffer.c
1 /*
2  * rwbuffer.c
3  * common definitions for readbuffer/writebuffer
4  *
5  * readbuffer and writebuffer are:
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 #ifndef RWBUFFER_SIZE_MB_DEF
33 #define RWBUFFER_SIZE_MB_DEF 16
34 #endif
35
36 #ifndef RWBUFFER_SIZE_MB_MAX
37 #define RWBUFFER_SIZE_MB_MAX 512
38 #endif
39
40 unsigned char *buf, *wp, *rp;
41 int used, seeneof, maxselfd;
42 size_t buffersize;
43 fd_set readfds;
44 fd_set writefds;
45
46 static int opt_mlock=0;
47
48 int min(int a, int b) { return a<=b ? a : b; }
49
50 static void usage(FILE *f) {
51   if (fprintf(f,"usage: %s [--mlock] [<megabytes>]\n",progname) < 0)
52     { perror("print usage"); exit(16); }
53 }
54
55 static void usageerr(const char *what) {
56   fprintf(stderr,"%s: bad usage: %s\n",progname,what);
57   usage(stderr);
58   exit(12);
59 }
60
61 void nonblock(int fd, int yesno) {
62   int r;
63   r= fcntl(fd,F_GETFL,0); if (r == -1) { perror("fcntl getfl"); exit(8); }
64   if (yesno) r |= O_NDELAY;
65   else r &= ~O_NDELAY;
66   if (fcntl(fd,F_SETFL,r) == -1) { perror("fcntl setfl"); exit(8); }
67 }
68
69 static void unnonblock(void) {
70   nonblock(0,0); nonblock(1,0);
71 }
72
73 void startupcore(void) {
74   buf= xmalloc(buffersize);
75
76   if (opt_mlock) {
77     if (mlock(buf,buffersize)) { perror("mlock"); exit(2); }
78   }
79
80   used=0; wp=rp=buf; seeneof=0;
81   if (atexit(unnonblock)) { perror("atexit"); exit(16); }
82 }
83
84 void startup(const char *const *argv) {
85   const char *arg;
86   char *ep;
87   unsigned long opt_buffersize=RWBUFFER_SIZE_MB_DEF;
88   
89   assert(argv[0]);
90   
91   while ((arg= *++argv)) {
92     if (!strcmp(arg,"--mlock")) {
93       opt_mlock= 1;
94     } else if (isdigit((unsigned char)arg[0])) {
95       opt_buffersize= strtoul(arg,&ep,0);
96       if (opt_buffersize > RWBUFFER_SIZE_MB_MAX)
97         usageerr("buffer size too big");
98     } else {
99       usageerr("invalid option");
100     }
101   }
102
103   buffersize= opt_buffersize*1024*1024;
104   startupcore();
105   nonblock(0,1); nonblock(1,1);
106 }
107
108 void *xmalloc(size_t sz) {
109   void *r= malloc(sz); if (!r) { perror("malloc"); exit(6); }; return r;
110 }
111
112 void callselect(void) {
113   int r;
114   
115   for (;;) {
116     r= select(maxselfd,&readfds,&writefds,0,0);
117     if (r != -1) return;
118     if (errno != EINTR) {
119       perror("select"); exit(4);
120     }
121   }
122 }