Tor-Routing/Network_Consesus.py at master · Amoghaithal/Tor-Routing · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
'''
Created on 16-Mar-2015

@author: aithal
'''
from stem import CircStatus, UNDEFINED
from stem.control import Controller
from stem.descriptor import Descriptor
from stem import descriptor
from stem.descriptor.remote import DescriptorDownloader
from stem import CircStatus
from stem.descriptor import parse_file, DocumentHandler
from stem.descriptor.networkstatus import NetworkStatusDocumentV3,\
    NetworkStatusDocumentV2

import os
import StringIO
import socket
import urllib
import time
import socks  # SocksiPy module
import stem.process
from stem import Signal
from stem.util import term
from stem.control import Controller
from stem.descriptor import parse_file

SOCKS_PORT = 7000
socks.setdefaultproxy(socks.PROXY_TYPE_SOCKS5, '127.0.0.1', SOCKS_PORT)
__originalSocket = socket.socket


def getaddrinfo(*args):
  return [(socket.AF_INET, socket.SOCK_STREAM, 6, '', (args[0], args[1]))]

socket.getaddrinfo = getaddrinfo


def query(url):
  try:
    return urllib.urlopen(url).read()
  except:
    return "Unable to reach %s" % url
  finally:
      socket.socket=__originalSocket

def print_bootstrap_lines(line):
  if "Bootstrapped " in line:
    print term.format(line)

def get_key(item):
    return item[3]
exit_digests = set()
entry_digests=set()
entry_fing=[]
exit_fing=[]

def get_entry_exit_fing():
    with Controller.from_port(port=9151) as controller:
        controller.authenticate()
        data_dir = controller.get_conf('DataDirectory')

        for desc in controller.get_microdescriptors():
            if desc.exit_policy.is_exiting_allowed():
                exit_digests.add(desc.digest)
            else:
                entry_digests.add(desc.digest)
        global entry_fing,exit_fing
        for desc in parse_file(os.path.join(data_dir, 'cached-microdesc-consensus')):
            if desc.digest in exit_digests :
                exit_fing.append((desc.fingerprint,desc.address,desc.nickname,desc.bandwidth))
            else:
                entry_fing.append((desc.fingerprint,desc.address,desc.nickname,desc.bandwidth))

        exit_fing=sorted(exit_fing,key=get_key,reverse=True)
        entry_fing=sorted(entry_fing,key=get_key,reverse=True)

get_entry_exit_fing()


tor_process = stem.process.launch_tor_with_config(
config = {
    'SocksPort': str(SOCKS_PORT),
},
  init_msg_handler = print_bootstrap_lines,
)
socket.socket = socks.socksocket
print term.format("\nChecking our endpoint:\n")
start_time=time.time()
print term.format(query("https://www.atagar.com/echo.php"))
print "time taken %s secs"%(time.time()-start_time)
tor_process.kill()
socket.socket=__originalSocket

tor_process = stem.process.launch_tor_with_config(
config = {
    'SocksPort': str(SOCKS_PORT),
},
  init_msg_handler = print_bootstrap_lines,
)

with Controller.from_port(port=9151) as controller:
    controller.authenticate()
    controller.signal(Signal.NEWNYM)
    controller.extend_circuit('0', [entry_fing[0][0],entry_fing[1][0],exit_fing[0][0]])
    print(controller.get_info('circuit-status'))

socket.socket = socks.socksocket
print term.format("\nChecking our endpoint:\n")
start_time=time.time()
print term.format(query("https://www.atagar.com/echo.php"))
print "time taken %s secs"%(time.time()-start_time)
socket.socket=__originalSocket
tor_process.kill()