python base64 crypto
from Crypto.Cipher import AES
from binascii import b2a_hex, a2b_hex
import JSON
import urllib2
import time
import random, string
import base64
import argparse
import sys
import os
import datetime
class prpcrypt():
def __init__(self, key, iv):
self.key = key
self.iv = iv
self.mode = AES.MODE_CBC
self.BS = AES.block_size
# 补位
self.pad = lambda s: s + (self.BS - len(s) % self.BS) * chr(self.BS - len(s) % self.BS)
self.unpad = lambda s: s[0:-ord(s[-1])]
def encrypt(self, text):
a = (self.BS - len(text) % self.BS)
b = chr(self.BS - len(text) % self.BS)
text = self.pad(text)
cryptor = AES.new(self.key, self.mode, self.iv)
# 目前AES-128 足够目前使用
ciphertext = cryptor.encrypt(text)
# 把加密后的字符串使用base64编码
return base64.b64encode(ciphertext)
# 解密后,去掉补足的空格用strip() 去掉
def decrypt(self, text):
text = base64.b64decode(text)
cryptor = AES.new(self.key, self.mode, self.iv)
plain_text = cryptor.decrypt(text)
return self.unpad(plain_text.rstrip('\0'))
class mock():
def init(self, acct_id):
key_len = 30
listStr = ['Http://','cdit.waa.cn','/crit-w/serce']
self.url = ''.join(listStr)
self.headers = {'Content-type': 'application/json', 'X_WANDA_ACCT_ID': acct_id,
'Cache-Control': 'no-cache'}
self.keylist = [random.choice(string.letters + string.digits) for i in range(key_len)]
self.reqsn = "".join(self.keylist)
self.reQtime = str(int(time.time())) + '000'
def fORMat(self, acct_id, acct_data):
data = {"req_time": self.reqtime, "acct_id": acct_id, "inf_id": "P_C_B016", "prod_id": "P_C_B016",
"request_sn": self.reqsn, "req_data": acct_data}
return data
def send(self, data):
req = urllib2.Request(self.url, data, self.headers)
try:
f = urllib2.urlopen(req)
except urllib2.HTTPError, e:
#print e.code
#print e.read()
return None
else:
return f.read()
start_time = datetime.datetime.now()
iv = "0000000000000000"
key = '5d8f090cb6196248145266334a'
acct_id = 'CRT_TET_ACCTID'
acct_data =
key = a2b_hex(key)
req_data = mock(acct_id).format(acct_id, acct_data)
req_data = json.dumps(req_data)
#print "请求原文:", req_data
req_data = prpcrypt(key, iv).encrypt(req_data)
#print "请求密文:", req_data
respon_data = mock(acct_id).send(req_data)
#print "响应密文:", respon_data
#if respon_data != None:
respon_data = prpcrypt(key, iv).decrypt(respon_data)
respon_data = json.loads(respon_data)
retcode = respon_data["retcode"]
end_time = datetime.datetime.now()
#print end_time
result_end = str(end_time - start_time)
#print result_end
response_sn = respon_data['response_sn']
if name == 'main':
iv = "0000000000000000"
key = '5d8f090cb6196245266334a'
acct_id = 'CRIT_TET_ACCTID'
acct_data = {"name": "周是", "cardNo": "51222219120017", "idType": "101", "reasonCode": "01","mobile": "13924623448", "qq": "544564567@qq.com", "wechat": "dgfh0890","weibo": "dgkhs0120@sina.com"}
key = a2b_hex(key)
req_data = mock(acct_id).format(acct_id, acct_data)
req_sn = req_data['request_sn']
acct_id = req_data['acct_id']
req_data = json.dumps(req_data)
#print "请求原文:", req_data
req_data = prpcrypt(key, iv).encrypt(req_data)
#print "请求密文:", req_data
respon_data = mock(acct_id).send(req_data)
#print "响应密文:", respon_data
if respon_data != None:
respon_data = prpcrypt(key, iv).decrypt(respon_data)
respon_data = json.loads(respon_data)
#print "响应原文:", respon_data
if respon_data['retcode'] == '000000':
print "credit service running"
else:
print "fail"
else:
print "fail"
相关文章