[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.121

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