random936/up
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added /shells route (wip)

Browse Source
This commit is contained in:
Random936 2024-12-22 18:48:08 -08:00
parent 4c4b3bc202
commit 8dc4b91ba2
1 changed files with 29 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/shells.rs
View File

@ -0,0 +1,29 @@
use std::net::SocketAddr;
use axum::{
body::Body,
http::{Request, StatusCode},
response::IntoResponse
};
use crate::config;
fn replace_shell_params(cmd: &str, lhost: &str, lport: &u16) -> String {
cmd.replace("{LHOST}", lhost)
.replace("{LPORT}", &lport.to_string())
}
pub async fn shells_handler(req: Request<Body>) -> impl IntoResponse {
let conf = config::load_config();
let Some(host) = req.headers()
.get("host")
.and_then(|v| v.to_str().ok()) else {
return (StatusCode::BAD_REQUEST, "Unable to determine LHOST.".to_string());
};
let ip = host.split(":").next().unwrap();
match conf.get_shell("bash") {
Some(res) => (StatusCode::OK, replace_shell_params(&res, ip, &conf.shell_port)),
None => (StatusCode::NOT_FOUND, "Invalid shell name.".to_string()),
}
}