hueso/gajim-plugins
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[omemo] Sanitize BLOBs in secret table

Browse Source
This commit is contained in:
lovetox
2020-05-23 10:21:10 +02:00
parent 3f38b77977
commit 66c49f5e7b
1 changed files with 22 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
omemo/backend/liteaxolotlstore.py
View File

@@ -147,7 +147,7 @@ class LiteAxolotlStore(AxolotlStore):
create_db_sql = """
BEGIN TRANSACTION;
%s
PRAGMA user_version=8;
PRAGMA user_version=9;
END TRANSACTION;
""" % (create_tables)
self._con.executescript(create_db_sql)
@@ -278,6 +278,27 @@ class LiteAxolotlStore(AxolotlStore):
self._con.execute('PRAGMA user_version=8')
self._con.commit()
if self.user_version() < 9:
# Sanitize invalid BLOBs from the python2 days
query_keys = '''SELECT device_id,
CAST(public_key as BLOB) as public_key,
CAST(private_key as BLOB) as private_key
FROM secret'''
rows = self._con.execute(query_keys).fetchall()
delete = 'DELETE FROM secret'
self._con.execute(delete)
insert = '''INSERT INTO secret (device_id, public_key, private_key)
VALUES (?, ?, ?)'''
for row in rows:
try:
self._con.execute(insert, row)
except Exception as error:
self._log.warning(error)
self._con.execute('PRAGMA user_version=9')
self._con.commit()
def loadSignedPreKey(self, signedPreKeyId):
query = 'SELECT record FROM signed_prekeys WHERE prekey_id = ?'
result = self._con.execute(query, (signedPreKeyId, )).fetchone()