128 lines
3.6 KiB
Python
128 lines
3.6 KiB
Python
# -*- coding:utf-8 -*-
|
|
## src/secrets.py
|
|
##
|
|
## Copyright (C) 2007-2008 Brendan Taylor <whateley AT gmail.com>
|
|
## Copyright (C) 2008 Jonathan Schleifer <js-gajim AT webkeks.org>
|
|
##
|
|
## This file is part of Gajim.
|
|
##
|
|
## Gajim 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; version 3 only.
|
|
##
|
|
## Gajim 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 Gajim. If not, see <http://www.gnu.org/licenses/>.
|
|
##
|
|
|
|
from common.configpaths import gajimpaths
|
|
|
|
import Crypto
|
|
from common import crypto
|
|
from common import exceptions
|
|
|
|
import os
|
|
import pickle
|
|
|
|
secrets_filename = gajimpaths['SECRETS_FILE']
|
|
secrets_cache = None
|
|
|
|
class Secrets():
|
|
def __init__(self, filename):
|
|
self.filename = filename
|
|
self.srs = {}
|
|
self.pubkeys = {}
|
|
self.privkeys = {}
|
|
|
|
def cancel(self):
|
|
raise exceptions.Cancelled
|
|
|
|
def save(self):
|
|
f = open(secrets_filename, 'wb')
|
|
pickle.dump(self, f, protocol=2)
|
|
f.close()
|
|
|
|
def retained_secrets(self, account, bare_jid):
|
|
try:
|
|
return self.srs[account][bare_jid]
|
|
except KeyError:
|
|
return []
|
|
|
|
# retained secrets are stored as a tuple of the secret and whether the user
|
|
# has verified it
|
|
def save_new_srs(self, account, jid, secret, verified):
|
|
if not account in self.srs:
|
|
self.srs[account] = {}
|
|
|
|
if not jid in self.srs[account]:
|
|
self.srs[account][jid] = []
|
|
|
|
self.srs[account][jid].append((secret, verified))
|
|
|
|
self.save()
|
|
|
|
def find_srs(self, account, jid, srs):
|
|
our_secrets = self.srs[account][jid]
|
|
return [(x, y) for x, y in our_secrets if x == srs][0]
|
|
|
|
# has the user verified this retained secret?
|
|
def srs_verified(self, account, jid, srs):
|
|
return self.find_srs(account, jid, srs)[1]
|
|
|
|
def replace_srs(self, account, jid, old_secret, new_secret, verified):
|
|
our_secrets = self.srs[account][jid]
|
|
|
|
idx = our_secrets.index(self.find_srs(account, jid, old_secret))
|
|
|
|
our_secrets[idx] = (new_secret, verified)
|
|
|
|
self.save()
|
|
|
|
# the public key associated with 'account'
|
|
def my_pubkey(self, account):
|
|
try:
|
|
pk = self.privkeys[account]
|
|
except KeyError:
|
|
pk = Crypto.PublicKey.RSA.generate(2048, crypto.random_bytes)
|
|
|
|
self.privkeys[account] = pk
|
|
self.save()
|
|
|
|
return pk
|
|
|
|
def load_secrets(filename):
|
|
f = open(filename, 'rb')
|
|
|
|
try:
|
|
secrets = pickle.load(f, encoding='latin1')
|
|
# We do that to be able to read files written in py2
|
|
for acct in secrets.srs:
|
|
for jid in secrets.srs[acct]:
|
|
for (secret, verified) in list(secrets.srs[acct][jid]):
|
|
if type(secret) is str:
|
|
secrets.srs[acct][jid].remove((secret, verified))
|
|
secrets.srs[acct][jid].append((secret.encode('latin1'), verified))
|
|
except (KeyError, EOFError):
|
|
f.close()
|
|
secrets = Secrets(filename)
|
|
|
|
f.close()
|
|
return secrets
|
|
|
|
def secrets():
|
|
global secrets_cache
|
|
|
|
if secrets_cache:
|
|
return secrets_cache
|
|
|
|
if os.path.exists(secrets_filename):
|
|
secrets_cache = load_secrets(secrets_filename)
|
|
else:
|
|
secrets_cache = Secrets(secrets_filename)
|
|
|
|
return secrets_cache
|