1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
from __future__ import absolute_import
from leap.common.events import (server as events_server)
from leap.soledad.common.errors import InvalidAuthTokenError
from pixelated.config import credentials
from pixelated.bitmask_libraries.config import LeapConfig
from pixelated.bitmask_libraries.certs import LeapCertificate
from pixelated.bitmask_libraries.provider import LeapProvider
from pixelated.bitmask_libraries.session import LeapSessionFactory
from twisted.internet import defer
import logging
log = logging.getLogger(__name__)
def initialize_leap_provider(provider_hostname, provider_cert, provider_fingerprint, leap_home):
LeapCertificate.set_cert_and_fingerprint(provider_cert,
provider_fingerprint)
config = LeapConfig(leap_home=leap_home, start_background_jobs=True)
provider = LeapProvider(provider_hostname, config)
provider.download_certificate()
LeapCertificate(provider).setup_ca_bundle()
return config, provider
@defer.inlineCallbacks
def initialize_leap_multi_user(provider_hostname,
leap_provider_cert,
leap_provider_cert_fingerprint,
credentials_file,
organization_mode,
leap_home):
config, provider = initialize_leap_provider(provider_hostname, leap_provider_cert, leap_provider_cert_fingerprint, leap_home)
defer.returnValue((config, provider))
def _create_session(provider, username, password, auth):
return LeapSessionFactory(provider).create(username, password, auth)
def _force_close_session(session):
try:
session.close()
except Exception, e:
log.error(e)
@defer.inlineCallbacks
def authenticate_user(provider, username, password, initial_sync=True, auth=None):
leap_session = _create_session(provider, username, password, auth)
try:
if initial_sync:
yield leap_session.initial_sync()
except InvalidAuthTokenError:
_force_close_session(leap_session)
leap_session = _create_session(provider, username, password, auth)
if initial_sync:
yield leap_session.initial_sync()
defer.returnValue(leap_session)
@defer.inlineCallbacks
def initialize_leap_single_user(leap_provider_cert,
leap_provider_cert_fingerprint,
credentials_file,
organization_mode,
leap_home,
initial_sync=True):
init_monkeypatches()
events_server.ensure_server()
provider, username, password = credentials.read(organization_mode,
credentials_file)
config, provider = initialize_leap_provider(provider, leap_provider_cert, leap_provider_cert_fingerprint, leap_home)
leap_session = yield authenticate_user(provider, username, password, initial_sync=initial_sync)
defer.returnValue(leap_session)
def init_monkeypatches():
import pixelated.extensions.requests_urllib3
|
