def setRemoteNetwork(service, ip, port):
# set default service #
if not service:
service = 'PRINT'
errMsg = ("wrong value for remote server, "
"input in the format [%s]@IP:PORT") % \
'|'.join(ThreadAnalyzer.requestType)
if not ip or not SysMgr.isEffectiveRequest(service):
SysMgr.printErr(errMsg)
sys.exit(0)
if not port:
port = SysMgr.defaultPort
networkObject = NetworkMgr('client', ip, port)
if not networkObject.ip:
sys.exit(0)
else:
networkObject.status = 'ALWAYS'
networkObject.request = service
naddr = '%s:%s' % (ip, str(port))
if service == 'PRINT':
SysMgr.addrListForPrint[naddr] = networkObject
elif service.startswith('REPORT_'):
SysMgr.reportEnable = True
SysMgr.addrListForReport[naddr] = networkObject
else:
SysMgr.printErr(errMsg)
SysMgr.printInfo(\
"use %s:%d as remote address to request %s" % \
(ip, port, service))
def setRemoteServer(value, tcp=False):
# receive mode #
if value and len(value) == 0:
SysMgr.remoteServObj = 'NONE'
return
# request mode #
service, ip, port = NetworkMgr.parseAddr(value)
# set PRINT as default #
if not service:
service = 'PRINT'
if not ip:
ip = NetworkMgr.getPublicIp()
if not port:
port = SysMgr.defaultPort
# check server addresses #
if SysMgr.localServObj and \
SysMgr.localServObj.ip == ip and \
SysMgr.localServObj.port == port:
SysMgr.printErr((\
"wrong option value with -X, "
"local address and remote address are same "
"with %s:%s") % (ip, port))
sys.exit(0)
if not ip or not port or \
not SysMgr.isEffectiveRequest(service):
reqList = ''
for req in ThreadAnalyzer.requestType:
reqList += req + '|'
SysMgr.printErr(\
("wrong option value with -X, "
"input [%s]@IP:PORT as remote address") % \
reqList[:-1])
sys.exit(0)
# create a socket #
networkObject = NetworkMgr('client', ip, port, tcp=tcp)
if not networkObject.ip:
sys.exit(0)
else:
networkObject.request = service
SysMgr.remoteServObj = networkObject
if tcp:
proto = 'TCP'
else:
proto = 'UDP'
SysMgr.printInfo(\
"use %s:%d(%s) as remote address" % (ip, port, proto))
def getData(self):
try:
data = b''
# receive and composite packets #
while 1:
output = self.recvfrom(noTimeout=True)
# handle timeout #
if not output:
continue
# get only data #
output = output[0]
# composite packets #
data = data + output
if len(output) == 0:
break
# decode data #
try:
output = output.decode()
except:
pass
if len(output) < self.recvSize and \
output[-1] == '\n':
break
except SystemExit:
sys.exit(0)
except:
err = SysMgr.getErrReason()
SysMgr.printErr(\
"Fail to get data from %s:%d as client because %s" % \
(self.ip, self.port, err))
return None
# decode data #
try:
retstr = data.decode()
return retstr
except:
return data
def sendto(self, message, ip, port):
if not ip or not port:
SysMgr.printErr(\
"Fail to use IP address for client because it is not set")
return False
elif not self.socket:
SysMgr.printErr(\
"Fail to use socket for client because it is not set")
return False
# encode message #
if UtilMgr.isString(message):
message = UtilMgr.encodeStr(message)
try:
self.socket.sendto(message, (ip, port))
return True
except SystemExit:
sys.exit(0)
except:
err = SysMgr.getErrReason()
SysMgr.printErr(\
"Fail to send data to %s:%d as client because %s" % \
(self.ip, self.port, err))
return False
def recvfrom(self, size=0, noTimeout=False, verbose=True):
if self.ip is None or self.port is None:
SysMgr.printErr(\
"Fail to use IP address for server because it is not set")
return False
elif not self.socket:
SysMgr.printErr(\
"Fail to use socket for client because it is not set")
return False
# get socket object #
socket = SysMgr.getPkg('socket', False)
# set recv size #
if size == 0:
size = self.recvSize
while 1:
try:
message, address = self.socket.recvfrom(size)
return (message, address)
except socket.timeout:
if noTimeout:
continue
SysMgr.printWarn(\
"Fail to receive data from %s:%d as client because %s" % \
(self.ip, self.port, 'timeout'))
return None
except KeyboardInterrupt:
sys.exit(0)
except SystemExit:
sys.exit(0)
except:
if verbose:
SysMgr.printWarn(\
"Fail to receive data from %s:%d as client because %s" % \
(self.ip, self.port, SysMgr.getErrReason()))
return None
def recv(self, size=0):
if self.ip is None or self.port is None:
SysMgr.printErr(\
"Fail to use IP address for server because it is not set")
return False
elif not self.socket:
SysMgr.printErr(\
"Fail to use socket for client because it is not set")
return False
# set recv size #
if size == 0:
size = self.recvSize
try:
return self.socket.recv(size)
except SystemExit:
sys.exit(0)
except:
err = SysMgr.getErrReason()
SysMgr.printWarn(\
"Fail to receive data from %s:%d as client because %s" % \
(self.ip, self.port, err))
return False
def send(self, message, write=False):
if self.ip is None or self.port is None:
SysMgr.printErr(\
"Fail to use IP address for client because it is not set")
return False
elif not self.socket:
SysMgr.printErr(\
"Fail to use socket for client because it is not set")
return False
# encode message #
if UtilMgr.isString(message):
message = UtilMgr.encodeStr(message)
try:
# check protocol #
if self.tcp:
ret = self.socket.send(message)
elif not write and SysMgr.localServObj:
ret = SysMgr.localServObj.socket.sendto(\
message, (self.ip, self.port))
else:
ret = self.socket.sendto(message, (self.ip, self.port))
if ret < 0:
raise Exception()
if self.status != 'ALWAYS':
self.status = 'SENT'
return True
except SystemExit:
sys.exit(0)
except:
err = SysMgr.getErrReason()
SysMgr.printErr(\
"Fail to send data to %s:%d as server because %s" % \
(self.ip, self.port, err))
return False
def printErr():
SysMgr.printErr(\
"No running server or wrong server address")
def getServerConn():
def printErr():
SysMgr.printErr(\
"No running server or wrong server address")
# set server address in local #
if SysMgr.isLinux and not SysMgr.remoteServObj:
try:
addr = SysMgr.getProcAddrs(__module__)
except:
addr = None
if not addr:
return None
# classify ip and port #
service, ip, port = NetworkMgr.parseAddr(addr)
if service == ip == port == None:
printErr()
return None
else:
NetworkMgr.setRemoteServer(addr, tcp=True)
# set server address again #
elif SysMgr.remoteServObj:
servObj = SysMgr.remoteServObj
ip = servObj.ip
port = servObj.port
NetworkMgr.setRemoteServer('%s:%s' % (ip, port), tcp=True)
# check server address #
if not SysMgr.remoteServObj:
printErr()
return None
# bind local socket for UDP #
try:
if not SysMgr.remoteServObj.tcp and \
SysMgr.localServObj:
lip = SysMgr.localServObj.ip
lport = SysMgr.localServObj.port
SysMgr.remoteServObj.socket.bind((lip, lport))
except:
err = SysMgr.getErrReason()
SysMgr.printErr(\
"Fail to bind socket to %s:%s for connection because %s" % \
(lip, lport, err))
# do connect to server #
try:
connObj = SysMgr.remoteServObj
connObj.timeout()
# connect with handling CLOSE_WAIT #
while 1:
try:
connObj.connect()
break
except:
err = SysMgr.getErrReason()
SysMgr.printWarn(\
"Fail to connect to %s:%s because %s" % \
(ip, port, err))
if err.startswith('99'):
time.sleep(0.1)
continue
break
return connObj
except:
err = SysMgr.getErrReason()
SysMgr.printErr(\
"Fail to set socket for connection because %s" % err)
return None
def onUpload(req):
# parse path #
plist = req.split('|', 1)[1]
path = plist.split(',')
origPath = path[0].strip()
targetPath = path[1].strip()
sender = self
targetIp = self.ip
targetPort = self.port
addr = '%s:%s' % (targetIp, targetPort)
# check file #
if not os.path.isfile(origPath):
SysMgr.printErr(\
'Failed to find %s to transfer' % origPath)
return
convert = UtilMgr.convertSize2Unit
try:
# receive file size #
stat = os.stat(origPath)
st_size = '%s' % stat.st_size
sender.send(st_size)
# read for ACK #
while 1:
ret = sender.recv(3)
if ret is None:
continue
elif ret is False:
sys.exit(0)
else:
break
# transfer file #
curSize = long(0)
totalSize = long(st_size)
with open(origPath, 'rb') as fd:
buf = fd.read(sender.sendSize)
while buf:
# print progress #
UtilMgr.printProgress(curSize, totalSize)
ret = sender.send(buf)
if not ret:
raise Exception()
else:
curSize = len(buf)
buf = fd.read(sender.sendSize)
UtilMgr.deleteProgress()
SysMgr.printInfo(\
"%s [%s] is uploaded to %s:%s successfully\n" % \
(origPath, convert(os.path.getsize(origPath)), \
addr, targetPath))
except:
err = SysMgr.getErrReason()
SysMgr.printErr(\
"Fail to upload %s to %s:%s because %s" % \
(origPath, addr, targetPath, err))
finally:
sender.close()
def onDownload(req):
# parse path #
plist = req.split('|', 1)[1]
path = plist.split(',')
origPath = path[0].strip()
targetPath = path[1].strip()
receiver = self
targetIp = self.ip
targetPort = self.port
# get select object #
selectObj = SysMgr.getPkg('select')
# receive file #
try:
curSize = long(0)
totalSize = None
dirPos = targetPath.rfind('/')
if dirPos >= 0 and \
not os.path.isdir(targetPath[:dirPos]):
os.makedirs(targetPath[:dirPos])
# receive file size #
while 1:
size = receiver.recv(receiver.recvSize)
if not size:
continue
else:
totalSize = long(size.decode())
receiver.send('ACK'.encode())
break
# receive file #
with open(targetPath, 'wb') as fd:
while 1:
selectObj.select([receiver.socket], [], [], 3)
buf = receiver.recv(receiver.recvSize)
if buf:
curSize += len(buf)
fd.write(buf)
else:
break
# print progress #
UtilMgr.printProgress(curSize, totalSize)
UtilMgr.deleteProgress()
SysMgr.printInfo(\
"%s [%s] is downloaded from %s:%s:%s successfully\n" % \
(targetPath, \
UtilMgr.convertSize2Unit(os.path.getsize(targetPath)),
targetIp, targetPort, origPath))
except:
err = SysMgr.getErrReason()
SysMgr.printErr(\
'Fail to download %s from %s:%s:%s because %s' % \
(origPath, targetIp, targetPort, targetPath, err))
finally:
receiver.close()
def handleServerRequest(self, req, onlySocket=False):
def onDownload(req):
# parse path #
plist = req.split('|', 1)[1]
path = plist.split(',')
origPath = path[0].strip()
targetPath = path[1].strip()
receiver = self
targetIp = self.ip
targetPort = self.port
# get select object #
selectObj = SysMgr.getPkg('select')
# receive file #
try:
curSize = long(0)
totalSize = None
dirPos = targetPath.rfind('/')
if dirPos >= 0 and \
not os.path.isdir(targetPath[:dirPos]):
os.makedirs(targetPath[:dirPos])
# receive file size #
while 1:
size = receiver.recv(receiver.recvSize)
if not size:
continue
else:
totalSize = long(size.decode())
receiver.send('ACK'.encode())
break
# receive file #
with open(targetPath, 'wb') as fd:
while 1:
selectObj.select([receiver.socket], [], [], 3)
buf = receiver.recv(receiver.recvSize)
if buf:
curSize += len(buf)
fd.write(buf)
else:
break
# print progress #
UtilMgr.printProgress(curSize, totalSize)
UtilMgr.deleteProgress()
SysMgr.printInfo(\
"%s [%s] is downloaded from %s:%s:%s successfully\n" % \
(targetPath, \
UtilMgr.convertSize2Unit(os.path.getsize(targetPath)),
targetIp, targetPort, origPath))
except:
err = SysMgr.getErrReason()
SysMgr.printErr(\
'Fail to download %s from %s:%s:%s because %s' % \
(origPath, targetIp, targetPort, targetPath, err))
finally:
receiver.close()
def onUpload(req):
# parse path #
plist = req.split('|', 1)[1]
path = plist.split(',')
origPath = path[0].strip()
targetPath = path[1].strip()
sender = self
targetIp = self.ip
targetPort = self.port
addr = '%s:%s' % (targetIp, targetPort)
# check file #
if not os.path.isfile(origPath):
SysMgr.printErr(\
'Failed to find %s to transfer' % origPath)
return
convert = UtilMgr.convertSize2Unit
try:
# receive file size #
stat = os.stat(origPath)
st_size = '%s' % stat.st_size
sender.send(st_size)
# read for ACK #
while 1:
ret = sender.recv(3)
if ret is None:
continue
elif ret is False:
sys.exit(0)
else:
break
# transfer file #
curSize = long(0)
totalSize = long(st_size)
with open(origPath, 'rb') as fd:
buf = fd.read(sender.sendSize)
while buf:
# print progress #
UtilMgr.printProgress(curSize, totalSize)
ret = sender.send(buf)
if not ret:
raise Exception()
else:
curSize = len(buf)
buf = fd.read(sender.sendSize)
UtilMgr.deleteProgress()
SysMgr.printInfo(\
"%s [%s] is uploaded to %s:%s successfully\n" % \
(origPath, convert(os.path.getsize(origPath)), \
addr, targetPath))
except:
err = SysMgr.getErrReason()
SysMgr.printErr(\
"Fail to upload %s to %s:%s because %s" % \
(origPath, addr, targetPath, err))
finally:
sender.close()
def onRun(req, onlySocket):
# parse command #
origReq = req
command = req.split('|', 1)[1]
# parse addr #
addr = '%s:%s' % (self.ip, self.port)
if not onlySocket:
SysMgr.printInfo(\
"'%s' is executed from %s\n" % (command, addr))
# return just the connected socket #
if onlySocket:
return self
# get select object #
selectObj = SysMgr.getPkg('select')
print(oneLine)
# run mainloop #
isPrint = False
while 1:
try:
[readSock, writeSock, errorSock] = \
selectObj.select([self.socket], [], [])
# receive packet #
output = self.getData()
if not output:
break
print(output[:-1])
isPrint = True
except:
break
# print output from server #
if not isPrint:
print('No response')
print(oneLine)
# close connection #
try:
self.close()
except:
pass
# get select object to check #
SysMgr.getPkg('select')
# unmarshalling #
if type(req) is tuple:
try:
req = req[0].decode()
except:
req = req[0]
# handle request #
if not req:
return
elif req.upper().startswith('DOWNLOAD'):
return onDownload(req)
elif req.upper().startswith('UPLOAD'):
return onUpload(req)
elif req.upper().startswith('RUN'):
return onRun(req, onlySocket)
elif req.startswith('ERROR'):
err = req.split('|', 1)[1]
errMsg = err.split(':', 1)[0]
SysMgr.printErr(errMsg)
else:
SysMgr.printErr(\
"Fail to recognize '%s' request" % req)
elif not req:
SysMgr.printErr(\
"No response from server")
else:
SysMgr.printErr(\
"received wrong reply '%s'" % req)
def __init__(\
self, mode, ip, port, blocking=True, tcp=False, anyPort=False):
self.mode = mode
self.ip = None
self.port = None
self.socket = None
self.request = None
self.status = None
self.ignore = long(0)
self.fileno = -1
self.time = None
self.sendSize = 32767
self.recvSize = 32767
self.tcp = tcp
self.connected = False
# get socket object #
socket = SysMgr.getPkg('socket')
try:
from socket import socket, AF_INET, SOCK_DGRAM, \
SOCK_STREAM, SOL_SOCKET, SO_REUSEADDR, SO_SNDBUF, SO_RCVBUF, \
SOL_TCP, TCP_NODELAY, SO_RCVTIMEO, SO_SNDTIMEO
except:
return None
try:
# set socket type #
if tcp:
self.socket = socket(AF_INET, SOCK_STREAM)
else:
self.socket = socket(AF_INET, SOCK_DGRAM)
self.fileno = self.socket.fileno()
# increate socket buffer size to 1MB #
self.socket.setsockopt(SOL_SOCKET, SO_SNDBUF, 1 << 20)
self.socket.setsockopt(SOL_SOCKET, SO_RCVBUF, 1 << 20)
# get buffer size #
self.sendSize = self.socket.getsockopt(SOL_SOCKET, SO_SNDBUF)
self.recvSize = self.socket.getsockopt(SOL_SOCKET, SO_RCVBUF)
# set REUSEADDR #
self.socket.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
# set SENDTIMEOUT #
'''
sec = 1
usec = long(0)
timeval = struct.pack('ll', sec, usec)
self.socket.setsockopt(SOL_SOCKET, SO_SNDTIMEO, timeval)
'''
# set NODELAY #
'''
self.socket.setsockopt(SOL_TCP, TCP_NODELAY, 1)
'''
# set IP & PORT #
self.ip = ip
self.port = port
if mode == 'server':
# IP #
if not ip:
self.ip = '0.0.0.0'
# PORT #
if anyPort:
self.port = long(0)
elif not port:
self.port = SysMgr.defaultPort
# bind #
try:
self.socket.bind((self.ip, self.port))
except:
self.socket.bind((self.ip, self.port))
# get bind port #
self.port = self.socket.getsockname()[1]
if not blocking:
self.socket.setblocking(0)
except:
err = SysMgr.getErrReason()
if err.startswith('13') and \
not SysMgr.isRoot() and \
port < 1024:
feedback = ', use port bigger than 1024'
else:
feedback = ''
SysMgr.printErr(\
"Fail to create socket with %s:%s as server because %s%s" % \
(self.ip, self.port, err, feedback))
'''
if error "99 Cannot assign requested address" occurs:
add "net.ipv4.ip_nonlocal_bind = 1" in /etc/sysctl.conf
execute sysctl -p /etc/sysctl.conf
'''
self.ip = None
self.port = None
return None