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