gongor/gongor.rs

56 lines
2.0 KiB
Rust
Raw Normal View History

2023-11-29 17:28:35 -08:00
use std::io::BufReader;
2023-11-29 16:42:35 -08:00
use std::fs::File;
use std::io::prelude::*;
2023-11-29 16:54:10 -08:00
use rand::rngs::OsRng;
use ed25519_dalek::SigningKey;
2023-11-29 17:28:35 -08:00
use ed25519_dalek::Signature;
use ed25519_dalek::Signer;
2023-11-29 16:54:10 -08:00
2023-11-29 16:42:35 -08:00
fn main() -> std::io::Result<()>{
2023-11-29 17:28:35 -08:00
let action = std::env::args().nth(1).expect("no command given");
if action == "sign" {
let keyfilepath = std::env::args().nth(2).expect("no keyfilepath given");
let messagefilepath = std::env::args().nth(3).expect("no message given");
let maybe_signature_filepath = std::env::args().nth(4);
2023-11-29 16:44:13 -08:00
2023-11-29 17:28:35 -08:00
let mut keyfile = File::open(&keyfilepath).expect("no file found");
let mut keybuffer: [u8; 32] = [0; 32];
keyfile.read(&mut keybuffer).expect("buffer overflow");
let signing_key: SigningKey = SigningKey::from_bytes(&keybuffer);
2023-11-29 16:54:10 -08:00
2023-11-29 17:28:35 -08:00
let messagefile= File::open(&messagefilepath).expect("no file found");
let mut messagereader = BufReader::new(messagefile);
let mut messagebuf = Vec::new();
messagereader.read_to_end(&mut messagebuf)?;
2023-11-29 17:06:38 -08:00
2023-11-29 17:28:35 -08:00
println!("Using key {keyfilepath}");
let signature: Signature = signing_key.sign(&messagebuf);
if maybe_signature_filepath.is_some() {
let filename = maybe_signature_filepath.expect("Not possible");
let mut file = File::create(filename.clone())?;
println!("Writing signature to {filename}");
file.write_all(&signature.to_bytes())?;
} else {
println!("Signture: {signature}");
}
2023-11-29 17:28:35 -08:00
return Ok(())
} else if action == "create" {
let keyfilepath = std::env::args().nth(2).expect("no keyfilepath given");
let mut csprng = OsRng;
let signing_key: SigningKey = SigningKey::generate(&mut csprng);
println!("Writing new key to {keyfilepath}");
let mut file = File::create(keyfilepath)?;
file.write_all(&signing_key.to_bytes())?;
return Ok(());
} else {
println!("Unrecognized command {action}");
return Ok(());
}
}