about summary refs log tree commit homepage
path: root/lib/unicorn/tmpio.rb
blob: 0bbf6ec5cb38ea0acd39136fd6f3ec4b3429035d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# -*- encoding: binary -*-
# :stopdoc:
require 'tmpdir'

# some versions of Ruby had a broken Tempfile which didn't work
# well with unlinked files.  This one is much shorter, easier
# to understand, and slightly faster.
class Unicorn::TmpIO < File

  # creates and returns a new File object.  The File is unlinked
  # immediately, switched to binary mode, and userspace output
  # buffering is disabled
  def self.new
    path = nil

    # workaround File#path being tainted:
    # https://bugs.ruby-lang.org/issues/14485
    fp = begin
      path = "#{Dir::tmpdir}/#{rand}"
      super(path, RDWR|CREAT|EXCL, 0600)
    rescue Errno::EEXIST
      retry
    end

    unlink(path)
    fp.binmode
    fp.sync = true
    fp
  end

  # pretend we're Tempfile for Rack::TempfileReaper
  alias close! close
end