2 Copyright (C) Andrew Tridgell 1996
3 Copyright (C) Paul Mackerras 1996
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation; either version 2 of the License, or
8 (at your option) any later version.
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with this program; if not, write to the Free Software
17 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 Utilities used in rsync
27 static int total_written = 0;
28 static int total_read = 0;
42 static int buffer_f_in = -1;
44 void setup_nonblocking(int f_in,int f_out)
46 set_blocking(f_out,0);
51 static char *read_buffer = NULL;
52 static char *read_buffer_p = NULL;
53 static int read_buffer_len = 0;
54 static int read_buffer_size = 0;
57 /* This function was added to overcome a deadlock problem when using
58 * ssh. It looks like we can't allow our receive queue to get full or
59 * ssh will clag up. Uggh. */
60 static void read_check(int f)
64 if (read_buffer_len == 0) {
65 read_buffer_p = read_buffer;
68 if ((n=num_waiting(f)) <= 0)
71 if (read_buffer_p != read_buffer) {
72 memmove(read_buffer,read_buffer_p,read_buffer_len);
73 read_buffer_p = read_buffer;
76 if (n > (read_buffer_size - read_buffer_len)) {
77 read_buffer_size += n;
79 read_buffer = (char *)malloc(read_buffer_size);
81 read_buffer = (char *)realloc(read_buffer,read_buffer_size);
82 if (!read_buffer) out_of_memory("read check");
83 read_buffer_p = read_buffer;
86 n = read(f,read_buffer+read_buffer_len,n);
93 static int readfd(int fd,char *buffer,int N)
100 if (read_buffer_len > 0) {
101 ret = MIN(read_buffer_len,N-total);
102 memcpy(buffer+total,read_buffer_p,ret);
103 read_buffer_p += ret;
104 read_buffer_len -= ret;
106 ret = read(fd,buffer + total,N - total);
120 if (readfd(f,b,4) != 4) {
122 fprintf(stderr,"Error reading %d bytes : %s\n",4,strerror(errno));
129 void read_buf(int f,char *buf,int len)
131 if (readfd(f,buf,len) != len) {
133 fprintf(stderr,"Error reading %d bytes : %s\n",len,strerror(errno));
139 unsigned char read_byte(int f)
143 return (unsigned char)c;
146 int read_write(int fd_in,int fd_out,int size)
148 static char *buf=NULL;
149 static int bufsize = WRITE_BLOCK_SIZE;
153 buf = (char *)malloc(bufsize);
154 if (!buf) out_of_memory("read_write");
157 while (total < size) {
158 int n = MIN(size-total,bufsize);
159 read_buf(fd_in,buf,n);
160 if (write(fd_out,buf,n) != n)
168 static int writefd(int fd,char *buf,int len)
174 if (buffer_f_in == -1)
175 return write(fd,buf,len);
177 while (total < len) {
178 int ret = write(fd,buf+total,len-total);
180 if (ret == 0) return total;
182 if (ret == -1 && errno != EWOULDBLOCK)
183 return total?total:-1;
186 read_check(buffer_f_in);
190 tv.tv_sec = BLOCKING_TIMEOUT;
192 select(16,NULL,&fds,NULL,&tv);
203 void write_int(int f,int x)
207 if (writefd(f,b,4) != 4) {
208 fprintf(stderr,"write_int failed : %s\n",strerror(errno));
214 void write_buf(int f,char *buf,int len)
216 if (writefd(f,buf,len) != len) {
217 fprintf(stderr,"write_buf failed : %s\n",strerror(errno));
220 total_written += len;
224 void write_byte(int f,unsigned char c)
226 write_buf(f,(char *)&c,1);
229 void write_flush(int f)