#----------------------------------------------------------
# prstmidx.rb -  Print all the streams in the capture file
#  Alternate approach demostrating Ruby Classes
#
#  usage: prstmidx <capture-filename>
#
#---------------------------------------------------------
require 'win32ole'
 
USAGE = "prstmidx <capture-filename>"
endl  = "\n"
tab   = "\t"
 
 
class UStream
	def initialize(usnfstream)
		@stm = usnfstream
	end
 
	def to_s
		"#{@stm.ID}\t#{@stm.StartTimeStamp}\t#{@stm.SourceAddress}\t" +
		"#{@stm.DestinationAddress}\t#{@stm.InByteCount}\t#{@stm.OutByteCount}\t" +
		"#{@stm.Description}\n" 
	end
end
 
class StreamList
	def initialize(stmindex)
		@streamList = Array.new
 
		count = stmindex.Count
		(0..count-1).each { |idx| @streamList.push( UStream.new(stmindex.Item(idx)) ) }
	end
 
	def to_s
		s = "Number of streams in capture file = #{@streamList.length}\n"
		@streamList.each { |itm| s += itm.to_s }
		s
	end
end
 
 
if ARGV.length != 1
	puts USAGE
	exit 1
end
 
UnsniffDB = WIN32OLE.new("Unsniff.Database")
UnsniffDB.Open(ARGV[0])
 
slist = StreamList.new(UnsniffDB.StreamIndex)
print slist
 
UnsniffDB.Close()