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
|
#!/usr/bin/env python
# Simple usage example of TorInfo. This class does some magic so that
# once it's set up, all the attributes it has (or appears to) are
# GETINFO ones, in a heirarchy. So where GETINFO specifies
# "net/listeners/dns" TorInfo will have a "net" attribute that
# contains at least "listeners", etcetera. The leaves are all methods
# which return a Deferred. If the corresponding GETINFO takes an
# argument, so does the leaf.
#
# Go straight to "setup_complete" for the goods -- this is called
# after TorInfo and the underlying TorControlProtocol are set up.
#
# If you want to issue multiple GETINFO calls in one network
# transaction, you'll have to use TorControlProtocol's get_info
# instead.
from __future__ import print_function
import sys
from twisted.internet import reactor, defer
from txtorcon import TorInfo, build_local_tor_connection
def error(x):
print("ERROR", x)
return x
@defer.inlineCallbacks
def recursive_dump(indent, obj, depth=0):
if callable(obj):
try:
print("%s: " % obj, end=' ')
sys.stdout.flush()
if obj.takes_arg:
v = yield obj('arrrrrg')
v = yield obj()
v = v.replace('\n', '\\')
if len(v) > 60:
v = v[:50] + '...' + v[-7:]
except Exception as e:
v = 'ERROR: ' + str(e)
print(v)
else:
indent = indent + ' '
for x in obj:
yield recursive_dump(indent, x, depth + 1)
@defer.inlineCallbacks
def setup_complete(info):
print("Top-Level Things:", dir(info))
if True:
# some examples of getting specific GETINFO callbacks
v = yield info.version()
ip = yield info.ip_to_country('1.2.3.4')
boot_phase = yield info.status.bootstrap_phase()
ns = yield info.ns.name('moria1')
guards = yield info.entry_guards()
print('version:', v)
print('1.2.3.4 is in', ip)
print('bootstrap-phase:', boot_phase)
print('moria1:', ns)
print('entry guards:', guards)
# now we dump everything, one at a time
d = recursive_dump('', info)
d.addCallback(lambda x: reactor.stop())
d.addErrback(error)
def setup_failed(arg):
print("SETUP FAILED", arg)
reactor.stop()
def bootstrap(c):
info = TorInfo(c)
info.post_bootstrap.addCallback(setup_complete).addErrback(setup_failed)
d = build_local_tor_connection(reactor, build_state=False)
# do not use addCallbacks() here, in case bootstrap has an error
d.addCallback(bootstrap).addErrback(setup_failed)
reactor.run()
|