/*
* Recursively send the contents of a directory.
*/
static void rsource(char *src)
{
char buf[2048];
char *last;
HANDLE dir;
WIN32_FIND_DATA fdat;
int ok;
if ((last = strrchr(src, '/')) == NULL)
last = src;
else
last++;
if (strrchr(last, '\\') != NULL)
last = strrchr(last, '\\') + 1;
if (last == src && strchr(src, ':') != NULL)
last = strchr(src, ':') + 1;
/* maybe send filetime */
sprintf(buf, "D0755 0 %s\n", last);
if (verbose)
fprintf(stderr, "Entering directory: %s", buf);
ssh_send(buf, strlen(buf));
if (response())
return;
sprintf(buf, "%s/*", src);
dir = FindFirstFile(buf, &fdat);
ok = (dir != INVALID_HANDLE_VALUE);
while (ok) {
if (strcmp(fdat.cFileName, ".") == 0 ||
strcmp(fdat.cFileName, "..") == 0) {
} else if (strlen(src) + 1 + strlen(fdat.cFileName) >=
sizeof(buf)) {
run_err("%s/%s: Name too long", src, fdat.cFileName);
} else {
sprintf(buf, "%s/%s", src, fdat.cFileName);
source(buf);
}
ok = FindNextFile(dir, &fdat);
}
FindClose(dir);
sprintf(buf, "E\n");
ssh_send(buf, strlen(buf));
(void) response();
}
/*
* Send an error message to the other side and to the screen.
* Increment error counter.
*/
static void run_err(const char *fmt, ...)
{
char str[2048];
va_list ap;
va_start(ap, fmt);
errs++;
strcpy(str, "\01scp: ");
vsprintf(str+strlen(str), fmt, ap);
strcat(str, "\n");
ssh_send(str, strlen(str));
vfprintf(stderr, fmt, ap);
fprintf(stderr, "\n");
va_end(ap);
}
/*
* Execute the sink part of the SCP protocol.
*/
static void sink(char *targ)
{
char buf[2048];
char namebuf[2048];
char ch;
int targisdir = 0;
int settime;
int exists;
DWORD attr;
HANDLE f;
unsigned long mtime, atime;
unsigned int mode;
unsigned long size, i;
int wrerror = 0;
unsigned long stat_bytes;
time_t stat_starttime, stat_lasttime;
char *stat_name;
attr = GetFileAttributes(targ);
if (attr != (DWORD)-1 && (attr & FILE_ATTRIBUTE_DIRECTORY) != 0)
targisdir = 1;
if (targetshouldbedirectory && !targisdir)
bump("%s: Not a directory", targ);
ssh_send("", 1);
while (1) {
settime = 0;
gottime:
if (ssh_recv(&ch, 1) <= 0)
return;
if (ch == '\n')
bump("Protocol error: Unexpected newline");
i = 0;
buf[i++] = ch;
do {
if (ssh_recv(&ch, 1) <= 0)
bump("Lost connection");
buf[i++] = ch;
} while (i < sizeof(buf) && ch != '\n');
buf[i-1] = '\0';
switch (buf[0]) {
case '\01': /* error */
fprintf(stderr, "%s\n", buf+1);
errs++;
continue;
case '\02': /* fatal error */
bump("%s", buf+1);
case 'E':
ssh_send("", 1);
return;
case 'T':
if (sscanf(buf, "T%ld %*d %ld %*d",
&mtime, &atime) == 2) {
settime = 1;
ssh_send("", 1);
goto gottime;
}
bump("Protocol error: Illegal time format");
case 'C':
case 'D':
break;
default:
bump("Protocol error: Expected control record");
}
if (sscanf(buf+1, "%u %lu %[^\n]", &mode, &size, namebuf) != 3)
bump("Protocol error: Illegal file descriptor format");
if (targisdir) {
char t[2048];
strcpy(t, targ);
if (targ[0] != '\0')
strcat(t, "/");
strcat(t, namebuf);
strcpy(namebuf, t);
} else {
strcpy(namebuf, targ);
}
attr = GetFileAttributes(namebuf);
exists = (attr != (DWORD)-1);
if (buf[0] == 'D') {
if (exists && (attr & FILE_ATTRIBUTE_DIRECTORY) == 0) {
run_err("%s: Not a directory", namebuf);
continue;
}
if (!exists) {
if (! CreateDirectory(namebuf, NULL)) {
run_err("%s: Cannot create directory",
namebuf);
continue;
}
}
sink(namebuf);
/* can we set the timestamp for directories ? */
continue;
}
f = CreateFile(namebuf, GENERIC_WRITE, 0, NULL,
CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
if (f == INVALID_HANDLE_VALUE) {
run_err("%s: Cannot create file", namebuf);
continue;
}
ssh_send("", 1);
if (statistics) {
stat_bytes = 0;
stat_starttime = time(NULL);
stat_lasttime = 0;
if ((stat_name = strrchr(namebuf, '/')) == NULL)
stat_name = namebuf;
else
stat_name++;
if (strrchr(stat_name, '\\') != NULL)
stat_name = strrchr(stat_name, '\\') + 1;
}
for (i = 0; i < size; i += 4096) {
char transbuf[4096];
DWORD j, k = 4096;
if (i + k > size) k = size - i;
if (ssh_recv(transbuf, k) == 0)
bump("Lost connection");
if (wrerror) continue;
if (! WriteFile(f, transbuf, k, &j, NULL) || j != k) {
wrerror = 1;
if (statistics)
printf("\r%-25.25s | %50s\n",
stat_name,
"Write error.. waiting for end of file");
continue;
}
if (statistics) {
stat_bytes += k;
if (time(NULL) > stat_lasttime ||
i + k == size) {
stat_lasttime = time(NULL);
print_stats(stat_name, size, stat_bytes,
stat_starttime, stat_lasttime);
}
}
}
(void) response();
if (settime) {
FILETIME actime, wrtime;
TIME_POSIX_TO_WIN(atime, actime);
TIME_POSIX_TO_WIN(mtime, wrtime);
SetFileTime(f, NULL, &actime, &wrtime);
}
CloseHandle(f);
if (wrerror) {
run_err("%s: Write error", namebuf);
continue;
}
ssh_send("", 1);
}
}
/*
* Execute the source part of the SCP protocol.
*/
static void source(char *src)
{
char buf[2048];
unsigned long size;
char *last;
HANDLE f;
DWORD attr;
unsigned long i;
unsigned long stat_bytes;
time_t stat_starttime, stat_lasttime;
attr = GetFileAttributes(src);
if (attr == (DWORD)-1) {
run_err("%s: No such file or directory", src);
return;
}
if ((attr & FILE_ATTRIBUTE_DIRECTORY) != 0) {
if (recursive) {
/*
* Avoid . and .. directories.
*/
char *p;
p = strrchr(src, '/');
if (!p)
p = strrchr(src, '\\');
if (!p)
p = src;
else
p++;
if (!strcmp(p, ".") || !strcmp(p, ".."))
/* skip . and .. */;
else
rsource(src);
} else {
run_err("%s: not a regular file", src);
}
return;
}
if ((last = strrchr(src, '/')) == NULL)
last = src;
else
last++;
if (strrchr(last, '\\') != NULL)
last = strrchr(last, '\\') + 1;
if (last == src && strchr(src, ':') != NULL)
last = strchr(src, ':') + 1;
f = CreateFile(src, GENERIC_READ, FILE_SHARE_READ, NULL,
OPEN_EXISTING, 0, 0);
if (f == INVALID_HANDLE_VALUE) {
run_err("%s: Cannot open file", src);
return;
}
if (preserve) {
FILETIME actime, wrtime;
unsigned long mtime, atime;
GetFileTime(f, NULL, &actime, &wrtime);
TIME_WIN_TO_POSIX(actime, atime);
TIME_WIN_TO_POSIX(wrtime, mtime);
sprintf(buf, "T%lu 0 %lu 0\n", mtime, atime);
ssh_send(buf, strlen(buf));
if (response())
return;
}
size = GetFileSize(f, NULL);
sprintf(buf, "C0644 %lu %s\n", size, last);
if (verbose)
fprintf(stderr, "Sending file modes: %s", buf);
ssh_send(buf, strlen(buf));
if (response())
return;
if (statistics) {
stat_bytes = 0;
stat_starttime = time(NULL);
stat_lasttime = 0;
}
for (i = 0; i < size; i += 4096) {
char transbuf[4096];
DWORD j, k = 4096;
if (i + k > size) k = size - i;
if (! ReadFile(f, transbuf, k, &j, NULL) || j != k) {
if (statistics) printf("\n");
bump("%s: Read error", src);
}
ssh_send(transbuf, k);
if (statistics) {
stat_bytes += k;
if (time(NULL) != stat_lasttime ||
i + k == size) {
stat_lasttime = time(NULL);
print_stats(last, size, stat_bytes,
stat_starttime, stat_lasttime);
}
}
}
CloseHandle(f);
ssh_send("", 1);
(void) response();
}
int ftp_putfile(const char *infile, const char *outfile, putmode_t how,
transfer_mode_t mode, ftp_transfer_func hookf)
{
FILE *fp;
int r;
struct stat statbuf;
if(stat(infile, &statbuf) != 0) {
perror(infile);
return -1;
}
if(S_ISDIR(statbuf.st_mode)) {
ftp_err(_("%s: is a directory\n"), infile);
return -1;
}
fp = fopen(infile, "r");
if(fp == 0) {
perror(infile);
return -1;
}
ftp->ti.total_size = statbuf.st_size;
free(ftp->ti.remote_name);
free(ftp->ti.local_name);
ftp->ti.remote_name = xstrdup(infile); /* actually local file, or _target_ */
ftp->ti.local_name = xstrdup(outfile); /* actually remote file, or _source_ */
if(how == putResume) {
rfile *f;
f = ftp_get_file(outfile);
if(f && f->size != (unsigned long long)-1)
ftp->restart_offset = f->size;
else {
ftp->restart_offset = ftp_filesize(outfile);
if(ftp->restart_offset == (unsigned long long)-1) {
ftp_err(_("unable to get remote filesize of '%s',"
" unable to resume\n"),
outfile);
ftp->restart_offset = 0L;
}
}
} else
ftp->restart_offset = 0L;
if(ftp->restart_offset > 0L) {
if(fseek(fp, ftp->restart_offset, SEEK_SET) != 0) {
ftp_err(_("%s: %s, transfer cancelled\n"),
outfile, strerror(errno));
fclose(fp);
return -1;
}
}
foo_hookf = hookf;
#ifdef HAVE_LIBSSH
if(ftp->session)
r = ssh_send(outfile, fp, how, mode, hookf);
else
#endif
r = ftp_send(outfile, fp, how, mode, hookf);
fclose(fp);
return r;
}
