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

1.1       markus      1: /*
                      2:  * Copyright (c) 2000 Markus Friedl.  All rights reserved.
                      3:  *
                      4:  * Redistribution and use in source and binary forms, with or without
                      5:  * modification, are permitted provided that the following conditions
                      6:  * are met:
                      7:  * 1. Redistributions of source code must retain the above copyright
                      8:  *    notice, this list of conditions and the following disclaimer.
                      9:  * 2. Redistributions in binary form must reproduce the above copyright
                     10:  *    notice, this list of conditions and the following disclaimer in the
                     11:  *    documentation and/or other materials provided with the distribution.
                     12:  *
                     13:  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
                     14:  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
                     15:  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
                     16:  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
                     17:  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
                     18:  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
                     19:  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
                     20:  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
                     21:  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
                     22:  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
                     23:  */
                     24: #include "includes.h"
1.15    ! djm        25: RCSID("$OpenBSD: sftp-server.c,v 1.14 2001/01/21 19:05:56 markus Exp $");
1.1       markus     26:
                     27: #include "buffer.h"
                     28: #include "bufaux.h"
                     29: #include "getput.h"
1.14      markus     30: #include "log.h"
1.1       markus     31: #include "xmalloc.h"
                     32:
1.10      markus     33: #include "sftp.h"
1.15    ! djm        34: #include "sftp-common.h"
1.1       markus     35:
                     36: /* helper */
1.10      markus     37: #define get_int64()                    buffer_get_int64(&iqueue);
1.1       markus     38: #define get_int()                      buffer_get_int(&iqueue);
                     39: #define get_string(lenp)               buffer_get_string(&iqueue, lenp);
1.10      markus     40: #define TRACE                          debug
1.1       markus     41:
                     42: /* input and output queue */
                     43: Buffer iqueue;
                     44: Buffer oqueue;
                     45:
                     46: /* portable attibutes, etc. */
                     47:
                     48: typedef struct Stat Stat;
                     49:
1.15    ! djm        50: struct Stat {
1.1       markus     51:        char *name;
                     52:        char *long_name;
                     53:        Attrib attrib;
                     54: };
                     55:
                     56: int
1.2       markus     57: errno_to_portable(int unixerrno)
1.1       markus     58: {
                     59:        int ret = 0;
1.2       markus     60:        switch (unixerrno) {
1.1       markus     61:        case 0:
1.10      markus     62:                ret = SSH2_FX_OK;
1.1       markus     63:                break;
                     64:        case ENOENT:
                     65:        case ENOTDIR:
                     66:        case EBADF:
                     67:        case ELOOP:
1.10      markus     68:                ret = SSH2_FX_NO_SUCH_FILE;
1.1       markus     69:                break;
                     70:        case EPERM:
                     71:        case EACCES:
                     72:        case EFAULT:
1.10      markus     73:                ret = SSH2_FX_PERMISSION_DENIED;
1.1       markus     74:                break;
                     75:        case ENAMETOOLONG:
                     76:        case EINVAL:
1.10      markus     77:                ret = SSH2_FX_BAD_MESSAGE;
1.1       markus     78:                break;
                     79:        default:
1.10      markus     80:                ret = SSH2_FX_FAILURE;
1.1       markus     81:                break;
                     82:        }
                     83:        return ret;
                     84: }
                     85:
                     86: int
                     87: flags_from_portable(int pflags)
                     88: {
                     89:        int flags = 0;
1.10      markus     90:        if (pflags & SSH2_FXF_READ &&
                     91:            pflags & SSH2_FXF_WRITE) {
1.1       markus     92:                flags = O_RDWR;
1.10      markus     93:        } else if (pflags & SSH2_FXF_READ) {
1.1       markus     94:                flags = O_RDONLY;
1.10      markus     95:        } else if (pflags & SSH2_FXF_WRITE) {
1.1       markus     96:                flags = O_WRONLY;
                     97:        }
1.10      markus     98:        if (pflags & SSH2_FXF_CREAT)
1.1       markus     99:                flags |= O_CREAT;
1.10      markus    100:        if (pflags & SSH2_FXF_TRUNC)
1.1       markus    101:                flags |= O_TRUNC;
1.10      markus    102:        if (pflags & SSH2_FXF_EXCL)
1.1       markus    103:                flags |= O_EXCL;
                    104:        return flags;
                    105: }
                    106:
                    107: Attrib *
                    108: get_attrib(void)
                    109: {
                    110:        return decode_attrib(&iqueue);
                    111: }
                    112:
                    113: /* handle handles */
                    114:
                    115: typedef struct Handle Handle;
                    116: struct Handle {
                    117:        int use;
                    118:        DIR *dirp;
                    119:        int fd;
                    120:        char *name;
                    121: };
                    122: enum {
                    123:        HANDLE_UNUSED,
                    124:        HANDLE_DIR,
                    125:        HANDLE_FILE
                    126: };
                    127: Handle handles[100];
                    128:
                    129: void
                    130: handle_init(void)
                    131: {
                    132:        int i;
                    133:        for(i = 0; i < sizeof(handles)/sizeof(Handle); i++)
                    134:                handles[i].use = HANDLE_UNUSED;
                    135: }
                    136:
                    137: int
                    138: handle_new(int use, char *name, int fd, DIR *dirp)
                    139: {
                    140:        int i;
                    141:        for(i = 0; i < sizeof(handles)/sizeof(Handle); i++) {
                    142:                if (handles[i].use == HANDLE_UNUSED) {
                    143:                        handles[i].use = use;
                    144:                        handles[i].dirp = dirp;
                    145:                        handles[i].fd = fd;
                    146:                        handles[i].name = name;
                    147:                        return i;
                    148:                }
                    149:        }
                    150:        return -1;
                    151: }
                    152:
                    153: int
                    154: handle_is_ok(int i, int type)
                    155: {
1.10      markus    156:        return i >= 0 && i < sizeof(handles)/sizeof(Handle) &&
                    157:            handles[i].use == type;
1.1       markus    158: }
                    159:
                    160: int
                    161: handle_to_string(int handle, char **stringp, int *hlenp)
                    162: {
                    163:        if (stringp == NULL || hlenp == NULL)
                    164:                return -1;
1.13      markus    165:        *stringp = xmalloc(sizeof(int32_t));
                    166:        PUT_32BIT(*stringp, handle);
                    167:        *hlenp = sizeof(int32_t);
1.1       markus    168:        return 0;
                    169: }
                    170:
                    171: int
1.5       markus    172: handle_from_string(char *handle, u_int hlen)
1.1       markus    173: {
1.13      markus    174:        int val;
                    175:        if (hlen != sizeof(int32_t))
1.1       markus    176:                return -1;
1.13      markus    177:        val = GET_32BIT(handle);
1.1       markus    178:        if (handle_is_ok(val, HANDLE_FILE) ||
                    179:            handle_is_ok(val, HANDLE_DIR))
                    180:                return val;
                    181:        return -1;
                    182: }
                    183:
                    184: char *
                    185: handle_to_name(int handle)
                    186: {
                    187:        if (handle_is_ok(handle, HANDLE_DIR)||
                    188:            handle_is_ok(handle, HANDLE_FILE))
                    189:                return handles[handle].name;
                    190:        return NULL;
                    191: }
                    192:
                    193: DIR *
                    194: handle_to_dir(int handle)
                    195: {
                    196:        if (handle_is_ok(handle, HANDLE_DIR))
                    197:                return handles[handle].dirp;
                    198:        return NULL;
                    199: }
                    200:
                    201: int
                    202: handle_to_fd(int handle)
                    203: {
                    204:        if (handle_is_ok(handle, HANDLE_FILE))
                    205:                return handles[handle].fd;
                    206:        return -1;
                    207: }
                    208:
                    209: int
                    210: handle_close(int handle)
                    211: {
                    212:        int ret = -1;
                    213:        if (handle_is_ok(handle, HANDLE_FILE)) {
                    214:                ret = close(handles[handle].fd);
                    215:                handles[handle].use = HANDLE_UNUSED;
                    216:        } else if (handle_is_ok(handle, HANDLE_DIR)) {
                    217:                ret = closedir(handles[handle].dirp);
                    218:                handles[handle].use = HANDLE_UNUSED;
                    219:        } else {
                    220:                errno = ENOENT;
                    221:        }
                    222:        return ret;
                    223: }
                    224:
                    225: int
                    226: get_handle(void)
                    227: {
                    228:        char *handle;
1.10      markus    229:        int val = -1;
1.5       markus    230:        u_int hlen;
1.1       markus    231:        handle = get_string(&hlen);
1.10      markus    232:        if (hlen < 256)
                    233:                val = handle_from_string(handle, hlen);
1.1       markus    234:        xfree(handle);
                    235:        return val;
                    236: }
                    237:
                    238: /* send replies */
                    239:
                    240: void
                    241: send_msg(Buffer *m)
                    242: {
                    243:        int mlen = buffer_len(m);
                    244:        buffer_put_int(&oqueue, mlen);
                    245:        buffer_append(&oqueue, buffer_ptr(m), mlen);
                    246:        buffer_consume(m, mlen);
                    247: }
                    248:
                    249: void
                    250: send_status(u_int32_t id, u_int32_t error)
                    251: {
                    252:        Buffer msg;
                    253:        TRACE("sent status id %d error %d", id, error);
                    254:        buffer_init(&msg);
1.10      markus    255:        buffer_put_char(&msg, SSH2_FXP_STATUS);
1.1       markus    256:        buffer_put_int(&msg, id);
                    257:        buffer_put_int(&msg, error);
                    258:        send_msg(&msg);
                    259:        buffer_free(&msg);
                    260: }
                    261: void
                    262: send_data_or_handle(char type, u_int32_t id, char *data, int dlen)
                    263: {
                    264:        Buffer msg;
                    265:        buffer_init(&msg);
                    266:        buffer_put_char(&msg, type);
                    267:        buffer_put_int(&msg, id);
                    268:        buffer_put_string(&msg, data, dlen);
                    269:        send_msg(&msg);
                    270:        buffer_free(&msg);
                    271: }
                    272:
                    273: void
                    274: send_data(u_int32_t id, char *data, int dlen)
                    275: {
                    276:        TRACE("sent data id %d len %d", id, dlen);
1.10      markus    277:        send_data_or_handle(SSH2_FXP_DATA, id, data, dlen);
1.1       markus    278: }
                    279:
                    280: void
                    281: send_handle(u_int32_t id, int handle)
                    282: {
                    283:        char *string;
                    284:        int hlen;
                    285:        handle_to_string(handle, &string, &hlen);
                    286:        TRACE("sent handle id %d handle %d", id, handle);
1.10      markus    287:        send_data_or_handle(SSH2_FXP_HANDLE, id, string, hlen);
1.1       markus    288:        xfree(string);
                    289: }
                    290:
                    291: void
                    292: send_names(u_int32_t id, int count, Stat *stats)
                    293: {
                    294:        Buffer msg;
                    295:        int i;
                    296:        buffer_init(&msg);
1.10      markus    297:        buffer_put_char(&msg, SSH2_FXP_NAME);
1.1       markus    298:        buffer_put_int(&msg, id);
                    299:        buffer_put_int(&msg, count);
                    300:        TRACE("sent names id %d count %d", id, count);
                    301:        for (i = 0; i < count; i++) {
                    302:                buffer_put_cstring(&msg, stats[i].name);
                    303:                buffer_put_cstring(&msg, stats[i].long_name);
                    304:                encode_attrib(&msg, &stats[i].attrib);
                    305:        }
                    306:        send_msg(&msg);
                    307:        buffer_free(&msg);
                    308: }
                    309:
                    310: void
                    311: send_attrib(u_int32_t id, Attrib *a)
                    312: {
                    313:        Buffer msg;
                    314:        TRACE("sent attrib id %d have 0x%x", id, a->flags);
                    315:        buffer_init(&msg);
1.10      markus    316:        buffer_put_char(&msg, SSH2_FXP_ATTRS);
1.1       markus    317:        buffer_put_int(&msg, id);
                    318:        encode_attrib(&msg, a);
                    319:        send_msg(&msg);
                    320:        buffer_free(&msg);
                    321: }
                    322:
                    323: /* parse incoming */
                    324:
                    325: void
                    326: process_init(void)
                    327: {
                    328:        Buffer msg;
                    329:        int version = buffer_get_int(&iqueue);
                    330:
                    331:        TRACE("client version %d", version);
                    332:        buffer_init(&msg);
1.10      markus    333:        buffer_put_char(&msg, SSH2_FXP_VERSION);
                    334:        buffer_put_int(&msg, SSH2_FILEXFER_VERSION);
1.1       markus    335:        send_msg(&msg);
                    336:        buffer_free(&msg);
                    337: }
                    338:
                    339: void
                    340: process_open(void)
                    341: {
                    342:        u_int32_t id, pflags;
                    343:        Attrib *a;
                    344:        char *name;
1.10      markus    345:        int handle, fd, flags, mode, status = SSH2_FX_FAILURE;
1.1       markus    346:
                    347:        id = get_int();
                    348:        name = get_string(NULL);
1.10      markus    349:        pflags = get_int();             /* portable flags */
1.1       markus    350:        a = get_attrib();
                    351:        flags = flags_from_portable(pflags);
1.10      markus    352:        mode = (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) ? a->perm : 0666;
1.1       markus    353:        TRACE("open id %d name %s flags %d mode 0%o", id, name, pflags, mode);
                    354:        fd = open(name, flags, mode);
                    355:        if (fd < 0) {
                    356:                status = errno_to_portable(errno);
                    357:        } else {
                    358:                handle = handle_new(HANDLE_FILE, xstrdup(name), fd, NULL);
                    359:                if (handle < 0) {
                    360:                        close(fd);
                    361:                } else {
                    362:                        send_handle(id, handle);
1.10      markus    363:                        status = SSH2_FX_OK;
1.1       markus    364:                }
                    365:        }
1.10      markus    366:        if (status != SSH2_FX_OK)
1.1       markus    367:                send_status(id, status);
                    368:        xfree(name);
                    369: }
                    370:
                    371: void
                    372: process_close(void)
                    373: {
                    374:        u_int32_t id;
1.10      markus    375:        int handle, ret, status = SSH2_FX_FAILURE;
1.1       markus    376:
                    377:        id = get_int();
                    378:        handle = get_handle();
                    379:        TRACE("close id %d handle %d", id, handle);
                    380:        ret = handle_close(handle);
1.10      markus    381:        status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
1.1       markus    382:        send_status(id, status);
                    383: }
                    384:
                    385: void
                    386: process_read(void)
                    387: {
                    388:        char buf[64*1024];
1.10      markus    389:        u_int32_t id, len;
                    390:        int handle, fd, ret, status = SSH2_FX_FAILURE;
1.1       markus    391:        u_int64_t off;
                    392:
                    393:        id = get_int();
                    394:        handle = get_handle();
1.10      markus    395:        off = get_int64();
1.1       markus    396:        len = get_int();
                    397:
                    398:        TRACE("read id %d handle %d off %qd len %d", id, handle, off, len);
                    399:        if (len > sizeof buf) {
                    400:                len = sizeof buf;
                    401:                log("read change len %d", len);
                    402:        }
                    403:        fd = handle_to_fd(handle);
                    404:        if (fd >= 0) {
                    405:                if (lseek(fd, off, SEEK_SET) < 0) {
                    406:                        error("process_read: seek failed");
                    407:                        status = errno_to_portable(errno);
                    408:                } else {
                    409:                        ret = read(fd, buf, len);
                    410:                        if (ret < 0) {
                    411:                                status = errno_to_portable(errno);
                    412:                        } else if (ret == 0) {
1.10      markus    413:                                status = SSH2_FX_EOF;
1.1       markus    414:                        } else {
                    415:                                send_data(id, buf, ret);
1.10      markus    416:                                status = SSH2_FX_OK;
1.1       markus    417:                        }
                    418:                }
                    419:        }
1.10      markus    420:        if (status != SSH2_FX_OK)
1.1       markus    421:                send_status(id, status);
                    422: }
                    423:
                    424: void
                    425: process_write(void)
                    426: {
1.10      markus    427:        u_int32_t id;
1.1       markus    428:        u_int64_t off;
1.5       markus    429:        u_int len;
1.10      markus    430:        int handle, fd, ret, status = SSH2_FX_FAILURE;
1.1       markus    431:        char *data;
                    432:
                    433:        id = get_int();
                    434:        handle = get_handle();
1.10      markus    435:        off = get_int64();
1.1       markus    436:        data = get_string(&len);
                    437:
                    438:        TRACE("write id %d handle %d off %qd len %d", id, handle, off, len);
                    439:        fd = handle_to_fd(handle);
                    440:        if (fd >= 0) {
                    441:                if (lseek(fd, off, SEEK_SET) < 0) {
                    442:                        status = errno_to_portable(errno);
                    443:                        error("process_write: seek failed");
                    444:                } else {
                    445: /* XXX ATOMICIO ? */
                    446:                        ret = write(fd, data, len);
                    447:                        if (ret == -1) {
                    448:                                error("process_write: write failed");
                    449:                                status = errno_to_portable(errno);
                    450:                        } else if (ret == len) {
1.10      markus    451:                                status = SSH2_FX_OK;
1.1       markus    452:                        } else {
                    453:                                log("nothing at all written");
                    454:                        }
                    455:                }
                    456:        }
                    457:        send_status(id, status);
                    458:        xfree(data);
                    459: }
                    460:
                    461: void
                    462: process_do_stat(int do_lstat)
                    463: {
1.13      markus    464:        Attrib a;
1.1       markus    465:        struct stat st;
                    466:        u_int32_t id;
                    467:        char *name;
1.10      markus    468:        int ret, status = SSH2_FX_FAILURE;
1.1       markus    469:
                    470:        id = get_int();
                    471:        name = get_string(NULL);
                    472:        TRACE("%sstat id %d name %s", do_lstat ? "l" : "", id, name);
                    473:        ret = do_lstat ? lstat(name, &st) : stat(name, &st);
                    474:        if (ret < 0) {
                    475:                status = errno_to_portable(errno);
                    476:        } else {
1.13      markus    477:                stat_to_attrib(&st, &a);
                    478:                send_attrib(id, &a);
1.10      markus    479:                status = SSH2_FX_OK;
1.1       markus    480:        }
1.10      markus    481:        if (status != SSH2_FX_OK)
1.1       markus    482:                send_status(id, status);
                    483:        xfree(name);
                    484: }
                    485:
                    486: void
                    487: process_stat(void)
                    488: {
                    489:        process_do_stat(0);
                    490: }
                    491:
                    492: void
                    493: process_lstat(void)
                    494: {
                    495:        process_do_stat(1);
                    496: }
                    497:
                    498: void
                    499: process_fstat(void)
                    500: {
1.13      markus    501:        Attrib a;
1.1       markus    502:        struct stat st;
                    503:        u_int32_t id;
1.10      markus    504:        int fd, ret, handle, status = SSH2_FX_FAILURE;
1.1       markus    505:
                    506:        id = get_int();
                    507:        handle = get_handle();
                    508:        TRACE("fstat id %d handle %d", id, handle);
                    509:        fd = handle_to_fd(handle);
                    510:        if (fd  >= 0) {
                    511:                ret = fstat(fd, &st);
                    512:                if (ret < 0) {
                    513:                        status = errno_to_portable(errno);
                    514:                } else {
1.13      markus    515:                        stat_to_attrib(&st, &a);
                    516:                        send_attrib(id, &a);
1.10      markus    517:                        status = SSH2_FX_OK;
1.1       markus    518:                }
                    519:        }
1.10      markus    520:        if (status != SSH2_FX_OK)
1.1       markus    521:                send_status(id, status);
                    522: }
                    523:
                    524: struct timeval *
                    525: attrib_to_tv(Attrib *a)
                    526: {
                    527:        static struct timeval tv[2];
                    528:        tv[0].tv_sec = a->atime;
                    529:        tv[0].tv_usec = 0;
                    530:        tv[1].tv_sec = a->mtime;
                    531:        tv[1].tv_usec = 0;
                    532:        return tv;
                    533: }
                    534:
                    535: void
                    536: process_setstat(void)
                    537: {
                    538:        Attrib *a;
                    539:        u_int32_t id;
                    540:        char *name;
                    541:        int ret;
1.10      markus    542:        int status = SSH2_FX_OK;
1.1       markus    543:
                    544:        id = get_int();
                    545:        name = get_string(NULL);
                    546:        a = get_attrib();
                    547:        TRACE("setstat id %d name %s", id, name);
1.10      markus    548:        if (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) {
1.1       markus    549:                ret = chmod(name, a->perm & 0777);
                    550:                if (ret == -1)
                    551:                        status = errno_to_portable(errno);
                    552:        }
1.10      markus    553:        if (a->flags & SSH2_FILEXFER_ATTR_ACMODTIME) {
1.1       markus    554:                ret = utimes(name, attrib_to_tv(a));
                    555:                if (ret == -1)
                    556:                        status = errno_to_portable(errno);
                    557:        }
                    558:        send_status(id, status);
                    559:        xfree(name);
                    560: }
                    561:
                    562: void
                    563: process_fsetstat(void)
                    564: {
                    565:        Attrib *a;
                    566:        u_int32_t id;
                    567:        int handle, fd, ret;
1.10      markus    568:        int status = SSH2_FX_OK;
1.1       markus    569:
                    570:        id = get_int();
                    571:        handle = get_handle();
                    572:        a = get_attrib();
                    573:        TRACE("fsetstat id %d handle %d", id, handle);
                    574:        fd = handle_to_fd(handle);
                    575:        if (fd < 0) {
1.10      markus    576:                status = SSH2_FX_FAILURE;
1.1       markus    577:        } else {
1.10      markus    578:                if (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) {
1.1       markus    579:                        ret = fchmod(fd, a->perm & 0777);
                    580:                        if (ret == -1)
                    581:                                status = errno_to_portable(errno);
                    582:                }
1.10      markus    583:                if (a->flags & SSH2_FILEXFER_ATTR_ACMODTIME) {
1.1       markus    584:                        ret = futimes(fd, attrib_to_tv(a));
                    585:                        if (ret == -1)
                    586:                                status = errno_to_portable(errno);
                    587:                }
                    588:        }
                    589:        send_status(id, status);
                    590: }
                    591:
                    592: void
                    593: process_opendir(void)
                    594: {
                    595:        DIR *dirp = NULL;
                    596:        char *path;
1.10      markus    597:        int handle, status = SSH2_FX_FAILURE;
1.1       markus    598:        u_int32_t id;
                    599:
                    600:        id = get_int();
                    601:        path = get_string(NULL);
                    602:        TRACE("opendir id %d path %s", id, path);
                    603:        dirp = opendir(path);
                    604:        if (dirp == NULL) {
                    605:                status = errno_to_portable(errno);
                    606:        } else {
                    607:                handle = handle_new(HANDLE_DIR, xstrdup(path), 0, dirp);
                    608:                if (handle < 0) {
                    609:                        closedir(dirp);
                    610:                } else {
                    611:                        send_handle(id, handle);
1.10      markus    612:                        status = SSH2_FX_OK;
1.1       markus    613:                }
                    614:
                    615:        }
1.10      markus    616:        if (status != SSH2_FX_OK)
1.1       markus    617:                send_status(id, status);
                    618:        xfree(path);
                    619: }
                    620:
1.10      markus    621: /*
1.12      markus    622:  * drwxr-xr-x    5 markus   markus       1024 Jan 13 18:39 .ssh
1.10      markus    623:  */
1.1       markus    624: char *
                    625: ls_file(char *name, struct stat *st)
                    626: {
1.12      markus    627:        int sz = 0;
                    628:        struct passwd *pw;
                    629:        struct group *gr;
                    630:        struct tm *ltime = localtime(&st->st_mtime);
                    631:        char *user, *group;
                    632:        char buf[1024], mode[11+1], tbuf[12+1], ubuf[11+1], gbuf[11+1];
                    633:
                    634:        strmode(st->st_mode, mode);
                    635:        if ((pw = getpwuid(st->st_uid)) != NULL) {
                    636:                user = pw->pw_name;
                    637:        } else {
                    638:                snprintf(ubuf, sizeof ubuf, "%d", st->st_uid);
                    639:                user = ubuf;
                    640:        }
                    641:        if ((gr = getgrgid(st->st_gid)) != NULL) {
                    642:                group = gr->gr_name;
                    643:        } else {
                    644:                snprintf(gbuf, sizeof gbuf, "%d", st->st_gid);
                    645:                group = gbuf;
                    646:        }
                    647:        if (ltime != NULL) {
                    648:                if (time(NULL) - st->st_mtime < (365*24*60*60)/2)
                    649:                        sz = strftime(tbuf, sizeof tbuf, "%b %e %H:%M", ltime);
                    650:                else
                    651:                        sz = strftime(tbuf, sizeof tbuf, "%b %e  %Y", ltime);
                    652:        }
                    653:        if (sz == 0)
                    654:                tbuf[0] = '\0';
                    655:        snprintf(buf, sizeof buf, "%s %3d %-8.8s %-8.8s %8qd %s %s", mode,
                    656:            st->st_nlink, user, group, (long long)st->st_size, tbuf, name);
1.1       markus    657:        return xstrdup(buf);
                    658: }
                    659:
                    660: void
                    661: process_readdir(void)
                    662: {
                    663:        DIR *dirp;
                    664:        struct dirent *dp;
                    665:        char *path;
                    666:        int handle;
                    667:        u_int32_t id;
                    668:
                    669:        id = get_int();
                    670:        handle = get_handle();
                    671:        TRACE("readdir id %d handle %d", id, handle);
                    672:        dirp = handle_to_dir(handle);
                    673:        path = handle_to_name(handle);
                    674:        if (dirp == NULL || path == NULL) {
1.10      markus    675:                send_status(id, SSH2_FX_FAILURE);
1.1       markus    676:        } else {
                    677:                struct stat st;
                    678:                char pathname[1024];
                    679:                Stat *stats;
                    680:                int nstats = 10, count = 0, i;
                    681:                stats = xmalloc(nstats * sizeof(Stat));
                    682:                while ((dp = readdir(dirp)) != NULL) {
                    683:                        if (count >= nstats) {
                    684:                                nstats *= 2;
                    685:                                stats = xrealloc(stats, nstats * sizeof(Stat));
                    686:                        }
                    687: /* XXX OVERFLOW ? */
                    688:                        snprintf(pathname, sizeof pathname,
                    689:                            "%s/%s", path, dp->d_name);
                    690:                        if (lstat(pathname, &st) < 0)
                    691:                                continue;
1.13      markus    692:                        stat_to_attrib(&st, &(stats[count].attrib));
1.1       markus    693:                        stats[count].name = xstrdup(dp->d_name);
                    694:                        stats[count].long_name = ls_file(dp->d_name, &st);
                    695:                        count++;
                    696:                        /* send up to 100 entries in one message */
1.11      markus    697:                        /* XXX check packet size instead */
1.1       markus    698:                        if (count == 100)
                    699:                                break;
                    700:                }
1.10      markus    701:                if (count > 0) {
                    702:                        send_names(id, count, stats);
                    703:                        for(i = 0; i < count; i++) {
                    704:                                xfree(stats[i].name);
                    705:                                xfree(stats[i].long_name);
                    706:                        }
                    707:                } else {
                    708:                        send_status(id, SSH2_FX_EOF);
1.1       markus    709:                }
                    710:                xfree(stats);
                    711:        }
                    712: }
                    713:
                    714: void
                    715: process_remove(void)
                    716: {
                    717:        char *name;
                    718:        u_int32_t id;
1.10      markus    719:        int status = SSH2_FX_FAILURE;
1.1       markus    720:        int ret;
                    721:
                    722:        id = get_int();
                    723:        name = get_string(NULL);
                    724:        TRACE("remove id %d name %s", id, name);
1.8       markus    725:        ret = unlink(name);
1.10      markus    726:        status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
1.1       markus    727:        send_status(id, status);
                    728:        xfree(name);
                    729: }
                    730:
                    731: void
                    732: process_mkdir(void)
                    733: {
                    734:        Attrib *a;
                    735:        u_int32_t id;
                    736:        char *name;
1.10      markus    737:        int ret, mode, status = SSH2_FX_FAILURE;
1.1       markus    738:
                    739:        id = get_int();
                    740:        name = get_string(NULL);
                    741:        a = get_attrib();
1.10      markus    742:        mode = (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) ?
                    743:            a->perm & 0777 : 0777;
1.1       markus    744:        TRACE("mkdir id %d name %s mode 0%o", id, name, mode);
                    745:        ret = mkdir(name, mode);
1.10      markus    746:        status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
1.1       markus    747:        send_status(id, status);
                    748:        xfree(name);
                    749: }
                    750:
                    751: void
                    752: process_rmdir(void)
                    753: {
                    754:        u_int32_t id;
                    755:        char *name;
                    756:        int ret, status;
                    757:
                    758:        id = get_int();
                    759:        name = get_string(NULL);
                    760:        TRACE("rmdir id %d name %s", id, name);
                    761:        ret = rmdir(name);
1.10      markus    762:        status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
1.1       markus    763:        send_status(id, status);
                    764:        xfree(name);
                    765: }
                    766:
                    767: void
                    768: process_realpath(void)
                    769: {
                    770:        char resolvedname[MAXPATHLEN];
                    771:        u_int32_t id;
                    772:        char *path;
                    773:
                    774:        id = get_int();
                    775:        path = get_string(NULL);
1.7       markus    776:        if (path[0] == '\0') {
                    777:                xfree(path);
                    778:                path = xstrdup(".");
                    779:        }
1.1       markus    780:        TRACE("realpath id %d path %s", id, path);
                    781:        if (realpath(path, resolvedname) == NULL) {
                    782:                send_status(id, errno_to_portable(errno));
                    783:        } else {
                    784:                Stat s;
                    785:                attrib_clear(&s.attrib);
                    786:                s.name = s.long_name = resolvedname;
                    787:                send_names(id, 1, &s);
                    788:        }
                    789:        xfree(path);
                    790: }
                    791:
                    792: void
                    793: process_rename(void)
                    794: {
                    795:        u_int32_t id;
1.11      markus    796:        struct stat st;
1.1       markus    797:        char *oldpath, *newpath;
1.11      markus    798:        int ret, status = SSH2_FX_FAILURE;
1.1       markus    799:
                    800:        id = get_int();
                    801:        oldpath = get_string(NULL);
                    802:        newpath = get_string(NULL);
                    803:        TRACE("rename id %d old %s new %s", id, oldpath, newpath);
1.11      markus    804:        /* fail if 'newpath' exists */
                    805:        if (stat(newpath, &st) == -1) {
                    806:                ret = rename(oldpath, newpath);
                    807:                status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
                    808:        }
1.1       markus    809:        send_status(id, status);
                    810:        xfree(oldpath);
                    811:        xfree(newpath);
                    812: }
                    813:
1.10      markus    814: void
                    815: process_extended(void)
                    816: {
                    817:        u_int32_t id;
                    818:        char *request;
                    819:
                    820:        id = get_int();
                    821:        request = get_string(NULL);
                    822:        send_status(id, SSH2_FX_OP_UNSUPPORTED);                /* MUST */
                    823:        xfree(request);
                    824: }
1.1       markus    825:
                    826: /* stolen from ssh-agent */
                    827:
                    828: void
                    829: process(void)
                    830: {
1.9       markus    831:        u_int msg_len;
                    832:        u_int type;
                    833:        u_char *cp;
1.1       markus    834:
                    835:        if (buffer_len(&iqueue) < 5)
                    836:                return;         /* Incomplete message. */
1.9       markus    837:        cp = (u_char *) buffer_ptr(&iqueue);
1.1       markus    838:        msg_len = GET_32BIT(cp);
                    839:        if (msg_len > 256 * 1024) {
                    840:                error("bad message ");
                    841:                exit(11);
                    842:        }
                    843:        if (buffer_len(&iqueue) < msg_len + 4)
                    844:                return;
                    845:        buffer_consume(&iqueue, 4);
                    846:        type = buffer_get_char(&iqueue);
                    847:        switch (type) {
1.10      markus    848:        case SSH2_FXP_INIT:
1.1       markus    849:                process_init();
                    850:                break;
1.10      markus    851:        case SSH2_FXP_OPEN:
1.1       markus    852:                process_open();
                    853:                break;
1.10      markus    854:        case SSH2_FXP_CLOSE:
1.1       markus    855:                process_close();
                    856:                break;
1.10      markus    857:        case SSH2_FXP_READ:
1.1       markus    858:                process_read();
                    859:                break;
1.10      markus    860:        case SSH2_FXP_WRITE:
1.1       markus    861:                process_write();
                    862:                break;
1.10      markus    863:        case SSH2_FXP_LSTAT:
1.1       markus    864:                process_lstat();
                    865:                break;
1.10      markus    866:        case SSH2_FXP_FSTAT:
1.1       markus    867:                process_fstat();
                    868:                break;
1.10      markus    869:        case SSH2_FXP_SETSTAT:
1.1       markus    870:                process_setstat();
                    871:                break;
1.10      markus    872:        case SSH2_FXP_FSETSTAT:
1.1       markus    873:                process_fsetstat();
                    874:                break;
1.10      markus    875:        case SSH2_FXP_OPENDIR:
1.1       markus    876:                process_opendir();
                    877:                break;
1.10      markus    878:        case SSH2_FXP_READDIR:
1.1       markus    879:                process_readdir();
                    880:                break;
1.10      markus    881:        case SSH2_FXP_REMOVE:
1.1       markus    882:                process_remove();
                    883:                break;
1.10      markus    884:        case SSH2_FXP_MKDIR:
1.1       markus    885:                process_mkdir();
                    886:                break;
1.10      markus    887:        case SSH2_FXP_RMDIR:
1.1       markus    888:                process_rmdir();
                    889:                break;
1.10      markus    890:        case SSH2_FXP_REALPATH:
1.1       markus    891:                process_realpath();
                    892:                break;
1.10      markus    893:        case SSH2_FXP_STAT:
1.1       markus    894:                process_stat();
                    895:                break;
1.10      markus    896:        case SSH2_FXP_RENAME:
1.1       markus    897:                process_rename();
                    898:                break;
1.10      markus    899:        case SSH2_FXP_EXTENDED:
                    900:                process_extended();
                    901:                break;
1.1       markus    902:        default:
                    903:                error("Unknown message %d", type);
                    904:                break;
                    905:        }
                    906: }
                    907:
                    908: int
                    909: main(int ac, char **av)
                    910: {
                    911:        fd_set rset, wset;
                    912:        int in, out, max;
1.5       markus    913:        ssize_t len, olen;
1.1       markus    914:
                    915:        handle_init();
1.10      markus    916:
1.11      markus    917: #ifdef DEBUG_SFTP_SERVER
1.10      markus    918:         log_init("sftp-server", SYSLOG_LEVEL_DEBUG1, SYSLOG_FACILITY_AUTH, 0);
1.11      markus    919: #endif
1.1       markus    920:
                    921:        in = dup(STDIN_FILENO);
                    922:        out = dup(STDOUT_FILENO);
                    923:
                    924:        max = 0;
                    925:        if (in > max)
                    926:                max = in;
                    927:        if (out > max)
                    928:                max = out;
                    929:
                    930:        buffer_init(&iqueue);
                    931:        buffer_init(&oqueue);
                    932:
                    933:        for (;;) {
                    934:                FD_ZERO(&rset);
                    935:                FD_ZERO(&wset);
                    936:
                    937:                FD_SET(in, &rset);
                    938:                olen = buffer_len(&oqueue);
                    939:                if (olen > 0)
                    940:                        FD_SET(out, &wset);
                    941:
                    942:                if (select(max+1, &rset, &wset, NULL, NULL) < 0) {
                    943:                        if (errno == EINTR)
                    944:                                continue;
                    945:                        exit(2);
                    946:                }
                    947:
                    948:                /* copy stdin to iqueue */
                    949:                if (FD_ISSET(in, &rset)) {
                    950:                        char buf[4*4096];
                    951:                        len = read(in, buf, sizeof buf);
                    952:                        if (len == 0) {
                    953:                                debug("read eof");
                    954:                                exit(0);
                    955:                        } else if (len < 0) {
                    956:                                error("read error");
                    957:                                exit(1);
                    958:                        } else {
                    959:                                buffer_append(&iqueue, buf, len);
                    960:                        }
                    961:                }
                    962:                /* send oqueue to stdout */
                    963:                if (FD_ISSET(out, &wset)) {
                    964:                        len = write(out, buffer_ptr(&oqueue), olen);
                    965:                        if (len < 0) {
                    966:                                error("write error");
                    967:                                exit(1);
                    968:                        } else {
                    969:                                buffer_consume(&oqueue, len);
                    970:                        }
                    971:                }
                    972:                /* process requests from client */
                    973:                process();
                    974:        }
                    975: }