diff options
Diffstat (limited to 'gl/malloc/dynarray.h')
| -rw-r--r-- | gl/malloc/dynarray.h | 177 |
1 files changed, 177 insertions, 0 deletions
diff --git a/gl/malloc/dynarray.h b/gl/malloc/dynarray.h new file mode 100644 index 00000000..a9a3b085 --- /dev/null +++ b/gl/malloc/dynarray.h | |||
| @@ -0,0 +1,177 @@ | |||
| 1 | /* Type-safe arrays which grow dynamically. Shared definitions. | ||
| 2 | Copyright (C) 2017-2023 Free Software Foundation, Inc. | ||
| 3 | This file is part of the GNU C Library. | ||
| 4 | |||
| 5 | The GNU C Library is free software; you can redistribute it and/or | ||
| 6 | modify it under the terms of the GNU Lesser General Public | ||
| 7 | License as published by the Free Software Foundation; either | ||
| 8 | version 2.1 of the License, or (at your option) any later version. | ||
| 9 | |||
| 10 | The GNU C Library is distributed in the hope that it will be useful, | ||
| 11 | but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| 12 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
| 13 | Lesser General Public License for more details. | ||
| 14 | |||
| 15 | You should have received a copy of the GNU Lesser General Public | ||
| 16 | License along with the GNU C Library; if not, see | ||
| 17 | <https://www.gnu.org/licenses/>. */ | ||
| 18 | |||
| 19 | /* To use the dynarray facility, you need to include | ||
| 20 | <malloc/dynarray-skeleton.c> and define the parameter macros | ||
| 21 | documented in that file. | ||
| 22 | |||
| 23 | A minimal example which provides a growing list of integers can be | ||
| 24 | defined like this: | ||
| 25 | |||
| 26 | struct int_array | ||
| 27 | { | ||
| 28 | // Pointer to result array followed by its length, | ||
| 29 | // as required by DYNARRAY_FINAL_TYPE. | ||
| 30 | int *array; | ||
| 31 | size_t length; | ||
| 32 | }; | ||
| 33 | |||
| 34 | #define DYNARRAY_STRUCT dynarray_int | ||
| 35 | #define DYNARRAY_ELEMENT int | ||
| 36 | #define DYNARRAY_PREFIX dynarray_int_ | ||
| 37 | #define DYNARRAY_FINAL_TYPE struct int_array | ||
| 38 | #include <malloc/dynarray-skeleton.c> | ||
| 39 | |||
| 40 | To create a three-element array with elements 1, 2, 3, use this | ||
| 41 | code: | ||
| 42 | |||
| 43 | struct dynarray_int dyn; | ||
| 44 | dynarray_int_init (&dyn); | ||
| 45 | for (int i = 1; i <= 3; ++i) | ||
| 46 | { | ||
| 47 | int *place = dynarray_int_emplace (&dyn); | ||
| 48 | assert (place != NULL); | ||
| 49 | *place = i; | ||
| 50 | } | ||
| 51 | struct int_array result; | ||
| 52 | bool ok = dynarray_int_finalize (&dyn, &result); | ||
| 53 | assert (ok); | ||
| 54 | assert (result.length == 3); | ||
| 55 | assert (result.array[0] == 1); | ||
| 56 | assert (result.array[1] == 2); | ||
| 57 | assert (result.array[2] == 3); | ||
| 58 | free (result.array); | ||
| 59 | |||
| 60 | If the elements contain resources which must be freed, define | ||
| 61 | DYNARRAY_ELEMENT_FREE appropriately, like this: | ||
| 62 | |||
| 63 | struct str_array | ||
| 64 | { | ||
| 65 | char **array; | ||
| 66 | size_t length; | ||
| 67 | }; | ||
| 68 | |||
| 69 | #define DYNARRAY_STRUCT dynarray_str | ||
| 70 | #define DYNARRAY_ELEMENT char * | ||
| 71 | #define DYNARRAY_ELEMENT_FREE(ptr) free (*ptr) | ||
| 72 | #define DYNARRAY_PREFIX dynarray_str_ | ||
| 73 | #define DYNARRAY_FINAL_TYPE struct str_array | ||
| 74 | #include <malloc/dynarray-skeleton.c> | ||
| 75 | |||
| 76 | Compared to scratch buffers, dynamic arrays have the following | ||
| 77 | features: | ||
| 78 | |||
| 79 | - They have an element type, and are not just an untyped buffer of | ||
| 80 | bytes. | ||
| 81 | |||
| 82 | - When growing, previously stored elements are preserved. (It is | ||
| 83 | expected that scratch_buffer_grow_preserve and | ||
| 84 | scratch_buffer_set_array_size eventually go away because all | ||
| 85 | current users are moved to dynamic arrays.) | ||
| 86 | |||
| 87 | - Scratch buffers have a more aggressive growth policy because | ||
| 88 | growing them typically means a retry of an operation (across an | ||
| 89 | NSS service module boundary), which is expensive. | ||
| 90 | |||
| 91 | - For the same reason, scratch buffers have a much larger initial | ||
| 92 | stack allocation. */ | ||
| 93 | |||
| 94 | #ifndef _DYNARRAY_H | ||
| 95 | #define _DYNARRAY_H | ||
| 96 | |||
| 97 | #include <stddef.h> | ||
| 98 | #include <string.h> | ||
| 99 | |||
| 100 | struct dynarray_header | ||
| 101 | { | ||
| 102 | size_t used; | ||
| 103 | size_t allocated; | ||
| 104 | void *array; | ||
| 105 | }; | ||
| 106 | |||
| 107 | /* Marker used in the allocated member to indicate that an error was | ||
| 108 | encountered. */ | ||
| 109 | static inline size_t | ||
| 110 | __dynarray_error_marker (void) | ||
| 111 | { | ||
| 112 | return -1; | ||
| 113 | } | ||
| 114 | |||
| 115 | /* Internal function. See the has_failed function in | ||
| 116 | dynarray-skeleton.c. */ | ||
| 117 | static inline bool | ||
| 118 | __dynarray_error (struct dynarray_header *list) | ||
| 119 | { | ||
| 120 | return list->allocated == __dynarray_error_marker (); | ||
| 121 | } | ||
| 122 | |||
| 123 | /* Internal function. Enlarge the dynamically allocated area of the | ||
| 124 | array to make room for one more element. SCRATCH is a pointer to | ||
| 125 | the scratch area (which is not heap-allocated and must not be | ||
| 126 | freed). ELEMENT_SIZE is the size, in bytes, of one element. | ||
| 127 | Return false on failure, true on success. */ | ||
| 128 | bool __libc_dynarray_emplace_enlarge (struct dynarray_header *, | ||
| 129 | void *scratch, size_t element_size); | ||
| 130 | |||
| 131 | /* Internal function. Enlarge the dynamically allocated area of the | ||
| 132 | array to make room for at least SIZE elements (which must be larger | ||
| 133 | than the existing used part of the dynamic array). SCRATCH is a | ||
| 134 | pointer to the scratch area (which is not heap-allocated and must | ||
| 135 | not be freed). ELEMENT_SIZE is the size, in bytes, of one element. | ||
| 136 | Return false on failure, true on success. */ | ||
| 137 | bool __libc_dynarray_resize (struct dynarray_header *, size_t size, | ||
| 138 | void *scratch, size_t element_size); | ||
| 139 | |||
| 140 | /* Internal function. Like __libc_dynarray_resize, but clear the new | ||
| 141 | part of the dynamic array. */ | ||
| 142 | bool __libc_dynarray_resize_clear (struct dynarray_header *, size_t size, | ||
| 143 | void *scratch, size_t element_size); | ||
| 144 | |||
| 145 | /* Internal type. */ | ||
| 146 | struct dynarray_finalize_result | ||
| 147 | { | ||
| 148 | void *array; | ||
| 149 | size_t length; | ||
| 150 | }; | ||
| 151 | |||
| 152 | /* Internal function. Copy the dynamically-allocated area to an | ||
| 153 | explicitly-sized heap allocation. SCRATCH is a pointer to the | ||
| 154 | embedded scratch space. ELEMENT_SIZE is the size, in bytes, of the | ||
| 155 | element type. On success, true is returned, and pointer and length | ||
| 156 | are written to *RESULT. On failure, false is returned. The caller | ||
| 157 | has to take care of some of the memory management; this function is | ||
| 158 | expected to be called from dynarray-skeleton.c. */ | ||
| 159 | bool __libc_dynarray_finalize (struct dynarray_header *list, void *scratch, | ||
| 160 | size_t element_size, | ||
| 161 | struct dynarray_finalize_result *result); | ||
| 162 | |||
| 163 | |||
| 164 | /* Internal function. Terminate the process after an index error. | ||
| 165 | SIZE is the number of elements of the dynamic array. INDEX is the | ||
| 166 | lookup index which triggered the failure. */ | ||
| 167 | _Noreturn void __libc_dynarray_at_failure (size_t size, size_t index); | ||
| 168 | |||
| 169 | #ifndef _ISOMAC | ||
| 170 | libc_hidden_proto (__libc_dynarray_emplace_enlarge) | ||
| 171 | libc_hidden_proto (__libc_dynarray_resize) | ||
| 172 | libc_hidden_proto (__libc_dynarray_resize_clear) | ||
| 173 | libc_hidden_proto (__libc_dynarray_finalize) | ||
| 174 | libc_hidden_proto (__libc_dynarray_at_failure) | ||
| 175 | #endif | ||
| 176 | |||
| 177 | #endif /* _DYNARRAY_H */ | ||
