orgasm 0.0.1a2 → 0.0.1a3

data/lib/orgasm.rb

@@ -17,6 +17,12 @@
 # along with orgasm. If not, see <http://www.gnu.org/licenses/>.
 #++
 
-module Orgasm
+require 'orgasm/base'
 
-end
+require 'orgasm/piece'
+require 'orgasm/styles'
+require 'orgasm/disassembler'
+require 'orgasm/generator'
+require 'orgasm/assembler'
+
+require 'orgasm/architecture'

data/lib/orgasm/arch/i386.rb

@@ -0,0 +1,27 @@
+#--
+# Copyleft meh. [http://meh.paranoid.pk | meh@paranoici.org]
+#
+# This file is part of orgasm.
+#
+# orgasm is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# orgasm is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with orgasm. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'orgasm/arch/i386/base'
+
+Orgasm::Architecture.for 'i386' do
+  instructions 'orgasm/arch/i386/instructions'
+  disassembler 'orgasm/arch/i386/disassembler'
+  generator    'orgasm/arch/i386/generator'
+  styles       'orgasm/arch/i386/styles'
+end

data/lib/orgasm/{style.rb → arch/i386/base.rb}

@@ -17,3 +17,7 @@
 # along with orgasm. If not, see <http://www.gnu.org/licenses/>.
 #++
 
+require 'orgasm/arch/i386/base/instruction'
+require 'orgasm/arch/i386/base/address'
+require 'orgasm/arch/i386/base/register'
+require 'orgasm/arch/i386/base/immediate'

data/lib/orgasm/{common → arch/i386/base}/address.rb

@@ -17,33 +17,26 @@
 # along with orgasm. If not, see <http://www.gnu.org/licenses/>.
 #++
 
-module Orgasm
+module Orgasm; module I386
 
-class Address
-  attr_reader :start
-
-  def initialize (value, offset=nil)
-    if offset
-      @start = value
-      @value = offset.to_i
+class Address < Orgasm::Address
+  def initialize (value=nil, options={})
+    if value.respond_to? :to_i
+      super(value)
     else
-      @value = value.to_i
+      super()
     end
 
-    yield self if block_given?
-  end
-
-  def offset?
-    !!start
+    @options = options
   end
 
-  def to_i
-    @value
+  def relative?
+    !!@options[:relative]
   end
 
-  def to_s
-    offset? ? "[#{start}+#{to_i}]" : "0x%x" % to_i
+  def offset?
+    !!@options[:offset]
   end
 end
 
-end
+end; end

data/lib/orgasm/{common/unknown.rb → arch/i386/base/immediate.rb}

@@ -17,18 +17,14 @@
 # along with orgasm. If not, see <http://www.gnu.org/licenses/>.
 #++
 
-module Orgasm
+module Orgasm; module I386
 
-class Unknown
-  def initialize (size)
-    @size = size.to_i
+class Immediate < Orgasm::Constant
+  attr_accessor :size
 
-    yield self if block_given?
-  end
-
-  def to_i
-    @size
+  def initialize (value=nil, size=nil)
+    super(value)
   end
 end
 
-end
+end; end

data/lib/orgasm/arch/i386/base/instruction.rb

@@ -0,0 +1,41 @@
+#--
+# Copyleft meh. [http://meh.paranoid.pk | meh@paranoici.org]
+#
+# This file is part of orgasm.
+#
+# orgasm is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# orgasm is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with orgasm. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+module Orgasm; module I386
+
+class Instruction < Orgasm::Instruction
+  extend Forwardable
+
+  def_delegator :@parameters, :first, :destination
+  def_delegator :@parameters, :last,  :source
+
+  def initialize (name=nil, destination=nil, source=nil)
+    super(name, destination, source)
+  end
+
+  def destination= (value)
+    parameters[0] = value
+  end
+
+  def source= (value)
+    parameters[1] = value
+  end
+end
+
+end; end

data/lib/orgasm/arch/i386/base/register.rb

@@ -0,0 +1,40 @@
+#--
+# Copyleft meh. [http://meh.paranoid.pk | meh@paranoici.org]
+#
+# This file is part of orgasm.
+#
+# orgasm is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# orgasm is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with orgasm. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+module Orgasm; module I386
+
+class Register < Orgasm::Register
+  attr_accessor :size
+
+  def initialize (name=nil)
+    super(name, Architecture.i386.instructions.register?(name))
+  end
+
+  def name= (value)
+    value = value.to_s.downcase.to_sym
+
+    unless Architecture.i386.instructions.register?(value)
+      raise ArgumentError, "#{value} isn't a valid i386 register"
+    end
+
+    @name = value
+  end
+end
+
+end; end

data/lib/orgasm/arch/i386/disassembler.rb

@@ -17,162 +17,34 @@
 # along with orgasm. If not, see <http://www.gnu.org/licenses/>.
 #++
 
-module Orgasm
-
-Disassembler.for('i386') {
-  reg = registers = Class.new(Hash) {
-    def initialize
-      merge!(
-        32 => {
-          EAX: 0x0,
-          ECX: 0x1,
-          EDX: 0x2,
-          EBX: 0x3,
-          ESP: 0x4,
-          EBP: 0x5,
-          ESI: 0x6,
-          EDI: 0x7
-        },
-
-        16 => {
-          AX: 0x0,
-          CX: 0x1,
-          DX: 0x2,
-          BX: 0x3,
-          SP: 0x4,
-          BP: 0x5,
-          SI: 0x6,
-          DI: 0x7
-        },
-
-        8 => {
-          AL: 0x0,
-          CL: 0x1,
-          DL: 0x2,
-          BL: 0x3,
-          AH: 0x4,
-          CH: 0x5,
-          DH: 0x6,
-          BH: 0x7
-        }
-      )
-    end
-
-    def source (byte, bits=32)
-      self[bits].key((byte & 0x38) >> 3)
-    end
-
-    def destination (byte, bits=32)
-      self[bits].key(byte & 0x07)
-    end; alias dest destination
-  }.new
-
-  on ?\x01, ?\x09, ?\x11, ?\x19, ?\x21, ?\x25, ?\x29, ?\x31, ?\x39, ?\x85, ?\x86, ?\x87, ?\x89, ?\xA1, ?\xA3 do
-    increment = 1
-
-    seek 1 do
-      read 1 do |data|
-        increment += 1 if data.to_byte & 0x07 == reg[32][:ESP]
-        increment += 1 if (data.to_byte & 0xC0) >> 6 == 0x01
-
-        if (data.to_byte & 0xC0) >> 6 == 0x10
-          Unknown.new(1)
-        end
-      end
-    end
-
-    on ?\x01 do
-      Instruction.new(:add) {|i|
-        seek +1
-
-        read 1 do |data|
-          i.parameters << Register.new(reg.source(data.to_byte), 32)
-          i.parameters << Register.new(reg.destination(data.to_byte), 32)
-        end
-
-        seek increment
-      }
-    end
-
-    on ?\x09 do
-      Instruction.new(:or) {
-        seek +1
-      }
-    end
-
-    on ?\x11 do
-      Instruction.new(:adc) {
-        seek +1
-      }
-    end
-
-    on ?\x19 do
-      Instruction.new(:sbb) {
-        seek +1
-      }
-    end
-
-    on ?\x21, ?\x25 do
-      Instruction.new(:ad) {
-        seek +1
-      }
-    end
-
-    on ?\x29 do
-      Instruction.new(:sub) {
-        seek +1
-      }
-    end
-
-    on ?\x31 do
-      Instruction.new(:xor) {
-        seek +1
-      }
-    end
-
-    on ?\x19 do
-      Instruction.new(:cmp) {
-        seek +1
-      }
-    end
-
-    on ?\x85 do
-      Instruction.new(:test) {
-        seek +1
-      }
-    end
-
-    on ?\x86 do
-      Instruction.new(:xchg) {
-        seek +1
-
-        # 8bit
-      }
-    end
-
-    on ?\x87 do
-      Instruction.new(:xchg) {
-        seek +1
-      }
-    end
-
-    on ?\x89 do
-      Instruction.new(:mov) {
-        seek +1
-
-        read 1 do |data|
-          increment = 5 if data.to_byte & 0x07 == 0x05 && data.to_byte < 0x40
+instructions.to_hash.each {|name, description|
+  description.each {|description|
+    if description.is_a?(Hash)
+      description.each {|params, opcodes|
+        opcodes = opcodes.clone
+        known   = opcodes.reverse.drop_while {|x| !x.is_a?(Integer)}.reverse.map {|x| x.chr}.join
+        opcodes.slice! known.length
+
+        on known do |whole, which|
+          seek which.length do
+            if opcodes.first.is_a?(String)
+              check = opcodes.shift.to_i
+
+              read 1 do |data|
+                skip unless ((data.to_byte & '00111000'.to_i(2)) >> 3) == check
+              end
+
+              opcodes.shift
+            end
+          end
         end
       }
-    end
+    else
+      on description.map {|b| b.chr}.join do |whole, which|
+        seek which.length
 
-    on ?\xA1, ?\xA3 do
-      # increment = 4
-      Instruction.new(:mov) {
-        seek +1
-      }
+        I386::Instruction.new(name)
+      end
     end
-  end
+  }
 }
-
-end

data/lib/orgasm/arch/i386/generator.rb

@@ -0,0 +1,44 @@
+#--
+# Copyleft meh. [http://meh.paranoid.pk | meh@paranoici.org]
+#
+# This file is part of orgasm.
+#
+# orgasm is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# orgasm is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with orgasm. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+instructions.registers.each {|register|
+  define_singleton_method register do
+    register
+  end
+}
+
+generator.for I386::Instruction do |name, &block|
+  I386::Instruction.new(name, &block)
+end
+
+generator.for I386::Register do |name|
+  I386::Register.new(name)
+end
+
+generator.for I386::Address do |data|
+  if data.is_a?(Array)
+    Address.new(data)
+  else
+    Address.new(data)
+  end
+end
+
+generator.for I386::Immediate do |data|
+  I386::Immediate.new(data, 32)
+end

data/lib/orgasm/arch/i386/instructions.rb

@@ -0,0 +1,150 @@
+#--
+# Copyleft meh. [http://meh.paranoid.pk | meh@paranoici.org]
+#
+# This file is part of orgasm.
+#
+# orgasm is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# orgasm is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with orgasm. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'orgasm/arch/i386/instructions/dsl'
+require 'orgasm/arch/i386/instructions/instructions'
+
+I386::Instructions[I386::DSL.new {
+  # ASCII Adjust After Addition
+  AAA [0x37]
+
+  # ASCII Adjust AX Before Division
+  AAD [0xD5, 0x0A],
+      [imm8] => [0xD5, ib]
+
+  # ASCII Adjust AX After Multiply
+  AAM [0xD4, 0x0A],
+      [imm8] => [0xD4, ib]
+
+  # ASCII Adjust AL After Substraction
+  AAS [0x3F]
+
+  # Add with Carry
+  ADC [al,      imm8]    => [0x14, ib],
+      [ax,      imm16]   => [0x15, iw],
+      [eax,     imm32]   => [0x15, id],
+      [r8|m8,   imm8]    => [0x80, ?2, ib],
+      [r16|m16, imm16]   => [0x81, ?2, iw],
+      [r32|m32, imm32]   => [0x81, ?2, id],
+      [r16|m16, imm8]    => [0x83, ?2, ib],
+      [r32|m32, imm8]    => [0x83, ?2, ib],
+      [r8|m8,   r8]      => [0x10, r],
+      [r16|m16, r16]     => [0x11, r],
+      [r32|m32, r32]     => [0x11, r],
+      [r8,      r8|m8]   => [0x12, r],
+      [r16,     r16|m16] => [0x13, r],
+      [r32,     r32|m32] => [0x13, r]
+
+  # Add
+  ADD [al,      imm8]    => [0x04, ib],
+      [ax,      imm16]   => [0x05, iw],
+      [eax,     imm32]   => [0x05, id],
+      [r8|m8,   imm8]    => [0x80, ?0, ib],
+      [r16|m16, imm16]   => [0x81, ?0, iw],
+      [r32|m32, imm32]   => [0x81, ?0, id],
+      [r16|m16, imm8]    => [0x83, ?0, ib],
+      [r32|m32, imm8]    => [0x83, ?0, ib],
+      [r8|m8,   r8]      => [0x00, r],
+      [r16|m16, r16]     => [0x01, r],
+      [r32|m32, r32]     => [0x01, r],
+      [r8,      r8|m8]   => [0x02, r],
+      [r16,     r16|m16] => [0x03, r],
+      [r32,     r32|m32] => [0x03, r]
+
+  # Logical AND
+  AND [al,      imm8]    => [0x24, ib],
+      [ax,      imm16]   => [0x25, iw],
+      [eax,     imm32]   => [0x25, id],
+      [r8|m8,   imm8]    => [0x80, ?4, ib],
+      [r16|m16, imm16]   => [0x81, ?4, iw],
+      [r32|m32, imm32]   => [0x81, ?4, id],
+      [r16|m16, imm8]    => [0x83, ?4, ib],
+      [r32|m32, imm8]    => [0x83, ?4, ib],
+      [r8|m8,   r8]      => [0x20, r],
+      [r16|m16, r16]     => [0x21, r],
+      [r32|m32, r32]     => [0x21, r],
+      [r8,      r8|m8]   => [0x22, r],
+      [r16,     r16|m16] => [0x23, r],
+      [r32,     r32|m32] => [0x23, r]
+
+  # Adjust RPL Field of Segment Selector
+  ARPL [r16|m16, r16] => [0x63, r]
+
+  # Check Array Index Against Bounds
+  BOUND [r16, m16&16] => [0x62, r],
+        [r32, m32&32] => [0x62, r]
+
+  # Bit Scan Forward
+  # BFS [r16, r16|m16] => [0x0F, 0xBC],
+  #     [r32, r32|m32] => [0x0F, 0xBC]
+  # TODO: find out what the fuck is this
+  
+  # Bit Scan Reverse
+  # BSR [r16, r16|m16] => [0x0F, 0xBD],
+  #     [r32, r32|m32] => [0x0F, 0xBD]
+  # TODO: find out what the fuck is this
+
+  # Byte Swap
+  BSWAP [r32] => [0x0F, 0xC8, rd]
+  # FIXME: not available on i386, only i486+
+
+  # Bit Test
+  BT [r16|m16, r16]  => [0x0F, 0xA3],
+     [r32|m32, r32]  => [0x0F, 0xA3],
+     [r16|m16, imm8] => [0x0F, 0xBA, ?4, ib],
+     [r32|m32, imm8] => [0x0F, 0xBA, ?4, ib]
+
+  # Bit Test and Complement
+  BTC [r16|m16, r16]  => [0x0F, 0xBB],
+      [r32|m32, r32]  => [0x0F, 0xBB],
+      [r16|m16, imm8] => [0x0F, 0xBA, ?7, ib],
+      [r32|m32, imm8] => [0x0F, 0xBA, ?7, ib]
+
+  # Bit Test and Reset
+  BTR [r16|m16, r16]  => [0x0F, 0xB3],
+      [r32|m32, r32]  => [0x0F, 0xB3],
+      [r16|m16, imm8] => [0x0F, 0xBA, ?6, ib],
+      [r32|m32, imm8] => [0x0F, 0xBA, ?6, ib]
+
+  # Call Procedure
+  CALL [rel16]    => [0xE8, cw],
+       [rel32]    => [0xE8, cd],
+       [r16|m16]  => [0xFF, ?2],
+       [r32|m32]  => [0xFF, ?2],
+       [ptr16^16] => [0x9A, cd],
+       [ptr16^32] => [0x9A, cp],
+       [m16^16]   => [0xFF, ?3],
+       [m16^32]   => [0xFF, ?3]
+        
+
+  # -- x87 FPU --
+
+  # Packed Single-FP Add
+  ADDPS [xmm1, xmm2|m128] => [0x0F, 0x58, r]
+
+  # Scalar Single-FP Add
+  ADDSS [xmm1, xmm2|m32] => [0xF3, 0x0F, 0x58, r]
+
+  # Bit-wise Logical And ot For Single-FP
+  ANDNPS [xmm1, xmm2|m128] => [0x0F, 0x55, r]
+
+  # Bit-wise Logical And For Single FP
+  ANDPS [xmm1, xmm2|m128] => [0x0F, 0x54, r]
+
+}.to_hash]