2016-07-31 31 views
0

我已经编写了以下类来测试不同的加密方案。但是,我在通过不同的加密方案实例化对象时遇到了问题。有人能指出一些没有意义的东西,我不赶上atm吗?我不确定它为什么不起作用。它给出了一个TypeError: encrypt() takes exactly 3 arguments (2 given),但它确实有自我传递,所以我不知道如何解决它在其他人的基础上。实例和类:当x-1给出时需要x个参数

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)) 
+0

你究竟在哪里传递了两个参数(除了'self')呢?所有'encrypt'调用都有一个参数。 –

+0

请确保包含* full * traceback。我可以猜出错误发生的位置,但是完整的追溯会告诉我们一定的错误,并确保我没有错过其他情况。 –

回答

1

您传递一个参数绑定方法:

return self.bc1.encrypt(
    self.bc2.encrypt(tweak) * self.bc3.encrypt(message) + 
    self.bc4.encrypt(tweak)) 

这是一个参数传递给每个BC.encrypt()方法,而这种方法需要2超出selfmessagekey

要么传递值为key,要么从BC.encrypt()方法定义中删除该参数(并从其他位置获取密钥;或许从__init__中设置的实例属性)。

相关问题