src/test/test_rebind.py

   1 from __future__ import print_function
   2
   3 import sys
   4 import subprocess
   5 import socket
   6 import os
   7 import time
   8 import random
   9
  10 def try_connecting_to_socksport():
  11     socks_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  12     if socks_socket.connect_ex(('127.0.0.1', socks_port)):
  13         tor_process.terminate()
  14         print('FAIL')
  15         sys.exit('Cannot connect to SOCKSPort')
  16     socks_socket.close()
  17
  18 def wait_for_log(s):
  19     while True:
  20         l = tor_process.stdout.readline()
  21         if s in l.decode('utf8'):
  22             return
  23
  24 def pick_random_port():
  25     port = 0
  26     random.seed()
  27
  28     for i in range(8):
  29         port = random.randint(10000, 60000)
  30         s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  31         if s.connect_ex(('127.0.0.1', port)) == 0:
  32             s.close()
  33         else:
  34             break
  35
  36     return port
  37
  38 if sys.hexversion < 0x02070000:
  39     sys.exit("ERROR: unsupported Python version (should be >= 2.7)")
  40
  41 if sys.hexversion > 0x03000000 and sys.hexversion < 0x03010000:
  42     sys.exit("ERROR: unsupported Python3 version (should be >= 3.1)")
  43
  44 control_port = pick_random_port()
  45 socks_port = pick_random_port()
  46
  47 assert control_port != 0
  48 assert socks_port != 0
  49
  50 if not os.path.exists(sys.argv[1]):
  51     sys.exit('ERROR: cannot find tor at %s' % sys.argv[1])
  52
  53 tor_path = sys.argv[1]
  54
  55 tor_process = subprocess.Popen([tor_path,
  56                                '-ControlPort', '127.0.0.1:{}'.format(control_port),
  57                                '-SOCKSPort', '127.0.0.1:{}'.format(socks_port),
  58                                '-FetchServerDescriptors', '0'],
  59                                stdout=subprocess.PIPE,
  60                                stderr=subprocess.PIPE)
  61
  62 if tor_process == None:
  63     sys.exit('ERROR: running tor failed')
  64
  65 if len(sys.argv) < 2:
  66      sys.exit('Usage: %s <path-to-tor>' % sys.argv[0])
  67
  68 wait_for_log('Opened Control listener on')
  69
  70 try_connecting_to_socksport()
  71
  72 control_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  73 if control_socket.connect_ex(('127.0.0.1', control_port)):
  74     tor_process.terminate()
  75     print('FAIL')
  76     sys.exit('Cannot connect to ControlPort')
  77
  78 control_socket.sendall('AUTHENTICATE \r\n'.encode('utf8'))
  79 control_socket.sendall('SETCONF SOCKSPort=0.0.0.0:{}\r\n'.format(socks_port).encode('utf8'))
  80 wait_for_log('Opened Socks listener')
  81
  82 try_connecting_to_socksport()
  83
  84 control_socket.sendall('SETCONF SOCKSPort=127.0.0.1:{}\r\n'.format(socks_port).encode('utf8'))
  85 wait_for_log('Opened Socks listener')
  86
  87 try_connecting_to_socksport()
  88
  89 control_socket.sendall('SIGNAL HALT\r\n'.encode('utf8'))
  90
  91 time.sleep(0.1)
  92 print('OK')
  93 tor_process.terminate()