support of reportMessageTypes() function

src/libcharon/plugins/tnc_imc/tnc_imc.c

@@ -38,6 +38,16 @@ struct private_tnc_imc_t {
 	 * ID of loaded IMC
 	 */
 	TNC_IMCID id;
+
+	/**
+	 * List of message types supported by IMC
+	 */
+	TNC_MessageTypeList supported_types;
+
+	/**
+	 * Number of supported message types
+	 */
+	TNC_UInt32 type_count;
 };
 
 METHOD(imc_t, set_id, void,
@@ -58,10 +68,27 @@ METHOD(imc_t, get_name, char*,
 	return this->name;
 }
 
+METHOD(imc_t, set_message_types, void,
+	private_tnc_imc_t *this, TNC_MessageTypeList supported_types,
+							 TNC_UInt32 type_count)
+{
+	free(this->supported_types);
+	this->supported_types = NULL;
+	this->type_count = type_count;
+	if (type_count && supported_types)
+	{
+		size_t size = type_count * sizeof(TNC_MessageType);
+
+		this->supported_types = malloc(size);
+		memcpy(this->supported_types, supported_types, size);
+	}
+}
+
 METHOD(imc_t, destroy, void,
 	private_tnc_imc_t *this)
 {
 	free(this->name);
+	free(this->supported_types);
 	free(this);
 }
 
@@ -78,6 +105,7 @@ imc_t* tnc_imc_create(char* name, char *filename)
 			.set_id = _set_id,
 			.get_id = _get_id,
 			.get_name = _get_name,
+			.set_message_types = _set_message_types,
 			.destroy = _destroy,
         },
 	);

src/libcharon/plugins/tnc_imc/tnc_imc_bind_function.c

@@ -27,7 +27,8 @@ TNC_Result TNC_TNCC_ReportMessageTypes(TNC_IMCID imc_id,
 									   TNC_UInt32 type_count)
 {
 	DBG2(DBG_TNC,"TNCC_ReportMessageTypes %u %u", imc_id, type_count);
-	return TNC_RESULT_SUCCESS;
+	return charon->imcs->set_message_types(charon->imcs, imc_id,
+										   supported_types, type_count);
 }
 
 /**

src/libcharon/plugins/tnc_imc/tnc_imc_manager.c

@@ -101,6 +101,30 @@ METHOD(imc_manager_t, begin_handshake, void,
 	enumerator->destroy(enumerator);
 }
 
+METHOD(imc_manager_t, set_message_types, TNC_Result,
+	private_tnc_imc_manager_t *this, TNC_IMCID id,
+									 TNC_MessageTypeList supported_types,
+									 TNC_UInt32 type_count)
+{
+	enumerator_t *enumerator;
+	imc_t *imc;
+	TNC_Result result = TNC_RESULT_FATAL;
+
+	enumerator = this->imcs->create_enumerator(this->imcs);
+	while (enumerator->enumerate(enumerator, &imc))
+	{
+		if (id == imc->get_id(imc))
+		{
+			imc->set_message_types(imc, supported_types, type_count);
+			result = TNC_RESULT_SUCCESS;
+			break;
+		}
+	}
+	enumerator->destroy(enumerator);
+	return result;
+}
+
+
 METHOD(imc_manager_t, destroy, void,
 	private_tnc_imc_manager_t *this)
 {
@@ -132,6 +156,7 @@ imc_manager_t* tnc_imc_manager_create(void)
 			.add = _add,
 			.notify_connection_change = _notify_connection_change,
 			.begin_handshake = _begin_handshake,
+			.set_message_types = _set_message_types,
 			.destroy = _destroy,
         },
 		.imcs = linked_list_create(),

src/libcharon/plugins/tnc_imv/tnc_imv.c

@@ -38,6 +38,16 @@ struct private_tnc_imv_t {
 	 * ID of loaded IMV
 	 */
 	TNC_IMVID id;
+
+	/**
+	 * List of message types supported by IMC
+	 */
+	TNC_MessageTypeList supported_types;
+
+	/**
+	 * Number of supported message types
+	 */
+	TNC_UInt32 type_count;
 };
 
 METHOD(imv_t, set_id, void,
@@ -58,10 +68,27 @@ METHOD(imv_t, get_name, char*,
 	return this->name;
 }
 
+METHOD(imv_t, set_message_types, void,
+	private_tnc_imv_t *this, TNC_MessageTypeList supported_types,
+							 TNC_UInt32 type_count)
+{
+	free(this->supported_types);
+	this->supported_types = NULL;
+	this->type_count = type_count;
+	if (type_count && supported_types)
+	{
+		size_t size = type_count * sizeof(TNC_MessageType);
+
+		this->supported_types = malloc(size);
+		memcpy(this->supported_types, supported_types, size);
+	}
+}
+
 METHOD(imv_t, destroy, void,
 	private_tnc_imv_t *this)
 {
 	free(this->name);
+	free(this->supported_types);
 	free(this);
 }
 
@@ -78,6 +105,7 @@ imv_t* tnc_imv_create(char *name, char *filename)
 			.set_id = _set_id,
 			.get_id = _get_id,
 			.get_name = _get_name,
+			.set_message_types = _set_message_types,
 			.destroy = _destroy,
         },
 	);

src/libcharon/plugins/tnc_imv/tnc_imv_bind_function.c

@@ -27,7 +27,8 @@ TNC_Result TNC_TNCS_ReportMessageTypes(TNC_IMVID imv_id,
 									   TNC_UInt32 type_count)
 {
 	DBG2(DBG_TNC,"TNCS_ReportMessageTypes %u %u", imv_id, type_count);
-	return TNC_RESULT_SUCCESS;
+	return charon->imvs->set_message_types(charon->imvs, imv_id,
+										   supported_types, type_count);
 }
 
 /**

src/libcharon/plugins/tnc_imv/tnc_imv_manager.c

@@ -70,7 +70,7 @@ METHOD(imv_manager_t, add, bool,
 
 METHOD(imv_manager_t, notify_connection_change, void,
 	private_tnc_imv_manager_t *this, TNC_ConnectionID id,
-	TNC_ConnectionState state)
+									 TNC_ConnectionState state)
 {
 	enumerator_t *enumerator;
 	imv_t *imv;
@@ -86,6 +86,29 @@ METHOD(imv_manager_t, notify_connection_change, void,
 	enumerator->destroy(enumerator);
 }
 
+METHOD(imv_manager_t, set_message_types, TNC_Result,
+	private_tnc_imv_manager_t *this, TNC_IMVID id,
+									 TNC_MessageTypeList supported_types,
+									 TNC_UInt32 type_count)
+{
+	enumerator_t *enumerator;
+	imv_t *imv;
+	TNC_Result result = TNC_RESULT_FATAL;
+
+	enumerator = this->imvs->create_enumerator(this->imvs);
+	while (enumerator->enumerate(enumerator, &imv))
+	{
+		if (id == imv->get_id(imv))
+		{
+			imv->set_message_types(imv, supported_types, type_count);
+			result = TNC_RESULT_SUCCESS;
+			break;
+		}
+	}
+	enumerator->destroy(enumerator);
+	return result;
+}
+
 METHOD(imv_manager_t, destroy, void,
 	private_tnc_imv_manager_t *this)
 {
@@ -116,6 +139,7 @@ imv_manager_t* tnc_imv_manager_create(void)
 		.public = {
 			.add = _add,
 			.notify_connection_change = _notify_connection_change,
+			.set_message_types = _set_message_types,
 			.destroy = _destroy,
         },
 		.imvs = linked_list_create(),