librex 0.0.20 → 0.0.21

data/lib/rex/ui/text/table.rb

@@ -0,0 +1,279 @@
+require 'rex/ui'
+
+module Rex
+module Ui
+module Text
+
+###
+#
+# Prints text in a tablized format.  Pretty lame at the moment, but 
+# whatever.
+#
+###
+class Table
+
+	#
+	# Initializes a text table instance using the supplied properties.  The
+	# Table class supports the following hash attributes:
+	#
+	# Header 
+	#
+	#	The string to display as a heading above the table.  If none is
+	#	specified, no header will be displayed.
+	#
+	# HeaderIndent
+	#
+	# 	The amount of space to indent the header.  The default is zero.
+	#
+	# Columns
+	#
+	# 	The array of columns that will exist within the table.
+	#
+	# Rows
+	#
+	# 	The array of rows that will exist.
+	#
+	# Width
+	#
+	# 	The maximum width of the table in characters.
+	#
+	# Indent
+	#
+	# 	The number of characters to indent the table.
+	#
+	# CellPad
+	#
+	# 	The number of characters to put between each horizontal cell.
+	#
+	# Prefix
+	#
+	# 	The text to prefix before the table.
+	#
+	# Postfix
+	#
+	# 	The text to affix to the end of the table.
+	#
+	def initialize(opts = {})
+		self.header   = opts['Header']
+		self.headeri  = opts['HeaderIndent'] || 0
+		self.columns  = opts['Columns'] || []
+		# updated below if we got a "Rows" option
+		self.rows     = []
+
+		self.width    = opts['Width']   || 80
+		self.indent   = opts['Indent']  || 0
+		self.cellpad  = opts['CellPad'] || 2
+		self.prefix   = opts['Prefix']  || ''
+		self.postfix  = opts['Postfix'] || ''
+		self.colprops = []
+
+		# Default column properties
+		self.columns.length.times { |idx|
+			self.colprops[idx] = {}
+			self.colprops[idx]['MaxWidth'] = self.columns[idx].length
+		}
+
+		# ensure all our internal state gets updated with the given rows by
+		# using add_row instead of just adding them to self.rows.  See #3825.
+		opts['Rows'].each { |row| add_row(row) } if opts['Rows']
+
+		# Merge in options
+		if (opts['ColProps'])
+			opts['ColProps'].each_key { |col|
+				idx = self.columns.index(col)
+
+				if (idx)
+					self.colprops[idx].merge!(opts['ColProps'][col])
+				end
+			}
+		end
+
+	end
+
+	#
+	# Converts table contents to a string.
+	# 
+	def to_s
+		str  = prefix
+		str << header_to_s || ''
+		str << columns_to_s || ''
+		str << hr_to_s || ''
+		
+		rows.each { |row|
+			if (is_hr(row))
+				str << hr_to_s
+			else
+				str << row_to_s(row)
+			end
+		}
+
+		str << postfix
+
+		return str
+	end
+	
+	#
+	# Converts table contents to a csv
+	#
+	def to_csv
+		str = ''
+		str << ( columns.join(",") + "\n" )
+		rows.each { |row|
+			next if is_hr(row)
+			str << ( row.map{|x| 
+				x.gsub(/[\r\n]/, ' ').gsub(/\s+/, ' ').gsub('"', '""')
+			}.map{|x| "\"#{x}\"" }.join(",") + "\n" )
+		}
+		str
+	end
+
+	#
+	#
+	# Returns the header string.
+	#
+	def header_to_s # :nodoc:
+		if (header)
+			pad = " " * headeri
+
+			return pad + header + "\n" + pad + "=" * header.length + "\n\n"
+		end
+
+		return ''
+	end
+
+	#
+	# Prints the contents of the table.
+	#
+	def print
+		puts to_s
+	end
+
+	#
+	# Adds a row using the supplied fields.
+	#
+	def <<(fields)
+		add_row(fields)
+	end
+
+	#
+	# Adds a row with the supplied fields.
+	#
+	def add_row(fields = [])
+		if fields.length != self.columns.length
+			raise RuntimeError, 'Invalid number of columns!'
+		end
+		fields.each_with_index { |field, idx|
+			if (colprops[idx]['MaxWidth'] < field.to_s.length)
+				colprops[idx]['MaxWidth'] = field.to_s.length
+			end
+		}
+
+		rows << fields	
+	end
+
+	#
+	# Adds a horizontal line.
+	#
+	def add_hr
+		rows << '__hr__'
+	end
+
+	alias p print
+
+	attr_accessor :header, :headeri # :nodoc:
+	attr_accessor :columns, :rows, :colprops # :nodoc:
+	attr_accessor :width, :indent, :cellpad # :nodoc:
+	attr_accessor :prefix, :postfix # :nodoc:
+
+protected
+
+	#
+	# Defaults cell widths and alignments.
+	#
+	def defaults # :nodoc:
+		self.columns.length.times { |idx|
+		}
+	end
+
+	#
+	# Checks to see if the row is an hr.
+	#
+	def is_hr(row) # :nodoc:
+		return ((row.kind_of?(String)) && (row == '__hr__'))
+	end
+
+	#
+	# Converts the columns to a string.
+	#
+	def columns_to_s # :nodoc:
+		nameline = ' ' * indent
+		barline  = nameline.dup
+		last_col = nil
+		last_idx = nil
+		columns.each_with_index { |col,idx|
+			if (last_col)
+				nameline << pad(' ', last_col, last_idx)
+
+				remainder = colprops[last_idx]['MaxWidth'] - last_col.length
+			if (remainder < 0)
+				remainder = 0
+			end
+				barline << (' ' * (cellpad + remainder))
+			end
+			nameline << col
+			barline << ('-' * col.length)
+				
+			last_col = col
+			last_idx = idx
+		}
+
+		return "#{nameline}\n#{barline}"
+	end
+
+	#
+	# Converts an hr to a string.
+	#
+	def hr_to_s # :nodoc:
+		return "\n"
+	end
+
+	#
+	# Converts a row to a string.
+	#
+	def row_to_s(row) # :nodoc:
+		line = ' ' * indent
+		last_cell = nil
+		last_idx = nil
+		row.each_with_index { |cell, idx|
+			if (last_cell)
+				line << pad(' ', last_cell.to_s, last_idx)
+			end
+			line << cell.to_s
+			# line << pad(' ', cell.to_s, idx)
+			last_cell = cell
+			last_idx = idx
+		}
+
+		return line + "\n"
+	end
+
+	#
+	# Pads out with the supplied character for the remainder of the space given
+	# some text and a column index.
+	#
+	def pad(chr, buf, colidx, use_cell_pad = true) # :nodoc:
+		remainder = colprops[colidx]['MaxWidth'] - buf.length
+		val       = chr * remainder;
+
+		if (use_cell_pad)
+			val << ' ' * cellpad
+		end
+
+		return val
+	end
+
+end
+
+end
+end
+end

data/lib/rex/ui/text/table.rb.ut.rb

@@ -0,0 +1,55 @@
+#!/usr/bin/env ruby
+
+$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))
+
+require 'test/unit'
+require 'rex/ui/text/table'
+
+class Rex::Ui::Text::Table::UnitTest < Test::Unit::TestCase
+
+	def new_table(opts = {})
+		if (opts['Columns'] == nil)
+			opts['Columns'] = 
+				[
+					'col1',
+					'col2',
+					'col3'
+				]
+		end
+
+		tbl = Rex::Ui::Text::Table.new(opts)
+
+		tbl << [ "r1cell1", "r1cell2", "r1cell3" ]
+		tbl << [ "r2cell1", "r2cell2", "r2cell3" ]
+
+		return tbl
+	end
+
+	def test_basic
+		tbl = new_table
+
+		dstr = <<End 
+col1     col2     col3     
+----     ----     ----     
+r1cell1  r1cell2  r1cell3  
+r2cell1  r2cell2  r2cell3  
+End
+
+		assert_equal(tbl.to_s, dstr)
+	end
+
+	def test_indent
+		tbl = new_table(
+			'Indent' => 4)
+
+		dstr = <<End 
+    col1     col2     col3     
+    ----     ----     ----     
+    r1cell1  r1cell2  r1cell3  
+    r2cell1  r2cell2  r2cell3  
+End
+
+		assert_equal(tbl.to_s, dstr)
+	end
+
+end

data/lib/rex/zip.rb

@@ -0,0 +1,93 @@
+#!/usr/bin/env ruby
+#
+# Zip library
+#
+# Written by Joshua J. Drake <jduck [at] metasploit.com>
+#
+# Based on code contributed by bannedit, and the following SPEC:
+# Reference: http://www.pkware.com/documents/casestudies/APPNOTE.TXT
+#
+
+require 'zlib'
+
+module Rex
+module Zip
+
+ZIP_VERSION = 0x14
+
+# general purpose bit flag values
+#
+# bit 0
+GPBF_ENCRYPTED      = 0x0001
+# bits 1 & 2
+# implode only
+GPBF_IMP_8KDICT     = 0x0002
+GPBF_IMP_3SFT       = 0x0004
+# deflate only
+GPBF_DEF_MAX        = 0x0002
+GPBF_DEF_FAST       = 0x0004
+GPBF_DEF_SUPERFAST  = 0x0006
+# lzma only
+GPBF_LZMA_EOSUSED   = 0x0002
+# bit 3
+GPBF_USE_DATADESC   = 0x0008
+# bit 4
+GPBF_DEF_ENHANCED   = 0x0010
+# bit 5
+GPBF_COMP_PATHCED   = 0x0020
+# bit 6
+GPBF_STRONG_ENC     = 0x0040
+# bit 7-10 unused
+# bit 11
+GPBF_STRS_UTF8      = 0x0800
+# bit 12 (reserved)
+# bit 13
+GPBF_DIR_ENCRYPTED  = 0x2000
+# bit 14,15 (reserved)
+
+
+# compression methods
+CM_STORE            = 0
+CM_SHRINK           = 1
+CM_REDUCE1          = 2
+CM_REDUCE2          = 3
+CM_REDUCE3          = 4
+CM_REDUCE4          = 5
+CM_IMPLODE          = 6
+CM_TOKENIZE         = 7
+CM_DEFLATE          = 8
+CM_DEFLATE64        = 9
+CM_PKWARE_IMPLODE   = 10
+# 11 - reserved
+CM_BZIP2            = 12
+# 13 - reserved
+CM_LZMA_EFS         = 14
+# 15-17 reserved
+CM_IBM_TERSE        = 18
+CM_IBM_LZ77         = 19
+# 20-96 reserved
+CM_WAVPACK          = 97
+CM_PPMD_V1R1        = 98
+
+
+# internal file attributes
+IFA_ASCII           = 0x0001
+# bits 2 & 3 are reserved
+IFA_MAINFRAME_MODE  = 0x0002 # ??
+
+
+# external file attributes
+EFA_ISDIR           = 0x0001
+
+
+# various parts of the structure
+require 'rex/zip/blocks'
+
+# an entry in a zip file
+require 'rex/zip/entry'
+
+# the archive class
+require 'rex/zip/archive'
+
+end
+end

data/lib/rex/zip/archive.rb

@@ -0,0 +1,184 @@
+##
+# $Id: archive.rb 11175 2010-11-30 07:10:57Z egypt $
+##
+
+module Rex
+module Zip
+
+#
+# This represents an entire archive.
+#
+class Archive
+	attr_reader :entries
+
+	def initialize(compmeth=CM_DEFLATE)
+		@compmeth = compmeth
+		@entries = []
+	end
+
+
+	def add_file(fname, fdata=nil, xtra=nil, comment=nil)
+		if (not fdata)
+			begin
+				st = File.stat(fname)
+			rescue
+				return nil
+			end
+
+			ts = st.mtime
+			if (st.directory?)
+				attrs = EFA_ISDIR
+				fname += '/'
+			else
+				f = File.open(fname, 'rb')
+				fdata = f.read(f.stat.size)
+				f.close
+			end
+		end
+
+		@entries << Entry.new(fname, fdata, @compmeth, ts, attrs, xtra, comment)
+	end
+
+
+	def set_comment(comment)
+		@comment = comment
+	end
+
+
+	def save_to(fname)
+		f = File.open(fname, 'wb')
+		f.write(pack)
+		f.close
+	end
+
+
+	def pack
+		ret = ''
+
+		# save the offests
+		offsets = []
+
+		# file 1 .. file n
+		@entries.each { |ent|
+			offsets << ret.length
+			ret << ent.pack
+		}
+
+		# archive decryption header (unsupported)
+		# archive extra data record (unsupported)
+
+		# central directory
+		cfd_offset = ret.length
+		idx = 0
+		@entries.each { |ent|
+			cfd = CentralDir.new(ent, offsets[idx])
+			ret << cfd.pack
+			idx += 1
+		}
+
+		# zip64 end of central dir record (unsupported)
+		# zip64 end of central dir locator (unsupported)
+
+		# end of central directory record
+		cur_offset = ret.length - cfd_offset
+		ret << CentralDirEnd.new(@entries.length, cur_offset, cfd_offset, @comment).pack
+
+		ret
+	end
+
+	def inspect
+		"#<#{self.class} entries = [#{@entries.map{|e| e.name}.join(",")}]>"
+	end
+
+end
+
+class Jar < Archive
+	attr_accessor :manifest
+
+	def build_manifest(opts={})
+		main_class = opts[:main_class] || nil
+		existing_manifest = nil
+
+		@manifest =  "Manifest-Version: 1.0\r\n"
+		@manifest << "Main-Class: #{main_class}\r\n" if main_class
+		@manifest << "\r\n"
+		@entries.each { |e|
+			existing_manifest = e if e.name == "META-INF/MANIFEST.MF" 
+			next unless e.name =~ /\.class$/
+			@manifest << "Name: #{e.name}\r\n"
+			#@manifest << "SHA1-Digest: #{Digest::SHA1.base64digest(e.data)}\r\n"
+			@manifest << "\r\n"
+		}
+		if existing_manifest
+			existing_manifest.data = @manifest
+		else
+			add_file("META-INF/", '')
+			add_file("META-INF/MANIFEST.MF", @manifest)
+		end
+	end
+
+	def to_s
+		pack
+	end
+
+	def length
+		pack.length
+	end
+
+	#
+	# Add multiple files from an array
+	#
+	# +files+ should be structured like so:
+	#   [
+	#     [ "path", "to", "file1" ],
+	#     [ "path", "to", "file2" ]
+	#   ]
+	# and +path+ should be the location on the file system to find the files to
+	# add.  +base_dir+ will be prepended to the path inside the jar.
+	#
+	# Example:
+	# <code>
+	# war = Rex::Zip::Jar.new
+	# war.add_file("WEB-INF/", '')
+	# war.add_file("WEB-INF/", "web.xml", web_xml)
+	# war.add_file("WEB-INF/classes/", '')
+	# files = [
+	#	[ "servlet", "examples", "HelloWorld.class" ],
+	#	[ "Foo.class" ],
+	#	[ "servlet", "Bar.class" ],
+	# ]
+	# war.add_files(files, "./class_files/", "WEB-INF/classes/")
+	# </code>
+	#
+	# The above code would create a jar with the following structure from files
+	# found in ./class_files/ :
+	#
+	# +- WEB-INF/
+	#   +- web.xml
+	#   +- classes/
+	#     +- Foo.class
+	#     +- servlet/
+	#       +- Bar.class
+	#       +- examples/
+	#         +- HelloWorld.class
+	#
+	def add_files(files, path, base_dir="")
+		files.each do |file|
+			# Add all of the subdirectories if they don't already exist
+			1.upto(file.length - 1) do |idx|
+				full = base_dir + file[0,idx].join("/") + "/"
+				if !(entries.map{|e|e.name}.include?(full))
+					add_file(full, '')
+				end
+			end
+			# Now add the actual file, grabbing data from the filesystem
+			fd = File.open(File.join( path, file ), "rb")
+			data = fd.read(fd.stat.size)
+			fd.close
+			add_file(base_dir + file.join("/"), data)
+		end
+	end
+end
+
+end
+end