[omemo] Add encrypt_file method

The HTTPUploadPlugin calls that if encryption is enabled
This commit is contained in:
Philipp Hörist
2017-06-09 23:32:42 +02:00
parent c4f2c658d7
commit 3e287d2a46
2 changed files with 29 additions and 0 deletions

View File

@@ -19,6 +19,7 @@
import logging
import time
import os
from base64 import b64encode
from axolotl.ecc.djbec import DjbECPublicKey
@@ -216,6 +217,14 @@ class OmemoState:
log.debug("Decrypted Message => " + result)
return result
def encrypt_file(self, data):
key = os.urandom(32)
iv = os.urandom(16)
payload, tag = encrypt(key, iv, data)
encrypted_data = payload + tag
return (encrypted_data, key, iv)
def create_msg(self, from_jid, jid, plaintext):
key = get_random_bytes(16)
iv = get_random_bytes(16)