elfcat 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 2af0dcf2cdde5ac558395bb31783939784fabbfb
-  data.tar.gz: 98f6a7933894897939d8f700ab8189c657ff0a00
+  metadata.gz: a9bfb99297cef606da82c0eee802dc08d6893e5e
+  data.tar.gz: a19611b558d5791421be87e8cb4ceb2b988b8792
 SHA512:
-  metadata.gz: 5b78661ce261c7b70ef2c6cc45f1e19bd9e419bf91fdcbb836e7763c39989b00423b422af612279e0f360fc13fb322eac56f32b8d61af86b2cdf95f2cecfc573
-  data.tar.gz: 64aa56a6cb44cec4f1a16113a3bc6214d7bbce22381ba8c4c3dc57b8151cde54711341dea520289d701d9cde6af45756496322b0bb3bd9f99886dc43e4d14daf
+  metadata.gz: b10fa9efc667e85b9ce04ecde07d1ea852bf2ca5a517907a80f7fa682dd8380e38d332a7e21c83fe77dc73cdaf2345ad3a93536a2d1ee45ea64bece1351bb13e
+  data.tar.gz: eb5376b246c450682bee140f92d962a7f7af587b3d5cad9a619ede519624c41efa0e8c0f9fe7e365fd968c1eb81a6da38b8c7967592de933b7daac3ecdcc9809

data/lib/elfcat.rb

@@ -5,6 +5,8 @@ require "elfcat/base"
 require "elfcat/fileheader"
 require "elfcat/sectiontable"
 require "elfcat/sectionname"
+require "elfcat/stringtable"
+require "elfcat/programheader"
 
 module Elfcat
 end

data/lib/elfcat/base.rb

@@ -13,4 +13,25 @@ class Base
   def [](_name)
     return @data[_name]
   end
+
+  def debug
+    raise RuntimeError, "Base::debug() is stubbed. Please override this method"
+  end
+
+  def parse_struct _struct, _resource, _element_count, _element_size
+    _element_count.times do |i|
+      st = Hash.new
+      struct_address = i * _element_size
+
+      _struct.each do |k, v|
+        st[k] = Util.concatenate(_resource, struct_address + v[0], v[1])
+      end
+
+      @data[i] = st.dup
+    end
+  end
+
+  def parse_slice _address, _length
+    return $resource.slice(_address, _length).split(/\x0/).dup
+  end
 end

data/lib/elfcat/elf.rb

@@ -13,6 +13,10 @@ class Elf
     return $resource
   end
 
+  def program_header
+    @program_header ||= ProgramHeader.new(file_header, section_table_with_names)
+  end
+
   def section_table
     @section_table ||= SectionTable.new(file_header)
   end
@@ -24,4 +28,8 @@ class Elf
   def section_table_with_names
     section_table.populate(section_names)
   end
+
+  def string_table
+    @string_table ||= StringTable.new(section_table_with_names)
+  end
 end

data/lib/elfcat/programheader.rb

@@ -0,0 +1,75 @@
+class ProgramHeader < Base
+  PH = {
+    p_type:      [0x00, 4],
+    p_offset:    [0x04, 4],
+    p_vaddr:     [0x08, 4],
+    p_paddr:     [0x0C, 4],
+    p_filesz:    [0x10, 4],
+    p_memsz:     [0x14, 4],
+    p_flags:     [0x18, 4],
+    p_align:     [0x1C, 4]
+  }
+
+  def initialize _fh, _st
+    super()
+    parse _fh
+    populate _st
+  end
+
+  def debug
+    @data.each_with_index do |(k, d), i|
+      print_debug_header if i % 64 == 0
+      index_s = sprintf("%4d", k)
+
+      real_name_s = sprintf("%-16.16s", d.real_name)
+
+      elf_address_s = CuteHex.x d.p_offset
+      elf_size_s = sprintf("%8d", d.p_filesz)
+
+      virtual_address_s = CuteHex.x d.p_vaddr
+      physical_address_s = CuteHex.x d.p_paddr
+      mem_size_s = sprintf("%8d", d.p_memsz)
+
+      puts "#{index_s} | #{real_name_s} #{elf_address_s} (#{elf_size_s}) | #{virtual_address_s} (#{mem_size_s})"
+    end
+  end
+
+private
+  def print_debug_header
+    puts "----------------------------------------------------------------------------"
+    puts " IDX | section_name     elf_addr    (  size B) | mem_addr    (  size B)"
+    puts "----------------------------------------------------------------------------"
+  end
+
+  def parse _fh
+    base_elf_address = _fh.e_phoff
+    struct_element_size = _fh.e_phentsize
+    struct_count = _fh.e_phnum
+    total_program_header_length = struct_count * struct_element_size
+
+    x = $resource.slice_with_index(base_elf_address, total_program_header_length)
+
+    parse_struct(PH, x, struct_count, struct_element_size)
+  end
+
+  def populate _st
+    @data.each do |k, d|
+      @data[k][:real_name] = Proc.new {
+        if d.p_filesz == 0 and d.memsz == 0
+          raise RuntimeError, 'Unexpected empty elf and mem'
+        elsif d.p_filesz > 0 and d.p_offset > 0
+          _st.addresses[d.p_vaddr]
+        elsif d.p_memsz > 0 and d.p_vaddr > 0
+          _st.addresses[d.p_offset]
+        # else
+        #   name_by_elf = _st.addresses[d.p_offset]
+        #   name_by_mem = _st.addresses[d.p_vaddr]
+
+        #   if d.p_vaddr > 0 and d.p_offset > 0
+        #     raise RuntimeError, "Mismatch elf and mem name #{name_by_elf} | #{name_by_mem}" if name_by_elf != name_by_mem
+        #   end
+        end
+      }.call
+    end
+  end
+end

data/lib/elfcat/sectionname.rb

@@ -19,6 +19,6 @@ private
     base_elf_address = entry.sh_offset
     length = entry.sh_size
 
-    @data = $resource.slice(base_elf_address, length).split(/\x0/).dup
+    @data = parse_slice(base_elf_address, length)
   end
 end

data/lib/elfcat/sectiontable.rb

@@ -1,4 +1,6 @@
 class SectionTable < Base
+  attr_reader :index, :addresses
+
   SH = {
     sh_name:       [0x00, 4],
     sh_type:       [0x04, 4],
@@ -99,6 +101,7 @@ class SectionTable < Base
   def initialize _fh
     super()
     @index = Hash.new
+    @addresses = Hash.new
 
     parse _fh
   end
@@ -106,11 +109,18 @@ class SectionTable < Base
   def populate _sn
     @data.each do |k, d|
       @data[k][:sh_real_name] = _sn[k]
+      @index[_sn[k]] = k
+      @addresses[d.sh_addr] = _sn[k]
+      @addresses[d.sh_offset] = _sn[k]
     end
 
     return self
   end
 
+  def get_by_index _n
+    return @data[@index[_n]]
+  end
+
   def debug
     @data.each_with_index do |(k, d), i|
       print_debug_header(d.sh_real_name) if i % 64 == 0
@@ -142,21 +152,12 @@ private
 
   def parse _fh
     base_elf_address = _fh.e_shoff
-    entry_element_size = _fh.e_shentsize
-    entry_count = _fh.e_shnum
-    section_table_length = entry_element_size * entry_count
+    struct_element_size = _fh.e_shentsize
+    struct_count = _fh.e_shnum
+    section_table_length = struct_element_size * struct_count
 
     x = $resource.slice_with_index(base_elf_address, section_table_length)
 
-    entry_count.times do |i|
-      st = Hash.new
-      struct_address = i * entry_element_size
-
-      SH.each do |k, v|
-        st[k] = Util.concatenate(x, struct_address + v[0], v[1])
-      end
-
-      @data[i] = st.dup
-    end
+    parse_struct(SH, x, struct_count, struct_element_size)
   end
 end

data/lib/elfcat/stringtable.rb

@@ -0,0 +1,24 @@
+class StringTable < Base
+  def initialize _st
+    super()
+    parse _st
+  end
+
+  def debug
+    @data.each_with_index do |x, i|
+      index_s = sprintf("%4d", i)
+      name_s = sprintf("%s", x)
+
+      puts "#{index_s} | #{name_s}"
+    end
+  end
+
+private
+  def parse _st
+    entry = _st.get_by_index('.strtab')
+    base_elf_address = entry.sh_offset
+    length = entry.sh_size
+
+    @data = parse_slice(base_elf_address, length)
+  end
+end

data/lib/elfcat/version.rb

@@ -1,3 +1,3 @@
 module Elfcat
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: elfcat
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Gloria Budiman
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-02-23 00:00:00.000000000 Z
+date: 2016-02-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: cute-hex
@@ -88,8 +88,10 @@ files:
 - lib/elfcat/base.rb
 - lib/elfcat/elf.rb
 - lib/elfcat/fileheader.rb
+- lib/elfcat/programheader.rb
 - lib/elfcat/sectionname.rb
 - lib/elfcat/sectiontable.rb
+- lib/elfcat/stringtable.rb
 - lib/elfcat/util.rb
 - lib/elfcat/version.rb
 homepage: https://github.com/gbudiman/elfcat