int main(int argc, char **argv) {
skod_t skod;
ftp_t ftp;
skod_init(&skod);
skod_parse_cla(argc, argv, &skod);
signal(SIGINT, signal_handler);
/* Init FTP*/
ftp.user = skod.user;
ftp.password = skod.password;
ftp.server = skod.server;
ftp.port = skod.port;
ftp.alarm_sec = 1; /* High risk */
ftp_mkcon(&ftp);
/* --dest, -e*/
if ( skod.dest != NULL )
ftp_cwd(&ftp, skod.dest);
else if ( skod.dest == NULL && flag == 4 )
die("You need to pass --dest (destination folder) with --upload.");
switch(flag) {
case 1:
ftp_list(&ftp, skod.path, 1);
break;
case 2:
ftp_remove(&ftp, skod.path);
break;
case 3:
ftp_download_single(&ftp, skod.path);
break;
case 4:
ftp_upload_single(&ftp, skod.path);
break;
case 5:
ftp_cat(&ftp, skod.path);
break;
case 6:
printf("%d\n", ftp_size(&ftp, skod.path));
break;
case 7:
printf("%s\n", ftp_pwd(&ftp));
break;
case 8:
ftp_delete(&ftp, skod.path);
break;
case 10:
ftp_mdtm(&ftp, skod.path);
break;
case 99:
skod_scan(&skod, &ftp);
break;
}
ftp_close(&ftp);
return 0;
}
int conn_init( conn_t *conn )
{
char *proxy = conn->conf->http_proxy, *host = conn->conf->no_proxy;
int i;
if( *conn->conf->http_proxy == 0 )
{
proxy = NULL;
}
else if( *conn->conf->no_proxy != 0 )
{
for( i = 0; ; i ++ )
if( conn->conf->no_proxy[i] == 0 )
{
if( strstr( conn->host, host ) != NULL )
proxy = NULL;
host = &conn->conf->no_proxy[i+1];
if( conn->conf->no_proxy[i+1] == 0 )
break;
}
}
conn->proxy = proxy != NULL;
if( conn->proto == PROTO_FTP && !conn->proxy )
{
conn->ftp->local_if = conn->local_if;
conn->ftp->ftp_mode = FTP_PASSIVE;
if( !ftp_connect( conn->ftp, conn->host, conn->port, conn->user, conn->pass ) )
{
conn->message = conn->ftp->message;
conn_disconnect( conn );
return( 0 );
}
conn->message = conn->ftp->message;
if( !ftp_cwd( conn->ftp, conn->dir ) )
{
conn_disconnect( conn );
return( 0 );
}
}
else
{
conn->http->local_if = conn->local_if;
if( !http_connect( conn->http, conn->proto, proxy, conn->host, conn->port, conn->user, conn->pass ) )
{
conn->message = conn->http->headers;
conn_disconnect( conn );
return( 0 );
}
conn->message = conn->http->headers;
conn->fd = conn->http->fd;
}
return( 1 );
}
/**
* Generates response message for client
* @param cmd Current command
* @param state Current connection state
*/
void response(Command *cmd, State *state)
{
switch(lookup_cmd(cmd->command)){
case USER: ftp_user(cmd,state); break;
case PASS: ftp_pass(cmd,state); break;
case PASV: ftp_pasv(cmd,state); break;
case LIST: ftp_list(cmd,state); break;
case CWD: ftp_cwd(cmd,state); break;
case PWD: ftp_pwd(cmd,state); break;
case MKD: ftp_mkd(cmd,state); break;
case RMD: ftp_rmd(cmd,state); break;
case RETR: ftp_retr(cmd,state); break;
case STOR: ftp_stor(cmd,state); break;
case DELE: ftp_dele(cmd,state); break;
case SIZE: ftp_size(cmd,state); break;
case ABOR: ftp_abor(state); break;
case QUIT: ftp_quit(state); break;
case TYPE: ftp_type(cmd,state); break;
case CDUP: ftp_cdup(state); break;
case HELP: ftp_help(cmd, state); break;
case NLST: ftp_nlst(cmd, state); break;
case RNFR: ftp_rnfr(cmd, state); break;
case RNTO: ftp_rnto(cmd, state); break;
case APPE: ftp_appe(cmd, state); break;
case NOOP:
if(state->logged_in){
state->message = "200 Zzz...\n";
}else{
state->message = "530 Please login with USER and PASS\n";
}
write_state(state);
break;
default:
state->message = "500 Unknown command\n";
write_state(state);
break;
}
}
/* Get file size and other information */
int conn_info( conn_t *conn )
{
/* It's all a bit messed up.. But it works. */
if( conn->proto == PROTO_FTP && !conn->proxy )
{
ftp_command( conn->ftp, "REST %lld", 1 );
if( ftp_wait( conn->ftp ) / 100 == 3 ||
conn->ftp->status / 100 == 2 )
{
conn->supported = 1;
ftp_command( conn->ftp, "REST %lld", 0 );
ftp_wait( conn->ftp );
}
else
{
conn->supported = 0;
}
if( !ftp_cwd( conn->ftp, conn->dir ) )
return( 0 );
conn->size = ftp_size( conn->ftp, conn->file, MAX_REDIR );
if( conn->size < 0 )
conn->supported = 0;
if( conn->size == -1 )
return( 0 );
else if( conn->size == -2 )
conn->size = INT_MAX;
}
else
{
char s[MAX_STRING], *t;
long long int i = 0;
do
{
conn->currentbyte = 1;
if( !conn_setup( conn ) )
return( 0 );
conn_exec( conn );
conn_disconnect( conn );
/* Code 3xx == redirect */
if( conn->http->status / 100 != 3 )
break;
if( ( t = http_header( conn->http, "location:" ) ) == NULL )
return( 0 );
sscanf( t, "%1023s", s ); // Warning: truncating to MAX_STRING
if( strstr( s, "://" ) == NULL)
{
sprintf( conn->http->headers, "%s%s",
conn_url( conn ), s );
strncpy( s, conn->http->headers, MAX_STRING );
}
else if( s[0] == '/' )
{
sprintf( conn->http->headers, "http://%s:%i%s",
conn->host, conn->port, s );
strncpy( s, conn->http->headers, MAX_STRING );
}
conn_set( conn, s );
i ++;
}
while( conn->http->status / 100 == 3 && i < MAX_REDIR );
if( i == MAX_REDIR )
{
sprintf( conn->message, _("Too many redirects.\n") );
return( 0 );
}
conn->size = http_size( conn->http );
if( conn->http->status == 206 && conn->size >= 0 )
{
conn->supported = 1;
conn->size ++;
}
else if( conn->http->status == 200 || conn->http->status == 206 )
{
conn->supported = 0;
conn->size = INT_MAX;
}
else
{
t = strchr( conn->message, '\n' );
if( t == NULL )
sprintf( conn->message, _("Unknown HTTP error.\n") );
else
*t = 0;
return( 0 );
}
}
return( 1 );
}
int main(int argc, char* argv[])
{
if (argc != 2)
{
print_usage(argv[0]);
return EXIT_FAILURE;
}
URL url;
url_init(&url);
int error = url_from_string(&url, argv[1]);
if (error)
{
fprintf(stderr, "Could not sucessfully parse FTP url (error code: %d)\n", error);
url_destroy(&url);
return EXIT_FAILURE;
}
error = url_host_to_ip(&url);
if (error)
{
fprintf(stderr, "Could not get an IPv4 IP address from hostname %s (error code: %d)\n", url.host, error);
return EXIT_FAILURE;
}
FTP ftp;
error = ftp_connect(&ftp, url.ip, url.port ? url.port : 21);
if (error)
{
fprintf(stderr, "Could not connect to ftp at IP %s, port %d\n", url.ip, url.port ? url.port : 21);
return EXIT_FAILURE;
}
const char* user = strlen(url.user) ? url.user : "******";
const char* pass = strlen(url.password) ? url.password : "******";
error = ftp_login(&ftp, user, pass);
if (error)
{
fprintf(stderr, "Could not login with user %s and pass %s\n", user, pass);
return EXIT_FAILURE;
}
char path[1024] = "";
for (int i = 0; i < url.num_parts - 1; ++i) {
strcat(path, url.parts[i]);
strcat(path, "/");
}
if (path[0] != 0) {
error = ftp_cwd(&ftp, path);
if (error) {
perror("ftp_cwd");
return error;
}
}
error = ftp_pasv(&ftp);
if (error) {
perror("ftp_pasv");
return error;
}
const char* file_name = url.num_parts ? url.parts[url.num_parts - 1] : "";
error = ftp_retr(&ftp, file_name);
if (error) {
perror("ftp_retr");
return error;
}
error = ftp_download(&ftp, file_name);
if (error) {
perror("ftp_download");
return error;
}
error = ftp_disconnect(&ftp);
if (error) {
perror("ftp_disconnect");
return error;
}
url_destroy(&url);
return EXIT_SUCCESS;
}
