1685a6bf8SThomas Gleixner // SPDX-License-Identifier: GPL-2.0-only
2b484b26cSGeorge Zhang /*
3b484b26cSGeorge Zhang * VMware VMCI Driver
4b484b26cSGeorge Zhang *
5b484b26cSGeorge Zhang * Copyright (C) 2012 VMware, Inc. All rights reserved.
6b484b26cSGeorge Zhang */
7b484b26cSGeorge Zhang
8b484b26cSGeorge Zhang #include <linux/slab.h>
9b484b26cSGeorge Zhang #include "vmci_handle_array.h"
10b484b26cSGeorge Zhang
vmci_handle_arr_create(u32 capacity,u32 max_capacity)111c2eb5b2SVishnu DASA struct vmci_handle_arr *vmci_handle_arr_create(u32 capacity, u32 max_capacity)
12b484b26cSGeorge Zhang {
13b484b26cSGeorge Zhang struct vmci_handle_arr *array;
14b484b26cSGeorge Zhang
151c2eb5b2SVishnu DASA if (max_capacity == 0 || capacity > max_capacity)
161c2eb5b2SVishnu DASA return NULL;
171c2eb5b2SVishnu DASA
18b484b26cSGeorge Zhang if (capacity == 0)
191c2eb5b2SVishnu DASA capacity = min((u32)VMCI_HANDLE_ARRAY_DEFAULT_CAPACITY,
201c2eb5b2SVishnu DASA max_capacity);
21b484b26cSGeorge Zhang
22*bda910d8SChristophe JAILLET array = kmalloc(struct_size(array, entries, capacity), GFP_ATOMIC);
23b484b26cSGeorge Zhang if (!array)
24b484b26cSGeorge Zhang return NULL;
25b484b26cSGeorge Zhang
26b484b26cSGeorge Zhang array->capacity = capacity;
271c2eb5b2SVishnu DASA array->max_capacity = max_capacity;
28b484b26cSGeorge Zhang array->size = 0;
29b484b26cSGeorge Zhang
30b484b26cSGeorge Zhang return array;
31b484b26cSGeorge Zhang }
32b484b26cSGeorge Zhang
vmci_handle_arr_destroy(struct vmci_handle_arr * array)33b484b26cSGeorge Zhang void vmci_handle_arr_destroy(struct vmci_handle_arr *array)
34b484b26cSGeorge Zhang {
35b484b26cSGeorge Zhang kfree(array);
36b484b26cSGeorge Zhang }
37b484b26cSGeorge Zhang
vmci_handle_arr_append_entry(struct vmci_handle_arr ** array_ptr,struct vmci_handle handle)381c2eb5b2SVishnu DASA int vmci_handle_arr_append_entry(struct vmci_handle_arr **array_ptr,
39b484b26cSGeorge Zhang struct vmci_handle handle)
40b484b26cSGeorge Zhang {
41b484b26cSGeorge Zhang struct vmci_handle_arr *array = *array_ptr;
42b484b26cSGeorge Zhang
43b484b26cSGeorge Zhang if (unlikely(array->size >= array->capacity)) {
44b484b26cSGeorge Zhang /* reallocate. */
45b484b26cSGeorge Zhang struct vmci_handle_arr *new_array;
461c2eb5b2SVishnu DASA u32 capacity_bump = min(array->max_capacity - array->capacity,
471c2eb5b2SVishnu DASA array->capacity);
48*bda910d8SChristophe JAILLET size_t new_size = struct_size(array, entries,
49*bda910d8SChristophe JAILLET size_add(array->capacity, capacity_bump));
501c2eb5b2SVishnu DASA
511c2eb5b2SVishnu DASA if (array->size >= array->max_capacity)
521c2eb5b2SVishnu DASA return VMCI_ERROR_NO_MEM;
53b484b26cSGeorge Zhang
54b484b26cSGeorge Zhang new_array = krealloc(array, new_size, GFP_ATOMIC);
55b484b26cSGeorge Zhang if (!new_array)
561c2eb5b2SVishnu DASA return VMCI_ERROR_NO_MEM;
57b484b26cSGeorge Zhang
581c2eb5b2SVishnu DASA new_array->capacity += capacity_bump;
59b484b26cSGeorge Zhang *array_ptr = array = new_array;
60b484b26cSGeorge Zhang }
61b484b26cSGeorge Zhang
62b484b26cSGeorge Zhang array->entries[array->size] = handle;
63b484b26cSGeorge Zhang array->size++;
641c2eb5b2SVishnu DASA
651c2eb5b2SVishnu DASA return VMCI_SUCCESS;
66b484b26cSGeorge Zhang }
67b484b26cSGeorge Zhang
68b484b26cSGeorge Zhang /*
69b484b26cSGeorge Zhang * Handle that was removed, VMCI_INVALID_HANDLE if entry not found.
70b484b26cSGeorge Zhang */
vmci_handle_arr_remove_entry(struct vmci_handle_arr * array,struct vmci_handle entry_handle)71b484b26cSGeorge Zhang struct vmci_handle vmci_handle_arr_remove_entry(struct vmci_handle_arr *array,
72b484b26cSGeorge Zhang struct vmci_handle entry_handle)
73b484b26cSGeorge Zhang {
74b484b26cSGeorge Zhang struct vmci_handle handle = VMCI_INVALID_HANDLE;
751c2eb5b2SVishnu DASA u32 i;
76b484b26cSGeorge Zhang
77b484b26cSGeorge Zhang for (i = 0; i < array->size; i++) {
78b484b26cSGeorge Zhang if (vmci_handle_is_equal(array->entries[i], entry_handle)) {
79b484b26cSGeorge Zhang handle = array->entries[i];
80b484b26cSGeorge Zhang array->size--;
81b484b26cSGeorge Zhang array->entries[i] = array->entries[array->size];
82b484b26cSGeorge Zhang array->entries[array->size] = VMCI_INVALID_HANDLE;
83b484b26cSGeorge Zhang break;
84b484b26cSGeorge Zhang }
85b484b26cSGeorge Zhang }
86b484b26cSGeorge Zhang
87b484b26cSGeorge Zhang return handle;
88b484b26cSGeorge Zhang }
89b484b26cSGeorge Zhang
90b484b26cSGeorge Zhang /*
91b484b26cSGeorge Zhang * Handle that was removed, VMCI_INVALID_HANDLE if array was empty.
92b484b26cSGeorge Zhang */
vmci_handle_arr_remove_tail(struct vmci_handle_arr * array)93b484b26cSGeorge Zhang struct vmci_handle vmci_handle_arr_remove_tail(struct vmci_handle_arr *array)
94b484b26cSGeorge Zhang {
95b484b26cSGeorge Zhang struct vmci_handle handle = VMCI_INVALID_HANDLE;
96b484b26cSGeorge Zhang
97b484b26cSGeorge Zhang if (array->size) {
98b484b26cSGeorge Zhang array->size--;
99b484b26cSGeorge Zhang handle = array->entries[array->size];
100b484b26cSGeorge Zhang array->entries[array->size] = VMCI_INVALID_HANDLE;
101b484b26cSGeorge Zhang }
102b484b26cSGeorge Zhang
103b484b26cSGeorge Zhang return handle;
104b484b26cSGeorge Zhang }
105b484b26cSGeorge Zhang
106b484b26cSGeorge Zhang /*
107b484b26cSGeorge Zhang * Handle at given index, VMCI_INVALID_HANDLE if invalid index.
108b484b26cSGeorge Zhang */
109b484b26cSGeorge Zhang struct vmci_handle
vmci_handle_arr_get_entry(const struct vmci_handle_arr * array,u32 index)1101c2eb5b2SVishnu DASA vmci_handle_arr_get_entry(const struct vmci_handle_arr *array, u32 index)
111b484b26cSGeorge Zhang {
112b484b26cSGeorge Zhang if (unlikely(index >= array->size))
113b484b26cSGeorge Zhang return VMCI_INVALID_HANDLE;
114b484b26cSGeorge Zhang
115b484b26cSGeorge Zhang return array->entries[index];
116b484b26cSGeorge Zhang }
117b484b26cSGeorge Zhang
vmci_handle_arr_has_entry(const struct vmci_handle_arr * array,struct vmci_handle entry_handle)118b484b26cSGeorge Zhang bool vmci_handle_arr_has_entry(const struct vmci_handle_arr *array,
119b484b26cSGeorge Zhang struct vmci_handle entry_handle)
120b484b26cSGeorge Zhang {
1211c2eb5b2SVishnu DASA u32 i;
122b484b26cSGeorge Zhang
123b484b26cSGeorge Zhang for (i = 0; i < array->size; i++)
124b484b26cSGeorge Zhang if (vmci_handle_is_equal(array->entries[i], entry_handle))
125b484b26cSGeorge Zhang return true;
126b484b26cSGeorge Zhang
127b484b26cSGeorge Zhang return false;
128b484b26cSGeorge Zhang }
129b484b26cSGeorge Zhang
130b484b26cSGeorge Zhang /*
131b484b26cSGeorge Zhang * NULL if the array is empty. Otherwise, a pointer to the array
132b484b26cSGeorge Zhang * of VMCI handles in the handle array.
133b484b26cSGeorge Zhang */
vmci_handle_arr_get_handles(struct vmci_handle_arr * array)134b484b26cSGeorge Zhang struct vmci_handle *vmci_handle_arr_get_handles(struct vmci_handle_arr *array)
135b484b26cSGeorge Zhang {
136b484b26cSGeorge Zhang if (array->size)
137b484b26cSGeorge Zhang return array->entries;
138b484b26cSGeorge Zhang
139b484b26cSGeorge Zhang return NULL;
140b484b26cSGeorge Zhang }
141