98 lines
2.2 KiB
Ruby
98 lines
2.2 KiB
Ruby
require 'find'
|
|
require 'rex/compat'
|
|
require 'tempfile'
|
|
|
|
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)
|
|
|
|
# Check for the absolute fast first
|
|
if (file_name[0,1] == "/" and ::File.exists?(file_name) and ::File::Stat.new(file_name))
|
|
return file_name
|
|
end
|
|
|
|
path = Rex::Compat.getenv('PATH')
|
|
if (path)
|
|
path.split(::File::PATH_SEPARATOR).each { |base|
|
|
begin
|
|
# Deal with Windows paths surrounded by quotes. Prevents
|
|
# silliness like trying to look for
|
|
# '"C:\\framework\\nmap"\\nmap.exe' which will always fail.
|
|
base = $1 if base =~ /^"(.*)"$/
|
|
path = base + ::File::SEPARATOR + file_name
|
|
if (::File::Stat.new(path) and not ::File.directory?(path))
|
|
return path
|
|
end
|
|
rescue
|
|
end
|
|
}
|
|
end
|
|
return nil
|
|
end
|
|
|
|
end
|
|
|
|
class Quickfile < ::Tempfile
|
|
def initialize(*args)
|
|
super(*args)
|
|
self.binmode
|
|
ObjectSpace.undefine_finalizer(self)
|
|
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
|