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