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
handle_arr_calc_size(u32 capacity)111c2eb5b2SVishnu DASA static size_t handle_arr_calc_size(u32 capacity)
12b484b26cSGeorge Zhang {
131c2eb5b2SVishnu DASA return VMCI_HANDLE_ARRAY_HEADER_SIZE +
14b484b26cSGeorge Zhang capacity * sizeof(struct vmci_handle);
15b484b26cSGeorge Zhang }
16b484b26cSGeorge Zhang
vmci_handle_arr_create(u32 capacity,u32 max_capacity)171c2eb5b2SVishnu DASA struct vmci_handle_arr *vmci_handle_arr_create(u32 capacity, u32 max_capacity)
18b484b26cSGeorge Zhang {
19b484b26cSGeorge Zhang struct vmci_handle_arr *array;
20b484b26cSGeorge Zhang
211c2eb5b2SVishnu DASA if (max_capacity == 0 || capacity > max_capacity)
221c2eb5b2SVishnu DASA return NULL;
231c2eb5b2SVishnu DASA
24b484b26cSGeorge Zhang if (capacity == 0)
251c2eb5b2SVishnu DASA capacity = min((u32)VMCI_HANDLE_ARRAY_DEFAULT_CAPACITY,
261c2eb5b2SVishnu DASA max_capacity);
27b484b26cSGeorge Zhang
28b484b26cSGeorge Zhang array = kmalloc(handle_arr_calc_size(capacity), GFP_ATOMIC);
29b484b26cSGeorge Zhang if (!array)
30b484b26cSGeorge Zhang return NULL;
31b484b26cSGeorge Zhang
32b484b26cSGeorge Zhang array->capacity = capacity;
331c2eb5b2SVishnu DASA array->max_capacity = max_capacity;
34b484b26cSGeorge Zhang array->size = 0;
35b484b26cSGeorge Zhang
36b484b26cSGeorge Zhang return array;
37b484b26cSGeorge Zhang }
38b484b26cSGeorge Zhang
vmci_handle_arr_destroy(struct vmci_handle_arr * array)39b484b26cSGeorge Zhang void vmci_handle_arr_destroy(struct vmci_handle_arr *array)
40b484b26cSGeorge Zhang {
41b484b26cSGeorge Zhang kfree(array);
42b484b26cSGeorge Zhang }
43b484b26cSGeorge Zhang
vmci_handle_arr_append_entry(struct vmci_handle_arr ** array_ptr,struct vmci_handle handle)441c2eb5b2SVishnu DASA int vmci_handle_arr_append_entry(struct vmci_handle_arr **array_ptr,
45b484b26cSGeorge Zhang struct vmci_handle handle)
46b484b26cSGeorge Zhang {
47b484b26cSGeorge Zhang struct vmci_handle_arr *array = *array_ptr;
48b484b26cSGeorge Zhang
49b484b26cSGeorge Zhang if (unlikely(array->size >= array->capacity)) {
50b484b26cSGeorge Zhang /* reallocate. */
51b484b26cSGeorge Zhang struct vmci_handle_arr *new_array;
521c2eb5b2SVishnu DASA u32 capacity_bump = min(array->max_capacity - array->capacity,
531c2eb5b2SVishnu DASA array->capacity);
541c2eb5b2SVishnu DASA size_t new_size = handle_arr_calc_size(array->capacity +
551c2eb5b2SVishnu DASA capacity_bump);
561c2eb5b2SVishnu DASA
571c2eb5b2SVishnu DASA if (array->size >= array->max_capacity)
581c2eb5b2SVishnu DASA return VMCI_ERROR_NO_MEM;
59b484b26cSGeorge Zhang
60b484b26cSGeorge Zhang new_array = krealloc(array, new_size, GFP_ATOMIC);
61b484b26cSGeorge Zhang if (!new_array)
621c2eb5b2SVishnu DASA return VMCI_ERROR_NO_MEM;
63b484b26cSGeorge Zhang
641c2eb5b2SVishnu DASA new_array->capacity += capacity_bump;
65b484b26cSGeorge Zhang *array_ptr = array = new_array;
66b484b26cSGeorge Zhang }
67b484b26cSGeorge Zhang
68b484b26cSGeorge Zhang array->entries[array->size] = handle;
69b484b26cSGeorge Zhang array->size++;
701c2eb5b2SVishnu DASA
711c2eb5b2SVishnu DASA return VMCI_SUCCESS;
72b484b26cSGeorge Zhang }
73b484b26cSGeorge Zhang
74b484b26cSGeorge Zhang /*
75b484b26cSGeorge Zhang * Handle that was removed, VMCI_INVALID_HANDLE if entry not found.
76b484b26cSGeorge Zhang */
vmci_handle_arr_remove_entry(struct vmci_handle_arr * array,struct vmci_handle entry_handle)77b484b26cSGeorge Zhang struct vmci_handle vmci_handle_arr_remove_entry(struct vmci_handle_arr *array,
78b484b26cSGeorge Zhang struct vmci_handle entry_handle)
79b484b26cSGeorge Zhang {
80b484b26cSGeorge Zhang struct vmci_handle handle = VMCI_INVALID_HANDLE;
811c2eb5b2SVishnu DASA u32 i;
82b484b26cSGeorge Zhang
83b484b26cSGeorge Zhang for (i = 0; i < array->size; i++) {
84b484b26cSGeorge Zhang if (vmci_handle_is_equal(array->entries[i], entry_handle)) {
85b484b26cSGeorge Zhang handle = array->entries[i];
86b484b26cSGeorge Zhang array->size--;
87b484b26cSGeorge Zhang array->entries[i] = array->entries[array->size];
88b484b26cSGeorge Zhang array->entries[array->size] = VMCI_INVALID_HANDLE;
89b484b26cSGeorge Zhang break;
90b484b26cSGeorge Zhang }
91b484b26cSGeorge Zhang }
92b484b26cSGeorge Zhang
93b484b26cSGeorge Zhang return handle;
94b484b26cSGeorge Zhang }
95b484b26cSGeorge Zhang
96b484b26cSGeorge Zhang /*
97b484b26cSGeorge Zhang * Handle that was removed, VMCI_INVALID_HANDLE if array was empty.
98b484b26cSGeorge Zhang */
vmci_handle_arr_remove_tail(struct vmci_handle_arr * array)99b484b26cSGeorge Zhang struct vmci_handle vmci_handle_arr_remove_tail(struct vmci_handle_arr *array)
100b484b26cSGeorge Zhang {
101b484b26cSGeorge Zhang struct vmci_handle handle = VMCI_INVALID_HANDLE;
102b484b26cSGeorge Zhang
103b484b26cSGeorge Zhang if (array->size) {
104b484b26cSGeorge Zhang array->size--;
105b484b26cSGeorge Zhang handle = array->entries[array->size];
106b484b26cSGeorge Zhang array->entries[array->size] = VMCI_INVALID_HANDLE;
107b484b26cSGeorge Zhang }
108b484b26cSGeorge Zhang
109b484b26cSGeorge Zhang return handle;
110b484b26cSGeorge Zhang }
111b484b26cSGeorge Zhang
112b484b26cSGeorge Zhang /*
113b484b26cSGeorge Zhang * Handle at given index, VMCI_INVALID_HANDLE if invalid index.
114b484b26cSGeorge Zhang */
115b484b26cSGeorge Zhang struct vmci_handle
vmci_handle_arr_get_entry(const struct vmci_handle_arr * array,u32 index)1161c2eb5b2SVishnu DASA vmci_handle_arr_get_entry(const struct vmci_handle_arr *array, u32 index)
117b484b26cSGeorge Zhang {
118b484b26cSGeorge Zhang if (unlikely(index >= array->size))
119b484b26cSGeorge Zhang return VMCI_INVALID_HANDLE;
120b484b26cSGeorge Zhang
121b484b26cSGeorge Zhang return array->entries[index];
122b484b26cSGeorge Zhang }
123b484b26cSGeorge Zhang
vmci_handle_arr_has_entry(const struct vmci_handle_arr * array,struct vmci_handle entry_handle)124b484b26cSGeorge Zhang bool vmci_handle_arr_has_entry(const struct vmci_handle_arr *array,
125b484b26cSGeorge Zhang struct vmci_handle entry_handle)
126b484b26cSGeorge Zhang {
1271c2eb5b2SVishnu DASA u32 i;
128b484b26cSGeorge Zhang
129b484b26cSGeorge Zhang for (i = 0; i < array->size; i++)
130b484b26cSGeorge Zhang if (vmci_handle_is_equal(array->entries[i], entry_handle))
131b484b26cSGeorge Zhang return true;
132b484b26cSGeorge Zhang
133b484b26cSGeorge Zhang return false;
134b484b26cSGeorge Zhang }
135b484b26cSGeorge Zhang
136b484b26cSGeorge Zhang /*
137b484b26cSGeorge Zhang * NULL if the array is empty. Otherwise, a pointer to the array
138b484b26cSGeorge Zhang * of VMCI handles in the handle array.
139b484b26cSGeorge Zhang */
vmci_handle_arr_get_handles(struct vmci_handle_arr * array)140b484b26cSGeorge Zhang struct vmci_handle *vmci_handle_arr_get_handles(struct vmci_handle_arr *array)
141b484b26cSGeorge Zhang {
142b484b26cSGeorge Zhang if (array->size)
143b484b26cSGeorge Zhang return array->entries;
144b484b26cSGeorge Zhang
145b484b26cSGeorge Zhang return NULL;
146b484b26cSGeorge Zhang }
147