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