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