require 'find' module Rex ### # # This class provides helper methods for dealing with files that are not # supplied by the standard ruby API. # ### module FileUtils # # This method searches the PATH environment variable for # a fully qualified path to the supplied file name. # def self.find_full_path(file_name) path = Rex::Compat.getenv('PATH') if (path) path.split(::File::PATH_SEPARATOR).each { |base| begin path = base + ::File::SEPARATOR + file_name if (::File::Stat.new(path)) return path end rescue end } end return nil end end module Find # # Identical to Find.find from Ruby, but follows symlinks to directories. # See http://blade.nagaokaut.ac.jp/cgi-bin/scat.rb/ruby/ruby-talk/68671 # def self.find(*paths) paths.collect!{|d| d.dup} while file = paths.shift catch(:prune) do yield file.dup.taint next unless File.exist? file begin if File.stat(file).directory? then d = Dir.open(file) begin for f in d next if f == "." or f == ".." if File::ALT_SEPARATOR and file =~ /^(?:[\/\\]|[A-Za-z]:[\/\\]?)$/ then f = file + f elsif file == "/" then f = "/" + f else f = File.join(file, f) end paths.unshift f.untaint end ensure d.close end end rescue Errno::ENOENT, Errno::EACCES end end end end def self.prune throw :prune end end end