Main server file.
This commit is contained in:
parent
e814174d7b
commit
35f574d160
3 changed files with 17 additions and 166 deletions
|
@ -1,115 +0,0 @@
|
||||||
#!/usr/bin/env python
|
|
||||||
|
|
||||||
import sys, logging, os
|
|
||||||
from optparse import OptionParser
|
|
||||||
import tftpy
|
|
||||||
|
|
||||||
def main():
|
|
||||||
usage=""
|
|
||||||
parser = OptionParser(usage=usage)
|
|
||||||
parser.add_option('-H',
|
|
||||||
'--host',
|
|
||||||
help='remote host or ip address')
|
|
||||||
parser.add_option('-p',
|
|
||||||
'--port',
|
|
||||||
help='remote port to use (default: 69)',
|
|
||||||
default=69)
|
|
||||||
parser.add_option('-f',
|
|
||||||
'--filename',
|
|
||||||
help='filename to fetch (deprecated, use download)')
|
|
||||||
parser.add_option('-D',
|
|
||||||
'--download',
|
|
||||||
help='filename to download')
|
|
||||||
parser.add_option('-u',
|
|
||||||
'--upload',
|
|
||||||
help='filename to upload')
|
|
||||||
parser.add_option('-b',
|
|
||||||
'--blksize',
|
|
||||||
help='udp packet size to use (default: 512)')
|
|
||||||
parser.add_option('-o',
|
|
||||||
'--output',
|
|
||||||
help='output file, - for stdout (default: same as download)')
|
|
||||||
parser.add_option('-i',
|
|
||||||
'--input',
|
|
||||||
help='input file, - for stdin (default: same as upload)')
|
|
||||||
parser.add_option('-d',
|
|
||||||
'--debug',
|
|
||||||
action='store_true',
|
|
||||||
default=False,
|
|
||||||
help='upgrade logging from info to debug')
|
|
||||||
parser.add_option('-q',
|
|
||||||
'--quiet',
|
|
||||||
action='store_true',
|
|
||||||
default=False,
|
|
||||||
help="downgrade logging from info to warning")
|
|
||||||
parser.add_option('-t',
|
|
||||||
'--tsize',
|
|
||||||
action='store_true',
|
|
||||||
default=False,
|
|
||||||
help="ask client to send tsize option in download")
|
|
||||||
options, args = parser.parse_args()
|
|
||||||
# Handle legacy --filename argument.
|
|
||||||
if options.filename:
|
|
||||||
options.download = options.filename
|
|
||||||
if not options.host or (not options.download and not options.upload):
|
|
||||||
sys.stderr.write("Both the --host and --filename options "
|
|
||||||
"are required.\n")
|
|
||||||
parser.print_help()
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
if options.debug and options.quiet:
|
|
||||||
sys.stderr.write("The --debug and --quiet options are "
|
|
||||||
"mutually exclusive.\n")
|
|
||||||
parser.print_help()
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
class Progress(object):
|
|
||||||
def __init__(self, out):
|
|
||||||
self.progress = 0
|
|
||||||
self.out = out
|
|
||||||
def progresshook(self, pkt):
|
|
||||||
if isinstance(pkt, tftpy.TftpPacketDAT):
|
|
||||||
self.progress += len(pkt.data)
|
|
||||||
self.out("Transferred %d bytes" % self.progress)
|
|
||||||
elif isinstance(pkt, tftpy.TftpPacketOACK):
|
|
||||||
self.out("Received OACK, options are: %s" % pkt.options)
|
|
||||||
|
|
||||||
if options.debug:
|
|
||||||
tftpy.setLogLevel(logging.DEBUG)
|
|
||||||
elif options.quiet:
|
|
||||||
tftpy.setLogLevel(logging.WARNING)
|
|
||||||
else:
|
|
||||||
tftpy.setLogLevel(logging.INFO)
|
|
||||||
|
|
||||||
progresshook = Progress(tftpy.log.info).progresshook
|
|
||||||
|
|
||||||
tftp_options = {}
|
|
||||||
if options.blksize:
|
|
||||||
tftp_options['blksize'] = int(options.blksize)
|
|
||||||
if options.tsize:
|
|
||||||
tftp_options['tsize'] = 0
|
|
||||||
|
|
||||||
tclient = tftpy.TftpClient(options.host,
|
|
||||||
int(options.port),
|
|
||||||
tftp_options)
|
|
||||||
try:
|
|
||||||
if options.download:
|
|
||||||
if not options.output:
|
|
||||||
options.output = os.path.basename(options.download)
|
|
||||||
tclient.download(options.download,
|
|
||||||
options.output,
|
|
||||||
progresshook)
|
|
||||||
elif options.upload:
|
|
||||||
if not options.input:
|
|
||||||
options.input = os.path.basename(options.upload)
|
|
||||||
tclient.upload(options.upload,
|
|
||||||
options.input,
|
|
||||||
progresshook)
|
|
||||||
except tftpy.TftpException, err:
|
|
||||||
sys.stderr.write("%s\n" % str(err))
|
|
||||||
sys.exit(1)
|
|
||||||
except KeyboardInterrupt:
|
|
||||||
pass
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
main()
|
|
|
@ -1,51 +0,0 @@
|
||||||
#!/usr/bin/env python
|
|
||||||
|
|
||||||
import sys, logging
|
|
||||||
from optparse import OptionParser
|
|
||||||
import tftpy
|
|
||||||
|
|
||||||
def main():
|
|
||||||
usage=""
|
|
||||||
parser = OptionParser(usage=usage)
|
|
||||||
parser.add_option('-i',
|
|
||||||
'--ip',
|
|
||||||
type='string',
|
|
||||||
help='ip address to bind to (default: INADDR_ANY)',
|
|
||||||
default="")
|
|
||||||
parser.add_option('-p',
|
|
||||||
'--port',
|
|
||||||
type='int',
|
|
||||||
help='local port to use (default: 69)',
|
|
||||||
default=69)
|
|
||||||
parser.add_option('-r',
|
|
||||||
'--root',
|
|
||||||
type='string',
|
|
||||||
help='path to serve from',
|
|
||||||
default=None)
|
|
||||||
parser.add_option('-d',
|
|
||||||
'--debug',
|
|
||||||
action='store_true',
|
|
||||||
default=False,
|
|
||||||
help='upgrade logging from info to debug')
|
|
||||||
options, args = parser.parse_args()
|
|
||||||
|
|
||||||
if options.debug:
|
|
||||||
tftpy.setLogLevel(logging.DEBUG)
|
|
||||||
else:
|
|
||||||
tftpy.setLogLevel(logging.INFO)
|
|
||||||
|
|
||||||
if not options.root:
|
|
||||||
parser.print_help()
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
server = tftpy.TftpServer(options.root)
|
|
||||||
try:
|
|
||||||
server.listen(options.ip, options.port)
|
|
||||||
except tftpy.TftpException, err:
|
|
||||||
sys.stderr.write("%s\n" % str(err))
|
|
||||||
sys.exit(1)
|
|
||||||
except KeyboardInterrupt:
|
|
||||||
pass
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
main()
|
|
17
run.py
Executable file
17
run.py
Executable file
|
@ -0,0 +1,17 @@
|
||||||
|
#!/usr/bin/env python
|
||||||
|
|
||||||
|
import sys, logging
|
||||||
|
import tftpy
|
||||||
|
|
||||||
|
def main():
|
||||||
|
server = tftpy.TftpServer("/var/khepri/tftp-root/")
|
||||||
|
try:
|
||||||
|
server.listen("10.8.1.1", 5000)
|
||||||
|
except tftpy.TftpException, err:
|
||||||
|
sys.stderr.write("%s\n" % str(err))
|
||||||
|
sys.exit(1)
|
||||||
|
except KeyboardInterrupt:
|
||||||
|
pass
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|
Reference in a new issue