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