Module: Veewee::Provider::Core::Helper::Iso
- Included in:
- Definition, Box
- Defined in:
- lib/veewee/provider/core/helper/iso.rb
Instance Method Summary collapse
- #download_iso(url, filename) ⇒ Object
- #download_progress(url, localfile) ⇒ Object
-
#hashsum(filename, type) ⇒ Object
Compute hash code.
- #verify_iso(options) ⇒ Object
- #verify_sum(full_path, type) ⇒ Object
Instance Method Details
#download_iso(url, filename) ⇒ Object
13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
# File 'lib/veewee/provider/core/helper/iso.rb', line 13 def download_iso(url,filename) if !File.exists?(env.config.veewee.iso_dir) ui.info "Creating an iso directory" FileUtils.mkdir(env.config.veewee.iso_dir) end ui.info "Checking if isofile #{filename} already exists." full_path=File.join(env.config.veewee.iso_dir,filename) ui.info "Full path: #{full_path}" if File.exists?(full_path) ui.info "" ui.info "The isofile #{filename} already exists." else download_progress(url,full_path) end end |
#download_progress(url, localfile) ⇒ Object
29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
# File 'lib/veewee/provider/core/helper/iso.rb', line 29 def download_progress(url,localfile) = nil URI.parse(url).open( :content_length_proc => lambda {|t| if t && 0 < t = ProgressBar.new("Fetching file", t) .file_transfer_mode end }, :progress_proc => lambda {|s| .set s if }) { |src| # We assume large 10K files, so this is tempfile object env.logger.info "#{src.class}" ui.info "Moving #{src.path} to #{localfile}" # Force the close of the src stream to release handle before moving # Not forcing the close may cause an issue on windows (Permission Denied) src.close FileUtils.mv(src.path,localfile) #open(localfile,"wb") { |dst| #dst.write(src.read) #} } end |
#hashsum(filename, type) ⇒ Object
Compute hash code
55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
# File 'lib/veewee/provider/core/helper/iso.rb', line 55 def hashsum(filename,type) case type when :md5 checksum=Digest::MD5.new when :sha1 checksum=Digest::SHA1.new else raise Veewee::Error, "Unknown checksum type #{type}" end buflen=1024 open(filename, "rb") do |io| counter = 0 while (!io.eof) readBuf = io.readpartial(buflen) env.ui.info('.',{:new_line => false,:prefix => false }) if ((counter+=1) % 20000 == 0) checksum.update(readBuf) end end return checksum.hexdigest end |
#verify_iso(options) ⇒ Object
90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 |
# File 'lib/veewee/provider/core/helper/iso.rb', line 90 def verify_iso() filename=self.iso_file full_path=File.join(env.config.veewee.iso_dir,filename) if File.exists?(full_path) ui.info "" ui.info "The isofile #{filename} already exists." else path1=Pathname.new(full_path) path2=Pathname.new(Dir.pwd) rel_path=path1.relative_path_from(path2).to_s ui.info "" ui.info "We did not find an isofile here : #{full_path}. \n\nThe definition provided the following download information:" unless "#{self.iso_src}"=="" ui.info "- Download url: #{self.iso_src}" end ui.info "- Md5 Checksum: #{self.iso_md5}" if self.iso_md5 ui.info "- Sha1 Checksum: #{self.iso_sha1}" if self.iso_sha1 ui.info "#{self.iso_download_instructions}" ui.info "" if self.iso_src == "" ui.info "Please follow the instructions above:" ui.info "- to get the ISO" ui.info" - put it in <currentdir>/iso" ui.info "- then re-run the command" ui.info "" raise Veewee::Error, "No ISO src is available, can't download it automatically" else answer=nil answer="yes" if ["auto"]==true env.logger.info "Auto download enabled?#{answer} #{!['auto'].nil?}" if answer.nil? answer=ui.ask("Download? (Yes/No)") {|q| q.default="No"} end if answer.downcase == "yes" begin download_iso(self.iso_src,filename) rescue OpenURI::HTTPError => ex ui.error "There was an error downloading #{self.iso_src}:" ui.error "#{ex}" raise Veewee::Error, "There was an error downloading #{self.iso_src}:\n#{ex}" end else ui.info "You have selected manual download: " ui.info "curl -C - -L '#{self.iso_src}' -o '#{rel_path}'" ui.info "md5 '#{rel_path}' " if self.iso_md5 ui.info "shasum '#{rel_path}' " if self.iso_sha1 ui.info "" exit end unless File.readable?(full_path) raise Veewee::Error, "The provided iso #{full_path} is not readable" end end end verify_sum(full_path,:md5) if ["checksum"] && !self.iso_md5.nil? verify_sum(full_path,:sha1) if ["checksum"] && !self.iso_sha1.nil? end |
#verify_sum(full_path, type) ⇒ Object
77 78 79 80 81 82 83 84 85 86 87 88 |
# File 'lib/veewee/provider/core/helper/iso.rb', line 77 def verify_sum(full_path,type) filename = File.basename(full_path) required_sum = self.instance_variable_get('@iso_'+type.to_s) ui.info "Verifying #{type} checksum : #{self.required_sum}" file_sum = hashsum(full_path,type) unless file_sum == required_sum ui.error "The #{type} checksum for file #{filename } do not match: " ui.error "- #{file_sum} (current) vs #{required_sum} (specified)" raise Veewee::Error, "The #{type} checksum for file #{filename } do not match: \n"+ "- #{file_sum} (current) vs #{required_sum} (specified)" end end |