fix: sign generate

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

View File

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

View File

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