David Maloney
3742d9b687
because we pulled in the new metasm repo, retabbed everything to avoid whitespace conflicts
107 lines
2.3 KiB
Ruby
107 lines
2.3 KiB
Ruby
# This file is part of Metasm, the Ruby assembly manipulation suite
|
|
# Copyright (C) 2006-2009 Yoann GUILLOT
|
|
#
|
|
# Licence is LGPL, see LICENCE in the top-level directory
|
|
|
|
|
|
require 'metasm/main'
|
|
|
|
module Metasm
|
|
|
|
# a Renderable element has a method #render that returns an array of [String or Renderable]
|
|
module Renderable
|
|
def to_s
|
|
render.join
|
|
end
|
|
|
|
# yields each Expr seen in #render (recursive)
|
|
def each_expr
|
|
r = proc { |e|
|
|
case e
|
|
when Expression
|
|
r[e.lexpr] ; r[e.rexpr]
|
|
yield e
|
|
when ExpressionType
|
|
yield e
|
|
when Renderable
|
|
e.render.each { |re| r[re] }
|
|
end
|
|
}
|
|
r[self]
|
|
end
|
|
end
|
|
|
|
|
|
class Instruction
|
|
include Renderable
|
|
def render
|
|
@cpu.render_instruction(self)
|
|
end
|
|
end
|
|
|
|
class Label
|
|
include Renderable
|
|
def render
|
|
[@name + ':']
|
|
end
|
|
end
|
|
|
|
class CPU
|
|
# renders an instruction
|
|
# may use instruction-global properties to render an argument (eg specify pointer size if not implicit)
|
|
def render_instruction(i)
|
|
r = []
|
|
r << i.opname
|
|
r << ' '
|
|
i.args.each { |a| r << a << ', ' }
|
|
r.pop
|
|
r
|
|
end
|
|
|
|
# ease debugging in irb
|
|
def inspect
|
|
"#<#{self.class}:#{'%x' % object_id} ... >"
|
|
end
|
|
end
|
|
|
|
class Expression
|
|
include Renderable
|
|
|
|
def render_integer(e)
|
|
if e < 0
|
|
neg = true
|
|
e = -e
|
|
end
|
|
if e < 10; e = e.to_s
|
|
else
|
|
e = '%xh' % e
|
|
e = '0' << e unless (?0..?9).include? e[0]
|
|
end
|
|
e = '-' << e if neg
|
|
e
|
|
end
|
|
|
|
NOSQ1 = NOSQ2 = {:* => [:*], :+ => [:+, :-, :*], :- => [:+, :-, :*]}
|
|
NOSQ2[:-] = [:*]
|
|
def render
|
|
l = @lexpr.kind_of?(::Integer) ? render_integer(@lexpr) : @lexpr
|
|
r = @rexpr.kind_of?(::Integer) ? render_integer(@rexpr) : @rexpr
|
|
l = ['(', l, ')'] if @lexpr.kind_of?(Expression) and (not oa = NOSQ1[@op] or not oa.include?(@lexpr.op))
|
|
r = ['(', r, ')'] if @rexpr.kind_of?(Expression) and (not oa = NOSQ2[@op] or not oa.include?(@rexpr.op))
|
|
op = @op if l or @op != :+
|
|
if op == :+
|
|
r0 = [r].flatten.first
|
|
r0 = r0.render.flatten.first while r0.kind_of? Renderable
|
|
op = nil if (r0.kind_of?(::Integer) and r0 < 0) or (r0.kind_of?(::String) and r0[0] == ?-) or r0 == :-
|
|
end
|
|
[l, op, r].compact
|
|
end
|
|
end
|
|
|
|
class ExpressionString
|
|
include Renderable
|
|
|
|
def render; hide_str ? @expr.render : render_str ; end
|
|
end
|
|
end
|