sparse-v2/flake.nix

{
  description = "Sparse C2 framework";

  inputs = {
    nixpkgs.url = "github:nixos/nixpkgs/nixos-unstable";

    rust-overlay = {
      url = "github:oxalica/rust-overlay";
      inputs.nixpkgs.follows = "nixpkgs";
    };
    crane.url = "github:ipetkov/crane";
    flake-utils.url = "github:numtide/flake-utils";
    advisory-db = {
      url = "github:rustsec/advisory-db";
      flake = false;
    };

    libpcap-src = {
      url = "git+https://github.com/the-tcpdump-group/libpcap";
      flake = false;
    };
    libnl-src = {
      url = "git+https://github.com/thom311/libnl";
      flake = false;
    };
    winpcap-libs = {
      url = "https://www.winpcap.org/install/bin/WpdPack_4_1_2.zip";
      flake = false;
    };
    winpcap-installer = {
      url = "https://www.winpcap.org/install/bin/WinPcap_4_1_3.exe";
      flake = false;
    };
    freebsd-libs-packed = {
      url = "https://download.freebsd.org/releases/amd64/14.1-RELEASE/base.txz";
      flake = false;
    };
  };

  outputs = { self, nixpkgs, flake-utils, crane, rust-overlay, advisory-db
    , libpcap-src, libnl-src, winpcap-libs, winpcap-installer
    , freebsd-libs-packed }:
    flake-utils.lib.eachSystem [ "x86_64-linux" ] (system:
      let
        pkgs = import nixpkgs {
          inherit system;
          overlays = [ (import rust-overlay) ];
        };

        system-libs = import ./system-libs.nix {
          inherit pkgs libnl-src libpcap-src winpcap-installer
            freebsd-libs-packed;
        };
        inherit (system-libs)
          freebsd-toolchain winpcap-drivers freebsd-libs libnl libpcap-linux-gnu
          libpcap-linux-musl libpcap-freebsd;

        buildTools = with pkgs; rec {
          linux = [
            # Tools for local compilation
            lld
            zig
            clang
            glibc
            libclang
            mold

            # Tools for building the web UI
            cargo-leptos
            wasm-bindgen-cli
            dart-sass
            binaryen
            sqlx-cli
          ];
          freebsd = buildTools.linux
            ++ [ pkgsCross.x86_64-freebsd.buildPackages.clang ];
          windows = buildTools.linux ++ [
            pkgsCross.mingwW64.stdenv.cc
            pkgsCross.mingwW64.windows.pthreads
          ];

          all = buildTools.linux ++ buildTools.freebsd ++ buildTools.windows;
        };

        devShellTools = with pkgs; [
          # Language servers
          rust-analyzer
          zls

          # Cargo lint tools
          taplo
          cargo-deny
        ];

        craneLib = (crane.mkLib pkgs).overrideToolchain (p:
          p.rust-bin.nightly.latest.default.override {
            targets = [
              "x86_64-unknown-linux-gnu"
              "x86_64-unknown-linux-musl"
              "x86_64-pc-windows-gnu"
              "x86_64-unknown-freebsd"
              "wasm32-unknown-unknown"
            ];
          });

        buildEnvironment = {
          SPARSE_BUILD_WINPCAP_LIBS = "${winpcap-libs}/Lib";
          SPARSE_BUILD_WINPCAP_DRIVERS = "${winpcap-drivers}/npf.sys";
          SPARSE_BUILD_LIBPCAP_LINUX_GNU = libpcap-linux-gnu;
          SPARSE_BUILD_LIBPCAP_LINUX_MUSL = libpcap-linux-musl;
          SPARSE_BUILD_LIBPCAP_FREEBSD = libpcap-freebsd;
          SPARSE_BUILD_LIBNL = libnl;

          LIBCLANG_PATH = "${pkgs.libclang.lib}/lib";

          FREEBSD_LIBS = freebsd-libs;
          GLIBC_LIBS = pkgs.glibc;
          GLIBC_LIBS_STATIC = pkgs.glibc.static;
        };

        outputs = import ./packages.nix ({
          inherit pkgs buildTools buildEnvironment craneLib advisory-db
            winpcap-libs;
        } // system-libs);

        setup-zig-freebsd = pkgs.writeScriptBin "setup-zig-freebsd" ''
          set -eux

          cp ${outputs.packages.freebsd-zig-libc} $(${pkgs.git}/bin/git rev-parse --show-toplevel)/unix-loader/freebsd.txt
        '';

        setup-dev-environment = pkgs.writeScriptBin "setup-dev-environment" ''
          set -eux

          ${setup-zig-freebsd}/bin/setup-zig-freebsd
          cargo build -p sparse-windows-beacon --target=x86_64-pc-windows-gnu
          cargo build -p sparse-unix-beacon --target=x86_64-unknown-freebsd
          cargo build -p sparse-unix-beacon --target=x86_64-unknown-linux-musl
        '';
      in {
        packages = outputs.packages;
        checks = outputs.checks;

        devShells = {
          default = craneLib.devShell (buildEnvironment // {
            name = "sparse-default";

            packages = buildTools.linux ++ devShellTools
              ++ [ setup-zig-freebsd setup-dev-environment ];

            # Added to make development easier
            SPARSE_INSTALLER_LINUX =
              "../../target/x86_64-unknown-linux-musl/debug/sparse-unix-beacon";
            SPARSE_INSTALLER_FREEBSD =
              "../../target/x86_64-unknown-freebsd/debug/sparse-unix-beacon";
            SPARSE_BEACON_LINUX =
              "../../target/x86_64-unknown-linux-musl/debug/sparse-unix-beacon";
            SPARSE_BEACON_FREEBSD =
              "../../target/x86_64-unknown-freebsd/debug/sparse-unix-beacon";
            SPARSE_BEACON_WINDOWS =
              "../../target/x86_64-pc-windows-gnu/debug/sparse-windows-beacon.exe";

            shellHook = ''
              export DATABASE_URL="sqlite://$(${pkgs.git}/bin/git rev-parse --show-toplevel)/sparse-server/db.sqlite"
            '';
          });

          windows = craneLib.devShell (buildEnvironment // {
            name = "sparse-default";

            packages = buildTools.windows ++ devShellTools
              ++ [ setup-zig-freebsd setup-dev-environment ];

            # No point adding above environment variables, since web server can't
            # be built with windows tools available
          });

          freebsd = craneLib.devShell (buildEnvironment // {
            name = "sparse-default";

            packages = buildTools.freebsd ++ devShellTools
              ++ [ setup-zig-freebsd setup-dev-environment ];
          });
        };
      });
}