QingdaoU/OnlineJudge
1
0
Fork 0
mirror of https://github.com/QingdaoU/OnlineJudge.git synced 2025-01-16 01:13:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

auto generate client token

Browse Source
This commit is contained in:
virusdefender 2017-01-26 13:55:31 +08:00
parent 27b2c92d8d
commit 16616ed217
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
conf/views.py
View File

@ -96,11 +96,13 @@ class JudgeServerHeartbeatAPI(CSRFExemptAPIView):
def post(self, request): def post(self, request):
judge_server_token = JudgeServerToken.objects.first() judge_server_token = JudgeServerToken.objects.first()
if not judge_server_token: if not judge_server_token:
return self.error("Web server token not set") token = rand_str(12)
token = judge_server_token.token JudgeServerToken.objects.create(token=token)
else:
token = judge_server_token.token
data = request.data data = request.data
judge_server_token = request.META.get("HTTP_X_JUDGE_SERVER_TOKEN") client_token = request.META.get("HTTP_X_JUDGE_SERVER_TOKEN")
if hashlib.sha256(token.encode("utf-8")).hexdigest() != judge_server_token: if hashlib.sha256(token.encode("utf-8")).hexdigest() != client_token:
return self.error("Invalid token") return self.error("Invalid token")
service_url = data.get("service_url") service_url = data.get("service_url")