malloc/dynarray.h

   1 /* Type-safe arrays which grow dynamically.  Shared definitions.
   2    Copyright (C) 2017 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    <http://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 <stdbool.h>
  98 #include <stddef.h>
  99 #include <string.h>
 100
 101 struct dynarray_header
 102 {
 103   size_t used;
 104   size_t allocated;
 105   void *array;
 106 };
 107
 108 /* Marker used in the allocated member to indicate that an error was
 109    encountered.  */
 110 static inline size_t
 111 __dynarray_error_marker (void)
 112 {
 113   return -1;
 114 }
 115
 116 /* Internal function.  See the has_failed function in
 117    dynarray-skeleton.c.  */
 118 static inline bool
 119 __dynarray_error (struct dynarray_header *list)
 120 {
 121   return list->allocated == __dynarray_error_marker ();
 122 }
 123
 124 /* Internal function.  Enlarge the dynamically allocated area of the
 125    array to make room for one more element.  SCRATCH is a pointer to
 126    the scratch area (which is not heap-allocated and must not be
 127    freed).  ELEMENT_SIZE is the size, in bytes, of one element.
 128    Return false on failure, true on success.  */
 129 bool __libc_dynarray_emplace_enlarge (struct dynarray_header *,
 130                                       void *scratch, size_t element_size);
 131 libc_hidden_proto (__libc_dynarray_emplace_enlarge)
 132
 133 /* Internal function.  Enlarge the dynamically allocated area of the
 134    array to make room for at least SIZE elements (which must be larger
 135    than the existing used part of the dynamic array).  SCRATCH is a
 136    pointer to the scratch area (which is not heap-allocated and must
 137    not be freed).  ELEMENT_SIZE is the size, in bytes, of one element.
 138    Return false on failure, true on success.  */
 139 bool __libc_dynarray_resize (struct dynarray_header *, size_t size,
 140                              void *scratch, size_t element_size);
 141 libc_hidden_proto (__libc_dynarray_resize)
 142
 143 /* Internal function.  Like __libc_dynarray_resize, but clear the new
 144    part of the dynamic array.  */
 145 bool __libc_dynarray_resize_clear (struct dynarray_header *, size_t size,
 146                                    void *scratch, size_t element_size);
 147 libc_hidden_proto (__libc_dynarray_resize_clear)
 148
 149 /* Internal type.  */
 150 struct dynarray_finalize_result
 151 {
 152   void *array;
 153   size_t length;
 154 };
 155
 156 /* Internal function.  Copy the dynamically-allocated area to an
 157    explicitly-sized heap allocation.  SCRATCH is a pointer to the
 158    embedded scratch space.  ELEMENT_SIZE is the size, in bytes, of the
 159    element type.  On success, true is returned, and pointer and length
 160    are written to *RESULT.  On failure, false is returned.  The caller
 161    has to take care of some of the memory management; this function is
 162    expected to be called from dynarray-skeleton.c.  */
 163 bool __libc_dynarray_finalize (struct dynarray_header *list, void *scratch,
 164                                size_t element_size,
 165                                struct dynarray_finalize_result *result);
 166 libc_hidden_proto (__libc_dynarray_finalize)
 167
 168
 169 /* Internal function.  Terminate the process after an index error.
 170    SIZE is the number of elements of the dynamic array.  INDEX is the
 171    lookup index which triggered the failure.  */
 172 void __libc_dynarray_at_failure (size_t size, size_t index)
 173   __attribute__ ((noreturn));
 174 libc_hidden_proto (__libc_dynarray_at_failure)
 175
 176 #endif /* _DYNARRAY_H */