The-Homebrew-Cloud/ftpiiu_plugin
2
0
Fork 0
mirror of https://github.com/wiiu-env/ftpiiu_plugin.git synced 2024-11-17 18:29:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

significantly increase throughput

Browse Source
This commit is contained in:
mtheall 2014-11-25 02:30:55 -06:00
parent a217c5a3b4
commit 38a2b11e86
1 changed files with 181 additions and 99 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

204
source/ftp.c
View File

@ -2,6 +2,7 @@
#include <arpa/inet.h> #include <arpa/inet.h>
#include <ctype.h> #include <ctype.h>
#include <errno.h> #include <errno.h>
#include <fcntl.h>
#include <malloc.h> #include <malloc.h>
#include <netinet/in.h> #include <netinet/in.h>
#include <poll.h> #include <poll.h>
@ -14,7 +15,6 @@
#ifdef _3DS #ifdef _3DS
#include <3ds.h> #include <3ds.h>
#else #else
#include <fcntl.h>
#include <dirent.h> #include <dirent.h>
#include <sys/stat.h> #include <sys/stat.h>
#endif #endif
@ -36,6 +36,8 @@
#endif #endif
#define POLL_UNKNOWN (~(POLLIN|POLLOUT)) #define POLL_UNKNOWN (~(POLLIN|POLLOUT))
#define XFER_BUFFERSIZE 4096
#define CMD_BUFFERSIZE 1024
#define SOC_ALIGN 0x1000 #define SOC_ALIGN 0x1000
#define SOC_BUFFERSIZE 0x100000 #define SOC_BUFFERSIZE 0x100000
#define LISTEN_PORT 5000 #define LISTEN_PORT 5000
@ -110,8 +112,8 @@ struct ftp_session_t
ftp_session_t *next; /*!< link to next session */ ftp_session_t *next; /*!< link to next session */
ftp_session_t *prev; /*!< link to prev session */ ftp_session_t *prev; /*!< link to prev session */
void (*transfer)(ftp_session_t*); /*! data transfer callback */ int (*transfer)(ftp_session_t*); /*! data transfer callback */
char buffer[1024]; /*! persistent data between callbacks */ char buffer[XFER_BUFFERSIZE]; /*! persistent data between callbacks */
size_t bufferpos; /*! persistent buffer position between callbacks */ size_t bufferpos; /*! persistent buffer position between callbacks */
size_t buffersize; /*! persistent buffer size between callbacks */ size_t buffersize; /*! persistent buffer size between callbacks */
uint64_t filepos; /*! persistent file position between callbacks */ uint64_t filepos; /*! persistent file position between callbacks */
@ -251,6 +253,34 @@ next_data_port(void)
#endif #endif
} }
/*! set a socket to non-blocking
*
* @param[in] fd socket
*
* @returns error
*/
static int
ftp_set_socket_nonblocking(int fd)
{
int rc, flags;
flags = fcntl(fd, F_GETFL, 0);
if(flags == -1)
{
console_print("fcntl: %s\n", strerror(SOC_GetErrno()));
return -1;
}
rc = fcntl(fd, F_SETFL, flags | O_NONBLOCK);
if(rc != 0)
{
console_print("fcntl: %s\n", strerror(SOC_GetErrno()));
return -1;
}
return 0;
}
/*! close a socket /*! close a socket
* *
* @param[in] fd socket to close * @param[in] fd socket to close
@ -654,6 +684,16 @@ ftp_session_set_state(ftp_session_t *session,
} }
} }
static void
ftp_session_transfer(ftp_session_t *session)
{
int rc;
do
{
rc = session->transfer(session);
} while(rc == 0);
}
__attribute__((format(printf,3,4))) __attribute__((format(printf,3,4)))
/*! send ftp response to ftp session's peer /*! send ftp response to ftp session's peer
* *
@ -669,7 +709,7 @@ ftp_send_response(ftp_session_t *session,
int code, int code,
const char *fmt, ...) const char *fmt, ...)
{ {
static char buffer[1024]; static char buffer[CMD_BUFFERSIZE];
ssize_t rc, to_send; ssize_t rc, to_send;
va_list ap; va_list ap;
@ -823,7 +863,7 @@ ftp_session_new(int listen_fd)
static int static int
ftp_session_accept(ftp_session_t *session) ftp_session_accept(ftp_session_t *session)
{ {
int new_fd; int rc, new_fd;
struct sockaddr_in addr; struct sockaddr_in addr;
socklen_t addrlen = sizeof(addr); socklen_t addrlen = sizeof(addr);
@ -840,7 +880,16 @@ ftp_session_accept(ftp_session_t *session)
if(new_fd < 0) if(new_fd < 0)
{ {
console_print("accept: %s\n", strerror(SOC_GetErrno())); console_print("accept: %s\n", strerror(SOC_GetErrno()));
ftp_session_close_pasv(session); ftp_session_set_state(session, COMMAND_STATE);
ftp_send_response(session, 425, "Failed to establish connection\r\n");
return -1;
}
rc = ftp_set_socket_nonblocking(new_fd);
if(rc != 0)
{
ftp_closesocket(new_fd, 1);
ftp_session_set_state(session, COMMAND_STATE);
ftp_send_response(session, 425, "Failed to establish connection\r\n"); ftp_send_response(session, 425, "Failed to establish connection\r\n");
return -1; return -1;
} }
@ -872,8 +921,6 @@ ftp_session_connect(ftp_session_t *session)
{ {
int rc; int rc;
if(session->flags & SESSION_PORT)
{
/* clear PORT flag */ /* clear PORT flag */
session->flags &= ~SESSION_PORT; session->flags &= ~SESSION_PORT;
@ -882,7 +929,6 @@ ftp_session_connect(ftp_session_t *session)
if(session->data_fd < 0) if(session->data_fd < 0)
{ {
console_print("socket: %s\n", strerror(SOC_GetErrno())); console_print("socket: %s\n", strerror(SOC_GetErrno()));
ftp_send_response(session, 425, "Failed to establish connection\r\n");
return -1; return -1;
} }
@ -894,25 +940,19 @@ ftp_session_connect(ftp_session_t *session)
console_print("connect: %s\n", strerror(SOC_GetErrno())); console_print("connect: %s\n", strerror(SOC_GetErrno()));
ftp_closesocket(session->data_fd, 0); ftp_closesocket(session->data_fd, 0);
session->data_fd = -1; session->data_fd = -1;
ftp_send_response(session, 425, "Failed to establish connection\r\n");
return -1; return -1;
} }
rc = ftp_set_socket_nonblocking(session->data_fd);
if(rc != 0)
return -1;
console_print("connected to %s:%u\n", console_print("connected to %s:%u\n",
inet_ntoa(session->peer_addr.sin_addr), inet_ntoa(session->peer_addr.sin_addr),
ntohs(session->peer_addr.sin_port)); ntohs(session->peer_addr.sin_port));
/* tell peer that connection has been established */
ftp_send_response(session, 150, "Ready\r\n");
return 0; return 0;
} }
else
{
/* peer didn't send PORT command */
ftp_send_response(session, 503, "Bad sequence of commands\r\n");
return -1;
}
}
/*! read command for ftp session /*! read command for ftp session
* *
@ -921,7 +961,7 @@ ftp_session_connect(ftp_session_t *session)
static void static void
ftp_session_read_command(ftp_session_t *session) ftp_session_read_command(ftp_session_t *session)
{ {
static char buffer[1024]; static char buffer[CMD_BUFFERSIZE];
ssize_t rc; ssize_t rc;
char *args; char *args;
ftp_command_t key, *command; ftp_command_t key, *command;
@ -1071,7 +1111,7 @@ ftp_session_poll(ftp_session_t *session)
ftp_send_response(session, 426, "Data connection failed\r\n"); ftp_send_response(session, 426, "Data connection failed\r\n");
} }
else if(pollinfo.revents & (POLLIN|POLLOUT)) else if(pollinfo.revents & (POLLIN|POLLOUT))
session->transfer(session); ftp_session_transfer(session);
break; break;
} }
} }
@ -1380,7 +1420,7 @@ build_path(ftp_session_t *session,
return 0; return 0;
} }
static void static int
list_transfer(ftp_session_t *session) list_transfer(ftp_session_t *session)
{ {
#ifdef _3DS #ifdef _3DS
@ -1402,7 +1442,7 @@ list_transfer(ftp_session_t *session)
ftp_session_close_cwd(session); ftp_session_close_cwd(session);
ftp_session_set_state(session, COMMAND_STATE); ftp_session_set_state(session, COMMAND_STATE);
ftp_send_response(session, 450, "failed to read directory\r\n"); ftp_send_response(session, 450, "failed to read directory\r\n");
return; return -1;
} }
if(entries == 0) if(entries == 0)
@ -1410,12 +1450,12 @@ list_transfer(ftp_session_t *session)
ftp_session_close_cwd(session); ftp_session_close_cwd(session);
ftp_session_set_state(session, COMMAND_STATE); ftp_session_set_state(session, COMMAND_STATE);
ftp_send_response(session, 226, "OK\r\n"); ftp_send_response(session, 226, "OK\r\n");
return; return -1;
} }
convert_name(name, dent.name); convert_name(name, dent.name);
if(strcmp(name, ".") == 0 || strcmp(name, "..") == 0) if(strcmp(name, ".") == 0 || strcmp(name, "..") == 0)
return; return 0;
session->buffersize = session->buffersize =
sprintf(session->buffer, sprintf(session->buffer,
@ -1431,11 +1471,11 @@ list_transfer(ftp_session_t *session)
ftp_session_close_cwd(session); ftp_session_close_cwd(session);
ftp_session_set_state(session, COMMAND_STATE); ftp_session_set_state(session, COMMAND_STATE);
ftp_send_response(session, 226, "OK\r\n"); ftp_send_response(session, 226, "OK\r\n");
return; return -1;
} }
if(strcmp(dent->d_name, ".") == 0 || strcmp(dent->d_name, "..") == 0) if(strcmp(dent->d_name, ".") == 0 || strcmp(dent->d_name, "..") == 0)
return; return 0;
snprintf(session->buffer, sizeof(session->buffer), snprintf(session->buffer, sizeof(session->buffer),
"%s/%s", session->cwd, dent->d_name); "%s/%s", session->cwd, dent->d_name);
@ -1446,7 +1486,7 @@ list_transfer(ftp_session_t *session)
ftp_session_close_cwd(session); ftp_session_close_cwd(session);
ftp_session_set_state(session, COMMAND_STATE); ftp_session_set_state(session, COMMAND_STATE);
ftp_send_response(session, 550, "unavailable\r\n"); ftp_send_response(session, 550, "unavailable\r\n");
return; return -1;
} }
session->buffersize = session->buffersize =
@ -1465,20 +1505,25 @@ list_transfer(ftp_session_t *session)
if(rc <= 0) if(rc <= 0)
{ {
if(rc < 0) if(rc < 0)
{
if(SOC_GetErrno() == EWOULDBLOCK)
return -1;
console_print("send: %s\n", strerror(SOC_GetErrno())); console_print("send: %s\n", strerror(SOC_GetErrno()));
}
else else
console_print("send: %s\n", strerror(ECONNRESET)); console_print("send: %s\n", strerror(ECONNRESET));
ftp_session_close_cwd(session); ftp_session_close_cwd(session);
ftp_session_set_state(session, COMMAND_STATE); ftp_session_set_state(session, COMMAND_STATE);
ftp_send_response(session, 426, "Connection broken during transfer\r\n"); ftp_send_response(session, 426, "Connection broken during transfer\r\n");
return; return -1;
} }
session->bufferpos += rc; session->bufferpos += rc;
return 0;
} }
static void static int
retrieve_transfer(ftp_session_t *session) retrieve_transfer(ftp_session_t *session)
{ {
ssize_t rc; ssize_t rc;
@ -1494,7 +1539,7 @@ retrieve_transfer(ftp_session_t *session)
ftp_send_response(session, 451, "Failed to read file\r\n"); ftp_send_response(session, 451, "Failed to read file\r\n");
else else
ftp_send_response(session, 226, "OK\r\n"); ftp_send_response(session, 226, "OK\r\n");
return; return -1;
} }
session->bufferpos = 0; session->bufferpos = 0;
@ -1506,20 +1551,25 @@ retrieve_transfer(ftp_session_t *session)
if(rc <= 0) if(rc <= 0)
{ {
if(rc < 0) if(rc < 0)
{
if(SOC_GetErrno() == EWOULDBLOCK)
return -1;
console_print("send: %s\n", strerror(SOC_GetErrno())); console_print("send: %s\n", strerror(SOC_GetErrno()));
}
else else
console_print("send: %s\n", strerror(ECONNRESET)); console_print("send: %s\n", strerror(ECONNRESET));
ftp_session_close_file(session); ftp_session_close_file(session);
ftp_session_set_state(session, COMMAND_STATE); ftp_session_set_state(session, COMMAND_STATE);
ftp_send_response(session, 426, "Connection broken during transfer\r\n"); ftp_send_response(session, 426, "Connection broken during transfer\r\n");
return; return -1;
} }
session->bufferpos += rc; session->bufferpos += rc;
return 0;
} }
static void static int
store_transfer(ftp_session_t *session) store_transfer(ftp_session_t *session)
{ {
ssize_t rc; ssize_t rc;
@ -1530,7 +1580,11 @@ store_transfer(ftp_session_t *session)
if(rc <= 0) if(rc <= 0)
{ {
if(rc < 0) if(rc < 0)
{
if(SOC_GetErrno() == EWOULDBLOCK)
return -1;
console_print("recv: %s\n", strerror(SOC_GetErrno())); console_print("recv: %s\n", strerror(SOC_GetErrno()));
}
ftp_session_close_file(session); ftp_session_close_file(session);
ftp_session_set_state(session, COMMAND_STATE); ftp_session_set_state(session, COMMAND_STATE);
@ -1539,7 +1593,7 @@ store_transfer(ftp_session_t *session)
ftp_send_response(session, 226, "OK\r\n"); ftp_send_response(session, 226, "OK\r\n");
else else
ftp_send_response(session, 426, "Connection broken during transfer\r\n"); ftp_send_response(session, 426, "Connection broken during transfer\r\n");
return; return -1;
} }
session->bufferpos = 0; session->bufferpos = 0;
@ -1552,10 +1606,11 @@ store_transfer(ftp_session_t *session)
ftp_session_close_file(session); ftp_session_close_file(session);
ftp_session_set_state(session, COMMAND_STATE); ftp_session_set_state(session, COMMAND_STATE);
ftp_send_response(session, 451, "Failed to write file\r\n"); ftp_send_response(session, 451, "Failed to write file\r\n");
return; return -1;
} }
session->bufferpos += rc; session->bufferpos += rc;
return 0;
} }
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
@ -1707,14 +1762,6 @@ FTP_DECLARE(LIST)
ftp_session_set_state(session, COMMAND_STATE); ftp_session_set_state(session, COMMAND_STATE);
return ftp_send_response(session, 425, "can't open data connection\r\n"); return ftp_send_response(session, 425, "can't open data connection\r\n");
} }
}
else if(session->flags & SESSION_PASV)
ftp_session_set_state(session, DATA_CONNECT_STATE);
else
{
ftp_session_set_state(session, COMMAND_STATE);
return ftp_send_response(session, 503, "Bad sequence of commands\r\n");
}
session->flags &= ~(SESSION_RECV|SESSION_SEND); session->flags &= ~(SESSION_RECV|SESSION_SEND);
session->flags |= SESSION_SEND; session->flags |= SESSION_SEND;
@ -1722,9 +1769,26 @@ FTP_DECLARE(LIST)
session->transfer = list_transfer; session->transfer = list_transfer;
session->bufferpos = 0; session->bufferpos = 0;
session->buffersize = 0; session->buffersize = 0;
return ftp_send_response(session, 150, "Ready\r\n");
}
else if(session->flags & SESSION_PASV)
{
session->flags &= ~(SESSION_RECV|SESSION_SEND);
session->flags |= SESSION_SEND;
session->transfer = list_transfer;
session->bufferpos = 0;
session->buffersize = 0;
ftp_session_set_state(session, DATA_CONNECT_STATE);
return 0; return 0;
} }
ftp_session_set_state(session, COMMAND_STATE);
return ftp_send_response(session, 503, "Bad sequence of commands\r\n");
}
FTP_DECLARE(MKD) FTP_DECLARE(MKD)
{ {
#ifdef _3DS #ifdef _3DS
@ -2021,14 +2085,6 @@ FTP_DECLARE(RETR)
ftp_session_set_state(session, COMMAND_STATE); ftp_session_set_state(session, COMMAND_STATE);
return ftp_send_response(session, 425, "can't open data connection\r\n"); return ftp_send_response(session, 425, "can't open data connection\r\n");
} }
}
else if(session->flags & SESSION_PASV)
ftp_session_set_state(session, DATA_CONNECT_STATE);
else
{
ftp_session_set_state(session, COMMAND_STATE);
return ftp_send_response(session, 503, "Bad sequence of commands\r\n");
}
session->flags &= ~(SESSION_RECV|SESSION_SEND); session->flags &= ~(SESSION_RECV|SESSION_SEND);
session->flags |= SESSION_SEND; session->flags |= SESSION_SEND;
@ -2036,9 +2092,26 @@ FTP_DECLARE(RETR)
session->transfer = retrieve_transfer; session->transfer = retrieve_transfer;
session->bufferpos = 0; session->bufferpos = 0;
session->buffersize = 0; session->buffersize = 0;
return ftp_send_response(session, 150, "Ready\r\n");
}
else if(session->flags & SESSION_PASV)
{
session->flags &= ~(SESSION_RECV|SESSION_SEND);
session->flags |= SESSION_SEND;
session->transfer = retrieve_transfer;
session->bufferpos = 0;
session->buffersize = 0;
ftp_session_set_state(session, DATA_CONNECT_STATE);
return 0; return 0;
} }
ftp_session_set_state(session, COMMAND_STATE);
return ftp_send_response(session, 503, "Bad sequence of commands\r\n");
}
FTP_DECLARE(RMD) FTP_DECLARE(RMD)
{ {
#ifdef _3DS #ifdef _3DS
@ -2122,7 +2195,7 @@ FTP_DECLARE(RNTO)
#else #else
int rc; int rc;
#endif #endif
char buffer[1024]; char buffer[XFER_BUFFERSIZE];
console_print("%s %s\n", __func__, args ? args : ""); console_print("%s %s\n", __func__, args ? args : "");
@ -2133,7 +2206,7 @@ FTP_DECLARE(RNTO)
session->flags &= ~SESSION_RENAME; session->flags &= ~SESSION_RENAME;
memcpy(buffer, session->buffer, 1024); memcpy(buffer, session->buffer, XFER_BUFFERSIZE);
if(build_path(session, args) != 0) if(build_path(session, args) != 0)
return ftp_send_response(session, 554, "%s\r\n", strerror(errno)); return ftp_send_response(session, 554, "%s\r\n", strerror(errno));
@ -2191,14 +2264,6 @@ FTP_DECLARE(STOR)
ftp_session_set_state(session, COMMAND_STATE); ftp_session_set_state(session, COMMAND_STATE);
return ftp_send_response(session, 425, "can't open data connection\r\n"); return ftp_send_response(session, 425, "can't open data connection\r\n");
} }
}
else if(session->flags & SESSION_PASV)
ftp_session_set_state(session, DATA_CONNECT_STATE);
else
{
ftp_session_set_state(session, COMMAND_STATE);
return ftp_send_response(session, 503, "Bad sequence of commands\r\n");
}
session->flags &= ~(SESSION_RECV|SESSION_SEND); session->flags &= ~(SESSION_RECV|SESSION_SEND);
session->flags |= SESSION_RECV; session->flags |= SESSION_RECV;
@ -2206,9 +2271,26 @@ FTP_DECLARE(STOR)
session->transfer = store_transfer; session->transfer = store_transfer;
session->bufferpos = 0; session->bufferpos = 0;
session->buffersize = 0; session->buffersize = 0;
return ftp_send_response(session, 150, "Ready\r\n");
}
else if(session->flags & SESSION_PASV)
{
session->flags &= ~(SESSION_RECV|SESSION_SEND);
session->flags |= SESSION_RECV;
session->transfer = store_transfer;
session->bufferpos = 0;
session->buffersize = 0;
ftp_session_set_state(session, DATA_CONNECT_STATE);
return 0; return 0;
} }
ftp_session_set_state(session, COMMAND_STATE);
return ftp_send_response(session, 503, "Bad sequence of commands\r\n");
}
FTP_DECLARE(STOU) FTP_DECLARE(STOU)
{ {
console_print("%s %s\n", __func__, args ? args : ""); console_print("%s %s\n", __func__, args ? args : "");