Small script to simplify network communication.
Something like telnetlib, but for clean TCP/UDP (no command sequences, \r\n newlines, etc.)
As an alternative, consider pwntools which contains unified interface (tubes) for communications with sockets, processes, etc. See tubes API.
Current development only supports python 3. Old python 2 version is available at the py2 branch.
$ pip3 install sockFor development or building from this repository, poetry is needed.
fromsockimport*f=Sock("some.cool.servi.ce:3123", timeout=10)
# or IPv6f=Sock6("::1 3123", timeout=3)
# or already existing socketf=Sock.from_socket(some_socket) # or toSock(some_socket)# or UDP/IPv6f=SockU6("::1 3123", timeout=3)
# or WebSocket# For using WebSock, the websocket-client module must be installed (pip install websocket-client).f=WebSock("ws://localhost:3123")
# wait for prompt (skip banner for example)# the prompt itself will be skipped (and returned) toof.read_until("> ", timeout=3) # read_until_re also existsf.send("flip coin\n")
# skip until regexpresult1=f.skip_until_re(r"You've got (heads|tails)") # skip_until(str) also exists# read until also consumes matched partf.read_until_re(r"You've g[oe]t ") # read_until(str) also exists# read specific number of bytesresult2=f.read_nbytes(5)
assertresult1==result2# alias for f.send(s + "\n")f.send_line("random please")
# read one packet and flush buffersprint(f.read_one())
# non-blocking read (flush buffers)print(f.read_one(0))
# read until disconnectprint(f.read_all())Author: hellman
License: MIT License