You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

218 lines
9.3 KiB

  1. #!/bin/sh
  2. # Cero3 Shaper
  3. # A 3 bin tc_codel and ipv6 enabled shaping script for
  4. # ethernet gateways
  5. # This program is free software; you can redistribute it and/or modify
  6. # it under the terms of the GNU General Public License version 2 as
  7. # published by the Free Software Foundation.
  8. #
  9. # Copyright (C) 2012-4 Michael D. Taht, Toke Høiland-Jørgensen, Sebastian Moeller
  10. # Compared to the complexity that debloat had become
  11. # this cleanly shows a means of going from diffserv marking
  12. # to prioritization using the current tools (ip(6)tables
  13. # and tc. I note that the complexity of debloat exists for
  14. # a reason, and it is expected that script is run first
  15. # to setup various other parameters such as BQL and ethtool.
  16. # (And that the debloat script has setup the other interfaces)
  17. # You need to jiggle these parameters. Note limits are tuned towards a <10Mbit uplink <60Mbup down
  18. . /usr/lib/sqm/functions.sh
  19. ipt_setup() {
  20. ipt -t mangle -N QOS_MARK_${IFACE}
  21. ipt -t mangle -A QOS_MARK_${IFACE} -j MARK --set-mark 0x2${IPT_MASK_STRING}
  22. # You can go further with classification but...
  23. ipt -t mangle -A QOS_MARK_${IFACE} -m dscp --dscp-class CS1 -j MARK --set-mark 0x3${IPT_MASK_STRING}
  24. ipt -t mangle -A QOS_MARK_${IFACE} -m dscp --dscp-class CS6 -j MARK --set-mark 0x1${IPT_MASK_STRING}
  25. ipt -t mangle -A QOS_MARK_${IFACE} -m dscp --dscp-class EF -j MARK --set-mark 0x1${IPT_MASK_STRING}
  26. ipt -t mangle -A QOS_MARK_${IFACE} -m dscp --dscp-class AF42 -j MARK --set-mark 0x1${IPT_MASK_STRING}
  27. ipt -t mangle -A QOS_MARK_${IFACE} -m tos --tos Minimize-Delay -j MARK --set-mark 0x1${IPT_MASK_STRING}
  28. # and it might be a good idea to do it for udp tunnels too
  29. # Turn it on. Preserve classification if already performed
  30. #sm: is it correct to do this in $IFACE? Should ingress not be on $DEV? since HTB acts on $DEV?
  31. # SQUASH also does not work on $DEV (that is the IFB will still see the incoming ToS bits whether we squash or not)
  32. # SQUASH is still useful to protect internal machines...
  33. if [ "$SQUASH_DSCP" = "1" ]
  34. then
  35. sqm_logger "Squashing differentiated services code points (DSCP) from ingress."
  36. ipt -t mangle -I PREROUTING -i $IFACE -m dscp ! --dscp 0 -j DSCP --set-dscp-class be
  37. else
  38. sqm_logger "Keeping differentiated services code points (DSCP) from ingress."
  39. ipt -t mangle -A PREROUTING -i $IFACE -m mark --mark 0x00${IPT_MASK_STRING} -g QOS_MARK_${IFACE}
  40. fi
  41. ipt -t mangle -A POSTROUTING -o $IFACE -m mark --mark 0x00${IPT_MASK_STRING} -g QOS_MARK_${IFACE}
  42. # The Syn optimization was nice but fq_codel does it for us
  43. # ipt -t mangle -A PREROUTING -i s+ -p tcp -m tcp --tcp-flags SYN,RST,ACK SYN -j MARK --set-mark 0x01
  44. # Not sure if this will work. Encapsulation is a problem period
  45. ipt -t mangle -I PREROUTING -i vtun+ -p tcp -j MARK --set-mark 0x2${IPT_MASK_STRING} # tcp tunnels need ordering
  46. # Emanating from router, do a little more optimization
  47. # but don't bother with it too much.
  48. ipt -t mangle -A OUTPUT -p udp -m multiport --ports 123,53 -j DSCP --set-dscp-class AF42
  49. #Not clear if the second line is needed
  50. #ipt -t mangle -A OUTPUT -o $IFACE -g QOS_MARK_${IFACE}
  51. }
  52. # TC rules
  53. egress() {
  54. CEIL=${UPLINK}
  55. PRIO_RATE=`expr $CEIL / 3` # Ceiling for prioirty
  56. BE_RATE=`expr $CEIL / 6` # Min for best effort
  57. BK_RATE=`expr $CEIL / 6` # Min for background
  58. BE_CEIL=`expr $CEIL - 16` # A little slop at the top
  59. LQ="quantum `get_mtu $IFACE $CEIL`"
  60. $TC qdisc del dev $IFACE root 2> /dev/null
  61. $TC qdisc add dev $IFACE root handle 1: `get_stab_string` htb default 12
  62. $TC class add dev $IFACE parent 1: classid 1:1 htb $LQ rate ${CEIL}kbit ceil ${CEIL}kbit `get_htb_adsll_string`
  63. $TC class add dev $IFACE parent 1:1 classid 1:10 htb $LQ rate ${CEIL}kbit ceil ${CEIL}kbit prio 0 `get_htb_adsll_string`
  64. $TC class add dev $IFACE parent 1:1 classid 1:11 htb $LQ rate 128kbit ceil ${PRIO_RATE}kbit prio 1 `get_htb_adsll_string`
  65. $TC class add dev $IFACE parent 1:1 classid 1:12 htb $LQ rate ${BE_RATE}kbit ceil ${BE_CEIL}kbit prio 2 `get_htb_adsll_string`
  66. $TC class add dev $IFACE parent 1:1 classid 1:13 htb $LQ rate ${BK_RATE}kbit ceil ${BE_CEIL}kbit prio 3 `get_htb_adsll_string`
  67. $TC qdisc add dev $IFACE parent 1:11 handle 110: $QDISC `get_limit ${ELIMIT}` `get_target "${ETARGET}" ${UPLINK}` `get_ecn ${EECN}` `get_quantum 300` `get_flows ${PRIO_RATE}` ${EQDISC_OPTS}
  68. $TC qdisc add dev $IFACE parent 1:12 handle 120: $QDISC `get_limit ${ELIMIT}` `get_target "${ETARGET}" ${UPLINK}` `get_ecn ${EECN}` `get_quantum 300` `get_flows ${BE_RATE}` ${EQDISC_OPTS}
  69. $TC qdisc add dev $IFACE parent 1:13 handle 130: $QDISC `get_limit ${ELIMIT}` `get_target "${ETARGET}" ${UPLINK}` `get_ecn ${EECN}` `get_quantum 300` `get_flows ${BK_RATE}` ${EQDISC_OPTS}
  70. # Need a catchall rule
  71. $TC filter add dev $IFACE parent 1:0 protocol all prio 999 u32 \
  72. match ip protocol 0 0x00 flowid 1:12
  73. # FIXME should probably change the filter here to do pre-nat
  74. $TC filter add dev $IFACE parent 1:0 protocol ip prio 1 handle 1 fw classid 1:11
  75. $TC filter add dev $IFACE parent 1:0 protocol ip prio 2 handle 2 fw classid 1:12
  76. $TC filter add dev $IFACE parent 1:0 protocol ip prio 3 handle 3 fw classid 1:13
  77. # ipv6 support. Note that the handle indicates the fw mark bucket that is looked for
  78. $TC filter add dev $IFACE parent 1:0 protocol ipv6 prio 4 handle 1 fw classid 1:11
  79. $TC filter add dev $IFACE parent 1:0 protocol ipv6 prio 5 handle 2 fw classid 1:12
  80. $TC filter add dev $IFACE parent 1:0 protocol ipv6 prio 6 handle 3 fw classid 1:13
  81. # Arp traffic
  82. $TC filter add dev $IFACE parent 1:0 protocol arp prio 7 handle 1 fw classid 1:11
  83. # ICMP traffic - Don't impress your friends. Deoptimize to manage ping floods
  84. # better instead
  85. $TC filter add dev $IFACE parent 1:0 protocol ip prio 8 \
  86. u32 match ip protocol 1 0xff flowid 1:13
  87. $TC filter add dev $IFACE parent 1:0 protocol ipv6 prio 9 \
  88. u32 match ip protocol 1 0xff flowid 1:13
  89. #diffserv $IFACE
  90. }
  91. ingress() {
  92. CEIL=$DOWNLINK
  93. PRIO_RATE=`expr $CEIL / 3` # Ceiling for prioirty
  94. BE_RATE=`expr $CEIL / 6` # Min for best effort
  95. BK_RATE=`expr $CEIL / 6` # Min for background
  96. BE_CEIL=`expr $CEIL - 16` # A little slop at the top
  97. LQ="quantum `get_mtu $IFACE $CEIL`"
  98. $TC qdisc del dev $IFACE handle ffff: ingress 2> /dev/null
  99. $TC qdisc add dev $IFACE handle ffff: ingress
  100. $TC qdisc del dev $DEV root 2> /dev/null
  101. if [ "$SQUASH_INGRESS" = "1" ]
  102. then
  103. sqm_logger "Do not perform DSCP based filtering on ingress. (1-tier classification)"
  104. # Revert to no dscp based filtering
  105. $TC qdisc del dev $DEV root 2>/dev/null
  106. $TC qdisc add dev $DEV root handle 1: `get_stab_string` htb default 10
  107. $TC class add dev $DEV parent 1: classid 1:1 htb $LQ rate ${DOWNLINK}kbit ceil ${DOWNLINK}kbit `get_htb_adsll_string`
  108. $TC class add dev $DEV parent 1:1 classid 1:10 htb $LQ rate ${DOWNLINK}kbit ceil ${DOWNLINK}kbit prio 0 `get_htb_adsll_string`
  109. $TC qdisc add dev $DEV parent 1:10 handle 110: $QDISC `get_limit ${ILIMIT}` `get_target "${ITARGET}" ${DOWNLINK}` `get_ecn ${IECN}` `get_flows ${DOWNLINK}` ${IQDISC_OPTS}
  110. else
  111. sqm_logger "Perform DSCP based filtering on ingress. (3-tier classification)"
  112. $TC qdisc add dev $DEV root handle 1: `get_stab_string` htb default 12
  113. $TC class add dev $DEV parent 1: classid 1:1 htb $LQ rate ${CEIL}kbit ceil ${CEIL}kbit `get_htb_adsll_string`
  114. $TC class add dev $DEV parent 1:1 classid 1:10 htb $LQ rate ${CEIL}kbit ceil ${CEIL}kbit prio 0 `get_htb_adsll_string`
  115. $TC class add dev $DEV parent 1:1 classid 1:11 htb $LQ rate 32kbit ceil ${PRIO_RATE}kbit prio 1 `get_htb_adsll_string`
  116. $TC class add dev $DEV parent 1:1 classid 1:12 htb $LQ rate ${BE_RATE}kbit ceil ${BE_CEIL}kbit prio 2 `get_htb_adsll_string`
  117. $TC class add dev $DEV parent 1:1 classid 1:13 htb $LQ rate ${BK_RATE}kbit ceil ${BE_CEIL}kbit prio 3 `get_htb_adsll_string`
  118. # I'd prefer to use a pre-nat filter but that causes permutation...
  119. $TC qdisc add dev $DEV parent 1:11 handle 110: $QDISC `get_limit ${ILIMIT}` `get_target "${ITARGET}" ${DOWNLINK}` `get_ecn ${IECN}` `get_quantum 500` `get_flows ${PRIO_RATE}` ${IQDISC_OPTS}
  120. $TC qdisc add dev $DEV parent 1:12 handle 120: $QDISC `get_limit ${ILIMIT}` `get_target "${ITARGET}" ${DOWNLINK}` `get_ecn ${IECN}` `get_quantum 1500` `get_flows ${BE_RATE}` ${IQDISC_OPTS}
  121. $TC qdisc add dev $DEV parent 1:13 handle 130: $QDISC `get_limit ${ILIMIT}` `get_target "${ITARGET}" ${DOWNLINK}` `get_ecn ${IECN}` `get_quantum 300` `get_flows ${BK_RATE}` ${IQDISC_OPTS}
  122. diffserv $DEV
  123. fi
  124. ifconfig $DEV up
  125. # redirect all IP packets arriving in $IFACE to ifb0
  126. $TC filter add dev $IFACE parent ffff: protocol all prio 10 u32 \
  127. match u32 0 0 flowid 1:1 action mirred egress redirect dev $DEV
  128. }
  129. do_modules
  130. ipt_setup
  131. if [ "$UPLINK" -ne 0 ];
  132. then
  133. egress
  134. sqm_logger "egress shaping activated"
  135. else
  136. sqm_logger "egress shaping deactivated"
  137. tc qdisc del dev $IFACE root 2> /dev/null
  138. fi
  139. if [ "$DOWNLINK" -ne 0 ];
  140. then
  141. ingress
  142. sqm_logger "ingress shaping activated"
  143. else
  144. sqm_logger "ingress shaping deactivated"
  145. tc qdisc del dev $DEV root 2> /dev/null
  146. tc qdisc del dev $IFACE ingress 2> /dev/null
  147. fi
  148. # References:
  149. # This alternate shaper attempts to go for 1/u performance in a clever way
  150. # http://git.coverfire.com/?p=linux-qos-scripts.git;a=blob;f=src-3tos.sh;hb=HEAD
  151. # Comments
  152. # This does the right thing with ipv6 traffic.
  153. # It also tries to leverage diffserv to some sane extent. In particular,
  154. # the 'priority' queue is limited to 33% of the total, so EF, and IMM traffic
  155. # cannot starve other types. The rfc suggested 30%. 30% is probably
  156. # a lot in today's world.
  157. # Flaws
  158. # Many!