globals.py 5.5 KB
Newer Older
Matt Traudt's avatar
Matt Traudt committed
1
import os
2
import logging
3
4
5
6
7
8
9
import platform

from requests import __version__ as requests_version
from stem import __version__ as stem_version

from sbws import __version__

Matt Traudt's avatar
Matt Traudt committed
10

juga's avatar
juga committed
11
12
13
from collections import OrderedDict


14
log = logging.getLogger(__name__)
Matt Traudt's avatar
Matt Traudt committed
15

16
RESULT_VERSION = 4
17
WIRE_VERSION = 1
juga's avatar
juga committed
18
SPEC_VERSION = '1.2.0'
19

20
21
# This is a dictionary of torrc options we always want to set when launching
# Tor and that do not depend on any runtime configuration
22
23
# Options that are known at runtime (from configuration file) are added
# in utils/stem.py launch_tor
24
25
26
27
28
29
30
31
TORRC_STARTING_POINT = {
    # We will find out via the ControlPort and not setting something static
    # means a lower chance of conflict
    'SocksPort': 'auto',
    # Easier than password authentication
    'CookieAuthentication': '1',
    # To avoid path bias warnings
    'UseEntryGuards': '0',
32
33
34
    # Because we need things from full server descriptors (namely for now: the
    # bandwidth line)
    'UseMicrodescriptors': '0',
juga's avatar
juga committed
35
36
37
38
    # useful logging options for clients that don't care about anonymity
    'SafeLogging': '0',
    'LogTimeGranularity': '1',
    'ProtocolWarnings': '1',
39
}
juga's avatar
juga committed
40
# Options that need to be set at runtime.
41
TORRC_RUNTIME_OPTIONS = {
juga's avatar
juga committed
42
43
    # The scanner builds the circuits to download the data itself,
    # so do not let Tor to build them.
44
    '__DisablePredictedCircuits': '1',
juga's avatar
juga committed
45
46
    # The scanner attach the streams to the circuit itself,
    # so do not let Tor to attache them.
47
48
    '__LeaveStreamsUnattached': '1',
}
juga's avatar
juga committed
49
50
51
52
53
54
55
# Options that can be set at runtime and can fail with some Tor versions
# The ones that fail will be ignored..
TORRC_OPTIONS_CAN_FAIL = OrderedDict({
    # Since currently scanner anonymity is not the goal, ConnectionPadding
    # is disable to do not send extra traffic
    'ConnectionPadding': '0'
    })
56

juga's avatar
juga committed
57
58
59
60
PKG_DIR = os.path.abspath(os.path.dirname(__file__))
DEFAULT_CONFIG_PATH = os.path.join(PKG_DIR, 'config.default.ini')
DEFAULT_LOG_CONFIG_PATH = os.path.join(PKG_DIR, 'config.log.default.ini')
USER_CONFIG_PATH = os.path.join(os.path.expanduser('~'), '.sbws.ini')
61
SUPERVISED_USER_CONFIG_PATH = "/etc/sbws/sbws.ini"
62
63
SUPERVISED_RUN_DPATH = "/run/sbws/tor"

juga's avatar
juga committed
64
SOCKET_TIMEOUT = 60  # seconds
65
TIMEOUT_MEASUREMENTS = 60 * 3  # 3 minutes
juga's avatar
juga committed
66
67
68
69
70
71
72
73

SBWS_SCALE_CONSTANT = 7500
TORFLOW_SCALING = 1
SBWS_SCALING = 2
TORFLOW_BW_MARGIN = 0.05
TORFLOW_OBS_LAST = 0
TORFLOW_OBS_MEAN = 1
TORFLOW_OBS_DECAYING = 3
74
TORFLOW_ROUND_DIG = 3
75
PROP276_ROUND_DIG = 2
juga's avatar
juga committed
76
77
DAY_SECS = 86400
NUM_MIN_RESULTS = 2
78
MIN_REPORT = 60
juga's avatar
juga committed
79
80
81
# Maximum difference between the total consensus bandwidth and the total in
# in the bandwidth lines in percentage
MAX_BW_DIFF_PERC = 50
juga's avatar
juga committed
82

83
84
85
# With the new KeyValues in #29591, the lines are greater than 510
# Tor already accept lines of any size, but leaving the limit anyway.
BW_LINE_SIZE = 1022
juga's avatar
juga committed
86

87
88
89
90
91
92
93
94
95
96
# RelayList, ResultDump, v3bwfile
# For how many seconds in the past the relays and measurements data is keep/
# considered valid.
# This is currently set by default in config.default.ini as ``date_period``,
# and used in ResultDump and v3bwfile.
# In a future refactor, constants in config.default.ini should be moved here,
# or calculated in settings, so that there's no need to pass the configuration
# to all the functions.
MEASUREMENTS_PERIOD = 5 * 24 * 60 * 60

97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
# Metadata to send in every requests, so that data servers can know which
# scanners are using them.
# In Requests these keys are case insensitive.
HTTP_HEADERS = {
    # This would be ignored if changing to HTTP/2
    'Connection': 'keep-alive',
    # Needs to get Tor version from the controller
    'User-Agent': 'sbws/{} ({}) Python/{} Requests/{} Stem/{} Tor/'.format(
                    __version__, platform.platform(),
                    platform.python_version(),
                    requests_version, stem_version),
    # Organization defined names (:rfc:`7239`)
    # Needs to get the nickname from the user config file.
    'Tor-Bandwidth-Scanner-Nickname': '{}',
    'Tor-Bandwidth-Scanner-UUID': '{}',
    # In case of including IP address.
    # 'Forwarded': 'for={}'  # IPv6 part, if there's
    }
# In the case of having ipv6 it's concatenated to forwarder.
IPV6_FORWARDED = ', for="[{}]"'

118
119
120
121
HTTP_GET_HEADERS = {
    'Range': '{}',
    'Accept-Encoding': 'identity',
}
122
DESTINATION_VERIFY_CERTIFICATE = True
123
124
# This number might need adjusted depending on the percentage of circuits and
# HTTP requests failures.
125
126
127
128
129

# Number of attempts to use a destination, that are stored, in order to decide
# whether the destination is functional or not.
NUM_DESTINATION_ATTEMPTS_STORED = 10
# Time to wait before trying again a destination that wasn't functional.
130
131
132
# Because intermitent failures with CDN destinations, start trying again
# after 5 min.
DELTA_SECONDS_RETRY_DESTINATION = 60 * 5
133
134
135
136
137
138
# Number of consecutive times a destination can fail before considering it
# not functional.
MAX_NUM_DESTINATION_FAILURES = 3
# By which factor to multiply DELTA_SECONDS_RETRY_DESTINATION when the
# destination fail again.
FACTOR_INCREMENT_DESTINATION_RETRY = 2
139

Matt Traudt's avatar
Matt Traudt committed
140

141
def fail_hard(*a, **kw):
Matt Traudt's avatar
Matt Traudt committed
142
    ''' Log something ... and then exit as fast as possible '''
Matt Traudt's avatar
Matt Traudt committed
143
    log.critical(*a, **kw)
144
    exit(1)
Matt Traudt's avatar
Matt Traudt committed
145
146


147
148
149
150
151
152
153
154
155
156
157
158
159
def touch_file(fname, times=None):
    '''
    If **fname** exists, update its last access and modified times to now. If
    **fname** does not exist, create it. If **times** are specified, pass them
    to os.utime for use.

    :param str fname: Name of file to update or create
    :param tuple times: 2-tuple of floats for access time and modified time
        respectively
    '''
    log.debug('Touching %s', fname)
    with open(fname, 'a') as fd:
        os.utime(fd.fileno(), times=times)