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

Added request logging middleware

Browse Source
This commit is contained in:
Random936
2024-12-22 20:40:04 -08:00
parent ceb1ab44d3
commit 0b2a4f130e
5 changed files with 49 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/config.rs
View File

@@ -31,6 +31,14 @@ impl Config {
serde_yaml::from_reader(reader).unwrap()
}
pub fn get_download_path(&self) -> PathBuf {
self.download_path.clone()
}
pub fn get_upload_path(&self) -> PathBuf {
self.upload_path.clone()
}
pub fn get_shell<S: Into<String>>(&self, key: S) -> Option<String> {
let key_str = key.into();
self.shells.get(&key_str).cloned()

17
src/logging.rs Normal file
View File

@@ -0,0 +1,17 @@
use std::net::SocketAddr;
use axum::{
extract::ConnectInfo,
response::Response,
middleware::Next,
http::Request,
body::Body
};
pub async fn logging_middleware(
ConnectInfo(addr): ConnectInfo<SocketAddr>,
req: Request<Body>,
next: Next
) -> Response {
println!("[{}] {} {}", addr, req.method(), req.uri());
next.run(req).await
}

24
src/main.rs
View File

@@ -1,14 +1,20 @@
use std::{
env,
net::SocketAddr
};
use tokio::net::TcpListener;
use tower_http::services::ServeDir;
use axum::routing::{get, post};
use axum::Router;
use std::env;
use std::net::SocketAddr;
use axum::{
middleware,
routing::get,
Router
};
mod argparse;
mod print_dir;
mod config;
mod shells;
mod logging;
#[tokio::main]
async fn main() {
@@ -24,14 +30,18 @@ async fn main() {
print_dir::print_interface(&args.interface, &port, &args.directory);
let app = Router::new()
//.route("/download/:path", get(download_handler))
//.route("/upload", post(upload_handler))
.route("/shells/:shell", get(shells::shells_handler))
.nest_service("/", ServeDir::new(cwd));
.nest_service("/download", ServeDir::new(conf.get_download_path()))
.nest_service("/", ServeDir::new(cwd))
.layer(middleware::from_fn(logging::logging_middleware));
let listener = TcpListener::bind(format!("0.0.0.0:{}", port))
.await
.unwrap();
axum::serve(listener, app).await.unwrap();
axum::serve(
listener,
app.into_make_service_with_connect_info::<SocketAddr>()
).await.unwrap();
}