This commit is contained in:
Carneiro committed 2023-10-04 23:08:52 -03:00
1 parent 122ba38beb
commit 2c9224cfeb
2 files changed
+6 -4

No files matched your search

+4 -3
View File
@@ -25,10 +25,11 @@ def create_token(username: str, stream_key: str, exp: int = settings.JWT_EXP_TIM
def verify(token: str): def verify(token: str):
try:
jwt.decode(token, JWT_PUB_KEY, algorithms=["RS256"]) jwt.decode(token, JWT_PUB_KEY, algorithms=["RS256"])
except (jwt.exceptions.ExpiredSignatureError, jwt.InvalidTokenError): # try:
return False # jwt.decode(token, JWT_PUB_KEY, algorithms=["RS256"])
# except (jwt.exceptions.ExpiredSignatureError, jwt.InvalidTokenError):
# return False
return True return True
+2 -1
View File
@@ -32,11 +32,12 @@ def publish_check():
stream_key = request.form.get('stream_key') stream_key = request.form.get('stream_key')
username = request.form.get('name') username = request.form.get('name')
stream_user = user.search_stream_key(stream_key)[0] stream_user = user.search_stream_key(stream_key)[0]
if username != stream_user['username'] or jwt.verify(stream_key.encode('utf-8')): if username != stream_user['username']:
raise ValueError raise ValueError
except (IndexError, ValueError): except (IndexError, ValueError):
print('vish') print('vish')
return Response('Invalid Stream Key', 401) return Response('Invalid Stream Key', 401)
jwt.verify(stream_key.encode('utf-8'))
return Response('OK', 200) return Response('OK', 200)