feat: define FFI interfaces to mcaptcha_pow_sha256

This commit is contained in:
Aravinth Manivannan 2024-01-08 04:19:36 +05:30
commit 1519ed69ca
Signed by: realaravinth
GPG key ID: F8F50389936984FF
7 changed files with 555 additions and 0 deletions

2
.gitignore vendored Normal file
View file

@ -0,0 +1,2 @@
lib/
main

9
Makefile Normal file
View file

@ -0,0 +1,9 @@
build-rust:
@-rm -rf lib
mkdir lib/
cd ./mcaptcha-pow-ffi/ && cargo build --release \
&& cp ./target/release/libmcaptcha_pow_ffi.a ../lib/ \
&& cp ./target/release/libmcaptcha_pow_ffi.so ../lib/
test:
cd ./mcaptcha-pow-ffi/ && cargo test

1
mcaptcha-pow-ffi/.gitignore vendored Normal file
View file

@ -0,0 +1 @@
/target

351
mcaptcha-pow-ffi/Cargo.lock generated Normal file
View file

@ -0,0 +1,351 @@
# This file is automatically @generated by Cargo.
# It is not intended for manual editing.
version = 3
[[package]]
name = "autocfg"
version = "1.1.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d468802bab17cbc0cc575e9b053f41e72aa36bfa6b7f55e3529ffa43161b97fa"
[[package]]
name = "bincode"
version = "1.3.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b1f45e9417d87227c7a56d22e471c6206462cba514c7590c09aff4cf6d1ddcad"
dependencies = [
"serde",
]
[[package]]
name = "block-buffer"
version = "0.10.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3078c7629b62d3f0439517fa394996acacc5cbc91c5a20d8c658e77abd503a71"
dependencies = [
"generic-array",
]
[[package]]
name = "cfg-if"
version = "1.0.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
[[package]]
name = "cpufeatures"
version = "0.2.12"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "53fe5e26ff1b7aef8bca9c6080520cfb8d9333c7568e1829cef191a9723e5504"
dependencies = [
"libc",
]
[[package]]
name = "crypto-common"
version = "0.1.6"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1bfb12502f3fc46cca1bb51ac28df9d618d813cdc3d2f25b9fe775a34af26bb3"
dependencies = [
"generic-array",
"typenum",
]
[[package]]
name = "darling"
version = "0.14.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "7b750cb3417fd1b327431a470f388520309479ab0bf5e323505daf0290cd3850"
dependencies = [
"darling_core",
"darling_macro",
]
[[package]]
name = "darling_core"
version = "0.14.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "109c1ca6e6b7f82cc233a97004ea8ed7ca123a9af07a8230878fcfda9b158bf0"
dependencies = [
"fnv",
"ident_case",
"proc-macro2",
"quote",
"strsim",
"syn 1.0.109",
]
[[package]]
name = "darling_macro"
version = "0.14.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a4aab4dbc9f7611d8b55048a3a16d2d010c2c8334e46304b40ac1cc14bf3b48e"
dependencies = [
"darling_core",
"quote",
"syn 1.0.109",
]
[[package]]
name = "derive_builder"
version = "0.12.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8d67778784b508018359cbc8696edb3db78160bab2c2a28ba7f56ef6932997f8"
dependencies = [
"derive_builder_macro",
]
[[package]]
name = "derive_builder_core"
version = "0.12.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "c11bdc11a0c47bc7d37d582b5285da6849c96681023680b906673c5707af7b0f"
dependencies = [
"darling",
"proc-macro2",
"quote",
"syn 1.0.109",
]
[[package]]
name = "derive_builder_macro"
version = "0.12.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ebcda35c7a396850a55ffeac740804b40ffec779b98fffbb1738f4033f0ee79e"
dependencies = [
"derive_builder_core",
"syn 1.0.109",
]
[[package]]
name = "digest"
version = "0.10.7"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "9ed9a281f7bc9b7576e61468ba615a66a5c8cfdff42420a70aa82701a3b1e292"
dependencies = [
"block-buffer",
"crypto-common",
]
[[package]]
name = "fnv"
version = "1.0.7"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3f9eec918d3f24069decb9af1554cad7c880e2da24a9afd88aca000531ab82c1"
[[package]]
name = "generic-array"
version = "0.14.7"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "85649ca51fd72272d7821adaf274ad91c288277713d9c18820d8499a7ff69e9a"
dependencies = [
"typenum",
"version_check",
]
[[package]]
name = "ident_case"
version = "1.0.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b9e0384b61958566e926dc50660321d12159025e767c18e043daf26b70104c39"
[[package]]
name = "libc"
version = "0.2.151"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "302d7ab3130588088d277783b1e2d2e10c9e9e4a16dd9050e6ec93fb3e7048f4"
[[package]]
name = "mcaptcha-pow-ffi"
version = "0.1.0"
dependencies = [
"bincode",
"libc",
"mcaptcha_pow_sha256",
]
[[package]]
name = "mcaptcha_pow_sha256"
version = "0.5.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "88b007473b7f3ef294022e8034f309b31ff0ba4c891c3bdf9dea7cd223f8ba45"
dependencies = [
"bincode",
"derive_builder",
"num",
"serde",
"sha2",
]
[[package]]
name = "num"
version = "0.4.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b05180d69e3da0e530ba2a1dae5110317e49e3b7f3d41be227dc5f92e49ee7af"
dependencies = [
"num-bigint",
"num-complex",
"num-integer",
"num-iter",
"num-rational",
"num-traits",
]
[[package]]
name = "num-bigint"
version = "0.4.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "608e7659b5c3d7cba262d894801b9ec9d00de989e8a82bd4bef91d08da45cdc0"
dependencies = [
"autocfg",
"num-integer",
"num-traits",
"serde",
]
[[package]]
name = "num-complex"
version = "0.4.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1ba157ca0885411de85d6ca030ba7e2a83a28636056c7c699b07c8b6f7383214"
dependencies = [
"num-traits",
"serde",
]
[[package]]
name = "num-integer"
version = "0.1.45"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "225d3389fb3509a24c93f5c29eb6bde2586b98d9f016636dff58d7c6f7569cd9"
dependencies = [
"autocfg",
"num-traits",
]
[[package]]
name = "num-iter"
version = "0.1.43"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "7d03e6c028c5dc5cac6e2dec0efda81fc887605bb3d884578bb6d6bf7514e252"
dependencies = [
"autocfg",
"num-integer",
"num-traits",
]
[[package]]
name = "num-rational"
version = "0.4.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "0638a1c9d0a3c0914158145bc76cff373a75a627e6ecbfb71cbe6f453a5a19b0"
dependencies = [
"autocfg",
"num-integer",
"num-traits",
"serde",
]
[[package]]
name = "num-traits"
version = "0.2.17"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "39e3200413f237f41ab11ad6d161bc7239c84dcb631773ccd7de3dfe4b5c267c"
dependencies = [
"autocfg",
]
[[package]]
name = "proc-macro2"
version = "1.0.75"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "907a61bd0f64c2f29cd1cf1dc34d05176426a3f504a78010f08416ddb7b13708"
dependencies = [
"unicode-ident",
]
[[package]]
name = "quote"
version = "1.0.35"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "291ec9ab5efd934aaf503a6466c5d5251535d108ee747472c3977cc5acc868ef"
dependencies = [
"proc-macro2",
]
[[package]]
name = "serde"
version = "1.0.194"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "0b114498256798c94a0689e1a15fec6005dee8ac1f41de56404b67afc2a4b773"
dependencies = [
"serde_derive",
]
[[package]]
name = "serde_derive"
version = "1.0.194"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a3385e45322e8f9931410f01b3031ec534c3947d0e94c18049af4d9f9907d4e0"
dependencies = [
"proc-macro2",
"quote",
"syn 2.0.48",
]
[[package]]
name = "sha2"
version = "0.10.8"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "793db75ad2bcafc3ffa7c68b215fee268f537982cd901d132f89c6343f3a3dc8"
dependencies = [
"cfg-if",
"cpufeatures",
"digest",
]
[[package]]
name = "strsim"
version = "0.10.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "73473c0e59e6d5812c5dfe2a064a6444949f089e20eec9a2e5506596494e4623"
[[package]]
name = "syn"
version = "1.0.109"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "72b64191b275b66ffe2469e8af2c1cfe3bafa67b529ead792a6d0160888b4237"
dependencies = [
"proc-macro2",
"quote",
"unicode-ident",
]
[[package]]
name = "syn"
version = "2.0.48"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "0f3531638e407dfc0814761abb7c00a5b54992b849452a0646b7f65c9f770f3f"
dependencies = [
"proc-macro2",
"quote",
"unicode-ident",
]
[[package]]
name = "typenum"
version = "1.17.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "42ff0bf0c66b8238c6f3b578df37d0b7848e55df8577b3f74f92a69acceeb825"
[[package]]
name = "unicode-ident"
version = "1.0.12"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3354b9ac3fae1ff6755cb6db53683adb661634f67557942dea4facebec0fee4b"
[[package]]
name = "version_check"
version = "0.9.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "49874b5167b65d7193b8aba1567f5c7d93d001cafc34600cee003eda787e483f"

View file

@ -0,0 +1,24 @@
[package]
name = "mcaptcha-pow-ffi"
version = "0.1.0"
edition = "2021"
authors = ["realaravinth <realaravinth@batsense.net>"]
description = "FFI library to compute mCaptcha Proof-of-Work"
keywords = ["DDoS", "mcaptcha", "captcha", "pow"]
homepage = "https://mcaptcha.org"
repository = "https://git.batsense.net/mCaptcha/mcaptcha-pow-rs"
documentation = "https://docs.rs/mcaptcha-api-rs"
license = "MIT OR Apache-2.0"
readme = "README.md"
[lib]
# If you only wanted dynamic library, you'd use only "cdylib".
# If you only wanted static library, you'd use only "staticlib".
# This demo shows both. See https://doc.rust-lang.org/reference/linkage.html
# for more information.
crate-type = ["cdylib", "staticlib"]
[dependencies]
bincode = "1.3.3"
libc = "0.2.151"
mcaptcha_pow_sha256 = "0.5.0"

View file

@ -0,0 +1,13 @@
#include <stdbool.h>
struct ProofOfWork {
unsigned long int nonce;
char *result;
};
bool is_valid_proof(unsigned long int nonce, char *result, char *phrase,
char *salt);
bool is_sufficient_difficulty(unsigned long int nonce, char *result, char *salt,
unsigned difficulty);
struct ProofOfWork prove_work(char *salt, char *phrase, unsigned difficulty);

155
mcaptcha-pow-ffi/src/lib.rs Normal file
View file

@ -0,0 +1,155 @@
use mcaptcha_pow_sha256::ConfigBuilder;
use std::ffi::CStr;
use std::ffi::CString;
#[repr(C)]
pub struct ProofOfWork {
pub nonce: u64,
pub result: *const libc::c_char,
}
impl From<mcaptcha_pow_sha256::PoW<String>> for ProofOfWork {
fn from(value: mcaptcha_pow_sha256::PoW<String>) -> Self {
Self {
nonce: value.nonce,
result: CString::new(value.result).unwrap().into_raw(),
}
}
}
#[no_mangle]
pub extern "C" fn prove_work(
salt: *const libc::c_char,
phrase: *const libc::c_char,
difficulty: u32,
) -> ProofOfWork {
let (salt, phrase) = unsafe { (CStr::from_ptr(salt), CStr::from_ptr(phrase)) };
let salt = salt.to_str().unwrap();
let phrase = phrase.to_str().unwrap().to_string();
let config = ConfigBuilder::default()
.salt(salt.to_string())
.build()
.unwrap();
config.prove_work(&phrase, difficulty).unwrap().into()
}
#[no_mangle]
pub extern "C" fn is_valid_proof(
nonce: u64,
result: *const libc::c_char,
phrase: *const libc::c_char,
salt: *const libc::c_char,
) -> bool {
let (result, phrase, salt) = unsafe {
(
CStr::from_ptr(result),
CStr::from_ptr(phrase),
CStr::from_ptr(salt),
)
};
let result = result.to_str().unwrap().to_string();
let pow: mcaptcha_pow_sha256::PoW<String> = mcaptcha_pow_sha256::PoWBuilder::default()
.nonce(nonce)
.result(result)
.build()
.unwrap();
let config = ConfigBuilder::default()
.salt(salt.to_str().unwrap().to_string())
.build()
.unwrap();
config.is_valid_proof(&pow, &phrase.to_str().unwrap().to_string())
}
#[no_mangle]
pub extern "C" fn is_sufficient_difficulty(
nonce: u64,
result: *const libc::c_char,
salt: *const libc::c_char,
difficulty: u32,
) -> bool {
let (result, salt) = unsafe { (CStr::from_ptr(result), CStr::from_ptr(salt)) };
let result = result.to_str().unwrap().to_string();
let pow: mcaptcha_pow_sha256::PoW<String> = mcaptcha_pow_sha256::PoWBuilder::default()
.nonce(nonce)
.result(result)
.build()
.unwrap();
let config = ConfigBuilder::default()
.salt(salt.to_str().unwrap().to_string())
.build()
.unwrap();
config.is_sufficient_difficulty(&pow, difficulty)
}
#[cfg(test)]
pub mod test {
use super::*;
use std::ffi::CString;
#[test]
fn test_pow() {
let salt: String =
"27b7eb6be3437b7da0d9329dea9a7c76f1640cd16d39fd94b5d0dd42c763c63c".into();
let phrase: String =
"b641c711c01079805a2a0888bf716b66c83419ab0c65455a49cb4d8b3e36c4ec".into();
let difficulty = 50_000;
let c = mcaptcha_pow_sha256::ConfigBuilder::default()
.salt(salt.clone())
.build()
.unwrap();
// calculate pow
let mcaptcha_work = c.prove_work(&phrase, difficulty).unwrap();
let ffi_work = super::prove_work(
CString::new(salt.clone()).unwrap().into_raw(),
CString::new(phrase.clone()).unwrap().into_raw(),
difficulty,
);
assert_eq!(ffi_work.nonce, mcaptcha_work.nonce);
let res = unsafe { CStr::from_ptr(ffi_work.result) };
let res = res.to_str().unwrap();
assert_eq!(res, mcaptcha_work.result);
// verify pow
assert!(super::is_valid_proof(
mcaptcha_work.nonce,
CString::new(mcaptcha_work.result.clone())
.unwrap()
.into_raw(),
CString::new(phrase.clone()).unwrap().into_raw(),
CString::new(salt.clone()).unwrap().into_raw(),
));
assert!(!super::is_valid_proof(
mcaptcha_work.nonce,
CString::new(phrase.clone()).unwrap().into_raw(),
CString::new(phrase.clone()).unwrap().into_raw(),
CString::new(salt.clone()).unwrap().into_raw(),
));
// is sufficient difficulty
assert!(super::is_sufficient_difficulty(
mcaptcha_work.nonce,
CString::new(mcaptcha_work.result.clone())
.unwrap()
.into_raw(),
CString::new(salt.clone()).unwrap().into_raw(),
difficulty
));
// is sufficient difficulty
assert!(super::is_sufficient_difficulty(
mcaptcha_work.nonce,
CString::new(mcaptcha_work.result.clone())
.unwrap()
.into_raw(),
CString::new(salt.clone()).unwrap().into_raw(),
difficulty
));
}
}