]> andersk Git - openssh.git/blobdiff - sftp-server.c
- djm@cvs.openbsd.org 2010/01/30 02:54:53
[openssh.git] / sftp-server.c
index 1cbd1847cdd837619d70770ed48f15135caa28dc..a98ac2b6dd28c9074771181e93a3757d0cebda98 100644 (file)
@@ -1,34 +1,52 @@
+/* $OpenBSD: sftp-server.c,v 1.91 2010/01/13 01:40:16 djm Exp $ */
 /*
- * Copyright (c) 2000 Markus Friedl.  All rights reserved.
+ * Copyright (c) 2000-2004 Markus Friedl.  All rights reserved.
  *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- * 1. Redistributions of source code must retain the above copyright
- *    notice, this list of conditions and the following disclaimer.
- * 2. Redistributions in binary form must reproduce the above copyright
- *    notice, this list of conditions and the following disclaimer in the
- *    documentation and/or other materials provided with the distribution.
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
  *
- * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
- * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
- * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
- * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
- * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
- * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
- * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  */
+
 #include "includes.h"
-RCSID("$OpenBSD: sftp-server.c,v 1.26 2001/05/12 19:53:13 markus Exp $");
 
+#include <sys/types.h>
+#include <sys/param.h>
+#include <sys/stat.h>
+#ifdef HAVE_SYS_TIME_H
+# include <sys/time.h>
+#endif
+#ifdef HAVE_SYS_MOUNT_H
+#include <sys/mount.h>
+#endif
+#ifdef HAVE_SYS_STATVFS_H
+#include <sys/statvfs.h>
+#endif
+
+#include <dirent.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <pwd.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <pwd.h>
+#include <time.h>
+#include <unistd.h>
+#include <stdarg.h>
+
+#include "xmalloc.h"
 #include "buffer.h"
-#include "bufaux.h"
-#include "getput.h"
 #include "log.h"
-#include "xmalloc.h"
+#include "misc.h"
+#include "uidswap.h"
 
 #include "sftp.h"
 #include "sftp-common.h"
@@ -37,13 +55,13 @@ RCSID("$OpenBSD: sftp-server.c,v 1.26 2001/05/12 19:53:13 markus Exp $");
 #define get_int64()                    buffer_get_int64(&iqueue);
 #define get_int()                      buffer_get_int(&iqueue);
 #define get_string(lenp)               buffer_get_string(&iqueue, lenp);
-#define TRACE                          debug
 
-#ifdef HAVE___PROGNAME
-extern char *__progname;
-#else
-char *__progname;
-#endif
+/* Our verbosity */
+LogLevel log_level = SYSLOG_LEVEL_ERROR;
+
+/* Our client */
+struct passwd *pw = NULL;
+char *client_addr = NULL;
 
 /* input and output queue */
 Buffer iqueue;
@@ -52,7 +70,10 @@ Buffer oqueue;
 /* Version of client */
 int version;
 
-/* portable attibutes, etc. */
+/* Disable writes */
+int readonly;
+
+/* portable attributes, etc. */
 
 typedef struct Stat Stat;
 
@@ -62,7 +83,7 @@ struct Stat {
        Attrib attrib;
 };
 
-int
+static int
 errno_to_portable(int unixerrno)
 {
        int ret = 0;
@@ -86,6 +107,9 @@ errno_to_portable(int unixerrno)
        case EINVAL:
                ret = SSH2_FX_BAD_MESSAGE;
                break;
+       case ENOSYS:
+               ret = SSH2_FX_OP_UNSUPPORTED;
+               break;
        default:
                ret = SSH2_FX_FAILURE;
                break;
@@ -93,7 +117,7 @@ errno_to_portable(int unixerrno)
        return ret;
 }
 
-int
+static int
 flags_from_portable(int pflags)
 {
        int flags = 0;
@@ -115,7 +139,34 @@ flags_from_portable(int pflags)
        return flags;
 }
 
-Attrib *
+static const char *
+string_from_portable(int pflags)
+{
+       static char ret[128];
+
+       *ret = '\0';
+
+#define PAPPEND(str)   {                               \
+               if (*ret != '\0')                       \
+                       strlcat(ret, ",", sizeof(ret)); \
+               strlcat(ret, str, sizeof(ret));         \
+       }
+
+       if (pflags & SSH2_FXF_READ)
+               PAPPEND("READ")
+       if (pflags & SSH2_FXF_WRITE)
+               PAPPEND("WRITE")
+       if (pflags & SSH2_FXF_CREAT)
+               PAPPEND("CREATE")
+       if (pflags & SSH2_FXF_TRUNC)
+               PAPPEND("TRUNCATE")
+       if (pflags & SSH2_FXF_EXCL)
+               PAPPEND("EXCL")
+
+       return ret;
+}
+
+static Attrib *
 get_attrib(void)
 {
        return decode_attrib(&iqueue);
@@ -129,6 +180,8 @@ struct Handle {
        DIR *dirp;
        int fd;
        char *name;
+       u_int64_t bytes_read, bytes_write;
+       int next_unused;
 };
 
 enum {
@@ -137,67 +190,74 @@ enum {
        HANDLE_FILE
 };
 
-Handle handles[100];
+Handle *handles = NULL;
+u_int num_handles = 0;
+int first_unused_handle = -1;
 
-void
-handle_init(void)
+static void handle_unused(int i)
 {
-       int i;
-
-       for(i = 0; i < sizeof(handles)/sizeof(Handle); i++)
-               handles[i].use = HANDLE_UNUSED;
+       handles[i].use = HANDLE_UNUSED;
+       handles[i].next_unused = first_unused_handle;
+       first_unused_handle = i;
 }
 
-int
-handle_new(int use, char *name, int fd, DIR *dirp)
+static int
+handle_new(int use, const char *name, int fd, DIR *dirp)
 {
        int i;
 
-       for(i = 0; i < sizeof(handles)/sizeof(Handle); i++) {
-               if (handles[i].use == HANDLE_UNUSED) {
-                       handles[i].use = use;
-                       handles[i].dirp = dirp;
-                       handles[i].fd = fd;
-                       handles[i].name = name;
-                       return i;
-               }
+       if (first_unused_handle == -1) {
+               if (num_handles + 1 <= num_handles)
+                       return -1;
+               num_handles++;
+               handles = xrealloc(handles, num_handles, sizeof(Handle));
+               handle_unused(num_handles - 1);
        }
-       return -1;
+
+       i = first_unused_handle;
+       first_unused_handle = handles[i].next_unused;
+
+       handles[i].use = use;
+       handles[i].dirp = dirp;
+       handles[i].fd = fd;
+       handles[i].name = xstrdup(name);
+       handles[i].bytes_read = handles[i].bytes_write = 0;
+
+       return i;
 }
 
-int
+static int
 handle_is_ok(int i, int type)
 {
-       return i >= 0 && i < sizeof(handles)/sizeof(Handle) &&
-           handles[i].use == type;
+       return i >= 0 && (u_int)i < num_handles && handles[i].use == type;
 }
 
-int
+static int
 handle_to_string(int handle, char **stringp, int *hlenp)
 {
        if (stringp == NULL || hlenp == NULL)
                return -1;
        *stringp = xmalloc(sizeof(int32_t));
-       PUT_32BIT(*stringp, handle);
+       put_u32(*stringp, handle);
        *hlenp = sizeof(int32_t);
        return 0;
 }
 
-int
-handle_from_string(char *handle, u_int hlen)
+static int
+handle_from_string(const char *handle, u_int hlen)
 {
        int val;
 
        if (hlen != sizeof(int32_t))
                return -1;
-       val = GET_32BIT(handle);
+       val = get_u32(handle);
        if (handle_is_ok(val, HANDLE_FILE) ||
            handle_is_ok(val, HANDLE_DIR))
                return val;
        return -1;
 }
 
-char *
+static char *
 handle_to_name(int handle)
 {
        if (handle_is_ok(handle, HANDLE_DIR)||
@@ -206,7 +266,7 @@ handle_to_name(int handle)
        return NULL;
 }
 
-DIR *
+static DIR *
 handle_to_dir(int handle)
 {
        if (handle_is_ok(handle, HANDLE_DIR))
@@ -214,7 +274,7 @@ handle_to_dir(int handle)
        return NULL;
 }
 
-int
+static int
 handle_to_fd(int handle)
 {
        if (handle_is_ok(handle, HANDLE_FILE))
@@ -222,24 +282,82 @@ handle_to_fd(int handle)
        return -1;
 }
 
-int
+static void
+handle_update_read(int handle, ssize_t bytes)
+{
+       if (handle_is_ok(handle, HANDLE_FILE) && bytes > 0)
+               handles[handle].bytes_read += bytes;
+}
+
+static void
+handle_update_write(int handle, ssize_t bytes)
+{
+       if (handle_is_ok(handle, HANDLE_FILE) && bytes > 0)
+               handles[handle].bytes_write += bytes;
+}
+
+static u_int64_t
+handle_bytes_read(int handle)
+{
+       if (handle_is_ok(handle, HANDLE_FILE))
+               return (handles[handle].bytes_read);
+       return 0;
+}
+
+static u_int64_t
+handle_bytes_write(int handle)
+{
+       if (handle_is_ok(handle, HANDLE_FILE))
+               return (handles[handle].bytes_write);
+       return 0;
+}
+
+static int
 handle_close(int handle)
 {
        int ret = -1;
 
        if (handle_is_ok(handle, HANDLE_FILE)) {
                ret = close(handles[handle].fd);
-               handles[handle].use = HANDLE_UNUSED;
+               xfree(handles[handle].name);
+               handle_unused(handle);
        } else if (handle_is_ok(handle, HANDLE_DIR)) {
                ret = closedir(handles[handle].dirp);
-               handles[handle].use = HANDLE_UNUSED;
+               xfree(handles[handle].name);
+               handle_unused(handle);
        } else {
                errno = ENOENT;
        }
        return ret;
 }
 
-int
+static void
+handle_log_close(int handle, char *emsg)
+{
+       if (handle_is_ok(handle, HANDLE_FILE)) {
+               logit("%s%sclose \"%s\" bytes read %llu written %llu",
+                   emsg == NULL ? "" : emsg, emsg == NULL ? "" : " ",
+                   handle_to_name(handle),
+                   (unsigned long long)handle_bytes_read(handle),
+                   (unsigned long long)handle_bytes_write(handle));
+       } else {
+               logit("%s%sclosedir \"%s\"",
+                   emsg == NULL ? "" : emsg, emsg == NULL ? "" : " ",
+                   handle_to_name(handle));
+       }
+}
+
+static void
+handle_log_exit(void)
+{
+       u_int i;
+
+       for (i = 0; i < num_handles; i++)
+               if (handles[i].use != HANDLE_UNUSED)
+                       handle_log_close(i, "forced");
+}
+
+static int
 get_handle(void)
 {
        char *handle;
@@ -255,7 +373,7 @@ get_handle(void)
 
 /* send replies */
 
-void
+static void
 send_msg(Buffer *m)
 {
        int mlen = buffer_len(m);
@@ -265,10 +383,9 @@ send_msg(Buffer *m)
        buffer_consume(m, mlen);
 }
 
-void
-send_status(u_int32_t id, u_int32_t error)
+static const char *
+status_to_message(u_int32_t status)
 {
-       Buffer msg;
        const char *status_messages[] = {
                "Success",                      /* SSH_FX_OK */
                "End of file",                  /* SSH_FX_EOF */
@@ -281,22 +398,31 @@ send_status(u_int32_t id, u_int32_t error)
                "Operation unsupported",        /* SSH_FX_OP_UNSUPPORTED */
                "Unknown error"                 /* Others */
        };
+       return (status_messages[MIN(status,SSH2_FX_MAX)]);
+}
+
+static void
+send_status(u_int32_t id, u_int32_t status)
+{
+       Buffer msg;
 
-       TRACE("sent status id %d error %d", id, error);
+       debug3("request %u: sent status %u", id, status);
+       if (log_level > SYSLOG_LEVEL_VERBOSE ||
+           (status != SSH2_FX_OK && status != SSH2_FX_EOF))
+               logit("sent status %s", status_to_message(status));
        buffer_init(&msg);
        buffer_put_char(&msg, SSH2_FXP_STATUS);
        buffer_put_int(&msg, id);
-       buffer_put_int(&msg, error);
+       buffer_put_int(&msg, status);
        if (version >= 3) {
-               buffer_put_cstring(&msg,
-                   status_messages[MIN(error,SSH2_FX_MAX)]);
+               buffer_put_cstring(&msg, status_to_message(status));
                buffer_put_cstring(&msg, "");
        }
        send_msg(&msg);
        buffer_free(&msg);
 }
-void
-send_data_or_handle(char type, u_int32_t id, char *data, int dlen)
+static void
+send_data_or_handle(char type, u_int32_t id, const char *data, int dlen)
 {
        Buffer msg;
 
@@ -308,27 +434,27 @@ send_data_or_handle(char type, u_int32_t id, char *data, int dlen)
        buffer_free(&msg);
 }
 
-void
-send_data(u_int32_t id, char *data, int dlen)
+static void
+send_data(u_int32_t id, const char *data, int dlen)
 {
-       TRACE("sent data id %d len %d", id, dlen);
+       debug("request %u: sent data len %d", id, dlen);
        send_data_or_handle(SSH2_FXP_DATA, id, data, dlen);
 }
 
-void
+static void
 send_handle(u_int32_t id, int handle)
 {
        char *string;
        int hlen;
 
        handle_to_string(handle, &string, &hlen);
-       TRACE("sent handle id %d handle %d", id, handle);
+       debug("request %u: sent handle handle %d", id, handle);
        send_data_or_handle(SSH2_FXP_HANDLE, id, string, hlen);
        xfree(string);
 }
 
-void
-send_names(u_int32_t id, int count, Stat *stats)
+static void
+send_names(u_int32_t id, int count, const Stat *stats)
 {
        Buffer msg;
        int i;
@@ -337,7 +463,7 @@ send_names(u_int32_t id, int count, Stat *stats)
        buffer_put_char(&msg, SSH2_FXP_NAME);
        buffer_put_int(&msg, id);
        buffer_put_int(&msg, count);
-       TRACE("sent names id %d count %d", id, count);
+       debug("request %u: sent names count %d", id, count);
        for (i = 0; i < count; i++) {
                buffer_put_cstring(&msg, stats[i].name);
                buffer_put_cstring(&msg, stats[i].long_name);
@@ -347,12 +473,12 @@ send_names(u_int32_t id, int count, Stat *stats)
        buffer_free(&msg);
 }
 
-void
-send_attrib(u_int32_t id, Attrib *a)
+static void
+send_attrib(u_int32_t id, const Attrib *a)
 {
        Buffer msg;
 
-       TRACE("sent attrib id %d have 0x%x", id, a->flags);
+       debug("request %u: sent attrib have 0x%x", id, a->flags);
        buffer_init(&msg);
        buffer_put_char(&msg, SSH2_FXP_ATTRS);
        buffer_put_int(&msg, id);
@@ -361,23 +487,59 @@ send_attrib(u_int32_t id, Attrib *a)
        buffer_free(&msg);
 }
 
+static void
+send_statvfs(u_int32_t id, struct statvfs *st)
+{
+       Buffer msg;
+       u_int64_t flag;
+
+       flag = (st->f_flag & ST_RDONLY) ? SSH2_FXE_STATVFS_ST_RDONLY : 0;
+       flag |= (st->f_flag & ST_NOSUID) ? SSH2_FXE_STATVFS_ST_NOSUID : 0;
+
+       buffer_init(&msg);
+       buffer_put_char(&msg, SSH2_FXP_EXTENDED_REPLY);
+       buffer_put_int(&msg, id);
+       buffer_put_int64(&msg, st->f_bsize);
+       buffer_put_int64(&msg, st->f_frsize);
+       buffer_put_int64(&msg, st->f_blocks);
+       buffer_put_int64(&msg, st->f_bfree);
+       buffer_put_int64(&msg, st->f_bavail);
+       buffer_put_int64(&msg, st->f_files);
+       buffer_put_int64(&msg, st->f_ffree);
+       buffer_put_int64(&msg, st->f_favail);
+       buffer_put_int64(&msg, FSID_TO_ULONG(st->f_fsid));
+       buffer_put_int64(&msg, flag);
+       buffer_put_int64(&msg, st->f_namemax);
+       send_msg(&msg);
+       buffer_free(&msg);
+}
+
 /* parse incoming */
 
-void
+static void
 process_init(void)
 {
        Buffer msg;
 
-       version = buffer_get_int(&iqueue);
-       TRACE("client version %d", version);
+       version = get_int();
+       verbose("received client version %d", version);
        buffer_init(&msg);
        buffer_put_char(&msg, SSH2_FXP_VERSION);
        buffer_put_int(&msg, SSH2_FILEXFER_VERSION);
+       /* POSIX rename extension */
+       buffer_put_cstring(&msg, "posix-rename@openssh.com");
+       buffer_put_cstring(&msg, "1"); /* version */
+       /* statvfs extension */
+       buffer_put_cstring(&msg, "statvfs@openssh.com");
+       buffer_put_cstring(&msg, "2"); /* version */
+       /* fstatvfs extension */
+       buffer_put_cstring(&msg, "fstatvfs@openssh.com");
+       buffer_put_cstring(&msg, "2"); /* version */
        send_msg(&msg);
        buffer_free(&msg);
 }
 
-void
+static void
 process_open(void)
 {
        u_int32_t id, pflags;
@@ -388,20 +550,27 @@ process_open(void)
        id = get_int();
        name = get_string(NULL);
        pflags = get_int();             /* portable flags */
+       debug3("request %u: open flags %d", id, pflags);
        a = get_attrib();
        flags = flags_from_portable(pflags);
        mode = (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) ? a->perm : 0666;
-       TRACE("open id %d name %s flags %d mode 0%o", id, name, pflags, mode);
-       fd = open(name, flags, mode);
-       if (fd < 0) {
-               status = errno_to_portable(errno);
-       } else {
-               handle = handle_new(HANDLE_FILE, xstrdup(name), fd, NULL);
-               if (handle < 0) {
-                       close(fd);
+       logit("open \"%s\" flags %s mode 0%o",
+           name, string_from_portable(pflags), mode);
+       if (readonly &&
+           ((flags & O_ACCMODE) == O_WRONLY || (flags & O_ACCMODE) == O_RDWR))
+               status = SSH2_FX_PERMISSION_DENIED;
+       else {
+               fd = open(name, flags, mode);
+               if (fd < 0) {
+                       status = errno_to_portable(errno);
                } else {
-                       send_handle(id, handle);
-                       status = SSH2_FX_OK;
+                       handle = handle_new(HANDLE_FILE, name, fd, NULL);
+                       if (handle < 0) {
+                               close(fd);
+                       } else {
+                               send_handle(id, handle);
+                               status = SSH2_FX_OK;
+                       }
                }
        }
        if (status != SSH2_FX_OK)
@@ -409,7 +578,7 @@ process_open(void)
        xfree(name);
 }
 
-void
+static void
 process_close(void)
 {
        u_int32_t id;
@@ -417,13 +586,14 @@ process_close(void)
 
        id = get_int();
        handle = get_handle();
-       TRACE("close id %d handle %d", id, handle);
+       debug3("request %u: close handle %u", id, handle);
+       handle_log_close(handle, NULL);
        ret = handle_close(handle);
        status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
        send_status(id, status);
 }
 
-void
+static void
 process_read(void)
 {
        char buf[64*1024];
@@ -436,11 +606,11 @@ process_read(void)
        off = get_int64();
        len = get_int();
 
-       TRACE("read id %d handle %d off %llu len %d", id, handle,
-           (u_int64_t)off, len);
+       debug("request %u: read \"%s\" (handle %d) off %llu len %d",
+           id, handle_to_name(handle), handle, (unsigned long long)off, len);
        if (len > sizeof buf) {
                len = sizeof buf;
-               log("read change len %d", len);
+               debug2("read change len %d", len);
        }
        fd = handle_to_fd(handle);
        if (fd >= 0) {
@@ -456,6 +626,7 @@ process_read(void)
                        } else {
                                send_data(id, buf, ret);
                                status = SSH2_FX_OK;
+                               handle_update_read(handle, ret);
                        }
                }
        }
@@ -463,13 +634,13 @@ process_read(void)
                send_status(id, status);
 }
 
-void
+static void
 process_write(void)
 {
        u_int32_t id;
        u_int64_t off;
        u_int len;
-       int handle, fd, ret, status = SSH2_FX_FAILURE;
+       int handle, fd, ret, status;
        char *data;
 
        id = get_int();
@@ -477,23 +648,30 @@ process_write(void)
        off = get_int64();
        data = get_string(&len);
 
-       TRACE("write id %d handle %d off %llu len %d", id, handle,
-           (u_int64_t)off, len);
+       debug("request %u: write \"%s\" (handle %d) off %llu len %d",
+           id, handle_to_name(handle), handle, (unsigned long long)off, len);
        fd = handle_to_fd(handle);
-       if (fd >= 0) {
+       
+       if (fd < 0)
+               status = SSH2_FX_FAILURE;
+       else if (readonly)
+               status = SSH2_FX_PERMISSION_DENIED;
+       else {
                if (lseek(fd, off, SEEK_SET) < 0) {
                        status = errno_to_portable(errno);
                        error("process_write: seek failed");
                } else {
 /* XXX ATOMICIO ? */
                        ret = write(fd, data, len);
-                       if (ret == -1) {
+                       if (ret < 0) {
                                error("process_write: write failed");
                                status = errno_to_portable(errno);
-                       } else if (ret == len) {
+                       } else if ((size_t)ret == len) {
                                status = SSH2_FX_OK;
+                               handle_update_write(handle, ret);
                        } else {
-                               log("nothing at all written");
+                               debug2("nothing at all written");
+                               status = SSH2_FX_FAILURE;
                        }
                }
        }
@@ -501,7 +679,7 @@ process_write(void)
        xfree(data);
 }
 
-void
+static void
 process_do_stat(int do_lstat)
 {
        Attrib a;
@@ -512,7 +690,8 @@ process_do_stat(int do_lstat)
 
        id = get_int();
        name = get_string(NULL);
-       TRACE("%sstat id %d name %s", do_lstat ? "l" : "", id, name);
+       debug3("request %u: %sstat", id, do_lstat ? "l" : "");
+       verbose("%sstat name \"%s\"", do_lstat ? "l" : "", name);
        ret = do_lstat ? lstat(name, &st) : stat(name, &st);
        if (ret < 0) {
                status = errno_to_portable(errno);
@@ -526,19 +705,19 @@ process_do_stat(int do_lstat)
        xfree(name);
 }
 
-void
+static void
 process_stat(void)
 {
        process_do_stat(0);
 }
 
-void
+static void
 process_lstat(void)
 {
        process_do_stat(1);
 }
 
-void
+static void
 process_fstat(void)
 {
        Attrib a;
@@ -548,9 +727,10 @@ process_fstat(void)
 
        id = get_int();
        handle = get_handle();
-       TRACE("fstat id %d handle %d", id, handle);
+       debug("request %u: fstat \"%s\" (handle %u)",
+           id, handle_to_name(handle), handle);
        fd = handle_to_fd(handle);
-       if (fd  >= 0) {
+       if (fd >= 0) {
                ret = fstat(fd, &st);
                if (ret < 0) {
                        status = errno_to_portable(errno);
@@ -564,8 +744,8 @@ process_fstat(void)
                send_status(id, status);
 }
 
-struct timeval *
-attrib_to_tv(Attrib *a)
+static struct timeval *
+attrib_to_tv(const Attrib *a)
 {
        static struct timeval tv[2];
 
@@ -576,30 +756,49 @@ attrib_to_tv(Attrib *a)
        return tv;
 }
 
-void
+static void
 process_setstat(void)
 {
        Attrib *a;
        u_int32_t id;
        char *name;
-       int ret;
-       int status = SSH2_FX_OK;
+       int status = SSH2_FX_OK, ret;
 
        id = get_int();
        name = get_string(NULL);
        a = get_attrib();
-       TRACE("setstat id %d name %s", id, name);
+       debug("request %u: setstat name \"%s\"", id, name);
+       if (readonly) {
+               status = SSH2_FX_PERMISSION_DENIED;
+               a->flags = 0;
+       }
+       if (a->flags & SSH2_FILEXFER_ATTR_SIZE) {
+               logit("set \"%s\" size %llu",
+                   name, (unsigned long long)a->size);
+               ret = truncate(name, a->size);
+               if (ret == -1)
+                       status = errno_to_portable(errno);
+       }
        if (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) {
-               ret = chmod(name, a->perm & 0777);
+               logit("set \"%s\" mode %04o", name, a->perm);
+               ret = chmod(name, a->perm & 07777);
                if (ret == -1)
                        status = errno_to_portable(errno);
        }
        if (a->flags & SSH2_FILEXFER_ATTR_ACMODTIME) {
+               char buf[64];
+               time_t t = a->mtime;
+
+               strftime(buf, sizeof(buf), "%Y%m%d-%H:%M:%S",
+                   localtime(&t));
+               logit("set \"%s\" modtime %s", name, buf);
                ret = utimes(name, attrib_to_tv(a));
                if (ret == -1)
                        status = errno_to_portable(errno);
        }
        if (a->flags & SSH2_FILEXFER_ATTR_UIDGID) {
+               logit("set \"%s\" owner %lu group %lu", name,
+                   (u_long)a->uid, (u_long)a->gid);
                ret = chown(name, a->uid, a->gid);
                if (ret == -1)
                        status = errno_to_portable(errno);
@@ -608,34 +807,50 @@ process_setstat(void)
        xfree(name);
 }
 
-void
+static void
 process_fsetstat(void)
 {
        Attrib *a;
        u_int32_t id;
        int handle, fd, ret;
        int status = SSH2_FX_OK;
-       char *name;
 
        id = get_int();
        handle = get_handle();
        a = get_attrib();
-       TRACE("fsetstat id %d handle %d", id, handle);
+       debug("request %u: fsetstat handle %d", id, handle);
        fd = handle_to_fd(handle);
-       name = handle_to_name(handle);
-       if (fd < 0 || name == NULL) {
+       if (fd < 0)
                status = SSH2_FX_FAILURE;
-       } else {
+       else if (readonly)
+               status = SSH2_FX_PERMISSION_DENIED;
+       else {
+               char *name = handle_to_name(handle);
+
+               if (a->flags & SSH2_FILEXFER_ATTR_SIZE) {
+                       logit("set \"%s\" size %llu",
+                           name, (unsigned long long)a->size);
+                       ret = ftruncate(fd, a->size);
+                       if (ret == -1)
+                               status = errno_to_portable(errno);
+               }
                if (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) {
+                       logit("set \"%s\" mode %04o", name, a->perm);
 #ifdef HAVE_FCHMOD
-                       ret = fchmod(fd, a->perm & 0777);
+                       ret = fchmod(fd, a->perm & 07777);
 #else
-                       ret = chmod(name, a->perm & 0777);
+                       ret = chmod(name, a->perm & 07777);
 #endif
                        if (ret == -1)
                                status = errno_to_portable(errno);
                }
                if (a->flags & SSH2_FILEXFER_ATTR_ACMODTIME) {
+                       char buf[64];
+                       time_t t = a->mtime;
+
+                       strftime(buf, sizeof(buf), "%Y%m%d-%H:%M:%S",
+                           localtime(&t));
+                       logit("set \"%s\" modtime %s", name, buf);
 #ifdef HAVE_FUTIMES
                        ret = futimes(fd, attrib_to_tv(a));
 #else
@@ -645,6 +860,8 @@ process_fsetstat(void)
                                status = errno_to_portable(errno);
                }
                if (a->flags & SSH2_FILEXFER_ATTR_UIDGID) {
+                       logit("set \"%s\" owner %lu group %lu", name,
+                           (u_long)a->uid, (u_long)a->gid);
 #ifdef HAVE_FCHOWN
                        ret = fchown(fd, a->uid, a->gid);
 #else
@@ -657,7 +874,7 @@ process_fsetstat(void)
        send_status(id, status);
 }
 
-void
+static void
 process_opendir(void)
 {
        DIR *dirp = NULL;
@@ -667,12 +884,13 @@ process_opendir(void)
 
        id = get_int();
        path = get_string(NULL);
-       TRACE("opendir id %d path %s", id, path);
+       debug3("request %u: opendir", id);
+       logit("opendir \"%s\"", path);
        dirp = opendir(path);
        if (dirp == NULL) {
                status = errno_to_portable(errno);
        } else {
-               handle = handle_new(HANDLE_DIR, xstrdup(path), 0, dirp);
+               handle = handle_new(HANDLE_DIR, path, 0, dirp);
                if (handle < 0) {
                        closedir(dirp);
                } else {
@@ -686,46 +904,7 @@ process_opendir(void)
        xfree(path);
 }
 
-/*
- * drwxr-xr-x    5 markus   markus       1024 Jan 13 18:39 .ssh
- */
-char *
-ls_file(char *name, struct stat *st)
-{
-       int sz = 0;
-       struct passwd *pw;
-       struct group *gr;
-       struct tm *ltime = localtime(&st->st_mtime);
-       char *user, *group;
-       char buf[1024], mode[11+1], tbuf[12+1], ubuf[11+1], gbuf[11+1];
-
-       strmode(st->st_mode, mode);
-       if ((pw = getpwuid(st->st_uid)) != NULL) {
-               user = pw->pw_name;
-       } else {
-               snprintf(ubuf, sizeof ubuf, "%d", st->st_uid);
-               user = ubuf;
-       }
-       if ((gr = getgrgid(st->st_gid)) != NULL) {
-               group = gr->gr_name;
-       } else {
-               snprintf(gbuf, sizeof gbuf, "%d", st->st_gid);
-               group = gbuf;
-       }
-       if (ltime != NULL) {
-               if (time(NULL) - st->st_mtime < (365*24*60*60)/2)
-                       sz = strftime(tbuf, sizeof tbuf, "%b %e %H:%M", ltime);
-               else
-                       sz = strftime(tbuf, sizeof tbuf, "%b %e  %Y", ltime);
-       }
-       if (sz == 0)
-               tbuf[0] = '\0';
-       snprintf(buf, sizeof buf, "%s %3d %-8.8s %-8.8s %8llu %s %s", mode,
-           st->st_nlink, user, group, (u_int64_t)st->st_size, tbuf, name);
-       return xstrdup(buf);
-}
-
-void
+static void
 process_readdir(void)
 {
        DIR *dirp;
@@ -736,30 +915,32 @@ process_readdir(void)
 
        id = get_int();
        handle = get_handle();
-       TRACE("readdir id %d handle %d", id, handle);
+       debug("request %u: readdir \"%s\" (handle %d)", id,
+           handle_to_name(handle), handle);
        dirp = handle_to_dir(handle);
        path = handle_to_name(handle);
        if (dirp == NULL || path == NULL) {
                send_status(id, SSH2_FX_FAILURE);
        } else {
                struct stat st;
-               char pathname[1024];
+               char pathname[MAXPATHLEN];
                Stat *stats;
                int nstats = 10, count = 0, i;
-               stats = xmalloc(nstats * sizeof(Stat));
+
+               stats = xcalloc(nstats, sizeof(Stat));
                while ((dp = readdir(dirp)) != NULL) {
                        if (count >= nstats) {
                                nstats *= 2;
-                               stats = xrealloc(stats, nstats * sizeof(Stat));
+                               stats = xrealloc(stats, nstats, sizeof(Stat));
                        }
 /* XXX OVERFLOW ? */
-                       snprintf(pathname, sizeof pathname,
-                           "%s/%s", path, dp->d_name);
+                       snprintf(pathname, sizeof pathname, "%s%s%s", path,
+                           strcmp(path, "/") ? "/" : "", dp->d_name);
                        if (lstat(pathname, &st) < 0)
                                continue;
                        stat_to_attrib(&st, &(stats[count].attrib));
                        stats[count].name = xstrdup(dp->d_name);
-                       stats[count].long_name = ls_file(dp->d_name, &st);
+                       stats[count].long_name = ls_file(dp->d_name, &st, 0, 0);
                        count++;
                        /* send up to 100 entries in one message */
                        /* XXX check packet size instead */
@@ -768,7 +949,7 @@ process_readdir(void)
                }
                if (count > 0) {
                        send_names(id, count, stats);
-                       for(i = 0; i < count; i++) {
+                       for (i = 0; i < count; i++) {
                                xfree(stats[i].name);
                                xfree(stats[i].long_name);
                        }
@@ -779,7 +960,7 @@ process_readdir(void)
        }
 }
 
-void
+static void
 process_remove(void)
 {
        char *name;
@@ -789,14 +970,19 @@ process_remove(void)
 
        id = get_int();
        name = get_string(NULL);
-       TRACE("remove id %d name %s", id, name);
-       ret = unlink(name);
-       status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
+       debug3("request %u: remove", id);
+       logit("remove name \"%s\"", name);
+       if (readonly)
+               status = SSH2_FX_PERMISSION_DENIED;
+       else {
+               ret = unlink(name);
+               status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
+       }
        send_status(id, status);
        xfree(name);
 }
 
-void
+static void
 process_mkdir(void)
 {
        Attrib *a;
@@ -808,15 +994,20 @@ process_mkdir(void)
        name = get_string(NULL);
        a = get_attrib();
        mode = (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) ?
-           a->perm & 0777 : 0777;
-       TRACE("mkdir id %d name %s mode 0%o", id, name, mode);
-       ret = mkdir(name, mode);
-       status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
+           a->perm & 07777 : 0777;
+       debug3("request %u: mkdir", id);
+       logit("mkdir name \"%s\" mode 0%o", name, mode);
+       if (readonly)
+               status = SSH2_FX_PERMISSION_DENIED;
+       else {
+               ret = mkdir(name, mode);
+               status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
+       }
        send_status(id, status);
        xfree(name);
 }
 
-void
+static void
 process_rmdir(void)
 {
        u_int32_t id;
@@ -825,14 +1016,19 @@ process_rmdir(void)
 
        id = get_int();
        name = get_string(NULL);
-       TRACE("rmdir id %d name %s", id, name);
-       ret = rmdir(name);
-       status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
+       debug3("request %u: rmdir", id);
+       logit("rmdir name \"%s\"", name);
+       if (readonly)
+               status = SSH2_FX_PERMISSION_DENIED;
+       else {
+               ret = rmdir(name);
+               status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
+       }
        send_status(id, status);
        xfree(name);
 }
 
-void
+static void
 process_realpath(void)
 {
        char resolvedname[MAXPATHLEN];
@@ -845,7 +1041,8 @@ process_realpath(void)
                xfree(path);
                path = xstrdup(".");
        }
-       TRACE("realpath id %d path %s", id, path);
+       debug3("request %u: realpath", id);
+       verbose("realpath \"%s\"", path);
        if (realpath(path, resolvedname) == NULL) {
                send_status(id, errno_to_portable(errno));
        } else {
@@ -857,66 +1054,109 @@ process_realpath(void)
        xfree(path);
 }
 
-void
+static void
 process_rename(void)
 {
        u_int32_t id;
-       struct stat st;
        char *oldpath, *newpath;
-       int ret, status = SSH2_FX_FAILURE;
+       int status;
+       struct stat sb;
 
        id = get_int();
        oldpath = get_string(NULL);
        newpath = get_string(NULL);
-       TRACE("rename id %d old %s new %s", id, oldpath, newpath);
-       /* fail if 'newpath' exists */
-       if (stat(newpath, &st) == -1) {
-               ret = rename(oldpath, newpath);
-               status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
+       debug3("request %u: rename", id);
+       logit("rename old \"%s\" new \"%s\"", oldpath, newpath);
+       status = SSH2_FX_FAILURE;
+       if (readonly)
+               status = SSH2_FX_PERMISSION_DENIED;
+       else if (lstat(oldpath, &sb) == -1)
+               status = errno_to_portable(errno);
+       else if (S_ISREG(sb.st_mode)) {
+               /* Race-free rename of regular files */
+               if (link(oldpath, newpath) == -1) {
+                       if (errno == EOPNOTSUPP || errno == ENOSYS
+#ifdef EXDEV
+                           || errno == EXDEV
+#endif
+#ifdef LINK_OPNOTSUPP_ERRNO
+                           || errno == LINK_OPNOTSUPP_ERRNO
+#endif
+                           ) {
+                               struct stat st;
+
+                               /*
+                                * fs doesn't support links, so fall back to
+                                * stat+rename.  This is racy.
+                                */
+                               if (stat(newpath, &st) == -1) {
+                                       if (rename(oldpath, newpath) == -1)
+                                               status =
+                                                   errno_to_portable(errno);
+                                       else
+                                               status = SSH2_FX_OK;
+                               }
+                       } else {
+                               status = errno_to_portable(errno);
+                       }
+               } else if (unlink(oldpath) == -1) {
+                       status = errno_to_portable(errno);
+                       /* clean spare link */
+                       unlink(newpath);
+               } else
+                       status = SSH2_FX_OK;
+       } else if (stat(newpath, &sb) == -1) {
+               if (rename(oldpath, newpath) == -1)
+                       status = errno_to_portable(errno);
+               else
+                       status = SSH2_FX_OK;
        }
        send_status(id, status);
        xfree(oldpath);
        xfree(newpath);
 }
 
-void
+static void
 process_readlink(void)
 {
        u_int32_t id;
        int len;
-       char link[MAXPATHLEN];
+       char buf[MAXPATHLEN];
        char *path;
 
        id = get_int();
        path = get_string(NULL);
-       TRACE("readlink id %d path %s", id, path);
-       if ((len = readlink(path, link, sizeof(link) - 1)) == -1)
+       debug3("request %u: readlink", id);
+       verbose("readlink \"%s\"", path);
+       if ((len = readlink(path, buf, sizeof(buf) - 1)) == -1)
                send_status(id, errno_to_portable(errno));
        else {
                Stat s;
-               
-               link[len] = '\0';
+
+               buf[len] = '\0';
                attrib_clear(&s.attrib);
-               s.name = s.long_name = link;
+               s.name = s.long_name = buf;
                send_names(id, 1, &s);
        }
        xfree(path);
 }
 
-void
+static void
 process_symlink(void)
 {
        u_int32_t id;
-       struct stat st;
        char *oldpath, *newpath;
-       int ret, status = SSH2_FX_FAILURE;
+       int ret, status;
 
        id = get_int();
        oldpath = get_string(NULL);
        newpath = get_string(NULL);
-       TRACE("symlink id %d old %s new %s", id, oldpath, newpath);
-       /* fail if 'newpath' exists */
-       if (stat(newpath, &st) == -1) {
+       debug3("request %u: symlink", id);
+       logit("symlink old \"%s\" new \"%s\"", oldpath, newpath);
+       /* this will fail if 'newpath' exists */
+       if (readonly)
+               status = SSH2_FX_PERMISSION_DENIED;
+       else {
                ret = symlink(oldpath, newpath);
                status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
        }
@@ -925,7 +1165,64 @@ process_symlink(void)
        xfree(newpath);
 }
 
-void
+static void
+process_extended_posix_rename(u_int32_t id)
+{
+       char *oldpath, *newpath;
+       int ret, status;
+
+       oldpath = get_string(NULL);
+       newpath = get_string(NULL);
+       debug3("request %u: posix-rename", id);
+       logit("posix-rename old \"%s\" new \"%s\"", oldpath, newpath);
+       if (readonly)
+               status = SSH2_FX_PERMISSION_DENIED;
+       else {
+               ret = rename(oldpath, newpath);
+               status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
+       }
+       send_status(id, status);
+       xfree(oldpath);
+       xfree(newpath);
+}
+
+static void
+process_extended_statvfs(u_int32_t id)
+{
+       char *path;
+       struct statvfs st;
+
+       path = get_string(NULL);
+       debug3("request %u: statfs", id);
+       logit("statfs \"%s\"", path);
+
+       if (statvfs(path, &st) != 0)
+               send_status(id, errno_to_portable(errno));
+       else
+               send_statvfs(id, &st);
+        xfree(path);
+}
+
+static void
+process_extended_fstatvfs(u_int32_t id)
+{
+       int handle, fd;
+       struct statvfs st;
+
+       handle = get_handle();
+       debug("request %u: fstatvfs \"%s\" (handle %u)",
+           id, handle_to_name(handle), handle);
+       if ((fd = handle_to_fd(handle)) < 0) {
+               send_status(id, SSH2_FX_FAILURE);
+               return;
+       }
+       if (fstatvfs(fd, &st) != 0)
+               send_status(id, errno_to_portable(errno));
+       else
+               send_statvfs(id, &st);
+}
+
+static void
 process_extended(void)
 {
        u_int32_t id;
@@ -933,30 +1230,42 @@ process_extended(void)
 
        id = get_int();
        request = get_string(NULL);
-       send_status(id, SSH2_FX_OP_UNSUPPORTED);                /* MUST */
+       if (strcmp(request, "posix-rename@openssh.com") == 0)
+               process_extended_posix_rename(id);
+       else if (strcmp(request, "statvfs@openssh.com") == 0)
+               process_extended_statvfs(id);
+       else if (strcmp(request, "fstatvfs@openssh.com") == 0)
+               process_extended_fstatvfs(id);
+       else
+               send_status(id, SSH2_FX_OP_UNSUPPORTED);        /* MUST */
        xfree(request);
 }
 
 /* stolen from ssh-agent */
 
-void
+static void
 process(void)
 {
        u_int msg_len;
+       u_int buf_len;
+       u_int consumed;
        u_int type;
        u_char *cp;
 
-       if (buffer_len(&iqueue) < 5)
+       buf_len = buffer_len(&iqueue);
+       if (buf_len < 5)
                return;         /* Incomplete message. */
-       cp = (u_char *) buffer_ptr(&iqueue);
-       msg_len = GET_32BIT(cp);
-       if (msg_len > 256 * 1024) {
-               error("bad message ");
-               exit(11);
+       cp = buffer_ptr(&iqueue);
+       msg_len = get_u32(cp);
+       if (msg_len > SFTP_MAX_MSG_LENGTH) {
+               error("bad message from %s local user %s",
+                   client_addr, pw->pw_name);
+               sftp_server_cleanup_exit(11);
        }
-       if (buffer_len(&iqueue) < msg_len + 4)
+       if (buf_len < msg_len + 4)
                return;
        buffer_consume(&iqueue, 4);
+       buf_len -= 4;
        type = buffer_get_char(&iqueue);
        switch (type) {
        case SSH2_FXP_INIT:
@@ -1023,26 +1332,118 @@ process(void)
                error("Unknown message %d", type);
                break;
        }
+       /* discard the remaining bytes from the current packet */
+       if (buf_len < buffer_len(&iqueue)) {
+               error("iqueue grew unexpectedly");
+               sftp_server_cleanup_exit(255);
+       }
+       consumed = buf_len - buffer_len(&iqueue);
+       if (msg_len < consumed) {
+               error("msg_len %d < consumed %d", msg_len, consumed);
+               sftp_server_cleanup_exit(255);
+       }
+       if (msg_len > consumed)
+               buffer_consume(&iqueue, msg_len - consumed);
+}
+
+/* Cleanup handler that logs active handles upon normal exit */
+void
+sftp_server_cleanup_exit(int i)
+{
+       if (pw != NULL && client_addr != NULL) {
+               handle_log_exit();
+               logit("session closed for local user %s from [%s]",
+                   pw->pw_name, client_addr);
+       }
+       _exit(i);
+}
+
+static void
+sftp_server_usage(void)
+{
+       extern char *__progname;
+
+       fprintf(stderr,
+           "usage: %s [-ehR] [-f log_facility] [-l log_level] [-u umask]\n",
+           __progname);
+       exit(1);
 }
 
 int
-main(int ac, char **av)
+sftp_server_main(int argc, char **argv, struct passwd *user_pw)
 {
        fd_set *rset, *wset;
-       int in, out, max;
+       int in, out, max, ch, skipargs = 0, log_stderr = 0;
        ssize_t len, olen, set_size;
+       SyslogFacility log_facility = SYSLOG_FACILITY_AUTH;
+       char *cp, buf[4*4096];
+       const char *errmsg;
+       mode_t mask;
+
+       extern char *optarg;
+       extern char *__progname;
+
+       __progname = ssh_get_progname(argv[0]);
+       log_init(__progname, log_level, log_facility, log_stderr);
+
+       while (!skipargs && (ch = getopt(argc, argv, "f:l:u:cehR")) != -1) {
+               switch (ch) {
+               case 'R':
+                       readonly = 1;
+                       break;
+               case 'c':
+                       /*
+                        * Ignore all arguments if we are invoked as a
+                        * shell using "sftp-server -c command"
+                        */
+                       skipargs = 1;
+                       break;
+               case 'e':
+                       log_stderr = 1;
+                       break;
+               case 'l':
+                       log_level = log_level_number(optarg);
+                       if (log_level == SYSLOG_LEVEL_NOT_SET)
+                               error("Invalid log level \"%s\"", optarg);
+                       break;
+               case 'f':
+                       log_facility = log_facility_number(optarg);
+                       if (log_facility == SYSLOG_FACILITY_NOT_SET)
+                               error("Invalid log facility \"%s\"", optarg);
+                       break;
+               case 'u':
+                       mask = (mode_t)strtonum(optarg, 0, 0777, &errmsg);
+                       if (errmsg != NULL)
+                               fatal("Invalid umask \"%s\": %s",
+                                   optarg, errmsg);
+                       (void)umask(mask);
+                       break;
+               case 'h':
+               default:
+                       sftp_server_usage();
+               }
+       }
 
-       /* XXX should use getopt */
+       log_init(__progname, log_level, log_facility, log_stderr);
 
-       __progname = get_progname(av[0]);
-       handle_init();
+       if ((cp = getenv("SSH_CONNECTION")) != NULL) {
+               client_addr = xstrdup(cp);
+               if ((cp = strchr(client_addr, ' ')) == NULL) {
+                       error("Malformed SSH_CONNECTION variable: \"%s\"",
+                           getenv("SSH_CONNECTION"));
+                       sftp_server_cleanup_exit(255);
+               }
+               *cp = '\0';
+       } else
+               client_addr = xstrdup("UNKNOWN");
 
-#ifdef DEBUG_SFTP_SERVER
-       log_init("sftp-server", SYSLOG_LEVEL_DEBUG1, SYSLOG_FACILITY_AUTH, 0);
-#endif
+       pw = pwcopy(user_pw);
 
-       in = dup(STDIN_FILENO);
-       out = dup(STDOUT_FILENO);
+       logit("session opened for local user %s from [%s]",
+           pw->pw_name, client_addr);
+
+       in = STDIN_FILENO;
+       out = STDOUT_FILENO;
 
 #ifdef HAVE_CYGWIN
        setmode(in, O_BINARY);
@@ -1066,7 +1467,15 @@ main(int ac, char **av)
                memset(rset, 0, set_size);
                memset(wset, 0, set_size);
 
-               FD_SET(in, rset);
+               /*
+                * Ensure that we can read a full buffer and handle
+                * the worst-case length packet it can generate,
+                * otherwise apply backpressure by stopping reads.
+                */
+               if (buffer_check_alloc(&iqueue, sizeof(buf)) &&
+                   buffer_check_alloc(&oqueue, SFTP_MAX_MSG_LENGTH))
+                       FD_SET(in, rset);
+
                olen = buffer_len(&oqueue);
                if (olen > 0)
                        FD_SET(out, wset);
@@ -1074,19 +1483,19 @@ main(int ac, char **av)
                if (select(max+1, rset, wset, NULL, NULL) < 0) {
                        if (errno == EINTR)
                                continue;
-                       exit(2);
+                       error("select: %s", strerror(errno));
+                       sftp_server_cleanup_exit(2);
                }
 
                /* copy stdin to iqueue */
                if (FD_ISSET(in, rset)) {
-                       char buf[4*4096];
                        len = read(in, buf, sizeof buf);
                        if (len == 0) {
                                debug("read eof");
-                               exit(0);
+                               sftp_server_cleanup_exit(0);
                        } else if (len < 0) {
-                               error("read error");
-                               exit(1);
+                               error("read: %s", strerror(errno));
+                               sftp_server_cleanup_exit(1);
                        } else {
                                buffer_append(&iqueue, buf, len);
                        }
@@ -1095,13 +1504,19 @@ main(int ac, char **av)
                if (FD_ISSET(out, wset)) {
                        len = write(out, buffer_ptr(&oqueue), olen);
                        if (len < 0) {
-                               error("write error");
-                               exit(1);
+                               error("write: %s", strerror(errno));
+                               sftp_server_cleanup_exit(1);
                        } else {
                                buffer_consume(&oqueue, len);
                        }
                }
-               /* process requests from client */
-               process();
+
+               /*
+                * Process requests from client if we can fit the results
+                * into the output buffer, otherwise stop processing input
+                * and let the output queue drain.
+                */
+               if (buffer_check_alloc(&oqueue, SFTP_MAX_MSG_LENGTH))
+                       process();
        }
 }
This page took 0.246254 seconds and 4 git commands to generate.