aboutsummaryrefslogtreecommitdiff
path: root/zkvms/jolt/default.nix
blob: 2c6399cfa5c25893aafd62b1a0a98b81a97c45a3 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
{ zkVM-helpers,
  stdenv,
  lib,
  just,
  metacraft-labs,
  pkg-config,
  openssl,
  craneLib-default,
}:
let
  commonArgs = {
    pname = "jolt";
    version = "infdev";

    src = with lib.fileset; toSource {
      root = ../..;
      fileset = intersection (gitTracked ../..) (unions [
          ./.
          ../../guests
          ../../guests_macro
          ../../zkvms_host_io
          ../../Vertices-010.in
      ]);
    };

    cargoLock = ./Cargo.lock;

    nativeBuildInputs = [
      metacraft-labs.jolt
      openssl
      pkg-config
    ];
  };

  craneLib = craneLib-default.overrideToolchain metacraft-labs.jolt;
  cargoArtifacts = craneLib.buildDepsOnly (zkVM-helpers.fixDeps (commonArgs // {
    postConfigure = ''
      sed -i 's/"guest",//' zkvms/jolt/Cargo.toml
      sed -i '/dependencies.guest/,+1d' zkvms/jolt/host/Cargo.toml
    '';
  }));
in
  craneLib.buildPackage (zkVM-helpers.withCustomPhases (commonArgs
    // {
      inherit cargoArtifacts;

      postPatch = ''
        ln -s ../../../../guests/graph_coloring ./zkvms/jolt/guest/src/zkp
        sed -i '/guest\/guests/d' ./zkvms/jolt/Cargo.toml
      '';

      hostBin = "host-jolt";
      guestTarget = "riscv32im-jolt-zkvm-elf";
      extraGuestArgs = "--features guest";

      preBuildGuest = ''
        export RUSTUP_TOOLCHAIN="x"
        export RUSTFLAGS="-C link-arg=-T${./guest/guest.ld} -C passes=lower-atomic -C panic=abort -C strip=symbols -C opt-level=z"
      '';

      preBuild = ''
        unset RUSTUP_TOOLCHAIN
        export RUSTFLAGS="-Z macro-backtrace"
      '';

      preRun = ''
        export ELF_PATH="$out/bin/guest"
        export PATH="$PATH:${metacraft-labs.jolt}/bin"
      '';

      doCheck = false;
    }))