metasploit-framework/lib/zip/samples/zipfind.rb

#!/usr/bin/env ruby

$VERBOSE = true

$: << "../lib"

require 'zip/zip'
require 'find'

module Zip
  module ZipFind
    def self.find(path, zipFilePattern = /\.zip$/i)
      Find.find(path) {
  |fileName|
  yield(fileName)
  if zipFilePattern.match(fileName)  && File.file?(fileName)
    begin
      Zip::ZipFile.foreach(fileName)  {
        |zipEntry|
        yield(fileName + File::SEPARATOR + zipEntry.to_s)
      }
    rescue Errno::EACCES => ex
      puts ex
    end
  end
      }
    end

    def self.find_file(path, fileNamePattern, zipFilePattern = /\.zip$/i)
      self.find(path, zipFilePattern) {
  |fileName|
  yield(fileName) if fileNamePattern.match(fileName)
      }
    end

  end
end

if __FILE__ == $0
  module ZipFindConsoleRunner
    
    PATH_ARG_INDEX = 0;
    FILENAME_PATTERN_ARG_INDEX = 1;
    ZIPFILE_PATTERN_ARG_INDEX = 2;
    
    def self.run(args)
      check_args(args)
      Zip::ZipFind.find_file(args[PATH_ARG_INDEX], 
          args[FILENAME_PATTERN_ARG_INDEX],
          args[ZIPFILE_PATTERN_ARG_INDEX]) {
  |fileName|
  report_entry_found fileName
      }
    end
    
    def self.check_args(args)
      if (args.size != 3)
  usage
  exit
      end
    end

    def self.usage
      puts "Usage: #{$0} PATH ZIPFILENAME_PATTERN FILNAME_PATTERN"
    end
    
    def self.report_entry_found(fileName)
      puts fileName
    end
    
  end

  ZipFindConsoleRunner.run(ARGV)
end