sync_mikrotik.pl 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744
  1. #!/usr/bin/perl -w
  2. #
  3. # Copyright (C) Roman Dmitiriev, rnd@rajven.ru
  4. #
  5. use FindBin '$Bin';
  6. use lib "$Bin/";
  7. use strict;
  8. use Time::Local;
  9. use FileHandle;
  10. use Data::Dumper;
  11. use Rstat::config;
  12. use Rstat::main;
  13. use Rstat::mikrotik;
  14. use Rstat::cmd;
  15. use Net::Patricia;
  16. use Date::Parse;
  17. use Rstat::net_utils;
  18. use Rstat::mysql;
  19. use DBI;
  20. use utf8;
  21. use open ":encoding(utf8)";
  22. $|=1;
  23. if (IsNotRun($SPID)) { Add_PID($SPID); } else { die "Warning!!! $SPID already runnning!\n"; }
  24. my %fields=('device_name'=>'1', 'ip'=>'1', 'device_model'=>'1', 'wan_int'=>'1', 'lan_int'=>'1', 'dhcp'=>'1', 'internet_gateway'=>'1', 'queue_enabled'=>'1', 'connected_user_only'=>'1');
  25. my @gateways =();
  26. #select undeleted mikrotik routers only
  27. if ($ARGV[0]) {
  28. my $router = get_record($dbh,'devices',\%fields,"(internet_gateway=1 or dhcp=1) and deleted=0 and vendor_id=9 and id='".$ARGV[0]."'");
  29. if ($router) { push(@gateways,$router); }
  30. } else {
  31. @gateways = get_records($dbh,'devices',\%fields,"(internet_gateway=1 or dhcp=1) and deleted=0 and vendor_id=9");
  32. }
  33. my $dhcp_networks = new Net::Patricia;
  34. my %dhcp_conf;
  35. my @subnets=get_custom_records($dbh,'SELECT * FROM subnets WHERE dhcp=1 and office=1 and vpn=0 ORDER BY ip_int_start');
  36. foreach my $subnet (@subnets) {
  37. next if (!$subnet->{gateway});
  38. my $subnet_name = $subnet->{subnet};
  39. $subnet_name=~s/\/\d+$//g;
  40. $dhcp_networks->add_string($subnet->{subnet},$subnet_name);
  41. $dhcp_conf{$subnet_name}->{first_pool_ip}=IpToStr($subnet->{dhcp_start});
  42. $dhcp_conf{$subnet_name}->{last_pool_ip}=IpToStr($subnet->{dhcp_stop});
  43. $dhcp_conf{$subnet_name}->{relay_ip}=IpToStr($subnet->{gateway});
  44. my $dhcp_info=GetDhcpRange($subnet->{subnet});
  45. $dhcp_conf{$subnet_name}->{first_ip} = $dhcp_info->{first_ip};
  46. $dhcp_conf{$subnet_name}->{last_ip} = $dhcp_info->{last_ip};
  47. $dhcp_conf{$subnet_name}->{first_ip_aton}=StrToIp($dhcp_info->{first_ip});
  48. $dhcp_conf{$subnet_name}->{last_ip_aton}=StrToIp($dhcp_info->{last_ip});
  49. }
  50. foreach my $gate (@gateways) {
  51. next if (!$gate);
  52. my $router_name=$gate->{device_name};
  53. my $router_ip=$gate->{ip};
  54. my $wan_dev = $gate->{wan_int};
  55. my $lan_dev = $gate->{lan_int};
  56. my $shaper_enabled = $gate->{queue_enabled};
  57. my $connected_users_only = $gate->{connected_user_only};
  58. my $connected_users = new Net::Patricia;
  59. #lan interfaces
  60. my @lan_int=split(/;/,$lan_dev);
  61. my @cmd_list=();
  62. my $t = Login_Mikrotik($router_ip);
  63. foreach my $int (@lan_int) { #interface dhcp loop
  64. next if (!$int);
  65. $int=trim($int);
  66. #get ip addr at interface
  67. my @int_addr=log_cmd4($t,'/ip address print terse without-paging where interface='.$int);
  68. my $found_subnet;
  69. foreach my $int_str(@int_addr) {
  70. $int_str=trim($int_str);
  71. next if (!$int_str);
  72. if ($int_str=~/\s+address=(\S*)\s+/i) {
  73. my $gate_interface=$1;
  74. if ($gate_interface) {
  75. my $gate_ip=$gate_interface;
  76. $gate_ip=~s/\/.*$//;
  77. #search for first match
  78. if (!$found_subnet) { $found_subnet=$dhcp_networks->match_string($gate_ip); }
  79. #all subnets match
  80. if ($connected_users_only) { $connected_users->add_string($gate_interface); }
  81. }
  82. }
  83. }
  84. if (!$found_subnet) { db_log_verbose($dbh,"DHCP subnet for interface $int not found! Skip interface."); next; }
  85. db_log_verbose($dbh,"Analyze interface $int. Found: ".Dumper($dhcp_conf{$found_subnet}));
  86. #dhcp config
  87. if ($gate->{dhcp}) {
  88. #fetch current dhcp records
  89. my @ret_static_leases=log_cmd4($t,'/ip dhcp-server lease print terse without-paging where server=dhcp-'.$int);
  90. my @current_static_leases=();
  91. foreach my $str (@ret_static_leases) {
  92. next if (!$str);
  93. $str=trim($str);
  94. if ($str=~/^\d/) {
  95. log_debug("Found current static lease record: ".$str);
  96. push(@current_static_leases,$str);
  97. }
  98. }
  99. #select users for this interface
  100. my @auth_records=get_custom_records($dbh,"SELECT * from User_auth WHERE dhcp=1 and `ip_int`>=".$dhcp_conf{$found_subnet}->{first_ip_aton}." and `ip_int`<=".$dhcp_conf{$found_subnet}->{last_ip_aton}." and deleted=0 and user_id<>".$default_user_id." and user_id<>".$hotspot_user_id." ORDER BY ip_int");
  101. my %leases;
  102. foreach my $lease (@auth_records) {
  103. next if (!$lease);
  104. next if (!$lease->{mac});
  105. next if (!$lease->{ip});
  106. next if ($lease->{ip} eq $dhcp_conf{$found_subnet}->{relay_ip});
  107. $leases{$lease->{ip}}{ip}=$lease->{ip};
  108. $leases{$lease->{ip}}{comment}=$lease->{id};
  109. $leases{$lease->{ip}}{id}=$lease->{id};
  110. $leases{$lease->{ip}}{dns_name}=$lease->{dns_name};
  111. if ($lease->{comments}) { $leases{$lease->{ip}}{comment}=translit($lease->{comments}); }
  112. $leases{$lease->{ip}}{mac}=uc(mac_splitted($lease->{mac}));
  113. if ($lease->{dhcp_acl}) {
  114. $leases{$lease->{ip}}{acl}=trim($lease->{dhcp_acl});
  115. $leases{$lease->{ip}}{acl}=~s/;/,/g;
  116. }
  117. $leases{$lease->{ip}}{acl}='' if (!$leases{$lease->{ip}}{acl});
  118. }
  119. my %active_leases;
  120. foreach my $lease (@current_static_leases) {
  121. my @words = split(/\s+/,$lease);
  122. my %tmp_lease;
  123. if ($lease=~/^(\d*)\s+/) { $tmp_lease{id}=$1; };
  124. next if (!defined($tmp_lease{id}));
  125. foreach my $option (@words) {
  126. next if (!$option);
  127. $option=trim($option);
  128. next if (!$option);
  129. my @tmp = split(/\=/,$option);
  130. my $token = trim($tmp[0]);
  131. my $value = trim($tmp[1]);
  132. next if (!$token);
  133. next if (!$value);
  134. $value=~s/\"//g;
  135. if ($token=~/^address$/i) { $tmp_lease{ip}=GetIP($value); }
  136. if ($token=~/^mac-address$/i) { $tmp_lease{mac}=uc(mac_splitted($value)); }
  137. if ($token=~/^host-name$/i) { $tmp_lease{dhcp_hostname}=$value; }
  138. if ($token=~/^address-lists$/i) { $tmp_lease{acl}=$value; }
  139. }
  140. next if (!$tmp_lease{ip});
  141. next if (!$tmp_lease{mac});
  142. if ($tmp_lease{dhcp_hostname}) {
  143. my $dSQL="Update User_auth set dhcp_hostname='".$tmp_lease{dhcp_hostname}."' where deleted=0 and ip_int=".StrToIp($tmp_lease{ip})." and mac='".lc($tmp_lease{mac})."'";
  144. db_log_debug($dbh,"Update dhcp hostname $tmp_lease{dhcp_hostname} for $tmp_lease{ip} [$tmp_lease{mac}] from mikrotik dhcp server");
  145. do_sql($dbh,$dSQL);
  146. }
  147. #skip dynamic leases. this check MUST BE After update dhcp hostname!!!
  148. next if ($lease=~/^(\d*)\s+D\s+/);
  149. $active_leases{$tmp_lease{ip}}{ip}=$tmp_lease{ip};
  150. $active_leases{$tmp_lease{ip}}{mac}=$tmp_lease{mac};
  151. $active_leases{$tmp_lease{ip}}{id}=$tmp_lease{id};
  152. $active_leases{$tmp_lease{ip}}{acl}='';
  153. if ($tmp_lease{acl}) {
  154. $active_leases{$tmp_lease{ip}}{acl}=$tmp_lease{acl};
  155. }
  156. }
  157. if ($debug) { log_debug("Active leases: ".Dumper(\%active_leases)); }
  158. #sync state
  159. foreach my $ip (keys %active_leases) {
  160. if (!exists $leases{$ip}) {
  161. db_log_verbose($dbh,"Address $ip not found in stat. Remove from router.");
  162. push(@cmd_list,':foreach i in [/ip dhcp-server lease find where address='.$ip.' ] do={/ip dhcp-server lease remove $i};');
  163. push(@cmd_list,'/ip dhcp-server lease remove [find address='.$ip.']');
  164. next;
  165. }
  166. if ($leases{$ip}{mac}!~/$active_leases{$ip}{mac}/i) {
  167. db_log_verbose($dbh,"Mac-address mismatch for ip $ip. stat: $leases{$ip}{mac} active: $active_leases{$ip}{mac}. Remove lease from router.");
  168. push(@cmd_list,':foreach i in [/ip dhcp-server lease find where address='.$ip.' ] do={/ip dhcp-server lease remove $i};');
  169. push(@cmd_list,'/ip dhcp-server lease remove [find address='.$ip.']');
  170. next;
  171. }
  172. next if (!$leases{$ip}{acl} and !$active_leases{$ip}{acl});
  173. if ($leases{$ip}{acl}!~/$active_leases{$ip}{acl}/) {
  174. db_log_error($dbh,"Acl mismatch for ip $ip. stat: $leases{$ip}{acl} active: $active_leases{$ip}{acl}. Remove lease from router.");
  175. push(@cmd_list,':foreach i in [/ip dhcp-server lease find where address='.$ip.' ] do={/ip dhcp-server lease remove $i};');
  176. push(@cmd_list,'/ip dhcp-server lease remove [find address='.$ip.']');
  177. next;
  178. }
  179. }
  180. foreach my $ip (keys %leases) {
  181. my $acl='';
  182. if ($leases{$ip}{acl}) { $acl = 'address-lists='.$leases{$ip}{acl}; }
  183. my $comment = $leases{$ip}{comment};
  184. $comment =~s/\=//g;
  185. my $dns_name='';
  186. if ($leases{$ip}{dns_name}) { $dns_name = $leases{$ip}{dns_name}; }
  187. $dns_name =~s/\=//g;
  188. if ($dns_name) { $comment = 'comment="'.$dns_name." - ".$comment.'"'; } else { $comment = 'comment="'.$comment.'"'; }
  189. if (!exists $active_leases{$ip}) {
  190. db_log_verbose($dbh,"Address $ip not found in router. Create static lease record.");
  191. #remove static and dynamic records for mac
  192. push(@cmd_list,':foreach i in [/ip dhcp-server lease find where mac-address='.uc($leases{$ip}{mac}).' ] do={/ip dhcp-server lease remove $i};');
  193. push(@cmd_list,'/ip dhcp-server lease remove [find mac-address='.uc($leases{$ip}{mac}).']');
  194. #remove current ip binding
  195. push(@cmd_list,':foreach i in [/ip dhcp-server lease find where address='.$ip.' ] do={/ip dhcp-server lease remove $i};');
  196. push(@cmd_list,'/ip dhcp-server lease remove [find address='.$ip.']');
  197. #add new bind
  198. push(@cmd_list,'/ip dhcp-server lease add address='.$ip.' mac-address='.$leases{$ip}{mac}.' '.$acl.' server=dhcp-'.$int.' '.$comment);
  199. next;
  200. }
  201. if ($leases{$ip}{mac}!~/$active_leases{$ip}{mac}/i) {
  202. db_log_error($dbh,"Mac-address mismatch for ip $ip. stat: $leases{$ip}{mac} active: $active_leases{$ip}{mac}. Create static lease record.");
  203. #remove static and dynamic records for mac
  204. push(@cmd_list,':foreach i in [/ip dhcp-server lease find where mac-address='.uc($leases{$ip}{mac}).' ] do={/ip dhcp-server lease remove $i};');
  205. push(@cmd_list,'/ip dhcp-server lease remove [find mac-address='.uc($leases{$ip}{mac}).']');
  206. #remove current ip binding
  207. push(@cmd_list,':foreach i in [/ip dhcp-server lease find where address='.$ip.' ] do={/ip dhcp-server lease remove $i};');
  208. push(@cmd_list,'/ip dhcp-server lease remove [find address='.$ip.']');
  209. #add new bind
  210. push(@cmd_list,'/ip dhcp-server lease add address='.$ip.' mac-address='.$leases{$ip}{mac}.' '.$acl.' server=dhcp-'.$int.' '.$comment);
  211. next;
  212. }
  213. next if (!$leases{$ip}{acl} and !$active_leases{$ip}{acl});
  214. if ($leases{$ip}{acl}!~/$active_leases{$ip}{acl}/) {
  215. db_log_error($dbh,"Acl mismatch for ip $ip. stat: $leases{$ip}{acl} active: $active_leases{$ip}{acl}. Create static lease record.");
  216. push(@cmd_list,':foreach i in [/ip dhcp-server lease find where mac-address='.uc($leases{$ip}{mac}).' ] do={/ip dhcp-server lease remove $i};');
  217. push(@cmd_list,'/ip dhcp-server lease remove [find mac-address='.uc($leases{$ip}{mac}).']');
  218. push(@cmd_list,'/ip dhcp-server lease add address='.$ip.' mac-address='.$leases{$ip}{mac}.' '.$acl.' server=dhcp-'.$int.' '.$comment);
  219. next;
  220. }
  221. }
  222. }#end interface dhcp loop
  223. }#end dhcp config
  224. #access lists config
  225. if ($gate->{internet_gateway}) {
  226. db_log_verbose($dbh,"Sync user state at router $router_name [".$router_ip."] started.");
  227. #get userid list
  228. my $user_auth_sql="SELECT User_auth.ip, User_auth.filter_group_id, User_auth.queue_id
  229. FROM User_auth, User_list
  230. WHERE User_auth.user_id = User_list.id
  231. AND User_auth.deleted =0
  232. AND User_auth.enabled =1
  233. AND User_auth.blocked =0
  234. AND User_list.blocked =0
  235. AND User_auth.user_id <> $hotspot_user_id
  236. ORDER BY ip_int";
  237. my $user_auth_list = $dbh->prepare($user_auth_sql);
  238. if ( !defined $user_auth_list ) { die "Cannot prepare statement: $DBI::errstr\n"; }
  239. $user_auth_list->execute;
  240. # user auth list
  241. my $authlist_ref = $user_auth_list->fetchall_arrayref();
  242. $user_auth_list->finish();
  243. my %users;
  244. my %lists;
  245. my @squid_users=();
  246. foreach my $row (@$authlist_ref) {
  247. if ($connected_users_only) {
  248. next if (!$connected_users->match_string($row->[0]));
  249. }
  250. $users{'group_'.$row->[1]}->{$row->[0]}=1;
  251. $users{'group_all'}->{$row->[0]}=1;
  252. $lists{'group_'.$row->[1]}=1;
  253. if ($row->[2]) { $users{'queue_'.$row->[2]}->{$row->[0]}=1; }
  254. }
  255. #full list
  256. $lists{'group_all'}=1;
  257. #get queue list
  258. my $queue_list = $dbh->prepare( "SELECT id,queue_name,Download,Upload FROM Queue_list" );
  259. if ( !defined $queue_list ) { die "Cannot prepare statement: $DBI::errstr\n"; }
  260. $queue_list->execute;
  261. # user auth list
  262. my $queuelist_ref = $queue_list->fetchall_arrayref();
  263. $queue_list->finish();
  264. my %queues;
  265. foreach my $row (@$queuelist_ref) {
  266. $lists{'queue_'.$row->[0]}=1;
  267. next if ((!$row->[2]) and !($row->[3]));
  268. $queues{'queue_'.$row->[0]}{id}=$row->[0];
  269. $queues{'queue_'.$row->[0]}{down}=$row->[2];
  270. $queues{'queue_'.$row->[0]}{up}=$row->[3];
  271. }
  272. #print Dumper(\%users) if ($debug);
  273. #get filters
  274. my $filter_list = $dbh->prepare( "SELECT id,name,proto,dst,dstport,action FROM Filter_list where type=0" );
  275. if ( !defined $filter_list ) { die "Cannot prepare statement: $DBI::errstr\n"; }
  276. $filter_list->execute;
  277. # user auth list
  278. my $filterlist_ref = $filter_list->fetchall_arrayref();
  279. $filter_list->finish();
  280. my %filters;
  281. foreach my $row (@$filterlist_ref) {
  282. $filters{$row->[0]}->{id}=$row->[0];
  283. $filters{$row->[0]}->{proto}=$row->[2];
  284. $filters{$row->[0]}->{dst}=$row->[3];
  285. $filters{$row->[0]}->{port}=$row->[4];
  286. $filters{$row->[0]}->{action}=$row->[5];
  287. }
  288. #print Dumper(\%filters) if ($debug);
  289. #get groups
  290. my $group_list = $dbh->prepare( "SELECT group_id,filter_id,Group_filters.order FROM Group_filters order by Group_filters.group_id,Group_filters.order" );
  291. if ( !defined $group_list ) { die "Cannot prepare statement: $DBI::errstr\n"; }
  292. $group_list->execute;
  293. # user auth list
  294. my $grouplist_ref = $group_list->fetchall_arrayref();
  295. $group_list->finish();
  296. my %group_filters;
  297. my $index=1;
  298. foreach my $row (@$grouplist_ref) {
  299. #{group-name}->{filter_id}=order
  300. $group_filters{'group_'.$row->[0]}->{$index}=$row->[1];
  301. $index++;
  302. }
  303. #print Dumper(\%group_filters) if ($debug);
  304. my %cur_users;
  305. foreach my $group_name (keys %lists) {
  306. my @address_lists=log_cmd4($t,'/ip firewall address-list print terse without-paging where list='.$group_name);
  307. foreach my $row (@address_lists) {
  308. $row=trim($row);
  309. next if (!$row);
  310. my @address=split(' ',$row);
  311. foreach my $row (@address) {
  312. if ($row=~/address\=(.*)/i) { $cur_users{$group_name}{$1}=1; }
  313. }
  314. }
  315. }
  316. #new-ips
  317. foreach my $group_name (keys %users) {
  318. foreach my $user_ip (keys %{$users{$group_name}}) {
  319. if (!exists($cur_users{$group_name}{$user_ip})) {
  320. db_log_verbose($dbh,"Add user with ip: $user_ip to access-list $group_name");
  321. push(@cmd_list,"/ip firewall address-list add address=".$user_ip." list=".$group_name);
  322. }
  323. }
  324. }
  325. #old-ips
  326. foreach my $group_name (keys %cur_users) {
  327. foreach my $user_ip (keys %{$cur_users{$group_name}}) {
  328. if (!exists($users{$group_name}{$user_ip})) {
  329. db_log_verbose($dbh,"Remove user with ip: $user_ip from access-list $group_name");
  330. push(@cmd_list,":foreach i in [/ip firewall address-list find where address=".$user_ip." and list=".$group_name."] do={/ip firewall address-list remove \$i};");
  331. }
  332. }
  333. }
  334. timestamp;
  335. #sync firewall rules
  336. #sync group chains
  337. my @chain_list=log_cmd4($t,'/ip firewall filter print terse without-paging where chain=Users and action=jump');
  338. my %cur_chain;
  339. foreach my $jump_list (@chain_list) {
  340. next if (!$jump_list);
  341. $jump_list=trim($jump_list);
  342. if ($jump_list=~/jump-target=(\S*)\s+/i) {
  343. if ($1) { $cur_chain{$1}++; }
  344. }
  345. }
  346. #old chains
  347. foreach my $group_name (keys %cur_chain) {
  348. if (!exists($group_filters{$group_name})) {
  349. push (@cmd_list,":foreach i in [/ip firewall filter find where chain=Users and action=jump and jump-target=".$group_name."] do={/ip firewall filter remove \$i};");
  350. } else {
  351. if ($cur_chain{$group_name} != 2) {
  352. push (@cmd_list,":foreach i in [/ip firewall filter find where chain=Users and action=jump and jump-target=".$group_name."] do={/ip firewall filter remove \$i};");
  353. push (@cmd_list,"/ip firewall filter add chain=Users action=jump jump-target=".$group_name." src-address-list=".$group_name);
  354. push (@cmd_list,"/ip firewall filter add chain=Users action=jump jump-target=".$group_name." dst-address-list=".$group_name);
  355. }
  356. }
  357. }
  358. #new chains
  359. foreach my $group_name (keys %group_filters) {
  360. if (!exists($cur_chain{$group_name})) {
  361. push (@cmd_list,"/ip firewall filter add chain=Users action=jump jump-target=".$group_name." src-address-list=".$group_name);
  362. push (@cmd_list,"/ip firewall filter add chain=Users action=jump jump-target=".$group_name." dst-address-list=".$group_name);
  363. }
  364. }
  365. my %chain_rules;
  366. foreach my $group_name (keys %group_filters) {
  367. next if (!$group_name);
  368. next if (!exists($group_filters{$group_name}));
  369. foreach my $filter_index (sort keys %{$group_filters{$group_name}}) {
  370. my $filter_id=$group_filters{$group_name}->{$filter_index};
  371. next if (!$filters{$filter_id});
  372. my $src_rule='chain='.$group_name;
  373. my $dst_rule='chain='.$group_name;
  374. if ($filters{$filter_id}->{action}) {
  375. $src_rule=$src_rule." action=accept";
  376. $dst_rule=$dst_rule." action=accept";
  377. } else {
  378. $src_rule=$src_rule." action=reject";
  379. $dst_rule=$dst_rule." action=reject";
  380. }
  381. if ($filters{$filter_id}->{proto} and ($filters{$filter_id}->{proto}!~/all/i)) {
  382. $src_rule=$src_rule." protocol=".$filters{$filter_id}->{proto};
  383. $dst_rule=$dst_rule." protocol=".$filters{$filter_id}->{proto};
  384. }
  385. if ($filters{$filter_id}->{dst} and $filters{$filter_id}->{dst} ne '0/0') {
  386. $src_rule=$src_rule." src-address=".trim($filters{$filter_id}->{dst});
  387. $dst_rule=$dst_rule." dst-address=".trim($filters{$filter_id}->{dst});
  388. }
  389. if ($filters{$filter_id}->{port} and $filters{$filter_id}->{port} ne '0') {
  390. $src_rule=$src_rule." src-port=".trim($filters{$filter_id}->{port});
  391. $dst_rule=$dst_rule." dst-port=".trim($filters{$filter_id}->{port});
  392. }
  393. if ($src_rule ne $dst_rule) {
  394. push(@{$chain_rules{$group_name}},$src_rule);
  395. push(@{$chain_rules{$group_name}},$dst_rule);
  396. } else {
  397. push(@{$chain_rules{$group_name}},$src_rule);
  398. }
  399. }
  400. }
  401. #print Dumper(\%chain_rules) if ($debug);
  402. #chain filters
  403. foreach my $group_name (keys %group_filters) {
  404. next if (!$group_name);
  405. my @get_filter=log_cmd4($t,'/ip firewall filter print terse without-paging where chain='.$group_name,1);
  406. my @cur_filter=();
  407. my $chain_ok=1;
  408. foreach (my $f_index=0; $f_index<scalar(@get_filter); $f_index++) {
  409. my $filter_str=trim($get_filter[$f_index]);
  410. next if (!$filter_str);
  411. next if ($filter_str!~/^(\d){1,3}/);
  412. $filter_str=~s/^\d{1,3}\s+//;
  413. $filter_str=trim($filter_str);
  414. next if (!$filter_str);
  415. push(@cur_filter,$filter_str);
  416. }
  417. #current state rules
  418. foreach (my $f_index=0; $f_index<scalar(@cur_filter); $f_index++) {
  419. my $filter_str=trim($cur_filter[$f_index]);
  420. if (!$chain_rules{$group_name}[$f_index] or $filter_str!~/$chain_rules{$group_name}[$f_index]/i) {
  421. print "Check chain $group_name error! $filter_str not found in new config. Recreate chain.\n";
  422. $chain_ok=0;
  423. last;
  424. }
  425. }
  426. #new rules
  427. if ($chain_ok and $chain_rules{$group_name} and scalar(@{$chain_rules{$group_name}})) {
  428. foreach (my $f_index=0; $f_index<scalar(@{$chain_rules{$group_name}}); $f_index++) {
  429. my $filter_str=trim($cur_filter[$f_index]);
  430. if (!$filter_str) {
  431. print "Check chain $group_name error! Not found: $chain_rules{$group_name}[$f_index]. Recreate chain.\n";
  432. $chain_ok=0;
  433. last;
  434. }
  435. $filter_str=~s/^\d//;
  436. $filter_str=trim($filter_str);
  437. if ($filter_str!~/$chain_rules{$group_name}[$f_index]/i) {
  438. print "Check chain $group_name error! Expected: $chain_rules{$group_name}[$f_index] Found: $filter_str. Recreate chain.\n";
  439. $chain_ok=0;
  440. last;
  441. }
  442. }
  443. }
  444. if (!$chain_ok) {
  445. push(@cmd_list,":foreach i in [/ip firewall filter find where chain=".$group_name." ] do={/ip firewall filter remove \$i};");
  446. foreach my $filter_str (@{$chain_rules{$group_name}}) {
  447. push(@cmd_list,'/ip firewall filter add '.$filter_str);
  448. }
  449. }
  450. }
  451. if ($shaper_enabled) {
  452. #shapers
  453. my %get_queue_type=();
  454. my %get_queue_tree=();
  455. my %get_filter_mangle=();
  456. my @tmp=log_cmd4($t,'/queue type print terse without-paging where name~"pcq_(down|up)load"');
  457. # 0 name=pcq_upload_3 kind=pcq pcq-rate=102401k pcq-limit=500KiB pcq-classifier=src-address pcq-total-limit=2000KiB pcq-burst-rate=0 pcq-burst-threshold=0 pcq-burst-time=10s
  458. #pcq-src-address-mask=32 pcq-dst-address-mask=32 pcq-src-address6-mask=64 pcq-dst-address6-mask=64
  459. foreach my $row (@tmp) {
  460. next if (!$row);
  461. $row = trim($row);
  462. next if ($row!~/^(\d){1,3}/);
  463. $row=~s/^\d{1,3}\s+//;
  464. next if (!$row);
  465. if ($row=~/name=pcq_(down|up)load_(\d){1,3}\s+/i) {
  466. next if (!$1);
  467. next if (!$2);
  468. my $direct = $1;
  469. my $index = $2;
  470. $get_queue_type{$index}{$direct}=$row;
  471. if ($row=~/pcq-rate=(\S*)\s+\S/i) {
  472. my $rate = $1;
  473. if ($rate=~/k$/i) { $rate =~s/k$//i; }
  474. $get_queue_type{$index}{$direct."-rate"}=$rate;
  475. }
  476. if ($row=~/pcq-classifier=(\S*)\s+\S/i) { $get_queue_type{$index}{$direct."-classifier"}=$1; }
  477. if ($row=~/pcq-src-address-mask=(\S*)\s+\S/i) { $get_queue_type{$index}{$direct."-src-address-mask"}=$1; }
  478. if ($row=~/pcq-dst-address-mask=(\S*)\s+\S/i) { $get_queue_type{$index}{$direct."-dst-address-mask"}=$1; }
  479. }
  480. }
  481. @tmp=();
  482. @tmp=log_cmd4($t,'/queue tree print terse without-paging where parent~"(download|upload)_root"');
  483. # 0 I name=queue_3_out parent=upload_root packet-mark=upload_3 limit-at=0 queue=*2A priority=8 max-limit=0 burst-limit=0 burst-threshold=0 burst-time=0s bucket-size=0.1
  484. # 5 I name=queue_3_vlan2_in parent=download_root_vlan2 packet-mark=download_3_vlan2 limit-at=0 queue=*2B priority=8 max-limit=0 burst-limit=0 burst-threshold=0 burst-time=0s bucket-size=0.1
  485. foreach my $row (@tmp) {
  486. next if (!$row);
  487. $row = trim($row);
  488. next if ($row!~/^(\d)/);
  489. $row=~s/^(\d*)\s+//;
  490. next if (!$row);
  491. if ($row=~/queue=pcq_(down|up)load_(\d){1,3}/i) {
  492. if ($row=~/name=queue_(\d){1,3}_out/i) {
  493. next if (!$1);
  494. my $index = $1;
  495. $get_queue_tree{$index}{up}=$row;
  496. if ($row=~/parent=(\S*)\s+\S/i) { $get_queue_tree{$index}{'up-parent'}=$1; }
  497. if ($row=~/packet-mark=(\S*)\s+\S/i) { $get_queue_tree{$index}{'up-mark'}=$1; }
  498. if ($row=~/queue=(\S*)\s+\S/i) { $get_queue_tree{$index}{'up-queue'}=$1; }
  499. }
  500. if ($row=~/name=queue_(\d){1,3}_(\S*)_in\s+/i) {
  501. next if (!$1);
  502. next if (!$2);
  503. my $index = $1;
  504. my $int_name = $2;
  505. $get_queue_tree{$index}{$int_name}{down}=$row;
  506. if ($row=~/parent=(\S*)\s+\S/i) { $get_queue_tree{$index}{$int_name}{'down-parent'}=$1; }
  507. if ($row=~/packet-mark=(\S*)\s+\S/i) { $get_queue_tree{$index}{$int_name}{'down-mark'}=$1; }
  508. if ($row=~/queue=(\S*)\s+\S/i) { $get_queue_tree{$index}{$int_name}{'down-queue'}=$1; }
  509. }
  510. }
  511. }
  512. @tmp=();
  513. @tmp=log_cmd4($t,'/ip firewall mangle print terse without-paging where action=mark-packet and new-packet-mark~"(upload|download)_[0-9]{1,3}"');
  514. # 0 chain=forward action=mark-packet new-packet-mark=upload_0 passthrough=yes src-address-list=queue_0 out-interface=sfp-sfpplus1-wan log=no log-prefix=""
  515. # 0 chain=forward action=mark-packet new-packet-mark=download_3_vlan2 passthrough=yes dst-address-list=queue_3 out-interface=vlan2 in-interface-list=WAN log=no log-prefix=""
  516. foreach my $row (@tmp) {
  517. next if (!$row);
  518. $row = trim($row);
  519. next if ($row!~/^(\d){1,3}/);
  520. $row=~s/^\d{1,3}\s+//;
  521. next if (!$row);
  522. if ($row=~/new-packet-mark=upload_(\d){1,3}\s+/i) {
  523. next if (!$1);
  524. my $index = $1;
  525. $get_filter_mangle{$index}{up}=$row;
  526. if ($row=~/src-address-list=(\S*)\s+\S/i) { $get_filter_mangle{$index}{'up-list'}=$1; }
  527. if ($row=~/out-interface=(\S*)\s+\S/i) { $get_filter_mangle{$index}{'up-dev'}=$1; }
  528. if ($row=~/new-packet-mark=(\S*)\s+\S/i) { $get_filter_mangle{$index}{'up-mark'}=$1; }
  529. }
  530. if ($row=~/new-packet-mark=download_(\d){1,3}_(\S*)\s+/i) {
  531. next if (!$1);
  532. next if (!$2);
  533. my $index = $1;
  534. my $int_name = $2;
  535. $get_filter_mangle{$index}{$int_name}{down}=$row;
  536. if ($row=~/dst-address-list=(\S*)\s+\S/i) { $get_filter_mangle{$index}{$int_name}{'down-list'}=$1; }
  537. if ($row=~/new-packet-mark=(\S*)\s+\S/i) { $get_filter_mangle{$index}{$int_name}{'down-mark'}=$1; }
  538. if ($row=~/out-interface=(\S*)\s+\S/i) { $get_filter_mangle{$index}{$int_name}{'down-dev'}=$1; }
  539. }
  540. }
  541. #print Dumper(\%get_queue_type) if ($debug);
  542. #print Dumper(\%get_queue_tree) if ($debug);
  543. #print Dumper(\%get_filter_mangle) if ($debug);
  544. my %queue_type;
  545. my %queue_tree;
  546. my %filter_mangle;
  547. #generate new config
  548. foreach my $queue_name (keys %queues) {
  549. my $q_id=$queues{$queue_name}{id};
  550. my $q_up=$queues{$queue_name}{up}+1;
  551. my $q_down=$queues{$queue_name}{down}+1;
  552. #queue_types
  553. $queue_type{$q_id}{up}="name=pcq_upload_".$q_id." kind=pcq pcq-rate=".$q_up."k pcq-limit=500KiB pcq-classifier=src-address pcq-total-limit=2000KiB pcq-burst-rate=0 pcq-burst-threshold=0 pcq-burst-time=10s pcq-src-address-mask=32 pcq-dst-address-mask=32 pcq-src-address6-mask=64 pcq-dst-address6-mask=64";
  554. $queue_type{$q_id}{down}="name=pcq_download_".$q_id." kind=pcq pcq-rate=".$q_down."k pcq-limit=500KiB pcq-classifier=dst-address pcq-total-limit=2000KiB pcq-burst-rate=0 pcq-burst-threshold=0 pcq-burst-time=10s pcq-src-address-mask=32 pcq-dst-address-mask=32 pcq-src-address6-mask=64 pcq-dst-address6-mask=64";
  555. my $queue_ok=1;
  556. if (!$get_queue_type{$q_id}{up}) { $queue_ok=0; }
  557. if ($queue_ok and abs($q_up - $get_queue_type{$q_id}{'up-rate'})>10) { $queue_ok=0; }
  558. if ($queue_ok and $get_queue_type{$q_id}{'up-classifier'}!~/src-address/i) { $queue_ok=0; }
  559. if (!$queue_ok) {
  560. push(@cmd_list,':foreach i in [/queue type find where name~"pcq_upload_'.$q_id.'" ] do={/queue type remove $i};');
  561. push(@cmd_list,'/queue type add '.$queue_type{$q_id}{up});
  562. }
  563. $queue_ok=1;
  564. if (!$get_queue_type{$q_id}{down}) { $queue_ok=0; }
  565. if ($queue_ok and abs($q_up - $get_queue_type{$q_id}{'down-rate'})>10) { $queue_ok=0; }
  566. if ($queue_ok and $get_queue_type{$q_id}{'down-classifier'}!~/dst-address/i) { $queue_ok=0; }
  567. if (!$queue_ok) {
  568. push(@cmd_list,':foreach i in [/queue type find where name~"pcq_download_'.$q_id.'" ] do={/queue type remove $i};');
  569. push(@cmd_list,'/queue type add '.$queue_type{$q_id}{down});
  570. }
  571. #upload queue
  572. $queue_tree{$q_id}{up}="name=queue_".$q_id."_out parent=upload_root packet-mark=upload_".$q_id." limit-at=0 queue=pcq_upload_".$q_id." priority=8 max-limit=0 burst-limit=0 burst-threshold=0 burst-time=0s bucket-size=0.1";
  573. $queue_ok=1;
  574. if (!$get_queue_tree{$q_id}{up}) { $queue_ok=0; }
  575. if ($queue_ok and ($get_queue_tree{$q_id}{'up-parent'} ne "upload_root")) { $queue_ok=0; print "$get_queue_tree{$q_id}{'up-parent'} ==== upload_root \n"; }
  576. if ($queue_ok and ($get_queue_tree{$q_id}{'up-mark'} ne "upload_".$q_id)) { $queue_ok=0; print "$get_queue_tree{$q_id}{'up-mark'} ==== upload_".$q_id."\n"; }
  577. if ($queue_ok and ($get_queue_tree{$q_id}{'up-queue'} ne "pcq_upload_".$q_id)) { $queue_ok=0; print "$get_queue_tree{$q_id}{'up-queue'} ==== pcq_upload_".$q_id."\n"; }
  578. if (!$queue_ok) {
  579. push(@cmd_list,':foreach i in [/queue tree find where name~"queue_'.$q_id.'_out" ] do={/queue tree remove $i};');
  580. push(@cmd_list,'/queue tree add '.$queue_tree{$q_id}{up});
  581. }
  582. $filter_mangle{$q_id}{up}="chain=forward action=mark-packet new-packet-mark=upload_".$q_id." passthrough=yes src-address-list=queue_".$q_id." out-interface=".$wan_dev." log=no log-prefix=\"\"";
  583. $queue_ok=1;
  584. if (!$get_filter_mangle{$q_id}{up}) { $queue_ok=0; }
  585. if ($queue_ok and ($get_filter_mangle{$q_id}{'up-mark'} ne "upload_".$q_id)) { $queue_ok=0; }
  586. if ($queue_ok and ($get_filter_mangle{$q_id}{'up-list'} ne "queue_".$q_id)) { $queue_ok=0; }
  587. if ($queue_ok and ($get_filter_mangle{$q_id}{'up-dev'} ne $wan_dev)) { $queue_ok=0; }
  588. if (!$queue_ok) {
  589. push(@cmd_list,':foreach i in [/ip firewall mangle find where action=mark-packet and new-packet-mark~"upload_'.$q_id.'" ] do={/ip firewall mangle remove $i};');
  590. push(@cmd_list,'/ip firewall mangle add '.$filter_mangle{$q_id}{up});
  591. }
  592. #download
  593. my @lan_int=split(/;/,$lan_dev);
  594. foreach my $int (@lan_int) {
  595. next if (!$int);
  596. $queue_tree{$q_id}{$int}{down}="name=queue_".$q_id."_".$int."_in parent=download_root_".$int." packet-mark=download_".$q_id."_".$int." limit-at=0 queue=pcq_download_".$q_id." priority=8 max-limit=0 burst-limit=0 burst-threshold=0 burst-time=0s bucket-size=0.1";
  597. $filter_mangle{$q_id}{$int}{down}="chain=forward action=mark-packet new-packet-mark=download_".$q_id."_".$int." passthrough=yes dst-address-list=queue_".$q_id." out-interface=".$int." in-interface-list=WAN log=no log-prefix=\"\"";
  598. $queue_ok=1;
  599. if (!$get_queue_tree{$q_id}{$int}{down}) { $queue_ok=0; }
  600. if ($queue_ok and ($get_queue_tree{$q_id}{$int}{'down-parent'} ne "download_root_".$int)) { $queue_ok=0; }
  601. if ($queue_ok and ($get_queue_tree{$q_id}{$int}{'down-mark'} ne "download_".$q_id."_".$int)) { $queue_ok=0; }
  602. if ($queue_ok and ($get_queue_tree{$q_id}{$int}{'down-queue'} ne "pcq_download_".$q_id)) { $queue_ok=0; }
  603. if (!$queue_ok) {
  604. push(@cmd_list,':foreach i in [/queue tree find where name~"queue_'.$q_id."_".$int."_in".'" ] do={/queue tree remove $i};');
  605. push(@cmd_list,'/queue tree add '.$queue_tree{$q_id}{$int}{down});
  606. }
  607. $queue_ok=1;
  608. if (!$get_filter_mangle{$q_id}{$int}{down}) { $queue_ok=0; }
  609. if ($queue_ok and ($get_filter_mangle{$q_id}{$int}{'down-mark'} ne "download_".$q_id."_".$int)) { $queue_ok=0; }
  610. if ($queue_ok and ($get_filter_mangle{$q_id}{$int}{'down-list'} ne "queue_".$q_id)) { $queue_ok=0; }
  611. if ($queue_ok and ($get_filter_mangle{$q_id}{$int}{'down-dev'} ne $int)) { $queue_ok=0; }
  612. if (!$queue_ok) {
  613. push(@cmd_list,':foreach i in [/ip firewall mangle find where action=mark-packet and new-packet-mark~"download_'.$q_id."_".$int.'" ] do={/ip firewall mangle remove $i};');
  614. push(@cmd_list,'/ip firewall mangle add '.$filter_mangle{$q_id}{$int}{down});
  615. }
  616. }
  617. #end shaper
  618. }
  619. }
  620. }#end access lists config
  621. if (scalar(@cmd_list)) {
  622. foreach my $cmd (@cmd_list) {
  623. log_info("$cmd");
  624. # print "$cmd\n" if ($debug);
  625. log_cmd($t,$cmd);
  626. }
  627. }
  628. db_log_verbose($dbh,"Sync user state at router $router_name [".$router_ip."] stopped.");
  629. }
  630. $dbh->disconnect();
  631. if (IsMyPID($SPID)) { Remove_PID($SPID); };
  632. do_exit 0;