12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364 |
- import sys
- import subprocess
- import socket
- import os
- import time
- def try_connecting_to_socksport():
- socks_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
- if socks_socket.connect_ex(('127.0.0.1', 9052)):
- tor_process.terminate()
- print 'FAIL'
- sys.exit('Cannot connect to SOCKSPort')
- socks_socket.close()
- def wait_for_log(s):
- while True:
- l = tor_process.stdout.readline()
- if s in l:
- return
- if not os.path.exists(sys.argv[1]):
- sys.exit('ERROR: cannot find tor at %s' % sys.argv[1])
- tor_path = sys.argv[1]
- tor_process = subprocess.Popen([tor_path,
- '-ControlPort', '127.0.0.1:9053',
- '-SOCKSPort', '127.0.0.1:9052',
- '-FetchServerDescriptors', '0'],
- stdout=subprocess.PIPE,
- stderr=subprocess.PIPE)
- if tor_process == None:
- sys.exit('ERROR: running tor failed')
- if len(sys.argv) < 2:
- sys.exit('Usage: %s <path-to-tor>' % sys.argv[0])
- wait_for_log('Opened Control listener on')
- try_connecting_to_socksport()
- control_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
- if control_socket.connect_ex(('127.0.0.1', 9053)):
- tor_process.terminate()
- print 'FAIL'
- sys.exit('Cannot connect to ControlPort')
- control_socket.sendall('AUTHENTICATE \r\n')
- control_socket.sendall('SETCONF SOCKSPort=0.0.0.0:9052\r\n')
- wait_for_log('Opened Socks listener')
- try_connecting_to_socksport()
- control_socket.sendall('SETCONF SOCKSPort=127.0.0.1:9052\r\n')
- wait_for_log('Opened Socks listener')
- try_connecting_to_socksport()
- control_socket.sendall('SIGNAL HALT\r\n')
- time.sleep(0.1)
- print 'OK'
- tor_process.terminate()
|