src/libs/compat/freebsd_network/unit.c

   1 /*
   2  * Copyright 2009 Colin Günther, coling@gmx.de
   3  * All Rights Reserved. Distributed under the terms of the MIT License.
   4  *
   5  */
   6
   7
   8 /*! Implementation of a number allocator.*/
   9
  10
  11 #include "unit.h"
  12
  13 #include <compat/sys/mutex.h>
  14
  15 #include <stdlib.h>
  16
  17
  18 extern struct mtx gIdStoreLock;
  19
  20
  21 struct unrhdr*
  22 new_unrhdr(int low, int high, struct mtx* mutex)
  23 {
  24         struct unrhdr* idStore;
  25         uint32 maxIdCount = high - low + 1;
  26
  27         KASSERT(low <= high,
  28                 ("ID-Store: use error: %s(%u, %u)", __func__, low, high));
  29
  30         idStore = malloc(sizeof *idStore);
  31         if (idStore == NULL)
  32                 return NULL;
  33
  34         if (_new_unrhdr_buffer(idStore, maxIdCount) != B_OK) {
  35                 free(idStore);
  36                 return NULL;
  37         }
  38
  39         if (mutex)
  40                 idStore->storeMutex = mutex;
  41         else
  42                 idStore->storeMutex = &gIdStoreLock;
  43
  44         idStore->idBias = low;
  45
  46         return idStore;
  47 }
  48
  49
  50 void
  51 delete_unrhdr(struct unrhdr* idStore)
  52 {
  53         KASSERT(uh != NULL,
  54                 ("ID-Store: %s: NULL pointer as argument.", __func__));
  55
  56         mtx_lock(idStore->storeMutex);
  57
  58         KASSERT(uh->idBuffer->root_size == 0,
  59                 ("ID-Store: %s: some ids are still in use..", __func__));
  60
  61         _delete_unrhdr_buffer_locked(idStore);
  62         mtx_unlock(idStore->storeMutex);
  63
  64         free(idStore);
  65         idStore = NULL;
  66 }
  67
  68
  69 int
  70 alloc_unr(struct unrhdr* idStore)
  71 {
  72         int id;
  73
  74         KASSERT(uh != NULL,
  75                 ("ID-Store: %s: NULL pointer as argument.", __func__));
  76
  77         mtx_lock(idStore->storeMutex);
  78         id = _alloc_unr_locked(idStore);
  79         mtx_unlock(idStore->storeMutex);
  80
  81         return id;
  82 }
  83
  84
  85 void
  86 free_unr(struct unrhdr* idStore, u_int identity)
  87 {
  88         KASSERT(uh != NULL,
  89                 ("ID-Store: %s: NULL pointer as argument.", __func__));
  90
  91         mtx_lock(idStore->storeMutex);
  92
  93         KASSERT((int32)item - uh->idBias >= 0, ("ID-Store: %s(%p, %u): second "
  94                 + "parameter is not in interval.", __func__, uh, item));
  95
  96         _free_unr_locked(idStore, identity);
  97
  98         mtx_unlock(idStore->storeMutex);
  99 }