FFI
This commit is contained in:
randogoth
parent
7ed1e5e4e6
commit
bdf3e8dde5
4 changed files with 86 additions and 150 deletions
|
|
@ -6,8 +6,6 @@ edition = "2021"
|
|||
[lib]
|
||||
name = "ocelli"
|
||||
path = "src/lib.rs"
|
||||
crate-type = ["cdylib"]
|
||||
|
||||
[dependencies]
|
||||
v4l = "0.14"
|
||||
chrono = "0.4"
|
||||
image = "0.25.5"
|
||||
|
|
|
|||
4
justfile
4
justfile
|
|
@ -1,2 +1,6 @@
|
|||
diehard file:
|
||||
dieharder -a -g 201 -f {{file}}
|
||||
|
||||
build arg:
|
||||
cargo build {{arg}}
|
||||
cargo ndk -t arm64-v8a -t armeabi-v7a -t x86_64 build {{arg}}
|
||||
|
|
|
|||
146
src/bin/main.rs
146
src/bin/main.rs
|
|
@ -1,146 +0,0 @@
|
|||
use ocelli::Ocelli;
|
||||
use v4l::prelude::*;
|
||||
use v4l::video::Capture;
|
||||
use v4l::buffer::Type;
|
||||
use v4l::format::FourCC;
|
||||
use v4l::io::traits::CaptureStream;
|
||||
use image::ImageReader;
|
||||
use std::fs::File;
|
||||
use std::io::{stdin, Write};
|
||||
use chrono::Local;
|
||||
use std::time::Instant;
|
||||
|
||||
fn frame_to_grayscale(data: &[u8]) -> Vec<u8> {
|
||||
let img = ImageReader::new(std::io::Cursor::new(data))
|
||||
.with_guessed_format()
|
||||
.expect("Failed to guess format")
|
||||
.decode()
|
||||
.expect("Failed to decode image");
|
||||
let gray = img.into_luma8(); // Convert to grayscale
|
||||
gray.into_raw() // Return raw pixel data as Vec<u8>
|
||||
}
|
||||
|
||||
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
let args: Vec<String> = std::env::args().collect();
|
||||
if args.len() < 3 {
|
||||
eprintln!("Usage: {} <camera index> <entropy length in bytes>", args[0]);
|
||||
std::process::exit(1);
|
||||
}
|
||||
|
||||
// Check if the quick flag is set
|
||||
let quick = args.contains(&String::from("-q"));
|
||||
|
||||
let camera_index: usize = args[1].parse().expect("Failed to parse camera index as a number");
|
||||
let length: usize = args[2].parse().expect("Failed to parse entropy length as a number");
|
||||
|
||||
let dev = Device::new(camera_index).expect("Failed to open camera");
|
||||
|
||||
// Set the desired format and resolution
|
||||
let mut format = dev.format().expect("Failed to get camera format");
|
||||
format.fourcc = FourCC::new(b"MJPG"); // Use MJPG for higher resolutions
|
||||
format.width = 1920;
|
||||
format.height = 1080;
|
||||
|
||||
if let Err(_) = dev.set_format(&format) {
|
||||
println!("Failed to set resolution to 1920x1080. Falling back to 1280x720.");
|
||||
format.width = 1280;
|
||||
format.height = 720;
|
||||
dev.set_format(&format)
|
||||
.expect("Failed to set resolution to 1280x720");
|
||||
}
|
||||
|
||||
println!(
|
||||
"Using resolution: {}x{} (FourCC: {})",
|
||||
format.width, format.height, format.fourcc
|
||||
);
|
||||
|
||||
let mut stream = MmapStream::with_buffers(&dev, Type::VideoCapture, 4)
|
||||
.expect("Failed to create stream");
|
||||
|
||||
let ocelli = Ocelli;
|
||||
let mut total_entropy = Vec::new();
|
||||
let start_time = Instant::now();
|
||||
let shannon_threshold = 7.9;
|
||||
let mut frame_count = 0;
|
||||
|
||||
while total_entropy.len() < length {
|
||||
// Capture first frame
|
||||
let (data1, _) = stream.next().expect("Failed to capture frame");
|
||||
|
||||
// Skip the first 30 frames
|
||||
if frame_count <= 30 {
|
||||
frame_count += 1;
|
||||
} else {
|
||||
|
||||
let grayscale_data1 = frame_to_grayscale(&data1);
|
||||
|
||||
let mut entropy: Vec<u8> = [0].to_vec();
|
||||
|
||||
if quick {
|
||||
// Quicker capture using Pick and Flip
|
||||
entropy = ocelli.whiten(&ocelli.pick_and_flip(&grayscale_data1, frame_count as usize));
|
||||
} else {
|
||||
if ocelli.is_covered(&grayscale_data1, 50) {
|
||||
// Capture second frame
|
||||
let (data2, _) = stream.next().expect("Failed to capture second frame");
|
||||
let grayscale_data2 = frame_to_grayscale(&data2);
|
||||
|
||||
// Generate entropy using chop_and_tack
|
||||
entropy = ocelli.chop_and_tack(&grayscale_data1, &grayscale_data2, format.width as usize, 30);
|
||||
} else {
|
||||
println!("Camera is not covered. Please cover the camera.");
|
||||
frame_count = 0;
|
||||
}
|
||||
}
|
||||
|
||||
let shannon_entropy = ocelli.shannon(&entropy);
|
||||
|
||||
if shannon_entropy >= shannon_threshold {
|
||||
total_entropy.extend(entropy);
|
||||
println!(
|
||||
"Collected {} of {} bytes of entropy (Shannon entropy: {:.3})",
|
||||
total_entropy.len(),
|
||||
length,
|
||||
shannon_entropy
|
||||
);
|
||||
} else {
|
||||
println!("Rejected entropy for frame {} (Shannon entropy: {:.3})", frame_count, shannon_entropy);
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
// Trim total_entropy to the exact length
|
||||
total_entropy.truncate(length);
|
||||
|
||||
// Print elapsed time
|
||||
let elapsed_time = start_time.elapsed();
|
||||
println!(
|
||||
"Process completed in {:.3} seconds.",
|
||||
elapsed_time.as_secs_f64()
|
||||
);
|
||||
|
||||
// Final Shannon entropy test
|
||||
let final_shannon_entropy = ocelli.shannon(&total_entropy);
|
||||
println!("Final Shannon entropy: {:.3}", final_shannon_entropy);
|
||||
|
||||
// Ask if the result should be saved
|
||||
println!("Save result to a file or print as hex string? (file/print):");
|
||||
let mut input = String::new();
|
||||
stdin().read_line(&mut input).expect("Failed to read input");
|
||||
|
||||
if input.trim().eq_ignore_ascii_case("file") {
|
||||
let timestamp = Local::now().format("%Y%m%d_%H%M%S").to_string();
|
||||
let filename = format!("entropy_{}.bin", timestamp);
|
||||
|
||||
let mut file = File::create(&filename).expect("Failed to create file");
|
||||
file.write_all(&total_entropy).expect("Failed to write data to file");
|
||||
|
||||
println!("Entropy saved to file: {}", filename);
|
||||
} else if input.trim().eq_ignore_ascii_case("print") {
|
||||
let entropy_hex: String = total_entropy.iter().map(|b| format!("{:02x}", b)).collect();
|
||||
println!("Generated entropy (hex): {}", entropy_hex);
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
80
src/lib.rs
80
src/lib.rs
|
|
@ -1,4 +1,5 @@
|
|||
use std::collections::{HashMap, HashSet};
|
||||
use std::slice;
|
||||
|
||||
pub struct Ocelli;
|
||||
|
||||
|
|
@ -135,3 +136,82 @@ impl Ocelli {
|
|||
unique_values.len() < threshold
|
||||
}
|
||||
}
|
||||
|
||||
#[no_mangle]
|
||||
pub extern "C" fn chop_and_tack(
|
||||
current_ptr: *const u8,
|
||||
current_len: usize,
|
||||
previous_ptr: *const u8,
|
||||
previous_len: usize,
|
||||
width: usize,
|
||||
minimum_distance: usize,
|
||||
result_ptr: *mut u8,
|
||||
result_len: &mut usize,
|
||||
) {
|
||||
let current = unsafe { slice::from_raw_parts(current_ptr, current_len) };
|
||||
let previous = unsafe { slice::from_raw_parts(previous_ptr, previous_len) };
|
||||
|
||||
let ocelli = Ocelli;
|
||||
let result = ocelli.chop_and_tack(¤t.to_vec(), &previous.to_vec(), width, minimum_distance);
|
||||
|
||||
unsafe {
|
||||
let result_slice = slice::from_raw_parts_mut(result_ptr, result.len());
|
||||
result_slice.copy_from_slice(&result);
|
||||
*result_len = result.len();
|
||||
}
|
||||
}
|
||||
|
||||
#[no_mangle]
|
||||
pub extern "C" fn pick_and_flip(
|
||||
data_ptr: *const u8,
|
||||
data_len: usize,
|
||||
current_frame_index: usize,
|
||||
result_ptr: *mut u8,
|
||||
result_len: &mut usize,
|
||||
) {
|
||||
let data = unsafe { slice::from_raw_parts(data_ptr, data_len) };
|
||||
|
||||
let ocelli = Ocelli;
|
||||
let result = ocelli.pick_and_flip(data, current_frame_index);
|
||||
|
||||
unsafe {
|
||||
let result_slice = slice::from_raw_parts_mut(result_ptr, result.len());
|
||||
result_slice.copy_from_slice(&result);
|
||||
*result_len = result.len();
|
||||
}
|
||||
}
|
||||
|
||||
#[no_mangle]
|
||||
pub extern "C" fn shannon(data_ptr: *const u8, data_len: usize) -> f64 {
|
||||
let data = unsafe { slice::from_raw_parts(data_ptr, data_len) };
|
||||
|
||||
let ocelli = Ocelli;
|
||||
ocelli.shannon(&data.to_vec())
|
||||
}
|
||||
|
||||
#[no_mangle]
|
||||
pub extern "C" fn whiten(
|
||||
entropy_ptr: *const u8,
|
||||
entropy_len: usize,
|
||||
result_ptr: *mut u8,
|
||||
result_len: &mut usize,
|
||||
) {
|
||||
let entropy = unsafe { slice::from_raw_parts(entropy_ptr, entropy_len) };
|
||||
|
||||
let ocelli = Ocelli;
|
||||
let result = ocelli.whiten(entropy);
|
||||
|
||||
unsafe {
|
||||
let result_slice = slice::from_raw_parts_mut(result_ptr, result.len());
|
||||
result_slice.copy_from_slice(&result);
|
||||
*result_len = result.len();
|
||||
}
|
||||
}
|
||||
|
||||
#[no_mangle]
|
||||
pub extern "C" fn is_covered(grayscale_ptr: *const u8, grayscale_len: usize, threshold: usize) -> bool {
|
||||
let grayscale = unsafe { slice::from_raw_parts(grayscale_ptr, grayscale_len) };
|
||||
|
||||
let ocelli = Ocelli;
|
||||
ocelli.is_covered(grayscale, threshold)
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue