def __init__(self, encoded_key): tink_config.register() aead.register() if (encoded_key==None): self.keyset_handle = tink.new_keyset_handle(aead.aead_key_templates.AES256_GCM) else: reader = tink.BinaryKeysetReader(base64.b64decode(encoded_key)) self.keyset_handle = cleartext_keyset_handle.read(reader) self.key=self.keyset_handle.keyset_info() self.aead_primitive = self.keyset_handle.primitive(aead.Aead)
def setUpModule(): tink_config.register()
return base64.b64encode( self.public_key.encrypt( raw, OAEP(mgf=MGF1(algorithm=hashes.SHA256()), algorithm=hashes.SHA256(), label=None))) def decrypt(self, raw): return self.private_key.decrypt( base64.b64decode(raw), OAEP(mgf=MGF1(algorithm=hashes.SHA256()), algorithm=hashes.SHA256(), label=None)).decode('utf-8').strip() tink_config.register() aead.register() mac.register() class AESCipher(object): def __init__(self, encoded_key): if (encoded_key == None): self.keyset_handle = tink.new_keyset_handle( aead.aead_key_templates.AES256_GCM) else: reader = tink.BinaryKeysetReader(base64.b64decode(encoded_key)) self.keyset_handle = cleartext_keyset_handle.read(reader) self.key = self.keyset_handle.keyset_info() self.aead_primitive = self.keyset_handle.primitive(aead.Aead)