require 'find'

module Rex

###
#
# This class provides helper mehods 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)
		
		case RUBY_PLATFORM
		when /mswin32/
			if (ENV['PATH'])
				ENV['PATH'].split(';').each { |base|
					begin
						path = base + ::File::SEPARATOR + file_name
						if (::File::Stat.new(path))
							return path
						end
					rescue
					end
				}
			end
		else
			if (ENV['PATH'])
				ENV['PATH'].split(':').each { |base|
					begin
						path = base + ::File::SEPARATOR + file_name
						if (::File::Stat.new(path))
							return path
						end
					rescue
					end
				}
			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