aniani/elinks
1
0
Fork 0
mirror of https://github.com/rkd77/elinks.git synced 2024-12-04 14:46:47 -05:00
Code

1008: upload of big files.

Browse Source 
Files bigger than 65536 bytes are loaded on "demand".
TODO: Add a progress bar of uploads.
This commit is contained in:
Witold Filipczyk 2008-05-02 20:48:10 +02:00 committed by Witold Filipczyk
parent 0b2edbd33a
commit ec382345c8
5 changed files with 345 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

107
src/protocol/file/cgi.c
View File

@ -80,22 +80,120 @@ close_pipe_and_read(struct socket *data_socket)
read_from_socket(conn->socket, rb, S_SENT, http_got_header);
}
#define POST_BUFFER_SIZE 4096
#define BIG_READ 65536
static void send_big_files2(struct socket *socket);
static void
send_big_files(struct socket *socket)
{
struct connection *conn = socket->conn;
struct http_connection_info *http = conn->info;
unsigned char *post = http->post_data;
unsigned char buffer[POST_BUFFER_SIZE];
unsigned char *big_file = strchr(post, BIG_FILE_CHAR);
struct string data;
int n = 0;
int finish = 0;
if (!init_string(&data)) {
abort_connection(conn, S_OUT_OF_MEM);
return;
}
if (!big_file) {
finish = 1;
big_file = strchr(post, '\0');
}
while (post < big_file) {
int h1, h2;
h1 = unhx(post[0]);
assertm(h1 >= 0 && h1 < 16, "h1 in the POST buffer is %d (%d/%c)", h1, post[0], post[0]);
if_assert_failed h1 = 0;
h2 = unhx(post[1]);
assertm(h2 >= 0 && h2 < 16, "h2 in the POST buffer is %d (%d/%c)", h2, post[1], post[1]);
if_assert_failed h2 = 0;
buffer[n++] = (h1<<4) + h2;
post += 2;
if (n == POST_BUFFER_SIZE) {
add_bytes_to_string(&data, buffer, n);
n = 0;
}
}
if (n) add_bytes_to_string(&data, buffer, n);
if (finish) {
write_to_socket(socket, data.source, data.length, S_SENT,
close_pipe_and_read);
} else {
unsigned char *end = strchr(big_file + 1, BIG_FILE_CHAR);
assert(end);
*end = '\0';
http->post_fd = open(big_file + 1, O_RDONLY);
*end = BIG_FILE_CHAR;
http->post_data = end + 1;
socket->state = SOCKET_END_ONCLOSE;
write_to_socket(socket, data.source, data.length, S_TRANS,
send_big_files2);
}
done_string(&data);
}
static void
send_big_files2(struct socket *socket)
{
struct connection *conn = socket->conn;
struct http_connection_info *http = conn->info;
unsigned char buffer[BIG_READ];
int n = safe_read(http->post_fd, buffer, BIG_READ);
if (n > 0) {
socket->state = SOCKET_END_ONCLOSE;
write_to_socket(socket, buffer, n, S_TRANS,
send_big_files2);
} else {
close(http->post_fd);
http->post_fd = -1;
send_big_files(socket);
}
}
static void
send_post_data(struct connection *conn)
{
#define POST_BUFFER_SIZE 4096
unsigned char *post = conn->uri->post;
unsigned char *postend;
unsigned char buffer[POST_BUFFER_SIZE];
struct string data;
int n = 0;
postend = strchr(post, '\n');
if (postend) post = postend + 1;
if (post) {
unsigned char *big_file = strchr(post, BIG_FILE_CHAR);
if (big_file) {
struct http_connection_info *http = conn->info;
http->post_data = post;
send_big_files(conn->data_socket);
return;
}
}
if (!init_string(&data)) {
abort_connection(conn, S_OUT_OF_MEM);
return;
}
postend = strchr(post, '\n');
if (postend) post = postend + 1;
/* FIXME: Code duplication with protocol/http/http.c! --witekfl */
while (post[0] && post[1]) {
@ -130,8 +228,9 @@ send_post_data(struct connection *conn)
close_pipe_and_read(conn->data_socket);
done_string(&data);
#undef POST_BUFFER_SIZE
}
#undef POST_BUFFER_SIZE
#undef BIG_READ
static void
send_request(struct connection *conn)

167
src/protocol/http/http.c
View File

@ -8,6 +8,7 @@
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
@ -51,11 +52,6 @@
#include "http_negotiate.h"
#endif
struct http_version {
int major;
int minor;
};
#define HTTP_0_9(x) ((x).major == 0 && (x).minor == 9)
#define HTTP_1_0(x) ((x).major == 1 && (x).minor == 0)
#define HTTP_1_1(x) ((x).major == 1 && (x).minor == 1)
@ -65,26 +61,13 @@ struct http_version {
#define POST_HTTP_1_1(x) ((x).major > 1 || ((x).major == 1 && (x).minor > 1))
struct http_connection_info {
enum blacklist_flags bl_flags;
struct http_version recv_version;
struct http_version sent_version;
int close;
#define LEN_CHUNKED -2 /* == we get data in unknown number of chunks */
#define LEN_FINISHED 0
int length;
/* Either bytes coming in this chunk yet or "parser state". */
#define CHUNK_DATA_END -3
#define CHUNK_ZERO_SIZE -2
#define CHUNK_SIZE -1
int chunk_remaining;
int code;
};
static struct auth_entry proxy_auth;
@ -488,6 +471,14 @@ http_end_request(struct connection *conn, enum connection_state state,
int notrunc)
{
shutdown_connection_stream(conn);
if (conn->info) {
struct http_connection_info *http = conn->info;
if (http->post_fd != -1) {
close(http->post_fd);
http->post_fd = -1;
}
}
if (conn->info && !((struct http_connection_info *) conn->info)->close
&& (!conn->socket->ssl) /* We won't keep alive ssl connections */
@ -543,6 +534,8 @@ init_http_connection_info(struct connection *conn, int major, int minor, int clo
http->sent_version.minor = minor;
http->close = close;
http->post_fd = -1;
/* The CGI code uses this too and blacklisting expects a host name. */
if (conn->proxied_uri->protocol != PROTOCOL_FILE)
http->bl_flags = get_blacklist_flags(conn->proxied_uri);
@ -587,6 +580,123 @@ accept_encoding_header(struct string *header)
#endif
}
/* This sets the Content-Length of POST data and counts big files. */
static size_t
post_length(unsigned char *post_data, unsigned int *count)
{
size_t size = 0;
size_t length = strlen(post_data);
unsigned char *end = post_data;
*count = 0;
while (1) {
struct stat sb;
unsigned char *begin;
int res;
begin = strchr(end, BIG_FILE_CHAR);
if (!begin) break;
end = strchr(begin + 1, BIG_FILE_CHAR);
if (!end) break;
*end = '\0';
res = stat(begin + 1, &sb);
*end = BIG_FILE_CHAR;
if (res) break;
(*count)++;
size += sb.st_size;
length -= (end - begin + 1);
end++;
}
size += (length / 2);
return size;
}
#define POST_BUFFER_SIZE 4096
#define BIG_READ 655360
static void send_big_files2(struct socket *socket);
static void
send_big_files(struct socket *socket)
{
struct connection *conn = socket->conn;
struct http_connection_info *http = conn->info;
unsigned char *post = http->post_data;
unsigned char buffer[POST_BUFFER_SIZE];
unsigned char *big_file = strchr(post, BIG_FILE_CHAR);
struct string data;
int n = 0;
int finish = 0;
if (!init_string(&data)) {
http_end_request(conn, S_OUT_OF_MEM, 0);
return;
}
if (!big_file) {
finish = 1;
big_file = strchr(post, '\0');
}
while (post < big_file) {
int h1, h2;
h1 = unhx(post[0]);
assertm(h1 >= 0 && h1 < 16, "h1 in the POST buffer is %d (%d/%c)", h1, post[0], post[0]);
if_assert_failed h1 = 0;
h2 = unhx(post[1]);
assertm(h2 >= 0 && h2 < 16, "h2 in the POST buffer is %d (%d/%c)", h2, post[1], post[1]);
if_assert_failed h2 = 0;
buffer[n++] = (h1<<4) + h2;
post += 2;
if (n == POST_BUFFER_SIZE) {
add_bytes_to_string(&data, buffer, n);
n = 0;
}
}
if (n) add_bytes_to_string(&data, buffer, n);
if (finish) {
request_from_socket(socket, data.source, data.length, S_SENT,
SOCKET_END_ONCLOSE, http_got_header);
} else {
unsigned char *end = strchr(big_file + 1, BIG_FILE_CHAR);
assert(end);
*end = '\0';
http->post_fd = open(big_file + 1, O_RDONLY);
*end = BIG_FILE_CHAR;
http->post_data = end + 1;
socket->state = SOCKET_END_ONCLOSE;
write_to_socket(socket, data.source, data.length, S_TRANS,
send_big_files2);
}
done_string(&data);
}
static void
send_big_files2(struct socket *socket)
{
struct connection *conn = socket->conn;
struct http_connection_info *http = conn->info;
unsigned char buffer[BIG_READ];
int n = safe_read(http->post_fd, buffer, BIG_READ);
if (n > 0) {
socket->state = SOCKET_END_ONCLOSE;
write_to_socket(socket, buffer, n, S_TRANS,
send_big_files2);
} else {
close(http->post_fd);
http->post_fd = -1;
send_big_files(socket);
}
}
static void
http_send_header(struct socket *socket)
{
@ -599,6 +709,7 @@ http_send_header(struct socket *socket)
struct uri *uri = conn->proxied_uri; /* Set to the real uri */
unsigned char *optstr;
int use_connect, talking_to_proxy;
int big_files = 0;
/* Sanity check for a host */
if (!uri || !uri->host || !*uri->host || !uri->hostlen) {
@ -932,6 +1043,7 @@ http_send_header(struct socket *socket)
* as set by get_form_uri(). This '\n' is dropped if any
* and replaced by correct '\r\n' termination here. */
unsigned char *postend = strchr(uri->post, '\n');
size_t size;
if (postend) {
add_to_string(&header, "Content-Type: ");
@ -941,7 +1053,8 @@ http_send_header(struct socket *socket)
post_data = postend ? postend + 1 : uri->post;
add_to_string(&header, "Content-Length: ");
add_long_to_string(&header, strlen(post_data) / 2);
size = post_length(post_data, &big_files);
add_long_to_string(&header, size);
add_crlf_to_string(&header);
}
@ -961,11 +1074,20 @@ http_send_header(struct socket *socket)
add_crlf_to_string(&header);
if (big_files) {
assert(!use_connect && post_data);
assert(http->post_fd == -1);
http->post_data = post_data;
socket->state = SOCKET_END_ONCLOSE;
write_to_socket(socket, header.source, header.length, S_TRANS,
send_big_files);
done_string(&header);
return;
}
/* CONNECT: Any POST data is for the origin server only.
* This was already checked above and post_data is NULL
* in that case. Verified with an assertion below. */
if (post_data) {
#define POST_BUFFER_SIZE 4096
unsigned char *post = post_data;
unsigned char buffer[POST_BUFFER_SIZE];
int n = 0;
@ -993,7 +1115,6 @@ http_send_header(struct socket *socket)
if (n)
add_bytes_to_string(&header, buffer, n);
#undef POST_BUFFER_SIZE
}
request_from_socket(socket, header.source, header.length, S_SENT,
@ -1001,6 +1122,8 @@ http_send_header(struct socket *socket)
done_string(&header);
}
#undef POST_BUFFER_SIZE
/* This function decompresses the data block given in @data (if it was
* compressed), which is long @len bytes. The decompressed data block is given
@ -1026,7 +1149,6 @@ decompress_data(struct connection *conn, unsigned char *data, int len,
int *length_of_block;
unsigned char *output = NULL;
#define BIG_READ 65536
if (http->length == LEN_CHUNKED) {
if (http->chunk_remaining == CHUNK_ZERO_SIZE)
@ -1110,6 +1232,7 @@ decompress_data(struct connection *conn, unsigned char *data, int len,
if (state == FINISHING) shutdown_connection_stream(conn);
return output;
}
#undef BIG_READ
static int
is_line_in_buffer(struct read_buffer *rb)

25
src/protocol/http/http.h
View File

@ -3,13 +3,36 @@
#define EL__PROTOCOL_HTTP_HTTP_H
#include "main/module.h"
#include "protocol/http/blacklist.h"
#include "protocol/protocol.h"
struct connection;
struct http_connection_info;
struct read_buffer;
struct socket;
struct http_version {
int major;
int minor;
};
struct http_connection_info {
enum blacklist_flags bl_flags;
struct http_version recv_version;
struct http_version sent_version;
int close;
int length;
int chunk_remaining;
int code;
/* Used by big files upload. */
unsigned char *post_data;
int post_fd;
};
extern struct module http_protocol_module;
extern protocol_handler_T http_protocol_handler;

11
src/protocol/uri.h
View File

@ -7,6 +7,7 @@ struct string;
#define POST_CHAR 1
#define POST_CHAR_S "\001"
#define BIG_FILE_CHAR '\002'
/* The uri structure is used to store the start position and length of commonly
* used uri fields. It is initialized by parse_uri(). It is possible that the
@ -54,6 +55,9 @@ struct uri {
unsigned int datalen:16;
unsigned int fragmentlen:16;
/* Number of files bigger than 1M */
unsigned int big_files:8;
/* Flags */
unsigned int ipv6:1; /* URI contains IPv6 host */
unsigned int form:1; /* URI originated from form */
@ -189,6 +193,13 @@ struct uri_list {
struct uri **uris;
};
/* I don't know where to put it. */
struct big_files_offset {
LIST_HEAD(struct big_files_offset);
int begin;
int end;
};
#define foreach_uri(uri, index, list) \
for (index = 0; index < (list)->size; index++) \
if ((uri = (list)->uris[index]))

72
src/viewer/text/form.c
View File

@ -59,6 +59,7 @@
* in viewer/common/. --pasky */
/** @relates submitted_value */
struct submitted_value *
init_submitted_value(unsigned char *name, unsigned char *value, enum form_type type,
struct form_control *fc, int position)
@ -906,8 +907,8 @@ check_boundary(struct string *data, struct boundary_info *boundary)
/** @todo FIXME: shouldn't we encode data at send time (in http.c) ? --Zas */
static void
encode_multipart(struct session *ses, LIST_OF(struct submitted_value) *l,
struct string *data,
struct boundary_info *boundary, int cp_from, int cp_to)
struct string *data, struct boundary_info *boundary,
LIST_OF(struct big_files_offset) *bfs, int cp_from, int cp_to)
{
struct conv_table *convert_table = NULL;
struct submitted_value *sv;
@ -967,6 +968,7 @@ encode_multipart(struct session *ses, LIST_OF(struct submitted_value) *l,
add_crlf_to_string(data);
if (*sv->value) {
struct stat stat_buf;
unsigned char *filename;
if (get_cmd_opt_bool("anonymous")) {
@ -979,9 +981,30 @@ encode_multipart(struct session *ses, LIST_OF(struct submitted_value) *l,
if (!filename) goto encode_error;
fh = open(filename, O_RDONLY);
if (fh == -1) {
mem_free(filename);
goto encode_error;
}
fstat(fh, &stat_buf);
if (stat_buf.st_size > (1L << 16)) { /* 65536 bytes */
struct big_files_offset *bfs_new = mem_calloc(1, sizeof(*bfs_new));
if (!bfs_new) {
mem_free(filename);
close(fh);
goto encode_error;
}
bfs_new->begin = data->length;
add_char_to_string(data, BIG_FILE_CHAR);
add_to_string(data, filename);
add_char_to_string(data, BIG_FILE_CHAR);
bfs_new->end = data->length;
add_to_list_end(*bfs, bfs_new);
mem_free(filename);
goto close_handle;
}
mem_free(filename);
if (fh == -1) goto encode_error;
set_bin(fh);
while (1) {
ssize_t rd = safe_read(fh, buffer, F_BUFLEN);
@ -998,6 +1021,7 @@ encode_multipart(struct session *ses, LIST_OF(struct submitted_value) *l,
break;
}
};
close_handle:
close(fh);
}
#undef F_BUFLEN
@ -1151,6 +1175,7 @@ get_form_uri(struct session *ses, struct document_view *doc_view,
{
struct boundary_info boundary;
INIT_LIST_OF(struct submitted_value, submit);
INIT_LIST_OF(struct big_files_offset, bfs);
struct string data;
struct string go;
int cp_from, cp_to;
@ -1184,7 +1209,8 @@ get_form_uri(struct session *ses, struct document_view *doc_view,
break;
case FORM_METHOD_POST_MP:
encode_multipart(ses, &submit, &data, &boundary, cp_from, cp_to);
encode_multipart(ses, &submit, &data, &boundary,
&bfs, cp_from, cp_to);
break;
case FORM_METHOD_POST_TEXT_PLAIN:
@ -1236,7 +1262,6 @@ get_form_uri(struct session *ses, struct document_view *doc_view,
{
/* Note that we end content type here by a simple '\n',
* replaced later by correct '\r\n' in http_send_header(). */
int i;
add_to_string(&go, form->action);
add_char_to_string(&go, POST_CHAR);
@ -1256,11 +1281,35 @@ get_form_uri(struct session *ses, struct document_view *doc_view,
add_char_to_string(&go, '\n');
}
for (i = 0; i < data.length; i++) {
unsigned char p[3];
if (list_empty(bfs)) {
int i;
ulonghexcat(p, NULL, (int) data.source[i], 2, '0', 0);
add_to_string(&go, p);
for (i = 0; i < data.length; i++) {
unsigned char p[3];
ulonghexcat(p, NULL, (int) data.source[i], 2, '0', 0);
add_to_string(&go, p);
}
} else {
struct big_files_offset *b;
int i = 0;
foreach (b, bfs) {
for (; i < b->begin; i++) {
unsigned char p[3];
ulonghexcat(p, NULL, (int) data.source[i], 2, '0', 0);
add_to_string(&go, p);
}
add_bytes_to_string(&go, data.source + i, b->end - b->begin);
i = b->end;
}
for (; i < data.length; i++) {
unsigned char p[3];
ulonghexcat(p, NULL, (int) data.source[i], 2, '0', 0);
add_to_string(&go, p);
}
}
}
}
@ -1269,7 +1318,10 @@ get_form_uri(struct session *ses, struct document_view *doc_view,
uri = get_uri(go.source, 0);
done_string(&go);
if (uri) uri->form = 1;
if (uri) {
uri->form = 1;
}
free_list(bfs);
return uri;
}