metasploit-framework/lib/rex/zip/archive.rb

# -*- coding: binary -*-

module Rex
module Zip

#
# This represents an entire archive.
#
class Archive

  # An array of the Entry objects stored in this Archive.
  attr_reader :entries


  def initialize(compmeth=CM_DEFLATE)
    @compmeth = compmeth
    @entries = []
  end


  #
  # Create a new Entry and add it to the archive.
  #
  # If fdata is set, the file is populated with that data
  # from the calling method. If fdata is nil, then the
  # fs is checked for the file.
  #
  def add_file(fname, fdata=nil, xtra=nil, comment=nil)
    if (not fdata)
      begin
        st = File.stat(fname)
      rescue
        return nil
      end

      ts = st.mtime
      if (st.directory?)
        attrs = EFA_ISDIR
        fdata = ''
        unless fname[-1,1] == '/'
          fname += '/'
        end
      else
        f = File.open(fname, 'rb')
        fdata = f.read(f.stat.size)
        f.close
      end
    end

    @entries << Entry.new(fname, fdata, @compmeth, ts, attrs, xtra, comment)
  end


  def set_comment(comment)
    @comment = comment
  end


  #
  # Write the compressed file to +fname+.
  #
  def save_to(fname)
    f = File.open(fname, 'wb')
    f.write(pack)
    f.close
  end


  #
  # Compress this archive and return the resulting zip file as a String.
  #
  def pack
    ret = ''

    # save the offests
    offsets = []

    # file 1 .. file n
    @entries.each { |ent|
      offsets << ret.length
      ret << ent.pack
    }

    # archive decryption header (unsupported)
    # archive extra data record (unsupported)

    # central directory
    cfd_offset = ret.length
    idx = 0
    @entries.each { |ent|
      cfd = CentralDir.new(ent, offsets[idx])
      ret << cfd.pack
      idx += 1
    }

    # zip64 end of central dir record (unsupported)
    # zip64 end of central dir locator (unsupported)

    # end of central directory record
    cur_offset = ret.length - cfd_offset
    ret << CentralDirEnd.new(@entries.length, cur_offset, cfd_offset, @comment).pack

    ret
  end

  def inspect
    "#<#{self.class} entries = [#{@entries.map{|e| e.name}.join(",")}]>"
  end

end

end
end