commit b8775f36754a6fd6f80e9e60ea27a0941c903ac1 Author: theBreadCompany Date: Tue Dec 3 23:27:00 2024 +0100 add pollard-rho diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..ea8c4bf --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +/target diff --git a/Cargo.lock b/Cargo.lock new file mode 100644 index 0000000..efc010e --- /dev/null +++ b/Cargo.lock @@ -0,0 +1,147 @@ +# This file is automatically @generated by Cargo. +# It is not intended for manual editing. +version = 3 + +[[package]] +name = "byteorder" +version = "1.5.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1fd0f2584146f6f2ef48085050886acf353beff7305ebd1ae69500e27c67f64b" + +[[package]] +name = "cfg-if" +version = "1.0.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd" + +[[package]] +name = "ec_crypto" +version = "0.1.0" +dependencies = [ + "gcd", + "rand", +] + +[[package]] +name = "gcd" +version = "2.3.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1d758ba1b47b00caf47f24925c0074ecb20d6dfcffe7f6d53395c0465674841a" + +[[package]] +name = "getrandom" +version = "0.2.15" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "c4567c8db10ae91089c99af84c68c38da3ec2f087c3f82960bcdbf3656b6f4d7" +dependencies = [ + "cfg-if", + "libc", + "wasi", +] + +[[package]] +name = "libc" +version = "0.2.167" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "09d6582e104315a817dff97f75133544b2e094ee22447d2acf4a74e189ba06fc" + +[[package]] +name = "ppv-lite86" +version = "0.2.20" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "77957b295656769bb8ad2b6a6b09d897d94f05c41b069aede1fcdaa675eaea04" +dependencies = [ + "zerocopy", +] + +[[package]] +name = "proc-macro2" +version = "1.0.92" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "37d3544b3f2748c54e147655edb5025752e2303145b5aefb3c3ea2c78b973bb0" +dependencies = [ + "unicode-ident", +] + +[[package]] +name = "quote" +version = "1.0.37" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "b5b9d34b8991d19d98081b46eacdd8eb58c6f2b201139f7c5f643cc155a633af" +dependencies = [ + "proc-macro2", +] + +[[package]] +name = "rand" +version = "0.8.5" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "34af8d1a0e25924bc5b7c43c079c942339d8f0a8b57c39049bef581b46327404" +dependencies = [ + "libc", + "rand_chacha", + "rand_core", +] + +[[package]] +name = "rand_chacha" +version = "0.3.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "e6c10a63a0fa32252be49d21e7709d4d4baf8d231c2dbce1eaa8141b9b127d88" +dependencies = [ + "ppv-lite86", + "rand_core", +] + +[[package]] +name = "rand_core" +version = "0.6.4" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "ec0be4795e2f6a28069bec0b5ff3e2ac9bafc99e6a9a7dc3547996c5c816922c" +dependencies = [ + "getrandom", +] + +[[package]] +name = "syn" +version = "2.0.90" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "919d3b74a5dd0ccd15aeb8f93e7006bd9e14c295087c9896a110f490752bcf31" +dependencies = [ + "proc-macro2", + "quote", + "unicode-ident", +] + +[[package]] +name = "unicode-ident" +version = "1.0.14" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "adb9e6ca4f869e1180728b7950e35922a7fc6397f7b641499e8f3ef06e50dc83" + +[[package]] +name = "wasi" +version = "0.11.0+wasi-snapshot-preview1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "9c8d87e72b64a3b4db28d11ce29237c246188f4f51057d65a7eab63b7987e423" + +[[package]] +name = "zerocopy" +version = "0.7.35" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1b9b4fd18abc82b8136838da5d50bae7bdea537c574d8dc1a34ed098d6c166f0" +dependencies = [ + "byteorder", + "zerocopy-derive", +] + +[[package]] +name = "zerocopy-derive" +version = "0.7.35" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "fa4f8080344d4671fb4e831a13ad1e68092748387dfc4f55e356242fae12ce3e" +dependencies = [ + "proc-macro2", + "quote", + "syn", +] diff --git a/Cargo.toml b/Cargo.toml new file mode 100644 index 0000000..84d9883 --- /dev/null +++ b/Cargo.toml @@ -0,0 +1,8 @@ +[package] +name = "ec_crypto" +version = "0.1.0" +edition = "2021" + +[dependencies] +rand = "0.8.5" +gcd = "2.3.0" diff --git a/src/main.rs b/src/main.rs new file mode 100644 index 0000000..fd0ca05 --- /dev/null +++ b/src/main.rs @@ -0,0 +1,7 @@ +mod pollard_rho; + +fn main() { + let n = rand::random::(); + let n_primediv = pollard_rho::pollard_rho(n.into()); + eprintln!("Generated random number {}, got prime divisor {}", n, n_primediv); +} diff --git a/src/pollard_rho.rs b/src/pollard_rho.rs new file mode 100644 index 0000000..91ca656 --- /dev/null +++ b/src/pollard_rho.rs @@ -0,0 +1,53 @@ +use gcd::Gcd; + +/** + * Calculate the primedivisor for some number `n` +*/ +pub fn pollard_rho(n: u16) -> u16 { + if n == 1 { return 1; } // 1 only has 1 as prime divisor + if n % 2 == 0 { return 2; } // even numbers have at least 2 as prime divisor + if is_prime(n) { return n; } + + let mut x = rand::random::() % 2; + let mut y = x; + + let c = rand::random::() % 2; + let mut div = 1; + + while div == 1 { + x = (mod_pow(x, 2, n) + c + n) % n; + y = (mod_pow(y, 2, n) + c + n) % n; + y = (mod_pow(y, 2, n) + c + n) % n; + div = u16::try_from((i16::try_from(x).unwrap()-i16::try_from(y).unwrap()).abs()).unwrap().gcd(n); + eprintln!("Got div {}", div); + if div == n { return pollard_rho(n); } + } + + return div; +} + +/** + * Discrete/Modular exponentiation + * + * Highly memory efficient because the full result is never stored, but shortened by defined modulo instead. + * We can use that because the prime divisor required for our algorithm is guarenteed to be smaller + * than n. + * + * Counterpart function to the discrete logarithm. + */ +fn mod_pow(base: u16, exp: u16, r#mod: u16) -> u16 { + let mut result = 1; + + for _ in 0..exp-1 { + result = (result * base) % r#mod; + } + + result +} + +fn is_prime(n: u16) -> bool { + for i in (3..=(n as f32).sqrt() as u16).step_by(2) { + if n % i == 0 { return false; } + } + true +}