exerb 6.0.1

data/lib/exerb/resource.rb

@@ -0,0 +1,190 @@
+
+#==============================================================================#
+# $Id: resource.rb,v 1.30 2005/05/05 02:45:41 yuya Exp $
+#==============================================================================#
+
+require 'stringio'
+require 'exerb/error'
+require 'exerb/win32/resource_directory'
+require 'exerb/win32/resource_directory_root'
+require 'exerb/win32/resource_entry'
+require 'exerb/win32/pe_file'
+require 'exerb/win32/const/resource'
+require 'exerb/resource/icon'
+require 'exerb/resource/group_icon'
+require 'exerb/resource/version_info'
+require 'exerb/resource/dialog'
+require 'exerb/resource/binary'
+
+#==============================================================================#
+
+module Exerb
+end # Exerb
+
+#==============================================================================#
+
+class Exerb::Resource
+
+  include Enumerable
+
+  DEFAULT_LANG_ID = 0x0400 # Neutral
+  RT_EXERB = 100
+  ID_EXERB = 1
+
+  def initialize
+    @entries = {}
+  end
+
+  attr_reader :entries
+
+  def self.read(io, base)
+    return self.new.read(io, base)
+  end
+
+  def self.new_from_binary(bin, base)
+    return self.read(StringIO.new(bin), base)
+  end
+
+  def self.new_from_pe_binary(bin)
+    pe     = Exerb::Win32::PeFile.new_from_binary(bin)
+    rsrc   = pe.sections.find { |sec| sec.name == '.rsrc' }
+    raise(ExerbError, "a resource section was not found in the core") if rsrc.nil?
+
+    return self.new_from_binary(bin[rsrc.pointer_to_raw_data, rsrc.virtual_size], rsrc.virtual_address)
+  end
+
+  def self.new_from_pe_file(filepath)
+    return File.open(filepath) { |file| self.new_from_pe_binary(file.read) }
+  end
+
+  def each
+    @entries.keys.sort.each { |type|
+      @entries[type].keys.sort.each { |id|
+        @entries[type][id].keys.sort.each { |lang|
+          yield(@entries[type][id][lang])
+        }
+      }
+    }
+  end
+
+  def add(type, id, data, lang = DEFAULT_LANG_ID)
+    @entries[type] ||= {}
+    @entries[type][id] ||= {}
+    @entries[type][id][lang] = Exerb::Resource::Entry.new(type, id, lang, data)
+
+    return self
+  end
+
+  def add_icon(id, data, lang = DEFAULT_LANG_ID)
+    return self.add(Exerb::Win32::Const::RT_ICON, id, data, lang)
+  end
+
+  def add_group_icon(id, data, lang = DEFAULT_LANG_ID)
+    return self.add(Exerb::Win32::Const::RT_GROUP_ICON, id, data, lang)
+  end
+
+  def add_dialog(id, data, lang = DEFAULT_LANG_ID)
+    return self.add(Exerb::Win32::Const::RT_DIALOG, id, data, lang)
+  end
+
+  def add_version(id, data, lang = DEFAULT_LANG_ID)
+    return self.add(Exerb::Win32::Const::RT_VERSION, id, data, lang)
+  end
+
+  def add_archive(archive)
+    return self.add(Exerb::Resource::RT_EXERB, Exerb::Resource::ID_EXERB, archive)
+  end
+
+  def pack(base, reloc = [])
+    root_dir = Exerb::Win32::ResourceDirectoryRoot.new
+
+    @entries.keys.sort.each { |type|
+      root_dir << Exerb::Win32::ResourceDirectory.new(type) { |type_dir|
+        @entries[type].keys.sort.each { |id|
+          type_dir << Exerb::Win32::ResourceDirectory.new(id) { |item_dir|
+            @entries[type][id].keys.sort.each { |lang|
+              item_dir << @entries[type][id][lang].to_resource_entry
+            }
+          }
+        }
+      }
+    }
+
+    return root_dir.pack_all(base, reloc)
+  end
+
+  def read(io, base)
+    root_dir = Exerb::Win32::ResourceDirectoryRoot.read(io, base)
+    root_dir.entries.each { |type_dir|
+      type_dir.entries.each { |item_dir|
+        item_dir.entries.each { |item|
+          type, id, lang = type_dir.name, item_dir.name, item.lang
+          @entries[type] ||= {}
+          @entries[type][id] ||= {}
+          @entries[type][id][lang] = Exerb::Resource::Entry.new(type, id, lang, Exerb::Resource::Binary.new(item.entry_data.data))
+        }
+      }
+    }
+
+    return self
+  end
+
+  def remove(type, id = nil, lang = nil)
+    if type && id && lang
+      if @entries[type] && @entries[type][id] && @entries[type][id][lang]
+        @entries[type][id].delete(lang)
+      end
+    elsif type && id
+      if @entries[type] && @entries[type][id]
+        @entries[type].delete(id)
+      end
+    elsif type
+      if @entries[type]
+        @entries.delete(type)
+      end
+    end
+
+    if type && id && @entries[type] && @entries[type][id] && @entries[type][id].empty?
+      @entries[type].delete(id)
+    end
+
+    if type && @entries[type] && @entries[type].empty?
+      @entries.delete(type)
+    end
+
+    return self
+  end
+
+  def merge(res)
+    res.each { |entry|
+      @entries[entry.type] ||= {}
+      @entries[entry.type][entry.id] ||= {}
+      @entries[entry.type][entry.id][entry.lang] = entry
+    }
+
+    return self
+  end
+
+end # Exerb::Resource
+
+#==============================================================================#
+
+class Exerb::Resource::Entry
+
+  def initialize(type, id, lang, data)
+    @type = type
+    @id   = id
+    @lang = lang
+    @data = data
+  end
+
+  attr_accessor :type, :id, :lang, :data
+
+  def to_resource_entry
+    return Exerb::Win32::ResourceEntry.new(@data.pack, lang)
+  end
+
+end # Exerb::Resource::Entry
+
+#==============================================================================#
+#==============================================================================#

data/lib/exerb/resource/base.rb

@@ -0,0 +1,29 @@
+
+#==============================================================================#
+# $Id: base.rb,v 1.2 2005/04/30 15:16:04 yuya Exp $
+#==============================================================================#
+
+module Exerb
+  class Resource
+  end # Resource
+end # Exerb
+
+#==============================================================================#
+
+class Exerb::Resource::Base
+
+  def initialize
+  end
+
+  def pack
+    raise NotImplementedError
+  end
+
+  def size
+    return self.pack.size
+  end
+
+end # Exerb::Resource::Base
+
+#==============================================================================#
+#==============================================================================#

data/lib/exerb/resource/binary.rb

@@ -0,0 +1,31 @@
+
+#==============================================================================#
+# $Id: binary.rb,v 1.2 2005/04/30 15:37:02 yuya Exp $
+#==============================================================================#
+
+require 'exerb/resource/base'
+
+#==============================================================================#
+
+module Exerb
+  class Resource
+  end # Resource
+end # Exerb
+
+#==============================================================================#
+
+class Exerb::Resource::Binary < Exerb::Resource::Base
+
+  def initialize(bin)
+    super()
+    @bin = bin
+  end
+
+  def pack
+    return @bin
+  end
+
+end # Exerb::Resource::Binary
+
+#==============================================================================#
+#==============================================================================#

data/lib/exerb/resource/dialog.rb

@@ -0,0 +1,30 @@
+
+#==============================================================================#
+# $Id: dialog.rb,v 1.4 2005/04/30 15:37:02 yuya Exp $
+#==============================================================================#
+
+require 'exerb/resource/base'
+
+#==============================================================================#
+
+module Exerb
+  class Resource
+  end # Resource
+end # Exerb
+
+#==============================================================================#
+
+class Exerb::Resource::Dialog < Exerb::Resource::Base
+
+  def initialize
+    super()
+  end
+
+  def pack
+    raise(NotImplementedError)
+  end
+
+end # Exerb::Resource::Dialog
+
+#==============================================================================#
+#==============================================================================#

data/lib/exerb/resource/group_icon.rb

@@ -0,0 +1,85 @@
+
+#==============================================================================#
+# $Id: group_icon.rb,v 1.9 2010/06/26 03:13:36 arton Exp $
+#==============================================================================#
+
+require 'exerb/win32/struct/icon_header'
+require 'exerb/win32/struct/icon_res_entry'
+require 'exerb/resource/base'
+
+#==============================================================================#
+
+module Exerb
+  class Resource
+  end # Resource
+end # Exerb
+
+#==============================================================================#
+
+class Exerb::Resource::GroupIcon < Exerb::Resource::Base
+
+  def initialize
+    super()
+    @entries = []
+  end
+
+  attr_reader :entries
+
+  def add(id, icon)
+    @entries << Entry.new(id, icon.width, icon.height, icon.bit_count, icon.size)
+    return self
+  end
+
+  def pack
+    header = Exerb::Win32::Struct::IconHeader.new
+    header.reserved = 0
+    header.type     = 1
+    header.count    = @entries.size
+
+    packed_header  = header.pack
+    packed_entries = @entries.collect { |entry| entry.pack }.join
+
+    return packed_header + packed_entries
+  end
+
+  class Entry
+
+    def initialize(id, width, height, bit_count, size)
+      @id        = id
+      @widht     = width
+      @height    = height
+      @bit_count = bit_count
+      @size      = size
+    end
+
+    attr_accessor :id, :width, :height, :bit_count, :size
+
+    def pack
+      case @bit_count
+      when 1 then cc, bc = 2, 1
+      when 4 then cc, bc = 16, 4
+      when 8 then cc, bc =  0, 8
+      when 24 then cc, bc = 0, 24
+      when 32 then cc, bc =  0, 32
+      else raise "invalid bit count -- #{@bit_count}"
+      end
+
+      icon_dir_entry = Exerb::Win32::Struct::IconResEntry.new
+      icon_dir_entry.width        = @widht
+      icon_dir_entry.height       = @height
+      icon_dir_entry.color_count  = cc
+      icon_dir_entry.reserved     = 0
+      icon_dir_entry.planes       = 1
+      icon_dir_entry.bit_count    = bc
+      icon_dir_entry.bytes_in_res = @size
+      icon_dir_entry.image_offset = @id
+
+      return icon_dir_entry.pack
+    end
+
+  end # Entry
+
+end # Exerb::Resource::GroupIcon
+
+#==============================================================================#
+#==============================================================================#

data/lib/exerb/resource/icon.rb

@@ -0,0 +1,52 @@
+
+#==============================================================================#
+# $Id: icon.rb,v 1.6 2005/05/05 02:26:29 yuya Exp $
+#==============================================================================#
+
+require 'exerb/error'
+require 'exerb/win32/icon_file'
+require 'exerb/resource/base'
+
+#==============================================================================#
+
+module Exerb
+  class Resource
+  end # Resource
+end # Exerb
+
+#==============================================================================#
+
+class Exerb::Resource::Icon < Exerb::Resource::Base
+
+  def initialize
+    super()
+    @width     = nil
+    @height    = nil
+    @bit_count = nil
+    @value     = nil
+  end
+
+  attr_accessor :width, :height, :bit_count, :value
+
+  def self.read(input, width, height, bit_count)
+    icon_file  = Exerb::Win32::IconFile.read(input)
+    icon_entry = icon_file.entries.find { |entry| entry.width == width && entry.height == height && entry.bit_count == bit_count }
+    raise(Exerb::ExerbError, "no such icon #{width}x#{height}x#{bit_count} in #{input}") unless icon_entry
+
+    icon = self.new
+    icon.width     = icon_entry.width
+    icon.height    = icon_entry.height
+    icon.bit_count = icon_entry.bit_count
+    icon.value     = icon_entry.value
+
+    return icon
+  end
+
+  def pack
+    return @value
+  end
+
+end # Exerb::Resource::Icon
+
+#==============================================================================#
+#==============================================================================#

data/lib/exerb/resource/version_info.rb

@@ -0,0 +1,155 @@
+
+#==============================================================================#
+# $Id: version_info.rb,v 1.15 2006/06/02 00:47:54 yuya Exp $
+#==============================================================================#
+
+require 'exerb/utility'
+require 'exerb/resource/base'
+require 'exerb/win32/const/resource'
+require 'exerb/win32/struct/vs_fixed_file_info'
+require 'exerb/win32/struct/version_info_block'
+
+#==============================================================================#
+
+module Exerb
+  class Resource
+  end # Resource
+end # Exerb
+
+#==============================================================================#
+
+class Exerb::Resource::VersionInfo < Exerb::Resource::Base
+
+  def initialize
+    @file_version_number    = 0
+    @product_version_number = 0
+    @comments               = ""
+    @company_name           = ""
+    @legal_copyright        = ""
+    @legal_trademarks       = ""
+    @file_version           = ""
+    @product_version        = ""
+    @product_name           = ""
+    @file_description       = ""
+    @internal_name          = ""
+    @original_filename      = ""
+    @private_build          = ""
+    @special_build          = ""
+
+    @is_dll = false
+  end
+
+  attr_accessor :file_version_number, :product_version_number, :comments, :company_name, :legal_copyright, :legal_trademarks, :file_version, :product_version, :product_name, :file_description, :internal_name, :original_filename, :private_build, :special_build, :is_dll
+
+  def self.make_version(n1, n2, n3, n4)
+    version  = (n1 & 0xFFFF) << 48
+    version |= (n2 & 0xFFFF) << 32
+    version |= (n3 & 0xFFFF) << 16
+    version |= (n4 & 0xFFFF)
+
+    return version
+  end
+
+  def pack
+    vsffi = Exerb::Win32::Struct::VsFixedFileInfo.new
+    vsffi.signature          = Exerb::Win32::Const::VS_FFI_SIGNATURE
+    vsffi.struct_version     = Exerb::Win32::Const::VS_FFI_STRUCVERSION
+    vsffi.file_version_ms    = @file_version_number >> 32
+    vsffi.file_version_ls    = @file_version_number & 0xFFFFFFFF
+    vsffi.product_version_ms = @product_version_number >> 32
+    vsffi.product_version_ls = @product_version_number & 0xFFFFFFFF
+    vsffi.file_flags_mask    = Exerb::Win32::Const::VS_FFI_FILEFLAGSMASK
+    vsffi.file_flags         = 0x00000000
+    vsffi.file_os            = Exerb::Win32::Const::VOS_NT_WINDOWS32
+    vsffi.file_type          = (@is_dll ? Exerb::Win32::Const::VFT_DLL : Exerb::Win32::Const::VFT_APP)
+    vsffi.file_subtype       = Exerb::Win32::Const::VFT2_UNKNOWN
+    vsffi.file_date_ms       = 0x00000000
+    vsffi.file_date_ls       = 0x00000000
+
+    block = BlockType0.new('VS_VERSION_INFO', vsffi.pack) { |vvi|
+      vvi << BlockType1.new('StringFileInfo') { |sfi|
+        # Language:Neutral CodePage:Unicode
+        sfi << BlockType1.new('000004b0') { |neutral|
+          neutral << BlockType1String.new('Comments',         @comments)
+          neutral << BlockType1String.new('CompanyName',      @company_name)
+          neutral << BlockType1String.new('LegalCopyright',   @legal_copyright)
+          neutral << BlockType1String.new('LegalTrademarks',  @legal_trademarks)
+          neutral << BlockType1String.new('FileVersion',      @file_version)
+          neutral << BlockType1String.new('ProductVersion',   @product_version)
+          neutral << BlockType1String.new('ProductName',      @product_name)
+          neutral << BlockType1String.new('FileDescription',  @file_description)
+          neutral << BlockType1String.new('InternalName',     @internal_name)
+          neutral << BlockType1String.new('OriginalFilename', @original_filename)
+          neutral << BlockType1String.new('PrivateBuild',     @private_build)
+          neutral << BlockType1String.new('SpecialBuild',     @special_build)
+        }
+      }
+      vvi << BlockType1.new('VarFileInfo') { |vfi|
+        vfi << BlockType0.new('Translation', [0x04B00000].pack('L'))
+      }
+    }
+
+    return block.pack
+  end
+
+  class BlockBase
+
+    def initialize(type, key, value = '', unicode = false)
+      @type     = type
+      @key      = key
+      @value    = value
+      @unicode  = unicode
+      @children = []
+
+      yield(self) if block_given?
+    end
+
+    attr_accessor :type, :key, :value, :unicode, :children
+
+    def <<(child)
+      @children << child
+    end
+
+    def pack
+      packed_children = @children.collect { |child| child.pack }.join
+
+      block = Exerb::Win32::Struct::VersionInfoBlock.new
+      block.type         = @type
+      block.key          = @key.unpack("U*").pack("v*") # UTF-8 to UTF-16LE
+      block.data         = @value
+      block.value_length = (@unicode ? @value.size / 2 : @value.size)
+      block.total_length = block.pack.size + packed_children.size
+
+      return Exerb::Utility.alignment(block.pack, 4) + packed_children
+    end
+
+  end # Block
+
+  class BlockType0 < BlockBase
+
+    def initialize(key, value = '', unicode = false)
+      super(0x0000, key, value, unicode)
+    end
+
+  end # BlockType0
+
+  class BlockType1 < BlockBase
+
+    def initialize(key, value = '', unicode = false)
+      super(0x0001, key, value, unicode)
+    end
+
+  end # BlockType1
+
+  class BlockType1String < BlockType1
+
+    def initialize(key, value)
+      super(key, value.to_s.unpack("U*").pack("v*") + "\0\0", true) # UTF-8 to UTF-16LE
+    end
+
+  end # BlockType1String
+
+end # Exerb::Resource::VersionInfo
+
+#==============================================================================#
+#==============================================================================#