updateFallbackDirs.py 92 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312
  1. #!/usr/bin/python
  2. # Usage:
  3. #
  4. # Regenerate the list:
  5. # scripts/maint/updateFallbackDirs.py > src/or/fallback_dirs.inc 2> fallback_dirs.log
  6. #
  7. # Check the existing list:
  8. # scripts/maint/updateFallbackDirs.py check_existing > fallback_dirs.inc.ok 2> fallback_dirs.log
  9. # mv fallback_dirs.inc.ok src/or/fallback_dirs.inc
  10. #
  11. # This script should be run from a stable, reliable network connection,
  12. # with no other network activity (and not over tor).
  13. # If this is not possible, please disable:
  14. # PERFORM_IPV4_DIRPORT_CHECKS and PERFORM_IPV6_DIRPORT_CHECKS
  15. #
  16. # Needs dateutil (and potentially other python packages)
  17. # Needs stem available in your PYTHONPATH, or just ln -s ../stem/stem .
  18. # Optionally uses ipaddress (python 3 builtin) or py2-ipaddress (package)
  19. # for netblock analysis, in PYTHONPATH, or just
  20. # ln -s ../py2-ipaddress-3.4.1/ipaddress.py .
  21. #
  22. # Then read the logs to make sure the fallbacks aren't dominated by a single
  23. # netblock or port
  24. # Script by weasel, April 2015
  25. # Portions by gsathya & karsten, 2013
  26. # https://trac.torproject.org/projects/tor/attachment/ticket/8374/dir_list.2.py
  27. # Modifications by teor, 2015
  28. import StringIO
  29. import string
  30. import re
  31. import datetime
  32. import gzip
  33. import os.path
  34. import json
  35. import math
  36. import sys
  37. import urllib
  38. import urllib2
  39. import hashlib
  40. import dateutil.parser
  41. # bson_lazy provides bson
  42. #from bson import json_util
  43. import copy
  44. import re
  45. from stem.descriptor import DocumentHandler
  46. from stem.descriptor.remote import get_consensus, get_server_descriptors, MAX_FINGERPRINTS
  47. import logging
  48. logging.root.name = ''
  49. HAVE_IPADDRESS = False
  50. try:
  51. # python 3 builtin, or install package py2-ipaddress
  52. # there are several ipaddress implementations for python 2
  53. # with slightly different semantics with str typed text
  54. # fortunately, all our IP addresses are in unicode
  55. import ipaddress
  56. HAVE_IPADDRESS = True
  57. except ImportError:
  58. # if this happens, we avoid doing netblock analysis
  59. logging.warning('Unable to import ipaddress, please install py2-ipaddress.' +
  60. ' A fallback list will be created, but optional netblock' +
  61. ' analysis will not be performed.')
  62. ## Top-Level Configuration
  63. # We use semantic versioning: https://semver.org
  64. # In particular:
  65. # * major changes include removing a mandatory field, or anything else that
  66. # would break an appropriately tolerant parser,
  67. # * minor changes include adding a field,
  68. # * patch changes include changing header comments or other unstructured
  69. # content
  70. FALLBACK_FORMAT_VERSION = '2.0.0'
  71. # Output all candidate fallbacks, or only output selected fallbacks?
  72. OUTPUT_CANDIDATES = False
  73. # Perform DirPort checks over IPv4?
  74. # Change this to False if IPv4 doesn't work for you, or if you don't want to
  75. # download a consensus for each fallback
  76. # Don't check ~1000 candidates when OUTPUT_CANDIDATES is True
  77. PERFORM_IPV4_DIRPORT_CHECKS = False if OUTPUT_CANDIDATES else True
  78. # Perform DirPort checks over IPv6?
  79. # If you know IPv6 works for you, set this to True
  80. # This will exclude IPv6 relays without an IPv6 DirPort configured
  81. # So it's best left at False until #18394 is implemented
  82. # Don't check ~1000 candidates when OUTPUT_CANDIDATES is True
  83. PERFORM_IPV6_DIRPORT_CHECKS = False if OUTPUT_CANDIDATES else False
  84. # Must relays be running now?
  85. MUST_BE_RUNNING_NOW = (PERFORM_IPV4_DIRPORT_CHECKS
  86. or PERFORM_IPV6_DIRPORT_CHECKS)
  87. # Clients have been using microdesc consensuses by default for a while now
  88. DOWNLOAD_MICRODESC_CONSENSUS = True
  89. # If a relay delivers an expired consensus, if it expired less than this many
  90. # seconds ago, we still allow the relay. This should never be less than -90,
  91. # as all directory mirrors should have downloaded a consensus 90 minutes
  92. # before it expires. It should never be more than 24 hours, because clients
  93. # reject consensuses that are older than REASONABLY_LIVE_TIME.
  94. # For the consensus expiry check to be accurate, the machine running this
  95. # script needs an accurate clock.
  96. #
  97. # Relays on 0.3.0 and later return a 404 when they are about to serve an
  98. # expired consensus. This makes them fail the download check.
  99. # We use a tolerance of 0, so that 0.2.x series relays also fail the download
  100. # check if they serve an expired consensus.
  101. CONSENSUS_EXPIRY_TOLERANCE = 0
  102. # Output fallback name, flags, bandwidth, and ContactInfo in a C comment?
  103. OUTPUT_COMMENTS = True if OUTPUT_CANDIDATES else False
  104. # Output matching ContactInfo in fallbacks list or the blacklist?
  105. # Useful if you're trying to contact operators
  106. CONTACT_COUNT = True if OUTPUT_CANDIDATES else False
  107. CONTACT_BLACKLIST_COUNT = True if OUTPUT_CANDIDATES else False
  108. # How the list should be sorted:
  109. # fingerprint: is useful for stable diffs of fallback lists
  110. # measured_bandwidth: is useful when pruning the list based on bandwidth
  111. # contact: is useful for contacting operators once the list has been pruned
  112. OUTPUT_SORT_FIELD = 'contact' if OUTPUT_CANDIDATES else 'fingerprint'
  113. ## OnionOO Settings
  114. ONIONOO = 'https://onionoo.torproject.org/'
  115. #ONIONOO = 'https://onionoo.thecthulhu.com/'
  116. # Don't bother going out to the Internet, just use the files available locally,
  117. # even if they're very old
  118. LOCAL_FILES_ONLY = False
  119. ## Whitelist / Blacklist Filter Settings
  120. # The whitelist contains entries that are included if all attributes match
  121. # (IPv4, dirport, orport, id, and optionally IPv6 and IPv6 orport)
  122. # The blacklist contains (partial) entries that are excluded if any
  123. # sufficiently specific group of attributes matches:
  124. # IPv4 & DirPort
  125. # IPv4 & ORPort
  126. # ID
  127. # IPv6 & DirPort
  128. # IPv6 & IPv6 ORPort
  129. # If neither port is included in the blacklist, the entire IP address is
  130. # blacklisted.
  131. # What happens to entries in neither list?
  132. # When True, they are included, when False, they are excluded
  133. INCLUDE_UNLISTED_ENTRIES = True if OUTPUT_CANDIDATES else False
  134. # If an entry is in both lists, what happens?
  135. # When True, it is excluded, when False, it is included
  136. BLACKLIST_EXCLUDES_WHITELIST_ENTRIES = True
  137. WHITELIST_FILE_NAME = 'scripts/maint/fallback.whitelist'
  138. BLACKLIST_FILE_NAME = 'scripts/maint/fallback.blacklist'
  139. FALLBACK_FILE_NAME = 'src/or/fallback_dirs.inc'
  140. # The number of bytes we'll read from a filter file before giving up
  141. MAX_LIST_FILE_SIZE = 1024 * 1024
  142. ## Eligibility Settings
  143. # Require fallbacks to have the same address and port for a set amount of time
  144. # We used to have this at 1 week, but that caused many fallback failures, which
  145. # meant that we had to rebuild the list more often.
  146. #
  147. # There was a bug in Tor 0.2.8.1-alpha and earlier where a relay temporarily
  148. # submits a 0 DirPort when restarted.
  149. # This causes OnionOO to (correctly) reset its stability timer.
  150. # Affected relays should upgrade to Tor 0.2.8.7 or later, which has a fix
  151. # for this issue.
  152. ADDRESS_AND_PORT_STABLE_DAYS = 30
  153. # We ignore relays that have been down for more than this period
  154. MAX_DOWNTIME_DAYS = 0 if MUST_BE_RUNNING_NOW else 7
  155. # What time-weighted-fraction of these flags must FallbackDirs
  156. # Equal or Exceed?
  157. CUTOFF_RUNNING = .90
  158. CUTOFF_V2DIR = .90
  159. # Tolerate lower guard flag averages, as guard flags are removed for some time
  160. # after a relay restarts
  161. CUTOFF_GUARD = .80
  162. # What time-weighted-fraction of these flags must FallbackDirs
  163. # Equal or Fall Under?
  164. # .00 means no bad exits
  165. PERMITTED_BADEXIT = .00
  166. # older entries' weights are adjusted with ALPHA^(age in days)
  167. AGE_ALPHA = 0.99
  168. # this factor is used to scale OnionOO entries to [0,1]
  169. ONIONOO_SCALE_ONE = 999.
  170. ## Fallback Count Limits
  171. # The target for these parameters is 20% of the guards in the network
  172. # This is around 200 as of October 2015
  173. _FB_POG = 0.2
  174. FALLBACK_PROPORTION_OF_GUARDS = None if OUTPUT_CANDIDATES else _FB_POG
  175. # Limit the number of fallbacks (eliminating lowest by advertised bandwidth)
  176. MAX_FALLBACK_COUNT = None if OUTPUT_CANDIDATES else 200
  177. # Emit a C #error if the number of fallbacks is less than expected
  178. MIN_FALLBACK_COUNT = 0 if OUTPUT_CANDIDATES else MAX_FALLBACK_COUNT*0.5
  179. # The maximum number of fallbacks on the same address, contact, or family
  180. # With 200 fallbacks, this means each operator can see 1% of client bootstraps
  181. # (The directory authorities used to see ~12% of client bootstraps each.)
  182. MAX_FALLBACKS_PER_IP = 1
  183. MAX_FALLBACKS_PER_IPV4 = MAX_FALLBACKS_PER_IP
  184. MAX_FALLBACKS_PER_IPV6 = MAX_FALLBACKS_PER_IP
  185. MAX_FALLBACKS_PER_CONTACT = 3
  186. MAX_FALLBACKS_PER_FAMILY = 3
  187. ## Fallback Bandwidth Requirements
  188. # Any fallback with the Exit flag has its bandwidth multipled by this fraction
  189. # to make sure we aren't further overloading exits
  190. # (Set to 1.0, because we asked that only lightly loaded exits opt-in,
  191. # and the extra load really isn't that much for large relays.)
  192. EXIT_BANDWIDTH_FRACTION = 1.0
  193. # If a single fallback's bandwidth is too low, it's pointless adding it
  194. # We expect fallbacks to handle an extra 10 kilobytes per second of traffic
  195. # Make sure they can support a hundred times the expected extra load
  196. # (Use 102.4 to make it come out nicely in MByte/s)
  197. # We convert this to a consensus weight before applying the filter,
  198. # because all the bandwidth amounts are specified by the relay
  199. MIN_BANDWIDTH = 102.4 * 10.0 * 1024.0
  200. # Clients will time out after 30 seconds trying to download a consensus
  201. # So allow fallback directories half that to deliver a consensus
  202. # The exact download times might change based on the network connection
  203. # running this script, but only by a few seconds
  204. # There is also about a second of python overhead
  205. CONSENSUS_DOWNLOAD_SPEED_MAX = 15.0
  206. # If the relay fails a consensus check, retry the download
  207. # This avoids delisting a relay due to transient network conditions
  208. CONSENSUS_DOWNLOAD_RETRY = True
  209. ## Parsing Functions
  210. def parse_ts(t):
  211. return datetime.datetime.strptime(t, "%Y-%m-%d %H:%M:%S")
  212. def remove_bad_chars(raw_string, bad_char_list):
  213. # Remove each character in the bad_char_list
  214. cleansed_string = raw_string
  215. for c in bad_char_list:
  216. cleansed_string = cleansed_string.replace(c, '')
  217. return cleansed_string
  218. def cleanse_unprintable(raw_string):
  219. # Remove all unprintable characters
  220. cleansed_string = ''
  221. for c in raw_string:
  222. if c in string.printable:
  223. cleansed_string += c
  224. return cleansed_string
  225. def cleanse_whitespace(raw_string):
  226. # Replace all whitespace characters with a space
  227. cleansed_string = raw_string
  228. for c in string.whitespace:
  229. cleansed_string = cleansed_string.replace(c, ' ')
  230. return cleansed_string
  231. def cleanse_c_multiline_comment(raw_string):
  232. cleansed_string = raw_string
  233. # Embedded newlines should be removed by tor/onionoo, but let's be paranoid
  234. cleansed_string = cleanse_whitespace(cleansed_string)
  235. # ContactInfo and Version can be arbitrary binary data
  236. cleansed_string = cleanse_unprintable(cleansed_string)
  237. # Prevent a malicious / unanticipated string from breaking out
  238. # of a C-style multiline comment
  239. # This removes '/*' and '*/' and '//'
  240. bad_char_list = '*/'
  241. # Prevent a malicious string from using C nulls
  242. bad_char_list += '\0'
  243. # Avoid confusing parsers by making sure there is only one comma per fallback
  244. bad_char_list += ','
  245. # Avoid confusing parsers by making sure there is only one equals per field
  246. bad_char_list += '='
  247. # Be safer by removing bad characters entirely
  248. cleansed_string = remove_bad_chars(cleansed_string, bad_char_list)
  249. # Some compilers may further process the content of comments
  250. # There isn't much we can do to cover every possible case
  251. # But comment-based directives are typically only advisory
  252. return cleansed_string
  253. def cleanse_c_string(raw_string):
  254. cleansed_string = raw_string
  255. # Embedded newlines should be removed by tor/onionoo, but let's be paranoid
  256. cleansed_string = cleanse_whitespace(cleansed_string)
  257. # ContactInfo and Version can be arbitrary binary data
  258. cleansed_string = cleanse_unprintable(cleansed_string)
  259. # Prevent a malicious address/fingerprint string from breaking out
  260. # of a C-style string
  261. bad_char_list = '"'
  262. # Prevent a malicious string from using escapes
  263. bad_char_list += '\\'
  264. # Prevent a malicious string from using C nulls
  265. bad_char_list += '\0'
  266. # Avoid confusing parsers by making sure there is only one comma per fallback
  267. bad_char_list += ','
  268. # Avoid confusing parsers by making sure there is only one equals per field
  269. bad_char_list += '='
  270. # Be safer by removing bad characters entirely
  271. cleansed_string = remove_bad_chars(cleansed_string, bad_char_list)
  272. # Some compilers may further process the content of strings
  273. # There isn't much we can do to cover every possible case
  274. # But this typically only results in changes to the string data
  275. return cleansed_string
  276. ## OnionOO Source Functions
  277. # a dictionary of source metadata for each onionoo query we've made
  278. fetch_source = {}
  279. # register source metadata for 'what'
  280. # assumes we only retrieve one document for each 'what'
  281. def register_fetch_source(what, url, relays_published, version):
  282. fetch_source[what] = {}
  283. fetch_source[what]['url'] = url
  284. fetch_source[what]['relays_published'] = relays_published
  285. fetch_source[what]['version'] = version
  286. # list each registered source's 'what'
  287. def fetch_source_list():
  288. return sorted(fetch_source.keys())
  289. # given 'what', provide a multiline C comment describing the source
  290. def describe_fetch_source(what):
  291. desc = '/*'
  292. desc += '\n'
  293. desc += 'Onionoo Source: '
  294. desc += cleanse_c_multiline_comment(what)
  295. desc += ' Date: '
  296. desc += cleanse_c_multiline_comment(fetch_source[what]['relays_published'])
  297. desc += ' Version: '
  298. desc += cleanse_c_multiline_comment(fetch_source[what]['version'])
  299. desc += '\n'
  300. desc += 'URL: '
  301. desc += cleanse_c_multiline_comment(fetch_source[what]['url'])
  302. desc += '\n'
  303. desc += '*/'
  304. return desc
  305. ## File Processing Functions
  306. def write_to_file(str, file_name, max_len):
  307. try:
  308. with open(file_name, 'w') as f:
  309. f.write(str[0:max_len])
  310. except EnvironmentError, error:
  311. logging.error('Writing file %s failed: %d: %s'%
  312. (file_name,
  313. error.errno,
  314. error.strerror)
  315. )
  316. def read_from_file(file_name, max_len):
  317. try:
  318. if os.path.isfile(file_name):
  319. with open(file_name, 'r') as f:
  320. return f.read(max_len)
  321. except EnvironmentError, error:
  322. logging.info('Loading file %s failed: %d: %s'%
  323. (file_name,
  324. error.errno,
  325. error.strerror)
  326. )
  327. return None
  328. def parse_fallback_file(file_name):
  329. file_data = read_from_file(file_name, MAX_LIST_FILE_SIZE)
  330. file_data = cleanse_unprintable(file_data)
  331. file_data = remove_bad_chars(file_data, '\n"\0')
  332. file_data = re.sub('/\*.*?\*/', '', file_data)
  333. file_data = file_data.replace(',', '\n')
  334. file_data = file_data.replace(' weight=10', '')
  335. return file_data
  336. def load_possibly_compressed_response_json(response):
  337. if response.info().get('Content-Encoding') == 'gzip':
  338. buf = StringIO.StringIO( response.read() )
  339. f = gzip.GzipFile(fileobj=buf)
  340. return json.load(f)
  341. else:
  342. return json.load(response)
  343. def load_json_from_file(json_file_name):
  344. # An exception here may be resolved by deleting the .last_modified
  345. # and .json files, and re-running the script
  346. try:
  347. with open(json_file_name, 'r') as f:
  348. return json.load(f)
  349. except EnvironmentError, error:
  350. raise Exception('Reading not-modified json file %s failed: %d: %s'%
  351. (json_file_name,
  352. error.errno,
  353. error.strerror)
  354. )
  355. ## OnionOO Functions
  356. def datestr_to_datetime(datestr):
  357. # Parse datetimes like: Fri, 02 Oct 2015 13:34:14 GMT
  358. if datestr is not None:
  359. dt = dateutil.parser.parse(datestr)
  360. else:
  361. # Never modified - use start of epoch
  362. dt = datetime.datetime.utcfromtimestamp(0)
  363. # strip any timezone out (in case they're supported in future)
  364. dt = dt.replace(tzinfo=None)
  365. return dt
  366. def onionoo_fetch(what, **kwargs):
  367. params = kwargs
  368. params['type'] = 'relay'
  369. #params['limit'] = 10
  370. params['first_seen_days'] = '%d-'%(ADDRESS_AND_PORT_STABLE_DAYS)
  371. params['last_seen_days'] = '-%d'%(MAX_DOWNTIME_DAYS)
  372. params['flag'] = 'V2Dir'
  373. url = ONIONOO + what + '?' + urllib.urlencode(params)
  374. # Unfortunately, the URL is too long for some OS filenames,
  375. # but we still don't want to get files from different URLs mixed up
  376. base_file_name = what + '-' + hashlib.sha1(url).hexdigest()
  377. full_url_file_name = base_file_name + '.full_url'
  378. MAX_FULL_URL_LENGTH = 1024
  379. last_modified_file_name = base_file_name + '.last_modified'
  380. MAX_LAST_MODIFIED_LENGTH = 64
  381. json_file_name = base_file_name + '.json'
  382. if LOCAL_FILES_ONLY:
  383. # Read from the local file, don't write to anything
  384. response_json = load_json_from_file(json_file_name)
  385. else:
  386. # store the full URL to a file for debugging
  387. # no need to compare as long as you trust SHA-1
  388. write_to_file(url, full_url_file_name, MAX_FULL_URL_LENGTH)
  389. request = urllib2.Request(url)
  390. request.add_header('Accept-encoding', 'gzip')
  391. # load the last modified date from the file, if it exists
  392. last_mod_date = read_from_file(last_modified_file_name,
  393. MAX_LAST_MODIFIED_LENGTH)
  394. if last_mod_date is not None:
  395. request.add_header('If-modified-since', last_mod_date)
  396. # Parse last modified date
  397. last_mod = datestr_to_datetime(last_mod_date)
  398. # Not Modified and still recent enough to be useful
  399. # Onionoo / Globe used to use 6 hours, but we can afford a day
  400. required_freshness = datetime.datetime.utcnow()
  401. # strip any timezone out (to match dateutil.parser)
  402. required_freshness = required_freshness.replace(tzinfo=None)
  403. required_freshness -= datetime.timedelta(hours=24)
  404. # Make the OnionOO request
  405. response_code = 0
  406. try:
  407. response = urllib2.urlopen(request)
  408. response_code = response.getcode()
  409. except urllib2.HTTPError, error:
  410. response_code = error.code
  411. if response_code == 304: # not modified
  412. pass
  413. else:
  414. raise Exception("Could not get " + url + ": "
  415. + str(error.code) + ": " + error.reason)
  416. if response_code == 200: # OK
  417. last_mod = datestr_to_datetime(response.info().get('Last-Modified'))
  418. # Check for freshness
  419. if last_mod < required_freshness:
  420. if last_mod_date is not None:
  421. # This check sometimes fails transiently, retry the script if it does
  422. date_message = "Outdated data: last updated " + last_mod_date
  423. else:
  424. date_message = "No data: never downloaded "
  425. raise Exception(date_message + " from " + url)
  426. # Process the data
  427. if response_code == 200: # OK
  428. response_json = load_possibly_compressed_response_json(response)
  429. with open(json_file_name, 'w') as f:
  430. # use the most compact json representation to save space
  431. json.dump(response_json, f, separators=(',',':'))
  432. # store the last modified date in its own file
  433. if response.info().get('Last-modified') is not None:
  434. write_to_file(response.info().get('Last-Modified'),
  435. last_modified_file_name,
  436. MAX_LAST_MODIFIED_LENGTH)
  437. elif response_code == 304: # Not Modified
  438. response_json = load_json_from_file(json_file_name)
  439. else: # Unexpected HTTP response code not covered in the HTTPError above
  440. raise Exception("Unexpected HTTP response code to " + url + ": "
  441. + str(response_code))
  442. register_fetch_source(what,
  443. url,
  444. response_json['relays_published'],
  445. response_json['version'])
  446. return response_json
  447. def fetch(what, **kwargs):
  448. #x = onionoo_fetch(what, **kwargs)
  449. # don't use sort_keys, as the order of or_addresses is significant
  450. #print json.dumps(x, indent=4, separators=(',', ': '))
  451. #sys.exit(0)
  452. return onionoo_fetch(what, **kwargs)
  453. ## Fallback Candidate Class
  454. class Candidate(object):
  455. CUTOFF_ADDRESS_AND_PORT_STABLE = (datetime.datetime.utcnow()
  456. - datetime.timedelta(ADDRESS_AND_PORT_STABLE_DAYS))
  457. def __init__(self, details):
  458. for f in ['fingerprint', 'nickname', 'last_changed_address_or_port',
  459. 'consensus_weight', 'or_addresses', 'dir_address']:
  460. if not f in details: raise Exception("Document has no %s field."%(f,))
  461. if not 'contact' in details:
  462. details['contact'] = None
  463. if not 'flags' in details or details['flags'] is None:
  464. details['flags'] = []
  465. if (not 'advertised_bandwidth' in details
  466. or details['advertised_bandwidth'] is None):
  467. # relays without advertised bandwdith have it calculated from their
  468. # consensus weight
  469. details['advertised_bandwidth'] = 0
  470. if (not 'effective_family' in details
  471. or details['effective_family'] is None):
  472. details['effective_family'] = []
  473. if not 'platform' in details:
  474. details['platform'] = None
  475. details['last_changed_address_or_port'] = parse_ts(
  476. details['last_changed_address_or_port'])
  477. self._data = details
  478. self._stable_sort_or_addresses()
  479. self._fpr = self._data['fingerprint']
  480. self._running = self._guard = self._v2dir = 0.
  481. self._split_dirport()
  482. self._compute_orport()
  483. if self.orport is None:
  484. raise Exception("Failed to get an orport for %s."%(self._fpr,))
  485. self._compute_ipv6addr()
  486. if not self.has_ipv6():
  487. logging.debug("Failed to get an ipv6 address for %s."%(self._fpr,))
  488. self._compute_version()
  489. self._extra_info_cache = None
  490. def _stable_sort_or_addresses(self):
  491. # replace self._data['or_addresses'] with a stable ordering,
  492. # sorting the secondary addresses in string order
  493. # leave the received order in self._data['or_addresses_raw']
  494. self._data['or_addresses_raw'] = self._data['or_addresses']
  495. or_address_primary = self._data['or_addresses'][:1]
  496. # subsequent entries in the or_addresses array are in an arbitrary order
  497. # so we stabilise the addresses by sorting them in string order
  498. or_addresses_secondaries_stable = sorted(self._data['or_addresses'][1:])
  499. or_addresses_stable = or_address_primary + or_addresses_secondaries_stable
  500. self._data['or_addresses'] = or_addresses_stable
  501. def get_fingerprint(self):
  502. return self._fpr
  503. # is_valid_ipv[46]_address by gsathya, karsten, 2013
  504. @staticmethod
  505. def is_valid_ipv4_address(address):
  506. if not isinstance(address, (str, unicode)):
  507. return False
  508. # check if there are four period separated values
  509. if address.count(".") != 3:
  510. return False
  511. # checks that each value in the octet are decimal values between 0-255
  512. for entry in address.split("."):
  513. if not entry.isdigit() or int(entry) < 0 or int(entry) > 255:
  514. return False
  515. elif entry[0] == "0" and len(entry) > 1:
  516. return False # leading zeros, for instance in "1.2.3.001"
  517. return True
  518. @staticmethod
  519. def is_valid_ipv6_address(address):
  520. if not isinstance(address, (str, unicode)):
  521. return False
  522. # remove brackets
  523. address = address[1:-1]
  524. # addresses are made up of eight colon separated groups of four hex digits
  525. # with leading zeros being optional
  526. # https://en.wikipedia.org/wiki/IPv6#Address_format
  527. colon_count = address.count(":")
  528. if colon_count > 7:
  529. return False # too many groups
  530. elif colon_count != 7 and not "::" in address:
  531. return False # not enough groups and none are collapsed
  532. elif address.count("::") > 1 or ":::" in address:
  533. return False # multiple groupings of zeros can't be collapsed
  534. found_ipv4_on_previous_entry = False
  535. for entry in address.split(":"):
  536. # If an IPv6 address has an embedded IPv4 address,
  537. # it must be the last entry
  538. if found_ipv4_on_previous_entry:
  539. return False
  540. if not re.match("^[0-9a-fA-f]{0,4}$", entry):
  541. if not Candidate.is_valid_ipv4_address(entry):
  542. return False
  543. else:
  544. found_ipv4_on_previous_entry = True
  545. return True
  546. def _split_dirport(self):
  547. # Split the dir_address into dirip and dirport
  548. (self.dirip, _dirport) = self._data['dir_address'].split(':', 2)
  549. self.dirport = int(_dirport)
  550. def _compute_orport(self):
  551. # Choose the first ORPort that's on the same IPv4 address as the DirPort.
  552. # In rare circumstances, this might not be the primary ORPort address.
  553. # However, _stable_sort_or_addresses() ensures we choose the same one
  554. # every time, even if onionoo changes the order of the secondaries.
  555. self._split_dirport()
  556. self.orport = None
  557. for i in self._data['or_addresses']:
  558. if i != self._data['or_addresses'][0]:
  559. logging.debug('Secondary IPv4 Address Used for %s: %s'%(self._fpr, i))
  560. (ipaddr, port) = i.rsplit(':', 1)
  561. if (ipaddr == self.dirip) and Candidate.is_valid_ipv4_address(ipaddr):
  562. self.orport = int(port)
  563. return
  564. def _compute_ipv6addr(self):
  565. # Choose the first IPv6 address that uses the same port as the ORPort
  566. # Or, choose the first IPv6 address in the list
  567. # _stable_sort_or_addresses() ensures we choose the same IPv6 address
  568. # every time, even if onionoo changes the order of the secondaries.
  569. self.ipv6addr = None
  570. self.ipv6orport = None
  571. # Choose the first IPv6 address that uses the same port as the ORPort
  572. for i in self._data['or_addresses']:
  573. (ipaddr, port) = i.rsplit(':', 1)
  574. if (port == self.orport) and Candidate.is_valid_ipv6_address(ipaddr):
  575. self.ipv6addr = ipaddr
  576. self.ipv6orport = int(port)
  577. return
  578. # Choose the first IPv6 address in the list
  579. for i in self._data['or_addresses']:
  580. (ipaddr, port) = i.rsplit(':', 1)
  581. if Candidate.is_valid_ipv6_address(ipaddr):
  582. self.ipv6addr = ipaddr
  583. self.ipv6orport = int(port)
  584. return
  585. def _compute_version(self):
  586. # parse the version out of the platform string
  587. # The platform looks like: "Tor 0.2.7.6 on Linux"
  588. self._data['version'] = None
  589. if self._data['platform'] is None:
  590. return
  591. # be tolerant of weird whitespacing, use a whitespace split
  592. tokens = self._data['platform'].split()
  593. for token in tokens:
  594. vnums = token.split('.')
  595. # if it's at least a.b.c.d, with potentially an -alpha-dev, -alpha, -rc
  596. if (len(vnums) >= 4 and vnums[0].isdigit() and vnums[1].isdigit() and
  597. vnums[2].isdigit()):
  598. self._data['version'] = token
  599. return
  600. # From #20509
  601. # bug #20499 affects versions from 0.2.9.1-alpha-dev to 0.2.9.4-alpha-dev
  602. # and version 0.3.0.0-alpha-dev
  603. # Exhaustive lists are hard to get wrong
  604. STALE_CONSENSUS_VERSIONS = ['0.2.9.1-alpha-dev',
  605. '0.2.9.2-alpha',
  606. '0.2.9.2-alpha-dev',
  607. '0.2.9.3-alpha',
  608. '0.2.9.3-alpha-dev',
  609. '0.2.9.4-alpha',
  610. '0.2.9.4-alpha-dev',
  611. '0.3.0.0-alpha-dev'
  612. ]
  613. def is_valid_version(self):
  614. # call _compute_version before calling this
  615. # is the version of the relay a version we want as a fallback?
  616. # checks both recommended versions and bug #20499 / #20509
  617. #
  618. # if the relay doesn't have a recommended version field, exclude the relay
  619. if not self._data.has_key('recommended_version'):
  620. log_excluded('%s not a candidate: no recommended_version field',
  621. self._fpr)
  622. return False
  623. if not self._data['recommended_version']:
  624. log_excluded('%s not a candidate: version not recommended', self._fpr)
  625. return False
  626. # if the relay doesn't have version field, exclude the relay
  627. if not self._data.has_key('version'):
  628. log_excluded('%s not a candidate: no version field', self._fpr)
  629. return False
  630. if self._data['version'] in Candidate.STALE_CONSENSUS_VERSIONS:
  631. logging.warning('%s not a candidate: version delivers stale consensuses',
  632. self._fpr)
  633. return False
  634. return True
  635. @staticmethod
  636. def _extract_generic_history(history, which='unknown'):
  637. # given a tree like this:
  638. # {
  639. # "1_month": {
  640. # "count": 187,
  641. # "factor": 0.001001001001001001,
  642. # "first": "2015-02-27 06:00:00",
  643. # "interval": 14400,
  644. # "last": "2015-03-30 06:00:00",
  645. # "values": [
  646. # 999,
  647. # 999
  648. # ]
  649. # },
  650. # "1_week": {
  651. # "count": 169,
  652. # "factor": 0.001001001001001001,
  653. # "first": "2015-03-23 07:30:00",
  654. # "interval": 3600,
  655. # "last": "2015-03-30 07:30:00",
  656. # "values": [ ...]
  657. # },
  658. # "1_year": {
  659. # "count": 177,
  660. # "factor": 0.001001001001001001,
  661. # "first": "2014-04-11 00:00:00",
  662. # "interval": 172800,
  663. # "last": "2015-03-29 00:00:00",
  664. # "values": [ ...]
  665. # },
  666. # "3_months": {
  667. # "count": 185,
  668. # "factor": 0.001001001001001001,
  669. # "first": "2014-12-28 06:00:00",
  670. # "interval": 43200,
  671. # "last": "2015-03-30 06:00:00",
  672. # "values": [ ...]
  673. # }
  674. # },
  675. # extract exactly one piece of data per time interval,
  676. # using smaller intervals where available.
  677. #
  678. # returns list of (age, length, value) dictionaries.
  679. generic_history = []
  680. periods = history.keys()
  681. periods.sort(key = lambda x: history[x]['interval'])
  682. now = datetime.datetime.utcnow()
  683. newest = now
  684. for p in periods:
  685. h = history[p]
  686. interval = datetime.timedelta(seconds = h['interval'])
  687. this_ts = parse_ts(h['last'])
  688. if (len(h['values']) != h['count']):
  689. logging.warning('Inconsistent value count in %s document for %s'
  690. %(p, which))
  691. for v in reversed(h['values']):
  692. if (this_ts <= newest):
  693. agt1 = now - this_ts
  694. agt2 = interval
  695. agetmp1 = (agt1.microseconds + (agt1.seconds + agt1.days * 24 * 3600)
  696. * 10**6) / 10**6
  697. agetmp2 = (agt2.microseconds + (agt2.seconds + agt2.days * 24 * 3600)
  698. * 10**6) / 10**6
  699. generic_history.append(
  700. { 'age': agetmp1,
  701. 'length': agetmp2,
  702. 'value': v
  703. })
  704. newest = this_ts
  705. this_ts -= interval
  706. if (this_ts + interval != parse_ts(h['first'])):
  707. logging.warning('Inconsistent time information in %s document for %s'
  708. %(p, which))
  709. #print json.dumps(generic_history, sort_keys=True,
  710. # indent=4, separators=(',', ': '))
  711. return generic_history
  712. @staticmethod
  713. def _avg_generic_history(generic_history):
  714. a = []
  715. for i in generic_history:
  716. if i['age'] > (ADDRESS_AND_PORT_STABLE_DAYS * 24 * 3600):
  717. continue
  718. if (i['length'] is not None
  719. and i['age'] is not None
  720. and i['value'] is not None):
  721. w = i['length'] * math.pow(AGE_ALPHA, i['age']/(3600*24))
  722. a.append( (i['value'] * w, w) )
  723. sv = math.fsum(map(lambda x: x[0], a))
  724. sw = math.fsum(map(lambda x: x[1], a))
  725. if sw == 0.0:
  726. svw = 0.0
  727. else:
  728. svw = sv/sw
  729. return svw
  730. def _add_generic_history(self, history):
  731. periods = r['read_history'].keys()
  732. periods.sort(key = lambda x: r['read_history'][x]['interval'] )
  733. print periods
  734. def add_running_history(self, history):
  735. pass
  736. def add_uptime(self, uptime):
  737. logging.debug('Adding uptime %s.'%(self._fpr,))
  738. # flags we care about: Running, V2Dir, Guard
  739. if not 'flags' in uptime:
  740. logging.debug('No flags in document for %s.'%(self._fpr,))
  741. return
  742. for f in ['Running', 'Guard', 'V2Dir']:
  743. if not f in uptime['flags']:
  744. logging.debug('No %s in flags for %s.'%(f, self._fpr,))
  745. return
  746. running = self._extract_generic_history(uptime['flags']['Running'],
  747. '%s-Running'%(self._fpr))
  748. guard = self._extract_generic_history(uptime['flags']['Guard'],
  749. '%s-Guard'%(self._fpr))
  750. v2dir = self._extract_generic_history(uptime['flags']['V2Dir'],
  751. '%s-V2Dir'%(self._fpr))
  752. if 'BadExit' in uptime['flags']:
  753. badexit = self._extract_generic_history(uptime['flags']['BadExit'],
  754. '%s-BadExit'%(self._fpr))
  755. self._running = self._avg_generic_history(running) / ONIONOO_SCALE_ONE
  756. self._guard = self._avg_generic_history(guard) / ONIONOO_SCALE_ONE
  757. self._v2dir = self._avg_generic_history(v2dir) / ONIONOO_SCALE_ONE
  758. self._badexit = None
  759. if 'BadExit' in uptime['flags']:
  760. self._badexit = self._avg_generic_history(badexit) / ONIONOO_SCALE_ONE
  761. def is_candidate(self):
  762. try:
  763. if (MUST_BE_RUNNING_NOW and not self.is_running()):
  764. log_excluded('%s not a candidate: not running now, unable to check ' +
  765. 'DirPort consensus download', self._fpr)
  766. return False
  767. if (self._data['last_changed_address_or_port'] >
  768. self.CUTOFF_ADDRESS_AND_PORT_STABLE):
  769. log_excluded('%s not a candidate: changed address/port recently (%s)',
  770. self._fpr, self._data['last_changed_address_or_port'])
  771. return False
  772. if self._running < CUTOFF_RUNNING:
  773. log_excluded('%s not a candidate: running avg too low (%lf)',
  774. self._fpr, self._running)
  775. return False
  776. if self._v2dir < CUTOFF_V2DIR:
  777. log_excluded('%s not a candidate: v2dir avg too low (%lf)',
  778. self._fpr, self._v2dir)
  779. return False
  780. if self._badexit is not None and self._badexit > PERMITTED_BADEXIT:
  781. log_excluded('%s not a candidate: badexit avg too high (%lf)',
  782. self._fpr, self._badexit)
  783. return False
  784. # this function logs a message depending on which check fails
  785. if not self.is_valid_version():
  786. return False
  787. if self._guard < CUTOFF_GUARD:
  788. log_excluded('%s not a candidate: guard avg too low (%lf)',
  789. self._fpr, self._guard)
  790. return False
  791. if (not self._data.has_key('consensus_weight')
  792. or self._data['consensus_weight'] < 1):
  793. log_excluded('%s not a candidate: consensus weight invalid', self._fpr)
  794. return False
  795. except BaseException as e:
  796. logging.warning("Exception %s when checking if fallback is a candidate",
  797. str(e))
  798. return False
  799. return True
  800. def is_in_whitelist(self, relaylist):
  801. """ A fallback matches if each key in the whitelist line matches:
  802. ipv4
  803. dirport
  804. orport
  805. id
  806. ipv6 address and port (if present)
  807. If the fallback has an ipv6 key, the whitelist line must also have
  808. it, and vice versa, otherwise they don't match. """
  809. ipv6 = None
  810. if self.has_ipv6():
  811. ipv6 = '%s:%d'%(self.ipv6addr, self.ipv6orport)
  812. for entry in relaylist:
  813. if entry['id'] != self._fpr:
  814. # can't log here unless we match an IP and port, because every relay's
  815. # fingerprint is compared to every entry's fingerprint
  816. if entry['ipv4'] == self.dirip and int(entry['orport']) == self.orport:
  817. logging.warning('%s excluded: has OR %s:%d changed fingerprint to ' +
  818. '%s?', entry['id'], self.dirip, self.orport,
  819. self._fpr)
  820. if self.has_ipv6() and entry.has_key('ipv6') and entry['ipv6'] == ipv6:
  821. logging.warning('%s excluded: has OR %s changed fingerprint to ' +
  822. '%s?', entry['id'], ipv6, self._fpr)
  823. continue
  824. if entry['ipv4'] != self.dirip:
  825. logging.warning('%s excluded: has it changed IPv4 from %s to %s?',
  826. self._fpr, entry['ipv4'], self.dirip)
  827. continue
  828. if int(entry['dirport']) != self.dirport:
  829. logging.warning('%s excluded: has it changed DirPort from %s:%d to ' +
  830. '%s:%d?', self._fpr, self.dirip, int(entry['dirport']),
  831. self.dirip, self.dirport)
  832. continue
  833. if int(entry['orport']) != self.orport:
  834. logging.warning('%s excluded: has it changed ORPort from %s:%d to ' +
  835. '%s:%d?', self._fpr, self.dirip, int(entry['orport']),
  836. self.dirip, self.orport)
  837. continue
  838. if entry.has_key('ipv6') and self.has_ipv6():
  839. # if both entry and fallback have an ipv6 address, compare them
  840. if entry['ipv6'] != ipv6:
  841. logging.warning('%s excluded: has it changed IPv6 ORPort from %s ' +
  842. 'to %s?', self._fpr, entry['ipv6'], ipv6)
  843. continue
  844. # if the fallback has an IPv6 address but the whitelist entry
  845. # doesn't, or vice versa, the whitelist entry doesn't match
  846. elif entry.has_key('ipv6') and not self.has_ipv6():
  847. logging.warning('%s excluded: has it lost its former IPv6 address %s?',
  848. self._fpr, entry['ipv6'])
  849. continue
  850. elif not entry.has_key('ipv6') and self.has_ipv6():
  851. logging.warning('%s excluded: has it gained an IPv6 address %s?',
  852. self._fpr, ipv6)
  853. continue
  854. return True
  855. return False
  856. def is_in_blacklist(self, relaylist):
  857. """ A fallback matches a blacklist line if a sufficiently specific group
  858. of attributes matches:
  859. ipv4 & dirport
  860. ipv4 & orport
  861. id
  862. ipv6 & dirport
  863. ipv6 & ipv6 orport
  864. If the fallback and the blacklist line both have an ipv6 key,
  865. their values will be compared, otherwise, they will be ignored.
  866. If there is no dirport and no orport, the entry matches all relays on
  867. that ip. """
  868. for entry in relaylist:
  869. for key in entry:
  870. value = entry[key]
  871. if key == 'id' and value == self._fpr:
  872. log_excluded('%s is in the blacklist: fingerprint matches',
  873. self._fpr)
  874. return True
  875. if key == 'ipv4' and value == self.dirip:
  876. # if the dirport is present, check it too
  877. if entry.has_key('dirport'):
  878. if int(entry['dirport']) == self.dirport:
  879. log_excluded('%s is in the blacklist: IPv4 (%s) and ' +
  880. 'DirPort (%d) match', self._fpr, self.dirip,
  881. self.dirport)
  882. return True
  883. # if the orport is present, check it too
  884. elif entry.has_key('orport'):
  885. if int(entry['orport']) == self.orport:
  886. log_excluded('%s is in the blacklist: IPv4 (%s) and ' +
  887. 'ORPort (%d) match', self._fpr, self.dirip,
  888. self.orport)
  889. return True
  890. else:
  891. log_excluded('%s is in the blacklist: IPv4 (%s) matches, and ' +
  892. 'entry has no DirPort or ORPort', self._fpr,
  893. self.dirip)
  894. return True
  895. ipv6 = None
  896. if self.has_ipv6():
  897. ipv6 = '%s:%d'%(self.ipv6addr, self.ipv6orport)
  898. if (key == 'ipv6' and self.has_ipv6()):
  899. # if both entry and fallback have an ipv6 address, compare them,
  900. # otherwise, disregard ipv6 addresses
  901. if value == ipv6:
  902. # if the dirport is present, check it too
  903. if entry.has_key('dirport'):
  904. if int(entry['dirport']) == self.dirport:
  905. log_excluded('%s is in the blacklist: IPv6 (%s) and ' +
  906. 'DirPort (%d) match', self._fpr, ipv6,
  907. self.dirport)
  908. return True
  909. # we've already checked the ORPort, it's part of entry['ipv6']
  910. else:
  911. log_excluded('%s is in the blacklist: IPv6 (%s) matches, and' +
  912. 'entry has no DirPort', self._fpr, ipv6)
  913. return True
  914. elif (key == 'ipv6' or self.has_ipv6()):
  915. # only log if the fingerprint matches but the IPv6 doesn't
  916. if entry.has_key('id') and entry['id'] == self._fpr:
  917. log_excluded('%s skipping IPv6 blacklist comparison: relay ' +
  918. 'has%s IPv6%s, but entry has%s IPv6%s', self._fpr,
  919. '' if self.has_ipv6() else ' no',
  920. (' (' + ipv6 + ')') if self.has_ipv6() else '',
  921. '' if key == 'ipv6' else ' no',
  922. (' (' + value + ')') if key == 'ipv6' else '')
  923. logging.warning('Has %s %s IPv6 address %s?', self._fpr,
  924. 'gained an' if self.has_ipv6() else 'lost its former',
  925. ipv6 if self.has_ipv6() else value)
  926. return False
  927. def cw_to_bw_factor(self):
  928. # any relays with a missing or zero consensus weight are not candidates
  929. # any relays with a missing advertised bandwidth have it set to zero
  930. return self._data['advertised_bandwidth'] / self._data['consensus_weight']
  931. # since advertised_bandwidth is reported by the relay, it can be gamed
  932. # to avoid this, use the median consensus weight to bandwidth factor to
  933. # estimate this relay's measured bandwidth, and make that the upper limit
  934. def measured_bandwidth(self, median_cw_to_bw_factor):
  935. cw_to_bw= median_cw_to_bw_factor
  936. # Reduce exit bandwidth to make sure we're not overloading them
  937. if self.is_exit():
  938. cw_to_bw *= EXIT_BANDWIDTH_FRACTION
  939. measured_bandwidth = self._data['consensus_weight'] * cw_to_bw
  940. if self._data['advertised_bandwidth'] != 0:
  941. # limit advertised bandwidth (if available) to measured bandwidth
  942. return min(measured_bandwidth, self._data['advertised_bandwidth'])
  943. else:
  944. return measured_bandwidth
  945. def set_measured_bandwidth(self, median_cw_to_bw_factor):
  946. self._data['measured_bandwidth'] = self.measured_bandwidth(
  947. median_cw_to_bw_factor)
  948. def is_exit(self):
  949. return 'Exit' in self._data['flags']
  950. def is_guard(self):
  951. return 'Guard' in self._data['flags']
  952. def is_running(self):
  953. return 'Running' in self._data['flags']
  954. # does this fallback have an IPv6 address and orport?
  955. def has_ipv6(self):
  956. return self.ipv6addr is not None and self.ipv6orport is not None
  957. # strip leading and trailing brackets from an IPv6 address
  958. # safe to use on non-bracketed IPv6 and on IPv4 addresses
  959. # also convert to unicode, and make None appear as ''
  960. @staticmethod
  961. def strip_ipv6_brackets(ip):
  962. if ip is None:
  963. return unicode('')
  964. if len(ip) < 2:
  965. return unicode(ip)
  966. if ip[0] == '[' and ip[-1] == ']':
  967. return unicode(ip[1:-1])
  968. return unicode(ip)
  969. # are ip_a and ip_b in the same netblock?
  970. # mask_bits is the size of the netblock
  971. # takes both IPv4 and IPv6 addresses
  972. # the versions of ip_a and ip_b must be the same
  973. # the mask must be valid for the IP version
  974. @staticmethod
  975. def netblocks_equal(ip_a, ip_b, mask_bits):
  976. if ip_a is None or ip_b is None:
  977. return False
  978. ip_a = Candidate.strip_ipv6_brackets(ip_a)
  979. ip_b = Candidate.strip_ipv6_brackets(ip_b)
  980. a = ipaddress.ip_address(ip_a)
  981. b = ipaddress.ip_address(ip_b)
  982. if a.version != b.version:
  983. raise Exception('Mismatching IP versions in %s and %s'%(ip_a, ip_b))
  984. if mask_bits > a.max_prefixlen:
  985. logging.error('Bad IP mask %d for %s and %s'%(mask_bits, ip_a, ip_b))
  986. mask_bits = a.max_prefixlen
  987. if mask_bits < 0:
  988. logging.error('Bad IP mask %d for %s and %s'%(mask_bits, ip_a, ip_b))
  989. mask_bits = 0
  990. a_net = ipaddress.ip_network('%s/%d'%(ip_a, mask_bits), strict=False)
  991. return b in a_net
  992. # is this fallback's IPv4 address (dirip) in the same netblock as other's
  993. # IPv4 address?
  994. # mask_bits is the size of the netblock
  995. def ipv4_netblocks_equal(self, other, mask_bits):
  996. return Candidate.netblocks_equal(self.dirip, other.dirip, mask_bits)
  997. # is this fallback's IPv6 address (ipv6addr) in the same netblock as
  998. # other's IPv6 address?
  999. # Returns False if either fallback has no IPv6 address
  1000. # mask_bits is the size of the netblock
  1001. def ipv6_netblocks_equal(self, other, mask_bits):
  1002. if not self.has_ipv6() or not other.has_ipv6():
  1003. return False
  1004. return Candidate.netblocks_equal(self.ipv6addr, other.ipv6addr, mask_bits)
  1005. # is this fallback's IPv4 DirPort the same as other's IPv4 DirPort?
  1006. def dirport_equal(self, other):
  1007. return self.dirport == other.dirport
  1008. # is this fallback's IPv4 ORPort the same as other's IPv4 ORPort?
  1009. def ipv4_orport_equal(self, other):
  1010. return self.orport == other.orport
  1011. # is this fallback's IPv6 ORPort the same as other's IPv6 ORPort?
  1012. # Returns False if either fallback has no IPv6 address
  1013. def ipv6_orport_equal(self, other):
  1014. if not self.has_ipv6() or not other.has_ipv6():
  1015. return False
  1016. return self.ipv6orport == other.ipv6orport
  1017. # does this fallback have the same DirPort, IPv4 ORPort, or
  1018. # IPv6 ORPort as other?
  1019. # Ignores IPv6 ORPort if either fallback has no IPv6 address
  1020. def port_equal(self, other):
  1021. return (self.dirport_equal(other) or self.ipv4_orport_equal(other)
  1022. or self.ipv6_orport_equal(other))
  1023. # return a list containing IPv4 ORPort, DirPort, and IPv6 ORPort (if present)
  1024. def port_list(self):
  1025. ports = [self.dirport, self.orport]
  1026. if self.has_ipv6() and not self.ipv6orport in ports:
  1027. ports.append(self.ipv6orport)
  1028. return ports
  1029. # does this fallback share a port with other, regardless of whether the
  1030. # port types match?
  1031. # For example, if self's IPv4 ORPort is 80 and other's DirPort is 80,
  1032. # return True
  1033. def port_shared(self, other):
  1034. for p in self.port_list():
  1035. if p in other.port_list():
  1036. return True
  1037. return False
  1038. # log how long it takes to download a consensus from dirip:dirport
  1039. # returns True if the download failed, False if it succeeded within max_time
  1040. @staticmethod
  1041. def fallback_consensus_download_speed(dirip, dirport, nickname, fingerprint,
  1042. max_time):
  1043. download_failed = False
  1044. # some directory mirrors respond to requests in ways that hang python
  1045. # sockets, which is why we log this line here
  1046. logging.info('Initiating %sconsensus download from %s (%s:%d) %s.',
  1047. 'microdesc ' if DOWNLOAD_MICRODESC_CONSENSUS else '',
  1048. nickname, dirip, dirport, fingerprint)
  1049. # there appears to be about 1 second of overhead when comparing stem's
  1050. # internal trace time and the elapsed time calculated here
  1051. TIMEOUT_SLOP = 1.0
  1052. start = datetime.datetime.utcnow()
  1053. try:
  1054. consensus = get_consensus(
  1055. endpoints = [(dirip, dirport)],
  1056. timeout = (max_time + TIMEOUT_SLOP),
  1057. validate = True,
  1058. retries = 0,
  1059. fall_back_to_authority = False,
  1060. document_handler = DocumentHandler.BARE_DOCUMENT,
  1061. microdescriptor = DOWNLOAD_MICRODESC_CONSENSUS
  1062. ).run()[0]
  1063. end = datetime.datetime.utcnow()
  1064. time_since_expiry = (end - consensus.valid_until).total_seconds()
  1065. except Exception, stem_error:
  1066. end = datetime.datetime.utcnow()
  1067. log_excluded('Unable to retrieve a consensus from %s: %s', nickname,
  1068. stem_error)
  1069. status = 'error: "%s"' % (stem_error)
  1070. level = logging.WARNING
  1071. download_failed = True
  1072. elapsed = (end - start).total_seconds()
  1073. if download_failed:
  1074. # keep the error failure status, and avoid using the variables
  1075. pass
  1076. elif elapsed > max_time:
  1077. status = 'too slow'
  1078. level = logging.WARNING
  1079. download_failed = True
  1080. elif (time_since_expiry > 0):
  1081. status = 'outdated consensus, expired %ds ago'%(int(time_since_expiry))
  1082. if time_since_expiry <= CONSENSUS_EXPIRY_TOLERANCE:
  1083. status += ', tolerating up to %ds'%(CONSENSUS_EXPIRY_TOLERANCE)
  1084. level = logging.INFO
  1085. else:
  1086. status += ', invalid'
  1087. level = logging.WARNING
  1088. download_failed = True
  1089. else:
  1090. status = 'ok'
  1091. level = logging.DEBUG
  1092. logging.log(level, 'Consensus download: %0.1fs %s from %s (%s:%d) %s, ' +
  1093. 'max download time %0.1fs.', elapsed, status, nickname,
  1094. dirip, dirport, fingerprint, max_time)
  1095. return download_failed
  1096. # does this fallback download the consensus fast enough?
  1097. def check_fallback_download_consensus(self):
  1098. # include the relay if we're not doing a check, or we can't check (IPv6)
  1099. ipv4_failed = False
  1100. ipv6_failed = False
  1101. if PERFORM_IPV4_DIRPORT_CHECKS:
  1102. ipv4_failed = Candidate.fallback_consensus_download_speed(self.dirip,
  1103. self.dirport,
  1104. self._data['nickname'],
  1105. self._fpr,
  1106. CONSENSUS_DOWNLOAD_SPEED_MAX)
  1107. if self.has_ipv6() and PERFORM_IPV6_DIRPORT_CHECKS:
  1108. # Clients assume the IPv6 DirPort is the same as the IPv4 DirPort
  1109. ipv6_failed = Candidate.fallback_consensus_download_speed(self.ipv6addr,
  1110. self.dirport,
  1111. self._data['nickname'],
  1112. self._fpr,
  1113. CONSENSUS_DOWNLOAD_SPEED_MAX)
  1114. return ((not ipv4_failed) and (not ipv6_failed))
  1115. # if this fallback has not passed a download check, try it again,
  1116. # and record the result, available in get_fallback_download_consensus
  1117. def try_fallback_download_consensus(self):
  1118. if not self.get_fallback_download_consensus():
  1119. self._data['download_check'] = self.check_fallback_download_consensus()
  1120. # did this fallback pass the download check?
  1121. def get_fallback_download_consensus(self):
  1122. # if we're not performing checks, return True
  1123. if not PERFORM_IPV4_DIRPORT_CHECKS and not PERFORM_IPV6_DIRPORT_CHECKS:
  1124. return True
  1125. # if we are performing checks, but haven't done one, return False
  1126. if not self._data.has_key('download_check'):
  1127. return False
  1128. return self._data['download_check']
  1129. # output an optional header comment and info for this fallback
  1130. # try_fallback_download_consensus before calling this
  1131. def fallbackdir_line(self, fallbacks, prefilter_fallbacks):
  1132. s = ''
  1133. if OUTPUT_COMMENTS:
  1134. s += self.fallbackdir_comment(fallbacks, prefilter_fallbacks)
  1135. # if the download speed is ok, output a C string
  1136. # if it's not, but we OUTPUT_COMMENTS, output a commented-out C string
  1137. if self.get_fallback_download_consensus() or OUTPUT_COMMENTS:
  1138. s += self.fallbackdir_info(self.get_fallback_download_consensus())
  1139. return s
  1140. # output a header comment for this fallback
  1141. def fallbackdir_comment(self, fallbacks, prefilter_fallbacks):
  1142. # /*
  1143. # nickname
  1144. # flags
  1145. # adjusted bandwidth, consensus weight
  1146. # [contact]
  1147. # [identical contact counts]
  1148. # */
  1149. # Multiline C comment
  1150. s = '/*'
  1151. s += '\n'
  1152. s += cleanse_c_multiline_comment(self._data['nickname'])
  1153. s += '\n'
  1154. s += 'Flags: '
  1155. s += cleanse_c_multiline_comment(' '.join(sorted(self._data['flags'])))
  1156. s += '\n'
  1157. # this is an adjusted bandwidth, see calculate_measured_bandwidth()
  1158. bandwidth = self._data['measured_bandwidth']
  1159. weight = self._data['consensus_weight']
  1160. s += 'Bandwidth: %.1f MByte/s, Consensus Weight: %d'%(
  1161. bandwidth/(1024.0*1024.0),
  1162. weight)
  1163. s += '\n'
  1164. if self._data['contact'] is not None:
  1165. s += cleanse_c_multiline_comment(self._data['contact'])
  1166. if CONTACT_COUNT or CONTACT_BLACKLIST_COUNT:
  1167. fallback_count = len([f for f in fallbacks
  1168. if f._data['contact'] == self._data['contact']])
  1169. if fallback_count > 1:
  1170. s += '\n'
  1171. s += '%d identical contacts listed' % (fallback_count)
  1172. if CONTACT_BLACKLIST_COUNT:
  1173. prefilter_count = len([f for f in prefilter_fallbacks
  1174. if f._data['contact'] == self._data['contact']])
  1175. filter_count = prefilter_count - fallback_count
  1176. if filter_count > 0:
  1177. if fallback_count > 1:
  1178. s += ' '
  1179. else:
  1180. s += '\n'
  1181. s += '%d blacklisted' % (filter_count)
  1182. s += '\n'
  1183. s += '*/'
  1184. s += '\n'
  1185. return s
  1186. # output the fallback info C string for this fallback
  1187. # this is the text that would go after FallbackDir in a torrc
  1188. # if this relay failed the download test and we OUTPUT_COMMENTS,
  1189. # comment-out the returned string
  1190. def fallbackdir_info(self, dl_speed_ok):
  1191. # "address:dirport orport=port id=fingerprint"
  1192. # "[ipv6=addr:orport]"
  1193. # /* nickname=name */
  1194. # /* extrainfo={0,1} */
  1195. # ,
  1196. #
  1197. # Do we want a C string, or a commented-out string?
  1198. c_string = dl_speed_ok
  1199. comment_string = not dl_speed_ok and OUTPUT_COMMENTS
  1200. # If we don't want either kind of string, bail
  1201. if not c_string and not comment_string:
  1202. return ''
  1203. s = ''
  1204. # Comment out the fallback directory entry if it's too slow
  1205. # See the debug output for which address and port is failing
  1206. if comment_string:
  1207. s += '/* Consensus download failed or was too slow:\n'
  1208. # Multi-Line C string with trailing comma (part of a string list)
  1209. # This makes it easier to diff the file, and remove IPv6 lines using grep
  1210. # Integers don't need escaping
  1211. s += '"%s orport=%d id=%s"'%(
  1212. cleanse_c_string(self._data['dir_address']),
  1213. self.orport,
  1214. cleanse_c_string(self._fpr))
  1215. s += '\n'
  1216. if self.has_ipv6():
  1217. s += '" ipv6=%s:%d"'%(cleanse_c_string(self.ipv6addr), self.ipv6orport)
  1218. s += '\n'
  1219. if not comment_string:
  1220. s += '/* '
  1221. s += 'nickname=%s'%(cleanse_c_string(self._data['nickname']))
  1222. if not comment_string:
  1223. s += ' */'
  1224. s += '\n'
  1225. # if we know that the fallback is an extrainfo cache, flag it
  1226. # and if we don't know, assume it is not
  1227. if not comment_string:
  1228. s += '/* '
  1229. s += 'extrainfo=%d'%(1 if self._extra_info_cache else 0)
  1230. if not comment_string:
  1231. s += ' */'
  1232. s += '\n'
  1233. s += ','
  1234. if comment_string:
  1235. s += '\n'
  1236. s += '*/'
  1237. return s
  1238. ## Fallback Candidate List Class
  1239. class CandidateList(dict):
  1240. def __init__(self):
  1241. pass
  1242. def _add_relay(self, details):
  1243. if not 'dir_address' in details: return
  1244. c = Candidate(details)
  1245. self[ c.get_fingerprint() ] = c
  1246. def _add_uptime(self, uptime):
  1247. try:
  1248. fpr = uptime['fingerprint']
  1249. except KeyError:
  1250. raise Exception("Document has no fingerprint field.")
  1251. try:
  1252. c = self[fpr]
  1253. except KeyError:
  1254. logging.debug('Got unknown relay %s in uptime document.'%(fpr,))
  1255. return
  1256. c.add_uptime(uptime)
  1257. def _add_details(self):
  1258. logging.debug('Loading details document.')
  1259. d = fetch('details',
  1260. fields=('fingerprint,nickname,contact,last_changed_address_or_port,' +
  1261. 'consensus_weight,advertised_bandwidth,or_addresses,' +
  1262. 'dir_address,recommended_version,flags,effective_family,' +
  1263. 'platform'))
  1264. logging.debug('Loading details document done.')
  1265. if not 'relays' in d: raise Exception("No relays found in document.")
  1266. for r in d['relays']: self._add_relay(r)
  1267. def _add_uptimes(self):
  1268. logging.debug('Loading uptime document.')
  1269. d = fetch('uptime')
  1270. logging.debug('Loading uptime document done.')
  1271. if not 'relays' in d: raise Exception("No relays found in document.")
  1272. for r in d['relays']: self._add_uptime(r)
  1273. def add_relays(self):
  1274. self._add_details()
  1275. self._add_uptimes()
  1276. def count_guards(self):
  1277. guard_count = 0
  1278. for fpr in self.keys():
  1279. if self[fpr].is_guard():
  1280. guard_count += 1
  1281. return guard_count
  1282. # Find fallbacks that fit the uptime, stability, and flags criteria,
  1283. # and make an array of them in self.fallbacks
  1284. def compute_fallbacks(self):
  1285. self.fallbacks = map(lambda x: self[x],
  1286. filter(lambda x: self[x].is_candidate(),
  1287. self.keys()))
  1288. # sort fallbacks by their consensus weight to advertised bandwidth factor,
  1289. # lowest to highest
  1290. # used to find the median cw_to_bw_factor()
  1291. def sort_fallbacks_by_cw_to_bw_factor(self):
  1292. self.fallbacks.sort(key=lambda f: f.cw_to_bw_factor())
  1293. # sort fallbacks by their measured bandwidth, highest to lowest
  1294. # calculate_measured_bandwidth before calling this
  1295. # this is useful for reviewing candidates in priority order
  1296. def sort_fallbacks_by_measured_bandwidth(self):
  1297. self.fallbacks.sort(key=lambda f: f._data['measured_bandwidth'],
  1298. reverse=True)
  1299. # sort fallbacks by the data field data_field, lowest to highest
  1300. def sort_fallbacks_by(self, data_field):
  1301. self.fallbacks.sort(key=lambda f: f._data[data_field])
  1302. @staticmethod
  1303. def load_relaylist(file_obj):
  1304. """ Read each line in the file, and parse it like a FallbackDir line:
  1305. an IPv4 address and optional port:
  1306. <IPv4 address>:<port>
  1307. which are parsed into dictionary entries:
  1308. ipv4=<IPv4 address>
  1309. dirport=<port>
  1310. followed by a series of key=value entries:
  1311. orport=<port>
  1312. id=<fingerprint>
  1313. ipv6=<IPv6 address>:<IPv6 orport>
  1314. each line's key/value pairs are placed in a dictonary,
  1315. (of string -> string key/value pairs),
  1316. and these dictionaries are placed in an array.
  1317. comments start with # and are ignored """
  1318. file_data = file_obj['data']
  1319. file_name = file_obj['name']
  1320. relaylist = []
  1321. if file_data is None:
  1322. return relaylist
  1323. for line in file_data.split('\n'):
  1324. relay_entry = {}
  1325. # ignore comments
  1326. line_comment_split = line.split('#')
  1327. line = line_comment_split[0]
  1328. # cleanup whitespace
  1329. line = cleanse_whitespace(line)
  1330. line = line.strip()
  1331. if len(line) == 0:
  1332. continue
  1333. for item in line.split(' '):
  1334. item = item.strip()
  1335. if len(item) == 0:
  1336. continue
  1337. key_value_split = item.split('=')
  1338. kvl = len(key_value_split)
  1339. if kvl < 1 or kvl > 2:
  1340. print '#error Bad %s item: %s, format is key=value.'%(
  1341. file_name, item)
  1342. if kvl == 1:
  1343. # assume that entries without a key are the ipv4 address,
  1344. # perhaps with a dirport
  1345. ipv4_maybe_dirport = key_value_split[0]
  1346. ipv4_maybe_dirport_split = ipv4_maybe_dirport.split(':')
  1347. dirl = len(ipv4_maybe_dirport_split)
  1348. if dirl < 1 or dirl > 2:
  1349. print '#error Bad %s IPv4 item: %s, format is ipv4:port.'%(
  1350. file_name, item)
  1351. if dirl >= 1:
  1352. relay_entry['ipv4'] = ipv4_maybe_dirport_split[0]
  1353. if dirl == 2:
  1354. relay_entry['dirport'] = ipv4_maybe_dirport_split[1]
  1355. elif kvl == 2:
  1356. relay_entry[key_value_split[0]] = key_value_split[1]
  1357. relaylist.append(relay_entry)
  1358. return relaylist
  1359. # apply the fallback whitelist and blacklist
  1360. def apply_filter_lists(self, whitelist_obj, blacklist_obj):
  1361. excluded_count = 0
  1362. logging.debug('Applying whitelist and blacklist.')
  1363. # parse the whitelist and blacklist
  1364. whitelist = self.load_relaylist(whitelist_obj)
  1365. blacklist = self.load_relaylist(blacklist_obj)
  1366. filtered_fallbacks = []
  1367. for f in self.fallbacks:
  1368. in_whitelist = f.is_in_whitelist(whitelist)
  1369. in_blacklist = f.is_in_blacklist(blacklist)
  1370. if in_whitelist and in_blacklist:
  1371. if BLACKLIST_EXCLUDES_WHITELIST_ENTRIES:
  1372. # exclude
  1373. excluded_count += 1
  1374. logging.warning('Excluding %s: in both blacklist and whitelist.',
  1375. f._fpr)
  1376. else:
  1377. # include
  1378. filtered_fallbacks.append(f)
  1379. elif in_whitelist:
  1380. # include
  1381. filtered_fallbacks.append(f)
  1382. elif in_blacklist:
  1383. # exclude
  1384. excluded_count += 1
  1385. log_excluded('Excluding %s: in blacklist.', f._fpr)
  1386. else:
  1387. if INCLUDE_UNLISTED_ENTRIES:
  1388. # include
  1389. filtered_fallbacks.append(f)
  1390. else:
  1391. # exclude
  1392. excluded_count += 1
  1393. log_excluded('Excluding %s: in neither blacklist nor whitelist.',
  1394. f._fpr)
  1395. self.fallbacks = filtered_fallbacks
  1396. return excluded_count
  1397. @staticmethod
  1398. def summarise_filters(initial_count, excluded_count):
  1399. return '/* Whitelist & blacklist excluded %d of %d candidates. */'%(
  1400. excluded_count, initial_count)
  1401. # calculate each fallback's measured bandwidth based on the median
  1402. # consensus weight to advertised bandwdith ratio
  1403. def calculate_measured_bandwidth(self):
  1404. self.sort_fallbacks_by_cw_to_bw_factor()
  1405. median_fallback = self.fallback_median(True)
  1406. if median_fallback is not None:
  1407. median_cw_to_bw_factor = median_fallback.cw_to_bw_factor()
  1408. else:
  1409. # this will never be used, because there are no fallbacks
  1410. median_cw_to_bw_factor = None
  1411. for f in self.fallbacks:
  1412. f.set_measured_bandwidth(median_cw_to_bw_factor)
  1413. # remove relays with low measured bandwidth from the fallback list
  1414. # calculate_measured_bandwidth for each relay before calling this
  1415. def remove_low_bandwidth_relays(self):
  1416. if MIN_BANDWIDTH is None:
  1417. return
  1418. above_min_bw_fallbacks = []
  1419. for f in self.fallbacks:
  1420. if f._data['measured_bandwidth'] >= MIN_BANDWIDTH:
  1421. above_min_bw_fallbacks.append(f)
  1422. else:
  1423. # the bandwidth we log here is limited by the relay's consensus weight
  1424. # as well as its adverttised bandwidth. See set_measured_bandwidth
  1425. # for details
  1426. log_excluded('%s not a candidate: bandwidth %.1fMByte/s too low, ' +
  1427. 'must be at least %.1fMByte/s', f._fpr,
  1428. f._data['measured_bandwidth']/(1024.0*1024.0),
  1429. MIN_BANDWIDTH/(1024.0*1024.0))
  1430. self.fallbacks = above_min_bw_fallbacks
  1431. # the minimum fallback in the list
  1432. # call one of the sort_fallbacks_* functions before calling this
  1433. def fallback_min(self):
  1434. if len(self.fallbacks) > 0:
  1435. return self.fallbacks[-1]
  1436. else:
  1437. return None
  1438. # the median fallback in the list
  1439. # call one of the sort_fallbacks_* functions before calling this
  1440. def fallback_median(self, require_advertised_bandwidth):
  1441. # use the low-median when there are an evan number of fallbacks,
  1442. # for consistency with the bandwidth authorities
  1443. if len(self.fallbacks) > 0:
  1444. median_position = (len(self.fallbacks) - 1) / 2
  1445. if not require_advertised_bandwidth:
  1446. return self.fallbacks[median_position]
  1447. # if we need advertised_bandwidth but this relay doesn't have it,
  1448. # move to a fallback with greater consensus weight until we find one
  1449. while not self.fallbacks[median_position]._data['advertised_bandwidth']:
  1450. median_position += 1
  1451. if median_position >= len(self.fallbacks):
  1452. return None
  1453. return self.fallbacks[median_position]
  1454. else:
  1455. return None
  1456. # the maximum fallback in the list
  1457. # call one of the sort_fallbacks_* functions before calling this
  1458. def fallback_max(self):
  1459. if len(self.fallbacks) > 0:
  1460. return self.fallbacks[0]
  1461. else:
  1462. return None
  1463. # return a new bag suitable for storing attributes
  1464. @staticmethod
  1465. def attribute_new():
  1466. return dict()
  1467. # get the count of attribute in attribute_bag
  1468. # if attribute is None or the empty string, return 0
  1469. @staticmethod
  1470. def attribute_count(attribute, attribute_bag):
  1471. if attribute is None or attribute == '':
  1472. return 0
  1473. if attribute not in attribute_bag:
  1474. return 0
  1475. return attribute_bag[attribute]
  1476. # does attribute_bag contain more than max_count instances of attribute?
  1477. # if so, return False
  1478. # if not, return True
  1479. # if attribute is None or the empty string, or max_count is invalid,
  1480. # always return True
  1481. @staticmethod
  1482. def attribute_allow(attribute, attribute_bag, max_count=1):
  1483. if attribute is None or attribute == '' or max_count <= 0:
  1484. return True
  1485. elif CandidateList.attribute_count(attribute, attribute_bag) >= max_count:
  1486. return False
  1487. else:
  1488. return True
  1489. # add attribute to attribute_bag, incrementing the count if it is already
  1490. # present
  1491. # if attribute is None or the empty string, or count is invalid,
  1492. # do nothing
  1493. @staticmethod
  1494. def attribute_add(attribute, attribute_bag, count=1):
  1495. if attribute is None or attribute == '' or count <= 0:
  1496. pass
  1497. attribute_bag.setdefault(attribute, 0)
  1498. attribute_bag[attribute] += count
  1499. # make sure there are only MAX_FALLBACKS_PER_IP fallbacks per IPv4 address,
  1500. # and per IPv6 address
  1501. # there is only one IPv4 address on each fallback: the IPv4 DirPort address
  1502. # (we choose the IPv4 ORPort which is on the same IPv4 as the DirPort)
  1503. # there is at most one IPv6 address on each fallback: the IPv6 ORPort address
  1504. # we try to match the IPv4 ORPort, but will use any IPv6 address if needed
  1505. # (clients only use the IPv6 ORPort)
  1506. # if there is no IPv6 address, only the IPv4 address is checked
  1507. # return the number of candidates we excluded
  1508. def limit_fallbacks_same_ip(self):
  1509. ip_limit_fallbacks = []
  1510. ip_list = CandidateList.attribute_new()
  1511. for f in self.fallbacks:
  1512. if (CandidateList.attribute_allow(f.dirip, ip_list,
  1513. MAX_FALLBACKS_PER_IPV4)
  1514. and CandidateList.attribute_allow(f.ipv6addr, ip_list,
  1515. MAX_FALLBACKS_PER_IPV6)):
  1516. ip_limit_fallbacks.append(f)
  1517. CandidateList.attribute_add(f.dirip, ip_list)
  1518. if f.has_ipv6():
  1519. CandidateList.attribute_add(f.ipv6addr, ip_list)
  1520. elif not CandidateList.attribute_allow(f.dirip, ip_list,
  1521. MAX_FALLBACKS_PER_IPV4):
  1522. log_excluded('Eliminated %s: already have %d fallback(s) on IPv4 %s'
  1523. %(f._fpr, CandidateList.attribute_count(f.dirip, ip_list),
  1524. f.dirip))
  1525. elif (f.has_ipv6() and
  1526. not CandidateList.attribute_allow(f.ipv6addr, ip_list,
  1527. MAX_FALLBACKS_PER_IPV6)):
  1528. log_excluded('Eliminated %s: already have %d fallback(s) on IPv6 %s'
  1529. %(f._fpr, CandidateList.attribute_count(f.ipv6addr,
  1530. ip_list),
  1531. f.ipv6addr))
  1532. original_count = len(self.fallbacks)
  1533. self.fallbacks = ip_limit_fallbacks
  1534. return original_count - len(self.fallbacks)
  1535. # make sure there are only MAX_FALLBACKS_PER_CONTACT fallbacks for each
  1536. # ContactInfo
  1537. # if there is no ContactInfo, allow the fallback
  1538. # this check can be gamed by providing no ContactInfo, or by setting the
  1539. # ContactInfo to match another fallback
  1540. # However, given the likelihood that relays with the same ContactInfo will
  1541. # go down at similar times, its usefulness outweighs the risk
  1542. def limit_fallbacks_same_contact(self):
  1543. contact_limit_fallbacks = []
  1544. contact_list = CandidateList.attribute_new()
  1545. for f in self.fallbacks:
  1546. if CandidateList.attribute_allow(f._data['contact'], contact_list,
  1547. MAX_FALLBACKS_PER_CONTACT):
  1548. contact_limit_fallbacks.append(f)
  1549. CandidateList.attribute_add(f._data['contact'], contact_list)
  1550. else:
  1551. log_excluded(
  1552. 'Eliminated %s: already have %d fallback(s) on ContactInfo %s'
  1553. %(f._fpr, CandidateList.attribute_count(f._data['contact'],
  1554. contact_list),
  1555. f._data['contact']))
  1556. original_count = len(self.fallbacks)
  1557. self.fallbacks = contact_limit_fallbacks
  1558. return original_count - len(self.fallbacks)
  1559. # make sure there are only MAX_FALLBACKS_PER_FAMILY fallbacks per effective
  1560. # family
  1561. # if there is no family, allow the fallback
  1562. # we use effective family, which ensures mutual family declarations
  1563. # but the check can be gamed by not declaring a family at all
  1564. # if any indirect families exist, the result depends on the order in which
  1565. # fallbacks are sorted in the list
  1566. def limit_fallbacks_same_family(self):
  1567. family_limit_fallbacks = []
  1568. fingerprint_list = CandidateList.attribute_new()
  1569. for f in self.fallbacks:
  1570. if CandidateList.attribute_allow(f._fpr, fingerprint_list,
  1571. MAX_FALLBACKS_PER_FAMILY):
  1572. family_limit_fallbacks.append(f)
  1573. CandidateList.attribute_add(f._fpr, fingerprint_list)
  1574. for family_fingerprint in f._data['effective_family']:
  1575. CandidateList.attribute_add(family_fingerprint, fingerprint_list)
  1576. else:
  1577. # we already have a fallback with this fallback in its effective
  1578. # family
  1579. log_excluded(
  1580. 'Eliminated %s: already have %d fallback(s) in effective family'
  1581. %(f._fpr, CandidateList.attribute_count(f._fpr, fingerprint_list)))
  1582. original_count = len(self.fallbacks)
  1583. self.fallbacks = family_limit_fallbacks
  1584. return original_count - len(self.fallbacks)
  1585. # try once to get the descriptors for fingerprint_list using stem
  1586. # returns an empty list on exception
  1587. @staticmethod
  1588. def get_fallback_descriptors_once(fingerprint_list):
  1589. desc_list = get_server_descriptors(fingerprints=fingerprint_list).run(suppress=True)
  1590. return desc_list
  1591. # try up to max_retries times to get the descriptors for fingerprint_list
  1592. # using stem. Stops retrying when all descriptors have been retrieved.
  1593. # returns a list containing the descriptors that were retrieved
  1594. @staticmethod
  1595. def get_fallback_descriptors(fingerprint_list, max_retries=5):
  1596. # we can't use stem's retries=, because we want to support more than 96
  1597. # descriptors
  1598. #
  1599. # add an attempt for every MAX_FINGERPRINTS (or part thereof) in the list
  1600. max_retries += (len(fingerprint_list) + MAX_FINGERPRINTS - 1) / MAX_FINGERPRINTS
  1601. remaining_list = fingerprint_list
  1602. desc_list = []
  1603. for _ in xrange(max_retries):
  1604. if len(remaining_list) == 0:
  1605. break
  1606. new_desc_list = CandidateList.get_fallback_descriptors_once(remaining_list[0:MAX_FINGERPRINTS])
  1607. for d in new_desc_list:
  1608. try:
  1609. remaining_list.remove(d.fingerprint)
  1610. except ValueError:
  1611. # warn and ignore if a directory mirror returned a bad descriptor
  1612. logging.warning("Directory mirror returned unwanted descriptor %s, ignoring",
  1613. d.fingerprint)
  1614. continue
  1615. desc_list.append(d)
  1616. return desc_list
  1617. # find the fallbacks that cache extra-info documents
  1618. # Onionoo doesn't know this, so we have to use stem
  1619. def mark_extra_info_caches(self):
  1620. fingerprint_list = [ f._fpr for f in self.fallbacks ]
  1621. logging.info("Downloading fallback descriptors to find extra-info caches")
  1622. desc_list = CandidateList.get_fallback_descriptors(fingerprint_list)
  1623. for d in desc_list:
  1624. self[d.fingerprint]._extra_info_cache = d.extra_info_cache
  1625. missing_descriptor_list = [ f._fpr for f in self.fallbacks
  1626. if f._extra_info_cache is None ]
  1627. for f in missing_descriptor_list:
  1628. logging.warning("No descriptor for {}. Assuming extrainfo=0.".format(f))
  1629. # try a download check on each fallback candidate in order
  1630. # stop after max_count successful downloads
  1631. # but don't remove any candidates from the array
  1632. def try_download_consensus_checks(self, max_count):
  1633. dl_ok_count = 0
  1634. for f in self.fallbacks:
  1635. f.try_fallback_download_consensus()
  1636. if f.get_fallback_download_consensus():
  1637. # this fallback downloaded a consensus ok
  1638. dl_ok_count += 1
  1639. if dl_ok_count >= max_count:
  1640. # we have enough fallbacks
  1641. return
  1642. # put max_count successful candidates in the fallbacks array:
  1643. # - perform download checks on each fallback candidate
  1644. # - retry failed candidates if CONSENSUS_DOWNLOAD_RETRY is set
  1645. # - eliminate failed candidates
  1646. # - if there are more than max_count candidates, eliminate lowest bandwidth
  1647. # - if there are fewer than max_count candidates, leave only successful
  1648. # Return the number of fallbacks that failed the consensus check
  1649. def perform_download_consensus_checks(self, max_count):
  1650. self.sort_fallbacks_by_measured_bandwidth()
  1651. self.try_download_consensus_checks(max_count)
  1652. if CONSENSUS_DOWNLOAD_RETRY:
  1653. # try unsuccessful candidates again
  1654. # we could end up with more than max_count successful candidates here
  1655. self.try_download_consensus_checks(max_count)
  1656. # now we have at least max_count successful candidates,
  1657. # or we've tried them all
  1658. original_count = len(self.fallbacks)
  1659. self.fallbacks = filter(lambda x: x.get_fallback_download_consensus(),
  1660. self.fallbacks)
  1661. # some of these failed the check, others skipped the check,
  1662. # if we already had enough successful downloads
  1663. failed_count = original_count - len(self.fallbacks)
  1664. self.fallbacks = self.fallbacks[:max_count]
  1665. return failed_count
  1666. # return a string that describes a/b as a percentage
  1667. @staticmethod
  1668. def describe_percentage(a, b):
  1669. if b != 0:
  1670. return '%d/%d = %.0f%%'%(a, b, (a*100.0)/b)
  1671. else:
  1672. # technically, 0/0 is undefined, but 0.0% is a sensible result
  1673. return '%d/%d = %.0f%%'%(a, b, 0.0)
  1674. # return a dictionary of lists of fallbacks by IPv4 netblock
  1675. # the dictionary is keyed by the fingerprint of an arbitrary fallback
  1676. # in each netblock
  1677. # mask_bits is the size of the netblock
  1678. def fallbacks_by_ipv4_netblock(self, mask_bits):
  1679. netblocks = {}
  1680. for f in self.fallbacks:
  1681. found_netblock = False
  1682. for b in netblocks.keys():
  1683. # we found an existing netblock containing this fallback
  1684. if f.ipv4_netblocks_equal(self[b], mask_bits):
  1685. # add it to the list
  1686. netblocks[b].append(f)
  1687. found_netblock = True
  1688. break
  1689. # make a new netblock based on this fallback's fingerprint
  1690. if not found_netblock:
  1691. netblocks[f._fpr] = [f]
  1692. return netblocks
  1693. # return a dictionary of lists of fallbacks by IPv6 netblock
  1694. # where mask_bits is the size of the netblock
  1695. def fallbacks_by_ipv6_netblock(self, mask_bits):
  1696. netblocks = {}
  1697. for f in self.fallbacks:
  1698. # skip fallbacks without IPv6 addresses
  1699. if not f.has_ipv6():
  1700. continue
  1701. found_netblock = False
  1702. for b in netblocks.keys():
  1703. # we found an existing netblock containing this fallback
  1704. if f.ipv6_netblocks_equal(self[b], mask_bits):
  1705. # add it to the list
  1706. netblocks[b].append(f)
  1707. found_netblock = True
  1708. break
  1709. # make a new netblock based on this fallback's fingerprint
  1710. if not found_netblock:
  1711. netblocks[f._fpr] = [f]
  1712. return netblocks
  1713. # log a message about the proportion of fallbacks in each IPv4 netblock,
  1714. # where mask_bits is the size of the netblock
  1715. def describe_fallback_ipv4_netblock_mask(self, mask_bits):
  1716. fallback_count = len(self.fallbacks)
  1717. shared_netblock_fallback_count = 0
  1718. most_frequent_netblock = None
  1719. netblocks = self.fallbacks_by_ipv4_netblock(mask_bits)
  1720. for b in netblocks.keys():
  1721. if len(netblocks[b]) > 1:
  1722. # how many fallbacks are in a netblock with other fallbacks?
  1723. shared_netblock_fallback_count += len(netblocks[b])
  1724. # what's the netblock with the most fallbacks?
  1725. if (most_frequent_netblock is None
  1726. or len(netblocks[b]) > len(netblocks[most_frequent_netblock])):
  1727. most_frequent_netblock = b
  1728. logging.debug('Fallback IPv4 addresses in the same /%d:'%(mask_bits))
  1729. for f in netblocks[b]:
  1730. logging.debug('%s - %s', f.dirip, f._fpr)
  1731. if most_frequent_netblock is not None:
  1732. logging.warning('There are %s fallbacks in the IPv4 /%d containing %s'%(
  1733. CandidateList.describe_percentage(
  1734. len(netblocks[most_frequent_netblock]),
  1735. fallback_count),
  1736. mask_bits,
  1737. self[most_frequent_netblock].dirip))
  1738. if shared_netblock_fallback_count > 0:
  1739. logging.warning(('%s of fallbacks are in an IPv4 /%d with other ' +
  1740. 'fallbacks')%(CandidateList.describe_percentage(
  1741. shared_netblock_fallback_count,
  1742. fallback_count),
  1743. mask_bits))
  1744. # log a message about the proportion of fallbacks in each IPv6 netblock,
  1745. # where mask_bits is the size of the netblock
  1746. def describe_fallback_ipv6_netblock_mask(self, mask_bits):
  1747. fallback_count = len(self.fallbacks_with_ipv6())
  1748. shared_netblock_fallback_count = 0
  1749. most_frequent_netblock = None
  1750. netblocks = self.fallbacks_by_ipv6_netblock(mask_bits)
  1751. for b in netblocks.keys():
  1752. if len(netblocks[b]) > 1:
  1753. # how many fallbacks are in a netblock with other fallbacks?
  1754. shared_netblock_fallback_count += len(netblocks[b])
  1755. # what's the netblock with the most fallbacks?
  1756. if (most_frequent_netblock is None
  1757. or len(netblocks[b]) > len(netblocks[most_frequent_netblock])):
  1758. most_frequent_netblock = b
  1759. logging.debug('Fallback IPv6 addresses in the same /%d:'%(mask_bits))
  1760. for f in netblocks[b]:
  1761. logging.debug('%s - %s', f.ipv6addr, f._fpr)
  1762. if most_frequent_netblock is not None:
  1763. logging.warning('There are %s fallbacks in the IPv6 /%d containing %s'%(
  1764. CandidateList.describe_percentage(
  1765. len(netblocks[most_frequent_netblock]),
  1766. fallback_count),
  1767. mask_bits,
  1768. self[most_frequent_netblock].ipv6addr))
  1769. if shared_netblock_fallback_count > 0:
  1770. logging.warning(('%s of fallbacks are in an IPv6 /%d with other ' +
  1771. 'fallbacks')%(CandidateList.describe_percentage(
  1772. shared_netblock_fallback_count,
  1773. fallback_count),
  1774. mask_bits))
  1775. # log a message about the proportion of fallbacks in each IPv4 /8, /16,
  1776. # and /24
  1777. def describe_fallback_ipv4_netblocks(self):
  1778. # this doesn't actually tell us anything useful
  1779. #self.describe_fallback_ipv4_netblock_mask(8)
  1780. self.describe_fallback_ipv4_netblock_mask(16)
  1781. self.describe_fallback_ipv4_netblock_mask(24)
  1782. # log a message about the proportion of fallbacks in each IPv6 /12 (RIR),
  1783. # /23 (smaller RIR blocks), /32 (LIR), /48 (Customer), and /64 (Host)
  1784. # https://www.iana.org/assignments/ipv6-unicast-address-assignments/
  1785. def describe_fallback_ipv6_netblocks(self):
  1786. # these don't actually tell us anything useful
  1787. #self.describe_fallback_ipv6_netblock_mask(12)
  1788. #self.describe_fallback_ipv6_netblock_mask(23)
  1789. self.describe_fallback_ipv6_netblock_mask(32)
  1790. self.describe_fallback_ipv6_netblock_mask(48)
  1791. self.describe_fallback_ipv6_netblock_mask(64)
  1792. # log a message about the proportion of fallbacks in each IPv4 and IPv6
  1793. # netblock
  1794. def describe_fallback_netblocks(self):
  1795. self.describe_fallback_ipv4_netblocks()
  1796. self.describe_fallback_ipv6_netblocks()
  1797. # return a list of fallbacks which are on the IPv4 ORPort port
  1798. def fallbacks_on_ipv4_orport(self, port):
  1799. return filter(lambda x: x.orport == port, self.fallbacks)
  1800. # return a list of fallbacks which are on the IPv6 ORPort port
  1801. def fallbacks_on_ipv6_orport(self, port):
  1802. return filter(lambda x: x.ipv6orport == port, self.fallbacks_with_ipv6())
  1803. # return a list of fallbacks which are on the DirPort port
  1804. def fallbacks_on_dirport(self, port):
  1805. return filter(lambda x: x.dirport == port, self.fallbacks)
  1806. # log a message about the proportion of fallbacks on IPv4 ORPort port
  1807. # and return that count
  1808. def describe_fallback_ipv4_orport(self, port):
  1809. port_count = len(self.fallbacks_on_ipv4_orport(port))
  1810. fallback_count = len(self.fallbacks)
  1811. logging.warning('%s of fallbacks are on IPv4 ORPort %d'%(
  1812. CandidateList.describe_percentage(port_count,
  1813. fallback_count),
  1814. port))
  1815. return port_count
  1816. # log a message about the proportion of IPv6 fallbacks on IPv6 ORPort port
  1817. # and return that count
  1818. def describe_fallback_ipv6_orport(self, port):
  1819. port_count = len(self.fallbacks_on_ipv6_orport(port))
  1820. fallback_count = len(self.fallbacks_with_ipv6())
  1821. logging.warning('%s of IPv6 fallbacks are on IPv6 ORPort %d'%(
  1822. CandidateList.describe_percentage(port_count,
  1823. fallback_count),
  1824. port))
  1825. return port_count
  1826. # log a message about the proportion of fallbacks on DirPort port
  1827. # and return that count
  1828. def describe_fallback_dirport(self, port):
  1829. port_count = len(self.fallbacks_on_dirport(port))
  1830. fallback_count = len(self.fallbacks)
  1831. logging.warning('%s of fallbacks are on DirPort %d'%(
  1832. CandidateList.describe_percentage(port_count,
  1833. fallback_count),
  1834. port))
  1835. return port_count
  1836. # log a message about the proportion of fallbacks on each dirport,
  1837. # each IPv4 orport, and each IPv6 orport
  1838. def describe_fallback_ports(self):
  1839. fallback_count = len(self.fallbacks)
  1840. ipv4_or_count = fallback_count
  1841. ipv4_or_count -= self.describe_fallback_ipv4_orport(443)
  1842. ipv4_or_count -= self.describe_fallback_ipv4_orport(9001)
  1843. logging.warning('%s of fallbacks are on other IPv4 ORPorts'%(
  1844. CandidateList.describe_percentage(ipv4_or_count,
  1845. fallback_count)))
  1846. ipv6_fallback_count = len(self.fallbacks_with_ipv6())
  1847. ipv6_or_count = ipv6_fallback_count
  1848. ipv6_or_count -= self.describe_fallback_ipv6_orport(443)
  1849. ipv6_or_count -= self.describe_fallback_ipv6_orport(9001)
  1850. logging.warning('%s of IPv6 fallbacks are on other IPv6 ORPorts'%(
  1851. CandidateList.describe_percentage(ipv6_or_count,
  1852. ipv6_fallback_count)))
  1853. dir_count = fallback_count
  1854. dir_count -= self.describe_fallback_dirport(80)
  1855. dir_count -= self.describe_fallback_dirport(9030)
  1856. logging.warning('%s of fallbacks are on other DirPorts'%(
  1857. CandidateList.describe_percentage(dir_count,
  1858. fallback_count)))
  1859. # return a list of fallbacks which cache extra-info documents
  1860. def fallbacks_with_extra_info_cache(self):
  1861. return filter(lambda x: x._extra_info_cache, self.fallbacks)
  1862. # log a message about the proportion of fallbacks that cache extra-info docs
  1863. def describe_fallback_extra_info_caches(self):
  1864. extra_info_falback_count = len(self.fallbacks_with_extra_info_cache())
  1865. fallback_count = len(self.fallbacks)
  1866. logging.warning('%s of fallbacks cache extra-info documents'%(
  1867. CandidateList.describe_percentage(extra_info_falback_count,
  1868. fallback_count)))
  1869. # return a list of fallbacks which have the Exit flag
  1870. def fallbacks_with_exit(self):
  1871. return filter(lambda x: x.is_exit(), self.fallbacks)
  1872. # log a message about the proportion of fallbacks with an Exit flag
  1873. def describe_fallback_exit_flag(self):
  1874. exit_falback_count = len(self.fallbacks_with_exit())
  1875. fallback_count = len(self.fallbacks)
  1876. logging.warning('%s of fallbacks have the Exit flag'%(
  1877. CandidateList.describe_percentage(exit_falback_count,
  1878. fallback_count)))
  1879. # return a list of fallbacks which have an IPv6 address
  1880. def fallbacks_with_ipv6(self):
  1881. return filter(lambda x: x.has_ipv6(), self.fallbacks)
  1882. # log a message about the proportion of fallbacks on IPv6
  1883. def describe_fallback_ip_family(self):
  1884. ipv6_falback_count = len(self.fallbacks_with_ipv6())
  1885. fallback_count = len(self.fallbacks)
  1886. logging.warning('%s of fallbacks are on IPv6'%(
  1887. CandidateList.describe_percentage(ipv6_falback_count,
  1888. fallback_count)))
  1889. def summarise_fallbacks(self, eligible_count, operator_count, failed_count,
  1890. guard_count, target_count):
  1891. s = ''
  1892. s += '/* To comment-out entries in this file, use C comments, and add *'
  1893. s += ' to the start of each line. (stem finds fallback entries using "'
  1894. s += ' at the start of a line.) */'
  1895. s += '\n'
  1896. # Report:
  1897. # whether we checked consensus download times
  1898. # the number of fallback directories (and limits/exclusions, if relevant)
  1899. # min & max fallback bandwidths
  1900. # #error if below minimum count
  1901. if PERFORM_IPV4_DIRPORT_CHECKS or PERFORM_IPV6_DIRPORT_CHECKS:
  1902. s += '/* Checked %s%s%s DirPorts served a consensus within %.1fs. */'%(
  1903. 'IPv4' if PERFORM_IPV4_DIRPORT_CHECKS else '',
  1904. ' and ' if (PERFORM_IPV4_DIRPORT_CHECKS
  1905. and PERFORM_IPV6_DIRPORT_CHECKS) else '',
  1906. 'IPv6' if PERFORM_IPV6_DIRPORT_CHECKS else '',
  1907. CONSENSUS_DOWNLOAD_SPEED_MAX)
  1908. else:
  1909. s += '/* Did not check IPv4 or IPv6 DirPort consensus downloads. */'
  1910. s += '\n'
  1911. # Multiline C comment with #error if things go bad
  1912. s += '/*'
  1913. s += '\n'
  1914. # Integers don't need escaping in C comments
  1915. fallback_count = len(self.fallbacks)
  1916. if FALLBACK_PROPORTION_OF_GUARDS is None:
  1917. fallback_proportion = ''
  1918. else:
  1919. fallback_proportion = ', Target %d (%d * %.2f)'%(target_count,
  1920. guard_count,
  1921. FALLBACK_PROPORTION_OF_GUARDS)
  1922. s += 'Final Count: %d (Eligible %d%s'%(fallback_count, eligible_count,
  1923. fallback_proportion)
  1924. if MAX_FALLBACK_COUNT is not None:
  1925. s += ', Max %d'%(MAX_FALLBACK_COUNT)
  1926. s += ')\n'
  1927. if eligible_count != fallback_count:
  1928. removed_count = eligible_count - fallback_count
  1929. excess_to_target_or_max = (eligible_count - operator_count - failed_count
  1930. - fallback_count)
  1931. # some 'Failed' failed the check, others 'Skipped' the check,
  1932. # if we already had enough successful downloads
  1933. s += ('Excluded: %d (Same Operator %d, Failed/Skipped Download %d, ' +
  1934. 'Excess %d)')%(removed_count, operator_count, failed_count,
  1935. excess_to_target_or_max)
  1936. s += '\n'
  1937. min_fb = self.fallback_min()
  1938. min_bw = min_fb._data['measured_bandwidth']
  1939. max_fb = self.fallback_max()
  1940. max_bw = max_fb._data['measured_bandwidth']
  1941. s += 'Bandwidth Range: %.1f - %.1f MByte/s'%(min_bw/(1024.0*1024.0),
  1942. max_bw/(1024.0*1024.0))
  1943. s += '\n'
  1944. s += '*/'
  1945. if fallback_count < MIN_FALLBACK_COUNT:
  1946. # We must have a minimum number of fallbacks so they are always
  1947. # reachable, and are in diverse locations
  1948. s += '\n'
  1949. s += '#error Fallback Count %d is too low. '%(fallback_count)
  1950. s += 'Must be at least %d for diversity. '%(MIN_FALLBACK_COUNT)
  1951. s += 'Try adding entries to the whitelist, '
  1952. s += 'or setting INCLUDE_UNLISTED_ENTRIES = True.'
  1953. return s
  1954. def process_existing():
  1955. logging.basicConfig(level=logging.INFO)
  1956. logging.getLogger('stem').setLevel(logging.INFO)
  1957. whitelist = {'data': parse_fallback_file(FALLBACK_FILE_NAME),
  1958. 'name': FALLBACK_FILE_NAME}
  1959. blacklist = {'data': read_from_file(BLACKLIST_FILE_NAME, MAX_LIST_FILE_SIZE),
  1960. 'name': BLACKLIST_FILE_NAME}
  1961. list_fallbacks(whitelist, blacklist)
  1962. def process_default():
  1963. logging.basicConfig(level=logging.WARNING)
  1964. logging.getLogger('stem').setLevel(logging.WARNING)
  1965. whitelist = {'data': read_from_file(WHITELIST_FILE_NAME, MAX_LIST_FILE_SIZE),
  1966. 'name': WHITELIST_FILE_NAME}
  1967. blacklist = {'data': read_from_file(BLACKLIST_FILE_NAME, MAX_LIST_FILE_SIZE),
  1968. 'name': BLACKLIST_FILE_NAME}
  1969. list_fallbacks(whitelist, blacklist)
  1970. ## Main Function
  1971. def main():
  1972. if get_command() == 'check_existing':
  1973. process_existing()
  1974. else:
  1975. process_default()
  1976. def get_command():
  1977. if len(sys.argv) == 2:
  1978. return sys.argv[1]
  1979. else:
  1980. return None
  1981. def log_excluded(msg, *args):
  1982. if get_command() == 'check_existing':
  1983. logging.warning(msg, *args)
  1984. else:
  1985. logging.info(msg, *args)
  1986. def list_fallbacks(whitelist, blacklist):
  1987. """ Fetches required onionoo documents and evaluates the
  1988. fallback directory criteria for each of the relays """
  1989. print "/* type=fallback */"
  1990. print ("/* version={} */"
  1991. .format(cleanse_c_multiline_comment(FALLBACK_FORMAT_VERSION)))
  1992. logging.warning('Downloading and parsing Onionoo data. ' +
  1993. 'This may take some time.')
  1994. # find relays that could be fallbacks
  1995. candidates = CandidateList()
  1996. candidates.add_relays()
  1997. # work out how many fallbacks we want
  1998. guard_count = candidates.count_guards()
  1999. if FALLBACK_PROPORTION_OF_GUARDS is None:
  2000. target_count = guard_count
  2001. else:
  2002. target_count = int(guard_count * FALLBACK_PROPORTION_OF_GUARDS)
  2003. # the maximum number of fallbacks is the least of:
  2004. # - the target fallback count (FALLBACK_PROPORTION_OF_GUARDS * guard count)
  2005. # - the maximum fallback count (MAX_FALLBACK_COUNT)
  2006. if MAX_FALLBACK_COUNT is None:
  2007. max_count = target_count
  2008. else:
  2009. max_count = min(target_count, MAX_FALLBACK_COUNT)
  2010. candidates.compute_fallbacks()
  2011. prefilter_fallbacks = copy.copy(candidates.fallbacks)
  2012. # filter with the whitelist and blacklist
  2013. # if a relay has changed IPv4 address or ports recently, it will be excluded
  2014. # as ineligible before we call apply_filter_lists, and so there will be no
  2015. # warning that the details have changed from those in the whitelist.
  2016. # instead, there will be an info-level log during the eligibility check.
  2017. initial_count = len(candidates.fallbacks)
  2018. excluded_count = candidates.apply_filter_lists(whitelist, blacklist)
  2019. print candidates.summarise_filters(initial_count, excluded_count)
  2020. eligible_count = len(candidates.fallbacks)
  2021. # calculate the measured bandwidth of each relay,
  2022. # then remove low-bandwidth relays
  2023. candidates.calculate_measured_bandwidth()
  2024. candidates.remove_low_bandwidth_relays()
  2025. # print the raw fallback list
  2026. #for x in candidates.fallbacks:
  2027. # print x.fallbackdir_line(True)
  2028. # print json.dumps(candidates[x]._data, sort_keys=True, indent=4,
  2029. # separators=(',', ': '), default=json_util.default)
  2030. # impose mandatory conditions here, like one per contact, family, IP
  2031. # in measured bandwidth order
  2032. candidates.sort_fallbacks_by_measured_bandwidth()
  2033. operator_count = 0
  2034. # only impose these limits on the final list - operators can nominate
  2035. # multiple candidate fallbacks, and then we choose the best set
  2036. if not OUTPUT_CANDIDATES:
  2037. operator_count += candidates.limit_fallbacks_same_ip()
  2038. operator_count += candidates.limit_fallbacks_same_contact()
  2039. operator_count += candidates.limit_fallbacks_same_family()
  2040. # check if each candidate can serve a consensus
  2041. # there's a small risk we've eliminated relays from the same operator that
  2042. # can serve a consensus, in favour of one that can't
  2043. # but given it takes up to 15 seconds to check each consensus download,
  2044. # the risk is worth it
  2045. if PERFORM_IPV4_DIRPORT_CHECKS or PERFORM_IPV6_DIRPORT_CHECKS:
  2046. logging.warning('Checking consensus download speeds. ' +
  2047. 'This may take some time.')
  2048. failed_count = candidates.perform_download_consensus_checks(max_count)
  2049. # work out which fallbacks cache extra-infos
  2050. candidates.mark_extra_info_caches()
  2051. # analyse and log interesting diversity metrics
  2052. # like netblock, ports, exit, IPv4-only
  2053. # (we can't easily analyse AS, and it's hard to accurately analyse country)
  2054. candidates.describe_fallback_ip_family()
  2055. # if we can't import the ipaddress module, we can't do netblock analysis
  2056. if HAVE_IPADDRESS:
  2057. candidates.describe_fallback_netblocks()
  2058. candidates.describe_fallback_ports()
  2059. candidates.describe_fallback_extra_info_caches()
  2060. candidates.describe_fallback_exit_flag()
  2061. # output C comments summarising the fallback selection process
  2062. if len(candidates.fallbacks) > 0:
  2063. print candidates.summarise_fallbacks(eligible_count, operator_count,
  2064. failed_count, guard_count,
  2065. target_count)
  2066. else:
  2067. print '/* No Fallbacks met criteria */'
  2068. # output C comments specifying the OnionOO data used to create the list
  2069. for s in fetch_source_list():
  2070. print describe_fetch_source(s)
  2071. # sort the list differently depending on why we've created it:
  2072. # if we're outputting the final fallback list, sort by fingerprint
  2073. # this makes diffs much more stable
  2074. # otherwise, if we're trying to find a bandwidth cutoff, or we want to
  2075. # contact operators in priority order, sort by bandwidth (not yet
  2076. # implemented)
  2077. # otherwise, if we're contacting operators, sort by contact
  2078. candidates.sort_fallbacks_by(OUTPUT_SORT_FIELD)
  2079. for x in candidates.fallbacks:
  2080. print x.fallbackdir_line(candidates.fallbacks, prefilter_fallbacks)
  2081. if __name__ == "__main__":
  2082. main()