net/mwan3/files/lib/mwan3/common.sh

   1 #!/bin/sh
   2
   3 IP4="ip -4"
   4 IP6="ip -6"
   5 SCRIPTNAME="$(basename "$0")"
   6
   7 MWAN3_STATUS_DIR="/var/run/mwan3"
   8 MWAN3_STATUS_IPTABLES_LOG_DIR="${MWAN3_STATUS_DIR}/iptables_log"
   9 MWAN3TRACK_STATUS_DIR="/var/run/mwan3track"
  10
  11 MWAN3_INTERFACE_MAX=""
  12
  13 MMX_MASK=""
  14 MMX_DEFAULT=""
  15 MMX_BLACKHOLE=""
  16 MM_BLACKHOLE=""
  17
  18 MMX_UNREACHABLE=""
  19 MM_UNREACHABLE=""
  20 MAX_SLEEP=$(((1<<31)-1))
  21
  22 command -v ip6tables > /dev/null
  23 NO_IPV6=$?
  24
  25 IPS="ipset"
  26 IPT4="iptables -t mangle -w"
  27 IPT6="ip6tables -t mangle -w"
  28 IPT4R="iptables-restore -T mangle -w -n"
  29 IPT6R="ip6tables-restore -T mangle -w -n"
  30
  31 LOG()
  32 {
  33         local facility=$1; shift
  34         # in development, we want to show 'debug' level logs
  35         # when this release is out of beta, the comment in the line below
  36         # should be removed
  37         [ "$facility" = "debug" ] && return
  38         logger -t "${SCRIPTNAME}[$$]" -p $facility "$*"
  39 }
  40
  41 mwan3_get_true_iface()
  42 {
  43         local family V
  44         _true_iface=$2
  45         config_get family "$2" family ipv4
  46         if [ "$family" = "ipv4" ]; then
  47                 V=4
  48         elif [ "$family" = "ipv6" ]; then
  49                 V=6
  50         fi
  51         ubus call "network.interface.${2}_${V}" status &>/dev/null && _true_iface="${2}_${V}"
  52         export "$1=$_true_iface"
  53 }
  54
  55 mwan3_get_src_ip()
  56 {
  57         local family _src_ip interface true_iface device addr_cmd default_ip IP sed_str
  58         interface=$2
  59         mwan3_get_true_iface true_iface $interface
  60
  61         unset "$1"
  62         config_get family "$interface" family ipv4
  63         if [ "$family" = "ipv4" ]; then
  64                 addr_cmd='network_get_ipaddr'
  65                 default_ip="0.0.0.0"
  66                 sed_str='s/ *inet \([^ \/]*\).*/\1/;T; pq'
  67                 IP="$IP4"
  68         elif [ "$family" = "ipv6" ]; then
  69                 addr_cmd='network_get_ipaddr6'
  70                 default_ip="::"
  71                 sed_str='s/ *inet6 \([^ \/]*\).* scope.*/\1/;T; pq'
  72                 IP="$IP6"
  73         fi
  74
  75         $addr_cmd _src_ip "$true_iface"
  76         if [ -z "$_src_ip" ]; then
  77                 network_get_device device $true_iface
  78                 _src_ip=$($IP address ls dev $device 2>/dev/null | sed -ne "$sed_str")
  79                 if [ -n "$_src_ip" ]; then
  80                         LOG warn "no src $family address found from netifd for interface '$true_iface' dev '$device' guessing $_src_ip"
  81                 else
  82                         _src_ip="$default_ip"
  83                         LOG warn "no src $family address found for interface '$true_iface' dev '$device'"
  84                 fi
  85         fi
  86         export "$1=$_src_ip"
  87 }
  88
  89 mwan3_get_mwan3track_status()
  90 {
  91         local track_ips pid
  92         mwan3_list_track_ips()
  93         {
  94                 track_ips="$1 $track_ips"
  95         }
  96         config_list_foreach "$1" track_ip mwan3_list_track_ips
  97
  98         if [ -n "$track_ips" ]; then
  99                 pid="$(pgrep -f "mwan3track $1$")"
 100                 if [ -n "$pid" ]; then
 101                         if [ "$(cat /proc/"$(pgrep -P $pid)"/cmdline)" = "sleep${MAX_SLEEP}" ]; then
 102                                 tracking="paused"
 103                         else
 104                                 tracking="active"
 105                         fi
 106                 else
 107                         tracking="down"
 108                 fi
 109         else
 110                 tracking="not enabled"
 111         fi
 112         echo "$tracking"
 113 }
 114
 115 mwan3_init()
 116 {
 117         local bitcnt mmdefault source_routing
 118
 119         config_load mwan3
 120
 121         [ -d $MWAN3_STATUS_DIR ] || mkdir -p $MWAN3_STATUS_DIR/iface_state
 122         [ -d "$MWAN3_STATUS_IPTABLES_LOG_DIR" ] || mkdir -p "$MWAN3_STATUS_IPTABLES_LOG_DIR"
 123
 124         # mwan3's MARKing mask (at least 3 bits should be set)
 125         if [ -e "${MWAN3_STATUS_DIR}/mmx_mask" ]; then
 126                 MMX_MASK=$(cat "${MWAN3_STATUS_DIR}/mmx_mask")
 127                 MWAN3_INTERFACE_MAX=$(uci_get_state mwan3 globals iface_max)
 128         else
 129                 config_get MMX_MASK globals mmx_mask '0x3F00'
 130                 echo "$MMX_MASK"| tr 'A-F' 'a-f' > "${MWAN3_STATUS_DIR}/mmx_mask"
 131                 LOG debug "Using firewall mask ${MMX_MASK}"
 132
 133                 bitcnt=$(mwan3_count_one_bits MMX_MASK)
 134                 mmdefault=$(((1<<bitcnt)-1))
 135                 MWAN3_INTERFACE_MAX=$((mmdefault-3))
 136                 uci_toggle_state mwan3 globals iface_max "$MWAN3_INTERFACE_MAX"
 137                 LOG debug "Max interface count is ${MWAN3_INTERFACE_MAX}"
 138         fi
 139
 140         # remove "linkdown", expiry and source based routing modifiers from route lines
 141         config_get_bool source_routing globals source_routing 0
 142         [ $source_routing -eq 1 ] && unset source_routing
 143         MWAN3_ROUTE_LINE_EXP="s/linkdown //; s/expires [0-9]\+sec//; s/error [0-9]\+//; ${source_routing:+s/default\(.*\) from [^ ]*/default\1/;} p"
 144
 145         # mark mask constants
 146         bitcnt=$(mwan3_count_one_bits MMX_MASK)
 147         mmdefault=$(((1<<bitcnt)-1))
 148         MM_BLACKHOLE=$((mmdefault-2))
 149         MM_UNREACHABLE=$((mmdefault-1))
 150
 151         # MMX_DEFAULT should equal MMX_MASK
 152         MMX_DEFAULT=$(mwan3_id2mask mmdefault MMX_MASK)
 153         MMX_BLACKHOLE=$(mwan3_id2mask MM_BLACKHOLE MMX_MASK)
 154         MMX_UNREACHABLE=$(mwan3_id2mask MM_UNREACHABLE MMX_MASK)
 155 }
 156
 157 # maps the 1st parameter so it only uses the bits allowed by the bitmask (2nd parameter)
 158 # which means spreading the bits of the 1st parameter to only use the bits that are set to 1 in the 2nd parameter
 159 # 0 0 0 0 0 1 0 1 (0x05) 1st parameter
 160 # 1 0 1 0 1 0 1 0 (0xAA) 2nd parameter
 161 #     1   0   1          result
 162 mwan3_id2mask()
 163 {
 164         local bit_msk bit_val result
 165         bit_val=0
 166         result=0
 167         for bit_msk in $(seq 0 31); do
 168                 if [ $((($2>>bit_msk)&1)) = "1" ]; then
 169                         if [ $((($1>>bit_val)&1)) = "1" ]; then
 170                                 result=$((result|(1<<bit_msk)))
 171                         fi
 172                         bit_val=$((bit_val+1))
 173                 fi
 174         done
 175         printf "0x%x" $result
 176 }
 177
 178 # counts how many bits are set to 1
 179 # n&(n-1) clears the lowest bit set to 1
 180 mwan3_count_one_bits()
 181 {
 182         local count n
 183         count=0
 184         n=$(($1))
 185         while [ "$n" -gt "0" ]; do
 186                 n=$((n&(n-1)))
 187                 count=$((count+1))
 188         done
 189         echo $count
 190 }
 191
 192 get_uptime() {
 193         local uptime=$(cat /proc/uptime)
 194         echo "${uptime%%.*}"
 195 }
 196
 197 get_online_time() {
 198         local time_n time_u iface
 199         iface="$1"
 200         time_u="$(cat "$MWAN3TRACK_STATUS_DIR/${iface}/ONLINE" 2>/dev/null)"
 201         [ -z "${time_u}" ] || [ "${time_u}" = "0" ] || {
 202                 time_n="$(get_uptime)"
 203                 echo $((time_n-time_u))
 204         }
 205 }