#coding=utf_8
from Crypto.Cipher import AES
from Crypto.Random import random
import socket
def padding(message):
paddingLength = 16 - len(message) % 16
paddingByte = chr(paddingLength)
message = message + paddingByte * paddingLength
return message
def depadding(message):
if(len(message) % 16 != 0):
raise Exception("Error: length of the message should be multiple of 16!")
paddingLength = ord(message[-1])
message = message[ : -paddingLength]
return message
def genKey():
key = ""
for i in range(16):
byte = random.randint(0, 255)
key = key + chr(byte)
return key
def connect2Server():
host = "localhost"
port = 11111
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.connect((host, port))
#生成并发送密钥key和初始化向量iv
key = genKey()
iv = genKey()
aes = AES.new(key, AES.MODE_CBC, iv)
message = "key:" + key + ";iv:" + iv
print "明文发送key和iv", message
s.send(message)
# 收到确认消息后，验证是否是ok
message = s.recv(1024).strip()
message = aes.decrypt(message)
message = depadding(message)
print message
if message != "ok":
print("会话确认错误！断开连接！")
s.close()
return123
print "确认开始会话"
while True:
print "请输入消息..."
message = raw_input().strip()
message = padding(message)
message = aes.encrypt(message)
s.send(message)
print "等待对方消息..."
message = s.recv(1024)
message = aes.decrypt(message)
message = depadding(message)
print(message)
s.close()
if __name__ == "__main__":
'''
加密通信协议：
1.A方client随机生成AES加密密钥key和初始化向量iv
2.以明文发送key和iv，报文格式："key:xxxxxxxxxxxxxxxx;iv:xxxxxxxxxxxxxxxx"
3.B方server收到并解析key和iv，以AES/CBC加密"ok"发送给A方
4.A方收到消息解密验证是否是ok，如果是开始会话，否则断开链接
:return:
'''
connect2Server()
