Basic tor runner

This commit is contained in:
Rémi Oudin 2018-02-25 19:42:58 +01:00
parent f7313ff659
commit ae5699c089

58
histories/tor_runner.py Normal file
View file

@ -0,0 +1,58 @@
"""
Modules that handles tor instaces creations in order to safely run histories
"""
import stem.process as tor
import shutil
import asyncio
class TorInstance():
"""
A tor instance object, with some useful information.
It is designed to be used as a worker in order to replay an history.
"""
BASE_SOCKS_PORT = 40000
BASE_CONTROL_PORT = 20000
BASE_DATA_DIR = "/tmp/tor{}/"
TOR_RUNNER = 0
@classmethod
async def create(cls):
""" Factory creation of tor processes"""
socks_port = cls.BASE_SOCKS_PORT + cls.TOR_RUNNER
control_port = cls.BASE_CONTROL_PORT + cls.TOR_RUNNER
data_dir = cls.BASE_DATA_DIR.format(cls.TOR_RUNNER)
TorInstance.TOR_RUNNER += 1
self = TorInstance()
self.socks_port = socks_port
self.control_port = control_port
self.data_dir = data_dir
self.process = tor.launch_tor_with_config(
config = {
'ControlPort' : str(control_port),
'SocksPort' : str(socks_port),
'DataDir' : data_dir
}
)
return self
def __str__(self):
""" Utility function """
return "[TOR] SOCKSPort: {0.socks_port}, ControlPort: "
"{0.control_port}, DataDir: {0.data_dir}".format(self)
async def kill(self):
""" Kills the process and remove the data dir"""
self.process.kill()
shutil.rmtree(self.data_dir)
async def main():
""" Test function """
for i in range(3):
a = await TorInstance.create()
print(a)
await a.kill()
if __name__ == "__main__":
loop = asyncio.get_event_loop()
loop.run_until_complete(main())