snow/nazrin
snow/nazrin
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

omyacid: properly detect unregistered IPs

Browse source
This commit is contained in:
snow flurry 2024-08-18 19:41:51 -07:00
parent 0cb3aea62e
commit 19a08abb52
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
omyacid/src/main.rs
View file

@ -99,7 +99,7 @@ async fn get_vendor_data(
// admin username from an unknown instance. // admin username from an unknown instance.
if let IpAddr::V4(ip) = addr.ip() { if let IpAddr::V4(ip) = addr.ip() {
match ctx.get_instance(ip).await { match ctx.get_instance(ip).await {
Ok(_) => { Ok(Some(_)) => {
let data = model::VendorData { let data = model::VendorData {
username: Some(&ctx.cfg().cloud.admin_user), username: Some(&ctx.cfg().cloud.admin_user),
}; };
@ -108,14 +108,14 @@ async fn get_vendor_data(
StatusCode::INTERNAL_SERVER_ERROR StatusCode::INTERNAL_SERVER_ERROR
}) })
} }
Ok(None) => {
tracing::warn!("Request from unregistered server {ip}");
Err(StatusCode::FORBIDDEN)
}
Err(err) => { Err(err) => {
tracing::error!("{err}"); tracing::error!("{err}");
Err(StatusCode::INTERNAL_SERVER_ERROR) Err(StatusCode::INTERNAL_SERVER_ERROR)
} }
_ => {
tracing::warn!("Request from unregistered server {ip}");
Err(StatusCode::FORBIDDEN)
}
} }
} else { } else {
Err(StatusCode::BAD_REQUEST) Err(StatusCode::BAD_REQUEST)