Blame view
include/linux/flex_array.h
4.31 KB
b24413180
|
1 |
/* SPDX-License-Identifier: GPL-2.0 */ |
534acc057
|
2 3 4 5 |
#ifndef _FLEX_ARRAY_H #define _FLEX_ARRAY_H #include <linux/types.h> |
809fa972f
|
6 |
#include <linux/reciprocal_div.h> |
534acc057
|
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
#include <asm/page.h> #define FLEX_ARRAY_PART_SIZE PAGE_SIZE #define FLEX_ARRAY_BASE_SIZE PAGE_SIZE struct flex_array_part; /* * This is meant to replace cases where an array-like * structure has gotten too big to fit into kmalloc() * and the developer is getting tempted to use * vmalloc(). */ struct flex_array { union { struct { int element_size; int total_nr_elements; |
704f15ddb
|
26 |
int elems_per_part; |
809fa972f
|
27 |
struct reciprocal_value reciprocal_elems; |
8e7ee2709
|
28 |
struct flex_array_part *parts[]; |
534acc057
|
29 30 31 32 33 34 35 36 |
}; /* * This little trick makes sure that * sizeof(flex_array) == PAGE_SIZE */ char padding[FLEX_ARRAY_BASE_SIZE]; }; }; |
45b588d6e
|
37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
/* Number of bytes left in base struct flex_array, excluding metadata */ #define FLEX_ARRAY_BASE_BYTES_LEFT \ (FLEX_ARRAY_BASE_SIZE - offsetof(struct flex_array, parts)) /* Number of pointers in base to struct flex_array_part pages */ #define FLEX_ARRAY_NR_BASE_PTRS \ (FLEX_ARRAY_BASE_BYTES_LEFT / sizeof(struct flex_array_part *)) /* Number of elements of size that fit in struct flex_array_part */ #define FLEX_ARRAY_ELEMENTS_PER_PART(size) \ (FLEX_ARRAY_PART_SIZE / size) /* * Defines a statically allocated flex array and ensures its parameters are * valid. */ #define DEFINE_FLEX_ARRAY(__arrayname, __element_size, __total) \ struct flex_array __arrayname = { { { \ .element_size = (__element_size), \ .total_nr_elements = (__total), \ } } }; \ static inline void __arrayname##_invalid_parameter(void) \ { \ BUILD_BUG_ON((__total) > FLEX_ARRAY_NR_BASE_PTRS * \ FLEX_ARRAY_ELEMENTS_PER_PART(__element_size)); \ } |
534acc057
|
63 |
|
b2e33536c
|
64 65 66 67 68 69 70 71 |
/** * flex_array_alloc() - Creates a flexible array. * @element_size: individual object size. * @total: maximum number of objects which can be stored. * @flags: GFP flags * * Return: Returns an object of structure flex_array. */ |
b62e408c0
|
72 73 |
struct flex_array *flex_array_alloc(int element_size, unsigned int total, gfp_t flags); |
b2e33536c
|
74 75 76 77 78 79 80 81 82 83 |
/** * flex_array_prealloc() - Ensures that memory for the elements indexed in the * range defined by start and nr_elements has been allocated. * @fa: array to allocate memory to. * @start: start address * @nr_elements: number of elements to be allocated. * @flags: GFP flags * */ |
b62e408c0
|
84 |
int flex_array_prealloc(struct flex_array *fa, unsigned int start, |
5d30b10bd
|
85 |
unsigned int nr_elements, gfp_t flags); |
b2e33536c
|
86 87 88 89 90 |
/** * flex_array_free() - Removes all elements of a flexible array. * @fa: array to be freed. */ |
534acc057
|
91 |
void flex_array_free(struct flex_array *fa); |
b2e33536c
|
92 93 94 95 96 97 |
/** * flex_array_free_parts() - Removes all elements of a flexible array, but * leaves the array itself in place. * @fa: array to be emptied. */ |
534acc057
|
98 |
void flex_array_free_parts(struct flex_array *fa); |
b2e33536c
|
99 100 101 102 103 104 105 106 107 108 109 |
/** * flex_array_put() - Stores data into a flexible array. * @fa: array where element is to be stored. * @element_nr: position to copy, must be less than the maximum specified when * the array was created. * @src: data source to be copied into the array. * @flags: GFP flags * * Return: Returns zero on success, a negative error code otherwise. */ |
b62e408c0
|
110 |
int flex_array_put(struct flex_array *fa, unsigned int element_nr, void *src, |
534acc057
|
111 |
gfp_t flags); |
b2e33536c
|
112 113 114 115 116 117 118 119 120 |
/** * flex_array_clear() - Clears an individual element in the array, sets the * given element to FLEX_ARRAY_FREE. * @element_nr: element position to clear. * @fa: array to which element to be cleared belongs. * * Return: Returns zero on success, -EINVAL otherwise. */ |
e6de3988a
|
121 |
int flex_array_clear(struct flex_array *fa, unsigned int element_nr); |
b2e33536c
|
122 123 124 125 126 127 128 129 130 131 |
/** * flex_array_get() - Retrieves data into a flexible array. * * @element_nr: Element position to retrieve data from. * @fa: array from which data is to be retrieved. * * Return: Returns a pointer to the data element, or NULL if that * particular element has never been allocated. */ |
b62e408c0
|
132 |
void *flex_array_get(struct flex_array *fa, unsigned int element_nr); |
b2e33536c
|
133 134 135 136 137 138 139 140 |
/** * flex_array_shrink() - Reduces the allocated size of an array. * @fa: array to shrink. * * Return: Returns number of pages of memory actually freed. * */ |
4af5a2f77
|
141 |
int flex_array_shrink(struct flex_array *fa); |
534acc057
|
142 |
|
ea98eed9b
|
143 |
#define flex_array_put_ptr(fa, nr, src, gfp) \ |
c41ab6a1b
|
144 |
flex_array_put(fa, nr, (void *)&(src), gfp) |
ea98eed9b
|
145 146 |
void *flex_array_get_ptr(struct flex_array *fa, unsigned int element_nr); |
534acc057
|
147 |
#endif /* _FLEX_ARRAY_H */ |