|
|
#import tftpy
|
|
|
import socket
|
|
|
|
|
|
class UDPComm:
|
|
|
|
|
|
__HEADER = "ABS"
|
|
|
|
|
|
def __init__(self, ipSource, ipDestino, portDestino, asServer=False):
|
|
|
|
|
|
self.ipSource = ipSource
|
|
|
self.ipDestino = ipDestino
|
|
|
self.portDestino = portDestino
|
|
|
self.addr = (ipDestino,portDestino)
|
|
|
self.answer = "none" #test
|
|
|
self.mode = "none"
|
|
|
|
|
|
|
|
|
self.openSocket(asServer)
|
|
|
|
|
|
def openSocket(self, asServer):
|
|
|
|
|
|
#self.socket_c = socket.socket(AF_INET,SOCK_DGRAM)
|
|
|
self.socket_c = socket.socket(socket.AF_INET, socket.SOCK_DGRAM,0)
|
|
|
# self.socket_c.connect((self.ipDestino, self.portDestino))
|
|
|
|
|
|
if asServer:
|
|
|
self.configAsServer()
|
|
|
self.mode = "server"
|
|
|
else:
|
|
|
self.configAsClient()
|
|
|
self.mode = "client"
|
|
|
|
|
|
def configAsClient(self):
|
|
|
#Configurar broadcast
|
|
|
self.socket_c.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
|
|
|
|
|
|
def configAsServer(self):
|
|
|
|
|
|
self.socket_c.bind(self.addr)
|
|
|
print "\nServer initialized"
|
|
|
|
|
|
def waitData(self, nbytes = 16384):
|
|
|
|
|
|
print "\nWaiting some data"
|
|
|
trama_rx, self.answer = self.socket_c.recvfrom(nbytes)
|
|
|
print "\nThis socket has received some data from:"
|
|
|
print self.answer
|
|
|
ipSource, ipDestino, cmd, data = self.__getTrama(trama_rx)
|
|
|
|
|
|
return ipSource, ipDestino, cmd, data
|
|
|
|
|
|
def sendData(self, cmd, data):
|
|
|
|
|
|
if self.portDestino == 7000:
|
|
|
trama_tx = self.__HEADER + ":" + str(self.ipSource) + ":" + str(self.ipDestino) + ":" + str(cmd) + ":" + str(data) + ":"
|
|
|
else:
|
|
|
trama_tx = data
|
|
|
|
|
|
if self.mode == "client":
|
|
|
destiny = self.addr
|
|
|
else:
|
|
|
destiny = self.answer
|
|
|
# Send messages
|
|
|
if(self.socket_c.sendto(trama_tx, destiny)):
|
|
|
print "Sending message:[" + trama_tx + "] to " + str(destiny)
|
|
|
|
|
|
def __getTrama(self, trama):
|
|
|
|
|
|
FrameList = trama.split(':')
|
|
|
|
|
|
header = FrameList[0]
|
|
|
ipSource = FrameList[1]
|
|
|
ipDestino = FrameList[2]
|
|
|
cmd = FrameList[3]
|
|
|
data = FrameList[4]
|
|
|
trash = FrameList[5]
|
|
|
|
|
|
return ipSource, ipDestino, cmd, data
|
|
|
|
|
|
class TCPComm:
|
|
|
|
|
|
__HEADER = "JRO"
|
|
|
__TYPE = "ABS"
|
|
|
|
|
|
def __init__(self, ipSource, ipDestino, portDestino, asServer=False):
|
|
|
|
|
|
self.ipSource = ipSource
|
|
|
self.ipDestino = ipDestino
|
|
|
self.portDestino = portDestino
|
|
|
self.addr = (ipDestino,portDestino)
|
|
|
|
|
|
self.sc = "none"
|
|
|
self.answer = "none" #test
|
|
|
self.mode = "none"
|
|
|
self.len = 0
|
|
|
self.crc = 0
|
|
|
|
|
|
self.openSocket(asServer)
|
|
|
|
|
|
def openSocket(self, asServer):
|
|
|
|
|
|
#self.socket_c = socket.socket(AF_INET,SOCK_DGRAM)
|
|
|
# self.socket_c = socket.socket()
|
|
|
# self.socket_c.connect((self.ipDestino, self.portDestino))
|
|
|
|
|
|
if asServer:
|
|
|
self.socket_c = socket.socket()
|
|
|
# self.configAsServer()
|
|
|
self.socket_c.bind(self.addr)
|
|
|
self.mode = "server"
|
|
|
else:
|
|
|
# self.configAsClient()
|
|
|
self.mode = "client" #Socket is opened at the sendData function
|
|
|
|
|
|
# def configAsClient(self):
|
|
|
#Buscar broadcast TCP
|
|
|
# self.socket_c.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
|
|
|
# self.socket_c.connect(self.addr)
|
|
|
# pass
|
|
|
|
|
|
# def configAsServer(self):
|
|
|
#
|
|
|
# self.socket_c.bind(self.addr)
|
|
|
|
|
|
def waitData(self, nbytes = 1024):
|
|
|
|
|
|
print "\nWaiting some client."
|
|
|
|
|
|
if self.mode == "client":
|
|
|
# Short data through ethernet
|
|
|
trama_rx = self.socket_c.recv(nbytes)
|
|
|
else:
|
|
|
self.socket_c.listen(1)
|
|
|
sc, addr = self.socket_c.accept()
|
|
|
self.sc = sc
|
|
|
self.answer = addr
|
|
|
# Big data through ethernet
|
|
|
trama_rx = ""
|
|
|
while True:
|
|
|
tmp = self.sc.recv(nbytes)
|
|
|
trama_rx = trama_rx + tmp
|
|
|
if trama_rx[-4:] == "quit":
|
|
|
break
|
|
|
|
|
|
print "\nThis socket has received some data."
|
|
|
|
|
|
ipSource, ipDestino, cmd, data = self.__getTrama(trama_rx)
|
|
|
|
|
|
return ipSource, ipDestino, cmd, data
|
|
|
|
|
|
def sendData(self, cmd, data, id):
|
|
|
|
|
|
trama_tx = self.__HEADER + ":" + self.__TYPE + ":" + str(self.ipSource) + ":" + str(self.ipDestino) + ":" + str(self.len) + ":" + str(cmd) + ":" + str(data) + ":" + str(self.crc)
|
|
|
|
|
|
if self.portDestino == 7000:
|
|
|
trama_tx = trama_tx + ":" + "quit"
|
|
|
# Send messages
|
|
|
if self.mode == "client":
|
|
|
host = "192.168.1." + str(id)
|
|
|
self.socket_c.connect((host, self.portDestino))
|
|
|
self.socket_c.send(trama_tx)
|
|
|
else:
|
|
|
self.sc.send(trama_tx)
|
|
|
print "Sending message:[" + trama_tx + "]"
|
|
|
|
|
|
def __getTrama(self, trama):
|
|
|
|
|
|
FrameList = trama.split(':')
|
|
|
|
|
|
header = FrameList[0]
|
|
|
TypeOfInstrument = FrameList[1]
|
|
|
ipSource = FrameList[2]
|
|
|
ipDestino = FrameList[3]
|
|
|
len = FrameList[4]
|
|
|
cmd = FrameList[5]
|
|
|
data = FrameList[6]
|
|
|
crc = FrameList[7]
|
|
|
trash = FrameList[8]
|
|
|
|
|
|
return ipSource, ipDestino, cmd, data
|
|
|
|
|
|
def close_socket(self):
|
|
|
self.socket_c.close()
|
|
|
|
|
|
def open_socket(self):
|
|
|
self.socket_c = socket.socket()
|
|
|
|
|
|
#class FTPComm:
|
|
|
#
|
|
|
# ftp_servidor = 'ftp.servidor.com'
|
|
|
# ftp_usuario = 'miusuario'
|
|
|
# ftp_clave = 'miclave'
|
|
|
# ftp_raiz = '/public_html'
|
|
|
#
|
|
|
# def __init__(self):
|
|
|
#
|
|
|
# self.client = tftpy.TftpClient(self.ftp_servidor, '69')
|
|
|
#
|
|
|
#
|
|
|
# def sendFile(self, filename):
|
|
|
#
|
|
|
# self.client.upload(filename)
|
|
|
#
|
|
|
#if __name__ == '__main__':
|
|
|
#
|
|
|
# obj = FTPComm()
|
|
|
|
|
|
|
|
|
|
