shiroyashik/sculptor
1
0
Fork 0
mirror of https://github.com/shiroyashik/sculptor.git synced 2025-12-06 04:51:13 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add check for max_avatar_size

Browse source
This commit is contained in:
Shiroyasha 2024-06-10 07:23:53 +03:00
parent 83533919f3
commit 34905701a0
Signed by: shiroyashik
GPG key ID: E4953D3940D7860A
1 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/main.rs
View file

@ -1,4 +1,4 @@
use anyhow::Result; use anyhow::{anyhow, Result};
use axum::{ use axum::{
extract::DefaultBodyLimit, middleware::from_extractor, routing::{delete, get, post, put}, Router extract::DefaultBodyLimit, middleware::from_extractor, routing::{delete, get, post, put}, Router
}; };
@ -6,7 +6,7 @@ use dashmap::DashMap;
use std::sync::Arc; use std::sync::Arc;
use tokio::sync::{broadcast, Mutex}; use tokio::sync::{broadcast, Mutex};
use tower_http::trace::TraceLayer; use tower_http::trace::TraceLayer;
use tracing::info; use tracing::{error, info};
use uuid::Uuid; use uuid::Uuid;
// WebSocket worker // WebSocket worker
@ -138,7 +138,14 @@ async fn main() -> Result<()> {
} }
}); });
let max_body_size = state.config.clone().lock().await.limitations.max_avatar_size as usize; let max_body_size = {
let mbs = state.config.clone().lock().await.limitations.max_avatar_size as usize;
if mbs >= 1024 {
mbs
} else {
return Err(anyhow!("maxAvatarSize {mbs} smaller than 1024!"));
}
};
let api = Router::new() let api = Router::new()
.nest("//auth", api_auth::router()) .nest("//auth", api_auth::router())
.route("/limits", get(api_info::limits)) .route("/limits", get(api_info::limits))