diff options
| -rw-r--r-- | flake.nix | 1 | ||||
| -rw-r--r-- | lib/default.nix | 30 | ||||
| -rw-r--r-- | lib/ipv4.nix | 38 | ||||
| -rw-r--r-- | nixos/configurations.nix | 5 |
4 files changed, 74 insertions, 0 deletions
@@ -35,6 +35,7 @@ }; nixosModules = import ./nixos self; nixosConfigurations = import ./nixos/configurations.nix self; + lib = import ./lib nixpkgs.lib; } // eachDefaultSystem (system: { packages = filterPackages system (flattenTree ( diff --git a/lib/default.nix b/lib/default.nix new file mode 100644 index 0000000..036af97 --- /dev/null +++ b/lib/default.nix @@ -0,0 +1,30 @@ +nixpkgslib: let + lib = nixpkgslib // nlib; + + nlib = with builtins; + with lib; { + # Generic power of number + pow = base: e: + if e == 0 + then 1 + else foldl (a: b: a * base) 1 (genList id (e - 1)); + # Power of 2 + pow2 = pow 2; + + # Convert integer to list of bits + int2bits = len: e: reverseList (genList (x: bitAnd e (pow2 (x + 1)) > 0) len); + # Reverse operation for int2bits + bits2int = l: let + len = length l; + zf = a: b: + if a + then pow2 b + else 0; + in + foldl add 0 (zipListsWith zf l (genList (i: len - i) len)); + + # IPv4 utilities + ipv4 = import ./ipv4.nix lib; + }; +in + nlib diff --git a/lib/ipv4.nix b/lib/ipv4.nix new file mode 100644 index 0000000..c843cb6 --- /dev/null +++ b/lib/ipv4.nix @@ -0,0 +1,38 @@ +lib: +with builtins; +with lib; rec { + # Converts string representation of IPv4 address to 32 bits + ip2bits = ip: let + perBits = map (x: int2bits 8 (toInt x)) (splitString "." ip); + in + flatten perBits; + # Converts 32 bits to IPv4 + bits2ip = bits: let + bts = i: toString (bits2int (sublist (i * 8) 8 bits)); + in "${bts 0}.${bts 1}.${bts 2}.${bts 3}"; + + # Convert IPv4 to number + ip2int = ip: bits2int (ip2bits ip); + # Convert number to IPv4 + int2ip = ip: bits2ip (int2bits 32 ip); + + # Generate bits for netmas of gitven prefix length + netmaskBits = prefixLength: genList (x: x < prefixLength) 32; + # Convert IP network prefix length to network mask + prefix2netmask = prefixLength: bits2ip (netmaskBits prefixLength); + # Mask IP by network mask specified by given network prefix length + prefix2ip = ip: prefixLength: let + a = netmaskBits prefixLength; + b = ip2bits ip; + in + bits2ip (zipListsWith (a: b: a && b) a b); + # Last address in the range + prefix2broadcast = ip: prefixLength: let + a = netmaskBits prefixLength; + b = ip2bits ip; + in + bits2ip (zipListsWith (a: b: !a || b) a b); + + # Offset address in network + ipAdd = ip: prefixLength: off: int2ip ((ip2int (prefix2ip ip prefixLength)) + off); +} diff --git a/nixos/configurations.nix b/nixos/configurations.nix index 24256b3..943b59a 100644 --- a/nixos/configurations.nix +++ b/nixos/configurations.nix @@ -18,6 +18,9 @@ with nixpkgs.lib; let } ] ++ (optional (hasAttr "machine-${hostname}" self.nixosModules) self.nixosModules."machine-${hostname}"); + specialArgs = { + lib = nixpkgs.lib.extend (prev: final: import ../lib prev); + }; genericSystem = { system ? "x86_64-linux", @@ -26,6 +29,7 @@ with nixpkgs.lib; let ${hostname} = nixpkgs.lib.nixosSystem { system = system; modules = (modules hostname) ++ extra_modules; + specialArgs = specialArgs; }; }; amd64System = genericSystem {}; @@ -96,6 +100,7 @@ with nixpkgs.lib; let nixpkgs = nixpkgs; board = board; modules = modules hostname; + override.specialArgs = specialArgs; }; }; turrisMoxSystem = turrisSystem "mox"; |