97 lines
3.5 KiB
Python
97 lines
3.5 KiB
Python
import sys
|
|
import socket
|
|
from time import time, sleep
|
|
from threading import Thread
|
|
|
|
from stem.control import Controller
|
|
|
|
try:
|
|
from youandme.tor import launch_tor
|
|
except ModuleNotFoundError:
|
|
pass
|
|
|
|
from youandme.server import server
|
|
from youandme.client import client
|
|
|
|
class _Address:
|
|
address = ""
|
|
|
|
|
|
def connector(host, send_data, recv_data, address="", control_port=1337, socks_port=1338):
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
result = sock.connect_ex(('127.0.0.1', socks_port))
|
|
if result != 0:
|
|
launch_tor(control_port=control_port, socks_port=socks_port)
|
|
if host:
|
|
with Controller.from_port(port=control_port) as controller:
|
|
controller.authenticate()
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
|
ip = '127.0.0.1'
|
|
s.bind((ip, 0))
|
|
s.listen(1)
|
|
port = s.getsockname()[1]
|
|
serv = controller.create_ephemeral_hidden_service(
|
|
{1337: '127.0.0.1:' + str(port)},
|
|
key_content='ED25519-V3',
|
|
await_publication=True,
|
|
)
|
|
_Address.address = serv.service_id
|
|
conn, addr = s.accept()
|
|
server(0.01, controller, conn, send_data, recv_data)
|
|
else:
|
|
if not address.endswith('.onion'):
|
|
address += '.onion'
|
|
client(0.01, address, socks_port, send_data, recv_data)
|
|
|
|
|
|
def chat(mode, send_data, recv_data):
|
|
display_buffer = []
|
|
if mode == 'host':
|
|
while _Address.address == "":
|
|
sleep(0.01)
|
|
print(_Address.address)
|
|
def display_new():
|
|
while True:
|
|
try:
|
|
char = chr(recv_data.pop(0))
|
|
display_buffer.append(char)
|
|
if char == "\n" or char == "\r\n" or len(display_buffer) > 100:
|
|
#print("\033[1;33m", char, "\033[0m", end="\n")
|
|
while len(display_buffer) != 0:
|
|
#print("\033[1;33m", display_buffer.pop(0), "\033[0m", end='')
|
|
print("\033[1;33m" + display_buffer.pop(0) + "\033[0m", end="")
|
|
|
|
except IndexError:
|
|
pass
|
|
sleep(0.1)
|
|
Thread(target=display_new, daemon=True).start()
|
|
def make_message():
|
|
while True:
|
|
new = input("\033[0m").encode('utf-8')
|
|
for b in new:
|
|
send_data.append(b)
|
|
send_data.append(ord(b"\n"))
|
|
Thread(target=make_message, daemon=True).start()
|
|
while True:
|
|
try:
|
|
if send_data is None:
|
|
print("Well crap, we lost connection.")
|
|
break
|
|
sleep(1)
|
|
except KeyboardInterrupt:
|
|
pass
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) >= 2:
|
|
if sys.argv[1] == 'chat':
|
|
send_data = bytearray()
|
|
recv_data = bytearray()
|
|
if sys.argv[2] == 'host':
|
|
Thread(target=connector, args=[True, send_data, recv_data], kwargs={'socks_port': int(input("socks")), 'control_port': int(input('control port'))}, daemon=True).start()
|
|
elif sys.argv[2].startswith('conn'):
|
|
Thread(target=connector, args=[False, send_data, recv_data], kwargs={'address': sys.argv[3], 'socks_port': int(input("socks")), 'control_port': int(input('control port'))}, daemon=True).start()
|
|
else:
|
|
print('Must specify host or conn')
|
|
sys.exit(1)
|
|
chat(sys.argv[2], send_data, recv_data)
|