api/encrypt_views.py

@@ -15,11 +15,11 @@ from pywe_storage import RedisStorage

                 from account.models import UserInfo
              

                 from logs.models import MchInfoDecryptLogInfo, MchInfoEncryptLogInfo
              

                 from marketcode.models import MarketCodeInfo
              

                -from mch.models import ActivityInfo, BrandInfo, ModelInfo, ConsumeShotUnbindingInfo
              

                +from mch.models import ActivityInfo, BrandInfo, ConsumeShotUnbindingInfo, ModelInfo
              

                 from utils.algorithm.b64 import b64_decrypt, b64_encrypt
              

                 from utils.algorithm.caesar import caesar_decrypt, caesar_encrypt
              

                 from utils.algorithm.rsalg import rsa_decrypt, rsa_encrypt
              

                -from utils.error.errno_utils import MarketCodeStatusCode, UserStatusCode
              

                +from utils.error.errno_utils import CiphertextStatusCode, MarketCodeStatusCode, UserStatusCode
              

                 from utils.redis.connect import r
              

                 
              

                 
              
@@ -108,7 +108,10 @@ def encrypt(request):

                 def decrypt(request, v='v2'):
              

                     ciphertext = request.POST.get('ciphertext', '')
              

                 
              

                -    prefix, cipherlen, ciphertext = ciphertext.split('+', 2)
              

                +    try:
              

                +        prefix, cipherlen, ciphertext = ciphertext.split('+', 2)
              

                +    except Exception:
              

                +        return response(CiphertextStatusCode.CIPHERTEXT_INVALID)
              

                 
              

                     ciphertext = ciphertext[:int(cipherlen)]
              

                 
              

utils/error/errno_utils.py

@@ -197,6 +197,11 @@ class MarketCodeStatusCode(BaseStatusCode):

                     MARKET_CODE_NOT_FOUND = StatusCodeField(505001, 'Market Code Not Found', description=u'一物一码不存在')
              

                 
              

                 
              

                +class CiphertextStatusCode(BaseStatusCode):
              

                +    """ 加解密相关错误码 5050xx """
              

                +    CIPHERTEXT_INVALID = StatusCodeField(505011, 'Ciphertext Invalid', description=u'密文无效')
              

                +
              

                +
              

                 class MaintenancePointStatusCode(BaseStatusCode):
              

                     """ 维修点相关错误码 5060xx """
              

                     MAINTENACE_POINT_NOT_FOUND = StatusCodeField(506001, 'Maintenance Point Not Found', description=u'维修点不存在')