furnace-avm2 0.0.1

data/lib/avm2/abc/metadata/klass_info.rb

@@ -0,0 +1,7 @@
+module AVM2::ABC
+  class KlassInfo < Record
+    root_ref     :initializer, :method
+
+    abc_array_of :trait, :nested, :class => TraitInfo
+  end
+end

data/lib/avm2/abc/metadata/metadata_info.rb

@@ -0,0 +1,9 @@
+module AVM2::ABC
+  class MetadataInfo < Record
+    vuint30      :name
+
+    vuint30      :item_count,  :value => lambda { item_keys.count }
+    array        :item_keys,   :type => :vuint30, :initial_length => :item_count
+    array        :item_values, :type => :vuint30, :initial_length => :item_count
+  end
+end

data/lib/avm2/abc/metadata/method_body_info.rb

@@ -0,0 +1,31 @@
+module AVM2::ABC
+  class MethodBodyInfo < Record
+    root_ref     :method
+    vuint30      :max_stack
+    vuint30      :local_count
+    vuint30      :init_scope_depth
+    vuint30      :max_scope_depth
+
+    vuint30      :code_length, :value => lambda { code.byte_length }
+    nested       :code, :class => OpcodeSequence
+
+    abc_array_of :exception, :nested, :class => ExceptionInfo
+
+    abc_array_of :trait, :nested, :class => TraitInfo
+
+    def after_read(io)
+      exceptions.each do |exception|
+        exception.resolve!
+      end
+    end
+
+    def build_ast
+      pipeline = Furnace::Transform::Pipeline.new([
+        AVM2::Transform::ASTBuild.new(validate: true),
+        #AVM2::Transform::ASTNormalize.new
+      ])
+
+      pipeline.run(code, self)
+    end
+  end
+end

data/lib/avm2/abc/metadata/method_info.rb

@@ -0,0 +1,57 @@
+module AVM2::ABC
+  class MethodInfo < Record
+    NEED_ARGUMENTS  = 0x01
+    NEED_ACTIVATION = 0x02
+    NEED_REST       = 0x04
+    HAS_OPTIONAL    = 0x08
+    SET_DXNS        = 0x40
+    HAS_PARAM_NAMES = 0x80
+
+    vuint30     :param_count, :value => lambda { param_types.count }
+
+    const_ref   :return_type, :multiname
+    const_array :param_types, :multiname, :initial_length => :param_count
+
+    const_ref   :name, :string
+
+    uint8       :flags
+    flag        :needs_arguments,  :flags, NEED_ARGUMENTS
+    flag        :needs_activation, :flags, NEED_ACTIVATION
+    flag        :needs_rest,       :flags, NEED_REST
+    flag        :has_optional,     :flags, HAS_OPTIONAL
+    flag        :set_dxns,         :flags, SET_DXNS
+    flag        :has_param_names,  :flags, HAS_PARAM_NAMES
+
+    nested      :options, :class => OptionInfo, :if => lambda { flags & HAS_OPTIONAL != 0 }
+    const_array :param_names, :string, :initial_length => :param_count, :if => lambda { flags & HAS_PARAM_NAMES != 0 }
+
+    def to_astlet(name=nil)
+      root = AST::Node.new(:method)
+      root.metadata = { origin: self }
+
+      root.children << name || self.name
+
+      if return_type
+        root.children << return_type.to_astlet
+      else
+        root.children << nil
+      end
+
+      if has_param_names?
+        names = param_names
+      else
+        names = param_count.times.map { |n| "a#{n}" }
+      end
+
+      root.children << names.each_with_index.map do |name, index|
+        if param_types[index]
+          [ name, param_types[index].to_astlet ]
+        else
+          [ name, nil ]
+        end
+      end
+
+      root.normalize_hierarchy!
+    end
+  end
+end

data/lib/avm2/abc/metadata/multiname_info.rb

@@ -0,0 +1,42 @@
+module AVM2::ABC
+  class MultinameInfo < Record
+    XlatTable = {
+      :QName       => 0x07,
+      :QNameA      => 0x0D,
+      :RTQName     => 0x0F,
+      :RTQNameA    => 0x10,
+      :RTQNameL    => 0x11,
+      :RTQNameLA   => 0x12,
+      :Multiname   => 0x09,
+      :MultinameA  => 0x0E,
+      :MultinameL  => 0x1B,
+      :MultinameLA => 0x1C,
+      :GenericName => 0x1D # Undocumented
+    }
+
+    uint8      :kind_raw
+    xlat_field :kind
+
+    choice     :data, :selection => :kind do
+      variant :QName,       :nested, :class => MultinameKindQName
+      variant :QNameA,      :nested, :class => MultinameKindQName
+      variant :RTQName,     :nested, :class => MultinameKindRTQName
+      variant :RTQNameA,    :nested, :class => MultinameKindRTQName
+      variant :RTQNameL,    :nested, :class => MultinameKindRTQNameL
+      variant :RTQNameLA,   :nested, :class => MultinameKindRTQNameL
+      variant :Multiname,   :nested, :class => MultinameKindMultiname
+      variant :MultinameA,  :nested, :class => MultinameKindMultiname
+      variant :MultinameL,  :nested, :class => MultinameKindMultinameL
+      variant :MultinameLA, :nested, :class => MultinameKindMultinameL
+      variant :GenericName, :nested, :class => MultinameKindGenericName
+    end
+
+    def method_missing(method, *args, &block)
+      data.send(method, *args, &block)
+    end
+
+    def to_s
+      data.to_s
+    end
+  end
+end

data/lib/avm2/abc/metadata/multiname_kind_genericname.rb

@@ -0,0 +1,16 @@
+module AVM2::ABC
+  # Undocumented
+  class MultinameKindGenericName < Record
+    vuint30      :name_type
+
+    abc_array_of :parameter, :vuint30
+
+    def to_s
+      "(genericname)"
+    end
+
+    def to_astlet
+      AST::Node.new(:generic, [ name_type, parameters ])
+    end
+  end
+end

data/lib/avm2/abc/metadata/multiname_kind_multiname.rb

@@ -0,0 +1,18 @@
+module AVM2::ABC
+  class MultinameKindMultiname < Record
+    const_ref :name,   :string
+    const_ref :ns_set, :ns_set
+
+    def to_s
+      "#{ns_set}::#{name || '*'}"
+    end
+
+    def to_astlet
+      AST::Node.new(:m, [ ns_set.to_astlet, name ])
+    end
+
+    def context_size
+      0
+    end
+  end
+end

data/lib/avm2/abc/metadata/multiname_kind_multinamel.rb

@@ -0,0 +1,17 @@
+module AVM2::ABC
+  class MultinameKindMultinameL < Record
+    const_ref :ns_set, :ns_set
+
+    def to_s
+      "#{ns_set}::(rt)"
+    end
+
+    def to_astlet(name)
+      node = AST::Node.new(:m, [ ns_set.to_astlet, name ])
+    end
+
+    def context_size
+      1
+    end
+  end
+end

data/lib/avm2/abc/metadata/multiname_kind_qname.rb

@@ -0,0 +1,44 @@
+module AVM2::ABC
+  class MultinameKindQName < Record
+    const_ref :ns,   :namespace
+    const_ref :name, :string
+
+    def to_s
+      if ns
+        "#{ns}::#{name || '*'}"
+      else
+        name || '*'
+      end
+    end
+
+    def to_astlet
+      node = AST::Node.new(:q)
+
+      if ns && ns.name == ""
+      elsif ns
+        node.children << ns.name
+      else
+        node.children << "*"
+      end
+
+      node.children << name
+
+      node
+    end
+
+    def to_astpat(type=:qname)
+      case type
+      when :qname
+        [ :q, (ns.name if ns), name ].compact
+      when :multiname
+        [ :m, [ ns.name ], name ]
+      else
+        raise "unknown astpat type #{type}"
+      end
+    end
+
+    def context_size
+      0
+    end
+  end
+end

data/lib/avm2/abc/metadata/multiname_kind_rtqname.rb

@@ -0,0 +1,17 @@
+module AVM2::ABC
+  class MultinameKindRTQName < Record
+    const_ref :name, :string
+
+    def to_s
+      "(rt)::#{name || '*'}"
+    end
+
+    def to_astlet(ns)
+      node = AST::Node.new(:q, [ ns, name ])
+    end
+
+    def context_size
+      1
+    end
+  end
+end

data/lib/avm2/abc/metadata/multiname_kind_rtqnamel.rb

@@ -0,0 +1,11 @@
+module AVM2::ABC
+  class MultinameKindRTQNameL < Record
+    def to_s
+      "(rt)::(rt)"
+    end
+
+    def context_size
+      2
+    end
+  end
+end

data/lib/avm2/abc/metadata/namespace_info.rb

@@ -0,0 +1,22 @@
+module AVM2::ABC
+  class NamespaceInfo < Record
+    XlatTable = {
+      :Namespace          => 0x08,
+      :PackageNamespace   => 0x16,
+      :PackageInternalNs  => 0x17,
+      :ProtectedNamespace => 0x18,
+      :ExplicitNamespace  => 0x19,
+      :StaticProtectedNs  => 0x1A,
+      :PrivateNs          => 0x05,
+    }
+
+    uint8      :kind_raw
+    xlat_field :kind
+
+    const_ref  :name, :string
+
+    def to_s
+      name.to_s
+    end
+  end
+end

data/lib/avm2/abc/metadata/ns_set_info.rb

@@ -0,0 +1,13 @@
+module AVM2::ABC
+  class NsSetInfo < Record
+    const_array_of :ns, :namespace, :plural => :ns
+
+    def to_s
+      "{#{ns.map(&:to_s).join(",")}}"
+    end
+
+    def to_astlet
+      ns.map(&:name).uniq
+    end
+  end
+end

data/lib/avm2/abc/metadata/option_detail.rb

@@ -0,0 +1,26 @@
+module AVM2::ABC
+  class OptionDetail < Record
+    XlatTable = {
+      :Int                => 0x03,
+      :UInt               => 0x04,
+      :Double             => 0x06,
+      :Utf8               => 0x01,
+      :True               => 0x0B,
+      :False              => 0x0A,
+      :Null               => 0x0C,
+      :Undefined          => 0x00,
+      :Namespace          => 0x08,
+      :PackageNamespace   => 0x16,
+      :PackageInternalNs  => 0x17,
+      :ProtectedNamespace => 0x18,
+      :ExplicitNamespace  => 0x19,
+      :StaticProtectedNs  => 0x1A,
+      :PrivateNs          => 0x05,
+    }
+
+    vuint30    :val
+
+    uint8      :kind_raw
+    xlat_field :kind
+  end
+end

data/lib/avm2/abc/metadata/option_info.rb

@@ -0,0 +1,5 @@
+module AVM2::ABC
+  class OptionInfo < Record
+    abc_array_of :option, :nested, :class => OptionDetail
+  end
+end

data/lib/avm2/abc/metadata/script_info.rb

@@ -0,0 +1,7 @@
+module AVM2::ABC
+  class ScriptInfo < Record
+    vuint30 :init
+
+    abc_array_of :trait, :nested, :class => TraitInfo
+  end
+end

data/lib/avm2/abc/metadata/trait_class.rb

@@ -0,0 +1,9 @@
+module AVM2::ABC
+  class TraitClass < Record
+    vuint30 :slot_id
+    vuint30 :classi
+
+    def to_astlet(trait)
+    end
+  end
+end

data/lib/avm2/abc/metadata/trait_function.rb

@@ -0,0 +1,9 @@
+module AVM2::ABC
+  class TraitFunction < Record
+    vuint30  :slot_id
+    root_ref :method
+
+    def to_astlet(trait)
+    end
+  end
+end

data/lib/avm2/abc/metadata/trait_info.rb

@@ -0,0 +1,55 @@
+module AVM2::ABC
+  class TraitInfo < Record
+    XlatTable = {
+      :Slot     => 0,
+      :Method   => 1,
+      :Getter   => 2,
+      :Setter   => 3,
+      :Class    => 4,
+      :Function => 5,
+      :Const    => 6
+    }
+
+    FINAL    = 0x01
+    OVERRIDE = 0x02
+    METADATA = 0x04
+
+    attr_accessor :kind_raw
+
+    def read_attributes(io, options)
+      byte = io.read(1).unpack("C").at(0)
+
+      value     = byte >> 4
+      @kind_raw = byte & 0xF
+
+      value
+    end
+
+    def write_attributes(io, value, options)
+      byte = (value << 4) | (@kind_raw & 0xf)
+
+      io.write([ byte ].pack("C"))
+    end
+
+    const_ref  :name, :multiname
+
+    attributes :attributes
+    xlat_field :kind
+
+    choice     :data, :selection => :kind do
+      variant :Slot,     :nested, :class => TraitSlot
+      variant :Method,   :nested, :class => TraitMethod
+      variant :Getter,   :nested, :class => TraitMethod
+      variant :Setter,   :nested, :class => TraitMethod
+      variant :Class,    :nested, :class => TraitClass
+      variant :Function, :nested, :class => TraitFunction
+      variant :Const,    :nested, :class => TraitSlot
+    end
+
+    abc_array_of :metadata, :vuint30, :plural => :metadata, :if => lambda { attributes & METADATA != 0 }
+
+    def to_astlet
+      data.to_astlet(self)
+    end
+  end
+end