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