为了能够测试不同的加密方案,我编写了以下类。但是,我在实例化来自不同加密方案的对象时遇到了问题。有人能指出一些毫无意义的事情吗?我没有赶上自动取款机?我不知道为什么它不起作用。它给出了一个TypeError: encrypt() takes exactly 3 arguments (2 given)
,但它确实有自通过,所以我不知道如何在其他的基础上修复它。
class AXU:
def __init__(self, sec_param):
self.sec_param = sec_param
def getHash(self):
# sample a, b and return hash function
a = random.randrange(self.sec_param)
b = random.randrange(self.sec_param)
return lambda x : a*x+b % sec_param
class BC(object):
def __init__(self, sec_param):
# generate a key
self.sec_param = sec_param
def encrypt(self, message, key):
#encrypt with AES?
cipher = AES.new(key, MODE_CFB, sec_param)
msg = iv + cipher.encrypt(message)
return msg
class tBC(object):
def __init__(self, sec_param):
self.sec_param = sec_param
def encrypt(self, tweak, message):
#pass
return AES.new(message, tweak)
class Trivial(tBC):
def __init__(self):
self.bcs = {}
def encrypt(self, tweak, message):
if tweak not in self.bcs.keys():
bc = BC()
self.bcs[tweak] = bc
return self.bcs[tweak].encrypt(message)
class Our(tBC):
def __init__(self, sec_param):
self.bc1 = BC(sec_param)
self.bc2 = BC(sec_param)
self.bc3 = BC(sec_param)
self.bc4 = BC(sec_param)
# encryption over GF field
def encrypt(self, tweak, message):
return self.bc1.encrypt(self.bc2.encrypt(tweak) * self.bc3.encrypt(message) + self.bc4.encrypt(tweak))
您正在向绑定方法传递一个参数:
return self.bc1.encrypt(
self.bc2.encrypt(tweak) * self.bc3.encrypt(message) +
self.bc4.encrypt(tweak))
这是BC.encrypt()
方法的一个参数,该方法在self
、message
和key
之外取2。
传入key
的值,或者从BC.encrypt()
方法定义中删除该参数(并从其他位置获取键;可能从__init__
中的实例属性集获取)。