Add some stuffs
This commit is contained in:
36
python/utils/connect.py
Normal file
36
python/utils/connect.py
Normal file
@ -0,0 +1,36 @@
|
||||
#!/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()
|
Reference in New Issue
Block a user