import datetime
import random
import requests
import hashlib
import json
import base64
from Crypto.PublicKey import RSA
from Crypto.Signature import PKCS1_v1_5
from Crypto.Hash import SHA256
from Crypto.Cipher import AES

def sign(signflag,keypath,baseRequest):
 #http請(qǐng)求body
  print(baseRequest)
  #加簽標(biāo)志
  if not signflag: return baseRequest
  else:
   #取請(qǐng)求體中的業(yè)務(wù)數(shù)據(jù)
    businessdata = json.dumps(baseRequest["data"])
    #讀取私鑰（.key格式，可使用openssl或java.keytools產(chǎn)生）
    with open(keypath,'r') as rsaKeyFile:
      rsaKey = rsaKeyFile.read().replace("\n",'')
      print(rsaKey)
    rsaKeyBytes = base64.b64decode(rsaKey)
    print(rsaKeyBytes)
    #SHA256摘要，RSA加密
    priKey = RSA.importKey(rsaKeyBytes)
    signer = PKCS1_v1_5.new(priKey)
    hash_obj = SHA256.new(business_data.encode('utf-8'))
    signature = base64.b64encode(signer.sign(hash_obj))
    print(signature)
    #把簽名加進(jìn)請(qǐng)求體并返回
    baseRequest['sign'] = signature.decode()
    print(baseRequest)
    return baseRequest

def validata(signflag,cerpath,res):
  if not signflag: return res
  else:
   #取業(yè)務(wù)數(shù)據(jù)和簽名
    data = res['data']
    sign = res['sign']
    #此處cer已轉(zhuǎn)換成pem格式，使用openssl工具
    #openssl x509 -inform der -pubkey -noout -in xxxxx.cer>xxxxx.pem
    cert = open(cerpath).read().replace("-----BEGIN PUBLIC KEY-----\n","").replace("-----END PUBLIC KEY-----\n","").replace("\n","")
    print(cert)
 #驗(yàn)簽邏輯同加簽
    pubBytes = base64.b64decode(cert)
    pubKey = RSA.importKey(pubBytes)
    signer = SHA256.new(json.dumps(data).encode("utf-8"))
    verifier = PKCS1_v1_5.new(pubKey)
    return verifier.verify(signer,base64.b64decode(sign))