Pass context limits on to id manager

diff --git a/src/charon-tkm/src/tkm/tkm.c b/src/charon-tkm/src/tkm/tkm.c
index e913ddbdf03d10b45b44448fbeb507d130ac264c..f78d49be5f87236809ec0dc164c92bfd46298bac 100644 (file)
--- a/src/charon-tkm/src/tkm/tkm.c
+++ b/src/charon-tkm/src/tkm/tkm.c
@@ -75,9 +75,12 @@ bool tkm_init()
                return FALSE;
        }
 
+       /* only nonce and DH context limits are currently needed */
+       const tkm_limits_t limits = {nc, dh};
+
        INIT(this,
                .public = {
-                       .idmgr = tkm_id_manager_create(),
+                       .idmgr = tkm_id_manager_create(limits),
                },
        );
        tkm = &this->public;

diff --git a/src/charon-tkm/src/tkm/tkm_id_manager.c b/src/charon-tkm/src/tkm/tkm_id_manager.c
index 5a657fa43327e48b7ae7dc52e507e293c6efad82..8426a2cef93a9a2c9d0e8c717c8e22dd7f4f8aa4 100644 (file)
--- a/src/charon-tkm/src/tkm/tkm_id_manager.c
+++ b/src/charon-tkm/src/tkm/tkm_id_manager.c
@@ -132,7 +132,7 @@ METHOD(tkm_id_manager_t, destroy, void,
 /*
  * see header file
  */
-tkm_id_manager_t *tkm_id_manager_create()
+tkm_id_manager_t *tkm_id_manager_create(const tkm_limits_t limits)
 {
        private_tkm_id_manager_t *this;
        int i, j;

diff --git a/src/charon-tkm/src/tkm/tkm_id_manager.h b/src/charon-tkm/src/tkm/tkm_id_manager.h
index 9842979dca732a6b1c7ba421bf5f0045ea28a018..9fbdb2d722765670ff764f97235e90093a0a970b 100644 (file)
--- a/src/charon-tkm/src/tkm/tkm_id_manager.h
+++ b/src/charon-tkm/src/tkm/tkm_id_manager.h
@@ -40,6 +40,11 @@ enum tkm_context_kind_t {
  */
 extern enum_name_t *tkm_context_kind_names;
 
+/**
+ * TKM context limits.
+ */
+typedef uint64_t tkm_limits_t[TKM_CTX_MAX];
+
 /**
  * The tkm id manager hands out context ids for all context kinds (e.g. nonce).
  */
@@ -74,8 +79,8 @@ struct tkm_id_manager_t {
 };
 
 /**
- * Create a tkm id manager instance.
+ * Create a tkm id manager instance using the given context limits.
  */
-tkm_id_manager_t *tkm_id_manager_create();
+tkm_id_manager_t *tkm_id_manager_create(const tkm_limits_t limits);
 
 #endif /** TKM_ID_MANAGER_H_ */

diff --git a/src/charon-tkm/tests/id_manager_tests.c b/src/charon-tkm/tests/id_manager_tests.c
index a71727a383dcc9c4f8e8e6e9d44d6830454f513a..86e80aee0f256d72a9e9e75ae0c36249466b0ce5 100644 (file)
--- a/src/charon-tkm/tests/id_manager_tests.c
+++ b/src/charon-tkm/tests/id_manager_tests.c
@@ -14,16 +14,17 @@
  * for more details.
  */
 
-#include <stdlib.h>
 #include <check.h>
 
 #include "tkm_id_manager.h"
 
+static const tkm_limits_t limits = {125, 100};
+
 START_TEST(test_id_mgr_creation)
 {
        tkm_id_manager_t *idmgr = NULL;
 
-       idmgr = tkm_id_manager_create();
+       idmgr = tkm_id_manager_create(limits);
        fail_if(idmgr == NULL, "Error creating tkm id manager");
 
        idmgr->destroy(idmgr);
@@ -33,7 +34,7 @@ END_TEST
 START_TEST(test_acquire_id)
 {
        int i, id = 0;
-       tkm_id_manager_t *idmgr = tkm_id_manager_create();
+       tkm_id_manager_t *idmgr = tkm_id_manager_create(limits);
 
        for (i = 0; i < TKM_CTX_MAX; i++)
        {
@@ -51,7 +52,7 @@ END_TEST
 START_TEST(test_acquire_id_invalid_kind)
 {
        int id = 0;
-       tkm_id_manager_t *idmgr = tkm_id_manager_create();
+       tkm_id_manager_t *idmgr = tkm_id_manager_create(limits);
 
        id = idmgr->acquire_id(idmgr, TKM_CTX_MAX);
        fail_unless(id == 0, "Acquired id for invalid context kind %d", TKM_CTX_MAX);
@@ -69,7 +70,7 @@ END_TEST
 START_TEST(test_acquire_id_same)
 {
        int id1 = 0, id2 = 0;
-       tkm_id_manager_t *idmgr = tkm_id_manager_create();
+       tkm_id_manager_t *idmgr = tkm_id_manager_create(limits);
 
        id1 = idmgr->acquire_id(idmgr, TKM_CTX_NONCE);
        fail_unless(id1 > 0, "Unable to acquire first id");
@@ -87,7 +88,7 @@ START_TEST(test_release_id)
 {
        int i, id = 0;
        bool released = false;
-       tkm_id_manager_t *idmgr = tkm_id_manager_create();
+       tkm_id_manager_t *idmgr = tkm_id_manager_create(limits);
 
        for (i = 0; i < TKM_CTX_MAX; i++)
        {
@@ -107,7 +108,7 @@ END_TEST
 START_TEST(test_release_id_invalid_kind)
 {
        bool released = TRUE;
-       tkm_id_manager_t *idmgr = tkm_id_manager_create();
+       tkm_id_manager_t *idmgr = tkm_id_manager_create(limits);
 
        released = idmgr->release_id(idmgr, TKM_CTX_MAX, 1);
        fail_if(released, "Released id for invalid context kind %d", TKM_CTX_MAX);
@@ -125,7 +126,7 @@ END_TEST
 START_TEST(test_release_id_nonexistent)
 {
        bool released = FALSE;
-       tkm_id_manager_t *idmgr = tkm_id_manager_create();
+       tkm_id_manager_t *idmgr = tkm_id_manager_create(limits);
 
        released = idmgr->release_id(idmgr, TKM_CTX_NONCE, 1);
        fail_unless(released, "Release of nonexistent id failed");