moved management of additional IMC/IMV IDs to agent

src/libimcv/plugins/imc_test/imc_test.c

@@ -73,7 +73,6 @@ TNC_Result TNC_IMC_NotifyConnectionChange(TNC_IMCID imc_id,
 	imc_state_t *state;
 	imc_test_state_t *test_state;
 	TNC_Result result;
-	TNC_UInt32 new_imc_id;
 	char *command;
 	bool retry;
 	int additional_ids;
@@ -92,7 +91,6 @@ TNC_Result TNC_IMC_NotifyConnectionChange(TNC_IMCID imc_id,
 			retry = lib->settings->get_bool(lib->settings,
 								"libimcv.plugins.imc-test.retry", FALSE);
 			state = imc_test_state_create(connection_id, command, retry);
-			test_state = (imc_test_state_t*)state;
 
 			result = imc_test->create_state(imc_test, state);
 			if (result != TNC_RESULT_SUCCESS)
@@ -100,35 +98,12 @@ TNC_Result TNC_IMC_NotifyConnectionChange(TNC_IMCID imc_id,
 				return result;
 			}
 
-			/* Do we want to reserve additional IMC IDs? */
+			/* Optionally reserve additional IMC IDs */
 			additional_ids = lib->settings->get_int(lib->settings,
-						 		"libimcv.plugins.imc-test.additional_ids", 0);
-			if (additional_ids < 1)
-			{
-				return TNC_RESULT_SUCCESS;
-			}
+								"libimcv.plugins.imc-test.additional_ids", 0);
+			imc_test->reserve_additional_ids(imc_test, additional_ids -
+								imc_test->count_additional_ids(imc_test));
 
-			if (!state->has_long(state))
-			{
-				DBG1(DBG_IMC, "IMC %u \"%s\" did not detect support of "
-							   "multiple IMC IDs", imc_id, imc_name);
-				return TNC_RESULT_SUCCESS;
-			}
-
-			while (additional_ids-- > 0)
-			{
-				if (imc_test->reserve_additional_id(imc_test, &new_imc_id) !=
-					TNC_RESULT_SUCCESS)
-				{
-					DBG1(DBG_IMC, "IMC %u \"%s\" failed to reserve "
-								  "%d additional IMC IDs",
-								   imc_id, imc_name, additional_ids);
-					break;
-				}
-				DBG2(DBG_IMC, "IMC %u \"%s\" reserved additional ID %u",
-							   imc_id, imc_name, new_imc_id);
-				test_state->add_id(test_state, new_imc_id);
-			}
 			return TNC_RESULT_SUCCESS;
 
 		case TNC_CONNECTION_STATE_HANDSHAKE:
@@ -211,7 +186,6 @@ TNC_Result TNC_IMC_BeginHandshake(TNC_IMCID imc_id,
 								  TNC_ConnectionID connection_id)
 {
 	imc_state_t *state;
-	imc_test_state_t *test_state;
 	enumerator_t *enumerator;
 	void *pointer;
 	TNC_UInt32 additional_id;
@@ -228,13 +202,26 @@ TNC_Result TNC_IMC_BeginHandshake(TNC_IMCID imc_id,
 	{
 		return TNC_RESULT_FATAL;
 	}
-	test_state = (imc_test_state_t*)state;
 
 	/* send PA message for primary IMC ID */
 	result = send_message(state, imc_id, TNC_IMVID_ANY);
-	
+
+	/* Exit if there are no additional IMC IDs */
+	if (!imc_test->count_additional_ids(imc_test))
+	{
+		return result;
+	}
+
+	/* Do we have support for transporting multiple IMC IDs? */
+	if (!state->has_long(state))
+	{
+		DBG1(DBG_IMC, "IMC %u \"%s\" did not detect support for transporting "
+					  "multiple IMC IDs", imc_id, imc_name);
+		return result;
+	}
+
 	/* send PA messages for additional IMC IDs */
-	enumerator = test_state->create_id_enumerator(test_state);
+	enumerator = imc_test->create_id_enumerator(imc_test);
 	while (result == TNC_RESULT_SUCCESS &&
 		   enumerator->enumerate(enumerator, &pointer))
 	{

src/libimcv/plugins/imc_test/imc_test_state.c

@@ -63,11 +63,6 @@ struct private_imc_test_state_t {
 	 * Do a handshake retry
 	 */
 	bool handshake_retry;
-
-	/**
-	 * List of additional IMC IDs
-	 */
-	linked_list_t *additional_ids;
 	
 };
 
@@ -105,7 +100,6 @@ METHOD(imc_state_t, change_state, void,
 METHOD(imc_state_t, destroy, void,
 	private_imc_test_state_t *this)
 {
-	this->additional_ids->destroy(this->additional_ids);
 	free(this->command);
 	free(this);
 }
@@ -148,22 +142,6 @@ METHOD(imc_test_state_t, do_handshake_retry, bool,
 	return retry;
 }
 
-METHOD(imc_test_state_t, add_id, void,
-	private_imc_test_state_t *this, TNC_IMCID id)
-{
-	void *pointer;
-
-	/* store the scalar value in the pointer */
-	pointer = (void*)id;
-	this->additional_ids->insert_last(this->additional_ids, pointer);
-}
-
-METHOD(imc_test_state_t, create_id_enumerator, enumerator_t*,
-	private_imc_test_state_t *this)
-{
-	return this->additional_ids->create_enumerator(this->additional_ids);
-}
-
 /**
  * Described in header.
  */
@@ -186,15 +164,12 @@ imc_state_t *imc_test_state_create(TNC_ConnectionID connection_id,
 			.set_command = _set_command,
 			.is_first_handshake = _is_first_handshake,
 			.do_handshake_retry = _do_handshake_retry,
-			.add_id = _add_id,
-			.create_id_enumerator = _create_id_enumerator,
 		},
 		.state = TNC_CONNECTION_STATE_CREATE,
 		.connection_id = connection_id,
 		.command = strdup(command),
 		.first_handshake = TRUE,
 		.handshake_retry = retry,
-		.additional_ids = linked_list_create(),
 	);
 	
 	return &this->public.interface;

src/libimcv/plugins/imc_test/imc_test_state.h

@@ -65,17 +65,6 @@ struct imc_test_state_t {
 	 */
 	bool (*do_handshake_retry)(imc_test_state_t *this);
 
-	/**
-	 * Add and additional IMC ID
-	 *
-	 * @param id			Additional IMC ID
-	 */
-	void (*add_id)(imc_test_state_t *this, TNC_IMCID id);
-
-	/**
-	 * Create an enumerator for additional IMC IDs
-	 */
-	enumerator_t* (*create_id_enumerator)(imc_test_state_t *this);
 };
 
 /**