Python socket.socket() Examples
The following are 30
code examples of socket.socket().
You can vote up the ones you like or vote down the ones you don't like,
and go to the original project or source file by following the links above each example.
You may also want to check out all available functions/classes of the module
socket
, or try the search function
.
.
Example #1
| Source File: server.py From RF-Monitor with GNU General Public License v2.0 | 10 votes |
|
def __init__(self, eventHandler):
threading.Thread.__init__(self)
self.name = 'Server'
self.daemon = True
self._eventHandler = eventHandler
self._client = None
self._server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self._server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
try:
self._server.bind(('', PORT))
self._server.listen(5)
except socket.error:
event = Event(Events.SCAN_ERROR, msg='Could not start server')
post_event(eventHandler, event)
return
self._cancel = False
self.start()
Example #2
| Source File: SockPuppet.py From ALF with Apache License 2.0 | 8 votes |
|
def connect(self):
if self.is_server:
log.debug("waiting for client to connect...")
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind(('', self.port))
s.settimeout(0.1)
start_time = time.time()
s.listen(0)
while True:
try:
conn, _ = s.accept()
self.conn = conn
break
except socket.timeout:
pass
if self.timeout > 0 and time.time() - start_time >= self.timeout:
s.close()
raise RuntimeError("Timeout exceeded (%ds)" % self.timeout)
self.conn.setblocking(True)
else:
log.debug("connecting to server (%s:%d)...", self.ip, self.port)
self.conn = socket.create_connection((self.ip, self.port), self.timeout)
Example #3
| Source File: start.py From Starx_Pixiv_Collector with MIT License | 8 votes |
|
def ip_latency_test(ip, port=443):
tag = 'IP_Latency_TEST'
print_with_tag(tag, ['Prepare IP latency test for ip', ip, 'Port', str(port)])
s_test = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s_test.settimeout(10)
s_start = time.time()
try:
s_test.connect((ip, port))
s_test.shutdown(socket.SHUT_RD)
except Exception as e:
print_with_tag(tag, ['Error:', e])
return None
s_stop = time.time()
s_runtime = '%.2f' % (1000 * (s_stop - s_start))
print_with_tag(tag, [ip, 'Latency:', s_runtime])
return float(s_runtime)
Example #4
| Source File: interface.py From XFLTReaT with MIT License | 6 votes |
|
def mac_set_ip_address(self, dev, ip, serverip, netmask):
ifr = struct.pack('<16sBBHIIIBBHIIIBBHIII',
self.iface_name,
16, socket.AF_INET, 0, struct.unpack('<L', socket.inet_pton(socket.AF_INET, ip))[0], 0, 0,
16, socket.AF_INET, 0, struct.unpack('<L', socket.inet_pton(socket.AF_INET, serverip))[0], 0, 0,
16, 0, 0, struct.unpack('<L', socket.inet_pton(socket.AF_INET, "255.255.255.255"))[0], 0, 0)
try:
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
fcntl.ioctl(sock, self.IOCTL_MACOSX_SIOCAIFADDR, ifr)
except Exception as e:
common.internal_print("Something went wrong with setting up the interface.", -1)
print(e)
sys.exit(-1)
# adding new route for forwarding packets properly.
integer_ip = struct.unpack(">I", socket.inet_pton(socket.AF_INET, serverip))[0]
rangeip = socket.inet_ntop(socket.AF_INET, struct.pack(">I", integer_ip & ((2**int(netmask))-1)<<32-int(netmask)))
ps = subprocess.Popen(["route", "add", "-net", rangeip+"/"+netmask, serverip], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
(stdout, stderr) = ps.communicate()
if stderr:
if not "File exists" in stderr:
common.internal_print("Error: adding client route: {0}".format(stderr), -1)
sys.exit(-1)
return
Example #5
| Source File: sdnip.py From iSDX with Apache License 2.0 | 6 votes |
|
def __init__(self, name, intfDict,
asNum, neighbors, routes=[],
quaggaConfFile=None,
zebraConfFile=None,
runDir='/var/run/quagga', *args, **kwargs):
super(BgpRouter, self).__init__(name, intfDict, **kwargs)
self.runDir = runDir
self.routes = routes
if quaggaConfFile is not None:
self.quaggaConfFile = quaggaConfFile
self.zebraConfFile = zebraConfFile
else:
self.quaggaConfFile = '%s/quagga%s.conf' % (runDir, name)
self.zebraConfFile = '%s/zebra%s.conf' % (runDir, name)
self.asNum = asNum
self.neighbors = neighbors
self.generateConfig()
self.socket = '%s/zebra%s.api' % (self.runDir, self.name)
self.quaggaPidFile = '%s/quagga%s.pid' % (self.runDir, self.name)
self.zebraPidFile = '%s/zebra%s.pid' % (self.runDir, self.name)
Example #6
| Source File: sdnip.py From iSDX with Apache License 2.0 | 6 votes |
|
def __init__(self, name, intfDict,
asNum, neighbors, routes=[],
quaggaConfFile=None,
zebraConfFile=None,
runDir='/var/run/quagga', *args, **kwargs):
super(BgpRouter, self).__init__(name, intfDict, **kwargs)
self.runDir = runDir
self.routes = routes
if quaggaConfFile is not None:
self.quaggaConfFile = quaggaConfFile
self.zebraConfFile = zebraConfFile
else:
self.quaggaConfFile = '%s/quagga%s.conf' % (runDir, name)
self.zebraConfFile = '%s/zebra%s.conf' % (runDir, name)
self.asNum = asNum
self.neighbors = neighbors
self.generateConfig()
self.socket = '%s/zebra%s.api' % (self.runDir, self.name)
self.quaggaPidFile = '%s/quagga%s.pid' % (self.runDir, self.name)
self.zebraPidFile = '%s/zebra%s.pid' % (self.runDir, self.name)
Example #7
| Source File: __init__.py From aegea with Apache License 2.0 | 6 votes |
|
def wait_for_port(host, port, timeout=600, print_progress=True):
if print_progress:
sys.stderr.write("Waiting for {}:{}...".format(host, port))
sys.stderr.flush()
start_time = time.time()
while True:
try:
socket.socket().connect((host, port))
if print_progress:
sys.stderr.write(GREEN("OK") + "\n")
return
except Exception:
time.sleep(1)
if print_progress:
sys.stderr.write(".")
sys.stderr.flush()
if time.time() - start_time > timeout:
raise
Example #8
| Source File: sdnip.py From iSDX with Apache License 2.0 | 6 votes |
|
def __init__(self, name, intfDict,
asNum, neighbors, routes=[],
quaggaConfFile=None,
zebraConfFile=None,
runDir='/var/run/quagga', *args, **kwargs):
super(BgpRouter, self).__init__(name, intfDict, **kwargs)
self.runDir = runDir
self.routes = routes
if quaggaConfFile is not None:
self.quaggaConfFile = quaggaConfFile
self.zebraConfFile = zebraConfFile
else:
self.quaggaConfFile = '%s/quagga%s.conf' % (runDir, name)
self.zebraConfFile = '%s/zebra%s.conf' % (runDir, name)
self.asNum = asNum
self.neighbors = neighbors
self.generateConfig()
self.socket = '%s/zebra%s.api' % (self.runDir, self.name)
self.quaggaPidFile = '%s/quagga%s.pid' % (self.runDir, self.name)
self.zebraPidFile = '%s/zebra%s.pid' % (self.runDir, self.name)
Example #9
| Source File: sdnip.py From iSDX with Apache License 2.0 | 6 votes |
|
def __init__(self, name, intfDict,
asNum, neighbors, routes=[],
quaggaConfFile=None,
zebraConfFile=None,
runDir='/var/run/quagga', *args, **kwargs):
super(BgpRouter, self).__init__(name, intfDict, **kwargs)
self.runDir = runDir
self.routes = routes
if quaggaConfFile is not None:
self.quaggaConfFile = quaggaConfFile
self.zebraConfFile = zebraConfFile
else:
self.quaggaConfFile = '%s/quagga%s.conf' % (runDir, name)
self.zebraConfFile = '%s/zebra%s.conf' % (runDir, name)
self.asNum = asNum
self.neighbors = neighbors
self.generateConfig()
self.socket = '%s/zebra%s.api' % (self.runDir, self.name)
self.quaggaPidFile = '%s/quagga%s.pid' % (self.runDir, self.name)
self.zebraPidFile = '%s/zebra%s.pid' % (self.runDir, self.name)
Example #10
| Source File: sdnip.py From iSDX with Apache License 2.0 | 6 votes |
|
def config(self, **kwargs):
super(BgpRouter, self).config(**kwargs)
self.cmd('%s/zebra -d -f %s -z %s -i %s'
% (BgpRouter.binDir, self.zebraConfFile, self.socket, self.zebraPidFile))
while True:
try:
s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) # @UndefinedVariable
s.connect(self.socket)
#print 'connected - breaking'
s.close()
break
except Exception, e:
#print' ERROR: ' + repr(e)
time.sleep(.1)
#print 'zebra ready'
Example #11
| Source File: tnode.py From iSDX with Apache License 2.0 | 6 votes |
|
def create_command_listener (baddr, port):
try:
if port is None:
try:
if os.path.exists(baddr):
os.unlink(baddr)
except OSError:
print 'could not remove old unix socket ' + baddr
return
s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) # @UndefinedVariable
s.bind(baddr)
else:
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind((baddr, int(port)))
except socket.error , msg:
print 'Bind failed on command interface ' + baddr + ' port ' + str(port) + ' Error Code : ' + str(msg[0]) + ' Message ' + msg[1] + '\n'
return
Example #12
| Source File: tmgr.py From iSDX with Apache License 2.0 | 6 votes |
|
def connect (host, why):
# should be either a listener host or a router host (edge-router)
if host not in bgprouters and host not in hosts and host not in participants:
log.error('MM:' + host + ' ERROR: ' + why + ': Unknown host: ' + host)
return None
try:
hostdata = hosts[host]
except:
try:
hostdata = bgprouters[host]
except:
hostdata = participants[host]
#print 'MM:' + host + ' INFO: ' + why + ': Connecting to ' + host + ' at ' + hostdata.host + ':' + str(hostdata.port)
try:
if hostdata.port is None:
s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) # @UndefinedVariable
s.connect(hostdata.host)
else:
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect((hostdata.host, int(hostdata.port)))
except Exception, e:
log.error('MM:' + host + ' ERROR: ' + why + ': ' + repr(e))
return None
Example #13
| Source File: tmgr.py From iSDX with Apache License 2.0 | 6 votes |
|
def router (args):
if len(args) < 2:
log.error('MM:00 ROUTER: ERROR: usage: router arg arg ...')
return
host = args[0]
if host not in bgprouters:
log.error('MM:' + host + ' ERROR: ' + 'ROUTER' + ' ' + host + ' : must be a BGP router')
return
del args[0]
cmd = ''
for arg in args:
cmd += '"' + arg + '" '
log.info('MM:' + host + ' ROUTER: ' + cmd)
r = generic(host, 'ROUTER', 'router ' + cmd + '\n')
if r is not None:
log.debug('MM:' + host + ' ROUTER: output = \n' + r.strip())
# generic command interface to a tnode - send cmd, capture data
# return None id cannot connect or socket error
# return '' if no data
Example #14
| Source File: tmgr.py From iSDX with Apache License 2.0 | 6 votes |
|
def genericObjNW (host, label, cmd):
if host not in participants:
log.error('MM:' + host + ' ERROR: ' + label + ': Can only send to a participant: ' + host)
return None
try:
hostdata = hosts[host]
except:
try:
hostdata = bgprouters[host]
except:
hostdata = participants[host]
#print 'MM:' + host + ' INFO: ' + why + ': Connecting to ' + host + ' at ' + hostdata.host + ':' + str(hostdata.port)
try:
if hostdata.port is None:
s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) # @UndefinedVariable
s.connect(hostdata.host)
else:
s = Client((hostdata.host, int(hostdata.port)))
except Exception, e:
log.error('MM:' + host + ' ERROR: ' + label + ': ' + repr(e))
return None
Example #15
| Source File: client-test2.py From The-chat-room with MIT License | 6 votes |
|
def fileGet(name):
PORT3 = 50009
ss2 = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
ss2.connect((IP, PORT3))
message = 'get ' + name
ss2.send(message.encode())
fileName = '.\\Client_image_cache\\' + name
print('Start downloading image!')
print('Waiting.......')
with open(fileName, 'wb') as f:
while True:
data = ss2.recv(1024)
if data == 'EOF'.encode():
print('Download completed!')
break
f.write(data)
time.sleep(0.1)
ss2.send('quit'.encode())
# 将图片上传到图片服务端的缓存文件夹中
Example #16
| Source File: client-test2.py From The-chat-room with MIT License | 6 votes |
|
def video_invite():
global IsOpen, Version, AudioOpen
if Version == 4:
host_name = socket.gethostbyname(socket.getfqdn(socket.gethostname()))
else:
host_name = [i['addr'] for i in ifaddresses(interfaces()[-2]).setdefault(AF_INET6, [{'addr': 'No IP addr'}])][
-1]
invite = 'INVITE' + host_name + ':;' + user + ':;' + chat
s.send(invite.encode())
if not IsOpen:
vserver = vachat.Video_Server(10087, Version)
if AudioOpen:
aserver = vachat.Audio_Server(10088, Version)
aserver.start()
vserver.start()
IsOpen = True
Example #17
| Source File: client.py From The-chat-room with MIT License | 6 votes |
|
def video_invite():
global IsOpen, Version, AudioOpen
if Version == 4:
host_name = socket.gethostbyname(socket.getfqdn(socket.gethostname()))
else:
host_name = [i['addr'] for i in ifaddresses(interfaces()[-2]).setdefault(AF_INET6, [{'addr': 'No IP addr'}])][
-1]
invite = 'INVITE' + host_name + ':;' + user + ':;' + chat
s.send(invite.encode())
if not IsOpen:
vserver = vachat.Video_Server(10087, Version)
if AudioOpen:
aserver = vachat.Audio_Server(10088, Version)
aserver.start()
vserver.start()
IsOpen = True
Example #18
| Source File: client-test.py From The-chat-room with MIT License | 6 votes |
|
def fileGet(name):
PORT3 = 50009
ss2 = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
ss2.connect((IP, PORT3))
message = 'get ' + name
ss2.send(message.encode())
fileName = '.\\Client_image_cache\\' + name
print('Start downloading image!')
print('Waiting.......')
with open(fileName, 'wb') as f:
while True:
data = ss2.recv(1024)
if data == 'EOF'.encode():
print('Download completed!')
break
f.write(data)
time.sleep(0.1)
ss2.send('quit'.encode())
# 将图片上传到图片服务端的缓存文件夹中
Example #19
| Source File: TCP_generic.py From XFLTReaT with MIT License | 6 votes |
|
def send(self, channel_type, message, additional_data):
if channel_type == common.CONTROL_CHANNEL_BYTE:
transformed_message = self.transform(self.encryption, common.CONTROL_CHANNEL_BYTE+message, 1)
else:
transformed_message = self.transform(self.encryption, common.DATA_CHANNEL_BYTE+message, 1)
common.internal_print("{0} sent: {1}".format(self.module_short, len(transformed_message)), 0, self.verbosity, common.DEBUG)
# WORKAROUND?!
# Windows: It looks like when the buffer fills up the OS does not do
# congestion control, instead throws and exception/returns with
# WSAEWOULDBLOCK which means that we need to try it again later.
# So we sleep 100ms and hope that the buffer has more space for us.
# If it does then it sends the data, otherwise tries it in an infinite
# loop...
while True:
try:
return self.comms_socket.send(struct.pack(">H", len(transformed_message))+transformed_message)
except socket.error as se:
if se.args[0] == 10035: # WSAEWOULDBLOCK
time.sleep(0.1)
pass
else:
raise
Example #20
| Source File: TCP_generic.py From XFLTReaT with MIT License | 6 votes |
|
def stop(self):
self._stop = True
if self.threads:
for t in self.threads:
t.stop()
# not so nice solution to get rid of the block of listen()
# unfortunately close() does not help on the block
try:
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
serverbind = self.config.get("Global", "serverbind")
if serverbind == "0.0.0.0":
# windows does not like to connect to 0.0.0.0
serverbind = "127.0.0.1"
server_socket.connect((serverbind, int(self.config.get(self.get_module_configname(), "serverport"))))
except:
pass
return
Example #21
| Source File: TCP_generic.py From XFLTReaT with MIT License | 6 votes |
|
def check(self):
try:
common.internal_print("Checking module on server: {0}".format(self.get_module_name()))
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_socket.settimeout(3)
server_socket.connect((self.config.get("Global", "remoteserverip"), int(self.config.get(self.get_module_configname(), "serverport"))))
client_fake_thread = TCP_generic_thread(0, 0, None, None, server_socket, None, self.authentication, self.encryption_module, self.verbosity, self.config, self.get_module_name())
client_fake_thread.do_check()
client_fake_thread.communication(True)
self.cleanup(server_socket)
except socket.timeout:
common.internal_print("Checking failed: {0}".format(self.get_module_name()), -1)
self.cleanup(server_socket)
except socket.error as exception:
if exception.args[0] == 111:
common.internal_print("Checking failed: {0}".format(self.get_module_name()), -1)
else:
common.internal_print("Connection error: {0}".format(self.get_module_name()), -1)
self.cleanup(server_socket)
return
Example #22
| Source File: UDP_generic.py From XFLTReaT with MIT License | 6 votes |
|
def serve(self):
server_socket = None
try:
common.internal_print("Starting module: {0} on {1}:{2}".format(self.get_module_name(), self.config.get("Global", "serverbind"), int(self.config.get(self.get_module_configname(), "serverport"))))
server_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
whereto = (self.config.get("Global", "serverbind"), int(self.config.get(self.get_module_configname(), "serverport")))
server_socket.bind(whereto)
self.comms_socket = server_socket
self.serverorclient = 1
self.authenticated = False
self.communication_initialization()
self.communication(False)
except KeyboardInterrupt:
self.cleanup()
return
self.cleanup()
return
Example #23
| Source File: UDP_generic.py From XFLTReaT with MIT License | 6 votes |
|
def connect(self):
try:
common.internal_print("Starting client: {0}".format(self.get_module_name()))
server_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
self.server_tuple = (self.config.get("Global", "remoteserverip"), int(self.config.get(self.get_module_configname(), "serverport")))
self.comms_socket = server_socket
self.serverorclient = 0
self.authenticated = False
self.do_hello()
self.communication(False)
except KeyboardInterrupt:
self.do_logoff()
self.cleanup()
raise
except socket.error:
self.cleanup()
raise
self.cleanup()
return
Example #24
| Source File: interface.py From XFLTReaT with MIT License | 6 votes |
|
def lin_set_ip_address(self, dev, ip, serverip, netmask):
sockfd = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
try:
# set IP
ifr = struct.pack('<16sH2s4s8s', dev, socket.AF_INET, "\x00"*2, socket.inet_aton(ip), "\x00"*8)
fcntl.ioctl(sockfd, self.IOCTL_LINUX_SIOCSIFADDR, ifr)
# get flags
ifr = struct.pack('<16sh', dev, 0)
flags = struct.unpack('<16sh', fcntl.ioctl(sockfd, self.IOCTL_LINUX_SIOCSIFFLAGS, ifr))[1]
# set new flags
flags = flags | self.IOCTL_LINUX_IFF_UP
ifr = struct.pack('<16sh', dev, flags)
# iface up
fcntl.ioctl(sockfd, self.IOCTL_LINUX_SIOCSIFFLAGS, ifr)
except Exception as e:
common.internal_print("Something went wrong with setting up the interface.", -1)
print(e)
sys.exit(-1)
# adding new route for forwarding packets properly.
integer_ip = struct.unpack(">I", socket.inet_pton(socket.AF_INET, serverip))[0]
rangeip = socket.inet_ntop(socket.AF_INET, struct.pack(">I", integer_ip & ((2**int(netmask))-1)<<32-int(netmask)))
integer_netmask = struct.pack(">I", ((2**int(netmask))-1)<<32-int(netmask))
netmask = socket.inet_ntoa(integer_netmask)
ps = subprocess.Popen(["route", "add", "-net", rangeip, "netmask", netmask, "dev", dev], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
(stdout, stderr) = ps.communicate()
if stderr:
if not "File exists" in stderr:
common.internal_print("Error: adding client route: {0}".format(stderr), -1)
sys.exit(-1)
return
Example #25
| Source File: client.py From The-chat-room with MIT License | 6 votes |
|
def fileGet(name):
PORT3 = 50009
ss2 = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
ss2.connect((IP, PORT3))
message = 'get ' + name
ss2.send(message.encode())
fileName = '.\\Client_image_cache\\' + name
print('Start downloading image!')
print('Waiting.......')
with open(fileName, 'wb') as f:
while True:
data = ss2.recv(1024)
if data == 'EOF'.encode():
print('Download completed!')
break
f.write(data)
time.sleep(0.1)
ss2.send('quit'.encode())
# 将图片上传到图片服务端的缓存文件夹中
Example #26
| Source File: DNS.py From XFLTReaT with MIT License | 5 votes |
|
def check(self):
try:
common.internal_print("Using nameserver: {0}".format(self.nameserver))
common.internal_print("Checking module on server: {0}".format(self.get_module_name()))
server_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
self.server_tuple = (self.nameserver, self.serverport)
self.comms_socket = server_socket
self.serverorclient = 0
self.authenticated = False
self.qMTU = self.DNS_proto.reverse_RR_type("A")[4](255, self.hostname, 4, self.upload_encoding_class)
self.do_check()
self.communication(True)
except KeyboardInterrupt:
self.cleanup()
raise
except socket.timeout:
common.internal_print("Checking failed: {0}".format(self.get_module_name()), -1)
except socket.error:
self.cleanup()
raise
self.cleanup()
return
Example #27
| Source File: DNS.py From XFLTReaT with MIT License | 5 votes |
|
def serve(self):
server_socket = None
if self.zonefile:
(hostname, self.ttl, self.zone) = self.DNS_common.parse_zone_file(self.zonefile)
if hostname and (hostname+"." != self.hostname):
common.internal_print("'hostname' in '{0}' section does not match with the zonefile's origin".format(self.get_module_configname()), -1)
return
try:
common.internal_print("Starting module: {0} on {1}:{2}".format(self.get_module_name(), self.config.get("Global", "serverbind"), self.serverport))
server_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
whereto = (self.config.get("Global", "serverbind"), self.serverport)
server_socket.bind(whereto)
self.comms_socket = server_socket
self.serverorclient = 1
self.authenticated = False
self.communication_initialization()
self.communication(False)
except KeyboardInterrupt:
self.cleanup()
return
self.cleanup()
return
Example #28
| Source File: ICMP.py From XFLTReaT with MIT License | 5 votes |
|
def lookup_client_pub(self, additional_data): addr = additional_data[0] identifier = additional_data[1] client_public_ip = socket.inet_aton(addr[0]) for c in self.clients: if (c.get_public_ip_addr() == client_public_ip) and (c.get_ICMP_received_identifier() == identifier): return c return None
Example #29
| Source File: ICMP.py From XFLTReaT with MIT License | 5 votes |
|
def send(self, channel_type, message, additional_data):
addr = additional_data[0]
identifier = additional_data[1]
sequence = additional_data[2]
queue_length = additional_data[3]
if queue_length < 256:
ql = chr(queue_length)
else:
ql = chr(255)
if channel_type == common.CONTROL_CHANNEL_BYTE:
transformed_message = self.transform(self.get_client_encryption(additional_data), ql+common.CONTROL_CHANNEL_BYTE+message, 1)
else:
transformed_message = self.transform(self.get_client_encryption(additional_data), ql+common.DATA_CHANNEL_BYTE+message, 1)
common.internal_print("ICMP sent: {0} seq: {1} id: {2}".format(len(transformed_message), sequence, identifier), 0, self.verbosity, common.DEBUG)
packet = self.icmp.create_packet(self.ICMP_send, identifier, sequence,
self.ICMP_prefix+struct.pack(">H", len(transformed_message))+transformed_message)
# WORKAROUND?!
# Windows: It looks like when the buffer fills up the OS does not do
# congestion control, instead throws and exception/returns with
# WSAEWOULDBLOCK which means that we need to try it again later.
# So we sleep 100ms and hope that the buffer has more space for us.
# If it does then it sends the data, otherwise tries it in an infinite
# loop...
while True:
try:
return self.comms_socket.sendto(packet, addr)
except socket.error as se:
if se.args[0] == 10035: # WSAEWOULDBLOCK
time.sleep(0.1)
pass
else:
raise
Example #30
| Source File: ICMP.py From XFLTReaT with MIT License | 5 votes |
|
def serve(self):
server_socket = None
self.serverorclient = 1
try:
common.internal_print("Starting module: {0} on {1}".format(self.get_module_name(), self.config.get("Global", "serverbind")))
server_socket = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_ICMP)
if (self.os_type == common.OS_WINDOWS) or (self.os_type == common.OS_MACOSX):
common.internal_print("This module can be run in client mode only on this operating system.", -1)
self.cleanup()
return
self.comms_socket = server_socket
self.authenticated = False
self.communication_initialization()
self.communication(False)
except KeyboardInterrupt:
self.cleanup()
return
self.cleanup()
return
