141 lines
4.4 KiB
Python
141 lines
4.4 KiB
Python
##
|
|
## Copyright (C) 2006 Gustavo J. A. M. Carneiro <gjcarneiro@gmail.com>
|
|
## Copyright (C) 2006 Nikos Kouremenos <kourem@gmail.com>
|
|
##
|
|
## 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/>.
|
|
##
|
|
|
|
__all__ = ['get_password', 'save_password']
|
|
|
|
from common import gajim
|
|
|
|
USER_HAS_GNOMEKEYRING = False
|
|
USER_USES_GNOMEKEYRING = False
|
|
gnomekeyring = None
|
|
|
|
class PasswordStorage(object):
|
|
def get_password(self, account_name):
|
|
raise NotImplementedError
|
|
def save_password(self, account_name, password):
|
|
raise NotImplementedError
|
|
|
|
|
|
class SimplePasswordStorage(PasswordStorage):
|
|
def get_password(self, account_name):
|
|
passwd = gajim.config.get_per('accounts', account_name, 'password')
|
|
if passwd and passwd.startswith('gnomekeyring:'):
|
|
return None # this is not a real password, it's a gnome keyring token
|
|
else:
|
|
return passwd
|
|
|
|
def save_password(self, account_name, password):
|
|
gajim.config.set_per('accounts', account_name, 'password', password)
|
|
gajim.connections[account_name].password = password
|
|
|
|
|
|
class GnomePasswordStorage(PasswordStorage):
|
|
def __init__(self):
|
|
# self.keyring = gnomekeyring.get_default_keyring_sync()
|
|
|
|
## above line commented and code below inserted as workaround
|
|
## for the bug http://bugzilla.gnome.org/show_bug.cgi?id=363019
|
|
self.keyring = "default"
|
|
try:
|
|
gnomekeyring.create_sync(self.keyring, None)
|
|
except gnomekeyring.AlreadyExistsError:
|
|
pass
|
|
|
|
def get_password(self, account_name):
|
|
conf = gajim.config.get_per('accounts', account_name, 'password')
|
|
if conf is None:
|
|
return None
|
|
try:
|
|
unused, auth_token = conf.split('gnomekeyring:')
|
|
auth_token = int(auth_token)
|
|
except ValueError:
|
|
password = conf
|
|
## migrate the password over to keyring
|
|
try:
|
|
self.save_password(account_name, password, update=False)
|
|
except gnomekeyring.NoKeyringDaemonError:
|
|
## no keyring daemon: in the future, stop using it
|
|
set_storage(SimplePasswordStorage())
|
|
return password
|
|
try:
|
|
return gnomekeyring.item_get_info_sync(self.keyring,
|
|
auth_token).get_secret()
|
|
except gnomekeyring.DeniedError:
|
|
return None
|
|
except gnomekeyring.NoKeyringDaemonError:
|
|
## no keyring daemon: in the future, stop using it
|
|
set_storage(SimplePasswordStorage())
|
|
return None
|
|
|
|
def save_password(self, account_name, password, update=True):
|
|
display_name = _('Gajim account %s') % account_name
|
|
attributes = dict(account_name=str(account_name), gajim=1)
|
|
try:
|
|
auth_token = gnomekeyring.item_create_sync(
|
|
self.keyring, gnomekeyring.ITEM_GENERIC_SECRET,
|
|
display_name, attributes, password, update)
|
|
except gnomekeyring.DeniedError:
|
|
set_storage(SimplePasswordStorage())
|
|
storage.save_password(account_name, password)
|
|
return
|
|
token = 'gnomekeyring:%i' % auth_token
|
|
gajim.config.set_per('accounts', account_name, 'password', token)
|
|
if gajim.connections.has_key(account_name):
|
|
gajim.connections[account_name].password = password
|
|
|
|
storage = None
|
|
def get_storage():
|
|
global storage
|
|
if storage is None: # None is only in first time get_storage is called
|
|
if gajim.config.get('use_gnomekeyring'):
|
|
global gnomekeyring
|
|
try:
|
|
import gnomekeyring
|
|
except ImportError:
|
|
pass
|
|
else:
|
|
global USER_HAS_GNOMEKEYRING
|
|
global USER_USES_GNOMEKEYRING
|
|
USER_HAS_GNOMEKEYRING = True
|
|
if gnomekeyring.is_available():
|
|
USER_USES_GNOMEKEYRING = True
|
|
else:
|
|
USER_USES_GNOMEKEYRING = False
|
|
if USER_USES_GNOMEKEYRING:
|
|
try:
|
|
storage = GnomePasswordStorage()
|
|
except gnomekeyring.NoKeyringDaemonError:
|
|
storage = SimplePasswordStorage()
|
|
except gnomekeyring.DeniedError:
|
|
storage = SimplePasswordStorage()
|
|
else:
|
|
storage = SimplePasswordStorage()
|
|
return storage
|
|
|
|
def set_storage(storage_):
|
|
global storage
|
|
storage = storage_
|
|
|
|
|
|
def get_password(account_name):
|
|
return get_storage().get_password(account_name)
|
|
|
|
def save_password(account_name, password):
|
|
return get_storage().save_password(account_name, password)
|