我正在尝试使用带有AES 256的加密TCP套接字进行文件传输.
>如果我传输的文件没有加密,则可以正常工作.
>如果我将小的命令(例如“ ipconfig”)发送到客户端或服务器,则加密工作正常.
>无论文件大小如何,我都会不断收到以下错误消息:
DecodeAES = lambda c, e: c.decrypt(base64.b64decode(e)).rstrip(PADDING)
File "/usr/lib/python2.6/base64.py", line 76, in b64decode
raise TypeError(msg)
TypeError: Incorrect padding
我的编码和解码功能如下(hat变量是消息):
def AESENC(hat,typ):
BLOCK_SIZE = 32
PADDING = '{'
pad = lambda s: s + (BLOCK_SIZE - len(s) % BLOCK_SIZE) * PADDING
EncodeAES = lambda c, s: base64.b64encode(c.encrypt(pad(s)))
DecodeAES = lambda c, e: c.decrypt(base64.b64decode(e)).rstrip(PADDING)
secret = '01234567890123456789012345678912'
IV = 'wir&/>H54mgd9a";'
cipher = AES.new(secret,AES.MODE_CFB,IV)
if typ == 0:
encoded = EncodeAES(cipher, hat)
return encoded
else:
decoded = DecodeAES(cipher, hat)
return decoded
客户端
if os.path.exists(df):
print ' found the file '
f = open(df, 'rb')
packet = f.read(1024)
while packet != '':
s.send(AESENC(packet,0))
s.send( AESENC('123XXX',0) )
s.send('123XXX')
f.close()
服务器端
f = open('/root/Desktop/Transfer.mp3','wb')
while True:
bits = AESENC ( conn.recv(1024) , 1 )
while (bits):
f.write(bits)
bits = AESENC ( conn.recv(1024) , 1 )
if bits.endswith('123XXX'):
print '[+] Transfer completed '
break
f.close()
break
return
有人知道如何解决此问题吗?
解决方法:
对于您对示例代码的误解,我们深表歉意,并感谢您添加服务器和客户端调用!
请注意,您收到的错误消息与加密无关.您可以阅读base64.py库的相关部分:错误消息表明base64数据无效.特别是,最后没有正确填充.
有了附加的呼叫站点信息,我相信问题是您正在加密,然后分别对客户端上数据的每个1024字节块进行编码.然后,在服务器上,您正在从网络读取1024个字节并尝试对其进行解码.但是,base64编码将增加原始数据的长度,因此您将仅读取编码形式的前1024个字节,这将是被截断的base64消息(因此未正确填充).
至于如何解决,base64消息需要全部解码为一条.因此,您要么需要计算要从网络读取多少数据即可获得完整的编码块(这种计算可以可靠地执行,但最后一个数据包可能很短,因为base64的长度始终为(length 2)/ 3 * 4 ),立即对整个数据流进行编码(如果要处理任意大小的文件,可能会由于使用内存而出现问题),或者定义您的网络协议,以便服务器可以判断出可以看到一个完整的块用base64解码.