##
# This file is part of the Metasploit Framework and may be subject to
# redistribution and commercial restrictions. Please see the Metasploit
# web site for more information on licensing and terms of use.
#   http://metasploit.com/
##

require 'msf/core'
require 'rex/proto/tftp'

class Metasploit3 < Msf::Auxiliary

	include Msf::Exploit::Remote::TFTPServer
	include Msf::Auxiliary::Report

	def initialize
		super(
			'Name'        => 'TFTP File Server',
			'Description'    => %q{
				This module provides a TFTP service
			},
			'Author'      => [ 'jduck' ],
			'License'     => MSF_LICENSE,
			'Actions'     =>
				[
					[ 'Service' ]
				],
			'PassiveActions' =>
				[
					'Service'
				],
			'DefaultAction'  => 'Service'
		)

		register_options(
			[
				OptString.new('TFTPROOT',   [ false,  "The TFTP root directory to serve files from" ]),
				OptString.new('OUTPUTPATH', [ false, "The directory in which uploaded files will be written." ])
			], self.class)
	end

	def run
		if not datastore['OUTPUTPATH'] and not datastore['TFTPROOT']
			print_error("You must set TFTPROOT and/or OUTPUTPATH to use this module.")
			return
		end

		@tftp = Rex::Proto::TFTP::Server.new

		print_status("Starting TFTP server...")

		if datastore['TFTPROOT']
			print_status("Files will be served from #{datastore['TFTPROOT']}")
			@tftp.set_tftproot(datastore['TFTPROOT'])
		end

		# register output directory
		if datastore['OUTPUTPATH']
			print_status("Uploaded files will be saved in #{datastore['OUTPUTPATH']}")
			@tftp.set_output_dir(datastore['OUTPUTPATH'])
		end

		# Individual virtual files can be served here -
		#@tftp.register_file("ays", "A" * 2048) # multiple of 512 on purpose

		@tftp.start
		add_socket(@tftp.sock)

		# Wait for finish..
		while @tftp.thread.alive?
			select(nil, nil, nil, 2)
		end

		vprint_status("Stopping TFTP server")
		@tftp.stop
	end

end