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

[omemo] Add sqlite SessionRecord converter

Browse Source
This commit is contained in:
Philipp Hörist
2019-02-22 16:06:35 +01:00
parent 0ee9b2fa78
commit 5aaa6957c9
1 changed files with 15 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
omemo/backend/liteaxolotlstore.py
View File

@@ -42,14 +42,20 @@ log = logging.getLogger('gajim.plugin_system.omemo')
def _convert_to_string(text): def _convert_to_string(text):
return text.decode() return text.decode()
def _convert_identity_key(key): def _convert_identity_key(key):
if not key: if not key:
return return
return IdentityKeyExtended(DjbECPublicKey(key[1:])) return IdentityKeyExtended(DjbECPublicKey(key[1:]))
def _convert_record(record):
return SessionRecord(serialized=record)
sqlite3.register_converter('jid', _convert_to_string) sqlite3.register_converter('jid', _convert_to_string)
sqlite3.register_converter('pk', _convert_identity_key) sqlite3.register_converter('pk', _convert_identity_key)
sqlite3.register_converter('session_record', _convert_record)
class LiteAxolotlStore(AxolotlStore): class LiteAxolotlStore(AxolotlStore):
@@ -273,12 +279,10 @@ class LiteAxolotlStore(AxolotlStore):
self._con.commit() self._con.commit()
def loadSession(self, recipientId, deviceId): def loadSession(self, recipientId, deviceId):
query = '''SELECT record FROM sessions WHERE query = '''SELECT record as "record [session_record]"
recipient_id = ? AND device_id = ?''' FROM sessions WHERE recipient_id = ? AND device_id = ?'''
result = self._con.execute(query, (recipientId, deviceId)).fetchone() result = self._con.execute(query, (recipientId, deviceId)).fetchone()
if result is None: return result.record if result is not None else SessionRecord()
return SessionRecord()
return SessionRecord(serialized=result.record)
def getJidFromDevice(self, device_id): def getJidFromDevice(self, device_id):
query = '''SELECT recipient_id as "recipient_id [jid]" query = '''SELECT recipient_id as "recipient_id [jid]"
@@ -350,17 +354,10 @@ class LiteAxolotlStore(AxolotlStore):
self._con.commit() self._con.commit()
def getInactiveSessionsKeys(self, recipientId): def getInactiveSessionsKeys(self, recipientId):
query = '''SELECT record FROM sessions query = '''SELECT record as "record [session_record]" FROM sessions
WHERE active = 0 AND recipient_id = ?''' WHERE active = 0 AND recipient_id = ?'''
result = self._con.execute(query, (recipientId,)).fetchall() result = self._con.execute(query, (recipientId,)).fetchall()
return [row.record.getSessionState().getRemoteIdentityKey() for row in result]
results = []
for row in result:
public_key = (SessionRecord(serialized=row.record).
getSessionState().getRemoteIdentityKey().
getPublicKey())
results.append(public_key.serialize())
return results
def loadPreKey(self, preKeyId): def loadPreKey(self, preKeyId):
query = '''SELECT record FROM prekeys WHERE prekey_id = ?''' query = '''SELECT record FROM prekeys WHERE prekey_id = ?'''
@@ -465,9 +462,9 @@ class LiteAxolotlStore(AxolotlStore):
return result.trust if result is not None else None return result.trust if result is not None else None
def getFingerprints(self, jid): def getFingerprints(self, jid):
query = '''SELECT _id, recipient_id as "recipient_id [jid]", query = '''SELECT recipient_id as "recipient_id [jid]",
public_key as "public_key [pk]", trust FROM identities public_key as "public_key [pk]", trust FROM identities
WHERE recipient_id =? ORDER BY trust ASC''' WHERE recipient_id = ? ORDER BY trust ASC'''
return self._con.execute(query, (jid,)).fetchall() return self._con.execute(query, (jid,)).fetchall()
def getTrustedFingerprints(self, jid): def getTrustedFingerprints(self, jid):
@@ -530,5 +527,7 @@ class LiteAxolotlStore(AxolotlStore):
def getUnacknowledgedCount(self, recipient_id, device_id): def getUnacknowledgedCount(self, recipient_id, device_id):
record = self.loadSession(recipient_id, device_id) record = self.loadSession(recipient_id, device_id)
if record.isFresh():
return 0
state = record.getSessionState() state = record.getSessionState()
return state.getSenderChainKey().getIndex() return state.getSenderChainKey().getIndex()