static int ftp_do_receive(FILE *fp,
transfer_mode_t mode, ftp_transfer_func hookf)
{
int r;
if(mode == tmBinary)
r = FILE_recv_binary(ftp->data, fp);
else
r = FILE_recv_ascii(ftp->data, fp);
sock_destroy(ftp->data);
ftp->data = 0;
if(r == 0) {
transfer_finished();
ftp_read_reply();
ftp->ti.ioerror = (ftp->code != ctComplete);
if(ftp->code != ctComplete) {
ftp_trace("transfer failed\n");
return -1;
}
} else
transfer_finished();
return (r == 0 && !ftp->ti.ioerror && !ftp->ti.interrupted) ? 0 : -1;
}
int ssh_send(const char *path, FILE *fp, putmode_t how,
transfer_mode_t mode, ftp_transfer_func hookf)
{
uint64_t offset = ftp->restart_offset;
reset_transfer_info();
ftp->ti.size = ftp->ti.restart_size = offset;
ftp->restart_offset = 0L;
ftp->ti.transfer_is_put = true;
if(how == putUnique) {
ftp_err(_("Unique put with SSH not implemented yet\n"));
return -1;
}
char* p = ftp_path_absolute(path);
stripslash(p);
ftp_cache_flush_mark_for(p);
if(how == putAppend) {
ftp_set_tmp_verbosity(vbNone);
offset = ftp_filesize(p);
}
int r = do_write(p, fp, hookf, offset);
free(p);
transfer_finished();
return r;
}
int ssh_send(const char *path, FILE *fp, putmode_t how,
transfer_mode_t mode, ftp_transfer_func hookf)
{
int r;
long offset = ftp->restart_offset;
char *p;
reset_transfer_info();
ftp->ti.size = ftp->ti.restart_size = offset;
ftp->restart_offset = 0L;
ftp->ti.transfer_is_put = true;
if(how == putUnique) {
ftp_err("Unique put with SSH not implemented yet\n");
return -1;
#if 0
/* try to figure out remote filename */
char *e = strstr(ftp->reply, " for ");
if(e) {
int l;
e += 5;
l = strlen(e);
if(l) {
free(ftp->ti.local_name);
if(*e == '\'')
ftp->ti.local_name = xstrndup(e+1, l-3);
else
ftp->ti.local_name = xstrndup(e, l-1);
ftp_trace("parsed unique filename as '%s'\n",
ftp->ti.local_name);
}
}
#endif
}
p = ftp_path_absolute(path);
stripslash(p);
ftp_cache_flush_mark_for(p);
if(how == putAppend) {
ftp_set_tmp_verbosity(vbNone);
offset = ftp_filesize(p);
}
r = ssh_send_binary(p, fp, hookf, offset);
free(p);
transfer_finished();
return 0;
}
int ssh_do_receive(const char *infile, FILE *fp, getmode_t mode,
ftp_transfer_func hookf)
{
uint64_t offset = ftp->restart_offset;
ftp->ti.size = ftp->ti.restart_size = offset;
ftp->restart_offset = 0L;
rfile* f = ftp_cache_get_file(infile);
if (f)
ftp->ti.total_size = f->size;
else
ftp->ti.total_size = ssh_filesize(infile);
int r = do_read(infile, fp, mode, hookf, offset);
transfer_finished();
return (r == 0 && !ftp->ti.ioerror && !ftp->ti.interrupted) ? 0 : -1;
}
void abort_transfer() {
transfer_finished();
}
static int ftp_send(const char *path, FILE *fp, putmode_t how,
transfer_mode_t mode, ftp_transfer_func hookf)
{
int r;
long rp = ftp->restart_offset;
ftp->restart_offset = 0L;
if(how == putUnique && !ftp->has_stou_command)
return -1;
if (how == putTryUnique && !ftp->has_stou_command)
how = putNormal;
reset_transfer_info();
ftp->ti.transfer_is_put = true;
if(ftp_init_transfer() != 0)
return -1;
ftp_type(mode);
if(rp > 0) {
/* fp is assumed to be fseek'd already */
ftp_cmd("REST %ld", rp);
if(ftp->code != ctContinue)
return -1;
ftp->ti.size = rp;
ftp->ti.restart_size = rp;
}
ftp_set_tmp_verbosity(vbError);
switch (how) {
case putAppend:
ftp_cmd("APPE %s", path);
break;
case putTryUnique:
case putUnique:
ftp_cmd("STOU %s", path);
if (ftp->fullcode == 502 || ftp->fullcode == 504) {
ftp->has_stou_command = false;
if (how == putTryUnique)
how = putNormal;
else
break;
}
else
break;
default:
ftp_cmd("STOR %s", path);
break;
}
if(ftp->code != ctPrelim)
return -1;
if(how == putUnique) {
/* try to figure out remote filename */
char *e = strstr(ftp->reply, " for ");
if(e) {
int l;
e += 5;
l = strlen(e);
if(l) {
free(ftp->ti.local_name);
if(*e == '\'')
ftp->ti.local_name = xstrndup(e+1, l-3);
else
ftp->ti.local_name = xstrndup(e, l-1);
ftp_trace("parsed unique filename as '%s'\n",
ftp->ti.local_name);
}
}
}
if(!sock_accept(ftp->data, "w", ftp_is_passive())) {
ftp_err(_("data connection not accepted\n"));
return -1;
}
ftp_cache_flush_mark_for(path);
if(mode == tmBinary)
r = FILE_send_binary(fp, ftp->data);
else
r = FILE_send_ascii(fp, ftp->data);
sock_flush(ftp->data);
sock_destroy(ftp->data);
ftp->data = 0;
if(r == 0) {
transfer_finished();
ftp_read_reply();
ftp->ti.ioerror = (ftp->code != ctComplete);
if(ftp->code != ctComplete) {
ftp_trace("transfer failed\n");
return -1;
}
} else
transfer_finished();
return 0;
}