Large file support for psftp and pscp on both Windows and Unix. On Unix

[u/mdw/putty] / unix / uxsftp.c

diff --git a/unix/uxsftp.c b/unix/uxsftp.c
index 7045c32..8db397a 100644 (file)
--- a/unix/uxsftp.c
+++ b/unix/uxsftp.c
@@ -13,9 +13,13 @@
 #include <errno.h>
 #include <assert.h>
 #include <glob.h>
+#ifndef HAVE_NO_SYS_SELECT_H
+#include <sys/select.h>
+#endif
 
 #include "putty.h"
 #include "psftp.h"
+#include "int64.h"
 
 /*
  * In PSFTP our selects are synchronous, so these functions are
@@ -65,17 +69,16 @@ Filename platform_default_filename(const char *name)
     return ret;
 }
 
-/*
- * Stubs for the GUI feedback mechanism in Windows PSCP.
- */
-void gui_update_stats(char *name, unsigned long size,
-                     int percentage, unsigned long elapsed,
-                     unsigned long done, unsigned long eta,
-                     unsigned long ratebs) {}
-void gui_send_errcount(int list, int errs) {}
-void gui_send_char(int is_stderr, int c) {}
-void gui_enable(char *arg) {}
+char *get_ttymode(void *frontend, const char *mode) { return NULL; }
 
+int get_userpass_input(prompts_t *p, unsigned char *in, int inlen)
+{
+    int ret;
+    ret = cmdline_get_passwd_input(p, in, inlen);
+    if (ret == -1)
+       ret = console_get_userpass_input(p, in, inlen);
+    return ret;
+}
 
 /*
  * Set local current directory. Returns NULL on success, or else an
@@ -120,7 +123,7 @@ struct RFile {
     int fd;
 };
 
-RFile *open_existing_file(char *name, unsigned long *size,
+RFile *open_existing_file(char *name, uint64 *size,
                          unsigned long *mtime, unsigned long *atime)
 {
     int fd;
@@ -140,9 +143,15 @@ RFile *open_existing_file(char *name, unsigned long *size,
            memset(&statbuf, 0, sizeof(statbuf));
        }
 
-       if (size)
-           *size = statbuf.st_size;
-
+       if (size) {
+           if (sizeof(statbuf.st_size) == 8) {
+               size->hi = statbuf.st_size >> 32;
+               size->lo = (long) statbuf.st_size;
+           } else {
+               *size = uint64_make(0, statbuf.st_size);
+           }
+       }
+               
        if (mtime)
            *mtime = statbuf.st_mtime;
 
@@ -185,6 +194,37 @@ WFile *open_new_file(char *name)
     return ret;
 }
 
+
+WFile *open_existing_wfile(char *name, uint64 *size)
+{
+    int fd;
+    WFile *ret;
+
+    fd = open(name, O_APPEND | O_WRONLY);
+    if (fd < 0)
+       return NULL;
+
+    ret = snew(WFile);
+    ret->fd = fd;
+
+    if (size) {
+       struct stat statbuf;
+       if (fstat(fd, &statbuf) < 0) {
+           fprintf(stderr, "%s: stat: %s\n", name, strerror(errno));
+           memset(&statbuf, 0, sizeof(statbuf));
+       }
+
+       if (sizeof(statbuf.st_size) == 8) {
+           size->hi = statbuf.st_size >> 32;
+           size->lo = (long) statbuf.st_size;
+       } else {
+           *size = uint64_make(0, statbuf.st_size);
+       }
+    }
+
+    return ret;
+}
+
 int write_to_file(WFile *f, void *buffer, int length)
 {
     char *p = (char *)buffer;
@@ -226,6 +266,53 @@ void close_wfile(WFile *f)
     sfree(f);
 }
 
+/* Seek offset bytes through file, from whence, where whence is
+   FROM_START, FROM_CURRENT, or FROM_END */
+int seek_file(WFile *f, uint64 offset, int whence)
+{
+    off_t fileofft;
+    int lseek_whence;
+    
+    if (sizeof(off_t) == 8) {
+       fileofft = ((off_t) offset.hi << 32) + offset.lo;
+    } else {
+       fileofft = offset.lo;
+    }
+
+    switch (whence) {
+    case FROM_START:
+       lseek_whence = SEEK_SET;
+       break;
+    case FROM_CURRENT:
+       lseek_whence = SEEK_CUR;
+       break;
+    case FROM_END:
+       lseek_whence = SEEK_END;
+       break;
+    default:
+       return -1;
+    }
+
+    return lseek(f->fd, fileofft, lseek_whence) >= 0 ? 0 : -1;
+}
+
+uint64 get_file_posn(WFile *f)
+{
+    off_t fileofft;
+    uint64 ret;
+
+    fileofft = lseek(f->fd, (off_t) 0, SEEK_CUR);
+
+    if (sizeof(off_t) == 8) {
+       ret.hi = fileofft >> 32;
+       ret.lo = (long) fileofft;
+    } else {
+       ret = uint64_make(0, fileofft);
+    }
+
+    return ret;
+}
+
 int file_type(char *name)
 {
     struct stat statbuf;
@@ -433,8 +520,27 @@ static int ssh_sftp_do_select(int include_stdin, int no_fds_ok)
            ret = select(maxfd, &rset, &wset, &xset, ptv);
            if (ret == 0)
                now = next;
-           else
-               now = GETTICKCOUNT();
+           else {
+               long newnow = GETTICKCOUNT();
+               /*
+                * Check to see whether the system clock has
+                * changed massively during the select.
+                */
+               if (newnow - now < 0 || newnow - now > next - now) {
+                   /*
+                    * If so, look at the elapsed time in the
+                    * select and use it to compute a new
+                    * tickcount_offset.
+                    */
+                   long othernow = now + tv.tv_sec * 1000 + tv.tv_usec / 1000;
+                   /* So we'd like GETTICKCOUNT to have returned othernow,
+                    * but instead it return newnow. Hence ... */
+                   tickcount_offset += othernow - newnow;
+                   now = othernow;
+               } else {
+                   now = newnow;
+               }
+           }
        } while (ret < 0 && errno != EINTR);
     } while (ret == 0);