jmug/stoatchat
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

fix: reimplement max redirects

Browse Source
This commit is contained in:
IAmTomahawkx
2026-05-06 18:01:44 -07:00
parent f0a6ed78fe
commit 1feec707c1
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
crates/services/january/src/requests.rs
View File

@@ -266,6 +266,7 @@ impl Request {
let url_host_str = url.host_str().ok_or(create_error!(ProxyError))?.to_string(); let url_host_str = url.host_str().ok_or(create_error!(ProxyError))?.to_string();
Request::url_is_blacklisted(&url).await?; Request::url_is_blacklisted(&url).await?;
let mut redirect_count = 0;
loop { loop {
let response = CLIENT let response = CLIENT
@@ -284,6 +285,11 @@ impl Request {
.map_err(|_| create_error!(ProxyError))?; .map_err(|_| create_error!(ProxyError))?;
if response.status().is_redirection() { if response.status().is_redirection() {
redirect_count += 1;
if redirect_count > 5 {
return Err(create_error!(ProxyError));
}
if let Some(location) = response.headers().get("location") { if let Some(location) = response.headers().get("location") {
let location = location.to_str().map_err(|_| create_error!(ProxyError))?; let location = location.to_str().map_err(|_| create_error!(ProxyError))?;
url = Url::from_str(location).to_internal_error()?; url = Url::from_str(location).to_internal_error()?;