p2p_connection.py

# p2p_connection.py
import socket
import threading
from stun_discovery import discover_public_ip_port

ext_ip, ext_port, nat_type = discover_public_ip_port()
print(f"My IP: {ext_ip}, My port: {ext_port}, NAT Type: {nat_type}")

# Ask for the peer's IP and port
peer_ip = input("Enter the peer's IP address: ")
peer_port = int(input("Enter the peer's port: "))

talking_port = 42424


def receive_messages():
    print(f"Start listening..")
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)  # Enable address reuse
    sock.bind(('0.0.0.0', peer_port))
    
    while True:
        data, addr = sock.recvfrom(1024)
        print(f"\nPeer {addr}: {data.decode()}\n")
        print("Enter message to send: ", end=">")

def main():
    
    # Start the receiving thread
    recv_thread = threading.Thread(target=receive_messages, daemon=True)
    recv_thread.start()
    
    print('Start punching hole..')
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)  # Enable address reuse
    sock.bind(('0.0.0.0', peer_port))
    sock.sendto(b'0', (peer_ip, talking_port))
    
    
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)  # Enable address reuse
    sock.bind(('0.0.0.0', talking_port))
    
    while True:
        message = input("\nEnter message to send: ")
        sock.sendto(message.encode(), (peer_ip, peer_port))

if __name__ == "__main__":
    main()