X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/e92338c82deba1576d23f22e11d95b0b34432e08..3cb6f5d6cbe2b1fead0453ffb31488c2ffc2def2:/io.c diff --git a/io.c b/io.c index dbe21005..75127f01 100644 --- a/io.c +++ b/io.c @@ -24,18 +24,18 @@ */ #include "rsync.h" -static int total_written = 0; -static int total_read = 0; +static off_t total_written; +static off_t total_read; extern int verbose; extern int sparse_files; -int write_total(void) +off_t write_total(void) { return total_written; } -int read_total(void) +off_t read_total(void) { return total_read; } @@ -49,10 +49,10 @@ void setup_nonblocking(int f_in,int f_out) } -static char *read_buffer = NULL; -static char *read_buffer_p = NULL; -static int read_buffer_len = 0; -static int read_buffer_size = 0; +static char *read_buffer; +static char *read_buffer_p; +static int read_buffer_len; +static int read_buffer_size; /* This function was added to overcome a deadlock problem when using @@ -145,6 +145,29 @@ int read_int(int f) return IVAL(b,0); } +off_t read_longint(int f) +{ + extern int remote_version; + off_t ret; + char b[8]; + ret = read_int(f); + if (ret == -1 && remote_version >= 16) { + if (sizeof(off_t) <= 4) { + fprintf(FERROR,"Integer overflow - attempted 64 bit offset\n"); + exit_cleanup(1); + } + if ((ret=readfd(f,b,8)) != 8) { + if (verbose > 1) + fprintf(FERROR,"(%d) Error reading %d bytes : %s\n", + getpid(),8,ret==-1?strerror(errno):"EOF"); + exit_cleanup(1); + } + total_read += 8; + ret = IVAL(b,0) | (((off_t)IVAL(b,4))<<32); + } + return ret; +} + void read_buf(int f,char *buf,int len) { int ret; @@ -165,8 +188,8 @@ unsigned char read_byte(int f) } -static char last_byte=0; -static int last_sparse = 0; +static char last_byte; +static int last_sparse; int sparse_end(int f) { @@ -211,27 +234,6 @@ int write_sparse(int f,char *buf,int len) return len; } -int read_write(int fd_in,int fd_out,int size) -{ - static char *buf=NULL; - int bufsize = sparse_files?SPARSE_WRITE_SIZE:WRITE_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_sparse(fd_out,buf,n) != n) - return total; - total += n; - } - return total; -} - static int writefd(int fd,char *buf,int len) { @@ -252,10 +254,19 @@ static int writefd(int fd,char *buf,int len) return -1; if (ret == -1 && got_select) { + /* hmmm, we got a write select on the fd and then failed to write. + Why doesn't that mean that the fd is dead? It doesn't on some + systems it seems (eg. IRIX) */ + u_sleep(1000); +#if 0 fprintf(FERROR,"write exception\n"); exit_cleanup(1); +#endif } + got_select = 0; + + if (ret == -1) { read_check(buffer_f_in); @@ -269,8 +280,6 @@ static int writefd(int fd,char *buf,int len) fd_count = buffer_f_in+1; } - got_select = 0; - tv.tv_sec = BLOCKING_TIMEOUT; tv.tv_usec = 0; count = select(fd_count,buffer_f_in == -1? NULL: &r_fds, @@ -309,6 +318,29 @@ void write_int(int f,int x) total_written += 4; } +void write_longint(int f, off_t x) +{ + extern int remote_version; + char b[8]; + int ret; + + if (remote_version < 16 || x <= 0x7FFFFFFF) { + write_int(f, (int)x); + return; + } + + write_int(f, -1); + SIVAL(b,0,(x&0xFFFFFFFF)); + SIVAL(b,4,((x>>32)&0xFFFFFFFF)); + + if ((ret=writefd(f,b,8)) != 8) { + fprintf(FERROR,"write_longint failed : %s\n", + ret==-1?strerror(errno):"EOF"); + exit_cleanup(1); + } + total_written += 8; +} + void write_buf(int f,char *buf,int len) { int ret;