package/qos-scripts/files/usr/lib/qos/generate.sh

   1 #!/bin/sh
   2 [ -e /etc/functions.sh ] && . /etc/functions.sh || . ./functions.sh
   3 [ -x /sbin/modprobe ] && insmod="modprobe" || insmod="insmod"
   4
   5 add_insmod() {
   6         eval "export isset=\${insmod_$1}"
   7         case "$isset" in
   8                 1) ;;
   9                 *) append INSMOD "$insmod $* >&- 2>&-" "$N"; export insmod_$1=1;;
  10         esac
  11 }
  12
  13 [ -e /etc/config/network ] && {
  14         # only try to parse network config on openwrt
  15
  16         find_ifname() {(
  17                 include /lib/network
  18                 scan_interfaces
  19                 config_get "$1" ifname
  20         )}
  21 } || {
  22         find_ifname() {
  23                 echo "Interface not found."
  24                 exit 1
  25         }
  26 }
  27
  28 parse_matching_rule() {
  29         local var="$1"
  30         local section="$2"
  31         local options="$3"
  32         local prefix="$4"
  33         local suffix="$5"
  34         local proto="$6"
  35         local mport=""
  36         local ports=""
  37
  38         append "$var" "$prefix" "$N"
  39         for option in $options; do
  40                 case "$option" in
  41                         proto) config_get value "$section" proto; proto="${proto:-$value}";;
  42                 esac
  43         done
  44         config_get type "$section" TYPE
  45         case "$type" in
  46                 classify) unset pkt; append "$var" "-m mark --mark 0";;
  47                 default) pkt=1; append "$var" "-m mark --mark 0";;
  48                 reclassify) pkt=1;;
  49         esac
  50         append "$var" "${proto:+-p $proto}"
  51         for option in $options; do
  52                 config_get value "$section" "$option"
  53
  54                 case "$pkt:$option" in
  55                         *:srchost)
  56                                 append "$var" "-s $value"
  57                         ;;
  58                         *:dsthost)
  59                                 append "$var" "-d $value"
  60                         ;;
  61                         *:ipp2p)
  62                                 add_insmod ipt_ipp2p
  63                                 append "$var" "-m ipp2p"
  64                                 case "$value" in
  65                                         all) append "$var" "--edk --dc --kazaa --gnu --bit";;
  66                                         *) append "$var" "--$value";;
  67                                 esac
  68                         ;;
  69                         *:layer7)
  70                                 add_insmod ipt_layer7
  71                                 append "$var" "-m layer7 --l7proto $value${pkt:+ --l7pkt}"
  72                         ;;
  73                         *:ports|*:srcports|*:dstports)
  74                                 value="$(echo "$value" | sed -e 's,-,:,g')"
  75                                 lproto=${lproto:-tcp}
  76                                 case "$proto" in
  77                                         ""|tcp|udp) append "$var" "-m ${proto:-tcp -p tcp} -m multiport";;
  78                                         *) unset "$var"; return 0;;
  79                                 esac
  80                                 case "$option" in
  81                                         ports)
  82                                                 config_set "$section" srcports ""
  83                                                 config_set "$section" dstports ""
  84                                                 config_set "$section" portrange ""
  85                                                 append "$var" "--ports $value"
  86                                         ;;
  87                                         srcports)
  88                                                 config_set "$section" ports ""
  89                                                 config_set "$section" dstports ""
  90                                                 config_set "$section" portrange ""
  91                                                 append "$var" "--sports $value"
  92                                         ;;
  93                                         dstports)
  94                                                 config_set "$section" ports ""
  95                                                 config_set "$section" srcports ""
  96                                                 config_set "$section" portrange ""
  97                                                 append "$var" "--dports $value"
  98                                         ;;
  99                                 esac
 100                                 ports=1
 101                         ;;
 102                         *:portrange)
 103                                 config_set "$section" ports ""
 104                                 config_set "$section" srcports ""
 105                                 config_set "$section" dstports ""
 106                                 value="$(echo "$value" | sed -e 's,-,:,g')"
 107                                 case "$proto" in
 108                                         ""|tcp|udp) append "$var" "-m ${proto:-tcp -p tcp} --sport $value --dport $value";;
 109                                         *) unset "$var"; return 0;;
 110                                 esac
 111                                 ports=1
 112                         ;;
 113                         1:pktsize)
 114                                 value="$(echo "$value" | sed -e 's,-,:,g')"
 115                                 add_insmod ipt_length
 116                                 append "$var" "-m length --length $value"
 117                         ;;
 118                         1:limit)
 119                                 add_insmod ipt_limit
 120                                 append "$var" "-m limit --limit $value"
 121                         ;;
 122                         1:tcpflags)
 123                                 case "$proto" in
 124                                         tcp) append "$var" "-m tcp --tcp-flags ALL $value";;
 125                                         *) unset $var; return 0;;
 126                                 esac
 127                         ;;
 128                         1:mark)
 129                                 config_get class "${value##!}" classnr
 130                                 [ -z "$class" ] && continue;
 131                                 case "$value" in
 132                                         !*) append "$var" "-m mark ! --mark $class";;
 133                                         *) append "$var" "-m mark --mark $class";;
 134                                 esac
 135                 esac
 136         done
 137         append "$var" "$suffix"
 138         case "$ports:$proto" in
 139                 1:)     parse_matching_rule "$var" "$section" "$options" "$prefix" "$suffix" "udp";;
 140         esac
 141 }
 142
 143 config_cb() {
 144         option_cb() {
 145                 return 0
 146         }
 147
 148         # Section start
 149         case "$1" in
 150                 interface)
 151                         config_set "$1" "classgroup" "Default"
 152                         config_set "$1" "upload" "128"
 153                 ;;
 154                 classify|default|reclassify)
 155                         option_cb() {
 156                                 append options "$1"
 157                         }
 158                 ;;
 159         esac
 160
 161     # Section end
 162         config_get TYPE "$CONFIG_SECTION" TYPE
 163         case "$TYPE" in
 164                 interface)
 165                         config_get enabled "$CONFIG_SECTION" enabled
 166                         config_get download "$CONFIG_SECTION" download
 167                         config_get classgroup "$CONFIG_SECTION" classgroup
 168                         config_set "$CONFIG_SECTION" imqdev "$C"
 169                         [ -z "$enabled" -o "$(($enabled))" -eq 0 ] || {
 170                                 C=$(($C+1))
 171                                 INTERFACES="$INTERFACES $CONFIG_SECTION"
 172                                 config_set "$classgroup" enabled 1
 173                         }
 174                         config_get device "$CONFIG_SECTION" device
 175                         [ -z "$device" ] && device="$(find_ifname ${CONFIG_SECTION})"
 176                         config_set "$CONFIG_SECTION" device "${device:-eth0}"
 177                 ;;
 178                 classgroup) append CG "$CONFIG_SECTION";;
 179                 classify|default|reclassify)
 180                         case "$TYPE" in
 181                                 classify) var="ctrules";;
 182                                 *) var="rules";;
 183                         esac
 184                         config_get target "$CONFIG_SECTION" target
 185                         config_set "$CONFIG_SECTION" options "$options"
 186                         append "$var" "$CONFIG_SECTION"
 187                         unset options
 188                 ;;
 189         esac
 190 }
 191
 192
 193 enum_classes() {
 194         local c="0"
 195         config_get classes "$1" classes
 196         config_get default "$1" default
 197         for class in $classes; do
 198                 c="$(($c + 1))"
 199                 config_set "${class}" classnr $c
 200                 case "$class" in
 201                         $default) class_default=$c;;
 202                 esac
 203         done
 204         class_default="${class_default:-$c}"
 205 }
 206
 207 cls_var() {
 208         local varname="$1"
 209         local class="$2"
 210         local name="$3"
 211         local type="$4"
 212         local default="$5"
 213         local tmp tmp1 tmp2
 214         config_get tmp1 "$class" "$name"
 215         config_get tmp2 "${class}_${type}" "$name"
 216         tmp="${tmp2:-$tmp1}"
 217         tmp="${tmp:-$tmp2}"
 218         export ${varname}="${tmp:-$default}"
 219 }
 220
 221 tcrules() {
 222         dir=/usr/lib/qos
 223         [ -e $dir/tcrules.awk ] || dir=.
 224         echo "$cstr" | awk \
 225                 -v device="$dev" \
 226                 -v linespeed="$rate" \
 227                 -f $dir/tcrules.awk
 228 }
 229
 230 start_interface() {
 231         local iface="$1"
 232         local num_imq="$2"
 233         config_get device "$iface" device
 234         config_get enabled "$iface" enabled
 235         [ -z "$device" -o -z "$enabled" ] && exit
 236         config_get upload "$iface" upload
 237         config_get halfduplex "$iface" halfduplex
 238         config_get download "$iface" download
 239         config_get classgroup "$iface" classgroup
 240
 241         download="${download:-${halfduplex:+$upload}}"
 242         enum_classes "$classgroup"
 243         for dir in up${halfduplex} ${download:+down}; do
 244                 case "$dir" in
 245                         up)
 246                                 upload=$(($upload * 98 / 100 - (32 * 128 / $upload)))
 247                                 dev="$device"
 248                                 rate="$upload"
 249                                 dl_mode=""
 250                                 prefix="cls"
 251                         ;;
 252                         down)
 253                                 add_insmod imq numdevs="$num_imq"
 254                                 config_get imqdev "$iface" imqdev
 255                                 download=$(($download * 98 / 100 - (100 * 1024 / $download)))
 256                                 dev="imq$imqdev"
 257                                 rate="$download"
 258                                 dl_mode=1
 259                                 prefix="d_cls"
 260                         ;;
 261                         *) continue;;
 262                 esac
 263                 cstr=
 264                 for class in $classes; do
 265                         cls_var pktsize "$class" packetsize $dir 1500
 266                         cls_var pktdelay "$class" packetdelay $dir 0
 267                         cls_var maxrate "$class" limitrate $dir 100
 268                         cls_var prio "$class" priority $dir 1
 269                         cls_var avgrate "$class" avgrate $dir 0
 270                         config_get classnr "$class" classnr
 271                         append cstr "$classnr:$prio:$avgrate:$pktsize:$pktdelay:$maxrate" "$N"
 272                 done
 273                 append ${prefix}q "$(tcrules)" "$N"
 274                 export dev_${dir}="ifconfig $dev up txqueuelen 5 >&- 2>&-
 275 tc qdisc del dev $dev root >&- 2>&-
 276 tc qdisc add dev $dev root handle 1: hfsc default ${class_default}0
 277 tc class add dev $dev parent 1: classid 1:1 hfsc sc rate ${rate}kbit ul rate ${rate}kbit"
 278         done
 279         add_insmod cls_fw
 280         add_insmod sch_hfsc
 281         add_insmod sch_sfq
 282         add_insmod sch_red
 283
 284         cat <<EOF
 285 ${INSMOD:+$INSMOD$N}${dev_up:+$dev_up
 286 $clsq
 287 }${imqdev:+$dev_down
 288 $d_clsq
 289 $d_clsl
 290 $d_clsf
 291 }
 292 EOF
 293         unset INSMOD clsq clsf clsl d_clsq d_clsl d_clsf dev_up dev_down
 294 }
 295
 296 start_interfaces() {
 297         local C="$1"
 298         for iface in $INTERFACES; do
 299                 start_interface "$iface" "$C"
 300         done
 301 }
 302
 303 add_rules() {
 304         local var="$1"
 305         local rules="$2"
 306         local prefix="$3"
 307
 308         for rule in $rules; do
 309                 unset iptrule
 310                 config_get target "$rule" target
 311                 config_get target "$target" classnr
 312                 config_get options "$rule" options
 313                 parse_matching_rule iptrule "$rule" "$options" "$prefix" "-j MARK --set-mark $target"
 314                 append "$var" "$iptrule" "$N"
 315         done
 316 }
 317
 318 start_cg() {
 319         local cg="$1"
 320         local iptrules
 321         local pktrules
 322         local sizerules
 323         local download
 324         enum_classes "$cg"
 325         add_rules iptrules "$ctrules" "iptables -t mangle -A ${cg}_ct"
 326         config_get classes "$cg" classes
 327         for class in $classes; do
 328                 config_get mark "$class" classnr
 329                 config_get maxsize "$class" maxsize
 330                 [ -z "$maxsize" -o -z "$mark" ] || {
 331                         add_insmod ipt_length
 332                         append pktrules "iptables -t mangle -A ${cg} -m mark --mark $mark -m length --length $maxsize: -j MARK --set-mark 0" "$N"
 333                 }
 334         done
 335         add_rules pktrules "$rules" "iptables -t mangle -A ${cg}"
 336         for iface in $INTERFACES; do
 337                 config_get classgroup "$iface" classgroup
 338                 config_get device "$iface" device
 339                 config_get imqdev "$iface" imqdev
 340                 config_get dl "$iface" download
 341                 config_get halfduplex "$iface" halfduplex
 342                 add_insmod ipt_IMQ
 343                 append up "iptables -t mangle -A OUTPUT -o $device -j ${cg}" "$N"
 344                 append up "iptables -t mangle -A FORWARD -o $device -j ${cg}" "$N"
 345                 [ -z "$dl" ] || {
 346                         [ -z "$halfduplex" ] || {
 347                                 append down "iptables -t mangle -A POSTROUTING -o $device -j IMQ --todev $imqdev" "$N"
 348                         }
 349                         append down "iptables -t mangle -A PREROUTING -i $device -j ${cg}" "$N"
 350                         append down "iptables -t mangle -A PREROUTING -i $device -j IMQ --todev $imqdev" "$N"
 351                 }
 352         done
 353         cat <<EOF
 354 $INSMOD
 355 iptables -t mangle -N ${cg} >&- 2>&-
 356 iptables -t mangle -N ${cg}_ct >&- 2>&-
 357 ${iptrules:+${iptrules}${N}iptables -t mangle -A ${cg}_ct -j CONNMARK --save-mark}
 358 iptables -t mangle -A ${cg} -j CONNMARK --restore-mark
 359 iptables -t mangle -A ${cg} -m mark --mark 0 -j ${cg}_ct
 360 $pktrules
 361 $up$N${down:+${down}$N}
 362 EOF
 363         unset INSMOD
 364 }
 365
 366 start_firewall() {
 367         add_insmod ipt_multiport
 368         add_insmod ipt_CONNMARK
 369         cat <<EOF
 370 iptables -t mangle -F
 371 iptables -t mangle -X
 372 EOF
 373         for group in $CG; do
 374                 start_cg $group
 375         done
 376 }
 377
 378 C="0"
 379 INTERFACES=""
 380 [ -e ./qos.conf ] && {
 381         . ./qos.conf
 382         config_cb
 383 } || config_load qos
 384
 385 C="0"
 386 for iface in $INTERFACES; do
 387         export C="$(($C + 1))"
 388 done
 389
 390 case "$1" in
 391         all)
 392                 start_interfaces "$C"
 393                 start_firewall
 394         ;;
 395         interface)
 396                 start_interface "$2" "$C"
 397         ;;
 398         interfaces)
 399                 start_interfaces
 400         ;;
 401         firewall)
 402                 start_firewall
 403         ;;
 404 esac