2 * scp.c - Scp (Secure Copy) client for PuTTY.
3 * Joris van Rantwijk, Simon Tatham
5 * This is mainly based on ssh-1.2.26/scp.c by Timo Rinne & Tatu Ylonen.
6 * They, in turn, used stuff from BSD rcp.
8 * Adaptations to enable connecting a GUI by L. Gunnarsson - Sept 2000
25 /* GUI Adaptation - Sept 2000 */
29 #define PUTTY_DO_GLOBALS
36 #define TIME_POSIX_TO_WIN(t, ft) (*(LONGLONG*)&(ft) = \
37 ((LONGLONG) (t) + (LONGLONG) 11644473600) * (LONGLONG) 10000000)
38 #define TIME_WIN_TO_POSIX(ft, t) ((t) = (unsigned long) \
39 ((*(LONGLONG*)&(ft)) / (LONGLONG) 10000000 - (LONGLONG) 11644473600))
41 /* GUI Adaptation - Sept 2000 */
42 #define WM_APP_BASE 0x8000
43 #define WM_STD_OUT_CHAR ( WM_APP_BASE+400 )
44 #define WM_STD_ERR_CHAR ( WM_APP_BASE+401 )
45 #define WM_STATS_CHAR ( WM_APP_BASE+402 )
46 #define WM_STATS_SIZE ( WM_APP_BASE+403 )
47 #define WM_STATS_PERCENT ( WM_APP_BASE+404 )
48 #define WM_STATS_ELAPSED ( WM_APP_BASE+405 )
49 #define WM_RET_ERR_CNT ( WM_APP_BASE+406 )
50 #define WM_LS_RET_ERR_CNT ( WM_APP_BASE+407 )
53 static int verbose = 0;
54 static int recursive = 0;
55 static int preserve = 0;
56 static int targetshouldbedirectory = 0;
57 static int statistics = 1;
58 static int portnumber = 0;
59 static int prev_stats_len = 0;
60 static int scp_unsafe_mode = 0;
61 static char *password = NULL;
63 /* GUI Adaptation - Sept 2000 */
64 #define NAME_STR_MAX 2048
65 static char statname[NAME_STR_MAX + 1];
66 static unsigned long statsize = 0;
67 static int statperct = 0;
68 static unsigned long statelapsed = 0;
69 static int gui_mode = 0;
70 static char *gui_hwnd = NULL;
71 static int using_sftp = 0;
73 static void source(char *src);
74 static void rsource(char *src);
75 static void sink(char *targ, char *src);
76 /* GUI Adaptation - Sept 2000 */
77 static void tell_char(FILE * stream, char c);
78 static void tell_str(FILE * stream, char *str);
79 static void tell_user(FILE * stream, char *fmt, ...);
80 static void gui_update_stats(char *name, unsigned long size,
81 int percentage, unsigned long elapsed);
84 * The maximum amount of queued data we accept before we stop and
85 * wait for the server to process some.
87 #define MAX_SCP_BUFSIZE 16384
89 void logevent(char *string)
93 void ldisc_send(char *buf, int len)
96 * This is only here because of the calls to ldisc_send(NULL,
97 * 0) in ssh.c. Nothing in PSCP actually needs to use the ldisc
98 * as an ldisc. So if we get called with any real data, I want
104 void verify_ssh_host_key(char *host, int port, char *keytype,
105 char *keystr, char *fingerprint)
111 static const char absentmsg[] =
112 "The server's host key is not cached in the registry. You\n"
113 "have no guarantee that the server is the computer you\n"
115 "The server's key fingerprint is:\n"
117 "If you trust this host, enter \"y\" to add the key to\n"
118 "PuTTY's cache and carry on connecting.\n"
119 "If you want to carry on connecting just once, without\n"
120 "adding the key to the cache, enter \"n\".\n"
121 "If you do not trust this host, press Return to abandon the\n"
123 "Store key in cache? (y/n) ";
125 static const char wrongmsg[] =
126 "WARNING - POTENTIAL SECURITY BREACH!\n"
127 "The server's host key does not match the one PuTTY has\n"
128 "cached in the registry. This means that either the\n"
129 "server administrator has changed the host key, or you\n"
130 "have actually connected to another computer pretending\n"
131 "to be the server.\n"
132 "The new key fingerprint is:\n"
134 "If you were expecting this change and trust the new key,\n"
135 "enter \"y\" to update PuTTY's cache and continue connecting.\n"
136 "If you want to carry on connecting but without updating\n"
137 "the cache, enter \"n\".\n"
138 "If you want to abandon the connection completely, press\n"
139 "Return to cancel. Pressing Return is the ONLY guaranteed\n"
141 "Update cached key? (y/n, Return cancels connection) ";
143 static const char abandoned[] = "Connection abandoned.\n";
148 * Verify the key against the registry.
150 ret = verify_host_key(host, port, keytype, keystr);
152 if (ret == 0) /* success - key matched OK */
155 if (ret == 2) { /* key was different */
156 fprintf(stderr, wrongmsg, fingerprint);
159 if (ret == 1) { /* key was absent */
160 fprintf(stderr, absentmsg, fingerprint);
164 hin = GetStdHandle(STD_INPUT_HANDLE);
165 GetConsoleMode(hin, &savemode);
166 SetConsoleMode(hin, (savemode | ENABLE_ECHO_INPUT |
167 ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT));
168 ReadFile(hin, line, sizeof(line) - 1, &i, NULL);
169 SetConsoleMode(hin, savemode);
171 if (line[0] != '\0' && line[0] != '\r' && line[0] != '\n') {
172 if (line[0] == 'y' || line[0] == 'Y')
173 store_host_key(host, port, keytype, keystr);
175 fprintf(stderr, abandoned);
181 * Ask whether the selected cipher is acceptable (since it was
182 * below the configured 'warn' threshold).
183 * cs: 0 = both ways, 1 = client->server, 2 = server->client
185 void askcipher(char *ciphername, int cs)
190 static const char msg[] =
191 "The first %scipher supported by the server is\n"
192 "%s, which is below the configured warning threshold.\n"
193 "Continue with connection? (y/n) ";
194 static const char abandoned[] = "Connection abandoned.\n";
200 (cs == 1) ? "client-to-server " :
205 hin = GetStdHandle(STD_INPUT_HANDLE);
206 GetConsoleMode(hin, &savemode);
207 SetConsoleMode(hin, (savemode | ENABLE_ECHO_INPUT |
208 ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT));
209 ReadFile(hin, line, sizeof(line) - 1, &i, NULL);
210 SetConsoleMode(hin, savemode);
212 if (line[0] == 'y' || line[0] == 'Y') {
215 fprintf(stderr, abandoned);
220 /* GUI Adaptation - Sept 2000 */
221 static void send_msg(HWND h, UINT message, WPARAM wParam)
223 while (!PostMessage(h, message, wParam, 0))
227 static void tell_char(FILE * stream, char c)
232 unsigned int msg_id = WM_STD_OUT_CHAR;
233 if (stream == stderr)
234 msg_id = WM_STD_ERR_CHAR;
235 send_msg((HWND) atoi(gui_hwnd), msg_id, (WPARAM) c);
239 static void tell_str(FILE * stream, char *str)
243 for (i = 0; i < strlen(str); ++i)
244 tell_char(stream, str[i]);
247 static void tell_user(FILE * stream, char *fmt, ...)
249 char str[0x100]; /* Make the size big enough */
252 vsprintf(str, fmt, ap);
255 tell_str(stream, str);
258 static void gui_update_stats(char *name, unsigned long size,
259 int percentage, unsigned long elapsed)
263 if (strcmp(name, statname) != 0) {
264 for (i = 0; i < strlen(name); ++i)
265 send_msg((HWND) atoi(gui_hwnd), WM_STATS_CHAR,
267 send_msg((HWND) atoi(gui_hwnd), WM_STATS_CHAR, (WPARAM) '\n');
268 strcpy(statname, name);
270 if (statsize != size) {
271 send_msg((HWND) atoi(gui_hwnd), WM_STATS_SIZE, (WPARAM) size);
274 if (statelapsed != elapsed) {
275 send_msg((HWND) atoi(gui_hwnd), WM_STATS_ELAPSED,
277 statelapsed = elapsed;
279 if (statperct != percentage) {
280 send_msg((HWND) atoi(gui_hwnd), WM_STATS_PERCENT,
281 (WPARAM) percentage);
282 statperct = percentage;
287 * Print an error message and perform a fatal exit.
289 void fatalbox(char *fmt, ...)
291 char str[0x100]; /* Make the size big enough */
294 strcpy(str, "Fatal: ");
295 vsprintf(str + strlen(str), fmt, ap);
298 tell_str(stderr, str);
302 unsigned int msg_id = WM_RET_ERR_CNT;
304 msg_id = WM_LS_RET_ERR_CNT;
306 ((HWND) atoi(gui_hwnd), msg_id, (WPARAM) errs,
307 0 /*lParam */ ))SleepEx(1000, TRUE);
312 void connection_fatal(char *fmt, ...)
314 char str[0x100]; /* Make the size big enough */
317 strcpy(str, "Fatal: ");
318 vsprintf(str + strlen(str), fmt, ap);
321 tell_str(stderr, str);
325 unsigned int msg_id = WM_RET_ERR_CNT;
327 msg_id = WM_LS_RET_ERR_CNT;
329 ((HWND) atoi(gui_hwnd), msg_id, (WPARAM) errs,
330 0 /*lParam */ ))SleepEx(1000, TRUE);
337 * Be told what socket we're supposed to be using.
339 static SOCKET scp_ssh_socket;
340 char *do_select(SOCKET skt, int startup)
343 scp_ssh_socket = skt;
345 scp_ssh_socket = INVALID_SOCKET;
348 extern int select_result(WPARAM, LPARAM);
351 * Receive a block of data from the SSH link. Block until all data
354 * To do this, we repeatedly call the SSH protocol module, with our
355 * own trap in from_backend() to catch the data that comes back. We
356 * do this until we have enough data.
359 static unsigned char *outptr; /* where to put the data */
360 static unsigned outlen; /* how much data required */
361 static unsigned char *pending = NULL; /* any spare data */
362 static unsigned pendlen = 0, pendsize = 0; /* length and phys. size of buffer */
363 int from_backend(int is_stderr, char *data, int datalen)
365 unsigned char *p = (unsigned char *) data;
366 unsigned len = (unsigned) datalen;
369 * stderr data is just spouted to local stderr and otherwise
373 fwrite(data, 1, len, stderr);
380 * If this is before the real session begins, just return.
386 unsigned used = outlen;
389 memcpy(outptr, p, used);
397 if (pendsize < pendlen + len) {
398 pendsize = pendlen + len + 4096;
399 pending = (pending ? srealloc(pending, pendsize) :
402 fatalbox("Out of memory");
404 memcpy(pending + pendlen, p, len);
410 static int scp_process_network_event(void)
415 FD_SET(scp_ssh_socket, &readfds);
416 if (select(1, &readfds, NULL, NULL, NULL) < 0)
418 select_result((WPARAM) scp_ssh_socket, (LPARAM) FD_READ);
421 static int ssh_scp_recv(unsigned char *buf, int len)
427 * See if the pending-input block contains some of what we
431 unsigned pendused = pendlen;
432 if (pendused > outlen)
434 memcpy(outptr, pending, pendused);
435 memmove(pending, pending + pendused, pendlen - pendused);
449 if (!scp_process_network_event())
457 * Loop through the ssh connection and authentication process.
459 static void ssh_scp_init(void)
461 if (scp_ssh_socket == INVALID_SOCKET)
463 while (!back->sendok()) {
466 FD_SET(scp_ssh_socket, &readfds);
467 if (select(1, &readfds, NULL, NULL, NULL) < 0)
469 select_result((WPARAM) scp_ssh_socket, (LPARAM) FD_READ);
471 using_sftp = !ssh_fallback_cmd;
475 * Print an error message and exit after closing the SSH link.
477 static void bump(char *fmt, ...)
479 char str[0x100]; /* Make the size big enough */
482 strcpy(str, "Fatal: ");
483 vsprintf(str + strlen(str), fmt, ap);
486 tell_str(stderr, str);
489 if (back != NULL && back->socket() != NULL) {
491 back->special(TS_EOF);
492 ssh_scp_recv(&ch, 1);
496 unsigned int msg_id = WM_RET_ERR_CNT;
498 msg_id = WM_LS_RET_ERR_CNT;
500 ((HWND) atoi(gui_hwnd), msg_id, (WPARAM) errs,
501 0 /*lParam */ ))SleepEx(1000, TRUE);
507 static int get_line(const char *prompt, char *str, int maxlen, int is_pw)
510 DWORD savemode, newmode, i;
512 if (is_pw && password) {
513 static int tried_once = 0;
518 strncpy(str, password, maxlen);
519 str[maxlen - 1] = '\0';
525 /* GUI Adaptation - Sept 2000 */
530 hin = GetStdHandle(STD_INPUT_HANDLE);
531 hout = GetStdHandle(STD_OUTPUT_HANDLE);
532 if (hin == INVALID_HANDLE_VALUE || hout == INVALID_HANDLE_VALUE)
533 bump("Cannot get standard input/output handles");
535 GetConsoleMode(hin, &savemode);
536 newmode = savemode | ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT;
538 newmode &= ~ENABLE_ECHO_INPUT;
540 newmode |= ENABLE_ECHO_INPUT;
541 SetConsoleMode(hin, newmode);
543 WriteFile(hout, prompt, strlen(prompt), &i, NULL);
544 ReadFile(hin, str, maxlen - 1, &i, NULL);
546 SetConsoleMode(hin, savemode);
548 if ((int) i > maxlen)
555 WriteFile(hout, "\r\n", 2, &i, NULL);
562 * Open an SSH connection to user@host and execute cmd.
564 static void do_cmd(char *host, char *user, char *cmd)
566 char *err, *realhost;
569 if (host == NULL || host[0] == '\0')
570 bump("Empty host name");
572 /* Try to load settings for this host */
573 do_defaults(host, &cfg);
574 if (cfg.host[0] == '\0') {
575 /* No settings for this host; use defaults */
576 do_defaults(NULL, &cfg);
577 strncpy(cfg.host, host, sizeof(cfg.host) - 1);
578 cfg.host[sizeof(cfg.host) - 1] = '\0';
583 if (user != NULL && user[0] != '\0') {
584 strncpy(cfg.username, user, sizeof(cfg.username) - 1);
585 cfg.username[sizeof(cfg.username) - 1] = '\0';
586 } else if (cfg.username[0] == '\0') {
588 if (GetUserName(user, &namelen) == FALSE)
589 bump("Empty user name");
590 user = smalloc(namelen * sizeof(char));
591 GetUserName(user, &namelen);
593 tell_user(stderr, "Guessing user name: %s", user);
594 strncpy(cfg.username, user, sizeof(cfg.username) - 1);
595 cfg.username[sizeof(cfg.username) - 1] = '\0';
599 if (cfg.protocol != PROT_SSH)
603 cfg.port = portnumber;
606 * Attempt to start the SFTP subsystem as a first choice,
607 * falling back to the provided scp command if that fails.
609 strcpy(cfg.remote_cmd, "sftp");
610 cfg.ssh_subsys = TRUE;
611 cfg.remote_cmd_ptr2 = cmd;
612 cfg.ssh_subsys2 = FALSE;
617 err = back->init(cfg.host, cfg.port, &realhost);
619 bump("ssh_init: %s", err);
621 if (verbose && realhost != NULL)
622 tell_user(stderr, "Connected to %s\n", realhost);
627 * Update statistic information about current file.
629 static void print_stats(char *name, unsigned long size, unsigned long done,
630 time_t start, time_t now)
638 /* GUI Adaptation - Sept 2000 */
640 gui_update_stats(name, size, (int) (100 * (done * 1.0 / size)),
641 (unsigned long) difftime(now, start));
644 ratebs = (float) done / (now - start);
646 ratebs = (float) done;
651 eta = (unsigned long) ((size - done) / ratebs);
652 sprintf(etastr, "%02ld:%02ld:%02ld",
653 eta / 3600, (eta % 3600) / 60, eta % 60);
655 pct = (int) (100.0 * (float) done / size);
657 len = printf("\r%-25.25s | %10ld kB | %5.1f kB/s | ETA: %8s | %3d%%",
658 name, done / 1024, ratebs / 1024.0, etastr, pct);
659 if (len < prev_stats_len)
660 printf("%*s", prev_stats_len - len, "");
661 prev_stats_len = len;
669 * Find a colon in str and return a pointer to the colon.
670 * This is used to separate hostname from filename.
672 static char *colon(char *str)
674 /* We ignore a leading colon, since the hostname cannot be
675 empty. We also ignore a colon as second character because
676 of filenames like f:myfile.txt. */
677 if (str[0] == '\0' || str[0] == ':' || str[1] == ':')
679 while (*str != '\0' && *str != ':' && *str != '/' && *str != '\\')
688 * Return a pointer to the portion of str that comes after the last
689 * slash (or backslash or colon, if `local' is TRUE).
691 static char *stripslashes(char *str, int local)
696 p = strchr(str, ':');
700 p = strrchr(str, '/');
704 p = strrchr(str, '\\');
712 * Determine whether a string is entirely composed of dots.
714 static int is_dots(char *str)
716 return str[strspn(str, ".")] == '\0';
720 * Wait for a response from the other side.
721 * Return 0 if ok, -1 if error.
723 static int response(void)
725 char ch, resp, rbuf[2048];
728 if (ssh_scp_recv(&resp, 1) <= 0)
729 bump("Lost connection");
739 case 2: /* fatal error */
741 if (ssh_scp_recv(&ch, 1) <= 0)
742 bump("Protocol error: Lost connection");
744 } while (p < sizeof(rbuf) && ch != '\n');
747 tell_user(stderr, "%s\n", rbuf);
755 int sftp_recvdata(char *buf, int len)
757 return ssh_scp_recv(buf, len);
759 int sftp_senddata(char *buf, int len)
761 back->send((unsigned char *) buf, len);
765 /* ----------------------------------------------------------------------
766 * sftp-based replacement for the hacky `pscp -ls'.
768 static int sftp_ls_compare(const void *av, const void *bv)
770 const struct fxp_name *a = (const struct fxp_name *) av;
771 const struct fxp_name *b = (const struct fxp_name *) bv;
772 return strcmp(a->filename, b->filename);
774 void scp_sftp_listdir(char *dirname)
776 struct fxp_handle *dirh;
777 struct fxp_names *names;
778 struct fxp_name *ournames;
779 int nnames, namesize;
782 printf("Listing directory %s\n", dirname);
784 dirh = fxp_opendir(dirname);
786 printf("Unable to open %s: %s\n", dirname, fxp_error());
788 nnames = namesize = 0;
793 names = fxp_readdir(dirh);
795 if (fxp_error_type() == SSH_FX_EOF)
797 printf("Reading directory %s: %s\n", dirname, fxp_error());
800 if (names->nnames == 0) {
801 fxp_free_names(names);
805 if (nnames + names->nnames >= namesize) {
806 namesize += names->nnames + 128;
808 srealloc(ournames, namesize * sizeof(*ournames));
811 for (i = 0; i < names->nnames; i++)
812 ournames[nnames++] = names->names[i];
814 names->nnames = 0; /* prevent free_names */
815 fxp_free_names(names);
820 * Now we have our filenames. Sort them by actual file
821 * name, and then output the longname parts.
823 qsort(ournames, nnames, sizeof(*ournames), sftp_ls_compare);
828 for (i = 0; i < nnames; i++)
829 printf("%s\n", ournames[i].longname);
833 /* ----------------------------------------------------------------------
834 * Helper routines that contain the actual SCP protocol elements,
835 * implemented both as SCP1 and SFTP.
838 static struct scp_sftp_dirstack {
839 struct scp_sftp_dirstack *next;
840 struct fxp_name *names;
841 int namepos, namelen;
844 int matched_something; /* wildcard match set was non-empty */
845 } *scp_sftp_dirstack_head;
846 static char *scp_sftp_remotepath, *scp_sftp_currentname;
847 static char *scp_sftp_wildcard;
848 static int scp_sftp_targetisdir, scp_sftp_donethistarget;
849 static int scp_sftp_preserve, scp_sftp_recursive;
850 static unsigned long scp_sftp_mtime, scp_sftp_atime;
851 static int scp_has_times;
852 static struct fxp_handle *scp_sftp_filehandle;
853 static uint64 scp_sftp_fileoffset;
855 void scp_source_setup(char *target, int shouldbedir)
859 * Find out whether the target filespec is in fact a
862 struct fxp_attrs attrs;
864 if (!fxp_stat(target, &attrs) ||
865 !(attrs.flags & SSH_FILEXFER_ATTR_PERMISSIONS))
866 scp_sftp_targetisdir = 0;
868 scp_sftp_targetisdir = (attrs.permissions & 0040000) != 0;
870 if (shouldbedir && !scp_sftp_targetisdir) {
871 bump("pscp: remote filespec %s: not a directory\n", target);
874 scp_sftp_remotepath = dupstr(target);
882 int scp_send_errmsg(char *str)
885 /* do nothing; we never need to send our errors to the server */
887 back->send("\001", 1); /* scp protocol error prefix */
888 back->send(str, strlen(str));
890 return 0; /* can't fail */
893 int scp_send_filetimes(unsigned long mtime, unsigned long atime)
896 scp_sftp_mtime = mtime;
897 scp_sftp_atime = atime;
902 sprintf(buf, "T%lu 0 %lu 0\n", mtime, atime);
903 back->send(buf, strlen(buf));
908 int scp_send_filename(char *name, unsigned long size, int modes)
912 if (scp_sftp_targetisdir) {
913 fullname = dupcat(scp_sftp_remotepath, "/", name, NULL);
915 fullname = dupstr(scp_sftp_remotepath);
917 scp_sftp_filehandle =
918 fxp_open(fullname, SSH_FXF_WRITE | SSH_FXF_CREAT | SSH_FXF_TRUNC);
919 if (!scp_sftp_filehandle) {
920 tell_user(stderr, "pscp: unable to open %s: %s",
921 fullname, fxp_error());
925 scp_sftp_fileoffset = uint64_make(0, 0);
930 sprintf(buf, "C%04o %lu ", modes, size);
931 back->send(buf, strlen(buf));
932 back->send(name, strlen(name));
938 int scp_send_filedata(char *data, int len)
941 if (!scp_sftp_filehandle) {
944 if (!fxp_write(scp_sftp_filehandle, data, scp_sftp_fileoffset, len)) {
945 tell_user(stderr, "error while writing: %s\n", fxp_error());
949 scp_sftp_fileoffset = uint64_add32(scp_sftp_fileoffset, len);
952 int bufsize = back->send(data, len);
955 * If the network transfer is backing up - that is, the
956 * remote site is not accepting data as fast as we can
957 * produce it - then we must loop on network events until
958 * we have space in the buffer again.
960 while (bufsize > MAX_SCP_BUFSIZE) {
961 if (!scp_process_network_event())
963 bufsize = back->sendbuffer();
970 int scp_send_finish(void)
973 struct fxp_attrs attrs;
974 if (!scp_sftp_filehandle) {
978 attrs.flags = SSH_FILEXFER_ATTR_ACMODTIME;
979 attrs.atime = scp_sftp_atime;
980 attrs.mtime = scp_sftp_mtime;
981 if (!fxp_fsetstat(scp_sftp_filehandle, attrs)) {
982 tell_user(stderr, "unable to set file times: %s\n", fxp_error());
986 fxp_close(scp_sftp_filehandle);
995 char *scp_save_remotepath(void)
998 return scp_sftp_remotepath;
1003 void scp_restore_remotepath(char *data)
1006 scp_sftp_remotepath = data;
1009 int scp_send_dirname(char *name, int modes)
1014 struct fxp_attrs attrs;
1015 if (scp_sftp_targetisdir) {
1016 fullname = dupcat(scp_sftp_remotepath, "/", name, NULL);
1018 fullname = dupstr(scp_sftp_remotepath);
1022 * We don't worry about whether we managed to create the
1023 * directory, because if it exists already it's OK just to
1024 * use it. Instead, we will stat it afterwards, and if it
1025 * exists and is a directory we will assume we were either
1026 * successful or it didn't matter.
1028 if (!fxp_mkdir(fullname))
1031 err = "server reported no error";
1032 if (!fxp_stat(fullname, &attrs) ||
1033 !(attrs.flags & SSH_FILEXFER_ATTR_PERMISSIONS) ||
1034 !(attrs.permissions & 0040000)) {
1035 tell_user(stderr, "unable to create directory %s: %s",
1041 scp_sftp_remotepath = fullname;
1046 sprintf(buf, "D%04o 0 ", modes);
1047 back->send(buf, strlen(buf));
1048 back->send(name, strlen(name));
1049 back->send("\n", 1);
1054 int scp_send_enddir(void)
1057 sfree(scp_sftp_remotepath);
1060 back->send("E\n", 2);
1066 * Yes, I know; I have an scp_sink_setup _and_ an scp_sink_init.
1067 * That's bad. The difference is that scp_sink_setup is called once
1068 * right at the start, whereas scp_sink_init is called to
1069 * initialise every level of recursion in the protocol.
1071 int scp_sink_setup(char *source, int preserve, int recursive)
1076 * It's possible that the source string we've been given
1077 * contains a wildcard. If so, we must split the directory
1078 * away from the wildcard itself (throwing an error if any
1079 * wildcardness comes before the final slash) and arrange
1080 * things so that a dirstack entry will be set up.
1082 newsource = smalloc(1+strlen(source));
1083 if (!wc_unescape(newsource, source)) {
1084 /* Yes, here we go; it's a wildcard. Bah. */
1085 char *dupsource, *lastpart, *dirpart, *wildcard;
1086 dupsource = dupstr(source);
1087 lastpart = stripslashes(dupsource, 0);
1088 wildcard = dupstr(lastpart);
1090 if (*dupsource && dupsource[1]) {
1092 * The remains of dupsource are at least two
1093 * characters long, meaning the pathname wasn't
1094 * empty or just `/'. Hence, we remove the trailing
1097 lastpart[-1] = '\0';
1098 } else if (!*dupsource) {
1100 * The remains of dupsource are _empty_ - the whole
1101 * pathname was a wildcard. Hence we need to
1102 * replace it with ".".
1105 dupsource = dupstr(".");
1109 * Now we have separated our string into dupsource (the
1110 * directory part) and wildcard. Both of these will
1111 * need freeing at some point. Next step is to remove
1112 * wildcard escapes from the directory part, throwing
1113 * an error if it contains a real wildcard.
1115 dirpart = smalloc(1+strlen(dupsource));
1116 if (!wc_unescape(dirpart, dupsource)) {
1117 tell_user(stderr, "%s: multiple-level wildcards unsupported",
1127 * Now we have dirpart (unescaped, ie a valid remote
1128 * path), and wildcard (a wildcard). This will be
1129 * sufficient to arrange a dirstack entry.
1131 scp_sftp_remotepath = dirpart;
1132 scp_sftp_wildcard = wildcard;
1135 scp_sftp_remotepath = newsource;
1136 scp_sftp_wildcard = NULL;
1138 scp_sftp_preserve = preserve;
1139 scp_sftp_recursive = recursive;
1140 scp_sftp_donethistarget = 0;
1141 scp_sftp_dirstack_head = NULL;
1146 int scp_sink_init(void)
1154 #define SCP_SINK_FILE 1
1155 #define SCP_SINK_DIR 2
1156 #define SCP_SINK_ENDDIR 3
1157 #define SCP_SINK_RETRY 4 /* not an action; just try again */
1158 struct scp_sink_action {
1159 int action; /* FILE, DIR, ENDDIR */
1160 char *buf; /* will need freeing after use */
1161 char *name; /* filename or dirname (not ENDDIR) */
1162 int mode; /* access mode (not ENDDIR) */
1163 unsigned long size; /* file size (not ENDDIR) */
1164 int settime; /* 1 if atime and mtime are filled */
1165 unsigned long atime, mtime; /* access times for the file */
1168 int scp_get_sink_action(struct scp_sink_action *act)
1172 int must_free_fname;
1173 struct fxp_attrs attrs;
1176 if (!scp_sftp_dirstack_head) {
1177 if (!scp_sftp_donethistarget) {
1179 * Simple case: we are only dealing with one file.
1181 fname = scp_sftp_remotepath;
1182 must_free_fname = 0;
1183 scp_sftp_donethistarget = 1;
1186 * Even simpler case: one file _which we've done_.
1187 * Return 1 (finished).
1193 * We're now in the middle of stepping through a list
1194 * of names returned from fxp_readdir(); so let's carry
1197 struct scp_sftp_dirstack *head = scp_sftp_dirstack_head;
1198 while (head->namepos < head->namelen &&
1199 (is_dots(head->names[head->namepos].filename) ||
1201 !wc_match(head->wildcard,
1202 head->names[head->namepos].filename))))
1203 head->namepos++; /* skip . and .. */
1204 if (head->namepos < head->namelen) {
1205 head->matched_something = 1;
1206 fname = dupcat(head->dirpath, "/",
1207 head->names[head->namepos++].filename,
1209 must_free_fname = 1;
1212 * We've come to the end of the list; pop it off
1213 * the stack and return an ENDDIR action (or RETRY
1214 * if this was a wildcard match).
1216 if (head->wildcard) {
1217 act->action = SCP_SINK_RETRY;
1218 if (!head->matched_something) {
1219 tell_user(stderr, "pscp: wildcard '%s' matched "
1220 "no files", head->wildcard);
1223 sfree(head->wildcard);
1226 act->action = SCP_SINK_ENDDIR;
1229 sfree(head->dirpath);
1231 scp_sftp_dirstack_head = head->next;
1239 * Now we have a filename. Stat it, and see if it's a file
1242 ret = fxp_stat(fname, &attrs);
1243 if (!ret || !(attrs.flags & SSH_FILEXFER_ATTR_PERMISSIONS)) {
1244 tell_user(stderr, "unable to identify %s: %s", fname,
1245 ret ? "file type not supplied" : fxp_error());
1250 if (attrs.permissions & 0040000) {
1251 struct scp_sftp_dirstack *newitem;
1252 struct fxp_handle *dirhandle;
1253 int nnames, namesize;
1254 struct fxp_name *ournames;
1255 struct fxp_names *names;
1258 * It's a directory. If we're not in recursive mode,
1259 * this merits a complaint (which is fatal if the name
1260 * was specified directly, but not if it was matched by
1263 * We skip this complaint completely if
1264 * scp_sftp_wildcard is set, because that's an
1265 * indication that we're not actually supposed to
1266 * _recursively_ transfer the dir, just scan it for
1267 * things matching the wildcard.
1269 if (!scp_sftp_recursive && !scp_sftp_wildcard) {
1270 tell_user(stderr, "pscp: %s: is a directory", fname);
1272 if (must_free_fname) sfree(fname);
1273 if (scp_sftp_dirstack_head) {
1274 act->action = SCP_SINK_RETRY;
1282 * Otherwise, the fun begins. We must fxp_opendir() the
1283 * directory, slurp the filenames into memory, return
1284 * SCP_SINK_DIR (unless this is a wildcard match), and
1285 * set targetisdir. The next time we're called, we will
1286 * run through the list of filenames one by one,
1287 * matching them against a wildcard if present.
1289 * If targetisdir is _already_ set (meaning we're
1290 * already in the middle of going through another such
1291 * list), we must push the other (target,namelist) pair
1294 dirhandle = fxp_opendir(fname);
1296 tell_user(stderr, "scp: unable to open directory %s: %s",
1297 fname, fxp_error());
1298 if (must_free_fname) sfree(fname);
1302 nnames = namesize = 0;
1307 names = fxp_readdir(dirhandle);
1308 if (names == NULL) {
1309 if (fxp_error_type() == SSH_FX_EOF)
1311 tell_user(stderr, "scp: reading directory %s: %s\n",
1312 fname, fxp_error());
1313 if (must_free_fname) sfree(fname);
1318 if (names->nnames == 0) {
1319 fxp_free_names(names);
1322 if (nnames + names->nnames >= namesize) {
1323 namesize += names->nnames + 128;
1325 srealloc(ournames, namesize * sizeof(*ournames));
1327 for (i = 0; i < names->nnames; i++)
1328 ournames[nnames++] = names->names[i];
1329 names->nnames = 0; /* prevent free_names */
1330 fxp_free_names(names);
1332 fxp_close(dirhandle);
1334 newitem = smalloc(sizeof(struct scp_sftp_dirstack));
1335 newitem->next = scp_sftp_dirstack_head;
1336 newitem->names = ournames;
1337 newitem->namepos = 0;
1338 newitem->namelen = nnames;
1339 if (must_free_fname)
1340 newitem->dirpath = fname;
1342 newitem->dirpath = dupstr(fname);
1343 if (scp_sftp_wildcard) {
1344 newitem->wildcard = scp_sftp_wildcard;
1345 newitem->matched_something = 0;
1346 scp_sftp_wildcard = NULL;
1348 newitem->wildcard = NULL;
1350 scp_sftp_dirstack_head = newitem;
1352 if (newitem->wildcard) {
1353 act->action = SCP_SINK_RETRY;
1355 act->action = SCP_SINK_DIR;
1356 act->buf = dupstr(stripslashes(fname, 0));
1357 act->name = act->buf;
1358 act->size = 0; /* duhh, it's a directory */
1359 act->mode = 07777 & attrs.permissions;
1360 if (scp_sftp_preserve &&
1361 (attrs.flags & SSH_FILEXFER_ATTR_ACMODTIME)) {
1362 act->atime = attrs.atime;
1363 act->mtime = attrs.mtime;
1372 * It's a file. Return SCP_SINK_FILE.
1374 act->action = SCP_SINK_FILE;
1375 act->buf = dupstr(stripslashes(fname, 0));
1376 act->name = act->buf;
1377 if (attrs.flags & SSH_FILEXFER_ATTR_SIZE) {
1378 if (uint64_compare(attrs.size,
1379 uint64_make(0, ULONG_MAX)) > 0) {
1380 act->size = ULONG_MAX; /* *boggle* */
1382 act->size = attrs.size.lo;
1384 act->size = ULONG_MAX; /* no idea */
1385 act->mode = 07777 & attrs.permissions;
1386 if (scp_sftp_preserve &&
1387 (attrs.flags & SSH_FILEXFER_ATTR_ACMODTIME)) {
1388 act->atime = attrs.atime;
1389 act->mtime = attrs.mtime;
1393 if (must_free_fname)
1394 scp_sftp_currentname = fname;
1396 scp_sftp_currentname = dupstr(fname);
1411 if (ssh_scp_recv(&ch, 1) <= 0)
1414 bump("Protocol error: Unexpected newline");
1418 if (ssh_scp_recv(&ch, 1) <= 0)
1419 bump("Lost connection");
1422 act->buf = srealloc(act->buf, bufsize);
1425 } while (ch != '\n');
1426 act->buf[i - 1] = '\0';
1428 case '\01': /* error */
1429 tell_user(stderr, "%s\n", act->buf);
1431 continue; /* go round again */
1432 case '\02': /* fatal error */
1433 bump("%s", act->buf);
1436 act->action = SCP_SINK_ENDDIR;
1439 if (sscanf(act->buf, "%ld %*d %ld %*d",
1440 &act->mtime, &act->atime) == 2) {
1443 continue; /* go round again */
1445 bump("Protocol error: Illegal time format");
1448 act->action = (action == 'C' ? SCP_SINK_FILE : SCP_SINK_DIR);
1451 bump("Protocol error: Expected control record");
1454 * We will go round this loop only once, unless we hit
1461 * If we get here, we must have seen SCP_SINK_FILE or
1464 if (sscanf(act->buf, "%o %lu %n", &act->mode, &act->size, &i) != 2)
1465 bump("Protocol error: Illegal file descriptor format");
1466 act->name = act->buf + i;
1471 int scp_accept_filexfer(void)
1474 scp_sftp_filehandle =
1475 fxp_open(scp_sftp_currentname, SSH_FXF_READ);
1476 if (!scp_sftp_filehandle) {
1477 tell_user(stderr, "pscp: unable to open %s: %s",
1478 scp_sftp_currentname, fxp_error());
1482 scp_sftp_fileoffset = uint64_make(0, 0);
1483 sfree(scp_sftp_currentname);
1487 return 0; /* can't fail */
1491 int scp_recv_filedata(char *data, int len)
1494 int actuallen = fxp_read(scp_sftp_filehandle, data,
1495 scp_sftp_fileoffset, len);
1496 if (actuallen == -1 && fxp_error_type() != SSH_FX_EOF) {
1497 tell_user(stderr, "pscp: error while reading: %s", fxp_error());
1504 scp_sftp_fileoffset = uint64_add32(scp_sftp_fileoffset, actuallen);
1508 return ssh_scp_recv(data, len);
1512 int scp_finish_filerecv(void)
1515 fxp_close(scp_sftp_filehandle);
1523 /* ----------------------------------------------------------------------
1524 * Send an error message to the other side and to the screen.
1525 * Increment error counter.
1527 static void run_err(const char *fmt, ...)
1533 strcpy(str, "scp: ");
1534 vsprintf(str + strlen(str), fmt, ap);
1536 scp_send_errmsg(str);
1537 tell_user(stderr, "%s", str);
1542 * Execute the source part of the SCP protocol.
1544 static void source(char *src)
1551 unsigned long stat_bytes;
1552 time_t stat_starttime, stat_lasttime;
1554 attr = GetFileAttributes(src);
1555 if (attr == (DWORD) - 1) {
1556 run_err("%s: No such file or directory", src);
1560 if ((attr & FILE_ATTRIBUTE_DIRECTORY) != 0) {
1563 * Avoid . and .. directories.
1566 p = strrchr(src, '/');
1568 p = strrchr(src, '\\');
1573 if (!strcmp(p, ".") || !strcmp(p, ".."))
1574 /* skip . and .. */ ;
1578 run_err("%s: not a regular file", src);
1583 if ((last = strrchr(src, '/')) == NULL)
1587 if (strrchr(last, '\\') != NULL)
1588 last = strrchr(last, '\\') + 1;
1589 if (last == src && strchr(src, ':') != NULL)
1590 last = strchr(src, ':') + 1;
1592 f = CreateFile(src, GENERIC_READ, FILE_SHARE_READ, NULL,
1593 OPEN_EXISTING, 0, 0);
1594 if (f == INVALID_HANDLE_VALUE) {
1595 run_err("%s: Cannot open file", src);
1600 FILETIME actime, wrtime;
1601 unsigned long mtime, atime;
1602 GetFileTime(f, NULL, &actime, &wrtime);
1603 TIME_WIN_TO_POSIX(actime, atime);
1604 TIME_WIN_TO_POSIX(wrtime, mtime);
1605 if (scp_send_filetimes(mtime, atime))
1609 size = GetFileSize(f, NULL);
1611 tell_user(stderr, "Sending file %s, size=%lu", last, size);
1612 if (scp_send_filename(last, size, 0644))
1616 stat_starttime = time(NULL);
1619 for (i = 0; i < size; i += 4096) {
1620 char transbuf[4096];
1625 if (!ReadFile(f, transbuf, k, &j, NULL) || j != k) {
1628 bump("%s: Read error", src);
1630 if (scp_send_filedata(transbuf, k))
1631 bump("%s: Network error occurred", src);
1635 if (time(NULL) != stat_lasttime || i + k == size) {
1636 stat_lasttime = time(NULL);
1637 print_stats(last, size, stat_bytes,
1638 stat_starttime, stat_lasttime);
1645 (void) scp_send_finish();
1649 * Recursively send the contents of a directory.
1651 static void rsource(char *src)
1653 char *last, *findfile;
1656 WIN32_FIND_DATA fdat;
1659 if ((last = strrchr(src, '/')) == NULL)
1663 if (strrchr(last, '\\') != NULL)
1664 last = strrchr(last, '\\') + 1;
1665 if (last == src && strchr(src, ':') != NULL)
1666 last = strchr(src, ':') + 1;
1668 /* maybe send filetime */
1670 save_target = scp_save_remotepath();
1673 tell_user(stderr, "Entering directory: %s", last);
1674 if (scp_send_dirname(last, 0755))
1677 findfile = dupcat(src, "/*", NULL);
1678 dir = FindFirstFile(findfile, &fdat);
1679 ok = (dir != INVALID_HANDLE_VALUE);
1681 if (strcmp(fdat.cFileName, ".") == 0 ||
1682 strcmp(fdat.cFileName, "..") == 0) {
1683 /* ignore . and .. */
1685 char *foundfile = dupcat(src, "/", fdat.cFileName, NULL);
1689 ok = FindNextFile(dir, &fdat);
1694 (void) scp_send_enddir();
1696 scp_restore_remotepath(save_target);
1700 * Execute the sink part of the SCP protocol.
1702 static void sink(char *targ, char *src)
1709 unsigned long received;
1711 unsigned long stat_bytes;
1712 time_t stat_starttime, stat_lasttime;
1715 attr = GetFileAttributes(targ);
1716 if (attr != (DWORD) - 1 && (attr & FILE_ATTRIBUTE_DIRECTORY) != 0)
1719 if (targetshouldbedirectory && !targisdir)
1720 bump("%s: Not a directory", targ);
1724 struct scp_sink_action act;
1725 if (scp_get_sink_action(&act))
1728 if (act.action == SCP_SINK_ENDDIR)
1731 if (act.action == SCP_SINK_RETRY)
1736 * Prevent the remote side from maliciously writing to
1737 * files outside the target area by sending a filename
1738 * containing `../'. In fact, it shouldn't be sending
1739 * filenames with any slashes or colons in at all; so
1740 * we'll find the last slash, backslash or colon in the
1741 * filename and use only the part after that. (And
1744 * In addition, we also ensure here that if we're
1745 * copying a single file and the target is a directory
1746 * (common usage: `pscp host:filename .') the remote
1747 * can't send us a _different_ file name. We can
1748 * distinguish this case because `src' will be non-NULL
1749 * and the last component of that will fail to match
1750 * (the last component of) the name sent.
1752 * Well, not always; if `src' is a wildcard, we do
1753 * expect to get back filenames that don't correspond
1754 * exactly to it. Ideally in this case, we would like
1755 * to ensure that the returned filename actually
1756 * matches the wildcard pattern - but one of SCP's
1757 * protocol infelicities is that wildcard matching is
1758 * done at the server end _by the server's rules_ and
1759 * so in general this is infeasible. Hence, we only
1760 * accept filenames that don't correspond to `src' if
1761 * unsafe mode is enabled or we are using SFTP (which
1762 * resolves remote wildcards on the client side and can
1765 char *striptarget, *stripsrc;
1767 striptarget = stripslashes(act.name, 1);
1768 if (striptarget != act.name) {
1769 tell_user(stderr, "warning: remote host sent a compound"
1770 " pathname '%s'", act.name);
1771 tell_user(stderr, " renaming local file to '%s'",
1776 * Also check to see if the target filename is '.' or
1777 * '..', or indeed '...' and so on because Windows
1778 * appears to interpret those like '..'.
1780 if (is_dots(striptarget)) {
1781 bump("security violation: remote host attempted to write to"
1782 " a '.' or '..' path!");
1786 stripsrc = stripslashes(src, 1);
1787 if (strcmp(striptarget, stripsrc) &&
1788 !using_sftp && !scp_unsafe_mode) {
1789 tell_user(stderr, "warning: remote host tried to write "
1790 "to a file called '%s'", striptarget);
1791 tell_user(stderr, " when we requested a file "
1792 "called '%s'.", stripsrc);
1793 tell_user(stderr, " If this is a wildcard, "
1794 "consider upgrading to SSH 2 or using");
1795 tell_user(stderr, " the '-unsafe' option. Renaming"
1796 " of this file has been disallowed.");
1797 /* Override the name the server provided with our own. */
1798 striptarget = stripsrc;
1802 if (targ[0] != '\0')
1803 destfname = dupcat(targ, "\\", striptarget, NULL);
1805 destfname = dupstr(striptarget);
1808 * In this branch of the if, the target area is a
1809 * single file with an explicitly specified name in any
1810 * case, so there's no danger.
1812 destfname = dupstr(targ);
1814 attr = GetFileAttributes(destfname);
1815 exists = (attr != (DWORD) - 1);
1817 if (act.action == SCP_SINK_DIR) {
1818 if (exists && (attr & FILE_ATTRIBUTE_DIRECTORY) == 0) {
1819 run_err("%s: Not a directory", destfname);
1823 if (!CreateDirectory(destfname, NULL)) {
1824 run_err("%s: Cannot create directory", destfname);
1828 sink(destfname, NULL);
1829 /* can we set the timestamp for directories ? */
1833 f = CreateFile(destfname, GENERIC_WRITE, 0, NULL,
1834 CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
1835 if (f == INVALID_HANDLE_VALUE) {
1836 run_err("%s: Cannot create file", destfname);
1840 if (scp_accept_filexfer())
1844 stat_starttime = time(NULL);
1846 stat_name = stripslashes(destfname, 1);
1849 while (received < act.size) {
1850 char transbuf[4096];
1851 DWORD blksize, read, written;
1853 if (blksize > act.size - received)
1854 blksize = act.size - received;
1855 read = scp_recv_filedata(transbuf, blksize);
1857 bump("Lost connection");
1860 if (!WriteFile(f, transbuf, read, &written, NULL) ||
1863 /* FIXME: in sftp we can actually abort the transfer */
1865 printf("\r%-25.25s | %50s\n",
1867 "Write error.. waiting for end of file");
1872 if (time(NULL) > stat_lasttime ||
1873 received + read == act.size) {
1874 stat_lasttime = time(NULL);
1875 print_stats(stat_name, act.size, stat_bytes,
1876 stat_starttime, stat_lasttime);
1882 FILETIME actime, wrtime;
1883 TIME_POSIX_TO_WIN(act.atime, actime);
1884 TIME_POSIX_TO_WIN(act.mtime, wrtime);
1885 SetFileTime(f, NULL, &actime, &wrtime);
1890 run_err("%s: Write error", destfname);
1893 (void) scp_finish_filerecv();
1900 * We will copy local files to a remote server.
1902 static void toremote(int argc, char *argv[])
1904 char *src, *targ, *host, *user;
1908 targ = argv[argc - 1];
1910 /* Separate host from filename */
1914 bump("targ == NULL in toremote()");
1918 /* Substitute "." for emtpy target */
1920 /* Separate host and username */
1922 host = strrchr(host, '@');
1933 /* Find out if the source filespec covers multiple files
1934 if so, we should set the targetshouldbedirectory flag */
1936 WIN32_FIND_DATA fdat;
1937 if (colon(argv[0]) != NULL)
1938 bump("%s: Remote to remote not supported", argv[0]);
1939 fh = FindFirstFile(argv[0], &fdat);
1940 if (fh == INVALID_HANDLE_VALUE)
1941 bump("%s: No such file or directory\n", argv[0]);
1942 if (FindNextFile(fh, &fdat))
1943 targetshouldbedirectory = 1;
1947 cmd = smalloc(strlen(targ) + 100);
1948 sprintf(cmd, "scp%s%s%s%s -t %s",
1949 verbose ? " -v" : "",
1950 recursive ? " -r" : "",
1951 preserve ? " -p" : "",
1952 targetshouldbedirectory ? " -d" : "", targ);
1953 do_cmd(host, user, cmd);
1956 scp_source_setup(targ, targetshouldbedirectory);
1958 for (i = 0; i < argc - 1; i++) {
1959 char *srcpath, *last;
1961 WIN32_FIND_DATA fdat;
1963 if (colon(src) != NULL) {
1964 tell_user(stderr, "%s: Remote to remote not supported\n", src);
1970 * Trim off the last pathname component of `src', to
1971 * provide the base pathname which will be prepended to
1972 * filenames returned from Find{First,Next}File.
1974 srcpath = dupstr(src);
1975 last = stripslashes(srcpath, 1);
1978 dir = FindFirstFile(src, &fdat);
1979 if (dir == INVALID_HANDLE_VALUE) {
1980 run_err("%s: No such file or directory", src);
1986 * Ensure that . and .. are never matched by wildcards,
1987 * but only by deliberate action.
1989 if (!strcmp(fdat.cFileName, ".") ||
1990 !strcmp(fdat.cFileName, "..")) {
1992 * Find*File has returned a special dir. We require
1993 * that _either_ `src' ends in a backslash followed
1994 * by that string, _or_ `src' is precisely that
1997 int len = strlen(src), dlen = strlen(fdat.cFileName);
1998 if (len == dlen && !strcmp(src, fdat.cFileName)) {
2000 } else if (len > dlen + 1 && src[len - dlen - 1] == '\\' &&
2001 !strcmp(src + len - dlen, fdat.cFileName)) {
2004 continue; /* ignore this one */
2006 filename = dupcat(srcpath, fdat.cFileName, NULL);
2009 } while (FindNextFile(dir, &fdat));
2016 * We will copy files from a remote server to the local machine.
2018 static void tolocal(int argc, char *argv[])
2020 char *src, *targ, *host, *user;
2024 bump("More than one remote source not supported");
2029 /* Separate host from filename */
2033 bump("Local to local copy not supported");
2037 /* Substitute "." for empty filename */
2039 /* Separate username and hostname */
2041 host = strrchr(host, '@');
2051 cmd = smalloc(strlen(src) + 100);
2052 sprintf(cmd, "scp%s%s%s%s -f %s",
2053 verbose ? " -v" : "",
2054 recursive ? " -r" : "",
2055 preserve ? " -p" : "",
2056 targetshouldbedirectory ? " -d" : "", src);
2057 do_cmd(host, user, cmd);
2060 if (scp_sink_setup(src, preserve, recursive))
2067 * We will issue a list command to get a remote directory.
2069 static void get_dir_list(int argc, char *argv[])
2071 char *src, *host, *user;
2077 /* Separate host from filename */
2081 bump("Local to local copy not supported");
2085 /* Substitute "." for empty filename */
2087 /* Separate username and hostname */
2089 host = strrchr(host, '@');
2099 cmd = smalloc(4 * strlen(src) + 100);
2100 strcpy(cmd, "ls -la '");
2101 p = cmd + strlen(cmd);
2102 for (q = src; *q; q++) {
2115 do_cmd(host, user, cmd);
2119 scp_sftp_listdir(src);
2121 while (ssh_scp_recv(&c, 1) > 0)
2122 tell_char(stdout, c);
2127 * Initialize the Win$ock driver.
2129 static void init_winsock(void)
2134 winsock_ver = MAKEWORD(1, 1);
2135 if (WSAStartup(winsock_ver, &wsadata))
2136 bump("Unable to initialise WinSock");
2137 if (LOBYTE(wsadata.wVersion) != 1 || HIBYTE(wsadata.wVersion) != 1)
2138 bump("WinSock version is incompatible with 1.1");
2142 * Short description of parameters.
2144 static void usage(void)
2146 printf("PuTTY Secure Copy client\n");
2147 printf("%s\n", ver);
2148 printf("Usage: pscp [options] [user@]host:source target\n");
2150 (" pscp [options] source [source...] [user@]host:target\n");
2151 printf(" pscp [options] -ls user@host:filespec\n");
2152 printf("Options:\n");
2153 printf(" -p preserve file attributes\n");
2154 printf(" -q quiet, don't show statistics\n");
2155 printf(" -r copy directories recursively\n");
2156 printf(" -v show verbose messages\n");
2157 printf(" -P port connect to specified port\n");
2158 printf(" -pw passw login with specified password\n");
2159 printf(" -unsafe allow server-side wildcards (DANGEROUS)\n");
2162 * -gui is an internal option, used by GUI front ends to get
2163 * pscp to pass progress reports back to them. It's not an
2164 * ordinary user-accessible option, so it shouldn't be part of
2165 * the command-line help. The only people who need to know
2166 * about it are programmers, and they can read the source.
2169 (" -gui hWnd GUI mode with the windows handle for receiving messages\n");
2175 * Main program (no, really?)
2177 int main(int argc, char *argv[])
2181 default_protocol = PROT_TELNET;
2183 flags = FLAG_STDERR;
2184 ssh_get_line = &get_line;
2188 for (i = 1; i < argc; i++) {
2189 if (argv[i][0] != '-')
2191 if (strcmp(argv[i], "-v") == 0)
2192 verbose = 1, flags |= FLAG_VERBOSE;
2193 else if (strcmp(argv[i], "-r") == 0)
2195 else if (strcmp(argv[i], "-p") == 0)
2197 else if (strcmp(argv[i], "-q") == 0)
2199 else if (strcmp(argv[i], "-h") == 0 || strcmp(argv[i], "-?") == 0)
2201 else if (strcmp(argv[i], "-P") == 0 && i + 1 < argc)
2202 portnumber = atoi(argv[++i]);
2203 else if (strcmp(argv[i], "-pw") == 0 && i + 1 < argc)
2204 password = argv[++i];
2205 else if (strcmp(argv[i], "-gui") == 0 && i + 1 < argc) {
2206 gui_hwnd = argv[++i];
2208 } else if (strcmp(argv[i], "-ls") == 0)
2210 else if (strcmp(argv[i], "-unsafe") == 0)
2211 scp_unsafe_mode = 1;
2212 else if (strcmp(argv[i], "--") == 0) {
2225 get_dir_list(argc, argv);
2232 targetshouldbedirectory = 1;
2234 if (colon(argv[argc - 1]) != NULL)
2235 toremote(argc, argv);
2237 tolocal(argc, argv);
2240 if (back != NULL && back->socket() != NULL) {
2242 back->special(TS_EOF);
2243 ssh_scp_recv(&ch, 1);
2248 /* GUI Adaptation - August 2000 */
2250 unsigned int msg_id = WM_RET_ERR_CNT;
2252 msg_id = WM_LS_RET_ERR_CNT;
2254 ((HWND) atoi(gui_hwnd), msg_id, (WPARAM) errs,
2255 0 /*lParam */ ))SleepEx(1000, TRUE);
2257 return (errs == 0 ? 0 : 1);