SboxTable = [
0xd6,0x90,0xe9,0xfe,0xcc,0xe1,0x3d,0xb7,0x16,0xb6,0x14,0xc2,0x28,0xfb,0x2c,0x05,
0x2b,0x67,0x9a,0x76,0x2a,0xbe,0x04,0xc3,0xaa,0x44,0x13,0x26,0x49,0x86,0x06,0x99,
0x9c,0x42,0x50,0xf4,0x91,0xef,0x98,0x7a,0x33,0x54,0x0b,0x43,0xed,0xcf,0xac,0x62,
0xe4,0xb3,0x1c,0xa9,0xc9,0x08,0xe8,0x95,0x80,0xdf,0x94,0xfa,0x75,0x8f,0x3f,0xa6,
0x47,0x07,0xa7,0xfc,0xf3,0x73,0x17,0xba,0x83,0x59,0x3c,0x19,0xe6,0x85,0x4f,0xa8,
0x68,0x6b,0x81,0xb2,0x71,0x64,0xda,0x8b,0xf8,0xeb,0x0f,0x4b,0x70,0x56,0x9d,0x35,
0x1e,0x24,0x0e,0x5e,0x63,0x58,0xd1,0xa2,0x25,0x22,0x7c,0x3b,0x01,0x21,0x78,0x87,
0xd4,0x00,0x46,0x57,0x9f,0xd3,0x27,0x52,0x4c,0x36,0x02,0xe7,0xa0,0xc4,0xc8,0x9e,
0xea,0xbf,0x8a,0xd2,0x40,0xc7,0x38,0xb5,0xa3,0xf7,0xf2,0xce,0xf9,0x61,0x15,0xa1,
0xe0,0xae,0x5d,0xa4,0x9b,0x34,0x1a,0x55,0xad,0x93,0x32,0x30,0xf5,0x8c,0xb1,0xe3,
0x1d,0xf6,0xe2,0x2e,0x82,0x66,0xca,0x60,0xc0,0x29,0x23,0xab,0x0d,0x53,0x4e,0x6f,
0xd5,0xdb,0x37,0x45,0xde,0xfd,0x8e,0x2f,0x03,0xff,0x6a,0x72,0x6d,0x6c,0x5b,0x51,
0x8d,0x1b,0xaf,0x92,0xbb,0xdd,0xbc,0x7f,0x11,0xd9,0x5c,0x41,0x1f,0x10,0x5a,0xd8,
0x0a,0xc1,0x31,0x88,0xa5,0xcd,0x7b,0xbd,0x2d,0x74,0xd0,0x12,0xb8,0xe5,0xb4,0xb0,
0x89,0x69,0x97,0x4a,0x0c,0x96,0x77,0x7e,0x65,0xb9,0xf1,0x09,0xc5,0x6e,0xc6,0x84,
0x18,0xf0,0x7d,0xec,0x3a,0xdc,0x4d,0x20,0x79,0xee,0x5f,0x3e,0xd7,0xcb,0x39,0x48,
]
def sm4sbox(idx):
return SboxTable[idx]
# System parameter
FK = [0xa3b1bac6,0x56aa3350,0x677d9197,0xb27022dc]
# fixed parameter
CK = [
0x00070e15,0x1c232a31,0x383f464d,0x545b6269,
0x70777e85,0x8c939aa1,0xa8afb6bd,0xc4cbd2d9,
0xe0e7eef5,0xfc030a11,0x181f262d,0x343b4249,
0x50575e65,0x6c737a81,0x888f969d,0xa4abb2b9,
0xc0c7ced5,0xdce3eaf1,0xf8ff060d,0x141b2229,
0x30373e45,0x4c535a61,0x686f767d,0x848b9299,
0xa0a7aeb5,0xbcc3cad1,0xd8dfe6ed,0xf4fb0209,
0x10171e25,0x2c333a41,0x484f565d,0x646b7279
]
#输入为32比特，分为4块，每块为8比特的数
#输出为32比特输入对应的大整数
def hex2int_32(key_data):
tmp_data = int((key_data[0] << 24) | (key_data[1] << 16) | (key_data[2] << 8) | (key_data[3]))
return tmp_data
#输入为128比特，4个分块，每块为32比特的整数
#输出为128比特对应的十六进制表示
def final_transfer(data):
C=(data[0]<<96)|(data[1]<<64)|(data[2]<<32)|data[3]
return ('{:032x}'.format(C))#把整数格式化为32位的16进制字符串，高位用零占位
#输入为32比特的大整数
#输出为4块，每块为8比特对应的整数（取值范围为0-255）
def int2hex_32(n):
return [int((n>>24)&0xff), int((n>>16)&0xff), int((n>>8)&0xff), int((n)&0xff)]
# 输入为两个列表，每个元素为十六进制表示的数
#输出为两个列表位置对应的数模2加
def XOR(a, b):
return list(map(lambda x, y: x ^ y, a, b))
#输入为一个32比特的整数
#输出为循环左移n比特后的32比特整数
def SHL(x, n):
xx = int(int(x << n) & 0xffffffff)
return xx
def ROTL(x, n):
xx = SHL(x, n)
yy = xx | int((x >> (32 - n)) & 0xffffffff)
return yy
#轮函数T变换,输入为32比特的整数
#输出为32比特的大整数
def sm4FT(xdata):
a=[0,0,0,0]
b=int2hex_32(xdata)
a[0]=sm4sbox(b[0])
a[1]=sm4sbox(b[1])
a[2]=sm4sbox(b[2])
a[3]=sm4sbox(b[3])
bb=hex2int_32(a)
c=bb^(ROTL(bb,2))^(ROTL(bb,10))^(ROTL(bb,18))^(ROTL(bb,24))
return c
#轮函数F，输入为160比特，4个分组，一个轮密钥，分5块，十六进制表示
#输出为32比特的大整数
def sm4F(x0,x1,x2,x3,rk):
a=x0^sm4FT(x1^x2^x3^rk)
return a
#密钥扩张T变换，输入为32比特的十六进制数，输出为32比特的整数
def keyexpandT(keydata):
a=[0,0,0,0]
b=int2hex_32(keydata)
a[0]=sm4sbox(b[0])
a[1]=sm4sbox(b[1])
a[2]=sm4sbox(b[2])
a[3]=sm4sbox(b[3])
bb=hex2int_32(a)
c=bb^(ROTL(bb,13))^(ROTL(bb,23))
return c
#密钥扩展算法，输入为128比特，4个分组
#输出为32比特整数
def keyexpand(MK0,MK1,MK2,MK3):
K=[0]*36
rk=[0]*32
K[0]=MK0^FK[0]
K[1]=MK1^FK[1]
K[2]=MK2^FK[2]
K[3]=MK3^FK[3]
for i in range(32):
K[i+4]=K[i]^(keyexpandT(K[i+1]^K[i+2]^K[i+3]^CK[i]))
rk[i]=K[i+4]
return rk
#sm4一轮加密，输入为160比特，5块，每块32比特，明文4块，密钥1块
#输出为128比特，4块，每块为32比特十六进制表示
def sm4_32_round(Xdata,KEY,mode):
X=[0]*36
rk=[0]*32
mk=[0,0,0,0]
Cb=[0,0,0,0]
for i in range (4):
mk[i]=KEY[i]
X[i]=Xdata[i]
rk=keyexpand(mk[0],mk[1],mk[2],mk[3])
if mode==0:
for i in range(32):
X[i+4]=sm4F(X[i],X[i+1],X[i+2],X[i+3],rk[i])
for i in range (4):
Cb[i]=X[35-i]
return Cb
else:
for i in range(32):
X[i+4]=sm4F(X[i],X[i+1],X[i+2],X[i+3],rk[31-i])
for i in range (4):
Cb[i]=X[35-i]
return Cb
