/* * Copyright (C) 2011-2012 Sansar Choinyambuu * Copyright (C) 2011-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_attestation_state.h" #include #include #include #include typedef struct private_imc_attestation_state_t private_imc_attestation_state_t; typedef struct func_comp_t func_comp_t; /** * Private data of an imc_attestation_state_t object. */ struct private_imc_attestation_state_t { /** * Public members of imc_attestation_state_t */ imc_attestation_state_t public; /** * TNCCS connection ID */ TNC_ConnectionID connection_id; /** * TNCCS connection state */ TNC_ConnectionState state; /** * Assessment/Evaluation Result */ TNC_IMV_Evaluation_Result result; /** * Does the TNCCS connection support long message types? */ bool has_long; /** * Does the TNCCS connection support exclusive delivery? */ bool has_excl; /** * Maximum PA-TNC message size for this TNCCS connection */ uint32_t max_msg_len; /** * PA-TNC attribute segmentation contracts associated with TNCCS connection */ seg_contract_manager_t *contracts; /** * PTS object */ pts_t *pts; /** * List of Functional Components */ linked_list_t *components; /** * Functional Component Evidence cache list */ linked_list_t *list; }; METHOD(imc_state_t, get_connection_id, TNC_ConnectionID, private_imc_attestation_state_t *this) { return this->connection_id; } METHOD(imc_state_t, has_long, bool, private_imc_attestation_state_t *this) { return this->has_long; } METHOD(imc_state_t, has_excl, bool, private_imc_attestation_state_t *this) { return this->has_excl; } METHOD(imc_state_t, set_flags, void, private_imc_attestation_state_t *this, bool has_long, bool has_excl) { this->has_long = has_long; this->has_excl = has_excl; } METHOD(imc_state_t, set_max_msg_len, void, private_imc_attestation_state_t *this, uint32_t max_msg_len) { this->max_msg_len = max_msg_len; } METHOD(imc_state_t, get_max_msg_len, uint32_t, private_imc_attestation_state_t *this) { return this->max_msg_len; } METHOD(imc_state_t, get_contracts, seg_contract_manager_t*, private_imc_attestation_state_t *this) { return this->contracts; } METHOD(imc_state_t, change_state, TNC_ConnectionState, private_imc_attestation_state_t *this, TNC_ConnectionState new_state) { TNC_ConnectionState old_state; old_state = this->state; this->state = new_state; return old_state; } METHOD(imc_state_t, set_result, void, private_imc_attestation_state_t *this, TNC_IMCID id, TNC_IMV_Evaluation_Result result) { this->result = result; } METHOD(imc_state_t, get_result, bool, private_imc_attestation_state_t *this, TNC_IMCID id, TNC_IMV_Evaluation_Result *result) { if (result) { *result = this->result; } return this->result != TNC_IMV_EVALUATION_RESULT_DONT_KNOW; } METHOD(imc_state_t, reset, void, private_imc_attestation_state_t *this) { this->result = TNC_IMV_EVALUATION_RESULT_DONT_KNOW; this->components->destroy_offset(this->components, offsetof(pts_component_t, destroy)); this->components = linked_list_create(); this->list->destroy_offset(this->list, offsetof(pts_comp_evidence_t, destroy)); this->list = linked_list_create(); this->pts->destroy(this->pts); this->pts = pts_create(TRUE); } METHOD(imc_state_t, destroy, void, private_imc_attestation_state_t *this) { this->pts->destroy(this->pts); this->components->destroy_offset(this->components, offsetof(pts_component_t, destroy)); this->list->destroy_offset(this->list, offsetof(pts_comp_evidence_t, destroy)); this->contracts->destroy(this->contracts); free(this); } METHOD(imc_attestation_state_t, get_pts, pts_t*, private_imc_attestation_state_t *this) { return this->pts; } METHOD(imc_attestation_state_t, create_component, pts_component_t*, private_imc_attestation_state_t *this, pts_comp_func_name_t *name, uint32_t depth) { enumerator_t *enumerator; pts_component_t *component; bool found = FALSE; enumerator = this->components->create_enumerator(this->components); while (enumerator->enumerate(enumerator, &component)) { if (name->equals(name, component->get_comp_func_name(component))) { found = TRUE; break; } } enumerator->destroy(enumerator); if (!found) { component = imcv_pts_components->create(imcv_pts_components, name, depth, NULL); if (!component) { return NULL; } this->components->insert_last(this->components, component); } return component; } METHOD(imc_attestation_state_t, add_evidence, void, private_imc_attestation_state_t *this, pts_comp_evidence_t *evid) { this->list->insert_last(this->list, evid); } METHOD(imc_attestation_state_t, next_evidence, bool, private_imc_attestation_state_t *this, pts_comp_evidence_t **evid) { return this->list->remove_first(this->list, (void**)evid) == SUCCESS; } /** * Described in header. */ imc_state_t *imc_attestation_state_create(TNC_ConnectionID connection_id) { private_imc_attestation_state_t *this; INIT(this, .public = { .interface = { .get_connection_id = _get_connection_id, .has_long = _has_long, .has_excl = _has_excl, .set_flags = _set_flags, .set_max_msg_len = _set_max_msg_len, .get_max_msg_len = _get_max_msg_len, .get_contracts = _get_contracts, .change_state = _change_state, .set_result = _set_result, .get_result = _get_result, .reset = _reset, .destroy = _destroy, }, .get_pts = _get_pts, .create_component = _create_component, .add_evidence = _add_evidence, .next_evidence = _next_evidence, }, .connection_id = connection_id, .state = TNC_CONNECTION_STATE_CREATE, .result = TNC_IMV_EVALUATION_RESULT_DONT_KNOW, .contracts = seg_contract_manager_create(), .pts = pts_create(TRUE), .components = linked_list_create(), .list = linked_list_create(), ); return &this->public.interface; }