2 * psftp.c: (platform-independent) front end for PSFTP.
11 #define PUTTY_DO_GLOBALS
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
26 static int psftp_connect(char *userhost, char *user, int portnumber);
27 static int do_sftp_init(void);
28 void do_sftp_cleanup();
30 /* ----------------------------------------------------------------------
36 static void *backhandle;
39 /* ----------------------------------------------------------------------
40 * Higher-level helper functions used in commands.
44 * Attempt to canonify a pathname starting from the pwd. If
45 * canonification fails, at least fall back to returning a _valid_
46 * pathname (though it may be ugly, eg /home/simon/../foobar).
48 char *canonify(char *name)
50 char *fullname, *canonname;
51 struct sftp_packet *pktin;
52 struct sftp_request *req, *rreq;
55 fullname = dupstr(name);
58 if (pwd[strlen(pwd) - 1] == '/')
62 fullname = dupcat(pwd, slash, name, NULL);
65 sftp_register(req = fxp_realpath_send(fullname));
66 rreq = sftp_find_request(pktin = sftp_recv());
68 canonname = fxp_realpath_recv(pktin, rreq);
75 * Attempt number 2. Some FXP_REALPATH implementations
76 * (glibc-based ones, in particular) require the _whole_
77 * path to point to something that exists, whereas others
78 * (BSD-based) only require all but the last component to
79 * exist. So if the first call failed, we should strip off
80 * everything from the last slash onwards and try again,
81 * then put the final component back on.
85 * - if the last component is "/." or "/..", then we don't
86 * bother trying this because there's no way it can work.
88 * - if the thing actually ends with a "/", we remove it
89 * before we start. Except if the string is "/" itself
90 * (although I can't see why we'd have got here if so,
91 * because surely "/" would have worked the first
92 * time?), in which case we don't bother.
94 * - if there's no slash in the string at all, give up in
95 * confusion (we expect at least one because of the way
96 * we constructed the string).
102 i = strlen(fullname);
103 if (i > 2 && fullname[i - 1] == '/')
104 fullname[--i] = '\0'; /* strip trailing / unless at pos 0 */
105 while (i > 0 && fullname[--i] != '/');
108 * Give up on special cases.
110 if (fullname[i] != '/' || /* no slash at all */
111 !strcmp(fullname + i, "/.") || /* ends in /. */
112 !strcmp(fullname + i, "/..") || /* ends in /.. */
113 !strcmp(fullname, "/")) {
118 * Now i points at the slash. Deal with the final special
119 * case i==0 (ie the whole path was "/nonexistentfile").
121 fullname[i] = '\0'; /* separate the string */
123 sftp_register(req = fxp_realpath_send("/"));
125 sftp_register(req = fxp_realpath_send(fullname));
127 rreq = sftp_find_request(pktin = sftp_recv());
129 canonname = fxp_realpath_recv(pktin, rreq);
132 return fullname; /* even that failed; give up */
135 * We have a canonical name for all but the last path
136 * component. Concatenate the last component and return.
138 returnname = dupcat(canonname,
139 canonname[strlen(canonname) - 1] ==
140 '/' ? "" : "/", fullname + i + 1, NULL);
148 * Return a pointer to the portion of str that comes after the last
149 * slash (or backslash or colon, if `local' is TRUE).
151 static char *stripslashes(char *str, int local)
156 p = strchr(str, ':');
160 p = strrchr(str, '/');
164 p = strrchr(str, '\\');
172 * qsort comparison routine for fxp_name structures. Sorts by real
175 static int sftp_name_compare(const void *av, const void *bv)
177 const struct fxp_name *const *a = (const struct fxp_name *const *) av;
178 const struct fxp_name *const *b = (const struct fxp_name *const *) bv;
179 return strcmp((*a)->filename, (*b)->filename);
183 * Likewise, but for a bare char *.
185 static int bare_name_compare(const void *av, const void *bv)
187 const char **a = (const char **) av;
188 const char **b = (const char **) bv;
189 return strcmp(*a, *b);
192 /* ----------------------------------------------------------------------
193 * The meat of the `get' and `put' commands.
195 int sftp_get_file(char *fname, char *outfname, int recurse, int restart)
197 struct fxp_handle *fh;
198 struct sftp_packet *pktin;
199 struct sftp_request *req, *rreq;
200 struct fxp_xfer *xfer;
203 int ret, shown_err = FALSE;
206 * In recursive mode, see if we're dealing with a directory.
207 * (If we're not in recursive mode, we need not even check: the
208 * subsequent FXP_OPEN will return a usable error message.)
211 struct fxp_attrs attrs;
214 sftp_register(req = fxp_stat_send(fname));
215 rreq = sftp_find_request(pktin = sftp_recv());
217 result = fxp_stat_recv(pktin, rreq, &attrs);
220 (attrs.flags & SSH_FILEXFER_ATTR_PERMISSIONS) &&
221 (attrs.permissions & 0040000)) {
223 struct fxp_handle *dirhandle;
224 int nnames, namesize;
225 struct fxp_name **ournames;
226 struct fxp_names *names;
230 * First, attempt to create the destination directory,
231 * unless it already exists.
233 if (file_type(outfname) != FILE_TYPE_DIRECTORY &&
234 !create_directory(outfname)) {
235 printf("%s: Cannot create directory\n", outfname);
240 * Now get the list of filenames in the remote
243 sftp_register(req = fxp_opendir_send(fname));
244 rreq = sftp_find_request(pktin = sftp_recv());
246 dirhandle = fxp_opendir_recv(pktin, rreq);
249 printf("%s: unable to open directory: %s\n",
253 nnames = namesize = 0;
258 sftp_register(req = fxp_readdir_send(dirhandle));
259 rreq = sftp_find_request(pktin = sftp_recv());
261 names = fxp_readdir_recv(pktin, rreq);
264 if (fxp_error_type() == SSH_FX_EOF)
266 printf("%s: reading directory: %s\n", fname, fxp_error());
270 if (names->nnames == 0) {
271 fxp_free_names(names);
274 if (nnames + names->nnames >= namesize) {
275 namesize += names->nnames + 128;
276 ournames = sresize(ournames, namesize, struct fxp_name *);
278 for (i = 0; i < names->nnames; i++)
279 if (strcmp(names->names[i].filename, ".") &&
280 strcmp(names->names[i].filename, "..")) {
281 if (!vet_filename(names->names[i].filename)) {
282 printf("ignoring potentially dangerous server-"
283 "supplied filename '%s'\n",
284 names->names[i].filename);
287 fxp_dup_name(&names->names[i]);
290 fxp_free_names(names);
292 sftp_register(req = fxp_close_send(dirhandle));
293 rreq = sftp_find_request(pktin = sftp_recv());
295 fxp_close_recv(pktin, rreq);
298 * Sort the names into a clear order. This ought to
299 * make things more predictable when we're doing a
300 * reget of the same directory, just in case two
301 * readdirs on the same remote directory return a
304 qsort(ournames, nnames, sizeof(*ournames), sftp_name_compare);
307 * If we're in restart mode, find the last filename on
308 * this list that already exists. We may have to do a
309 * reget on _that_ file, but shouldn't have to do
310 * anything on the previous files.
312 * If none of them exists, of course, we start at 0.
319 nextoutfname = dir_file_cat(outfname,
320 ournames[i]->filename);
322 nextoutfname = dupstr(ournames[i]->filename);
323 ret = (file_type(nextoutfname) == FILE_TYPE_NONEXISTENT);
333 * Now we're ready to recurse. Starting at ournames[i]
334 * and continuing on to the end of the list, we
335 * construct a new source and target file name, and
336 * call sftp_get_file again.
338 for (; i < nnames; i++) {
339 char *nextfname, *nextoutfname;
342 nextfname = dupcat(fname, "/", ournames[i]->filename, NULL);
344 nextoutfname = dir_file_cat(outfname,
345 ournames[i]->filename);
347 nextoutfname = dupstr(ournames[i]->filename);
348 ret = sftp_get_file(nextfname, nextoutfname, recurse, restart);
349 restart = FALSE; /* after first partial file, do full */
353 for (i = 0; i < nnames; i++) {
354 fxp_free_name(ournames[i]);
362 * Done this recursion level. Free everything.
364 for (i = 0; i < nnames; i++) {
365 fxp_free_name(ournames[i]);
373 sftp_register(req = fxp_open_send(fname, SSH_FXF_READ));
374 rreq = sftp_find_request(pktin = sftp_recv());
376 fh = fxp_open_recv(pktin, rreq);
379 printf("%s: %s\n", fname, fxp_error());
384 fp = fopen(outfname, "rb+");
386 fp = fopen(outfname, "wb");
390 printf("local: unable to open %s\n", outfname);
392 sftp_register(req = fxp_close_send(fh));
393 rreq = sftp_find_request(pktin = sftp_recv());
395 fxp_close_recv(pktin, rreq);
402 fseek(fp, 0L, SEEK_END);
404 printf("reget: restarting at file position %ld\n", posn);
405 offset = uint64_make(0, posn);
407 offset = uint64_make(0, 0);
410 printf("remote:%s => local:%s\n", fname, outfname);
413 * FIXME: we can use FXP_FSTAT here to get the file size, and
414 * thus put up a progress bar.
417 xfer = xfer_download_init(fh, offset);
418 while (!xfer_done(xfer)) {
423 xfer_download_queue(xfer);
425 ret = xfer_download_gotpkt(xfer, pktin);
429 printf("error while reading: %s\n", fxp_error());
435 while (xfer_download_data(xfer, &vbuf, &len)) {
436 unsigned char *buf = (unsigned char *)vbuf;
440 wlen = fwrite(buf + wpos, 1, len - wpos, fp);
442 printf("error while writing local file\n");
444 xfer_set_error(xfer);
448 if (wpos < len) { /* we had an error */
450 xfer_set_error(xfer);
461 sftp_register(req = fxp_close_send(fh));
462 rreq = sftp_find_request(pktin = sftp_recv());
464 fxp_close_recv(pktin, rreq);
469 int sftp_put_file(char *fname, char *outfname, int recurse, int restart)
471 struct fxp_handle *fh;
472 struct fxp_xfer *xfer;
473 struct sftp_packet *pktin;
474 struct sftp_request *req, *rreq;
480 * In recursive mode, see if we're dealing with a directory.
481 * (If we're not in recursive mode, we need not even check: the
482 * subsequent fopen will return an error message.)
484 if (recurse && file_type(fname) == FILE_TYPE_DIRECTORY) {
485 struct fxp_attrs attrs;
487 int nnames, namesize;
488 char *name, **ournames;
493 * First, attempt to create the destination directory,
494 * unless it already exists.
496 sftp_register(req = fxp_stat_send(outfname));
497 rreq = sftp_find_request(pktin = sftp_recv());
499 result = fxp_stat_recv(pktin, rreq, &attrs);
501 !(attrs.flags & SSH_FILEXFER_ATTR_PERMISSIONS) ||
502 !(attrs.permissions & 0040000)) {
503 sftp_register(req = fxp_mkdir_send(outfname));
504 rreq = sftp_find_request(pktin = sftp_recv());
506 result = fxp_mkdir_recv(pktin, rreq);
509 printf("%s: create directory: %s\n",
510 outfname, fxp_error());
516 * Now get the list of filenames in the local directory.
518 nnames = namesize = 0;
521 dh = open_directory(fname);
523 printf("%s: unable to open directory\n", fname);
526 while ((name = read_filename(dh)) != NULL) {
527 if (nnames >= namesize) {
529 ournames = sresize(ournames, namesize, char *);
531 ournames[nnames++] = name;
536 * Sort the names into a clear order. This ought to make
537 * things more predictable when we're doing a reput of the
538 * same directory, just in case two readdirs on the same
539 * local directory return a different order.
541 qsort(ournames, nnames, sizeof(*ournames), bare_name_compare);
544 * If we're in restart mode, find the last filename on this
545 * list that already exists. We may have to do a reput on
546 * _that_ file, but shouldn't have to do anything on the
549 * If none of them exists, of course, we start at 0.
554 nextoutfname = dupcat(outfname, "/", ournames[i], NULL);
555 sftp_register(req = fxp_stat_send(nextoutfname));
556 rreq = sftp_find_request(pktin = sftp_recv());
558 result = fxp_stat_recv(pktin, rreq, &attrs);
568 * Now we're ready to recurse. Starting at ournames[i]
569 * and continuing on to the end of the list, we
570 * construct a new source and target file name, and
571 * call sftp_put_file again.
573 for (; i < nnames; i++) {
574 char *nextfname, *nextoutfname;
578 nextfname = dir_file_cat(fname, ournames[i]);
580 nextfname = dupstr(ournames[i]);
581 nextoutfname = dupcat(outfname, "/", ournames[i], NULL);
582 ret = sftp_put_file(nextfname, nextoutfname, recurse, restart);
583 restart = FALSE; /* after first partial file, do full */
587 for (i = 0; i < nnames; i++) {
596 * Done this recursion level. Free everything.
598 for (i = 0; i < nnames; i++) {
606 fp = fopen(fname, "rb");
608 printf("local: unable to open %s\n", fname);
612 sftp_register(req = fxp_open_send(outfname, SSH_FXF_WRITE));
614 sftp_register(req = fxp_open_send(outfname, SSH_FXF_WRITE |
615 SSH_FXF_CREAT | SSH_FXF_TRUNC));
617 rreq = sftp_find_request(pktin = sftp_recv());
619 fh = fxp_open_recv(pktin, rreq);
622 printf("%s: %s\n", outfname, fxp_error());
628 struct fxp_attrs attrs;
631 sftp_register(req = fxp_fstat_send(fh));
632 rreq = sftp_find_request(pktin = sftp_recv());
634 ret = fxp_fstat_recv(pktin, rreq, &attrs);
637 printf("read size of %s: %s\n", outfname, fxp_error());
640 if (!(attrs.flags & SSH_FILEXFER_ATTR_SIZE)) {
641 printf("read size of %s: size was not given\n", outfname);
645 uint64_decimal(offset, decbuf);
646 printf("reput: restarting at file position %s\n", decbuf);
647 if (uint64_compare(offset, uint64_make(0, LONG_MAX)) > 0) {
648 printf("reput: remote file is larger than we can deal with\n");
651 if (fseek(fp, offset.lo, SEEK_SET) != 0)
652 fseek(fp, 0, SEEK_END); /* *shrug* */
654 offset = uint64_make(0, 0);
657 printf("local:%s => remote:%s\n", fname, outfname);
660 * FIXME: we can use FXP_FSTAT here to get the file size, and
661 * thus put up a progress bar.
664 xfer = xfer_upload_init(fh, offset);
666 while ((!err && !eof) || !xfer_done(xfer)) {
670 while (xfer_upload_ready(xfer) && !err && !eof) {
671 len = fread(buffer, 1, sizeof(buffer), fp);
673 printf("error while reading local file\n");
675 } else if (len == 0) {
678 xfer_upload_data(xfer, buffer, len);
682 if (!xfer_done(xfer)) {
684 ret = xfer_upload_gotpkt(xfer, pktin);
686 printf("error while writing: %s\n", fxp_error());
694 sftp_register(req = fxp_close_send(fh));
695 rreq = sftp_find_request(pktin = sftp_recv());
697 fxp_close_recv(pktin, rreq);
704 /* ----------------------------------------------------------------------
705 * A remote wildcard matcher, providing a similar interface to the
706 * local one in psftp.h.
709 typedef struct SftpWildcardMatcher {
710 struct fxp_handle *dirh;
711 struct fxp_names *names;
713 char *wildcard, *prefix;
714 } SftpWildcardMatcher;
716 SftpWildcardMatcher *sftp_begin_wildcard_matching(char *name)
718 struct sftp_packet *pktin;
719 struct sftp_request *req, *rreq;
721 char *unwcdir, *tmpdir, *cdir;
723 SftpWildcardMatcher *swcm;
724 struct fxp_handle *dirh;
727 * We don't handle multi-level wildcards; so we expect to find
728 * a fully specified directory part, followed by a wildcard
731 wildcard = stripslashes(name, 0);
733 unwcdir = dupstr(name);
734 len = wildcard - name;
736 if (len > 0 && unwcdir[len-1] == '/')
737 unwcdir[len-1] = '\0';
738 tmpdir = snewn(1 + len, char);
739 check = wc_unescape(tmpdir, unwcdir);
743 printf("Multiple-level wildcards are not supported\n");
748 cdir = canonify(unwcdir);
750 sftp_register(req = fxp_opendir_send(cdir));
751 rreq = sftp_find_request(pktin = sftp_recv());
753 dirh = fxp_opendir_recv(pktin, rreq);
756 swcm = snew(SftpWildcardMatcher);
759 swcm->wildcard = dupstr(wildcard);
760 swcm->prefix = unwcdir;
762 printf("Unable to open %s: %s\n", cdir, fxp_error());
772 char *sftp_wildcard_get_filename(SftpWildcardMatcher *swcm)
774 struct fxp_name *name;
775 struct sftp_packet *pktin;
776 struct sftp_request *req, *rreq;
779 if (swcm->names && swcm->namepos >= swcm->names->nnames) {
780 fxp_free_names(swcm->names);
785 sftp_register(req = fxp_readdir_send(swcm->dirh));
786 rreq = sftp_find_request(pktin = sftp_recv());
788 swcm->names = fxp_readdir_recv(pktin, rreq);
791 if (fxp_error_type() != SSH_FX_EOF)
792 printf("%s: reading directory: %s\n", swcm->prefix,
800 assert(swcm->names && swcm->namepos < swcm->names->nnames);
802 name = &swcm->names->names[swcm->namepos++];
804 if (!strcmp(name->filename, ".") || !strcmp(name->filename, ".."))
805 continue; /* expected bad filenames */
807 if (!vet_filename(name->filename)) {
808 printf("ignoring potentially dangerous server-"
809 "supplied filename '%s'\n", name->filename);
810 continue; /* unexpected bad filename */
813 if (!wc_match(swcm->wildcard, name->filename))
814 continue; /* doesn't match the wildcard */
817 * We have a working filename. Return it.
819 return dupprintf("%s%s%s", swcm->prefix,
820 swcm->prefix[strlen(swcm->prefix)-1]=='/' ? "" : "/",
825 void sftp_finish_wildcard_matching(SftpWildcardMatcher *swcm)
827 struct sftp_packet *pktin;
828 struct sftp_request *req, *rreq;
830 sftp_register(req = fxp_close_send(swcm->dirh));
831 rreq = sftp_find_request(pktin = sftp_recv());
833 fxp_close_recv(pktin, rreq);
836 fxp_free_names(swcm->names);
839 sfree(swcm->wildcard);
844 /* ----------------------------------------------------------------------
845 * Actual sftp commands.
847 struct sftp_command {
849 int nwords, wordssize;
850 int (*obey) (struct sftp_command *); /* returns <0 to quit */
853 int sftp_cmd_null(struct sftp_command *cmd)
855 return 1; /* success */
858 int sftp_cmd_unknown(struct sftp_command *cmd)
860 printf("psftp: unknown command \"%s\"\n", cmd->words[0]);
861 return 0; /* failure */
864 int sftp_cmd_quit(struct sftp_command *cmd)
869 int sftp_cmd_close(struct sftp_command *cmd)
872 printf("psftp: not connected to a host; use \"open host.name\"\n");
876 if (back != NULL && back->socket(backhandle) != NULL) {
878 back->special(backhandle, TS_EOF);
879 sftp_recvdata(&ch, 1);
887 * List a directory. If no arguments are given, list pwd; otherwise
888 * list the directory given in words[1].
890 int sftp_cmd_ls(struct sftp_command *cmd)
892 struct fxp_handle *dirh;
893 struct fxp_names *names;
894 struct fxp_name **ournames;
895 int nnames, namesize;
896 char *dir, *cdir, *unwcdir, *wildcard;
897 struct sftp_packet *pktin;
898 struct sftp_request *req, *rreq;
902 printf("psftp: not connected to a host; use \"open host.name\"\n");
911 unwcdir = snewn(1 + strlen(dir), char);
912 if (wc_unescape(unwcdir, dir)) {
919 wildcard = stripslashes(dir, 0);
920 unwcdir = dupstr(dir);
921 len = wildcard - dir;
923 if (len > 0 && unwcdir[len-1] == '/')
924 unwcdir[len-1] = '\0';
925 tmpdir = snewn(1 + len, char);
926 check = wc_unescape(tmpdir, unwcdir);
929 printf("Multiple-level wildcards are not supported\n");
936 cdir = canonify(dir);
938 printf("%s: %s\n", dir, fxp_error());
943 printf("Listing directory %s\n", cdir);
945 sftp_register(req = fxp_opendir_send(cdir));
946 rreq = sftp_find_request(pktin = sftp_recv());
948 dirh = fxp_opendir_recv(pktin, rreq);
951 printf("Unable to open %s: %s\n", dir, fxp_error());
953 nnames = namesize = 0;
958 sftp_register(req = fxp_readdir_send(dirh));
959 rreq = sftp_find_request(pktin = sftp_recv());
961 names = fxp_readdir_recv(pktin, rreq);
964 if (fxp_error_type() == SSH_FX_EOF)
966 printf("Reading directory %s: %s\n", dir, fxp_error());
969 if (names->nnames == 0) {
970 fxp_free_names(names);
974 if (nnames + names->nnames >= namesize) {
975 namesize += names->nnames + 128;
976 ournames = sresize(ournames, namesize, struct fxp_name *);
979 for (i = 0; i < names->nnames; i++)
980 if (!wildcard || wc_match(wildcard, names->names[i].filename))
981 ournames[nnames++] = fxp_dup_name(&names->names[i]);
983 fxp_free_names(names);
985 sftp_register(req = fxp_close_send(dirh));
986 rreq = sftp_find_request(pktin = sftp_recv());
988 fxp_close_recv(pktin, rreq);
991 * Now we have our filenames. Sort them by actual file
992 * name, and then output the longname parts.
994 qsort(ournames, nnames, sizeof(*ournames), sftp_name_compare);
999 for (i = 0; i < nnames; i++) {
1000 printf("%s\n", ournames[i]->longname);
1001 fxp_free_name(ournames[i]);
1013 * Change directories. We do this by canonifying the new name, then
1014 * trying to OPENDIR it. Only if that succeeds do we set the new pwd.
1016 int sftp_cmd_cd(struct sftp_command *cmd)
1018 struct fxp_handle *dirh;
1019 struct sftp_packet *pktin;
1020 struct sftp_request *req, *rreq;
1024 printf("psftp: not connected to a host; use \"open host.name\"\n");
1028 if (cmd->nwords < 2)
1029 dir = dupstr(homedir);
1031 dir = canonify(cmd->words[1]);
1034 printf("%s: %s\n", dir, fxp_error());
1038 sftp_register(req = fxp_opendir_send(dir));
1039 rreq = sftp_find_request(pktin = sftp_recv());
1040 assert(rreq == req);
1041 dirh = fxp_opendir_recv(pktin, rreq);
1044 printf("Directory %s: %s\n", dir, fxp_error());
1049 sftp_register(req = fxp_close_send(dirh));
1050 rreq = sftp_find_request(pktin = sftp_recv());
1051 assert(rreq == req);
1052 fxp_close_recv(pktin, rreq);
1056 printf("Remote directory is now %s\n", pwd);
1062 * Print current directory. Easy as pie.
1064 int sftp_cmd_pwd(struct sftp_command *cmd)
1067 printf("psftp: not connected to a host; use \"open host.name\"\n");
1071 printf("Remote directory is %s\n", pwd);
1076 * Get a file and save it at the local end. We have three very
1077 * similar commands here. The basic one is `get'; `reget' differs
1078 * in that it checks for the existence of the destination file and
1079 * starts from where a previous aborted transfer left off; `mget'
1080 * differs in that it interprets all its arguments as files to
1081 * transfer (never as a different local name for a remote file) and
1082 * can handle wildcards.
1084 int sftp_general_get(struct sftp_command *cmd, int restart, int multiple)
1086 char *fname, *unwcfname, *origfname, *origwfname, *outfname;
1088 int recurse = FALSE;
1091 printf("psftp: not connected to a host; use \"open host.name\"\n");
1096 while (i < cmd->nwords && cmd->words[i][0] == '-') {
1097 if (!strcmp(cmd->words[i], "--")) {
1098 /* finish processing options */
1101 } else if (!strcmp(cmd->words[i], "-r")) {
1104 printf("%s: unrecognised option '%s'\n", cmd->words[0], cmd->words[i]);
1110 if (i >= cmd->nwords) {
1111 printf("%s: expects a filename\n", cmd->words[0]);
1117 SftpWildcardMatcher *swcm;
1119 origfname = cmd->words[i++];
1120 unwcfname = snewn(strlen(origfname)+1, char);
1122 if (multiple && !wc_unescape(unwcfname, origfname)) {
1123 swcm = sftp_begin_wildcard_matching(origfname);
1128 origwfname = sftp_wildcard_get_filename(swcm);
1130 /* Politely warn the user that nothing matched. */
1131 printf("%s: nothing matched\n", origfname);
1132 sftp_finish_wildcard_matching(swcm);
1137 origwfname = origfname;
1141 while (origwfname) {
1142 fname = canonify(origwfname);
1145 printf("%s: %s\n", origwfname, fxp_error());
1150 if (!multiple && i < cmd->nwords)
1151 outfname = cmd->words[i++];
1153 outfname = stripslashes(origwfname, 0);
1155 ret = sftp_get_file(fname, outfname, recurse, restart);
1161 origwfname = sftp_wildcard_get_filename(swcm);
1168 sftp_finish_wildcard_matching(swcm);
1172 } while (multiple && i < cmd->nwords);
1176 int sftp_cmd_get(struct sftp_command *cmd)
1178 return sftp_general_get(cmd, 0, 0);
1180 int sftp_cmd_mget(struct sftp_command *cmd)
1182 return sftp_general_get(cmd, 0, 1);
1184 int sftp_cmd_reget(struct sftp_command *cmd)
1186 return sftp_general_get(cmd, 1, 0);
1190 * Send a file and store it at the remote end. We have three very
1191 * similar commands here. The basic one is `put'; `reput' differs
1192 * in that it checks for the existence of the destination file and
1193 * starts from where a previous aborted transfer left off; `mput'
1194 * differs in that it interprets all its arguments as files to
1195 * transfer (never as a different remote name for a local file) and
1196 * can handle wildcards.
1198 int sftp_general_put(struct sftp_command *cmd, int restart, int multiple)
1200 char *fname, *wfname, *origoutfname, *outfname;
1202 int recurse = FALSE;
1205 printf("psftp: not connected to a host; use \"open host.name\"\n");
1210 while (i < cmd->nwords && cmd->words[i][0] == '-') {
1211 if (!strcmp(cmd->words[i], "--")) {
1212 /* finish processing options */
1215 } else if (!strcmp(cmd->words[i], "-r")) {
1218 printf("%s: unrecognised option '%s'\n", cmd->words[0], cmd->words[i]);
1224 if (i >= cmd->nwords) {
1225 printf("%s: expects a filename\n", cmd->words[0]);
1231 WildcardMatcher *wcm;
1232 fname = cmd->words[i++];
1234 if (multiple && test_wildcard(fname, FALSE) == WCTYPE_WILDCARD) {
1235 wcm = begin_wildcard_matching(fname);
1236 wfname = wildcard_get_filename(wcm);
1238 /* Politely warn the user that nothing matched. */
1239 printf("%s: nothing matched\n", fname);
1240 finish_wildcard_matching(wcm);
1249 if (!multiple && i < cmd->nwords)
1250 origoutfname = cmd->words[i++];
1252 origoutfname = stripslashes(wfname, 1);
1254 outfname = canonify(origoutfname);
1256 printf("%s: %s\n", origoutfname, fxp_error());
1259 finish_wildcard_matching(wcm);
1263 ret = sftp_put_file(wfname, outfname, recurse, restart);
1268 wfname = wildcard_get_filename(wcm);
1275 finish_wildcard_matching(wcm);
1280 } while (multiple && i < cmd->nwords);
1284 int sftp_cmd_put(struct sftp_command *cmd)
1286 return sftp_general_put(cmd, 0, 0);
1288 int sftp_cmd_mput(struct sftp_command *cmd)
1290 return sftp_general_put(cmd, 0, 1);
1292 int sftp_cmd_reput(struct sftp_command *cmd)
1294 return sftp_general_put(cmd, 1, 0);
1297 int sftp_cmd_mkdir(struct sftp_command *cmd)
1300 struct sftp_packet *pktin;
1301 struct sftp_request *req, *rreq;
1305 printf("psftp: not connected to a host; use \"open host.name\"\n");
1309 if (cmd->nwords < 2) {
1310 printf("mkdir: expects a directory\n");
1314 dir = canonify(cmd->words[1]);
1316 printf("%s: %s\n", dir, fxp_error());
1320 sftp_register(req = fxp_mkdir_send(dir));
1321 rreq = sftp_find_request(pktin = sftp_recv());
1322 assert(rreq == req);
1323 result = fxp_mkdir_recv(pktin, rreq);
1326 printf("mkdir %s: %s\n", dir, fxp_error());
1335 int sftp_cmd_rmdir(struct sftp_command *cmd)
1338 struct sftp_packet *pktin;
1339 struct sftp_request *req, *rreq;
1343 printf("psftp: not connected to a host; use \"open host.name\"\n");
1347 if (cmd->nwords < 2) {
1348 printf("rmdir: expects a directory\n");
1352 dir = canonify(cmd->words[1]);
1354 printf("%s: %s\n", dir, fxp_error());
1358 sftp_register(req = fxp_rmdir_send(dir));
1359 rreq = sftp_find_request(pktin = sftp_recv());
1360 assert(rreq == req);
1361 result = fxp_rmdir_recv(pktin, rreq);
1364 printf("rmdir %s: %s\n", dir, fxp_error());
1373 int sftp_cmd_rm(struct sftp_command *cmd)
1376 struct sftp_packet *pktin;
1377 struct sftp_request *req, *rreq;
1381 printf("psftp: not connected to a host; use \"open host.name\"\n");
1385 if (cmd->nwords < 2) {
1386 printf("rm: expects a filename\n");
1390 fname = canonify(cmd->words[1]);
1392 printf("%s: %s\n", fname, fxp_error());
1396 sftp_register(req = fxp_remove_send(fname));
1397 rreq = sftp_find_request(pktin = sftp_recv());
1398 assert(rreq == req);
1399 result = fxp_remove_recv(pktin, rreq);
1402 printf("rm %s: %s\n", fname, fxp_error());
1411 int sftp_cmd_mv(struct sftp_command *cmd)
1413 char *srcfname, *dstfname;
1414 struct sftp_packet *pktin;
1415 struct sftp_request *req, *rreq;
1419 printf("psftp: not connected to a host; use \"open host.name\"\n");
1423 if (cmd->nwords < 3) {
1424 printf("mv: expects two filenames\n");
1427 srcfname = canonify(cmd->words[1]);
1429 printf("%s: %s\n", srcfname, fxp_error());
1433 dstfname = canonify(cmd->words[2]);
1435 printf("%s: %s\n", dstfname, fxp_error());
1439 sftp_register(req = fxp_rename_send(srcfname, dstfname));
1440 rreq = sftp_find_request(pktin = sftp_recv());
1441 assert(rreq == req);
1442 result = fxp_rename_recv(pktin, rreq);
1445 char const *error = fxp_error();
1446 struct fxp_attrs attrs;
1449 * The move might have failed because dstfname pointed at a
1450 * directory. We check this possibility now: if dstfname
1451 * _is_ a directory, we re-attempt the move by appending
1452 * the basename of srcfname to dstfname.
1454 sftp_register(req = fxp_stat_send(dstfname));
1455 rreq = sftp_find_request(pktin = sftp_recv());
1456 assert(rreq == req);
1457 result = fxp_stat_recv(pktin, rreq, &attrs);
1460 (attrs.flags & SSH_FILEXFER_ATTR_PERMISSIONS) &&
1461 (attrs.permissions & 0040000)) {
1463 char *newname, *newcanon;
1464 printf("(destination %s is a directory)\n", dstfname);
1465 p = srcfname + strlen(srcfname);
1466 while (p > srcfname && p[-1] != '/') p--;
1467 newname = dupcat(dstfname, "/", p, NULL);
1468 newcanon = canonify(newname);
1472 dstfname = newcanon;
1474 sftp_register(req = fxp_rename_send(srcfname, dstfname));
1475 rreq = sftp_find_request(pktin = sftp_recv());
1476 assert(rreq == req);
1477 result = fxp_rename_recv(pktin, rreq);
1479 error = result ? NULL : fxp_error();
1483 printf("mv %s %s: %s\n", srcfname, dstfname, error);
1489 printf("%s -> %s\n", srcfname, dstfname);
1496 int sftp_cmd_chmod(struct sftp_command *cmd)
1500 struct fxp_attrs attrs;
1501 unsigned attrs_clr, attrs_xor, oldperms, newperms;
1502 struct sftp_packet *pktin;
1503 struct sftp_request *req, *rreq;
1506 printf("psftp: not connected to a host; use \"open host.name\"\n");
1510 if (cmd->nwords < 3) {
1511 printf("chmod: expects a mode specifier and a filename\n");
1516 * Attempt to parse the mode specifier in cmd->words[1]. We
1517 * don't support the full horror of Unix chmod; instead we
1518 * support a much simpler syntax in which the user can either
1519 * specify an octal number, or a comma-separated sequence of
1520 * [ugoa]*[-+=][rwxst]+. (The initial [ugoa] sequence may
1521 * _only_ be omitted if the only attribute mentioned is t,
1522 * since all others require a user/group/other specification.
1523 * Additionally, the s attribute may not be specified for any
1524 * [ugoa] specifications other than exactly u or exactly g.
1526 attrs_clr = attrs_xor = 0;
1527 mode = cmd->words[1];
1528 if (mode[0] >= '0' && mode[0] <= '9') {
1529 if (mode[strspn(mode, "01234567")]) {
1530 printf("chmod: numeric file modes should"
1531 " contain digits 0-7 only\n");
1535 sscanf(mode, "%o", &attrs_xor);
1536 attrs_xor &= attrs_clr;
1539 char *modebegin = mode;
1540 unsigned subset, perms;
1544 while (*mode && *mode != ',' &&
1545 *mode != '+' && *mode != '-' && *mode != '=') {
1547 case 'u': subset |= 04700; break; /* setuid, user perms */
1548 case 'g': subset |= 02070; break; /* setgid, group perms */
1549 case 'o': subset |= 00007; break; /* just other perms */
1550 case 'a': subset |= 06777; break; /* all of the above */
1552 printf("chmod: file mode '%.*s' contains unrecognised"
1553 " user/group/other specifier '%c'\n",
1554 (int)strcspn(modebegin, ","), modebegin, *mode);
1559 if (!*mode || *mode == ',') {
1560 printf("chmod: file mode '%.*s' is incomplete\n",
1561 (int)strcspn(modebegin, ","), modebegin);
1565 if (!*mode || *mode == ',') {
1566 printf("chmod: file mode '%.*s' is incomplete\n",
1567 (int)strcspn(modebegin, ","), modebegin);
1571 while (*mode && *mode != ',') {
1573 case 'r': perms |= 00444; break;
1574 case 'w': perms |= 00222; break;
1575 case 'x': perms |= 00111; break;
1576 case 't': perms |= 01000; subset |= 01000; break;
1578 if ((subset & 06777) != 04700 &&
1579 (subset & 06777) != 02070) {
1580 printf("chmod: file mode '%.*s': set[ug]id bit should"
1581 " be used with exactly one of u or g only\n",
1582 (int)strcspn(modebegin, ","), modebegin);
1588 printf("chmod: file mode '%.*s' contains unrecognised"
1589 " permission specifier '%c'\n",
1590 (int)strcspn(modebegin, ","), modebegin, *mode);
1595 if (!(subset & 06777) && (perms &~ subset)) {
1596 printf("chmod: file mode '%.*s' contains no user/group/other"
1597 " specifier and permissions other than 't' \n",
1598 (int)strcspn(modebegin, ","), modebegin);
1609 attrs_xor &= ~perms;
1612 attrs_clr |= subset;
1616 if (*mode) mode++; /* eat comma */
1620 fname = canonify(cmd->words[2]);
1622 printf("%s: %s\n", fname, fxp_error());
1626 sftp_register(req = fxp_stat_send(fname));
1627 rreq = sftp_find_request(pktin = sftp_recv());
1628 assert(rreq == req);
1629 result = fxp_stat_recv(pktin, rreq, &attrs);
1631 if (!result || !(attrs.flags & SSH_FILEXFER_ATTR_PERMISSIONS)) {
1632 printf("get attrs for %s: %s\n", fname,
1633 result ? "file permissions not provided" : fxp_error());
1638 attrs.flags = SSH_FILEXFER_ATTR_PERMISSIONS; /* perms _only_ */
1639 oldperms = attrs.permissions & 07777;
1640 attrs.permissions &= ~attrs_clr;
1641 attrs.permissions ^= attrs_xor;
1642 newperms = attrs.permissions & 07777;
1644 sftp_register(req = fxp_setstat_send(fname, attrs));
1645 rreq = sftp_find_request(pktin = sftp_recv());
1646 assert(rreq == req);
1647 result = fxp_setstat_recv(pktin, rreq);
1650 printf("set attrs for %s: %s\n", fname, fxp_error());
1655 printf("%s: %04o -> %04o\n", fname, oldperms, newperms);
1661 static int sftp_cmd_open(struct sftp_command *cmd)
1666 printf("psftp: already connected\n");
1670 if (cmd->nwords < 2) {
1671 printf("open: expects a host name\n");
1675 if (cmd->nwords > 2) {
1676 portnumber = atoi(cmd->words[2]);
1677 if (portnumber == 0) {
1678 printf("open: invalid port number\n");
1684 if (psftp_connect(cmd->words[1], NULL, portnumber)) {
1685 back = NULL; /* connection is already closed */
1686 return -1; /* this is fatal */
1692 static int sftp_cmd_lcd(struct sftp_command *cmd)
1694 char *currdir, *errmsg;
1696 if (cmd->nwords < 2) {
1697 printf("lcd: expects a local directory name\n");
1701 errmsg = psftp_lcd(cmd->words[1]);
1703 printf("lcd: unable to change directory: %s\n", errmsg);
1708 currdir = psftp_getcwd();
1709 printf("New local directory is %s\n", currdir);
1715 static int sftp_cmd_lpwd(struct sftp_command *cmd)
1719 currdir = psftp_getcwd();
1720 printf("Current local directory is %s\n", currdir);
1726 static int sftp_cmd_pling(struct sftp_command *cmd)
1730 exitcode = system(cmd->words[1]);
1731 return (exitcode == 0);
1734 static int sftp_cmd_help(struct sftp_command *cmd);
1736 static struct sftp_cmd_lookup {
1739 * For help purposes, there are two kinds of command:
1741 * - primary commands, in which `longhelp' is non-NULL. In
1742 * this case `shorthelp' is descriptive text, and `longhelp'
1743 * is longer descriptive text intended to be printed after
1746 * - alias commands, in which `longhelp' is NULL. In this case
1747 * `shorthelp' is the name of a primary command, which
1748 * contains the help that should double up for this command.
1750 int listed; /* do we list this in primary help? */
1753 int (*obey) (struct sftp_command *);
1756 * List of sftp commands. This is binary-searched so it MUST be
1760 "!", TRUE, "run a local command",
1762 /* FIXME: this example is crap for non-Windows. */
1763 " Runs a local command. For example, \"!del myfile\".\n",
1767 "bye", TRUE, "finish your SFTP session",
1769 " Terminates your SFTP session and quits the PSFTP program.\n",
1773 "cd", TRUE, "change your remote working directory",
1774 " [ <New working directory> ]\n"
1775 " Change the remote working directory for your SFTP session.\n"
1776 " If a new working directory is not supplied, you will be\n"
1777 " returned to your home directory.\n",
1781 "chmod", TRUE, "change file permissions and modes",
1782 " ( <octal-digits> | <modifiers> ) <filename>\n"
1783 " Change the file permissions on a file or directory.\n"
1784 " <octal-digits> can be any octal Unix permission specifier.\n"
1785 " Alternatively, <modifiers> can include:\n"
1786 " u+r make file readable by owning user\n"
1787 " u+w make file writable by owning user\n"
1788 " u+x make file executable by owning user\n"
1789 " u-r make file not readable by owning user\n"
1790 " [also u-w, u-x]\n"
1791 " g+r make file readable by members of owning group\n"
1792 " [also g+w, g+x, g-r, g-w, g-x]\n"
1793 " o+r make file readable by all other users\n"
1794 " [also o+w, o+x, o-r, o-w, o-x]\n"
1795 " a+r make file readable by absolutely everybody\n"
1796 " [also a+w, a+x, a-r, a-w, a-x]\n"
1797 " u+s enable the Unix set-user-ID bit\n"
1798 " u-s disable the Unix set-user-ID bit\n"
1799 " g+s enable the Unix set-group-ID bit\n"
1800 " g-s disable the Unix set-group-ID bit\n"
1801 " +t enable the Unix \"sticky bit\"\n"
1802 " You can give more than one modifier for the same user (\"g-rwx\"), and\n"
1803 " more than one user for the same modifier (\"ug+w\"). You can\n"
1804 " use commas to separate different modifiers (\"u+rwx,g+s\").\n",
1808 "close", TRUE, "finish your SFTP session but do not quit PSFTP",
1810 " Terminates your SFTP session, but does not quit the PSFTP\n"
1811 " program. You can then use \"open\" to start another SFTP\n"
1812 " session, to the same server or to a different one.\n",
1816 "del", TRUE, "delete a file",
1818 " Delete a file.\n",
1822 "delete", FALSE, "del", NULL, sftp_cmd_rm
1825 "dir", TRUE, "list contents of a remote directory",
1826 " [ <directory-name> ]/[ <wildcard> ]\n"
1827 " List the contents of a specified directory on the server.\n"
1828 " If <directory-name> is not given, the current working directory\n"
1830 " If <wildcard> is given, it is treated as a set of files to\n"
1831 " list; otherwise, all files are listed.\n",
1835 "exit", TRUE, "bye", NULL, sftp_cmd_quit
1838 "get", TRUE, "download a file from the server to your local machine",
1839 " [ -r ] [ -- ] <filename> [ <local-filename> ]\n"
1840 " Downloads a file on the server and stores it locally under\n"
1841 " the same name, or under a different one if you supply the\n"
1842 " argument <local-filename>.\n"
1843 " If -r specified, recursively fetch a directory.\n",
1847 "help", TRUE, "give help",
1848 " [ <command> [ <command> ... ] ]\n"
1849 " Give general help if no commands are specified.\n"
1850 " If one or more commands are specified, give specific help on\n"
1851 " those particular commands.\n",
1855 "lcd", TRUE, "change local working directory",
1856 " <local-directory-name>\n"
1857 " Change the local working directory of the PSFTP program (the\n"
1858 " default location where the \"get\" command will save files).\n",
1862 "lpwd", TRUE, "print local working directory",
1864 " Print the local working directory of the PSFTP program (the\n"
1865 " default location where the \"get\" command will save files).\n",
1869 "ls", TRUE, "dir", NULL,
1873 "mget", TRUE, "download multiple files at once",
1874 " [ -r ] [ -- ] <filename-or-wildcard> [ <filename-or-wildcard>... ]\n"
1875 " Downloads many files from the server, storing each one under\n"
1876 " the same name it has on the server side. You can use wildcards\n"
1877 " such as \"*.c\" to specify lots of files at once.\n"
1878 " If -r specified, recursively fetch files and directories.\n",
1882 "mkdir", TRUE, "create a directory on the remote server",
1883 " <directory-name>\n"
1884 " Creates a directory with the given name on the server.\n",
1888 "mput", TRUE, "upload multiple files at once",
1889 " [ -r ] [ -- ] <filename-or-wildcard> [ <filename-or-wildcard>... ]\n"
1890 " Uploads many files to the server, storing each one under the\n"
1891 " same name it has on the client side. You can use wildcards\n"
1892 " such as \"*.c\" to specify lots of files at once.\n"
1893 " If -r specified, recursively store files and directories.\n",
1897 "mv", TRUE, "move or rename a file on the remote server",
1898 " <source-filename> <destination-filename>\n"
1899 " Moves or renames the file <source-filename> on the server,\n"
1900 " so that it is accessible under the name <destination-filename>.\n",
1904 "open", TRUE, "connect to a host",
1905 " [<user>@]<hostname> [<port>]\n"
1906 " Establishes an SFTP connection to a given host. Only usable\n"
1907 " when you did not already specify a host name on the command\n"
1912 "put", TRUE, "upload a file from your local machine to the server",
1913 " [ -r ] [ -- ] <filename> [ <remote-filename> ]\n"
1914 " Uploads a file to the server and stores it there under\n"
1915 " the same name, or under a different one if you supply the\n"
1916 " argument <remote-filename>.\n"
1917 " If -r specified, recursively store a directory.\n",
1921 "pwd", TRUE, "print your remote working directory",
1923 " Print the current remote working directory for your SFTP session.\n",
1927 "quit", TRUE, "bye", NULL,
1931 "reget", TRUE, "continue downloading a file",
1932 " [ -r ] [ -- ] <filename> [ <local-filename> ]\n"
1933 " Works exactly like the \"get\" command, but the local file\n"
1934 " must already exist. The download will begin at the end of the\n"
1935 " file. This is for resuming a download that was interrupted.\n"
1936 " If -r specified, resume interrupted \"get -r\".\n",
1940 "ren", TRUE, "mv", NULL,
1944 "rename", FALSE, "mv", NULL,
1948 "reput", TRUE, "continue uploading a file",
1949 " [ -r ] [ -- ] <filename> [ <remote-filename> ]\n"
1950 " Works exactly like the \"put\" command, but the remote file\n"
1951 " must already exist. The upload will begin at the end of the\n"
1952 " file. This is for resuming an upload that was interrupted.\n"
1953 " If -r specified, resume interrupted \"put -r\".\n",
1957 "rm", TRUE, "del", NULL,
1961 "rmdir", TRUE, "remove a directory on the remote server",
1962 " <directory-name>\n"
1963 " Removes the directory with the given name on the server.\n"
1964 " The directory will not be removed unless it is empty.\n",
1969 const struct sftp_cmd_lookup *lookup_command(char *name)
1974 j = sizeof(sftp_lookup) / sizeof(*sftp_lookup);
1977 cmp = strcmp(name, sftp_lookup[k].name);
1983 return &sftp_lookup[k];
1989 static int sftp_cmd_help(struct sftp_command *cmd)
1992 if (cmd->nwords == 1) {
1994 * Give short help on each command.
1998 for (i = 0; i < sizeof(sftp_lookup) / sizeof(*sftp_lookup); i++) {
2000 if (!sftp_lookup[i].listed)
2002 len = strlen(sftp_lookup[i].name);
2006 for (i = 0; i < sizeof(sftp_lookup) / sizeof(*sftp_lookup); i++) {
2007 const struct sftp_cmd_lookup *lookup;
2008 if (!sftp_lookup[i].listed)
2010 lookup = &sftp_lookup[i];
2011 printf("%-*s", maxlen+2, lookup->name);
2012 if (lookup->longhelp == NULL)
2013 lookup = lookup_command(lookup->shorthelp);
2014 printf("%s\n", lookup->shorthelp);
2018 * Give long help on specific commands.
2020 for (i = 1; i < cmd->nwords; i++) {
2021 const struct sftp_cmd_lookup *lookup;
2022 lookup = lookup_command(cmd->words[i]);
2024 printf("help: %s: command not found\n", cmd->words[i]);
2026 printf("%s", lookup->name);
2027 if (lookup->longhelp == NULL)
2028 lookup = lookup_command(lookup->shorthelp);
2029 printf("%s", lookup->longhelp);
2036 /* ----------------------------------------------------------------------
2037 * Command line reading and parsing.
2039 struct sftp_command *sftp_getcmd(FILE *fp, int mode, int modeflags)
2042 struct sftp_command *cmd;
2046 cmd = snew(struct sftp_command);
2056 line = fgetline(fp);
2058 line = ssh_sftp_get_cmdline("psftp> ", back == NULL);
2061 if (!line || !*line) {
2062 cmd->obey = sftp_cmd_quit;
2063 if ((mode == 0) || (modeflags & 1))
2065 return cmd; /* eof */
2068 line[strcspn(line, "\r\n")] = '\0';
2070 if (modeflags & 1) {
2071 printf("%s\n", line);
2075 while (*p && (*p == ' ' || *p == '\t'))
2080 * Special case: the ! command. This is always parsed as
2081 * exactly two words: one containing the !, and the second
2082 * containing everything else on the line.
2084 cmd->nwords = cmd->wordssize = 2;
2085 cmd->words = sresize(cmd->words, cmd->wordssize, char *);
2086 cmd->words[0] = dupstr("!");
2087 cmd->words[1] = dupstr(p+1);
2091 * Parse the command line into words. The syntax is:
2092 * - double quotes are removed, but cause spaces within to be
2093 * treated as non-separating.
2094 * - a double-doublequote pair is a literal double quote, inside
2095 * _or_ outside quotes. Like this:
2097 * firstword "second word" "this has ""quotes"" in" and""this""
2103 * >this has "quotes" in<
2107 /* skip whitespace */
2108 while (*p && (*p == ' ' || *p == '\t'))
2110 /* mark start of word */
2111 q = r = p; /* q sits at start, r writes word */
2114 if (!quoting && (*p == ' ' || *p == '\t'))
2115 break; /* reached end of word */
2116 else if (*p == '"' && p[1] == '"')
2117 p += 2, *r++ = '"'; /* a literal quote */
2119 p++, quoting = !quoting;
2124 p++; /* skip over the whitespace */
2126 if (cmd->nwords >= cmd->wordssize) {
2127 cmd->wordssize = cmd->nwords + 16;
2128 cmd->words = sresize(cmd->words, cmd->wordssize, char *);
2130 cmd->words[cmd->nwords++] = dupstr(q);
2137 * Now parse the first word and assign a function.
2140 if (cmd->nwords == 0)
2141 cmd->obey = sftp_cmd_null;
2143 const struct sftp_cmd_lookup *lookup;
2144 lookup = lookup_command(cmd->words[0]);
2146 cmd->obey = sftp_cmd_unknown;
2148 cmd->obey = lookup->obey;
2154 static int do_sftp_init(void)
2156 struct sftp_packet *pktin;
2157 struct sftp_request *req, *rreq;
2160 * Do protocol initialisation.
2164 "Fatal: unable to initialise SFTP: %s\n", fxp_error());
2165 return 1; /* failure */
2169 * Find out where our home directory is.
2171 sftp_register(req = fxp_realpath_send("."));
2172 rreq = sftp_find_request(pktin = sftp_recv());
2173 assert(rreq == req);
2174 homedir = fxp_realpath_recv(pktin, rreq);
2178 "Warning: failed to resolve home directory: %s\n",
2180 homedir = dupstr(".");
2182 printf("Remote working directory is %s\n", homedir);
2184 pwd = dupstr(homedir);
2188 void do_sftp_cleanup()
2192 back->special(backhandle, TS_EOF);
2193 sftp_recvdata(&ch, 1);
2194 back->free(backhandle);
2195 sftp_cleanup_request();
2209 void do_sftp(int mode, int modeflags, char *batchfile)
2219 /* ------------------------------------------------------------------
2220 * Now we're ready to do Real Stuff.
2223 struct sftp_command *cmd;
2224 cmd = sftp_getcmd(NULL, 0, 0);
2227 ret = cmd->obey(cmd);
2230 for(i = 0; i < cmd->nwords; i++)
2231 sfree(cmd->words[i]);
2239 fp = fopen(batchfile, "r");
2241 printf("Fatal: unable to open %s\n", batchfile);
2245 struct sftp_command *cmd;
2246 cmd = sftp_getcmd(fp, mode, modeflags);
2249 ret = cmd->obey(cmd);
2253 if (!(modeflags & 2))
2262 /* ----------------------------------------------------------------------
2263 * Dirty bits: integration with PuTTY.
2266 static int verbose = 0;
2269 * Print an error message and perform a fatal exit.
2271 void fatalbox(char *fmt, ...)
2276 str = dupvprintf(fmt, ap);
2277 str2 = dupcat("Fatal: ", str, "\n", NULL);
2280 fputs(str2, stderr);
2285 void modalfatalbox(char *fmt, ...)
2290 str = dupvprintf(fmt, ap);
2291 str2 = dupcat("Fatal: ", str, "\n", NULL);
2294 fputs(str2, stderr);
2299 void connection_fatal(void *frontend, char *fmt, ...)
2304 str = dupvprintf(fmt, ap);
2305 str2 = dupcat("Fatal: ", str, "\n", NULL);
2308 fputs(str2, stderr);
2314 void ldisc_send(void *handle, char *buf, int len, int interactive)
2317 * This is only here because of the calls to ldisc_send(NULL,
2318 * 0) in ssh.c. Nothing in PSFTP actually needs to use the
2319 * ldisc as an ldisc. So if we get called with any real data, I
2320 * want to know about it.
2326 * In psftp, all agent requests should be synchronous, so this is a
2327 * never-called stub.
2329 void agent_schedule_callback(void (*callback)(void *, void *, int),
2330 void *callback_ctx, void *data, int len)
2332 assert(!"We shouldn't be here");
2336 * Receive a block of data from the SSH link. Block until all data
2339 * To do this, we repeatedly call the SSH protocol module, with our
2340 * own trap in from_backend() to catch the data that comes back. We
2341 * do this until we have enough data.
2344 static unsigned char *outptr; /* where to put the data */
2345 static unsigned outlen; /* how much data required */
2346 static unsigned char *pending = NULL; /* any spare data */
2347 static unsigned pendlen = 0, pendsize = 0; /* length and phys. size of buffer */
2348 int from_backend(void *frontend, int is_stderr, const char *data, int datalen)
2350 unsigned char *p = (unsigned char *) data;
2351 unsigned len = (unsigned) datalen;
2354 * stderr data is just spouted to local stderr and otherwise
2359 fwrite(data, 1, len, stderr);
2364 * If this is before the real session begins, just return.
2369 if ((outlen > 0) && (len > 0)) {
2370 unsigned used = outlen;
2373 memcpy(outptr, p, used);
2381 if (pendsize < pendlen + len) {
2382 pendsize = pendlen + len + 4096;
2383 pending = sresize(pending, pendsize, unsigned char);
2385 memcpy(pending + pendlen, p, len);
2391 int sftp_recvdata(char *buf, int len)
2393 outptr = (unsigned char *) buf;
2397 * See if the pending-input block contains some of what we
2401 unsigned pendused = pendlen;
2402 if (pendused > outlen)
2404 memcpy(outptr, pending, pendused);
2405 memmove(pending, pending + pendused, pendlen - pendused);
2408 pendlen -= pendused;
2418 while (outlen > 0) {
2419 if (ssh_sftp_loop_iteration() < 0)
2420 return 0; /* doom */
2425 int sftp_senddata(char *buf, int len)
2427 back->send(backhandle, buf, len);
2432 * Short description of parameters.
2434 static void usage(void)
2436 printf("PuTTY Secure File Transfer (SFTP) client\n");
2437 printf("%s\n", ver);
2438 printf("Usage: psftp [options] [user@]host\n");
2439 printf("Options:\n");
2440 printf(" -b file use specified batchfile\n");
2441 printf(" -bc output batchfile commands\n");
2442 printf(" -be don't stop batchfile processing if errors\n");
2443 printf(" -v show verbose messages\n");
2444 printf(" -load sessname Load settings from saved session\n");
2445 printf(" -l user connect with specified username\n");
2446 printf(" -P port connect to specified port\n");
2447 printf(" -pw passw login with specified password\n");
2448 printf(" -1 -2 force use of particular SSH protocol version\n");
2449 printf(" -C enable compression\n");
2450 printf(" -i key private key file for authentication\n");
2451 printf(" -batch disable all interactive prompts\n");
2452 printf(" -V print version information\n");
2456 static void version(void)
2458 printf("psftp: %s\n", ver);
2463 * Connect to a host.
2465 static int psftp_connect(char *userhost, char *user, int portnumber)
2467 char *host, *realhost;
2471 /* Separate host and username */
2473 host = strrchr(host, '@');
2479 printf("psftp: multiple usernames specified; using \"%s\"\n",
2486 * If we haven't loaded session details already (e.g., from -load),
2487 * try looking for a session called "host".
2489 if (!loaded_session) {
2490 /* Try to load settings for `host' into a temporary config */
2492 cfg2.host[0] = '\0';
2493 do_defaults(host, &cfg2);
2494 if (cfg2.host[0] != '\0') {
2495 /* Settings present and include hostname */
2496 /* Re-load data into the real config. */
2497 do_defaults(host, &cfg);
2499 /* Session doesn't exist or mention a hostname. */
2500 /* Use `host' as a bare hostname. */
2501 strncpy(cfg.host, host, sizeof(cfg.host) - 1);
2502 cfg.host[sizeof(cfg.host) - 1] = '\0';
2505 /* Patch in hostname `host' to session details. */
2506 strncpy(cfg.host, host, sizeof(cfg.host) - 1);
2507 cfg.host[sizeof(cfg.host) - 1] = '\0';
2511 * Force use of SSH. (If they got the protocol wrong we assume the
2512 * port is useless too.)
2514 if (cfg.protocol != PROT_SSH) {
2515 cfg.protocol = PROT_SSH;
2520 * If saved session / Default Settings says SSH-1 (`1 only' or `1'),
2521 * then change it to SSH-2, on the grounds that that's more likely to
2522 * work for SFTP. (Can be overridden with `-1' option.)
2523 * But if it says `2 only' or `2', respect which.
2525 if (cfg.sshprot != 2 && cfg.sshprot != 3)
2529 * Enact command-line overrides.
2531 cmdline_run_saved(&cfg);
2534 * Trim leading whitespace off the hostname if it's there.
2537 int space = strspn(cfg.host, " \t");
2538 memmove(cfg.host, cfg.host+space, 1+strlen(cfg.host)-space);
2541 /* See if host is of the form user@host */
2542 if (cfg.host[0] != '\0') {
2543 char *atsign = strrchr(cfg.host, '@');
2544 /* Make sure we're not overflowing the user field */
2546 if (atsign - cfg.host < sizeof cfg.username) {
2547 strncpy(cfg.username, cfg.host, atsign - cfg.host);
2548 cfg.username[atsign - cfg.host] = '\0';
2550 memmove(cfg.host, atsign + 1, 1 + strlen(atsign + 1));
2555 * Trim a colon suffix off the hostname if it's there.
2557 cfg.host[strcspn(cfg.host, ":")] = '\0';
2560 * Remove any remaining whitespace from the hostname.
2564 while (cfg.host[p2] != '\0') {
2565 if (cfg.host[p2] != ' ' && cfg.host[p2] != '\t') {
2566 cfg.host[p1] = cfg.host[p2];
2571 cfg.host[p1] = '\0';
2575 if (user != NULL && user[0] != '\0') {
2576 strncpy(cfg.username, user, sizeof(cfg.username) - 1);
2577 cfg.username[sizeof(cfg.username) - 1] = '\0';
2579 if (!cfg.username[0]) {
2580 if (!console_get_line("login as: ",
2581 cfg.username, sizeof(cfg.username), FALSE)) {
2582 fprintf(stderr, "psftp: no username, aborting\n");
2585 int len = strlen(cfg.username);
2586 if (cfg.username[len - 1] == '\n')
2587 cfg.username[len - 1] = '\0';
2592 cfg.port = portnumber;
2595 * Disable scary things which shouldn't be enabled for simple
2596 * things like SCP and SFTP: agent forwarding, port forwarding,
2599 cfg.x11_forward = 0;
2601 cfg.portfwd[0] = cfg.portfwd[1] = '\0';
2603 /* Set up subsystem name. */
2604 strcpy(cfg.remote_cmd, "sftp");
2605 cfg.ssh_subsys = TRUE;
2609 * Set up fallback option, for SSH1 servers or servers with the
2610 * sftp subsystem not enabled but the server binary installed
2611 * in the usual place. We only support fallback on Unix
2612 * systems, and we use a kludgy piece of shellery which should
2613 * try to find sftp-server in various places (the obvious
2614 * systemwide spots /usr/lib and /usr/local/lib, and then the
2615 * user's PATH) and finally give up.
2617 * test -x /usr/lib/sftp-server && exec /usr/lib/sftp-server
2618 * test -x /usr/local/lib/sftp-server && exec /usr/local/lib/sftp-server
2621 * the idea being that this will attempt to use either of the
2622 * obvious pathnames and then give up, and when it does give up
2623 * it will print the preferred pathname in the error messages.
2625 cfg.remote_cmd_ptr2 =
2626 "test -x /usr/lib/sftp-server && exec /usr/lib/sftp-server\n"
2627 "test -x /usr/local/lib/sftp-server && exec /usr/local/lib/sftp-server\n"
2629 cfg.ssh_subsys2 = FALSE;
2631 back = &ssh_backend;
2633 err = back->init(NULL, &backhandle, &cfg, cfg.host, cfg.port, &realhost,
2634 0, cfg.tcp_keepalives);
2636 fprintf(stderr, "ssh_init: %s\n", err);
2639 logctx = log_init(NULL, &cfg);
2640 back->provide_logctx(backhandle, logctx);
2641 console_provide_logctx(logctx);
2642 while (!back->sendok(backhandle)) {
2643 if (ssh_sftp_loop_iteration() < 0) {
2644 fprintf(stderr, "ssh_init: error during SSH connection setup\n");
2648 if (verbose && realhost != NULL)
2649 printf("Connected to %s\n", realhost);
2650 if (realhost != NULL)
2655 void cmdline_error(char *p, ...)
2658 fprintf(stderr, "psftp: ");
2660 vfprintf(stderr, p, ap);
2662 fprintf(stderr, "\n try typing \"psftp -h\" for help\n");
2667 * Main program. Parse arguments etc.
2669 int psftp_main(int argc, char *argv[])
2673 char *userhost, *user;
2676 char *batchfile = NULL;
2679 flags = FLAG_STDERR | FLAG_INTERACTIVE
2680 #ifdef FLAG_SYNCAGENT
2684 cmdline_tooltype = TOOLTYPE_FILETRANSFER;
2685 ssh_get_line = &console_get_line;
2688 userhost = user = NULL;
2690 /* Load Default Settings before doing anything else. */
2691 do_defaults(NULL, &cfg);
2692 loaded_session = FALSE;
2695 for (i = 1; i < argc; i++) {
2697 if (argv[i][0] != '-') {
2701 userhost = dupstr(argv[i]);
2704 ret = cmdline_process_param(argv[i], i+1<argc?argv[i+1]:NULL, 1, &cfg);
2706 cmdline_error("option \"%s\" requires an argument", argv[i]);
2707 } else if (ret == 2) {
2708 i++; /* skip next argument */
2709 } else if (ret == 1) {
2710 /* We have our own verbosity in addition to `flags'. */
2711 if (flags & FLAG_VERBOSE)
2713 } else if (strcmp(argv[i], "-h") == 0 ||
2714 strcmp(argv[i], "-?") == 0) {
2716 } else if (strcmp(argv[i], "-V") == 0) {
2718 } else if (strcmp(argv[i], "-batch") == 0) {
2719 console_batch_mode = 1;
2720 } else if (strcmp(argv[i], "-b") == 0 && i + 1 < argc) {
2722 batchfile = argv[++i];
2723 } else if (strcmp(argv[i], "-bc") == 0) {
2724 modeflags = modeflags | 1;
2725 } else if (strcmp(argv[i], "-be") == 0) {
2726 modeflags = modeflags | 2;
2727 } else if (strcmp(argv[i], "--") == 0) {
2731 cmdline_error("unknown option \"%s\"", argv[i]);
2739 * If the loaded session provides a hostname, and a hostname has not
2740 * otherwise been specified, pop it in `userhost' so that
2741 * `psftp -load sessname' is sufficient to start a session.
2743 if (!userhost && cfg.host[0] != '\0') {
2744 userhost = dupstr(cfg.host);
2748 * If a user@host string has already been provided, connect to
2753 ret = psftp_connect(userhost, user, portnumber);
2760 printf("psftp: no hostname specified; use \"open host.name\""
2764 do_sftp(mode, modeflags, batchfile);
2766 if (back != NULL && back->socket(backhandle) != NULL) {
2768 back->special(backhandle, TS_EOF);
2769 sftp_recvdata(&ch, 1);
2774 console_provide_logctx(NULL);