crypto/openssh-4/sftp-server.c

   1 /* $OpenBSD: sftp-server.c,v 1.73 2007/05/17 07:55:29 djm Exp $ */
   2 /*
   3  * Copyright (c) 2000-2004 Markus Friedl.  All rights reserved.
   4  *
   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.
   8  *
   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.
  16  */
  17
  18 #include "includes.h"
  19
  20 #include <sys/types.h>
  21 #include <sys/param.h>
  22 #include <sys/stat.h>
  23 #ifdef HAVE_SYS_TIME_H
  24 # include <sys/time.h>
  25 #endif
  26
  27 #include <dirent.h>
  28 #include <errno.h>
  29 #include <fcntl.h>
  30 #include <pwd.h>
  31 #include <stdlib.h>
  32 #include <stdio.h>
  33 #include <string.h>
  34 #include <pwd.h>
  35 #include <time.h>
  36 #include <unistd.h>
  37 #include <stdarg.h>
  38
  39 #include "xmalloc.h"
  40 #include "buffer.h"
  41 #include "log.h"
  42 #include "misc.h"
  43 #include "uidswap.h"
  44
  45 #include "sftp.h"
  46 #include "sftp-common.h"
  47
  48 /* helper */
  49 #define get_int64()                     buffer_get_int64(&iqueue);
  50 #define get_int()                       buffer_get_int(&iqueue);
  51 #define get_string(lenp)                buffer_get_string(&iqueue, lenp);
  52
  53 /* Our verbosity */
  54 LogLevel log_level = SYSLOG_LEVEL_ERROR;
  55
  56 /* Our client */
  57 struct passwd *pw = NULL;
  58 char *client_addr = NULL;
  59
  60 /* input and output queue */
  61 Buffer iqueue;
  62 Buffer oqueue;
  63
  64 /* Version of client */
  65 int version;
  66
  67 /* portable attributes, etc. */
  68
  69 typedef struct Stat Stat;
  70
  71 struct Stat {
  72         char *name;
  73         char *long_name;
  74         Attrib attrib;
  75 };
  76
  77 static int
  78 errno_to_portable(int unixerrno)
  79 {
  80         int ret = 0;
  81
  82         switch (unixerrno) {
  83         case 0:
  84                 ret = SSH2_FX_OK;
  85                 break;
  86         case ENOENT:
  87         case ENOTDIR:
  88         case EBADF:
  89         case ELOOP:
  90                 ret = SSH2_FX_NO_SUCH_FILE;
  91                 break;
  92         case EPERM:
  93         case EACCES:
  94         case EFAULT:
  95                 ret = SSH2_FX_PERMISSION_DENIED;
  96                 break;
  97         case ENAMETOOLONG:
  98         case EINVAL:
  99                 ret = SSH2_FX_BAD_MESSAGE;
 100                 break;
 101         default:
 102                 ret = SSH2_FX_FAILURE;
 103                 break;
 104         }
 105         return ret;
 106 }
 107
 108 static int
 109 flags_from_portable(int pflags)
 110 {
 111         int flags = 0;
 112
 113         if ((pflags & SSH2_FXF_READ) &&
 114             (pflags & SSH2_FXF_WRITE)) {
 115                 flags = O_RDWR;
 116         } else if (pflags & SSH2_FXF_READ) {
 117                 flags = O_RDONLY;
 118         } else if (pflags & SSH2_FXF_WRITE) {
 119                 flags = O_WRONLY;
 120         }
 121         if (pflags & SSH2_FXF_CREAT)
 122                 flags |= O_CREAT;
 123         if (pflags & SSH2_FXF_TRUNC)
 124                 flags |= O_TRUNC;
 125         if (pflags & SSH2_FXF_EXCL)
 126                 flags |= O_EXCL;
 127         return flags;
 128 }
 129
 130 static const char *
 131 string_from_portable(int pflags)
 132 {
 133         static char ret[128];
 134
 135         *ret = '\0';
 136
 137 #define PAPPEND(str)    {                               \
 138                 if (*ret != '\0')                       \
 139                         strlcat(ret, ",", sizeof(ret)); \
 140                 strlcat(ret, str, sizeof(ret));         \
 141         }
 142
 143         if (pflags & SSH2_FXF_READ)
 144                 PAPPEND("READ")
 145         if (pflags & SSH2_FXF_WRITE)
 146                 PAPPEND("WRITE")
 147         if (pflags & SSH2_FXF_CREAT)
 148                 PAPPEND("CREATE")
 149         if (pflags & SSH2_FXF_TRUNC)
 150                 PAPPEND("TRUNCATE")
 151         if (pflags & SSH2_FXF_EXCL)
 152                 PAPPEND("EXCL")
 153
 154         return ret;
 155 }
 156
 157 static Attrib *
 158 get_attrib(void)
 159 {
 160         return decode_attrib(&iqueue);
 161 }
 162
 163 /* handle handles */
 164
 165 typedef struct Handle Handle;
 166 struct Handle {
 167         int use;
 168         DIR *dirp;
 169         int fd;
 170         char *name;
 171         u_int64_t bytes_read, bytes_write;
 172 };
 173
 174 enum {
 175         HANDLE_UNUSED,
 176         HANDLE_DIR,
 177         HANDLE_FILE
 178 };
 179
 180 Handle  handles[100];
 181
 182 static void
 183 handle_init(void)
 184 {
 185         u_int i;
 186
 187         for (i = 0; i < sizeof(handles)/sizeof(Handle); i++)
 188                 handles[i].use = HANDLE_UNUSED;
 189 }
 190
 191 static int
 192 handle_new(int use, const char *name, int fd, DIR *dirp)
 193 {
 194         u_int i;
 195
 196         for (i = 0; i < sizeof(handles)/sizeof(Handle); i++) {
 197                 if (handles[i].use == HANDLE_UNUSED) {
 198                         handles[i].use = use;
 199                         handles[i].dirp = dirp;
 200                         handles[i].fd = fd;
 201                         handles[i].name = xstrdup(name);
 202                         handles[i].bytes_read = handles[i].bytes_write = 0;
 203                         return i;
 204                 }
 205         }
 206         return -1;
 207 }
 208
 209 static int
 210 handle_is_ok(int i, int type)
 211 {
 212         return i >= 0 && (u_int)i < sizeof(handles)/sizeof(Handle) &&
 213             handles[i].use == type;
 214 }
 215
 216 static int
 217 handle_to_string(int handle, char **stringp, int *hlenp)
 218 {
 219         if (stringp == NULL || hlenp == NULL)
 220                 return -1;
 221         *stringp = xmalloc(sizeof(int32_t));
 222         put_u32(*stringp, handle);
 223         *hlenp = sizeof(int32_t);
 224         return 0;
 225 }
 226
 227 static int
 228 handle_from_string(const char *handle, u_int hlen)
 229 {
 230         int val;
 231
 232         if (hlen != sizeof(int32_t))
 233                 return -1;
 234         val = get_u32(handle);
 235         if (handle_is_ok(val, HANDLE_FILE) ||
 236             handle_is_ok(val, HANDLE_DIR))
 237                 return val;
 238         return -1;
 239 }
 240
 241 static char *
 242 handle_to_name(int handle)
 243 {
 244         if (handle_is_ok(handle, HANDLE_DIR)||
 245             handle_is_ok(handle, HANDLE_FILE))
 246                 return handles[handle].name;
 247         return NULL;
 248 }
 249
 250 static DIR *
 251 handle_to_dir(int handle)
 252 {
 253         if (handle_is_ok(handle, HANDLE_DIR))
 254                 return handles[handle].dirp;
 255         return NULL;
 256 }
 257
 258 static int
 259 handle_to_fd(int handle)
 260 {
 261         if (handle_is_ok(handle, HANDLE_FILE))
 262                 return handles[handle].fd;
 263         return -1;
 264 }
 265
 266 static void
 267 handle_update_read(int handle, ssize_t bytes)
 268 {
 269         if (handle_is_ok(handle, HANDLE_FILE) && bytes > 0)
 270                 handles[handle].bytes_read += bytes;
 271 }
 272
 273 static void
 274 handle_update_write(int handle, ssize_t bytes)
 275 {
 276         if (handle_is_ok(handle, HANDLE_FILE) && bytes > 0)
 277                 handles[handle].bytes_write += bytes;
 278 }
 279
 280 static u_int64_t
 281 handle_bytes_read(int handle)
 282 {
 283         if (handle_is_ok(handle, HANDLE_FILE))
 284                 return (handles[handle].bytes_read);
 285         return 0;
 286 }
 287
 288 static u_int64_t
 289 handle_bytes_write(int handle)
 290 {
 291         if (handle_is_ok(handle, HANDLE_FILE))
 292                 return (handles[handle].bytes_write);
 293         return 0;
 294 }
 295
 296 static int
 297 handle_close(int handle)
 298 {
 299         int ret = -1;
 300
 301         if (handle_is_ok(handle, HANDLE_FILE)) {
 302                 ret = close(handles[handle].fd);
 303                 handles[handle].use = HANDLE_UNUSED;
 304                 xfree(handles[handle].name);
 305         } else if (handle_is_ok(handle, HANDLE_DIR)) {
 306                 ret = closedir(handles[handle].dirp);
 307                 handles[handle].use = HANDLE_UNUSED;
 308                 xfree(handles[handle].name);
 309         } else {
 310                 errno = ENOENT;
 311         }
 312         return ret;
 313 }
 314
 315 static void
 316 handle_log_close(int handle, char *emsg)
 317 {
 318         if (handle_is_ok(handle, HANDLE_FILE)) {
 319                 logit("%s%sclose \"%s\" bytes read %llu written %llu",
 320                     emsg == NULL ? "" : emsg, emsg == NULL ? "" : " ",
 321                     handle_to_name(handle),
 322                     (unsigned long long)handle_bytes_read(handle),
 323                     (unsigned long long)handle_bytes_write(handle));
 324         } else {
 325                 logit("%s%sclosedir \"%s\"",
 326                     emsg == NULL ? "" : emsg, emsg == NULL ? "" : " ",
 327                     handle_to_name(handle));
 328         }
 329 }
 330
 331 static void
 332 handle_log_exit(void)
 333 {
 334         u_int i;
 335
 336         for (i = 0; i < sizeof(handles)/sizeof(Handle); i++)
 337                 if (handles[i].use != HANDLE_UNUSED)
 338                         handle_log_close(i, "forced");
 339 }
 340
 341 static int
 342 get_handle(void)
 343 {
 344         char *handle;
 345         int val = -1;
 346         u_int hlen;
 347
 348         handle = get_string(&hlen);
 349         if (hlen < 256)
 350                 val = handle_from_string(handle, hlen);
 351         xfree(handle);
 352         return val;
 353 }
 354
 355 /* send replies */
 356
 357 static void
 358 send_msg(Buffer *m)
 359 {
 360         int mlen = buffer_len(m);
 361
 362         buffer_put_int(&oqueue, mlen);
 363         buffer_append(&oqueue, buffer_ptr(m), mlen);
 364         buffer_consume(m, mlen);
 365 }
 366
 367 static const char *
 368 status_to_message(u_int32_t status)
 369 {
 370         const char *status_messages[] = {
 371                 "Success",                      /* SSH_FX_OK */
 372                 "End of file",                  /* SSH_FX_EOF */
 373                 "No such file",                 /* SSH_FX_NO_SUCH_FILE */
 374                 "Permission denied",            /* SSH_FX_PERMISSION_DENIED */
 375                 "Failure",                      /* SSH_FX_FAILURE */
 376                 "Bad message",                  /* SSH_FX_BAD_MESSAGE */
 377                 "No connection",                /* SSH_FX_NO_CONNECTION */
 378                 "Connection lost",              /* SSH_FX_CONNECTION_LOST */
 379                 "Operation unsupported",        /* SSH_FX_OP_UNSUPPORTED */
 380                 "Unknown error"                 /* Others */
 381         };
 382         return (status_messages[MIN(status,SSH2_FX_MAX)]);
 383 }
 384
 385 static void
 386 send_status(u_int32_t id, u_int32_t status)
 387 {
 388         Buffer msg;
 389
 390         debug3("request %u: sent status %u", id, status);
 391         if (log_level > SYSLOG_LEVEL_VERBOSE ||
 392             (status != SSH2_FX_OK && status != SSH2_FX_EOF))
 393                 logit("sent status %s", status_to_message(status));
 394         buffer_init(&msg);
 395         buffer_put_char(&msg, SSH2_FXP_STATUS);
 396         buffer_put_int(&msg, id);
 397         buffer_put_int(&msg, status);
 398         if (version >= 3) {
 399                 buffer_put_cstring(&msg, status_to_message(status));
 400                 buffer_put_cstring(&msg, "");
 401         }
 402         send_msg(&msg);
 403         buffer_free(&msg);
 404 }
 405 static void
 406 send_data_or_handle(char type, u_int32_t id, const char *data, int dlen)
 407 {
 408         Buffer msg;
 409
 410         buffer_init(&msg);
 411         buffer_put_char(&msg, type);
 412         buffer_put_int(&msg, id);
 413         buffer_put_string(&msg, data, dlen);
 414         send_msg(&msg);
 415         buffer_free(&msg);
 416 }
 417
 418 static void
 419 send_data(u_int32_t id, const char *data, int dlen)
 420 {
 421         debug("request %u: sent data len %d", id, dlen);
 422         send_data_or_handle(SSH2_FXP_DATA, id, data, dlen);
 423 }
 424
 425 static void
 426 send_handle(u_int32_t id, int handle)
 427 {
 428         char *string;
 429         int hlen;
 430
 431         handle_to_string(handle, &string, &hlen);
 432         debug("request %u: sent handle handle %d", id, handle);
 433         send_data_or_handle(SSH2_FXP_HANDLE, id, string, hlen);
 434         xfree(string);
 435 }
 436
 437 static void
 438 send_names(u_int32_t id, int count, const Stat *stats)
 439 {
 440         Buffer msg;
 441         int i;
 442
 443         buffer_init(&msg);
 444         buffer_put_char(&msg, SSH2_FXP_NAME);
 445         buffer_put_int(&msg, id);
 446         buffer_put_int(&msg, count);
 447         debug("request %u: sent names count %d", id, count);
 448         for (i = 0; i < count; i++) {
 449                 buffer_put_cstring(&msg, stats[i].name);
 450                 buffer_put_cstring(&msg, stats[i].long_name);
 451                 encode_attrib(&msg, &stats[i].attrib);
 452         }
 453         send_msg(&msg);
 454         buffer_free(&msg);
 455 }
 456
 457 static void
 458 send_attrib(u_int32_t id, const Attrib *a)
 459 {
 460         Buffer msg;
 461
 462         debug("request %u: sent attrib have 0x%x", id, a->flags);
 463         buffer_init(&msg);
 464         buffer_put_char(&msg, SSH2_FXP_ATTRS);
 465         buffer_put_int(&msg, id);
 466         encode_attrib(&msg, a);
 467         send_msg(&msg);
 468         buffer_free(&msg);
 469 }
 470
 471 /* parse incoming */
 472
 473 static void
 474 process_init(void)
 475 {
 476         Buffer msg;
 477
 478         version = get_int();
 479         verbose("received client version %d", version);
 480         buffer_init(&msg);
 481         buffer_put_char(&msg, SSH2_FXP_VERSION);
 482         buffer_put_int(&msg, SSH2_FILEXFER_VERSION);
 483         send_msg(&msg);
 484         buffer_free(&msg);
 485 }
 486
 487 static void
 488 process_open(void)
 489 {
 490         u_int32_t id, pflags;
 491         Attrib *a;
 492         char *name;
 493         int handle, fd, flags, mode, status = SSH2_FX_FAILURE;
 494
 495         id = get_int();
 496         name = get_string(NULL);
 497         pflags = get_int();             /* portable flags */
 498         debug3("request %u: open flags %d", id, pflags);
 499         a = get_attrib();
 500         flags = flags_from_portable(pflags);
 501         mode = (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) ? a->perm : 0666;
 502         logit("open \"%s\" flags %s mode 0%o",
 503             name, string_from_portable(pflags), mode);
 504         fd = open(name, flags, mode);
 505         if (fd < 0) {
 506                 status = errno_to_portable(errno);
 507         } else {
 508                 handle = handle_new(HANDLE_FILE, name, fd, NULL);
 509                 if (handle < 0) {
 510                         close(fd);
 511                 } else {
 512                         send_handle(id, handle);
 513                         status = SSH2_FX_OK;
 514                 }
 515         }
 516         if (status != SSH2_FX_OK)
 517                 send_status(id, status);
 518         xfree(name);
 519 }
 520
 521 static void
 522 process_close(void)
 523 {
 524         u_int32_t id;
 525         int handle, ret, status = SSH2_FX_FAILURE;
 526
 527         id = get_int();
 528         handle = get_handle();
 529         debug3("request %u: close handle %u", id, handle);
 530         handle_log_close(handle, NULL);
 531         ret = handle_close(handle);
 532         status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
 533         send_status(id, status);
 534 }
 535
 536 static void
 537 process_read(void)
 538 {
 539         char buf[64*1024];
 540         u_int32_t id, len;
 541         int handle, fd, ret, status = SSH2_FX_FAILURE;
 542         u_int64_t off;
 543
 544         id = get_int();
 545         handle = get_handle();
 546         off = get_int64();
 547         len = get_int();
 548
 549         debug("request %u: read \"%s\" (handle %d) off %llu len %d",
 550             id, handle_to_name(handle), handle, (unsigned long long)off, len);
 551         if (len > sizeof buf) {
 552                 len = sizeof buf;
 553                 debug2("read change len %d", len);
 554         }
 555         fd = handle_to_fd(handle);
 556         if (fd >= 0) {
 557                 if (lseek(fd, off, SEEK_SET) < 0) {
 558                         error("process_read: seek failed");
 559                         status = errno_to_portable(errno);
 560                 } else {
 561                         ret = read(fd, buf, len);
 562                         if (ret < 0) {
 563                                 status = errno_to_portable(errno);
 564                         } else if (ret == 0) {
 565                                 status = SSH2_FX_EOF;
 566                         } else {
 567                                 send_data(id, buf, ret);
 568                                 status = SSH2_FX_OK;
 569                                 handle_update_read(handle, ret);
 570                         }
 571                 }
 572         }
 573         if (status != SSH2_FX_OK)
 574                 send_status(id, status);
 575 }
 576
 577 static void
 578 process_write(void)
 579 {
 580         u_int32_t id;
 581         u_int64_t off;
 582         u_int len;
 583         int handle, fd, ret, status = SSH2_FX_FAILURE;
 584         char *data;
 585
 586         id = get_int();
 587         handle = get_handle();
 588         off = get_int64();
 589         data = get_string(&len);
 590
 591         debug("request %u: write \"%s\" (handle %d) off %llu len %d",
 592             id, handle_to_name(handle), handle, (unsigned long long)off, len);
 593         fd = handle_to_fd(handle);
 594         if (fd >= 0) {
 595                 if (lseek(fd, off, SEEK_SET) < 0) {
 596                         status = errno_to_portable(errno);
 597                         error("process_write: seek failed");
 598                 } else {
 599 /* XXX ATOMICIO ? */
 600                         ret = write(fd, data, len);
 601                         if (ret < 0) {
 602                                 error("process_write: write failed");
 603                                 status = errno_to_portable(errno);
 604                         } else if ((size_t)ret == len) {
 605                                 status = SSH2_FX_OK;
 606                                 handle_update_write(handle, ret);
 607                         } else {
 608                                 debug2("nothing at all written");
 609                         }
 610                 }
 611         }
 612         send_status(id, status);
 613         xfree(data);
 614 }
 615
 616 static void
 617 process_do_stat(int do_lstat)
 618 {
 619         Attrib a;
 620         struct stat st;
 621         u_int32_t id;
 622         char *name;
 623         int ret, status = SSH2_FX_FAILURE;
 624
 625         id = get_int();
 626         name = get_string(NULL);
 627         debug3("request %u: %sstat", id, do_lstat ? "l" : "");
 628         verbose("%sstat name \"%s\"", do_lstat ? "l" : "", name);
 629         ret = do_lstat ? lstat(name, &st) : stat(name, &st);
 630         if (ret < 0) {
 631                 status = errno_to_portable(errno);
 632         } else {
 633                 stat_to_attrib(&st, &a);
 634                 send_attrib(id, &a);
 635                 status = SSH2_FX_OK;
 636         }
 637         if (status != SSH2_FX_OK)
 638                 send_status(id, status);
 639         xfree(name);
 640 }
 641
 642 static void
 643 process_stat(void)
 644 {
 645         process_do_stat(0);
 646 }
 647
 648 static void
 649 process_lstat(void)
 650 {
 651         process_do_stat(1);
 652 }
 653
 654 static void
 655 process_fstat(void)
 656 {
 657         Attrib a;
 658         struct stat st;
 659         u_int32_t id;
 660         int fd, ret, handle, status = SSH2_FX_FAILURE;
 661
 662         id = get_int();
 663         handle = get_handle();
 664         debug("request %u: fstat \"%s\" (handle %u)",
 665             id, handle_to_name(handle), handle);
 666         fd = handle_to_fd(handle);
 667         if (fd >= 0) {
 668                 ret = fstat(fd, &st);
 669                 if (ret < 0) {
 670                         status = errno_to_portable(errno);
 671                 } else {
 672                         stat_to_attrib(&st, &a);
 673                         send_attrib(id, &a);
 674                         status = SSH2_FX_OK;
 675                 }
 676         }
 677         if (status != SSH2_FX_OK)
 678                 send_status(id, status);
 679 }
 680
 681 static struct timeval *
 682 attrib_to_tv(const Attrib *a)
 683 {
 684         static struct timeval tv[2];
 685
 686         tv[0].tv_sec = a->atime;
 687         tv[0].tv_usec = 0;
 688         tv[1].tv_sec = a->mtime;
 689         tv[1].tv_usec = 0;
 690         return tv;
 691 }
 692
 693 static void
 694 process_setstat(void)
 695 {
 696         Attrib *a;
 697         u_int32_t id;
 698         char *name;
 699         int status = SSH2_FX_OK, ret;
 700
 701         id = get_int();
 702         name = get_string(NULL);
 703         a = get_attrib();
 704         debug("request %u: setstat name \"%s\"", id, name);
 705         if (a->flags & SSH2_FILEXFER_ATTR_SIZE) {
 706                 logit("set \"%s\" size %llu",
 707                     name, (unsigned long long)a->size);
 708                 ret = truncate(name, a->size);
 709                 if (ret == -1)
 710                         status = errno_to_portable(errno);
 711         }
 712         if (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) {
 713                 logit("set \"%s\" mode %04o", name, a->perm);
 714                 ret = chmod(name, a->perm & 0777);
 715                 if (ret == -1)
 716                         status = errno_to_portable(errno);
 717         }
 718         if (a->flags & SSH2_FILEXFER_ATTR_ACMODTIME) {
 719                 char buf[64];
 720                 time_t t = a->mtime;
 721
 722                 strftime(buf, sizeof(buf), "%Y%m%d-%H:%M:%S",
 723                     localtime(&t));
 724                 logit("set \"%s\" modtime %s", name, buf);
 725                 ret = utimes(name, attrib_to_tv(a));
 726                 if (ret == -1)
 727                         status = errno_to_portable(errno);
 728         }
 729         if (a->flags & SSH2_FILEXFER_ATTR_UIDGID) {
 730                 logit("set \"%s\" owner %lu group %lu", name,
 731                     (u_long)a->uid, (u_long)a->gid);
 732                 ret = chown(name, a->uid, a->gid);
 733                 if (ret == -1)
 734                         status = errno_to_portable(errno);
 735         }
 736         send_status(id, status);
 737         xfree(name);
 738 }
 739
 740 static void
 741 process_fsetstat(void)
 742 {
 743         Attrib *a;
 744         u_int32_t id;
 745         int handle, fd, ret;
 746         int status = SSH2_FX_OK;
 747
 748         id = get_int();
 749         handle = get_handle();
 750         a = get_attrib();
 751         debug("request %u: fsetstat handle %d", id, handle);
 752         fd = handle_to_fd(handle);
 753         if (fd < 0) {
 754                 status = SSH2_FX_FAILURE;
 755         } else {
 756                 char *name = handle_to_name(handle);
 757
 758                 if (a->flags & SSH2_FILEXFER_ATTR_SIZE) {
 759                         logit("set \"%s\" size %llu",
 760                             name, (unsigned long long)a->size);
 761                         ret = ftruncate(fd, a->size);
 762                         if (ret == -1)
 763                                 status = errno_to_portable(errno);
 764                 }
 765                 if (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) {
 766                         logit("set \"%s\" mode %04o", name, a->perm);
 767 #ifdef HAVE_FCHMOD
 768                         ret = fchmod(fd, a->perm & 0777);
 769 #else
 770                         ret = chmod(name, a->perm & 0777);
 771 #endif
 772                         if (ret == -1)
 773                                 status = errno_to_portable(errno);
 774                 }
 775                 if (a->flags & SSH2_FILEXFER_ATTR_ACMODTIME) {
 776                         char buf[64];
 777                         time_t t = a->mtime;
 778
 779                         strftime(buf, sizeof(buf), "%Y%m%d-%H:%M:%S",
 780                             localtime(&t));
 781                         logit("set \"%s\" modtime %s", name, buf);
 782 #ifdef HAVE_FUTIMES
 783                         ret = futimes(fd, attrib_to_tv(a));
 784 #else
 785                         ret = utimes(name, attrib_to_tv(a));
 786 #endif
 787                         if (ret == -1)
 788                                 status = errno_to_portable(errno);
 789                 }
 790                 if (a->flags & SSH2_FILEXFER_ATTR_UIDGID) {
 791                         logit("set \"%s\" owner %lu group %lu", name,
 792                             (u_long)a->uid, (u_long)a->gid);
 793 #ifdef HAVE_FCHOWN
 794                         ret = fchown(fd, a->uid, a->gid);
 795 #else
 796                         ret = chown(name, a->uid, a->gid);
 797 #endif
 798                         if (ret == -1)
 799                                 status = errno_to_portable(errno);
 800                 }
 801         }
 802         send_status(id, status);
 803 }
 804
 805 static void
 806 process_opendir(void)
 807 {
 808         DIR *dirp = NULL;
 809         char *path;
 810         int handle, status = SSH2_FX_FAILURE;
 811         u_int32_t id;
 812
 813         id = get_int();
 814         path = get_string(NULL);
 815         debug3("request %u: opendir", id);
 816         logit("opendir \"%s\"", path);
 817         dirp = opendir(path);
 818         if (dirp == NULL) {
 819                 status = errno_to_portable(errno);
 820         } else {
 821                 handle = handle_new(HANDLE_DIR, path, 0, dirp);
 822                 if (handle < 0) {
 823                         closedir(dirp);
 824                 } else {
 825                         send_handle(id, handle);
 826                         status = SSH2_FX_OK;
 827                 }
 828
 829         }
 830         if (status != SSH2_FX_OK)
 831                 send_status(id, status);
 832         xfree(path);
 833 }
 834
 835 static void
 836 process_readdir(void)
 837 {
 838         DIR *dirp;
 839         struct dirent *dp;
 840         char *path;
 841         int handle;
 842         u_int32_t id;
 843
 844         id = get_int();
 845         handle = get_handle();
 846         debug("request %u: readdir \"%s\" (handle %d)", id,
 847             handle_to_name(handle), handle);
 848         dirp = handle_to_dir(handle);
 849         path = handle_to_name(handle);
 850         if (dirp == NULL || path == NULL) {
 851                 send_status(id, SSH2_FX_FAILURE);
 852         } else {
 853                 struct stat st;
 854                 char pathname[MAXPATHLEN];
 855                 Stat *stats;
 856                 int nstats = 10, count = 0, i;
 857
 858                 stats = xcalloc(nstats, sizeof(Stat));
 859                 while ((dp = readdir(dirp)) != NULL) {
 860                         if (count >= nstats) {
 861                                 nstats *= 2;
 862                                 stats = xrealloc(stats, nstats, sizeof(Stat));
 863                         }
 864 /* XXX OVERFLOW ? */
 865                         snprintf(pathname, sizeof pathname, "%s%s%s", path,
 866                             strcmp(path, "/") ? "/" : "", dp->d_name);
 867                         if (lstat(pathname, &st) < 0)
 868                                 continue;
 869                         stat_to_attrib(&st, &(stats[count].attrib));
 870                         stats[count].name = xstrdup(dp->d_name);
 871                         stats[count].long_name = ls_file(dp->d_name, &st, 0);
 872                         count++;
 873                         /* send up to 100 entries in one message */
 874                         /* XXX check packet size instead */
 875                         if (count == 100)
 876                                 break;
 877                 }
 878                 if (count > 0) {
 879                         send_names(id, count, stats);
 880                         for (i = 0; i < count; i++) {
 881                                 xfree(stats[i].name);
 882                                 xfree(stats[i].long_name);
 883                         }
 884                 } else {
 885                         send_status(id, SSH2_FX_EOF);
 886                 }
 887                 xfree(stats);
 888         }
 889 }
 890
 891 static void
 892 process_remove(void)
 893 {
 894         char *name;
 895         u_int32_t id;
 896         int status = SSH2_FX_FAILURE;
 897         int ret;
 898
 899         id = get_int();
 900         name = get_string(NULL);
 901         debug3("request %u: remove", id);
 902         logit("remove name \"%s\"", name);
 903         ret = unlink(name);
 904         status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
 905         send_status(id, status);
 906         xfree(name);
 907 }
 908
 909 static void
 910 process_mkdir(void)
 911 {
 912         Attrib *a;
 913         u_int32_t id;
 914         char *name;
 915         int ret, mode, status = SSH2_FX_FAILURE;
 916
 917         id = get_int();
 918         name = get_string(NULL);
 919         a = get_attrib();
 920         mode = (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) ?
 921             a->perm & 0777 : 0777;
 922         debug3("request %u: mkdir", id);
 923         logit("mkdir name \"%s\" mode 0%o", name, mode);
 924         ret = mkdir(name, mode);
 925         status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
 926         send_status(id, status);
 927         xfree(name);
 928 }
 929
 930 static void
 931 process_rmdir(void)
 932 {
 933         u_int32_t id;
 934         char *name;
 935         int ret, status;
 936
 937         id = get_int();
 938         name = get_string(NULL);
 939         debug3("request %u: rmdir", id);
 940         logit("rmdir name \"%s\"", name);
 941         ret = rmdir(name);
 942         status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
 943         send_status(id, status);
 944         xfree(name);
 945 }
 946
 947 static void
 948 process_realpath(void)
 949 {
 950         char resolvedname[MAXPATHLEN];
 951         u_int32_t id;
 952         char *path;
 953
 954         id = get_int();
 955         path = get_string(NULL);
 956         if (path[0] == '\0') {
 957                 xfree(path);
 958                 path = xstrdup(".");
 959         }
 960         debug3("request %u: realpath", id);
 961         verbose("realpath \"%s\"", path);
 962         if (realpath(path, resolvedname) == NULL) {
 963                 send_status(id, errno_to_portable(errno));
 964         } else {
 965                 Stat s;
 966                 attrib_clear(&s.attrib);
 967                 s.name = s.long_name = resolvedname;
 968                 send_names(id, 1, &s);
 969         }
 970         xfree(path);
 971 }
 972
 973 static void
 974 process_rename(void)
 975 {
 976         u_int32_t id;
 977         char *oldpath, *newpath;
 978         int status;
 979         struct stat sb;
 980
 981         id = get_int();
 982         oldpath = get_string(NULL);
 983         newpath = get_string(NULL);
 984         debug3("request %u: rename", id);
 985         logit("rename old \"%s\" new \"%s\"", oldpath, newpath);
 986         status = SSH2_FX_FAILURE;
 987         if (lstat(oldpath, &sb) == -1)
 988                 status = errno_to_portable(errno);
 989         else if (S_ISREG(sb.st_mode)) {
 990                 /* Race-free rename of regular files */
 991                 if (link(oldpath, newpath) == -1) {
 992                         if (errno == EOPNOTSUPP
 993 #ifdef LINK_OPNOTSUPP_ERRNO
 994                             || errno == LINK_OPNOTSUPP_ERRNO
 995 #endif
 996                             ) {
 997                                 struct stat st;
 998
 999                                 /*
1000                                  * fs doesn't support links, so fall back to
1001                                  * stat+rename.  This is racy.
1002                                  */
1003                                 if (stat(newpath, &st) == -1) {
1004                                         if (rename(oldpath, newpath) == -1)
1005                                                 status =
1006                                                     errno_to_portable(errno);
1007                                         else
1008                                                 status = SSH2_FX_OK;
1009                                 }
1010                         } else {
1011                                 status = errno_to_portable(errno);
1012                         }
1013                 } else if (unlink(oldpath) == -1) {
1014                         status = errno_to_portable(errno);
1015                         /* clean spare link */
1016                         unlink(newpath);
1017                 } else
1018                         status = SSH2_FX_OK;
1019         } else if (stat(newpath, &sb) == -1) {
1020                 if (rename(oldpath, newpath) == -1)
1021                         status = errno_to_portable(errno);
1022                 else
1023                         status = SSH2_FX_OK;
1024         }
1025         send_status(id, status);
1026         xfree(oldpath);
1027         xfree(newpath);
1028 }
1029
1030 static void
1031 process_readlink(void)
1032 {
1033         u_int32_t id;
1034         int len;
1035         char buf[MAXPATHLEN];
1036         char *path;
1037
1038         id = get_int();
1039         path = get_string(NULL);
1040         debug3("request %u: readlink", id);
1041         verbose("readlink \"%s\"", path);
1042         if ((len = readlink(path, buf, sizeof(buf) - 1)) == -1)
1043                 send_status(id, errno_to_portable(errno));
1044         else {
1045                 Stat s;
1046
1047                 buf[len] = '\0';
1048                 attrib_clear(&s.attrib);
1049                 s.name = s.long_name = buf;
1050                 send_names(id, 1, &s);
1051         }
1052         xfree(path);
1053 }
1054
1055 static void
1056 process_symlink(void)
1057 {
1058         u_int32_t id;
1059         char *oldpath, *newpath;
1060         int ret, status;
1061
1062         id = get_int();
1063         oldpath = get_string(NULL);
1064         newpath = get_string(NULL);
1065         debug3("request %u: symlink", id);
1066         logit("symlink old \"%s\" new \"%s\"", oldpath, newpath);
1067         /* this will fail if 'newpath' exists */
1068         ret = symlink(oldpath, newpath);
1069         status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
1070         send_status(id, status);
1071         xfree(oldpath);
1072         xfree(newpath);
1073 }
1074
1075 static void
1076 process_extended(void)
1077 {
1078         u_int32_t id;
1079         char *request;
1080
1081         id = get_int();
1082         request = get_string(NULL);
1083         send_status(id, SSH2_FX_OP_UNSUPPORTED);                /* MUST */
1084         xfree(request);
1085 }
1086
1087 /* stolen from ssh-agent */
1088
1089 static void
1090 process(void)
1091 {
1092         u_int msg_len;
1093         u_int buf_len;
1094         u_int consumed;
1095         u_int type;
1096         u_char *cp;
1097
1098         buf_len = buffer_len(&iqueue);
1099         if (buf_len < 5)
1100                 return;         /* Incomplete message. */
1101         cp = buffer_ptr(&iqueue);
1102         msg_len = get_u32(cp);
1103         if (msg_len > SFTP_MAX_MSG_LENGTH) {
1104                 error("bad message from %s local user %s",
1105                     client_addr, pw->pw_name);
1106                 cleanup_exit(11);
1107         }
1108         if (buf_len < msg_len + 4)
1109                 return;
1110         buffer_consume(&iqueue, 4);
1111         buf_len -= 4;
1112         type = buffer_get_char(&iqueue);
1113         switch (type) {
1114         case SSH2_FXP_INIT:
1115                 process_init();
1116                 break;
1117         case SSH2_FXP_OPEN:
1118                 process_open();
1119                 break;
1120         case SSH2_FXP_CLOSE:
1121                 process_close();
1122                 break;
1123         case SSH2_FXP_READ:
1124                 process_read();
1125                 break;
1126         case SSH2_FXP_WRITE:
1127                 process_write();
1128                 break;
1129         case SSH2_FXP_LSTAT:
1130                 process_lstat();
1131                 break;
1132         case SSH2_FXP_FSTAT:
1133                 process_fstat();
1134                 break;
1135         case SSH2_FXP_SETSTAT:
1136                 process_setstat();
1137                 break;
1138         case SSH2_FXP_FSETSTAT:
1139                 process_fsetstat();
1140                 break;
1141         case SSH2_FXP_OPENDIR:
1142                 process_opendir();
1143                 break;
1144         case SSH2_FXP_READDIR:
1145                 process_readdir();
1146                 break;
1147         case SSH2_FXP_REMOVE:
1148                 process_remove();
1149                 break;
1150         case SSH2_FXP_MKDIR:
1151                 process_mkdir();
1152                 break;
1153         case SSH2_FXP_RMDIR:
1154                 process_rmdir();
1155                 break;
1156         case SSH2_FXP_REALPATH:
1157                 process_realpath();
1158                 break;
1159         case SSH2_FXP_STAT:
1160                 process_stat();
1161                 break;
1162         case SSH2_FXP_RENAME:
1163                 process_rename();
1164                 break;
1165         case SSH2_FXP_READLINK:
1166                 process_readlink();
1167                 break;
1168         case SSH2_FXP_SYMLINK:
1169                 process_symlink();
1170                 break;
1171         case SSH2_FXP_EXTENDED:
1172                 process_extended();
1173                 break;
1174         default:
1175                 error("Unknown message %d", type);
1176                 break;
1177         }
1178         /* discard the remaining bytes from the current packet */
1179         if (buf_len < buffer_len(&iqueue))
1180                 fatal("iqueue grew unexpectedly");
1181         consumed = buf_len - buffer_len(&iqueue);
1182         if (msg_len < consumed)
1183                 fatal("msg_len %d < consumed %d", msg_len, consumed);
1184         if (msg_len > consumed)
1185                 buffer_consume(&iqueue, msg_len - consumed);
1186 }
1187
1188 /* Cleanup handler that logs active handles upon normal exit */
1189 void
1190 cleanup_exit(int i)
1191 {
1192         if (pw != NULL && client_addr != NULL) {
1193                 handle_log_exit();
1194                 logit("session closed for local user %s from [%s]",
1195                     pw->pw_name, client_addr);
1196         }
1197         _exit(i);
1198 }
1199
1200 static void
1201 usage(void)
1202 {
1203         extern char *__progname;
1204
1205         fprintf(stderr,
1206             "usage: %s [-he] [-l log_level] [-f log_facility]\n", __progname);
1207         exit(1);
1208 }
1209
1210 int
1211 main(int argc, char **argv)
1212 {
1213         fd_set *rset, *wset;
1214         int in, out, max, ch, skipargs = 0, log_stderr = 0;
1215         ssize_t len, olen, set_size;
1216         SyslogFacility log_facility = SYSLOG_FACILITY_AUTH;
1217         char *cp, buf[4*4096];
1218
1219         extern char *optarg;
1220         extern char *__progname;
1221
1222         /* Ensure that fds 0, 1 and 2 are open or directed to /dev/null */
1223         sanitise_stdfd();
1224
1225         __progname = ssh_get_progname(argv[0]);
1226         log_init(__progname, log_level, log_facility, log_stderr);
1227
1228         while (!skipargs && (ch = getopt(argc, argv, "C:f:l:che")) != -1) {
1229                 switch (ch) {
1230                 case 'c':
1231                         /*
1232                          * Ignore all arguments if we are invoked as a
1233                          * shell using "sftp-server -c command"
1234                          */
1235                         skipargs = 1;
1236                         break;
1237                 case 'e':
1238                         log_stderr = 1;
1239                         break;
1240                 case 'l':
1241                         log_level = log_level_number(optarg);
1242                         if (log_level == SYSLOG_LEVEL_NOT_SET)
1243                                 error("Invalid log level \"%s\"", optarg);
1244                         break;
1245                 case 'f':
1246                         log_facility = log_facility_number(optarg);
1247                         if (log_level == SYSLOG_FACILITY_NOT_SET)
1248                                 error("Invalid log facility \"%s\"", optarg);
1249                         break;
1250                 case 'h':
1251                 default:
1252                         usage();
1253                 }
1254         }
1255
1256         log_init(__progname, log_level, log_facility, log_stderr);
1257
1258         if ((cp = getenv("SSH_CONNECTION")) != NULL) {
1259                 client_addr = xstrdup(cp);
1260                 if ((cp = strchr(client_addr, ' ')) == NULL)
1261                         fatal("Malformed SSH_CONNECTION variable: \"%s\"",
1262                             getenv("SSH_CONNECTION"));
1263                 *cp = '\0';
1264         } else
1265                 client_addr = xstrdup("UNKNOWN");
1266
1267         if ((pw = getpwuid(getuid())) == NULL)
1268                 fatal("No user found for uid %lu", (u_long)getuid());
1269         pw = pwcopy(pw);
1270
1271         logit("session opened for local user %s from [%s]",
1272             pw->pw_name, client_addr);
1273
1274         handle_init();
1275
1276         in = dup(STDIN_FILENO);
1277         out = dup(STDOUT_FILENO);
1278
1279 #ifdef HAVE_CYGWIN
1280         setmode(in, O_BINARY);
1281         setmode(out, O_BINARY);
1282 #endif
1283
1284         max = 0;
1285         if (in > max)
1286                 max = in;
1287         if (out > max)
1288                 max = out;
1289
1290         buffer_init(&iqueue);
1291         buffer_init(&oqueue);
1292
1293         set_size = howmany(max + 1, NFDBITS) * sizeof(fd_mask);
1294         rset = (fd_set *)xmalloc(set_size);
1295         wset = (fd_set *)xmalloc(set_size);
1296
1297         for (;;) {
1298                 memset(rset, 0, set_size);
1299                 memset(wset, 0, set_size);
1300
1301                 /*
1302                  * Ensure that we can read a full buffer and handle
1303                  * the worst-case length packet it can generate,
1304                  * otherwise apply backpressure by stopping reads.
1305                  */
1306                 if (buffer_check_alloc(&iqueue, sizeof(buf)) &&
1307                     buffer_check_alloc(&oqueue, SFTP_MAX_MSG_LENGTH))
1308                         FD_SET(in, rset);
1309
1310                 olen = buffer_len(&oqueue);
1311                 if (olen > 0)
1312                         FD_SET(out, wset);
1313
1314                 if (select(max+1, rset, wset, NULL, NULL) < 0) {
1315                         if (errno == EINTR)
1316                                 continue;
1317                         error("select: %s", strerror(errno));
1318                         cleanup_exit(2);
1319                 }
1320
1321                 /* copy stdin to iqueue */
1322                 if (FD_ISSET(in, rset)) {
1323                         len = read(in, buf, sizeof buf);
1324                         if (len == 0) {
1325                                 debug("read eof");
1326                                 cleanup_exit(0);
1327                         } else if (len < 0) {
1328                                 error("read: %s", strerror(errno));
1329                                 cleanup_exit(1);
1330                         } else {
1331                                 buffer_append(&iqueue, buf, len);
1332                         }
1333                 }
1334                 /* send oqueue to stdout */
1335                 if (FD_ISSET(out, wset)) {
1336                         len = write(out, buffer_ptr(&oqueue), olen);
1337                         if (len < 0) {
1338                                 error("write: %s", strerror(errno));
1339                                 cleanup_exit(1);
1340                         } else {
1341                                 buffer_consume(&oqueue, len);
1342                         }
1343                 }
1344
1345                 /*
1346                  * Process requests from client if we can fit the results
1347                  * into the output buffer, otherwise stop processing input
1348                  * and let the output queue drain.
1349                  */
1350                 if (buffer_check_alloc(&oqueue, SFTP_MAX_MSG_LENGTH))
1351                         process();
1352         }
1353 }