|
| 1 | +/** |
| 2 | + * \file nettisa.cpp |
| 3 | + * \brief Plugin for creating NetTiSA flow. |
| 4 | + * \author Josef Koumar [email protected] |
| 5 | + * \date 2023 |
| 6 | + */ |
| 7 | + |
| 8 | +#include "nettisa.hpp" |
| 9 | + |
| 10 | +#include <cmath> |
| 11 | +#include <iostream> |
| 12 | + |
| 13 | +namespace ipxp { |
| 14 | + |
| 15 | +int RecordExtNETTISA::REGISTERED_ID = -1; |
| 16 | + |
| 17 | +__attribute__((constructor)) static void register_this_plugin() |
| 18 | +{ |
| 19 | + static PluginRecord rec = PluginRecord("nettisa", []() { return new NETTISAPlugin(); }); |
| 20 | + register_plugin(&rec); |
| 21 | + RecordExtNETTISA::REGISTERED_ID = register_extension(); |
| 22 | +} |
| 23 | + |
| 24 | +NETTISAPlugin::NETTISAPlugin() {} |
| 25 | + |
| 26 | +NETTISAPlugin::~NETTISAPlugin() {} |
| 27 | + |
| 28 | +void NETTISAPlugin::init(const char* params) {} |
| 29 | + |
| 30 | +void NETTISAPlugin::close() {} |
| 31 | + |
| 32 | +ProcessPlugin* NETTISAPlugin::copy() |
| 33 | +{ |
| 34 | + return new NETTISAPlugin(*this); |
| 35 | +} |
| 36 | + |
| 37 | +void NETTISAPlugin::update_record( |
| 38 | + RecordExtNETTISA* nettisa_data, |
| 39 | + const Packet& pkt, |
| 40 | + const Flow& rec) |
| 41 | +{ |
| 42 | + float variation_from_mean = pkt.payload_len_wire - nettisa_data->mean; |
| 43 | + uint32_t n = rec.dst_packets + rec.src_packets; |
| 44 | + long diff_time = pkt.ts.tv_usec - nettisa_data->prev_time; |
| 45 | + nettisa_data->prev_time = pkt.ts.tv_usec; |
| 46 | + // MEAN |
| 47 | + nettisa_data->mean += (variation_from_mean) / n; |
| 48 | + // MIN |
| 49 | + nettisa_data->min = std::min(nettisa_data->min, pkt.payload_len_wire); |
| 50 | + // MAX |
| 51 | + nettisa_data->max = std::max(nettisa_data->max, pkt.payload_len_wire); |
| 52 | + // ROOT MEAN SQUARE |
| 53 | + nettisa_data->root_mean_square += pow(pkt.payload_len_wire, 2); |
| 54 | + // AVERAGE DISPERSION |
| 55 | + nettisa_data->average_dispersion += abs(variation_from_mean); |
| 56 | + // KURTOSIS |
| 57 | + nettisa_data->kurtosis += pow(variation_from_mean, 4); |
| 58 | + // MEAN SCALED TIME |
| 59 | + nettisa_data->mean_scaled_time |
| 60 | + += (pkt.ts.tv_usec - rec.time_first.tv_usec - nettisa_data->mean_scaled_time) / n; |
| 61 | + // MEAN TIME DIFFERENCES |
| 62 | + nettisa_data->mean_difftimes += (diff_time - nettisa_data->mean_difftimes) / n; |
| 63 | + // MIN |
| 64 | + nettisa_data->min_difftimes = fmin(nettisa_data->min_difftimes, diff_time); |
| 65 | + // MAX |
| 66 | + nettisa_data->max_difftimes = fmax(nettisa_data->max_difftimes, diff_time); |
| 67 | + // TIME DISTRIBUTION |
| 68 | + nettisa_data->time_distribution += abs(nettisa_data->mean_difftimes - diff_time); |
| 69 | + // SWITCHING RATIO |
| 70 | + if (nettisa_data->prev_payload != pkt.packet_len_wire) { |
| 71 | + nettisa_data->switching_ratio += 1; |
| 72 | + nettisa_data->prev_payload = pkt.packet_len_wire; |
| 73 | + } |
| 74 | +} |
| 75 | + |
| 76 | +int NETTISAPlugin::post_create(Flow& rec, const Packet& pkt) |
| 77 | +{ |
| 78 | + RecordExtNETTISA* nettisa_data = new RecordExtNETTISA(); |
| 79 | + rec.add_extension(nettisa_data); |
| 80 | + |
| 81 | + nettisa_data->prev_time = pkt.ts.tv_usec; |
| 82 | + |
| 83 | + update_record(nettisa_data, pkt, rec); |
| 84 | + return 0; |
| 85 | +} |
| 86 | + |
| 87 | +int NETTISAPlugin::post_update(Flow& rec, const Packet& pkt) |
| 88 | +{ |
| 89 | + RecordExtNETTISA* nettisa_data |
| 90 | + = (RecordExtNETTISA*) rec.get_extension(RecordExtNETTISA::REGISTERED_ID); |
| 91 | + update_record(nettisa_data, pkt, rec); |
| 92 | + return 0; |
| 93 | +} |
| 94 | + |
| 95 | +void NETTISAPlugin::pre_export(Flow& rec) |
| 96 | +{ |
| 97 | + RecordExtNETTISA* nettisa_data |
| 98 | + = (RecordExtNETTISA*) rec.get_extension(RecordExtNETTISA::REGISTERED_ID); |
| 99 | + uint32_t n = rec.src_packets + rec.dst_packets; |
| 100 | + nettisa_data->switching_ratio = nettisa_data->switching_ratio / ((n - 1) / 2); |
| 101 | + nettisa_data->stdev = pow( |
| 102 | + (nettisa_data->root_mean_square / n) - pow((rec.src_bytes + rec.dst_bytes) / n, 2), |
| 103 | + 0.5); |
| 104 | + nettisa_data->root_mean_square = pow(nettisa_data->root_mean_square / n, 0.5); |
| 105 | + nettisa_data->average_dispersion = nettisa_data->average_dispersion / n; |
| 106 | + nettisa_data->kurtosis = nettisa_data->kurtosis / (n * pow(nettisa_data->stdev, 4)); |
| 107 | + nettisa_data->time_distribution = (nettisa_data->time_distribution / (n - 1)) |
| 108 | + / (nettisa_data->max_difftimes - nettisa_data->min); |
| 109 | +} |
| 110 | + |
| 111 | +} // namespace ipxp |
0 commit comments