0
|
1
|
|
2 /*
|
|
3 * Copyright (C) Igor Sysoev
|
|
4 */
|
|
5
|
|
6
|
|
7 #include <ngx_config.h>
|
|
8 #include <ngx_core.h>
|
|
9
|
|
10
|
154
|
11 ngx_array_t *
|
|
12 ngx_array_create(ngx_pool_t *p, ngx_uint_t n, size_t size)
|
0
|
13 {
|
|
14 ngx_array_t *a;
|
|
15
|
50
|
16 a = ngx_palloc(p, sizeof(ngx_array_t));
|
|
17 if (a == NULL) {
|
34
|
18 return NULL;
|
|
19 }
|
0
|
20
|
50
|
21 a->elts = ngx_palloc(p, n * size);
|
|
22 if (a->elts == NULL) {
|
34
|
23 return NULL;
|
|
24 }
|
|
25
|
0
|
26 a->nelts = 0;
|
34
|
27 a->size = size;
|
0
|
28 a->nalloc = n;
|
34
|
29 a->pool = p;
|
0
|
30
|
|
31 return a;
|
|
32 }
|
|
33
|
|
34
|
154
|
35 void
|
|
36 ngx_array_destroy(ngx_array_t *a)
|
0
|
37 {
|
|
38 ngx_pool_t *p;
|
|
39
|
|
40 p = a->pool;
|
|
41
|
382
|
42 if ((u_char *) a->elts + a->size * a->nalloc == p->d.last) {
|
|
43 p->d.last -= a->size * a->nalloc;
|
0
|
44 }
|
|
45
|
382
|
46 if ((u_char *) a + sizeof(ngx_array_t) == p->d.last) {
|
|
47 p->d.last = (u_char *) a;
|
0
|
48 }
|
|
49 }
|
|
50
|
|
51
|
154
|
52 void *
|
|
53 ngx_array_push(ngx_array_t *a)
|
0
|
54 {
|
|
55 void *elt, *new;
|
34
|
56 size_t size;
|
0
|
57 ngx_pool_t *p;
|
|
58
|
|
59 if (a->nelts == a->nalloc) {
|
34
|
60
|
|
61 /* the array is full */
|
|
62
|
|
63 size = a->size * a->nalloc;
|
|
64
|
0
|
65 p = a->pool;
|
|
66
|
382
|
67 if ((u_char *) a->elts + size == p->d.last
|
|
68 && p->d.last + a->size <= p->d.end)
|
0
|
69 {
|
34
|
70 /*
|
|
71 * the array allocation is the last in the pool
|
|
72 * and there is space for new allocation
|
|
73 */
|
|
74
|
382
|
75 p->d.last += a->size;
|
0
|
76 a->nalloc++;
|
|
77
|
|
78 } else {
|
34
|
79 /* allocate a new array */
|
0
|
80
|
50
|
81 new = ngx_palloc(p, 2 * size);
|
|
82 if (new == NULL) {
|
34
|
83 return NULL;
|
|
84 }
|
|
85
|
|
86 ngx_memcpy(new, a->elts, size);
|
0
|
87 a->elts = new;
|
|
88 a->nalloc *= 2;
|
|
89 }
|
|
90 }
|
|
91
|
34
|
92 elt = (u_char *) a->elts + a->size * a->nelts;
|
0
|
93 a->nelts++;
|
|
94
|
|
95 return elt;
|
|
96 }
|
34
|
97
|
|
98
|
154
|
99 void *
|
|
100 ngx_array_push_n(ngx_array_t *a, ngx_uint_t n)
|
34
|
101 {
|
|
102 void *elt, *new;
|
|
103 size_t size;
|
|
104 ngx_uint_t nalloc;
|
|
105 ngx_pool_t *p;
|
|
106
|
|
107 size = n * a->size;
|
|
108
|
|
109 if (a->nelts + n > a->nalloc) {
|
|
110
|
|
111 /* the array is full */
|
|
112
|
|
113 p = a->pool;
|
|
114
|
382
|
115 if ((u_char *) a->elts + a->size * a->nalloc == p->d.last
|
|
116 && p->d.last + size <= p->d.end)
|
34
|
117 {
|
|
118 /*
|
|
119 * the array allocation is the last in the pool
|
|
120 * and there is space for new allocation
|
|
121 */
|
|
122
|
382
|
123 p->d.last += size;
|
34
|
124 a->nalloc += n;
|
|
125
|
|
126 } else {
|
|
127 /* allocate a new array */
|
|
128
|
|
129 nalloc = 2 * ((n >= a->nalloc) ? n : a->nalloc);
|
|
130
|
50
|
131 new = ngx_palloc(p, nalloc * a->size);
|
|
132 if (new == NULL) {
|
34
|
133 return NULL;
|
|
134 }
|
|
135
|
|
136 ngx_memcpy(new, a->elts, a->nelts * a->size);
|
|
137 a->elts = new;
|
|
138 a->nalloc = nalloc;
|
|
139 }
|
|
140 }
|
|
141
|
|
142 elt = (u_char *) a->elts + a->size * a->nelts;
|
|
143 a->nelts += n;
|
|
144
|
|
145 return elt;
|
|
146 }
|