mirror of
https://github.com/php/pecl-php-uploadprogress.git
synced 2026-03-24 07:02:06 +01:00
581 lines
16 KiB
C
581 lines
16 KiB
C
/*
|
||
+----------------------------------------------------------------------+
|
||
| Uploadprogress extension |
|
||
+----------------------------------------------------------------------+
|
||
| Copyright (c) The PHP Group |
|
||
+----------------------------------------------------------------------+
|
||
| This source file is subject to version 3.01 of the PHP license, |
|
||
| that is bundled with this package in the file LICENSE, and is |
|
||
| available through the world-wide-web at the following url: |
|
||
| http://www.php.net/license/3_01.txt. |
|
||
| If you did not receive a copy of the PHP license and are unable to |
|
||
| obtain it through the world-wide-web, please send a note to |
|
||
| license@php.net so we can mail you a copy immediately. |
|
||
+----------------------------------------------------------------------+
|
||
| Author: Christian Stocker (chregu@liip.ch) |
|
||
| Derived from: Doru Petrescu (pdoru-php-upm@kappa.ro) |
|
||
| http://pdoru.from.ro/upload-progress-meter/ |
|
||
+----------------------------------------------------------------------+
|
||
*/
|
||
|
||
#include "php_uploadprogress.h"
|
||
#include "rfc1867.h"
|
||
|
||
#if HAVE_UPLOADPROGRESS
|
||
|
||
#ifdef P_tmpdir
|
||
#define TMPDIR P_tmpdir
|
||
#else
|
||
#define TMPDIR "/tmp"
|
||
#endif
|
||
|
||
/* {{{ uploadprogress_functions[] */
|
||
zend_function_entry uploadprogress_functions[] = {
|
||
PHP_FE(uploadprogress_get_info, NULL)
|
||
PHP_FE(uploadprogress_get_contents, NULL)
|
||
{ NULL, NULL, NULL }
|
||
};
|
||
/* }}} */
|
||
|
||
PHP_INI_BEGIN()
|
||
PHP_INI_ENTRY("uploadprogress.file.filename_template", TMPDIR"/upt_%s.txt", PHP_INI_ALL, NULL)
|
||
PHP_INI_ENTRY("uploadprogress.file.contents_template", TMPDIR"/upload_contents_%s", PHP_INI_ALL, NULL)
|
||
PHP_INI_ENTRY("uploadprogress.get_contents", "0", PHP_INI_ALL, NULL)
|
||
PHP_INI_END()
|
||
|
||
/* {{{ uploadprogress_module_entry
|
||
*/
|
||
zend_module_entry uploadprogress_module_entry = {
|
||
STANDARD_MODULE_HEADER,
|
||
"uploadprogress",
|
||
uploadprogress_functions,
|
||
PHP_MINIT(uploadprogress),
|
||
PHP_MSHUTDOWN(uploadprogress),
|
||
PHP_RINIT(uploadprogress),
|
||
PHP_RSHUTDOWN(uploadprogress),
|
||
PHP_MINFO(uploadprogress),
|
||
PHP_UPLOADPROGRESS_VERSION,
|
||
STANDARD_MODULE_PROPERTIES
|
||
};
|
||
/* }}} */
|
||
|
||
#ifdef COMPILE_DL_UPLOADPROGRESS
|
||
ZEND_GET_MODULE(uploadprogress)
|
||
#endif
|
||
|
||
PHPAPI extern int (*php_rfc1867_callback)(unsigned int , void *, void ** TSRMLS_DC);
|
||
|
||
/* {{{ uploadprogress_php_rfc1867_file
|
||
*/
|
||
static int uploadprogress_php_rfc1867_file(unsigned int event, void *event_data, void **data TSRMLS_DC)
|
||
{
|
||
char *callable = NULL;
|
||
uploadprogress_data *progress;
|
||
int read_bytes;
|
||
zend_bool get_contents = INI_BOOL("uploadprogress.get_contents");
|
||
|
||
progress = *data;
|
||
|
||
if (event == MULTIPART_EVENT_START) {
|
||
multipart_event_start *e_data;
|
||
e_data = (multipart_event_start*) event_data;
|
||
progress = emalloc(sizeof(uploadprogress_data));
|
||
progress->upload_id = NULL;
|
||
progress->fieldname = NULL;
|
||
progress->data_filename = NULL;
|
||
progress->bytes_total = e_data->content_length;
|
||
progress->identifier = NULL;
|
||
progress->identifier_tmp = NULL;
|
||
progress->time_start = time(NULL);
|
||
*data = progress;
|
||
} else if (event == MULTIPART_EVENT_FORMDATA) {
|
||
multipart_event_formdata *e_data;
|
||
e_data = (multipart_event_formdata*) event_data;
|
||
read_bytes = e_data->post_bytes_processed;
|
||
|
||
if (e_data->newlength) {
|
||
*e_data->newlength = e_data->length;
|
||
}
|
||
|
||
if (strcmp(e_data->name, "UPLOAD_IDENTIFIER") == 0) {
|
||
char **upload_id;
|
||
char *template = INI_STR("uploadprogress.file.filename_template");
|
||
|
||
if (strcmp(template, "") == 0) {
|
||
return FAILURE;
|
||
}
|
||
|
||
upload_id = emalloc(strlen(*e_data->value) + 1);
|
||
strcpy(*upload_id, *e_data->value);
|
||
|
||
progress->upload_id = *upload_id;
|
||
progress->time_last = time(NULL);
|
||
progress->speed_average = 0;
|
||
progress->speed_last = 0;
|
||
progress->bytes_uploaded = read_bytes;
|
||
progress->files_uploaded = 0;
|
||
progress->est_sec = 0;
|
||
progress->identifier = uploadprogress_mk_filename(*upload_id, template);
|
||
progress->identifier_tmp = emalloc(strlen( progress->identifier) + 4);
|
||
sprintf(progress->identifier_tmp, "%s.wr", progress->identifier);
|
||
|
||
if (upload_id) {
|
||
efree(upload_id);
|
||
}
|
||
}
|
||
}
|
||
|
||
if (progress->identifier) {
|
||
time_t crtime = time(NULL);
|
||
int d, dt, ds;
|
||
|
||
if (event == MULTIPART_EVENT_FILE_START) {
|
||
char *data_identifier;
|
||
multipart_event_file_start *e_data;
|
||
|
||
e_data = (multipart_event_file_start*) event_data;
|
||
read_bytes = e_data->post_bytes_processed;
|
||
|
||
progress->fieldname = e_data->name;
|
||
progress->filename = *e_data->filename;
|
||
|
||
data_identifier = emalloc(strlen(progress->upload_id) + strlen(progress->fieldname) + 2);
|
||
sprintf(data_identifier, "%s-%s", progress->upload_id, progress->fieldname);
|
||
|
||
if (get_contents) {
|
||
char *data_template = INI_STR("uploadprogress.file.contents_template");
|
||
|
||
if (strcmp(data_template, "") == 0) {
|
||
return FAILURE;
|
||
}
|
||
|
||
progress->data_filename = uploadprogress_mk_filename(data_identifier, data_template);
|
||
}
|
||
|
||
if (data_identifier) {
|
||
efree(data_identifier);
|
||
}
|
||
} else if (event == MULTIPART_EVENT_FILE_DATA) {
|
||
multipart_event_file_data *e_data;
|
||
|
||
e_data = (multipart_event_file_data*) event_data;
|
||
read_bytes = e_data->post_bytes_processed;
|
||
|
||
if (get_contents) {
|
||
php_stream *stream;
|
||
#if defined(ZEND_ENGINE_3)
|
||
int options = 0;
|
||
#else
|
||
int options = ENFORCE_SAFE_MODE;
|
||
#endif
|
||
|
||
stream = php_stream_open_wrapper(progress->data_filename, "ab", options, NULL);
|
||
|
||
if (stream) {
|
||
php_stream_write(stream, e_data->data, e_data->length);
|
||
}
|
||
|
||
php_stream_close(stream);
|
||
}
|
||
} else if (event == MULTIPART_EVENT_FILE_END) {
|
||
multipart_event_file_end *e_data;
|
||
|
||
e_data = (multipart_event_file_end*) event_data;
|
||
|
||
read_bytes = e_data->post_bytes_processed;
|
||
progress->files_uploaded++;
|
||
|
||
if (get_contents) {
|
||
VCWD_UNLINK(progress->data_filename);
|
||
efree(progress->data_filename);
|
||
}
|
||
} else if (event == MULTIPART_EVENT_END) {
|
||
VCWD_UNLINK(progress->identifier);
|
||
efree(progress->identifier);
|
||
efree(progress->identifier_tmp);
|
||
efree(progress);
|
||
|
||
return SUCCESS;
|
||
}
|
||
|
||
/* Just in case we encounter a fracture in time. */
|
||
if (progress->time_last > crtime) {
|
||
progress->time_last = crtime;
|
||
}
|
||
|
||
dt = crtime - progress->time_last;
|
||
ds = crtime - progress->time_start;
|
||
d = read_bytes - progress->bytes_uploaded;
|
||
|
||
if (dt) {
|
||
progress->speed_last = d/dt;
|
||
progress->time_last = crtime;
|
||
progress->bytes_uploaded = read_bytes;
|
||
progress->speed_average = ds ? read_bytes / ds : 0;
|
||
progress->est_sec = progress->speed_average ? (progress->bytes_total - read_bytes) / progress->speed_average : -1;
|
||
}
|
||
|
||
if (dt || event >= MULTIPART_EVENT_FILE_END) {
|
||
FILE *F;
|
||
|
||
F = VCWD_FOPEN(progress->identifier_tmp, "wb");
|
||
|
||
if (F) {
|
||
fprintf(F,
|
||
"upload_id=%s\nfieldname=%s\nfilename=%s\ntime_start=%ld\n"
|
||
"time_last=%ld\nspeed_average=%u\nspeed_last=%u\n"
|
||
"bytes_uploaded=%lu\nbytes_total=%lu\n"
|
||
"files_uploaded=%u\nest_sec=%d\n",
|
||
progress->upload_id,
|
||
progress->fieldname,
|
||
progress->filename,
|
||
progress->time_start,
|
||
progress->time_last,
|
||
progress->speed_average,
|
||
progress->speed_last,
|
||
progress->bytes_uploaded,
|
||
progress->bytes_total,
|
||
progress->files_uploaded,
|
||
progress->est_sec);
|
||
|
||
fclose(F);
|
||
|
||
/* VCWD_RENAME on WIN32 and PHP < 5.3 has a bug, if target does exist */
|
||
#ifdef PHP_WIN32
|
||
#if PHP_MAJOR_VERSION == 5 && PHP_MINOR_VERSION < 3
|
||
VCWD_UNLINK(progress->identifier);
|
||
#endif
|
||
#endif
|
||
|
||
VCWD_RENAME(progress->identifier_tmp, progress->identifier);
|
||
}
|
||
}
|
||
}
|
||
|
||
if (event == MULTIPART_EVENT_END ) {
|
||
if (progress->identifier) {
|
||
efree(progress->identifier);
|
||
}
|
||
|
||
if (progress->identifier_tmp) {
|
||
efree(progress->identifier_tmp);
|
||
}
|
||
|
||
if (get_contents && progress->data_filename) {
|
||
efree(progress->data_filename);
|
||
}
|
||
|
||
efree(progress);
|
||
}
|
||
}
|
||
/* }}} */
|
||
|
||
/* {{{ PHP_MINIT_FUNCTION */
|
||
PHP_MINIT_FUNCTION(uploadprogress)
|
||
{
|
||
REGISTER_INI_ENTRIES();
|
||
php_rfc1867_callback = uploadprogress_php_rfc1867_file;
|
||
|
||
return SUCCESS;
|
||
}
|
||
/* }}} */
|
||
|
||
/* {{{ PHP_MSHUTDOWN_FUNCTION */
|
||
PHP_MSHUTDOWN_FUNCTION(uploadprogress)
|
||
{
|
||
UNREGISTER_INI_ENTRIES();
|
||
php_rfc1867_callback = NULL;
|
||
|
||
return SUCCESS;
|
||
}
|
||
/* }}} */
|
||
|
||
/* {{{ PHP_RINIT_FUNCTION */
|
||
PHP_RINIT_FUNCTION(uploadprogress)
|
||
{
|
||
return SUCCESS;
|
||
}
|
||
/* }}} */
|
||
|
||
/* {{{ PHP_RSHUTDOWN_FUNCTION */
|
||
PHP_RSHUTDOWN_FUNCTION(uploadprogress)
|
||
{
|
||
return SUCCESS;
|
||
}
|
||
/* }}} */
|
||
|
||
/* {{{ PHP_MINFO_FUNCTION */
|
||
PHP_MINFO_FUNCTION(uploadprogress)
|
||
{
|
||
php_info_print_table_start();
|
||
php_info_print_table_header(2, "uploadprogress support", "enabled");
|
||
php_info_print_table_row(2, "Version", PHP_UPLOADPROGRESS_VERSION);
|
||
php_info_print_table_end();
|
||
DISPLAY_INI_ENTRIES();
|
||
}
|
||
/* }}} */
|
||
|
||
/* {{{ proto array uploadprogress_get_info(string identifier)
|
||
*/
|
||
PHP_FUNCTION(uploadprogress_get_info)
|
||
{
|
||
char *id;
|
||
#if defined(ZEND_ENGINE_3)
|
||
size_t id_lg;
|
||
#else
|
||
int id_lg;
|
||
#endif
|
||
|
||
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s", &id, &id_lg) == FAILURE) {
|
||
return;
|
||
}
|
||
|
||
uploadprogress_file_php_get_info(id, return_value);
|
||
|
||
return;
|
||
}
|
||
/* }}} */
|
||
|
||
/* {{{ proto string uploadprogress_get_contents(string identifier, string fieldname[, int maxlen])
|
||
*/
|
||
PHP_FUNCTION(uploadprogress_get_contents)
|
||
{
|
||
char *id, *fieldname;
|
||
#if defined(ZEND_ENGINE_3)
|
||
size_t id_len, fieldname_len;
|
||
#else
|
||
int id_len, fieldname_len;
|
||
#endif
|
||
long maxlen = PHP_STREAM_COPY_ALL;
|
||
zend_bool get_contents = INI_BOOL("uploadprogress.get_contents");
|
||
|
||
if (!get_contents) {
|
||
php_error_docref(NULL TSRMLS_CC, E_WARNING,
|
||
"this function is disabled; set uploadprogress.get_contents = On to enable it");
|
||
RETURN_FALSE;
|
||
|
||
return;
|
||
}
|
||
|
||
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "ss|l",
|
||
&id, &id_len, &fieldname, &fieldname_len, &maxlen) == FAILURE) {
|
||
return;
|
||
}
|
||
|
||
if (ZEND_NUM_ARGS() == 3 && maxlen < 0) {
|
||
php_error_docref(NULL TSRMLS_CC, E_WARNING, "length must be greater than or equal to zero");
|
||
RETURN_FALSE;
|
||
}
|
||
|
||
uploadprogress_file_php_get_contents(id, fieldname, maxlen, return_value);
|
||
return;
|
||
}
|
||
/* }}} */
|
||
|
||
/* {{{ uploadprogress_mk_filename
|
||
*/
|
||
static char *uploadprogress_mk_filename(char *identifier, char *template)
|
||
{
|
||
char *x;
|
||
char *filename;
|
||
|
||
filename = emalloc(strlen(template) + strlen(identifier) + 3);
|
||
x = strstr(template, "%s");
|
||
|
||
if (x == NULL) {
|
||
sprintf(filename, "%s/%s", template, identifier);
|
||
} else {
|
||
strcpy(filename, template);
|
||
strcpy(filename + (x - template), identifier);
|
||
strcat(filename, x + 2);
|
||
}
|
||
|
||
return filename;
|
||
}
|
||
/* }}} */
|
||
|
||
/* {{{ uploadprogress_file_php_get_info
|
||
*/
|
||
static void uploadprogress_file_php_get_info(char *id, zval *return_value)
|
||
{
|
||
char s[1024];
|
||
char *filename;
|
||
char *template;
|
||
FILE *F;
|
||
TSRMLS_FETCH();
|
||
|
||
template = INI_STR("uploadprogress.file.filename_template");
|
||
|
||
if (strcmp(template, "") == 0) {
|
||
return;
|
||
} else {
|
||
filename = uploadprogress_mk_filename(id, template);
|
||
|
||
if (!filename) {
|
||
return;
|
||
}
|
||
|
||
F = VCWD_FOPEN(filename, "rb");
|
||
|
||
if (F) {
|
||
array_init(return_value);
|
||
|
||
while (fgets(s, 1000, F)) {
|
||
char *k, *v, *e;
|
||
int index = 0;
|
||
|
||
e = strchr(s, '=');
|
||
|
||
if (!e) {
|
||
continue;
|
||
}
|
||
|
||
/* Break the line into 2 parts. */
|
||
*e = 0;
|
||
v = e + 1;
|
||
k = s;
|
||
|
||
/* Trim spaces in front of the name/value. */
|
||
while (*k && *k <= 32) {
|
||
k++;
|
||
}
|
||
|
||
while (*v && *v <= 32) {
|
||
v++;
|
||
}
|
||
|
||
/* Trim spaces everywhere in the name. */
|
||
for (e = k; *e; e++) {
|
||
if (*e <= 32) {
|
||
*e = 0;
|
||
break;
|
||
}
|
||
}
|
||
|
||
/* Trim spaces only at the end of the value. */
|
||
if (v != NULL) {
|
||
for (index = strlen(v); index > 0; index--) {
|
||
if (v[index] > 32) {
|
||
break;
|
||
}
|
||
|
||
v[index] = 0;
|
||
}
|
||
}
|
||
|
||
#if defined(ZEND_ENGINE_3)
|
||
add_assoc_string(return_value, k, v);
|
||
#else
|
||
add_assoc_string(return_value, k, v, 1);
|
||
#endif
|
||
}
|
||
|
||
fclose(F);
|
||
}
|
||
|
||
if (filename) {
|
||
efree(filename);
|
||
}
|
||
|
||
return;
|
||
}
|
||
}
|
||
/* }}} */
|
||
|
||
/* {{{ uploadprogress_file_php_get_contents
|
||
*/
|
||
static void uploadprogress_file_php_get_contents(char *id, char *fieldname, long maxlen, zval *return_value)
|
||
{
|
||
#if defined(ZEND_ENGINE_3)
|
||
char *filename, *template, *data_identifier;
|
||
zend_string *contents;
|
||
int options = 0;
|
||
size_t len;
|
||
#else
|
||
char *filename, *template, *contents, *data_identifier;
|
||
int options = ENFORCE_SAFE_MODE;
|
||
int len;
|
||
#endif
|
||
php_stream *stream;
|
||
#if PHP_API_VERSION < 20100412
|
||
int newlen;
|
||
#endif
|
||
TSRMLS_FETCH();
|
||
|
||
template = INI_STR("uploadprogress.file.contents_template");
|
||
|
||
if (strcmp(template, "") == 0) {
|
||
return;
|
||
} else {
|
||
data_identifier = emalloc(strlen(id) + strlen(fieldname) + 2);
|
||
sprintf(data_identifier, "%s-%s", id, fieldname);
|
||
|
||
filename = uploadprogress_mk_filename(data_identifier, template);
|
||
|
||
if (!filename) {
|
||
if (data_identifier) {
|
||
efree(data_identifier);
|
||
}
|
||
|
||
return;
|
||
}
|
||
|
||
stream = php_stream_open_wrapper(filename, "rb", options, NULL);
|
||
|
||
if (!stream) {
|
||
if (data_identifier) {
|
||
efree(data_identifier);
|
||
}
|
||
|
||
if (filename) {
|
||
efree(filename);
|
||
}
|
||
|
||
RETURN_FALSE;
|
||
}
|
||
|
||
/* Uses mmap if possible. */
|
||
#if defined(ZEND_ENGINE_3)
|
||
contents = php_stream_copy_to_mem(stream, maxlen, 0);
|
||
|
||
if (contents) {
|
||
len = contents->len;
|
||
}
|
||
|
||
if (contents && len > 0) {
|
||
#else
|
||
if ((len = php_stream_copy_to_mem(stream, &contents, maxlen, 0)) > 0) {
|
||
#endif
|
||
|
||
#if PHP_API_VERSION < 20100412
|
||
if (PG(magic_quotes_runtime)) {
|
||
contents = php_addslashes(contents, len, &newlen, 1 TSRMLS_CC);
|
||
len = newlen;
|
||
}
|
||
#endif
|
||
#if defined(ZEND_ENGINE_3)
|
||
RETVAL_STR(contents);
|
||
#else
|
||
RETVAL_STRINGL(contents, len, 0);
|
||
#endif
|
||
} else {
|
||
RETVAL_EMPTY_STRING();
|
||
}
|
||
|
||
php_stream_close(stream);
|
||
|
||
if (data_identifier) {
|
||
efree(data_identifier);
|
||
}
|
||
|
||
if (filename) {
|
||
efree(filename);
|
||
}
|
||
|
||
return;
|
||
}
|
||
}
|
||
/* }}} */
|
||
|
||
#endif /* HAVE_UPLOADPROGRESS */
|