# -----------------------------------------------------------------
# batchimp.rb	-  Batch import a set of libpcap files into a
#		   single Unsniff capture file
#
#	usage  : batchimp <directory>  <libpcapfiles> <unsniff-file>
#	exmple : batchimp ..\tdata a*.pcap output.usnf
#------------------------------------------------------------------
require 'win32ole'
 
USAGE = "batchimp <directory> <import-files> <unsniff-file>"
 
if ARGV.length != 3
	puts USAGE
	exit 1
end
 
FromDirectory = ARGV[0]
FromWildcards = ARGV[1]
ToFile        = ARGV[2]
 
#------ Check all parameters ------------
if ! File.directory?(FromDirectory)
	print "#{FromDirectory} is not a directory"
	exit 1
end 
 
#------ Create or Open the Unsniff file -----
UnsniffDB = WIN32OLE.new("Unsniff.Database")
if File.exist?(ToFile)
	UnsniffDB.Open(ToFile)
else
	UnsniffDB.New(ToFile)
end
 
 
#------ Start importing -------------
ipkts =UnsniffDB.PacketCount
OldDir = Dir.pwd
Dir.chdir(FromDirectory)
FilesToImport = Dir[FromWildcards]
Dir.chdir(OldDir)
 
FilesToImport.each { |fil|
	impfile = File.join(FromDirectory,fil)
	impfile.sub!(/\//,'\\')
	print "Importing from #{impfile}..."
 
	begin
		UnsniffDB.Import("libpcap",impfile)
	rescue
	end
 
	ipkts = UnsniffDB.PacketCount - ipkts
	print "(#{ipkts} packets) done.\n"
}
 
print "wait! saving capture file"
UnsniffDB.Save
print "done"
 
print "Unsniff Filename is  #{ToFile}"