import os,sys; sys.path.append(os.path.abspath(os.path.join(os.path.abspath(os.path.join(os.path.dirname(__file__),'..')),'..'))) from komrade import * # import requests_async as requests log=print from logging import Handler class MazeWalker(Handler): def __init__(self,callbacks={},*x,**y): super().__init__(*x,**y) self._callbacks=callbacks self.walk = [] def emit(self, record): from torpy.documents.network_status import Router walk=self.walk for arg in record.args: if type(arg)==Router: router = arg if router.ip in walk: continue walk.append(router.ip) # print('! Found router:',record.msg % record.args) # print(router.ip,walk) # print() f=None if record.msg.startswith('Connecting to guard node'): f = self._callbacks.get('torpy_guard_node_connect') elif record.msg.startswith('Extending the circuit'): f = self._callbacks.get('torpy_extend_circuit') # import asyncio # if f: # asyncio.create_task(f(router)) # if f: # texec = ThreadExecutor() # texec(f, router) # f(router) import asyncio asyncio.run(f(router)) pass # def komrade_request(url,allow_clearnet = ALLOW_CLEARNET): # if '.onion' in url or not allow_clearnet: # return tor_request(url) # return requests.get(url,timeout=600) # def tor_request(url): # return tor_request_in_python(url) # # return tor_request_in_proxy(url) # async def tor_request_async(url): # return await tor_request_in_python_async(url) # def tor_request_in_proxy(url): # with get_tor_proxy_session() as s: # return s.get(url,timeout=60) # async def tor_request_in_python_async(url): # import requests_async as requests # tor = TorClient() # with tor.get_guard() as guard: # adapter = TorHttpAdapter(guard, 3, retries=RETRIES) # async with requests.Session() as s: # # await s # s.headers.update({'User-Agent': 'Mozilla/5.0'}) # s.mount('http://', adapter) # s.mount('https://', adapter) # return await s.get(url, timeout=60) # def tor_request_in_python(url): # tor = TorClient() # with tor.get_guard() as guard: # adapter = TorHttpAdapter(guard, 3, retries=RETRIES) # with requests.Session() as s: # s.headers.update({'User-Agent': 'Mozilla/5.0'}) # s.mount('http://', adapter) # s.mount('https://', adapter) # # r = s.get(url, timeout=30) # # r = s.post('http://u7spnj3dmwumzoa4.onion/op/',data=b'hello world', timeout=30) # #_dat = 'Z29iYmxlZHlnb29rZ29iYmxlZHlnb29rZ29iYmxlZHlnb29rZ29iYmxlZHlnb29rZ29iYmxlZHlnb29rZ29iYmxlZHlnb29rZ29iYmxlZHlnb29rZ29iYmxlZHlnb29rZ29iYmxlZHlnb29r' # r = s.get(url, timeout=60) # return r # #return r # # #r = s.get('http://u7spnj3dmwumzoa4.onion',timeout=30) # # print (r, r.text[:1000]) # # return s # def get_tor_proxy_session(): # session = requests.session() # # Tor uses the 9050 port as the default socks port # session.proxies = {'http': 'socks5://127.0.0.1:9050', # 'https': 'socks5://127.0.0.1:9050'} # return session # def get_async_tor_proxy_session(): # import requests_futures # from requests_futures.sessions import FuturesSession # session = FuturesSession() # # Tor uses the 9050 port as the default socks port # session.proxies = {'http': 'socks5://127.0.0.1:9050', # 'https': 'socks5://127.0.0.1:9050'} # return session def test_torpy(): hostname = KOMRADE_ONION from torpy import TorClient tor = TorClient() # Choose random guard node and create 3-hops circuit with tor.create_circuit(3) as circuit: # Create tor stream to host with circuit.create_stream((hostname, 80)) as stream: # Send some data to it stream.send(b'GET / HTTP/1.0\r\nHost: %s\r\n\r\n' % hostname.encode()) recv = recv_all(stream).decode() #logger.warning('recv: %s', recv) print('RECEIVED:',recv) import random def test_slow_func(n,t=1,m=10): for x in range(m): print(f'Thread #{n}: it #{x} of {m}, sleeping {m} sec..') yield x # time.sleep(t) def test_threads(): from multiprocessing.pool import ThreadPool pool = ThreadPool(processes=2) for x in pool.imap_unordered(test_slow_func, [1,2,3]): for y in x: print(y,'?') print(x,'...') async def test_slow_func(n,t=1,m=10): for x in range(m): print(f'Thread #{n}: it #{x} of {m}, sleeping {m} sec..') yield x asyncio.sleep(t) # test async async def test_async(): return await asyncio.gather(*[await test_slow_func(n) for n in range(3)]) #### CODE FROM TORPY # Copyright 2019 James Brown # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. # import os import socket import logging from threading import Event from selectors import EVENT_READ from multiprocessing.pool import ThreadPool import requests from torpy import TorClient from torpy.stream import TorStream from torpy.utils import AuthType, recv_all, retry from torpy.http.adapter import TorHttpAdapter from torpy.hiddenservice import HiddenService from torpy.http.requests import TorRequests, tor_requests_session, do_request as requests_request from torpy.http.urlopener import do_request as urllib_request logging.getLogger('requests').setLevel(logging.CRITICAL) logging.basicConfig(format='[%(asctime)s] [%(threadName)-16s] %(message)s', level=logging.DEBUG) logger = logging.getLogger('komrade') HS_BASIC_HOST = os.getenv('HS_BASIC_HOST') HS_BASIC_AUTH = os.getenv('HS_BASIC_AUTH') HS_STEALTH_HOST = os.getenv('HS_STEALTH_HOST') HS_STEALTH_AUTH = os.getenv('HS_STEALTH_AUTH') RETRIES = 3 @retry(RETRIES, (TimeoutError, ConnectionError, )) def test_clearnet_raw(): hostname = 'ifconfig.me' tor = TorClient() # Choose random guard node and create 3-hops circuit with tor.create_circuit(3) as circuit: # Create tor stream to host with circuit.create_stream((hostname, 80)) as stream: # Send some data to it stream.send(b'GET / HTTP/1.0\r\nHost: %s\r\n\r\n' % hostname.encode()) recv = recv_all(stream).decode() logger.warning('recv: %s', recv) search_ip = '.'.join(circuit.last_node.router.ip.split('.')[:-1]) + '.' assert search_ip in recv, 'wrong data received' @retry(RETRIES, (TimeoutError, ConnectionError, )) def test_onion_raw(): hostname = 'u7spnj3dmwumzoa4.onion' tor = TorClient() # Choose random guard node and create 3-hops circuit with tor.create_circuit(3) as circuit: # Create tor stream to host with circuit.create_stream((hostname, 80)) as stream: # Send some data to it stream.send(b'GET / HTTP/1.0\r\nHost: %s\r\n\r\n' % hostname.encode()) recv = recv_all(stream).decode() logger.warning('recv: %s', recv) assert 'StickyNotes' in recv, 'wrong data received' def test_requests_no_agent(): data = requests_request('https://httpbin.org/headers', retries=RETRIES) assert 'User-Agent' not in data def test_requests(): data = requests_request('https://httpbin.org/headers', headers={'User-Agent': 'Mozilla/5.0'}, retries=RETRIES) assert 'Mozilla' in data def test_requests_session(): tor = TorClient() with tor.get_guard() as guard: adapter = TorHttpAdapter(guard, 3, retries=RETRIES) with requests.Session() as s: s.headers.update({'User-Agent': 'Mozilla/5.0'}) s.mount('http://', adapter) s.mount('https://', adapter) r = s.get('https://google.com', timeout=30) logger.warning(r) logger.warning(r.text) assert r.text.rstrip().endswith('') r = s.get('https://stackoverflow.com/questions/tagged/python') assert r.text.rstrip().endswith('') logger.warning(r) logger.warning(r.text) def test_requests_session2(): tor = TorClient() with tor.get_guard() as guard: adapter = TorHttpAdapter(guard, 3, retries=RETRIES) with requests.Session() as s: s.headers.update({'User-Agent': 'Mozilla/5.0'}) s.mount('http://', adapter) s.mount('https://', adapter) #r = s.get('https://google.com', timeout=30) r = s.get('http://u7spnj3dmwumzoa4.onion',timeout=30) print(r) print(r.text[:1000]) def test_urlopener_no_agent(): data = urllib_request('https://httpbin.org/headers', verbose=1, retries=RETRIES) assert 'User-Agent' not in data def test_urlopener(): data = urllib_request('https://httpbin.org/headers', headers=[('User-Agent', 'Mozilla/5.0')], verbose=1, retries=RETRIES) assert 'Mozilla' in data def test_multi_threaded(): auth_data = {HS_BASIC_HOST: (HS_BASIC_AUTH, AuthType.Basic)} if HS_BASIC_HOST and HS_BASIC_AUTH else None with TorRequests(auth_data=auth_data) as tor_requests: links = [ 'https://httpbin.org/headers', 'https://google.com', 'https://ifconfig.me', 'http://facebookcorewwwi.onion', ] if HS_BASIC_HOST: links.append('http://' + HS_BASIC_HOST) links = links * 10 with tor_requests.get_session(retries=RETRIES) as sess: def process(link): try: logger.debug('get link: %s', link) r = sess.get(link, timeout=30) logger.warning('get link %s finish: %s', link, r) return r except BaseException: logger.exception('get link %s error', link) pool = ThreadPool(10) for i, w in enumerate(pool._pool): w.name = 'Worker{}'.format(i) results = pool.map(process, links) pool.close() pool.join() logger.debug('test_multi_threaded ends: %r', results) @retry(RETRIES, (TimeoutError, ConnectionError, )) def test_basic_auth(): """Connecting to Hidden Service with 'Basic' authorization.""" if not HS_BASIC_HOST or not HS_BASIC_AUTH: logger.warning('Skip test_basic_auth()') return hs = HiddenService(HS_BASIC_HOST, HS_BASIC_AUTH, AuthType.Basic) tor = TorClient() # Choose random guard node and create 3-hops circuit with tor.create_circuit(3) as circuit: # Create tor stream to host with circuit.create_stream((hs, 80)) as stream: # Send some data to it stream.send(b'GET / HTTP/1.0\r\nHost: %s.onion\r\n\r\n' % hs.onion.encode()) recv = recv_all(stream).decode() logger.warning('recv: %s', recv) @retry(RETRIES, (TimeoutError, ConnectionError, )) def test_stealth_auth(): """Connecting to Hidden Service with 'Stealth' authorization.""" if not HS_STEALTH_HOST or not HS_STEALTH_AUTH: logger.warning('Skip test_stealth_auth()') return hs = HiddenService(HS_STEALTH_HOST, HS_STEALTH_AUTH, AuthType.Stealth) tor = TorClient() # Choose random guard node and create 3-hops circuit with tor.create_circuit(3) as circuit: # Create tor stream to host with circuit.create_stream((hs, 80)) as stream: # Send some data to it stream.send(b'GET / HTTP/1.0\r\nHost: %s\r\n\r\n' % hs.hostname.encode()) recv = recv_all(stream).decode() logger.warning('recv: %s', recv) @retry(RETRIES, (TimeoutError, ConnectionError, )) def test_basic_auth_pre(): """Using pre-defined authorization data for making HTTP requests.""" if not HS_BASIC_HOST or not HS_BASIC_AUTH: logger.warning('Skip test_basic_auth()') return hidden_service = HS_BASIC_HOST auth_data = {HS_BASIC_HOST: (HS_BASIC_AUTH, AuthType.Basic)} tor = TorClient(auth_data=auth_data) # Choose random guard node and create 3-hops circuit with tor.create_circuit(3) as circuit: # Create tor stream to host with circuit.create_stream((hidden_service, 80)) as stream: # Send some data to it stream.send(b'GET / HTTP/1.0\r\nHost: %s.onion\r\n\r\n' % hidden_service.encode()) recv = recv_all(stream).decode() logger.warning('recv: %s', recv) def test_requests_hidden(): """Using pre-defined authorization data for making HTTP requests by tor_requests_session.""" if not HS_BASIC_HOST or not HS_BASIC_AUTH: logger.warning('Skip test_requests_hidden()') return auth_data = {HS_BASIC_HOST: (HS_BASIC_AUTH, AuthType.Basic)} with tor_requests_session(auth_data=auth_data, retries=RETRIES) as sess: r = sess.get('http://{}/'.format(HS_BASIC_HOST), timeout=30) logger.warning(r) logger.warning(r.text) @retry(2, (TimeoutError, ConnectionError, )) def test_select(): sock_r, sock_w = socket.socketpair() events = {TorStream: {'data': Event(), 'close': Event()}, socket.socket: {'data': Event(), 'close': Event()}} hostname = 'ifconfig.me' tor = TorClient() with tor.get_guard() as guard: def recv_callback(sock_or_stream, mask): logger.debug(f'recv_callback {sock_or_stream}') kind = type(sock_or_stream) data = sock_or_stream.recv(1024) logger.info('%s: %r', kind.__name__, data.decode()) if data: events[kind]['data'].set() else: logger.debug('closing') guard.unregister(sock_or_stream) events[kind]['close'].set() with guard.create_circuit(3) as circuit: with circuit.create_stream((hostname, 80)) as stream: guard.register(sock_r, EVENT_READ, recv_callback) guard.register(stream, EVENT_READ, recv_callback) stream.send(b'GET / HTTP/1.0\r\nHost: %s\r\n\r\n' % hostname.encode()) sock_w.send(b'some data') assert events[socket.socket]['data'].wait(10), 'no sock data received' assert events[TorStream]['data'].wait(30), 'no stream data received' sock_w.close() assert events[socket.socket]['close'].wait(10), 'no sock close received' assert events[TorStream]['close'].wait(10), 'no stream close received' if __name__=='__main__': test_torpy()