librex 0.0.13 → 0.0.15

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

@@ -1,279 +0,0 @@
-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

@@ -1,55 +0,0 @@
-#!/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

@@ -1,93 +0,0 @@
-#!/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

@@ -1,184 +0,0 @@
-##
-# $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