CHIA-1106: Add Python bindings to chiavdf and chiapos

.github/workflows/build-wheels.yml

@@ -146,23 +146,24 @@ jobs:
         run: |
           python -m pip install maturin
 
-      - name: Build MacOs with maturin on Python ${{ matrix.python }}
+      - name: Build MacOS with maturin on Python ${{ matrix.python.major-dot-minor }}
         if: matrix.os.matrix == 'macos'
         env:
           MACOSX_DEPLOYMENT_TARGET: "11.0"
         run: |
+          export LIBRARY_PATH=/opt/homebrew/lib:$LIBRARY_PATH
           python${{ matrix.python.major-dot-minor }} -m venv venv
           . venv/bin/activate
           maturin build -i python --release -m wheel/Cargo.toml
 
-      - name: Build Linux with maturin on Python ${{ matrix.python }}
+      - name: Build Linux with maturin on Python ${{ matrix.python.major-dot-minor }}
         if: matrix.os.matrix == 'ubuntu'
         run: |
           docker run --rm --pull always \
             -v ${{ github.workspace }}:/ws --workdir=/ws \
             ${{ matrix.python.by-arch[matrix.arch.matrix].docker-url }} \
             bash -exc '\
-              yum -y install openssl-devel && \
+              yum -y install openssl-devel gmp-devel clang && \
               source $HOME/.cargo/env && \
               rustup target add ${{ matrix.python.by-arch[matrix.arch.matrix].rustup-target }} && \
               python${{ matrix.python.major-dot-minor }} -m venv /venv && \
@@ -172,7 +173,15 @@ jobs:
               CC=gcc maturin build --release --manylinux ${{ matrix.python.by-arch[matrix.arch.matrix].manylinux-version }} -m wheel/Cargo.toml \
             '
 
-      - name: Build Windows with maturin on Python ${{ matrix.python }}
+      - name: Checkout MPIR for Windows
+        if: matrix.os.matrix == 'windows'
+        uses: actions/checkout@v4
+        with:
+          repository: Chia-Network/mpir_gc_x64
+          fetch-depth: 1
+          path: mpir_gc_x64
+
+      - name: Build Windows with maturin on Python ${{ matrix.python.major-dot-minor }}
         if: matrix.os.matrix == 'windows'
         env:
           CC: "clang"

Cargo.toml

@@ -139,3 +139,5 @@ zstd = "0.13.2"
 blocking-threadpool = "1.0.1"
 libfuzzer-sys = "0.4"
 wasm-bindgen = "0.2.93"
+chiavdf = "1.1.5"
+chiapos = "2.0.6"

crates/chia-consensus/Cargo.toml

@@ -13,6 +13,8 @@ workspace = true
 
 [features]
 py-bindings = ["dep:pyo3", "dep:chia_py_streamable_macro"]
+chiavdf = ["dep:chiavdf"]
+chiapos = ["dep:chiapos"]
 
 [dependencies]
 clvmr = { workspace = true }
@@ -29,6 +31,8 @@ chia-puzzles = { workspace = true }
 chia-bls = { workspace = true }
 hex-literal = { workspace = true }
 thiserror = { workspace = true }
+chiavdf = { workspace = true, optional = true }
+chiapos = { workspace = true, optional = true }
 
 [dev-dependencies]
 num-traits = { workspace = true }

crates/chia-consensus/src/lib.rs

@@ -10,3 +10,9 @@ pub mod merkle_set;
 pub mod merkle_tree;
 pub mod spendbundle_conditions;
 pub mod spendbundle_validation;
+
+#[cfg(feature = "chiavdf")]
+pub mod vdf;
+
+#[cfg(feature = "chiapos")]
+pub mod proof_of_space;

crates/chia-consensus/src/proof_of_space.rs

@@ -0,0 +1,17 @@
+use chia_protocol::{Bytes32, ProofOfSpace};
+use chiapos::validate_proof;
+
+pub fn get_quality_string(pos: &ProofOfSpace, plot_id: Bytes32) -> Option<Bytes32> {
+    let mut quality = [0; 32];
+    if validate_proof(
+        &plot_id.to_bytes(),
+        pos.size,
+        &pos.challenge.to_bytes(),
+        &pos.proof,
+        &mut quality,
+    ) {
+        Some(Bytes32::from(quality))
+    } else {
+        None
+    }
+}