eye-statd.pl 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953
  1. #!/usr/bin/perl -w
  2. use utf8;
  3. use open ":encoding(utf8)";
  4. use Encode;
  5. no warnings 'utf8';
  6. use English;
  7. use base;
  8. use FindBin '$Bin';
  9. use lib "/opt/Eye/scripts";
  10. use strict;
  11. use DBI;
  12. use Time::Local;
  13. use Net::Patricia;
  14. use Data::Dumper;
  15. use Date::Parse;
  16. use DateTime;
  17. use eyelib::config;
  18. use eyelib::main;
  19. use eyelib::net_utils;
  20. use eyelib::database;
  21. use eyelib::common;
  22. use eyelib::snmp;
  23. use Socket qw(AF_INET6 inet_ntop);
  24. use IO::Socket;
  25. my $proc_name = $MY_NAME;
  26. $proc_name =~ s/\.[^.]+$//;
  27. my $pid_file = '/run/eye/'.$proc_name;
  28. my @router_ref = ();
  29. my @interfaces = ();
  30. my %proto_map = (
  31. 1 => 'ICMP', 2 => 'IGMP', 6 => 'TCP',
  32. 17 => 'UDP', 47 => 'GRE', 50 => 'ESP',
  33. 51 => 'AH', 89 => 'OSPF', 88 => 'EIGRP'
  34. );
  35. my %mute;
  36. my %routers_svi;
  37. my %routers_by_ip;
  38. my %routers;
  39. my %wan_dev;
  40. my %lan_dev;
  41. my @traffic = ();
  42. my $saving = 0;
  43. #user statistics for cached data
  44. my %user_stats;
  45. my %wan_stats;
  46. my $MAXREAD = 9216;
  47. my $timeshift = get_option($dbh,55)*60;
  48. my $save_path = get_option($dbh,72);
  49. my $thread_count = $cpu_count;
  50. #save traffic to DB
  51. my $traf_lastflush = time();
  52. # NetFlow
  53. my $server_port = 2055;
  54. my $netflow5_header_len = 24;
  55. my $netflow5_flowrec_len = 48;
  56. my $netflow9_header_len = 20;
  57. my $netflow9_templates = {};
  58. # reap dead children
  59. $SIG{CHLD} = \&REAPER;
  60. $SIG{TERM} = \&TERM;
  61. $SIG{INT} = \&TERM;
  62. $SIG{HUP} = \&INIT;
  63. if (IsNotRun($pid_file)) {
  64. Add_PID($pid_file);
  65. } else {
  66. print "Daemon $MY_NAME already running!\n";
  67. exit 100;
  68. }
  69. sub REAPER {
  70. wait;
  71. $saving = 0;
  72. $SIG{CHLD} = \&REAPER;
  73. }
  74. sub TERM {
  75. print "SIGTERM received\n";
  76. flush_traffic(1);
  77. while (wait() != -1) {}
  78. if (IsMyPID($pid_file)) { Remove_PID($pid_file); }
  79. exit 0;
  80. }
  81. sub INIT {
  82. # Create new database handle. If we can't connect, die()
  83. my $hdb = init_db();
  84. InitSubnets();
  85. init_option($hdb);
  86. #a directory for storing traffic details in text form
  87. $save_path = get_option($hdb,72);
  88. #the period for resetting statistics from netflow to billing
  89. $timeshift = get_option($hdb,55)*60;
  90. @router_ref = get_records_sql($hdb,"SELECT * FROM devices WHERE deleted=0 AND device_type=2 AND snmp_version>0 ORDER by ip" );
  91. @interfaces = get_records_sql($hdb,"SELECT * FROM device_l3_interfaces ORDER by device_id" );
  92. #router device_id by known device ip
  93. foreach my $row (@router_ref) {
  94. setCommunity($row);
  95. $routers{$row->{id}}=$row;
  96. my $l3_list = getIpAdEntIfIndex($row->{ip},$row->{snmp});
  97. #create hash for interface snmp index => ip-address at interface =1;
  98. foreach my $router_ip (keys %$l3_list) { $routers_svi{$row->{id}}{$l3_list->{$router_ip}}{$router_ip}=1; }
  99. #create hash by all ip-addresses for router
  100. foreach my $router_ip (keys %$l3_list) {
  101. $routers_by_ip{$router_ip}->{id}=$row->{id};
  102. if ($config_ref{save_detail}) {
  103. $routers_by_ip{$router_ip}->{save}=$row->{netflow_save};
  104. } else { $routers_by_ip{$router_ip}->{save}=0; }
  105. }
  106. }
  107. #snmp index for WAN/LAN interface by device id
  108. foreach my $row (@interfaces) {
  109. if ($row->{interface_type}) { $wan_dev{$row->{device_id}}{$row->{snmpin}}=1; } else { $lan_dev{$row->{device_id}}{$row->{snmpin}}=1; }
  110. }
  111. #get userid list
  112. my @auth_list_ref = get_records_sql($hdb,"SELECT id,ip,save_traf FROM user_auth where deleted=0 ORDER by id");
  113. foreach my $row (@auth_list_ref) {
  114. $user_stats{$row->{ip}}{auth_id}=$row->{id};
  115. if ($config_ref{save_detail}) {
  116. $user_stats{$row->{ip}}{save_traf}=$row->{save_traf};
  117. } else {
  118. $user_stats{$row->{ip}}{save_traf}=0;
  119. }
  120. }
  121. $hdb->disconnect();
  122. }
  123. ############### MAIN ##########################
  124. #close default database
  125. $dbh->disconnect();
  126. INIT();
  127. my $lsn_nflow;
  128. my $sel = IO::Select->new();
  129. # prepare to listen for NetFlow UDP packets
  130. if ($server_port > 0) {
  131. $lsn_nflow = IO::Socket::INET->new(LocalPort => $server_port, Proto => "udp")
  132. or die "Couldn't be a NetFlow UDP server on port $server_port : $@\n";
  133. $sel->add($lsn_nflow);
  134. }
  135. my ($him,$datagram,$flags);
  136. # main datagram receive loop
  137. while (1) {
  138. while (my @ready = $sel->can_read) {
  139. foreach my $server (@ready) {
  140. $him = $server->recv($datagram, $MAXREAD);
  141. next if (!$him);
  142. my ($port, $ipaddr) = sockaddr_in($server->peername);
  143. if (defined($lsn_nflow) && $server == $lsn_nflow) {
  144. my ($version) = unpack("n", $datagram);
  145. if ($version == 5) {
  146. parse_netflow_v5($datagram, $ipaddr);
  147. } elsif ($version == 9) {
  148. parse_netflow_v9($datagram, $ipaddr);
  149. } else {
  150. print "unknown NetFlow version: $version\n";
  151. }
  152. }
  153. }
  154. }
  155. }
  156. sub parse_netflow_v5 {
  157. my $datagram = shift;
  158. my $ipaddr = shift;
  159. my ($version, $count, $sysuptime, $unix_secs, $unix_nsecs,
  160. $flow_sequence, $engine_type, $engine_id, $aggregation,
  161. $agg_version) = unpack("nnNNNNCCCC", $datagram);
  162. my $flowrecs = substr($datagram, $netflow5_header_len);
  163. #0 - N 0-3 srcaddr Source IP address
  164. #1 - N 4-7 dstaddr Destination IP address
  165. #2 - N 8-11 nexthop IP address of next hop router
  166. #3 - n 12-13 input SNMP index of input interface
  167. #4 - n 14-15 output SNMP index of output interface
  168. #5 - N 16-19 dPkts Packets in the flow
  169. #6 - N 20-23 dOctets Total number of Layer 3 bytes in the packets of the flow
  170. #7 - N 24-27 First SysUptime at start of flow
  171. #8 - N 28-31 Last SysUptime at the time the last packet of the flow was received
  172. #9 - n 32-33 src_port TCP/UDP source port number or equivalent
  173. #10- n 34-35 dst_port TCP/UDP destination port number or equivalent
  174. #11- C 36 pad1 Unused (zero) byte
  175. #12- C 37 tcp_flags Cumulative OR of TCP flags
  176. #13- C 38 prot IP protocol type (for example, TCP = 6; UDP = 17)
  177. #14- C 39 tos IP type of service (ToS)
  178. #15- n 40-41 src_as Autonomous system number of the source, either origin or peer
  179. #16- n 42-43 dst_as Autonomous system number of the destination, either origin or peer
  180. #17- C 44 src_mask Source address prefix mask bits
  181. #18- C 45 dst_mask Destination address prefix mask bits
  182. #19- n 46-47 pad2 Unused (zero) bytes
  183. for (my $i = 0; $i < $count; $i++) {
  184. my $flowrec = substr($datagram, $netflow5_header_len + ($i*$netflow5_flowrec_len), $netflow5_flowrec_len);
  185. my @flowdata = unpack("NNNnnNNNNnnCCCCnnCCn", $flowrec);
  186. my %flow;
  187. $flow{src_ip} = join '.', unpack 'C4', pack 'N', $flowdata[0];
  188. $flow{dst_ip} = join '.', unpack 'C4', pack 'N', $flowdata[1];
  189. $flow{snmp_in} = $flowdata[3] || 0;
  190. $flow{snmp_out} = $flowdata[4] || 0;
  191. $flow{pkts} = $flowdata[5] || 0;
  192. $flow{octets} = $flowdata[6] || 0;
  193. $flow{src_port} = $flowdata[9] || 0;
  194. $flow{dst_port} = $flowdata[10] || 0;
  195. $flow{proto} = $flowdata[13] || 0;
  196. $flow{xsrc_ip} = $flow{src_ip};
  197. $flow{xdst_ip} = $flow{dst_ip};
  198. $flow{starttime} = time();
  199. $flow{netflow_v} = '5';
  200. $flow{ipv} = '4';
  201. save_flow($ipaddr, \%flow);
  202. }
  203. }
  204. sub parse_netflow_v9 {
  205. my $datagram = shift;
  206. my $ipaddr = shift;
  207. # Parse packet
  208. my ($version, $count, $sysuptime, $unix_secs, $seqno, $source_id, @flowsets) = unpack("nnNNNN(nnX4/a)*", $datagram);
  209. # Loop through FlowSets and take appropriate action
  210. for (my $i = 0; $i < scalar @flowsets; $i += 2) {
  211. my $flowsetid = $flowsets[$i];
  212. my $flowsetdata = substr($flowsets[$i+1], 4); # chop off id/length
  213. if ($flowsetid == 0) {
  214. # 0 = Template FlowSet
  215. parse_netflow_v9_template_flowset($flowsetdata, $ipaddr, $source_id);
  216. } elsif ($flowsetid == 1) {
  217. # 1 - Options Template FlowSet
  218. } elsif ($flowsetid > 255) {
  219. # > 255: Data FlowSet
  220. parse_netflow_v9_data_flowset($flowsetid, $flowsetdata, $ipaddr, $source_id);
  221. } else {
  222. # reserved FlowSet
  223. print "Unknown FlowSet ID $flowsetid found\n";
  224. }
  225. }
  226. }
  227. sub parse_netflow_v9_template_flowset {
  228. my $templatedata = shift;
  229. my $ipaddr = shift;
  230. my $source_id = shift;
  231. # Note: there may be multiple templates in a Template FlowSet
  232. my @template_ints = unpack("n*", $templatedata);
  233. my $i = 0;
  234. while ($i < scalar @template_ints) {
  235. my $template_id = $template_ints[$i];
  236. my $fldcount = $template_ints[$i+1];
  237. last if (!defined($template_id) || !defined($fldcount));
  238. # print "Updated template ID $template_id (source ID $source_id, from " . inet_ntoa($ipaddr) . ")\n" if ($debug);
  239. my $template = [@template_ints[($i+2) .. ($i+2+$fldcount*2-1)]];
  240. $netflow9_templates->{$ipaddr}->{$source_id}->{$template_id}->{'template'} = $template;
  241. # total length of template data
  242. my $totallen = 0;
  243. for (my $j = 1; $j < scalar @$template; $j += 2) {
  244. $totallen += $template->[$j];
  245. }
  246. $netflow9_templates->{$ipaddr}->{$source_id}->{$template_id}->{'len'} = $totallen;
  247. $i += (2 + $fldcount*2);
  248. }
  249. }
  250. sub parse_netflow_v9_data_flowset {
  251. my ($flowsetid, $flowsetdata, $ipaddr, $source_id) = @_;
  252. my $template = $netflow9_templates->{$ipaddr}->{$source_id}->{$flowsetid}->{'template'};
  253. unless (defined $template) {
  254. return;
  255. }
  256. my $len = $netflow9_templates->{$ipaddr}->{$source_id}->{$flowsetid}->{'len'};
  257. my $offset = 0;
  258. my $datalen = length($flowsetdata);
  259. while (($offset + $len) <= $datalen) {
  260. my %flow = (
  261. netflow_v => '9',
  262. starttime => time(),
  263. ipv => '4',
  264. snmp_in => 0,
  265. snmp_out => 0,
  266. octets => 0,
  267. pkts => 0
  268. );
  269. for (my $i = 0; $i < scalar @$template; $i += 2) {
  270. my $field_type = $template->[$i];
  271. my $field_length = $template->[$i+1];
  272. my $value = substr($flowsetdata, $offset, $field_length);
  273. $offset += $field_length;
  274. # IN_BYTES (1)
  275. if ($field_type == 1) {
  276. $flow{octets} = $field_length == 8 ? unpack("Q>", $value) : unpack("N", $value);
  277. }
  278. # IN_PACKETS (2)
  279. elsif ($field_type == 2) {
  280. $flow{pkts} = $field_length == 8 ? unpack("Q>", $value) : unpack("N", $value);
  281. }
  282. # IN_PROTOCOL (4)
  283. elsif ($field_type == 4) {
  284. $flow{proto} = unpack("C", $value);
  285. }
  286. # L4_SRC_PORT (7)
  287. elsif ($field_type == 7) {
  288. $flow{src_port} = unpack("n", $value);
  289. }
  290. # IPV4_SRC_ADDR (8)
  291. elsif ($field_type == 8) {
  292. $flow{src_ip} = inet_ntop(AF_INET, $value);
  293. }
  294. # INPUT_SNMP (10)
  295. elsif ($field_type == 10) {
  296. $flow{snmp_in} = $field_length == 4 ? unpack("N", $value) : unpack("n", $value);
  297. }
  298. # L4_DST_PORT (11)
  299. elsif ($field_type == 11) {
  300. $flow{dst_port} = unpack("n", $value);
  301. }
  302. # IPV4_DST_ADDR (12)
  303. elsif ($field_type == 12) {
  304. $flow{dst_ip} = inet_ntop(AF_INET, $value);
  305. }
  306. # OUTPUT_SNMP (14)
  307. elsif ($field_type == 14) {
  308. $flow{snmp_out} = $field_length == 4 ? unpack("N", $value) : unpack("n", $value);
  309. }
  310. # ICMP_TYPE (32)
  311. elsif ($field_type == 32) {
  312. $flow{icmp_type} = unpack("C", $value);
  313. }
  314. # ICMP_CODE (33)
  315. elsif ($field_type == 33) {
  316. $flow{icmp_code} = unpack("C", $value);
  317. }
  318. # IP_PROTOCOL_VERSION (60)
  319. elsif ($field_type == 60) {
  320. my $ipversion = unpack("C", $value);
  321. if ($ipversion == 6) {
  322. %flow = ();
  323. last;
  324. }
  325. $flow{ipv} = $ipversion;
  326. }
  327. # XLATE_SRC_ADDR_IPV4 (225)
  328. elsif ($field_type == 225) {
  329. $flow{xsrc_ip} = inet_ntop(AF_INET, $value);
  330. }
  331. # XLATE_DST_ADDR_IPV4 (226)
  332. elsif ($field_type == 226) {
  333. $flow{xdst_ip} = inet_ntop(AF_INET, $value);
  334. }
  335. }
  336. # Обработка не-TCP/UDP трафика
  337. if ($flow{proto} == 1) { # ICMP
  338. $flow{src_port} = $flow{icmp_type} || 0;
  339. $flow{dst_port} = $flow{icmp_code} || 0;
  340. }
  341. elsif ($flow{proto} == 2) { # IGMP
  342. $flow{src_port} = $flow{igmp_type} || 0;
  343. }
  344. elsif ($flow{proto} == 47) { # GRE
  345. $flow{src_port} = $flow{gre_version} || 0;
  346. $flow{dst_port} = $flow{gre_key} || 0;
  347. }
  348. elsif ($flow{proto} == 50 || # ESP
  349. $flow{proto} == 51) { # AH
  350. $flow{src_port} = $flow{dst_port} = 0; # Нет портов
  351. }
  352. elsif ($flow{proto} == 89) { # OSPF
  353. $flow{src_port} = $flow{dst_port} = 0;
  354. }
  355. # Сохраняем только валидные потоки
  356. if (%flow && exists $flow{src_ip} && exists $flow{dst_ip}) {
  357. save_flow($ipaddr, \%flow);
  358. }
  359. }
  360. }
  361. sub get_proto_name {
  362. my ($proto_num) = @_;
  363. return $proto_map{$proto_num} || "Proto-$proto_num";
  364. }
  365. sub save_flow {
  366. my $router_ip = shift;
  367. my $flow = shift;
  368. $router_ip = inet_ntoa($router_ip);
  369. #direction for user, 0 - in, 1 - out
  370. $flow->{direction} = '0';
  371. my $router_id;
  372. #skip unknown router
  373. if (exists $routers_by_ip{$router_ip}) {
  374. $router_id = $routers_by_ip{$router_ip}{id};
  375. $flow->{router_ip} = $router_ip;
  376. $flow->{device_id} = $router_id;
  377. $flow->{save} = $routers_by_ip{$router_ip}{save};
  378. } else {
  379. if (!exists $mute{$router_ip}) { $mute{$router_ip} = time(); }
  380. if (time() - $mute{$router_ip} >=3600) {
  381. $mute{$router_ip} = time();
  382. log_warning("Found unknown router ip [".$router_ip."] in netflow!");
  383. }
  384. return;
  385. }
  386. #skip local traffic for router
  387. if (!exists $wan_dev{$router_id}->{$flow->{snmp_out}} and ! exists $wan_dev{$router_id}->{$flow->{snmp_in}}) { return; }
  388. #detect traffic direction
  389. if (exists $wan_dev{$router_id}->{$flow->{snmp_out}}) { $flow->{direction} = 1; }
  390. push(@traffic,$flow);
  391. flush_traffic(0);
  392. }
  393. sub flush_traffic {
  394. my $force = shift || 0;
  395. if (!$force && ($saving || ((time - $traf_lastflush) < $timeshift))) { return; }
  396. $saving++;
  397. my $pid = fork();
  398. INIT();
  399. log_verbose("Start flush traffic to DB");
  400. log_debug("ROUTERS-SVI:".Dumper(\%routers_svi));
  401. log_debug("ROUTERS by IP::".Dumper(\%routers_by_ip));
  402. log_debug("ROUTERS:".Dumper(\%routers));
  403. log_debug("WAN-DEVS:".Dumper(\%wan_dev));
  404. log_debug("LAN-DEVS:".Dumper(\%lan_dev));
  405. if (!defined $pid) {
  406. $saving = 0;
  407. print "cannot fork! Save traffic and exit...\n";
  408. } elsif ($pid != 0) {
  409. # in parent
  410. $traf_lastflush = time();
  411. #clean main cache
  412. @traffic = ();
  413. return;
  414. }
  415. #create oper-cache
  416. my @flush_table = ();
  417. push(@flush_table,@traffic);
  418. my $hdb=init_db();
  419. #saved packet by users
  420. my @detail_traffic = ();
  421. my %saved_netflow = ();
  422. my %routers_found;
  423. #last packet timestamp
  424. my $last_time = time();
  425. my $start_time;
  426. log_debug("Netflow statistics calculation started");
  427. foreach my $traf_record (@flush_table) {
  428. #log_debug("RAW-DATA: ".hash_to_kv_csv($traf_record));
  429. my ($auth_id,$l_src_ip,$l_dst_ip,$user_ip,$router_id);
  430. #skip unknown router
  431. next if (!$traf_record->{device_id});
  432. $router_id = $traf_record->{device_id};
  433. #prepare router traffic detailization data only if traffic retention is enabled globally
  434. if ($config_ref{save_detail} and $traf_record->{save}) {
  435. push(@{$saved_netflow{$traf_record->{device_id}}},join(';',$traf_record->{starttime},$traf_record->{proto},$traf_record->{snmp_in},$traf_record->{snmp_out},$traf_record->{src_ip},$traf_record->{dst_ip},$traf_record->{xsrc_ip},$traf_record->{xdst_ip},$traf_record->{src_port},$traf_record->{dst_port},$traf_record->{octets},$traf_record->{pkts}));
  436. }
  437. $routers_found{$router_id} = 1;
  438. #save start netflow time
  439. if (!$start_time) { $start_time = $traf_record->{starttime}; }
  440. #--- router statistics
  441. #input traffic and traffic originated from router
  442. if (!$traf_record->{snmp_out} or !$traf_record->{snmp_in}) {
  443. #input
  444. if (!$traf_record->{snmp_out} and exists $routers_svi{$router_id}{$traf_record->{snmp_in}}{$traf_record->{dst_ip}}) {
  445. #log_debug("ROUTER id: $router_id I-DATA: ".hash_to_kv_csv($traf_record));
  446. #input
  447. if (!$free_networks->match_string($traf_record->{src_ip})) {
  448. if (exists $wan_stats{$router_id}{$traf_record->{snmp_in}}{in}) {
  449. $wan_stats{$router_id}{$traf_record->{snmp_in}}{in}+=$traf_record->{octets};
  450. } else {
  451. $wan_stats{$router_id}{$traf_record->{snmp_in}}{in}=$traf_record->{octets};
  452. }
  453. }
  454. next;
  455. }
  456. #output
  457. if (!$traf_record->{snmp_in} and exists $routers_svi{$router_id}{$traf_record->{snmp_out}}{$traf_record->{src_ip}}) {
  458. #log_debug("ROUTER id: $router_id O-DATA: ".hash_to_kv_csv($traf_record));
  459. #output
  460. if (!$free_networks->match_string($traf_record->{dst_ip})) {
  461. if (exists $wan_stats{$router_id}{$traf_record->{snmp_out}}{out}) {
  462. $wan_stats{$router_id}{$traf_record->{snmp_out}}{out}+=$traf_record->{octets};
  463. } else {
  464. $wan_stats{$router_id}{$traf_record->{snmp_out}}{out}=$traf_record->{octets};
  465. }
  466. }
  467. next;
  468. }
  469. #log_debug("ROUTER id: $router_id U-DATA: ".hash_to_kv_csv($traf_record));
  470. #unknown packet
  471. next;
  472. }
  473. #simple output traffic from router
  474. if (exists $wan_dev{$router_id}->{$traf_record->{snmp_out}} and exists $wan_dev{$router_id}->{$traf_record->{snmp_in}}) {
  475. if (exists $routers_svi{$router_id}{$traf_record->{snmp_out}}{$traf_record->{src_ip}}) {
  476. #log_debug("ROUTER id: $router_id O-SDATA: ".hash_to_kv_csv($traf_record));
  477. #output
  478. if (!$free_networks->match_string($traf_record->{dst_ip})) {
  479. if (exists $wan_stats{$router_id}{$traf_record->{snmp_out}}{out}) {
  480. $wan_stats{$router_id}{$traf_record->{snmp_out}}{out}+=$traf_record->{octets};
  481. } else {
  482. $wan_stats{$router_id}{$traf_record->{snmp_out}}{out}=$traf_record->{octets};
  483. }
  484. }
  485. next;
  486. }
  487. #It is unlikely that it will ever work out
  488. if (exists $routers_svi{$router_id}{$traf_record->{snmp_in}}{$traf_record->{dst_ip}}) {
  489. #log_debug("ROUTER id: $router_id I-SDATA: ".hash_to_kv_csv($traf_record));
  490. #input
  491. if (!$free_networks->match_string($traf_record->{src_ip})) {
  492. if (exists $wan_stats{$router_id}{$traf_record->{snmp_in}}{in}) {
  493. $wan_stats{$router_id}{$traf_record->{snmp_in}}{in}+=$traf_record->{octets};
  494. } else {
  495. $wan_stats{$router_id}{$traf_record->{snmp_in}}{in}=$traf_record->{octets};
  496. }
  497. }
  498. next;
  499. }
  500. #log_debug("ROUTER id: $router_id U-SDATA: ".hash_to_kv_csv($traf_record));
  501. #unknown packet
  502. next;
  503. } else {
  504. #forward
  505. if (!$free_networks->match_string($traf_record->{src_ip}) and !$free_networks->match_string($traf_record->{dst_ip})) {
  506. if ($traf_record->{direction}) {
  507. #log_debug("ROUTER id: $router_id FO-DATA: ".hash_to_kv_csv($traf_record));
  508. #out
  509. if (exists $wan_stats{$router_id}{$traf_record->{snmp_out}}{forward_out}) {
  510. $wan_stats{$router_id}{$traf_record->{snmp_out}}{forward_out}+=$traf_record->{octets};
  511. } else {
  512. $wan_stats{$router_id}{$traf_record->{snmp_out}}{forward_out}+=$traf_record->{octets};
  513. }
  514. } else {
  515. #log_debug("ROUTER id: $router_id FI-DATA: ".hash_to_kv_csv($traf_record));
  516. #in
  517. if (exists $wan_stats{$router_id}{$traf_record->{snmp_in}}{forward_in}) {
  518. $wan_stats{$router_id}{$traf_record->{snmp_in}}{forward_in}+=$traf_record->{octets};
  519. } else {
  520. $wan_stats{$router_id}{$traf_record->{snmp_in}}{forward_in}+=$traf_record->{octets};
  521. }
  522. }
  523. } else {
  524. #log_debug("ROUTER id: $router_id FREE-DATA: ".hash_to_kv_csv($traf_record));
  525. }
  526. }
  527. #--- user statistics
  528. my $free = 0;
  529. if ($traf_record->{direction}) {
  530. #outbound traffic
  531. if (exists $user_stats{$traf_record->{src_ip}}) {
  532. $user_ip = $traf_record->{src_ip};
  533. $l_src_ip = $traf_record->{src_ip};
  534. $l_dst_ip = $traf_record->{dst_ip};
  535. $free = $free_networks->match_string($l_dst_ip);
  536. #skip calculate free net
  537. if (!$free) {
  538. if (exists $user_stats{$user_ip}{$router_id}{out}) {
  539. $user_stats{$user_ip}{$router_id}{out}+=$traf_record->{octets};
  540. } else {
  541. $user_stats{$user_ip}{$router_id}{out}=$traf_record->{octets};
  542. }
  543. if (exists $user_stats{$user_ip}{$router_id}{pkt_out}) {
  544. $user_stats{$user_ip}{$router_id}{pkt_out}+=$traf_record->{pkts};
  545. } else {
  546. $user_stats{$user_ip}{$router_id}{pkt_out}=$traf_record->{pkts};
  547. }
  548. }
  549. }
  550. #a new user is created only by the presence of outgoing traffic
  551. if (!$user_ip and $config_ref{add_unknown_user}) {
  552. #skip create router interface as user
  553. if (exists $routers_by_ip{$traf_record->{src_ip}}) { next; }
  554. if (!$office_networks->match_string($traf_record->{src_ip})) {
  555. log_debug("Unknown src network ".hash_to_kv_csv($traf_record));
  556. next;
  557. }
  558. $user_ip = $traf_record->{src_ip};
  559. $auth_id = new_auth($hdb,$user_ip);
  560. if (!$auth_id) { next; }
  561. $l_src_ip = $traf_record->{src_ip};
  562. $l_dst_ip = $traf_record->{dst_ip};
  563. $user_stats{$user_ip}{auth_id}=$auth_id;
  564. $user_stats{$user_ip}{$router_id}{in}=0;
  565. $user_stats{$user_ip}{$router_id}{pkt_in}=0;
  566. $user_stats{$user_ip}{$router_id}{out}=0;
  567. $user_stats{$user_ip}{$router_id}{pkt_out}=0;
  568. $user_stats{$user_ip}{save_traf}=$config_ref{save_detail};
  569. $free = $free_networks->match_string($l_dst_ip);
  570. #skip calculate free net
  571. if (!$free) {
  572. $user_stats{$user_ip}{$router_id}{out}=$traf_record->{octets};
  573. $user_stats{$user_ip}{$router_id}{pkt_out}=$traf_record->{pkts};
  574. }
  575. }
  576. } else {
  577. #inbound traffic
  578. if (exists $user_stats{$traf_record->{xdst_ip}}) {
  579. $user_ip = $traf_record->{xdst_ip};
  580. $l_src_ip = $traf_record->{src_ip};
  581. $l_dst_ip = $traf_record->{xdst_ip};
  582. $free = $free_networks->match_string($l_src_ip);
  583. #skip calculate free net
  584. if (!$free) {
  585. if (exists $user_stats{$user_ip}{$router_id}{in}) {
  586. $user_stats{$user_ip}{$router_id}{in}+=$traf_record->{octets};
  587. } else {
  588. $user_stats{$user_ip}{$router_id}{in}=$traf_record->{octets};
  589. }
  590. if (exists $user_stats{$user_ip}{$router_id}{pkt_in}) {
  591. $user_stats{$user_ip}{$router_id}{pkt_in}+=$traf_record->{pkts};
  592. } else {
  593. $user_stats{$user_ip}{$router_id}{pkt_in}=$traf_record->{pkts};
  594. }
  595. }
  596. }
  597. }
  598. if (!$user_ip) {
  599. #log_debug("Unknown USER: ".hash_to_kv_csv($traf_record));
  600. next;
  601. }
  602. $last_time = $traf_record->{starttime};
  603. $user_stats{$user_ip}{last_found} = $last_time;
  604. next if (!$config_ref{save_detail} and !$user_stats{$user_ip}{save_traf});
  605. my $l_src_ip_aton=StrToIp($l_src_ip);
  606. my $l_dst_ip_aton=StrToIp($l_dst_ip);
  607. my ($sec,$min,$hour,$day,$month,$year,$zone) = (localtime($last_time))[0,1,2,3,4,5];
  608. my $full_time = sprintf "%04d-%02d-%02d %02d:%02d:%02d",$year+1900,$month+1,$day,$hour,$min,$sec;
  609. push @detail_traffic, [
  610. $user_stats{$user_ip}->{auth_id},
  611. $router_id,
  612. $full_time,
  613. $traf_record->{proto},
  614. $l_src_ip_aton,
  615. $l_dst_ip_aton,
  616. $traf_record->{src_port},
  617. $traf_record->{dst_port},
  618. $traf_record->{octets},
  619. $traf_record->{pkts}
  620. ];
  621. }
  622. log_debug("The netflow statistics calculation is finished");
  623. @flush_table=();
  624. #start hour
  625. my ($sec,$min,$hour,$day,$month,$year) = (localtime($last_time))[0,1,2,3,4,5];
  626. #save netflow
  627. if ($config_ref{save_detail}) {
  628. $save_path=~s/\/$//;
  629. log_debug("Write netflow started");
  630. foreach my $dev_id (keys %saved_netflow) {
  631. my $netflow_file_path = $save_path.'/'.$dev_id.'/'.sprintf "%04d/%02d/%02d/%02d/",$year+1900,$month+1,$day,$hour;
  632. my $nmin = int($min/10)*10;
  633. my $netflow_file_name = $netflow_file_path.sprintf "%04d%02d%02d-%02d%02d.csv",$year+1900,$month+1,$day,$hour,$nmin;
  634. if ($saved_netflow{$dev_id} and scalar @{$saved_netflow{$dev_id}}) {
  635. use File::Path;
  636. File::Path::make_path($netflow_file_path);
  637. if ( -e $netflow_file_name) {
  638. open (ND,">>$netflow_file_name") || die("Error open file $netflow_file_name!!! die...");
  639. binmode(ND,':utf8');
  640. } else {
  641. open (ND,">$netflow_file_name") || die("Error open file $netflow_file_name!!! die...");
  642. binmode(ND,':utf8');
  643. print ND join(';',"time","proto","snmp_in","snmp_out","src_ip","dst_ip","xsrc_ip","xdst_ip","src_port","dst_port","octets","pkts")."\n";
  644. }
  645. foreach my $row (@{$saved_netflow{$dev_id}}) {
  646. next if (!$row);
  647. print ND $row."\n";
  648. }
  649. close ND;
  650. @{$saved_netflow{$dev_id}}=();
  651. }
  652. }
  653. log_debug("Write netflow is finished");
  654. }
  655. undef %saved_netflow;
  656. #save statistics
  657. #start stat time
  658. #my $hour_date1 = sprintf "%04d-%02d-%02d %02d:00:00",$year+1900,$month+1,$day,$hour;
  659. #end hour
  660. #($hour,$day,$month,$year) = (localtime($last_time+3600))[2,3,4,5];
  661. #my $hour_date2 = sprintf "%04d-%02d-%02d %02d:00:00",$year+1900,$month+1,$day,$hour;
  662. my @batch_user_stats=();
  663. my @batch_user_stats_update=();
  664. my @batch_user_stats_full=();
  665. my @batch_auth_status=();
  666. my @batch_wan_stats=();
  667. #log_debug("User STATS: ".Dumper(\%user_stats));
  668. log_debug("The user statistics calculation started");
  669. # update database
  670. foreach my $user_ip (keys %user_stats) {
  671. next if (!exists $user_stats{$user_ip}{last_found});
  672. my $user_ip_aton=StrToIp($user_ip);
  673. my $auth_id = $user_stats{$user_ip}{auth_id};
  674. #last flow for user
  675. my ($sec,$min,$hour,$day,$month,$year) = (localtime($user_stats{$user_ip}{last_found}))[0,1,2,3,4,5];
  676. #flow time string
  677. my $flow_date = sprintf "%04d-%02d-%02d %02d:%02d:%02d",$year+1900,$month+1,$day,$hour,$min,$sec;
  678. my $flow_hour_start = sprintf "%04d-%02d-%02d %02d:00:00", $year + 1900, $month + 1, $day, $hour;
  679. #last found timestamp
  680. push @batch_auth_status, [ $flow_date, $flow_date, $auth_id ];
  681. #per router stats
  682. foreach my $router_id (keys %routers_found) {
  683. next if (!exists $user_stats{$user_ip}{$router_id});
  684. if (!exists $user_stats{$user_ip}{$router_id}{in}) { $user_stats{$user_ip}{$router_id}{in} = 0; }
  685. if (!exists $user_stats{$user_ip}{$router_id}{out}) { $user_stats{$user_ip}{$router_id}{out} = 0; }
  686. #skip empty stats
  687. if ($user_stats{$user_ip}{$router_id}{in} + $user_stats{$user_ip}{$router_id}{out} ==0) { next; }
  688. #packet count per router
  689. if (!exists $user_stats{$user_ip}{$router_id}{pkt_in}) { $user_stats{$user_ip}{$router_id}{pkt_in} = 0; }
  690. if (!exists $user_stats{$user_ip}{$router_id}{pkt_out}) { $user_stats{$user_ip}{$router_id}{pkt_out} = 0; }
  691. #current stats
  692. push @batch_user_stats_full, [
  693. $flow_date,
  694. $auth_id,
  695. $router_id,
  696. $user_stats{$user_ip}{$router_id}{in},
  697. $user_stats{$user_ip}{$router_id}{out},
  698. $user_stats{$user_ip}{$router_id}{pkt_in},
  699. $user_stats{$user_ip}{$router_id}{pkt_out},
  700. $timeshift ];
  701. #hour stats
  702. # get current stats
  703. my $sql = "SELECT byte_in, byte_out, pkt_in, pkt_out FROM user_stats WHERE ts = ? AND router_id = ? AND auth_id = ?";
  704. my $hour_stat = get_record_sql($hdb, $sql,
  705. $flow_hour_start,
  706. $router_id,
  707. $auth_id
  708. );
  709. if (!$hour_stat) {
  710. push @batch_user_stats, [
  711. $flow_hour_start,
  712. $auth_id,
  713. $router_id,
  714. $user_stats{$user_ip}{$router_id}{in},
  715. $user_stats{$user_ip}{$router_id}{out},
  716. $user_stats{$user_ip}{$router_id}{pkt_in},
  717. $user_stats{$user_ip}{$router_id}{pkt_out},
  718. 3600 ];
  719. next;
  720. }
  721. if (!$hour_stat->{byte_in}) { $hour_stat->{byte_in}=0; }
  722. if (!$hour_stat->{byte_out}) { $hour_stat->{byte_out}=0; }
  723. if (!$hour_stat->{pkt_in}) { $hour_stat->{pkt_in}=0; }
  724. if (!$hour_stat->{pkt_out}) { $hour_stat->{pkt_out}=0; }
  725. $hour_stat->{byte_in} += $user_stats{$user_ip}{$router_id}{in};
  726. $hour_stat->{byte_out} += $user_stats{$user_ip}{$router_id}{out};
  727. $hour_stat->{pkt_in} += $user_stats{$user_ip}{$router_id}{pkt_in};
  728. $hour_stat->{pkt_out} += $user_stats{$user_ip}{$router_id}{pkt_out};
  729. push @batch_user_stats_update, [
  730. $hour_stat->{byte_in},
  731. $hour_stat->{byte_out},
  732. $hour_stat->{pkt_in},
  733. $hour_stat->{pkt_out},
  734. $flow_hour_start,
  735. $auth_id,
  736. $router_id ];
  737. }
  738. }
  739. log_debug("User calculation is finished");
  740. #print Dumper(\%wan_stats) if ($debug);
  741. # update database
  742. log_debug("Routers statistics started");
  743. foreach my $router_id (keys %wan_stats) {
  744. #last flow for user
  745. my ($sec,$min,$hour,$day,$month,$year) = (localtime($start_time))[0,1,2,3,4,5];
  746. #flow time string
  747. my $flow_date = sprintf "%04d-%02d-%02d %02d:%02d:%02d",$year+1900,$month+1,$day,$hour,$min,$sec;
  748. #per interface stats
  749. foreach my $int_id (keys %{$wan_stats{$router_id}}) {
  750. if (!$wan_stats{$router_id}{$int_id}{in}) { $wan_stats{$router_id}{$int_id}{in} = 0; }
  751. if (!$wan_stats{$router_id}{$int_id}{out}) { $wan_stats{$router_id}{$int_id}{out} = 0; }
  752. if (!$wan_stats{$router_id}{$int_id}{forward_in}) { $wan_stats{$router_id}{$int_id}{forward_in} = 0; }
  753. if (!$wan_stats{$router_id}{$int_id}{forward_out}) { $wan_stats{$router_id}{$int_id}{forward_out} = 0; }
  754. #skip empty stats
  755. if ($wan_stats{$router_id}{$int_id}{in} + $wan_stats{$router_id}{$int_id}{out} + $wan_stats{$router_id}{$int_id}{forward_in} + $wan_stats{$router_id}{$int_id}{forward_out} ==0) { next; }
  756. #current wan stats
  757. push @batch_wan_stats, [
  758. $flow_date,
  759. $router_id,
  760. $int_id,
  761. $wan_stats{$router_id}{$int_id}{in},
  762. $wan_stats{$router_id}{$int_id}{out},
  763. $wan_stats{$router_id}{$int_id}{forward_in},
  764. $wan_stats{$router_id}{$int_id}{forward_out}
  765. ];
  766. }
  767. }
  768. log_debug("Router statistics is finished");
  769. log_verbose("Try update user_auth table for ".scalar @batch_auth_status. " records");
  770. my $tSQL="UPDATE user_auth SET arp_found= ?, last_found= ? WHERE id= ?";
  771. batch_db_sql_cached($tSQL,\@batch_auth_status);
  772. log_verbose("Finished");
  773. log_verbose("Try update user_stats_full table for ".scalar @batch_user_stats_full. " records");
  774. $tSQL="INSERT INTO user_stats_full (ts,auth_id,router_id,byte_in,byte_out,pkt_in,pkt_out,step) VALUES( ?, ?, ?, ?, ?, ?, ?, ?)";
  775. batch_db_sql_cached($tSQL,\@batch_user_stats_full);
  776. log_verbose("Finished");
  777. log_verbose("Try create new records in user_stats table for ".scalar @batch_user_stats. " records");
  778. $tSQL="INSERT INTO user_stats (ts,auth_id,router_id,byte_in,byte_out,pkt_in,pkt_out,step) VALUES( ?, ?, ?, ?, ?, ?, ? ,?)";
  779. batch_db_sql_cached($tSQL,\@batch_user_stats);
  780. log_verbose("Finished");
  781. log_verbose("Try update user_stats table for ".scalar @batch_user_stats_update. " records");
  782. $tSQL="UPDATE user_stats SET byte_in= ?, byte_out= ?, pkt_in = ?, pkt_out = ? WHERE ts = ? AND auth_id= ? AND router_id= ?";
  783. batch_db_sql_cached($tSQL,\@batch_user_stats_update);
  784. log_verbose("Finished");
  785. log_verbose("Try create new records in wan_stats table for ".scalar @batch_wan_stats. " records");
  786. $tSQL="INSERT INTO wan_stats (ts,router_id,interface_id,bytes_in,bytes_out,forward_in,forward_out) VALUES( ?, ?, ?, ?, ?, ?, ?)";
  787. batch_db_sql_cached($tSQL,\@batch_wan_stats);
  788. log_verbose("Finished");
  789. @batch_user_stats=();
  790. @batch_user_stats_update=();
  791. @batch_user_stats_full=();
  792. @batch_auth_status=();
  793. @batch_wan_stats=();
  794. if ($config_ref{enable_quotes}) {
  795. log_info($hdb,"Recalc quotes started");
  796. foreach my $router_id (keys %routers_found) { recalc_quotes($hdb,$router_id); }
  797. log_info($hdb,"Recalc quotes stopped");
  798. }
  799. if (scalar(@detail_traffic)) {
  800. db_log_debug($hdb,"Start write traffic detail to DB. ".scalar @detail_traffic." lines count") if ($debug);
  801. my $traffic_fields = ['auth_id', 'router_id', 'ts', 'proto', 'src_ip', 'dst_ip', 'src_port', 'dst_port', 'bytes', 'pkt'];
  802. unshift @detail_traffic, $traffic_fields;
  803. batch_db_sql_csv("traffic_detail",\@detail_traffic);
  804. @detail_traffic = ();
  805. db_log_debug($hdb,"Write traffic detail to DB stopped") if ($debug);
  806. }
  807. $hdb->disconnect();
  808. $saving = 0;
  809. log_verbose("Flush traffic to DB is finished");
  810. exit;
  811. }
  812. if (IsMyPID($pid_file)) { Remove_PID($pid_file); }
  813. exit;