yaclty/LightMirrors
1
0
Fork 0
mirror of https://github.com/NoCLin/LightMirrors synced 2025-08-02 15:24:49 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: report content length as is for HEAD, accept more headers

Browse Source
This commit is contained in:
Anonymous 2024-07-04 21:21:33 +08:00
parent 2e1ae11ad4
commit 39a78dbc69
1 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/mirrorsrun/proxy/direct.py
View File

@ -67,7 +67,7 @@ async def direct_proxy(
async with httpx.AsyncClient() as client:
req_headers = request.headers.mutablecopy()
for key in req_headers.keys():
if key not in ["user-agent", "accept", "authorization"]:
if key in ["host"]:
del req_headers[key]
httpx_req: HttpxRequest = client.build_request(
@ -79,17 +79,15 @@ async def direct_proxy(
httpx_req = await pre_process_request(request, httpx_req, pre_process)
upstream_response = await client.send(
httpx_req, follow_redirects=follow_redirects
httpx_req,
follow_redirects=follow_redirects,
)
res_headers = upstream_response.headers
res_headers = {key: value for key, value in upstream_response.headers.items()}
res_headers.pop("content-length", None)
res_headers.pop("content-encoding", None)
logger.info(
f"proxy {request.url} to {target_url} {upstream_response.status_code}"
)
if request.method != "HEAD":
res_headers.pop("content-length", None)
res_headers.pop("content-encoding", None)
content = upstream_response.content
response = Response(