feat: factordb attack

.github/workflows/ci.yml

@@ -9,6 +9,7 @@ on:
 
 env:
   RUST_BACKTRACE: 1
+  NO_FACTORDB: 1
 
 jobs:
   fmt:
@@ -192,8 +193,7 @@ jobs:
     runs-on: ubuntu-latest
     timeout-minutes: 30
 
-    needs:
-      [fmt, clippy, docs, tests, ctf-challenges, coverage, build, docker-build]
+    needs: [fmt, clippy, docs, tests, ctf-challenges, coverage, build, docker-build]
     if: github.event_name == 'push' && contains(github.ref, 'refs/tags/')
 
     steps:
@@ -211,8 +211,7 @@ jobs:
     runs-on: ubuntu-latest
     timeout-minutes: 120
 
-    needs:
-      [fmt, clippy, docs, tests, ctf-challenges, coverage, build, docker-build]
+    needs: [fmt, clippy, docs, tests, ctf-challenges, coverage, build, docker-build]
     if: github.event_name == 'push' && contains(github.ref, 'refs/tags/')
 
     env:

Cargo.toml

@@ -54,6 +54,7 @@ indicatif = "0.17"
 itertools = "0.13"
 discrete-logarithm = "1.0"
 base64 = "0.22"
+factordb = { version = "0.3.0", features = ["blocking"] }
 
 [dependencies.rug]
 version = "1.26"

src/attack/factorization/factordb.rs

@@ -0,0 +1,49 @@
+use std::str::FromStr;
+
+use factordb::FactorDbBlockingClient;
+use indicatif::ProgressBar;
+use rug::Integer;
+
+use crate::{key::PrivateKey, Attack, AttackSpeed, Error, Parameters, Solution};
+
+/// Factordb attack
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub struct FactorDbAttack;
+
+impl Attack for FactorDbAttack {
+    fn name(&self) -> &'static str {
+        "factordb"
+    }
+
+    fn speed(&self) -> AttackSpeed {
+        AttackSpeed::Fast
+    }
+
+    fn run(&self, params: &Parameters, _pb: Option<&ProgressBar>) -> Result<Solution, Error> {
+        // If `NO_FACTORDB` is set, ignore the factordb attack
+        // This is useful for testing
+        if std::env::var("NO_FACTORDB").is_ok() {
+            return Err(Error::NotFound);
+        }
+
+        let e = &params.e;
+        let n = params.n.as_ref().ok_or(Error::MissingParameters)?;
+
+        let factors = tokio::task::block_in_place(|| FactorDbBlockingClient::new().get(n))
+            .map_err(|_| Error::NotFound)?
+            .into_factors_flattened()
+            .iter()
+            .map(|f| Integer::from_str(&f.to_string()))
+            .collect::<Result<Vec<Integer>, _>>()
+            .map_err(|_| Error::NotFound)?;
+
+        if factors.len() < 2 {
+            return Err(Error::NotFound);
+        }
+
+        Ok(Solution::new_pk(
+            self.name(),
+            PrivateKey::from_factors(factors, e.clone())?,
+        ))
+    }
+}

src/attack/factorization/mod.rs

@@ -5,6 +5,7 @@ use lazy_static::lazy_static;
 mod brent;
 mod cunningham_chain;
 mod ecm;
+mod factordb;
 mod fermat;
 mod hart;
 mod known_factors;
@@ -23,6 +24,7 @@ mod twin_prime;
 pub use self::ecm::EcmAttack;
 pub use brent::BrentAttack;
 pub use cunningham_chain::CunninghamChainAttack;
+pub use factordb::FactorDbAttack;
 pub use fermat::FermatAttack;
 pub use hart::HartAttack;
 pub use known_factors::KnownFactorsAttack;
@@ -45,6 +47,7 @@ lazy_static! {
         Arc::new(BrentAttack),
         Arc::new(CunninghamChainAttack),
         Arc::new(EcmAttack),
+        Arc::new(FactorDbAttack),
         Arc::new(FermatAttack),
         Arc::new(HartAttack),
         Arc::new(KnownFactorsAttack),