swf_ruby 0.1.1 → 0.2.0

data/.gitignore

@@ -1,2 +1,3 @@
 pkg/*
-*.gem
+*.gem
+*.swp

data/README.md

@@ -9,7 +9,7 @@ SwfRuby is a utilities to dump or manipulate a SWF with Ruby.
     * Used from 'swf_dump' command.
   * SwfRuby::SwfTamperer
     * Manipulating(replagcing) resources in SWF.
-    * Used from 'swf_jpeg_replace' and 'swf_lossless_replace' command.
+    * Used from 'swf_jpeg_replace', 'swf_lossless_replace', 'swf_as_var_replace' and 'swf_sprite_replace' command.
   * compatible on ruby-1.8.7 and ruby-1.9.2.
 
 Dependencies
@@ -57,6 +57,18 @@ Replacing GIF/PNG in SWF
     $ swf_lossless_replace samples/sample.swf 120 samples/icon.gif > samples/sample3.swf
     # <120> is offset to DefineBitsLossless2 resource getting by 'swf_dump'.
 
+Replacing ActionScript Variable in SWF
+--------------------------------------
+
+    $ swf_as_var_replace foo.swf bar piyo > foo2.swf
+    # <bar> is variable name. <piyo> is new value to the variable.
+
+Replacing Sprite(internal movieclip) in SWF
+-------------------------------------------
+
+    $ swf_sprite_replace foo.swf bar piyo.swf > foo2.swf
+    # <bar> is instance variable name of movieclip. <piyo.swf> is new movieclip file to replace.
+
 Thanks
 ======
 

data/bin/swf_as_var_replace

@@ -0,0 +1,21 @@
+#!/usr/bin/env ruby
+#  vim: set fileencoding=utf-8 filetype=ruby ts=2 : 
+# SWF中のAS変数の値(foo = var; な箇所に限る)を差し替える.
+require 'swf_ruby'
+
+if ARGV.size != 3
+  print "Specify target swf path, AS var name, and new value for arguments.\n"
+  exit
+end
+
+swf_dumper = SwfRuby::SwfDumper.new
+swf_dumper.open(ARGV[0])
+
+repl_targets = SwfRuby::AsVarReplaceTarget.build_by_var_name(swf_dumper, ARGV[1]).collect do |t|
+  t.str = ARGV[2].dup
+  t
+end
+
+st = SwfRuby::SwfTamperer.new
+
+print st.replace(swf_dumper.swf.dup, repl_targets)

data/bin/swf_dump

@@ -12,14 +12,24 @@ end
 swf = SwfRuby::SwfDumper.new
 swf.open(ARGV[0])
 swf.tags.each_with_index do |tag, i|
-  print "#{SwfRuby::Swf::TAG_TYPE[tag.code]}, offset: #{swf.tags_addresses[i]}, length: #{tag.length}\n"
+  character_id = tag.character_id ? "character_id: #{tag.character_id}, " : ""
+  refer_character_id = tag.refer_character_id ? "refer_character_id: #{tag.refer_character_id}, " : ""
+  refer_character_inst_name = tag.refer_character_inst_name ? "refer_character_inst_name: #{tag.refer_character_inst_name}, " : ""
+  refer_bitmap_id = ""
+  tag.refer_bitmap_offsets_to_ids.each do |offset, bitmap_id|
+    refer_bitmap_id += "bitmap_id: #{bitmap_id}(offset: #{offset}), "
+  end if tag.refer_bitmap_offsets_to_ids
+  print "#{SwfRuby::Swf::TAG_TYPE[tag.code]}, offset: #{swf.tags_addresses[i]}, #{character_id}#{refer_character_id}#{refer_character_inst_name}#{refer_bitmap_id}length: #{tag.length}\n"
   if tag.code == 39
     # DefineSprite
     sd = SwfRuby::SpriteDumper.new
     sd.dump(tag)
     print "  Sprite ID: #{sd.sprite_id}, Frame Count: #{sd.frame_count}\n"
     sd.tags.each_with_index do |tag2, k|
-      print "    #{SwfRuby::Swf::TAG_TYPE[tag2.code]}, offset: #{sd.tags_addresses[k]}, length: #{tag2.length}\n"
+      character_id = tag2.character_id ? "character_id: #{tag2.character_id}, " : ""
+      refer_character_id = tag2.refer_character_id ? "refer_character_id: #{tag2.refer_character_id}, " : ""
+      refer_character_inst_name = tag2.refer_character_inst_name ? "refer_character_inst_name: #{tag2.refer_character_inst_name}, " : ""
+      print "    #{SwfRuby::Swf::TAG_TYPE[tag2.code]}, offset: #{sd.tags_addresses[k]}, #{character_id}#{refer_character_id}#{refer_character_inst_name}length: #{tag2.length}\n"
       if tag2.code == 12
         # DoAction
         dad = SwfRuby::DoActionDumper.new

data/bin/swf_sprite_replace

@@ -0,0 +1,19 @@
+#!/usr/bin/env ruby
+#  vim: set fileencoding=utf-8 filetype=ruby ts=2 : 
+# SWF中の指定したインスタンス変数名で設置されているDefineSpriteを入れ替える.
+require 'swf_ruby'
+
+if ARGV.size != 3
+  print "Specify target swf path, sprite instance variable name, and swf path to embed for arguments.\n"
+  exit
+end
+
+swf_dumper = SwfRuby::SwfDumper.new
+swf_dumper.open(ARGV[0])
+sprite_swf = File.open(ARGV[2], "rb").read
+
+repl_targets = SwfRuby::SpriteReplaceTarget.build_list_by_instance_var_names(swf_dumper, { ARGV[1] => sprite_swf })
+
+st = SwfRuby::SwfTamperer.new
+
+print st.replace(swf_dumper.swf.dup, repl_targets)

data/lib/swf_ruby/replace_target.rb

@@ -23,16 +23,162 @@ module SwfRuby
     end
   end
 
+  class SpriteReplaceTarget < ReplaceTarget
+    attr_accessor :swf
+    attr_accessor :frame_count
+    attr_accessor :define_tags
+    attr_accessor :control_tags
+    attr_accessor :idmap
+    attr_accessor :target_define_tags_string
+    attr_accessor :target_control_tags_string
+
+    def initialize(offset, swf)
+      @offset = offset
+      @swf = swf
+      @target_swf_dumper = SwfDumper.new.dump(@swf)
+      @frame_count = @target_swf_dumper.header.frame_count
+      @define_tags = @target_swf_dumper.tags.select { |t| t.define_tag? }
+      @control_tags = @target_swf_dumper.tags - @define_tags
+      @idmap = {}
+    end
+
+    def self.build_list_by_instance_var_names(swf_dumper, var_name_to_swf)
+      from_character_id = (swf_dumper.tags.collect { |t| t.define_tag? ? t.character_id : nil }).compact.max + 1
+      repl_targets = []
+      var_name_to_swf.each do |var_name, swf|
+        repl_target, from_character_id = SwfRuby::SpriteReplaceTarget.build_by_instance_var_name(swf_dumper, var_name, swf, from_character_id)
+        repl_targets << repl_target
+      end
+      repl_targets
+    end
+
+    # 指定したインスタンス変数名に対するSpriteReplaceTargetを生成する
+    def self.build_by_instance_var_name(swf_dumper, var_name, swf, from_character_id = nil)
+      from_character_id ||= (swf_dumper.tags.collect { |t| t.define_tag? ? t.character_id : nil }).compact.max + 1
+      refer_character_id = nil
+      sprite_indices = {}
+      swf_dumper.tags.each_with_index do |t,i|
+        if t.character_id
+          sprite_indices[t.character_id] = i
+        end
+        if var_name == t.refer_character_inst_name
+          refer_character_id = t.refer_character_id
+          break
+        end
+      end
+      raise ReplaceTargetError unless refer_character_id
+      offset = swf_dumper.tags_addresses[sprite_indices[refer_character_id]]
+      srt = SpriteReplaceTarget.new(offset, swf)
+      srt.target_define_tags_string, from_character_id = srt.build_define_tags_string(from_character_id)
+      srt.target_control_tags_string = srt.build_control_tags_string
+      [srt, from_character_id]
+    end
+
+    # 置換するSWFからCharacterIdを付け替えながらDefineタグを抽出する.
+    # 対象のSWFにBitmapIDの参照が含まれる場合、これも合わせて付け替える.
+    # 同時に、CharacterIdの対応付けマップを作成する.
+    def build_define_tags_string(from_character_id)
+      str = ""
+      @define_tags.each do |t|
+        if t.character_id
+          from_character_id += 1
+          @idmap[t.character_id] = from_character_id
+          str << t.rawdata_with_define_character_id(@idmap, @idmap[t.character_id])
+        end
+      end
+      [str, from_character_id+1]
+    end
+
+    # DefineSpriteに埋め込むためのControl tagsのみを抽出する.
+    # 参照先のcharacter_idを変更する必要がある場合は付け替える.
+    def build_control_tags_string
+      str = ""
+      valid_control_tag_codes = [0, 1, 4, 5, 12, 18, 19, 26, 28, 43, 45, 70, 72]
+      @control_tags.each do |t|
+        next unless valid_control_tag_codes.include? t.code
+        if @idmap[t.refer_character_id]
+          str << t.rawdata_with_refer_character_id(@idmap[t.refer_character_id])
+        else
+          str << t.rawdata
+        end
+      end
+      str
+    end
+  end
+
   class AsVarReplaceTarget < ReplaceTarget
     attr_accessor :do_action_offset
     attr_accessor :str
-    attr_accessor :parent_sprite_offsets
+    attr_accessor :parent_sprite_offset
 
-    def initialize(action_push_offset, do_action_offset, str, parent_sprite_offsets = [])
+    def initialize(action_push_offset, do_action_offset, str, parent_sprite_offset = nil)
       @offset = action_push_offset
       @do_action_offset = do_action_offset
       @str = str
-      @parent_sprite_offsets = parent_sprite_offsets
+      @parent_sprite_offset = parent_sprite_offset
+    end
+
+    # 指定したAS変数名に対するAsVarReplaceTargetのリストを生成する
+    def self.build_by_var_name(swf_dumper, var_name)
+      as_var_replace_targets = []
+      swf_dumper.tags.each_with_index do |t, i|
+        if t.code == 39
+          # DefineSprite
+          sd = SpriteDumper.new
+          sd.dump(t)
+          sd.tags.each_with_index do |u, j|
+            if u.code == 12
+              # DoAction in DefineSprite
+              as_var_replace_targets += AsVarReplaceTarget.generate_as_var_replace_target_by_do_action(var_name, swf_dumper, j, sd, swf_dumper.tags_addresses[i])
+            end
+          end
+        end
+        if t.code == 12
+          # DoAction
+          as_var_replace_targets += AsVarReplaceTarget.generate_as_var_replace_target_by_do_action(var_name, swf_dumper, i)
+        end
+      end
+      as_var_replace_targets
+    end
+
+    # 指定したインデックス(SWFまたはSpriteの先頭からカウント)にあるDoAction以下を走査し、
+    # 指定したAS変数名の代入部分を発見し、AsVarReplaceTargetのリストを生成する.
+    def self.generate_as_var_replace_target_by_do_action(var_name, swf_dumper, do_action_index, sprite_dumper = nil, parent_sprite_offset = nil)
+      as_var_replace_targets = []
+      action_records = []
+      do_action_offset = 0
+
+      dad = DoActionDumper.new
+      if sprite_dumper
+        do_action_offset = parent_sprite_offset + sprite_dumper.tags_addresses[do_action_index]
+        dad.dump(swf_dumper.swf[do_action_offset, sprite_dumper.tags[do_action_index].length])
+      else 
+        do_action_offset = swf_dumper.tags_addresses[do_action_index]
+        dad.dump(swf_dumper.swf[do_action_offset, swf_dumper.tags[do_action_index].length])
+      end
+      dad.actions.each_with_index do |ar, i|
+        # ActionPush, ActionPush, SetVariableの並びを検出したら変数名をチェック.
+        action_records.shift if action_records.length > 2
+        action_records << ar
+        if ar.code == 29 and
+          action_records[0] and
+          action_records[0].code == 150 and
+          action_records[0].data.delete("\0") == var_name and
+          action_records[1].code == 150
+          # 対象の代入式を発見.
+          ap = Swf::ActionPush.new(action_records[1])
+          as_var_replace_targets << AsVarReplaceTarget.new(
+            do_action_offset + dad.actions_addresses[i] - action_records[1].length,
+            do_action_offset,
+            "",
+            parent_sprite_offset
+          )
+        end
+      end
+      as_var_replace_targets
     end
   end
+
+  # 置換対象指定エラー.
+  class ReplaceTargetError < StandardError; end
 end

data/lib/swf_ruby/swf/cxformwithalpha.rb

@@ -0,0 +1,47 @@
+#  vim: set fileencoding=utf-8 filetype=ruby ts=2 : 
+
+module SwfRuby
+  module Swf
+    class Cxformwithalpha
+      attr_accessor :has_add_terms
+      attr_accessor :has_mult_terms
+      attr_accessor :nbits
+      attr_accessor :red_mult_term
+      attr_accessor :green_mult_term
+      attr_accessor :blue_mult_term
+      attr_accessor :alpha_mult_term
+      attr_accessor :red_add_term
+      attr_accessor :green_add_term
+      attr_accessor :blue_add_term
+      attr_accessor :alpha_add_term
+      attr_reader :length # byte_aligned
+
+      def initialize(bytearray)
+        bits_str = bytearray.unpack("B*").first
+        @has_add_terms = bits_str[0, 1].to_i(2)
+        @has_mult_terms = bits_str[1, 1].to_i(2)
+        @nbits = bits_str[2, 4].to_i(2)
+        offset = 6
+        if @has_mult_terms == 1
+          # TODO to get as signed bit values!!
+          @red_mult_term = bits_str[offset, @nbits].to_i(2)
+          @green_mult_term = bits_str[offset+@nbits, @nbits].to_i(2)
+          @blue_mult_term = bits_str[offset+2*@nbits, @nbits].to_i(2)
+          @alpha_mult_term = bits_str[offset+3*@nbits, @nbits].to_i(2)
+          offset += 4 * @nbits
+        end
+        if @has_add_terms == 1
+          # TODO to get as signed bit values!!
+          @red_add_term = bits_str[offset, @nbits].to_i(2)
+          @green_add_term = bits_str[offset+@nbits, @nbits].to_i(2)
+          @blue_add_term = bits_str[offset+2*@nbits, @nbits].to_i(2)
+          @alpha_add_term = bits_str[offset+3*@nbits, @nbits].to_i(2)
+          offset += 4 * @nbits
+        end
+        @length = offset >> 3
+        @length += 1 if offset & 7 > 0
+        self
+      end
+    end
+  end
+end

data/lib/swf_ruby/swf/fillstyle.rb

@@ -0,0 +1,51 @@
+#  vim: set fileencoding=utf-8 filetype=ruby ts=2 : 
+
+module SwfRuby
+  module Swf
+    class Fillstyle
+      attr_accessor :shape_version
+      attr_accessor :fill_style_type
+      attr_accessor :color
+      attr_accessor :gradient_matrix
+      attr_accessor :gradient
+      attr_accessor :bitmap_id
+      attr_accessor :bitmap_id_offset
+      attr_accessor :bitmap_matrix
+      attr_reader :length
+
+      def initialize(bytearray, shape_version)
+        @shape_version = shape_version
+        @fill_style_type = bytearray[0, 1].unpack("C").first
+        offset = 1
+        if @fill_style_type == 0x00
+          if @shape_version >= 3
+            @color = SwfRuby::Swf::Rgba.new(bytearray[offset..-1])
+          else
+            @color = SwfRuby::Swf::Rgb.new(bytearray[offset..-1])
+          end
+          offset += @color.length
+        end
+        if @fill_style_type == 0x10 or @fill_style_type == 0x12 or @fill_style_type == 0x13
+          @gradient_matrix = SwfRuby::Swf::Matrix.new(bytearray[offset..-1])
+          offset += @gradient_matrix.length
+        end
+        if @fill_style_type == 0x10 or @fill_style_type == 0x12
+          @gradient = SwfRuby::Swf::Gradient.new(bytearray[offset..-1], @shape_version)
+          offset += @gradient.length
+        elsif @fill_style_type == 0x13
+          @gradient = SwfRuby::Swf::Focalgradient.new(bytearray[offset..-1], @shape_version)
+          offset += @gradient.length
+        end
+        if @fill_style_type == 0x40 or @fill_style_type == 0x41 or @fill_style_type == 0x42 or @fill_style_type == 0x43
+          @bitmap_id = bytearray[offset, 2].unpack("v").first
+          @bitmap_id_offset = offset
+          offset += 2
+          @bitmap_matrix = SwfRuby::Swf::Matrix.new(bytearray[offset..-1])
+          offset += @bitmap_matrix.length
+        end
+        @length = offset
+        self
+      end
+    end
+  end
+end

data/lib/swf_ruby/swf/focalgradient.rb

@@ -0,0 +1,35 @@
+#  vim: set fileencoding=utf-8 filetype=ruby ts=2 : 
+
+module SwfRuby
+  module Swf
+    class Focalgradient
+      attr_accessor :shape_version
+      attr_accessor :spread_mode
+      attr_accessor :interpolation_mode
+      attr_accessor :num_gradients
+      attr_accessor :gradient_records
+      attr_accessor :focal_point
+      attr_reader :length
+
+      def initialize(bytearray, shape_version)
+        @shape_version = shape_version
+        bits_str = bytearray[0, 1].unpack("B").first
+        @spread_mode = bits_str[0, 2].to_i(2)
+        @interpolation_mode = bits_str[2, 2].to_i(2)
+        @num_gradients = bits_str[4, 4].to_i(2)
+        offset = 1
+        @gradient_records = []
+        @num_gradients.times do
+          gradient_record = SwfRuby::Swf::Gradrecord.new(bytearray[offset..-1], @shape_version)
+          @gradient_records << gradient_record
+          offset += gradient_record.length
+        end
+        # TODO focal_point to be 8bit.8bit fixed_point number.
+        @focal_point = bytearray[offset, 2]
+        offset += 2
+        @length = offset
+        self
+      end
+    end
+  end
+end

data/lib/swf_ruby/swf/gradient.rb

@@ -0,0 +1,31 @@
+#  vim: set fileencoding=utf-8 filetype=ruby ts=2 : 
+
+module SwfRuby
+  module Swf
+    class Gradient
+      attr_accessor :shape_version
+      attr_accessor :spread_mode
+      attr_accessor :interpolation_mode
+      attr_accessor :num_gradients
+      attr_accessor :gradient_records
+      attr_reader :length
+
+      def initialize(bytearray, shape_version)
+        @shape_version = shape_version
+        bits_str = bytearray[0, 1].unpack("B").first
+        @spread_mode = bits_str[0, 2].to_i(2)
+        @interpolation_mode = bits_str[2, 2].to_i(2)
+        @num_gradients = bits_str[4, 4].to_i(2)
+        offset = 1
+        @gradient_records = []
+        @num_gradients.times do
+          gradient_record = SwfRuby::Swf::Gradrecord.new(bytearray[offset..-1], @shape_version)
+          @gradient_records << gradient_record
+          offset += gradient_record.length
+        end
+        @length = offset
+        self
+      end
+    end
+  end
+end

data/lib/swf_ruby/swf/gradrecord.rb

@@ -0,0 +1,26 @@
+#  vim: set fileencoding=utf-8 filetype=ruby ts=2 : 
+
+module SwfRuby
+  module Swf
+    class Gradrecord
+      attr_accessor :shape_version
+      attr_accessor :ratio
+      attr_accessor :color
+      attr_reader :length
+
+      def initialize(bytearray, shape_version)
+        @shape_version = shape_version
+        @ratio = bytearray[0, 1].unpack("C").first
+        offset = 1
+        if @shape_version >= 3
+          @color = SwfRuby::Swf::Rgba.new(bytearray[offset..-1])
+        else
+          @color = SwfRuby::Swf::Rgb.new(bytearray[offset..-1])
+        end
+        offset += @color.length
+        @length = offset
+        self
+      end
+    end
+  end
+end

data/lib/swf_ruby/swf/header.rb

@@ -9,30 +9,39 @@ module SwfRuby
       attr_accessor :frame_size
       attr_accessor :frame_rate
       attr_accessor :frame_count
-      attr_reader :length
+      attr_accessor :length
 
-      def initialize(swf)
+      def self.parse(swf)
+        header = Header.new
         # check if compressed
         if swf[1].chr == "W" and swf[2].chr == "S"
           if swf[0].chr == "F"
-            @compressed = false
+            header.compressed = false
           elsif swf[0].chr == "C"
-            @compressed = true
+            header.compressed = true
           end
         end
         # version
-        @version = swf[3].chr.unpack("C").first
+        header.version = swf[3].chr.unpack("C").first
         # file size
-        @file_length = swf[4, 4].unpack("V").first
+        header.file_length = swf[4, 4].unpack("V").first
+
+        # Zlib inflate
+        if header.compressed
+          swf[8..-1] = Zlib::Inflate.inflate(swf[8..-1])
+        end
+
         # frame size
-        @frame_size = Rectangle.new(swf[8..-1])
+        header.frame_size = Rectangle.new(swf[8..-1])
         # frame rate
-        @frame_rate = (swf[8+@frame_size.length].chr.unpack("C").first).to_f / 0x100
-        @frame_rate += swf[9+@frame_size.length].chr.unpack("C").first
+        header.frame_rate = (swf[8+header.frame_size.length].chr.unpack("C").first).to_f / 0x100
+        header.frame_rate += swf[9+header.frame_size.length].chr.unpack("C").first
         # frame count
-        @frame_count = swf[10+@frame_size.length, 2].unpack("v").first
+        header.frame_count = swf[10+header.frame_size.length, 2].unpack("v").first
         # header length
-        @length = 12+@frame_size.length
+        header.length = 12+header.frame_size.length
+
+        [header, swf]
       end
     end
   end

data/lib/swf_ruby/swf/matrix.rb

@@ -0,0 +1,50 @@
+#  vim: set fileencoding=utf-8 filetype=ruby ts=2 : 
+
+module SwfRuby
+  module Swf
+    class Matrix
+      attr_accessor :has_scale
+      attr_accessor :n_scale_bits
+      attr_accessor :scale_x
+      attr_accessor :scale_y
+      attr_accessor :has_rotate
+      attr_accessor :n_rotate_bits
+      attr_accessor :rotate_skew_0
+      attr_accessor :rotate_skew_1
+      attr_accessor :n_translate_bits
+      attr_accessor :translate_x
+      attr_accessor :translate_y
+      attr_reader :length # byte_aligned
+
+      def initialize(bytearray)
+        bits_str = bytearray.unpack("B*").first
+        @has_scale = bits_str[0, 1].to_i(2)
+        offset = 1
+        if @has_scale == 1
+          @n_scale_bits = bits_str[offset, 5].to_i(2)
+          offset += 5
+          @scale_x = bits_str[offset, self.n_scale_bits].to_i(2)
+          @scale_y = bits_str[offset+self.n_scale_bits, self.n_scale_bits].to_i(2)
+          offset += 2 * self.n_scale_bits
+        end
+        @has_rotate = bits_str[offset, 1].to_i(2)
+        offset += 1
+        if @has_rotate == 1
+          @n_rotate_bits = bits_str[offset, 5].to_i(2)
+          offset += 5
+          @rotate_skew_0 = bits_str[offset, self.n_rotate_bits].to_i(2)
+          @rotate_skew_1 = bits_str[offset+self.n_rotate_bits, self.n_rotate_bits].to_i(2)
+          offset += 2 * self.n_rotate_bits
+        end
+        @n_translate_bits = bits_str[offset, 5].to_i(2)
+        offset += 5
+        @translate_x = bits_str[offset, self.n_translate_bits].to_i(2)
+        @translate_y = bits_str[offset+self.n_translate_bits, self.n_translate_bits].to_i(2)
+        offset += 2 * self.n_translate_bits
+        @length = offset >> 3
+        @length += 1 if offset & 7 > 0
+        self
+      end
+    end
+  end
+end

data/lib/swf_ruby/swf/rgb.rb

@@ -0,0 +1,22 @@
+#  vim: set fileencoding=utf-8 filetype=ruby ts=2 : 
+
+module SwfRuby
+  module Swf
+    class Rgb
+
+      attr_accessor :red
+      attr_accessor :green
+      attr_accessor :blue
+      attr_reader :length
+
+      def initialize(bytearray)
+        @red = bytearray[0, 1].unpack("C").first
+        @green = bytearray[1, 1].unpack("C").first
+        @blue = bytearray[2, 1].unpack("C").first
+        @length = 3
+        self
+      end
+
+    end
+  end
+end

data/lib/swf_ruby/swf/rgba.rb

@@ -0,0 +1,22 @@
+#  vim: set fileencoding=utf-8 filetype=ruby ts=2 : 
+
+module SwfRuby
+  module Swf
+    class Rgba
+      attr_accessor :red
+      attr_accessor :green
+      attr_accessor :blue
+      attr_accessor :alpha
+      attr_reader :length
+
+      def initialize(bytearray)
+        @red = bytearray[0, 1].unpack("C").first
+        @green = bytearray[1, 1].unpack("C").first
+        @blue = bytearray[2, 1].unpack("C").first
+        @alpha = bytearray[3, 1].unpack("C").first
+        @length = 4
+        self
+      end
+    end
+  end
+end

data/lib/swf_ruby/swf/shapewithstyle.rb

@@ -0,0 +1,30 @@
+#  vim: set fileencoding=utf-8 filetype=ruby ts=2 : 
+
+module SwfRuby
+  module Swf
+    class Shapewithstyle
+      attr_accessor :shape_version
+      attr_accessor :fill_styles_with_offset
+      attr_accessor :after_line_styles
+
+      def initialize(bytearray, shape_version)
+        @shape_version = shape_version
+        fill_style_count = bytearray[0, 1].unpack("C").first
+        offset = 1
+        if fill_style_count == 255
+          fill_style_count = bytearray[1, 2].unpack("v").first
+          offset += 2
+        end
+        @fill_styles_with_offset = {}
+        fill_style_count.times do
+          fill_style = SwfRuby::Swf::Fillstyle.new(bytearray[offset..-1], @shape_version)
+          @fill_styles_with_offset[offset] = fill_style
+          offset += fill_style.length
+        end
+        # TODO after_line_styles are including outside tag data..
+        @after_line_styles = bytearray[offset..-1]
+        self
+      end
+    end
+  end
+end

data/lib/swf_ruby/swf/swf_string.rb

@@ -0,0 +1,21 @@
+#  vim: set fileencoding=utf-8 filetype=ruby ts=2 : 
+
+module SwfRuby
+  module Swf
+    class SwfString
+      attr_accessor :string
+      attr_reader :length # including StringEndMark
+
+      def initialize(bytearray)
+        @length = 0
+        bytearray.each_byte do |byte|
+          break if byte == 0
+          @length += 1
+        end
+        @string = bytearray[0, @length]
+        @length += 1 # Endmark
+        self
+      end
+    end
+  end
+end

data/lib/swf_ruby/swf/tag.rb

@@ -3,10 +3,16 @@
 module SwfRuby
   module Swf
     class Tag
+      attr_reader :rawdata
       attr_reader :code
       attr_reader :length
       attr_reader :long_header
       attr_reader :data
+      attr_reader :character_id
+      attr_reader :refer_character_id
+      attr_reader :refer_character_id_offset
+      attr_reader :refer_character_inst_name
+      attr_reader :refer_bitmap_offsets_to_ids
 
       def initialize(swf)
         swf.force_encoding("ASCII-8BIT") if swf.respond_to? :force_encoding
@@ -26,9 +32,129 @@ module SwfRuby
           @data = swf[2, @length]
           @length += 2
         end
+        @character_id = nil
+        if self.define_tag?
+          @character_id = @data[0, 2].unpack("v").first
+        end
+        @refer_character_id = nil
+        @refer_character_id_offset = nil
+        @refer_character_inst_name = nil
+        data_offset = @long_header ? 6 : 2
+        case Swf::TAG_TYPE[@code]
+        when "PlaceObject"
+          @refer_character_id_offset = data_offset
+          @refer_character_id = @data[0, 2].unpack("v").first
+        when "PlaceObject2"
+          flags = @data[0, 1].unpack("C").first
+          offset = 3
+          if flags & 2 == 2
+            @refer_character_id_offset = data_offset + offset
+            @refer_character_id = @data[offset, 2].unpack("v").first
+            offset += 2
+          end
+          if flags & 32 == 32
+            if flags & 4 == 4
+              matrix = SwfRuby::Swf::Matrix.new(@data[offset..-1])
+              offset += matrix.length
+            end
+            if flags & 8 == 8
+              cxtfm = SwfRuby::Swf::Cxformwithalpha.new(@data[offset..-1])
+              offset += cxtfm.length
+            end
+            offset += 2 if flags & 16 == 16
+            @refer_character_inst_name = SwfRuby::Swf::SwfString.new(@data[offset..-1]).string
+          end
+        when "PlaceObject3"
+          flags = @data[0, 2].unpack("n").first
+          offset = 4
+          if flags & 8 == 8
+            class_name = SwfRuby::Swf::SwfString.new(@data[offset..-1])
+            offset += class_name.length
+          end
+          if flags & 512 == 512
+            @refer_character_id_offset = data_offset + offset
+            @refer_character_id = @data[offset, 2].unpack("v").first
+          end
+          if flags & 8192 == 8192
+            if flags & 1024 == 1024
+              matrix = SwfRuby::Swf::Matrix.new(@data[offset..-1])
+              offset += matrix.length
+            end
+            if flags & 2048 == 2048
+              cxtfm = SwfRuby::Swf::Cxformwithalpha.new(@data[offset..-1])
+              offset += cxtfm.length
+            end
+            offset += 2 if flags & 4096 == 4096
+            @refer_character_inst_name = SwfRuby::Swf::SwfString.new(@data[offset..-1]).string
+          end
+        when "RemoveObject"
+          @refer_character_id_offset = data_offset
+          @refer_character_id = @data[0, 2].unpack("v").first
+        when "DefineShape"
+        when "DefineShape2"
+        when "DefineShape3"
+          version = Swf::TAG_TYPE[@code][-1].to_i
+          version = 1 if version == 0
+          @refer_bitmap_offsets_to_ids = {}
+          rect = SwfRuby::Swf::Rectangle.new(@data[2..-1])
+          offset = 2+rect.length
+          shapewithstyle = SwfRuby::Swf::Shapewithstyle.new(@data[offset..-1], version)
+          shapewithstyle.fill_styles_with_offset.each do |fs_offset, fs|
+            if fs.bitmap_id
+              o = offset + fs_offset + fs.bitmap_id_offset
+              o += @long_header ? 6 : 2
+              @refer_bitmap_offsets_to_ids[o] = fs.bitmap_id
+            end
+          end
+        when "DefineShape4"
+          version = 4
+          @refer_bitmap_offsets_to_ids = {}
+          shape_rect = SwfRuby::Swf::Rectangle.new(@data[2..-1])
+          offset = 2+shape_rect.length
+          edge_rect = SwfRuby::Swf::Rectangle.new(@data[offset..-1])
+          offset += edge_rect.length
+          offset += 1
+          shapewithstyle = SwfRuby::Swf::Shapewithstyle.new(@data[offset..-1], version)
+          shapewithstyle.fill_styles_with_offset.each do |fs_offset, fs|
+            if fs.bitmap_id
+              o = offset + fs_offset + fs.bitmap_id_offset
+              o += @long_header ? 6 : 2
+              @refer_bitmap_offsets_to_ids[o] = fs.bitmap_id
+            end
+          end
+        else 
+          # do nothing.
+        end
+        @rawdata = swf[0, @length]
+        self
+      end
+
+      # Define系タグの冒頭のcharacter_id参照を指定のIDに書き換える.
+      # 内部にBitmapIDの参照を含む場合、渡されたidmapから読み替える値を参照し、書き換える.
+      def rawdata_with_define_character_id(idmap, character_id)
+        if self.define_tag?
+          offset = @long_header ? 6 : 2
+          @rawdata[offset, 2] = [character_id].pack("v")
+          @refer_bitmap_offsets_to_ids.each do |bitmap_id_offset, bitmap_id|
+            @rawdata[bitmap_id_offset, 2] = [idmap[bitmap_id]].pack("v")
+          end if @refer_bitmap_offsets_to_ids
+        end
+        @rawdata
       end
+
+      def rawdata_with_refer_character_id(character_id)
+        @rawdata[@refer_character_id_offset, 2] = [character_id].pack("v") if @refer_character_id_offset
+        @rawdata
+      end
+
+      def define_tag?
+        Swf::TAG_TYPE[@code] =~ /^Define/
+      end
+
     end
 
+    class TagError < StandardError; end
+
     TAG_TYPE = {
       0=>"End",
       1=>"ShowFrame",

data/lib/swf_ruby/swf_dumper.rb

@@ -7,6 +7,7 @@ module SwfRuby
     attr_reader :header
     attr_reader :tags
     attr_reader :tags_addresses
+    attr_accessor :character_ids
 
     # 初期化.
     def initialize
@@ -27,7 +28,7 @@ module SwfRuby
     # ダンプして構造をインスタンス変数に格納.
     def dump(swf)
       @swf = swf
-      @header = Swf::Header.new(@swf)
+      @header, @swf = Swf::Header.parse(@swf)
       @tags = []
       @tags_addresses = []
       tags_length = 0
@@ -40,5 +41,6 @@ module SwfRuby
       end
       self
     end
+
   end
 end

data/lib/swf_ruby/swf_tamperer.rb

@@ -8,20 +8,48 @@ module SwfRuby
       replace_targets.sort_by { |a| a.offset }.reverse.each do |rt|
         case rt
         when Jpeg2ReplaceTarget
-          swf = self.repl_jpeg2(swf, rt.offset, rt.jpeg)
+          swf = repl_jpeg2(swf, rt.offset, rt.jpeg)
         when Lossless2ReplaceTarget
-          swf = self.repl_lossless2(swf, rt.offset, rt.image)
+          swf = repl_lossless2(swf, rt.offset, rt.image)
         when AsVarReplaceTarget
-          swf = self.repl_action_push_string(swf, rt.do_action_offset, rt.offset, rt.str, rt.parent_sprite_offsets)
+          swf = repl_action_push_string(swf, rt.do_action_offset, rt.offset, rt.str, rt.parent_sprite_offset)
+        when SpriteReplaceTarget
+          swf = repl_sprite(swf, rt.offset, rt.target_define_tags_string, rt.frame_count, rt.target_control_tags_string)
         end
       end
       swf
     end
 
-    protected
+    private
 
-    # ActionScriptに含まれる文字列を置換.
-    def repl_action_push_string(swf, do_action_offset, action_push_offset, str, parent_sprite_offsets = [])
+    # 対象オフセット位置にあるDefineSpriteを指定したControlTags文字列で置換.
+    # 新しいSprite含むDefineTagsは、DefineSpriteの直前に挿入する.
+    def repl_sprite(swf, offset, define_tags, frame_count, control_tags)
+      swf.force_encoding("ASCII-8BIT") if swf.respond_to? :force_encoding
+      define_tags.force_encoding("ASCII-8BIT") if define_tags.respond_to? :force_encoding
+      control_tags.force_encoding("ASCII-8BIT") if control_tags.respond_to? :force_encoding
+      record_header = swf[offset, 2].unpack("v").first
+      # tag check
+      raise ReplaceTargetError if (record_header >> 6) & 1023 != 39
+      # error for short header (not implemented yet.)
+      raise ReplaceTargetError if record_header & 63 < 63
+      # rewrite frame count
+      swf[offset+8, 2] = [frame_count].pack("v")
+      # rewrite control tag and length
+      sprite_len = swf[offset+2, 4].unpack("i").first # without recordheader
+      old_control_tags_len = sprite_len - 4
+      delta_control_tags_len = control_tags.length - old_control_tags_len
+      swf[offset+10, old_control_tags_len] = control_tags
+      swf[offset+2, 4] = [sprite_len + delta_control_tags_len].pack("V")
+      # insert define tags before define sprite
+      swf[offset, 0] = define_tags
+      # rewrite swf header
+      swf[4, 4] = [swf[4, 4].unpack("V").first + define_tags.length + delta_control_tags_len].pack("V")
+      swf
+    end
+
+    # 対象オフセット位置にあるActionPushデータを置換.
+    def repl_action_push_string(swf, do_action_offset, action_push_offset, str, parent_sprite_offset = nil)
       swf.force_encoding("ASCII-8BIT") if swf.respond_to? :force_encoding
       str.force_encoding("ASCII-8BIT") if str.respond_to? :force_encoding
       record_header = swf[do_action_offset, 2].unpack("v").first
@@ -30,10 +58,10 @@ module SwfRuby
       # action check
       raise ReplaceTargetError if swf[action_push_offset].chr.unpack("C").first != 0x96
       raise ReplaceTargetError if swf[action_push_offset + 3].chr.unpack("C").first != 0
-      # error on short header
+      # error for short header (not implemented yet.)
       raise ReplaceTargetError if record_header & 63 < 63
       # calc length
-      do_action_len = swf[do_action_offset+2, 4].unpack("i").first
+      do_action_len = swf[do_action_offset+2, 4].unpack("i").first # without recordheader
       action_push_len = swf[action_push_offset+1, 2].unpack("v").first
       org_str_length = action_push_len - 2 # data type & terminated null
       delta_str_length = str.length - org_str_length
@@ -41,7 +69,7 @@ module SwfRuby
       swf[action_push_offset+4, org_str_length] = str
       swf[action_push_offset+1, 2] = [action_push_len + delta_str_length].pack("v")
       swf[do_action_offset+2, 4] = [do_action_len + delta_str_length].pack("V")
-      parent_sprite_offsets.sort.reverse.each do |parent_sprite_offset|
+      if parent_sprite_offset
         swf[parent_sprite_offset+2, 4] = [swf[parent_sprite_offset+2, 4].unpack("V").first + delta_str_length].pack("V")
       end
       swf[4, 4] = [swf[4, 4].unpack("V").first + delta_str_length].pack("V")
@@ -143,6 +171,4 @@ module SwfRuby
     end
   end
 
-  # 置換対象指定エラー.
-  class ReplaceTargetError < StandardError; end
 end

data/lib/swf_ruby/version.rb

@@ -1,3 +1,3 @@
 module SwfRuby
-  VERSION = "0.1.1"
+  VERSION = "0.2.0"
 end

data/swf_ruby.gemspec

@@ -19,7 +19,7 @@ Gem::Specification.new do |s|
   s.add_dependency "rmagick", ">= 2.13.0"
 
   s.bindir             = 'bin'
-  s.executables        = ['swf_dump', 'swf_jpeg_replace', 'swf_lossless_replace']
+  s.executables        = ['swf_dump', 'swf_jpeg_replace', 'swf_lossless_replace', 'swf_as_var_replace', 'swf_sprite_replace']
 
   s.files        = `git ls-files`.split("\n")
   s.require_path = 'lib'

metadata

@@ -4,9 +4,9 @@ version: !ruby/object:Gem::Version
   prerelease: false
   segments: 
   - 0
-  - 1
-  - 1
-  version: 0.1.1
+  - 2
+  - 0
+  version: 0.2.0
 platform: ruby
 authors: []
 
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-01-27 00:00:00 +09:00
+date: 2011-04-27 00:00:00 +09:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -56,6 +56,8 @@ executables:
 - swf_dump
 - swf_jpeg_replace
 - swf_lossless_replace
+- swf_as_var_replace
+- swf_sprite_replace
 extensions: []
 
 extra_rdoc_files: []
@@ -68,9 +70,11 @@ files:
 - Gemfile.lock
 - README.md
 - Rakefile
+- bin/swf_as_var_replace
 - bin/swf_dump
 - bin/swf_jpeg_replace
 - bin/swf_lossless_replace
+- bin/swf_sprite_replace
 - lib/swf_ruby.rb
 - lib/swf_ruby/do_action_dumper.rb
 - lib/swf_ruby/replace_target.rb
@@ -79,8 +83,18 @@ files:
 - lib/swf_ruby/swf/action_push.rb
 - lib/swf_ruby/swf/action_record.rb
 - lib/swf_ruby/swf/bits_lossless2.rb
+- lib/swf_ruby/swf/cxformwithalpha.rb
+- lib/swf_ruby/swf/fillstyle.rb
+- lib/swf_ruby/swf/focalgradient.rb
+- lib/swf_ruby/swf/gradient.rb
+- lib/swf_ruby/swf/gradrecord.rb
 - lib/swf_ruby/swf/header.rb
+- lib/swf_ruby/swf/matrix.rb
 - lib/swf_ruby/swf/rectangle.rb
+- lib/swf_ruby/swf/rgb.rb
+- lib/swf_ruby/swf/rgba.rb
+- lib/swf_ruby/swf/shapewithstyle.rb
+- lib/swf_ruby/swf/swf_string.rb
 - lib/swf_ruby/swf/tag.rb
 - lib/swf_ruby/swf_dumper.rb
 - lib/swf_ruby/swf_tamperer.rb