Sha256: 7fbf3721bcf5e3836f030c61c02cedf2d220cf1bd44f0536cf97e8094c651205
Contents?: true
Size: 1.55 KB
Versions: 25
Compression:
Stored size: 1.55 KB
Contents
# frozen_string_literal: true
module Hyrax
# Responsible for checking if the given file is a virus. Coordinates
# with the underlying system virus scanner.
class VirusCheckerService
attr_accessor :original_file, :system_virus_scanner
# @api public
# @param original_file [String, #path]
# @return true or false result from system_virus_scanner
def self.file_has_virus?(original_file)
new(original_file).file_has_virus?
end
def initialize(original_file, system_virus_scanner = Hyrax.config.virus_scanner)
self.original_file = original_file
self.system_virus_scanner = system_virus_scanner
end
# Default behavior is to raise a validation error and halt the save if a virus is found
def file_has_virus?
path = original_file.is_a?(String) ? original_file : local_path_for_file(original_file)
system_virus_scanner.infected?(path)
end
private
# Returns a path for reading the content of +file+
# @param [File] file object to retrieve a path for
def local_path_for_file(file)
return file.path if file.respond_to?(:path)
return file.content.path if file.content.respond_to?(:path)
Tempfile.open('') do |t|
t.binmode
write_to_temp_file(file, t)
t.close
t.path
end
end
def write_to_temp_file(file, temp_file)
if file.new_record?
temp_file.write(file.content.read)
file.content.rewind
else
file.stream.each do |chunk|
temp_file.write(chunk)
end
end
end
end
end
Version data entries
25 entries across 25 versions & 1 rubygems