mirror of
https://github.com/BlackLight/Snort_AIPreproc.git
synced 2024-11-24 04:35:11 +01:00
Avoiding duplicates in neural clusters
This commit is contained in:
parent
b49e9152ec
commit
593a752ecf
1 changed files with 38 additions and 3 deletions
|
@ -45,7 +45,7 @@
|
||||||
PRIVATE void
|
PRIVATE void
|
||||||
__AI_neural_clusters_to_xml ( kmeans_t *km, AI_alerts_per_neuron *alerts_per_neuron )
|
__AI_neural_clusters_to_xml ( kmeans_t *km, AI_alerts_per_neuron *alerts_per_neuron )
|
||||||
{
|
{
|
||||||
int i, j, k, l, are_equal;
|
int i, j, k, l, m, n, are_equal;
|
||||||
FILE *fp = NULL;
|
FILE *fp = NULL;
|
||||||
|
|
||||||
uint32_t src_addr = 0,
|
uint32_t src_addr = 0,
|
||||||
|
@ -57,8 +57,9 @@ __AI_neural_clusters_to_xml ( kmeans_t *km, AI_alerts_per_neuron *alerts_per_neu
|
||||||
*tmp = NULL,
|
*tmp = NULL,
|
||||||
*buf = NULL;
|
*buf = NULL;
|
||||||
|
|
||||||
AI_alerts_per_neuron_key key;
|
AI_alerts_per_neuron_key key, tmp_key;
|
||||||
AI_alerts_per_neuron *alert_iterator = NULL;
|
AI_alerts_per_neuron *alert_iterator = NULL,
|
||||||
|
*tmp_iterator = NULL;
|
||||||
|
|
||||||
if ( !( fp = fopen ( config->neural_clusters_log, "w" )))
|
if ( !( fp = fopen ( config->neural_clusters_log, "w" )))
|
||||||
{
|
{
|
||||||
|
@ -106,6 +107,40 @@ __AI_neural_clusters_to_xml ( kmeans_t *km, AI_alerts_per_neuron *alerts_per_neu
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/* If no duplicate alert was found on the same neuron, check
|
||||||
|
* that there is no duplicate alert on other neurons */
|
||||||
|
if ( !are_equal )
|
||||||
|
{
|
||||||
|
for ( l=0; l <= i && !are_equal; l++ )
|
||||||
|
{
|
||||||
|
for ( m=0; m < j && !are_equal; m++ )
|
||||||
|
{
|
||||||
|
tmp_key.x = km->clusters[l][m][0];
|
||||||
|
tmp_key.y = km->clusters[l][m][1];
|
||||||
|
HASH_FIND ( hh, alerts_per_neuron, &tmp_key, sizeof ( tmp_key ), tmp_iterator );
|
||||||
|
|
||||||
|
if ( tmp_iterator )
|
||||||
|
{
|
||||||
|
for ( n=0; n < tmp_iterator->n_alerts && !are_equal; n++ )
|
||||||
|
{
|
||||||
|
if (
|
||||||
|
alert_iterator->alerts[k].gid == tmp_iterator->alerts[n].gid &&
|
||||||
|
alert_iterator->alerts[k].sid == tmp_iterator->alerts[n].sid &&
|
||||||
|
alert_iterator->alerts[k].rev == tmp_iterator->alerts[n].rev &&
|
||||||
|
alert_iterator->alerts[k].src_ip_addr == tmp_iterator->alerts[n].src_ip_addr &&
|
||||||
|
alert_iterator->alerts[k].dst_ip_addr == tmp_iterator->alerts[n].dst_ip_addr &&
|
||||||
|
alert_iterator->alerts[k].src_port == tmp_iterator->alerts[n].src_port &&
|
||||||
|
alert_iterator->alerts[k].dst_port == tmp_iterator->alerts[n].dst_port &&
|
||||||
|
alert_iterator->alerts[k].timestamp == tmp_iterator->alerts[n].timestamp )
|
||||||
|
{
|
||||||
|
are_equal = 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if ( !are_equal )
|
if ( !are_equal )
|
||||||
{
|
{
|
||||||
src_addr = htonl ( alert_iterator->alerts[k].src_ip_addr );
|
src_addr = htonl ( alert_iterator->alerts[k].src_ip_addr );
|
||||||
|
|
Loading…
Reference in a new issue