37 lines
774 B
Python
37 lines
774 B
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import socket
|
||
|
|
import sys
|
||
|
|
|
||
|
|
def get_header(sock):
|
||
|
|
try:
|
||
|
|
data = sock.recv(4096, socket.MSG_DONTWAIT) # 0x40 : O_NONBLOCK
|
||
|
|
#sys.stdout.write(data)
|
||
|
|
print(data.decode("utf-8"), end="")
|
||
|
|
except socket.error as msg:
|
||
|
|
# Nothing left
|
||
|
|
pass
|
||
|
|
|
||
|
|
try:
|
||
|
|
sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
|
||
|
|
except socket.error as msg:
|
||
|
|
print(msg)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
try:
|
||
|
|
sock = socket.create_connection((sys.argv[1], sys.argv[2]),60)
|
||
|
|
except socket.error as msg:
|
||
|
|
print(msg)
|
||
|
|
sock.close()
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
if sock is None:
|
||
|
|
print('Unable to connect')
|
||
|
|
sys.exit(2)
|
||
|
|
|
||
|
|
#sock.setblocking(0)
|
||
|
|
sock.shutdown(socket.SHUT_WR)
|
||
|
|
print("Connected to %s:%s" % (sys.argv[1],sys.argv[2]))
|
||
|
|
get_header(sock)
|
||
|
|
sock.close()
|