mirror of
https://github.com/clearlinux/uwsgi.git
synced 2026-08-30 13:50:55 +00:00
Unbit
d4130dbc4b
preliminary api for language-independent body read another step completed body read language independent implementation ported gevent to the new read/write api ported websockets to the new read/write api removed channels subsystem removed channels subsystem ported lua to the new read/write api fixed post-buffering readline is still broken improved request body readline very difficult test for readline()/read() combo passed other improvements in postbuffering/read/readline ported --http-socket to the new api added X-Forwarded-SSL management removed old api more refactoring ported the RACK plugin to the new api ported psgi plugin to the new api defintely removed clustering simpified ifdel hell simpified ifdef hell removed useless configuration options
108 lines
2.7 KiB
C
108 lines
2.7 KiB
C
#include <uwsgi.h>
|
|
|
|
/*
|
|
|
|
upload progress facilities
|
|
|
|
*/
|
|
|
|
extern struct uwsgi_server uwsgi;
|
|
|
|
char *uwsgi_upload_progress_create(struct wsgi_request *wsgi_req, int *fd) {
|
|
const char *x_progress_id = "X-Progress-ID=";
|
|
char *xpi_ptr = (char *) x_progress_id;
|
|
uint16_t i;
|
|
char *upload_progress_filename = NULL;
|
|
|
|
if (wsgi_req->uri_len <= 51)
|
|
return NULL;
|
|
|
|
|
|
for (i = 0; i < wsgi_req->uri_len; i++) {
|
|
if (wsgi_req->uri[i] == xpi_ptr[0]) {
|
|
if (xpi_ptr[0] == '=') {
|
|
if (wsgi_req->uri + i + 36 <= wsgi_req->uri + wsgi_req->uri_len) {
|
|
upload_progress_filename = wsgi_req->uri + i + 1;
|
|
}
|
|
break;
|
|
}
|
|
xpi_ptr++;
|
|
}
|
|
else {
|
|
xpi_ptr = (char *) x_progress_id;
|
|
}
|
|
}
|
|
|
|
// now check for valid uuid (from spec available at http://en.wikipedia.org/wiki/Universally_unique_identifier)
|
|
if (!upload_progress_filename)
|
|
return NULL;
|
|
|
|
uwsgi_log("upload progress uuid = %.*s\n", 36, upload_progress_filename);
|
|
if (!check_hex(upload_progress_filename, 8))
|
|
return NULL;
|
|
if (upload_progress_filename[8] != '-')
|
|
return NULL;
|
|
|
|
if (!check_hex(upload_progress_filename + 9, 4))
|
|
return NULL;
|
|
if (upload_progress_filename[13] != '-')
|
|
return NULL;
|
|
|
|
if (!check_hex(upload_progress_filename + 14, 4))
|
|
return NULL;
|
|
if (upload_progress_filename[18] != '-')
|
|
return NULL;
|
|
|
|
if (!check_hex(upload_progress_filename + 19, 4))
|
|
return NULL;
|
|
if (upload_progress_filename[23] != '-')
|
|
return NULL;
|
|
|
|
if (!check_hex(upload_progress_filename + 24, 12))
|
|
return NULL;
|
|
|
|
upload_progress_filename = uwsgi_concat4n(uwsgi.upload_progress, strlen(uwsgi.upload_progress), "/", 1, upload_progress_filename, 36, ".js", 3);
|
|
// here we use O_EXCL to avoid eventual application bug in uuid generation/using
|
|
*fd = open(upload_progress_filename, O_WRONLY | O_CREAT | O_EXCL, S_IRUSR | S_IWUSR | S_IRGRP);
|
|
if (*fd < 0) {
|
|
uwsgi_error_open(upload_progress_filename);
|
|
free(upload_progress_filename);
|
|
return NULL;
|
|
}
|
|
|
|
return upload_progress_filename;
|
|
}
|
|
|
|
int uwsgi_upload_progress_update(struct wsgi_request *wsgi_req, int fd, size_t remains) {
|
|
char buf[4096];
|
|
|
|
int ret = snprintf(buf, 4096, "{ \"state\" : \"uploading\", \"received\" : %llu, \"size\" : %llu }\r\n", (unsigned long long) (wsgi_req->post_cl - remains), (unsigned long long) wsgi_req->post_cl);
|
|
if (ret < 0) {
|
|
return -1;
|
|
}
|
|
|
|
if (lseek(fd, 0, SEEK_SET)) {
|
|
uwsgi_error("uwsgi_upload_progress_update()/lseek()");
|
|
return -1;
|
|
}
|
|
|
|
if (write(fd, buf, ret) != ret) {
|
|
uwsgi_error("uwsgi_upload_progress_update()/write()");
|
|
return -1;
|
|
}
|
|
|
|
if (fsync(fd)) {
|
|
uwsgi_error("uwsgi_upload_progress_update()/fsync()");
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
void uwsgi_upload_progress_destroy(char *filename, int fd) {
|
|
close(fd);
|
|
if (unlink(filename)) {
|
|
uwsgi_error("uwsgi_upload_progress_destroy()/unlink()");
|
|
}
|
|
free(filename);
|
|
}
|