feat: zkVM packages

flake.nix

@@ -64,6 +64,12 @@
             pkgs = final;
             inherit config;
           };
+          devShells.nexus = import ./shells/nexus.nix { pkgs = final; };
+          devShells.jolt = import ./shells/jolt.nix { pkgs = final; };
+          devShells.zkm = import ./shells/zkm.nix { pkgs = final; };
+          devShells.zkwasm = import ./shells/zkwasm.nix { pkgs = final; };
+          devShells.sp1 = import ./shells/sp1.nix { pkgs = final; };
+          devShells.risc0 = import ./shells/risc0.nix { pkgs = final; };
         };
     };
 }

packages/all-packages.nix

@@ -18,8 +18,10 @@
       inherit (self'.legacyPackages)
         rustPlatformStable
         craneLib-stable
+        craneLib-nightly
         cardano-node
         cardano-cli
+        pkgs-with-rust-overlay
         ;
       python3Packages = pkgs.python3Packages;
 
@@ -127,6 +129,60 @@
       };
       polkadot = polkadot-generic { };
       polkadot-fast = polkadot-generic { enableFastRuntime = true; };
+
+      fetchGitHubFile =
+        {
+          owner,
+          repo,
+          rev,
+          file,
+          hash,
+        }:
+        pkgs.fetchurl {
+          url = "https://raw.githubusercontent.com/${owner}/${repo}/${rev}/${file}";
+          inherit hash;
+        };
+
+      fetchGitHubReleaseAsset =
+        {
+          owner,
+          repo,
+          tag,
+          asset,
+          hash,
+        }:
+        pkgs.fetchzip {
+          url = "https://github.com/${owner}/${repo}/releases/download/${tag}/${asset}";
+          inherit hash;
+          stripRoot = false;
+        };
+
+      installSourceAndCargo = rust-toolchain: rec {
+        installPhaseCommand = ''
+          mkdir -p "$out"/bin
+          # Install source code
+          cp -r /build/source/. "$out"
+          # Install cargo commands
+          ln -s "${rust-toolchain}"/bin/* "$out"/bin/
+          # Install binaries
+          for result in target/release/*
+          do
+            [ "''${result:15:5}" != 'crane' -a -f "$result" -a -x "$result" ] \
+              && ln -s "$out/$result" "$out"/bin/
+          done
+        '';
+      };
+
+      args-zkVM = {
+        inherit (pkgs-with-rust-overlay) rust-bin;
+        inherit craneLib-nightly;
+        inherit fetchGitHubFile;
+        inherit installSourceAndCargo;
+      };
+
+      args-zkVM-rust = {
+        inherit fetchGitHubReleaseAsset;
+      };
     in
     {
       legacyPackages.metacraft-labs =
@@ -186,6 +242,17 @@
           leap = callPackage ./leap/default.nix { };
           eos-vm = callPackage ./eos-vm/default.nix { };
           cdt = callPackage ./cdt/default.nix { };
+
+          zkwasm = callPackage ./zkwasm/default.nix args-zkVM;
+          jolt-guest-rust = callPackage ./jolt-guest-rust/default.nix args-zkVM-rust;
+          jolt = callPackage ./jolt/default.nix (args-zkVM // { inherit jolt-guest-rust; });
+          zkm-rust = callPackage ./zkm-rust/default.nix args-zkVM-rust;
+          zkm = callPackage ./zkm/default.nix (args-zkVM // { inherit zkm-rust; });
+          nexus = callPackage ./nexus/default.nix args-zkVM;
+          sp1-rust = callPackage ./sp1-rust/default.nix args-zkVM-rust;
+          sp1 = callPackage ./sp1/default.nix (args-zkVM // { inherit sp1-rust; });
+          risc0-rust = callPackage ./risc0-rust/default.nix args-zkVM-rust;
+          risc0 = callPackage ./risc0/default.nix (args-zkVM // { inherit risc0-rust; });
         }
         // lib.optionalAttrs hostPlatform.isx86 rec {
           inherit

packages/default.nix

@@ -33,7 +33,7 @@
       rust-nightly = pkgs-with-rust-overlay.rust-bin.nightly.latest.default.override {
         extensions = [ "rust-src" ];
         targets = [
-          "wasm32-wasi"
+          "wasm32-wasip1"
           "wasm32-unknown-unknown"
         ];
       };

packages/jolt-guest-rust/default.nix

@@ -0,0 +1,35 @@
+{
+  stdenv,
+  fetchGitHubReleaseAsset,
+  autoPatchelfHook,
+  zlib,
+  ...
+}:
+let
+  nightly-hash = "8af9d45d5e09a04832cc9b2e1df993fd1ce49d02";
+in
+stdenv.mkDerivation rec {
+  name = "jolt-guest-rust"; # Used when guest is compiled with std
+  version = "nightly-${nightly-hash}";
+
+  nativeBuildInputs = [
+    autoPatchelfHook
+    stdenv.cc.cc.lib
+    zlib
+  ];
+
+  installPhase = ''
+    runHook preInstall
+    mkdir -p $out
+    cp -r ./* $out/
+    runHook postInstall
+  '';
+
+  src = fetchGitHubReleaseAsset {
+    owner = "a16z";
+    repo = "rust";
+    tag = "${version}";
+    asset = "rust-toolchain-x86_64-unknown-linux-gnu.tar.gz";
+    hash = "sha256-aAhqLAvbeIh60R/E1c85KxWmYDH2SOpXhQChW3y3wgQ=";
+  };
+}

packages/jolt/default.nix

@@ -0,0 +1,71 @@
+{
+  rust-bin,
+  craneLib-nightly,
+  fetchFromGitHub,
+  fetchGitHubFile,
+  installSourceAndCargo,
+  pkg-config,
+  openssl,
+  jolt-guest-rust,
+  ...
+}:
+let
+  commonArgs = rec {
+    pname = "jolt";
+    version = "unstable-2024-12-18";
+
+    nativeBuildInputs = [
+      pkg-config
+      openssl
+    ];
+
+    preBuild = ''
+      sed -i 's/%2F/\//g' $CARGO_HOME/config.toml
+    '';
+
+    src = fetchFromGitHub {
+      owner = "a16z";
+      repo = "jolt";
+      rev = "f1e5ab3cb2f55eb49c17c249039294184d27fdba";
+      hash = "sha256-yr+JzZ3GIbwncEzfcrDmfxzph3YAI8dNLxcCXWT8520=";
+      fetchSubmodules = true;
+    };
+  };
+
+  rust-toolchain = rust-bin.fromRustupToolchainFile (fetchGitHubFile {
+    inherit (commonArgs.src) owner repo rev;
+    file = "rust-toolchain.toml";
+    hash = "sha256-Fyj+Bp/dt3epuTN9kXN+r7Z3gzXYCDrcVEPWTr1sQqk=";
+  });
+  craneLib = craneLib-nightly.overrideToolchain rust-toolchain;
+  cargoArtifacts = craneLib.buildDepsOnly commonArgs;
+in
+craneLib.buildPackage (
+  commonArgs
+  // (installSourceAndCargo rust-toolchain)
+  // rec {
+    inherit cargoArtifacts;
+
+    postPatch = ''
+      sed -i 's|package =.*git = "https://github.com/a16z/jolt"|path = "'$out'"|' src/main.rs
+      sed -i '44,46d' jolt-core/src/host/toolchain.rs
+    '';
+
+    # Different toolchain is used when guest has std features
+    # https://github.com/a16z/jolt/blob/fa45507aaddb1815bafd54332e4b14173a7f8699/jolt-core/src/host/mod.rs#L132-L134
+    postInstall = ''
+      rm $out/bin/cargo
+      cat <<EOF > $out/bin/cargo
+      #!/bin/sh
+      if [ -n "\''${RUSTUP_TOOLCHAIN+x}" ]
+      then
+          export PATH="${jolt-guest-rust}/rust/build/host/stage2/bin:\$PATH"
+      fi
+      ${rust-toolchain}/bin/cargo \$@
+      EOF
+      chmod +x $out/bin/cargo
+    '';
+
+    doCheck = false;
+  }
+)