/* * Copyright (C) 2013-2014 Andreas Steffen * HSR Hochschule fuer Technik Rapperswil * * This program is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License as published by the * Free Software Foundation; either version 2 of the License, or (at your * option) any later version. See . * * This program is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License * for more details. */ #include "imc_swid_state.h" #include "libpts.h" #include "swid/swid_inventory.h" #include "swid/swid_error.h" #include "tcg/swid/tcg_swid_attr_req.h" #include "tcg/swid/tcg_swid_attr_tag_inv.h" #include "tcg/swid/tcg_swid_attr_tag_id_inv.h" #include #include #include #include #include #include #include #define SWID_GENERATOR "/usr/local/bin/swid_generator" /* IMC definitions */ static const char imc_name[] = "SWID"; static pen_type_t msg_types[] = { { PEN_TCG, PA_SUBTYPE_TCG_SWID } }; static imc_agent_t *imc_swid; /** * see section 3.8.1 of TCG TNC IF-IMC Specification 1.3 */ TNC_Result TNC_IMC_Initialize(TNC_IMCID imc_id, TNC_Version min_version, TNC_Version max_version, TNC_Version *actual_version) { if (imc_swid) { DBG1(DBG_IMC, "IMC \"%s\" has already been initialized", imc_name); return TNC_RESULT_ALREADY_INITIALIZED; } imc_swid = imc_agent_create(imc_name, msg_types, countof(msg_types), imc_id, actual_version); if (!imc_swid) { return TNC_RESULT_FATAL; } libpts_init(); if (min_version > TNC_IFIMC_VERSION_1 || max_version < TNC_IFIMC_VERSION_1) { DBG1(DBG_IMC, "no common IF-IMC version"); return TNC_RESULT_NO_COMMON_VERSION; } return TNC_RESULT_SUCCESS; } /** * see section 3.8.2 of TCG TNC IF-IMC Specification 1.3 */ TNC_Result TNC_IMC_NotifyConnectionChange(TNC_IMCID imc_id, TNC_ConnectionID connection_id, TNC_ConnectionState new_state) { imc_state_t *state; if (!imc_swid) { DBG1(DBG_IMC, "IMC \"%s\" has not been initialized", imc_name); return TNC_RESULT_NOT_INITIALIZED; } switch (new_state) { case TNC_CONNECTION_STATE_CREATE: state = imc_swid_state_create(connection_id); return imc_swid->create_state(imc_swid, state); case TNC_CONNECTION_STATE_HANDSHAKE: if (imc_swid->change_state(imc_swid, connection_id, new_state, &state) != TNC_RESULT_SUCCESS) { return TNC_RESULT_FATAL; } state->set_result(state, imc_id, TNC_IMV_EVALUATION_RESULT_DONT_KNOW); return TNC_RESULT_SUCCESS; case TNC_CONNECTION_STATE_DELETE: return imc_swid->delete_state(imc_swid, connection_id); default: return imc_swid->change_state(imc_swid, connection_id, new_state, NULL); } } /** * see section 3.8.3 of TCG TNC IF-IMC Specification 1.3 */ TNC_Result TNC_IMC_BeginHandshake(TNC_IMCID imc_id, TNC_ConnectionID connection_id) { imc_state_t *state; if (!imc_swid) { DBG1(DBG_IMC, "IMC \"%s\" has not been initialized", imc_name); return TNC_RESULT_NOT_INITIALIZED; } if (!imc_swid->get_state(imc_swid, connection_id, &state)) { return TNC_RESULT_FATAL; } return TNC_RESULT_SUCCESS; } /** * Add one or multiple SWID Inventory attributes to the send queue */ static bool add_swid_inventory(imc_state_t *state, imc_msg_t *msg, uint32_t request_id, bool full_tags, swid_inventory_t *targets) { pa_tnc_attr_t *attr, *attr_angel, *attr_error; imc_swid_state_t *swid_state; swid_inventory_t *swid_inventory; char *swid_directory, *swid_generator; uint32_t eid_epoch; size_t max_attr_size, attr_size, entry_size; bool first = TRUE, swid_pretty, swid_full; enumerator_t *enumerator; swid_directory = lib->settings->get_str(lib->settings, "%s.plugins.imc-swid.swid_directory", SWID_DIRECTORY, lib->ns); swid_generator = lib->settings->get_str(lib->settings, "%s.plugins.imc-swid.swid_generator", SWID_GENERATOR, lib->ns); swid_pretty = lib->settings->get_bool(lib->settings, "%s.plugins.imc-swid.swid_pretty", FALSE, lib->ns); swid_full = lib->settings->get_bool(lib->settings, "%s.plugins.imc-swid.swid_full", FALSE, lib->ns); swid_inventory = swid_inventory_create(full_tags); if (!swid_inventory->collect(swid_inventory, swid_directory, swid_generator, targets, swid_pretty, swid_full)) { swid_inventory->destroy(swid_inventory); attr_error = swid_error_create(TCG_SWID_ERROR, request_id, 0, "error in SWID tag collection"); msg->add_attribute(msg, attr_error); return FALSE; } DBG1(DBG_IMC, "collected %d SWID tag%s%s", swid_inventory->get_count(swid_inventory), full_tags ? "" : " ID", swid_inventory->get_count(swid_inventory) == 1 ? "" : "s"); swid_state = (imc_swid_state_t*)state; eid_epoch = swid_state->get_eid_epoch(swid_state); /** * Compute the maximum TCG SWID Tag [ID] Inventory attribute size * leaving space for an additional ITA Angel attribute */ max_attr_size = state->get_max_msg_len(state) - PA_TNC_HEADER_SIZE - PA_TNC_ATTR_HEADER_SIZE; if (full_tags) { tcg_swid_attr_tag_inv_t *swid_attr; swid_tag_t *tag; chunk_t encoding, tag_file_path; /* At least one TCG Tag Inventory attribute is sent */ attr_size = PA_TNC_ATTR_HEADER_SIZE + TCG_SWID_TAG_INV_MIN_SIZE; attr = tcg_swid_attr_tag_inv_create(request_id, eid_epoch, 1); enumerator = swid_inventory->create_enumerator(swid_inventory); while (enumerator->enumerate(enumerator, &tag)) { tag_file_path = tag->get_tag_file_path(tag); encoding = tag->get_encoding(tag); entry_size = 2 + tag_file_path.len + 4 + encoding.len; /* Check for oversize tags that cannot be transported */ if (PA_TNC_ATTR_HEADER_SIZE + TCG_SWID_TAG_INV_MIN_SIZE + entry_size > max_attr_size) { attr_error = swid_error_create(TCG_SWID_RESPONSE_TOO_LARGE, request_id, max_attr_size, "oversize SWID tag omitted"); msg->add_attribute(msg, attr_error); continue; } if (attr_size + entry_size > max_attr_size) { if (first) { /** * Send an ITA Start Angel attribute to the IMV signalling * that multiple TGC SWID Tag Inventory attributes follow */ attr_angel = ita_attr_angel_create(TRUE); msg->add_attribute(msg, attr_angel); first = FALSE; } msg->add_attribute(msg, attr); /* create the next TCG SWID Tag Inventory attribute */ attr_size = PA_TNC_ATTR_HEADER_SIZE + TCG_SWID_TAG_INV_MIN_SIZE; attr = tcg_swid_attr_tag_inv_create(request_id, eid_epoch, 1); } swid_attr = (tcg_swid_attr_tag_inv_t*)attr; swid_attr->add(swid_attr, tag->get_ref(tag)); attr_size += entry_size; } enumerator->destroy(enumerator); } else { tcg_swid_attr_tag_id_inv_t *swid_id_attr; swid_tag_id_t *tag_id; chunk_t tag_creator, unique_sw_id, tag_file_path; /* At least one TCG Tag ID Inventory attribute is sent */ attr_size = PA_TNC_ATTR_HEADER_SIZE + TCG_SWID_TAG_ID_INV_MIN_SIZE; attr = tcg_swid_attr_tag_id_inv_create(request_id, eid_epoch, 1); swid_id_attr = (tcg_swid_attr_tag_id_inv_t*)attr; enumerator = swid_inventory->create_enumerator(swid_inventory); while (enumerator->enumerate(enumerator, &tag_id)) { tag_creator = tag_id->get_tag_creator(tag_id); unique_sw_id = tag_id->get_unique_sw_id(tag_id, &tag_file_path); entry_size = 2 + tag_creator.len + 2 + unique_sw_id.len + 2 + tag_file_path.len; if (attr_size + entry_size > max_attr_size) { if (first) { /** * Send an ITA Start Angel attribute to the IMV signalling * that multiple TGC SWID Tag ID Inventory attributes follow */ attr_angel = ita_attr_angel_create(TRUE); msg->add_attribute(msg, attr_angel); first = FALSE; } msg->add_attribute(msg, attr); /* create the next TCG SWID Tag ID Inventory attribute */ attr_size = PA_TNC_ATTR_HEADER_SIZE + TCG_SWID_TAG_ID_INV_MIN_SIZE; attr = tcg_swid_attr_tag_id_inv_create(request_id, eid_epoch, 1); } swid_id_attr = (tcg_swid_attr_tag_id_inv_t*)attr; swid_id_attr->add(swid_id_attr, tag_id->get_ref(tag_id)); attr_size += entry_size; } enumerator->destroy(enumerator); } msg->add_attribute(msg, attr); swid_inventory->destroy(swid_inventory); if (!first) { /** * If we sent an ITA Start Angel attribute in the first place, * terminate by appending a matching ITA Stop Angel attribute. */ attr_angel = ita_attr_angel_create(FALSE); msg->add_attribute(msg, attr_angel); } return TRUE; } static TNC_Result receive_message(imc_state_t *state, imc_msg_t *in_msg) { imc_msg_t *out_msg; pa_tnc_attr_t *attr; enumerator_t *enumerator; pen_type_t type; TNC_Result result; bool fatal_error = FALSE; /* parse received PA-TNC message and handle local and remote errors */ result = in_msg->receive(in_msg, &fatal_error); if (result != TNC_RESULT_SUCCESS) { return result; } out_msg = imc_msg_create_as_reply(in_msg); /* analyze PA-TNC attributes */ enumerator = in_msg->create_attribute_enumerator(in_msg); while (enumerator->enumerate(enumerator, &attr)) { tcg_swid_attr_req_t *attr_req; uint8_t flags; uint32_t request_id; bool full_tags; swid_inventory_t *targets; type = attr->get_type(attr); if (type.vendor_id != PEN_TCG || type.type != TCG_SWID_REQUEST) { continue; } attr_req = (tcg_swid_attr_req_t*)attr; flags = attr_req->get_flags(attr_req); request_id = attr_req->get_request_id(attr_req); targets = attr_req->get_targets(attr_req); if (flags & (TCG_SWID_ATTR_REQ_FLAG_S | TCG_SWID_ATTR_REQ_FLAG_C)) { attr = swid_error_create(TCG_SWID_SUBSCRIPTION_DENIED, request_id, 0, "no subscription available yet"); out_msg->add_attribute(out_msg, attr); break; } full_tags = (flags & TCG_SWID_ATTR_REQ_FLAG_R) == 0; if (!add_swid_inventory(state, out_msg, request_id, full_tags, targets)) { break; } } enumerator->destroy(enumerator); if (fatal_error) { result = TNC_RESULT_FATAL; } else { result = out_msg->send(out_msg, TRUE); } out_msg->destroy(out_msg); return result; } /** * see section 3.8.4 of TCG TNC IF-IMC Specification 1.3 */ TNC_Result TNC_IMC_ReceiveMessage(TNC_IMCID imc_id, TNC_ConnectionID connection_id, TNC_BufferReference msg, TNC_UInt32 msg_len, TNC_MessageType msg_type) { imc_state_t *state; imc_msg_t *in_msg; TNC_Result result; if (!imc_swid) { DBG1(DBG_IMC, "IMC \"%s\" has not been initialized", imc_name); return TNC_RESULT_NOT_INITIALIZED; } if (!imc_swid->get_state(imc_swid, connection_id, &state)) { return TNC_RESULT_FATAL; } in_msg = imc_msg_create_from_data(imc_swid, state, connection_id, msg_type, chunk_create(msg, msg_len)); result = receive_message(state, in_msg); in_msg->destroy(in_msg); return result; } /** * see section 3.8.6 of TCG TNC IF-IMV Specification 1.3 */ TNC_Result TNC_IMC_ReceiveMessageLong(TNC_IMCID imc_id, TNC_ConnectionID connection_id, TNC_UInt32 msg_flags, TNC_BufferReference msg, TNC_UInt32 msg_len, TNC_VendorID msg_vid, TNC_MessageSubtype msg_subtype, TNC_UInt32 src_imv_id, TNC_UInt32 dst_imc_id) { imc_state_t *state; imc_msg_t *in_msg; TNC_Result result; if (!imc_swid) { DBG1(DBG_IMC, "IMC \"%s\" has not been initialized", imc_name); return TNC_RESULT_NOT_INITIALIZED; } if (!imc_swid->get_state(imc_swid, connection_id, &state)) { return TNC_RESULT_FATAL; } in_msg = imc_msg_create_from_long_data(imc_swid, state, connection_id, src_imv_id, dst_imc_id,msg_vid, msg_subtype, chunk_create(msg, msg_len)); result =receive_message(state, in_msg); in_msg->destroy(in_msg); return result; } /** * see section 3.8.7 of TCG TNC IF-IMC Specification 1.3 */ TNC_Result TNC_IMC_BatchEnding(TNC_IMCID imc_id, TNC_ConnectionID connection_id) { if (!imc_swid) { DBG1(DBG_IMC, "IMC \"%s\" has not been initialized", imc_name); return TNC_RESULT_NOT_INITIALIZED; } return TNC_RESULT_SUCCESS; } /** * see section 3.8.8 of TCG TNC IF-IMC Specification 1.3 */ TNC_Result TNC_IMC_Terminate(TNC_IMCID imc_id) { if (!imc_swid) { DBG1(DBG_IMC, "IMC \"%s\" has not been initialized", imc_name); return TNC_RESULT_NOT_INITIALIZED; } libpts_deinit(); imc_swid->destroy(imc_swid); imc_swid = NULL; return TNC_RESULT_SUCCESS; } /** * see section 4.2.8.1 of TCG TNC IF-IMC Specification 1.3 */ TNC_Result TNC_IMC_ProvideBindFunction(TNC_IMCID imc_id, TNC_TNCC_BindFunctionPointer bind_function) { if (!imc_swid) { DBG1(DBG_IMC, "IMC \"%s\" has not been initialized", imc_name); return TNC_RESULT_NOT_INITIALIZED; } return imc_swid->bind_functions(imc_swid, bind_function); }