Merge pull request #13 from LimesKey/main

Gather user's system GPU specs

.github/workflows/rust.yml

@@ -2,27 +2,50 @@ name: Rust
 
 on:
   push:
-    branches: [ "main" ]
+    branches: [ "main", "master", "GPU-MD5-Crack" ]
   pull_request:
-    branches: [ "main" ]
+    branches: [ "main", "master", "GPU-MD5-Crack" ]
 
 env:
   CARGO_TERM_COLOR: always
 
 jobs:
   build:
-
-    runs-on: windows-latest
-
+    runs-on: ubuntu-latest
+
     steps:
+    - name: Load OpenCL
+      run: |
+        wget -qO - https://repositories.intel.com/graphics/intel-graphics.key |
+          sudo apt-key add -
+        sudo add-apt-repository \
+          'deb [arch=amd64] https://repositories.intel.com/graphics/ubuntu focal main'
+        sudo apt-get update
+        sudo apt-get install \
+          intel-opencl-icd \
+          intel-level-zero-gpu level-zero \
+          intel-media-va-driver-non-free libmfx1
+        sudo apt-get install ocl-icd-opencl-dev
     - uses: actions/checkout@v3
     - name: Build
       run: cargo build --verbose
-    
+
   test:
     runs-on: ubuntu-latest
-    
+
     steps:
+    - name: Load OpenCL
+      run: |
+        wget -qO - https://repositories.intel.com/graphics/intel-graphics.key |
+          sudo apt-key add -
+        sudo add-apt-repository \
+          'deb [arch=amd64] https://repositories.intel.com/graphics/ubuntu focal main'
+        sudo apt-get update
+        sudo apt-get install \
+          intel-opencl-icd \
+          intel-level-zero-gpu level-zero \
+          intel-media-va-driver-non-free libmfx1
+        sudo apt-get install ocl-icd-opencl-dev
     - uses: actions/checkout@v3
     - name: Run tests
       run: cargo test --verbose

Cargo.toml

@@ -11,4 +11,5 @@ rpassword = "7.2"
 tokio = { version = "1.28.2", features = ["full"] }
 pwned = { git = "https://github.com/wisespace-io/pwned-rs.git" }
 parselnk = "0.1.1"
-regex = "1.9.5"
+regex = "1.9.5"
+opencl3 = "0.9.3"

src/gpu/mod.rs

@@ -0,0 +1,49 @@
+#[allow(non_snake_case)]
+
+pub mod gpu { 
+    use opencl3::device::{get_all_devices, CL_DEVICE_TYPE_GPU};
+    use std::io;
+    pub fn obtainGPU() -> Result<(), ()> {
+        let devices: Vec<*mut std::ffi::c_void> = get_all_devices(CL_DEVICE_TYPE_GPU).expect("Cannot detect a GPU!");
+        let mut gpu_choice: u32 = 0;
+
+        if devices.len() > 1 {
+            loop {
+                println!("You have multiple ({}) GPUs, please select one!", devices.len());
+                for gpu in 0..devices.len() {
+                    let device = devices[gpu];
+                    let device = opencl3::device::Device::new(device);
+                    println!("\tGPU Device {}: {}", gpu, device.name().unwrap());
+                }
+                print!("Selection: ");
+
+                let mut tree = String::new();
+                io::stdin().read_line(&mut tree).expect("Failed to read line");
+                let water = tree.trim();
+
+                match water.parse::<u32>() {
+                    Ok(_) => {
+                        gpu_choice = water.parse().unwrap();
+                        if gpu_choice > devices.len() as u32 {
+                            panic!("Please select a valid GPU!")
+                        }
+                        break;
+                    }
+                    Err(_) => {
+                        println!("Please type a number!");
+                    }
+                }
+            }
+        }
+        println!("You have selected GPU Device {}", gpu_choice);
+        let gpu = opencl3::device::Device::new(devices[gpu_choice as usize]);
+
+        println!("Your {} has {} CUDA cores and {} stream multiprocessors", gpu.name().unwrap(), gpu.max_compute_units().unwrap(), (gpu.max_compute_units().unwrap() * 8));
+        //println!("Device: {:?}", device.name());
+        //println!("Device: {:?}", device.vendor());
+        //println!("Device: {:?}", device.version());
+        //println!("Device: {:?}", device.driver_version());
+        //println!("MAx compute units: {:?}", device.max_compute_units());
+        Ok(())
+    }
+}

src/main.rs

@@ -3,10 +3,14 @@
 
 mod tests;
 mod utils;
+mod gpu;
+
+use utils::pwned_api::pass_check;
+use gpu::gpu::obtainGPU;
+
 use std::env;
 use std::fs::File;
 use std::path::Path;
-use utils::pwned_api::pass_check;
 use round::round;
 use std::fs;
 use std::io::{BufRead, BufReader};
@@ -28,6 +32,7 @@ pub async fn main() {
     let pool_size = get_pool_size(password.clone());
     let entropy = calculate_entropy(pool_size); // calls functions
     let alphabet_match = regex_match(password.clone());
+    obtainGPU(); // todo
 
     check_if_pwned(password.clone()).await;
 
@@ -172,7 +177,7 @@ async fn password_list(password: String) -> Result<bool, ()> {
                             tokio::time::sleep(Duration::from_secs(1)).await;
                         }
                     }
-                    if line >= 2459760 {
+                    if line >= 2459760 { // if it's at the end of the file, stop the search and return password not found
                         return Ok(false);
                     }
                 }