store detected improper OS settings in database

This commit is contained in:
Andreas Steffen
2012-11-30 00:12:48 +01:00
parent 78b2a2b148
commit 2b61f7baa3
4 changed files with 29 additions and 20 deletions
+4 -3
View File
@@ -374,7 +374,9 @@ static TNC_Result receive_message(imv_state_t *state, imv_msg_t *in_msg)
!os_state->get_angel_count(os_state)) !os_state->get_angel_count(os_state))
{ {
int device_id, count, count_update, count_blacklist, count_ok; int device_id, count, count_update, count_blacklist, count_ok;
u_int os_settings;
os_settings = os_state->get_os_settings(os_state);
os_state->get_count(os_state, &count, &count_update, &count_blacklist, os_state->get_count(os_state, &count, &count_update, &count_blacklist,
&count_ok); &count_ok);
DBG1(DBG_IMV, "processed %d packages: %d not updated, %d blacklisted, " DBG1(DBG_IMV, "processed %d packages: %d not updated, %d blacklisted, "
@@ -387,11 +389,10 @@ static TNC_Result receive_message(imv_state_t *state, imv_msg_t *in_msg)
{ {
os_db->set_device_info(os_db, device_id, os_db->set_device_info(os_db, device_id,
os_state->get_info(os_state, NULL, NULL, NULL), os_state->get_info(os_state, NULL, NULL, NULL),
count, count_update, count_blacklist); count, count_update, count_blacklist, os_settings);
} }
if (count_update || count_blacklist || if (count_update || count_blacklist || os_settings)
os_state->get_os_settings(os_state))
{ {
state->set_recommendation(state, state->set_recommendation(state,
TNC_IMV_ACTION_RECOMMENDATION_ISOLATE, TNC_IMV_ACTION_RECOMMENDATION_ISOLATE,
+14 -10
View File
@@ -215,11 +215,12 @@ METHOD(imv_os_database_t, get_device_id, int,
METHOD(imv_os_database_t, set_device_info, void, METHOD(imv_os_database_t, set_device_info, void,
private_imv_os_database_t *this, int device_id, char *os_info, private_imv_os_database_t *this, int device_id, char *os_info,
int count, int count_update, int count_blacklist) int count, int count_update, int count_blacklist, u_int flags)
{ {
enumerator_t *e; enumerator_t *e;
time_t last_time; time_t last_time;
int pid = 0, last_pid = 0, last_count_update = 0, last_count_blacklist = 0; int pid = 0, last_pid = 0, last_count_update = 0, last_count_blacklist = 0;
u_int last_flags;
bool found = FALSE; bool found = FALSE;
/* get primary key of OS info string if it exists */ /* get primary key of OS info string if it exists */
@@ -241,32 +242,35 @@ METHOD(imv_os_database_t, set_device_info, void,
/* get latest device info record if it exists */ /* get latest device info record if it exists */
e = this->db->query(this->db, e = this->db->query(this->db,
"SELECT time, product, count_update, count_blacklist " "SELECT time, product, count_update, count_blacklist, flags "
"FROM device_infos WHERE device = ? ORDER BY time DESC", "FROM device_infos WHERE device = ? ORDER BY time DESC",
DB_INT, device_id, DB_UINT, DB_INT, DB_INT, DB_INT); DB_INT, device_id, DB_UINT, DB_INT, DB_INT, DB_INT, DB_UINT);
if (e) if (e)
{ {
found = e->enumerate(e, &last_time, &last_pid, &last_count_update, found = e->enumerate(e, &last_time, &last_pid, &last_count_update,
&last_count_blacklist); &last_count_blacklist, &last_flags);
e->destroy(e); e->destroy(e);
} }
if (found && !last_count_update && !last_count_blacklist && pid == last_pid) if (found && !last_count_update && !last_count_blacklist && !last_flags &&
pid == last_pid)
{ {
/* update device info */ /* update device info */
this->db->execute(this->db, NULL, this->db->execute(this->db, NULL,
"UPDATE device_infos SET time = ?, count = ?, count_update = ?, " "UPDATE device_infos SET time = ?, count = ?, count_update = ?, "
"count_blacklist = ? WHERE device = ? AND time = ?", "count_blacklist = ?, flags = ? WHERE device = ? AND time = ?",
DB_UINT, time(NULL), DB_INT, count, DB_INT, count_update, DB_UINT, time(NULL), DB_INT, count, DB_INT, count_update,
DB_INT, count_blacklist, DB_INT, device_id, DB_UINT, last_time); DB_INT, count_blacklist, DB_UINT, flags,
DB_INT, device_id, DB_UINT, last_time);
} }
else else
{ {
/* insert device info */ /* insert device info */
this->db->execute(this->db, NULL, this->db->execute(this->db, NULL,
"INSERT INTO device_infos (device, time, product, " "INSERT INTO device_infos (device, time, product, count, "
"count, count_update, count_blacklist) VALUES (?, ?, ?, ?, ?, ?)", "count_update, count_blacklist, flags) VALUES (?, ?, ?, ?, ?, ?, ?)",
DB_INT, device_id, DB_UINT, time(NULL), DB_INT, pid, DB_INT, device_id, DB_UINT, time(NULL), DB_INT, pid,
DB_INT, count, DB_INT, count_update, DB_INT, count_blacklist); DB_INT, count, DB_INT, count_update, DB_INT, count_blacklist,
DB_UINT, flags);
} }
} }
+3 -1
View File
@@ -57,9 +57,11 @@ struct imv_os_database_t {
* @param count Number of installed packages * @param count Number of installed packages
* @param count_update Number of packages to be updated * @param count_update Number of packages to be updated
* @param count_blacklist Number of blacklisted packages * @param count_blacklist Number of blacklisted packages
* @param flags Various flags, e.g. illegal OS settings
*/ */
void (*set_device_info)(imv_os_database_t *this, int device_id, char *os_info, void (*set_device_info)(imv_os_database_t *this, int device_id, char *os_info,
int count, int count_update, int count_blacklist); int count, int count_update, int count_blacklist,
u_int flags);
/** /**
* Destroys an imv_os_database_t object. * Destroys an imv_os_database_t object.
@@ -799,19 +799,20 @@ METHOD(attest_db_t, list_devices, void,
time_t timestamp; time_t timestamp;
int id, last_id = 0, device_count = 0; int id, last_id = 0, device_count = 0;
int count, count_update, count_blacklist; int count, count_update, count_blacklist;
u_int tstamp, flags = 0;
e = this->db->query(this->db, e = this->db->query(this->db,
"SELECT d.id, d.value, i.time, i.count, i.count_update, " "SELECT d.id, d.value, i.time, i.count, i.count_update, "
"i.count_blacklist, p.name FROM devices AS d " "i.count_blacklist, i.flags, p.name FROM devices AS d "
"JOIN device_infos AS i ON d.id = i.device " "JOIN device_infos AS i ON d.id = i.device "
"JOIN products AS p ON p.id = i.product " "JOIN products AS p ON p.id = i.product "
"ORDER BY d.value, i.time DESC", "ORDER BY d.value, i.time DESC",
DB_INT, DB_BLOB, DB_UINT, DB_INT, DB_INT, DB_INT, DB_TEXT); DB_INT, DB_BLOB, DB_UINT, DB_INT, DB_INT, DB_INT, DB_UINT, DB_TEXT);
if (e) if (e)
{ {
while (e->enumerate(e, &id, &value, &timestamp, &count, &count_update, while (e->enumerate(e, &id, &value, &tstamp, &count, &count_update,
&count_blacklist, &product)) &count_blacklist, &flags, &product))
{ {
if (id != last_id) if (id != last_id)
{ {
@@ -819,8 +820,9 @@ METHOD(attest_db_t, list_devices, void,
device_count++; device_count++;
last_id = id; last_id = id;
} }
printf(" %T, %4d, %3d, %3d, '%s'\n", &timestamp, TRUE, timestamp = tstamp;
count, count_update, count_blacklist, product); printf(" %T, %4d, %3d, %3d, %1u, '%s'\n", &timestamp, TRUE,
count, count_update, count_blacklist, flags, product);
} }
e->destroy(e); e->destroy(e);
printf("%d device%s found\n", device_count, printf("%d device%s found\n", device_count,