71 lines
1.9 KiB
Ruby
71 lines
1.9 KiB
Ruby
require "tempfile"
|
|
require "tmpdir"
|
|
require "fileutils"
|
|
|
|
class UploadedFile
|
|
InvalidPathError = Class.new(StandardError)
|
|
|
|
# The filename, *not* including the path, of the "uploaded" file
|
|
attr_reader :original_filename
|
|
|
|
# The tempfile
|
|
attr_reader :tempfile
|
|
|
|
# The content type of the "uploaded" file
|
|
attr_accessor :content_type
|
|
|
|
attr_reader :remote_id
|
|
attr_reader :sha256
|
|
|
|
def initialize(path, filename: nil, content_type: "application/octet-stream", sha256: nil, remote_id: nil)
|
|
raise InvalidPathError, "#{path} file does not exist" unless ::File.exist?(path)
|
|
|
|
@content_type = content_type
|
|
@original_filename = filename || ::File.basename(path)
|
|
@content_type = content_type
|
|
@sha256 = sha256
|
|
@remote_id = remote_id
|
|
@tempfile = File.new(path, 'rb')
|
|
end
|
|
|
|
def self.from_params(params, field, upload_paths)
|
|
unless params["#{field}.path"]
|
|
raise InvalidPathError, "file is invalid" if params["#{field}.remote_id"]
|
|
|
|
return
|
|
end
|
|
|
|
file_path = File.realpath(params["#{field}.path"])
|
|
|
|
paths = Array(upload_paths) << Dir.tmpdir
|
|
unless self.allowed_path?(file_path, paths.compact)
|
|
raise InvalidPathError, "insecure path used '#{file_path}'"
|
|
end
|
|
|
|
UploadedFile.new(file_path,
|
|
filename: params["#{field}.name"],
|
|
content_type: params["#{field}.type"] || 'application/octet-stream',
|
|
sha256: params["#{field}.sha256"],
|
|
remote_id: params["#{field}.remote_id"])
|
|
end
|
|
|
|
def self.allowed_path?(file_path, paths)
|
|
paths.any? do |path|
|
|
File.exist?(path) && file_path.start_with?(File.realpath(path))
|
|
end
|
|
end
|
|
|
|
def path
|
|
@tempfile.path
|
|
end
|
|
|
|
alias_method :local_path, :path
|
|
|
|
def method_missing(method_name, *args, &block) #:nodoc:
|
|
@tempfile.__send__(method_name, *args, &block) # rubocop:disable GitlabSecurity/PublicSend
|
|
end
|
|
|
|
def respond_to?(method_name, include_private = false) #:nodoc:
|
|
@tempfile.respond_to?(method_name, include_private) || super
|
|
end
|
|
end
|