| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865 |
- #!/usr/bin/perl
- #
- # Copyright (C) Roman Dmitiriev, rnd@rajven.ru
- #
- use utf8;
- use English;
- use base;
- use FindBin '$Bin';
- use lib "$Bin/";
- use strict;
- use Time::Local;
- use FileHandle;
- use Data::Dumper;
- use eyelib::config;
- use eyelib::main;
- use eyelib::cmd;
- use Net::Patricia;
- use Date::Parse;
- use eyelib::net_utils;
- use eyelib::mysql;
- use DBI;
- use Fcntl qw(:flock);
- use Parallel::ForkManager;
- use Net::DNS;
- #$debug = 1;
- open(SELF,"<",$0) or die "Cannot open $0 - $!";
- flock(SELF, LOCK_EX|LOCK_NB) or exit 1;
- $|=1;
- if (IsNotRun($SPID)) { Add_PID($SPID); } else { die "Warning!!! $SPID already runnning!\n"; }
- my $fork_count = $cpu_count*10;
- #flag for operation status
- my $all_ok = 1;
- my @gateways =();
- #select undeleted mikrotik routers only
- if ($ARGV[0]) {
- my $router = get_record_sql($dbh,'SELECT * FROM devices WHERE (device_type=2 OR device_type=0) and (user_acl=1 or dhcp=1) and deleted=0 and vendor_id=9 and id='.$ARGV[0]);
- if ($router) { push(@gateways,$router); }
- } else {
- @gateways = get_records_sql($dbh,'SELECT * FROM devices WHERE (device_type=2 OR device_type=0) and (user_acl=1 or dhcp=1) and deleted=0 and vendor_id=9');
- }
- my $dhcp_networks = new Net::Patricia;
- my %dhcp_conf;
- my @subnets=get_records_sql($dbh,'SELECT * FROM subnets WHERE dhcp=1 and office=1 and vpn=0 ORDER BY ip_int_start');
- foreach my $subnet (@subnets) {
- next if (!$subnet->{gateway});
- my $subnet_name = $subnet->{subnet};
- $subnet_name=~s/\/\d+$//g;
- $dhcp_networks->add_string($subnet->{subnet},$subnet_name);
- $dhcp_conf{$subnet_name}->{first_pool_ip}=IpToStr($subnet->{dhcp_start});
- $dhcp_conf{$subnet_name}->{last_pool_ip}=IpToStr($subnet->{dhcp_stop});
- $dhcp_conf{$subnet_name}->{relay_ip}=IpToStr($subnet->{gateway});
- my $dhcp_info=GetDhcpRange($subnet->{subnet});
- $dhcp_conf{$subnet_name}->{first_ip} = $dhcp_info->{first_ip};
- $dhcp_conf{$subnet_name}->{last_ip} = $dhcp_info->{last_ip};
- $dhcp_conf{$subnet_name}->{first_ip_aton}=StrToIp($dhcp_info->{first_ip});
- $dhcp_conf{$subnet_name}->{last_ip_aton}=StrToIp($dhcp_info->{last_ip});
- }
- my $pm = Parallel::ForkManager->new($fork_count);
- #save changed records
- my @changes_found = get_records_sql($dbh,"SELECT id FROM User_auth WHERE changed=1");
- foreach my $gate (@gateways) {
- next if (!$gate);
- $pm->start and next;
- $dbh = init_db();
- my $router_name=$gate->{device_name};
- my $router_ip=$gate->{ip};
- my $shaper_enabled = $gate->{queue_enabled};
- my $connected_users_only = $gate->{connected_user_only};
- my $connected_users = new Net::Patricia;
- my @changed_ref=();
- my @lan_int=();
- my @wan_int=();
- my @l3_int = get_records_sql($dbh,'SELECT * FROM device_l3_interfaces WHERE device_id='.$gate->{'id'});
- foreach my $l3 (@l3_int) {
- if ($l3->{'interface_type'} eq '0') { push(@lan_int,$l3->{'name'}); }
- if ($l3->{'interface_type'} eq '1') { push(@wan_int,$l3->{'name'}); }
- }
- my @cmd_list=();
- $gate = netdev_set_auth($gate);
- $gate->{login}.='+ct400w';
- my $t = netdev_login($gate);
- foreach my $int (@lan_int) { #interface dhcp loop
- next if (!$int);
- $int=trim($int);
- #get ip addr at interface
- my @int_addr=netdev_cmd($gate,$t,$gate->{proto},'/ip address print terse without-paging where interface='.$int,1);
- log_debug("Get interfaces: ".Dumper(\@int_addr));
- my $found_subnet;
- foreach my $int_str(@int_addr) {
- $int_str=trim($int_str);
- next if (!$int_str);
- if ($int_str=~/\s+address=(\S*)\s+/i) {
- my $gate_interface=$1;
- if ($gate_interface) {
- my $gate_ip=$gate_interface;
- $gate_ip=~s/\/.*$//;
- #search for first match
- if (!$found_subnet) { $found_subnet=$dhcp_networks->match_string($gate_ip); }
- #all subnets match
- if ($connected_users_only) { $connected_users->add_string($gate_interface); }
- }
- }
- }
- if (!$found_subnet) { db_log_verbose($dbh,"DHCP subnet for interface $int not found! Skip interface."); next; }
- db_log_verbose($dbh,"Analyze interface $int. Found: ".Dumper($dhcp_conf{$found_subnet}));
- #dhcp config
- if ($gate->{dhcp}) {
- #fetch current dhcp records
- my @ret_static_leases=netdev_cmd($gate,$t,$gate->{proto},'/ip dhcp-server lease print terse without-paging where server=dhcp-'.$int,1);
- log_debug("Get dhcp leases:".Dumper(\@ret_static_leases));
- my @current_static_leases=();
- foreach my $str (@ret_static_leases) {
- next if (!$str);
- $str=trim($str);
- if ($str=~/^\d/) {
- log_debug("Found current static lease record: ".$str);
- push(@current_static_leases,$str);
- }
- }
- #select users for this interface
- my @auth_records=get_records_sql($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 ou_id !=".$default_user_ou_id." and ou_id !=".$default_hotspot_ou_id." ORDER BY ip_int");
- my %leases;
- foreach my $lease (@auth_records) {
- next if (!$lease);
- next if (!$lease->{mac});
- next if (!$lease->{ip});
- next if ($lease->{ip} eq $dhcp_conf{$found_subnet}->{relay_ip});
- $leases{$lease->{ip}}{ip}=$lease->{ip};
- $leases{$lease->{ip}}{comment}=$lease->{id};
- $leases{$lease->{ip}}{id}=$lease->{id};
- $leases{$lease->{ip}}{dns_name}=$lease->{dns_name};
- if ($lease->{comments}) { $leases{$lease->{ip}}{comment}=translit($lease->{comments}); }
- $leases{$lease->{ip}}{mac}=uc(mac_splitted($lease->{mac}));
- if ($lease->{dhcp_acl}) {
- $leases{$lease->{ip}}{acl}=trim($lease->{dhcp_acl});
- $leases{$lease->{ip}}{acl}=~s/;/,/g;
- }
- $leases{$lease->{ip}}{acl}='' if (!$leases{$lease->{ip}}{acl});
- }
- my %active_leases;
- foreach my $lease (@current_static_leases) {
- my @words = split(/\s+/,$lease);
- my %tmp_lease;
- if ($lease=~/^(\d*)\s+/) { $tmp_lease{id}=$1; };
- next if (!defined($tmp_lease{id}));
- foreach my $option (@words) {
- next if (!$option);
- $option=trim($option);
- next if (!$option);
- my @tmp = split(/\=/,$option);
- my $token = trim($tmp[0]);
- my $value = trim($tmp[1]);
- next if (!$token);
- next if (!$value);
- $value=~s/\"//g;
- if ($token=~/^address$/i) { $tmp_lease{ip}=GetIP($value); }
- if ($token=~/^mac-address$/i) { $tmp_lease{mac}=uc(mac_splitted($value)); }
- if ($token=~/^address-lists$/i) { $tmp_lease{acl}=$value; }
- }
- next if (!$tmp_lease{ip});
- next if (!$tmp_lease{mac});
- next if ($lease=~/^(\d*)\s+D\s+/);
- $active_leases{$tmp_lease{ip}}{ip}=$tmp_lease{ip};
- $active_leases{$tmp_lease{ip}}{mac}=$tmp_lease{mac};
- $active_leases{$tmp_lease{ip}}{id}=$tmp_lease{id};
- $active_leases{$tmp_lease{ip}}{acl}='';
- if ($tmp_lease{acl}) {
- $active_leases{$tmp_lease{ip}}{acl}=$tmp_lease{acl};
- }
- }
- log_debug("Active leases: ".Dumper(\%active_leases));
- #sync state
- foreach my $ip (keys %active_leases) {
- if (!exists $leases{$ip}) {
- db_log_verbose($dbh,"Address $ip not found in stat. Remove from router.");
- push(@cmd_list,':foreach i in [/ip dhcp-server lease find where address='.$ip.' ] do={/ip dhcp-server lease remove $i};');
- push(@cmd_list,'/ip dhcp-server lease remove [find address='.$ip.']');
- push(@cmd_list,'/ip arp remove [find address='.$ip.']');
- next;
- }
- if ($leases{$ip}{mac}!~/$active_leases{$ip}{mac}/i) {
- db_log_verbose($dbh,"Mac-address mismatch for ip $ip. stat: $leases{$ip}{mac} active: $active_leases{$ip}{mac}. Remove lease from router.");
- push(@cmd_list,':foreach i in [/ip dhcp-server lease find where address='.$ip.' ] do={/ip dhcp-server lease remove $i};');
- push(@cmd_list,'/ip dhcp-server lease remove [find address='.$ip.']');
- push(@cmd_list,'/ip arp remove [find address='.$ip.']');
- next;
- }
- next if (!$leases{$ip}{acl} and !$active_leases{$ip}{acl});
- if ($leases{$ip}{acl}!~/$active_leases{$ip}{acl}/) {
- db_log_error($dbh,"Acl mismatch for ip $ip. stat: $leases{$ip}{acl} active: $active_leases{$ip}{acl}. Remove lease from router.");
- push(@cmd_list,':foreach i in [/ip dhcp-server lease find where address='.$ip.' ] do={/ip dhcp-server lease remove $i};');
- push(@cmd_list,'/ip dhcp-server lease remove [find address='.$ip.']');
- push(@cmd_list,'/ip arp remove [find address='.$ip.']');
- next;
- }
- }
- foreach my $ip (keys %leases) {
- my $acl='';
- if ($leases{$ip}{acl}) { $acl = 'address-lists='.$leases{$ip}{acl}; }
- my $comment = $leases{$ip}{comment};
- $comment =~s/\=//g;
- my $dns_name='';
- if ($leases{$ip}{dns_name}) { $dns_name = $leases{$ip}{dns_name}; }
- $dns_name =~s/\=//g;
- if ($dns_name) { $comment = 'comment="'.$dns_name." - ".$comment.'"'; } else { $comment = 'comment="'.$comment.'"'; }
- if (!exists $active_leases{$ip}) {
- db_log_verbose($dbh,"Address $ip not found in router. Create static lease record.");
- #remove static and dynamic records for mac
- 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};');
- push(@cmd_list,'/ip dhcp-server lease remove [find mac-address='.uc($leases{$ip}{mac}).']');
- #remove current ip binding
- push(@cmd_list,':foreach i in [/ip dhcp-server lease find where address='.$ip.' ] do={/ip dhcp-server lease remove $i};');
- push(@cmd_list,'/ip dhcp-server lease remove [find address='.$ip.']');
- #add new bind
- push(@cmd_list,'/ip dhcp-server lease add address='.$ip.' mac-address='.$leases{$ip}{mac}.' '.$acl.' server=dhcp-'.$int.' '.$comment);
- #clear arp record
- push(@cmd_list,'/ip arp remove [find mac-address='.uc($leases{$ip}{mac}).']');
- next;
- }
- if ($leases{$ip}{mac}!~/$active_leases{$ip}{mac}/i) {
- db_log_error($dbh,"Mac-address mismatch for ip $ip. stat: $leases{$ip}{mac} active: $active_leases{$ip}{mac}. Create static lease record.");
- #remove static and dynamic records for mac
- 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};');
- push(@cmd_list,'/ip dhcp-server lease remove [find mac-address='.uc($leases{$ip}{mac}).']');
- #remove current ip binding
- push(@cmd_list,':foreach i in [/ip dhcp-server lease find where address='.$ip.' ] do={/ip dhcp-server lease remove $i};');
- push(@cmd_list,'/ip dhcp-server lease remove [find address='.$ip.']');
- #add new bind
- push(@cmd_list,'/ip dhcp-server lease add address='.$ip.' mac-address='.$leases{$ip}{mac}.' '.$acl.' server=dhcp-'.$int.' '.$comment);
- #clear arp record
- push(@cmd_list,'/ip arp remove [find mac-address='.uc($leases{$ip}{mac}).']');
- next;
- }
- next if (!$leases{$ip}{acl} and !$active_leases{$ip}{acl});
- if ($leases{$ip}{acl}!~/$active_leases{$ip}{acl}/) {
- db_log_error($dbh,"Acl mismatch for ip $ip. stat: $leases{$ip}{acl} active: $active_leases{$ip}{acl}. Create static lease record.");
- 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};');
- push(@cmd_list,'/ip dhcp-server lease remove [find mac-address='.uc($leases{$ip}{mac}).']');
- push(@cmd_list,'/ip dhcp-server lease add address='.$ip.' mac-address='.$leases{$ip}{mac}.' '.$acl.' server=dhcp-'.$int.' '.$comment);
- #clear arp record
- push(@cmd_list,'/ip arp remove [find mac-address='.uc($leases{$ip}{mac}).']');
- next;
- }
- }
- }#end interface dhcp loop
- }#end dhcp config
- #access lists config
- if ($gate->{user_acl}) {
- db_log_verbose($dbh,"Sync user state at router $router_name [".$router_ip."] started.");
- #get userid list
- my $user_auth_sql="SELECT User_auth.ip, User_auth.filter_group_id, User_auth.queue_id, User_auth.id
- FROM User_auth, User_list
- WHERE User_auth.user_id = User_list.id
- AND User_auth.deleted =0
- AND User_auth.enabled =1
- AND User_auth.blocked =0
- AND User_list.blocked =0
- AND User_list.enabled =1
- AND User_auth.ou_id <> $default_hotspot_ou_id
- ORDER BY ip_int";
- my @authlist_ref = get_records_sql($dbh,$user_auth_sql);
- my %users;
- my %lists;
- my %found_users;
- foreach my $row (@authlist_ref) {
- if ($connected_users_only) { next if (!$connected_users->match_string($row->{ip})); }
- #skip not office ip's
- next if (!$office_networks->match_string($row->{ip}));
- $found_users{$row->{'id'}}=$row->{ip};
- #filter group acl's
- $users{'group_'.$row->{filter_group_id}}->{$row->{ip}}=1;
- $users{'group_all'}->{$row->{ip}}=1;
- $lists{'group_'.$row->{filter_group_id}}=1;
- #queue acl's
- if ($row->{queue_id}) { $users{'queue_'.$row->{queue_id}}->{$row->{ip}}=1; }
- }
- log_debug("Users status:".Dumper(\%users));
- #full list
- $lists{'group_all'}=1;
- #get queue list
- my @queuelist_ref = get_records_sql($dbh,"SELECT * FROM Queue_list");
- my %queues;
- foreach my $row (@queuelist_ref) {
- $lists{'queue_'.$row->{id}}=1;
- next if ((!$row->{Download}) and !($row->{Upload}));
- $queues{'queue_'.$row->{id}}{id}=$row->{id};
- $queues{'queue_'.$row->{id}}{down}=$row->{Download};
- $queues{'queue_'.$row->{id}}{up}=$row->{Upload};
- }
- log_debug("Queues status:".Dumper(\%queues));
- my @filterlist_ref = get_records_sql($dbh,"SELECT * FROM Filter_list where type=0");
- my %filters;
- my %dyn_filters;
- my $max_filter_rec = get_record_sql($dbh,"SELECT MAX(id) FROM Filter_list");
- my $max_filter_id = $max_filter_rec->{id};
- my $dyn_filters_base = $max_filter_id+1000;
- my $dyn_filters_index = $dyn_filters_base;
- foreach my $row (@filterlist_ref) {
- #if dst - ip address
- if (is_ip($row->{dst})) {
- $filters{$row->{id}}->{id}=$row->{id};
- $filters{$row->{id}}->{proto}=$row->{proto};
- $filters{$row->{id}}->{dst}=$row->{dst};
- $filters{$row->{id}}->{dstport}=$row->{dstport};
- $filters{$row->{id}}->{srcport}=$row->{srcport};
- #set false for dns dst flag
- $filters{$row->{id}}->{dns_dst}=0;
- } else {
- #if dst not ip - check dns record
- my @dns_record=ResolveNames($row->{dst},undef);
- my $resolved_ips = (scalar @dns_record>0);
- next if (!$resolved_ips);
- foreach my $resolved_ip (sort @dns_record) {
- next if (!$resolved_ip);
- #enable dns dst filters
- $filters{$row->{id}}->{dns_dst}=1;
- #add dynamic dns filter
- $filters{$dyn_filters_index}->{id}=$row->{id};
- $filters{$dyn_filters_index}->{proto}=$row->{proto};
- $filters{$dyn_filters_index}->{dst}=$resolved_ip;
- $filters{$dyn_filters_index}->{dstport}=$row->{dstport};
- $filters{$dyn_filters_index}->{srcport}=$row->{srcport};
- $filters{$dyn_filters_index}->{dns_dst}=0;
- #save new filter dns id for original filter id
- push(@{$dyn_filters{$row->{id}}},$dyn_filters_index);
- $dyn_filters_index++;
- }
- }
- }
- log_debug("Filters status:". Dumper(\%filters));
- log_debug("DNS-filters status:". Dumper(\%dyn_filters));
- #clean unused filter records
- do_sql($dbh,"DELETE FROM Group_filters WHERE group_id NOT IN (SELECT id FROM Group_list)");
- do_sql($dbh,"DELETE FROM Group_filters WHERE filter_id NOT IN (SELECT id FROM Filter_list)");
- my @grouplist_ref = get_records_sql($dbh,"SELECT `group_id`,`filter_id`,`order`,`action` FROM Group_filters ORDER BY Group_filters.group_id,Group_filters.order");
- my %group_filters;
- my $index=0;
- foreach my $row (@grouplist_ref) {
- #if dst dns filter not found
- if (!$filters{$row->{filter_id}}->{dns_dst}) {
- $group_filters{'group_'.$row->{group_id}}->{$index}->{filter_id}=$row->{filter_id};
- $group_filters{'group_'.$row->{group_id}}->{$index}->{action}=$row->{action};
- $index++;
- } else {
- #if found dns dst filters - add
- if (exists $dyn_filters{$row->{filter_id}}) {
- my @dyn_ips = @{$dyn_filters{$row->{filter_id}}};
- if (scalar @dyn_ips >0) {
- for (my $i = 0; $i < scalar @dyn_ips; $i++) {
- $group_filters{'group_'.$row->{group_id}}->{$index}->{filter_id}=$dyn_ips[$i];
- $group_filters{'group_'.$row->{group_id}}->{$index}->{action}=$row->{action};
- $index++;
- }
- }
- }
- }
- }
- log_debug("Group filters: ".Dumper(\%group_filters));
- my %cur_users;
- foreach my $group_name (keys %lists) {
- my @address_lists=netdev_cmd($gate,$t,$gate->{proto},'/ip firewall address-list print terse without-paging where list='.$group_name,1);
- log_debug("Get address lists:".Dumper(\@address_lists));
- foreach my $row (@address_lists) {
- $row=trim($row);
- next if (!$row);
- my @address=split(' ',$row);
- foreach my $row (@address) {
- if ($row=~/address\=(.*)/i) { $cur_users{$group_name}{$1}=1; }
- }
- }
- }
- #new-ips
- foreach my $group_name (keys %users) {
- foreach my $user_ip (keys %{$users{$group_name}}) {
- if (!exists($cur_users{$group_name}{$user_ip})) {
- db_log_verbose($dbh,"Add user with ip: $user_ip to access-list $group_name");
- push(@cmd_list,"/ip firewall address-list add address=".$user_ip." list=".$group_name);
- }
- }
- }
- #old-ips
- foreach my $group_name (keys %cur_users) {
- foreach my $user_ip (keys %{$cur_users{$group_name}}) {
- if (!exists($users{$group_name}{$user_ip})) {
- db_log_verbose($dbh,"Remove user with ip: $user_ip from access-list $group_name");
- 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};");
- }
- }
- }
- timestamp;
- #sync firewall rules
- #sync group chains
- my @chain_list=netdev_cmd($gate,$t,$gate->{proto},'/ip firewall filter print terse without-paging where chain=Users and action=jump',1);
- log_debug("Get firewall chains:".Dumper(\@chain_list));
- my %cur_chain;
- foreach my $jump_list (@chain_list) {
- next if (!$jump_list);
- $jump_list=trim($jump_list);
- if ($jump_list=~/jump-target=(\S*)\s+/i) {
- if ($1) { $cur_chain{$1}++; }
- }
- }
- #old chains
- foreach my $group_name (keys %cur_chain) {
- if (!exists($group_filters{$group_name})) {
- 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};");
- } else {
- if ($cur_chain{$group_name} != 2) {
- 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};");
- push (@cmd_list,"/ip firewall filter add chain=Users action=jump jump-target=".$group_name." src-address-list=".$group_name);
- push (@cmd_list,"/ip firewall filter add chain=Users action=jump jump-target=".$group_name." dst-address-list=".$group_name);
- }
- }
- }
- #new chains
- foreach my $group_name (keys %group_filters) {
- if (!exists($cur_chain{$group_name})) {
- push (@cmd_list,"/ip firewall filter add chain=Users action=jump jump-target=".$group_name." src-address-list=".$group_name);
- push (@cmd_list,"/ip firewall filter add chain=Users action=jump jump-target=".$group_name." dst-address-list=".$group_name);
- }
- }
- my %chain_rules;
- foreach my $group_name (keys %group_filters) {
- next if (!$group_name);
- next if (!exists($group_filters{$group_name}));
- foreach my $filter_index (sort keys %{$group_filters{$group_name}}) {
- my $filter_id=$group_filters{$group_name}->{$filter_index}->{filter_id};
- next if (!$filters{$filter_id});
- next if ($filters{$filter_id}->{dns_dst});
- my $src_rule='chain='.$group_name;
- my $dst_rule='chain='.$group_name;
- if ($group_filters{$group_name}->{$filter_index}->{action}) {
- $src_rule=$src_rule." action=accept";
- $dst_rule=$dst_rule." action=accept";
- } else {
- $src_rule=$src_rule." action=reject";
- $dst_rule=$dst_rule." action=reject";
- }
- if ($filters{$filter_id}->{proto} and ($filters{$filter_id}->{proto}!~/all/i)) {
- $src_rule=$src_rule." protocol=".$filters{$filter_id}->{proto};
- $dst_rule=$dst_rule." protocol=".$filters{$filter_id}->{proto};
- }
- if ($filters{$filter_id}->{dst} and $filters{$filter_id}->{dst} ne '0/0') {
- $src_rule=$src_rule." src-address=".trim($filters{$filter_id}->{dst});
- $dst_rule=$dst_rule." dst-address=".trim($filters{$filter_id}->{dst});
- }
- #dstport and srcport
- if (!$filters{$filter_id}->{dstport}) { $filters{$filter_id}->{dstport}=0; }
- if (!$filters{$filter_id}->{srcport}) { $filters{$filter_id}->{srcport}=0; }
- if ($filters{$filter_id}->{dstport} ne '0' and $filters{$filter_id}->{srcport} ne '0') {
- $src_rule=$src_rule." dst-port=".trim($filters{$filter_id}->{srcport})." src-port=".trim($filters{$filter_id}->{dstport});
- $dst_rule=$dst_rule." src-port=".trim($filters{$filter_id}->{srcport})." dst-port=".trim($filters{$filter_id}->{dstport});
- }
- if ($filters{$filter_id}->{dstport} eq '0' and $filters{$filter_id}->{srcport} ne '0') {
- $src_rule=$src_rule." dst-port=".trim($filters{$filter_id}->{srcport});
- $dst_rule=$dst_rule." src-port=".trim($filters{$filter_id}->{srcport});
- }
- if ($filters{$filter_id}->{dstport} ne '0' and $filters{$filter_id}->{srcport} eq '0') {
- $src_rule=$src_rule." src-port=".trim($filters{$filter_id}->{dstport});
- $dst_rule=$dst_rule." dst-port=".trim($filters{$filter_id}->{dstport});
- }
- if ($src_rule ne $dst_rule) {
- push(@{$chain_rules{$group_name}},$src_rule);
- push(@{$chain_rules{$group_name}},$dst_rule);
- } else {
- push(@{$chain_rules{$group_name}},$src_rule);
- }
- }
- }
- #chain filters
- foreach my $group_name (keys %group_filters) {
- next if (!$group_name);
- my @get_filter=netdev_cmd($gate,$t,$gate->{proto},'/ip firewall filter print terse without-paging where chain='.$group_name,1);
- chomp(@get_filter);
- my @cur_filter=();
- my $chain_ok=1;
- foreach (my $f_index=0; $f_index<scalar(@get_filter); $f_index++) {
- my $filter_str=trim($get_filter[$f_index]);
- next if (!$filter_str);
- next if ($filter_str!~/^(\d){1,3}/);
- $filter_str=~s/[^[:ascii:]]//g;
- $filter_str=~s/^\d{1,3}\s+//;
- $filter_str=trim($filter_str);
- next if (!$filter_str);
- push(@cur_filter,$filter_str);
- }
- log_debug("Current filters:".Dumper(\@cur_filter));
- log_debug("New filters:".Dumper($chain_rules{$group_name}));
- #current state rules
- foreach (my $f_index=0; $f_index<scalar(@cur_filter); $f_index++) {
- my $filter_str=trim($cur_filter[$f_index]);
- if (!$chain_rules{$group_name}[$f_index] or $filter_str!~/$chain_rules{$group_name}[$f_index]/i) {
- print "Check chain $group_name error! $filter_str not found in new config. Recreate chain.\n";
- $chain_ok=0;
- last;
- }
- }
- #new rules
- if ($chain_ok and $chain_rules{$group_name} and scalar(@{$chain_rules{$group_name}})) {
- foreach (my $f_index=0; $f_index<scalar(@{$chain_rules{$group_name}}); $f_index++) {
- my $filter_str=trim($cur_filter[$f_index]);
- if (!$filter_str) {
- print "Check chain $group_name error! Not found: $chain_rules{$group_name}[$f_index]. Recreate chain.\n";
- $chain_ok=0;
- last;
- }
- $filter_str=~s/^\d//;
- $filter_str=trim($filter_str);
- if ($filter_str!~/$chain_rules{$group_name}[$f_index]/i) {
- print "Check chain $group_name error! Expected: $chain_rules{$group_name}[$f_index] Found: $filter_str. Recreate chain.\n";
- $chain_ok=0;
- last;
- }
- }
- }
- if (!$chain_ok) {
- push(@cmd_list,":foreach i in [/ip firewall filter find where chain=".$group_name." ] do={/ip firewall filter remove \$i};");
- foreach my $filter_str (@{$chain_rules{$group_name}}) {
- push(@cmd_list,'/ip firewall filter add '.$filter_str);
- }
- }
- }
- if ($shaper_enabled) {
- #shapers
- my %get_queue_type=();
- my %get_queue_tree=();
- my %get_filter_mangle=();
- my @tmp=netdev_cmd($gate,$t,$gate->{proto},'/queue type print terse without-paging where name~"pcq_(down|up)load"',1);
- log_debug("Get queues: ".Dumper(\@tmp));
- # 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
- #pcq-src-address-mask=32 pcq-dst-address-mask=32 pcq-src-address6-mask=64 pcq-dst-address6-mask=64
- foreach my $row (@tmp) {
- next if (!$row);
- $row = trim($row);
- next if ($row!~/^(\d){1,3}/);
- $row=~s/^\d{1,3}\s+//;
- next if (!$row);
- if ($row=~/name=pcq_(down|up)load_(\d){1,3}\s+/i) {
- next if (!$1);
- next if (!$2);
- my $direct = $1;
- my $index = $2;
- $get_queue_type{$index}{$direct}=$row;
- if ($row=~/pcq-rate=(\S*)\s+\S/i) {
- my $rate = $1;
- if ($rate=~/k$/i) { $rate =~s/k$//i; }
- $get_queue_type{$index}{$direct."-rate"}=$rate;
- }
- if ($row=~/pcq-classifier=(\S*)\s+\S/i) { $get_queue_type{$index}{$direct."-classifier"}=$1; }
- if ($row=~/pcq-src-address-mask=(\S*)\s+\S/i) { $get_queue_type{$index}{$direct."-src-address-mask"}=$1; }
- if ($row=~/pcq-dst-address-mask=(\S*)\s+\S/i) { $get_queue_type{$index}{$direct."-dst-address-mask"}=$1; }
- }
- }
- @tmp=();
- @tmp=netdev_cmd($gate,$t,$gate->{proto},'/queue tree print terse without-paging where parent~"(download|upload)_root"',1);
- log_debug("Get root queues: ".Dumper(\@tmp));
- #print Dumper(\@tmp);
- # 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
- # 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
- foreach my $row (@tmp) {
- next if (!$row);
- $row = trim($row);
- next if ($row!~/^(\d)/);
- $row=~s/^(\d*)\s+//;
- next if (!$row);
- if ($row=~/queue=pcq_(down|up)load_(\d){1,3}/i) {
- if ($row=~/name=queue_(\d){1,3}_(\S*)_out\s+/i) {
- next if (!$1);
- next if (!$2);
- my $index = $1;
- my $int_name = $2;
- $get_queue_tree{$index}{$int_name}{up}=$row;
- if ($row=~/parent=(\S*)\s+\S/i) { $get_queue_tree{$index}{$int_name}{'up-parent'}=$1; }
- if ($row=~/packet-mark=(\S*)\s+\S/i) { $get_queue_tree{$index}{$int_name}{'up-mark'}=$1; }
- if ($row=~/queue=(\S*)\s+\S/i) { $get_queue_tree{$index}{$int_name}{'up-queue'}=$1; }
- }
- if ($row=~/name=queue_(\d){1,3}_(\S*)_in\s+/i) {
- next if (!$1);
- next if (!$2);
- my $index = $1;
- my $int_name = $2;
- $get_queue_tree{$index}{$int_name}{down}=$row;
- if ($row=~/parent=(\S*)\s+\S/i) { $get_queue_tree{$index}{$int_name}{'down-parent'}=$1; }
- if ($row=~/packet-mark=(\S*)\s+\S/i) { $get_queue_tree{$index}{$int_name}{'down-mark'}=$1; }
- if ($row=~/queue=(\S*)\s+\S/i) { $get_queue_tree{$index}{$int_name}{'down-queue'}=$1; }
- }
- }
- }
- @tmp=();
- @tmp=netdev_cmd($gate,$t,$gate->{proto},'/ip firewall mangle print terse without-paging where action=mark-packet and new-packet-mark~"(upload|download)_[0-9]{1,3}"',1);
- log_debug("Get firewall mangle rules for queues:".Dumper(\@tmp));
- # 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=""
- # 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=""
- foreach my $row (@tmp) {
- next if (!$row);
- $row = trim($row);
- next if ($row!~/^(\d){1,3}/);
- $row=~s/^\d{1,3}\s+//;
- next if (!$row);
- if ($row=~/new-packet-mark=upload_(\d){1,3}_(\S*)\s+/i) {
- next if (!$1);
- next if (!$2);
- my $index = $1;
- my $int_name = $2;
- $get_filter_mangle{$index}{$int_name}{up}=$row;
- if ($row=~/src-address-list=(\S*)\s+\S/i) { $get_filter_mangle{$index}{$int_name}{'up-list'}=$1; }
- if ($row=~/out-interface=(\S*)\s+\S/i) { $get_filter_mangle{$index}{$int_name}{'up-dev'}=$1; }
- if ($row=~/new-packet-mark=(\S*)\s+\S/i) { $get_filter_mangle{$index}{$int_name}{'up-mark'}=$1; }
- }
- if ($row=~/new-packet-mark=download_(\d){1,3}_(\S*)\s+/i) {
- next if (!$1);
- next if (!$2);
- my $index = $1;
- my $int_name = $2;
- $get_filter_mangle{$index}{$int_name}{down}=$row;
- if ($row=~/dst-address-list=(\S*)\s+\S/i) { $get_filter_mangle{$index}{$int_name}{'down-list'}=$1; }
- if ($row=~/new-packet-mark=(\S*)\s+\S/i) { $get_filter_mangle{$index}{$int_name}{'down-mark'}=$1; }
- if ($row=~/out-interface=(\S*)\s+\S/i) { $get_filter_mangle{$index}{$int_name}{'down-dev'}=$1; }
- }
- }
- log_debug("Queues type status:".Dumper(\%get_queue_type));
- log_debug("Queues tree status:".Dumper(\%get_queue_tree));
- log_debug("Firewall mangle status:".Dumper(\%get_filter_mangle));
- my %queue_type;
- my %queue_tree;
- my %filter_mangle;
- #generate new config
- foreach my $queue_name (keys %queues) {
- my $q_id=$queues{$queue_name}{id};
- my $q_up=$queues{$queue_name}{up}+1;
- my $q_down=$queues{$queue_name}{down}+1;
- #queue_types
- $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";
- $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";
- my $queue_ok=1;
- if (!$get_queue_type{$q_id}{up}) { $queue_ok=0; }
- if ($queue_ok and abs($q_up - $get_queue_type{$q_id}{'up-rate'})>10) { $queue_ok=0; }
- if ($queue_ok and $get_queue_type{$q_id}{'up-classifier'}!~/src-address/i) { $queue_ok=0; }
- if (!$queue_ok) {
- push(@cmd_list,':foreach i in [/queue type find where name~"pcq_upload_'.$q_id.'" ] do={/queue type remove $i};');
- push(@cmd_list,'/queue type add '.$queue_type{$q_id}{up});
- }
- $queue_ok=1;
- if (!$get_queue_type{$q_id}{down}) { $queue_ok=0; }
- if ($queue_ok and abs($q_up - $get_queue_type{$q_id}{'down-rate'})>10) { $queue_ok=0; }
- if ($queue_ok and $get_queue_type{$q_id}{'down-classifier'}!~/dst-address/i) { $queue_ok=0; }
- if (!$queue_ok) {
- push(@cmd_list,':foreach i in [/queue type find where name~"pcq_download_'.$q_id.'" ] do={/queue type remove $i};');
- push(@cmd_list,'/queue type add '.$queue_type{$q_id}{down});
- }
- #upload queue
- foreach my $int (@wan_int) {
- $queue_tree{$q_id}{$int}{up}="name=queue_".$q_id."_".$int."_out parent=upload_root_".$int." packet-mark=upload_".$q_id."_".$int." 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";
- $filter_mangle{$q_id}{$int}{up}="chain=forward action=mark-packet new-packet-mark=upload_".$q_id."_".$int." passthrough=yes src-address-list=queue_".$q_id." out-interface=".$int." log=no log-prefix=\"\"";
- $queue_ok=1;
- if (!$get_queue_tree{$q_id}{$int}{up}) { $queue_ok=0; }
- if ($queue_ok and ($get_queue_tree{$q_id}{$int}{'up-parent'} ne "upload_root_".$int)) { $queue_ok=0;}
- if ($queue_ok and ($get_queue_tree{$q_id}{$int}{'up-mark'} ne "upload_".$q_id."_".$int)) { $queue_ok=0; }
- if ($queue_ok and ($get_queue_tree{$q_id}{$int}{'up-queue'} ne "pcq_upload_".$q_id)) { $queue_ok=0; }
- if (!$queue_ok) {
- push(@cmd_list,':foreach i in [/queue tree find where name~"queue_'.$q_id."_".$int."_out".'" ] do={/queue tree remove $i};');
- push(@cmd_list,'/queue tree add '.$queue_tree{$q_id}{$int}{up});
- }
- $queue_ok=1;
- if (!$get_filter_mangle{$q_id}{$int}{up}) { $queue_ok=0; }
- if ($queue_ok and ($get_filter_mangle{$q_id}{$int}{'up-mark'} ne "upload_".$q_id."_".$int)) { $queue_ok=0; }
- if ($queue_ok and ($get_filter_mangle{$q_id}{$int}{'up-list'} ne "queue_".$q_id)) { $queue_ok=0; }
- if ($queue_ok and ($get_filter_mangle{$q_id}{$int}{'up-dev'} ne $int)) { $queue_ok=0; }
- if (!$queue_ok) {
- push(@cmd_list,':foreach i in [/ip firewall mangle find where action=mark-packet and new-packet-mark~"upload_'.$q_id."_".$int.'" ] do={/ip firewall mangle remove $i};');
- push(@cmd_list,'/ip firewall mangle add '.$filter_mangle{$q_id}{$int}{up});
- }
- }
- #download
- foreach my $int (@lan_int) {
- next if (!$int);
- $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";
- $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=\"\"";
- $queue_ok=1;
- if (!$get_queue_tree{$q_id}{$int}{down}) { $queue_ok=0; }
- if ($queue_ok and ($get_queue_tree{$q_id}{$int}{'down-parent'} ne "download_root_".$int)) { $queue_ok=0; }
- if ($queue_ok and ($get_queue_tree{$q_id}{$int}{'down-mark'} ne "download_".$q_id."_".$int)) { $queue_ok=0; }
- if ($queue_ok and ($get_queue_tree{$q_id}{$int}{'down-queue'} ne "pcq_download_".$q_id)) { $queue_ok=0; }
- if (!$queue_ok) {
- push(@cmd_list,':foreach i in [/queue tree find where name~"queue_'.$q_id."_".$int."_in".'" ] do={/queue tree remove $i};');
- push(@cmd_list,'/queue tree add '.$queue_tree{$q_id}{$int}{down});
- }
- $queue_ok=1;
- if (!$get_filter_mangle{$q_id}{$int}{down}) { $queue_ok=0; }
- if ($queue_ok and ($get_filter_mangle{$q_id}{$int}{'down-mark'} ne "download_".$q_id."_".$int)) { $queue_ok=0; }
- if ($queue_ok and ($get_filter_mangle{$q_id}{$int}{'down-list'} ne "queue_".$q_id)) { $queue_ok=0; }
- if ($queue_ok and ($get_filter_mangle{$q_id}{$int}{'down-dev'} ne $int)) { $queue_ok=0; }
- if (!$queue_ok) {
- 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};');
- push(@cmd_list,'/ip firewall mangle add '.$filter_mangle{$q_id}{$int}{down});
- }
- }
- #end shaper
- }
- }
- }#end access lists config
- if (scalar(@cmd_list)) {
- log_debug("Apply:");
- if ($debug) { foreach my $cmd (@cmd_list) { log_debug("$cmd"); } }
- eval {
- netdev_cmd($gate,$t,$gate->{proto},\@cmd_list,1);
- };
- if ($@) {
- $all_ok = 0;
- log_debug("Error programming gateway! Err: ".$@);
- }
- }
- db_log_verbose($dbh,"Sync user state at router $router_name [".$router_ip."] stopped.");
- $dbh->disconnect();
- $pm->finish;
- }
- $pm->wait_all_children;
- #clear changed
- if ($all_ok) {
- foreach my $row (@changes_found) {
- do_sql($dbh,"UPDATE User_auth SET changed=0 WHERE id=".$row->{id});
- }
- }
- if (IsMyPID($SPID)) { Remove_PID($SPID); };
- do_exit 0;
|