renil/lx-music-api-server
1
0
Fork 0
mirror of https://github.com/MeoProject/lx-music-api-server.git synced 2025-05-23 19:17:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: 存在nginx反代时报错的问题 #1

Browse Source
This commit is contained in:
helloplhm-qwq 2023-12-17 14:12:40 +08:00
parent 909abc3239
commit 630848f62e
No known key found for this signature in database
GPG Key ID: B7542212855B2C26
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
main.py
View File

@ -37,9 +37,11 @@ async def handle_before_request(app, handler):
async def handle_request(request): async def handle_request(request):
# nginx proxy header # nginx proxy header
if (request.headers.get("X-Real-IP")): if (request.headers.get("X-Real-IP")):
request.remote = request.headers.get("X-Real-IP") request.remote_addr = request.headers.get("X-Real-IP")
else:
request.remote_addr = request.remote
# check ip # check ip
if (config.check_ip_banned(request.remote)): if (config.check_ip_banned(request.remote_addr)):
return handleResult({"code": 1, "msg": "您的IP已被封禁", "data": None}, 403) return handleResult({"code": 1, "msg": "您的IP已被封禁", "data": None}, 403)
# check global rate limit # check global rate limit
if ( if (
@ -49,23 +51,23 @@ async def handle_before_request(app, handler):
): ):
return handleResult({"code": 5, "msg": "全局限速", "data": None}, 429) return handleResult({"code": 5, "msg": "全局限速", "data": None}, 429)
if ( if (
(time.time() - config.getRequestTime(request.remote)) (time.time() - config.getRequestTime(request.remote_addr))
< <
(config.read_config("security.rate_limit.ip")) (config.read_config("security.rate_limit.ip"))
): ):
return handleResult({"code": 5, "msg": "IP限速", "data": None}, 429) return handleResult({"code": 5, "msg": "IP限速", "data": None}, 429)
# update request time # update request time
config.updateRequestTime('global') config.updateRequestTime('global')
config.updateRequestTime(request.remote) config.updateRequestTime(request.remote_addr)
# check host # check host
if (config.read_config("security.allowed_host.enable")): if (config.read_config("security.allowed_host.enable")):
if request.remote_host.split(":")[0] not in config.read_config("security.allowed_host.list"): if request.remote_host.split(":")[0] not in config.read_config("security.allowed_host.list"):
if config.read_config("security.allowed_host.blacklist.enable"): if config.read_config("security.allowed_host.blacklist.enable"):
config.ban_ip(request.remote, int(config.read_config("security.allowed_host.blacklist.length"))) config.ban_ip(request.remote_addr, int(config.read_config("security.allowed_host.blacklist.length")))
return handleResult({'code': 6, 'msg': '未找到您所请求的资源', 'data': None}, 404) return handleResult({'code': 6, 'msg': '未找到您所请求的资源', 'data': None}, 404)
try: try:
resp = await handler(request) resp = await handler(request)
aiologger.info(f'{request.remote} - {request.method} "{request.path}", {resp.status}') aiologger.info(f'{request.remote_addr} - {request.method} "{request.path}", {resp.status}')
return resp return resp
except: except:
logger.error(traceback.format_exc()) logger.error(traceback.format_exc())