应用场景
cmdb 这类项目的资产入库等操作,当agent 与server 端通过api 进行数据交互时,为了安全采取了两项安全措施:1、server 端需要对agent 端进行身份验证(避免有冒充agent 请况);2、当agent 采集数据提交到server 端过程需要对数据加密,数据传输到agent 端时再数据解密(防止数据传输过程被截取,泄露数据)。
api 身份认证
认证原理
server 与 agent保存相同的字符串密钥,agent 通过密钥生成签名值,把数据与签名值发送给server 端,server 根据自己的配置文件保存的密钥生成签名值与agent 端签名值比对,两者一致则认证通过否则失败。
安全加强(只针对数据传输中签名被截取安全隐患,暂不考虑server与agent 本地保存的字符串密钥安全隐患)
签名值一次性
agent 每次发送过来的签名值都会被server 记录,每次发送的签名值如果已经使用过则认证失败(防止传输中被截取后黑客可以永久使用被截取的签名密钥来认证)
动态生成签名
由于“签名值一次性”所以签名密钥必须动态生成,每次用不同的密钥。agent 每次生成签名前生成一个时间值,把时间与保存的字符串密钥组合在一起再生成一个签名值。签名值变成了动态。
签名超时设置
agent 每次发送签名与生成签名的时间到server ,server 端接收时会再本段生成一个时间,将两个时间值比较,当时间差超过10s ,则认证失败。(“签名值一次性”中对认证过的签名值记录在了字典中以{"client_time":"签名密钥"}形式存储,为了避免此字典无线增大占用内存,所以会对字典中的时间判断,超过10s 的数据进行删除,但是删除的签名数据再有相同的签名数据来认证时就不能通过“签名一次性”来排除了,所以“签名超时设置”解决了此问题)
代码
agent 端
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
import requests import time import hashlib def gen_sign(ctime): key = "uiakjsdfasjdf898" #此字符串在server端也保存一份 val = "%s|%s" %(key,ctime) #加入时间,动态生成签名密钥 obj = hashlib.md5() #生成一个对象 obj.update(val.encode('utf-8')) #md5只能对字节形式的数据进行加密 return obj.hexdigest() ctime = int(time.time()*1000) #*1000单位变ms,int 转化为整数(去掉小数点后的数字,而不是四舍五入) result = requests.post( url = 'http://127.0.0.1:8000/api/test/', #注意路径结尾一定更要以/结尾 params = {'sign':gen_sign(ctime),'ctime':ctime} #相当于在url = 'http://.../?key="alskdjflskdfjkjk"',也就是在url中以get形式发送数据 ) print(result.url,result.text) #认证逻辑:agent 把签名密钥与生成签名密钥的时间一并发送到server端,server 端通过自身保存的key与agent 的时间生成签名值,两个签名比对进行认证
server 端(view.py)
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
import json import hashlib #此模块里有md5的类 import time from django.shortcuts import render,HttpResponse from django.views import View from django.conf import settings from rest_framework.views import APIView from rest_framework.response import Response #生成签名的函数 def gen_sign(ctime): """ 生成签名 :param ctime: :return: """ val = '%s|%s' %(settings.URL_AUTH_KEY,ctime,) obj = hashlib.md5() #实力化一个md5对象 obj.update(val.encode('utf-8')) #md5对象只能对bytes格式数据加密 return obj.hexdigest() #返回生成的签名值 #所有的验证成功的签名密钥都会记录在SIGN_RECORD 中 SIGN_RECORD = {} class TestView(APIView): def post(self,request): print("请求来了") #由于继承了APIView所以此处的request 不是django原生的request,原生的为request._request client_sign = request._request.GET.get('sign') #获取agent发来的签名密钥 client_ctime = int(request._request.GET.get('ctime')) #获取agent 发来的时间 server_time = int(time.time() * 1000) #记录当前时间,*1000单位变成ms #认证第一关 if server_time - client_ctime > 5000: #比对两个时间,超过5s,则认证失败 return Response({'status': 'false', 'error': "路上时间太久了"}) # 认证第二关 if client_sign in SIGN_RECORD: #已经认证通过的签名会保存在SING_RECORD,新来密钥如果被认证过了就会认证失败 return Response("签名已经被使用过了") #认证第三关 server_sign = gen_sign(client_ctime) #server 端生成签名 if client_sign != server_sign: #密钥比对来通过认证 return Response({'status':'false','error':403}) #认证通过进行以下操作 SIGN_RECORD[client_sign] = client_ctime #通过认证的密钥存储起来,下次防止密钥重复使用 #防止SIGN_RECORD无线增大,对于5s 之前的记录删除。对于已经认证通过并存储在该记录中的密钥,超过10s删除后, ## 那么认证第二关就失效了,所以认证第一关就是为此而设 for k in list(SIGN_RECORD.keys()): v = SIGN_RECORD[k] if server_time - v > 5000: print("已经超过5s") del SIGN_RECORD[k] return Response({'status':'true','data':666})
数据加密
加密原理
通过rsa 模块生成公钥和私钥分别保存在agent 与server 端,agent 通过公钥加密数据传输到server 端通过私钥解密。rsa 加密分为1024与2048两种,生成公私钥对象时指定参数即可 pub_key_obj, priv_key_obj = rsa.newkeys(1024) ,1024指的是能加密的数据的位数,换算成字节就是1024/8=128 bytes
代码
生成公私钥,分别保存在server/agent 两端
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
# ######### 1. 生成公钥私钥 ######### pub_key_obj, priv_key_obj = rsa.newkeys(1024) # 1024/8 = 128 ,128 - 11 = 117 # 公钥字符串 pub_key_str = pub_key_obj.save_pkcs1() pub_key_code = base64.standard_b64encode(pub_key_str) # 私钥字符串 priv_key_str = priv_key_obj.save_pkcs1() priv_key_code = base64.standard_b64encode(priv_key_str) 将生成的公私钥(经过base64.standard_b64encode处理的)分别保存在agent/server 配置文件settings.py中 原生的公钥: b'-----BEGIN RSA PUBLIC KEY-----\nMIGJAoGBAJo2DEaukeIBTvc5vscIrh0gU79N+XRrf6NBGxGi6eOh7muzH3VV7UIn\nZvfUE3Nxu97DiMAC1u2JEudM8iatMChSLxSh9qFNB36ejz7dCi9DrAH6Ce46JZ7h\n+iwlo9x7Qr4uLJrQsHhia4/i89aAooNV6I8Ne+WOe3V1PvEUs+PhAgMBAAE=\n-----END RSA PUBLIC KEY-----\n' base64.standard_b64encode编码处理后: b'LS0tLS1CRUdJTiBSU0EgUFVCTElDIEtFWS0tLS0tCk1JR0pBb0dCQUpvMkRFYXVrZUlCVHZjNXZzY0lyaDBnVTc5TitYUnJmNk5CR3hHaTZlT2g3bXV6SDNWVjdVSW4KWnZmVUUzTnh1OTdEaU1BQzF1MkpFdWRNOGlhdE1DaFNMeFNoOXFGTkIzNmVqejdkQ2k5RHJBSDZDZTQ2Slo3aAoraXdsbzl4N1FyNHVMSnJRc0hoaWE0L2k4OWFBb29OVjZJOE5lK1dPZTNWMVB2RVVzK1BoQWdNQkFBRT0KLS0tLS1FTkQgUlNBIFBVQkxJQyBLRVktLS0tLQo='
agent 端
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
import rsa import base64 from config import settings #数据加密函数 def encrypt(value_bytes): """ rsa 公钥加密 :param value_bytes: 要加密的字节 :return: """ key_str = base64.standard_b64decode(settings.PUB_KEY) #settings.py 中记录了公钥 pk = rsa.PublicKey.load_pkcs1(key_str) #rsa 1024 能加密的数据大小为128(1024/8)字节,rsa 本身数据占用11字节,所以对大于128字节的数据分批次加密(每次加密117(128-11)字节)最后加密的数据拼接即可。 data_list = [] for i in range(0,len(value_bytes),117): chunk = value_bytes[i:i+117] result = rsa.encrypt(chunk, pk) data_list.append(result) return b''.join(data_list) #进行数据加密,并且发送到agent ctime = int(time.time()*1000) r1 = requests.post( url=self.asset_api, params = {'sign':gen_sign(ctime),'ctime':ctime}, #传输签名密钥 data=encrypt(json.dumps(info).encode('utf-8')), #传输数据rsa 加密 headers={'Content-Type':'application/json'} )
server 端
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
#使用私钥解密的函数 import rsa import base64 def decrypt(bytes_value): """ rsa解密 :param bytes_value: 要解密的数据为bytes 类型,因为rsa 只能对bytes 类型数据加密,所以解密数据为bytes类型 :return: 解密完成的字节 """ key_str = base64.standard_b64decode(settings.PRIV_KEY) #生成密钥后使用base64.standard_b64encode对公钥进行了编码处理,所以此处要用decode 解码 pk = rsa.PrivateKey.load_pkcs1(key_str) #生成私钥证书 #rsa 1024 能加密的数据大小为128(1024/8)字节,所以需要对大于128字节的数据分批次加密(每次加密128字节)最后加密的数据拼接即可。所以解密时也是每次解密128字节,最后拼接即可 result = [] for i in range(0,len(bytes_value),128): chunk = bytes_value[i:i+128] val = rsa.decrypt(chunk, pk) #通过公钥证书对数据chunk 加密 result.append(val) return b''.join(result) #利用私钥解密的函数解密数据 class AssetView(APIAuthView): def post(self, request, *args, **kwargs): body = decrypt(request._request.body) #通过decrypt 解密数据,body 中才是原生的数据 asset_info = json.loads(body.decode('utf-8')) #解密后的数据解码再序列化加载