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
24 /* GUI Adaptation - Sept 2000 */
28 #define PUTTY_DO_GLOBALS
33 #define TIME_POSIX_TO_WIN(t, ft) (*(LONGLONG*)&(ft) = \
34 ((LONGLONG) (t) + (LONGLONG) 11644473600) * (LONGLONG) 10000000)
35 #define TIME_WIN_TO_POSIX(ft, t) ((t) = (unsigned long) \
36 ((*(LONGLONG*)&(ft)) / (LONGLONG) 10000000 - (LONGLONG) 11644473600))
38 /* GUI Adaptation - Sept 2000 */
39 #define WM_APP_BASE 0x8000
40 #define WM_STD_OUT_CHAR ( WM_APP_BASE+400 )
41 #define WM_STD_ERR_CHAR ( WM_APP_BASE+401 )
42 #define WM_STATS_CHAR ( WM_APP_BASE+402 )
43 #define WM_STATS_SIZE ( WM_APP_BASE+403 )
44 #define WM_STATS_PERCENT ( WM_APP_BASE+404 )
45 #define WM_STATS_ELAPSED ( WM_APP_BASE+405 )
46 #define WM_RET_ERR_CNT ( WM_APP_BASE+406 )
47 #define WM_LS_RET_ERR_CNT ( WM_APP_BASE+407 )
50 static int verbose
= 0;
51 static int recursive
= 0;
52 static int preserve
= 0;
53 static int targetshouldbedirectory
= 0;
54 static int statistics
= 1;
55 static int portnumber
= 0;
56 static int prev_stats_len
= 0;
57 static char *password
= NULL
;
59 /* GUI Adaptation - Sept 2000 */
60 #define NAME_STR_MAX 2048
61 static char statname
[NAME_STR_MAX
+ 1];
62 static unsigned long statsize
= 0;
63 static int statperct
= 0;
64 static unsigned long statelapsed
= 0;
65 static int gui_mode
= 0;
66 static char *gui_hwnd
= NULL
;
68 static void source(char *src
);
69 static void rsource(char *src
);
70 static void sink(char *targ
, char *src
);
71 /* GUI Adaptation - Sept 2000 */
72 static void tell_char(FILE * stream
, char c
);
73 static void tell_str(FILE * stream
, char *str
);
74 static void tell_user(FILE * stream
, char *fmt
, ...);
75 static void gui_update_stats(char *name
, unsigned long size
,
76 int percentage
, unsigned long elapsed
);
79 * The maximum amount of queued data we accept before we stop and
80 * wait for the server to process some.
82 #define MAX_SCP_BUFSIZE 16384
84 void logevent(char *string
)
88 void ldisc_send(char *buf
, int len
)
91 * This is only here because of the calls to ldisc_send(NULL,
92 * 0) in ssh.c. Nothing in PSCP actually needs to use the ldisc
93 * as an ldisc. So if we get called with any real data, I want
99 void verify_ssh_host_key(char *host
, int port
, char *keytype
,
100 char *keystr
, char *fingerprint
)
106 static const char absentmsg
[] =
107 "The server's host key is not cached in the registry. You\n"
108 "have no guarantee that the server is the computer you\n"
110 "The server's key fingerprint is:\n"
112 "If you trust this host, enter \"y\" to add the key to\n"
113 "PuTTY's cache and carry on connecting.\n"
114 "If you want to carry on connecting just once, without\n"
115 "adding the key to the cache, enter \"n\".\n"
116 "If you do not trust this host, press Return to abandon the\n"
118 "Store key in cache? (y/n) ";
120 static const char wrongmsg
[] =
121 "WARNING - POTENTIAL SECURITY BREACH!\n"
122 "The server's host key does not match the one PuTTY has\n"
123 "cached in the registry. This means that either the\n"
124 "server administrator has changed the host key, or you\n"
125 "have actually connected to another computer pretending\n"
126 "to be the server.\n"
127 "The new key fingerprint is:\n"
129 "If you were expecting this change and trust the new key,\n"
130 "enter \"y\" to update PuTTY's cache and continue connecting.\n"
131 "If you want to carry on connecting but without updating\n"
132 "the cache, enter \"n\".\n"
133 "If you want to abandon the connection completely, press\n"
134 "Return to cancel. Pressing Return is the ONLY guaranteed\n"
136 "Update cached key? (y/n, Return cancels connection) ";
138 static const char abandoned
[] = "Connection abandoned.\n";
143 * Verify the key against the registry.
145 ret
= verify_host_key(host
, port
, keytype
, keystr
);
147 if (ret
== 0) /* success - key matched OK */
150 if (ret
== 2) { /* key was different */
151 fprintf(stderr
, wrongmsg
, fingerprint
);
154 if (ret
== 1) { /* key was absent */
155 fprintf(stderr
, absentmsg
, fingerprint
);
159 hin
= GetStdHandle(STD_INPUT_HANDLE
);
160 GetConsoleMode(hin
, &savemode
);
161 SetConsoleMode(hin
, (savemode
| ENABLE_ECHO_INPUT
|
162 ENABLE_PROCESSED_INPUT
| ENABLE_LINE_INPUT
));
163 ReadFile(hin
, line
, sizeof(line
) - 1, &i
, NULL
);
164 SetConsoleMode(hin
, savemode
);
166 if (line
[0] != '\0' && line
[0] != '\r' && line
[0] != '\n') {
167 if (line
[0] == 'y' || line
[0] == 'Y')
168 store_host_key(host
, port
, keytype
, keystr
);
170 fprintf(stderr
, abandoned
);
176 * Ask whether the selected cipher is acceptable (since it was
177 * below the configured 'warn' threshold).
178 * cs: 0 = both ways, 1 = client->server, 2 = server->client
180 void askcipher(char *ciphername
, int cs
)
185 static const char msg
[] =
186 "The first %scipher supported by the server is\n"
187 "%s, which is below the configured warning threshold.\n"
188 "Continue with connection? (y/n) ";
189 static const char abandoned
[] = "Connection abandoned.\n";
195 (cs
== 1) ?
"client-to-server " :
200 hin
= GetStdHandle(STD_INPUT_HANDLE
);
201 GetConsoleMode(hin
, &savemode
);
202 SetConsoleMode(hin
, (savemode
| ENABLE_ECHO_INPUT
|
203 ENABLE_PROCESSED_INPUT
| ENABLE_LINE_INPUT
));
204 ReadFile(hin
, line
, sizeof(line
) - 1, &i
, NULL
);
205 SetConsoleMode(hin
, savemode
);
207 if (line
[0] == 'y' || line
[0] == 'Y') {
210 fprintf(stderr
, abandoned
);
215 /* GUI Adaptation - Sept 2000 */
216 static void send_msg(HWND h
, UINT message
, WPARAM wParam
)
218 while (!PostMessage(h
, message
, wParam
, 0))
222 static void tell_char(FILE * stream
, char c
)
227 unsigned int msg_id
= WM_STD_OUT_CHAR
;
228 if (stream
== stderr
)
229 msg_id
= WM_STD_ERR_CHAR
;
230 send_msg((HWND
) atoi(gui_hwnd
), msg_id
, (WPARAM
) c
);
234 static void tell_str(FILE * stream
, char *str
)
238 for (i
= 0; i
< strlen(str
); ++i
)
239 tell_char(stream
, str
[i
]);
242 static void tell_user(FILE * stream
, char *fmt
, ...)
244 char str
[0x100]; /* Make the size big enough */
247 vsprintf(str
, fmt
, ap
);
250 tell_str(stream
, str
);
253 static void gui_update_stats(char *name
, unsigned long size
,
254 int percentage
, unsigned long elapsed
)
258 if (strcmp(name
, statname
) != 0) {
259 for (i
= 0; i
< strlen(name
); ++i
)
260 send_msg((HWND
) atoi(gui_hwnd
), WM_STATS_CHAR
,
262 send_msg((HWND
) atoi(gui_hwnd
), WM_STATS_CHAR
, (WPARAM
) '\n');
263 strcpy(statname
, name
);
265 if (statsize
!= size
) {
266 send_msg((HWND
) atoi(gui_hwnd
), WM_STATS_SIZE
, (WPARAM
) size
);
269 if (statelapsed
!= elapsed
) {
270 send_msg((HWND
) atoi(gui_hwnd
), WM_STATS_ELAPSED
,
272 statelapsed
= elapsed
;
274 if (statperct
!= percentage
) {
275 send_msg((HWND
) atoi(gui_hwnd
), WM_STATS_PERCENT
,
276 (WPARAM
) percentage
);
277 statperct
= percentage
;
282 * Print an error message and perform a fatal exit.
284 void fatalbox(char *fmt
, ...)
286 char str
[0x100]; /* Make the size big enough */
289 strcpy(str
, "Fatal: ");
290 vsprintf(str
+ strlen(str
), fmt
, ap
);
293 tell_str(stderr
, str
);
297 unsigned int msg_id
= WM_RET_ERR_CNT
;
299 msg_id
= WM_LS_RET_ERR_CNT
;
301 ((HWND
) atoi(gui_hwnd
), msg_id
, (WPARAM
) errs
,
302 0 /*lParam */ ))SleepEx(1000, TRUE
);
307 void connection_fatal(char *fmt
, ...)
309 char str
[0x100]; /* Make the size big enough */
312 strcpy(str
, "Fatal: ");
313 vsprintf(str
+ strlen(str
), fmt
, ap
);
316 tell_str(stderr
, str
);
320 unsigned int msg_id
= WM_RET_ERR_CNT
;
322 msg_id
= WM_LS_RET_ERR_CNT
;
324 ((HWND
) atoi(gui_hwnd
), msg_id
, (WPARAM
) errs
,
325 0 /*lParam */ ))SleepEx(1000, TRUE
);
332 * Be told what socket we're supposed to be using.
334 static SOCKET scp_ssh_socket
;
335 char *do_select(SOCKET skt
, int startup
)
338 scp_ssh_socket
= skt
;
340 scp_ssh_socket
= INVALID_SOCKET
;
343 extern int select_result(WPARAM
, LPARAM
);
346 * Receive a block of data from the SSH link. Block until all data
349 * To do this, we repeatedly call the SSH protocol module, with our
350 * own trap in from_backend() to catch the data that comes back. We
351 * do this until we have enough data.
354 static unsigned char *outptr
; /* where to put the data */
355 static unsigned outlen
; /* how much data required */
356 static unsigned char *pending
= NULL
; /* any spare data */
357 static unsigned pendlen
= 0, pendsize
= 0; /* length and phys. size of buffer */
358 int from_backend(int is_stderr
, char *data
, int datalen
)
360 unsigned char *p
= (unsigned char *) data
;
361 unsigned len
= (unsigned) datalen
;
364 * stderr data is just spouted to local stderr and otherwise
368 fwrite(data
, 1, len
, stderr
);
375 * If this is before the real session begins, just return.
381 unsigned used
= outlen
;
384 memcpy(outptr
, p
, used
);
392 if (pendsize
< pendlen
+ len
) {
393 pendsize
= pendlen
+ len
+ 4096;
394 pending
= (pending ?
srealloc(pending
, pendsize
) :
397 fatalbox("Out of memory");
399 memcpy(pending
+ pendlen
, p
, len
);
405 static int scp_process_network_event(void)
410 FD_SET(scp_ssh_socket
, &readfds
);
411 if (select(1, &readfds
, NULL
, NULL
, NULL
) < 0)
413 select_result((WPARAM
) scp_ssh_socket
, (LPARAM
) FD_READ
);
416 static int ssh_scp_recv(unsigned char *buf
, int len
)
422 * See if the pending-input block contains some of what we
426 unsigned pendused
= pendlen
;
427 if (pendused
> outlen
)
429 memcpy(outptr
, pending
, pendused
);
430 memmove(pending
, pending
+ pendused
, pendlen
- pendused
);
444 if (!scp_process_network_event())
452 * Loop through the ssh connection and authentication process.
454 static void ssh_scp_init(void)
456 if (scp_ssh_socket
== INVALID_SOCKET
)
458 while (!back
->sendok()) {
461 FD_SET(scp_ssh_socket
, &readfds
);
462 if (select(1, &readfds
, NULL
, NULL
, NULL
) < 0)
464 select_result((WPARAM
) scp_ssh_socket
, (LPARAM
) FD_READ
);
469 * Print an error message and exit after closing the SSH link.
471 static void bump(char *fmt
, ...)
473 char str
[0x100]; /* Make the size big enough */
476 strcpy(str
, "Fatal: ");
477 vsprintf(str
+ strlen(str
), fmt
, ap
);
480 tell_str(stderr
, str
);
483 if (back
!= NULL
&& back
->socket() != NULL
) {
485 back
->special(TS_EOF
);
486 ssh_scp_recv(&ch
, 1);
490 unsigned int msg_id
= WM_RET_ERR_CNT
;
492 msg_id
= WM_LS_RET_ERR_CNT
;
494 ((HWND
) atoi(gui_hwnd
), msg_id
, (WPARAM
) errs
,
495 0 /*lParam */ ))SleepEx(1000, TRUE
);
501 static int get_line(const char *prompt
, char *str
, int maxlen
, int is_pw
)
504 DWORD savemode
, newmode
, i
;
506 if (is_pw
&& password
) {
507 static int tried_once
= 0;
512 strncpy(str
, password
, maxlen
);
513 str
[maxlen
- 1] = '\0';
519 /* GUI Adaptation - Sept 2000 */
524 hin
= GetStdHandle(STD_INPUT_HANDLE
);
525 hout
= GetStdHandle(STD_OUTPUT_HANDLE
);
526 if (hin
== INVALID_HANDLE_VALUE
|| hout
== INVALID_HANDLE_VALUE
)
527 bump("Cannot get standard input/output handles");
529 GetConsoleMode(hin
, &savemode
);
530 newmode
= savemode
| ENABLE_PROCESSED_INPUT
| ENABLE_LINE_INPUT
;
532 newmode
&= ~ENABLE_ECHO_INPUT
;
534 newmode
|= ENABLE_ECHO_INPUT
;
535 SetConsoleMode(hin
, newmode
);
537 WriteFile(hout
, prompt
, strlen(prompt
), &i
, NULL
);
538 ReadFile(hin
, str
, maxlen
- 1, &i
, NULL
);
540 SetConsoleMode(hin
, savemode
);
542 if ((int) i
> maxlen
)
549 WriteFile(hout
, "\r\n", 2, &i
, NULL
);
556 * Open an SSH connection to user@host and execute cmd.
558 static void do_cmd(char *host
, char *user
, char *cmd
)
560 char *err
, *realhost
;
563 if (host
== NULL
|| host
[0] == '\0')
564 bump("Empty host name");
566 /* Try to load settings for this host */
567 do_defaults(host
, &cfg
);
568 if (cfg
.host
[0] == '\0') {
569 /* No settings for this host; use defaults */
570 do_defaults(NULL
, &cfg
);
571 strncpy(cfg
.host
, host
, sizeof(cfg
.host
) - 1);
572 cfg
.host
[sizeof(cfg
.host
) - 1] = '\0';
577 if (user
!= NULL
&& user
[0] != '\0') {
578 strncpy(cfg
.username
, user
, sizeof(cfg
.username
) - 1);
579 cfg
.username
[sizeof(cfg
.username
) - 1] = '\0';
580 } else if (cfg
.username
[0] == '\0') {
582 if (GetUserName(user
, &namelen
) == FALSE
)
583 bump("Empty user name");
584 user
= smalloc(namelen
* sizeof(char));
585 GetUserName(user
, &namelen
);
587 tell_user(stderr
, "Guessing user name: %s", user
);
588 strncpy(cfg
.username
, user
, sizeof(cfg
.username
) - 1);
589 cfg
.username
[sizeof(cfg
.username
) - 1] = '\0';
593 if (cfg
.protocol
!= PROT_SSH
)
597 cfg
.port
= portnumber
;
599 strncpy(cfg
.remote_cmd
, cmd
, sizeof(cfg
.remote_cmd
));
600 cfg
.remote_cmd
[sizeof(cfg
.remote_cmd
) - 1] = '\0';
605 err
= back
->init(cfg
.host
, cfg
.port
, &realhost
);
607 bump("ssh_init: %s", err
);
609 if (verbose
&& realhost
!= NULL
)
610 tell_user(stderr
, "Connected to %s\n", realhost
);
615 * Update statistic information about current file.
617 static void print_stats(char *name
, unsigned long size
, unsigned long done
,
618 time_t start
, time_t now
)
626 /* GUI Adaptation - Sept 2000 */
628 gui_update_stats(name
, size
, (int) (100 * (done
* 1.0 / size
)),
629 (unsigned long) difftime(now
, start
));
632 ratebs
= (float) done
/ (now
- start
);
634 ratebs
= (float) done
;
639 eta
= (unsigned long) ((size
- done
) / ratebs
);
640 sprintf(etastr
, "%02ld:%02ld:%02ld",
641 eta
/ 3600, (eta
% 3600) / 60, eta
% 60);
643 pct
= (int) (100.0 * (float) done
/ size
);
645 len
= printf("\r%-25.25s | %10ld kB | %5.1f kB/s | ETA: %8s | %3d%%",
646 name
, done
/ 1024, ratebs
/ 1024.0, etastr
, pct
);
647 if (len
< prev_stats_len
)
648 printf("%*s", prev_stats_len
- len
, "");
649 prev_stats_len
= len
;
657 * Find a colon in str and return a pointer to the colon.
658 * This is used to separate hostname from filename.
660 static char *colon(char *str
)
662 /* We ignore a leading colon, since the hostname cannot be
663 empty. We also ignore a colon as second character because
664 of filenames like f:myfile.txt. */
665 if (str
[0] == '\0' || str
[0] == ':' || str
[1] == ':')
667 while (*str
!= '\0' && *str
!= ':' && *str
!= '/' && *str
!= '\\')
676 * Wait for a response from the other side.
677 * Return 0 if ok, -1 if error.
679 static int response(void)
681 char ch
, resp
, rbuf
[2048];
684 if (ssh_scp_recv(&resp
, 1) <= 0)
685 bump("Lost connection");
695 case 2: /* fatal error */
697 if (ssh_scp_recv(&ch
, 1) <= 0)
698 bump("Protocol error: Lost connection");
700 } while (p
< sizeof(rbuf
) && ch
!= '\n');
703 tell_user(stderr
, "%s\n", rbuf
);
711 /* ----------------------------------------------------------------------
712 * Helper routines that contain the actual SCP protocol elements,
713 * so they can be switched to use SFTP.
716 int scp_send_errmsg(char *str
)
718 back
->send("\001", 1); /* scp protocol error prefix */
719 back
->send(str
, strlen(str
));
720 return 0; /* can't fail */
723 int scp_send_filetimes(unsigned long mtime
, unsigned long atime
)
726 sprintf(buf
, "T%lu 0 %lu 0\n", mtime
, atime
);
727 back
->send(buf
, strlen(buf
));
731 int scp_send_filename(char *name
, unsigned long size
, int modes
)
734 sprintf(buf
, "C%04o %lu ", modes
, size
);
735 back
->send(buf
, strlen(buf
));
736 back
->send(name
, strlen(name
));
741 int scp_send_filedata(char *data
, int len
)
743 int bufsize
= back
->send(data
, len
);
746 * If the network transfer is backing up - that is, the remote
747 * site is not accepting data as fast as we can produce it -
748 * then we must loop on network events until we have space in
751 while (bufsize
> MAX_SCP_BUFSIZE
) {
752 if (!scp_process_network_event())
754 bufsize
= back
->sendbuffer();
760 int scp_send_finish(void)
766 int scp_send_dirname(char *name
, int modes
)
769 sprintf(buf
, "D%04o 0 ", modes
);
770 back
->send(buf
, strlen(buf
));
771 back
->send(name
, strlen(name
));
776 int scp_send_enddir(void)
778 back
->send("E\n", 2);
782 int scp_sink_init(void)
788 #define SCP_SINK_FILE 1
789 #define SCP_SINK_DIR 2
790 #define SCP_SINK_ENDDIR 3
791 struct scp_sink_action
{
792 int action
; /* FILE, DIR, ENDDIR */
793 char *buf
; /* will need freeing after use */
794 char *name
; /* filename or dirname (not ENDDIR) */
795 int mode
; /* access mode (not ENDDIR) */
796 unsigned long size
; /* file size (not ENDDIR) */
797 int settime
; /* 1 if atime and mtime are filled */
798 unsigned long atime
, mtime
; /* access times for the file */
801 int scp_get_sink_action(struct scp_sink_action
*act
)
813 if (ssh_scp_recv(&ch
, 1) <= 0)
816 bump("Protocol error: Unexpected newline");
820 if (ssh_scp_recv(&ch
, 1) <= 0)
821 bump("Lost connection");
824 act
->buf
= srealloc(act
->buf
, bufsize
);
827 } while (ch
!= '\n');
828 act
->buf
[i
- 1] = '\0';
830 case '\01': /* error */
831 tell_user(stderr
, "%s\n", act
->buf
);
833 continue; /* go round again */
834 case '\02': /* fatal error */
835 bump("%s", act
->buf
);
838 act
->action
= SCP_SINK_ENDDIR
;
841 if (sscanf(act
->buf
, "%ld %*d %ld %*d",
842 &act
->mtime
, &act
->atime
) == 2) {
845 continue; /* go round again */
847 bump("Protocol error: Illegal time format");
850 act
->action
= (action
== 'C' ? SCP_SINK_FILE
: SCP_SINK_DIR
);
853 bump("Protocol error: Expected control record");
856 * We will go round this loop only once, unless we hit
863 * If we get here, we must have seen SCP_SINK_FILE or
866 if (sscanf(act
->buf
, "%o %lu %n", &act
->mode
, &act
->size
, &i
) != 2)
867 bump("Protocol error: Illegal file descriptor format");
868 act
->name
= act
->buf
+ i
;
872 int scp_accept_filexfer(void)
875 return 0; /* can't fail */
878 int scp_recv_filedata(char *data
, int len
)
880 return ssh_scp_recv(data
, len
);
883 int scp_finish_filerecv(void)
889 /* ----------------------------------------------------------------------
890 * Send an error message to the other side and to the screen.
891 * Increment error counter.
893 static void run_err(const char *fmt
, ...)
899 strcpy(str
, "scp: ");
900 vsprintf(str
+ strlen(str
), fmt
, ap
);
902 scp_send_errmsg(str
);
903 tell_user(stderr
, "%s", str
);
908 * Execute the source part of the SCP protocol.
910 static void source(char *src
)
918 unsigned long stat_bytes
;
919 time_t stat_starttime
, stat_lasttime
;
921 attr
= GetFileAttributes(src
);
922 if (attr
== (DWORD
) - 1) {
923 run_err("%s: No such file or directory", src
);
927 if ((attr
& FILE_ATTRIBUTE_DIRECTORY
) != 0) {
930 * Avoid . and .. directories.
933 p
= strrchr(src
, '/');
935 p
= strrchr(src
, '\\');
940 if (!strcmp(p
, ".") || !strcmp(p
, ".."))
941 /* skip . and .. */ ;
945 run_err("%s: not a regular file", src
);
950 if ((last
= strrchr(src
, '/')) == NULL
)
954 if (strrchr(last
, '\\') != NULL
)
955 last
= strrchr(last
, '\\') + 1;
956 if (last
== src
&& strchr(src
, ':') != NULL
)
957 last
= strchr(src
, ':') + 1;
959 f
= CreateFile(src
, GENERIC_READ
, FILE_SHARE_READ
, NULL
,
960 OPEN_EXISTING
, 0, 0);
961 if (f
== INVALID_HANDLE_VALUE
) {
962 run_err("%s: Cannot open file", src
);
967 FILETIME actime
, wrtime
;
968 unsigned long mtime
, atime
;
969 GetFileTime(f
, NULL
, &actime
, &wrtime
);
970 TIME_WIN_TO_POSIX(actime
, atime
);
971 TIME_WIN_TO_POSIX(wrtime
, mtime
);
972 if (scp_send_filetimes(mtime
, atime
))
976 size
= GetFileSize(f
, NULL
);
978 tell_user(stderr
, "Sending file %s, size=%lu", last
, size
);
979 if (scp_send_filename(last
, size
, 0644))
983 stat_starttime
= time(NULL
);
986 for (i
= 0; i
< size
; i
+= 4096) {
992 if (!ReadFile(f
, transbuf
, k
, &j
, NULL
) || j
!= k
) {
995 bump("%s: Read error", src
);
997 if (scp_send_filedata(transbuf
, k
))
998 bump("%s: Network error occurred", src
);
1002 if (time(NULL
) != stat_lasttime
|| i
+ k
== size
) {
1003 stat_lasttime
= time(NULL
);
1004 print_stats(last
, size
, stat_bytes
,
1005 stat_starttime
, stat_lasttime
);
1012 (void) scp_send_finish();
1016 * Recursively send the contents of a directory.
1018 static void rsource(char *src
)
1023 WIN32_FIND_DATA fdat
;
1026 if ((last
= strrchr(src
, '/')) == NULL
)
1030 if (strrchr(last
, '\\') != NULL
)
1031 last
= strrchr(last
, '\\') + 1;
1032 if (last
== src
&& strchr(src
, ':') != NULL
)
1033 last
= strchr(src
, ':') + 1;
1035 /* maybe send filetime */
1038 tell_user(stderr
, "Entering directory: %s", last
);
1039 if (scp_send_dirname(last
, 0755))
1042 sprintf(buf
, "%s/*", src
);
1043 dir
= FindFirstFile(buf
, &fdat
);
1044 ok
= (dir
!= INVALID_HANDLE_VALUE
);
1046 if (strcmp(fdat
.cFileName
, ".") == 0 ||
1047 strcmp(fdat
.cFileName
, "..") == 0) {
1048 } else if (strlen(src
) + 1 + strlen(fdat
.cFileName
) >= sizeof(buf
)) {
1049 run_err("%s/%s: Name too long", src
, fdat
.cFileName
);
1051 sprintf(buf
, "%s/%s", src
, fdat
.cFileName
);
1054 ok
= FindNextFile(dir
, &fdat
);
1058 (void) scp_send_enddir();
1062 * Execute the sink part of the SCP protocol.
1064 static void sink(char *targ
, char *src
)
1074 unsigned long received
;
1076 unsigned long stat_bytes
;
1077 time_t stat_starttime
, stat_lasttime
;
1080 attr
= GetFileAttributes(targ
);
1081 if (attr
!= (DWORD
) - 1 && (attr
& FILE_ATTRIBUTE_DIRECTORY
) != 0)
1084 if (targetshouldbedirectory
&& !targisdir
)
1085 bump("%s: Not a directory", targ
);
1089 struct scp_sink_action act
;
1090 if (scp_get_sink_action(&act
))
1093 if (act
.action
== SCP_SINK_ENDDIR
)
1095 /* Security fix: ensure the file ends up where we asked for it. */
1100 if (targ
[0] != '\0')
1102 p
= act
.name
+ strlen(act
.name
);
1103 while (p
> act
.name
&& p
[-1] != '/' && p
[-1] != '\\')
1108 strcpy(namebuf
, targ
);
1110 attr
= GetFileAttributes(namebuf
);
1111 exists
= (attr
!= (DWORD
) - 1);
1113 if (act
.action
== SCP_SINK_DIR
) {
1114 if (exists
&& (attr
& FILE_ATTRIBUTE_DIRECTORY
) == 0) {
1115 run_err("%s: Not a directory", namebuf
);
1119 if (!CreateDirectory(namebuf
, NULL
)) {
1120 run_err("%s: Cannot create directory", namebuf
);
1124 sink(namebuf
, NULL
);
1125 /* can we set the timestamp for directories ? */
1129 f
= CreateFile(namebuf
, GENERIC_WRITE
, 0, NULL
,
1130 CREATE_ALWAYS
, FILE_ATTRIBUTE_NORMAL
, 0);
1131 if (f
== INVALID_HANDLE_VALUE
) {
1132 run_err("%s: Cannot create file", namebuf
);
1136 if (scp_accept_filexfer())
1140 stat_starttime
= time(NULL
);
1142 if ((stat_name
= strrchr(namebuf
, '/')) == NULL
)
1143 stat_name
= namebuf
;
1146 if (strrchr(stat_name
, '\\') != NULL
)
1147 stat_name
= strrchr(stat_name
, '\\') + 1;
1150 while (received
< act
.size
) {
1151 char transbuf
[4096];
1152 DWORD blksize
, read
, written
;
1154 if (blksize
> act
.size
- received
)
1155 blksize
= act
.size
- received
;
1156 read
= scp_recv_filedata(transbuf
, blksize
);
1158 bump("Lost connection");
1161 if (!WriteFile(f
, transbuf
, read
, &written
, NULL
) ||
1164 /* FIXME: in sftp we can actually abort the transfer */
1166 printf("\r%-25.25s | %50s\n",
1168 "Write error.. waiting for end of file");
1173 if (time(NULL
) > stat_lasttime
||
1174 received
+ read
== act
.size
) {
1175 stat_lasttime
= time(NULL
);
1176 print_stats(stat_name
, act
.size
, stat_bytes
,
1177 stat_starttime
, stat_lasttime
);
1183 FILETIME actime
, wrtime
;
1184 TIME_POSIX_TO_WIN(act
.atime
, actime
);
1185 TIME_POSIX_TO_WIN(act
.mtime
, wrtime
);
1186 SetFileTime(f
, NULL
, &actime
, &wrtime
);
1191 run_err("%s: Write error", namebuf
);
1194 (void) scp_finish_filerecv();
1199 * We will copy local files to a remote server.
1201 static void toremote(int argc
, char *argv
[])
1203 char *src
, *targ
, *host
, *user
;
1207 targ
= argv
[argc
- 1];
1209 /* Separate host from filename */
1213 bump("targ == NULL in toremote()");
1217 /* Substitute "." for emtpy target */
1219 /* Separate host and username */
1221 host
= strrchr(host
, '@');
1232 /* Find out if the source filespec covers multiple files
1233 if so, we should set the targetshouldbedirectory flag */
1235 WIN32_FIND_DATA fdat
;
1236 if (colon(argv
[0]) != NULL
)
1237 bump("%s: Remote to remote not supported", argv
[0]);
1238 fh
= FindFirstFile(argv
[0], &fdat
);
1239 if (fh
== INVALID_HANDLE_VALUE
)
1240 bump("%s: No such file or directory\n", argv
[0]);
1241 if (FindNextFile(fh
, &fdat
))
1242 targetshouldbedirectory
= 1;
1246 cmd
= smalloc(strlen(targ
) + 100);
1247 sprintf(cmd
, "scp%s%s%s%s -t %s",
1248 verbose ?
" -v" : "",
1249 recursive ?
" -r" : "",
1250 preserve ?
" -p" : "",
1251 targetshouldbedirectory ?
" -d" : "", targ
);
1252 do_cmd(host
, user
, cmd
);
1257 for (i
= 0; i
< argc
- 1; i
++) {
1259 WIN32_FIND_DATA fdat
;
1261 if (colon(src
) != NULL
) {
1262 tell_user(stderr
, "%s: Remote to remote not supported\n", src
);
1266 dir
= FindFirstFile(src
, &fdat
);
1267 if (dir
== INVALID_HANDLE_VALUE
) {
1268 run_err("%s: No such file or directory", src
);
1275 * Ensure that . and .. are never matched by wildcards,
1276 * but only by deliberate action.
1278 if (!strcmp(fdat
.cFileName
, ".") ||
1279 !strcmp(fdat
.cFileName
, "..")) {
1281 * Find*File has returned a special dir. We require
1282 * that _either_ `src' ends in a backslash followed
1283 * by that string, _or_ `src' is precisely that
1286 int len
= strlen(src
), dlen
= strlen(fdat
.cFileName
);
1287 if (len
== dlen
&& !strcmp(src
, fdat
.cFileName
)) {
1289 } else if (len
> dlen
+ 1 && src
[len
- dlen
- 1] == '\\' &&
1290 !strcmp(src
+ len
- dlen
, fdat
.cFileName
)) {
1293 continue; /* ignore this one */
1295 if (strlen(src
) + strlen(fdat
.cFileName
) >= sizeof(namebuf
)) {
1296 tell_user(stderr
, "%s: Name too long", src
);
1299 strcpy(namebuf
, src
);
1300 if ((last
= strrchr(namebuf
, '/')) == NULL
)
1304 if (strrchr(last
, '\\') != NULL
)
1305 last
= strrchr(last
, '\\') + 1;
1306 if (last
== namebuf
&& strrchr(namebuf
, ':') != NULL
)
1307 last
= strchr(namebuf
, ':') + 1;
1308 strcpy(last
, fdat
.cFileName
);
1310 } while (FindNextFile(dir
, &fdat
));
1316 * We will copy files from a remote server to the local machine.
1318 static void tolocal(int argc
, char *argv
[])
1320 char *src
, *targ
, *host
, *user
;
1324 bump("More than one remote source not supported");
1329 /* Separate host from filename */
1333 bump("Local to local copy not supported");
1337 /* Substitute "." for empty filename */
1339 /* Separate username and hostname */
1341 host
= strrchr(host
, '@');
1351 cmd
= smalloc(strlen(src
) + 100);
1352 sprintf(cmd
, "scp%s%s%s%s -f %s",
1353 verbose ?
" -v" : "",
1354 recursive ?
" -r" : "",
1355 preserve ?
" -p" : "",
1356 targetshouldbedirectory ?
" -d" : "", src
);
1357 do_cmd(host
, user
, cmd
);
1364 * We will issue a list command to get a remote directory.
1366 static void get_dir_list(int argc
, char *argv
[])
1368 char *src
, *host
, *user
;
1374 /* Separate host from filename */
1378 bump("Local to local copy not supported");
1382 /* Substitute "." for empty filename */
1384 /* Separate username and hostname */
1386 host
= strrchr(host
, '@');
1396 cmd
= smalloc(4 * strlen(src
) + 100);
1397 strcpy(cmd
, "ls -la '");
1398 p
= cmd
+ strlen(cmd
);
1399 for (q
= src
; *q
; q
++) {
1412 do_cmd(host
, user
, cmd
);
1415 while (ssh_scp_recv(&c
, 1) > 0)
1416 tell_char(stdout
, c
);
1420 * Initialize the Win$ock driver.
1422 static void init_winsock(void)
1427 winsock_ver
= MAKEWORD(1, 1);
1428 if (WSAStartup(winsock_ver
, &wsadata
))
1429 bump("Unable to initialise WinSock");
1430 if (LOBYTE(wsadata
.wVersion
) != 1 || HIBYTE(wsadata
.wVersion
) != 1)
1431 bump("WinSock version is incompatible with 1.1");
1435 * Short description of parameters.
1437 static void usage(void)
1439 printf("PuTTY Secure Copy client\n");
1440 printf("%s\n", ver
);
1441 printf("Usage: pscp [options] [user@]host:source target\n");
1443 (" pscp [options] source [source...] [user@]host:target\n");
1444 printf(" pscp [options] -ls user@host:filespec\n");
1445 printf("Options:\n");
1446 printf(" -p preserve file attributes\n");
1447 printf(" -q quiet, don't show statistics\n");
1448 printf(" -r copy directories recursively\n");
1449 printf(" -v show verbose messages\n");
1450 printf(" -P port connect to specified port\n");
1451 printf(" -pw passw login with specified password\n");
1454 * -gui is an internal option, used by GUI front ends to get
1455 * pscp to pass progress reports back to them. It's not an
1456 * ordinary user-accessible option, so it shouldn't be part of
1457 * the command-line help. The only people who need to know
1458 * about it are programmers, and they can read the source.
1461 (" -gui hWnd GUI mode with the windows handle for receiving messages\n");
1467 * Main program (no, really?)
1469 int main(int argc
, char *argv
[])
1473 default_protocol
= PROT_TELNET
;
1475 flags
= FLAG_STDERR
;
1476 ssh_get_line
= &get_line
;
1480 for (i
= 1; i
< argc
; i
++) {
1481 if (argv
[i
][0] != '-')
1483 if (strcmp(argv
[i
], "-v") == 0)
1484 verbose
= 1, flags
|= FLAG_VERBOSE
;
1485 else if (strcmp(argv
[i
], "-r") == 0)
1487 else if (strcmp(argv
[i
], "-p") == 0)
1489 else if (strcmp(argv
[i
], "-q") == 0)
1491 else if (strcmp(argv
[i
], "-h") == 0 || strcmp(argv
[i
], "-?") == 0)
1493 else if (strcmp(argv
[i
], "-P") == 0 && i
+ 1 < argc
)
1494 portnumber
= atoi(argv
[++i
]);
1495 else if (strcmp(argv
[i
], "-pw") == 0 && i
+ 1 < argc
)
1496 password
= argv
[++i
];
1497 else if (strcmp(argv
[i
], "-gui") == 0 && i
+ 1 < argc
) {
1498 gui_hwnd
= argv
[++i
];
1500 } else if (strcmp(argv
[i
], "-ls") == 0)
1502 else if (strcmp(argv
[i
], "--") == 0) {
1515 get_dir_list(argc
, argv
);
1522 targetshouldbedirectory
= 1;
1524 if (colon(argv
[argc
- 1]) != NULL
)
1525 toremote(argc
, argv
);
1527 tolocal(argc
, argv
);
1530 if (back
!= NULL
&& back
->socket() != NULL
) {
1532 back
->special(TS_EOF
);
1533 ssh_scp_recv(&ch
, 1);
1538 /* GUI Adaptation - August 2000 */
1540 unsigned int msg_id
= WM_RET_ERR_CNT
;
1542 msg_id
= WM_LS_RET_ERR_CNT
;
1544 ((HWND
) atoi(gui_hwnd
), msg_id
, (WPARAM
) errs
,
1545 0 /*lParam */ ))SleepEx(1000, TRUE
);
1547 return (errs
== 0 ?
0 : 1);