91 lines
2.2 KiB
Ruby
Executable File
91 lines
2.2 KiB
Ruby
Executable File
module Zip
|
|
class ExtraField < Hash
|
|
ID_MAP = {}
|
|
|
|
def initialize(binstr = nil)
|
|
binstr and merge(binstr)
|
|
end
|
|
|
|
def extra_field_type_exist(binstr, id, len, i)
|
|
field_name = ID_MAP[id].name
|
|
if self.member?(field_name)
|
|
self[field_name].merge(binstr[i, len + 4])
|
|
else
|
|
field_obj = ID_MAP[id].new(binstr[i, len + 4])
|
|
self[field_name] = field_obj
|
|
end
|
|
end
|
|
|
|
def extra_field_type_unknown(binstr, len, i)
|
|
create_unknown_item unless self['Unknown']
|
|
if !len || len + 4 > binstr[i..-1].bytesize
|
|
self['Unknown'] << binstr[i..-1]
|
|
return
|
|
end
|
|
self['Unknown'] << binstr[i, len + 4]
|
|
end
|
|
|
|
def create_unknown_item
|
|
s = ''
|
|
class << s
|
|
alias_method :to_c_dir_bin, :to_s
|
|
alias_method :to_local_bin, :to_s
|
|
end
|
|
self['Unknown'] = s
|
|
end
|
|
|
|
def merge(binstr)
|
|
return if binstr.empty?
|
|
i = 0
|
|
while i < binstr.bytesize
|
|
id = binstr[i, 2]
|
|
len = binstr[i + 2, 2].to_s.unpack('v').first
|
|
if id && ID_MAP.member?(id)
|
|
extra_field_type_exist(binstr, id, len, i)
|
|
elsif id
|
|
create_unknown_item unless self['Unknown']
|
|
break unless extra_field_type_unknown(binstr, len, i)
|
|
end
|
|
i += len + 4
|
|
end
|
|
end
|
|
|
|
def create(name)
|
|
unless field_class = ID_MAP.values.find { |k| k.name == name }
|
|
raise ZipError, "Unknown extra field '#{name}'"
|
|
end
|
|
self[name] = field_class.new
|
|
end
|
|
|
|
def to_local_bin
|
|
self.map { |_, v| v.to_local_bin }.join
|
|
end
|
|
|
|
alias :to_s :to_local_bin
|
|
|
|
def to_c_dir_bin
|
|
self.map { |_, v| v.to_c_dir_bin }.join
|
|
end
|
|
|
|
def c_dir_length
|
|
to_c_dir_bin.bytesize
|
|
end
|
|
|
|
def local_length
|
|
to_local_bin.bytesize
|
|
end
|
|
|
|
alias :c_dir_size :c_dir_length
|
|
alias :local_size :local_length
|
|
alias :length :local_length
|
|
alias :size :local_length
|
|
end
|
|
end
|
|
|
|
require 'zip/extra_field/generic'
|
|
require 'zip/extra_field/universal_time'
|
|
require 'zip/extra_field/unix'
|
|
# Copyright (C) 2002, 2003 Thomas Sondergaard
|
|
# rubyzip is free software; you can redistribute it and/or
|
|
# modify it under the terms of the ruby license.
|