hehesheng/TgToFileSystem
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: sign generate

Browse Source
This commit is contained in:
hehesheng 2024-06-18 22:52:26 +08:00
parent ff997c7434
commit f74a1bfa95
3 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/TgFileSystemClientManager.py
View File

@ -113,10 +113,14 @@ class TgFileSystemClientManager(object):
need_encrypt_bin = need_encrypt_str.encode() need_encrypt_bin = need_encrypt_str.encode()
sign_bin = rsa.encrypt(need_encrypt_bin, self.public_key) sign_bin = rsa.encrypt(need_encrypt_bin, self.public_key)
sign = base64.b64encode(sign_bin).decode() sign = base64.b64encode(sign_bin).decode()
sign = sign.replace("+", "-")
sign = sign.replace("/", "_")
logger.info(f"generate {sign_type.name} sign: {sign}") logger.info(f"generate {sign_type.name} sign: {sign}")
return sign return sign
def parse_sign(self, sign: str) -> dict[str, any] | None: def parse_sign(self, sign: str) -> dict[str, any] | None:
sign = sign.replace("-", "+")
sign = sign.replace("_", "/")
try: try:
res_dict = {} res_dict = {}
sign_bin = base64.b64decode(sign) sign_bin = base64.b64decode(sign)

1
backend/api.py
View File

@ -59,7 +59,6 @@ async def verify_post_sign(body: TgToFileListRequestBody):
async def verify_get_sign(sign: str): async def verify_get_sign(sign: str):
clients_mgr = TgFileSystemClientManager.get_instance() clients_mgr = TgFileSystemClientManager.get_instance()
sign = sign.replace(" ", "+")
if not clients_mgr.verify_sign(sign): if not clients_mgr.verify_sign(sign):
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=f"{sign}") raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=f"{sign}")
return sign return sign