embb_mtapi_id_pool_t.c 3.71 KB
Newer Older
1
/*
2
 * Copyright (c) 2014-2015, Siemens AG. All rights reserved.
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include <assert.h>

#include <embb_mtapi_alloc.h>
#include <embb_mtapi_log.h>
#include <embb_mtapi_id_pool_t.h>
32
#include <embb/base/c/mutex.h>
33 34 35 36 37 38 39 40

void embb_mtapi_id_pool_initialize(
  embb_mtapi_id_pool_t * that,
  mtapi_uint_t capacity) {
  mtapi_uint_t ii;

  that->capacity = capacity;
  that->id_buffer = (mtapi_uint_t*)
41
    embb_mtapi_alloc_allocate(sizeof(mtapi_uint_t)*(capacity + 1));
42
  that->id_buffer[0] = EMBB_MTAPI_IDPOOL_INVALID_ID;
43
  for (ii = 1; ii <= capacity; ii++) {
44 45
    that->id_buffer[ii] = ii;
  }
46
  that->ids_available = capacity;
47 48
  that->put_id_position = 0;
  that->get_id_position = 1;
49
  embb_spin_init(&that->lock);
50 51 52 53
}

void embb_mtapi_id_pool_finalize(embb_mtapi_id_pool_t * that) {
  that->capacity = 0;
54
  that->ids_available = 0;
55 56 57 58
  that->get_id_position = 0;
  that->put_id_position = 0;
  embb_mtapi_alloc_deallocate(that->id_buffer);
  that->id_buffer = NULL;
59
  embb_spin_destroy(&that->lock);
60 61 62 63 64 65 66
}

mtapi_uint_t embb_mtapi_id_pool_allocate(embb_mtapi_id_pool_t * that) {
  mtapi_uint_t id = EMBB_MTAPI_IDPOOL_INVALID_ID;

  assert(MTAPI_NULL != that);

67
  if (embb_spin_lock(&that->lock) == EMBB_SUCCESS) {
68
    if (0 < that->ids_available) {
69
      /* take away one id */
70
      that->ids_available--;
71 72 73 74

      /* acquire position to fetch id from */
      mtapi_uint_t id_position = that->get_id_position;
      that->get_id_position++;
75
      if (that->capacity < that->get_id_position) {
76 77 78 79 80 81 82 83 84
        that->get_id_position = 0;
      }

      /* fetch id */
      id = that->id_buffer[id_position];

      /* make id entry invalid just in case */
      that->id_buffer[id_position] = EMBB_MTAPI_IDPOOL_INVALID_ID;
    }
85
    embb_spin_unlock(&that->lock);
86 87 88 89 90 91 92 93 94 95
  }

  return id;
}

void embb_mtapi_id_pool_deallocate(
  embb_mtapi_id_pool_t * that,
  mtapi_uint_t id) {
  assert(MTAPI_NULL != that);

96
  if (embb_spin_lock(&that->lock) == EMBB_SUCCESS) {
97
    if (that->capacity > that->ids_available) {
98 99 100
      /* acquire position to put id to */
      mtapi_uint_t id_position = that->put_id_position;
      that->put_id_position++;
101
      if (that->capacity < that->put_id_position) {
102 103 104 105 106 107 108
        that->put_id_position = 0;
      }

      /* put id back into buffer */
      that->id_buffer[id_position] = id;

      /* make it available */
109
      that->ids_available++;
110
    }
111
    embb_spin_unlock(&that->lock);
112 113 114 115 116
  } else {
    embb_mtapi_log_error(
      "could not acquire lock in embb_mtapi_IdPool_deallocate\n");
  }
}