module Msf
module Serializer

###
#
# Text
# ----
#
# This class formats information in a plain-text format that
# is meant to be displayed on a console or some other non-GUI
# medium.
#
###
class ReadableText

	DefaultColumnWrap = 60
	DefaultIndent     = 4

	#
	# Returns a formatted string that contains information about
	# the supplied module instance.
	#
	def self.dump_module(mod, indent = "    ")
		case mod.type
			when MODULE_PAYLOAD
				return dump_payload_module(mod, indent)
			when MODULE_NOP
				return dump_basic_module(mod, indent)
			when MODULE_ENCODER
				return dump_basic_module(mod, indent)
			when MODULE_EXPLOIT
				return dump_exploit_module(mod, indent)
			else
				return dump_generic_module(mod, indent)
		end
	end

	#
	# Dumps information about an exploit module.
	#
	def self.dump_exploit_module(mod, indent = '')
		output  = "\n"
		output += "       Name: #{mod.name}\n"
		output += "    Version: #{mod.version}\n"
		output += "   Platform: #{mod.platform_to_s}\n"
		output += " Privileged: " + (mod.privileged? ? "Yes" : "No") + "\n"
		output += "\n"

		# Authors
		output += "Provided by:\n"
		mod.each_author { |author|
			output += indent + author.to_s + "\n"
		}
		output += "\n"

		# Targets
		tbl = Rex::Ui::Text::Table.new(
			'Indent'  => indent.length,
			'Columns' =>
				[
					'Id', 
					'Name',
				])

		output += "Available targets:\n"
		mod.targets.each_with_index { |target, idx|
			tbl << [ idx.to_s, target.name || 'All' ]	
		}
		output += tbl.to_s 
		output += "\n"

		# Options
		if (mod.options.has_options?)
			output += "Available options:\n"
			output += dump_options(mod)
			output += "\n"
		end

		# Advanced options
		if (mod.options.has_advanced_options?)
			output += "Advanced options:\n"
			output += dump_advanced_options(mod)
			output += "\n"
		end

		# Payload information
		if (mod.payload_info.length)
			output += "Payload information:\n"
			if (mod.payload_space)
				output += indent + "Space: " + mod.payload_space.to_s + "\n"
			end
			if (mod.payload_badchars)
				output += indent + "Avoid: " + mod.payload_badchars.length.to_s + " characters\n"
			end
			output += "\n"
		end
	
		# Description
		output += "Description:\n"
		output += word_wrap(mod.description)
		output += "\n"

		# References
		if (mod.references.length > 0)
			output += "References:\n"
			mod.references.each { |ref|
				output += indent + ref.to_s + "\n"
			}
			output += "\n"
		end
	
		return output

	end

	# 
	# Dumps information about a payload module.
	#
	def self.dump_payload_module(mod, indent = '')
		# General
		output  = "\n"
		output += "       Name: #{mod.name}\n"
		output += "    Version: #{mod.version}\n"
		output += "   Platform: #{mod.platform_to_s}\n"
		output += "       Arch: #{mod.arch.to_s}\n"
		output += "Needs Admin: " + (mod.privileged? ? "Yes" : "No") + "\n"
		output += " Total size: #{mod.size}\n"
		output += "\n"

		# Authors
		output += "Provided by:\n"
		mod.each_author { |author|
			output += indent + author.to_s + "\n"
		}
		output += "\n"

		# Options
		if (mod.options.has_options?)
			output += "Available options:\n"
			output += dump_options(mod)
			output += "\n"
		end

		# Advanced options
		if (mod.options.has_advanced_options?)
			output += "Advanced options:\n"
			output += dump_advanced_options(mod)
			output += "\n"
		end
	
		# Description
		output += "Description:\n"
		output += word_wrap(mod.description)
		output += "\n\n"
	
		return output
	end

	#
	# Dumps information about a module, just the basics.
	#
	def self.dump_basic_module(mod, indent = '')
		# General
		output  = "\n"
		output += "       Name: #{mod.name}\n"
		output += "    Version: #{mod.version}\n"
		output += "   Platform: #{mod.platform_to_s}\n"
		output += "       Arch: #{mod.arch.to_s}\n"
		output += "\n"

		# Authors
		output += "Provided by:\n"
		mod.each_author { |author|
			output += indent + author.to_s + "\n"
		}
		output += "\n"

		# Advanced options
		if (mod.options.has_advanced_options?)
			output += "Advanced options:\n"
			output += dump_advanced_options(mod)
			output += "\n"
		end

		# Description
		output += "Description:\n"
		output += word_wrap(mod.description)
		output += "\n\n"
	
		return output

	end

	def self.dump_generic_module(mod, indent = '')
	end

	#
	# Dumps the list of options associated with the
	# supplied module.
	#
	def self.dump_options(mod, indent = DefaultIndent)
		tbl = Rex::Ui::Text::Table.new(
			'Indent'  => indent,
			'Columns' =>
				[
					'Name', 
					'Default',
					'Required',
					'Description'
				])

		mod.options.sorted.each { |entry|
			name, opt = entry

			next if (opt.advanced?)

			val = mod.datastore[name] || opt.default || ''

			tbl << [ name, val.to_s, opt.required? ? "yes" : "no", opt.desc ]
		}

		return tbl.to_s
	end

	def self.dump_advanced_options(mod, indent = DefaultIndent)
		output = ''
		pad    = ' ' * indent

		mod.options.sorted.each { |entry|
			name, opt = entry

			next if (!opt.advanced?)

			val = mod.datastore[name] || opt.default || ''

			output += pad + "Name   : #{name}\n"
			output += pad + "Default: #{val}\n\n"
			output += word_wrap(opt.desc, indent + 3)
		}

		return output
	end

	#
	# Dumps the contents of a datastore
	#
	def self.dump_datastore(name, ds, indent = DefaultIndent, col = DefaultColumnWrap)
		tbl = Rex::Ui::Text::Table.new(
			'Indent'  => indent,
			'Header'  => name,
			'Columns' =>
				[
					'Name', 
					'Value'
				])

		ds.keys.sort.each { |k|
			tbl << [ k, ds[k] ]
		}

		return ds.length > 0 ? tbl.to_s : "#{tbl.header_to_s}No entries in data store.\n"
	end

	#
	# Dumps the list of active sessions
	#
	def self.dump_sessions(framework, indent = DefaultIndent, col = DefaultColumnWrap)
		tbl = Rex::Ui::Text::Table.new(
			'Indent'  => indent,
			'Header'  => "Active sessions",
			'Columns' =>
				[
					'Id',
					'Description',
					'Tunnel'
				])

		framework.sessions.each_sorted { |k|
			session = framework.sessions[k]

			tbl << [ session.sid.to_s, session.desc, session.tunnel_to_s ]
		}

		return framework.sessions.length > 0 ? tbl.to_s : "#{tbl.header_to_s}No active sessions.\n"
	end

	#
	# Dumps the list of running jobs
	#
	def self.dump_jobs(framework, indent = DefaultIndent, col = DefaultColumnWrap)
		tbl = Rex::Ui::Text::Table.new(
			'Indent'  => indent,
			'Header'  => "Jobs",
			'Columns' =>
				[
					'Name'
				])

		framework.jobs.keys.sort.each { |k|
			tbl << [ k ]
		}

		return framework.jobs.keys.length > 0 ? tbl.to_s : "#{tbl.header_to_s}No active jobs.\n"
	end

	#
	# Jacked from Ernest Ellingson <erne [at] powernav.com>, modified
	# a bit to add indention
	#
	def self.word_wrap(str, indent = 4, col = DefaultColumnWrap)
		return Rex::Text.wordwrap(str, indent, col)
	end

end

end end