nixos/networkd: add HierarchyTokenBucketClass options

NixOS · Mar 16, 2023 · 88d99a3 · 88d99a3
1 parent 24df07c
commit 88d99a3
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/nixos/modules/system/boot/networkd.nix b/nixos/modules/system/boot/networkd.nix
@@ -1371,6 +1371,21 @@ let
         ])
         (assertInt "RateToQuantum")
       ];
+
+      sectionHierarchyTokenBucketClass = checkUnitConfig "HierarchyTokenBucketClass" [
+        (assertOnlyFields [
+          "Parent"
+          "ClassId"
+          "Priority"
+          "QuantumBytes"
+          "MTUBytes"
+          "OverheadBytes"
+          "Rate"
+          "CeilRate"
+          "BufferBytes"
+          "CeilBufferBytes"
+        ])
+      ];
     };
   };
 
@@ -2270,6 +2285,17 @@ let
       '';
     };
 
+    hierarchyTokenBucketClassConfig = mkOption {
+      default = {};
+      example = { Parent = "root"; Rate = "10M"; };
+      type = types.addCheck (types.attrsOf unitOption) check.network.sectionHierarchyTokenBucketClass;
+      description = lib.mdDoc ''
+        Each attribute in this set specifies an option in the
+        `[HierarchyTokenBucketClass]` section of the unit.  See
+        {manpage}`systemd.network(5)` for details.
+      '';
+    };
+
     name = mkOption {
       type = types.nullOr types.str;
       default = null;
@@ -2785,6 +2811,10 @@ let
           [HierarchyTokenBucket]
           ${attrsToSection def.hierarchyTokenBucketConfig}
         ''
+        + optionalString (def.hierarchyTokenBucketClassConfig != { }) ''
+          [HierarchyTokenBucketClass]
+          ${attrsToSection def.hierarchyTokenBucketClassConfig}
+        ''
         + def.extraConfig;
     };