OMEMO GTK3 inital

omemo/omemo/liteidentitykeystore.py

@@ -0,0 +1,167 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright 2015 Tarek Galal <tare2.galal@gmail.com>
+#
+# This file is part of Gajim-OMEMO plugin.
+#
+# The Gajim-OMEMO plugin is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by the Free
+# Software Foundation, either version 3 of the License, or (at your option) any
+# later version.
+#
+# Gajim-OMEMO is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+# A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along with
+# the Gajim-OMEMO plugin.  If not, see <http://www.gnu.org/licenses/>.
+#
+
+from axolotl.ecc.djbec import DjbECPrivateKey, DjbECPublicKey
+from axolotl.identitykey import IdentityKey
+from axolotl.identitykeypair import IdentityKeyPair
+from axolotl.state.identitykeystore import IdentityKeyStore
+
+UNDECIDED = 2
+TRUSTED = 1
+UNTRUSTED = 0
+
+
+class LiteIdentityKeyStore(IdentityKeyStore):
+    def __init__(self, dbConn):
+        """
+        :type dbConn: Connection
+        """
+        self.dbConn = dbConn
+
+    def getIdentityKeyPair(self):
+        q = "SELECT public_key, private_key FROM identities " + \
+            "WHERE recipient_id = -1"
+        c = self.dbConn.cursor()
+        c.execute(q)
+        result = c.fetchone()
+
+        publicKey, privateKey = result
+        return IdentityKeyPair(
+            IdentityKey(DjbECPublicKey(publicKey[1:])),
+            DjbECPrivateKey(privateKey))
+
+    def getLocalRegistrationId(self):
+        q = "SELECT registration_id FROM identities WHERE recipient_id = -1"
+        c = self.dbConn.cursor()
+        c.execute(q)
+        result = c.fetchone()
+        return result[0] if result else None
+
+    def storeLocalData(self, registrationId, identityKeyPair):
+        q = "INSERT INTO identities( " + \
+            "recipient_id, registration_id, public_key, private_key) " + \
+            "VALUES(-1, ?, ?, ?)"
+        c = self.dbConn.cursor()
+        c.execute(q,
+                  (registrationId,
+                   identityKeyPair.getPublicKey().getPublicKey().serialize(),
+                   identityKeyPair.getPrivateKey().serialize()))
+
+        self.dbConn.commit()
+
+    def saveIdentity(self, recipientId, identityKey):
+        q = "INSERT INTO identities (recipient_id, public_key, trust) " \
+            "VALUES(?, ?, ?)"
+        c = self.dbConn.cursor()
+
+        if not self.getIdentity(recipientId, identityKey):
+            c.execute(q, (recipientId,
+                          identityKey.getPublicKey().serialize(),
+                          UNDECIDED))
+            self.dbConn.commit()
+
+    def getIdentity(self, recipientId, identityKey):
+        q = "SELECT * FROM identities WHERE recipient_id = ? " \
+            "AND public_key = ?"
+        c = self.dbConn.cursor()
+
+        c.execute(q, (recipientId, identityKey.getPublicKey().serialize()))
+        result = c.fetchone()
+
+        return result is not None
+
+    def isTrustedIdentity(self, recipientId, identityKey):
+        q = "SELECT trust FROM identities WHERE recipient_id = ? " \
+            "AND public_key = ?"
+        c = self.dbConn.cursor()
+
+        c.execute(q, (recipientId, identityKey.getPublicKey().serialize()))
+        result = c.fetchone()
+
+        states = [UNTRUSTED, TRUSTED, UNDECIDED]
+
+        if result and result[0] in states:
+            return result[0]
+        else:
+            return True
+
+    def getAllFingerprints(self):
+        q = "SELECT _id, recipient_id, public_key, trust FROM identities " \
+            "WHERE recipient_id != -1 ORDER BY recipient_id ASC"
+        c = self.dbConn.cursor()
+
+        result = []
+        for row in c.execute(q):
+            result.append((row[0], row[1], row[2], row[3]))
+        return result
+
+    def getFingerprints(self, jid):
+        q = "SELECT _id, recipient_id, public_key, trust FROM identities " \
+            "WHERE recipient_id =? ORDER BY trust ASC"
+        c = self.dbConn.cursor()
+
+        result = []
+        c.execute(q, (jid,))
+        rows = c.fetchall()
+        for row in rows:
+            result.append((row[0], row[1], row[2], row[3]))
+        return result
+
+    def getTrustedFingerprints(self, jid):
+        q = "SELECT public_key FROM identities WHERE recipient_id = ? AND trust = ?"
+        c = self.dbConn.cursor()
+
+        result = []
+        c.execute(q, (jid, TRUSTED))
+        rows = c.fetchall()
+        for row in rows:
+            result.append(row[0])
+        return result
+
+    def getUndecidedFingerprints(self, jid):
+        q = "SELECT trust FROM identities WHERE recipient_id = ? AND trust = ?"
+        c = self.dbConn.cursor()
+
+        result = []
+        c.execute(q, (jid, UNDECIDED))
+        result = c.fetchall()
+
+        return result
+
+    def getNewFingerprints(self, jid):
+        q = "SELECT _id FROM identities WHERE shown = 0 AND " \
+            "recipient_id = ?"
+        c = self.dbConn.cursor()
+        result = []
+        for row in c.execute(q, (jid,)):
+            result.append(row[0])
+        return result
+
+    def setShownFingerprints(self, fingerprints):
+        q = "UPDATE identities SET shown = 1 WHERE _id IN ({})" \
+            .format(', '.join(['?'] * len(fingerprints)))
+        c = self.dbConn.cursor()
+        c.execute(q, fingerprints)
+        self.dbConn.commit()
+
+    def setTrust(self, _id, trust):
+        q = "UPDATE identities SET trust = ? WHERE _id = ?"
+        c = self.dbConn.cursor()
+        c.execute(q, (trust, _id))
+        self.dbConn.commit()