0
|
1
|
|
2 /*
|
|
3 * Copyright (C) Igor Sysoev
|
|
4 */
|
|
5
|
|
6
|
|
7 #include <ngx_config.h>
|
|
8 #include <ngx_core.h>
|
|
9 #include <ngx_http.h>
|
503
|
10 #include <ngx_md5.h>
|
0
|
11
|
|
12
|
558
|
13 static ngx_int_t ngx_http_file_cache_read(ngx_http_request_t *r,
|
|
14 ngx_http_cache_t *c);
|
585
|
15 static ssize_t ngx_http_file_cache_aio_read(ngx_http_request_t *r,
|
|
16 ngx_http_cache_t *c);
|
558
|
17 #if (NGX_HAVE_FILE_AIO)
|
|
18 static void ngx_http_cache_aio_event_handler(ngx_event_t *ev);
|
|
19 #endif
|
507
|
20 static ngx_int_t ngx_http_file_cache_exists(ngx_http_file_cache_t *cache,
|
|
21 ngx_http_cache_t *c);
|
503
|
22 static ngx_http_file_cache_node_t *
|
|
23 ngx_http_file_cache_lookup(ngx_http_file_cache_t *cache, u_char *key);
|
|
24 static void ngx_http_file_cache_rbtree_insert_value(ngx_rbtree_node_t *temp,
|
|
25 ngx_rbtree_node_t *node, ngx_rbtree_node_t *sentinel);
|
|
26 static void ngx_http_file_cache_cleanup(void *data);
|
507
|
27 static time_t ngx_http_file_cache_forced_expire(ngx_http_file_cache_t *cache);
|
|
28 static time_t ngx_http_file_cache_expire(ngx_http_file_cache_t *cache);
|
|
29 static void ngx_http_file_cache_delete(ngx_http_file_cache_t *cache,
|
|
30 ngx_queue_t *q, u_char *name);
|
|
31 static ngx_int_t
|
|
32 ngx_http_file_cache_manager_sleep(ngx_http_file_cache_t *cache);
|
|
33 static ngx_int_t ngx_http_file_cache_noop(ngx_tree_ctx_t *ctx,
|
503
|
34 ngx_str_t *path);
|
507
|
35 static ngx_int_t ngx_http_file_cache_manage_file(ngx_tree_ctx_t *ctx,
|
|
36 ngx_str_t *path);
|
|
37 static ngx_int_t ngx_http_file_cache_add_file(ngx_tree_ctx_t *ctx,
|
|
38 ngx_str_t *path);
|
|
39 static ngx_int_t ngx_http_file_cache_add(ngx_http_file_cache_t *cache,
|
|
40 ngx_http_cache_t *c);
|
|
41 static ngx_int_t ngx_http_file_cache_delete_file(ngx_tree_ctx_t *ctx,
|
503
|
42 ngx_str_t *path);
|
0
|
43
|
|
44
|
542
|
45 ngx_str_t ngx_http_cache_status[] = {
|
|
46 ngx_string("MISS"),
|
|
47 ngx_string("EXPIRED"),
|
|
48 ngx_string("STALE"),
|
|
49 ngx_string("UPDATING"),
|
|
50 ngx_string("HIT")
|
|
51 };
|
|
52
|
|
53
|
503
|
54 static u_char ngx_http_file_cache_key[] = { LF, 'K', 'E', 'Y', ':', ' ' };
|
0
|
55
|
|
56
|
503
|
57 static ngx_int_t
|
|
58 ngx_http_file_cache_init(ngx_shm_zone_t *shm_zone, void *data)
|
0
|
59 {
|
503
|
60 ngx_http_file_cache_t *ocache = data;
|
0
|
61
|
507
|
62 size_t len;
|
554
|
63 ngx_uint_t n;
|
503
|
64 ngx_http_file_cache_t *cache;
|
0
|
65
|
503
|
66 cache = shm_zone->data;
|
0
|
67
|
503
|
68 if (ocache) {
|
|
69 if (ngx_strcmp(cache->path->name.data, ocache->path->name.data) != 0) {
|
|
70 ngx_log_error(NGX_LOG_EMERG, shm_zone->shm.log, 0,
|
|
71 "cache \"%V\" uses the \"%V\" cache path "
|
|
72 "while previously it used the \"%V\" cache path",
|
520
|
73 &shm_zone->shm.name, &cache->path->name,
|
503
|
74 &ocache->path->name);
|
0
|
75
|
|
76 return NGX_ERROR;
|
|
77 }
|
|
78
|
554
|
79 for (n = 0; n < 3; n++) {
|
|
80 if (cache->path->level[n] != ocache->path->level[n]) {
|
|
81 ngx_log_error(NGX_LOG_EMERG, shm_zone->shm.log, 0,
|
|
82 "cache \"%V\" had previously different levels",
|
|
83 &shm_zone->shm.name);
|
|
84 return NGX_ERROR;
|
|
85 }
|
|
86 }
|
|
87
|
520
|
88 cache->sh = ocache->sh;
|
|
89
|
503
|
90 cache->shpool = ocache->shpool;
|
507
|
91 cache->bsize = ocache->bsize;
|
|
92
|
|
93 cache->max_size /= cache->bsize;
|
503
|
94
|
554
|
95 if (!cache->sh->cold || cache->sh->loading) {
|
|
96 cache->path->loader = NULL;
|
|
97 }
|
|
98
|
503
|
99 return NGX_OK;
|
|
100 }
|
|
101
|
|
102 cache->shpool = (ngx_slab_pool_t *) shm_zone->shm.addr;
|
|
103
|
520
|
104 if (shm_zone->shm.exists) {
|
|
105 cache->sh = cache->shpool->data;
|
|
106 cache->bsize = ngx_fs_bsize(cache->path->name.data);
|
|
107
|
|
108 return NGX_OK;
|
503
|
109 }
|
|
110
|
520
|
111 cache->sh = ngx_slab_alloc(cache->shpool, sizeof(ngx_http_file_cache_sh_t));
|
|
112 if (cache->sh == NULL) {
|
503
|
113 return NGX_ERROR;
|
|
114 }
|
|
115
|
520
|
116 cache->shpool->data = cache->sh;
|
|
117
|
|
118 ngx_rbtree_init(&cache->sh->rbtree, &cache->sh->sentinel,
|
503
|
119 ngx_http_file_cache_rbtree_insert_value);
|
|
120
|
520
|
121 ngx_queue_init(&cache->sh->queue);
|
503
|
122
|
520
|
123 cache->sh->cold = 1;
|
554
|
124 cache->sh->loading = 0;
|
520
|
125 cache->sh->size = 0;
|
507
|
126
|
|
127 cache->bsize = ngx_fs_bsize(cache->path->name.data);
|
|
128
|
|
129 cache->max_size /= cache->bsize;
|
|
130
|
520
|
131 len = sizeof(" in cache keys zone \"\"") + shm_zone->shm.name.len;
|
507
|
132
|
|
133 cache->shpool->log_ctx = ngx_slab_alloc(cache->shpool, len);
|
|
134 if (cache->shpool->log_ctx == NULL) {
|
|
135 return NGX_ERROR;
|
|
136 }
|
|
137
|
|
138 ngx_sprintf(cache->shpool->log_ctx, " in cache keys zone \"%V\"%Z",
|
520
|
139 &shm_zone->shm.name);
|
503
|
140
|
|
141 return NGX_OK;
|
|
142 }
|
|
143
|
|
144
|
|
145 void
|
|
146 ngx_http_file_cache_create_key(ngx_http_request_t *r)
|
|
147 {
|
|
148 size_t len;
|
|
149 ngx_str_t *key;
|
|
150 ngx_uint_t i;
|
|
151 ngx_md5_t md5;
|
|
152 ngx_http_cache_t *c;
|
|
153
|
|
154 c = r->cache;
|
|
155
|
|
156 len = 0;
|
|
157
|
|
158 ngx_crc32_init(c->crc32);
|
|
159 ngx_md5_init(&md5);
|
|
160
|
|
161 key = c->keys.elts;
|
|
162 for (i = 0; i < c->keys.nelts; i++) {
|
|
163 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
164 "http cache key: \"%V\"", &key[i]);
|
|
165
|
|
166 len += key[i].len;
|
|
167
|
|
168 ngx_crc32_update(&c->crc32, key[i].data, key[i].len);
|
|
169 ngx_md5_update(&md5, key[i].data, key[i].len);
|
|
170 }
|
|
171
|
|
172 c->header_start = sizeof(ngx_http_file_cache_header_t)
|
|
173 + sizeof(ngx_http_file_cache_key) + len + 1;
|
|
174
|
|
175 ngx_crc32_final(c->crc32);
|
|
176 ngx_md5_final(c->key, &md5);
|
|
177 }
|
|
178
|
|
179
|
|
180 ngx_int_t
|
|
181 ngx_http_file_cache_open(ngx_http_request_t *r)
|
|
182 {
|
558
|
183 u_char *p;
|
|
184 ngx_int_t rc, rv;
|
|
185 ngx_uint_t cold, test;
|
|
186 ngx_path_t *path;
|
|
187 ngx_http_cache_t *c;
|
|
188 ngx_pool_cleanup_t *cln;
|
|
189 ngx_open_file_info_t of;
|
|
190 ngx_http_file_cache_t *cache;
|
|
191 ngx_http_core_loc_conf_t *clcf;
|
503
|
192
|
|
193 c = r->cache;
|
558
|
194
|
|
195 if (c->buf) {
|
|
196 return ngx_http_file_cache_read(r, c);
|
|
197 }
|
|
198
|
503
|
199 cache = c->file_cache;
|
|
200
|
|
201 cln = ngx_pool_cleanup_add(r->pool, 0);
|
|
202 if (cln == NULL) {
|
|
203 return NGX_ERROR;
|
|
204 }
|
|
205
|
507
|
206 rc = ngx_http_file_cache_exists(cache, c);
|
503
|
207
|
538
|
208 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
209 "http file cache exists: %i e:%d", rc, c->exists);
|
503
|
210
|
|
211 if (rc == NGX_ERROR) {
|
|
212 return rc;
|
|
213 }
|
|
214
|
|
215 cln->handler = ngx_http_file_cache_cleanup;
|
|
216 cln->data = c;
|
|
217
|
|
218 if (rc == NGX_AGAIN) {
|
558
|
219 return NGX_HTTP_CACHE_SCARCE;
|
503
|
220 }
|
|
221
|
520
|
222 cold = cache->sh->cold;
|
503
|
223
|
|
224 if (rc == NGX_OK) {
|
|
225
|
|
226 if (c->error) {
|
|
227 return c->error;
|
|
228 }
|
|
229
|
|
230 c->temp_file = 1;
|
|
231 test = c->exists ? 1 : 0;
|
|
232 rv = NGX_DECLINED;
|
|
233
|
|
234 } else { /* rc == NGX_DECLINED */
|
|
235
|
|
236 if (c->min_uses > 1) {
|
|
237
|
|
238 if (!cold) {
|
558
|
239 return NGX_HTTP_CACHE_SCARCE;
|
0
|
240 }
|
|
241
|
503
|
242 test = 1;
|
558
|
243 rv = NGX_HTTP_CACHE_SCARCE;
|
503
|
244
|
|
245 } else {
|
|
246 c->temp_file = 1;
|
|
247 test = cold ? 1 : 0;
|
|
248 rv = NGX_DECLINED;
|
0
|
249 }
|
|
250 }
|
|
251
|
503
|
252 path = cache->path;
|
|
253
|
|
254 c->file.name.len = path->name.len + 1 + path->len
|
|
255 + 2 * NGX_HTTP_CACHE_KEY_LEN;
|
|
256
|
|
257 c->file.name.data = ngx_pnalloc(r->pool, c->file.name.len + 1);
|
|
258 if (c->file.name.data == NULL) {
|
|
259 return NGX_ERROR;
|
|
260 }
|
|
261
|
|
262 ngx_memcpy(c->file.name.data, path->name.data, path->name.len);
|
|
263
|
|
264 p = c->file.name.data + path->name.len + 1 + path->len;
|
|
265 p = ngx_hex_dump(p, c->key, NGX_HTTP_CACHE_KEY_LEN);
|
|
266 *p = '\0';
|
|
267
|
|
268 ngx_create_hashed_filename(path, c->file.name.data, c->file.name.len);
|
|
269
|
|
270 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, ngx_cycle->log, 0,
|
|
271 "cache file: \"%s\"", c->file.name.data);
|
|
272
|
|
273 if (!test) {
|
|
274 return NGX_DECLINED;
|
|
275 }
|
|
276
|
|
277 clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module);
|
|
278
|
|
279 ngx_memzero(&of, sizeof(ngx_open_file_info_t));
|
0
|
280
|
503
|
281 of.uniq = c->uniq;
|
|
282 of.valid = clcf->open_file_cache_valid;
|
|
283 of.min_uses = clcf->open_file_cache_min_uses;
|
|
284 of.events = clcf->open_file_cache_events;
|
|
285 of.directio = NGX_OPEN_FILE_DIRECTIO_OFF;
|
575
|
286 of.read_ahead = clcf->read_ahead;
|
503
|
287
|
|
288 if (ngx_open_cached_file(clcf->open_file_cache, &c->file.name, &of, r->pool)
|
|
289 != NGX_OK)
|
|
290 {
|
|
291 switch (of.err) {
|
|
292
|
|
293 case 0:
|
|
294 return NGX_ERROR;
|
|
295
|
|
296 case NGX_ENOENT:
|
|
297 case NGX_ENOTDIR:
|
|
298 return rv;
|
|
299
|
|
300 default:
|
|
301 ngx_log_error(NGX_LOG_CRIT, r->connection->log, of.err,
|
|
302 ngx_open_file_n " \"%s\" failed", c->file.name.data);
|
|
303 return NGX_ERROR;
|
|
304 }
|
|
305 }
|
|
306
|
|
307 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
308 "http file cache fd: %d", of.fd);
|
|
309
|
|
310 c->file.fd = of.fd;
|
505
|
311 c->file.log = r->connection->log;
|
558
|
312 c->uniq = of.uniq;
|
|
313 c->length = of.size;
|
503
|
314
|
|
315 c->buf = ngx_create_temp_buf(r->pool, c->body_start);
|
|
316 if (c->buf == NULL) {
|
|
317 return NGX_ERROR;
|
|
318 }
|
|
319
|
558
|
320 return ngx_http_file_cache_read(r, c);
|
|
321 }
|
|
322
|
|
323
|
|
324 static ngx_int_t
|
|
325 ngx_http_file_cache_read(ngx_http_request_t *r, ngx_http_cache_t *c)
|
|
326 {
|
|
327 time_t now;
|
|
328 ssize_t n;
|
|
329 ngx_int_t rc;
|
|
330 ngx_http_file_cache_t *cache;
|
|
331 ngx_http_file_cache_header_t *h;
|
|
332
|
|
333 c = r->cache;
|
|
334
|
585
|
335 n = ngx_http_file_cache_aio_read(r, c);
|
558
|
336
|
585
|
337 if (n < 0) {
|
0
|
338 return n;
|
|
339 }
|
|
340
|
597
|
341 if ((size_t) n < c->header_start) {
|
503
|
342 ngx_log_error(NGX_LOG_CRIT, r->connection->log, 0,
|
26
|
343 "cache file \"%s\" is too small", c->file.name.data);
|
0
|
344 return NGX_ERROR;
|
|
345 }
|
|
346
|
503
|
347 h = (ngx_http_file_cache_header_t *) c->buf->pos;
|
0
|
348
|
607
|
349 if (h->crc32 != c->crc32) {
|
503
|
350 ngx_log_error(NGX_LOG_CRIT, r->connection->log, 0,
|
|
351 "cache file \"%s\" has md5 collision", c->file.name.data);
|
0
|
352 return NGX_DECLINED;
|
|
353 }
|
|
354
|
503
|
355 c->buf->last += n;
|
26
|
356
|
503
|
357 c->valid_sec = h->valid_sec;
|
|
358 c->last_modified = h->last_modified;
|
|
359 c->date = h->date;
|
|
360 c->valid_msec = h->valid_msec;
|
607
|
361 c->header_start = h->header_start;
|
503
|
362 c->body_start = h->body_start;
|
26
|
363
|
503
|
364 r->cached = 1;
|
|
365
|
558
|
366 cache = c->file_cache;
|
|
367
|
|
368 if (cache->sh->cold) {
|
503
|
369
|
|
370 ngx_shmtx_lock(&cache->shpool->mutex);
|
26
|
371
|
507
|
372 if (!c->node->exists) {
|
|
373 c->node->uses = 1;
|
|
374 c->node->body_start = c->body_start;
|
|
375 c->node->exists = 1;
|
558
|
376 c->node->uniq = c->uniq;
|
507
|
377
|
520
|
378 cache->sh->size += (c->length + cache->bsize - 1) / cache->bsize;
|
507
|
379 }
|
503
|
380
|
|
381 ngx_shmtx_unlock(&cache->shpool->mutex);
|
0
|
382 }
|
|
383
|
507
|
384 now = ngx_time();
|
|
385
|
503
|
386 if (c->valid_sec < now) {
|
0
|
387
|
538
|
388 ngx_shmtx_lock(&cache->shpool->mutex);
|
0
|
389
|
538
|
390 if (c->node->updating) {
|
|
391 rc = NGX_HTTP_CACHE_UPDATING;
|
503
|
392
|
538
|
393 } else {
|
|
394 c->node->updating = 1;
|
|
395 rc = NGX_HTTP_CACHE_STALE;
|
|
396 }
|
|
397
|
|
398 ngx_shmtx_unlock(&cache->shpool->mutex);
|
|
399
|
|
400 ngx_log_debug3(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
401 "http file cache expired: %i %T %T",
|
|
402 rc, c->valid_sec, now);
|
|
403
|
|
404 return rc;
|
0
|
405 }
|
|
406
|
503
|
407 return NGX_OK;
|
|
408 }
|
|
409
|
|
410
|
585
|
411 static ssize_t
|
|
412 ngx_http_file_cache_aio_read(ngx_http_request_t *r, ngx_http_cache_t *c)
|
|
413 {
|
558
|
414 #if (NGX_HAVE_FILE_AIO)
|
585
|
415 ssize_t n;
|
|
416 ngx_http_core_loc_conf_t *clcf;
|
558
|
417
|
585
|
418 if (!ngx_file_aio) {
|
|
419 goto noaio;
|
|
420 }
|
|
421
|
|
422 clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module);
|
|
423
|
|
424 if (!clcf->aio) {
|
|
425 goto noaio;
|
|
426 }
|
|
427
|
|
428 n = ngx_file_aio_read(&c->file, c->buf->pos, c->body_start, 0, r->pool);
|
|
429
|
|
430 if (n != NGX_AGAIN) {
|
|
431 return n;
|
|
432 }
|
|
433
|
|
434 c->file.aio->data = r;
|
|
435 c->file.aio->handler = ngx_http_cache_aio_event_handler;
|
|
436
|
|
437 r->main->blocked++;
|
|
438 r->aio = 1;
|
|
439
|
|
440 return NGX_AGAIN;
|
|
441
|
|
442 noaio:
|
|
443
|
|
444 #endif
|
|
445
|
|
446 return ngx_read_file(&c->file, c->buf->pos, c->body_start, 0);
|
|
447 }
|
|
448
|
|
449
|
|
450 #if (NGX_HAVE_FILE_AIO)
|
558
|
451
|
|
452 static void
|
|
453 ngx_http_cache_aio_event_handler(ngx_event_t *ev)
|
|
454 {
|
|
455 ngx_event_aio_t *aio;
|
|
456 ngx_http_request_t *r;
|
|
457
|
|
458 aio = ev->data;
|
|
459 r = aio->data;
|
|
460
|
|
461 r->main->blocked--;
|
|
462 r->aio = 0;
|
|
463
|
|
464 r->connection->write->handler(r->connection->write);
|
|
465 }
|
|
466
|
|
467 #endif
|
|
468
|
|
469
|
503
|
470 static ngx_int_t
|
507
|
471 ngx_http_file_cache_exists(ngx_http_file_cache_t *cache, ngx_http_cache_t *c)
|
503
|
472 {
|
|
473 ngx_int_t rc;
|
|
474 ngx_http_file_cache_node_t *fcn;
|
|
475
|
|
476 ngx_shmtx_lock(&cache->shpool->mutex);
|
|
477
|
507
|
478 fcn = ngx_http_file_cache_lookup(cache, c->key);
|
503
|
479
|
|
480 if (fcn) {
|
|
481 ngx_queue_remove(&fcn->queue);
|
|
482
|
|
483 if (fcn->error) {
|
|
484
|
|
485 if (fcn->valid_sec < ngx_time()) {
|
|
486 goto renew;
|
|
487 }
|
|
488
|
|
489 rc = NGX_OK;
|
|
490
|
|
491 goto done;
|
|
492 }
|
|
493
|
|
494 fcn->uses++;
|
|
495 fcn->count++;
|
|
496
|
|
497 if (fcn->exists) {
|
|
498
|
507
|
499 c->exists = fcn->exists;
|
|
500 c->body_start = fcn->body_start;
|
503
|
501
|
|
502 rc = NGX_OK;
|
|
503
|
|
504 goto done;
|
|
505 }
|
|
506
|
507
|
507 if (fcn->uses >= c->min_uses) {
|
503
|
508
|
507
|
509 c->exists = fcn->exists;
|
|
510 c->body_start = fcn->body_start;
|
503
|
511
|
|
512 rc = NGX_OK;
|
|
513
|
|
514 } else {
|
|
515 rc = NGX_AGAIN;
|
|
516 }
|
|
517
|
|
518 goto done;
|
|
519 }
|
|
520
|
|
521 fcn = ngx_slab_alloc_locked(cache->shpool,
|
|
522 sizeof(ngx_http_file_cache_node_t));
|
|
523 if (fcn == NULL) {
|
|
524 ngx_shmtx_unlock(&cache->shpool->mutex);
|
|
525
|
518
|
526 (void) ngx_http_file_cache_forced_expire(cache);
|
503
|
527
|
|
528 ngx_shmtx_lock(&cache->shpool->mutex);
|
|
529
|
|
530 fcn = ngx_slab_alloc_locked(cache->shpool,
|
|
531 sizeof(ngx_http_file_cache_node_t));
|
|
532 if (fcn == NULL) {
|
|
533 rc = NGX_ERROR;
|
|
534 goto failed;
|
|
535 }
|
|
536 }
|
|
537
|
507
|
538 ngx_memcpy((u_char *) &fcn->node.key, c->key, sizeof(ngx_rbtree_key_t));
|
503
|
539
|
507
|
540 ngx_memcpy(fcn->key, &c->key[sizeof(ngx_rbtree_key_t)],
|
503
|
541 NGX_HTTP_CACHE_KEY_LEN - sizeof(ngx_rbtree_key_t));
|
|
542
|
520
|
543 ngx_rbtree_insert(&cache->sh->rbtree, &fcn->node);
|
503
|
544
|
|
545 renew:
|
|
546
|
|
547 rc = NGX_DECLINED;
|
|
548
|
|
549 fcn->uses = 1;
|
|
550 fcn->count = 1;
|
|
551 fcn->valid_msec = 0;
|
|
552 fcn->error = 0;
|
|
553 fcn->exists = 0;
|
|
554 fcn->valid_sec = 0;
|
|
555 fcn->uniq = 0;
|
|
556 fcn->body_start = 0;
|
509
|
557 fcn->length = 0;
|
503
|
558
|
|
559 done:
|
|
560
|
|
561 fcn->expire = ngx_time() + cache->inactive;
|
|
562
|
520
|
563 ngx_queue_insert_head(&cache->sh->queue, &fcn->queue);
|
503
|
564
|
507
|
565 c->uniq = fcn->uniq;
|
|
566 c->error = fcn->error;
|
|
567 c->node = fcn;
|
503
|
568
|
|
569 failed:
|
|
570
|
|
571 ngx_shmtx_unlock(&cache->shpool->mutex);
|
|
572
|
|
573 return rc;
|
|
574 }
|
|
575
|
|
576
|
|
577 static ngx_http_file_cache_node_t *
|
|
578 ngx_http_file_cache_lookup(ngx_http_file_cache_t *cache, u_char *key)
|
|
579 {
|
|
580 ngx_int_t rc;
|
|
581 ngx_rbtree_key_t node_key;
|
|
582 ngx_rbtree_node_t *node, *sentinel;
|
|
583 ngx_http_file_cache_node_t *fcn;
|
|
584
|
|
585 ngx_memcpy((u_char *) &node_key, key, sizeof(ngx_rbtree_key_t));
|
|
586
|
520
|
587 node = cache->sh->rbtree.root;
|
|
588 sentinel = cache->sh->rbtree.sentinel;
|
503
|
589
|
|
590 while (node != sentinel) {
|
|
591
|
|
592 if (node_key < node->key) {
|
|
593 node = node->left;
|
|
594 continue;
|
|
595 }
|
|
596
|
|
597 if (node_key > node->key) {
|
|
598 node = node->right;
|
|
599 continue;
|
|
600 }
|
|
601
|
|
602 /* node_key == node->key */
|
|
603
|
|
604 do {
|
|
605 fcn = (ngx_http_file_cache_node_t *) node;
|
|
606
|
|
607 rc = ngx_memcmp(&key[sizeof(ngx_rbtree_key_t)], fcn->key,
|
|
608 NGX_HTTP_CACHE_KEY_LEN - sizeof(ngx_rbtree_key_t));
|
|
609
|
|
610 if (rc == 0) {
|
|
611 return fcn;
|
|
612 }
|
|
613
|
|
614 node = (rc < 0) ? node->left : node->right;
|
|
615
|
|
616 } while (node != sentinel && node_key == node->key);
|
|
617
|
|
618 break;
|
|
619 }
|
|
620
|
|
621 /* not found */
|
|
622
|
|
623 return NULL;
|
|
624 }
|
|
625
|
|
626
|
|
627 static void
|
|
628 ngx_http_file_cache_rbtree_insert_value(ngx_rbtree_node_t *temp,
|
|
629 ngx_rbtree_node_t *node, ngx_rbtree_node_t *sentinel)
|
|
630 {
|
|
631 ngx_rbtree_node_t **p;
|
|
632 ngx_http_file_cache_node_t *cn, *cnt;
|
|
633
|
|
634 for ( ;; ) {
|
|
635
|
|
636 if (node->key < temp->key) {
|
|
637
|
|
638 p = &temp->left;
|
|
639
|
|
640 } else if (node->key > temp->key) {
|
|
641
|
|
642 p = &temp->right;
|
|
643
|
|
644 } else { /* node->key == temp->key */
|
|
645
|
|
646 cn = (ngx_http_file_cache_node_t *) node;
|
|
647 cnt = (ngx_http_file_cache_node_t *) temp;
|
|
648
|
|
649 p = (ngx_memcmp(cn->key, cnt->key,
|
|
650 NGX_HTTP_CACHE_KEY_LEN - sizeof(ngx_rbtree_key_t))
|
|
651 < 0)
|
|
652 ? &temp->left : &temp->right;
|
|
653 }
|
|
654
|
|
655 if (*p == sentinel) {
|
|
656 break;
|
|
657 }
|
|
658
|
|
659 temp = *p;
|
|
660 }
|
|
661
|
|
662 *p = node;
|
|
663 node->parent = temp;
|
|
664 node->left = sentinel;
|
|
665 node->right = sentinel;
|
|
666 ngx_rbt_red(node);
|
|
667 }
|
|
668
|
|
669
|
|
670 void
|
|
671 ngx_http_file_cache_set_header(ngx_http_request_t *r, u_char *buf)
|
|
672 {
|
|
673 ngx_http_file_cache_header_t *h = (ngx_http_file_cache_header_t *) buf;
|
|
674
|
|
675 u_char *p;
|
|
676 ngx_str_t *key;
|
|
677 ngx_uint_t i;
|
|
678 ngx_http_cache_t *c;
|
|
679
|
|
680 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
681 "http file cache set header");
|
|
682
|
|
683 c = r->cache;
|
|
684
|
|
685 h->valid_sec = c->valid_sec;
|
|
686 h->last_modified = c->last_modified;
|
|
687 h->date = c->date;
|
|
688 h->crc32 = c->crc32;
|
|
689 h->valid_msec = (u_short) c->valid_msec;
|
|
690 h->header_start = (u_short) c->header_start;
|
|
691 h->body_start = (u_short) c->body_start;
|
|
692
|
|
693 p = buf + sizeof(ngx_http_file_cache_header_t);
|
|
694
|
|
695 p = ngx_cpymem(p, ngx_http_file_cache_key, sizeof(ngx_http_file_cache_key));
|
|
696
|
|
697 key = c->keys.elts;
|
|
698 for (i = 0; i < c->keys.nelts; i++) {
|
|
699 p = ngx_copy(p, key[i].data, key[i].len);
|
|
700 }
|
|
701
|
|
702 *p = LF;
|
|
703 }
|
|
704
|
|
705
|
|
706 void
|
|
707 ngx_http_file_cache_update(ngx_http_request_t *r, ngx_temp_file_t *tf)
|
|
708 {
|
579
|
709 off_t size, length;
|
503
|
710 ngx_int_t rc;
|
|
711 ngx_file_uniq_t uniq;
|
|
712 ngx_file_info_t fi;
|
|
713 ngx_http_cache_t *c;
|
|
714 ngx_ext_rename_file_t ext;
|
|
715 ngx_http_file_cache_t *cache;
|
|
716
|
|
717 c = r->cache;
|
|
718
|
|
719 if (c->updated) {
|
|
720 return;
|
|
721 }
|
|
722
|
|
723 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
724 "http file cache update");
|
|
725
|
|
726 c->updated = 1;
|
|
727
|
|
728 cache = c->file_cache;
|
|
729
|
|
730 uniq = 0;
|
579
|
731 length = 0;
|
503
|
732
|
|
733 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
734 "http file cache rename: \"%s\" to \"%s\"",
|
|
735 tf->file.name.data, c->file.name.data);
|
|
736
|
|
737 ext.access = NGX_FILE_OWNER_ACCESS;
|
|
738 ext.path_access = NGX_FILE_OWNER_ACCESS;
|
|
739 ext.time = -1;
|
|
740 ext.create_path = 1;
|
|
741 ext.delete_file = 1;
|
|
742 ext.log = r->connection->log;
|
|
743
|
|
744 rc = ngx_ext_rename_file(&tf->file.name, &c->file.name, &ext);
|
|
745
|
|
746 if (rc == NGX_OK) {
|
|
747
|
|
748 if (ngx_fd_info(tf->file.fd, &fi) == NGX_FILE_ERROR) {
|
|
749 ngx_log_error(NGX_LOG_CRIT, r->connection->log, ngx_errno,
|
|
750 ngx_fd_info_n " \"%s\" failed", tf->file.name.data);
|
|
751
|
|
752 rc = NGX_ERROR;
|
|
753
|
|
754 } else {
|
|
755 uniq = ngx_file_uniq(&fi);
|
579
|
756 length = ngx_file_size(&fi);
|
503
|
757 }
|
|
758 }
|
|
759
|
579
|
760 size = (length + cache->bsize - 1) / cache->bsize;
|
507
|
761
|
503
|
762 ngx_shmtx_lock(&cache->shpool->mutex);
|
|
763
|
|
764 c->node->count--;
|
|
765 c->node->uniq = uniq;
|
|
766 c->node->body_start = c->body_start;
|
|
767
|
507
|
768 size = size - (c->node->length + cache->bsize - 1) / cache->bsize;
|
|
769
|
579
|
770 c->node->length = length;
|
507
|
771
|
520
|
772 cache->sh->size += size;
|
507
|
773
|
503
|
774 if (rc == NGX_OK) {
|
|
775 c->node->exists = 1;
|
|
776 }
|
|
777
|
538
|
778 c->node->updating = 0;
|
|
779
|
503
|
780 ngx_shmtx_unlock(&cache->shpool->mutex);
|
|
781 }
|
|
782
|
|
783
|
|
784 ngx_int_t
|
|
785 ngx_http_cache_send(ngx_http_request_t *r)
|
|
786 {
|
|
787 ngx_int_t rc;
|
|
788 ngx_buf_t *b;
|
|
789 ngx_chain_t out;
|
|
790 ngx_http_cache_t *c;
|
|
791
|
|
792 c = r->cache;
|
|
793
|
|
794 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
795 "http file cache send: %s", c->file.name.data);
|
|
796
|
|
797 /* we need to allocate all before the header would be sent */
|
|
798
|
|
799 b = ngx_pcalloc(r->pool, sizeof(ngx_buf_t));
|
|
800 if (b == NULL) {
|
|
801 return NGX_HTTP_INTERNAL_SERVER_ERROR;
|
|
802 }
|
|
803
|
|
804 b->file = ngx_pcalloc(r->pool, sizeof(ngx_file_t));
|
|
805 if (b->file == NULL) {
|
|
806 return NGX_HTTP_INTERNAL_SERVER_ERROR;
|
|
807 }
|
|
808
|
601
|
809 r->header_only = (c->length - c->body_start) == 0;
|
|
810
|
503
|
811 rc = ngx_http_send_header(r);
|
|
812
|
|
813 if (rc == NGX_ERROR || rc > NGX_OK || r->header_only) {
|
|
814 return rc;
|
|
815 }
|
|
816
|
|
817 b->file_pos = c->body_start;
|
|
818 b->file_last = c->length;
|
|
819
|
601
|
820 b->in_file = 1;
|
503
|
821 b->last_buf = (r == r->main) ? 1: 0;
|
|
822 b->last_in_chain = 1;
|
|
823
|
|
824 b->file->fd = c->file.fd;
|
|
825 b->file->name = c->file.name;
|
|
826 b->file->log = r->connection->log;
|
|
827
|
|
828 out.buf = b;
|
|
829 out.next = NULL;
|
|
830
|
|
831 return ngx_http_output_filter(r, &out);
|
|
832 }
|
|
833
|
|
834
|
|
835 void
|
|
836 ngx_http_file_cache_free(ngx_http_request_t *r, ngx_temp_file_t *tf)
|
|
837 {
|
|
838 ngx_http_cache_t *c;
|
|
839 ngx_http_file_cache_t *cache;
|
|
840
|
|
841 c = r->cache;
|
|
842
|
|
843 if (c->updated) {
|
|
844 return;
|
|
845 }
|
|
846
|
|
847 c->updated = 1;
|
|
848
|
|
849 cache = c->file_cache;
|
|
850
|
|
851 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
852 "http file cache free");
|
|
853
|
|
854 ngx_shmtx_lock(&cache->shpool->mutex);
|
|
855
|
|
856 c->node->count--;
|
|
857
|
|
858 if (c->error) {
|
|
859 c->node->valid_sec = c->valid_sec;
|
|
860 c->node->valid_msec = c->valid_msec;
|
|
861 c->node->error = c->error;
|
|
862 }
|
|
863
|
538
|
864 c->node->updating = 0;
|
|
865
|
503
|
866 ngx_shmtx_unlock(&cache->shpool->mutex);
|
|
867
|
|
868 if (c->temp_file) {
|
|
869 if (tf && tf->file.fd != NGX_INVALID_FILE) {
|
|
870 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
871 "http file cache incomplete: \"%s\"",
|
|
872 tf->file.name.data);
|
|
873
|
|
874 if (ngx_delete_file(tf->file.name.data) == NGX_FILE_ERROR) {
|
|
875 ngx_log_error(NGX_LOG_CRIT, r->connection->log, ngx_errno,
|
|
876 ngx_delete_file_n " \"%s\" failed",
|
|
877 tf->file.name.data);
|
|
878 }
|
|
879 }
|
|
880 }
|
|
881 }
|
|
882
|
|
883
|
|
884 static void
|
|
885 ngx_http_file_cache_cleanup(void *data)
|
|
886 {
|
|
887 ngx_http_cache_t *c = data;
|
|
888
|
|
889 ngx_http_file_cache_t *cache;
|
|
890
|
|
891 if (c->updated) {
|
|
892 return;
|
|
893 }
|
|
894
|
|
895 c->updated = 1;
|
|
896
|
|
897 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, c->file.log, 0,
|
|
898 "http file cache cleanup");
|
|
899
|
|
900 if (c->error) {
|
|
901 return;
|
|
902 }
|
|
903
|
|
904 cache = c->file_cache;
|
|
905
|
|
906 ngx_shmtx_lock(&cache->shpool->mutex);
|
|
907
|
|
908 c->node->count--;
|
|
909
|
|
910 ngx_shmtx_unlock(&cache->shpool->mutex);
|
|
911 }
|
|
912
|
|
913
|
|
914 static time_t
|
507
|
915 ngx_http_file_cache_forced_expire(ngx_http_file_cache_t *cache)
|
|
916 {
|
|
917 u_char *name;
|
|
918 size_t len;
|
|
919 time_t wait;
|
|
920 ngx_uint_t tries;
|
|
921 ngx_path_t *path;
|
|
922 ngx_queue_t *q;
|
|
923 ngx_http_file_cache_node_t *fcn;
|
|
924
|
|
925 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, ngx_cycle->log, 0,
|
|
926 "http file cache forced expire");
|
|
927
|
|
928 path = cache->path;
|
|
929 len = path->name.len + 1 + path->len + 2 * NGX_HTTP_CACHE_KEY_LEN;
|
|
930
|
|
931 name = ngx_alloc(len + 1, ngx_cycle->log);
|
|
932 if (name == NULL) {
|
518
|
933 return 10;
|
507
|
934 }
|
|
935
|
|
936 ngx_memcpy(name, path->name.data, path->name.len);
|
|
937
|
518
|
938 wait = 10;
|
507
|
939 tries = 0;
|
|
940
|
|
941 ngx_shmtx_lock(&cache->shpool->mutex);
|
|
942
|
520
|
943 for (q = ngx_queue_last(&cache->sh->queue);
|
|
944 q != ngx_queue_sentinel(&cache->sh->queue);
|
507
|
945 q = ngx_queue_prev(q))
|
|
946 {
|
|
947 fcn = ngx_queue_data(q, ngx_http_file_cache_node_t, queue);
|
|
948
|
|
949 ngx_log_debug6(NGX_LOG_DEBUG_HTTP, ngx_cycle->log, 0,
|
|
950 "http file cache forced expire: #%d %d %02xd%02xd%02xd%02xd",
|
|
951 fcn->count, fcn->exists,
|
|
952 fcn->key[0], fcn->key[1], fcn->key[2], fcn->key[3]);
|
|
953
|
|
954 if (fcn->count) {
|
|
955
|
|
956 if (tries++ < 20) {
|
|
957 continue;
|
|
958 }
|
|
959
|
|
960 wait = 1;
|
|
961
|
|
962 break;
|
|
963 }
|
|
964
|
|
965 if (!fcn->exists) {
|
|
966
|
|
967 ngx_queue_remove(q);
|
520
|
968 ngx_rbtree_delete(&cache->sh->rbtree, &fcn->node);
|
507
|
969 ngx_slab_free_locked(cache->shpool, fcn);
|
|
970
|
|
971 break;
|
|
972 }
|
|
973
|
|
974 ngx_http_file_cache_delete(cache, q, name);
|
|
975
|
|
976 break;
|
|
977 }
|
|
978
|
|
979 ngx_shmtx_unlock(&cache->shpool->mutex);
|
|
980
|
|
981 ngx_free(name);
|
|
982
|
|
983 return wait;
|
|
984 }
|
|
985
|
|
986
|
|
987 static time_t
|
|
988 ngx_http_file_cache_expire(ngx_http_file_cache_t *cache)
|
503
|
989 {
|
|
990 u_char *name, *p;
|
|
991 size_t len;
|
|
992 time_t now, wait;
|
|
993 ngx_path_t *path;
|
|
994 ngx_queue_t *q;
|
|
995 ngx_http_file_cache_node_t *fcn;
|
|
996 u_char key[2 * NGX_HTTP_CACHE_KEY_LEN];
|
|
997
|
|
998 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, ngx_cycle->log, 0,
|
|
999 "http file cache expire");
|
|
1000
|
|
1001 path = cache->path;
|
|
1002 len = path->name.len + 1 + path->len + 2 * NGX_HTTP_CACHE_KEY_LEN;
|
|
1003
|
507
|
1004 name = ngx_alloc(len + 1, ngx_cycle->log);
|
|
1005 if (name == NULL) {
|
518
|
1006 return 10;
|
507
|
1007 }
|
|
1008
|
|
1009 ngx_memcpy(name, path->name.data, path->name.len);
|
|
1010
|
503
|
1011 now = ngx_time();
|
|
1012
|
|
1013 ngx_shmtx_lock(&cache->shpool->mutex);
|
|
1014
|
|
1015 for ( ;; ) {
|
|
1016
|
520
|
1017 if (ngx_queue_empty(&cache->sh->queue)) {
|
518
|
1018 wait = 10;
|
503
|
1019 break;
|
|
1020 }
|
|
1021
|
520
|
1022 q = ngx_queue_last(&cache->sh->queue);
|
503
|
1023
|
|
1024 fcn = ngx_queue_data(q, ngx_http_file_cache_node_t, queue);
|
|
1025
|
507
|
1026 wait = fcn->expire - now;
|
503
|
1027
|
507
|
1028 if (wait > 0) {
|
518
|
1029 wait = wait > 10 ? 10 : wait;
|
507
|
1030 break;
|
503
|
1031 }
|
|
1032
|
|
1033 ngx_log_debug6(NGX_LOG_DEBUG_HTTP, ngx_cycle->log, 0,
|
|
1034 "http file cache expire: #%d %d %02xd%02xd%02xd%02xd",
|
|
1035 fcn->count, fcn->exists,
|
|
1036 fcn->key[0], fcn->key[1], fcn->key[2], fcn->key[3]);
|
|
1037
|
|
1038 if (fcn->count) {
|
|
1039
|
507
|
1040 p = ngx_hex_dump(key, (u_char *) &fcn->node.key,
|
|
1041 sizeof(ngx_rbtree_key_t));
|
503
|
1042
|
507
|
1043 len = NGX_HTTP_CACHE_KEY_LEN - sizeof(ngx_rbtree_key_t);
|
|
1044 (void) ngx_hex_dump(p, fcn->key, len);
|
503
|
1045
|
507
|
1046 /*
|
|
1047 * abnormally exited workers may leave locked cache entries,
|
|
1048 * and although it may be safe to remove them completely,
|
|
1049 * we prefer to remove them from inactive queue and rbtree
|
|
1050 * only, and to allow other leaks
|
|
1051 */
|
503
|
1052
|
507
|
1053 ngx_queue_remove(q);
|
520
|
1054 ngx_rbtree_delete(&cache->sh->rbtree, &fcn->node);
|
503
|
1055
|
507
|
1056 ngx_log_error(NGX_LOG_ALERT, ngx_cycle->log, 0,
|
505
|
1057 "ignore long locked inactive cache entry %*s, count:%d",
|
|
1058 2 * NGX_HTTP_CACHE_KEY_LEN, key, fcn->count);
|
503
|
1059
|
507
|
1060 continue;
|
503
|
1061 }
|
|
1062
|
|
1063 if (!fcn->exists) {
|
|
1064
|
|
1065 ngx_queue_remove(q);
|
520
|
1066 ngx_rbtree_delete(&cache->sh->rbtree, &fcn->node);
|
503
|
1067 ngx_slab_free_locked(cache->shpool, fcn);
|
|
1068
|
|
1069 continue;
|
|
1070 }
|
|
1071
|
507
|
1072 ngx_http_file_cache_delete(cache, q, name);
|
503
|
1073 }
|
|
1074
|
|
1075 ngx_shmtx_unlock(&cache->shpool->mutex);
|
|
1076
|
507
|
1077 ngx_free(name);
|
|
1078
|
503
|
1079 return wait;
|
|
1080 }
|
|
1081
|
|
1082
|
507
|
1083 static void
|
|
1084 ngx_http_file_cache_delete(ngx_http_file_cache_t *cache, ngx_queue_t *q,
|
|
1085 u_char *name)
|
|
1086 {
|
|
1087 u_char *p;
|
|
1088 size_t len;
|
|
1089 ngx_path_t *path;
|
|
1090 ngx_http_file_cache_node_t *fcn;
|
|
1091
|
|
1092 fcn = ngx_queue_data(q, ngx_http_file_cache_node_t, queue);
|
|
1093
|
520
|
1094 cache->sh->size -= (fcn->length + cache->bsize - 1) / cache->bsize;
|
507
|
1095
|
|
1096 path = cache->path;
|
|
1097
|
|
1098 p = name + path->name.len + 1 + path->len;
|
|
1099
|
|
1100 p = ngx_hex_dump(p, (u_char *) &fcn->node.key, sizeof(ngx_rbtree_key_t));
|
|
1101
|
|
1102 len = NGX_HTTP_CACHE_KEY_LEN - sizeof(ngx_rbtree_key_t);
|
|
1103 p = ngx_hex_dump(p, fcn->key, len);
|
|
1104 *p = '\0';
|
|
1105
|
|
1106 ngx_queue_remove(q);
|
|
1107
|
520
|
1108 ngx_rbtree_delete(&cache->sh->rbtree, &fcn->node);
|
507
|
1109
|
|
1110 ngx_slab_free_locked(cache->shpool, fcn);
|
|
1111
|
|
1112 ngx_shmtx_unlock(&cache->shpool->mutex);
|
|
1113
|
|
1114 len = path->name.len + 1 + path->len + 2 * NGX_HTTP_CACHE_KEY_LEN;
|
|
1115
|
|
1116 ngx_create_hashed_filename(path, name, len);
|
|
1117
|
|
1118 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, ngx_cycle->log, 0,
|
|
1119 "http file cache expire: \"%s\"", name);
|
|
1120
|
|
1121 if (ngx_delete_file(name) == NGX_FILE_ERROR) {
|
|
1122 ngx_log_error(NGX_LOG_CRIT, ngx_cycle->log, ngx_errno,
|
|
1123 ngx_delete_file_n " \"%s\" failed", name);
|
|
1124 }
|
|
1125
|
|
1126 ngx_shmtx_lock(&cache->shpool->mutex);
|
|
1127 }
|
|
1128
|
|
1129
|
|
1130 static time_t
|
|
1131 ngx_http_file_cache_manager(void *data)
|
503
|
1132 {
|
|
1133 ngx_http_file_cache_t *cache = data;
|
|
1134
|
554
|
1135 off_t size;
|
|
1136 time_t next;
|
503
|
1137
|
507
|
1138 next = ngx_http_file_cache_expire(cache);
|
|
1139
|
|
1140 cache->last = ngx_current_msec;
|
|
1141 cache->files = 0;
|
|
1142
|
|
1143 for ( ;; ) {
|
|
1144 ngx_shmtx_lock(&cache->shpool->mutex);
|
|
1145
|
520
|
1146 size = cache->sh->size;
|
503
|
1147
|
507
|
1148 ngx_shmtx_unlock(&cache->shpool->mutex);
|
|
1149
|
509
|
1150 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, ngx_cycle->log, 0,
|
|
1151 "http file cache size: %O", size);
|
|
1152
|
507
|
1153 if (size < cache->max_size) {
|
|
1154 return next;
|
|
1155 }
|
503
|
1156
|
507
|
1157 next = ngx_http_file_cache_forced_expire(cache);
|
|
1158
|
|
1159 if (ngx_http_file_cache_manager_sleep(cache) != NGX_OK) {
|
|
1160 return next;
|
|
1161 }
|
|
1162 }
|
503
|
1163 }
|
|
1164
|
|
1165
|
554
|
1166 static void
|
|
1167 ngx_http_file_cache_loader(void *data)
|
|
1168 {
|
|
1169 ngx_http_file_cache_t *cache = data;
|
|
1170
|
|
1171 ngx_tree_ctx_t tree;
|
|
1172
|
|
1173 if (!cache->sh->cold || cache->sh->loading) {
|
|
1174 return;
|
|
1175 }
|
|
1176
|
|
1177 if (!ngx_atomic_cmp_set(&cache->sh->loading, 0, ngx_pid)) {
|
|
1178 return;
|
|
1179 }
|
|
1180
|
|
1181 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, ngx_cycle->log, 0,
|
|
1182 "http file cache loader");
|
|
1183
|
|
1184 tree.init_handler = NULL;
|
|
1185 tree.file_handler = ngx_http_file_cache_manage_file;
|
|
1186 tree.pre_tree_handler = ngx_http_file_cache_noop;
|
|
1187 tree.post_tree_handler = ngx_http_file_cache_noop;
|
|
1188 tree.spec_handler = ngx_http_file_cache_delete_file;
|
|
1189 tree.data = cache;
|
|
1190 tree.alloc = 0;
|
|
1191 tree.log = ngx_cycle->log;
|
|
1192
|
|
1193 cache->last = ngx_current_msec;
|
|
1194 cache->files = 0;
|
|
1195
|
|
1196 if (ngx_walk_tree(&tree, &cache->path->name) == NGX_ABORT) {
|
|
1197 cache->sh->loading = 0;
|
|
1198 return;
|
|
1199 }
|
|
1200
|
|
1201 cache->sh->cold = 0;
|
|
1202 cache->sh->loading = 0;
|
|
1203
|
|
1204 ngx_log_error(NGX_LOG_NOTICE, ngx_cycle->log, 0,
|
|
1205 "http file cache: %V %.3fM, bsize: %uz",
|
|
1206 &cache->path->name,
|
|
1207 ((double) cache->sh->size * cache->bsize) / (1024 * 1024),
|
|
1208 cache->bsize);
|
|
1209 }
|
|
1210
|
|
1211
|
503
|
1212 static ngx_int_t
|
507
|
1213 ngx_http_file_cache_manager_sleep(ngx_http_file_cache_t *cache)
|
|
1214 {
|
|
1215 ngx_msec_t elapsed;
|
|
1216
|
|
1217 if (cache->files++ > 100) {
|
|
1218
|
|
1219 ngx_time_update(0, 0);
|
|
1220
|
|
1221 elapsed = ngx_abs((ngx_msec_int_t) (ngx_current_msec - cache->last));
|
|
1222
|
|
1223 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, ngx_cycle->log, 0,
|
|
1224 "http file cache manager time: %M", elapsed);
|
|
1225
|
|
1226 if (elapsed > 200) {
|
|
1227
|
|
1228 /*
|
|
1229 * if processing 100 files takes more than 200ms,
|
|
1230 * it seems that many operations require disk i/o,
|
|
1231 * therefore sleep 200ms
|
|
1232 */
|
|
1233
|
|
1234 ngx_msleep(200);
|
|
1235
|
|
1236 ngx_time_update(0, 0);
|
|
1237 }
|
|
1238
|
|
1239 cache->last = ngx_current_msec;
|
|
1240 cache->files = 0;
|
|
1241 }
|
|
1242
|
|
1243 return (ngx_quit || ngx_terminate) ? NGX_ABORT : NGX_OK;
|
|
1244 }
|
|
1245
|
|
1246
|
|
1247 static ngx_int_t
|
|
1248 ngx_http_file_cache_noop(ngx_tree_ctx_t *ctx, ngx_str_t *path)
|
503
|
1249 {
|
|
1250 return NGX_OK;
|
|
1251 }
|
|
1252
|
|
1253
|
|
1254 static ngx_int_t
|
507
|
1255 ngx_http_file_cache_manage_file(ngx_tree_ctx_t *ctx, ngx_str_t *path)
|
503
|
1256 {
|
507
|
1257 ngx_http_file_cache_t *cache;
|
|
1258
|
|
1259 cache = ctx->data;
|
|
1260
|
|
1261 if (ngx_http_file_cache_add_file(ctx, path) != NGX_OK) {
|
|
1262 (void) ngx_http_file_cache_delete_file(ctx, path);
|
|
1263 }
|
|
1264
|
|
1265 return ngx_http_file_cache_manager_sleep(cache);
|
|
1266 }
|
|
1267
|
503
|
1268
|
507
|
1269 static ngx_int_t
|
|
1270 ngx_http_file_cache_add_file(ngx_tree_ctx_t *ctx, ngx_str_t *name)
|
|
1271 {
|
|
1272 u_char *p;
|
|
1273 ngx_fd_t fd;
|
|
1274 ngx_int_t n;
|
|
1275 ngx_uint_t i;
|
|
1276 ngx_file_info_t fi;
|
|
1277 ngx_http_cache_t c;
|
|
1278 ngx_http_file_cache_t *cache;
|
|
1279 ngx_http_file_cache_header_t h;
|
|
1280
|
|
1281 if (name->len < 2 * NGX_HTTP_CACHE_KEY_LEN) {
|
|
1282 return NGX_ERROR;
|
503
|
1283 }
|
|
1284
|
507
|
1285 ngx_memzero(&c, sizeof(ngx_http_cache_t));
|
|
1286
|
|
1287 fd = ngx_open_file(name->data, NGX_FILE_RDONLY, NGX_FILE_OPEN, 0);
|
|
1288
|
|
1289 if (fd == NGX_INVALID_FILE) {
|
|
1290 ngx_log_error(NGX_LOG_CRIT, ctx->log, ngx_errno,
|
|
1291 ngx_open_file_n " \"%s\" failed", name->data);
|
|
1292 return NGX_ERROR;
|
|
1293 }
|
|
1294
|
|
1295 c.file.fd = fd;
|
|
1296 c.file.name = *name;
|
|
1297 c.file.log = ctx->log;
|
|
1298
|
|
1299 n = ngx_read_file(&c.file, (u_char *) &h,
|
|
1300 sizeof(ngx_http_file_cache_header_t), 0);
|
|
1301 if (n == NGX_ERROR) {
|
|
1302 return NGX_ERROR;
|
|
1303 }
|
|
1304
|
|
1305 if ((size_t) n < sizeof(ngx_http_file_cache_header_t)) {
|
|
1306 ngx_log_error(NGX_LOG_CRIT, ctx->log, 0,
|
|
1307 "cache file \"%s\" is too small", name->data);
|
|
1308 return NGX_ERROR;
|
|
1309 }
|
|
1310
|
|
1311 if (ngx_fd_info(fd, &fi) == NGX_FILE_ERROR) {
|
|
1312 ngx_log_error(NGX_LOG_CRIT, ctx->log, ngx_errno,
|
|
1313 ngx_fd_info_n " \"%s\" failed", name->data);
|
|
1314
|
|
1315 } else {
|
|
1316 c.uniq = ngx_file_uniq(&fi);
|
|
1317 c.valid_sec = h.valid_sec;
|
|
1318 c.valid_msec = h.valid_msec;
|
|
1319 c.body_start = h.body_start;
|
|
1320 c.length = ngx_file_size(&fi);
|
|
1321 }
|
|
1322
|
|
1323 if (ngx_close_file(fd) == NGX_FILE_ERROR) {
|
|
1324 ngx_log_error(NGX_LOG_ALERT, ctx->log, ngx_errno,
|
|
1325 ngx_close_file_n " \"%s\" failed", name->data);
|
|
1326 }
|
|
1327
|
|
1328 if (c.body_start == 0) {
|
|
1329 return NGX_ERROR;
|
|
1330 }
|
|
1331
|
|
1332 p = &name->data[name->len - 2 * NGX_HTTP_CACHE_KEY_LEN];
|
503
|
1333
|
|
1334 for (i = 0; i < NGX_HTTP_CACHE_KEY_LEN; i++) {
|
|
1335 n = ngx_hextoi(p, 2);
|
|
1336
|
|
1337 if (n == NGX_ERROR) {
|
507
|
1338 return NGX_ERROR;
|
503
|
1339 }
|
|
1340
|
|
1341 p += 2;
|
|
1342
|
507
|
1343 c.key[i] = (u_char) n;
|
503
|
1344 }
|
|
1345
|
|
1346 cache = ctx->data;
|
|
1347
|
507
|
1348 return ngx_http_file_cache_add(cache, &c);
|
|
1349 }
|
|
1350
|
|
1351
|
|
1352 static ngx_int_t
|
|
1353 ngx_http_file_cache_add(ngx_http_file_cache_t *cache, ngx_http_cache_t *c)
|
|
1354 {
|
|
1355 ngx_http_file_cache_node_t *fcn;
|
|
1356
|
503
|
1357 ngx_shmtx_lock(&cache->shpool->mutex);
|
|
1358
|
507
|
1359 fcn = ngx_http_file_cache_lookup(cache, c->key);
|
|
1360
|
|
1361 if (fcn == NULL) {
|
|
1362
|
|
1363 fcn = ngx_slab_alloc_locked(cache->shpool,
|
|
1364 sizeof(ngx_http_file_cache_node_t));
|
|
1365 if (fcn == NULL) {
|
|
1366 ngx_shmtx_unlock(&cache->shpool->mutex);
|
|
1367 return NGX_ERROR;
|
|
1368 }
|
|
1369
|
|
1370 ngx_memcpy((u_char *) &fcn->node.key, c->key, sizeof(ngx_rbtree_key_t));
|
|
1371
|
|
1372 ngx_memcpy(fcn->key, &c->key[sizeof(ngx_rbtree_key_t)],
|
|
1373 NGX_HTTP_CACHE_KEY_LEN - sizeof(ngx_rbtree_key_t));
|
|
1374
|
520
|
1375 ngx_rbtree_insert(&cache->sh->rbtree, &fcn->node);
|
507
|
1376
|
|
1377 fcn->uses = 1;
|
|
1378 fcn->count = 0;
|
|
1379 fcn->valid_msec = c->valid_msec;
|
|
1380 fcn->error = 0;
|
|
1381 fcn->exists = 1;
|
|
1382 fcn->uniq = c->uniq;
|
|
1383 fcn->valid_sec = c->valid_sec;
|
|
1384 fcn->body_start = c->body_start;
|
|
1385 fcn->length = c->length;
|
|
1386
|
520
|
1387 cache->sh->size += (c->length + cache->bsize - 1) / cache->bsize;
|
507
|
1388
|
|
1389 } else {
|
|
1390 ngx_queue_remove(&fcn->queue);
|
|
1391 }
|
|
1392
|
|
1393 fcn->expire = ngx_time() + cache->inactive;
|
|
1394
|
520
|
1395 ngx_queue_insert_head(&cache->sh->queue, &fcn->queue);
|
503
|
1396
|
|
1397 ngx_shmtx_unlock(&cache->shpool->mutex);
|
|
1398
|
507
|
1399 return NGX_OK;
|
|
1400 }
|
|
1401
|
503
|
1402
|
507
|
1403 static ngx_int_t
|
|
1404 ngx_http_file_cache_delete_file(ngx_tree_ctx_t *ctx, ngx_str_t *path)
|
|
1405 {
|
503
|
1406 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, ctx->log, 0,
|
507
|
1407 "http file cache delete: \"%s\"", path->data);
|
503
|
1408
|
|
1409 if (ngx_delete_file(path->data) == NGX_FILE_ERROR) {
|
|
1410 ngx_log_error(NGX_LOG_CRIT, ctx->log, ngx_errno,
|
|
1411 ngx_delete_file_n " \"%s\" failed", path->data);
|
|
1412 }
|
26
|
1413
|
0
|
1414 return NGX_OK;
|
|
1415 }
|
|
1416
|
|
1417
|
503
|
1418 time_t
|
|
1419 ngx_http_file_cache_valid(ngx_array_t *cache_valid, ngx_uint_t status)
|
0
|
1420 {
|
503
|
1421 ngx_uint_t i;
|
|
1422 ngx_http_cache_valid_t *valid;
|
0
|
1423
|
509
|
1424 if (cache_valid == NULL) {
|
|
1425 return 0;
|
|
1426 }
|
|
1427
|
503
|
1428 valid = cache_valid->elts;
|
|
1429 for (i = 0; i < cache_valid->nelts; i++) {
|
0
|
1430
|
503
|
1431 if (valid[i].status == 0) {
|
|
1432 return valid[i].valid;
|
0
|
1433 }
|
|
1434
|
503
|
1435 if (valid[i].status == status) {
|
|
1436 return valid[i].valid;
|
0
|
1437 }
|
503
|
1438 }
|
0
|
1439
|
503
|
1440 return 0;
|
0
|
1441 }
|
|
1442
|
|
1443
|
503
|
1444 char *
|
|
1445 ngx_http_file_cache_set_slot(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
1446 {
|
507
|
1447 off_t max_size;
|
503
|
1448 u_char *last, *p;
|
507
|
1449 time_t inactive;
|
503
|
1450 ssize_t size;
|
|
1451 ngx_str_t s, name, *value;
|
|
1452 ngx_uint_t i, n;
|
|
1453 ngx_http_file_cache_t *cache;
|
|
1454
|
|
1455 cache = ngx_pcalloc(cf->pool, sizeof(ngx_http_file_cache_t));
|
|
1456 if (cache == NULL) {
|
|
1457 return NGX_CONF_ERROR;
|
|
1458 }
|
|
1459
|
|
1460 cache->path = ngx_pcalloc(cf->pool, sizeof(ngx_path_t));
|
|
1461 if (cache->path == NULL) {
|
|
1462 return NGX_CONF_ERROR;
|
|
1463 }
|
|
1464
|
|
1465 inactive = 600;
|
|
1466
|
|
1467 name.len = 0;
|
|
1468 size = 0;
|
507
|
1469 max_size = NGX_MAX_OFF_T_VALUE;
|
503
|
1470
|
|
1471 value = cf->args->elts;
|
|
1472
|
|
1473 cache->path->name = value[1];
|
|
1474
|
|
1475 if (cache->path->name.data[cache->path->name.len - 1] == '/') {
|
|
1476 cache->path->name.len--;
|
|
1477 }
|
|
1478
|
|
1479 if (ngx_conf_full_name(cf->cycle, &cache->path->name, 0) != NGX_OK) {
|
|
1480 return NGX_CONF_ERROR;
|
|
1481 }
|
|
1482
|
|
1483 for (i = 2; i < cf->args->nelts; i++) {
|
|
1484
|
|
1485 if (ngx_strncmp(value[i].data, "levels=", 7) == 0) {
|
|
1486
|
|
1487 p = value[i].data + 7;
|
|
1488 last = value[i].data + value[i].len;
|
|
1489
|
512
|
1490 for (n = 0; n < 3 && p < last; n++) {
|
503
|
1491
|
512
|
1492 if (*p > '0' && *p < '3') {
|
503
|
1493
|
|
1494 cache->path->level[n] = *p++ - '0';
|
|
1495 cache->path->len += cache->path->level[n] + 1;
|
|
1496
|
|
1497 if (p == last) {
|
|
1498 break;
|
|
1499 }
|
|
1500
|
512
|
1501 if (*p++ == ':' && n < 2 && p != last) {
|
503
|
1502 continue;
|
|
1503 }
|
512
|
1504
|
|
1505 goto invalid_levels;
|
503
|
1506 }
|
|
1507
|
|
1508 goto invalid_levels;
|
|
1509 }
|
|
1510
|
|
1511 if (cache->path->len < 10 + 3) {
|
|
1512 continue;
|
|
1513 }
|
|
1514
|
|
1515 invalid_levels:
|
|
1516
|
|
1517 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1518 "invalid \"levels\" \"%V\"", &value[i]);
|
|
1519 return NGX_CONF_ERROR;
|
|
1520 }
|
|
1521
|
|
1522 if (ngx_strncmp(value[i].data, "keys_zone=", 10) == 0) {
|
|
1523
|
|
1524 name.data = value[i].data + 10;
|
|
1525
|
|
1526 p = (u_char *) ngx_strchr(name.data, ':');
|
|
1527
|
|
1528 if (p) {
|
520
|
1529 *p = '\0';
|
|
1530
|
503
|
1531 name.len = p - name.data;
|
|
1532
|
|
1533 p++;
|
|
1534
|
|
1535 s.len = value[i].data + value[i].len - p;
|
|
1536 s.data = p;
|
|
1537
|
|
1538 size = ngx_parse_size(&s);
|
|
1539 if (size > 8191) {
|
|
1540 continue;
|
|
1541 }
|
|
1542 }
|
|
1543
|
|
1544 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1545 "invalid keys zone size \"%V\"", &value[i]);
|
|
1546 return NGX_CONF_ERROR;
|
|
1547 }
|
|
1548
|
|
1549 if (ngx_strncmp(value[i].data, "inactive=", 9) == 0) {
|
|
1550
|
|
1551 s.len = value[i].len - 9;
|
|
1552 s.data = value[i].data + 9;
|
|
1553
|
|
1554 inactive = ngx_parse_time(&s, 1);
|
|
1555 if (inactive < 0) {
|
|
1556 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1557 "invalid inactive value \"%V\"", &value[i]);
|
|
1558 return NGX_CONF_ERROR;
|
|
1559 }
|
|
1560
|
|
1561 continue;
|
|
1562 }
|
|
1563
|
507
|
1564 if (ngx_strncmp(value[i].data, "max_size=", 9) == 0) {
|
503
|
1565
|
507
|
1566 s.len = value[i].len - 9;
|
|
1567 s.data = value[i].data + 9;
|
503
|
1568
|
507
|
1569 max_size = ngx_parse_offset(&s);
|
|
1570 if (max_size < 0) {
|
503
|
1571 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
507
|
1572 "invalid max_size value \"%V\"", &value[i]);
|
503
|
1573 return NGX_CONF_ERROR;
|
|
1574 }
|
|
1575
|
|
1576 continue;
|
|
1577 }
|
|
1578
|
|
1579 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1580 "invalid parameter \"%V\"", &value[i]);
|
|
1581 return NGX_CONF_ERROR;
|
|
1582 }
|
|
1583
|
|
1584 if (name.len == 0 || size == 0) {
|
|
1585 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1586 "\"%V\" must have \"keys_zone\" parameter",
|
|
1587 &cmd->name);
|
|
1588 return NGX_CONF_ERROR;
|
|
1589 }
|
|
1590
|
507
|
1591 cache->path->manager = ngx_http_file_cache_manager;
|
554
|
1592 cache->path->loader = ngx_http_file_cache_loader;
|
503
|
1593 cache->path->data = cache;
|
|
1594
|
|
1595 if (ngx_add_path(cf, &cache->path) != NGX_OK) {
|
|
1596 return NGX_CONF_ERROR;
|
|
1597 }
|
|
1598
|
|
1599 cache->shm_zone = ngx_shared_memory_add(cf, &name, size, cmd->post);
|
|
1600 if (cache->shm_zone == NULL) {
|
|
1601 return NGX_CONF_ERROR;
|
|
1602 }
|
|
1603
|
|
1604 if (cache->shm_zone->data) {
|
|
1605 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1606 "duplicate zone \"%V\"", &name);
|
|
1607 return NGX_CONF_ERROR;
|
|
1608 }
|
|
1609
|
|
1610
|
|
1611 cache->shm_zone->init = ngx_http_file_cache_init;
|
|
1612 cache->shm_zone->data = cache;
|
|
1613
|
|
1614 cache->inactive = inactive;
|
507
|
1615 cache->max_size = max_size;
|
503
|
1616
|
|
1617 return NGX_CONF_OK;
|
|
1618 }
|
26
|
1619
|
|
1620
|
503
|
1621 char *
|
|
1622 ngx_http_file_cache_valid_set_slot(ngx_conf_t *cf, ngx_command_t *cmd,
|
|
1623 void *conf)
|
0
|
1624 {
|
503
|
1625 char *p = conf;
|
0
|
1626
|
503
|
1627 time_t valid;
|
|
1628 ngx_str_t *value;
|
|
1629 ngx_uint_t i, n, status;
|
|
1630 ngx_array_t **a;
|
|
1631 ngx_http_cache_valid_t *v;
|
|
1632 static ngx_uint_t statuses[] = { 200, 301, 302 };
|
26
|
1633
|
503
|
1634 a = (ngx_array_t **) (p + cmd->offset);
|
0
|
1635
|
503
|
1636 if (*a == NGX_CONF_UNSET_PTR) {
|
|
1637 *a = ngx_array_create(cf->pool, 1, sizeof(ngx_http_cache_valid_t));
|
|
1638 if (*a == NULL) {
|
|
1639 return NGX_CONF_ERROR;
|
|
1640 }
|
|
1641 }
|
0
|
1642
|
503
|
1643 value = cf->args->elts;
|
|
1644 n = cf->args->nelts - 1;
|
0
|
1645
|
503
|
1646 valid = ngx_parse_time(&value[n], 1);
|
|
1647 if (valid < 0) {
|
|
1648 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1649 "invalid time value \"%V\"", &value[n]);
|
|
1650 return NGX_CONF_ERROR;
|
0
|
1651 }
|
|
1652
|
503
|
1653 if (n == 1) {
|
|
1654
|
|
1655 for (i = 0; i < 3; i++) {
|
|
1656 v = ngx_array_push(*a);
|
|
1657 if (v == NULL) {
|
|
1658 return NGX_CONF_ERROR;
|
|
1659 }
|
|
1660
|
|
1661 v->status = statuses[i];
|
|
1662 v->valid = valid;
|
|
1663 }
|
|
1664
|
|
1665 return NGX_CONF_OK;
|
0
|
1666 }
|
|
1667
|
503
|
1668 for (i = 1; i < n; i++) {
|
|
1669
|
|
1670 if (ngx_strcmp(value[i].data, "any") == 0) {
|
|
1671
|
|
1672 status = 0;
|
|
1673
|
|
1674 } else {
|
0
|
1675
|
503
|
1676 status = ngx_atoi(value[i].data, value[i].len);
|
|
1677 if (status < 100) {
|
|
1678 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
1679 "invalid status \"%V\"", &value[i]);
|
|
1680 return NGX_CONF_ERROR;
|
|
1681 }
|
|
1682 }
|
|
1683
|
|
1684 v = ngx_array_push(*a);
|
|
1685 if (v == NULL) {
|
|
1686 return NGX_CONF_ERROR;
|
|
1687 }
|
|
1688
|
|
1689 v->status = status;
|
|
1690 v->valid = valid;
|
|
1691 }
|
|
1692
|
|
1693 return NGX_CONF_OK;
|
0
|
1694 }
|