mirror of
https://gitlab.torproject.org/tpo/core/tor.git
synced 2024-11-10 13:13:44 +01:00
99 lines
2.2 KiB
Python
Executable File
99 lines
2.2 KiB
Python
Executable File
#!/usr/bin/python2
|
|
#$Id$
|
|
|
|
import socket
|
|
import struct
|
|
import sys
|
|
|
|
MSG_TYPE_SETCONF = 0x0002
|
|
MSG_TYPE_GETCONF = 0x0003
|
|
MSG_TYPE_SETEVENTS = 0x0005
|
|
MSG_TYPE_AUTH = 0x0007
|
|
|
|
EVENT_TYPE_BANDWIDTH = 0x0004
|
|
EVENT_TYPE_WARN = 0x0005
|
|
|
|
def parseHostAndPort(h):
|
|
host, port = "localhost", 9051
|
|
if ":" in h:
|
|
i = h.index(":")
|
|
host = h[:i]
|
|
try:
|
|
port = int(h[i+1:])
|
|
except ValueError:
|
|
print "Bad hostname %r"%h
|
|
sys.exit(1)
|
|
elif h:
|
|
try:
|
|
port = int(h)
|
|
except ValueError:
|
|
host = h
|
|
|
|
return host, port
|
|
|
|
def receive_message(s):
|
|
body = ""
|
|
header = s.recv(4)
|
|
length,type = struct.unpack("!HH",header)
|
|
print "Got response length %d, type %d"%(length,type)
|
|
if length:
|
|
body = s.recv(length)
|
|
print "Got response length %d, type %d, body %s"%(length,type,body)
|
|
return length,type,body
|
|
|
|
def pack_message(type, body=""):
|
|
length = len(body)
|
|
reqheader = struct.pack("!HH", length, type)
|
|
return "%s%s"%(reqheader,body)
|
|
|
|
def authenticate(s):
|
|
s.sendall(pack_message(MSG_TYPE_AUTH))
|
|
length,type,body = receive_message(s)
|
|
return
|
|
|
|
def get_option(s,name):
|
|
s.sendall(pack_message(MSG_TYPE_GETCONF,name))
|
|
length,type,body = receive_message(s)
|
|
return
|
|
|
|
def set_option(s,msg):
|
|
s.sendall(pack_message(MSG_TYPE_SETCONF,msg))
|
|
length,type,body = receive_message(s)
|
|
return
|
|
|
|
def get_event(s,events):
|
|
eventbody = struct.pack("!H", events)
|
|
s.sendall(pack_message(MSG_TYPE_SETEVENTS,eventbody))
|
|
length,type,body = receive_message(s)
|
|
return
|
|
|
|
def listen_for_events(s):
|
|
while(1):
|
|
length,type,body = receive_message(s)
|
|
return
|
|
|
|
def do_main_loop(host,port):
|
|
print "host is %s:%d"%(host,port)
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
s.connect((host,port))
|
|
authenticate(s)
|
|
get_option(s,"nickname")
|
|
get_option(s,"DirFetchPostPeriod\n")
|
|
set_option(s,"1")
|
|
set_option(s,"bandwidthburstbytes 100000")
|
|
# set_option(s,"runasdaemon 1")
|
|
# get_event(s,EVENT_TYPE_WARN)
|
|
get_event(s,EVENT_TYPE_BANDWIDTH)
|
|
|
|
listen_for_events(s)
|
|
|
|
return
|
|
|
|
if __name__ == '__main__':
|
|
if len(sys.argv) != 2:
|
|
print "Syntax: tor-control.py torhost:torport"
|
|
sys.exit(0)
|
|
sh,sp = parseHostAndPort(sys.argv[1])
|
|
do_main_loop(sh,sp)
|
|
|