blob: ade2663e7cbdc03c32d5c678d8c26858eb2c4963 [file] [log] [blame]
Damien Miller7b28dc52000-09-05 13:34:53 +11001/*
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.
Damien Miller7b28dc52000-09-05 13:34:53 +110012 *
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"
Kevin Steves8e743932001-02-05 13:24:35 +000025RCSID("$OpenBSD: sftp-server.c,v 1.18 2001/02/04 22:21:19 stevesk Exp $");
Damien Miller7b28dc52000-09-05 13:34:53 +110026
Damien Miller7b28dc52000-09-05 13:34:53 +110027#include "buffer.h"
28#include "bufaux.h"
29#include "getput.h"
Ben Lindstrom226cfa02001-01-22 05:34:40 +000030#include "log.h"
Damien Miller7b28dc52000-09-05 13:34:53 +110031#include "xmalloc.h"
32
Ben Lindstrom2f959b42001-01-11 06:20:23 +000033#include "sftp.h"
Damien Miller33804262001-02-04 23:20:18 +110034#include "sftp-common.h"
Damien Miller7b28dc52000-09-05 13:34:53 +110035
36/* helper */
Ben Lindstrom2f959b42001-01-11 06:20:23 +000037#define get_int64() buffer_get_int64(&iqueue);
Damien Miller7b28dc52000-09-05 13:34:53 +110038#define get_int() buffer_get_int(&iqueue);
39#define get_string(lenp) buffer_get_string(&iqueue, lenp);
Ben Lindstrom2f959b42001-01-11 06:20:23 +000040#define TRACE debug
Damien Miller7b28dc52000-09-05 13:34:53 +110041
Ben Lindstrom49a79c02000-11-17 03:47:20 +000042#ifdef HAVE___PROGNAME
43extern char *__progname;
44#else
45char *__progname;
46#endif
47
Damien Miller7b28dc52000-09-05 13:34:53 +110048/* input and output queue */
49Buffer iqueue;
50Buffer oqueue;
51
52/* portable attibutes, etc. */
53
Damien Miller7b28dc52000-09-05 13:34:53 +110054typedef struct Stat Stat;
55
Damien Miller33804262001-02-04 23:20:18 +110056struct Stat {
Damien Miller7b28dc52000-09-05 13:34:53 +110057 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:
Ben Lindstrom2f959b42001-01-11 06:20:23 +000068 ret = SSH2_FX_OK;
Damien Miller7b28dc52000-09-05 13:34:53 +110069 break;
70 case ENOENT:
71 case ENOTDIR:
72 case EBADF:
73 case ELOOP:
Ben Lindstrom2f959b42001-01-11 06:20:23 +000074 ret = SSH2_FX_NO_SUCH_FILE;
Damien Miller7b28dc52000-09-05 13:34:53 +110075 break;
76 case EPERM:
77 case EACCES:
78 case EFAULT:
Ben Lindstrom2f959b42001-01-11 06:20:23 +000079 ret = SSH2_FX_PERMISSION_DENIED;
Damien Miller7b28dc52000-09-05 13:34:53 +110080 break;
81 case ENAMETOOLONG:
82 case EINVAL:
Ben Lindstrom2f959b42001-01-11 06:20:23 +000083 ret = SSH2_FX_BAD_MESSAGE;
Damien Miller7b28dc52000-09-05 13:34:53 +110084 break;
85 default:
Ben Lindstrom2f959b42001-01-11 06:20:23 +000086 ret = SSH2_FX_FAILURE;
Damien Miller7b28dc52000-09-05 13:34:53 +110087 break;
88 }
89 return ret;
90}
91
92int
93flags_from_portable(int pflags)
94{
95 int flags = 0;
Ben Lindstrom2f959b42001-01-11 06:20:23 +000096 if (pflags & SSH2_FXF_READ &&
97 pflags & SSH2_FXF_WRITE) {
Damien Miller7b28dc52000-09-05 13:34:53 +110098 flags = O_RDWR;
Ben Lindstrom2f959b42001-01-11 06:20:23 +000099 } else if (pflags & SSH2_FXF_READ) {
Damien Miller7b28dc52000-09-05 13:34:53 +1100100 flags = O_RDONLY;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000101 } else if (pflags & SSH2_FXF_WRITE) {
Damien Miller7b28dc52000-09-05 13:34:53 +1100102 flags = O_WRONLY;
103 }
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000104 if (pflags & SSH2_FXF_CREAT)
Damien Miller7b28dc52000-09-05 13:34:53 +1100105 flags |= O_CREAT;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000106 if (pflags & SSH2_FXF_TRUNC)
Damien Miller7b28dc52000-09-05 13:34:53 +1100107 flags |= O_TRUNC;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000108 if (pflags & SSH2_FXF_EXCL)
Damien Miller7b28dc52000-09-05 13:34:53 +1100109 flags |= O_EXCL;
110 return flags;
111}
112
Damien Miller7b28dc52000-09-05 13:34:53 +1100113Attrib *
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{
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000162 return i >= 0 && i < sizeof(handles)/sizeof(Handle) &&
163 handles[i].use == type;
Damien Miller7b28dc52000-09-05 13:34:53 +1100164}
165
166int
167handle_to_string(int handle, char **stringp, int *hlenp)
168{
Damien Miller7b28dc52000-09-05 13:34:53 +1100169 if (stringp == NULL || hlenp == NULL)
170 return -1;
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000171 *stringp = xmalloc(sizeof(int32_t));
172 PUT_32BIT(*stringp, handle);
173 *hlenp = sizeof(int32_t);
Damien Miller7b28dc52000-09-05 13:34:53 +1100174 return 0;
175}
176
177int
Damien Millere4340be2000-09-16 13:29:08 +1100178handle_from_string(char *handle, u_int hlen)
Damien Miller7b28dc52000-09-05 13:34:53 +1100179{
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000180 int val;
181 if (hlen != sizeof(int32_t))
Damien Miller7b28dc52000-09-05 13:34:53 +1100182 return -1;
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000183 val = GET_32BIT(handle);
Damien Miller7b28dc52000-09-05 13:34:53 +1100184 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{
Kevin Stevesef4eea92001-02-05 12:42:17 +0000210 if (handle_is_ok(handle, HANDLE_FILE))
Damien Miller7b28dc52000-09-05 13:34:53 +1100211 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;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000235 int val = -1;
Damien Millere4340be2000-09-16 13:29:08 +1100236 u_int hlen;
Damien Miller7b28dc52000-09-05 13:34:53 +1100237 handle = get_string(&hlen);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000238 if (hlen < 256)
239 val = handle_from_string(handle, hlen);
Damien Miller7b28dc52000-09-05 13:34:53 +1100240 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);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000261 buffer_put_char(&msg, SSH2_FXP_STATUS);
Damien Miller7b28dc52000-09-05 13:34:53 +1100262 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);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000283 send_data_or_handle(SSH2_FXP_DATA, id, data, dlen);
Damien Miller7b28dc52000-09-05 13:34:53 +1100284}
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);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000293 send_data_or_handle(SSH2_FXP_HANDLE, id, string, hlen);
Damien Miller7b28dc52000-09-05 13:34:53 +1100294 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);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000303 buffer_put_char(&msg, SSH2_FXP_NAME);
Damien Miller7b28dc52000-09-05 13:34:53 +1100304 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);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000322 buffer_put_char(&msg, SSH2_FXP_ATTRS);
Damien Miller7b28dc52000-09-05 13:34:53 +1100323 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);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000339 buffer_put_char(&msg, SSH2_FXP_VERSION);
340 buffer_put_int(&msg, SSH2_FILEXFER_VERSION);
Damien Miller7b28dc52000-09-05 13:34:53 +1100341 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;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000351 int handle, fd, flags, mode, status = SSH2_FX_FAILURE;
Damien Miller7b28dc52000-09-05 13:34:53 +1100352
353 id = get_int();
354 name = get_string(NULL);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000355 pflags = get_int(); /* portable flags */
Damien Miller7b28dc52000-09-05 13:34:53 +1100356 a = get_attrib();
357 flags = flags_from_portable(pflags);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000358 mode = (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) ? a->perm : 0666;
Damien Miller7b28dc52000-09-05 13:34:53 +1100359 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);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000369 status = SSH2_FX_OK;
Damien Miller7b28dc52000-09-05 13:34:53 +1100370 }
371 }
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000372 if (status != SSH2_FX_OK)
Damien Miller7b28dc52000-09-05 13:34:53 +1100373 send_status(id, status);
374 xfree(name);
375}
376
377void
378process_close(void)
379{
380 u_int32_t id;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000381 int handle, ret, status = SSH2_FX_FAILURE;
Damien Miller7b28dc52000-09-05 13:34:53 +1100382
383 id = get_int();
384 handle = get_handle();
385 TRACE("close id %d handle %d", id, handle);
386 ret = handle_close(handle);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000387 status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
Damien Miller7b28dc52000-09-05 13:34:53 +1100388 send_status(id, status);
389}
390
391void
392process_read(void)
393{
394 char buf[64*1024];
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000395 u_int32_t id, len;
396 int handle, fd, ret, status = SSH2_FX_FAILURE;
Damien Miller7b28dc52000-09-05 13:34:53 +1100397 u_int64_t off;
398
399 id = get_int();
400 handle = get_handle();
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000401 off = get_int64();
Damien Miller7b28dc52000-09-05 13:34:53 +1100402 len = get_int();
403
Damien Millerf056e232000-09-23 14:58:32 +1100404 TRACE("read id %d handle %d off %lld len %d", id, handle, off, len);
Damien Miller7b28dc52000-09-05 13:34:53 +1100405 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) {
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000419 status = SSH2_FX_EOF;
Damien Miller7b28dc52000-09-05 13:34:53 +1100420 } else {
421 send_data(id, buf, ret);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000422 status = SSH2_FX_OK;
Damien Miller7b28dc52000-09-05 13:34:53 +1100423 }
424 }
425 }
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000426 if (status != SSH2_FX_OK)
Damien Miller7b28dc52000-09-05 13:34:53 +1100427 send_status(id, status);
428}
429
430void
431process_write(void)
432{
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000433 u_int32_t id;
Damien Miller7b28dc52000-09-05 13:34:53 +1100434 u_int64_t off;
Damien Millere4340be2000-09-16 13:29:08 +1100435 u_int len;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000436 int handle, fd, ret, status = SSH2_FX_FAILURE;
Damien Miller7b28dc52000-09-05 13:34:53 +1100437 char *data;
438
439 id = get_int();
440 handle = get_handle();
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000441 off = get_int64();
Damien Miller7b28dc52000-09-05 13:34:53 +1100442 data = get_string(&len);
443
Damien Millerf056e232000-09-23 14:58:32 +1100444 TRACE("write id %d handle %d off %lld len %d", id, handle, off, len);
Damien Miller7b28dc52000-09-05 13:34:53 +1100445 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) {
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000457 status = SSH2_FX_OK;
Damien Miller7b28dc52000-09-05 13:34:53 +1100458 } 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{
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000470 Attrib a;
Damien Miller7b28dc52000-09-05 13:34:53 +1100471 struct stat st;
472 u_int32_t id;
473 char *name;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000474 int ret, status = SSH2_FX_FAILURE;
Damien Miller7b28dc52000-09-05 13:34:53 +1100475
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 {
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000483 stat_to_attrib(&st, &a);
484 send_attrib(id, &a);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000485 status = SSH2_FX_OK;
Damien Miller7b28dc52000-09-05 13:34:53 +1100486 }
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000487 if (status != SSH2_FX_OK)
Damien Miller7b28dc52000-09-05 13:34:53 +1100488 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{
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000507 Attrib a;
Damien Miller7b28dc52000-09-05 13:34:53 +1100508 struct stat st;
509 u_int32_t id;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000510 int fd, ret, handle, status = SSH2_FX_FAILURE;
Damien Miller7b28dc52000-09-05 13:34:53 +1100511
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 {
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000521 stat_to_attrib(&st, &a);
522 send_attrib(id, &a);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000523 status = SSH2_FX_OK;
Damien Miller7b28dc52000-09-05 13:34:53 +1100524 }
525 }
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000526 if (status != SSH2_FX_OK)
Damien Miller7b28dc52000-09-05 13:34:53 +1100527 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;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000548 int status = SSH2_FX_OK;
Damien Miller7b28dc52000-09-05 13:34:53 +1100549
550 id = get_int();
551 name = get_string(NULL);
552 a = get_attrib();
553 TRACE("setstat id %d name %s", id, name);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000554 if (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) {
Damien Miller7b28dc52000-09-05 13:34:53 +1100555 ret = chmod(name, a->perm & 0777);
556 if (ret == -1)
557 status = errno_to_portable(errno);
558 }
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000559 if (a->flags & SSH2_FILEXFER_ATTR_ACMODTIME) {
Damien Miller7b28dc52000-09-05 13:34:53 +1100560 ret = utimes(name, attrib_to_tv(a));
561 if (ret == -1)
562 status = errno_to_portable(errno);
563 }
Kevin Steves8e743932001-02-05 13:24:35 +0000564 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 }
Damien Miller7b28dc52000-09-05 13:34:53 +1100569 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;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000579 int status = SSH2_FX_OK;
Damien Millere4340be2000-09-16 13:29:08 +1100580 char *name;
Kevin Stevesf7ffab32001-01-24 20:11:06 +0000581
Damien Miller7b28dc52000-09-05 13:34:53 +1100582 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);
Kevin Stevesf7ffab32001-01-24 20:11:06 +0000588 if (fd < 0 || name == NULL) {
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000589 status = SSH2_FX_FAILURE;
Damien Miller7b28dc52000-09-05 13:34:53 +1100590 } else {
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000591 if (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) {
Ben Lindstrom200e3c92001-01-15 01:56:46 +0000592#ifdef HAVE_FCHMOD
Damien Miller7b28dc52000-09-05 13:34:53 +1100593 ret = fchmod(fd, a->perm & 0777);
Ben Lindstrom200e3c92001-01-15 01:56:46 +0000594#else
Kevin Stevesb6b37ba2001-01-24 20:01:44 +0000595 ret = chmod(name, a->perm & 0777);
Ben Lindstrom200e3c92001-01-15 01:56:46 +0000596#endif
Damien Miller7b28dc52000-09-05 13:34:53 +1100597 if (ret == -1)
598 status = errno_to_portable(errno);
599 }
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000600 if (a->flags & SSH2_FILEXFER_ATTR_ACMODTIME) {
Damien Miller7b28dc52000-09-05 13:34:53 +1100601#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 }
Kevin Steves8e743932001-02-05 13:24:35 +0000609 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 }
Damien Miller7b28dc52000-09-05 13:34:53 +1100614 }
615 send_status(id, status);
616}
617
618void
619process_opendir(void)
620{
621 DIR *dirp = NULL;
622 char *path;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000623 int handle, status = SSH2_FX_FAILURE;
Damien Miller7b28dc52000-09-05 13:34:53 +1100624 u_int32_t id;
625
626 id = get_int();
627 path = get_string(NULL);
628 TRACE("opendir id %d path %s", id, path);
Kevin Stevesef4eea92001-02-05 12:42:17 +0000629 dirp = opendir(path);
Damien Miller7b28dc52000-09-05 13:34:53 +1100630 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);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000638 status = SSH2_FX_OK;
Damien Miller7b28dc52000-09-05 13:34:53 +1100639 }
Kevin Stevesef4eea92001-02-05 12:42:17 +0000640
Damien Miller7b28dc52000-09-05 13:34:53 +1100641 }
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000642 if (status != SSH2_FX_OK)
Damien Miller7b28dc52000-09-05 13:34:53 +1100643 send_status(id, status);
644 xfree(path);
645}
646
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000647/*
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000648 * drwxr-xr-x 5 markus markus 1024 Jan 13 18:39 .ssh
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000649 */
Damien Miller7b28dc52000-09-05 13:34:53 +1100650char *
651ls_file(char *name, struct stat *st)
652{
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000653 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';
Kevin Stevesb71eb582001-01-29 16:57:27 +0000681 snprintf(buf, sizeof buf, "%s %3d %-8.8s %-8.8s %8lld %s %s", mode,
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000682 st->st_nlink, user, group, (long long)st->st_size, tbuf, name);
Damien Miller7b28dc52000-09-05 13:34:53 +1100683 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) {
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000701 send_status(id, SSH2_FX_FAILURE);
Damien Miller7b28dc52000-09-05 13:34:53 +1100702 } else {
Damien Miller7b28dc52000-09-05 13:34:53 +1100703 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;
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000718 stat_to_attrib(&st, &(stats[count].attrib));
Damien Miller7b28dc52000-09-05 13:34:53 +1100719 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 */
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000723 /* XXX check packet size instead */
Damien Miller7b28dc52000-09-05 13:34:53 +1100724 if (count == 100)
725 break;
726 }
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000727 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);
Damien Miller7b28dc52000-09-05 13:34:53 +1100735 }
736 xfree(stats);
737 }
738}
739
740void
741process_remove(void)
742{
743 char *name;
744 u_int32_t id;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000745 int status = SSH2_FX_FAILURE;
Damien Miller7b28dc52000-09-05 13:34:53 +1100746 int ret;
747
748 id = get_int();
749 name = get_string(NULL);
750 TRACE("remove id %d name %s", id, name);
Kevin Stevesa074feb2000-12-21 22:33:45 +0000751 ret = unlink(name);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000752 status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
Damien Miller7b28dc52000-09-05 13:34:53 +1100753 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;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000763 int ret, mode, status = SSH2_FX_FAILURE;
Damien Miller7b28dc52000-09-05 13:34:53 +1100764
765 id = get_int();
766 name = get_string(NULL);
767 a = get_attrib();
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000768 mode = (a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) ?
769 a->perm & 0777 : 0777;
Damien Miller7b28dc52000-09-05 13:34:53 +1100770 TRACE("mkdir id %d name %s mode 0%o", id, name, mode);
771 ret = mkdir(name, mode);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000772 status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
Damien Miller7b28dc52000-09-05 13:34:53 +1100773 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);
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000788 status = (ret == -1) ? errno_to_portable(errno) : SSH2_FX_OK;
Damien Miller7b28dc52000-09-05 13:34:53 +1100789 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);
Ben Lindstromfa1b3d02000-12-10 01:55:37 +0000802 if (path[0] == '\0') {
803 xfree(path);
804 path = xstrdup(".");
805 }
Damien Miller7b28dc52000-09-05 13:34:53 +1100806 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;
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000822 struct stat st;
Damien Miller7b28dc52000-09-05 13:34:53 +1100823 char *oldpath, *newpath;
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000824 int ret, status = SSH2_FX_FAILURE;
Damien Miller7b28dc52000-09-05 13:34:53 +1100825
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);
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000830 /* 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 }
Damien Miller7b28dc52000-09-05 13:34:53 +1100835 send_status(id, status);
836 xfree(oldpath);
837 xfree(newpath);
838}
839
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000840void
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}
Damien Miller7b28dc52000-09-05 13:34:53 +1100851
852/* stolen from ssh-agent */
853
854void
855process(void)
856{
Ben Lindstrom46c16222000-12-22 01:43:59 +0000857 u_int msg_len;
858 u_int type;
859 u_char *cp;
Damien Miller7b28dc52000-09-05 13:34:53 +1100860
861 if (buffer_len(&iqueue) < 5)
862 return; /* Incomplete message. */
Ben Lindstrom46c16222000-12-22 01:43:59 +0000863 cp = (u_char *) buffer_ptr(&iqueue);
Damien Miller7b28dc52000-09-05 13:34:53 +1100864 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) {
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000874 case SSH2_FXP_INIT:
Damien Miller7b28dc52000-09-05 13:34:53 +1100875 process_init();
876 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000877 case SSH2_FXP_OPEN:
Damien Miller7b28dc52000-09-05 13:34:53 +1100878 process_open();
879 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000880 case SSH2_FXP_CLOSE:
Damien Miller7b28dc52000-09-05 13:34:53 +1100881 process_close();
882 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000883 case SSH2_FXP_READ:
Damien Miller7b28dc52000-09-05 13:34:53 +1100884 process_read();
885 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000886 case SSH2_FXP_WRITE:
Damien Miller7b28dc52000-09-05 13:34:53 +1100887 process_write();
888 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000889 case SSH2_FXP_LSTAT:
Damien Miller7b28dc52000-09-05 13:34:53 +1100890 process_lstat();
891 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000892 case SSH2_FXP_FSTAT:
Damien Miller7b28dc52000-09-05 13:34:53 +1100893 process_fstat();
894 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000895 case SSH2_FXP_SETSTAT:
Damien Miller7b28dc52000-09-05 13:34:53 +1100896 process_setstat();
897 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000898 case SSH2_FXP_FSETSTAT:
Damien Miller7b28dc52000-09-05 13:34:53 +1100899 process_fsetstat();
900 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000901 case SSH2_FXP_OPENDIR:
Damien Miller7b28dc52000-09-05 13:34:53 +1100902 process_opendir();
903 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000904 case SSH2_FXP_READDIR:
Damien Miller7b28dc52000-09-05 13:34:53 +1100905 process_readdir();
906 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000907 case SSH2_FXP_REMOVE:
Damien Miller7b28dc52000-09-05 13:34:53 +1100908 process_remove();
909 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000910 case SSH2_FXP_MKDIR:
Damien Miller7b28dc52000-09-05 13:34:53 +1100911 process_mkdir();
912 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000913 case SSH2_FXP_RMDIR:
Damien Miller7b28dc52000-09-05 13:34:53 +1100914 process_rmdir();
915 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000916 case SSH2_FXP_REALPATH:
Damien Miller7b28dc52000-09-05 13:34:53 +1100917 process_realpath();
918 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000919 case SSH2_FXP_STAT:
Damien Miller7b28dc52000-09-05 13:34:53 +1100920 process_stat();
921 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000922 case SSH2_FXP_RENAME:
Damien Miller7b28dc52000-09-05 13:34:53 +1100923 process_rename();
924 break;
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000925 case SSH2_FXP_EXTENDED:
926 process_extended();
927 break;
Damien Miller7b28dc52000-09-05 13:34:53 +1100928 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;
Damien Millere4340be2000-09-16 13:29:08 +1100939 ssize_t len, olen;
Damien Miller7b28dc52000-09-05 13:34:53 +1100940
Ben Lindstrom49a79c02000-11-17 03:47:20 +0000941 __progname = get_progname(av[0]);
Damien Miller7b28dc52000-09-05 13:34:53 +1100942 handle_init();
943
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000944#ifdef DEBUG_SFTP_SERVER
Kevin Stevesef4eea92001-02-05 12:42:17 +0000945 log_init("sftp-server", SYSLOG_LEVEL_DEBUG1, SYSLOG_FACILITY_AUTH, 0);
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000946#endif
Ben Lindstrom2f959b42001-01-11 06:20:23 +0000947
Damien Miller7b28dc52000-09-05 13:34:53 +1100948 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}