coollsd commited on
Commit
a5feba9
·
verified ·
1 Parent(s): 9ad417d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -9
app.py CHANGED
@@ -885,9 +885,10 @@ async def handle_video_stream(path: str, request: Request):
885
  async def handle_encrypted_stream(encrypted_data: str, request: Request):
886
  try:
887
  browser_info = request.headers.get('User-Agent', '') + request.headers.get('Accept-Language', '') + str(request.client.host)
888
- key = hashlib.sha256(browser_info.encode()).hexdigest()[:32]
 
889
 
890
- decrypted_url = decrypt_url(encrypted_data, key)
891
 
892
  if not decrypted_url.startswith('/rbxg/'):
893
  raise ValueError("Invalid decrypted URL")
@@ -993,13 +994,6 @@ async def retry_upload(upload_url: str, file_content: bytes, content_type: str,
993
 
994
  return False
995
 
996
- def decrypt_url(encrypted_data: str, key: str) -> str:
997
- try:
998
- decrypted = base64.b64decode(encrypted_data).decode('utf-8')
999
- return decrypted
1000
- except:
1001
- raise ValueError("Invalid encrypted data")
1002
-
1003
  if __name__ == "__main__":
1004
  import uvicorn
1005
  uvicorn.run(app, host="0.0.0.0", port=7860)
 
885
  async def handle_encrypted_stream(encrypted_data: str, request: Request):
886
  try:
887
  browser_info = request.headers.get('User-Agent', '') + request.headers.get('Accept-Language', '') + str(request.client.host)
888
+ key = hashlib.sha256(browser_info.encode()).hexdigest()[:32].encode()
889
+ fernet = Fernet(base64.urlsafe_b64encode(key))
890
 
891
+ decrypted_url = fernet.decrypt(encrypted_data.encode()).decode()
892
 
893
  if not decrypted_url.startswith('/rbxg/'):
894
  raise ValueError("Invalid decrypted URL")
 
994
 
995
  return False
996
 
 
 
 
 
 
 
 
997
  if __name__ == "__main__":
998
  import uvicorn
999
  uvicorn.run(app, host="0.0.0.0", port=7860)