forked from krakjoe/apcu
-
Notifications
You must be signed in to change notification settings - Fork 0
/
apc_rfc1867.c
235 lines (207 loc) · 11 KB
/
apc_rfc1867.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
/*
+----------------------------------------------------------------------+
| APC |
+----------------------------------------------------------------------+
| Copyright (c) 2006-2011 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 |
| [email protected] so we can mail you a copy immediately. |
+----------------------------------------------------------------------+
| Authors: Rasmus Lerdorf <[email protected]> |
+----------------------------------------------------------------------+
This software was contributed to PHP by Community Connect Inc. in 2002
and revised in 2005 by Yahoo! Inc. to add support for PHP 5.1.
Future revisions and derivatives of this source code must acknowledge
Community Connect Inc. as the original contributor of this module by
leaving this note intact in the source code.
All other licensing and usage conditions are those of the PHP Group.
*/
/* $Id: apc_rfc1867.c 309694 2011-03-25 18:47:38Z rasmus $*/
#include "apc.h"
#include "apc_globals.h"
#include "rfc1867.h"
#ifdef PHP_WIN32
#include "win32/time.h"
#endif
extern apc_cache_t* apc_user_cache;
#ifdef MULTIPART_EVENT_FORMDATA
static zend_bool update_bytes_processed(apc_cache_t* cache, apc_cache_entry_t* entry, void* data) {
int *bytes_ptr = (int*)data;
zval* val = entry->val;
if(Z_TYPE_P(val) == IS_ARRAY) {
HashTable *ht = val->value.ht;
Bucket* curr = NULL;
for (curr = ht->pListHead; curr != NULL; curr = curr->pListNext) {
if(curr->nKeyLength == 8 &&
(!memcmp(curr->arKey, "current", curr->nKeyLength))) {
zval* current = ((zval**)curr->pData)[0];
current->value.lval = *bytes_ptr;
return 1;
}
}
}
return 0;
}
static double my_time() {
struct timeval a;
double t;
gettimeofday(&a, NULL);
t = a.tv_sec + (a.tv_usec/1000000.00);
return t;
}
#define RFC1867_DATA(name) \
((request_data)->name)
int apc_rfc1867_progress(uint event, void *event_data, void **extra TSRMLS_DC) {
apc_rfc1867_data *request_data = &APCG(rfc1867_data);
zval *track = NULL;
switch (event) {
case MULTIPART_EVENT_START:
{
multipart_event_start *data = (multipart_event_start *) event_data;
RFC1867_DATA(content_length) = data->content_length;
RFC1867_DATA(tracking_key)[0] = '\0';
RFC1867_DATA(name)[0] = '\0';
RFC1867_DATA(cancel_upload) = 0;
RFC1867_DATA(temp_filename) = NULL;
RFC1867_DATA(filename)[0] = '\0';
RFC1867_DATA(key_length) = 0;
RFC1867_DATA(start_time) = my_time();
RFC1867_DATA(bytes_processed) = 0;
RFC1867_DATA(prev_bytes_processed) = 0;
RFC1867_DATA(rate) = 0;
RFC1867_DATA(update_freq) = (int) APCG(rfc1867_freq);
RFC1867_DATA(started) = 0;
if(RFC1867_DATA(update_freq) < 0) { // frequency is a percentage, not bytes
RFC1867_DATA(update_freq) = (int) (RFC1867_DATA(content_length) * APCG(rfc1867_freq) / 100);
}
}
break;
case MULTIPART_EVENT_FORMDATA:
{
int prefix_len = strlen(APCG(rfc1867_prefix));
multipart_event_formdata *data = (multipart_event_formdata *) event_data;
if(data->name && !strncasecmp(data->name, APCG(rfc1867_name), strlen(APCG(rfc1867_name)))
&& data->value && data->length) {
if(data->length >= sizeof(RFC1867_DATA(tracking_key)) - prefix_len) {
apc_warning("Key too long for '%s'. Maximum size is '%d' characters." TSRMLS_CC,
APCG(rfc1867_name),
sizeof(RFC1867_DATA(tracking_key)) - prefix_len);
break;
}
if(RFC1867_DATA(started)) {
apc_warning("Upload progress key '%s' should be before the file upload entry in the form." TSRMLS_CC,
APCG(rfc1867_name));
break;
}
strlcat(RFC1867_DATA(tracking_key), APCG(rfc1867_prefix), 63);
strlcat(RFC1867_DATA(tracking_key), *data->value, 63);
RFC1867_DATA(key_length) = data->length + prefix_len;
RFC1867_DATA(bytes_processed) = data->post_bytes_processed;
}
}
break;
case MULTIPART_EVENT_FILE_START:
{
RFC1867_DATA(started) = 1;
if(*RFC1867_DATA(tracking_key)) {
multipart_event_file_start *data = (multipart_event_file_start *) event_data;
RFC1867_DATA(bytes_processed) = data->post_bytes_processed;
strlcpy(RFC1867_DATA(filename),*data->filename,128);
RFC1867_DATA(temp_filename) = NULL;
strlcpy(RFC1867_DATA(name),data->name,64);
ALLOC_INIT_ZVAL(track);
array_init(track);
add_assoc_long(track, "total", RFC1867_DATA(content_length));
add_assoc_long(track, "current", RFC1867_DATA(bytes_processed));
add_assoc_string(track, "filename", RFC1867_DATA(filename), 1);
add_assoc_string(track, "name", RFC1867_DATA(name), 1);
add_assoc_long(track, "done", 0);
add_assoc_double(track, "start_time", RFC1867_DATA(start_time));
apc_cache_store(apc_user_cache, RFC1867_DATA(tracking_key), RFC1867_DATA(key_length)+1, track, APCG(rfc1867_ttl), 0 TSRMLS_CC);
zval_ptr_dtor(&track);
}
}
break;
case MULTIPART_EVENT_FILE_DATA:
if(*RFC1867_DATA(tracking_key)) {
multipart_event_file_data *data = (multipart_event_file_data *) event_data;
RFC1867_DATA(bytes_processed) = data->post_bytes_processed;
if(RFC1867_DATA(bytes_processed) - RFC1867_DATA(prev_bytes_processed) > (uint) RFC1867_DATA(update_freq)) {
if(!apc_cache_update(apc_user_cache, RFC1867_DATA(tracking_key), RFC1867_DATA(key_length), update_bytes_processed, &RFC1867_DATA(bytes_processed) TSRMLS_CC)) {
ALLOC_INIT_ZVAL(track);
array_init(track);
add_assoc_long(track, "total", RFC1867_DATA(content_length));
add_assoc_long(track, "current", RFC1867_DATA(bytes_processed));
add_assoc_string(track, "filename", RFC1867_DATA(filename), 1);
add_assoc_string(track, "name", RFC1867_DATA(name), 1);
add_assoc_long(track, "done", 0);
add_assoc_double(track, "start_time", RFC1867_DATA(start_time));
apc_cache_store(apc_user_cache, RFC1867_DATA(tracking_key), RFC1867_DATA(key_length)+1, track, APCG(rfc1867_ttl), 0 TSRMLS_CC);
zval_ptr_dtor(&track);
}
RFC1867_DATA(prev_bytes_processed) = RFC1867_DATA(bytes_processed);
}
}
break;
case MULTIPART_EVENT_FILE_END:
if(*RFC1867_DATA(tracking_key)) {
multipart_event_file_end *data = (multipart_event_file_end *) event_data;
RFC1867_DATA(bytes_processed) = data->post_bytes_processed;
RFC1867_DATA(cancel_upload) = data->cancel_upload;
if(data->temp_filename) {
RFC1867_DATA(temp_filename) = data->temp_filename;
} else {
RFC1867_DATA(temp_filename) = "";
}
ALLOC_INIT_ZVAL(track);
array_init(track);
add_assoc_long(track, "total", RFC1867_DATA(content_length));
add_assoc_long(track, "current", RFC1867_DATA(bytes_processed));
add_assoc_string(track, "filename", RFC1867_DATA(filename), 1);
add_assoc_string(track, "name", RFC1867_DATA(name), 1);
add_assoc_string(track, "temp_filename", RFC1867_DATA(temp_filename), 1);
add_assoc_long(track, "cancel_upload", RFC1867_DATA(cancel_upload));
add_assoc_long(track, "done", 0);
add_assoc_double(track, "start_time", RFC1867_DATA(start_time));
apc_cache_store(apc_user_cache, RFC1867_DATA(tracking_key), RFC1867_DATA(key_length)+1, track, APCG(rfc1867_ttl), 0 TSRMLS_CC);
zval_ptr_dtor(&track);
}
break;
case MULTIPART_EVENT_END:
if(*RFC1867_DATA(tracking_key)) {
double now = my_time();
multipart_event_end *data = (multipart_event_end *) event_data;
RFC1867_DATA(bytes_processed) = data->post_bytes_processed;
if(now>RFC1867_DATA(start_time)) RFC1867_DATA(rate) = 8.0*RFC1867_DATA(bytes_processed)/(now-RFC1867_DATA(start_time));
else RFC1867_DATA(rate) = 8.0*RFC1867_DATA(bytes_processed); /* Too quick */
ALLOC_INIT_ZVAL(track);
array_init(track);
add_assoc_long(track, "total", RFC1867_DATA(content_length));
add_assoc_long(track, "current", RFC1867_DATA(bytes_processed));
add_assoc_double(track, "rate", RFC1867_DATA(rate));
add_assoc_string(track, "filename", RFC1867_DATA(filename), 1);
add_assoc_string(track, "name", RFC1867_DATA(name), 1);
add_assoc_long(track, "cancel_upload", RFC1867_DATA(cancel_upload));
add_assoc_long(track, "done", 1);
add_assoc_double(track, "start_time", RFC1867_DATA(start_time));
apc_cache_store(apc_user_cache, RFC1867_DATA(tracking_key), RFC1867_DATA(key_length)+1, track, APCG(rfc1867_ttl), 0 TSRMLS_CC);
zval_ptr_dtor(&track);
}
break;
}
return SUCCESS;
}
#endif
/*
* Local variables:
* tab-width: 4
* c-basic-offset: 4
* End:
* vim>600: expandtab sw=4 ts=4 sts=4 fdm=marker
* vim<600: expandtab sw=4 ts=4 sts=4
*/