SSRF Fix Updated

This commit is contained in:
root 2024-08-07 10:59:22 -04:00
parent 1f8d08eaa2
commit 590fd129c8

View File

@ -150,8 +150,7 @@ async def update_engine_url(
else:
url = form_data.AUTOMATIC1111_BASE_URL.strip("/")
try:
r = requests.head(url)
r.raise_for_status()
r = requests.head(url)
app.state.config.AUTOMATIC1111_BASE_URL = url
except Exception as e:
raise HTTPException(status_code=400, detail="Invalid URL provided.")