94 lines
4.4 KiB
Python
94 lines
4.4 KiB
Python
import time, types
|
|
from TftpShared import *
|
|
from TftpPacketFactory import *
|
|
from TftpStates import TftpContextClientDownload, TftpContextClientUpload
|
|
|
|
class TftpClient(TftpSession):
|
|
"""This class is an implementation of a tftp client. Once instantiated, a
|
|
download can be initiated via the download() method."""
|
|
def __init__(self, host, port, options={}):
|
|
"""This constructor returns an instance of TftpClient, taking the
|
|
remote host, the remote port, and the filename to fetch."""
|
|
TftpSession.__init__(self)
|
|
self.context = None
|
|
self.host = host
|
|
self.iport = port
|
|
self.filename = None
|
|
self.options = options
|
|
# FIXME: If the blksize is DEF_BLKSIZE, we should just skip sending
|
|
# it.
|
|
if self.options.has_key('blksize'):
|
|
size = self.options['blksize']
|
|
tftpassert(types.IntType == type(size), "blksize must be an int")
|
|
if size < MIN_BLKSIZE or size > MAX_BLKSIZE:
|
|
raise TftpException, "Invalid blksize: %d" % size
|
|
else:
|
|
self.options['blksize'] = DEF_BLKSIZE
|
|
|
|
def download(self, filename, output, packethook=None, timeout=SOCK_TIMEOUT):
|
|
"""This method initiates a tftp download from the configured remote
|
|
host, requesting the filename passed. It saves the file to a local
|
|
file specified in the output parameter. If a packethook is provided,
|
|
it must be a function that takes a single parameter, which will be a
|
|
copy of each DAT packet received in the form of a TftpPacketDAT
|
|
object. The timeout parameter may be used to override the default
|
|
SOCK_TIMEOUT setting, which is the amount of time that the client will
|
|
wait for a receive packet to arrive."""
|
|
# We're downloading.
|
|
log.debug("Creating download context with the following params:")
|
|
log.debug("host = %s, port = %s, filename = %s, output = %s"
|
|
% (self.host, self.iport, filename, output))
|
|
log.debug("options = %s, packethook = %s, timeout = %s"
|
|
% (self.options, packethook, timeout))
|
|
self.context = TftpContextClientDownload(self.host,
|
|
self.iport,
|
|
filename,
|
|
output,
|
|
self.options,
|
|
packethook,
|
|
timeout)
|
|
self.context.start()
|
|
# Download happens here
|
|
self.context.end()
|
|
|
|
metrics = self.context.metrics
|
|
|
|
# FIXME: Should we output this? Shouldn't we let the client control
|
|
# output? This should be in the sample client, but not in the download
|
|
# call.
|
|
if metrics.duration == 0:
|
|
log.info("Duration too short, rate undetermined")
|
|
else:
|
|
log.info('')
|
|
log.info("Downloaded %.2f bytes in %.2f seconds" % (metrics.bytes, metrics.duration))
|
|
log.info("Average rate: %.2f kbps" % metrics.kbps)
|
|
log.info("Received %d duplicate packets" % metrics.dupcount)
|
|
|
|
def upload(self, filename, input, packethook=None, timeout=SOCK_TIMEOUT):
|
|
# Open the input file.
|
|
# FIXME: As of the state machine, this is now broken. Need to
|
|
# implement with new state machine.
|
|
self.context = TftpContextClientUpload(self.host,
|
|
self.iport,
|
|
filename,
|
|
input,
|
|
self.options,
|
|
packethook,
|
|
timeout)
|
|
self.context.start()
|
|
# Upload happens here
|
|
self.context.end()
|
|
|
|
metrics = self.context.metrics
|
|
|
|
# FIXME: Should we output this? Shouldn't we let the client control
|
|
# output? This should be in the sample client, but not in the download
|
|
# call.
|
|
if metrics.duration == 0:
|
|
log.info("Duration too short, rate undetermined")
|
|
else:
|
|
log.info('')
|
|
log.info("Downloaded %.2f bytes in %.2f seconds" % (metrics.bytes, metrics.duration))
|
|
log.info("Average rate: %.2f kbps" % metrics.kbps)
|
|
log.info("Received %d duplicate packets" % metrics.dupcount)
|