blob: a6d8282c44196135859bca73db6faf5461c38081 (
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
|
{
description = "zkVMs benchmarks";
nixConfig = {
extra-substituters = "https://nix-blockchain-development.cachix.org";
extra-trusted-public-keys = "nix-blockchain-development.cachix.org-1:Ekei3RuW3Se+P/UIo6Q/oAgor/fVhFuuuX5jR8K/cdg=";
};
inputs = {
mcl-blockchain.url = "github:metacraft-labs/nix-blockchain-development?ref=zkvm-packages";
nixpkgs.follows = "mcl-blockchain/nixpkgs";
crane.follows = "mcl-blockchain/crane";
rust-overlay.follows = "mcl-blockchain/rust-overlay";
# flake-utils.follows = "mcl-blockchain/flake-utils";
};
outputs = { self, nixpkgs, mcl-blockchain, crane, rust-overlay, ... }:
let
system = "x86_64-linux";
pkgs = import nixpkgs { system = system; overlays = [
mcl-blockchain.overlays.default
rust-overlay.overlays.default
];
};
craneLib-default = crane.mkLib pkgs;
callPackage = pkgs.lib.callPackageWith pkgs;
fixDeps = commonArgs: commonArgs // {
postUnpack = ''
${commonArgs.postUnpack or ""}
ln -s ../../../guests ./source/zkvms/${commonArgs.pname}/guest/
ln -s ../../../guests_macro ./source/zkvms/${commonArgs.pname}/guest/
'';
preBuild = ''
${commonArgs.preBuild or ""}
cd zkvms/${commonArgs.pname}/host
cargo check --release --offline --all-targets
'';
};
withGeneratedLockfile = guest: commonArgs: with pkgs; {
cargoVendorDir = craneLib-default.vendorCargoDeps {
src = stdenv.mkDerivation {
name = "cargolock";
src = lib.fileset.toSource {
root = ./.;
fileset = ./.;
};
installPhase = ''
mkdir -p "$out"
cd zkvms/${commonArgs.pname}
cat ./host/Cargo.lock > "$out/Cargo.lock"
tail -n +4 ./guest/Cargo.lock >> "$out/Cargo.lock"
tail -n +4 ../../guests/${guest}/Cargo.lock >> "$out/Cargo.lock"
'';
};
};
} // commonArgs;
# Creates custom build and install phases
# Adds the "buildGuest" phase
# Adds the "run" pseudo-phase (running your zkVM is done with a shell script,
# this "phase" allows one to add things to the script)
# Requirements:
# - zkVM is inside zkvms/pname/
# - guest crate is located at zkvms/pname/guest and is named "guest"
withCustomPhases = guest: currentPackage: let
hostBin = currentPackage.hostBin or ("host-" + currentPackage.pname);
in with currentPackage; {
phases = [
"unpackPhase" # Standard phases
"linkGuest" # Custom phase
"patchPhase" "configurePhase" # Standard phases
"buildGuestPhase" # Custom phase
"buildPhase" "checkPhase" "installPhase" "fixupPhase" # Standard phases
];
linkGuest = ''
echo 'zkp = { path = "../../../guests/${guest}", package = "${guest}" }' >> zkvms/${currentPackage.pname}/guest/Cargo.toml
'';
buildGuestPhase = ''
pushd zkvms/${currentPackage.pname}/guest
runHook preBuildGuest
${currentPackage.buildGuestCommand or "cargo build --release"} \
${if currentPackage ? guestTarget then "--target " + currentPackage.guestTarget else ""} \
${currentPackage.guestExtraArgs or ""}
${if currentPackage ? guestTarget then "ln -s ../../guest/target/${currentPackage.guestTarget}/release/guest ../host/src/guest" else ""}
unset RUSTUP_TOOLCHAIN RUSTFLAGS CARGO_ENCODED_RUSTFLAGS
runHook postBuildGuest
popd
'';
buildPhase = ''
export INPUTS_DIR="$PWD/guests/${guest}"
pushd zkvms/${currentPackage.pname}/host
runHook preBuild
cargo build --release
runHook postBuild
popd
'';
installPhase = let
preRunBinaries =
if currentPackage ? preRunBinaries && builtins.length currentPackage.preRunBinaries > 0 then
"export PATH=\"\\$PATH:" + pkgs.lib.makeBinPath currentPackage.preRunBinaries + "\""
else
"";
preRunLibraries =
if currentPackage ? preRunLibraries && builtins.length currentPackage.preRunLibraries > 0 then
"export LD_LIBRARY_PATH=\"\\$LD_LIBRARY_PATH:" + pkgs.lib.makeLibraryPath currentPackage.preRunLibraries + "\""
else
"";
in ''
runHook preInstall
mkdir -p "$out"/bin
for bin in $(find . -type f -regex "./zkvms/.*release/[^/]*" -executable -print)
do
mv "$bin" "$out"/bin/
done
cat <<EOF > "$out"/bin/${pname}
#!/usr/bin/env sh
${preRunBinaries}
${preRunLibraries}
${currentPackage.preRun or ""}
"$out"/bin/${hostBin} \$@
EOF
chmod +x "$out"/bin/${pname}
runHook postInstall
'';
doNotPostBuildInstallCargoBinaries = true;
} // currentPackage;
createPackages = guestName: let
guest = if guestName == null then "graph_coloring" else guestName;
postfix = if guestName == null then "" else "/" + guest;
args-zkVM = {
inherit craneLib-default;
zkVM-helpers = {
inherit fixDeps;
withGeneratedLockfile = withGeneratedLockfile guest;
withCustomPhases = withCustomPhases guest;
};
};
in {
"risc0${postfix}" = callPackage ./zkvms/risc0/default.nix args-zkVM;
"sp1${postfix}" = callPackage ./zkvms/sp1/default.nix args-zkVM;
"zkwasm${postfix}" = callPackage ./zkvms/zkwasm/default.nix args-zkVM;
"zkm${postfix}" = callPackage ./zkvms/zkm/default.nix args-zkVM;
"jolt${postfix}" = callPackage ./zkvms/jolt/default.nix args-zkVM;
"nexus${postfix}" = callPackage ./zkvms/nexus/default.nix args-zkVM;
};
guests = [ null ] ++ (builtins.attrNames
(pkgs.lib.filterAttrs
(_: type: type == "directory")
(builtins.readDir ./guests)));
in {
packages.${system} = pkgs.lib.foldr
(guest: accum: accum // (createPackages guest))
{}
guests;
};
}
|
