import sys
_pythonMajorVersion = sys.version_info[0]
ECB =0
CBC =1
PAD_NORMAL =1
PAD_PKCS5 =2class_baseDes(object):def__init__(self, mode=ECB, IV=None, pad=None, padmode=PAD_NORMAL):if IV:
IV = self._guardAgainstUnicode(IV)if pad:
pad = self._guardAgainstUnicode(pad)
self.block_size =8if pad and padmode == PAD_PKCS5:raise ValueError("Cannot use a pad character with PAD_PKCS5")if IV andlen(IV)!= self.block_size:raise ValueError("Invalid Initial Value (IV), must be a multiple of "+str(self.block_size)+" bytes")
self._mode = mode
self._iv = IV
self._padding = pad
self._padmode = padmode
defgetKey(self):"""getKey() -> bytes"""return self.__key
defsetKey(self, key):"""Will set the crypting key for this object."""
key = self._guardAgainstUnicode(key)
self.__key = key
defgetMode(self):"""getMode() -> pyDes.ECB or pyDes.CBC"""return self._mode
defsetMode(self, mode):"""Sets the type of crypting mode, pyDes.ECB or pyDes.CBC"""
self._mode = mode
defgetPadding(self):"""getPadding() -> bytes of length 1. Padding character."""return self._padding
defsetPadding(self, pad):"""setPadding() -> bytes of length 1. Padding character."""if pad isnotNone:
pad = self._guardAgainstUnicode(pad)
self._padding = pad
defgetPadMode(self):"""getPadMode() -> pyDes.PAD_NORMAL or pyDes.PAD_PKCS5"""return self._padmode
defsetPadMode(self, mode):"""Sets the type of padding mode, pyDes.PAD_NORMAL or pyDes.PAD_PKCS5"""
self._padmode = mode
defgetIV(self):"""getIV() -> bytes"""return self._iv
defsetIV(self, IV):"""Will set the Initial Value, used in conjunction with CBC mode"""ifnot IV orlen(IV)!= self.block_size:raise ValueError("Invalid Initial Value (IV), must be a multiple of "+str(self.block_size)+" bytes")
IV = self._guardAgainstUnicode(IV)
self._iv = IV
def_padData(self, data, pad, padmode):if padmode isNone:
padmode = self.getPadMode()if pad and padmode == PAD_PKCS5:raise ValueError("Cannot use a pad character with PAD_PKCS5")if padmode == PAD_NORMAL:iflen(data)% self.block_size ==0:return data
ifnot pad:
pad = self.getPadding()ifnot pad:raise ValueError("Data must be a multiple of "+str(self.block_size)+" bytes in length. Use padmode=PAD_PKCS5 or set the pad character.")
data +=(self.block_size -(len(data)% self.block_size))* pad
elif padmode == PAD_PKCS5:
pad_len =8-(len(data)% self.block_size)if _pythonMajorVersion <3:
data += pad_len *chr(pad_len)else:
data +=bytes([pad_len]* pad_len)return data
def_unpadData(self, data, pad, padmode):ifnot data:return data
if pad and padmode == PAD_PKCS5:raise ValueError("Cannot use a pad character with PAD_PKCS5")if padmode isNone:
padmode = self.getPadMode()if padmode == PAD_NORMAL:ifnot pad:
pad = self.getPadding()if pad:
data = data[:-self.block_size]+ \
data[-self.block_size:].rstrip(pad)elif padmode == PAD_PKCS5:if _pythonMajorVersion <3:
pad_len =ord(data[-1])else:
pad_len = data[-1]
data = data[:-pad_len]return data
def_guardAgainstUnicode(self, data):if _pythonMajorVersion <3:ifisinstance(data,unicode):raise ValueError("pyDes can only work with bytes, not Unicode strings.")else:ifisinstance(data,str):try:return data.encode('ascii')except UnicodeEncodeError:passraise ValueError("pyDes can only work with encoded strings, not Unicode.")return data
classdes(_baseDes):# Permutation and translation tables for DES
__pc1 =[56,48,40,32,24,16,8,0,57,49,41,33,25,17,9,1,58,50,42,34,26,18,10,2,59,51,43,35,62,54,46,38,30,22,14,6,61,53,45,37,29,21,13,5,60,52,44,36,28,20,12,4,27,19,11,3]# number left rotations of pc1
__left_rotations =[1,1,2,2,2,2,2,2,1,2,2,2,2,2,2,1]# permuted choice key (table 2)
__pc2 =[13,16,10,23,0,4,2,27,14,5,20,9,22,18,11,3,25,7,15,6,26,19,12,1,40,51,30,36,46,54,29,39,50,44,32,47,43,48,38,55,33,52,45,41,49,35,28,31]# initial permutation IP
__ip =[57,49,41,33,25,17,9,1,59,51,43,35,27,19,11,3,61,53,45,37,29,21,13,5,63,55,47,39,31,23,15,7,56,48,40,32,24,16,8,0,58,50,42,34,26,18,10,2,60,52,44,36,28,20,12,4,62,54,46,38,30,22,14,6]# Expansion table for turning 32 bit blocks into 48 bits
__expansion_table =[31,0,1,2,3,4,3,4,5,6,7,8,7,8,9,10,11,12,11,12,13,14,15,16,15,16,17,18,19,20,19,20,21,22,23,24,23,24,25,26,27,28,27,28,29,30,31,0]# The (in)famous S-boxes
__sbox =[# S1[14,4,13,1,2,15,11,8,3,10,6,12,5,9,0,7,0,15,7,4,14,2,13,1,10,6,12,11,9,5,3,8,4,1,14,8,13,6,2,11,15,12,9,7,3,10,5,0,15,12,8,2,4,9,1,7,5,11,3,14,10,0,6,13],# S2[15,1,8,14,6,11,3,4,9,7,2,13,12,0,5,10,3,13,4,7,15,2,8,14,12,0,1,10,6,9,11,5,0,14,7,11,10,4,13,1,5,8,12,6,9,3,2,15,13,8,10,1,3,15,4,2,11,6,7,12,0,5,14,9],# S3[10,0,9,14,6,3,15,5,1,13,12,7,11,4,2,8,13,7,0,9,3,4,6,10,2,8,5,14,12,11,15,1,13,6,4,9,8,15,3,0,11,1,2,12,5,10,14,7,1,10,13,0,6,9,8,7,4,15,14,3,11,5,2,12],# S4[7,13,14,3,0,6,9,10,1,2,8,5,11,12,4,15,13,8,11,5,6,15,0,3,4,7,2,12,1,10,14,9,10,6,9,0,12,11,7,13,15,1,3,14,5,2,8,4,3,15,0,6,10,1,13,8,9,4,5,11,12,7,2,14],# S5[2,12,4,1,7,10,11,6,8,5,3,15,13,0,14,9,14,11,2,12,4,7,13,1,5,0,15,10,3,9,8,6,4,2,1,11,10,13,7,8,15,9,12,5,6,3,0,14,11,8,12,7,1,14,2,13,6,15,0,9,10,4,5,3],# S6[12,1,10,15,9,2,6,8,0,13,3,4,14,7,5,11,10,15,4,2,7,12,9,5,6,1,13,14,0,11,3,8,9,14,15,5,2,8,12,3,7,0,4,10,1,13,11,6,4,3,2,12,9,5,15,10,11,14,1,7,6,0,8,13],# S7[4,11,2,14,15,0,8,13,3,12,9,7,5,10,6,1,13,0,11,7,4,9,1,10,14,3,5,12,2,15,8,6,1,4,11,13,12,3,7,14,10,15,6,8,0,5,9,2,6,11,13,8,1,4,10,7,9,5,0,15,14,2,3,12],# S8[13,2,8,4,6,15,11,1,10,9,3,14,5,0,12,7,1,15,13,8,10,3,7,4,12,5,6,11,0,14,9,2,7,11,4,1,9,12,14,2,0,6,10,13,15,3,5,8,2,1,14,7,4,10,8,13,15,12,9,0,3,5,6,11],]# 32-bit permutation function P used on the output of the S-boxes
__p =[15,6,19,20,28,11,27,16,0,14,22,25,4,17,30,9,1,7,23,13,31,26,2,8,18,12,29,5,21,10,3,24]# final permutation IP^-1
__fp =[39,7,47,15,55,23,63,31,38,6,46,14,