[BACK]Return to sftp-server.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / ssh

Annotation of src/usr.bin/ssh/sftp-server.c, Revision 1.143

1.143   ! djm         1: /* $OpenBSD: sftp-server.c,v 1.142 2022/09/16 06:55:37 djm Exp $ */
1.1       markus      2: /*
1.45      markus      3:  * Copyright (c) 2000-2004 Markus Friedl.  All rights reserved.
1.1       markus      4:  *
1.45      markus      5:  * Permission to use, copy, modify, and distribute this software for any
                      6:  * purpose with or without fee is hereby granted, provided that the above
                      7:  * copyright notice and this permission notice appear in all copies.
1.1       markus      8:  *
1.45      markus      9:  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
                     10:  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
                     11:  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
                     12:  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
                     13:  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
                     14:  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
                     15:  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
1.1       markus     16:  */
1.52      stevesk    17:
                     18: #include <sys/types.h>
1.122     djm        19: #include <sys/resource.h>
1.52      stevesk    20: #include <sys/stat.h>
1.66      stevesk    21: #include <sys/time.h>
1.79      djm        22: #include <sys/mount.h>
                     23: #include <sys/statvfs.h>
1.51      stevesk    24:
                     25: #include <dirent.h>
1.62      stevesk    26: #include <errno.h>
1.59      stevesk    27: #include <fcntl.h>
1.132     deraadt    28: #include <poll.h>
1.68      stevesk    29: #include <stdlib.h>
1.69      stevesk    30: #include <stdio.h>
1.65      stevesk    31: #include <string.h>
1.70      deraadt    32: #include <pwd.h>
1.143   ! djm        33: #include <grp.h>
1.64      stevesk    34: #include <time.h>
1.63      stevesk    35: #include <unistd.h>
1.70      deraadt    36: #include <stdarg.h>
1.1       markus     37:
1.140     djm        38: #include "atomicio.h"
1.70      deraadt    39: #include "xmalloc.h"
1.104     djm        40: #include "sshbuf.h"
                     41: #include "ssherr.h"
1.14      markus     42: #include "log.h"
1.49      djm        43: #include "misc.h"
1.98      djm        44: #include "match.h"
1.58      djm        45: #include "uidswap.h"
1.1       markus     46:
1.10      markus     47: #include "sftp.h"
1.15      djm        48: #include "sftp-common.h"
1.1       markus     49:
1.117     djm        50: char *sftp_realpath(const char *, char *); /* sftp-realpath.c */
                     51:
1.122     djm        52: /* Maximum data read that we are willing to accept */
1.124     djm        53: #define SFTP_MAX_READ_LENGTH (SFTP_MAX_MSG_LENGTH - 1024)
1.122     djm        54:
1.58      djm        55: /* Our verbosity */
1.98      djm        56: static LogLevel log_level = SYSLOG_LEVEL_ERROR;
1.58      djm        57:
                     58: /* Our client */
1.98      djm        59: static struct passwd *pw = NULL;
                     60: static char *client_addr = NULL;
1.1       markus     61:
                     62: /* input and output queue */
1.104     djm        63: struct sshbuf *iqueue;
                     64: struct sshbuf *oqueue;
1.1       markus     65:
1.23      djm        66: /* Version of client */
1.98      djm        67: static u_int version;
                     68:
                     69: /* SSH2_FXP_INIT received */
                     70: static int init_done;
1.23      djm        71:
1.90      djm        72: /* Disable writes */
1.98      djm        73: static int readonly;
                     74:
                     75: /* Requests that are allowed/denied */
1.118     djm        76: static char *request_allowlist, *request_denylist;
1.90      djm        77:
1.43      miod       78: /* portable attributes, etc. */
1.1       markus     79: typedef struct Stat Stat;
                     80:
1.15      djm        81: struct Stat {
1.1       markus     82:        char *name;
                     83:        char *long_name;
                     84:        Attrib attrib;
                     85: };
                     86:
1.98      djm        87: /* Packet handlers */
                     88: static void process_open(u_int32_t id);
                     89: static void process_close(u_int32_t id);
                     90: static void process_read(u_int32_t id);
                     91: static void process_write(u_int32_t id);
                     92: static void process_stat(u_int32_t id);
                     93: static void process_lstat(u_int32_t id);
                     94: static void process_fstat(u_int32_t id);
                     95: static void process_setstat(u_int32_t id);
                     96: static void process_fsetstat(u_int32_t id);
                     97: static void process_opendir(u_int32_t id);
                     98: static void process_readdir(u_int32_t id);
                     99: static void process_remove(u_int32_t id);
                    100: static void process_mkdir(u_int32_t id);
                    101: static void process_rmdir(u_int32_t id);
                    102: static void process_realpath(u_int32_t id);
                    103: static void process_rename(u_int32_t id);
                    104: static void process_readlink(u_int32_t id);
                    105: static void process_symlink(u_int32_t id);
                    106: static void process_extended_posix_rename(u_int32_t id);
                    107: static void process_extended_statvfs(u_int32_t id);
                    108: static void process_extended_fstatvfs(u_int32_t id);
                    109: static void process_extended_hardlink(u_int32_t id);
1.102     djm       110: static void process_extended_fsync(u_int32_t id);
1.114     djm       111: static void process_extended_lsetstat(u_int32_t id);
1.122     djm       112: static void process_extended_limits(u_int32_t id);
1.129     djm       113: static void process_extended_expand(u_int32_t id);
1.140     djm       114: static void process_extended_copy_data(u_int32_t id);
1.141     djm       115: static void process_extended_home_directory(u_int32_t id);
1.143   ! djm       116: static void process_extended_get_users_groups_by_id(u_int32_t id);
1.98      djm       117: static void process_extended(u_int32_t id);
                    118:
                    119: struct sftp_handler {
                    120:        const char *name;       /* user-visible name for fine-grained perms */
                    121:        const char *ext_name;   /* extended request name */
                    122:        u_int type;             /* packet type, for non extended packets */
                    123:        void (*handler)(u_int32_t);
                    124:        int does_write;         /* if nonzero, banned for readonly mode */
                    125: };
                    126:
1.113     djm       127: static const struct sftp_handler handlers[] = {
1.98      djm       128:        /* NB. SSH2_FXP_OPEN does the readonly check in the handler itself */
                    129:        { "open", NULL, SSH2_FXP_OPEN, process_open, 0 },
                    130:        { "close", NULL, SSH2_FXP_CLOSE, process_close, 0 },
                    131:        { "read", NULL, SSH2_FXP_READ, process_read, 0 },
                    132:        { "write", NULL, SSH2_FXP_WRITE, process_write, 1 },
                    133:        { "lstat", NULL, SSH2_FXP_LSTAT, process_lstat, 0 },
                    134:        { "fstat", NULL, SSH2_FXP_FSTAT, process_fstat, 0 },
                    135:        { "setstat", NULL, SSH2_FXP_SETSTAT, process_setstat, 1 },
                    136:        { "fsetstat", NULL, SSH2_FXP_FSETSTAT, process_fsetstat, 1 },
                    137:        { "opendir", NULL, SSH2_FXP_OPENDIR, process_opendir, 0 },
                    138:        { "readdir", NULL, SSH2_FXP_READDIR, process_readdir, 0 },
                    139:        { "remove", NULL, SSH2_FXP_REMOVE, process_remove, 1 },
                    140:        { "mkdir", NULL, SSH2_FXP_MKDIR, process_mkdir, 1 },
                    141:        { "rmdir", NULL, SSH2_FXP_RMDIR, process_rmdir, 1 },
                    142:        { "realpath", NULL, SSH2_FXP_REALPATH, process_realpath, 0 },
                    143:        { "stat", NULL, SSH2_FXP_STAT, process_stat, 0 },
                    144:        { "rename", NULL, SSH2_FXP_RENAME, process_rename, 1 },
                    145:        { "readlink", NULL, SSH2_FXP_READLINK, process_readlink, 0 },
                    146:        { "symlink", NULL, SSH2_FXP_SYMLINK, process_symlink, 1 },
                    147:        { NULL, NULL, 0, NULL, 0 }
                    148: };
                    149:
                    150: /* SSH2_FXP_EXTENDED submessages */
1.113     djm       151: static const struct sftp_handler extended_handlers[] = {
1.98      djm       152:        { "posix-rename", "posix-rename@openssh.com", 0,
1.127     djm       153:            process_extended_posix_rename, 1 },
1.98      djm       154:        { "statvfs", "statvfs@openssh.com", 0, process_extended_statvfs, 0 },
                    155:        { "fstatvfs", "fstatvfs@openssh.com", 0, process_extended_fstatvfs, 0 },
                    156:        { "hardlink", "hardlink@openssh.com", 0, process_extended_hardlink, 1 },
1.102     djm       157:        { "fsync", "fsync@openssh.com", 0, process_extended_fsync, 1 },
1.114     djm       158:        { "lsetstat", "lsetstat@openssh.com", 0, process_extended_lsetstat, 1 },
1.128     djm       159:        { "limits", "limits@openssh.com", 0, process_extended_limits, 0 },
1.129     djm       160:        { "expand-path", "expand-path@openssh.com", 0,
                    161:            process_extended_expand, 0 },
1.140     djm       162:        { "copy-data", "copy-data", 0, process_extended_copy_data, 1 },
1.141     djm       163:        { "home-directory", "home-directory", 0,
                    164:            process_extended_home_directory, 0 },
1.143   ! djm       165:        { "users-groups-by-id", "users-groups-by-id@openssh.com", 0,
        !           166:            process_extended_get_users_groups_by_id, 0 },
1.98      djm       167:        { NULL, NULL, 0, NULL, 0 }
                    168: };
                    169:
1.125     djm       170: static const struct sftp_handler *
                    171: extended_handler_byname(const char *name)
                    172: {
                    173:        int i;
                    174:
                    175:        for (i = 0; extended_handlers[i].handler != NULL; i++) {
                    176:                if (strcmp(name, extended_handlers[i].ext_name) == 0)
                    177:                        return &extended_handlers[i];
                    178:        }
                    179:        return NULL;
                    180: }
                    181:
1.98      djm       182: static int
1.113     djm       183: request_permitted(const struct sftp_handler *h)
1.98      djm       184: {
                    185:        char *result;
                    186:
                    187:        if (readonly && h->does_write) {
                    188:                verbose("Refusing %s request in read-only mode", h->name);
                    189:                return 0;
                    190:        }
1.118     djm       191:        if (request_denylist != NULL &&
                    192:            ((result = match_list(h->name, request_denylist, NULL))) != NULL) {
1.98      djm       193:                free(result);
1.118     djm       194:                verbose("Refusing denylisted %s request", h->name);
1.98      djm       195:                return 0;
                    196:        }
1.118     djm       197:        if (request_allowlist != NULL &&
                    198:            ((result = match_list(h->name, request_allowlist, NULL))) != NULL) {
1.98      djm       199:                free(result);
1.118     djm       200:                debug2("Permitting allowlisted %s request", h->name);
1.98      djm       201:                return 1;
                    202:        }
1.118     djm       203:        if (request_allowlist != NULL) {
                    204:                verbose("Refusing non-allowlisted %s request", h->name);
1.98      djm       205:                return 0;
                    206:        }
                    207:        return 1;
                    208: }
                    209:
1.28      itojun    210: static int
1.2       markus    211: errno_to_portable(int unixerrno)
1.1       markus    212: {
                    213:        int ret = 0;
1.22      deraadt   214:
1.2       markus    215:        switch (unixerrno) {
1.1       markus    216:        case 0:
1.10      markus    217:                ret = SSH2_FX_OK;
1.1       markus    218:                break;
                    219:        case ENOENT:
                    220:        case ENOTDIR:
                    221:        case EBADF:
                    222:        case ELOOP:
1.10      markus    223:                ret = SSH2_FX_NO_SUCH_FILE;
1.1       markus    224:                break;
                    225:        case EPERM:
                    226:        case EACCES:
                    227:        case EFAULT:
1.10      markus    228:                ret = SSH2_FX_PERMISSION_DENIED;
1.1       markus    229:                break;
                    230:        case ENAMETOOLONG:
                    231:        case EINVAL:
1.10      markus    232:                ret = SSH2_FX_BAD_MESSAGE;
1.82      dtucker   233:                break;
                    234:        case ENOSYS:
                    235:                ret = SSH2_FX_OP_UNSUPPORTED;
1.1       markus    236:                break;
                    237:        default:
1.10      markus    238:                ret = SSH2_FX_FAILURE;
1.1       markus    239:                break;
                    240:        }
                    241:        return ret;
                    242: }
                    243:
1.28      itojun    244: static int
1.1       markus    245: flags_from_portable(int pflags)
                    246: {
                    247:        int flags = 0;
1.22      deraadt   248:
1.20      deraadt   249:        if ((pflags & SSH2_FXF_READ) &&
                    250:            (pflags & SSH2_FXF_WRITE)) {
1.1       markus    251:                flags = O_RDWR;
1.10      markus    252:        } else if (pflags & SSH2_FXF_READ) {
1.1       markus    253:                flags = O_RDONLY;
1.10      markus    254:        } else if (pflags & SSH2_FXF_WRITE) {
1.1       markus    255:                flags = O_WRONLY;
                    256:        }
1.101     djm       257:        if (pflags & SSH2_FXF_APPEND)
                    258:                flags |= O_APPEND;
1.10      markus    259:        if (pflags & SSH2_FXF_CREAT)
1.1       markus    260:                flags |= O_CREAT;
1.10      markus    261:        if (pflags & SSH2_FXF_TRUNC)
1.1       markus    262:                flags |= O_TRUNC;
1.10      markus    263:        if (pflags & SSH2_FXF_EXCL)
1.1       markus    264:                flags |= O_EXCL;
                    265:        return flags;
                    266: }
                    267:
1.58      djm       268: static const char *
                    269: string_from_portable(int pflags)
                    270: {
                    271:        static char ret[128];
                    272:
                    273:        *ret = '\0';
                    274:
                    275: #define PAPPEND(str)   {                               \
                    276:                if (*ret != '\0')                       \
                    277:                        strlcat(ret, ",", sizeof(ret)); \
1.70      deraadt   278:                strlcat(ret, str, sizeof(ret));         \
1.58      djm       279:        }
                    280:
                    281:        if (pflags & SSH2_FXF_READ)
                    282:                PAPPEND("READ")
                    283:        if (pflags & SSH2_FXF_WRITE)
                    284:                PAPPEND("WRITE")
1.101     djm       285:        if (pflags & SSH2_FXF_APPEND)
                    286:                PAPPEND("APPEND")
1.58      djm       287:        if (pflags & SSH2_FXF_CREAT)
                    288:                PAPPEND("CREATE")
                    289:        if (pflags & SSH2_FXF_TRUNC)
                    290:                PAPPEND("TRUNCATE")
                    291:        if (pflags & SSH2_FXF_EXCL)
                    292:                PAPPEND("EXCL")
                    293:
                    294:        return ret;
                    295: }
                    296:
1.1       markus    297: /* handle handles */
                    298:
                    299: typedef struct Handle Handle;
                    300: struct Handle {
                    301:        int use;
                    302:        DIR *dirp;
                    303:        int fd;
1.101     djm       304:        int flags;
1.1       markus    305:        char *name;
1.58      djm       306:        u_int64_t bytes_read, bytes_write;
1.75      djm       307:        int next_unused;
1.1       markus    308: };
1.22      deraadt   309:
1.1       markus    310: enum {
                    311:        HANDLE_UNUSED,
                    312:        HANDLE_DIR,
                    313:        HANDLE_FILE
                    314: };
1.22      deraadt   315:
1.113     djm       316: static Handle *handles = NULL;
                    317: static u_int num_handles = 0;
                    318: static int first_unused_handle = -1;
1.75      djm       319:
                    320: static void handle_unused(int i)
                    321: {
                    322:        handles[i].use = HANDLE_UNUSED;
                    323:        handles[i].next_unused = first_unused_handle;
                    324:        first_unused_handle = i;
1.1       markus    325: }
                    326:
1.28      itojun    327: static int
1.101     djm       328: handle_new(int use, const char *name, int fd, int flags, DIR *dirp)
1.1       markus    329: {
1.75      djm       330:        int i;
                    331:
                    332:        if (first_unused_handle == -1) {
                    333:                if (num_handles + 1 <= num_handles)
                    334:                        return -1;
                    335:                num_handles++;
1.106     deraadt   336:                handles = xreallocarray(handles, num_handles, sizeof(Handle));
1.75      djm       337:                handle_unused(num_handles - 1);
                    338:        }
                    339:
                    340:        i = first_unused_handle;
                    341:        first_unused_handle = handles[i].next_unused;
                    342:
                    343:        handles[i].use = use;
                    344:        handles[i].dirp = dirp;
                    345:        handles[i].fd = fd;
1.101     djm       346:        handles[i].flags = flags;
1.75      djm       347:        handles[i].name = xstrdup(name);
                    348:        handles[i].bytes_read = handles[i].bytes_write = 0;
1.22      deraadt   349:
1.75      djm       350:        return i;
1.1       markus    351: }
                    352:
1.28      itojun    353: static int
1.1       markus    354: handle_is_ok(int i, int type)
                    355: {
1.75      djm       356:        return i >= 0 && (u_int)i < num_handles && handles[i].use == type;
1.1       markus    357: }
                    358:
1.28      itojun    359: static int
1.104     djm       360: handle_to_string(int handle, u_char **stringp, int *hlenp)
1.1       markus    361: {
                    362:        if (stringp == NULL || hlenp == NULL)
                    363:                return -1;
1.13      markus    364:        *stringp = xmalloc(sizeof(int32_t));
1.57      djm       365:        put_u32(*stringp, handle);
1.13      markus    366:        *hlenp = sizeof(int32_t);
1.1       markus    367:        return 0;
                    368: }
                    369:
1.28      itojun    370: static int
1.104     djm       371: handle_from_string(const u_char *handle, u_int hlen)
1.1       markus    372: {
1.13      markus    373:        int val;
1.22      deraadt   374:
1.13      markus    375:        if (hlen != sizeof(int32_t))
1.1       markus    376:                return -1;
1.57      djm       377:        val = get_u32(handle);
1.1       markus    378:        if (handle_is_ok(val, HANDLE_FILE) ||
                    379:            handle_is_ok(val, HANDLE_DIR))
                    380:                return val;
                    381:        return -1;
                    382: }
                    383:
1.28      itojun    384: static char *
1.1       markus    385: handle_to_name(int handle)
                    386: {
                    387:        if (handle_is_ok(handle, HANDLE_DIR)||
                    388:            handle_is_ok(handle, HANDLE_FILE))
                    389:                return handles[handle].name;
                    390:        return NULL;
                    391: }
                    392:
1.28      itojun    393: static DIR *
1.1       markus    394: handle_to_dir(int handle)
                    395: {
                    396:        if (handle_is_ok(handle, HANDLE_DIR))
                    397:                return handles[handle].dirp;
                    398:        return NULL;
                    399: }
                    400:
1.28      itojun    401: static int
1.1       markus    402: handle_to_fd(int handle)
                    403: {
1.17      stevesk   404:        if (handle_is_ok(handle, HANDLE_FILE))
1.1       markus    405:                return handles[handle].fd;
                    406:        return -1;
                    407: }
                    408:
1.101     djm       409: static int
                    410: handle_to_flags(int handle)
                    411: {
                    412:        if (handle_is_ok(handle, HANDLE_FILE))
                    413:                return handles[handle].flags;
                    414:        return 0;
                    415: }
                    416:
1.58      djm       417: static void
                    418: handle_update_read(int handle, ssize_t bytes)
                    419: {
                    420:        if (handle_is_ok(handle, HANDLE_FILE) && bytes > 0)
                    421:                handles[handle].bytes_read += bytes;
                    422: }
                    423:
                    424: static void
                    425: handle_update_write(int handle, ssize_t bytes)
                    426: {
                    427:        if (handle_is_ok(handle, HANDLE_FILE) && bytes > 0)
                    428:                handles[handle].bytes_write += bytes;
                    429: }
                    430:
                    431: static u_int64_t
                    432: handle_bytes_read(int handle)
                    433: {
                    434:        if (handle_is_ok(handle, HANDLE_FILE))
                    435:                return (handles[handle].bytes_read);
                    436:        return 0;
                    437: }
                    438:
                    439: static u_int64_t
                    440: handle_bytes_write(int handle)
                    441: {
                    442:        if (handle_is_ok(handle, HANDLE_FILE))
                    443:                return (handles[handle].bytes_write);
                    444:        return 0;
                    445: }
                    446:
1.28      itojun    447: static int
1.1       markus    448: handle_close(int handle)
                    449: {
                    450:        int ret = -1;
1.22      deraadt   451:
1.1       markus    452:        if (handle_is_ok(handle, HANDLE_FILE)) {
                    453:                ret = close(handles[handle].fd);
1.97      djm       454:                free(handles[handle].name);
1.75      djm       455:                handle_unused(handle);
1.1       markus    456:        } else if (handle_is_ok(handle, HANDLE_DIR)) {
                    457:                ret = closedir(handles[handle].dirp);
1.97      djm       458:                free(handles[handle].name);
1.75      djm       459:                handle_unused(handle);
1.1       markus    460:        } else {
                    461:                errno = ENOENT;
                    462:        }
                    463:        return ret;
                    464: }
                    465:
1.58      djm       466: static void
                    467: handle_log_close(int handle, char *emsg)
                    468: {
                    469:        if (handle_is_ok(handle, HANDLE_FILE)) {
                    470:                logit("%s%sclose \"%s\" bytes read %llu written %llu",
                    471:                    emsg == NULL ? "" : emsg, emsg == NULL ? "" : " ",
                    472:                    handle_to_name(handle),
1.72      stevesk   473:                    (unsigned long long)handle_bytes_read(handle),
                    474:                    (unsigned long long)handle_bytes_write(handle));
1.58      djm       475:        } else {
                    476:                logit("%s%sclosedir \"%s\"",
                    477:                    emsg == NULL ? "" : emsg, emsg == NULL ? "" : " ",
                    478:                    handle_to_name(handle));
                    479:        }
                    480: }
                    481:
                    482: static void
                    483: handle_log_exit(void)
                    484: {
                    485:        u_int i;
                    486:
1.75      djm       487:        for (i = 0; i < num_handles; i++)
1.58      djm       488:                if (handles[i].use != HANDLE_UNUSED)
                    489:                        handle_log_close(i, "forced");
                    490: }
                    491:
1.28      itojun    492: static int
1.104     djm       493: get_handle(struct sshbuf *queue, int *hp)
1.1       markus    494: {
1.104     djm       495:        u_char *handle;
                    496:        int r;
                    497:        size_t hlen;
                    498:
                    499:        *hp = -1;
                    500:        if ((r = sshbuf_get_string(queue, &handle, &hlen)) != 0)
                    501:                return r;
1.10      markus    502:        if (hlen < 256)
1.104     djm       503:                *hp = handle_from_string(handle, hlen);
1.97      djm       504:        free(handle);
1.104     djm       505:        return 0;
1.1       markus    506: }
                    507:
                    508: /* send replies */
                    509:
1.28      itojun    510: static void
1.104     djm       511: send_msg(struct sshbuf *m)
1.1       markus    512: {
1.104     djm       513:        int r;
1.22      deraadt   514:
1.104     djm       515:        if ((r = sshbuf_put_stringb(oqueue, m)) != 0)
1.120     djm       516:                fatal_fr(r, "enqueue");
1.104     djm       517:        sshbuf_reset(m);
1.1       markus    518: }
                    519:
1.58      djm       520: static const char *
                    521: status_to_message(u_int32_t status)
1.1       markus    522: {
1.139     djm       523:        static const char * const status_messages[] = {
1.23      djm       524:                "Success",                      /* SSH_FX_OK */
                    525:                "End of file",                  /* SSH_FX_EOF */
                    526:                "No such file",                 /* SSH_FX_NO_SUCH_FILE */
                    527:                "Permission denied",            /* SSH_FX_PERMISSION_DENIED */
                    528:                "Failure",                      /* SSH_FX_FAILURE */
                    529:                "Bad message",                  /* SSH_FX_BAD_MESSAGE */
                    530:                "No connection",                /* SSH_FX_NO_CONNECTION */
                    531:                "Connection lost",              /* SSH_FX_CONNECTION_LOST */
                    532:                "Operation unsupported",        /* SSH_FX_OP_UNSUPPORTED */
                    533:                "Unknown error"                 /* Others */
                    534:        };
1.110     deraadt   535:        return (status_messages[MINIMUM(status,SSH2_FX_MAX)]);
1.58      djm       536: }
1.22      deraadt   537:
1.58      djm       538: static void
1.136     djm       539: send_status_errmsg(u_int32_t id, u_int32_t status, const char *errmsg)
1.58      djm       540: {
1.104     djm       541:        struct sshbuf *msg;
                    542:        int r;
1.58      djm       543:
                    544:        debug3("request %u: sent status %u", id, status);
                    545:        if (log_level > SYSLOG_LEVEL_VERBOSE ||
                    546:            (status != SSH2_FX_OK && status != SSH2_FX_EOF))
                    547:                logit("sent status %s", status_to_message(status));
1.104     djm       548:        if ((msg = sshbuf_new()) == NULL)
1.120     djm       549:                fatal_f("sshbuf_new failed");
1.104     djm       550:        if ((r = sshbuf_put_u8(msg, SSH2_FXP_STATUS)) != 0 ||
                    551:            (r = sshbuf_put_u32(msg, id)) != 0 ||
                    552:            (r = sshbuf_put_u32(msg, status)) != 0)
1.120     djm       553:                fatal_fr(r, "compose");
1.23      djm       554:        if (version >= 3) {
1.136     djm       555:                if ((r = sshbuf_put_cstring(msg, errmsg == NULL ?
                    556:                    status_to_message(status) : errmsg)) != 0 ||
1.104     djm       557:                    (r = sshbuf_put_cstring(msg, "")) != 0)
1.120     djm       558:                        fatal_fr(r, "compose message");
1.23      djm       559:        }
1.104     djm       560:        send_msg(msg);
                    561:        sshbuf_free(msg);
1.1       markus    562: }
1.136     djm       563:
                    564: static void
                    565: send_status(u_int32_t id, u_int32_t status)
                    566: {
1.137     dtucker   567:        send_status_errmsg(id, status, NULL);
1.136     djm       568: }
                    569:
1.28      itojun    570: static void
1.104     djm       571: send_data_or_handle(char type, u_int32_t id, const u_char *data, int dlen)
1.1       markus    572: {
1.104     djm       573:        struct sshbuf *msg;
                    574:        int r;
1.22      deraadt   575:
1.104     djm       576:        if ((msg = sshbuf_new()) == NULL)
1.120     djm       577:                fatal_f("sshbuf_new failed");
1.104     djm       578:        if ((r = sshbuf_put_u8(msg, type)) != 0 ||
                    579:            (r = sshbuf_put_u32(msg, id)) != 0 ||
                    580:            (r = sshbuf_put_string(msg, data, dlen)) != 0)
1.120     djm       581:                fatal_fr(r, "compose");
1.104     djm       582:        send_msg(msg);
                    583:        sshbuf_free(msg);
1.1       markus    584: }
                    585:
1.28      itojun    586: static void
1.104     djm       587: send_data(u_int32_t id, const u_char *data, int dlen)
1.1       markus    588: {
1.58      djm       589:        debug("request %u: sent data len %d", id, dlen);
1.10      markus    590:        send_data_or_handle(SSH2_FXP_DATA, id, data, dlen);
1.1       markus    591: }
                    592:
1.28      itojun    593: static void
1.1       markus    594: send_handle(u_int32_t id, int handle)
                    595: {
1.104     djm       596:        u_char *string;
1.1       markus    597:        int hlen;
1.22      deraadt   598:
1.1       markus    599:        handle_to_string(handle, &string, &hlen);
1.58      djm       600:        debug("request %u: sent handle handle %d", id, handle);
1.10      markus    601:        send_data_or_handle(SSH2_FXP_HANDLE, id, string, hlen);
1.97      djm       602:        free(string);
1.1       markus    603: }
                    604:
1.28      itojun    605: static void
1.44      jakob     606: send_names(u_int32_t id, int count, const Stat *stats)
1.1       markus    607: {
1.104     djm       608:        struct sshbuf *msg;
                    609:        int i, r;
1.22      deraadt   610:
1.104     djm       611:        if ((msg = sshbuf_new()) == NULL)
1.120     djm       612:                fatal_f("sshbuf_new failed");
1.104     djm       613:        if ((r = sshbuf_put_u8(msg, SSH2_FXP_NAME)) != 0 ||
                    614:            (r = sshbuf_put_u32(msg, id)) != 0 ||
                    615:            (r = sshbuf_put_u32(msg, count)) != 0)
1.120     djm       616:                fatal_fr(r, "compose");
1.58      djm       617:        debug("request %u: sent names count %d", id, count);
1.1       markus    618:        for (i = 0; i < count; i++) {
1.104     djm       619:                if ((r = sshbuf_put_cstring(msg, stats[i].name)) != 0 ||
                    620:                    (r = sshbuf_put_cstring(msg, stats[i].long_name)) != 0 ||
                    621:                    (r = encode_attrib(msg, &stats[i].attrib)) != 0)
1.120     djm       622:                        fatal_fr(r, "compose filenames/attrib");
1.1       markus    623:        }
1.104     djm       624:        send_msg(msg);
                    625:        sshbuf_free(msg);
1.1       markus    626: }
                    627:
1.28      itojun    628: static void
1.44      jakob     629: send_attrib(u_int32_t id, const Attrib *a)
1.1       markus    630: {
1.104     djm       631:        struct sshbuf *msg;
                    632:        int r;
1.22      deraadt   633:
1.58      djm       634:        debug("request %u: sent attrib have 0x%x", id, a->flags);
1.104     djm       635:        if ((msg = sshbuf_new()) == NULL)
1.120     djm       636:                fatal_f("sshbuf_new failed");
1.104     djm       637:        if ((r = sshbuf_put_u8(msg, SSH2_FXP_ATTRS)) != 0 ||
                    638:            (r = sshbuf_put_u32(msg, id)) != 0 ||
                    639:            (r = encode_attrib(msg, a)) != 0)
1.120     djm       640:                fatal_fr(r, "compose");
1.104     djm       641:        send_msg(msg);
                    642:        sshbuf_free(msg);
1.1       markus    643: }
                    644:
1.79      djm       645: static void
                    646: send_statvfs(u_int32_t id, struct statvfs *st)
                    647: {
1.104     djm       648:        struct sshbuf *msg;
1.79      djm       649:        u_int64_t flag;
1.104     djm       650:        int r;
1.79      djm       651:
                    652:        flag = (st->f_flag & ST_RDONLY) ? SSH2_FXE_STATVFS_ST_RDONLY : 0;
                    653:        flag |= (st->f_flag & ST_NOSUID) ? SSH2_FXE_STATVFS_ST_NOSUID : 0;
                    654:
1.104     djm       655:        if ((msg = sshbuf_new()) == NULL)
1.120     djm       656:                fatal_f("sshbuf_new failed");
1.104     djm       657:        if ((r = sshbuf_put_u8(msg, SSH2_FXP_EXTENDED_REPLY)) != 0 ||
                    658:            (r = sshbuf_put_u32(msg, id)) != 0 ||
                    659:            (r = sshbuf_put_u64(msg, st->f_bsize)) != 0 ||
                    660:            (r = sshbuf_put_u64(msg, st->f_frsize)) != 0 ||
                    661:            (r = sshbuf_put_u64(msg, st->f_blocks)) != 0 ||
                    662:            (r = sshbuf_put_u64(msg, st->f_bfree)) != 0 ||
                    663:            (r = sshbuf_put_u64(msg, st->f_bavail)) != 0 ||
                    664:            (r = sshbuf_put_u64(msg, st->f_files)) != 0 ||
                    665:            (r = sshbuf_put_u64(msg, st->f_ffree)) != 0 ||
                    666:            (r = sshbuf_put_u64(msg, st->f_favail)) != 0 ||
                    667:            (r = sshbuf_put_u64(msg, st->f_fsid)) != 0 ||
                    668:            (r = sshbuf_put_u64(msg, flag)) != 0 ||
                    669:            (r = sshbuf_put_u64(msg, st->f_namemax)) != 0)
1.120     djm       670:                fatal_fr(r, "compose");
1.104     djm       671:        send_msg(msg);
                    672:        sshbuf_free(msg);
1.79      djm       673: }
                    674:
1.125     djm       675: /*
                    676:  * Prepare SSH2_FXP_VERSION extension advertisement for a single extension.
1.135     jsg       677:  * The extension is checked for permission prior to advertisement.
1.125     djm       678:  */
                    679: static int
                    680: compose_extension(struct sshbuf *msg, const char *name, const char *ver)
                    681: {
                    682:        int r;
                    683:        const struct sftp_handler *exthnd;
                    684:
                    685:        if ((exthnd = extended_handler_byname(name)) == NULL)
                    686:                fatal_f("internal error: no handler for %s", name);
                    687:        if (!request_permitted(exthnd)) {
                    688:                debug2_f("refusing to advertise disallowed extension %s", name);
                    689:                return 0;
                    690:        }
                    691:        if ((r = sshbuf_put_cstring(msg, name)) != 0 ||
                    692:            (r = sshbuf_put_cstring(msg, ver)) != 0)
                    693:                fatal_fr(r, "compose %s", name);
                    694:        return 0;
                    695: }
                    696:
1.1       markus    697: /* parse incoming */
                    698:
1.28      itojun    699: static void
1.1       markus    700: process_init(void)
                    701: {
1.104     djm       702:        struct sshbuf *msg;
                    703:        int r;
1.1       markus    704:
1.104     djm       705:        if ((r = sshbuf_get_u32(iqueue, &version)) != 0)
1.120     djm       706:                fatal_fr(r, "parse");
1.94      djm       707:        verbose("received client version %u", version);
1.104     djm       708:        if ((msg = sshbuf_new()) == NULL)
1.120     djm       709:                fatal_f("sshbuf_new failed");
1.104     djm       710:        if ((r = sshbuf_put_u8(msg, SSH2_FXP_VERSION)) != 0 ||
1.125     djm       711:            (r = sshbuf_put_u32(msg, SSH2_FILEXFER_VERSION)) != 0)
1.120     djm       712:                fatal_fr(r, "compose");
1.125     djm       713:
1.135     jsg       714:         /* extension advertisements */
1.125     djm       715:        compose_extension(msg, "posix-rename@openssh.com", "1");
                    716:        compose_extension(msg, "statvfs@openssh.com", "2");
                    717:        compose_extension(msg, "fstatvfs@openssh.com", "2");
                    718:        compose_extension(msg, "hardlink@openssh.com", "1");
                    719:        compose_extension(msg, "fsync@openssh.com", "1");
                    720:        compose_extension(msg, "lsetstat@openssh.com", "1");
                    721:        compose_extension(msg, "limits@openssh.com", "1");
1.129     djm       722:        compose_extension(msg, "expand-path@openssh.com", "1");
1.140     djm       723:        compose_extension(msg, "copy-data", "1");
1.141     djm       724:        compose_extension(msg, "home-directory", "1");
1.143   ! djm       725:        compose_extension(msg, "users-groups-by-id@openssh.com", "1");
1.125     djm       726:
1.104     djm       727:        send_msg(msg);
                    728:        sshbuf_free(msg);
1.1       markus    729: }
                    730:
1.28      itojun    731: static void
1.98      djm       732: process_open(u_int32_t id)
1.1       markus    733: {
1.98      djm       734:        u_int32_t pflags;
1.104     djm       735:        Attrib a;
1.1       markus    736:        char *name;
1.104     djm       737:        int r, handle, fd, flags, mode, status = SSH2_FX_FAILURE;
                    738:
                    739:        if ((r = sshbuf_get_cstring(iqueue, &name, NULL)) != 0 ||
                    740:            (r = sshbuf_get_u32(iqueue, &pflags)) != 0 || /* portable flags */
                    741:            (r = decode_attrib(iqueue, &a)) != 0)
1.120     djm       742:                fatal_fr(r, "parse");
1.1       markus    743:
1.61      djm       744:        debug3("request %u: open flags %d", id, pflags);
1.1       markus    745:        flags = flags_from_portable(pflags);
1.104     djm       746:        mode = (a.flags & SSH2_FILEXFER_ATTR_PERMISSIONS) ? a.perm : 0666;
1.58      djm       747:        logit("open \"%s\" flags %s mode 0%o",
                    748:            name, string_from_portable(pflags), mode);
1.90      djm       749:        if (readonly &&
1.111     djm       750:            ((flags & O_ACCMODE) != O_RDONLY ||
                    751:            (flags & (O_CREAT|O_TRUNC)) != 0)) {
1.98      djm       752:                verbose("Refusing open request in read-only mode");
1.104     djm       753:                status = SSH2_FX_PERMISSION_DENIED;
1.98      djm       754:        } else {
1.90      djm       755:                fd = open(name, flags, mode);
1.116     deraadt   756:                if (fd == -1) {
1.90      djm       757:                        status = errno_to_portable(errno);
1.1       markus    758:                } else {
1.101     djm       759:                        handle = handle_new(HANDLE_FILE, name, fd, flags, NULL);
1.90      djm       760:                        if (handle < 0) {
                    761:                                close(fd);
                    762:                        } else {
                    763:                                send_handle(id, handle);
                    764:                                status = SSH2_FX_OK;
                    765:                        }
1.1       markus    766:                }
                    767:        }
1.10      markus    768:        if (status != SSH2_FX_OK)
1.1       markus    769:                send_status(id, status);
1.97      djm       770:        free(name);
1.1       markus    771: }
                    772:
1.28      itojun    773: static void
1.98      djm       774: process_close(u_int32_t id)
1.1       markus    775: {
1.104     djm       776:        int r, handle, ret, status = SSH2_FX_FAILURE;
                    777:
                    778:        if ((r = get_handle(iqueue, &handle)) != 0)
1.120     djm       779:                fatal_fr(r, "parse");
1.1       markus    780:
1.58      djm       781:        debug3("request %u: close handle %u", id, handle);
                    782:        handle_log_close(handle, NULL);
1.1       markus    783:        ret = handle_close(handle);
1.10      markus    784:        status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
1.1       markus    785:        send_status(id, status);
                    786: }
                    787:
1.28      itojun    788: static void
1.98      djm       789: process_read(u_int32_t id)
1.1       markus    790: {
1.124     djm       791:        static u_char *buf;
                    792:        static size_t buflen;
1.98      djm       793:        u_int32_t len;
1.104     djm       794:        int r, handle, fd, ret, status = SSH2_FX_FAILURE;
1.1       markus    795:        u_int64_t off;
                    796:
1.104     djm       797:        if ((r = get_handle(iqueue, &handle)) != 0 ||
                    798:            (r = sshbuf_get_u64(iqueue, &off)) != 0 ||
                    799:            (r = sshbuf_get_u32(iqueue, &len)) != 0)
1.120     djm       800:                fatal_fr(r, "parse");
1.1       markus    801:
1.124     djm       802:        debug("request %u: read \"%s\" (handle %d) off %llu len %u",
1.58      djm       803:            id, handle_to_name(handle), handle, (unsigned long long)off, len);
1.124     djm       804:        if ((fd = handle_to_fd(handle)) == -1)
                    805:                goto out;
                    806:        if (len > SFTP_MAX_READ_LENGTH) {
                    807:                debug2("read change len %u to %u", len, SFTP_MAX_READ_LENGTH);
                    808:                len = SFTP_MAX_READ_LENGTH;
                    809:        }
                    810:        if (len > buflen) {
                    811:                debug3_f("allocate %zu => %u", buflen, len);
                    812:                if ((buf = realloc(NULL, len)) == NULL)
                    813:                        fatal_f("realloc failed");
                    814:                buflen = len;
                    815:        }
                    816:        if (lseek(fd, off, SEEK_SET) == -1) {
                    817:                status = errno_to_portable(errno);
                    818:                error_f("seek \"%.100s\": %s", handle_to_name(handle),
                    819:                    strerror(errno));
                    820:                goto out;
                    821:        }
                    822:        if (len == 0) {
                    823:                /* weird, but not strictly disallowed */
                    824:                ret = 0;
                    825:        } else if ((ret = read(fd, buf, len)) == -1) {
                    826:                status = errno_to_portable(errno);
                    827:                error_f("read \"%.100s\": %s", handle_to_name(handle),
                    828:                    strerror(errno));
                    829:                goto out;
                    830:        } else if (ret == 0) {
                    831:                status = SSH2_FX_EOF;
                    832:                goto out;
                    833:        }
                    834:        send_data(id, buf, ret);
                    835:        handle_update_read(handle, ret);
                    836:        /* success */
                    837:        status = SSH2_FX_OK;
                    838:  out:
1.10      markus    839:        if (status != SSH2_FX_OK)
1.1       markus    840:                send_status(id, status);
                    841: }
                    842:
1.28      itojun    843: static void
1.98      djm       844: process_write(u_int32_t id)
1.1       markus    845: {
                    846:        u_int64_t off;
1.104     djm       847:        size_t len;
                    848:        int r, handle, fd, ret, status;
                    849:        u_char *data;
                    850:
                    851:        if ((r = get_handle(iqueue, &handle)) != 0 ||
                    852:            (r = sshbuf_get_u64(iqueue, &off)) != 0 ||
                    853:            (r = sshbuf_get_string(iqueue, &data, &len)) != 0)
1.120     djm       854:                fatal_fr(r, "parse");
1.1       markus    855:
1.104     djm       856:        debug("request %u: write \"%s\" (handle %d) off %llu len %zu",
1.58      djm       857:            id, handle_to_name(handle), handle, (unsigned long long)off, len);
1.1       markus    858:        fd = handle_to_fd(handle);
1.104     djm       859:
1.90      djm       860:        if (fd < 0)
                    861:                status = SSH2_FX_FAILURE;
                    862:        else {
1.101     djm       863:                if (!(handle_to_flags(handle) & O_APPEND) &&
1.123     djm       864:                    lseek(fd, off, SEEK_SET) == -1) {
1.1       markus    865:                        status = errno_to_portable(errno);
1.123     djm       866:                        error_f("seek \"%.100s\": %s", handle_to_name(handle),
                    867:                            strerror(errno));
1.1       markus    868:                } else {
                    869: /* XXX ATOMICIO ? */
                    870:                        ret = write(fd, data, len);
1.116     deraadt   871:                        if (ret == -1) {
1.1       markus    872:                                status = errno_to_portable(errno);
1.123     djm       873:                                error_f("write \"%.100s\": %s",
                    874:                                    handle_to_name(handle), strerror(errno));
1.48      djm       875:                        } else if ((size_t)ret == len) {
1.10      markus    876:                                status = SSH2_FX_OK;
1.58      djm       877:                                handle_update_write(handle, ret);
1.1       markus    878:                        } else {
1.120     djm       879:                                debug2_f("nothing at all written");
1.90      djm       880:                                status = SSH2_FX_FAILURE;
1.1       markus    881:                        }
                    882:                }
                    883:        }
                    884:        send_status(id, status);
1.97      djm       885:        free(data);
1.1       markus    886: }
                    887:
1.28      itojun    888: static void
1.98      djm       889: process_do_stat(u_int32_t id, int do_lstat)
1.1       markus    890: {
1.13      markus    891:        Attrib a;
1.1       markus    892:        struct stat st;
                    893:        char *name;
1.104     djm       894:        int r, status = SSH2_FX_FAILURE;
                    895:
                    896:        if ((r = sshbuf_get_cstring(iqueue, &name, NULL)) != 0)
1.120     djm       897:                fatal_fr(r, "parse");
1.1       markus    898:
1.58      djm       899:        debug3("request %u: %sstat", id, do_lstat ? "l" : "");
                    900:        verbose("%sstat name \"%s\"", do_lstat ? "l" : "", name);
1.104     djm       901:        r = do_lstat ? lstat(name, &st) : stat(name, &st);
1.116     deraadt   902:        if (r == -1) {
1.1       markus    903:                status = errno_to_portable(errno);
                    904:        } else {
1.13      markus    905:                stat_to_attrib(&st, &a);
                    906:                send_attrib(id, &a);
1.10      markus    907:                status = SSH2_FX_OK;
1.1       markus    908:        }
1.10      markus    909:        if (status != SSH2_FX_OK)
1.1       markus    910:                send_status(id, status);
1.97      djm       911:        free(name);
1.1       markus    912: }
                    913:
1.28      itojun    914: static void
1.98      djm       915: process_stat(u_int32_t id)
1.1       markus    916: {
1.98      djm       917:        process_do_stat(id, 0);
1.1       markus    918: }
                    919:
1.28      itojun    920: static void
1.98      djm       921: process_lstat(u_int32_t id)
1.1       markus    922: {
1.98      djm       923:        process_do_stat(id, 1);
1.1       markus    924: }
                    925:
1.28      itojun    926: static void
1.98      djm       927: process_fstat(u_int32_t id)
1.1       markus    928: {
1.13      markus    929:        Attrib a;
1.1       markus    930:        struct stat st;
1.104     djm       931:        int fd, r, handle, status = SSH2_FX_FAILURE;
1.1       markus    932:
1.104     djm       933:        if ((r = get_handle(iqueue, &handle)) != 0)
1.120     djm       934:                fatal_fr(r, "parse");
1.58      djm       935:        debug("request %u: fstat \"%s\" (handle %u)",
                    936:            id, handle_to_name(handle), handle);
1.1       markus    937:        fd = handle_to_fd(handle);
1.71      stevesk   938:        if (fd >= 0) {
1.104     djm       939:                r = fstat(fd, &st);
1.116     deraadt   940:                if (r == -1) {
1.1       markus    941:                        status = errno_to_portable(errno);
                    942:                } else {
1.13      markus    943:                        stat_to_attrib(&st, &a);
                    944:                        send_attrib(id, &a);
1.10      markus    945:                        status = SSH2_FX_OK;
1.1       markus    946:                }
                    947:        }
1.10      markus    948:        if (status != SSH2_FX_OK)
1.1       markus    949:                send_status(id, status);
                    950: }
                    951:
1.28      itojun    952: static struct timeval *
1.44      jakob     953: attrib_to_tv(const Attrib *a)
1.1       markus    954: {
                    955:        static struct timeval tv[2];
1.22      deraadt   956:
1.1       markus    957:        tv[0].tv_sec = a->atime;
                    958:        tv[0].tv_usec = 0;
                    959:        tv[1].tv_sec = a->mtime;
                    960:        tv[1].tv_usec = 0;
                    961:        return tv;
                    962: }
                    963:
1.114     djm       964: static struct timespec *
                    965: attrib_to_ts(const Attrib *a)
                    966: {
                    967:        static struct timespec ts[2];
                    968:
                    969:        ts[0].tv_sec = a->atime;
                    970:        ts[0].tv_nsec = 0;
                    971:        ts[1].tv_sec = a->mtime;
                    972:        ts[1].tv_nsec = 0;
                    973:        return ts;
                    974: }
                    975:
1.28      itojun    976: static void
1.98      djm       977: process_setstat(u_int32_t id)
1.1       markus    978: {
1.104     djm       979:        Attrib a;
1.1       markus    980:        char *name;
1.104     djm       981:        int r, status = SSH2_FX_OK;
                    982:
                    983:        if ((r = sshbuf_get_cstring(iqueue, &name, NULL)) != 0 ||
                    984:            (r = decode_attrib(iqueue, &a)) != 0)
1.120     djm       985:                fatal_fr(r, "parse");
1.1       markus    986:
1.58      djm       987:        debug("request %u: setstat name \"%s\"", id, name);
1.104     djm       988:        if (a.flags & SSH2_FILEXFER_ATTR_SIZE) {
1.72      stevesk   989:                logit("set \"%s\" size %llu",
1.104     djm       990:                    name, (unsigned long long)a.size);
                    991:                r = truncate(name, a.size);
                    992:                if (r == -1)
1.33      markus    993:                        status = errno_to_portable(errno);
                    994:        }
1.104     djm       995:        if (a.flags & SSH2_FILEXFER_ATTR_PERMISSIONS) {
                    996:                logit("set \"%s\" mode %04o", name, a.perm);
                    997:                r = chmod(name, a.perm & 07777);
                    998:                if (r == -1)
1.1       markus    999:                        status = errno_to_portable(errno);
                   1000:        }
1.104     djm      1001:        if (a.flags & SSH2_FILEXFER_ATTR_ACMODTIME) {
1.58      djm      1002:                char buf[64];
1.104     djm      1003:                time_t t = a.mtime;
1.58      djm      1004:
                   1005:                strftime(buf, sizeof(buf), "%Y%m%d-%H:%M:%S",
                   1006:                    localtime(&t));
                   1007:                logit("set \"%s\" modtime %s", name, buf);
1.104     djm      1008:                r = utimes(name, attrib_to_tv(&a));
                   1009:                if (r == -1)
1.1       markus   1010:                        status = errno_to_portable(errno);
                   1011:        }
1.104     djm      1012:        if (a.flags & SSH2_FILEXFER_ATTR_UIDGID) {
1.58      djm      1013:                logit("set \"%s\" owner %lu group %lu", name,
1.104     djm      1014:                    (u_long)a.uid, (u_long)a.gid);
                   1015:                r = chown(name, a.uid, a.gid);
                   1016:                if (r == -1)
1.18      stevesk  1017:                        status = errno_to_portable(errno);
                   1018:        }
1.1       markus   1019:        send_status(id, status);
1.97      djm      1020:        free(name);
1.1       markus   1021: }
                   1022:
1.28      itojun   1023: static void
1.98      djm      1024: process_fsetstat(u_int32_t id)
1.1       markus   1025: {
1.104     djm      1026:        Attrib a;
                   1027:        int handle, fd, r;
1.10      markus   1028:        int status = SSH2_FX_OK;
1.1       markus   1029:
1.104     djm      1030:        if ((r = get_handle(iqueue, &handle)) != 0 ||
                   1031:            (r = decode_attrib(iqueue, &a)) != 0)
1.120     djm      1032:                fatal_fr(r, "parse");
1.104     djm      1033:
1.58      djm      1034:        debug("request %u: fsetstat handle %d", id, handle);
1.1       markus   1035:        fd = handle_to_fd(handle);
1.90      djm      1036:        if (fd < 0)
1.10      markus   1037:                status = SSH2_FX_FAILURE;
1.90      djm      1038:        else {
1.58      djm      1039:                char *name = handle_to_name(handle);
                   1040:
1.104     djm      1041:                if (a.flags & SSH2_FILEXFER_ATTR_SIZE) {
1.72      stevesk  1042:                        logit("set \"%s\" size %llu",
1.104     djm      1043:                            name, (unsigned long long)a.size);
                   1044:                        r = ftruncate(fd, a.size);
                   1045:                        if (r == -1)
1.33      markus   1046:                                status = errno_to_portable(errno);
                   1047:                }
1.104     djm      1048:                if (a.flags & SSH2_FILEXFER_ATTR_PERMISSIONS) {
                   1049:                        logit("set \"%s\" mode %04o", name, a.perm);
                   1050:                        r = fchmod(fd, a.perm & 07777);
                   1051:                        if (r == -1)
1.1       markus   1052:                                status = errno_to_portable(errno);
                   1053:                }
1.104     djm      1054:                if (a.flags & SSH2_FILEXFER_ATTR_ACMODTIME) {
1.58      djm      1055:                        char buf[64];
1.104     djm      1056:                        time_t t = a.mtime;
1.58      djm      1057:
                   1058:                        strftime(buf, sizeof(buf), "%Y%m%d-%H:%M:%S",
                   1059:                            localtime(&t));
                   1060:                        logit("set \"%s\" modtime %s", name, buf);
1.104     djm      1061:                        r = futimes(fd, attrib_to_tv(&a));
                   1062:                        if (r == -1)
1.18      stevesk  1063:                                status = errno_to_portable(errno);
                   1064:                }
1.104     djm      1065:                if (a.flags & SSH2_FILEXFER_ATTR_UIDGID) {
1.58      djm      1066:                        logit("set \"%s\" owner %lu group %lu", name,
1.104     djm      1067:                            (u_long)a.uid, (u_long)a.gid);
                   1068:                        r = fchown(fd, a.uid, a.gid);
                   1069:                        if (r == -1)
1.1       markus   1070:                                status = errno_to_portable(errno);
                   1071:                }
                   1072:        }
                   1073:        send_status(id, status);
                   1074: }
                   1075:
1.28      itojun   1076: static void
1.98      djm      1077: process_opendir(u_int32_t id)
1.1       markus   1078: {
                   1079:        DIR *dirp = NULL;
                   1080:        char *path;
1.104     djm      1081:        int r, handle, status = SSH2_FX_FAILURE;
                   1082:
                   1083:        if ((r = sshbuf_get_cstring(iqueue, &path, NULL)) != 0)
1.120     djm      1084:                fatal_fr(r, "parse");
1.1       markus   1085:
1.58      djm      1086:        debug3("request %u: opendir", id);
                   1087:        logit("opendir \"%s\"", path);
1.17      stevesk  1088:        dirp = opendir(path);
1.1       markus   1089:        if (dirp == NULL) {
                   1090:                status = errno_to_portable(errno);
                   1091:        } else {
1.101     djm      1092:                handle = handle_new(HANDLE_DIR, path, 0, 0, dirp);
1.1       markus   1093:                if (handle < 0) {
                   1094:                        closedir(dirp);
                   1095:                } else {
                   1096:                        send_handle(id, handle);
1.10      markus   1097:                        status = SSH2_FX_OK;
1.1       markus   1098:                }
1.17      stevesk  1099:
1.1       markus   1100:        }
1.10      markus   1101:        if (status != SSH2_FX_OK)
1.1       markus   1102:                send_status(id, status);
1.97      djm      1103:        free(path);
1.1       markus   1104: }
                   1105:
1.28      itojun   1106: static void
1.98      djm      1107: process_readdir(u_int32_t id)
1.1       markus   1108: {
                   1109:        DIR *dirp;
                   1110:        struct dirent *dp;
                   1111:        char *path;
1.104     djm      1112:        int r, handle;
                   1113:
                   1114:        if ((r = get_handle(iqueue, &handle)) != 0)
1.120     djm      1115:                fatal_fr(r, "parse");
1.1       markus   1116:
1.58      djm      1117:        debug("request %u: readdir \"%s\" (handle %d)", id,
                   1118:            handle_to_name(handle), handle);
1.1       markus   1119:        dirp = handle_to_dir(handle);
                   1120:        path = handle_to_name(handle);
                   1121:        if (dirp == NULL || path == NULL) {
1.10      markus   1122:                send_status(id, SSH2_FX_FAILURE);
1.1       markus   1123:        } else {
                   1124:                struct stat st;
1.105     deraadt  1125:                char pathname[PATH_MAX];
1.1       markus   1126:                Stat *stats;
                   1127:                int nstats = 10, count = 0, i;
1.36      deraadt  1128:
1.54      djm      1129:                stats = xcalloc(nstats, sizeof(Stat));
1.1       markus   1130:                while ((dp = readdir(dirp)) != NULL) {
                   1131:                        if (count >= nstats) {
                   1132:                                nstats *= 2;
1.106     deraadt  1133:                                stats = xreallocarray(stats, nstats, sizeof(Stat));
1.1       markus   1134:                        }
                   1135: /* XXX OVERFLOW ? */
1.30      jakob    1136:                        snprintf(pathname, sizeof pathname, "%s%s%s", path,
                   1137:                            strcmp(path, "/") ? "/" : "", dp->d_name);
1.116     deraadt  1138:                        if (lstat(pathname, &st) == -1)
1.1       markus   1139:                                continue;
1.13      markus   1140:                        stat_to_attrib(&st, &(stats[count].attrib));
1.1       markus   1141:                        stats[count].name = xstrdup(dp->d_name);
1.91      djm      1142:                        stats[count].long_name = ls_file(dp->d_name, &st, 0, 0);
1.1       markus   1143:                        count++;
                   1144:                        /* send up to 100 entries in one message */
1.11      markus   1145:                        /* XXX check packet size instead */
1.1       markus   1146:                        if (count == 100)
                   1147:                                break;
                   1148:                }
1.10      markus   1149:                if (count > 0) {
                   1150:                        send_names(id, count, stats);
1.31      deraadt  1151:                        for (i = 0; i < count; i++) {
1.97      djm      1152:                                free(stats[i].name);
                   1153:                                free(stats[i].long_name);
1.10      markus   1154:                        }
                   1155:                } else {
                   1156:                        send_status(id, SSH2_FX_EOF);
1.1       markus   1157:                }
1.97      djm      1158:                free(stats);
1.1       markus   1159:        }
                   1160: }
                   1161:
1.28      itojun   1162: static void
1.98      djm      1163: process_remove(u_int32_t id)
1.1       markus   1164: {
                   1165:        char *name;
1.104     djm      1166:        int r, status = SSH2_FX_FAILURE;
                   1167:
                   1168:        if ((r = sshbuf_get_cstring(iqueue, &name, NULL)) != 0)
1.120     djm      1169:                fatal_fr(r, "parse");
1.1       markus   1170:
1.58      djm      1171:        debug3("request %u: remove", id);
                   1172:        logit("remove name \"%s\"", name);
1.104     djm      1173:        r = unlink(name);
                   1174:        status = (r == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
1.1       markus   1175:        send_status(id, status);
1.97      djm      1176:        free(name);
1.1       markus   1177: }
                   1178:
1.28      itojun   1179: static void
1.98      djm      1180: process_mkdir(u_int32_t id)
1.1       markus   1181: {
1.104     djm      1182:        Attrib a;
1.1       markus   1183:        char *name;
1.104     djm      1184:        int r, mode, status = SSH2_FX_FAILURE;
                   1185:
                   1186:        if ((r = sshbuf_get_cstring(iqueue, &name, NULL)) != 0 ||
                   1187:            (r = decode_attrib(iqueue, &a)) != 0)
1.120     djm      1188:                fatal_fr(r, "parse");
1.1       markus   1189:
1.104     djm      1190:        mode = (a.flags & SSH2_FILEXFER_ATTR_PERMISSIONS) ?
                   1191:            a.perm & 07777 : 0777;
1.58      djm      1192:        debug3("request %u: mkdir", id);
                   1193:        logit("mkdir name \"%s\" mode 0%o", name, mode);
1.104     djm      1194:        r = mkdir(name, mode);
                   1195:        status = (r == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
1.1       markus   1196:        send_status(id, status);
1.97      djm      1197:        free(name);
1.1       markus   1198: }
                   1199:
1.28      itojun   1200: static void
1.98      djm      1201: process_rmdir(u_int32_t id)
1.1       markus   1202: {
                   1203:        char *name;
1.104     djm      1204:        int r, status;
                   1205:
                   1206:        if ((r = sshbuf_get_cstring(iqueue, &name, NULL)) != 0)
1.120     djm      1207:                fatal_fr(r, "parse");
1.1       markus   1208:
1.58      djm      1209:        debug3("request %u: rmdir", id);
                   1210:        logit("rmdir name \"%s\"", name);
1.104     djm      1211:        r = rmdir(name);
                   1212:        status = (r == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
1.1       markus   1213:        send_status(id, status);
1.97      djm      1214:        free(name);
1.1       markus   1215: }
                   1216:
1.28      itojun   1217: static void
1.98      djm      1218: process_realpath(u_int32_t id)
1.1       markus   1219: {
1.105     deraadt  1220:        char resolvedname[PATH_MAX];
1.1       markus   1221:        char *path;
1.104     djm      1222:        int r;
                   1223:
                   1224:        if ((r = sshbuf_get_cstring(iqueue, &path, NULL)) != 0)
1.120     djm      1225:                fatal_fr(r, "parse");
1.1       markus   1226:
1.7       markus   1227:        if (path[0] == '\0') {
1.97      djm      1228:                free(path);
1.7       markus   1229:                path = xstrdup(".");
                   1230:        }
1.58      djm      1231:        debug3("request %u: realpath", id);
                   1232:        verbose("realpath \"%s\"", path);
1.117     djm      1233:        if (sftp_realpath(path, resolvedname) == NULL) {
1.1       markus   1234:                send_status(id, errno_to_portable(errno));
                   1235:        } else {
                   1236:                Stat s;
                   1237:                attrib_clear(&s.attrib);
                   1238:                s.name = s.long_name = resolvedname;
                   1239:                send_names(id, 1, &s);
                   1240:        }
1.97      djm      1241:        free(path);
1.1       markus   1242: }
                   1243:
1.28      itojun   1244: static void
1.98      djm      1245: process_rename(u_int32_t id)
1.1       markus   1246: {
                   1247:        char *oldpath, *newpath;
1.104     djm      1248:        int r, status;
1.41      deraadt  1249:        struct stat sb;
1.1       markus   1250:
1.104     djm      1251:        if ((r = sshbuf_get_cstring(iqueue, &oldpath, NULL)) != 0 ||
                   1252:            (r = sshbuf_get_cstring(iqueue, &newpath, NULL)) != 0)
1.120     djm      1253:                fatal_fr(r, "parse");
1.104     djm      1254:
1.58      djm      1255:        debug3("request %u: rename", id);
                   1256:        logit("rename old \"%s\" new \"%s\"", oldpath, newpath);
1.41      deraadt  1257:        status = SSH2_FX_FAILURE;
1.98      djm      1258:        if (lstat(oldpath, &sb) == -1)
1.39      markus   1259:                status = errno_to_portable(errno);
1.41      deraadt  1260:        else if (S_ISREG(sb.st_mode)) {
                   1261:                /* Race-free rename of regular files */
1.47      dtucker  1262:                if (link(oldpath, newpath) == -1) {
                   1263:                        if (errno == EOPNOTSUPP) {
                   1264:                                struct stat st;
                   1265:
                   1266:                                /*
                   1267:                                 * fs doesn't support links, so fall back to
                   1268:                                 * stat+rename.  This is racy.
                   1269:                                 */
                   1270:                                if (stat(newpath, &st) == -1) {
                   1271:                                        if (rename(oldpath, newpath) == -1)
                   1272:                                                status =
                   1273:                                                    errno_to_portable(errno);
                   1274:                                        else
                   1275:                                                status = SSH2_FX_OK;
                   1276:                                }
                   1277:                        } else {
                   1278:                                status = errno_to_portable(errno);
                   1279:                        }
                   1280:                } else if (unlink(oldpath) == -1) {
1.41      deraadt  1281:                        status = errno_to_portable(errno);
                   1282:                        /* clean spare link */
                   1283:                        unlink(newpath);
                   1284:                } else
                   1285:                        status = SSH2_FX_OK;
                   1286:        } else if (stat(newpath, &sb) == -1) {
                   1287:                if (rename(oldpath, newpath) == -1)
                   1288:                        status = errno_to_portable(errno);
                   1289:                else
                   1290:                        status = SSH2_FX_OK;
                   1291:        }
1.1       markus   1292:        send_status(id, status);
1.97      djm      1293:        free(oldpath);
                   1294:        free(newpath);
1.1       markus   1295: }
                   1296:
1.28      itojun   1297: static void
1.98      djm      1298: process_readlink(u_int32_t id)
1.23      djm      1299: {
1.104     djm      1300:        int r, len;
1.105     deraadt  1301:        char buf[PATH_MAX];
1.23      djm      1302:        char *path;
                   1303:
1.104     djm      1304:        if ((r = sshbuf_get_cstring(iqueue, &path, NULL)) != 0)
1.120     djm      1305:                fatal_fr(r, "parse");
1.104     djm      1306:
1.58      djm      1307:        debug3("request %u: readlink", id);
                   1308:        verbose("readlink \"%s\"", path);
1.46      avsm     1309:        if ((len = readlink(path, buf, sizeof(buf) - 1)) == -1)
1.23      djm      1310:                send_status(id, errno_to_portable(errno));
                   1311:        else {
                   1312:                Stat s;
1.31      deraadt  1313:
1.46      avsm     1314:                buf[len] = '\0';
1.23      djm      1315:                attrib_clear(&s.attrib);
1.46      avsm     1316:                s.name = s.long_name = buf;
1.23      djm      1317:                send_names(id, 1, &s);
                   1318:        }
1.97      djm      1319:        free(path);
1.23      djm      1320: }
                   1321:
1.28      itojun   1322: static void
1.98      djm      1323: process_symlink(u_int32_t id)
1.23      djm      1324: {
                   1325:        char *oldpath, *newpath;
1.104     djm      1326:        int r, status;
                   1327:
                   1328:        if ((r = sshbuf_get_cstring(iqueue, &oldpath, NULL)) != 0 ||
                   1329:            (r = sshbuf_get_cstring(iqueue, &newpath, NULL)) != 0)
1.120     djm      1330:                fatal_fr(r, "parse");
1.23      djm      1331:
1.58      djm      1332:        debug3("request %u: symlink", id);
                   1333:        logit("symlink old \"%s\" new \"%s\"", oldpath, newpath);
1.39      markus   1334:        /* this will fail if 'newpath' exists */
1.104     djm      1335:        r = symlink(oldpath, newpath);
                   1336:        status = (r == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
1.23      djm      1337:        send_status(id, status);
1.97      djm      1338:        free(oldpath);
                   1339:        free(newpath);
1.23      djm      1340: }
                   1341:
1.28      itojun   1342: static void
1.78      djm      1343: process_extended_posix_rename(u_int32_t id)
                   1344: {
                   1345:        char *oldpath, *newpath;
1.104     djm      1346:        int r, status;
                   1347:
                   1348:        if ((r = sshbuf_get_cstring(iqueue, &oldpath, NULL)) != 0 ||
                   1349:            (r = sshbuf_get_cstring(iqueue, &newpath, NULL)) != 0)
1.120     djm      1350:                fatal_fr(r, "parse");
1.78      djm      1351:
                   1352:        debug3("request %u: posix-rename", id);
                   1353:        logit("posix-rename old \"%s\" new \"%s\"", oldpath, newpath);
1.104     djm      1354:        r = rename(oldpath, newpath);
                   1355:        status = (r == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
1.90      djm      1356:        send_status(id, status);
1.97      djm      1357:        free(oldpath);
                   1358:        free(newpath);
1.78      djm      1359: }
                   1360:
                   1361: static void
1.79      djm      1362: process_extended_statvfs(u_int32_t id)
                   1363: {
                   1364:        char *path;
                   1365:        struct statvfs st;
1.104     djm      1366:        int r;
1.79      djm      1367:
1.104     djm      1368:        if ((r = sshbuf_get_cstring(iqueue, &path, NULL)) != 0)
1.120     djm      1369:                fatal_fr(r, "parse");
1.103     dtucker  1370:        debug3("request %u: statvfs", id);
                   1371:        logit("statvfs \"%s\"", path);
1.79      djm      1372:
                   1373:        if (statvfs(path, &st) != 0)
                   1374:                send_status(id, errno_to_portable(errno));
                   1375:        else
                   1376:                send_statvfs(id, &st);
1.127     djm      1377:        free(path);
1.79      djm      1378: }
                   1379:
                   1380: static void
                   1381: process_extended_fstatvfs(u_int32_t id)
                   1382: {
1.104     djm      1383:        int r, handle, fd;
1.79      djm      1384:        struct statvfs st;
                   1385:
1.104     djm      1386:        if ((r = get_handle(iqueue, &handle)) != 0)
1.120     djm      1387:                fatal_fr(r, "parse");
1.79      djm      1388:        debug("request %u: fstatvfs \"%s\" (handle %u)",
                   1389:            id, handle_to_name(handle), handle);
                   1390:        if ((fd = handle_to_fd(handle)) < 0) {
                   1391:                send_status(id, SSH2_FX_FAILURE);
                   1392:                return;
                   1393:        }
                   1394:        if (fstatvfs(fd, &st) != 0)
                   1395:                send_status(id, errno_to_portable(errno));
                   1396:        else
                   1397:                send_statvfs(id, &st);
                   1398: }
                   1399:
                   1400: static void
1.93      djm      1401: process_extended_hardlink(u_int32_t id)
                   1402: {
                   1403:        char *oldpath, *newpath;
1.104     djm      1404:        int r, status;
                   1405:
                   1406:        if ((r = sshbuf_get_cstring(iqueue, &oldpath, NULL)) != 0 ||
                   1407:            (r = sshbuf_get_cstring(iqueue, &newpath, NULL)) != 0)
1.120     djm      1408:                fatal_fr(r, "parse");
1.93      djm      1409:
                   1410:        debug3("request %u: hardlink", id);
                   1411:        logit("hardlink old \"%s\" new \"%s\"", oldpath, newpath);
1.104     djm      1412:        r = link(oldpath, newpath);
                   1413:        status = (r == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
1.93      djm      1414:        send_status(id, status);
1.97      djm      1415:        free(oldpath);
                   1416:        free(newpath);
1.102     djm      1417: }
                   1418:
                   1419: static void
                   1420: process_extended_fsync(u_int32_t id)
                   1421: {
1.104     djm      1422:        int handle, fd, r, status = SSH2_FX_OP_UNSUPPORTED;
1.102     djm      1423:
1.104     djm      1424:        if ((r = get_handle(iqueue, &handle)) != 0)
1.120     djm      1425:                fatal_fr(r, "parse");
1.102     djm      1426:        debug3("request %u: fsync (handle %u)", id, handle);
                   1427:        verbose("fsync \"%s\"", handle_to_name(handle));
                   1428:        if ((fd = handle_to_fd(handle)) < 0)
                   1429:                status = SSH2_FX_NO_SUCH_FILE;
                   1430:        else if (handle_is_ok(handle, HANDLE_FILE)) {
1.104     djm      1431:                r = fsync(fd);
                   1432:                status = (r == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
1.102     djm      1433:        }
                   1434:        send_status(id, status);
1.114     djm      1435: }
                   1436:
                   1437: static void
                   1438: process_extended_lsetstat(u_int32_t id)
                   1439: {
                   1440:        Attrib a;
                   1441:        char *name;
                   1442:        int r, status = SSH2_FX_OK;
                   1443:
                   1444:        if ((r = sshbuf_get_cstring(iqueue, &name, NULL)) != 0 ||
                   1445:            (r = decode_attrib(iqueue, &a)) != 0)
1.120     djm      1446:                fatal_fr(r, "parse");
1.114     djm      1447:
                   1448:        debug("request %u: lsetstat name \"%s\"", id, name);
                   1449:        if (a.flags & SSH2_FILEXFER_ATTR_SIZE) {
                   1450:                /* nonsensical for links */
                   1451:                status = SSH2_FX_BAD_MESSAGE;
                   1452:                goto out;
                   1453:        }
                   1454:        if (a.flags & SSH2_FILEXFER_ATTR_PERMISSIONS) {
                   1455:                logit("set \"%s\" mode %04o", name, a.perm);
                   1456:                r = fchmodat(AT_FDCWD, name,
                   1457:                    a.perm & 07777, AT_SYMLINK_NOFOLLOW);
                   1458:                if (r == -1)
                   1459:                        status = errno_to_portable(errno);
                   1460:        }
                   1461:        if (a.flags & SSH2_FILEXFER_ATTR_ACMODTIME) {
                   1462:                char buf[64];
                   1463:                time_t t = a.mtime;
                   1464:
                   1465:                strftime(buf, sizeof(buf), "%Y%m%d-%H:%M:%S",
                   1466:                    localtime(&t));
                   1467:                logit("set \"%s\" modtime %s", name, buf);
                   1468:                r = utimensat(AT_FDCWD, name,
                   1469:                    attrib_to_ts(&a), AT_SYMLINK_NOFOLLOW);
                   1470:                if (r == -1)
                   1471:                        status = errno_to_portable(errno);
                   1472:        }
                   1473:        if (a.flags & SSH2_FILEXFER_ATTR_UIDGID) {
                   1474:                logit("set \"%s\" owner %lu group %lu", name,
                   1475:                    (u_long)a.uid, (u_long)a.gid);
                   1476:                r = fchownat(AT_FDCWD, name, a.uid, a.gid,
                   1477:                    AT_SYMLINK_NOFOLLOW);
                   1478:                if (r == -1)
                   1479:                        status = errno_to_portable(errno);
                   1480:        }
                   1481:  out:
                   1482:        send_status(id, status);
                   1483:        free(name);
1.122     djm      1484: }
                   1485:
                   1486: static void
                   1487: process_extended_limits(u_int32_t id)
                   1488: {
                   1489:        struct sshbuf *msg;
                   1490:        int r;
                   1491:        uint64_t nfiles = 0;
                   1492:        struct rlimit rlim;
                   1493:
                   1494:        debug("request %u: limits", id);
                   1495:
                   1496:        if (getrlimit(RLIMIT_NOFILE, &rlim) != -1 && rlim.rlim_cur > 5)
                   1497:                nfiles = rlim.rlim_cur - 5; /* stdio(3) + syslog + spare */
                   1498:
                   1499:        if ((msg = sshbuf_new()) == NULL)
                   1500:                fatal_f("sshbuf_new failed");
                   1501:        if ((r = sshbuf_put_u8(msg, SSH2_FXP_EXTENDED_REPLY)) != 0 ||
                   1502:            (r = sshbuf_put_u32(msg, id)) != 0 ||
                   1503:            /* max-packet-length */
                   1504:            (r = sshbuf_put_u64(msg, SFTP_MAX_MSG_LENGTH)) != 0 ||
                   1505:            /* max-read-length */
                   1506:            (r = sshbuf_put_u64(msg, SFTP_MAX_READ_LENGTH)) != 0 ||
                   1507:            /* max-write-length */
                   1508:            (r = sshbuf_put_u64(msg, SFTP_MAX_MSG_LENGTH - 1024)) != 0 ||
                   1509:            /* max-open-handles */
                   1510:            (r = sshbuf_put_u64(msg, nfiles)) != 0)
                   1511:                fatal_fr(r, "compose");
                   1512:        send_msg(msg);
                   1513:        sshbuf_free(msg);
1.129     djm      1514: }
                   1515:
                   1516: static void
                   1517: process_extended_expand(u_int32_t id)
                   1518: {
                   1519:        char cwd[PATH_MAX], resolvedname[PATH_MAX];
                   1520:        char *path, *npath;
                   1521:        int r;
                   1522:        Stat s;
                   1523:
                   1524:        if ((r = sshbuf_get_cstring(iqueue, &path, NULL)) != 0)
                   1525:                fatal_fr(r, "parse");
                   1526:        if (getcwd(cwd, sizeof(cwd)) == NULL) {
                   1527:                send_status(id, errno_to_portable(errno));
                   1528:                goto out;
                   1529:        }
                   1530:
                   1531:        debug3("request %u: expand, original \"%s\"", id, path);
                   1532:        if (path[0] == '\0') {
                   1533:                /* empty path */
                   1534:                free(path);
                   1535:                path = xstrdup(".");
                   1536:        } else if (*path == '~') {
                   1537:                /* ~ expand path */
                   1538:                /* Special-case for "~" and "~/" to respect homedir flag */
                   1539:                if (strcmp(path, "~") == 0) {
                   1540:                        free(path);
                   1541:                        path = xstrdup(cwd);
                   1542:                } else if (strncmp(path, "~/", 2) == 0) {
                   1543:                        npath = xstrdup(path + 2);
                   1544:                        free(path);
                   1545:                        xasprintf(&path, "%s/%s", cwd, npath);
1.130     dtucker  1546:                        free(npath);
1.129     djm      1547:                } else {
                   1548:                        /* ~user expansions */
                   1549:                        if (tilde_expand(path, pw->pw_uid, &npath) != 0) {
1.138     djm      1550:                                send_status_errmsg(id,
                   1551:                                    errno_to_portable(ENOENT), "no such user");
1.129     djm      1552:                                goto out;
                   1553:                        }
                   1554:                        free(path);
                   1555:                        path = npath;
                   1556:                }
                   1557:        } else if (*path != '/') {
                   1558:                /* relative path */
                   1559:                xasprintf(&npath, "%s/%s", cwd, path);
                   1560:                free(path);
                   1561:                path = npath;
                   1562:        }
                   1563:        verbose("expand \"%s\"", path);
                   1564:        if (sftp_realpath(path, resolvedname) == NULL) {
                   1565:                send_status(id, errno_to_portable(errno));
                   1566:                goto out;
                   1567:        }
                   1568:        attrib_clear(&s.attrib);
                   1569:        s.name = s.long_name = resolvedname;
                   1570:        send_names(id, 1, &s);
                   1571:  out:
                   1572:        free(path);
1.140     djm      1573: }
                   1574:
                   1575: static void
                   1576: process_extended_copy_data(u_int32_t id)
                   1577: {
                   1578:        u_char buf[64*1024];
                   1579:        int read_handle, read_fd, write_handle, write_fd;
                   1580:        u_int64_t len, read_off, read_len, write_off;
                   1581:        int r, copy_until_eof, status = SSH2_FX_OP_UNSUPPORTED;
                   1582:        size_t ret;
                   1583:
                   1584:        if ((r = get_handle(iqueue, &read_handle)) != 0 ||
                   1585:            (r = sshbuf_get_u64(iqueue, &read_off)) != 0 ||
                   1586:            (r = sshbuf_get_u64(iqueue, &read_len)) != 0 ||
                   1587:            (r = get_handle(iqueue, &write_handle)) != 0 ||
                   1588:            (r = sshbuf_get_u64(iqueue, &write_off)) != 0)
                   1589:                fatal("%s: buffer error: %s", __func__, ssh_err(r));
                   1590:
                   1591:        debug("request %u: copy-data from \"%s\" (handle %d) off %llu len %llu "
                   1592:            "to \"%s\" (handle %d) off %llu",
                   1593:            id, handle_to_name(read_handle), read_handle,
                   1594:            (unsigned long long)read_off, (unsigned long long)read_len,
                   1595:            handle_to_name(write_handle), write_handle,
                   1596:            (unsigned long long)write_off);
                   1597:
                   1598:        /* For read length of 0, we read until EOF. */
                   1599:        if (read_len == 0) {
                   1600:                read_len = (u_int64_t)-1 - read_off;
                   1601:                copy_until_eof = 1;
                   1602:        } else
                   1603:                copy_until_eof = 0;
                   1604:
                   1605:        read_fd = handle_to_fd(read_handle);
                   1606:        write_fd = handle_to_fd(write_handle);
                   1607:
                   1608:        /* Disallow reading & writing to the same handle or same path or dirs */
                   1609:        if (read_handle == write_handle || read_fd < 0 || write_fd < 0 ||
                   1610:            !strcmp(handle_to_name(read_handle), handle_to_name(write_handle))) {
                   1611:                status = SSH2_FX_FAILURE;
                   1612:                goto out;
                   1613:        }
                   1614:
                   1615:        if (lseek(read_fd, read_off, SEEK_SET) < 0) {
                   1616:                status = errno_to_portable(errno);
                   1617:                error("%s: read_seek failed", __func__);
                   1618:                goto out;
                   1619:        }
                   1620:
                   1621:        if ((handle_to_flags(write_handle) & O_APPEND) == 0 &&
                   1622:            lseek(write_fd, write_off, SEEK_SET) < 0) {
                   1623:                status = errno_to_portable(errno);
                   1624:                error("%s: write_seek failed", __func__);
                   1625:                goto out;
                   1626:        }
                   1627:
                   1628:        /* Process the request in chunks. */
                   1629:        while (read_len > 0 || copy_until_eof) {
                   1630:                len = MINIMUM(sizeof(buf), read_len);
                   1631:                read_len -= len;
                   1632:
                   1633:                ret = atomicio(read, read_fd, buf, len);
                   1634:                if (ret == 0 && errno == EPIPE) {
                   1635:                        status = copy_until_eof ? SSH2_FX_OK : SSH2_FX_EOF;
                   1636:                        break;
                   1637:                } else if (ret == 0) {
                   1638:                        status = errno_to_portable(errno);
                   1639:                        error("%s: read failed: %s", __func__, strerror(errno));
                   1640:                        break;
                   1641:                }
                   1642:                len = ret;
                   1643:                handle_update_read(read_handle, len);
                   1644:
                   1645:                ret = atomicio(vwrite, write_fd, buf, len);
                   1646:                if (ret != len) {
                   1647:                        status = errno_to_portable(errno);
                   1648:                        error("%s: write failed: %llu != %llu: %s", __func__,
                   1649:                            (unsigned long long)ret, (unsigned long long)len,
                   1650:                            strerror(errno));
                   1651:                        break;
                   1652:                }
                   1653:                handle_update_write(write_handle, len);
                   1654:        }
                   1655:
                   1656:        if (read_len == 0)
                   1657:                status = SSH2_FX_OK;
                   1658:
                   1659:  out:
                   1660:        send_status(id, status);
1.141     djm      1661: }
                   1662:
                   1663: static void
                   1664: process_extended_home_directory(u_int32_t id)
                   1665: {
                   1666:        char *username;
                   1667:        struct passwd *user_pw;
                   1668:        int r;
                   1669:        Stat s;
                   1670:
                   1671:        if ((r = sshbuf_get_cstring(iqueue, &username, NULL)) != 0)
                   1672:                fatal_fr(r, "parse");
                   1673:
                   1674:        debug3("request %u: home-directory \"%s\"", id, username);
                   1675:        if ((user_pw = getpwnam(username)) == NULL) {
1.142     djm      1676:                send_status(id, SSH2_FX_FAILURE);
1.141     djm      1677:                goto out;
                   1678:        }
                   1679:
                   1680:        verbose("home-directory \"%s\"", pw->pw_dir);
                   1681:        attrib_clear(&s.attrib);
                   1682:        s.name = s.long_name = pw->pw_dir;
                   1683:        send_names(id, 1, &s);
                   1684:  out:
                   1685:        free(username);
1.143   ! djm      1686: }
        !          1687:
        !          1688: static void
        !          1689: process_extended_get_users_groups_by_id(u_int32_t id)
        !          1690: {
        !          1691:        struct passwd *user_pw;
        !          1692:        struct group *gr;
        !          1693:        struct sshbuf *uids, *gids, *usernames, *groupnames, *msg;
        !          1694:        int r;
        !          1695:        u_int n, nusers = 0, ngroups = 0;
        !          1696:        const char *name;
        !          1697:
        !          1698:        if ((usernames = sshbuf_new()) == NULL ||
        !          1699:            (groupnames = sshbuf_new()) == NULL ||
        !          1700:            (msg = sshbuf_new()) == NULL)
        !          1701:                fatal_f("sshbuf_new failed");
        !          1702:        if ((r = sshbuf_froms(iqueue, &uids)) != 0 ||
        !          1703:            (r = sshbuf_froms(iqueue, &gids)) != 0)
        !          1704:                fatal_fr(r, "parse");
        !          1705:        debug_f("uids len = %zu, gids len = %zu",
        !          1706:            sshbuf_len(uids), sshbuf_len(gids));
        !          1707:        while (sshbuf_len(uids) != 0) {
        !          1708:                if ((r = sshbuf_get_u32(uids, &n)) != 0)
        !          1709:                        fatal_fr(r, "parse inner uid");
        !          1710:                user_pw = getpwuid((uid_t)n);
        !          1711:                name = user_pw == NULL ? "" : user_pw->pw_name;
        !          1712:                debug3_f("uid %u => \"%s\"", n, name);
        !          1713:                if ((r = sshbuf_put_cstring(usernames, name)) != 0)
        !          1714:                        fatal_fr(r, "assemble gid reply");
        !          1715:                nusers++;
        !          1716:        }
        !          1717:        while (sshbuf_len(gids) != 0) {
        !          1718:                if ((r = sshbuf_get_u32(gids, &n)) != 0)
        !          1719:                        fatal_fr(r, "parse inner gid");
        !          1720:                gr = getgrgid((gid_t)n);
        !          1721:                name = gr == NULL ? "" : gr->gr_name;
        !          1722:                debug3_f("gid %u => \"%s\"", n, name);
        !          1723:                if ((r = sshbuf_put_cstring(groupnames, name)) != 0)
        !          1724:                        fatal_fr(r, "assemble gid reply");
        !          1725:                nusers++;
        !          1726:        }
        !          1727:        verbose("users-groups-by-id: %u users, %u groups", nusers, ngroups);
        !          1728:
        !          1729:        if ((r = sshbuf_put_u8(msg, SSH2_FXP_EXTENDED_REPLY)) != 0 ||
        !          1730:            (r = sshbuf_put_u32(msg, id)) != 0 ||
        !          1731:            (r = sshbuf_put_stringb(msg, usernames)) != 0 ||
        !          1732:            (r = sshbuf_put_stringb(msg, groupnames)) != 0)
        !          1733:                fatal_fr(r, "compose");
        !          1734:        send_msg(msg);
        !          1735:
        !          1736:        sshbuf_free(uids);
        !          1737:        sshbuf_free(gids);
        !          1738:        sshbuf_free(usernames);
        !          1739:        sshbuf_free(groupnames);
        !          1740:        sshbuf_free(msg);
1.93      djm      1741: }
                   1742:
                   1743: static void
1.98      djm      1744: process_extended(u_int32_t id)
1.10      markus   1745: {
                   1746:        char *request;
1.126     djm      1747:        int r;
1.125     djm      1748:        const struct sftp_handler *exthand;
1.10      markus   1749:
1.104     djm      1750:        if ((r = sshbuf_get_cstring(iqueue, &request, NULL)) != 0)
1.120     djm      1751:                fatal_fr(r, "parse");
1.125     djm      1752:        if ((exthand = extended_handler_byname(request)) == NULL) {
1.98      djm      1753:                error("Unknown extended request \"%.100s\"", request);
1.78      djm      1754:                send_status(id, SSH2_FX_OP_UNSUPPORTED);        /* MUST */
1.125     djm      1755:        } else {
                   1756:                if (!request_permitted(exthand))
                   1757:                        send_status(id, SSH2_FX_PERMISSION_DENIED);
                   1758:                else
                   1759:                        exthand->handler(id);
1.98      djm      1760:        }
1.97      djm      1761:        free(request);
1.10      markus   1762: }
1.1       markus   1763:
                   1764: /* stolen from ssh-agent */
                   1765:
1.28      itojun   1766: static void
1.1       markus   1767: process(void)
                   1768: {
1.104     djm      1769:        u_int msg_len;
                   1770:        u_int buf_len;
                   1771:        u_int consumed;
                   1772:        u_char type;
                   1773:        const u_char *cp;
                   1774:        int i, r;
1.98      djm      1775:        u_int32_t id;
1.1       markus   1776:
1.104     djm      1777:        buf_len = sshbuf_len(iqueue);
1.34      markus   1778:        if (buf_len < 5)
1.1       markus   1779:                return;         /* Incomplete message. */
1.104     djm      1780:        cp = sshbuf_ptr(iqueue);
1.57      djm      1781:        msg_len = get_u32(cp);
1.50      djm      1782:        if (msg_len > SFTP_MAX_MSG_LENGTH) {
1.58      djm      1783:                error("bad message from %s local user %s",
                   1784:                    client_addr, pw->pw_name);
1.76      markus   1785:                sftp_server_cleanup_exit(11);
1.1       markus   1786:        }
1.34      markus   1787:        if (buf_len < msg_len + 4)
1.1       markus   1788:                return;
1.104     djm      1789:        if ((r = sshbuf_consume(iqueue, 4)) != 0)
1.120     djm      1790:                fatal_fr(r, "consume");
1.34      markus   1791:        buf_len -= 4;
1.104     djm      1792:        if ((r = sshbuf_get_u8(iqueue, &type)) != 0)
1.120     djm      1793:                fatal_fr(r, "parse type");
1.98      djm      1794:
1.1       markus   1795:        switch (type) {
1.10      markus   1796:        case SSH2_FXP_INIT:
1.1       markus   1797:                process_init();
1.98      djm      1798:                init_done = 1;
1.1       markus   1799:                break;
1.10      markus   1800:        case SSH2_FXP_EXTENDED:
1.98      djm      1801:                if (!init_done)
                   1802:                        fatal("Received extended request before init");
1.104     djm      1803:                if ((r = sshbuf_get_u32(iqueue, &id)) != 0)
1.120     djm      1804:                        fatal_fr(r, "parse extended ID");
1.98      djm      1805:                process_extended(id);
1.10      markus   1806:                break;
1.1       markus   1807:        default:
1.98      djm      1808:                if (!init_done)
                   1809:                        fatal("Received %u request before init", type);
1.104     djm      1810:                if ((r = sshbuf_get_u32(iqueue, &id)) != 0)
1.120     djm      1811:                        fatal_fr(r, "parse ID");
1.98      djm      1812:                for (i = 0; handlers[i].handler != NULL; i++) {
                   1813:                        if (type == handlers[i].type) {
                   1814:                                if (!request_permitted(&handlers[i])) {
                   1815:                                        send_status(id,
                   1816:                                            SSH2_FX_PERMISSION_DENIED);
                   1817:                                } else {
                   1818:                                        handlers[i].handler(id);
                   1819:                                }
                   1820:                                break;
                   1821:                        }
                   1822:                }
                   1823:                if (handlers[i].handler == NULL)
                   1824:                        error("Unknown message %u", type);
1.1       markus   1825:        }
1.34      markus   1826:        /* discard the remaining bytes from the current packet */
1.104     djm      1827:        if (buf_len < sshbuf_len(iqueue)) {
1.76      markus   1828:                error("iqueue grew unexpectedly");
                   1829:                sftp_server_cleanup_exit(255);
                   1830:        }
1.104     djm      1831:        consumed = buf_len - sshbuf_len(iqueue);
1.76      markus   1832:        if (msg_len < consumed) {
1.98      djm      1833:                error("msg_len %u < consumed %u", msg_len, consumed);
1.76      markus   1834:                sftp_server_cleanup_exit(255);
                   1835:        }
1.104     djm      1836:        if (msg_len > consumed &&
                   1837:            (r = sshbuf_consume(iqueue, msg_len - consumed)) != 0)
1.120     djm      1838:                fatal_fr(r, "consume");
1.1       markus   1839: }
                   1840:
1.58      djm      1841: /* Cleanup handler that logs active handles upon normal exit */
                   1842: void
1.76      markus   1843: sftp_server_cleanup_exit(int i)
1.58      djm      1844: {
                   1845:        if (pw != NULL && client_addr != NULL) {
                   1846:                handle_log_exit();
                   1847:                logit("session closed for local user %s from [%s]",
                   1848:                    pw->pw_name, client_addr);
                   1849:        }
                   1850:        _exit(i);
                   1851: }
                   1852:
                   1853: static void
1.76      markus   1854: sftp_server_usage(void)
1.58      djm      1855: {
                   1856:        extern char *__progname;
                   1857:
                   1858:        fprintf(stderr,
1.96      jmc      1859:            "usage: %s [-ehR] [-d start_directory] [-f log_facility] "
1.118     djm      1860:            "[-l log_level]\n\t[-P denied_requests] "
                   1861:            "[-p allowed_requests] [-u umask]\n"
1.100     jmc      1862:            "       %s -Q protocol_feature\n",
                   1863:            __progname, __progname);
1.58      djm      1864:        exit(1);
                   1865: }
                   1866:
1.1       markus   1867: int
1.77      djm      1868: sftp_server_main(int argc, char **argv, struct passwd *user_pw)
1.1       markus   1869: {
1.132     deraadt  1870:        int i, r, in, out, ch, skipargs = 0, log_stderr = 0;
                   1871:        ssize_t len, olen;
1.58      djm      1872:        SyslogFacility log_facility = SYSLOG_FACILITY_AUTH;
1.112     djm      1873:        char *cp, *homedir = NULL, uidstr[32], buf[4*4096];
1.92      djm      1874:        long mask;
1.58      djm      1875:
                   1876:        extern char *optarg;
                   1877:        extern char *__progname;
1.49      djm      1878:
1.58      djm      1879:        log_init(__progname, log_level, log_facility, log_stderr);
                   1880:
1.95      djm      1881:        pw = pwcopy(user_pw);
                   1882:
1.98      djm      1883:        while (!skipargs && (ch = getopt(argc, argv,
                   1884:            "d:f:l:P:p:Q:u:cehR")) != -1) {
1.58      djm      1885:                switch (ch) {
1.98      djm      1886:                case 'Q':
                   1887:                        if (strcasecmp(optarg, "requests") != 0) {
                   1888:                                fprintf(stderr, "Invalid query type\n");
                   1889:                                exit(1);
                   1890:                        }
                   1891:                        for (i = 0; handlers[i].handler != NULL; i++)
                   1892:                                printf("%s\n", handlers[i].name);
                   1893:                        for (i = 0; extended_handlers[i].handler != NULL; i++)
                   1894:                                printf("%s\n", extended_handlers[i].name);
                   1895:                        exit(0);
                   1896:                        break;
1.90      djm      1897:                case 'R':
                   1898:                        readonly = 1;
                   1899:                        break;
1.58      djm      1900:                case 'c':
                   1901:                        /*
                   1902:                         * Ignore all arguments if we are invoked as a
1.70      deraadt  1903:                         * shell using "sftp-server -c command"
1.58      djm      1904:                         */
                   1905:                        skipargs = 1;
                   1906:                        break;
                   1907:                case 'e':
                   1908:                        log_stderr = 1;
                   1909:                        break;
                   1910:                case 'l':
                   1911:                        log_level = log_level_number(optarg);
                   1912:                        if (log_level == SYSLOG_LEVEL_NOT_SET)
                   1913:                                error("Invalid log level \"%s\"", optarg);
                   1914:                        break;
                   1915:                case 'f':
                   1916:                        log_facility = log_facility_number(optarg);
1.74      djm      1917:                        if (log_facility == SYSLOG_FACILITY_NOT_SET)
1.58      djm      1918:                                error("Invalid log facility \"%s\"", optarg);
1.86      djm      1919:                        break;
1.95      djm      1920:                case 'd':
                   1921:                        cp = tilde_expand_filename(optarg, user_pw->pw_uid);
1.112     djm      1922:                        snprintf(uidstr, sizeof(uidstr), "%llu",
                   1923:                            (unsigned long long)pw->pw_uid);
1.95      djm      1924:                        homedir = percent_expand(cp, "d", user_pw->pw_dir,
1.112     djm      1925:                            "u", user_pw->pw_name, "U", uidstr, (char *)NULL);
1.95      djm      1926:                        free(cp);
1.98      djm      1927:                        break;
                   1928:                case 'p':
1.118     djm      1929:                        if (request_allowlist != NULL)
1.98      djm      1930:                                fatal("Permitted requests already set");
1.118     djm      1931:                        request_allowlist = xstrdup(optarg);
1.98      djm      1932:                        break;
                   1933:                case 'P':
1.118     djm      1934:                        if (request_denylist != NULL)
1.98      djm      1935:                                fatal("Refused requests already set");
1.118     djm      1936:                        request_denylist = xstrdup(optarg);
1.95      djm      1937:                        break;
1.86      djm      1938:                case 'u':
1.92      djm      1939:                        errno = 0;
                   1940:                        mask = strtol(optarg, &cp, 8);
                   1941:                        if (mask < 0 || mask > 0777 || *cp != '\0' ||
                   1942:                            cp == optarg || (mask == 0 && errno != 0))
                   1943:                                fatal("Invalid umask \"%s\"", optarg);
                   1944:                        (void)umask((mode_t)mask);
1.58      djm      1945:                        break;
                   1946:                case 'h':
                   1947:                default:
1.76      markus   1948:                        sftp_server_usage();
1.58      djm      1949:                }
                   1950:        }
                   1951:
                   1952:        log_init(__progname, log_level, log_facility, log_stderr);
                   1953:
                   1954:        if ((cp = getenv("SSH_CONNECTION")) != NULL) {
                   1955:                client_addr = xstrdup(cp);
1.76      markus   1956:                if ((cp = strchr(client_addr, ' ')) == NULL) {
                   1957:                        error("Malformed SSH_CONNECTION variable: \"%s\"",
1.58      djm      1958:                            getenv("SSH_CONNECTION"));
1.76      markus   1959:                        sftp_server_cleanup_exit(255);
                   1960:                }
1.58      djm      1961:                *cp = '\0';
                   1962:        } else
                   1963:                client_addr = xstrdup("UNKNOWN");
                   1964:
                   1965:        logit("session opened for local user %s from [%s]",
                   1966:            pw->pw_name, client_addr);
1.10      markus   1967:
1.89      djm      1968:        in = STDIN_FILENO;
                   1969:        out = STDOUT_FILENO;
1.1       markus   1970:
1.104     djm      1971:        if ((iqueue = sshbuf_new()) == NULL)
1.120     djm      1972:                fatal_f("sshbuf_new failed");
1.104     djm      1973:        if ((oqueue = sshbuf_new()) == NULL)
1.120     djm      1974:                fatal_f("sshbuf_new failed");
1.1       markus   1975:
1.95      djm      1976:        if (homedir != NULL) {
                   1977:                if (chdir(homedir) != 0) {
                   1978:                        error("chdir to \"%s\" failed: %s", homedir,
                   1979:                            strerror(errno));
                   1980:                }
                   1981:        }
1.21      millert  1982:
1.1       markus   1983:        for (;;) {
1.132     deraadt  1984:                struct pollfd pfd[2];
                   1985:
                   1986:                memset(pfd, 0, sizeof pfd);
                   1987:                pfd[0].fd = pfd[1].fd = -1;
1.1       markus   1988:
1.73      djm      1989:                /*
                   1990:                 * Ensure that we can read a full buffer and handle
                   1991:                 * the worst-case length packet it can generate,
                   1992:                 * otherwise apply backpressure by stopping reads.
                   1993:                 */
1.104     djm      1994:                if ((r = sshbuf_check_reserve(iqueue, sizeof(buf))) == 0 &&
                   1995:                    (r = sshbuf_check_reserve(oqueue,
1.133     deraadt  1996:                    SFTP_MAX_MSG_LENGTH)) == 0) {
1.132     deraadt  1997:                        pfd[0].fd = in;
1.133     deraadt  1998:                        pfd[0].events = POLLIN;
                   1999:                }
1.104     djm      2000:                else if (r != SSH_ERR_NO_BUFFER_SPACE)
1.120     djm      2001:                        fatal_fr(r, "reserve");
1.73      djm      2002:
1.104     djm      2003:                olen = sshbuf_len(oqueue);
1.133     deraadt  2004:                if (olen > 0) {
1.132     deraadt  2005:                        pfd[1].fd = out;
1.133     deraadt  2006:                        pfd[1].events = POLLOUT;
                   2007:                }
1.1       markus   2008:
1.132     deraadt  2009:                if (poll(pfd, 2, -1) == -1) {
1.1       markus   2010:                        if (errno == EINTR)
                   2011:                                continue;
1.132     deraadt  2012:                        error("poll: %s", strerror(errno));
1.76      markus   2013:                        sftp_server_cleanup_exit(2);
1.1       markus   2014:                }
                   2015:
                   2016:                /* copy stdin to iqueue */
1.134     djm      2017:                if (pfd[0].revents & (POLLIN|POLLHUP)) {
1.1       markus   2018:                        len = read(in, buf, sizeof buf);
                   2019:                        if (len == 0) {
                   2020:                                debug("read eof");
1.76      markus   2021:                                sftp_server_cleanup_exit(0);
1.116     deraadt  2022:                        } else if (len == -1) {
1.134     djm      2023:                                if (errno != EAGAIN && errno != EINTR) {
                   2024:                                        error("read: %s", strerror(errno));
                   2025:                                        sftp_server_cleanup_exit(1);
                   2026:                                }
1.120     djm      2027:                        } else if ((r = sshbuf_put(iqueue, buf, len)) != 0)
                   2028:                                fatal_fr(r, "sshbuf_put");
1.1       markus   2029:                }
                   2030:                /* send oqueue to stdout */
1.134     djm      2031:                if (pfd[1].revents & (POLLOUT|POLLHUP)) {
1.104     djm      2032:                        len = write(out, sshbuf_ptr(oqueue), olen);
1.134     djm      2033:                        if (len == 0 || (len == -1 && errno == EPIPE)) {
                   2034:                                debug("write eof");
                   2035:                                sftp_server_cleanup_exit(0);
                   2036:                        } else if (len == -1) {
1.76      markus   2037:                                sftp_server_cleanup_exit(1);
1.134     djm      2038:                                if (errno != EAGAIN && errno != EINTR) {
                   2039:                                        error("write: %s", strerror(errno));
                   2040:                                        sftp_server_cleanup_exit(1);
                   2041:                                }
1.120     djm      2042:                        } else if ((r = sshbuf_consume(oqueue, len)) != 0)
                   2043:                                fatal_fr(r, "consume");
1.1       markus   2044:                }
1.73      djm      2045:
                   2046:                /*
                   2047:                 * Process requests from client if we can fit the results
                   2048:                 * into the output buffer, otherwise stop processing input
                   2049:                 * and let the output queue drain.
                   2050:                 */
1.104     djm      2051:                r = sshbuf_check_reserve(oqueue, SFTP_MAX_MSG_LENGTH);
                   2052:                if (r == 0)
1.73      djm      2053:                        process();
1.104     djm      2054:                else if (r != SSH_ERR_NO_BUFFER_SPACE)
1.120     djm      2055:                        fatal_fr(r, "reserve");
1.1       markus   2056:        }
                   2057: }