Introduced workitems to Attestation IMV
This commit is contained in:
@@ -184,8 +184,8 @@ static TNC_Result receive_msg(private_imv_os_agent_t *this, imv_state_t *state,
|
||||
ietf_attr_product_info_t *attr_cast;
|
||||
pen_t vendor_id;
|
||||
|
||||
os_state->set_received(os_state,
|
||||
IMV_OS_ATTR_PRODUCT_INFORMATION);
|
||||
state->set_action_flags(state,
|
||||
IMV_OS_ATTR_PRODUCT_INFORMATION);
|
||||
attr_cast = (ietf_attr_product_info_t*)attr;
|
||||
os_name = attr_cast->get_info(attr_cast, &vendor_id, NULL);
|
||||
if (vendor_id != PEN_IETF)
|
||||
@@ -205,8 +205,8 @@ static TNC_Result receive_msg(private_imv_os_agent_t *this, imv_state_t *state,
|
||||
{
|
||||
ietf_attr_string_version_t *attr_cast;
|
||||
|
||||
os_state->set_received(os_state,
|
||||
IMV_OS_ATTR_STRING_VERSION);
|
||||
state->set_action_flags(state,
|
||||
IMV_OS_ATTR_STRING_VERSION);
|
||||
attr_cast = (ietf_attr_string_version_t*)attr;
|
||||
os_version = attr_cast->get_version(attr_cast, NULL, NULL);
|
||||
if (os_version.len)
|
||||
@@ -221,8 +221,8 @@ static TNC_Result receive_msg(private_imv_os_agent_t *this, imv_state_t *state,
|
||||
ietf_attr_numeric_version_t *attr_cast;
|
||||
u_int32_t major, minor;
|
||||
|
||||
os_state->set_received(os_state,
|
||||
IMV_OS_ATTR_NUMERIC_VERSION);
|
||||
state->set_action_flags(state,
|
||||
IMV_OS_ATTR_NUMERIC_VERSION);
|
||||
attr_cast = (ietf_attr_numeric_version_t*)attr;
|
||||
attr_cast->get_version(attr_cast, &major, &minor);
|
||||
DBG1(DBG_IMV, "operating system numeric version is %d.%d",
|
||||
@@ -236,8 +236,8 @@ static TNC_Result receive_msg(private_imv_os_agent_t *this, imv_state_t *state,
|
||||
op_result_t op_result;
|
||||
time_t last_boot;
|
||||
|
||||
os_state->set_received(os_state,
|
||||
IMV_OS_ATTR_OPERATIONAL_STATUS);
|
||||
state->set_action_flags(state,
|
||||
IMV_OS_ATTR_OPERATIONAL_STATUS);
|
||||
attr_cast = (ietf_attr_op_status_t*)attr;
|
||||
op_status = attr_cast->get_status(attr_cast);
|
||||
op_result = attr_cast->get_result(attr_cast);
|
||||
@@ -252,8 +252,8 @@ static TNC_Result receive_msg(private_imv_os_agent_t *this, imv_state_t *state,
|
||||
ietf_attr_fwd_enabled_t *attr_cast;
|
||||
os_fwd_status_t fwd_status;
|
||||
|
||||
os_state->set_received(os_state,
|
||||
IMV_OS_ATTR_FORWARDING_ENABLED);
|
||||
state->set_action_flags(state,
|
||||
IMV_OS_ATTR_FORWARDING_ENABLED);
|
||||
attr_cast = (ietf_attr_fwd_enabled_t*)attr;
|
||||
fwd_status = attr_cast->get_status(attr_cast);
|
||||
DBG1(DBG_IMV, "IPv4 forwarding is %N",
|
||||
@@ -270,7 +270,7 @@ static TNC_Result receive_msg(private_imv_os_agent_t *this, imv_state_t *state,
|
||||
ietf_attr_default_pwd_enabled_t *attr_cast;
|
||||
bool default_pwd_status;
|
||||
|
||||
os_state->set_received(os_state,
|
||||
state->set_action_flags(state,
|
||||
IMV_OS_ATTR_FACTORY_DEFAULT_PWD_ENABLED);
|
||||
attr_cast = (ietf_attr_default_pwd_enabled_t*)attr;
|
||||
default_pwd_status = attr_cast->get_status(attr_cast);
|
||||
@@ -289,8 +289,8 @@ static TNC_Result receive_msg(private_imv_os_agent_t *this, imv_state_t *state,
|
||||
enumerator_t *e;
|
||||
status_t status;
|
||||
|
||||
os_state->set_received(os_state,
|
||||
IMV_OS_ATTR_INSTALLED_PACKAGES);
|
||||
state->set_action_flags(state,
|
||||
IMV_OS_ATTR_INSTALLED_PACKAGES);
|
||||
if (!this->db)
|
||||
{
|
||||
break;
|
||||
@@ -325,7 +325,7 @@ static TNC_Result receive_msg(private_imv_os_agent_t *this, imv_state_t *state,
|
||||
char *name;
|
||||
chunk_t value;
|
||||
|
||||
os_state->set_received(os_state, IMV_OS_ATTR_SETTINGS);
|
||||
state->set_action_flags(state, IMV_OS_ATTR_SETTINGS);
|
||||
|
||||
attr_cast = (ita_attr_settings_t*)attr;
|
||||
e = attr_cast->create_enumerator(attr_cast);
|
||||
@@ -349,7 +349,7 @@ static TNC_Result receive_msg(private_imv_os_agent_t *this, imv_state_t *state,
|
||||
int device_id;
|
||||
chunk_t value;
|
||||
|
||||
os_state->set_received(os_state, IMV_OS_ATTR_DEVICE_ID);
|
||||
state->set_action_flags(state, IMV_OS_ATTR_DEVICE_ID);
|
||||
|
||||
value = attr->get_value(attr);
|
||||
DBG1(DBG_IMV, "device ID is %.*s", value.len, value.ptr);
|
||||
@@ -464,7 +464,7 @@ METHOD(imv_agent_if_t, receive_message_long, TNC_Result,
|
||||
/**
|
||||
* Build an IETF Attribute Request attribute for missing attributes
|
||||
*/
|
||||
static pa_tnc_attr_t* build_attr_request(u_int received)
|
||||
static pa_tnc_attr_t* build_attr_request(u_int32_t received)
|
||||
{
|
||||
pa_tnc_attr_t *attr;
|
||||
ietf_attr_attr_request_t *attr_cast;
|
||||
@@ -517,7 +517,7 @@ METHOD(imv_agent_if_t, batch_ending, TNC_Result,
|
||||
TNC_Result result = TNC_RESULT_SUCCESS;
|
||||
bool no_workitems = TRUE;
|
||||
enumerator_t *enumerator;
|
||||
u_int received;
|
||||
u_int32_t received;
|
||||
|
||||
if (!this->agent->get_state(this->agent, id, &state))
|
||||
{
|
||||
@@ -525,14 +525,18 @@ METHOD(imv_agent_if_t, batch_ending, TNC_Result,
|
||||
}
|
||||
os_state = (imv_os_state_t*)state;
|
||||
handshake_state = os_state->get_handshake_state(os_state);
|
||||
received = os_state->get_received(os_state);
|
||||
received = state->get_action_flags(state);
|
||||
session = state->get_session(state);
|
||||
imv_id = this->agent->get_id(this->agent);
|
||||
|
||||
if (handshake_state == IMV_OS_STATE_END)
|
||||
{
|
||||
return TNC_RESULT_SUCCESS;
|
||||
}
|
||||
|
||||
/* create an empty out message - we might need it */
|
||||
out_msg = imv_msg_create(this->agent, state, id,
|
||||
this->agent->get_id(this->agent),
|
||||
TNC_IMCID_ANY, msg_types[0]);
|
||||
out_msg = imv_msg_create(this->agent, state, id, imv_id, TNC_IMCID_ANY,
|
||||
msg_types[0]);
|
||||
|
||||
if (handshake_state == IMV_OS_STATE_INIT)
|
||||
{
|
||||
@@ -627,7 +631,8 @@ METHOD(imv_agent_if_t, batch_ending, TNC_Result,
|
||||
|
||||
if (no_workitems)
|
||||
{
|
||||
DBG2(DBG_IMV, "no workitems generated - no evaluation requested");
|
||||
DBG2(DBG_IMV, "IMV %d has no workitems - "
|
||||
"no evaluation requested", imv_id);
|
||||
state->set_recommendation(state,
|
||||
TNC_IMV_ACTION_RECOMMENDATION_ALLOW,
|
||||
TNC_IMV_EVALUATION_RESULT_DONT_KNOW);
|
||||
@@ -727,6 +732,8 @@ METHOD(imv_agent_if_t, batch_ending, TNC_Result,
|
||||
/* finalized all workitems ? */
|
||||
if (session->get_workitem_count(session, imv_id) == 0)
|
||||
{
|
||||
os_state->set_handshake_state(os_state, IMV_OS_STATE_END);
|
||||
|
||||
result = out_msg->send_assessment(out_msg);
|
||||
out_msg->destroy(out_msg);
|
||||
if (result != TNC_RESULT_SUCCESS)
|
||||
|
||||
@@ -64,6 +64,11 @@ struct private_imv_os_state_t {
|
||||
*/
|
||||
u_int32_t max_msg_len;
|
||||
|
||||
/**
|
||||
* Flags set for completed actions
|
||||
*/
|
||||
u_int32_t action_flags;
|
||||
|
||||
/**
|
||||
* Access Requestor ID Type
|
||||
*/
|
||||
@@ -159,11 +164,6 @@ struct private_imv_os_state_t {
|
||||
*/
|
||||
int count_ok;
|
||||
|
||||
/**
|
||||
* Flags set for received attributes
|
||||
*/
|
||||
u_int received_flags;
|
||||
|
||||
/**
|
||||
* OS Settings
|
||||
*/
|
||||
@@ -337,6 +337,18 @@ METHOD(imv_state_t, get_max_msg_len, u_int32_t,
|
||||
return this->max_msg_len;
|
||||
}
|
||||
|
||||
METHOD(imv_state_t, set_action_flags, void,
|
||||
private_imv_os_state_t *this, u_int32_t flags)
|
||||
{
|
||||
this->action_flags |= flags;
|
||||
}
|
||||
|
||||
METHOD(imv_state_t, get_action_flags, u_int32_t,
|
||||
private_imv_os_state_t *this)
|
||||
{
|
||||
return this->action_flags;
|
||||
}
|
||||
|
||||
METHOD(imv_state_t, set_ar_id, void,
|
||||
private_imv_os_state_t *this, u_int32_t id_type, chunk_t id_value)
|
||||
{
|
||||
@@ -580,18 +592,6 @@ METHOD(imv_os_state_t, get_count, void,
|
||||
}
|
||||
}
|
||||
|
||||
METHOD(imv_os_state_t, set_received, void,
|
||||
private_imv_os_state_t *this, u_int flags)
|
||||
{
|
||||
this->received_flags |= flags;
|
||||
}
|
||||
|
||||
METHOD(imv_os_state_t, get_received, u_int,
|
||||
private_imv_os_state_t *this)
|
||||
{
|
||||
return this->received_flags;
|
||||
}
|
||||
|
||||
METHOD(imv_os_state_t, set_device_id, void,
|
||||
private_imv_os_state_t *this, int id)
|
||||
{
|
||||
@@ -660,6 +660,8 @@ imv_state_t *imv_os_state_create(TNC_ConnectionID connection_id)
|
||||
.set_flags = _set_flags,
|
||||
.set_max_msg_len = _set_max_msg_len,
|
||||
.get_max_msg_len = _get_max_msg_len,
|
||||
.set_action_flags = _set_action_flags,
|
||||
.get_action_flags = _get_action_flags,
|
||||
.set_ar_id = _set_ar_id,
|
||||
.get_ar_id = _get_ar_id,
|
||||
.set_session = _set_session,
|
||||
@@ -678,8 +680,6 @@ imv_state_t *imv_os_state_create(TNC_ConnectionID connection_id)
|
||||
.get_info = _get_info,
|
||||
.set_count = _set_count,
|
||||
.get_count = _get_count,
|
||||
.set_received = _set_received,
|
||||
.get_received = _get_received,
|
||||
.set_device_id = _set_device_id,
|
||||
.get_device_id = _get_device_id,
|
||||
.set_os_settings = _set_os_settings,
|
||||
|
||||
@@ -39,7 +39,8 @@ enum imv_os_handshake_state_t {
|
||||
IMV_OS_STATE_INIT,
|
||||
IMV_OS_STATE_ATTR_REQ,
|
||||
IMV_OS_STATE_POLICY_START,
|
||||
IMV_OS_STATE_WORKITEMS
|
||||
IMV_OS_STATE_WORKITEMS,
|
||||
IMV_OS_STATE_END
|
||||
};
|
||||
|
||||
/**
|
||||
@@ -119,20 +120,6 @@ struct imv_os_state_t {
|
||||
void (*get_count)(imv_os_state_t *this, int *count, int *count_update,
|
||||
int *count_blacklist, int *count_ok);
|
||||
|
||||
/**
|
||||
* Set flags for received attributes
|
||||
*
|
||||
* @param flags Flags to be set
|
||||
*/
|
||||
void (*set_received)(imv_os_state_t *this, u_int flags);
|
||||
|
||||
/**
|
||||
* Get flags set for received attributes
|
||||
*
|
||||
* @return Flags set for received attributes
|
||||
*/
|
||||
u_int (*get_received)(imv_os_state_t *this);
|
||||
|
||||
/**
|
||||
* Set device ID
|
||||
*
|
||||
|
||||
Reference in New Issue
Block a user