From 4773ddffc3f886329ef3407bd731524b6288bd8e Mon Sep 17 00:00:00 2001
From: Simon Gardling <titaniumtown@proton.me>
Date: Thu, 17 Apr 2025 18:16:53 -0400
Subject: [PATCH] traffic shaping improvements

---
 services/wg.nix | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/services/wg.nix b/services/wg.nix
index ce95ed9..5a4122a 100644
--- a/services/wg.nix
+++ b/services/wg.nix
@@ -29,8 +29,8 @@
 
   systemd.services."traffic-shaping" =
     let
-      upload_pipe = 20;
-      high_prio = 18;
+      upload_pipe = 22;
+      high_prio = 20;
       low_prio = 2;
     in
     {
@@ -56,6 +56,14 @@
           # Direct marked packets to low-priority class
           ${pkgs.iproute2}/bin/tc filter add dev ${eth_interface} parent 1: protocol ip prio 1 handle 1 fw flowid 1:20
         '';
+
+      ExecStop = pkgs.writeShellScript "tc-stop" ''
+        ${pkgs.iproute2}/bin/tc filter del dev ${eth_interface} parent 1:
+        ${pkgs.iproute2}/bin/tc class del dev ${eth_interface} parent 1: classid 1:20
+        ${pkgs.iproute2}/bin/tc class del dev ${eth_interface} parent 1: classid 1:10
+        ${pkgs.iproute2}/bin/tc class del dev ${eth_interface} parent 1: classid 1:1
+        ${pkgs.iproute2}/bin/tc qdisc del dev ${eth_interface} root
+      '';
       };
     };
 }