flakified KEVM / EVM

Author: craigem

flake.lock

@@ -0,0 +1,81 @@
+{
+  description = "Mantis flake";
+
+  inputs = {
+    flake-utils.url = "github:numtide/flake-utils";
+    nixpkgs.url =
+      "github:nixos/nixpkgs?rev=a98302aa9b9628915878a6ea9776c40a0bb02950";
+    nixpkgs-sbt = {
+      url = "github:nixos/nixpkgs?rev=e2bb73ce5f786b83e984b80199112f86b8a6cc9d";
+      flake = false;
+    };
+    sbtix = {
+      url =
+        "github:input-output-hk/Sbtix?rev=7b969a5641fce10500ca51cbe88af4ea160d7064";
+      flake = false;
+    };
+  };
+
+  outputs = { self, nixpkgs, flake-utils, nixpkgs-sbt, ... }@inputs:
+    let
+      overlay = final: prev: {
+        inherit (import nixpkgs-sbt { inherit (final) system; }) sbt;
+        sbtix = final.callPackage ./sbtix.nix { };
+
+        mantisPkgs = final.callPackage ./nix/pkgs/mantis {
+          src = builtins.fetchGit {
+            url = "https://github.com/input-output-hk/mantis";
+            rev = self.rev or "482340d5e6ab635e5a5047e9b670d59b4ad366c2";
+            ref = "develop";
+            submodules = true;
+          };
+        };
+
+        jdk = prev.openjdk8_headless;
+        jre = prev.openjdk8_headless.jre;
+
+        inherit (final.mantisPkgs) mantis;
+
+        kevm = final.callPackage ./nix/pkgs/kevm.nix { };
+
+        mantis-entrypoint = final.callPackage ./nix/entrypoint.nix { };
+      };
+
+      pkgsForSystem = system:
+        (import nixpkgs) {
+          inherit system;
+          overlays = [ overlay ];
+        };
+
+      mkHydraUtils = mkPkgs:
+        let
+          # nothing in lib should really depend on the system
+          libPkgs = mkPkgs "x86_64-linux";
+          # [attrset] -> attrset
+          recursiveMerge = libPkgs.lib.foldr libPkgs.lib.recursiveUpdate { };
+          mkHydraJobsForSystem = attrs: system:
+            recursiveMerge
+            (map (n: { "${n}"."${system}" = (mkPkgs system)."${n}"; }) attrs);
+        in {
+          collectHydraSets = jobSets: { hydraJobs = recursiveMerge jobSets; };
+          mkHydraSet = attrs: systems:
+            recursiveMerge (map (mkHydraJobsForSystem attrs) systems);
+        };
+
+      hydraUtils = mkHydraUtils pkgsForSystem;
+      inherit (hydraUtils) collectHydraSets mkHydraSet;
+
+    in flake-utils.lib.eachDefaultSystem (system: rec {
+      pkgs = pkgsForSystem system;
+      legacyPackages = pkgs;
+
+      defaultPackage = pkgs.mantis;
+      devShell = pkgs.mkShell { nativeBuildInputs = with pkgs; [ solc sbt ]; };
+      apps.mantis = flake-utils.lib.mkApp { drv = pkgs.mantis; };
+      defaultApp = apps.mantis;
+    }) // (collectHydraSets [
+      (mkHydraSet [ "mantis" ] [ "x86_64-linux" "x86_64-darwin" ])
+      (mkHydraSet [ "kevm" ] [ "x86_64-linux" ])
+      (mkHydraSet [ "mantis-entrypoint" ] [ "x86_64-linux" ])
+    ]);
+}

flake.nix

@@ -0,0 +1,8 @@
+{ lib, writeShellScript, awscli, mantis-kevm, coreutils, gnugrep }:
+writeShellScript "mantis" ''
+  set -exuo pipefail
+
+  export PATH="${lib.makeBinPath [ coreutils mantis-kevm awscli gnugrep ]}"
+
+  ${builtins.readFile ./entrypoint.sh}
+''

nix/default.nix

@@ -0,0 +1,33 @@
+#!/usr/bin/env bash
+
+set -exuo pipefail
+
+mkdir -p /tmp "$NOMAD_TASK_DIR/mantis"
+cd "$NOMAD_TASK_DIR"
+
+if [ -n "${DAG_NAME:-}" ]; then
+    if [ -f "ethash/$DAG_NAME" ]; then
+        echo "found existing DAG"
+        sha256sum "ethash/$DAG_NAME"
+    else
+        mkdir -p ethash
+        aws \
+            --endpoint-url "$MONITORING_ADDR" \
+            s3 cp \
+            "s3://mantis-kevm-dag/$DAG_NAME" \
+            "ethash/$DAG_NAME" \
+            || echo "Unable to download DAG, skipping."
+    fi
+fi
+
+set +x
+echo "Waiting for $REQUIRED_PEER_COUNT peers to show up in the config"
+until [ "$(grep -c enode mantis.conf)" -ge "$REQUIRED_PEER_COUNT" ]; do
+    sleep 1
+done
+set -x
+
+ulimit -c unlimited
+cp mantis.conf running.conf
+
+exec mantis "-Duser.home=$NOMAD_TASK_DIR" "$@"

nix/entrypoint.nix

@@ -0,0 +1,25 @@
+{ lib, stdenv, dockerTools, secp256k1, gmp5, mpfr, zlib }:
+let libPath = lib.makeLibraryPath [ secp256k1 gmp5 mpfr zlib ];
+in stdenv.mkDerivation {
+  name = "kevm";
+
+  src = dockerTools.pullImage {
+    imageName = "inputoutput/mantis";
+    imageDigest =
+      "sha256:594ed009f1bc1f12b86e11136441602107c3d580476002d9bae58b258a74ac1b";
+    sha256 = "sha256-JT+FarGQlgYoO392Ne1ofdqmishZLfH+OwV2CXTYwdA=";
+  };
+
+  installPhase = ''
+    mkdir -p tmp $out/bin
+    tar --delay-directory-restore -C tmp -xf layer.tar || true
+
+    cp tmp/bin/kevm-vm $out/bin
+    chmod 0755 $out/bin/kevm-vm
+
+    patchelf \
+      --set-interpreter "$(cat $NIX_CC/nix-support/dynamic-linker)" \
+      --set-rpath "${libPath}" \
+      "$out/bin/kevm-vm"
+  '';
+}

nix/entrypoint.sh

@@ -1,8 +1,6 @@
-{ src, mkSrc, lib, gitignoreSource, callPackage, jre }: rec {
+{ src, lib, callPackage, jre }: rec {
 
-  mantis-source = mkSrc src;
-
-  mantis-unwrapped = callPackage ./unwrapped.nix { src = mantis-source; };
+  mantis-unwrapped = callPackage ./unwrapped.nix { inherit src; };
 
   mantis = callPackage ./wrapped.nix {
     mantis = mantis-unwrapped;

nix/pkgs/default.nix

@@ -1,9 +1,9 @@
-{ lib, stdenv, mantis, makeWrapper, jre, gawk }:
+{ lib, stdenv, mantis, makeWrapper, jre, gawk, gnused, kevm }:
 let
   inherit (stdenv.lib) optionalString makeLibraryPath;
   inherit (stdenv) cc isDarwin;
   LD_LIBRARY_PATH = makeLibraryPath [ cc.cc.lib ];
-  PATH = lib.makeBinPath [ jre gawk ];
+  PATH = lib.makeBinPath [ jre gawk gnused kevm ];
 in stdenv.mkDerivation {
   pname = "mantis";
   version = let
@@ -23,11 +23,13 @@ in stdenv.mkDerivation {
 
     for p in $(find $out/bin/* -executable); do
       wrapProgram "$p" \
-        --prefix PATH : ${PATH} \
+        --set PATH ${PATH} \
         ${
           optionalString (!isDarwin)
           "--prefix LD_LIBRARY_PATH : ${LD_LIBRARY_PATH}"
         }
     done
+
+    ln -s ${kevm}/bin/kevm-vm $out/bin
   '';
 }