Commit 9b058963 authored by Philipp Hörist's avatar Philipp Hörist

[omemo] Sanitize BLOBs in sessions table

parent 66c49f5e
......@@ -147,7 +147,7 @@ class LiteAxolotlStore(AxolotlStore):
create_db_sql = """
BEGIN TRANSACTION;
%s
PRAGMA user_version=9;
PRAGMA user_version=10;
END TRANSACTION;
""" % (create_tables)
self._con.executescript(create_db_sql)
......@@ -299,6 +299,31 @@ class LiteAxolotlStore(AxolotlStore):
self._con.execute('PRAGMA user_version=9')
self._con.commit()
if self.user_version() < 10:
# Sanitize invalid BLOBs from the python2 days
query_keys = '''SELECT _id,
recipient_id,
device_id,
CAST(record as BLOB) as record,
timestamp,
active
FROM sessions'''
rows = self._con.execute(query_keys).fetchall()
delete = 'DELETE FROM sessions'
self._con.execute(delete)
insert = '''INSERT INTO sessions (_id, recipient_id, device_id,
record, timestamp, active)
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=10')
self._con.commit()
def loadSignedPreKey(self, signedPreKeyId):
query = 'SELECT record FROM signed_prekeys WHERE prekey_id = ?'
result = self._con.execute(query, (signedPreKeyId, )).fetchone()
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment