Skip to content
This repository has been archived by the owner on Apr 14, 2021. It is now read-only.

Commit

Permalink
handle Dir.mktmpdir failure outside of filesystem_access block
Browse files Browse the repository at this point in the history
  • Loading branch information
Adam Wanninger committed Aug 22, 2017
1 parent 4383414 commit 1813716
Show file tree
Hide file tree
Showing 2 changed files with 53 additions and 46 deletions.
98 changes: 53 additions & 45 deletions lib/bundler/compact_index_client/updater.rb
Original file line number Diff line number Diff line change
Expand Up @@ -26,60 +26,57 @@ def initialize(fetcher)
end

def update(local_path, remote_path, retrying = nil)
headers = {}
local_temp_dir = create_tmpdir

Bundler::SharedHelpers.filesystem_access(Dir.tmpdir, :write) do
Dir.mktmpdir("bundler-compact-index-") do |local_temp_dir|
local_temp_path = Pathname.new(local_temp_dir).join(local_path.basename)

# first try to fetch any new bytes on the existing file
if retrying.nil? && local_path.file?
FileUtils.cp local_path, local_temp_path
headers["If-None-Match"] = etag_for(local_temp_path)
headers["Range"] =
if local_temp_path.size.nonzero?
# Subtract a byte to ensure the range won't be empty.
# Avoids 416 (Range Not Satisfiable) responses.
"bytes=#{local_temp_path.size - 1}-"
else
"bytes=#{local_temp_path.size}-"
end
headers = {}
local_temp_path = Pathname.new(local_temp_dir).join(local_path.basename)

# first try to fetch any new bytes on the existing file
if retrying.nil? && local_path.file?
FileUtils.cp local_path, local_temp_path
headers["If-None-Match"] = etag_for(local_temp_path)
headers["Range"] =
if local_temp_path.size.nonzero?
# Subtract a byte to ensure the range won't be empty.
# Avoids 416 (Range Not Satisfiable) responses.
"bytes=#{local_temp_path.size - 1}-"
else
# Fastly ignores Range when Accept-Encoding: gzip is set
headers["Accept-Encoding"] = "gzip"
end

response = @fetcher.call(remote_path, headers)
return nil if response.is_a?(Net::HTTPNotModified)

content = response.body
if response["Content-Encoding"] == "gzip"
content = Zlib::GzipReader.new(StringIO.new(content)).read
"bytes=#{local_temp_path.size}-"
end
else
# Fastly ignores Range when Accept-Encoding: gzip is set
headers["Accept-Encoding"] = "gzip"
end

SharedHelpers.filesystem_access(local_temp_path) do
if response.is_a?(Net::HTTPPartialContent) && local_temp_path.size.nonzero?
local_temp_path.open("a") {|f| f << slice_body(content, 1..-1) }
else
local_temp_path.open("w") {|f| f << content }
end
end
response = @fetcher.call(remote_path, headers)
return nil if response.is_a?(Net::HTTPNotModified)

response_etag = (response["ETag"] || "").gsub(%r{\AW/}, "")
if etag_for(local_temp_path) == response_etag
SharedHelpers.filesystem_access(local_path) do
FileUtils.mv(local_temp_path, local_path)
end
return nil
end
content = response.body
if response["Content-Encoding"] == "gzip"
content = Zlib::GzipReader.new(StringIO.new(content)).read
end

if retrying
raise MisMatchedChecksumError.new(remote_path, response_etag, etag_for(local_temp_path))
end
SharedHelpers.filesystem_access(local_temp_path) do
if response.is_a?(Net::HTTPPartialContent) && local_temp_path.size.nonzero?
local_temp_path.open("a") {|f| f << slice_body(content, 1..-1) }
else
local_temp_path.open("w") {|f| f << content }
end
end

update(local_path, remote_path, :retrying)
response_etag = (response["ETag"] || "").gsub(%r{\AW/}, "")
if etag_for(local_temp_path) == response_etag
SharedHelpers.filesystem_access(local_path) do
FileUtils.mv(local_temp_path, local_path)
end
return nil
end

if retrying
raise MisMatchedChecksumError.new(remote_path, response_etag, etag_for(local_temp_path))
end

update(local_path, remote_path, :retrying)
end

def etag_for(path)
Expand All @@ -104,6 +101,17 @@ def checksum_for_file(path)
Digest::MD5.hexdigest(IO.read(path))
end
end

private

def create_tmpdir
Dir.mktmpdir("bundler-compact-index-")
rescue Errno::EACCES
raise Bundler::PermissionError,
"Bundler does not have write access to create a temp directory " \
"within #{Dir.tmpdir}. Bundler must have write access to your " \
"systems temp directory to function properly. " \
end
end
end
end
1 change: 0 additions & 1 deletion spec/bundler/compact_index_client/updater_spec.rb
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,6 @@
let(:remote_path) { double(:remote_path) }

it "Errno::EACCES is raised" do
allow(Dir).to receive(:tmpdir) { "/tmp" }
allow(Dir).to receive(:mktmpdir) { raise Errno::EACCES }

expect do
Expand Down

0 comments on commit 1813716

Please sign in to comment.