mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
4b8f0e7637
* lib/shell/system-command.rb: ditto. * lib/tmpdir.rb: ditto. * lib/uri/generic.rb: ditto. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@46969 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
152 lines
4.1 KiB
Ruby
152 lines
4.1 KiB
Ruby
#
|
|
# tmpdir - retrieve temporary directory path
|
|
#
|
|
# $Id$
|
|
#
|
|
|
|
require 'fileutils'
|
|
begin
|
|
require 'etc.so'
|
|
rescue LoadError # rescue LoadError for miniruby
|
|
end
|
|
|
|
class Dir
|
|
|
|
@@systmpdir ||= defined?(Etc.systmpdir) ? Etc.systmpdir : '/tmp'
|
|
|
|
##
|
|
# Returns the operating system's temporary file path.
|
|
|
|
def Dir::tmpdir
|
|
if $SAFE > 0
|
|
@@systmpdir
|
|
else
|
|
tmp = nil
|
|
for dir in [ENV['TMPDIR'], ENV['TMP'], ENV['TEMP'], @@systmpdir, '/tmp', '.']
|
|
next if !dir
|
|
dir = File.expand_path(dir)
|
|
if stat = File.stat(dir) and stat.directory? and stat.writable? and
|
|
(!stat.world_writable? or stat.sticky?)
|
|
tmp = dir
|
|
break
|
|
end rescue nil
|
|
end
|
|
raise ArgumentError, "could not find a temporary directory" if !tmp
|
|
tmp
|
|
end
|
|
end
|
|
|
|
# Dir.mktmpdir creates a temporary directory.
|
|
#
|
|
# The directory is created with 0700 permission.
|
|
# Application should not change the permission to make the temporary directory accessible from other users.
|
|
#
|
|
# The prefix and suffix of the name of the directory is specified by
|
|
# the optional first argument, <i>prefix_suffix</i>.
|
|
# - If it is not specified or nil, "d" is used as the prefix and no suffix is used.
|
|
# - If it is a string, it is used as the prefix and no suffix is used.
|
|
# - If it is an array, first element is used as the prefix and second element is used as a suffix.
|
|
#
|
|
# Dir.mktmpdir {|dir| dir is ".../d..." }
|
|
# Dir.mktmpdir("foo") {|dir| dir is ".../foo..." }
|
|
# Dir.mktmpdir(["foo", "bar"]) {|dir| dir is ".../foo...bar" }
|
|
#
|
|
# The directory is created under Dir.tmpdir or
|
|
# the optional second argument <i>tmpdir</i> if non-nil value is given.
|
|
#
|
|
# Dir.mktmpdir {|dir| dir is "#{Dir.tmpdir}/d..." }
|
|
# Dir.mktmpdir(nil, "/var/tmp") {|dir| dir is "/var/tmp/d..." }
|
|
#
|
|
# If a block is given,
|
|
# it is yielded with the path of the directory.
|
|
# The directory and its contents are removed
|
|
# using FileUtils.remove_entry before Dir.mktmpdir returns.
|
|
# The value of the block is returned.
|
|
#
|
|
# Dir.mktmpdir {|dir|
|
|
# # use the directory...
|
|
# open("#{dir}/foo", "w") { ... }
|
|
# }
|
|
#
|
|
# If a block is not given,
|
|
# The path of the directory is returned.
|
|
# In this case, Dir.mktmpdir doesn't remove the directory.
|
|
#
|
|
# dir = Dir.mktmpdir
|
|
# begin
|
|
# # use the directory...
|
|
# open("#{dir}/foo", "w") { ... }
|
|
# ensure
|
|
# # remove the directory.
|
|
# FileUtils.remove_entry dir
|
|
# end
|
|
#
|
|
def Dir.mktmpdir(prefix_suffix=nil, *rest)
|
|
path = Tmpname.create(prefix_suffix || "d", *rest) {|n| mkdir(n, 0700)}
|
|
if block_given?
|
|
begin
|
|
yield path
|
|
ensure
|
|
stat = File.stat(File.dirname(path))
|
|
if stat.world_writable? and !stat.sticky?
|
|
raise ArgumentError, "parent directory is world writable but not sticky"
|
|
end
|
|
FileUtils.remove_entry path
|
|
end
|
|
else
|
|
path
|
|
end
|
|
end
|
|
|
|
module Tmpname # :nodoc:
|
|
module_function
|
|
|
|
def tmpdir
|
|
Dir.tmpdir
|
|
end
|
|
|
|
def make_tmpname(prefix_suffix, n)
|
|
case prefix_suffix
|
|
when String
|
|
prefix = prefix_suffix
|
|
suffix = ""
|
|
when Array
|
|
prefix = prefix_suffix[0]
|
|
suffix = prefix_suffix[1]
|
|
else
|
|
raise ArgumentError, "unexpected prefix_suffix: #{prefix_suffix.inspect}"
|
|
end
|
|
t = Time.now.strftime("%Y%m%d")
|
|
path = "#{prefix}#{t}-#{$$}-#{rand(0x100000000).to_s(36)}"
|
|
path << "-#{n}" if n
|
|
path << suffix
|
|
end
|
|
|
|
def create(basename, *rest)
|
|
if opts = Hash.try_convert(rest[-1])
|
|
opts = opts.dup if rest.pop.equal?(opts)
|
|
max_try = opts.delete(:max_try)
|
|
opts = [opts]
|
|
else
|
|
opts = []
|
|
end
|
|
tmpdir, = *rest
|
|
if $SAFE > 0 and tmpdir.tainted?
|
|
tmpdir = '/tmp'
|
|
else
|
|
tmpdir ||= tmpdir()
|
|
end
|
|
n = nil
|
|
begin
|
|
path = File.join(tmpdir, make_tmpname(basename, n))
|
|
yield(path, n, *opts)
|
|
rescue Errno::EEXIST
|
|
n ||= 0
|
|
n += 1
|
|
retry if !max_try or n < max_try
|
|
raise "cannot generate temporary name using `#{basename}' under `#{tmpdir}'"
|
|
end
|
|
path
|
|
end
|
|
end
|
|
end
|