]> asedeno.scripts.mit.edu Git - PuTTY.git/blob - psftp.c
Jacob points out that I introduced a bug in PSFTP when I did the
[PuTTY.git] / psftp.c
1 /*
2  * psftp.c: (platform-independent) front end for PSFTP.
3  */
4
5 #include <stdio.h>
6 #include <stdlib.h>
7 #include <stdarg.h>
8 #include <assert.h>
9 #include <limits.h>
10
11 #define PUTTY_DO_GLOBALS
12 #include "putty.h"
13 #include "psftp.h"
14 #include "storage.h"
15 #include "ssh.h"
16 #include "sftp.h"
17 #include "int64.h"
18
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
26 static int psftp_connect(char *userhost, char *user, int portnumber);
27 static int do_sftp_init(void);
28 void do_sftp_cleanup();
29
30 /* ----------------------------------------------------------------------
31  * sftp client state.
32  */
33
34 char *pwd, *homedir;
35 static Backend *back;
36 static void *backhandle;
37 static Config cfg;
38
39 /* ----------------------------------------------------------------------
40  * Higher-level helper functions used in commands.
41  */
42
43 /*
44  * Determine whether a string is entirely composed of dots.
45  */
46 static int is_dots(char *str)
47 {
48     return str[strspn(str, ".")] == '\0';
49 }
50
51 /*
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).
55  */
56 char *canonify(char *name)
57 {
58     char *fullname, *canonname;
59     struct sftp_packet *pktin;
60     struct sftp_request *req, *rreq;
61
62     if (name[0] == '/') {
63         fullname = dupstr(name);
64     } else {
65         char *slash;
66         if (pwd[strlen(pwd) - 1] == '/')
67             slash = "";
68         else
69             slash = "/";
70         fullname = dupcat(pwd, slash, name, NULL);
71     }
72
73     sftp_register(req = fxp_realpath_send(fullname));
74     rreq = sftp_find_request(pktin = sftp_recv());
75     assert(rreq == req);
76     canonname = fxp_realpath_recv(pktin, rreq);
77
78     if (canonname) {
79         sfree(fullname);
80         return canonname;
81     } else {
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) {
131             sftp_register(req = fxp_realpath_send("/"));
132         } else {
133             sftp_register(req = fxp_realpath_send(fullname));
134         }
135         rreq = sftp_find_request(pktin = sftp_recv());
136         assert(rreq == req);
137         canonname = fxp_realpath_recv(pktin, rreq);
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;
152     }
153 }
154
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  */
159 static 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
179 /*
180  * qsort comparison routine for fxp_name structures. Sorts by real
181  * file name.
182  */
183 static int sftp_name_compare(const void *av, const void *bv)
184 {
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);
188 }
189
190 /*
191  * Likewise, but for a bare char *.
192  */
193 static int bare_name_compare(const void *av, const void *bv)
194 {
195     const char **a = (const char **) av;
196     const char **b = (const char **) bv;
197     return strcmp(*a, *b);
198 }
199
200 /* ----------------------------------------------------------------------
201  * The meat of the `get' and `put' commands.
202  */
203 int sftp_get_file(char *fname, char *outfname, int recurse, int restart,
204                   char *wildcard)
205 {
206     struct fxp_handle *fh;
207     struct sftp_packet *pktin;
208     struct sftp_request *req, *rreq;
209     struct fxp_xfer *xfer;
210     uint64 offset;
211     FILE *fp;
212     int ret;
213
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      */
219     if (wildcard || recurse) {
220         struct fxp_attrs attrs;
221         int result;
222
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))) {
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,
244              * unless it already exists (or this is a wildcard
245              * run).
246              */
247             if (!wildcard &&
248                 file_type(outfname) != FILE_TYPE_DIRECTORY &&
249                 !create_directory(outfname)) {
250                 printf("%s: Cannot create directory\n", outfname);
251                 return 0;
252             }
253
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++)
294                     if (!is_dots(names->names[i].filename) &&
295                         (!wildcard || wc_match(wildcard,
296                                                names->names[i].filename)))
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             /*
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             /*
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;
334                 if (outfname)
335                     nextoutfname = dir_file_cat(outfname,
336                                                 ournames[i]->filename);
337                 else
338                     nextoutfname = dupstr(ournames[i]->filename);
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);
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);
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         }
388     }
389
390     sftp_register(req = fxp_open_send(fname, SSH_FXF_READ));
391     rreq = sftp_find_request(pktin = sftp_recv());
392     assert(rreq == req);
393     fh = fxp_open_recv(pktin, rreq);
394
395     if (!fh) {
396         printf("%s: %s\n", fname, fxp_error());
397         return 0;
398     }
399
400     if (restart) {
401         fp = fopen(outfname, "rb+");
402     } else {
403         fp = fopen(outfname, "wb");
404     }
405
406     if (!fp) {
407         printf("local: unable to open %s\n", outfname);
408
409         sftp_register(req = fxp_close_send(fh));
410         rreq = sftp_find_request(pktin = sftp_recv());
411         assert(rreq == req);
412         fxp_close_recv(pktin, rreq);
413
414         return 0;
415     }
416
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     }
426
427     printf("remote:%s => local:%s\n", fname, outfname);
428
429     /*
430      * FIXME: we can use FXP_FSTAT here to get the file size, and
431      * thus put up a progress bar.
432      */
433     ret = 1;
434     xfer = xfer_download_init(fh, offset);
435     while (!xfer_done(xfer)) {
436         void *vbuf;
437         int ret, len;
438         int wpos, wlen;
439
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;
447         }
448
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 */
463                 ret = 0;
464                 xfer_set_error(xfer);
465             }
466
467             sfree(vbuf);
468         }
469     }
470
471     xfer_cleanup(xfer);
472
473     fclose(fp);
474
475     sftp_register(req = fxp_close_send(fh));
476     rreq = sftp_find_request(pktin = sftp_recv());
477     assert(rreq == req);
478     fxp_close_recv(pktin, rreq);
479
480     return ret;
481 }
482
483 int sftp_put_file(char *fname, char *outfname, int recurse, int restart,
484                   char *wildcard)
485 {
486     struct fxp_handle *fh;
487     struct fxp_xfer *xfer;
488     struct sftp_packet *pktin;
489     struct sftp_request *req, *rreq;
490     uint64 offset;
491     FILE *fp;
492     int ret, err, eof;
493
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      */
499     if (wildcard || (recurse && file_type(fname) == FILE_TYPE_DIRECTORY)) {
500         struct fxp_attrs attrs;
501         int result;
502         int nnames, namesize;
503         char *name, **ournames;
504         DirHandle *dh;
505         int i;
506
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));
513             rreq = sftp_find_request(pktin = sftp_recv());
514             assert(rreq == req);
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);
523
524                 if (!result) {
525                     printf("%s: create directory: %s\n",
526                            outfname, fxp_error());
527                     return 0;
528                 }
529             }
530         }
531
532         /*
533          * Now get the list of filenames in the local directory.
534          */
535         nnames = namesize = 0;
536         ournames = NULL;
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);
550             }
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);
573         }
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
617             if (fname)
618                 nextfname = dir_file_cat(fname, ournames[i]);
619             else
620                 nextfname = dupstr(ournames[i]);
621             nextoutfname = dupcat(outfname, "/", ournames[i], NULL);
622             ret = sftp_put_file(nextfname, nextoutfname,
623                                 recurse, restart, NULL);
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;
645     }
646
647     fp = fopen(fname, "rb");
648     if (!fp) {
649         printf("local: unable to open %s\n", fname);
650         return 0;
651     }
652     if (restart) {
653         sftp_register(req = fxp_open_send(outfname, SSH_FXF_WRITE));
654     } else {
655         sftp_register(req = fxp_open_send(outfname, SSH_FXF_WRITE |
656                                           SSH_FXF_CREAT | SSH_FXF_TRUNC));
657     }
658     rreq = sftp_find_request(pktin = sftp_recv());
659     assert(rreq == req);
660     fh = fxp_open_recv(pktin, rreq);
661
662     if (!fh) {
663         printf("%s: %s\n", outfname, fxp_error());
664         return 0;
665     }
666
667     if (restart) {
668         char decbuf[30];
669         struct fxp_attrs attrs;
670         int ret;
671
672         sftp_register(req = fxp_fstat_send(fh));
673         rreq = sftp_find_request(pktin = sftp_recv());
674         assert(rreq == req);
675         ret = fxp_fstat_recv(pktin, rreq, &attrs);
676
677         if (!ret) {
678             printf("read size of %s: %s\n", outfname, fxp_error());
679             return 0;
680         }
681         if (!(attrs.flags & SSH_FILEXFER_ATTR_SIZE)) {
682             printf("read size of %s: size was not given\n", outfname);
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");
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     }
697
698     printf("local:%s => remote:%s\n", fname, outfname);
699
700     /*
701      * FIXME: we can use FXP_FSTAT here to get the file size, and
702      * thus put up a progress bar.
703      */
704     ret = 1;
705     xfer = xfer_upload_init(fh, offset);
706     err = eof = 0;
707     while ((!err && !eof) || !xfer_done(xfer)) {
708         char buffer[4096];
709         int len, ret;
710
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             }
721         }
722
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             }
730         }
731     }
732
733     xfer_cleanup(xfer);
734
735     sftp_register(req = fxp_close_send(fh));
736     rreq = sftp_find_request(pktin = sftp_recv());
737     assert(rreq == req);
738     fxp_close_recv(pktin, rreq);
739
740     fclose(fp);
741
742     return ret;
743 }
744
745 /* ----------------------------------------------------------------------
746  * Actual sftp commands.
747  */
748 struct sftp_command {
749     char **words;
750     int nwords, wordssize;
751     int (*obey) (struct sftp_command *);        /* returns <0 to quit */
752 };
753
754 int sftp_cmd_null(struct sftp_command *cmd)
755 {
756     return 1;                          /* success */
757 }
758
759 int sftp_cmd_unknown(struct sftp_command *cmd)
760 {
761     printf("psftp: unknown command \"%s\"\n", cmd->words[0]);
762     return 0;                          /* failure */
763 }
764
765 int 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  */
774 int 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;
780     char *dir, *cdir, *unwcdir, *wildcard;
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
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
820     cdir = canonify(dir);
821     if (!cdir) {
822         printf("%s: %s\n", dir, fxp_error());
823         sfree(unwcdir);
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++)
864                 if (!wildcard || wc_match(wildcard, names->names[i].filename))
865                     ournames[nnames++] = fxp_dup_name(&names->names[i]);
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);
891     sfree(unwcdir);
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  */
900 int 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  */
948 int 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 /*
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.
967  */
968 int sftp_general_get(struct sftp_command *cmd, int restart, int multiple)
969 {
970     char *fname, *unwcfname, *origfname, *outfname;
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
999     do {
1000         unwcfname = NULL;
1001         origfname = cmd->words[i++];
1002
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             }
1014
1015             if (!multiple && i < cmd->nwords)
1016                 outfname = cmd->words[i++];
1017             else
1018                 outfname = stripslashes(origfname, 0);
1019
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);
1029
1030     return ret;
1031 }
1032 int sftp_cmd_get(struct sftp_command *cmd)
1033 {
1034     return sftp_general_get(cmd, 0, 0);
1035 }
1036 int sftp_cmd_mget(struct sftp_command *cmd)
1037 {
1038     return sftp_general_get(cmd, 0, 1);
1039 }
1040 int sftp_cmd_reget(struct sftp_command *cmd)
1041 {
1042     return sftp_general_get(cmd, 1, 0);
1043 }
1044
1045 /*
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.
1053  */
1054 int sftp_general_put(struct sftp_command *cmd, int restart, int multiple)
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
1085     do {
1086         fname = cmd->words[i++];
1087
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;
1106
1107     } while (multiple && i < cmd->nwords);
1108
1109     return ret;
1110 }
1111 int sftp_cmd_put(struct sftp_command *cmd)
1112 {
1113     return sftp_general_put(cmd, 0, 0);
1114 }
1115 int sftp_cmd_mput(struct sftp_command *cmd)
1116 {
1117     return sftp_general_put(cmd, 0, 1);
1118 }
1119 int sftp_cmd_reput(struct sftp_command *cmd)
1120 {
1121     return sftp_general_put(cmd, 1, 0);
1122 }
1123
1124 int sftp_cmd_mkdir(struct sftp_command *cmd)
1125 {
1126     char *dir;
1127     struct sftp_packet *pktin;
1128     struct sftp_request *req, *rreq;
1129     int result;
1130
1131     if (back == NULL) {
1132         printf("psftp: not connected to a host; use \"open host.name\"\n");
1133         return 0;
1134     }
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
1147     sftp_register(req = fxp_mkdir_send(dir));
1148     rreq = sftp_find_request(pktin = sftp_recv());
1149     assert(rreq == req);
1150     result = fxp_mkdir_recv(pktin, rreq);
1151
1152     if (!result) {
1153         printf("mkdir %s: %s\n", dir, fxp_error());
1154         sfree(dir);
1155         return 0;
1156     }
1157
1158     sfree(dir);
1159     return 1;
1160 }
1161
1162 int sftp_cmd_rmdir(struct sftp_command *cmd)
1163 {
1164     char *dir;
1165     struct sftp_packet *pktin;
1166     struct sftp_request *req, *rreq;
1167     int result;
1168
1169     if (back == NULL) {
1170         printf("psftp: not connected to a host; use \"open host.name\"\n");
1171         return 0;
1172     }
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
1185     sftp_register(req = fxp_rmdir_send(dir));
1186     rreq = sftp_find_request(pktin = sftp_recv());
1187     assert(rreq == req);
1188     result = fxp_rmdir_recv(pktin, rreq);
1189
1190     if (!result) {
1191         printf("rmdir %s: %s\n", dir, fxp_error());
1192         sfree(dir);
1193         return 0;
1194     }
1195
1196     sfree(dir);
1197     return 1;
1198 }
1199
1200 int sftp_cmd_rm(struct sftp_command *cmd)
1201 {
1202     char *fname;
1203     struct sftp_packet *pktin;
1204     struct sftp_request *req, *rreq;
1205     int result;
1206
1207     if (back == NULL) {
1208         printf("psftp: not connected to a host; use \"open host.name\"\n");
1209         return 0;
1210     }
1211
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
1223     sftp_register(req = fxp_remove_send(fname));
1224     rreq = sftp_find_request(pktin = sftp_recv());
1225     assert(rreq == req);
1226     result = fxp_remove_recv(pktin, rreq);
1227
1228     if (!result) {
1229         printf("rm %s: %s\n", fname, fxp_error());
1230         sfree(fname);
1231         return 0;
1232     }
1233
1234     sfree(fname);
1235     return 1;
1236 }
1237
1238 int sftp_cmd_mv(struct sftp_command *cmd)
1239 {
1240     char *srcfname, *dstfname;
1241     struct sftp_packet *pktin;
1242     struct sftp_request *req, *rreq;
1243     int result;
1244
1245     if (back == NULL) {
1246         printf("psftp: not connected to a host; use \"open host.name\"\n");
1247         return 0;
1248     }
1249
1250     if (cmd->nwords < 3) {
1251         printf("mv: expects two filenames\n");
1252         return 0;
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     }
1265
1266     sftp_register(req = fxp_rename_send(srcfname, dstfname));
1267     rreq = sftp_find_request(pktin = sftp_recv());
1268     assert(rreq == req);
1269     result = fxp_rename_recv(pktin, rreq);
1270
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          */
1281         sftp_register(req = fxp_stat_send(dstfname));
1282         rreq = sftp_find_request(pktin = sftp_recv());
1283         assert(rreq == req);
1284         result = fxp_stat_recv(pktin, rreq, &attrs);
1285
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;
1300
1301                 sftp_register(req = fxp_rename_send(srcfname, dstfname));
1302                 rreq = sftp_find_request(pktin = sftp_recv());
1303                 assert(rreq == req);
1304                 result = fxp_rename_recv(pktin, rreq);
1305
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);
1320     return 1;
1321 }
1322
1323 int 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;
1329     struct sftp_packet *pktin;
1330     struct sftp_request *req, *rreq;
1331
1332     if (back == NULL) {
1333         printf("psftp: not connected to a host; use \"open host.name\"\n");
1334         return 0;
1335     }
1336
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",
1381                            (int)strcspn(modebegin, ","), modebegin, *mode);
1382                     return 0;
1383                 }
1384                 mode++;
1385             }
1386             if (!*mode || *mode == ',') {
1387                 printf("chmod: file mode '%.*s' is incomplete\n",
1388                        (int)strcspn(modebegin, ","), modebegin);
1389                 return 0;
1390             }
1391             action = *mode++;
1392             if (!*mode || *mode == ',') {
1393                 printf("chmod: file mode '%.*s' is incomplete\n",
1394                        (int)strcspn(modebegin, ","), modebegin);
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",
1409                                (int)strcspn(modebegin, ","), modebegin);
1410                         return 0;
1411                     }
1412                     perms |= 06000;
1413                     break;
1414                   default:
1415                     printf("chmod: file mode '%.*s' contains unrecognised"
1416                            " permission specifier '%c'\n",
1417                            (int)strcspn(modebegin, ","), modebegin, *mode);
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",
1425                        (int)strcspn(modebegin, ","), modebegin);
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
1453     sftp_register(req = fxp_stat_send(fname));
1454     rreq = sftp_find_request(pktin = sftp_recv());
1455     assert(rreq == req);
1456     result = fxp_stat_recv(pktin, rreq, &attrs);
1457
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
1471     sftp_register(req = fxp_setstat_send(fname, attrs));
1472     rreq = sftp_find_request(pktin = sftp_recv());
1473     assert(rreq == req);
1474     result = fxp_setstat_recv(pktin, rreq);
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);
1485     return 1;
1486 }
1487
1488 static int sftp_cmd_open(struct sftp_command *cmd)
1489 {
1490     int portnumber;
1491
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
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)) {
1512         back = NULL;                   /* connection is already closed */
1513         return -1;                     /* this is fatal */
1514     }
1515     do_sftp_init();
1516     return 1;
1517 }
1518
1519 static int sftp_cmd_lcd(struct sftp_command *cmd)
1520 {
1521     char *currdir, *errmsg;
1522
1523     if (cmd->nwords < 2) {
1524         printf("lcd: expects a local directory name\n");
1525         return 0;
1526     }
1527
1528     errmsg = psftp_lcd(cmd->words[1]);
1529     if (errmsg) {
1530         printf("lcd: unable to change directory: %s\n", errmsg);
1531         sfree(errmsg);
1532         return 0;
1533     }
1534
1535     currdir = psftp_getcwd();
1536     printf("New local directory is %s\n", currdir);
1537     sfree(currdir);
1538
1539     return 1;
1540 }
1541
1542 static int sftp_cmd_lpwd(struct sftp_command *cmd)
1543 {
1544     char *currdir;
1545
1546     currdir = psftp_getcwd();
1547     printf("Current local directory is %s\n", currdir);
1548     sfree(currdir);
1549
1550     return 1;
1551 }
1552
1553 static 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
1561 static int sftp_cmd_help(struct sftp_command *cmd);
1562
1563 static struct sftp_cmd_lookup {
1564     char *name;
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      */
1577     int listed;                        /* do we list this in primary help? */
1578     char *shorthelp;
1579     char *longhelp;
1580     int (*obey) (struct sftp_command *);
1581 } sftp_lookup[] = {
1582     /*
1583      * List of sftp commands. This is binary-searched so it MUST be
1584      * in ASCII order.
1585      */
1586     {
1587         "!", TRUE, "run a local command",
1588             "<command>\n"
1589             /* FIXME: this example is crap for non-Windows. */
1590             "  Runs a local command. For example, \"!del myfile\".\n",
1591             sftp_cmd_pling
1592     },
1593     {
1594         "bye", TRUE, "finish your SFTP session",
1595             "\n"
1596             "  Terminates your SFTP session and quits the PSFTP program.\n",
1597             sftp_cmd_quit
1598     },
1599     {
1600         "cd", TRUE, "change your remote working directory",
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     {
1608         "chmod", TRUE, "change file permissions and modes",
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     {
1635         "del", TRUE, "delete a file",
1636             " <filename>\n"
1637             "  Delete a file.\n",
1638             sftp_cmd_rm
1639     },
1640     {
1641         "delete", FALSE, "del", NULL, sftp_cmd_rm
1642     },
1643     {
1644         "dir", TRUE, "list contents of a remote directory",
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     {
1652         "exit", TRUE, "bye", NULL, sftp_cmd_quit
1653     },
1654     {
1655         "get", TRUE, "download a file from the server to your local machine",
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     {
1663         "help", TRUE, "give help",
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     {
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,
1686             sftp_cmd_ls
1687     },
1688     {
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     {
1697         "mkdir", TRUE, "create a directory on the remote server",
1698             " <directory-name>\n"
1699             "  Creates a directory with the given name on the server.\n",
1700             sftp_cmd_mkdir
1701     },
1702     {
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     {
1711         "mv", TRUE, "move or rename a file on the remote server",
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     {
1718         "open", TRUE, "connect to a host",
1719             " [<user>@]<hostname> [<port>]\n"
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     {
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     {
1734         "pwd", TRUE, "print your remote working directory",
1735             "\n"
1736             "  Print the current remote working directory for your SFTP session.\n",
1737             sftp_cmd_pwd
1738     },
1739     {
1740         "quit", TRUE, "bye", NULL,
1741             sftp_cmd_quit
1742     },
1743     {
1744         "reget", TRUE, "continue downloading a file",
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     {
1752         "ren", TRUE, "mv", NULL,
1753             sftp_cmd_mv
1754     },
1755     {
1756         "rename", FALSE, "mv", NULL,
1757             sftp_cmd_mv
1758     },
1759     {
1760         "reput", TRUE, "continue uploading a file",
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     {
1768         "rm", TRUE, "del", NULL,
1769             sftp_cmd_rm
1770     },
1771     {
1772         "rmdir", TRUE, "remove a directory on the remote server",
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
1780 const 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
1800 static 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++) {
1810             int len;
1811             if (!sftp_lookup[i].listed)
1812                 continue;
1813             len = strlen(sftp_lookup[i].name);
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;
1819             if (!sftp_lookup[i].listed)
1820                 continue;
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     }
1844     return 1;
1845 }
1846
1847 /* ----------------------------------------------------------------------
1848  * Command line reading and parsing.
1849  */
1850 struct sftp_command *sftp_getcmd(FILE *fp, int mode, int modeflags)
1851 {
1852     char *line;
1853     struct sftp_command *cmd;
1854     char *p, *q, *r;
1855     int quoting;
1856
1857     cmd = snew(struct sftp_command);
1858     cmd->words = NULL;
1859     cmd->nwords = 0;
1860     cmd->wordssize = 0;
1861
1862     line = NULL;
1863
1864     if (fp) {
1865         if (modeflags & 1)
1866             printf("psftp> ");
1867         line = fgetline(fp);
1868     } else {
1869         line = ssh_sftp_get_cmdline("psftp> ", back == NULL);
1870     }
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
1881     if (modeflags & 1) {
1882         printf("%s\n", line);
1883     }
1884
1885     p = line;
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;
1896         cmd->words = sresize(cmd->words, cmd->wordssize, char *);
1897         cmd->words[0] = dupstr("!");
1898         cmd->words[1] = dupstr(p+1);
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          */
1917         while (*p) {
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;
1939                 cmd->words = sresize(cmd->words, cmd->wordssize, char *);
1940             }
1941             cmd->words[cmd->nwords++] = dupstr(q);
1942         }
1943     }
1944
1945     sfree(line);
1946
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 {
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;
1960     }
1961
1962     return cmd;
1963 }
1964
1965 static int do_sftp_init(void)
1966 {
1967     struct sftp_packet *pktin;
1968     struct sftp_request *req, *rreq;
1969
1970     /*
1971      * Do protocol initialisation. 
1972      */
1973     if (!fxp_init()) {
1974         fprintf(stderr,
1975                 "Fatal: unable to initialise SFTP: %s\n", fxp_error());
1976         return 1;                      /* failure */
1977     }
1978
1979     /*
1980      * Find out where our home directory is.
1981      */
1982     sftp_register(req = fxp_realpath_send("."));
1983     rreq = sftp_find_request(pktin = sftp_recv());
1984     assert(rreq == req);
1985     homedir = fxp_realpath_recv(pktin, rreq);
1986
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);
1996     return 0;
1997 }
1998
1999 void do_sftp_cleanup()
2000 {
2001     char ch;
2002     if (back) {
2003         back->special(backhandle, TS_EOF);
2004         sftp_recvdata(&ch, 1);
2005         back->free(backhandle);
2006         sftp_cleanup_request();
2007         back = NULL;
2008         backhandle = NULL;
2009     }
2010     if (pwd) {
2011         sfree(pwd);
2012         pwd = NULL;
2013     }
2014     if (homedir) {
2015         sfree(homedir);
2016         homedir = NULL;
2017     }
2018 }
2019
2020 void do_sftp(int mode, int modeflags, char *batchfile)
2021 {
2022     FILE *fp;
2023     int ret;
2024
2025     /*
2026      * Batch mode?
2027      */
2028     if (mode == 0) {
2029
2030         /* ------------------------------------------------------------------
2031          * Now we're ready to do Real Stuff.
2032          */
2033         while (1) {
2034             struct sftp_command *cmd;
2035             cmd = sftp_getcmd(NULL, 0, 0);
2036             if (!cmd)
2037                 break;
2038             ret = cmd->obey(cmd);
2039             if (cmd->words) {
2040                 int i;
2041                 for(i = 0; i < cmd->nwords; i++)
2042                     sfree(cmd->words[i]);
2043                 sfree(cmd->words);
2044             }
2045             sfree(cmd);
2046             if (ret < 0)
2047                 break;
2048         }
2049     } else {
2050         fp = fopen(batchfile, "r");
2051         if (!fp) {
2052             printf("Fatal: unable to open %s\n", batchfile);
2053             return;
2054         }
2055         while (1) {
2056             struct sftp_command *cmd;
2057             cmd = sftp_getcmd(fp, mode, modeflags);
2058             if (!cmd)
2059                 break;
2060             ret = cmd->obey(cmd);
2061             if (ret < 0)
2062                 break;
2063             if (ret == 0) {
2064                 if (!(modeflags & 2))
2065                     break;
2066             }
2067         }
2068         fclose(fp);
2069
2070     }
2071 }
2072
2073 /* ----------------------------------------------------------------------
2074  * Dirty bits: integration with PuTTY.
2075  */
2076
2077 static int verbose = 0;
2078
2079 /*
2080  *  Print an error message and perform a fatal exit.
2081  */
2082 void fatalbox(char *fmt, ...)
2083 {
2084     char *str, *str2;
2085     va_list ap;
2086     va_start(ap, fmt);
2087     str = dupvprintf(fmt, ap);
2088     str2 = dupcat("Fatal: ", str, "\n", NULL);
2089     sfree(str);
2090     va_end(ap);
2091     fputs(str2, stderr);
2092     sfree(str2);
2093
2094     cleanup_exit(1);
2095 }
2096 void modalfatalbox(char *fmt, ...)
2097 {
2098     char *str, *str2;
2099     va_list ap;
2100     va_start(ap, fmt);
2101     str = dupvprintf(fmt, ap);
2102     str2 = dupcat("Fatal: ", str, "\n", NULL);
2103     sfree(str);
2104     va_end(ap);
2105     fputs(str2, stderr);
2106     sfree(str2);
2107
2108     cleanup_exit(1);
2109 }
2110 void connection_fatal(void *frontend, char *fmt, ...)
2111 {
2112     char *str, *str2;
2113     va_list ap;
2114     va_start(ap, fmt);
2115     str = dupvprintf(fmt, ap);
2116     str2 = dupcat("Fatal: ", str, "\n", NULL);
2117     sfree(str);
2118     va_end(ap);
2119     fputs(str2, stderr);
2120     sfree(str2);
2121
2122     cleanup_exit(1);
2123 }
2124
2125 void ldisc_send(void *handle, char *buf, int len, int interactive)
2126 {
2127     /*
2128      * This is only here because of the calls to ldisc_send(NULL,
2129      * 0) in ssh.c. Nothing in PSFTP actually needs to use the
2130      * ldisc as an ldisc. So if we get called with any real data, I
2131      * want to know about it.
2132      */
2133     assert(len == 0);
2134 }
2135
2136 /*
2137  * In psftp, all agent requests should be synchronous, so this is a
2138  * never-called stub.
2139  */
2140 void agent_schedule_callback(void (*callback)(void *, void *, int),
2141                              void *callback_ctx, void *data, int len)
2142 {
2143     assert(!"We shouldn't be here");
2144 }
2145
2146 /*
2147  * Receive a block of data from the SSH link. Block until all data
2148  * is available.
2149  *
2150  * To do this, we repeatedly call the SSH protocol module, with our
2151  * own trap in from_backend() to catch the data that comes back. We
2152  * do this until we have enough data.
2153  */
2154
2155 static unsigned char *outptr;          /* where to put the data */
2156 static unsigned outlen;                /* how much data required */
2157 static unsigned char *pending = NULL;  /* any spare data */
2158 static unsigned pendlen = 0, pendsize = 0;      /* length and phys. size of buffer */
2159 int from_backend(void *frontend, int is_stderr, const char *data, int datalen)
2160 {
2161     unsigned char *p = (unsigned char *) data;
2162     unsigned len = (unsigned) datalen;
2163
2164     /*
2165      * stderr data is just spouted to local stderr and otherwise
2166      * ignored.
2167      */
2168     if (is_stderr) {
2169         if (len > 0)
2170             fwrite(data, 1, len, stderr);
2171         return 0;
2172     }
2173
2174     /*
2175      * If this is before the real session begins, just return.
2176      */
2177     if (!outptr)
2178         return 0;
2179
2180     if ((outlen > 0) && (len > 0)) {
2181         unsigned used = outlen;
2182         if (used > len)
2183             used = len;
2184         memcpy(outptr, p, used);
2185         outptr += used;
2186         outlen -= used;
2187         p += used;
2188         len -= used;
2189     }
2190
2191     if (len > 0) {
2192         if (pendsize < pendlen + len) {
2193             pendsize = pendlen + len + 4096;
2194             pending = sresize(pending, pendsize, unsigned char);
2195         }
2196         memcpy(pending + pendlen, p, len);
2197         pendlen += len;
2198     }
2199
2200     return 0;
2201 }
2202 int sftp_recvdata(char *buf, int len)
2203 {
2204     outptr = (unsigned char *) buf;
2205     outlen = len;
2206
2207     /*
2208      * See if the pending-input block contains some of what we
2209      * need.
2210      */
2211     if (pendlen > 0) {
2212         unsigned pendused = pendlen;
2213         if (pendused > outlen)
2214             pendused = outlen;
2215         memcpy(outptr, pending, pendused);
2216         memmove(pending, pending + pendused, pendlen - pendused);
2217         outptr += pendused;
2218         outlen -= pendused;
2219         pendlen -= pendused;
2220         if (pendlen == 0) {
2221             pendsize = 0;
2222             sfree(pending);
2223             pending = NULL;
2224         }
2225         if (outlen == 0)
2226             return 1;
2227     }
2228
2229     while (outlen > 0) {
2230         if (ssh_sftp_loop_iteration() < 0)
2231             return 0;                  /* doom */
2232     }
2233
2234     return 1;
2235 }
2236 int sftp_senddata(char *buf, int len)
2237 {
2238     back->send(backhandle, buf, len);
2239     return 1;
2240 }
2241
2242 /*
2243  *  Short description of parameters.
2244  */
2245 static void usage(void)
2246 {
2247     printf("PuTTY Secure File Transfer (SFTP) client\n");
2248     printf("%s\n", ver);
2249     printf("Usage: psftp [options] [user@]host\n");
2250     printf("Options:\n");
2251     printf("  -b file   use specified batchfile\n");
2252     printf("  -bc       output batchfile commands\n");
2253     printf("  -be       don't stop batchfile processing if errors\n");
2254     printf("  -v        show verbose messages\n");
2255     printf("  -load sessname  Load settings from saved session\n");
2256     printf("  -l user   connect with specified username\n");
2257     printf("  -P port   connect to specified port\n");
2258     printf("  -pw passw login with specified password\n");
2259     printf("  -1 -2     force use of particular SSH protocol version\n");
2260     printf("  -C        enable compression\n");
2261     printf("  -i key    private key file for authentication\n");
2262     printf("  -batch    disable all interactive prompts\n");
2263     printf("  -V        print version information\n");
2264     cleanup_exit(1);
2265 }
2266
2267 static void version(void)
2268 {
2269   printf("psftp: %s\n", ver);
2270   cleanup_exit(1);
2271 }
2272
2273 /*
2274  * Connect to a host.
2275  */
2276 static int psftp_connect(char *userhost, char *user, int portnumber)
2277 {
2278     char *host, *realhost;
2279     const char *err;
2280     void *logctx;
2281
2282     /* Separate host and username */
2283     host = userhost;
2284     host = strrchr(host, '@');
2285     if (host == NULL) {
2286         host = userhost;
2287     } else {
2288         *host++ = '\0';
2289         if (user) {
2290             printf("psftp: multiple usernames specified; using \"%s\"\n",
2291                    user);
2292         } else
2293             user = userhost;
2294     }
2295
2296     /*
2297      * If we haven't loaded session details already (e.g., from -load),
2298      * try looking for a session called "host".
2299      */
2300     if (!loaded_session) {
2301         /* Try to load settings for `host' into a temporary config */
2302         Config cfg2;
2303         cfg2.host[0] = '\0';
2304         do_defaults(host, &cfg2);
2305         if (cfg2.host[0] != '\0') {
2306             /* Settings present and include hostname */
2307             /* Re-load data into the real config. */
2308             do_defaults(host, &cfg);
2309         } else {
2310             /* Session doesn't exist or mention a hostname. */
2311             /* Use `host' as a bare hostname. */
2312             strncpy(cfg.host, host, sizeof(cfg.host) - 1);
2313             cfg.host[sizeof(cfg.host) - 1] = '\0';
2314         }
2315     } else {
2316         /* Patch in hostname `host' to session details. */
2317         strncpy(cfg.host, host, sizeof(cfg.host) - 1);
2318         cfg.host[sizeof(cfg.host) - 1] = '\0';
2319     }
2320
2321     /*
2322      * Force use of SSH. (If they got the protocol wrong we assume the
2323      * port is useless too.)
2324      */
2325     if (cfg.protocol != PROT_SSH) {
2326         cfg.protocol = PROT_SSH;
2327         cfg.port = 22;
2328     }
2329
2330     /*
2331      * If saved session / Default Settings says SSH-1 (`1 only' or `1'),
2332      * then change it to SSH-2, on the grounds that that's more likely to
2333      * work for SFTP. (Can be overridden with `-1' option.)
2334      * But if it says `2 only' or `2', respect which.
2335      */
2336     if (cfg.sshprot != 2 && cfg.sshprot != 3)
2337         cfg.sshprot = 2;
2338
2339     /*
2340      * Enact command-line overrides.
2341      */
2342     cmdline_run_saved(&cfg);
2343
2344     /*
2345      * Trim leading whitespace off the hostname if it's there.
2346      */
2347     {
2348         int space = strspn(cfg.host, " \t");
2349         memmove(cfg.host, cfg.host+space, 1+strlen(cfg.host)-space);
2350     }
2351
2352     /* See if host is of the form user@host */
2353     if (cfg.host[0] != '\0') {
2354         char *atsign = strrchr(cfg.host, '@');
2355         /* Make sure we're not overflowing the user field */
2356         if (atsign) {
2357             if (atsign - cfg.host < sizeof cfg.username) {
2358                 strncpy(cfg.username, cfg.host, atsign - cfg.host);
2359                 cfg.username[atsign - cfg.host] = '\0';
2360             }
2361             memmove(cfg.host, atsign + 1, 1 + strlen(atsign + 1));
2362         }
2363     }
2364
2365     /*
2366      * Trim a colon suffix off the hostname if it's there.
2367      */
2368     cfg.host[strcspn(cfg.host, ":")] = '\0';
2369
2370     /*
2371      * Remove any remaining whitespace from the hostname.
2372      */
2373     {
2374         int p1 = 0, p2 = 0;
2375         while (cfg.host[p2] != '\0') {
2376             if (cfg.host[p2] != ' ' && cfg.host[p2] != '\t') {
2377                 cfg.host[p1] = cfg.host[p2];
2378                 p1++;
2379             }
2380             p2++;
2381         }
2382         cfg.host[p1] = '\0';
2383     }
2384
2385     /* Set username */
2386     if (user != NULL && user[0] != '\0') {
2387         strncpy(cfg.username, user, sizeof(cfg.username) - 1);
2388         cfg.username[sizeof(cfg.username) - 1] = '\0';
2389     }
2390     if (!cfg.username[0]) {
2391         if (!console_get_line("login as: ",
2392                               cfg.username, sizeof(cfg.username), FALSE)) {
2393             fprintf(stderr, "psftp: no username, aborting\n");
2394             cleanup_exit(1);
2395         } else {
2396             int len = strlen(cfg.username);
2397             if (cfg.username[len - 1] == '\n')
2398                 cfg.username[len - 1] = '\0';
2399         }
2400     }
2401
2402     if (portnumber)
2403         cfg.port = portnumber;
2404
2405     /*
2406      * Disable scary things which shouldn't be enabled for simple
2407      * things like SCP and SFTP: agent forwarding, port forwarding,
2408      * X forwarding.
2409      */
2410     cfg.x11_forward = 0;
2411     cfg.agentfwd = 0;
2412     cfg.portfwd[0] = cfg.portfwd[1] = '\0';
2413
2414     /* Set up subsystem name. */
2415     strcpy(cfg.remote_cmd, "sftp");
2416     cfg.ssh_subsys = TRUE;
2417     cfg.nopty = TRUE;
2418
2419     /*
2420      * Set up fallback option, for SSH1 servers or servers with the
2421      * sftp subsystem not enabled but the server binary installed
2422      * in the usual place. We only support fallback on Unix
2423      * systems, and we use a kludgy piece of shellery which should
2424      * try to find sftp-server in various places (the obvious
2425      * systemwide spots /usr/lib and /usr/local/lib, and then the
2426      * user's PATH) and finally give up.
2427      * 
2428      *   test -x /usr/lib/sftp-server && exec /usr/lib/sftp-server
2429      *   test -x /usr/local/lib/sftp-server && exec /usr/local/lib/sftp-server
2430      *   exec sftp-server
2431      * 
2432      * the idea being that this will attempt to use either of the
2433      * obvious pathnames and then give up, and when it does give up
2434      * it will print the preferred pathname in the error messages.
2435      */
2436     cfg.remote_cmd_ptr2 =
2437         "test -x /usr/lib/sftp-server && exec /usr/lib/sftp-server\n"
2438         "test -x /usr/local/lib/sftp-server && exec /usr/local/lib/sftp-server\n"
2439         "exec sftp-server";
2440     cfg.ssh_subsys2 = FALSE;
2441
2442     back = &ssh_backend;
2443
2444     err = back->init(NULL, &backhandle, &cfg, cfg.host, cfg.port, &realhost,
2445                      0, cfg.tcp_keepalives);
2446     if (err != NULL) {
2447         fprintf(stderr, "ssh_init: %s\n", err);
2448         return 1;
2449     }
2450     logctx = log_init(NULL, &cfg);
2451     back->provide_logctx(backhandle, logctx);
2452     console_provide_logctx(logctx);
2453     while (!back->sendok(backhandle)) {
2454         if (ssh_sftp_loop_iteration() < 0) {
2455             fprintf(stderr, "ssh_init: error during SSH connection setup\n");
2456             return 1;
2457         }
2458     }
2459     if (verbose && realhost != NULL)
2460         printf("Connected to %s\n", realhost);
2461     if (realhost != NULL)
2462         sfree(realhost);
2463     return 0;
2464 }
2465
2466 void cmdline_error(char *p, ...)
2467 {
2468     va_list ap;
2469     fprintf(stderr, "psftp: ");
2470     va_start(ap, p);
2471     vfprintf(stderr, p, ap);
2472     va_end(ap);
2473     fprintf(stderr, "\n       try typing \"psftp -h\" for help\n");
2474     exit(1);
2475 }
2476
2477 /*
2478  * Main program. Parse arguments etc.
2479  */
2480 int psftp_main(int argc, char *argv[])
2481 {
2482     int i;
2483     int portnumber = 0;
2484     char *userhost, *user;
2485     int mode = 0;
2486     int modeflags = 0;
2487     char *batchfile = NULL;
2488     int errors = 0;
2489
2490     flags = FLAG_STDERR | FLAG_INTERACTIVE
2491 #ifdef FLAG_SYNCAGENT
2492         | FLAG_SYNCAGENT
2493 #endif
2494         ;
2495     cmdline_tooltype = TOOLTYPE_FILETRANSFER;
2496     ssh_get_line = &console_get_line;
2497     sk_init();
2498
2499     userhost = user = NULL;
2500
2501     /* Load Default Settings before doing anything else. */
2502     do_defaults(NULL, &cfg);
2503     loaded_session = FALSE;
2504
2505     errors = 0;
2506     for (i = 1; i < argc; i++) {
2507         int ret;
2508         if (argv[i][0] != '-') {
2509             if (userhost)
2510                 usage();
2511             else
2512                 userhost = dupstr(argv[i]);
2513             continue;
2514         }
2515         ret = cmdline_process_param(argv[i], i+1<argc?argv[i+1]:NULL, 1, &cfg);
2516         if (ret == -2) {
2517             cmdline_error("option \"%s\" requires an argument", argv[i]);
2518         } else if (ret == 2) {
2519             i++;               /* skip next argument */
2520         } else if (ret == 1) {
2521             /* We have our own verbosity in addition to `flags'. */
2522             if (flags & FLAG_VERBOSE)
2523                 verbose = 1;
2524         } else if (strcmp(argv[i], "-h") == 0 ||
2525                    strcmp(argv[i], "-?") == 0) {
2526             usage();
2527         } else if (strcmp(argv[i], "-V") == 0) {
2528             version();
2529         } else if (strcmp(argv[i], "-batch") == 0) {
2530             console_batch_mode = 1;
2531         } else if (strcmp(argv[i], "-b") == 0 && i + 1 < argc) {
2532             mode = 1;
2533             batchfile = argv[++i];
2534         } else if (strcmp(argv[i], "-bc") == 0) {
2535             modeflags = modeflags | 1;
2536         } else if (strcmp(argv[i], "-be") == 0) {
2537             modeflags = modeflags | 2;
2538         } else if (strcmp(argv[i], "--") == 0) {
2539             i++;
2540             break;
2541         } else {
2542             cmdline_error("unknown option \"%s\"", argv[i]);
2543         }
2544     }
2545     argc -= i;
2546     argv += i;
2547     back = NULL;
2548
2549     /*
2550      * If the loaded session provides a hostname, and a hostname has not
2551      * otherwise been specified, pop it in `userhost' so that
2552      * `psftp -load sessname' is sufficient to start a session.
2553      */
2554     if (!userhost && cfg.host[0] != '\0') {
2555         userhost = dupstr(cfg.host);
2556     }
2557
2558     /*
2559      * If a user@host string has already been provided, connect to
2560      * it now.
2561      */
2562     if (userhost) {
2563         int ret;
2564         ret = psftp_connect(userhost, user, portnumber);
2565         sfree(userhost);
2566         if (ret)
2567             return 1;
2568         if (do_sftp_init())
2569             return 1;
2570     } else {
2571         printf("psftp: no hostname specified; use \"open host.name\""
2572                " to connect\n");
2573     }
2574
2575     do_sftp(mode, modeflags, batchfile);
2576
2577     if (back != NULL && back->socket(backhandle) != NULL) {
2578         char ch;
2579         back->special(backhandle, TS_EOF);
2580         sftp_recvdata(&ch, 1);
2581     }
2582     random_save_seed();
2583     cmdline_cleanup();
2584     console_provide_logctx(NULL);
2585     do_sftp_cleanup();
2586     backhandle = NULL;
2587     back = NULL;
2588     sk_cleanup();
2589
2590     return 0;
2591 }