1e18a033bSKonstantin Ananyev
2e18a033bSKonstantin Ananyev/*
3e18a033bSKonstantin Ananyev * Copyright (C) Igor Sysoev
4e18a033bSKonstantin Ananyev * Copyright (C) Nginx, Inc.
5e18a033bSKonstantin Ananyev */
6e18a033bSKonstantin Ananyev
7e18a033bSKonstantin Ananyev
8e18a033bSKonstantin Ananyev#include <ngx_config.h>
9e18a033bSKonstantin Ananyev#include <ngx_core.h>
10e18a033bSKonstantin Ananyev
11e18a033bSKonstantin Ananyev
12e18a033bSKonstantin Ananyevngx_buf_t *
13e18a033bSKonstantin Ananyevngx_create_temp_buf(ngx_pool_t *pool, size_t size)
14e18a033bSKonstantin Ananyev{
15e18a033bSKonstantin Ananyev    ngx_buf_t *b;
16e18a033bSKonstantin Ananyev
17e18a033bSKonstantin Ananyev    b = ngx_calloc_buf(pool);
18e18a033bSKonstantin Ananyev    if (b == NULL) {
19e18a033bSKonstantin Ananyev        return NULL;
20e18a033bSKonstantin Ananyev    }
21e18a033bSKonstantin Ananyev
22e18a033bSKonstantin Ananyev    b->start = ngx_palloc(pool, size);
23e18a033bSKonstantin Ananyev    if (b->start == NULL) {
24e18a033bSKonstantin Ananyev        return NULL;
25e18a033bSKonstantin Ananyev    }
26e18a033bSKonstantin Ananyev
27e18a033bSKonstantin Ananyev    /*
28e18a033bSKonstantin Ananyev     * set by ngx_calloc_buf():
29e18a033bSKonstantin Ananyev     *
30e18a033bSKonstantin Ananyev     *     b->file_pos = 0;
31e18a033bSKonstantin Ananyev     *     b->file_last = 0;
32e18a033bSKonstantin Ananyev     *     b->file = NULL;
33e18a033bSKonstantin Ananyev     *     b->shadow = NULL;
34e18a033bSKonstantin Ananyev     *     b->tag = 0;
35e18a033bSKonstantin Ananyev     *     and flags
36e18a033bSKonstantin Ananyev     */
37e18a033bSKonstantin Ananyev
38e18a033bSKonstantin Ananyev    b->pos = b->start;
39e18a033bSKonstantin Ananyev    b->last = b->start;
40e18a033bSKonstantin Ananyev    b->end = b->last + size;
41e18a033bSKonstantin Ananyev    b->temporary = 1;
42e18a033bSKonstantin Ananyev
43e18a033bSKonstantin Ananyev    return b;
44e18a033bSKonstantin Ananyev}
45e18a033bSKonstantin Ananyev
46e18a033bSKonstantin Ananyev
47e18a033bSKonstantin Ananyevngx_chain_t *
48e18a033bSKonstantin Ananyevngx_alloc_chain_link(ngx_pool_t *pool)
49e18a033bSKonstantin Ananyev{
50e18a033bSKonstantin Ananyev    ngx_chain_t  *cl;
51e18a033bSKonstantin Ananyev
52e18a033bSKonstantin Ananyev    cl = pool->chain;
53e18a033bSKonstantin Ananyev
54e18a033bSKonstantin Ananyev    if (cl) {
55e18a033bSKonstantin Ananyev        pool->chain = cl->next;
56e18a033bSKonstantin Ananyev        return cl;
57e18a033bSKonstantin Ananyev    }
58e18a033bSKonstantin Ananyev
59e18a033bSKonstantin Ananyev    cl = ngx_palloc(pool, sizeof(ngx_chain_t));
60e18a033bSKonstantin Ananyev    if (cl == NULL) {
61e18a033bSKonstantin Ananyev        return NULL;
62e18a033bSKonstantin Ananyev    }
63e18a033bSKonstantin Ananyev
64e18a033bSKonstantin Ananyev    return cl;
65e18a033bSKonstantin Ananyev}
66e18a033bSKonstantin Ananyev
67e18a033bSKonstantin Ananyev
68e18a033bSKonstantin Ananyevngx_chain_t *
69e18a033bSKonstantin Ananyevngx_create_chain_of_bufs(ngx_pool_t *pool, ngx_bufs_t *bufs)
70e18a033bSKonstantin Ananyev{
71e18a033bSKonstantin Ananyev    u_char       *p;
72e18a033bSKonstantin Ananyev    ngx_int_t     i;
73e18a033bSKonstantin Ananyev    ngx_buf_t    *b;
74e18a033bSKonstantin Ananyev    ngx_chain_t  *chain, *cl, **ll;
75e18a033bSKonstantin Ananyev
76e18a033bSKonstantin Ananyev    p = ngx_palloc(pool, bufs->num * bufs->size);
77e18a033bSKonstantin Ananyev    if (p == NULL) {
78e18a033bSKonstantin Ananyev        return NULL;
79e18a033bSKonstantin Ananyev    }
80e18a033bSKonstantin Ananyev
81e18a033bSKonstantin Ananyev    ll = &chain;
82e18a033bSKonstantin Ananyev
83e18a033bSKonstantin Ananyev    for (i = 0; i < bufs->num; i++) {
84e18a033bSKonstantin Ananyev
85e18a033bSKonstantin Ananyev        b = ngx_calloc_buf(pool);
86e18a033bSKonstantin Ananyev        if (b == NULL) {
87e18a033bSKonstantin Ananyev            return NULL;
88e18a033bSKonstantin Ananyev        }
89e18a033bSKonstantin Ananyev
90e18a033bSKonstantin Ananyev        /*
91e18a033bSKonstantin Ananyev         * set by ngx_calloc_buf():
92e18a033bSKonstantin Ananyev         *
93e18a033bSKonstantin Ananyev         *     b->file_pos = 0;
94e18a033bSKonstantin Ananyev         *     b->file_last = 0;
95e18a033bSKonstantin Ananyev         *     b->file = NULL;
96e18a033bSKonstantin Ananyev         *     b->shadow = NULL;
97e18a033bSKonstantin Ananyev         *     b->tag = 0;
98e18a033bSKonstantin Ananyev         *     and flags
99e18a033bSKonstantin Ananyev         *
100e18a033bSKonstantin Ananyev         */
101e18a033bSKonstantin Ananyev
102e18a033bSKonstantin Ananyev        b->pos = p;
103e18a033bSKonstantin Ananyev        b->last = p;
104e18a033bSKonstantin Ananyev        b->temporary = 1;
105e18a033bSKonstantin Ananyev
106e18a033bSKonstantin Ananyev        b->start = p;
107e18a033bSKonstantin Ananyev        p += bufs->size;
108e18a033bSKonstantin Ananyev        b->end = p;
109e18a033bSKonstantin Ananyev
110e18a033bSKonstantin Ananyev        cl = ngx_alloc_chain_link(pool);
111e18a033bSKonstantin Ananyev        if (cl == NULL) {
112e18a033bSKonstantin Ananyev            return NULL;
113e18a033bSKonstantin Ananyev        }
114e18a033bSKonstantin Ananyev
115e18a033bSKonstantin Ananyev        cl->buf = b;
116e18a033bSKonstantin Ananyev        *ll = cl;
117e18a033bSKonstantin Ananyev        ll = &cl->next;
118e18a033bSKonstantin Ananyev    }
119e18a033bSKonstantin Ananyev
120e18a033bSKonstantin Ananyev    *ll = NULL;
121e18a033bSKonstantin Ananyev
122e18a033bSKonstantin Ananyev    return chain;
123e18a033bSKonstantin Ananyev}
124e18a033bSKonstantin Ananyev
125e18a033bSKonstantin Ananyev
126e18a033bSKonstantin Ananyevngx_int_t
127e18a033bSKonstantin Ananyevngx_chain_add_copy(ngx_pool_t *pool, ngx_chain_t **chain, ngx_chain_t *in)
128e18a033bSKonstantin Ananyev{
129e18a033bSKonstantin Ananyev    ngx_chain_t  *cl, **ll;
130e18a033bSKonstantin Ananyev
131e18a033bSKonstantin Ananyev    ll = chain;
132e18a033bSKonstantin Ananyev
133e18a033bSKonstantin Ananyev    for (cl = *chain; cl; cl = cl->next) {
134e18a033bSKonstantin Ananyev        ll = &cl->next;
135e18a033bSKonstantin Ananyev    }
136e18a033bSKonstantin Ananyev
137e18a033bSKonstantin Ananyev    while (in) {
138e18a033bSKonstantin Ananyev        cl = ngx_alloc_chain_link(pool);
139e18a033bSKonstantin Ananyev        if (cl == NULL) {
140e18a033bSKonstantin Ananyev            return NGX_ERROR;
141e18a033bSKonstantin Ananyev        }
142e18a033bSKonstantin Ananyev
143e18a033bSKonstantin Ananyev        cl->buf = in->buf;
144e18a033bSKonstantin Ananyev        *ll = cl;
145e18a033bSKonstantin Ananyev        ll = &cl->next;
146e18a033bSKonstantin Ananyev        in = in->next;
147e18a033bSKonstantin Ananyev    }
148e18a033bSKonstantin Ananyev
149e18a033bSKonstantin Ananyev    *ll = NULL;
150e18a033bSKonstantin Ananyev
151e18a033bSKonstantin Ananyev    return NGX_OK;
152e18a033bSKonstantin Ananyev}
153e18a033bSKonstantin Ananyev
154e18a033bSKonstantin Ananyev
155e18a033bSKonstantin Ananyevngx_chain_t *
156e18a033bSKonstantin Ananyevngx_chain_get_free_buf(ngx_pool_t *p, ngx_chain_t **free)
157e18a033bSKonstantin Ananyev{
158e18a033bSKonstantin Ananyev    ngx_chain_t  *cl;
159e18a033bSKonstantin Ananyev
160e18a033bSKonstantin Ananyev    if (*free) {
161e18a033bSKonstantin Ananyev        cl = *free;
162e18a033bSKonstantin Ananyev        *free = cl->next;
163e18a033bSKonstantin Ananyev        cl->next = NULL;
164e18a033bSKonstantin Ananyev        return cl;
165e18a033bSKonstantin Ananyev    }
166e18a033bSKonstantin Ananyev
167e18a033bSKonstantin Ananyev    cl = ngx_alloc_chain_link(p);
168e18a033bSKonstantin Ananyev    if (cl == NULL) {
169e18a033bSKonstantin Ananyev        return NULL;
170e18a033bSKonstantin Ananyev    }
171e18a033bSKonstantin Ananyev
172e18a033bSKonstantin Ananyev    cl->buf = ngx_calloc_buf(p);
173e18a033bSKonstantin Ananyev    if (cl->buf == NULL) {
174e18a033bSKonstantin Ananyev        return NULL;
175e18a033bSKonstantin Ananyev    }
176e18a033bSKonstantin Ananyev
177e18a033bSKonstantin Ananyev    cl->next = NULL;
178e18a033bSKonstantin Ananyev
179e18a033bSKonstantin Ananyev    return cl;
180e18a033bSKonstantin Ananyev}
181e18a033bSKonstantin Ananyev
182e18a033bSKonstantin Ananyev
183e18a033bSKonstantin Ananyevvoid
184e18a033bSKonstantin Ananyevngx_chain_update_chains(ngx_pool_t *p, ngx_chain_t **free, ngx_chain_t **busy,
185e18a033bSKonstantin Ananyev    ngx_chain_t **out, ngx_buf_tag_t tag)
186e18a033bSKonstantin Ananyev{
187e18a033bSKonstantin Ananyev    ngx_chain_t  *cl;
188e18a033bSKonstantin Ananyev
189e18a033bSKonstantin Ananyev    if (*out) {
190e18a033bSKonstantin Ananyev        if (*busy == NULL) {
191e18a033bSKonstantin Ananyev            *busy = *out;
192e18a033bSKonstantin Ananyev
193e18a033bSKonstantin Ananyev        } else {
194e18a033bSKonstantin Ananyev            for (cl = *busy; cl->next; cl = cl->next) { /* void */ }
195e18a033bSKonstantin Ananyev
196e18a033bSKonstantin Ananyev            cl->next = *out;
197e18a033bSKonstantin Ananyev        }
198e18a033bSKonstantin Ananyev
199e18a033bSKonstantin Ananyev        *out = NULL;
200e18a033bSKonstantin Ananyev    }
201e18a033bSKonstantin Ananyev
202e18a033bSKonstantin Ananyev    while (*busy) {
203e18a033bSKonstantin Ananyev        cl = *busy;
204e18a033bSKonstantin Ananyev
205e18a033bSKonstantin Ananyev        if (ngx_buf_size(cl->buf) != 0) {
206e18a033bSKonstantin Ananyev            break;
207e18a033bSKonstantin Ananyev        }
208e18a033bSKonstantin Ananyev
209e18a033bSKonstantin Ananyev        if (cl->buf->tag != tag) {
210e18a033bSKonstantin Ananyev            *busy = cl->next;
211e18a033bSKonstantin Ananyev            ngx_free_chain(p, cl);
212e18a033bSKonstantin Ananyev            continue;
213e18a033bSKonstantin Ananyev        }
214e18a033bSKonstantin Ananyev
215e18a033bSKonstantin Ananyev        cl->buf->pos = cl->buf->start;
216e18a033bSKonstantin Ananyev        cl->buf->last = cl->buf->start;
217e18a033bSKonstantin Ananyev
218e18a033bSKonstantin Ananyev        *busy = cl->next;
219e18a033bSKonstantin Ananyev        cl->next = *free;
220e18a033bSKonstantin Ananyev        *free = cl;
221e18a033bSKonstantin Ananyev    }
222e18a033bSKonstantin Ananyev}
223e18a033bSKonstantin Ananyev
224e18a033bSKonstantin Ananyev
225e18a033bSKonstantin Ananyevoff_t
226e18a033bSKonstantin Ananyevngx_chain_coalesce_file(ngx_chain_t **in, off_t limit)
227e18a033bSKonstantin Ananyev{
228e18a033bSKonstantin Ananyev    off_t         total, size, aligned, fprev;
229e18a033bSKonstantin Ananyev    ngx_fd_t      fd;
230e18a033bSKonstantin Ananyev    ngx_chain_t  *cl;
231e18a033bSKonstantin Ananyev
232e18a033bSKonstantin Ananyev    total = 0;
233e18a033bSKonstantin Ananyev
234e18a033bSKonstantin Ananyev    cl = *in;
235e18a033bSKonstantin Ananyev    fd = cl->buf->file->fd;
236e18a033bSKonstantin Ananyev
237e18a033bSKonstantin Ananyev    do {
238e18a033bSKonstantin Ananyev        size = cl->buf->file_last - cl->buf->file_pos;
239e18a033bSKonstantin Ananyev
240e18a033bSKonstantin Ananyev        if (size > limit - total) {
241e18a033bSKonstantin Ananyev            size = limit - total;
242e18a033bSKonstantin Ananyev
243e18a033bSKonstantin Ananyev            aligned = (cl->buf->file_pos + size + ngx_pagesize - 1)
244e18a033bSKonstantin Ananyev                       & ~((off_t) ngx_pagesize - 1);
245e18a033bSKonstantin Ananyev
246e18a033bSKonstantin Ananyev            if (aligned <= cl->buf->file_last) {
247e18a033bSKonstantin Ananyev                size = aligned - cl->buf->file_pos;
248e18a033bSKonstantin Ananyev            }
249e18a033bSKonstantin Ananyev
250e18a033bSKonstantin Ananyev            total += size;
251e18a033bSKonstantin Ananyev            break;
252e18a033bSKonstantin Ananyev        }
253e18a033bSKonstantin Ananyev
254e18a033bSKonstantin Ananyev        total += size;
255e18a033bSKonstantin Ananyev        fprev = cl->buf->file_pos + size;
256e18a033bSKonstantin Ananyev        cl = cl->next;
257e18a033bSKonstantin Ananyev
258e18a033bSKonstantin Ananyev    } while (cl
259e18a033bSKonstantin Ananyev             && cl->buf->in_file
260e18a033bSKonstantin Ananyev             && total < limit
261e18a033bSKonstantin Ananyev             && fd == cl->buf->file->fd
262e18a033bSKonstantin Ananyev             && fprev == cl->buf->file_pos);
263e18a033bSKonstantin Ananyev
264e18a033bSKonstantin Ananyev    *in = cl;
265e18a033bSKonstantin Ananyev
266e18a033bSKonstantin Ananyev    return total;
267e18a033bSKonstantin Ananyev}
268e18a033bSKonstantin Ananyev
269e18a033bSKonstantin Ananyev
270e18a033bSKonstantin Ananyevngx_chain_t *
271e18a033bSKonstantin Ananyevngx_chain_update_sent(ngx_chain_t *in, off_t sent)
272e18a033bSKonstantin Ananyev{
273e18a033bSKonstantin Ananyev    off_t  size;
274e18a033bSKonstantin Ananyev
275e18a033bSKonstantin Ananyev    for ( /* void */ ; in; in = in->next) {
276e18a033bSKonstantin Ananyev
277e18a033bSKonstantin Ananyev        if (ngx_buf_special(in->buf)) {
278e18a033bSKonstantin Ananyev            continue;
279e18a033bSKonstantin Ananyev        }
280e18a033bSKonstantin Ananyev
281e18a033bSKonstantin Ananyev        if (sent == 0) {
282e18a033bSKonstantin Ananyev            break;
283e18a033bSKonstantin Ananyev        }
284e18a033bSKonstantin Ananyev
285e18a033bSKonstantin Ananyev        size = ngx_buf_size(in->buf);
286e18a033bSKonstantin Ananyev
287e18a033bSKonstantin Ananyev        if (sent >= size) {
288e18a033bSKonstantin Ananyev            sent -= size;
289e18a033bSKonstantin Ananyev
290e18a033bSKonstantin Ananyev            if (ngx_buf_in_memory(in->buf)) {
291e18a033bSKonstantin Ananyev                in->buf->pos = in->buf->last;
292e18a033bSKonstantin Ananyev            }
293e18a033bSKonstantin Ananyev
294e18a033bSKonstantin Ananyev            if (in->buf->in_file) {
295e18a033bSKonstantin Ananyev                in->buf->file_pos = in->buf->file_last;
296e18a033bSKonstantin Ananyev            }
297e18a033bSKonstantin Ananyev
298e18a033bSKonstantin Ananyev            continue;
299e18a033bSKonstantin Ananyev        }
300e18a033bSKonstantin Ananyev
301e18a033bSKonstantin Ananyev        if (ngx_buf_in_memory(in->buf)) {
302e18a033bSKonstantin Ananyev            in->buf->pos += (size_t) sent;
303e18a033bSKonstantin Ananyev        }
304e18a033bSKonstantin Ananyev
305e18a033bSKonstantin Ananyev        if (in->buf->in_file) {
306e18a033bSKonstantin Ananyev            in->buf->file_pos += sent;
307e18a033bSKonstantin Ananyev        }
308e18a033bSKonstantin Ananyev
309e18a033bSKonstantin Ananyev        break;
310e18a033bSKonstantin Ananyev    }
311e18a033bSKonstantin Ananyev
312e18a033bSKonstantin Ananyev    return in;
313e18a033bSKonstantin Ananyev}
314