nxt_array.c (146:24354fd871d5) nxt_array.c (1489:4a3ec07f4b19)
1
2/*
3 * Copyright (C) Igor Sysoev
4 * Copyright (C) NGINX, Inc.
5 */
6
7#include <nxt_main.h>
8
9
10nxt_array_t *
11nxt_array_create(nxt_mp_t *mp, nxt_uint_t n, size_t size)
12{
13 nxt_array_t *array;
14
15 array = nxt_mp_alloc(mp, sizeof(nxt_array_t) + n * size);
16
17 if (nxt_slow_path(array == NULL)) {
18 return NULL;
19 }
20
21 array->elts = nxt_pointer_to(array, sizeof(nxt_array_t));
22 array->nelts = 0;
23 array->size = size;
24 array->nalloc = n;
25 array->mem_pool = mp;
26
27 return array;
28}
29
30
31void
32nxt_array_destroy(nxt_array_t *array)
33{
34 if (array->elts != nxt_pointer_to(array, sizeof(nxt_array_t))) {
35 nxt_mp_free(array->mem_pool, array->elts);
36 }
37
38 nxt_mp_free(array->mem_pool, array);
39}
40
41
42void *
43nxt_array_add(nxt_array_t *array)
44{
45 void *p;
46 uint32_t nalloc, new_alloc;
47
48 nalloc = array->nalloc;
49
50 if (array->nelts == nalloc) {
51
52 if (nalloc < 16) {
53 /* Allocate new array twice larger than current. */
54 new_alloc = nalloc * 2;
55
56 } else {
57 /* Allocate new array 1.5 times larger than current. */
58 new_alloc = nalloc + nalloc / 2;
59 }
60
61 p = nxt_mp_alloc(array->mem_pool, array->size * new_alloc);
62
63 if (nxt_slow_path(p == NULL)) {
64 return NULL;
65 }
66
67 nxt_memcpy(p, array->elts, array->size * nalloc);
68
69 if (array->elts != nxt_pointer_to(array, sizeof(nxt_array_t))) {
70 nxt_mp_free(array->mem_pool, array->elts);
71 }
72
73 array->elts = p;
74 array->nalloc = new_alloc;
75 }
76
77 p = nxt_pointer_to(array->elts, array->size * array->nelts);
78 array->nelts++;
79
80 return p;
81}
82
83
84void *
85nxt_array_zero_add(nxt_array_t *array)
86{
87 void *p;
88
89 p = nxt_array_add(array);
90
91 if (nxt_fast_path(p != NULL)) {
92 nxt_memzero(p, array->size);
93 }
94
95 return p;
96}
97
98
99void
100nxt_array_remove(nxt_array_t *array, void *elt)
101{
102 void *last;
103
104 last = nxt_array_last(array);
105
106 if (elt != last) {
107 nxt_memcpy(elt, last, array->size);
108 }
109
110 array->nelts--;
111}
1
2/*
3 * Copyright (C) Igor Sysoev
4 * Copyright (C) NGINX, Inc.
5 */
6
7#include <nxt_main.h>
8
9
10nxt_array_t *
11nxt_array_create(nxt_mp_t *mp, nxt_uint_t n, size_t size)
12{
13 nxt_array_t *array;
14
15 array = nxt_mp_alloc(mp, sizeof(nxt_array_t) + n * size);
16
17 if (nxt_slow_path(array == NULL)) {
18 return NULL;
19 }
20
21 array->elts = nxt_pointer_to(array, sizeof(nxt_array_t));
22 array->nelts = 0;
23 array->size = size;
24 array->nalloc = n;
25 array->mem_pool = mp;
26
27 return array;
28}
29
30
31void
32nxt_array_destroy(nxt_array_t *array)
33{
34 if (array->elts != nxt_pointer_to(array, sizeof(nxt_array_t))) {
35 nxt_mp_free(array->mem_pool, array->elts);
36 }
37
38 nxt_mp_free(array->mem_pool, array);
39}
40
41
42void *
43nxt_array_add(nxt_array_t *array)
44{
45 void *p;
46 uint32_t nalloc, new_alloc;
47
48 nalloc = array->nalloc;
49
50 if (array->nelts == nalloc) {
51
52 if (nalloc < 16) {
53 /* Allocate new array twice larger than current. */
54 new_alloc = nalloc * 2;
55
56 } else {
57 /* Allocate new array 1.5 times larger than current. */
58 new_alloc = nalloc + nalloc / 2;
59 }
60
61 p = nxt_mp_alloc(array->mem_pool, array->size * new_alloc);
62
63 if (nxt_slow_path(p == NULL)) {
64 return NULL;
65 }
66
67 nxt_memcpy(p, array->elts, array->size * nalloc);
68
69 if (array->elts != nxt_pointer_to(array, sizeof(nxt_array_t))) {
70 nxt_mp_free(array->mem_pool, array->elts);
71 }
72
73 array->elts = p;
74 array->nalloc = new_alloc;
75 }
76
77 p = nxt_pointer_to(array->elts, array->size * array->nelts);
78 array->nelts++;
79
80 return p;
81}
82
83
84void *
85nxt_array_zero_add(nxt_array_t *array)
86{
87 void *p;
88
89 p = nxt_array_add(array);
90
91 if (nxt_fast_path(p != NULL)) {
92 nxt_memzero(p, array->size);
93 }
94
95 return p;
96}
97
98
99void
100nxt_array_remove(nxt_array_t *array, void *elt)
101{
102 void *last;
103
104 last = nxt_array_last(array);
105
106 if (elt != last) {
107 nxt_memcpy(elt, last, array->size);
108 }
109
110 array->nelts--;
111}
112
113
114nxt_array_t *
115nxt_array_copy(nxt_mp_t *mp, nxt_array_t *dst, nxt_array_t *src)
116{
117 void *data;
118 uint32_t i, size;
119
120 size = src->size;
121
122 if (dst == NULL) {
123 dst = nxt_array_create(mp, src->nelts, size);
124 if (nxt_slow_path(dst == NULL)) {
125 return NULL;
126 }
127 }
128
129 nxt_assert(size == dst->size);
130
131 if (dst->nalloc >= src->nelts) {
132 nxt_memcpy(dst->elts, src->elts, src->nelts * size);
133
134 } else {
135 nxt_memcpy(dst->elts, src->elts, dst->nelts * size);
136
137 for (i = dst->nelts; i < src->nelts; i++) {
138 data = nxt_array_add(dst);
139 if (nxt_slow_path(data == NULL)) {
140 return NULL;
141 }
142
143 nxt_memcpy(data, src->elts + (i * size), size);
144 }
145 }
146
147 dst->nelts = src->nelts;
148
149 return dst;
150}