feat(core/objpool): add objpool_alloc_with_id and objpool_get_by_id

This commit introduces two new methods to the objpool API.
The first method, objpool_alloc_with_id, enables the allocation of an
object within a pool and returns the index where the object was allocated.
The second method, objpool_get_by_id, allows retrieving a previously
allocated object using the index of the pool where it was initially allocated.

Signed-off-by: João Peixoto <[email protected]>

Author: joaopeixoto13

src/core/inc/objpool.h

@@ -32,6 +32,15 @@ struct objpool {
 
 void objpool_init(struct objpool* objpool);
 void* objpool_alloc(struct objpool* objpool);
+void* objpool_alloc_with_id(struct objpool* objpool, objpool_id_t* id);
 void objpool_free(struct objpool* objpool, void* obj);
 
+inline void* objpool_get_by_id(struct objpool* objpool, objpool_id_t id)
+{
+    if (id < objpool->num) {
+        return (void*)((uintptr_t)objpool->pool + (objpool->objsize * id));
+    }
+    return NULL;
+}
+
 #endif /* OBJPOOL_H */

src/core/inc/types.h

@@ -48,6 +48,8 @@ typedef unsigned irqid_t;
 
 typedef unsigned deviceid_t;
 
+typedef size_t objpool_id_t;
+
 typedef enum AS_SEC {
     /*--- HYP AS SECTIONS -----*/
     SEC_HYP_GLOBAL = 0,

src/core/objpool.c

@@ -12,7 +12,7 @@ void objpool_init(struct objpool* objpool)
     memset(objpool->bitmap, 0, BITMAP_SIZE(objpool->num));
 }
 
-void* objpool_alloc(struct objpool* objpool)
+void* objpool_alloc_with_id(struct objpool* objpool, objpool_id_t* id)
 {
     void* obj = NULL;
     spin_lock(&objpool->lock);
@@ -21,10 +21,18 @@ void* objpool_alloc(struct objpool* objpool)
         bitmap_set(objpool->bitmap, (size_t)n);
         obj = (void*)((uintptr_t)objpool->pool + (objpool->objsize * (size_t)n));
     }
+    if (id != NULL) {
+        *id = (objpool_id_t)n;
+    }
     spin_unlock(&objpool->lock);
     return obj;
 }
 
+void* objpool_alloc(struct objpool* objpool)
+{
+    return objpool_alloc_with_id(objpool, NULL);
+}
+
 void objpool_free(struct objpool* objpool, void* obj)
 {
     vaddr_t obj_addr = (vaddr_t)obj;