114 lines
2.1 KiB
Ruby
114 lines
2.1 KiB
Ruby
# -*- 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
|