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

Made shell dynamic for /shells route

Browse Source
This commit is contained in:
Random936 2024-12-22 19:29:45 -08:00
parent a3b49ed7fc
commit ceb1ab44d3
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/shells.rs
View File

@ -1,12 +1,13 @@
use axum::{ use axum::{
body::Body, body::Body,
extract::Path,
http::{Request, StatusCode}, http::{Request, StatusCode},
response::IntoResponse response::IntoResponse
}; };
use crate::config; use crate::config;
pub async fn shells_handler(req: Request<Body>) -> impl IntoResponse { pub async fn shells_handler(Path(shell): Path<String>, req: Request<Body>) -> impl IntoResponse {
let conf = config::load_config(); let conf = config::load_config();
let Some(host) = req.headers() let Some(host) = req.headers()
.get("host") .get("host")
@ -15,7 +16,7 @@ pub async fn shells_handler(req: Request<Body>) -> impl IntoResponse {
}; };
let ip = host.split(":").next().unwrap(); let ip = host.split(":").next().unwrap();
match conf.get_shell("bash") { match conf.get_shell(&shell) {
Some(res) => { Some(res) => {
let cmd = res.replace("{LHOST}", ip) let cmd = res.replace("{LHOST}", ip)
.replace("{LPORT}", &conf.shell_port.to_string()); .replace("{LPORT}", &conf.shell_port.to_string());