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) ⇒ Object
Compute hash code.
- #verify_iso(options) ⇒ Object
- #verify_md5sum(full_path) ⇒ Object
Instance Method Details
#download_iso(url, filename) ⇒ Object
12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
# File 'lib/veewee/provider/core/helper/iso.rb', line 12 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
28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
# File 'lib/veewee/provider/core/helper/iso.rb', line 28 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) ⇒ Object
Compute hash code
54 55 56 57 58 59 60 61 62 63 64 65 66 |
# File 'lib/veewee/provider/core/helper/iso.rb', line 54 def hashsum(filename) checksum=Digest::MD5.new 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
80 81 82 83 84 85 86 87 88 89 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 |
# File 'lib/veewee/provider/core/helper/iso.rb', line 80 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}" 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}' " ui.info "" exit end unless File.readable?(full_path) raise Veewee::Error, "The provided iso #{full_path} is not readable" end end end verify_md5sum(full_path) if ["md5check"] && !self.iso_md5.nil? end |
#verify_md5sum(full_path) ⇒ Object
68 69 70 71 72 73 74 75 76 77 78 |
# File 'lib/veewee/provider/core/helper/iso.rb', line 68 def verify_md5sum(full_path) filename = File.basename(full_path) ui.info "Verifying md5 checksum : #{self.iso_md5}" file_md5=hashsum(full_path) unless file_md5==self.iso_md5 ui.error "The MD5 checksums for file #{filename } do not match: " ui.error "- #{file_md5} (current) vs #{self.iso_md5} (specified)" raise Veewee::Error, "The MD5 checksums for file #{filename } do not match: \n"+ "- #{file_md5} (current) vs #{self.iso_md5} (specified)" end end |