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