Wildcards in `ls'. I think that completes `psftp-multi'. Woo!
[u/mdw/putty] / psftp.c
CommitLineData
4c7f0d61 1/*
f4ff9455 2 * psftp.c: (platform-independent) front end for PSFTP.
4c7f0d61 3 */
4
5#include <stdio.h>
6#include <stdlib.h>
f9e162aa 7#include <stdarg.h>
4c7f0d61 8#include <assert.h>
d92624dc 9#include <limits.h>
4c7f0d61 10
4a8fc3c4 11#define PUTTY_DO_GLOBALS
12#include "putty.h"
d6cc41e6 13#include "psftp.h"
4a8fc3c4 14#include "storage.h"
15#include "ssh.h"
4c7f0d61 16#include "sftp.h"
17#include "int64.h"
18
5471d09a 19/*
20 * Since SFTP is a request-response oriented protocol, it requires
21 * no buffer management: when we send data, we stop and wait for an
22 * acknowledgement _anyway_, and so we can't possibly overfill our
23 * send buffer.
24 */
25
fa3db767 26static int psftp_connect(char *userhost, char *user, int portnumber);
07534184 27static int do_sftp_init(void);
679539d7 28void do_sftp_cleanup();
fa3db767 29
4c7f0d61 30/* ----------------------------------------------------------------------
4c7f0d61 31 * sftp client state.
32 */
33
34char *pwd, *homedir;
6b78788a 35static Backend *back;
36static void *backhandle;
3ea863a3 37static Config cfg;
4c7f0d61 38
39/* ----------------------------------------------------------------------
40 * Higher-level helper functions used in commands.
41 */
42
43/*
93e86a8b 44 * Determine whether a string is entirely composed of dots.
45 */
46static int is_dots(char *str)
47{
48 return str[strspn(str, ".")] == '\0';
49}
50
51/*
f9e162aa 52 * Attempt to canonify a pathname starting from the pwd. If
53 * canonification fails, at least fall back to returning a _valid_
54 * pathname (though it may be ugly, eg /home/simon/../foobar).
4c7f0d61 55 */
32874aea 56char *canonify(char *name)
57{
f9e162aa 58 char *fullname, *canonname;
1bc24185 59 struct sftp_packet *pktin;
60 struct sftp_request *req, *rreq;
4a8fc3c4 61
f9e162aa 62 if (name[0] == '/') {
63 fullname = dupstr(name);
64 } else {
4a8fc3c4 65 char *slash;
32874aea 66 if (pwd[strlen(pwd) - 1] == '/')
4a8fc3c4 67 slash = "";
68 else
69 slash = "/";
70 fullname = dupcat(pwd, slash, name, NULL);
f9e162aa 71 }
4a8fc3c4 72
1bc24185 73 sftp_register(req = fxp_realpath_send(fullname));
74 rreq = sftp_find_request(pktin = sftp_recv());
75 assert(rreq == req);
7b7de4f4 76 canonname = fxp_realpath_recv(pktin, rreq);
4a8fc3c4 77
f9e162aa 78 if (canonname) {
79 sfree(fullname);
80 return canonname;
50d7e054 81 } else {
32874aea 82 /*
83 * Attempt number 2. Some FXP_REALPATH implementations
84 * (glibc-based ones, in particular) require the _whole_
85 * path to point to something that exists, whereas others
86 * (BSD-based) only require all but the last component to
87 * exist. So if the first call failed, we should strip off
88 * everything from the last slash onwards and try again,
89 * then put the final component back on.
90 *
91 * Special cases:
92 *
93 * - if the last component is "/." or "/..", then we don't
94 * bother trying this because there's no way it can work.
95 *
96 * - if the thing actually ends with a "/", we remove it
97 * before we start. Except if the string is "/" itself
98 * (although I can't see why we'd have got here if so,
99 * because surely "/" would have worked the first
100 * time?), in which case we don't bother.
101 *
102 * - if there's no slash in the string at all, give up in
103 * confusion (we expect at least one because of the way
104 * we constructed the string).
105 */
106
107 int i;
108 char *returnname;
109
110 i = strlen(fullname);
111 if (i > 2 && fullname[i - 1] == '/')
112 fullname[--i] = '\0'; /* strip trailing / unless at pos 0 */
113 while (i > 0 && fullname[--i] != '/');
114
115 /*
116 * Give up on special cases.
117 */
118 if (fullname[i] != '/' || /* no slash at all */
119 !strcmp(fullname + i, "/.") || /* ends in /. */
120 !strcmp(fullname + i, "/..") || /* ends in /.. */
121 !strcmp(fullname, "/")) {
122 return fullname;
123 }
124
125 /*
126 * Now i points at the slash. Deal with the final special
127 * case i==0 (ie the whole path was "/nonexistentfile").
128 */
129 fullname[i] = '\0'; /* separate the string */
130 if (i == 0) {
1bc24185 131 sftp_register(req = fxp_realpath_send("/"));
32874aea 132 } else {
1bc24185 133 sftp_register(req = fxp_realpath_send(fullname));
32874aea 134 }
1bc24185 135 rreq = sftp_find_request(pktin = sftp_recv());
136 assert(rreq == req);
7b7de4f4 137 canonname = fxp_realpath_recv(pktin, rreq);
32874aea 138
139 if (!canonname)
140 return fullname; /* even that failed; give up */
141
142 /*
143 * We have a canonical name for all but the last path
144 * component. Concatenate the last component and return.
145 */
146 returnname = dupcat(canonname,
147 canonname[strlen(canonname) - 1] ==
148 '/' ? "" : "/", fullname + i + 1, NULL);
149 sfree(fullname);
150 sfree(canonname);
151 return returnname;
50d7e054 152 }
4c7f0d61 153}
154
dcf8495c 155/*
156 * Return a pointer to the portion of str that comes after the last
157 * slash (or backslash or colon, if `local' is TRUE).
158 */
159static char *stripslashes(char *str, int local)
160{
161 char *p;
162
163 if (local) {
164 p = strchr(str, ':');
165 if (p) str = p+1;
166 }
167
168 p = strrchr(str, '/');
169 if (p) str = p+1;
170
171 if (local) {
172 p = strrchr(str, '\\');
173 if (p) str = p+1;
174 }
175
176 return str;
177}
178
4c7f0d61 179/*
93e86a8b 180 * qsort comparison routine for fxp_name structures. Sorts by real
181 * file name.
4c7f0d61 182 */
93e86a8b 183static int sftp_name_compare(const void *av, const void *bv)
32874aea 184{
7d2c1789 185 const struct fxp_name *const *a = (const struct fxp_name *const *) av;
186 const struct fxp_name *const *b = (const struct fxp_name *const *) bv;
187 return strcmp((*a)->filename, (*b)->filename);
4c7f0d61 188}
4c7f0d61 189
190/*
93e86a8b 191 * Likewise, but for a bare char *.
4f2b387f 192 */
93e86a8b 193static int bare_name_compare(const void *av, const void *bv)
4f2b387f 194{
93e86a8b 195 const char **a = (const char **) av;
196 const char **b = (const char **) bv;
197 return strcmp(*a, *b);
4f2b387f 198}
199
93e86a8b 200/* ----------------------------------------------------------------------
201 * The meat of the `get' and `put' commands.
4c7f0d61 202 */
9c77ddf6 203int sftp_get_file(char *fname, char *outfname, int recurse, int restart,
204 char *wildcard)
32874aea 205{
4c7f0d61 206 struct fxp_handle *fh;
1bc24185 207 struct sftp_packet *pktin;
208 struct sftp_request *req, *rreq;
c606c42d 209 struct fxp_xfer *xfer;
4c7f0d61 210 uint64 offset;
211 FILE *fp;
df49ff19 212 int ret;
4c7f0d61 213
93e86a8b 214 /*
215 * In recursive mode, see if we're dealing with a directory.
216 * (If we're not in recursive mode, we need not even check: the
217 * subsequent FXP_OPEN will return a usable error message.)
218 */
9c77ddf6 219 if (wildcard || recurse) {
93e86a8b 220 struct fxp_attrs attrs;
221 int result;
fa3db767 222
9c77ddf6 223 if (!wildcard) {
224 sftp_register(req = fxp_stat_send(fname));
225 rreq = sftp_find_request(pktin = sftp_recv());
226 assert(rreq == req);
227 result = fxp_stat_recv(pktin, rreq, &attrs);
228 } else
229 result = 0; /* placate optimisers */
230
231 if (wildcard ||
232 (result &&
233 (attrs.flags & SSH_FILEXFER_ATTR_PERMISSIONS) &&
234 (attrs.permissions & 0040000))) {
93e86a8b 235
236 struct fxp_handle *dirhandle;
237 int nnames, namesize;
238 struct fxp_name **ournames;
239 struct fxp_names *names;
240 int i;
241
242 /*
243 * First, attempt to create the destination directory,
9c77ddf6 244 * unless it already exists (or this is a wildcard
245 * run).
93e86a8b 246 */
9c77ddf6 247 if (!wildcard &&
248 file_type(outfname) != FILE_TYPE_DIRECTORY &&
93e86a8b 249 !create_directory(outfname)) {
250 printf("%s: Cannot create directory\n", outfname);
251 return 0;
252 }
4c7f0d61 253
93e86a8b 254 /*
255 * Now get the list of filenames in the remote
256 * directory.
257 */
258 sftp_register(req = fxp_opendir_send(fname));
259 rreq = sftp_find_request(pktin = sftp_recv());
260 assert(rreq == req);
261 dirhandle = fxp_opendir_recv(pktin, rreq);
262
263 if (!dirhandle) {
264 printf("%s: unable to open directory: %s\n",
265 fname, fxp_error());
266 return 0;
267 }
268 nnames = namesize = 0;
269 ournames = NULL;
270 while (1) {
271 int i;
272
273 sftp_register(req = fxp_readdir_send(dirhandle));
274 rreq = sftp_find_request(pktin = sftp_recv());
275 assert(rreq == req);
276 names = fxp_readdir_recv(pktin, rreq);
277
278 if (names == NULL) {
279 if (fxp_error_type() == SSH_FX_EOF)
280 break;
281 printf("%s: reading directory: %s\n", fname, fxp_error());
282 sfree(ournames);
283 return 0;
284 }
285 if (names->nnames == 0) {
286 fxp_free_names(names);
287 break;
288 }
289 if (nnames + names->nnames >= namesize) {
290 namesize += names->nnames + 128;
291 ournames = sresize(ournames, namesize, struct fxp_name *);
292 }
293 for (i = 0; i < names->nnames; i++)
9c77ddf6 294 if (!is_dots(names->names[i].filename) &&
295 (!wildcard || wc_match(wildcard,
296 names->names[i].filename)))
93e86a8b 297 ournames[nnames++] = fxp_dup_name(&names->names[i]);
298 fxp_free_names(names);
299 }
300 sftp_register(req = fxp_close_send(dirhandle));
301 rreq = sftp_find_request(pktin = sftp_recv());
302 assert(rreq == req);
303 fxp_close_recv(pktin, rreq);
304
305 /*
9c77ddf6 306 * A polite warning if nothing at all matched the
307 * wildcard.
308 */
309 if (wildcard && !nnames) {
310 printf("%s: nothing matched\n", wildcard);
311 }
312
313 /*
93e86a8b 314 * Sort the names into a clear order. This ought to
315 * make things more predictable when we're doing a
316 * reget of the same directory, just in case two
317 * readdirs on the same remote directory return a
318 * different order.
319 */
320 qsort(ournames, nnames, sizeof(*ournames), sftp_name_compare);
321
322 /*
323 * If we're in restart mode, find the last filename on
324 * this list that already exists. We may have to do a
325 * reget on _that_ file, but shouldn't have to do
326 * anything on the previous files.
327 *
328 * If none of them exists, of course, we start at 0.
329 */
330 i = 0;
331 while (i < nnames) {
332 char *nextoutfname;
333 int ret;
9c77ddf6 334 if (outfname)
335 nextoutfname = dir_file_cat(outfname,
336 ournames[i]->filename);
337 else
338 nextoutfname = dupstr(ournames[i]->filename);
93e86a8b 339 ret = (file_type(nextoutfname) == FILE_TYPE_NONEXISTENT);
340 sfree(nextoutfname);
341 if (ret)
342 break;
343 i++;
344 }
345 if (i > 0)
346 i--;
347
348 /*
349 * Now we're ready to recurse. Starting at ournames[i]
350 * and continuing on to the end of the list, we
351 * construct a new source and target file name, and
352 * call sftp_get_file again.
353 */
354 for (; i < nnames; i++) {
355 char *nextfname, *nextoutfname;
356 int ret;
357
358 nextfname = dupcat(fname, "/", ournames[i]->filename, NULL);
9c77ddf6 359 if (outfname)
360 nextoutfname = dir_file_cat(outfname,
361 ournames[i]->filename);
362 else
363 nextoutfname = dupstr(ournames[i]->filename);
364 ret = sftp_get_file(nextfname, nextoutfname,
365 recurse, restart, NULL);
93e86a8b 366 restart = FALSE; /* after first partial file, do full */
367 sfree(nextoutfname);
368 sfree(nextfname);
369 if (!ret) {
370 for (i = 0; i < nnames; i++) {
371 fxp_free_name(ournames[i]);
372 }
373 sfree(ournames);
374 return 0;
375 }
376 }
377
378 /*
379 * Done this recursion level. Free everything.
380 */
381 for (i = 0; i < nnames; i++) {
382 fxp_free_name(ournames[i]);
383 }
384 sfree(ournames);
385
386 return 1;
387 }
4c7f0d61 388 }
4c7f0d61 389
1bc24185 390 sftp_register(req = fxp_open_send(fname, SSH_FXF_READ));
391 rreq = sftp_find_request(pktin = sftp_recv());
392 assert(rreq == req);
7b7de4f4 393 fh = fxp_open_recv(pktin, rreq);
1bc24185 394
4c7f0d61 395 if (!fh) {
396 printf("%s: %s\n", fname, fxp_error());
4c7f0d61 397 return 0;
398 }
d92624dc 399
400 if (restart) {
401 fp = fopen(outfname, "rb+");
402 } else {
403 fp = fopen(outfname, "wb");
404 }
405
4c7f0d61 406 if (!fp) {
407 printf("local: unable to open %s\n", outfname);
1bc24185 408
409 sftp_register(req = fxp_close_send(fh));
410 rreq = sftp_find_request(pktin = sftp_recv());
411 assert(rreq == req);
7b7de4f4 412 fxp_close_recv(pktin, rreq);
1bc24185 413
4c7f0d61 414 return 0;
415 }
416
d92624dc 417 if (restart) {
418 long posn;
419 fseek(fp, 0L, SEEK_END);
420 posn = ftell(fp);
421 printf("reget: restarting at file position %ld\n", posn);
422 offset = uint64_make(0, posn);
423 } else {
424 offset = uint64_make(0, 0);
425 }
4c7f0d61 426
d92624dc 427 printf("remote:%s => local:%s\n", fname, outfname);
4c7f0d61 428
429 /*
430 * FIXME: we can use FXP_FSTAT here to get the file size, and
431 * thus put up a progress bar.
432 */
df49ff19 433 ret = 1;
c606c42d 434 xfer = xfer_download_init(fh, offset);
df0870fc 435 while (!xfer_done(xfer)) {
c606c42d 436 void *vbuf;
437 int ret, len;
4c7f0d61 438 int wpos, wlen;
439
c606c42d 440 xfer_download_queue(xfer);
441 pktin = sftp_recv();
442 ret = xfer_download_gotpkt(xfer, pktin);
443
444 if (ret < 0) {
445 printf("error while reading: %s\n", fxp_error());
446 ret = 0;
4c7f0d61 447 }
32874aea 448
c606c42d 449 while (xfer_download_data(xfer, &vbuf, &len)) {
450 unsigned char *buf = (unsigned char *)vbuf;
451
452 wpos = 0;
453 while (wpos < len) {
454 wlen = fwrite(buf + wpos, 1, len - wpos, fp);
455 if (wlen <= 0) {
456 printf("error while writing local file\n");
457 ret = 0;
458 xfer_set_error(xfer);
459 }
460 wpos += wlen;
461 }
462 if (wpos < len) { /* we had an error */
df49ff19 463 ret = 0;
c606c42d 464 xfer_set_error(xfer);
4c7f0d61 465 }
9ff4e23d 466
467 sfree(vbuf);
df49ff19 468 }
4c7f0d61 469 }
470
c606c42d 471 xfer_cleanup(xfer);
472
4c7f0d61 473 fclose(fp);
1bc24185 474
475 sftp_register(req = fxp_close_send(fh));
476 rreq = sftp_find_request(pktin = sftp_recv());
477 assert(rreq == req);
7b7de4f4 478 fxp_close_recv(pktin, rreq);
1bc24185 479
df49ff19 480 return ret;
4c7f0d61 481}
482
9c77ddf6 483int sftp_put_file(char *fname, char *outfname, int recurse, int restart,
484 char *wildcard)
32874aea 485{
4c7f0d61 486 struct fxp_handle *fh;
df0870fc 487 struct fxp_xfer *xfer;
1bc24185 488 struct sftp_packet *pktin;
489 struct sftp_request *req, *rreq;
4c7f0d61 490 uint64 offset;
491 FILE *fp;
df0870fc 492 int ret, err, eof;
4c7f0d61 493
93e86a8b 494 /*
495 * In recursive mode, see if we're dealing with a directory.
496 * (If we're not in recursive mode, we need not even check: the
497 * subsequent fopen will return an error message.)
498 */
9c77ddf6 499 if (wildcard || (recurse && file_type(fname) == FILE_TYPE_DIRECTORY)) {
93e86a8b 500 struct fxp_attrs attrs;
501 int result;
502 int nnames, namesize;
503 char *name, **ournames;
504 DirHandle *dh;
505 int i;
4c7f0d61 506
9c77ddf6 507 if (!wildcard) {
508 /*
509 * First, attempt to create the destination directory,
510 * unless it already exists.
511 */
512 sftp_register(req = fxp_stat_send(outfname));
93e86a8b 513 rreq = sftp_find_request(pktin = sftp_recv());
514 assert(rreq == req);
9c77ddf6 515 result = fxp_stat_recv(pktin, rreq, &attrs);
516 if (!result ||
517 !(attrs.flags & SSH_FILEXFER_ATTR_PERMISSIONS) ||
518 !(attrs.permissions & 0040000)) {
519 sftp_register(req = fxp_mkdir_send(outfname));
520 rreq = sftp_find_request(pktin = sftp_recv());
521 assert(rreq == req);
522 result = fxp_mkdir_recv(pktin, rreq);
93e86a8b 523
9c77ddf6 524 if (!result) {
525 printf("%s: create directory: %s\n",
526 outfname, fxp_error());
527 return 0;
528 }
93e86a8b 529 }
530 }
531
532 /*
533 * Now get the list of filenames in the local directory.
534 */
93e86a8b 535 nnames = namesize = 0;
536 ournames = NULL;
9c77ddf6 537 if (wildcard) {
538 WildcardMatcher *wcm;
539
540 wcm = begin_wildcard_matching(wildcard);
541 if (wcm) {
542 while ((name = wildcard_get_filename(wcm)) != NULL) {
543 if (nnames >= namesize) {
544 namesize += 128;
545 ournames = sresize(ournames, namesize, char *);
546 }
547 ournames[nnames++] = name;
548 }
549 finish_wildcard_matching(wcm);
93e86a8b 550 }
9c77ddf6 551 } else {
552 dh = open_directory(fname);
553 if (!dh) {
554 printf("%s: unable to open directory\n", fname);
555 return 0;
556 }
557 while ((name = read_filename(dh)) != NULL) {
558 if (nnames >= namesize) {
559 namesize += 128;
560 ournames = sresize(ournames, namesize, char *);
561 }
562 ournames[nnames++] = name;
563 }
564 close_directory(dh);
565 }
566
567 /*
568 * A polite warning if nothing at all matched the
569 * wildcard.
570 */
571 if (wildcard && !nnames) {
572 printf("%s: nothing matched\n", wildcard);
93e86a8b 573 }
93e86a8b 574
575 /*
576 * Sort the names into a clear order. This ought to make
577 * things more predictable when we're doing a reput of the
578 * same directory, just in case two readdirs on the same
579 * local directory return a different order.
580 */
581 qsort(ournames, nnames, sizeof(*ournames), bare_name_compare);
582
583 /*
584 * If we're in restart mode, find the last filename on this
585 * list that already exists. We may have to do a reput on
586 * _that_ file, but shouldn't have to do anything on the
587 * previous files.
588 *
589 * If none of them exists, of course, we start at 0.
590 */
591 i = 0;
592 while (i < nnames) {
593 char *nextoutfname;
594 nextoutfname = dupcat(outfname, "/", ournames[i], NULL);
595 sftp_register(req = fxp_stat_send(nextoutfname));
596 rreq = sftp_find_request(pktin = sftp_recv());
597 assert(rreq == req);
598 result = fxp_stat_recv(pktin, rreq, &attrs);
599 sfree(nextoutfname);
600 if (!result)
601 break;
602 i++;
603 }
604 if (i > 0)
605 i--;
606
607 /*
608 * Now we're ready to recurse. Starting at ournames[i]
609 * and continuing on to the end of the list, we
610 * construct a new source and target file name, and
611 * call sftp_put_file again.
612 */
613 for (; i < nnames; i++) {
614 char *nextfname, *nextoutfname;
615 int ret;
616
9c77ddf6 617 if (fname)
618 nextfname = dir_file_cat(fname, ournames[i]);
619 else
620 nextfname = dupstr(ournames[i]);
93e86a8b 621 nextoutfname = dupcat(outfname, "/", ournames[i], NULL);
9c77ddf6 622 ret = sftp_put_file(nextfname, nextoutfname,
623 recurse, restart, NULL);
93e86a8b 624 restart = FALSE; /* after first partial file, do full */
625 sfree(nextoutfname);
626 sfree(nextfname);
627 if (!ret) {
628 for (i = 0; i < nnames; i++) {
629 sfree(ournames[i]);
630 }
631 sfree(ournames);
632 return 0;
633 }
634 }
635
636 /*
637 * Done this recursion level. Free everything.
638 */
639 for (i = 0; i < nnames; i++) {
640 sfree(ournames[i]);
641 }
642 sfree(ournames);
643
644 return 1;
4c7f0d61 645 }
646
647 fp = fopen(fname, "rb");
648 if (!fp) {
649 printf("local: unable to open %s\n", fname);
4c7f0d61 650 return 0;
651 }
d92624dc 652 if (restart) {
1bc24185 653 sftp_register(req = fxp_open_send(outfname, SSH_FXF_WRITE));
d92624dc 654 } else {
1bc24185 655 sftp_register(req = fxp_open_send(outfname, SSH_FXF_WRITE |
656 SSH_FXF_CREAT | SSH_FXF_TRUNC));
d92624dc 657 }
1bc24185 658 rreq = sftp_find_request(pktin = sftp_recv());
659 assert(rreq == req);
7b7de4f4 660 fh = fxp_open_recv(pktin, rreq);
1bc24185 661
4c7f0d61 662 if (!fh) {
663 printf("%s: %s\n", outfname, fxp_error());
4c7f0d61 664 return 0;
665 }
666
d92624dc 667 if (restart) {
668 char decbuf[30];
669 struct fxp_attrs attrs;
1bc24185 670 int ret;
671
672 sftp_register(req = fxp_fstat_send(fh));
673 rreq = sftp_find_request(pktin = sftp_recv());
674 assert(rreq == req);
7b7de4f4 675 ret = fxp_fstat_recv(pktin, rreq, &attrs);
1bc24185 676
677 if (!ret) {
d92624dc 678 printf("read size of %s: %s\n", outfname, fxp_error());
d92624dc 679 return 0;
680 }
681 if (!(attrs.flags & SSH_FILEXFER_ATTR_SIZE)) {
682 printf("read size of %s: size was not given\n", outfname);
d92624dc 683 return 0;
684 }
685 offset = attrs.size;
686 uint64_decimal(offset, decbuf);
687 printf("reput: restarting at file position %s\n", decbuf);
688 if (uint64_compare(offset, uint64_make(0, LONG_MAX)) > 0) {
689 printf("reput: remote file is larger than we can deal with\n");
d92624dc 690 return 0;
691 }
692 if (fseek(fp, offset.lo, SEEK_SET) != 0)
693 fseek(fp, 0, SEEK_END); /* *shrug* */
694 } else {
695 offset = uint64_make(0, 0);
696 }
4c7f0d61 697
d92624dc 698 printf("local:%s => remote:%s\n", fname, outfname);
4c7f0d61 699
700 /*
701 * FIXME: we can use FXP_FSTAT here to get the file size, and
702 * thus put up a progress bar.
703 */
df49ff19 704 ret = 1;
df0870fc 705 xfer = xfer_upload_init(fh, offset);
706 err = eof = 0;
707 while ((!err && !eof) || !xfer_done(xfer)) {
4c7f0d61 708 char buffer[4096];
1bc24185 709 int len, ret;
4c7f0d61 710
df0870fc 711 while (xfer_upload_ready(xfer) && !err && !eof) {
712 len = fread(buffer, 1, sizeof(buffer), fp);
713 if (len == -1) {
714 printf("error while reading local file\n");
715 err = 1;
716 } else if (len == 0) {
717 eof = 1;
718 } else {
719 xfer_upload_data(xfer, buffer, len);
720 }
4c7f0d61 721 }
1bc24185 722
2dbd915a 723 if (!xfer_done(xfer)) {
724 pktin = sftp_recv();
725 ret = xfer_upload_gotpkt(xfer, pktin);
726 if (!ret) {
727 printf("error while writing: %s\n", fxp_error());
728 err = 1;
729 }
4c7f0d61 730 }
4c7f0d61 731 }
732
df0870fc 733 xfer_cleanup(xfer);
734
1bc24185 735 sftp_register(req = fxp_close_send(fh));
736 rreq = sftp_find_request(pktin = sftp_recv());
737 assert(rreq == req);
7b7de4f4 738 fxp_close_recv(pktin, rreq);
1bc24185 739
4c7f0d61 740 fclose(fp);
93e86a8b 741
742 return ret;
743}
744
745/* ----------------------------------------------------------------------
746 * Actual sftp commands.
747 */
748struct sftp_command {
749 char **words;
750 int nwords, wordssize;
751 int (*obey) (struct sftp_command *); /* returns <0 to quit */
752};
753
754int sftp_cmd_null(struct sftp_command *cmd)
755{
756 return 1; /* success */
757}
758
759int sftp_cmd_unknown(struct sftp_command *cmd)
760{
761 printf("psftp: unknown command \"%s\"\n", cmd->words[0]);
762 return 0; /* failure */
763}
764
765int sftp_cmd_quit(struct sftp_command *cmd)
766{
767 return -1;
768}
769
770/*
771 * List a directory. If no arguments are given, list pwd; otherwise
772 * list the directory given in words[1].
773 */
774int sftp_cmd_ls(struct sftp_command *cmd)
775{
776 struct fxp_handle *dirh;
777 struct fxp_names *names;
778 struct fxp_name **ournames;
779 int nnames, namesize;
3394416c 780 char *dir, *cdir, *unwcdir, *wildcard;
93e86a8b 781 struct sftp_packet *pktin;
782 struct sftp_request *req, *rreq;
783 int i;
784
785 if (back == NULL) {
786 printf("psftp: not connected to a host; use \"open host.name\"\n");
787 return 0;
788 }
789
790 if (cmd->nwords < 2)
791 dir = ".";
792 else
793 dir = cmd->words[1];
794
3394416c 795 unwcdir = snewn(1 + strlen(dir), char);
796 if (wc_unescape(unwcdir, dir)) {
797 dir = unwcdir;
798 wildcard = NULL;
799 } else {
800 char *tmpdir;
801 int len, check;
802
803 wildcard = stripslashes(dir, 0);
804 unwcdir = dupstr(dir);
805 len = wildcard - dir;
806 unwcdir[len] = '\0';
807 if (len > 0 && unwcdir[len-1] == '/')
808 unwcdir[len-1] = '\0';
809 tmpdir = snewn(1 + len, char);
810 check = wc_unescape(tmpdir, unwcdir);
811 sfree(tmpdir);
812 if (!check) {
813 printf("Multiple-level wildcards are not supported\n");
814 sfree(unwcdir);
815 return 0;
816 }
817 dir = unwcdir;
818 }
819
93e86a8b 820 cdir = canonify(dir);
821 if (!cdir) {
822 printf("%s: %s\n", dir, fxp_error());
3394416c 823 sfree(unwcdir);
93e86a8b 824 return 0;
825 }
826
827 printf("Listing directory %s\n", cdir);
828
829 sftp_register(req = fxp_opendir_send(cdir));
830 rreq = sftp_find_request(pktin = sftp_recv());
831 assert(rreq == req);
832 dirh = fxp_opendir_recv(pktin, rreq);
833
834 if (dirh == NULL) {
835 printf("Unable to open %s: %s\n", dir, fxp_error());
836 } else {
837 nnames = namesize = 0;
838 ournames = NULL;
839
840 while (1) {
841
842 sftp_register(req = fxp_readdir_send(dirh));
843 rreq = sftp_find_request(pktin = sftp_recv());
844 assert(rreq == req);
845 names = fxp_readdir_recv(pktin, rreq);
846
847 if (names == NULL) {
848 if (fxp_error_type() == SSH_FX_EOF)
849 break;
850 printf("Reading directory %s: %s\n", dir, fxp_error());
851 break;
852 }
853 if (names->nnames == 0) {
854 fxp_free_names(names);
855 break;
856 }
857
858 if (nnames + names->nnames >= namesize) {
859 namesize += names->nnames + 128;
860 ournames = sresize(ournames, namesize, struct fxp_name *);
861 }
862
863 for (i = 0; i < names->nnames; i++)
3394416c 864 if (!wildcard || wc_match(wildcard, names->names[i].filename))
865 ournames[nnames++] = fxp_dup_name(&names->names[i]);
93e86a8b 866
867 fxp_free_names(names);
868 }
869 sftp_register(req = fxp_close_send(dirh));
870 rreq = sftp_find_request(pktin = sftp_recv());
871 assert(rreq == req);
872 fxp_close_recv(pktin, rreq);
873
874 /*
875 * Now we have our filenames. Sort them by actual file
876 * name, and then output the longname parts.
877 */
878 qsort(ournames, nnames, sizeof(*ournames), sftp_name_compare);
879
880 /*
881 * And print them.
882 */
883 for (i = 0; i < nnames; i++) {
884 printf("%s\n", ournames[i]->longname);
885 fxp_free_name(ournames[i]);
886 }
887 sfree(ournames);
888 }
889
890 sfree(cdir);
3394416c 891 sfree(unwcdir);
93e86a8b 892
893 return 1;
894}
895
896/*
897 * Change directories. We do this by canonifying the new name, then
898 * trying to OPENDIR it. Only if that succeeds do we set the new pwd.
899 */
900int sftp_cmd_cd(struct sftp_command *cmd)
901{
902 struct fxp_handle *dirh;
903 struct sftp_packet *pktin;
904 struct sftp_request *req, *rreq;
905 char *dir;
906
907 if (back == NULL) {
908 printf("psftp: not connected to a host; use \"open host.name\"\n");
909 return 0;
910 }
911
912 if (cmd->nwords < 2)
913 dir = dupstr(homedir);
914 else
915 dir = canonify(cmd->words[1]);
916
917 if (!dir) {
918 printf("%s: %s\n", dir, fxp_error());
919 return 0;
920 }
921
922 sftp_register(req = fxp_opendir_send(dir));
923 rreq = sftp_find_request(pktin = sftp_recv());
924 assert(rreq == req);
925 dirh = fxp_opendir_recv(pktin, rreq);
926
927 if (!dirh) {
928 printf("Directory %s: %s\n", dir, fxp_error());
929 sfree(dir);
930 return 0;
931 }
932
933 sftp_register(req = fxp_close_send(dirh));
934 rreq = sftp_find_request(pktin = sftp_recv());
935 assert(rreq == req);
936 fxp_close_recv(pktin, rreq);
937
938 sfree(pwd);
939 pwd = dir;
940 printf("Remote directory is now %s\n", pwd);
941
942 return 1;
943}
944
945/*
946 * Print current directory. Easy as pie.
947 */
948int sftp_cmd_pwd(struct sftp_command *cmd)
949{
950 if (back == NULL) {
951 printf("psftp: not connected to a host; use \"open host.name\"\n");
952 return 0;
953 }
954
955 printf("Remote directory is %s\n", pwd);
956 return 1;
957}
958
959/*
9c77ddf6 960 * Get a file and save it at the local end. We have three very
961 * similar commands here. The basic one is `get'; `reget' differs
962 * in that it checks for the existence of the destination file and
963 * starts from where a previous aborted transfer left off; `mget'
964 * differs in that it interprets all its arguments as files to
965 * transfer (never as a different local name for a remote file) and
966 * can handle wildcards.
93e86a8b 967 */
9c77ddf6 968int sftp_general_get(struct sftp_command *cmd, int restart, int multiple)
93e86a8b 969{
9c77ddf6 970 char *fname, *unwcfname, *origfname, *outfname;
93e86a8b 971 int i, ret;
972 int recurse = FALSE;
973
974 if (back == NULL) {
975 printf("psftp: not connected to a host; use \"open host.name\"\n");
976 return 0;
977 }
978
979 i = 1;
980 while (i < cmd->nwords && cmd->words[i][0] == '-') {
981 if (!strcmp(cmd->words[i], "--")) {
982 /* finish processing options */
983 i++;
984 break;
985 } else if (!strcmp(cmd->words[i], "-r")) {
986 recurse = TRUE;
987 } else {
988 printf("get: unrecognised option '%s'\n", cmd->words[i]);
989 return 0;
990 }
991 i++;
992 }
993
994 if (i >= cmd->nwords) {
995 printf("get: expects a filename\n");
996 return 0;
997 }
998
9c77ddf6 999 do {
1000 unwcfname = NULL;
1001 origfname = cmd->words[i++];
93e86a8b 1002
9c77ddf6 1003 if (multiple &&
1004 !wc_unescape(unwcfname = snewn(strlen(origfname)+1, char),
1005 origfname)) {
1006 ret = sftp_get_file(pwd, NULL, recurse, restart, origfname);
1007 } else {
1008 fname = canonify(origfname);
1009 if (!fname) {
1010 printf("%s: %s\n", origfname, fxp_error());
1011 sfree(unwcfname);
1012 return 0;
1013 }
93e86a8b 1014
9c77ddf6 1015 if (!multiple && i < cmd->nwords)
1016 outfname = cmd->words[i++];
1017 else
3394416c 1018 outfname = stripslashes(origfname, 0);
93e86a8b 1019
9c77ddf6 1020 ret = sftp_get_file(fname, outfname, recurse, restart, NULL);
1021
1022 sfree(fname);
1023 }
1024 sfree(unwcfname);
1025 if (!ret)
1026 return ret;
1027
1028 } while (multiple && i < cmd->nwords);
93e86a8b 1029
1030 return ret;
1031}
1032int sftp_cmd_get(struct sftp_command *cmd)
1033{
9c77ddf6 1034 return sftp_general_get(cmd, 0, 0);
1035}
1036int sftp_cmd_mget(struct sftp_command *cmd)
1037{
1038 return sftp_general_get(cmd, 0, 1);
93e86a8b 1039}
1040int sftp_cmd_reget(struct sftp_command *cmd)
1041{
9c77ddf6 1042 return sftp_general_get(cmd, 1, 0);
93e86a8b 1043}
1044
1045/*
9c77ddf6 1046 * Send a file and store it at the remote end. We have three very
1047 * similar commands here. The basic one is `put'; `reput' differs
1048 * in that it checks for the existence of the destination file and
1049 * starts from where a previous aborted transfer left off; `mput'
1050 * differs in that it interprets all its arguments as files to
1051 * transfer (never as a different remote name for a local file) and
1052 * can handle wildcards.
93e86a8b 1053 */
9c77ddf6 1054int sftp_general_put(struct sftp_command *cmd, int restart, int multiple)
93e86a8b 1055{
1056 char *fname, *origoutfname, *outfname;
1057 int i, ret;
1058 int recurse = FALSE;
1059
1060 if (back == NULL) {
1061 printf("psftp: not connected to a host; use \"open host.name\"\n");
1062 return 0;
1063 }
1064
1065 i = 1;
1066 while (i < cmd->nwords && cmd->words[i][0] == '-') {
1067 if (!strcmp(cmd->words[i], "--")) {
1068 /* finish processing options */
1069 i++;
1070 break;
1071 } else if (!strcmp(cmd->words[i], "-r")) {
1072 recurse = TRUE;
1073 } else {
1074 printf("put: unrecognised option '%s'\n", cmd->words[i]);
1075 return 0;
1076 }
1077 i++;
1078 }
1079
1080 if (i >= cmd->nwords) {
1081 printf("put: expects a filename\n");
1082 return 0;
1083 }
1084
9c77ddf6 1085 do {
1086 fname = cmd->words[i++];
93e86a8b 1087
9c77ddf6 1088 if (multiple && test_wildcard(fname, FALSE) == WCTYPE_WILDCARD) {
1089 ret = sftp_put_file(NULL, pwd, recurse, restart, fname);
1090 } else {
1091 if (!multiple && i < cmd->nwords)
1092 origoutfname = cmd->words[i++];
1093 else
1094 origoutfname = stripslashes(fname, 1);
1095
1096 outfname = canonify(origoutfname);
1097 if (!outfname) {
1098 printf("%s: %s\n", origoutfname, fxp_error());
1099 return 0;
1100 }
1101 ret = sftp_put_file(fname, outfname, recurse, restart, NULL);
1102 sfree(outfname);
1103 }
1104 if (!ret)
1105 return ret;
93e86a8b 1106
9c77ddf6 1107 } while (multiple && i < cmd->nwords);
4c7f0d61 1108
df49ff19 1109 return ret;
4c7f0d61 1110}
d92624dc 1111int sftp_cmd_put(struct sftp_command *cmd)
1112{
9c77ddf6 1113 return sftp_general_put(cmd, 0, 0);
1114}
1115int sftp_cmd_mput(struct sftp_command *cmd)
1116{
1117 return sftp_general_put(cmd, 0, 1);
d92624dc 1118}
1119int sftp_cmd_reput(struct sftp_command *cmd)
1120{
9c77ddf6 1121 return sftp_general_put(cmd, 1, 0);
d92624dc 1122}
4c7f0d61 1123
9954aaa3 1124int sftp_cmd_mkdir(struct sftp_command *cmd)
1125{
1126 char *dir;
1bc24185 1127 struct sftp_packet *pktin;
1128 struct sftp_request *req, *rreq;
9954aaa3 1129 int result;
1130
fa3db767 1131 if (back == NULL) {
1132 printf("psftp: not connected to a host; use \"open host.name\"\n");
1133 return 0;
1134 }
9954aaa3 1135
1136 if (cmd->nwords < 2) {
1137 printf("mkdir: expects a directory\n");
1138 return 0;
1139 }
1140
1141 dir = canonify(cmd->words[1]);
1142 if (!dir) {
1143 printf("%s: %s\n", dir, fxp_error());
1144 return 0;
1145 }
1146
1bc24185 1147 sftp_register(req = fxp_mkdir_send(dir));
1148 rreq = sftp_find_request(pktin = sftp_recv());
1149 assert(rreq == req);
7b7de4f4 1150 result = fxp_mkdir_recv(pktin, rreq);
1bc24185 1151
9954aaa3 1152 if (!result) {
1153 printf("mkdir %s: %s\n", dir, fxp_error());
1154 sfree(dir);
1155 return 0;
1156 }
1157
d92624dc 1158 sfree(dir);
df49ff19 1159 return 1;
9954aaa3 1160}
1161
1162int sftp_cmd_rmdir(struct sftp_command *cmd)
1163{
1164 char *dir;
1bc24185 1165 struct sftp_packet *pktin;
1166 struct sftp_request *req, *rreq;
9954aaa3 1167 int result;
1168
fa3db767 1169 if (back == NULL) {
1170 printf("psftp: not connected to a host; use \"open host.name\"\n");
1171 return 0;
1172 }
9954aaa3 1173
1174 if (cmd->nwords < 2) {
1175 printf("rmdir: expects a directory\n");
1176 return 0;
1177 }
1178
1179 dir = canonify(cmd->words[1]);
1180 if (!dir) {
1181 printf("%s: %s\n", dir, fxp_error());
1182 return 0;
1183 }
1184
1bc24185 1185 sftp_register(req = fxp_rmdir_send(dir));
1186 rreq = sftp_find_request(pktin = sftp_recv());
1187 assert(rreq == req);
7b7de4f4 1188 result = fxp_rmdir_recv(pktin, rreq);
1bc24185 1189
9954aaa3 1190 if (!result) {
1191 printf("rmdir %s: %s\n", dir, fxp_error());
1192 sfree(dir);
1193 return 0;
1194 }
1195
d92624dc 1196 sfree(dir);
df49ff19 1197 return 1;
9954aaa3 1198}
1199
1200int sftp_cmd_rm(struct sftp_command *cmd)
1201{
1202 char *fname;
1bc24185 1203 struct sftp_packet *pktin;
1204 struct sftp_request *req, *rreq;
9954aaa3 1205 int result;
1206
fa3db767 1207 if (back == NULL) {
1208 printf("psftp: not connected to a host; use \"open host.name\"\n");
1209 return 0;
1210 }
1211
9954aaa3 1212 if (cmd->nwords < 2) {
1213 printf("rm: expects a filename\n");
1214 return 0;
1215 }
1216
1217 fname = canonify(cmd->words[1]);
1218 if (!fname) {
1219 printf("%s: %s\n", fname, fxp_error());
1220 return 0;
1221 }
1222
1bc24185 1223 sftp_register(req = fxp_remove_send(fname));
1224 rreq = sftp_find_request(pktin = sftp_recv());
1225 assert(rreq == req);
7b7de4f4 1226 result = fxp_remove_recv(pktin, rreq);
1bc24185 1227
9954aaa3 1228 if (!result) {
1229 printf("rm %s: %s\n", fname, fxp_error());
1230 sfree(fname);
1231 return 0;
1232 }
1233
d92624dc 1234 sfree(fname);
df49ff19 1235 return 1;
d92624dc 1236}
1237
1238int sftp_cmd_mv(struct sftp_command *cmd)
1239{
1240 char *srcfname, *dstfname;
1bc24185 1241 struct sftp_packet *pktin;
1242 struct sftp_request *req, *rreq;
d92624dc 1243 int result;
1244
fa3db767 1245 if (back == NULL) {
1246 printf("psftp: not connected to a host; use \"open host.name\"\n");
1247 return 0;
1248 }
1249
d92624dc 1250 if (cmd->nwords < 3) {
1251 printf("mv: expects two filenames\n");
9954aaa3 1252 return 0;
d92624dc 1253 }
1254 srcfname = canonify(cmd->words[1]);
1255 if (!srcfname) {
1256 printf("%s: %s\n", srcfname, fxp_error());
1257 return 0;
1258 }
1259
1260 dstfname = canonify(cmd->words[2]);
1261 if (!dstfname) {
1262 printf("%s: %s\n", dstfname, fxp_error());
1263 return 0;
1264 }
9954aaa3 1265
1bc24185 1266 sftp_register(req = fxp_rename_send(srcfname, dstfname));
1267 rreq = sftp_find_request(pktin = sftp_recv());
1268 assert(rreq == req);
7b7de4f4 1269 result = fxp_rename_recv(pktin, rreq);
1bc24185 1270
d92624dc 1271 if (!result) {
1272 char const *error = fxp_error();
1273 struct fxp_attrs attrs;
1274
1275 /*
1276 * The move might have failed because dstfname pointed at a
1277 * directory. We check this possibility now: if dstfname
1278 * _is_ a directory, we re-attempt the move by appending
1279 * the basename of srcfname to dstfname.
1280 */
1bc24185 1281 sftp_register(req = fxp_stat_send(dstfname));
1282 rreq = sftp_find_request(pktin = sftp_recv());
1283 assert(rreq == req);
7b7de4f4 1284 result = fxp_stat_recv(pktin, rreq, &attrs);
1bc24185 1285
d92624dc 1286 if (result &&
1287 (attrs.flags & SSH_FILEXFER_ATTR_PERMISSIONS) &&
1288 (attrs.permissions & 0040000)) {
1289 char *p;
1290 char *newname, *newcanon;
1291 printf("(destination %s is a directory)\n", dstfname);
1292 p = srcfname + strlen(srcfname);
1293 while (p > srcfname && p[-1] != '/') p--;
1294 newname = dupcat(dstfname, "/", p, NULL);
1295 newcanon = canonify(newname);
1296 sfree(newname);
1297 if (newcanon) {
1298 sfree(dstfname);
1299 dstfname = newcanon;
1bc24185 1300
1301 sftp_register(req = fxp_rename_send(srcfname, dstfname));
1302 rreq = sftp_find_request(pktin = sftp_recv());
1303 assert(rreq == req);
7b7de4f4 1304 result = fxp_rename_recv(pktin, rreq);
1bc24185 1305
d92624dc 1306 error = result ? NULL : fxp_error();
1307 }
1308 }
1309 if (error) {
1310 printf("mv %s %s: %s\n", srcfname, dstfname, error);
1311 sfree(srcfname);
1312 sfree(dstfname);
1313 return 0;
1314 }
1315 }
1316 printf("%s -> %s\n", srcfname, dstfname);
1317
1318 sfree(srcfname);
1319 sfree(dstfname);
df49ff19 1320 return 1;
9954aaa3 1321}
1322
d92624dc 1323int sftp_cmd_chmod(struct sftp_command *cmd)
1324{
1325 char *fname, *mode;
1326 int result;
1327 struct fxp_attrs attrs;
1328 unsigned attrs_clr, attrs_xor, oldperms, newperms;
1bc24185 1329 struct sftp_packet *pktin;
1330 struct sftp_request *req, *rreq;
d92624dc 1331
fa3db767 1332 if (back == NULL) {
1333 printf("psftp: not connected to a host; use \"open host.name\"\n");
1334 return 0;
1335 }
1336
d92624dc 1337 if (cmd->nwords < 3) {
1338 printf("chmod: expects a mode specifier and a filename\n");
1339 return 0;
1340 }
1341
1342 /*
1343 * Attempt to parse the mode specifier in cmd->words[1]. We
1344 * don't support the full horror of Unix chmod; instead we
1345 * support a much simpler syntax in which the user can either
1346 * specify an octal number, or a comma-separated sequence of
1347 * [ugoa]*[-+=][rwxst]+. (The initial [ugoa] sequence may
1348 * _only_ be omitted if the only attribute mentioned is t,
1349 * since all others require a user/group/other specification.
1350 * Additionally, the s attribute may not be specified for any
1351 * [ugoa] specifications other than exactly u or exactly g.
1352 */
1353 attrs_clr = attrs_xor = 0;
1354 mode = cmd->words[1];
1355 if (mode[0] >= '0' && mode[0] <= '9') {
1356 if (mode[strspn(mode, "01234567")]) {
1357 printf("chmod: numeric file modes should"
1358 " contain digits 0-7 only\n");
1359 return 0;
1360 }
1361 attrs_clr = 07777;
1362 sscanf(mode, "%o", &attrs_xor);
1363 attrs_xor &= attrs_clr;
1364 } else {
1365 while (*mode) {
1366 char *modebegin = mode;
1367 unsigned subset, perms;
1368 int action;
1369
1370 subset = 0;
1371 while (*mode && *mode != ',' &&
1372 *mode != '+' && *mode != '-' && *mode != '=') {
1373 switch (*mode) {
1374 case 'u': subset |= 04700; break; /* setuid, user perms */
1375 case 'g': subset |= 02070; break; /* setgid, group perms */
1376 case 'o': subset |= 00007; break; /* just other perms */
1377 case 'a': subset |= 06777; break; /* all of the above */
1378 default:
1379 printf("chmod: file mode '%.*s' contains unrecognised"
1380 " user/group/other specifier '%c'\n",
b51259f6 1381 (int)strcspn(modebegin, ","), modebegin, *mode);
d92624dc 1382 return 0;
1383 }
1384 mode++;
1385 }
1386 if (!*mode || *mode == ',') {
1387 printf("chmod: file mode '%.*s' is incomplete\n",
b51259f6 1388 (int)strcspn(modebegin, ","), modebegin);
d92624dc 1389 return 0;
1390 }
1391 action = *mode++;
1392 if (!*mode || *mode == ',') {
1393 printf("chmod: file mode '%.*s' is incomplete\n",
b51259f6 1394 (int)strcspn(modebegin, ","), modebegin);
d92624dc 1395 return 0;
1396 }
1397 perms = 0;
1398 while (*mode && *mode != ',') {
1399 switch (*mode) {
1400 case 'r': perms |= 00444; break;
1401 case 'w': perms |= 00222; break;
1402 case 'x': perms |= 00111; break;
1403 case 't': perms |= 01000; subset |= 01000; break;
1404 case 's':
1405 if ((subset & 06777) != 04700 &&
1406 (subset & 06777) != 02070) {
1407 printf("chmod: file mode '%.*s': set[ug]id bit should"
1408 " be used with exactly one of u or g only\n",
b51259f6 1409 (int)strcspn(modebegin, ","), modebegin);
d92624dc 1410 return 0;
1411 }
1412 perms |= 06000;
1413 break;
1414 default:
1415 printf("chmod: file mode '%.*s' contains unrecognised"
1416 " permission specifier '%c'\n",
b51259f6 1417 (int)strcspn(modebegin, ","), modebegin, *mode);
d92624dc 1418 return 0;
1419 }
1420 mode++;
1421 }
1422 if (!(subset & 06777) && (perms &~ subset)) {
1423 printf("chmod: file mode '%.*s' contains no user/group/other"
1424 " specifier and permissions other than 't' \n",
b51259f6 1425 (int)strcspn(modebegin, ","), modebegin);
d92624dc 1426 return 0;
1427 }
1428 perms &= subset;
1429 switch (action) {
1430 case '+':
1431 attrs_clr |= perms;
1432 attrs_xor |= perms;
1433 break;
1434 case '-':
1435 attrs_clr |= perms;
1436 attrs_xor &= ~perms;
1437 break;
1438 case '=':
1439 attrs_clr |= subset;
1440 attrs_xor |= perms;
1441 break;
1442 }
1443 if (*mode) mode++; /* eat comma */
1444 }
1445 }
1446
1447 fname = canonify(cmd->words[2]);
1448 if (!fname) {
1449 printf("%s: %s\n", fname, fxp_error());
1450 return 0;
1451 }
1452
1bc24185 1453 sftp_register(req = fxp_stat_send(fname));
1454 rreq = sftp_find_request(pktin = sftp_recv());
1455 assert(rreq == req);
7b7de4f4 1456 result = fxp_stat_recv(pktin, rreq, &attrs);
1bc24185 1457
d92624dc 1458 if (!result || !(attrs.flags & SSH_FILEXFER_ATTR_PERMISSIONS)) {
1459 printf("get attrs for %s: %s\n", fname,
1460 result ? "file permissions not provided" : fxp_error());
1461 sfree(fname);
1462 return 0;
1463 }
1464
1465 attrs.flags = SSH_FILEXFER_ATTR_PERMISSIONS; /* perms _only_ */
1466 oldperms = attrs.permissions & 07777;
1467 attrs.permissions &= ~attrs_clr;
1468 attrs.permissions ^= attrs_xor;
1469 newperms = attrs.permissions & 07777;
1470
1bc24185 1471 sftp_register(req = fxp_setstat_send(fname, attrs));
1472 rreq = sftp_find_request(pktin = sftp_recv());
1473 assert(rreq == req);
7b7de4f4 1474 result = fxp_setstat_recv(pktin, rreq);
d92624dc 1475
1476 if (!result) {
1477 printf("set attrs for %s: %s\n", fname, fxp_error());
1478 sfree(fname);
1479 return 0;
1480 }
1481
1482 printf("%s: %04o -> %04o\n", fname, oldperms, newperms);
1483
1484 sfree(fname);
df49ff19 1485 return 1;
d92624dc 1486}
9954aaa3 1487
fa3db767 1488static int sftp_cmd_open(struct sftp_command *cmd)
1489{
f11233cb 1490 int portnumber;
1491
fa3db767 1492 if (back != NULL) {
1493 printf("psftp: already connected\n");
1494 return 0;
1495 }
1496
1497 if (cmd->nwords < 2) {
1498 printf("open: expects a host name\n");
1499 return 0;
1500 }
1501
f11233cb 1502 if (cmd->nwords > 2) {
1503 portnumber = atoi(cmd->words[2]);
1504 if (portnumber == 0) {
1505 printf("open: invalid port number\n");
1506 return 0;
1507 }
1508 } else
1509 portnumber = 0;
1510
1511 if (psftp_connect(cmd->words[1], NULL, portnumber)) {
fa3db767 1512 back = NULL; /* connection is already closed */
1513 return -1; /* this is fatal */
1514 }
1515 do_sftp_init();
df49ff19 1516 return 1;
fa3db767 1517}
1518
3af97463 1519static int sftp_cmd_lcd(struct sftp_command *cmd)
1520{
d6cc41e6 1521 char *currdir, *errmsg;
3af97463 1522
1523 if (cmd->nwords < 2) {
1524 printf("lcd: expects a local directory name\n");
1525 return 0;
1526 }
1527
d6cc41e6 1528 errmsg = psftp_lcd(cmd->words[1]);
1529 if (errmsg) {
1530 printf("lcd: unable to change directory: %s\n", errmsg);
1531 sfree(errmsg);
3af97463 1532 return 0;
1533 }
1534
d6cc41e6 1535 currdir = psftp_getcwd();
3af97463 1536 printf("New local directory is %s\n", currdir);
1537 sfree(currdir);
1538
1539 return 1;
1540}
1541
1542static int sftp_cmd_lpwd(struct sftp_command *cmd)
1543{
1544 char *currdir;
3af97463 1545
d6cc41e6 1546 currdir = psftp_getcwd();
3af97463 1547 printf("Current local directory is %s\n", currdir);
1548 sfree(currdir);
1549
1550 return 1;
1551}
1552
1553static int sftp_cmd_pling(struct sftp_command *cmd)
1554{
1555 int exitcode;
1556
1557 exitcode = system(cmd->words[1]);
1558 return (exitcode == 0);
1559}
1560
bf5240cd 1561static int sftp_cmd_help(struct sftp_command *cmd);
1562
4c7f0d61 1563static struct sftp_cmd_lookup {
1564 char *name;
bf5240cd 1565 /*
1566 * For help purposes, there are two kinds of command:
1567 *
1568 * - primary commands, in which `longhelp' is non-NULL. In
1569 * this case `shorthelp' is descriptive text, and `longhelp'
1570 * is longer descriptive text intended to be printed after
1571 * the command name.
1572 *
1573 * - alias commands, in which `longhelp' is NULL. In this case
1574 * `shorthelp' is the name of a primary command, which
1575 * contains the help that should double up for this command.
1576 */
3af97463 1577 int listed; /* do we list this in primary help? */
bf5240cd 1578 char *shorthelp;
1579 char *longhelp;
32874aea 1580 int (*obey) (struct sftp_command *);
4c7f0d61 1581} sftp_lookup[] = {
1582 /*
1583 * List of sftp commands. This is binary-searched so it MUST be
1584 * in ASCII order.
1585 */
32874aea 1586 {
d6cc41e6 1587 "!", TRUE, "run a local command",
3af97463 1588 "<command>\n"
d6cc41e6 1589 /* FIXME: this example is crap for non-Windows. */
1590 " Runs a local command. For example, \"!del myfile\".\n",
3af97463 1591 sftp_cmd_pling
1592 },
1593 {
1594 "bye", TRUE, "finish your SFTP session",
bf5240cd 1595 "\n"
1596 " Terminates your SFTP session and quits the PSFTP program.\n",
1597 sftp_cmd_quit
1598 },
1599 {
3af97463 1600 "cd", TRUE, "change your remote working directory",
bf5240cd 1601 " [ <New working directory> ]\n"
1602 " Change the remote working directory for your SFTP session.\n"
1603 " If a new working directory is not supplied, you will be\n"
1604 " returned to your home directory.\n",
1605 sftp_cmd_cd
1606 },
1607 {
3af97463 1608 "chmod", TRUE, "change file permissions and modes",
bf5240cd 1609 " ( <octal-digits> | <modifiers> ) <filename>\n"
1610 " Change the file permissions on a file or directory.\n"
1611 " <octal-digits> can be any octal Unix permission specifier.\n"
1612 " Alternatively, <modifiers> can include:\n"
1613 " u+r make file readable by owning user\n"
1614 " u+w make file writable by owning user\n"
1615 " u+x make file executable by owning user\n"
1616 " u-r make file not readable by owning user\n"
1617 " [also u-w, u-x]\n"
1618 " g+r make file readable by members of owning group\n"
1619 " [also g+w, g+x, g-r, g-w, g-x]\n"
1620 " o+r make file readable by all other users\n"
1621 " [also o+w, o+x, o-r, o-w, o-x]\n"
1622 " a+r make file readable by absolutely everybody\n"
1623 " [also a+w, a+x, a-r, a-w, a-x]\n"
1624 " u+s enable the Unix set-user-ID bit\n"
1625 " u-s disable the Unix set-user-ID bit\n"
1626 " g+s enable the Unix set-group-ID bit\n"
1627 " g-s disable the Unix set-group-ID bit\n"
1628 " +t enable the Unix \"sticky bit\"\n"
1629 " You can give more than one modifier for the same user (\"g-rwx\"), and\n"
1630 " more than one user for the same modifier (\"ug+w\"). You can\n"
1631 " use commas to separate different modifiers (\"u+rwx,g+s\").\n",
1632 sftp_cmd_chmod
1633 },
1634 {
3af97463 1635 "del", TRUE, "delete a file",
bf5240cd 1636 " <filename>\n"
1637 " Delete a file.\n",
1638 sftp_cmd_rm
1639 },
1640 {
3af97463 1641 "delete", FALSE, "del", NULL, sftp_cmd_rm
bf5240cd 1642 },
1643 {
3af97463 1644 "dir", TRUE, "list contents of a remote directory",
bf5240cd 1645 " [ <directory-name> ]\n"
1646 " List the contents of a specified directory on the server.\n"
1647 " If <directory-name> is not given, the current working directory\n"
1648 " will be listed.\n",
1649 sftp_cmd_ls
1650 },
1651 {
3af97463 1652 "exit", TRUE, "bye", NULL, sftp_cmd_quit
bf5240cd 1653 },
1654 {
3af97463 1655 "get", TRUE, "download a file from the server to your local machine",
bf5240cd 1656 " <filename> [ <local-filename> ]\n"
1657 " Downloads a file on the server and stores it locally under\n"
1658 " the same name, or under a different one if you supply the\n"
1659 " argument <local-filename>.\n",
1660 sftp_cmd_get
1661 },
1662 {
3af97463 1663 "help", TRUE, "give help",
bf5240cd 1664 " [ <command> [ <command> ... ] ]\n"
1665 " Give general help if no commands are specified.\n"
1666 " If one or more commands are specified, give specific help on\n"
1667 " those particular commands.\n",
1668 sftp_cmd_help
1669 },
1670 {
3af97463 1671 "lcd", TRUE, "change local working directory",
1672 " <local-directory-name>\n"
1673 " Change the local working directory of the PSFTP program (the\n"
1674 " default location where the \"get\" command will save files).\n",
1675 sftp_cmd_lcd
1676 },
1677 {
1678 "lpwd", TRUE, "print local working directory",
1679 "\n"
1680 " Print the local working directory of the PSFTP program (the\n"
1681 " default location where the \"get\" command will save files).\n",
1682 sftp_cmd_lpwd
1683 },
1684 {
1685 "ls", TRUE, "dir", NULL,
bf5240cd 1686 sftp_cmd_ls
1687 },
1688 {
9c77ddf6 1689 "mget", TRUE, "download multiple files at once",
1690 " <filename-or-wildcard> [ <filename-or-wildcard>... ]\n"
1691 " Downloads many files from the server, storing each one under\n"
1692 " the same name it has on the server side. You can use wildcards\n"
1693 " such as \"*.c\" to specify lots of files at once.\n",
1694 sftp_cmd_mget
1695 },
1696 {
3af97463 1697 "mkdir", TRUE, "create a directory on the remote server",
bf5240cd 1698 " <directory-name>\n"
1699 " Creates a directory with the given name on the server.\n",
1700 sftp_cmd_mkdir
1701 },
1702 {
9c77ddf6 1703 "mput", TRUE, "upload multiple files at once",
1704 " <filename-or-wildcard> [ <filename-or-wildcard>... ]\n"
1705 " Uploads many files to the server, storing each one under the\n"
1706 " same name it has on the client side. You can use wildcards\n"
1707 " such as \"*.c\" to specify lots of files at once.\n",
1708 sftp_cmd_mput
1709 },
1710 {
3af97463 1711 "mv", TRUE, "move or rename a file on the remote server",
bf5240cd 1712 " <source-filename> <destination-filename>\n"
1713 " Moves or renames the file <source-filename> on the server,\n"
1714 " so that it is accessible under the name <destination-filename>.\n",
1715 sftp_cmd_mv
1716 },
1717 {
3af97463 1718 "open", TRUE, "connect to a host",
f11233cb 1719 " [<user>@]<hostname> [<port>]\n"
fa3db767 1720 " Establishes an SFTP connection to a given host. Only usable\n"
1721 " when you did not already specify a host name on the command\n"
1722 " line.\n",
1723 sftp_cmd_open
1724 },
1725 {
56542985 1726 "put", TRUE, "upload a file from your local machine to the server",
1727 " <filename> [ <remote-filename> ]\n"
1728 " Uploads a file to the server and stores it there under\n"
1729 " the same name, or under a different one if you supply the\n"
1730 " argument <remote-filename>.\n",
1731 sftp_cmd_put
1732 },
1733 {
3af97463 1734 "pwd", TRUE, "print your remote working directory",
4f2b387f 1735 "\n"
1736 " Print the current remote working directory for your SFTP session.\n",
1737 sftp_cmd_pwd
1738 },
1739 {
3af97463 1740 "quit", TRUE, "bye", NULL,
bf5240cd 1741 sftp_cmd_quit
1742 },
1743 {
3af97463 1744 "reget", TRUE, "continue downloading a file",
bf5240cd 1745 " <filename> [ <local-filename> ]\n"
1746 " Works exactly like the \"get\" command, but the local file\n"
1747 " must already exist. The download will begin at the end of the\n"
1748 " file. This is for resuming a download that was interrupted.\n",
1749 sftp_cmd_reget
1750 },
1751 {
3af97463 1752 "ren", TRUE, "mv", NULL,
bf5240cd 1753 sftp_cmd_mv
1754 },
1755 {
3af97463 1756 "rename", FALSE, "mv", NULL,
bf5240cd 1757 sftp_cmd_mv
1758 },
1759 {
3af97463 1760 "reput", TRUE, "continue uploading a file",
bf5240cd 1761 " <filename> [ <remote-filename> ]\n"
1762 " Works exactly like the \"put\" command, but the remote file\n"
1763 " must already exist. The upload will begin at the end of the\n"
1764 " file. This is for resuming an upload that was interrupted.\n",
1765 sftp_cmd_reput
1766 },
1767 {
3af97463 1768 "rm", TRUE, "del", NULL,
bf5240cd 1769 sftp_cmd_rm
1770 },
1771 {
3af97463 1772 "rmdir", TRUE, "remove a directory on the remote server",
bf5240cd 1773 " <directory-name>\n"
1774 " Removes the directory with the given name on the server.\n"
1775 " The directory will not be removed unless it is empty.\n",
1776 sftp_cmd_rmdir
1777 }
1778};
1779
1780const struct sftp_cmd_lookup *lookup_command(char *name)
1781{
1782 int i, j, k, cmp;
1783
1784 i = -1;
1785 j = sizeof(sftp_lookup) / sizeof(*sftp_lookup);
1786 while (j - i > 1) {
1787 k = (j + i) / 2;
1788 cmp = strcmp(name, sftp_lookup[k].name);
1789 if (cmp < 0)
1790 j = k;
1791 else if (cmp > 0)
1792 i = k;
1793 else {
1794 return &sftp_lookup[k];
1795 }
1796 }
1797 return NULL;
1798}
1799
1800static int sftp_cmd_help(struct sftp_command *cmd)
1801{
1802 int i;
1803 if (cmd->nwords == 1) {
1804 /*
1805 * Give short help on each command.
1806 */
1807 int maxlen;
1808 maxlen = 0;
1809 for (i = 0; i < sizeof(sftp_lookup) / sizeof(*sftp_lookup); i++) {
3af97463 1810 int len;
1811 if (!sftp_lookup[i].listed)
1812 continue;
1813 len = strlen(sftp_lookup[i].name);
bf5240cd 1814 if (maxlen < len)
1815 maxlen = len;
1816 }
1817 for (i = 0; i < sizeof(sftp_lookup) / sizeof(*sftp_lookup); i++) {
1818 const struct sftp_cmd_lookup *lookup;
3af97463 1819 if (!sftp_lookup[i].listed)
1820 continue;
bf5240cd 1821 lookup = &sftp_lookup[i];
1822 printf("%-*s", maxlen+2, lookup->name);
1823 if (lookup->longhelp == NULL)
1824 lookup = lookup_command(lookup->shorthelp);
1825 printf("%s\n", lookup->shorthelp);
1826 }
1827 } else {
1828 /*
1829 * Give long help on specific commands.
1830 */
1831 for (i = 1; i < cmd->nwords; i++) {
1832 const struct sftp_cmd_lookup *lookup;
1833 lookup = lookup_command(cmd->words[i]);
1834 if (!lookup) {
1835 printf("help: %s: command not found\n", cmd->words[i]);
1836 } else {
1837 printf("%s", lookup->name);
1838 if (lookup->longhelp == NULL)
1839 lookup = lookup_command(lookup->shorthelp);
1840 printf("%s", lookup->longhelp);
1841 }
1842 }
1843 }
df49ff19 1844 return 1;
bf5240cd 1845}
4c7f0d61 1846
1847/* ----------------------------------------------------------------------
1848 * Command line reading and parsing.
1849 */
9954aaa3 1850struct sftp_command *sftp_getcmd(FILE *fp, int mode, int modeflags)
32874aea 1851{
4c7f0d61 1852 char *line;
4c7f0d61 1853 struct sftp_command *cmd;
1854 char *p, *q, *r;
1855 int quoting;
1856
3d88e64d 1857 cmd = snew(struct sftp_command);
4c7f0d61 1858 cmd->words = NULL;
1859 cmd->nwords = 0;
1860 cmd->wordssize = 0;
1861
1862 line = NULL;
39934deb 1863
1864 if (fp) {
1865 if (modeflags & 1)
1866 printf("psftp> ");
1867 line = fgetline(fp);
1868 } else {
1869 line = ssh_sftp_get_cmdline("psftp> ");
4c7f0d61 1870 }
39934deb 1871
1872 if (!line || !*line) {
1873 cmd->obey = sftp_cmd_quit;
1874 if ((mode == 0) || (modeflags & 1))
1875 printf("quit\n");
1876 return cmd; /* eof */
1877 }
1878
1879 line[strcspn(line, "\r\n")] = '\0';
1880
df49ff19 1881 if (modeflags & 1) {
1882 printf("%s\n", line);
1883 }
4c7f0d61 1884
4c7f0d61 1885 p = line;
3af97463 1886 while (*p && (*p == ' ' || *p == '\t'))
1887 p++;
1888
1889 if (*p == '!') {
1890 /*
1891 * Special case: the ! command. This is always parsed as
1892 * exactly two words: one containing the !, and the second
1893 * containing everything else on the line.
1894 */
1895 cmd->nwords = cmd->wordssize = 2;
3d88e64d 1896 cmd->words = sresize(cmd->words, cmd->wordssize, char *);
679539d7 1897 cmd->words[0] = dupstr("!");
1898 cmd->words[1] = dupstr(p+1);
3af97463 1899 } else {
1900
1901 /*
1902 * Parse the command line into words. The syntax is:
1903 * - double quotes are removed, but cause spaces within to be
1904 * treated as non-separating.
1905 * - a double-doublequote pair is a literal double quote, inside
1906 * _or_ outside quotes. Like this:
1907 *
1908 * firstword "second word" "this has ""quotes"" in" and""this""
1909 *
1910 * becomes
1911 *
1912 * >firstword<
1913 * >second word<
1914 * >this has "quotes" in<
1915 * >and"this"<
1916 */
4c7f0d61 1917 while (*p) {
3af97463 1918 /* skip whitespace */
1919 while (*p && (*p == ' ' || *p == '\t'))
1920 p++;
1921 /* mark start of word */
1922 q = r = p; /* q sits at start, r writes word */
1923 quoting = 0;
1924 while (*p) {
1925 if (!quoting && (*p == ' ' || *p == '\t'))
1926 break; /* reached end of word */
1927 else if (*p == '"' && p[1] == '"')
1928 p += 2, *r++ = '"'; /* a literal quote */
1929 else if (*p == '"')
1930 p++, quoting = !quoting;
1931 else
1932 *r++ = *p++;
1933 }
1934 if (*p)
1935 p++; /* skip over the whitespace */
1936 *r = '\0';
1937 if (cmd->nwords >= cmd->wordssize) {
1938 cmd->wordssize = cmd->nwords + 16;
3d88e64d 1939 cmd->words = sresize(cmd->words, cmd->wordssize, char *);
3af97463 1940 }
679539d7 1941 cmd->words[cmd->nwords++] = dupstr(q);
4c7f0d61 1942 }
4c7f0d61 1943 }
1944
39934deb 1945 sfree(line);
1946
4c7f0d61 1947 /*
1948 * Now parse the first word and assign a function.
1949 */
1950
1951 if (cmd->nwords == 0)
1952 cmd->obey = sftp_cmd_null;
1953 else {
bf5240cd 1954 const struct sftp_cmd_lookup *lookup;
1955 lookup = lookup_command(cmd->words[0]);
1956 if (!lookup)
1957 cmd->obey = sftp_cmd_unknown;
1958 else
1959 cmd->obey = lookup->obey;
4c7f0d61 1960 }
1961
1962 return cmd;
1963}
1964
774204f5 1965static int do_sftp_init(void)
32874aea 1966{
1bc24185 1967 struct sftp_packet *pktin;
1968 struct sftp_request *req, *rreq;
1969
4c7f0d61 1970 /*
1971 * Do protocol initialisation.
1972 */
1973 if (!fxp_init()) {
1974 fprintf(stderr,
32874aea 1975 "Fatal: unable to initialise SFTP: %s\n", fxp_error());
774204f5 1976 return 1; /* failure */
4c7f0d61 1977 }
1978
1979 /*
1980 * Find out where our home directory is.
1981 */
1bc24185 1982 sftp_register(req = fxp_realpath_send("."));
1983 rreq = sftp_find_request(pktin = sftp_recv());
1984 assert(rreq == req);
7b7de4f4 1985 homedir = fxp_realpath_recv(pktin, rreq);
1bc24185 1986
4c7f0d61 1987 if (!homedir) {
1988 fprintf(stderr,
1989 "Warning: failed to resolve home directory: %s\n",
1990 fxp_error());
1991 homedir = dupstr(".");
1992 } else {
1993 printf("Remote working directory is %s\n", homedir);
1994 }
1995 pwd = dupstr(homedir);
774204f5 1996 return 0;
fa3db767 1997}
1998
679539d7 1999void do_sftp_cleanup()
2000{
2001 char ch;
f11233cb 2002 if (back) {
2003 back->special(backhandle, TS_EOF);
2004 sftp_recvdata(&ch, 1);
2005 back->free(backhandle);
2006 sftp_cleanup_request();
2007 }
679539d7 2008 if (pwd) {
2009 sfree(pwd);
2010 pwd = NULL;
2011 }
2012 if (homedir) {
2013 sfree(homedir);
2014 homedir = NULL;
2015 }
2016}
2017
fa3db767 2018void do_sftp(int mode, int modeflags, char *batchfile)
2019{
2020 FILE *fp;
df49ff19 2021 int ret;
4c7f0d61 2022
9954aaa3 2023 /*
2024 * Batch mode?
4c7f0d61 2025 */
9954aaa3 2026 if (mode == 0) {
2027
2028 /* ------------------------------------------------------------------
2029 * Now we're ready to do Real Stuff.
2030 */
2031 while (1) {
df49ff19 2032 struct sftp_command *cmd;
39934deb 2033 cmd = sftp_getcmd(NULL, 0, 0);
df49ff19 2034 if (!cmd)
2035 break;
679539d7 2036 ret = cmd->obey(cmd);
2037 if (cmd->words) {
2038 int i;
2039 for(i = 0; i < cmd->nwords; i++)
2040 sfree(cmd->words[i]);
2041 sfree(cmd->words);
2042 }
2043 sfree(cmd);
2044 if (ret < 0)
df49ff19 2045 break;
bf5240cd 2046 }
9954aaa3 2047 } else {
2048 fp = fopen(batchfile, "r");
2049 if (!fp) {
bf5240cd 2050 printf("Fatal: unable to open %s\n", batchfile);
2051 return;
9954aaa3 2052 }
2053 while (1) {
bf5240cd 2054 struct sftp_command *cmd;
2055 cmd = sftp_getcmd(fp, mode, modeflags);
2056 if (!cmd)
2057 break;
df49ff19 2058 ret = cmd->obey(cmd);
2059 if (ret < 0)
bf5240cd 2060 break;
df49ff19 2061 if (ret == 0) {
bf5240cd 2062 if (!(modeflags & 2))
9954aaa3 2063 break;
bf5240cd 2064 }
9954aaa3 2065 }
bf5240cd 2066 fclose(fp);
9954aaa3 2067
4c7f0d61 2068 }
4a8fc3c4 2069}
4c7f0d61 2070
4a8fc3c4 2071/* ----------------------------------------------------------------------
2072 * Dirty bits: integration with PuTTY.
2073 */
2074
2075static int verbose = 0;
2076
7bedb13c 2077/*
4a8fc3c4 2078 * Print an error message and perform a fatal exit.
2079 */
2080void fatalbox(char *fmt, ...)
2081{
57356d63 2082 char *str, *str2;
4a8fc3c4 2083 va_list ap;
2084 va_start(ap, fmt);
57356d63 2085 str = dupvprintf(fmt, ap);
2086 str2 = dupcat("Fatal: ", str, "\n", NULL);
2087 sfree(str);
4a8fc3c4 2088 va_end(ap);
57356d63 2089 fputs(str2, stderr);
2090 sfree(str2);
4a8fc3c4 2091
93b581bd 2092 cleanup_exit(1);
4a8fc3c4 2093}
1709795f 2094void modalfatalbox(char *fmt, ...)
2095{
57356d63 2096 char *str, *str2;
1709795f 2097 va_list ap;
2098 va_start(ap, fmt);
57356d63 2099 str = dupvprintf(fmt, ap);
2100 str2 = dupcat("Fatal: ", str, "\n", NULL);
2101 sfree(str);
1709795f 2102 va_end(ap);
57356d63 2103 fputs(str2, stderr);
2104 sfree(str2);
1709795f 2105
2106 cleanup_exit(1);
2107}
a8327734 2108void connection_fatal(void *frontend, char *fmt, ...)
4a8fc3c4 2109{
57356d63 2110 char *str, *str2;
4a8fc3c4 2111 va_list ap;
2112 va_start(ap, fmt);
57356d63 2113 str = dupvprintf(fmt, ap);
2114 str2 = dupcat("Fatal: ", str, "\n", NULL);
2115 sfree(str);
4a8fc3c4 2116 va_end(ap);
57356d63 2117 fputs(str2, stderr);
2118 sfree(str2);
4a8fc3c4 2119
93b581bd 2120 cleanup_exit(1);
4a8fc3c4 2121}
2122
6b78788a 2123void ldisc_send(void *handle, char *buf, int len, int interactive)
32874aea 2124{
4a8fc3c4 2125 /*
2126 * This is only here because of the calls to ldisc_send(NULL,
2127 * 0) in ssh.c. Nothing in PSFTP actually needs to use the
2128 * ldisc as an ldisc. So if we get called with any real data, I
2129 * want to know about it.
4c7f0d61 2130 */
4a8fc3c4 2131 assert(len == 0);
2132}
2133
2134/*
c44bf5bd 2135 * In psftp, all agent requests should be synchronous, so this is a
2136 * never-called stub.
2137 */
2138void agent_schedule_callback(void (*callback)(void *, void *, int),
2139 void *callback_ctx, void *data, int len)
2140{
2141 assert(!"We shouldn't be here");
2142}
2143
2144/*
4a8fc3c4 2145 * Receive a block of data from the SSH link. Block until all data
2146 * is available.
2147 *
2148 * To do this, we repeatedly call the SSH protocol module, with our
2149 * own trap in from_backend() to catch the data that comes back. We
2150 * do this until we have enough data.
2151 */
2152
32874aea 2153static unsigned char *outptr; /* where to put the data */
2154static unsigned outlen; /* how much data required */
4a8fc3c4 2155static unsigned char *pending = NULL; /* any spare data */
32874aea 2156static unsigned pendlen = 0, pendsize = 0; /* length and phys. size of buffer */
9fab77dc 2157int from_backend(void *frontend, int is_stderr, const char *data, int datalen)
32874aea 2158{
2159 unsigned char *p = (unsigned char *) data;
2160 unsigned len = (unsigned) datalen;
4a8fc3c4 2161
2162 /*
2163 * stderr data is just spouted to local stderr and otherwise
2164 * ignored.
2165 */
2166 if (is_stderr) {
bfa5400d 2167 if (len > 0)
2168 fwrite(data, 1, len, stderr);
5471d09a 2169 return 0;
4a8fc3c4 2170 }
2171
2172 /*
2173 * If this is before the real session begins, just return.
2174 */
2175 if (!outptr)
5471d09a 2176 return 0;
4a8fc3c4 2177
bfa5400d 2178 if ((outlen > 0) && (len > 0)) {
32874aea 2179 unsigned used = outlen;
2180 if (used > len)
2181 used = len;
2182 memcpy(outptr, p, used);
2183 outptr += used;
2184 outlen -= used;
2185 p += used;
2186 len -= used;
4a8fc3c4 2187 }
2188
2189 if (len > 0) {
32874aea 2190 if (pendsize < pendlen + len) {
2191 pendsize = pendlen + len + 4096;
3d88e64d 2192 pending = sresize(pending, pendsize, unsigned char);
32874aea 2193 }
2194 memcpy(pending + pendlen, p, len);
2195 pendlen += len;
4a8fc3c4 2196 }
5471d09a 2197
2198 return 0;
4a8fc3c4 2199}
32874aea 2200int sftp_recvdata(char *buf, int len)
2201{
2202 outptr = (unsigned char *) buf;
4a8fc3c4 2203 outlen = len;
2204
2205 /*
2206 * See if the pending-input block contains some of what we
2207 * need.
2208 */
2209 if (pendlen > 0) {
32874aea 2210 unsigned pendused = pendlen;
2211 if (pendused > outlen)
2212 pendused = outlen;
4a8fc3c4 2213 memcpy(outptr, pending, pendused);
32874aea 2214 memmove(pending, pending + pendused, pendlen - pendused);
4a8fc3c4 2215 outptr += pendused;
2216 outlen -= pendused;
32874aea 2217 pendlen -= pendused;
2218 if (pendlen == 0) {
2219 pendsize = 0;
2220 sfree(pending);
2221 pending = NULL;
2222 }
2223 if (outlen == 0)
2224 return 1;
4a8fc3c4 2225 }
2226
2227 while (outlen > 0) {
d6cc41e6 2228 if (ssh_sftp_loop_iteration() < 0)
32874aea 2229 return 0; /* doom */
4a8fc3c4 2230 }
2231
2232 return 1;
2233}
32874aea 2234int sftp_senddata(char *buf, int len)
2235{
776792d7 2236 back->send(backhandle, buf, len);
4a8fc3c4 2237 return 1;
2238}
2239
2240/*
4a8fc3c4 2241 * Short description of parameters.
2242 */
2243static void usage(void)
2244{
2245 printf("PuTTY Secure File Transfer (SFTP) client\n");
2246 printf("%s\n", ver);
90767715 2247 printf("Usage: psftp [options] [user@]host\n");
4a8fc3c4 2248 printf("Options:\n");
9954aaa3 2249 printf(" -b file use specified batchfile\n");
2250 printf(" -bc output batchfile commands\n");
2251 printf(" -be don't stop batchfile processing if errors\n");
4a8fc3c4 2252 printf(" -v show verbose messages\n");
e2a197cf 2253 printf(" -load sessname Load settings from saved session\n");
2254 printf(" -l user connect with specified username\n");
4a8fc3c4 2255 printf(" -P port connect to specified port\n");
2256 printf(" -pw passw login with specified password\n");
e2a197cf 2257 printf(" -1 -2 force use of particular SSH protocol version\n");
2258 printf(" -C enable compression\n");
2259 printf(" -i key private key file for authentication\n");
2260 printf(" -batch disable all interactive prompts\n");
dc108ebc 2261 printf(" -V print version information\n");
93b581bd 2262 cleanup_exit(1);
4a8fc3c4 2263}
2264
dc108ebc 2265static void version(void)
2266{
2267 printf("psftp: %s\n", ver);
2268 cleanup_exit(1);
2269}
2270
4a8fc3c4 2271/*
fa3db767 2272 * Connect to a host.
4a8fc3c4 2273 */
fa3db767 2274static int psftp_connect(char *userhost, char *user, int portnumber)
4a8fc3c4 2275{
fa3db767 2276 char *host, *realhost;
cbe2d68f 2277 const char *err;
b51259f6 2278 void *logctx;
4a8fc3c4 2279
2280 /* Separate host and username */
2281 host = userhost;
2282 host = strrchr(host, '@');
2283 if (host == NULL) {
2284 host = userhost;
2285 } else {
2286 *host++ = '\0';
2287 if (user) {
32874aea 2288 printf("psftp: multiple usernames specified; using \"%s\"\n",
2289 user);
4a8fc3c4 2290 } else
2291 user = userhost;
2292 }
2293
18e62ad8 2294 /*
2295 * If we haven't loaded session details already (e.g., from -load),
2296 * try looking for a session called "host".
2297 */
2298 if (!loaded_session) {
2299 /* Try to load settings for `host' into a temporary config */
2300 Config cfg2;
2301 cfg2.host[0] = '\0';
2302 do_defaults(host, &cfg2);
2303 if (cfg2.host[0] != '\0') {
2304 /* Settings present and include hostname */
2305 /* Re-load data into the real config. */
2306 do_defaults(host, &cfg);
2307 } else {
2308 /* Session doesn't exist or mention a hostname. */
2309 /* Use `host' as a bare hostname. */
2310 strncpy(cfg.host, host, sizeof(cfg.host) - 1);
2311 cfg.host[sizeof(cfg.host) - 1] = '\0';
2312 }
2313 } else {
2314 /* Patch in hostname `host' to session details. */
32874aea 2315 strncpy(cfg.host, host, sizeof(cfg.host) - 1);
2316 cfg.host[sizeof(cfg.host) - 1] = '\0';
f133db8e 2317 }
2318
2319 /*
2320 * Force use of SSH. (If they got the protocol wrong we assume the
2321 * port is useless too.)
2322 */
2323 if (cfg.protocol != PROT_SSH) {
2324 cfg.protocol = PROT_SSH;
2325 cfg.port = 22;
4a8fc3c4 2326 }
2327
449925a6 2328 /*
4123fa9a 2329 * If saved session / Default Settings says SSH-1 (`1 only' or `1'),
2330 * then change it to SSH-2, on the grounds that that's more likely to
2331 * work for SFTP. (Can be overridden with `-1' option.)
2332 * But if it says `2 only' or `2', respect which.
2333 */
2334 if (cfg.sshprot != 2 && cfg.sshprot != 3)
2335 cfg.sshprot = 2;
2336
2337 /*
c0a81592 2338 * Enact command-line overrides.
2339 */
5555d393 2340 cmdline_run_saved(&cfg);
c0a81592 2341
2342 /*
449925a6 2343 * Trim leading whitespace off the hostname if it's there.
2344 */
2345 {
2346 int space = strspn(cfg.host, " \t");
2347 memmove(cfg.host, cfg.host+space, 1+strlen(cfg.host)-space);
2348 }
2349
2350 /* See if host is of the form user@host */
2351 if (cfg.host[0] != '\0') {
5dd103a8 2352 char *atsign = strrchr(cfg.host, '@');
449925a6 2353 /* Make sure we're not overflowing the user field */
2354 if (atsign) {
2355 if (atsign - cfg.host < sizeof cfg.username) {
2356 strncpy(cfg.username, cfg.host, atsign - cfg.host);
2357 cfg.username[atsign - cfg.host] = '\0';
2358 }
2359 memmove(cfg.host, atsign + 1, 1 + strlen(atsign + 1));
2360 }
2361 }
2362
2363 /*
2364 * Trim a colon suffix off the hostname if it's there.
2365 */
2366 cfg.host[strcspn(cfg.host, ":")] = '\0';
2367
cae0c023 2368 /*
2369 * Remove any remaining whitespace from the hostname.
2370 */
2371 {
2372 int p1 = 0, p2 = 0;
2373 while (cfg.host[p2] != '\0') {
2374 if (cfg.host[p2] != ' ' && cfg.host[p2] != '\t') {
2375 cfg.host[p1] = cfg.host[p2];
2376 p1++;
2377 }
2378 p2++;
2379 }
2380 cfg.host[p1] = '\0';
2381 }
2382
4a8fc3c4 2383 /* Set username */
2384 if (user != NULL && user[0] != '\0') {
32874aea 2385 strncpy(cfg.username, user, sizeof(cfg.username) - 1);
2386 cfg.username[sizeof(cfg.username) - 1] = '\0';
4a8fc3c4 2387 }
2388 if (!cfg.username[0]) {
b5af97b3 2389 if (!console_get_line("login as: ",
2390 cfg.username, sizeof(cfg.username), FALSE)) {
2391 fprintf(stderr, "psftp: no username, aborting\n");
93b581bd 2392 cleanup_exit(1);
4a8fc3c4 2393 } else {
2394 int len = strlen(cfg.username);
32874aea 2395 if (cfg.username[len - 1] == '\n')
2396 cfg.username[len - 1] = '\0';
4a8fc3c4 2397 }
2398 }
2399
4a8fc3c4 2400 if (portnumber)
2401 cfg.port = portnumber;
2402
d27b4a18 2403 /*
2404 * Disable scary things which shouldn't be enabled for simple
2405 * things like SCP and SFTP: agent forwarding, port forwarding,
2406 * X forwarding.
2407 */
2408 cfg.x11_forward = 0;
2409 cfg.agentfwd = 0;
2410 cfg.portfwd[0] = cfg.portfwd[1] = '\0';
2411
bebf22d0 2412 /* Set up subsystem name. */
4a8fc3c4 2413 strcpy(cfg.remote_cmd, "sftp");
2414 cfg.ssh_subsys = TRUE;
2415 cfg.nopty = TRUE;
2416
bebf22d0 2417 /*
2418 * Set up fallback option, for SSH1 servers or servers with the
2419 * sftp subsystem not enabled but the server binary installed
2420 * in the usual place. We only support fallback on Unix
248c0c5a 2421 * systems, and we use a kludgy piece of shellery which should
2422 * try to find sftp-server in various places (the obvious
2423 * systemwide spots /usr/lib and /usr/local/lib, and then the
2424 * user's PATH) and finally give up.
bebf22d0 2425 *
248c0c5a 2426 * test -x /usr/lib/sftp-server && exec /usr/lib/sftp-server
2427 * test -x /usr/local/lib/sftp-server && exec /usr/local/lib/sftp-server
2428 * exec sftp-server
bebf22d0 2429 *
2430 * the idea being that this will attempt to use either of the
2431 * obvious pathnames and then give up, and when it does give up
2432 * it will print the preferred pathname in the error messages.
2433 */
2434 cfg.remote_cmd_ptr2 =
248c0c5a 2435 "test -x /usr/lib/sftp-server && exec /usr/lib/sftp-server\n"
2436 "test -x /usr/local/lib/sftp-server && exec /usr/local/lib/sftp-server\n"
2437 "exec sftp-server";
bebf22d0 2438 cfg.ssh_subsys2 = FALSE;
2439
4a8fc3c4 2440 back = &ssh_backend;
2441
79bf227b 2442 err = back->init(NULL, &backhandle, &cfg, cfg.host, cfg.port, &realhost,
2443 0, cfg.tcp_keepalives);
4a8fc3c4 2444 if (err != NULL) {
fa3db767 2445 fprintf(stderr, "ssh_init: %s\n", err);
4a8fc3c4 2446 return 1;
2447 }
c229ef97 2448 logctx = log_init(NULL, &cfg);
a8327734 2449 back->provide_logctx(backhandle, logctx);
d3fef4a5 2450 console_provide_logctx(logctx);
d6cc41e6 2451 while (!back->sendok(backhandle)) {
2452 if (ssh_sftp_loop_iteration() < 0) {
2453 fprintf(stderr, "ssh_init: error during SSH connection setup\n");
2454 return 1;
2455 }
2456 }
4a8fc3c4 2457 if (verbose && realhost != NULL)
2458 printf("Connected to %s\n", realhost);
679539d7 2459 if (realhost != NULL)
2460 sfree(realhost);
fa3db767 2461 return 0;
2462}
2463
c0a81592 2464void cmdline_error(char *p, ...)
2465{
2466 va_list ap;
86256dc6 2467 fprintf(stderr, "psftp: ");
c0a81592 2468 va_start(ap, p);
2469 vfprintf(stderr, p, ap);
2470 va_end(ap);
86256dc6 2471 fprintf(stderr, "\n try typing \"psftp -h\" for help\n");
c0a81592 2472 exit(1);
2473}
2474
fa3db767 2475/*
2476 * Main program. Parse arguments etc.
2477 */
d6cc41e6 2478int psftp_main(int argc, char *argv[])
fa3db767 2479{
2480 int i;
2481 int portnumber = 0;
2482 char *userhost, *user;
2483 int mode = 0;
2484 int modeflags = 0;
2485 char *batchfile = NULL;
86256dc6 2486 int errors = 0;
fa3db767 2487
b51259f6 2488 flags = FLAG_STDERR | FLAG_INTERACTIVE
2489#ifdef FLAG_SYNCAGENT
2490 | FLAG_SYNCAGENT
2491#endif
2492 ;
c0a81592 2493 cmdline_tooltype = TOOLTYPE_FILETRANSFER;
ff2ae367 2494 ssh_get_line = &console_get_line;
fa3db767 2495 sk_init();
2496
2497 userhost = user = NULL;
2498
18e62ad8 2499 /* Load Default Settings before doing anything else. */
2500 do_defaults(NULL, &cfg);
2501 loaded_session = FALSE;
2502
86256dc6 2503 errors = 0;
fa3db767 2504 for (i = 1; i < argc; i++) {
c0a81592 2505 int ret;
fa3db767 2506 if (argv[i][0] != '-') {
c0a81592 2507 if (userhost)
2508 usage();
2509 else
2510 userhost = dupstr(argv[i]);
2511 continue;
2512 }
5555d393 2513 ret = cmdline_process_param(argv[i], i+1<argc?argv[i+1]:NULL, 1, &cfg);
c0a81592 2514 if (ret == -2) {
2515 cmdline_error("option \"%s\" requires an argument", argv[i]);
2516 } else if (ret == 2) {
2517 i++; /* skip next argument */
2518 } else if (ret == 1) {
2519 /* We have our own verbosity in addition to `flags'. */
2520 if (flags & FLAG_VERBOSE)
2521 verbose = 1;
fa3db767 2522 } else if (strcmp(argv[i], "-h") == 0 ||
2523 strcmp(argv[i], "-?") == 0) {
2524 usage();
dc108ebc 2525 } else if (strcmp(argv[i], "-V") == 0) {
2526 version();
c0a81592 2527 } else if (strcmp(argv[i], "-batch") == 0) {
2528 console_batch_mode = 1;
fa3db767 2529 } else if (strcmp(argv[i], "-b") == 0 && i + 1 < argc) {
2530 mode = 1;
2531 batchfile = argv[++i];
d13c2ee9 2532 } else if (strcmp(argv[i], "-bc") == 0) {
fa3db767 2533 modeflags = modeflags | 1;
d13c2ee9 2534 } else if (strcmp(argv[i], "-be") == 0) {
fa3db767 2535 modeflags = modeflags | 2;
2536 } else if (strcmp(argv[i], "--") == 0) {
2537 i++;
2538 break;
2539 } else {
86256dc6 2540 cmdline_error("unknown option \"%s\"", argv[i]);
fa3db767 2541 }
2542 }
2543 argc -= i;
2544 argv += i;
2545 back = NULL;
2546
2547 /*
e1bb41d1 2548 * If the loaded session provides a hostname, and a hostname has not
2549 * otherwise been specified, pop it in `userhost' so that
2550 * `psftp -load sessname' is sufficient to start a session.
2551 */
2552 if (!userhost && cfg.host[0] != '\0') {
2553 userhost = dupstr(cfg.host);
2554 }
2555
2556 /*
fa3db767 2557 * If a user@host string has already been provided, connect to
2558 * it now.
2559 */
2560 if (userhost) {
679539d7 2561 int ret;
2562 ret = psftp_connect(userhost, user, portnumber);
2563 sfree(userhost);
2564 if (ret)
fa3db767 2565 return 1;
774204f5 2566 if (do_sftp_init())
2567 return 1;
fa3db767 2568 } else {
2569 printf("psftp: no hostname specified; use \"open host.name\""
679539d7 2570 " to connect\n");
fa3db767 2571 }
4c7f0d61 2572
9954aaa3 2573 do_sftp(mode, modeflags, batchfile);
4a8fc3c4 2574
51470298 2575 if (back != NULL && back->socket(backhandle) != NULL) {
4a8fc3c4 2576 char ch;
51470298 2577 back->special(backhandle, TS_EOF);
4a8fc3c4 2578 sftp_recvdata(&ch, 1);
2579 }
4a8fc3c4 2580 random_save_seed();
679539d7 2581 cmdline_cleanup();
2582 console_provide_logctx(NULL);
2583 do_sftp_cleanup();
2584 backhandle = NULL;
2585 back = NULL;
2586 sk_cleanup();
4a8fc3c4 2587
4c7f0d61 2588 return 0;
2589}