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
|
# -*- coding: utf-8 -*-
# pinned.py
# Copyright (C) 2013-2014 LEAP
#
# This program 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.
#
# This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
"""
Pinned Providers
"""
import logging
from leap.bitmask.provider import pinned_demobitmask
from leap.bitmask.provider import pinned_riseup
logger = logging.getLogger(__name__)
class PinnedProviders(object):
"""
Represents the providers that are pinned in Bitmask
"""
CONFIG_KEY = "config"
CACERT_KEY = "cacert"
PREFERRED_PROVIDER = pinned_demobitmask.DOMAIN
PROVIDERS = {
pinned_demobitmask.DOMAIN: {
CONFIG_KEY: pinned_demobitmask.PROVIDER_JSON,
CACERT_KEY: pinned_demobitmask.CACERT_PEM,
},
pinned_riseup.DOMAIN: {
CONFIG_KEY: pinned_riseup.PROVIDER_JSON,
CACERT_KEY: pinned_riseup.CACERT_PEM,
}
}
def __init__(self):
pass
@classmethod
def domains(self):
"""
Return the domains that are pinned in here.
The first domain in the list is the preferred one.
:rtype: list of str
"""
domains = self.PROVIDERS.keys()
domains.remove(self.PREFERRED_PROVIDER)
domains.insert(0, self.PREFERRED_PROVIDER)
return domains
@classmethod
def save_hardcoded(self, domain, provider_path, cacert_path):
"""
Save the pinned content for provider.json and cacert.pem to
the specified paths
:param domain: domain of the pinned provider
:type domain: str
:param provider_path: path where the pinned provider.json will
be saved
:type provider_path: str
:param cacert_path: path where the pinned cacert.pem will be
saved
:type cacert_path: str
"""
with open(provider_path, "w") as f:
f.write(self.PROVIDERS[domain][self.CONFIG_KEY])
with open(cacert_path, "w") as f:
f.write(self.PROVIDERS[domain][self.CACERT_KEY])
|
