aboutsummaryrefslogtreecommitdiff
path: root/zkvms/zkm/default.nix
blob: da7224b404234378764350fc5004e209c66b3489 (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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
{ zkVM-helpers,
  stdenv,
  lib,
  just,
  metacraft-labs,
  rust-bin,
  protobuf,
  pkg-config,
  openssl,
  buildGoModule,
  craneLib-default,
}:
let
  zkm_libsnark = buildGoModule {
    pname = "zkm_libsnark";
    version = "0.1.0";
    src = with lib.fileset; toSource {
      root = ./sdk/src/local/libsnark;
      fileset = ./sdk/src/local/libsnark;
    };
    vendorHash = "sha256-tGajRfJ8G4M89QSiJnjpTzQ3+VA2RLkavD1ipANeOSI=";

    buildPhase = "sh ./compile.sh";
    installPhase = ''
      mkdir -p "$out"/lib
      mv libsnark.so "$out"/lib/
    '';
  };

  commonArgs = {
    pname = "zkm";
    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 = [
      pkg-config
      openssl
      protobuf
      metacraft-labs.zkm
    ];
  };

  craneLib = craneLib-default.overrideToolchain metacraft-labs.zkm;
  cargoArtifacts = craneLib.buildDepsOnly (zkVM-helpers.fixDeps commonArgs);
in
  craneLib.buildPackage (zkVM-helpers.withCustomPhases (commonArgs
    // {
      inherit cargoArtifacts;

      postPatch = ''
        ln -s ../../../../guests/graph_coloring ./zkvms/zkm/guest/src/zkp
      '';

      hostBin = "host-zkm";
      guestTarget = "mips-zkm-zkvm-elf";

      preBuildGuest = ''
        # https://github.com/zkMIPS/zkm/blob/0e62a053970eb25c81aa409d0c7234f5611a192d/build/src/command/utils.rs#L45-L61
        export RUSTFLAGS="-C target-cpu=mips2 -C target-feature=+crt-static -C link-arg=-nostdlib -C link-arg=-g -C link-arg=--entry=main"
      '';

      postBuildGuest = ''
        unset RUSTFLAGS
      '';

      preBuild = ''
        export RUSTFLAGS="-L ${zkm_libsnark}/lib $RUSTFLAGS"
      '';

      preRun = ''
        export ELF_PATH="$out/bin/guest"
        export PKG_CONFIG_PATH='${openssl.dev}/lib/pkgconfig' # Dirty hack
        export LD_LIBRARY_PATH='${lib.makeLibraryPath [ openssl zkm_libsnark ]}'
      '';

      doCheck = false;
    }))