dxs/onionr
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

bug fixes

Browse source
This commit is contained in:
Kevin Froman 2019-03-14 19:18:35 -05:00
parent 4757bb0963
commit 7f84cd0598
3 changed files with 9 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

11
onionr/onionrusers/onionrusers.py
View file

@ -33,7 +33,7 @@ def deleteExpiredKeys(coreInst):
return
def deleteTheirExpiredKeys(coreInst, pubkey):
conn = sqlite3.connect(self._core.peerDB, timeout=10)
conn = sqlite3.connect(coreInst.peerDB, timeout=10)
c = conn.cursor()
# Prepare the insert
@ -44,8 +44,8 @@ def deleteTheirExpiredKeys(coreInst, pubkey):
conn.commit()
conn.close()
DEFAULT_KEY_EXPIRE = 604800
#DEFAULT_KEY_EXPIRE = 600
#DEFAULT_KEY_EXPIRE = 604800
DEFAULT_KEY_EXPIRE = 600
class OnionrUser:
def __init__(self, coreInst, publicKey, saveUser=False):
@ -99,6 +99,7 @@ class OnionrUser:
return decrypted
def forwardEncrypt(self, data):
deleteTheirExpiredKeys(self._core, self.publicKey)
deleteExpiredKeys(self._core)
retData = ''
forwardKey = self._getLatestForwardKey()
@ -129,7 +130,7 @@ class OnionrUser:
c = conn.cursor()
# TODO: account for keys created at the same time (same epoch)
for row in c.execute("SELECT forwardKey, max(DATE) FROM forwardKeys WHERE peerKey = ?", (self.publicKey,)):
for row in c.execute("SELECT forwardKey, max(EXPIRE) FROM forwardKeys WHERE peerKey = ? ORDER BY expire DESC", (self.publicKey,)):
key = (row[0], row[1])
break
@ -143,7 +144,7 @@ class OnionrUser:
c = conn.cursor()
keyList = []
for row in c.execute("SELECT forwardKey, date FROM forwardKeys WHERE peerKey = ? ORDER BY date DESC", (self.publicKey,)):
for row in c.execute("SELECT forwardKey, date FROM forwardKeys WHERE peerKey = ? ORDER BY expire DESC", (self.publicKey,)):
keyList.append((row[0], row[1]))
conn.commit()