0
|
1
|
|
2 /*
|
|
3 * Copyright (C) Igor Sysoev
|
660
|
4 * Copyright (C) Nginx, Inc.
|
0
|
5 */
|
|
6
|
|
7
|
|
8 #include <ngx_config.h>
|
|
9 #include <ngx_core.h>
|
|
10 #include <ngx_event.h>
|
|
11 #include <ngx_event_pipe.h>
|
|
12
|
|
13
|
|
14 static ngx_int_t ngx_event_pipe_read_upstream(ngx_event_pipe_t *p);
|
|
15 static ngx_int_t ngx_event_pipe_write_to_downstream(ngx_event_pipe_t *p);
|
|
16
|
|
17 static ngx_int_t ngx_event_pipe_write_chain_to_temp_file(ngx_event_pipe_t *p);
|
16
|
18 static ngx_inline void ngx_event_pipe_remove_shadow_links(ngx_buf_t *buf);
|
0
|
19 static ngx_int_t ngx_event_pipe_drain_chains(ngx_event_pipe_t *p);
|
|
20
|
|
21
|
126
|
22 ngx_int_t
|
344
|
23 ngx_event_pipe(ngx_event_pipe_t *p, ngx_int_t do_write)
|
0
|
24 {
|
|
25 u_int flags;
|
518
|
26 ngx_int_t rc;
|
0
|
27 ngx_event_t *rev, *wev;
|
|
28
|
|
29 for ( ;; ) {
|
|
30 if (do_write) {
|
32
|
31 p->log->action = "sending to client";
|
|
32
|
518
|
33 rc = ngx_event_pipe_write_to_downstream(p);
|
|
34
|
|
35 if (rc == NGX_ABORT) {
|
0
|
36 return NGX_ABORT;
|
|
37 }
|
518
|
38
|
|
39 if (rc == NGX_BUSY) {
|
|
40 return NGX_OK;
|
|
41 }
|
0
|
42 }
|
|
43
|
|
44 p->read = 0;
|
|
45 p->upstream_blocked = 0;
|
|
46
|
32
|
47 p->log->action = "reading upstream";
|
|
48
|
0
|
49 if (ngx_event_pipe_read_upstream(p) == NGX_ABORT) {
|
|
50 return NGX_ABORT;
|
|
51 }
|
|
52
|
|
53 if (!p->read && !p->upstream_blocked) {
|
|
54 break;
|
|
55 }
|
|
56
|
|
57 do_write = 1;
|
|
58 }
|
|
59
|
|
60 if (p->upstream->fd != -1) {
|
|
61 rev = p->upstream->read;
|
|
62
|
|
63 flags = (rev->eof || rev->error) ? NGX_CLOSE_EVENT : 0;
|
|
64
|
430
|
65 if (ngx_handle_read_event(rev, flags) != NGX_OK) {
|
0
|
66 return NGX_ABORT;
|
|
67 }
|
|
68
|
162
|
69 if (rev->active && !rev->ready) {
|
0
|
70 ngx_add_timer(rev, p->read_timeout);
|
162
|
71
|
|
72 } else if (rev->timer_set) {
|
|
73 ngx_del_timer(rev);
|
0
|
74 }
|
|
75 }
|
|
76
|
58
|
77 if (p->downstream->fd != -1 && p->downstream->data == p->output_ctx) {
|
0
|
78 wev = p->downstream->write;
|
430
|
79 if (ngx_handle_write_event(wev, p->send_lowat) != NGX_OK) {
|
0
|
80 return NGX_ABORT;
|
|
81 }
|
|
82
|
162
|
83 if (!wev->delayed) {
|
|
84 if (wev->active && !wev->ready) {
|
|
85 ngx_add_timer(wev, p->send_timeout);
|
|
86
|
|
87 } else if (wev->timer_set) {
|
|
88 ngx_del_timer(wev);
|
|
89 }
|
0
|
90 }
|
|
91 }
|
|
92
|
|
93 return NGX_OK;
|
|
94 }
|
|
95
|
|
96
|
126
|
97 static ngx_int_t
|
|
98 ngx_event_pipe_read_upstream(ngx_event_pipe_t *p)
|
0
|
99 {
|
10
|
100 ssize_t n, size;
|
|
101 ngx_int_t rc;
|
0
|
102 ngx_buf_t *b;
|
56
|
103 ngx_chain_t *chain, *cl, *ln;
|
0
|
104
|
|
105 if (p->upstream_eof || p->upstream_error || p->upstream_done) {
|
|
106 return NGX_OK;
|
|
107 }
|
|
108
|
|
109 ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
|
110 "pipe read upstream: %d", p->upstream->read->ready);
|
|
111
|
|
112 for ( ;; ) {
|
|
113
|
|
114 if (p->upstream_eof || p->upstream_error || p->upstream_done) {
|
|
115 break;
|
|
116 }
|
|
117
|
|
118 if (p->preread_bufs == NULL && !p->upstream->read->ready) {
|
|
119 break;
|
|
120 }
|
|
121
|
|
122 if (p->preread_bufs) {
|
|
123
|
|
124 /* use the pre-read bufs if they exist */
|
|
125
|
|
126 chain = p->preread_bufs;
|
|
127 p->preread_bufs = NULL;
|
|
128 n = p->preread_size;
|
|
129
|
|
130 ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
10
|
131 "pipe preread: %z", n);
|
0
|
132
|
|
133 if (n) {
|
|
134 p->read = 1;
|
|
135 }
|
|
136
|
|
137 } else {
|
|
138
|
130
|
139 #if (NGX_HAVE_KQUEUE)
|
|
140
|
0
|
141 /*
|
|
142 * kqueue notifies about the end of file or a pending error.
|
|
143 * This test allows not to allocate a buf on these conditions
|
126
|
144 * and not to call c->recv_chain().
|
0
|
145 */
|
|
146
|
|
147 if (p->upstream->read->available == 0
|
136
|
148 && p->upstream->read->pending_eof)
|
0
|
149 {
|
|
150 p->upstream->read->ready = 0;
|
636
|
151 p->upstream->read->eof = 1;
|
0
|
152 p->upstream_eof = 1;
|
|
153 p->read = 1;
|
|
154
|
|
155 if (p->upstream->read->kq_errno) {
|
|
156 p->upstream->read->error = 1;
|
|
157 p->upstream_error = 1;
|
|
158 p->upstream_eof = 0;
|
|
159
|
|
160 ngx_log_error(NGX_LOG_ERR, p->log,
|
|
161 p->upstream->read->kq_errno,
|
130
|
162 "kevent() reported that upstream "
|
|
163 "closed connection");
|
0
|
164 }
|
|
165
|
|
166 break;
|
|
167 }
|
130
|
168 #endif
|
0
|
169
|
|
170 if (p->free_raw_bufs) {
|
|
171
|
|
172 /* use the free bufs if they exist */
|
|
173
|
|
174 chain = p->free_raw_bufs;
|
|
175 if (p->single_buf) {
|
|
176 p->free_raw_bufs = p->free_raw_bufs->next;
|
|
177 chain->next = NULL;
|
|
178 } else {
|
|
179 p->free_raw_bufs = NULL;
|
|
180 }
|
|
181
|
|
182 } else if (p->allocated < p->bufs.num) {
|
|
183
|
|
184 /* allocate a new buf if it's still allowed */
|
|
185
|
50
|
186 b = ngx_create_temp_buf(p->pool, p->bufs.size);
|
|
187 if (b == NULL) {
|
0
|
188 return NGX_ABORT;
|
|
189 }
|
|
190
|
|
191 p->allocated++;
|
|
192
|
50
|
193 chain = ngx_alloc_chain_link(p->pool);
|
|
194 if (chain == NULL) {
|
32
|
195 return NGX_ABORT;
|
|
196 }
|
|
197
|
|
198 chain->buf = b;
|
|
199 chain->next = NULL;
|
0
|
200
|
340
|
201 } else if (!p->cacheable
|
58
|
202 && p->downstream->data == p->output_ctx
|
76
|
203 && p->downstream->write->ready
|
|
204 && !p->downstream->write->delayed)
|
58
|
205 {
|
0
|
206 /*
|
|
207 * if the bufs are not needed to be saved in a cache and
|
|
208 * a downstream is ready then write the bufs to a downstream
|
|
209 */
|
|
210
|
|
211 p->upstream_blocked = 1;
|
|
212
|
|
213 ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
|
214 "pipe downstream ready");
|
|
215
|
|
216 break;
|
|
217
|
340
|
218 } else if (p->cacheable
|
0
|
219 || p->temp_file->offset < p->max_temp_file_size)
|
|
220 {
|
|
221
|
|
222 /*
|
32
|
223 * if it is allowed, then save some bufs from r->in
|
0
|
224 * to a temporary file, and add them to a r->out chain
|
|
225 */
|
|
226
|
|
227 rc = ngx_event_pipe_write_chain_to_temp_file(p);
|
|
228
|
|
229 ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
10
|
230 "pipe temp offset: %O", p->temp_file->offset);
|
0
|
231
|
70
|
232 if (rc == NGX_BUSY) {
|
|
233 break;
|
|
234 }
|
|
235
|
0
|
236 if (rc == NGX_AGAIN) {
|
|
237 if (ngx_event_flags & NGX_USE_LEVEL_EVENT
|
|
238 && p->upstream->read->active
|
|
239 && p->upstream->read->ready)
|
|
240 {
|
|
241 if (ngx_del_event(p->upstream->read, NGX_READ_EVENT, 0)
|
126
|
242 == NGX_ERROR)
|
0
|
243 {
|
|
244 return NGX_ABORT;
|
|
245 }
|
|
246 }
|
|
247 }
|
|
248
|
|
249 if (rc != NGX_OK) {
|
|
250 return rc;
|
|
251 }
|
|
252
|
|
253 chain = p->free_raw_bufs;
|
|
254 if (p->single_buf) {
|
|
255 p->free_raw_bufs = p->free_raw_bufs->next;
|
|
256 chain->next = NULL;
|
|
257 } else {
|
|
258 p->free_raw_bufs = NULL;
|
|
259 }
|
|
260
|
|
261 } else {
|
|
262
|
32
|
263 /* there are no bufs to read in */
|
0
|
264
|
|
265 ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
|
266 "no pipe bufs to read in");
|
126
|
267
|
0
|
268 break;
|
|
269 }
|
|
270
|
126
|
271 n = p->upstream->recv_chain(p->upstream, chain);
|
0
|
272
|
|
273 ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
10
|
274 "pipe recv chain: %z", n);
|
0
|
275
|
|
276 if (p->free_raw_bufs) {
|
|
277 chain->next = p->free_raw_bufs;
|
|
278 }
|
|
279 p->free_raw_bufs = chain;
|
|
280
|
|
281 if (n == NGX_ERROR) {
|
|
282 p->upstream_error = 1;
|
|
283 return NGX_ERROR;
|
|
284 }
|
|
285
|
|
286 if (n == NGX_AGAIN) {
|
|
287 if (p->single_buf) {
|
|
288 ngx_event_pipe_remove_shadow_links(chain->buf);
|
|
289 }
|
|
290
|
|
291 break;
|
|
292 }
|
|
293
|
|
294 p->read = 1;
|
|
295
|
|
296 if (n == 0) {
|
|
297 p->upstream_eof = 1;
|
|
298 break;
|
|
299 }
|
|
300 }
|
|
301
|
|
302 p->read_length += n;
|
|
303 cl = chain;
|
258
|
304 p->free_raw_bufs = NULL;
|
0
|
305
|
|
306 while (cl && n > 0) {
|
|
307
|
|
308 ngx_event_pipe_remove_shadow_links(cl->buf);
|
|
309
|
|
310 size = cl->buf->end - cl->buf->last;
|
|
311
|
|
312 if (n >= size) {
|
|
313 cl->buf->last = cl->buf->end;
|
|
314
|
|
315 /* STUB */ cl->buf->num = p->num++;
|
|
316
|
|
317 if (p->input_filter(p, cl->buf) == NGX_ERROR) {
|
|
318 return NGX_ABORT;
|
|
319 }
|
|
320
|
|
321 n -= size;
|
56
|
322 ln = cl;
|
0
|
323 cl = cl->next;
|
56
|
324 ngx_free_chain(p->pool, ln);
|
0
|
325
|
|
326 } else {
|
|
327 cl->buf->last += n;
|
|
328 n = 0;
|
|
329 }
|
|
330 }
|
|
331
|
258
|
332 if (cl) {
|
|
333 for (ln = cl; ln->next; ln = ln->next) { /* void */ }
|
|
334
|
|
335 ln->next = p->free_raw_bufs;
|
|
336 p->free_raw_bufs = cl;
|
|
337 }
|
0
|
338 }
|
|
339
|
|
340 #if (NGX_DEBUG)
|
|
341
|
2
|
342 for (cl = p->busy; cl; cl = cl->next) {
|
32
|
343 ngx_log_debug8(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
|
344 "pipe buf busy s:%d t:%d f:%d "
|
|
345 "%p, pos %p, size: %z "
|
|
346 "file: %O, size: %z",
|
|
347 (cl->buf->shadow ? 1 : 0),
|
|
348 cl->buf->temporary, cl->buf->in_file,
|
0
|
349 cl->buf->start, cl->buf->pos,
|
32
|
350 cl->buf->last - cl->buf->pos,
|
|
351 cl->buf->file_pos,
|
|
352 cl->buf->file_last - cl->buf->file_pos);
|
0
|
353 }
|
|
354
|
2
|
355 for (cl = p->out; cl; cl = cl->next) {
|
32
|
356 ngx_log_debug8(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
|
357 "pipe buf out s:%d t:%d f:%d "
|
|
358 "%p, pos %p, size: %z "
|
|
359 "file: %O, size: %z",
|
|
360 (cl->buf->shadow ? 1 : 0),
|
|
361 cl->buf->temporary, cl->buf->in_file,
|
|
362 cl->buf->start, cl->buf->pos,
|
|
363 cl->buf->last - cl->buf->pos,
|
|
364 cl->buf->file_pos,
|
|
365 cl->buf->file_last - cl->buf->file_pos);
|
2
|
366 }
|
|
367
|
|
368 for (cl = p->in; cl; cl = cl->next) {
|
32
|
369 ngx_log_debug8(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
|
370 "pipe buf in s:%d t:%d f:%d "
|
|
371 "%p, pos %p, size: %z "
|
|
372 "file: %O, size: %z",
|
|
373 (cl->buf->shadow ? 1 : 0),
|
|
374 cl->buf->temporary, cl->buf->in_file,
|
0
|
375 cl->buf->start, cl->buf->pos,
|
32
|
376 cl->buf->last - cl->buf->pos,
|
|
377 cl->buf->file_pos,
|
|
378 cl->buf->file_last - cl->buf->file_pos);
|
0
|
379 }
|
|
380
|
|
381 for (cl = p->free_raw_bufs; cl; cl = cl->next) {
|
32
|
382 ngx_log_debug8(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
|
383 "pipe buf free s:%d t:%d f:%d "
|
|
384 "%p, pos %p, size: %z "
|
|
385 "file: %O, size: %z",
|
|
386 (cl->buf->shadow ? 1 : 0),
|
|
387 cl->buf->temporary, cl->buf->in_file,
|
|
388 cl->buf->start, cl->buf->pos,
|
|
389 cl->buf->last - cl->buf->pos,
|
|
390 cl->buf->file_pos,
|
|
391 cl->buf->file_last - cl->buf->file_pos);
|
0
|
392 }
|
|
393
|
640
|
394 ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
|
395 "pipe length: %O", p->length);
|
|
396
|
0
|
397 #endif
|
|
398
|
640
|
399 if (p->free_raw_bufs && p->length != -1) {
|
|
400 cl = p->free_raw_bufs;
|
|
401
|
|
402 if (cl->buf->last - cl->buf->pos >= p->length) {
|
|
403
|
|
404 /* STUB */ cl->buf->num = p->num++;
|
|
405
|
|
406 if (p->input_filter(p, cl->buf) == NGX_ERROR) {
|
|
407 return NGX_ABORT;
|
|
408 }
|
|
409
|
|
410 p->free_raw_bufs = cl->next;
|
|
411 ngx_free_chain(p->pool, cl);
|
|
412 }
|
|
413 }
|
|
414
|
|
415 if (p->length == 0) {
|
|
416 p->upstream_done = 1;
|
|
417 p->read = 1;
|
|
418 }
|
|
419
|
0
|
420 if ((p->upstream_eof || p->upstream_error) && p->free_raw_bufs) {
|
|
421
|
|
422 /* STUB */ p->free_raw_bufs->buf->num = p->num++;
|
|
423
|
|
424 if (p->input_filter(p, p->free_raw_bufs->buf) == NGX_ERROR) {
|
|
425 return NGX_ABORT;
|
|
426 }
|
|
427
|
|
428 p->free_raw_bufs = p->free_raw_bufs->next;
|
|
429
|
486
|
430 if (p->free_bufs && p->buf_to_file == NULL) {
|
0
|
431 for (cl = p->free_raw_bufs; cl; cl = cl->next) {
|
2
|
432 if (cl->buf->shadow == NULL) {
|
126
|
433 ngx_pfree(p->pool, cl->buf->start);
|
2
|
434 }
|
0
|
435 }
|
|
436 }
|
|
437 }
|
|
438
|
340
|
439 if (p->cacheable && p->in) {
|
0
|
440 if (ngx_event_pipe_write_chain_to_temp_file(p) == NGX_ABORT) {
|
|
441 return NGX_ABORT;
|
|
442 }
|
|
443 }
|
|
444
|
|
445 return NGX_OK;
|
|
446 }
|
|
447
|
|
448
|
126
|
449 static ngx_int_t
|
|
450 ngx_event_pipe_write_to_downstream(ngx_event_pipe_t *p)
|
0
|
451 {
|
284
|
452 u_char *prev;
|
132
|
453 size_t bsize;
|
346
|
454 ngx_int_t rc;
|
518
|
455 ngx_uint_t flush, flushed, prev_last_shadow;
|
464
|
456 ngx_chain_t *out, **ll, *cl, file;
|
132
|
457 ngx_connection_t *downstream;
|
|
458
|
|
459 downstream = p->downstream;
|
0
|
460
|
|
461 ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
132
|
462 "pipe write downstream: %d", downstream->write->ready);
|
0
|
463
|
518
|
464 flushed = 0;
|
|
465
|
0
|
466 for ( ;; ) {
|
|
467 if (p->downstream_error) {
|
|
468 return ngx_event_pipe_drain_chains(p);
|
|
469 }
|
|
470
|
|
471 if (p->upstream_eof || p->upstream_error || p->upstream_done) {
|
|
472
|
|
473 /* pass the p->out and p->in chains to the output filter */
|
|
474
|
58
|
475 for (cl = p->busy; cl; cl = cl->next) {
|
|
476 cl->buf->recycled = 0;
|
|
477 }
|
|
478
|
0
|
479 if (p->out) {
|
|
480 ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
|
481 "pipe write downstream flush out");
|
|
482
|
58
|
483 for (cl = p->out; cl; cl = cl->next) {
|
|
484 cl->buf->recycled = 0;
|
|
485 }
|
|
486
|
346
|
487 rc = p->output_filter(p->output_ctx, p->out);
|
|
488
|
|
489 if (rc == NGX_ERROR) {
|
0
|
490 p->downstream_error = 1;
|
|
491 return ngx_event_pipe_drain_chains(p);
|
|
492 }
|
|
493
|
|
494 p->out = NULL;
|
|
495 }
|
|
496
|
|
497 if (p->in) {
|
|
498 ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
|
499 "pipe write downstream flush in");
|
|
500
|
58
|
501 for (cl = p->in; cl; cl = cl->next) {
|
|
502 cl->buf->recycled = 0;
|
|
503 }
|
|
504
|
346
|
505 rc = p->output_filter(p->output_ctx, p->in);
|
132
|
506
|
346
|
507 if (rc == NGX_ERROR) {
|
0
|
508 p->downstream_error = 1;
|
|
509 return ngx_event_pipe_drain_chains(p);
|
|
510 }
|
|
511
|
|
512 p->in = NULL;
|
|
513 }
|
|
514
|
464
|
515 if (p->cacheable && p->buf_to_file) {
|
|
516
|
|
517 file.buf = p->buf_to_file;
|
|
518 file.next = NULL;
|
|
519
|
|
520 if (ngx_write_chain_to_temp_file(p->temp_file, &file)
|
|
521 == NGX_ERROR)
|
|
522 {
|
|
523 return NGX_ABORT;
|
|
524 }
|
|
525 }
|
|
526
|
0
|
527 ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
|
528 "pipe write downstream done");
|
|
529
|
|
530 /* TODO: free unused bufs */
|
|
531
|
|
532 p->downstream_done = 1;
|
|
533 break;
|
|
534 }
|
|
535
|
132
|
536 if (downstream->data != p->output_ctx
|
|
537 || !downstream->write->ready
|
|
538 || downstream->write->delayed)
|
58
|
539 {
|
0
|
540 break;
|
|
541 }
|
|
542
|
28
|
543 /* bsize is the size of the busy recycled bufs */
|
0
|
544
|
284
|
545 prev = NULL;
|
0
|
546 bsize = 0;
|
|
547
|
|
548 for (cl = p->busy; cl; cl = cl->next) {
|
284
|
549
|
28
|
550 if (cl->buf->recycled) {
|
284
|
551 if (prev == cl->buf->start) {
|
|
552 continue;
|
|
553 }
|
|
554
|
28
|
555 bsize += cl->buf->end - cl->buf->start;
|
284
|
556 prev = cl->buf->start;
|
28
|
557 }
|
0
|
558 }
|
|
559
|
|
560 ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
28
|
561 "pipe write busy: %uz", bsize);
|
0
|
562
|
|
563 out = NULL;
|
284
|
564
|
|
565 if (bsize >= (size_t) p->busy_size) {
|
|
566 flush = 1;
|
|
567 goto flush;
|
|
568 }
|
|
569
|
|
570 flush = 0;
|
0
|
571 ll = NULL;
|
276
|
572 prev_last_shadow = 1;
|
0
|
573
|
|
574 for ( ;; ) {
|
|
575 if (p->out) {
|
|
576 cl = p->out;
|
|
577
|
646
|
578 if (cl->buf->recycled) {
|
|
579 ngx_log_error(NGX_LOG_ALERT, p->log, 0,
|
|
580 "recycled buffer in pipe out chain");
|
0
|
581 }
|
|
582
|
|
583 p->out = p->out->next;
|
28
|
584
|
340
|
585 } else if (!p->cacheable && p->in) {
|
0
|
586 cl = p->in;
|
|
587
|
274
|
588 ngx_log_debug3(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
|
589 "pipe write buf ls:%d %p %z",
|
|
590 cl->buf->last_shadow,
|
|
591 cl->buf->pos,
|
|
592 cl->buf->last - cl->buf->pos);
|
|
593
|
646
|
594 if (cl->buf->recycled && prev_last_shadow) {
|
|
595 if (bsize + cl->buf->end - cl->buf->start > p->busy_size) {
|
|
596 flush = 1;
|
|
597 break;
|
276
|
598 }
|
|
599
|
646
|
600 bsize += cl->buf->end - cl->buf->start;
|
0
|
601 }
|
|
602
|
276
|
603 prev_last_shadow = cl->buf->last_shadow;
|
|
604
|
0
|
605 p->in = p->in->next;
|
|
606
|
|
607 } else {
|
|
608 break;
|
|
609 }
|
|
610
|
|
611 cl->next = NULL;
|
50
|
612
|
|
613 if (out) {
|
|
614 *ll = cl;
|
|
615 } else {
|
|
616 out = cl;
|
|
617 }
|
|
618 ll = &cl->next;
|
0
|
619 }
|
|
620
|
284
|
621 flush:
|
|
622
|
0
|
623 ngx_log_debug2(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
10
|
624 "pipe write: out:%p, f:%d", out, flush);
|
0
|
625
|
518
|
626 if (out == NULL) {
|
|
627
|
|
628 if (!flush) {
|
|
629 break;
|
|
630 }
|
|
631
|
|
632 /* a workaround for AIO */
|
|
633 if (flushed++ > 10) {
|
|
634 return NGX_BUSY;
|
|
635 }
|
0
|
636 }
|
|
637
|
346
|
638 rc = p->output_filter(p->output_ctx, out);
|
|
639
|
640
|
640 ngx_chain_update_chains(p->pool, &p->free, &p->busy, &out, p->tag);
|
|
641
|
346
|
642 if (rc == NGX_ERROR) {
|
0
|
643 p->downstream_error = 1;
|
|
644 return ngx_event_pipe_drain_chains(p);
|
|
645 }
|
|
646
|
|
647 for (cl = p->free; cl; cl = cl->next) {
|
|
648
|
|
649 if (cl->buf->temp_file) {
|
340
|
650 if (p->cacheable || !p->cyclic_temp_file) {
|
0
|
651 continue;
|
|
652 }
|
|
653
|
|
654 /* reset p->temp_offset if all bufs had been sent */
|
|
655
|
|
656 if (cl->buf->file_last == p->temp_file->offset) {
|
|
657 p->temp_file->offset = 0;
|
|
658 }
|
|
659 }
|
|
660
|
|
661 /* TODO: free buf if p->free_bufs && upstream done */
|
|
662
|
|
663 /* add the free shadow raw buf to p->free_raw_bufs */
|
|
664
|
|
665 if (cl->buf->last_shadow) {
|
32
|
666 if (ngx_event_pipe_add_free_buf(p, cl->buf->shadow) != NGX_OK) {
|
|
667 return NGX_ABORT;
|
|
668 }
|
0
|
669
|
|
670 cl->buf->last_shadow = 0;
|
|
671 }
|
|
672
|
|
673 cl->buf->shadow = NULL;
|
|
674 }
|
|
675 }
|
|
676
|
|
677 return NGX_OK;
|
|
678 }
|
|
679
|
|
680
|
126
|
681 static ngx_int_t
|
|
682 ngx_event_pipe_write_chain_to_temp_file(ngx_event_pipe_t *p)
|
0
|
683 {
|
646
|
684 ssize_t size, bsize, n;
|
0
|
685 ngx_buf_t *b;
|
646
|
686 ngx_uint_t prev_last_shadow;
|
|
687 ngx_chain_t *cl, *tl, *next, *out, **ll, **last_out, **last_free, fl;
|
0
|
688
|
|
689 if (p->buf_to_file) {
|
|
690 fl.buf = p->buf_to_file;
|
|
691 fl.next = p->in;
|
|
692 out = &fl;
|
|
693
|
|
694 } else {
|
|
695 out = p->in;
|
|
696 }
|
|
697
|
340
|
698 if (!p->cacheable) {
|
0
|
699
|
|
700 size = 0;
|
|
701 cl = out;
|
|
702 ll = NULL;
|
646
|
703 prev_last_shadow = 1;
|
0
|
704
|
|
705 ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
10
|
706 "pipe offset: %O", p->temp_file->offset);
|
0
|
707
|
|
708 do {
|
|
709 bsize = cl->buf->last - cl->buf->pos;
|
|
710
|
646
|
711 ngx_log_debug4(NGX_LOG_DEBUG_EVENT, p->log, 0,
|
|
712 "pipe buf ls:%d %p, pos %p, size: %z",
|
|
713 cl->buf->last_shadow, cl->buf->start,
|
|
714 cl->buf->pos, bsize);
|
0
|
715
|
646
|
716 if (prev_last_shadow
|
|
717 && ((size + bsize > p->temp_file_write_size)
|
|
718 || (p->temp_file->offset + size + bsize
|
|
719 > p->max_temp_file_size)))
|
0
|
720 {
|
|
721 break;
|
|
722 }
|
|
723
|
646
|
724 prev_last_shadow = cl->buf->last_shadow;
|
|
725
|
0
|
726 size += bsize;
|
|
727 ll = &cl->next;
|
|
728 cl = cl->next;
|
|
729
|
|
730 } while (cl);
|
|
731
|
10
|
732 ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0, "size: %z", size);
|
0
|
733
|
70
|
734 if (ll == NULL) {
|
|
735 return NGX_BUSY;
|
|
736 }
|
|
737
|
0
|
738 if (cl) {
|
|
739 p->in = cl;
|
|
740 *ll = NULL;
|
|
741
|
|
742 } else {
|
|
743 p->in = NULL;
|
|
744 p->last_in = &p->in;
|
|
745 }
|
|
746
|
|
747 } else {
|
|
748 p->in = NULL;
|
|
749 p->last_in = &p->in;
|
|
750 }
|
|
751
|
646
|
752 n = ngx_write_chain_to_temp_file(p->temp_file, out);
|
|
753
|
|
754 if (n == NGX_ERROR) {
|
0
|
755 return NGX_ABORT;
|
|
756 }
|
|
757
|
646
|
758 if (p->buf_to_file) {
|
|
759 p->temp_file->offset = p->buf_to_file->last - p->buf_to_file->pos;
|
|
760 n -= p->buf_to_file->last - p->buf_to_file->pos;
|
|
761 p->buf_to_file = NULL;
|
|
762 out = out->next;
|
|
763 }
|
|
764
|
|
765 if (n > 0) {
|
|
766 /* update previous buffer or add new buffer */
|
|
767
|
|
768 if (p->out) {
|
|
769 for (cl = p->out; cl->next; cl = cl->next) { /* void */ }
|
|
770
|
|
771 b = cl->buf;
|
|
772
|
|
773 if (b->file_last == p->temp_file->offset) {
|
|
774 p->temp_file->offset += n;
|
|
775 b->file_last = p->temp_file->offset;
|
|
776 goto free;
|
|
777 }
|
|
778
|
|
779 last_out = &cl->next;
|
|
780
|
|
781 } else {
|
|
782 last_out = &p->out;
|
|
783 }
|
|
784
|
|
785 cl = ngx_chain_get_free_buf(p->pool, &p->free);
|
|
786 if (cl == NULL) {
|
|
787 return NGX_ABORT;
|
|
788 }
|
|
789
|
|
790 b = cl->buf;
|
|
791
|
|
792 ngx_memzero(b, sizeof(ngx_buf_t));
|
|
793
|
|
794 b->tag = p->tag;
|
|
795
|
|
796 b->file = &p->temp_file->file;
|
|
797 b->file_pos = p->temp_file->offset;
|
|
798 p->temp_file->offset += n;
|
|
799 b->file_last = p->temp_file->offset;
|
|
800
|
|
801 b->in_file = 1;
|
|
802 b->temp_file = 1;
|
|
803
|
|
804 *last_out = cl;
|
|
805 }
|
|
806
|
|
807 free:
|
|
808
|
0
|
809 for (last_free = &p->free_raw_bufs;
|
|
810 *last_free != NULL;
|
|
811 last_free = &(*last_free)->next)
|
|
812 {
|
|
813 /* void */
|
|
814 }
|
|
815
|
|
816 for (cl = out; cl; cl = next) {
|
|
817 next = cl->next;
|
646
|
818
|
|
819 cl->next = p->free;
|
|
820 p->free = cl;
|
0
|
821
|
|
822 b = cl->buf;
|
|
823
|
|
824 if (b->last_shadow) {
|
32
|
825
|
50
|
826 tl = ngx_alloc_chain_link(p->pool);
|
|
827 if (tl == NULL) {
|
32
|
828 return NGX_ABORT;
|
|
829 }
|
|
830
|
|
831 tl->buf = b->shadow;
|
|
832 tl->next = NULL;
|
|
833
|
|
834 *last_free = tl;
|
|
835 last_free = &tl->next;
|
|
836
|
2
|
837 b->shadow->pos = b->shadow->start;
|
|
838 b->shadow->last = b->shadow->start;
|
|
839
|
32
|
840 ngx_event_pipe_remove_shadow_links(b->shadow);
|
0
|
841 }
|
|
842 }
|
|
843
|
|
844 return NGX_OK;
|
|
845 }
|
|
846
|
|
847
|
|
848 /* the copy input filter */
|
|
849
|
126
|
850 ngx_int_t
|
|
851 ngx_event_pipe_copy_input_filter(ngx_event_pipe_t *p, ngx_buf_t *buf)
|
0
|
852 {
|
|
853 ngx_buf_t *b;
|
|
854 ngx_chain_t *cl;
|
|
855
|
|
856 if (buf->pos == buf->last) {
|
|
857 return NGX_OK;
|
|
858 }
|
|
859
|
|
860 if (p->free) {
|
56
|
861 cl = p->free;
|
|
862 b = cl->buf;
|
|
863 p->free = cl->next;
|
|
864 ngx_free_chain(p->pool, cl);
|
0
|
865
|
|
866 } else {
|
50
|
867 b = ngx_alloc_buf(p->pool);
|
|
868 if (b == NULL) {
|
0
|
869 return NGX_ERROR;
|
|
870 }
|
|
871 }
|
|
872
|
|
873 ngx_memcpy(b, buf, sizeof(ngx_buf_t));
|
|
874 b->shadow = buf;
|
|
875 b->tag = p->tag;
|
|
876 b->last_shadow = 1;
|
|
877 b->recycled = 1;
|
|
878 buf->shadow = b;
|
|
879
|
50
|
880 cl = ngx_alloc_chain_link(p->pool);
|
|
881 if (cl == NULL) {
|
32
|
882 return NGX_ERROR;
|
|
883 }
|
|
884
|
|
885 cl->buf = b;
|
|
886 cl->next = NULL;
|
0
|
887
|
28
|
888 ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0, "input buf #%d", b->num);
|
0
|
889
|
50
|
890 if (p->in) {
|
|
891 *p->last_in = cl;
|
|
892 } else {
|
|
893 p->in = cl;
|
|
894 }
|
|
895 p->last_in = &cl->next;
|
0
|
896
|
640
|
897 if (p->length == -1) {
|
|
898 return NGX_OK;
|
|
899 }
|
|
900
|
|
901 p->length -= b->last - b->pos;
|
|
902
|
0
|
903 return NGX_OK;
|
|
904 }
|
|
905
|
|
906
|
126
|
907 static ngx_inline void
|
|
908 ngx_event_pipe_remove_shadow_links(ngx_buf_t *buf)
|
0
|
909 {
|
|
910 ngx_buf_t *b, *next;
|
|
911
|
2
|
912 b = buf->shadow;
|
|
913
|
|
914 if (b == NULL) {
|
0
|
915 return;
|
|
916 }
|
|
917
|
|
918 while (!b->last_shadow) {
|
|
919 next = b->shadow;
|
|
920
|
2
|
921 b->temporary = 0;
|
|
922 b->recycled = 0;
|
0
|
923
|
|
924 b->shadow = NULL;
|
|
925 b = next;
|
|
926 }
|
|
927
|
2
|
928 b->temporary = 0;
|
|
929 b->recycled = 0;
|
0
|
930 b->last_shadow = 0;
|
|
931
|
|
932 b->shadow = NULL;
|
|
933
|
|
934 buf->shadow = NULL;
|
|
935 }
|
|
936
|
|
937
|
126
|
938 ngx_int_t
|
|
939 ngx_event_pipe_add_free_buf(ngx_event_pipe_t *p, ngx_buf_t *b)
|
0
|
940 {
|
32
|
941 ngx_chain_t *cl;
|
|
942
|
50
|
943 cl = ngx_alloc_chain_link(p->pool);
|
|
944 if (cl == NULL) {
|
32
|
945 return NGX_ERROR;
|
0
|
946 }
|
|
947
|
32
|
948 b->pos = b->start;
|
|
949 b->last = b->start;
|
|
950 b->shadow = NULL;
|
|
951
|
|
952 cl->buf = b;
|
|
953
|
|
954 if (p->free_raw_bufs == NULL) {
|
|
955 p->free_raw_bufs = cl;
|
|
956 cl->next = NULL;
|
|
957
|
|
958 return NGX_OK;
|
|
959 }
|
|
960
|
|
961 if (p->free_raw_bufs->buf->pos == p->free_raw_bufs->buf->last) {
|
0
|
962
|
32
|
963 /* add the free buf to the list start */
|
|
964
|
|
965 cl->next = p->free_raw_bufs;
|
|
966 p->free_raw_bufs = cl;
|
|
967
|
|
968 return NGX_OK;
|
0
|
969 }
|
32
|
970
|
|
971 /* the first free buf is partialy filled, thus add the free buf after it */
|
|
972
|
|
973 cl->next = p->free_raw_bufs->next;
|
|
974 p->free_raw_bufs->next = cl;
|
|
975
|
|
976 return NGX_OK;
|
0
|
977 }
|
|
978
|
|
979
|
126
|
980 static ngx_int_t
|
|
981 ngx_event_pipe_drain_chains(ngx_event_pipe_t *p)
|
0
|
982 {
|
|
983 ngx_chain_t *cl, *tl;
|
|
984
|
|
985 for ( ;; ) {
|
|
986 if (p->busy) {
|
|
987 cl = p->busy;
|
|
988 p->busy = NULL;
|
|
989
|
|
990 } else if (p->out) {
|
|
991 cl = p->out;
|
|
992 p->out = NULL;
|
|
993
|
|
994 } else if (p->in) {
|
|
995 cl = p->in;
|
|
996 p->in = NULL;
|
|
997
|
|
998 } else {
|
|
999 return NGX_OK;
|
|
1000 }
|
|
1001
|
|
1002 while (cl) {
|
|
1003 if (cl->buf->last_shadow) {
|
32
|
1004 if (ngx_event_pipe_add_free_buf(p, cl->buf->shadow) != NGX_OK) {
|
28
|
1005 return NGX_ABORT;
|
|
1006 }
|
|
1007
|
0
|
1008 cl->buf->last_shadow = 0;
|
|
1009 }
|
|
1010
|
|
1011 cl->buf->shadow = NULL;
|
|
1012 tl = cl->next;
|
|
1013 cl->next = p->free;
|
|
1014 p->free = cl;
|
|
1015 cl = tl;
|
|
1016 }
|
|
1017 }
|
|
1018 }
|