+/*
+ Copyright (C) Andrew Tridgell 1996
+ Copyright (C) Paul Mackerras 1996
+
+ This program is free software; you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation; either version 2 of the License, or
+ (at your option) any later version.
+
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with this program; if not, write to the Free Software
+ Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+
+/*
+ Utilities used in rsync
+
+ tridge, June 1996
+ */
+#include "rsync.h"
+
+static int total_written = 0;
+static int total_read = 0;
+
+extern int verbose;
+
+int write_total(void)
+{
+ return total_written;
+}
+
+int read_total(void)
+{
+ return total_read;
+}
+
+static int buffer_f_in = -1;
+
+void setup_nonblocking(int f_in,int f_out)
+{
+ set_blocking(f_out,0);
+ buffer_f_in = f_in;
+}
+
+
+static char *read_buffer = NULL;
+static char *read_buffer_p = NULL;
+static int read_buffer_len = 0;
+static int read_buffer_size = 0;
+
+
+/* This function was added to overcome a deadlock problem when using
+ * ssh. It looks like we can't allow our receive queue to get full or
+ * ssh will clag up. Uggh. */
+static void read_check(int f)
+{
+ int n;
+
+ if (read_buffer_len == 0) {
+ read_buffer_p = read_buffer;
+ }
+
+ if ((n=num_waiting(f)) <= 0)
+ return;
+
+ if (read_buffer_p != read_buffer) {
+ memmove(read_buffer,read_buffer_p,read_buffer_len);
+ read_buffer_p = read_buffer;
+ }
+
+ if (n > (read_buffer_size - read_buffer_len)) {
+ read_buffer_size += n;
+ if (!read_buffer)
+ read_buffer = (char *)malloc(read_buffer_size);
+ else
+ read_buffer = (char *)realloc(read_buffer,read_buffer_size);
+ if (!read_buffer) out_of_memory("read check");
+ read_buffer_p = read_buffer;
+ }
+
+ n = read(f,read_buffer+read_buffer_len,n);
+ if (n > 0) {
+ read_buffer_len += n;
+ }
+}
+
+
+static int readfd(int fd,char *buffer,int N)
+{
+ int ret;
+ int total=0;
+
+ while (total < N)
+ {
+ if (read_buffer_len > 0) {
+ ret = MIN(read_buffer_len,N-total);
+ memcpy(buffer+total,read_buffer_p,ret);
+ read_buffer_p += ret;
+ read_buffer_len -= ret;
+ } else {
+ ret = read(fd,buffer + total,N - total);
+ }
+
+ if (ret <= 0)
+ return total;
+ total += ret;
+ }
+ return total;
+}
+
+
+int read_int(int f)
+{
+ char b[4];
+ if (readfd(f,b,4) != 4) {
+ if (verbose > 1)
+ fprintf(stderr,"Error reading %d bytes : %s\n",4,strerror(errno));
+ exit(1);
+ }
+ total_read += 4;
+ return IVAL(b,0);
+}
+
+void read_buf(int f,char *buf,int len)
+{
+ if (readfd(f,buf,len) != len) {
+ if (verbose > 1)
+ fprintf(stderr,"Error reading %d bytes : %s\n",len,strerror(errno));
+ exit(1);
+ }
+ total_read += len;
+}
+
+
+int read_write(int fd_in,int fd_out,int size)
+{
+ static char *buf=NULL;
+ static int bufsize = WRITE_BLOCK_SIZE;
+ int total=0;
+
+ if (!buf) {
+ buf = (char *)malloc(bufsize);
+ if (!buf) out_of_memory("read_write");
+ }
+
+ while (total < size) {
+ int n = MIN(size-total,bufsize);
+ read_buf(fd_in,buf,n);
+ if (write(fd_out,buf,n) != n)
+ return total;
+ total += n;
+ }
+ return total;
+}
+
+
+static int writefd(int fd,char *buf,int len)
+{
+ int total = 0;
+ fd_set fds;
+
+ if (buffer_f_in == -1)
+ return write(fd,buf,len);
+
+ while (total < len) {
+ int ret = write(fd,buf+total,len-total);
+
+ if (ret == 0) return total;
+
+ if (ret == -1 && errno != EWOULDBLOCK)
+ return total?total:-1;
+
+ if (ret == -1) {
+ read_check(buffer_f_in);
+
+ FD_ZERO(&fds);
+ FD_SET(fd,&fds);
+ select(16,NULL,&fds,NULL,NULL);
+ } else {
+ total += ret;
+ }
+ }
+
+ return total;
+}
+
+
+
+void write_int(int f,int x)
+{
+ char b[4];
+ SIVAL(b,0,x);
+ if (writefd(f,b,4) != 4) {
+ fprintf(stderr,"write_int failed : %s\n",strerror(errno));
+ exit(1);
+ }
+ total_written += 4;
+}
+
+void write_buf(int f,char *buf,int len)
+{
+ if (writefd(f,buf,len) != len) {
+ fprintf(stderr,"write_buf failed : %s\n",strerror(errno));
+ exit(1);
+ }
+ total_written += len;
+}
+
+
+void write_flush(int f)
+{
+}
+
+
Matt McCutchen's Web Site / rsync / rsync.git / commitdiff