利用set_trace_func方法来判断一个方法是否被alias过

2014 年 8 月 26 日

其实就是囧叔跟我说的利用irb的help相同的技巧，在set_trace_func下调用，在合适的binding中取得信息并且直接用throw…catch来跳出（实际并不执行方法体），

详细方法见代码注释：

def custom_trace(sym, &b)
  u = method(sym)                            #得到方法
  v = u.arity < 0 ? [] : [0] * u.arity       #伪造方法参数
  x = lambda{u.call(*v)}                     #x.call将要调用这个方法，但是实际上不会真的调用
  set_trace_func Proc.new
  ret = catch(:gogogo) do                         #捕捉gogogo并且返回新的方法名                          
    x.call                                  #执行上述方法
  end
  set_trace_func nil#取消跟踪
  ret
end

def aliased?(sym)
  sym != custom_trace(sym) do |*e|
    next unless e[0] == "call"       #定位是call
    throw :gogogo, e[-3]             #退出, e[-3]就是当时的方法名
  end
end

def a u 
  puts "Hello"
end
alias b a

p aliased? :b #=> true
p aliased? :a #=> false

def custom_trace(sym, &b)

u = method(sym) #得到方法

v = u.arity < 0 ? [] : [0] * u.arity #伪造方法参数

x = lambda{u.call(*v)} #x.call将要调用这个方法，但是实际上不会真的调用

set_trace_func Proc.new

ret = catch(:gogogo) do #捕捉gogogo并且返回新的方法名

x.call #执行上述方法

end

set_trace_func nil#取消跟踪

ret

end

def aliased?(sym)

sym != custom_trace(sym) do |*e|

next unless e[0] == "call" #定位是call

throw :gogogo, e[-3] #退出, e[-3]就是当时的方法名

end

def a u

puts "Hello"

end

alias b a

p aliased? :b #=> true

p aliased? :a #=> false

顺便，在Ruby1.8的Method没有定义source_location的情况下取得方法定义的位置的方法，以及取得参数名的方法。

def source_location(sym)
  custom_trace(sym) do |*e|
    next unless e[0] == "call"      
    throw :gogogo, eval("[__FILE__, __LINE__]", e[-2])
  end
end

def parameters(sym)
    custom_trace(sym) do |*e|
    next unless e[0] == "call"      
    throw :gogogo, eval("local_variables", e[-2]) 
=begin
  第一行就出现的本地变量是一般就是参数
  除非有意写成def add(a,b); s = 3
  这个方法除了返回"a" "b"还会返回"s"
=end
  end
end

def source_location(sym)

custom_trace(sym) do |*e|

next unless e[0] == "call"

throw :gogogo, eval("[__FILE__, __LINE__]", e[-2])

end

def parameters(sym)

custom_trace(sym) do |*e|

next unless e[0] == "call"

throw :gogogo, eval("local_variables", e[-2])

=begin

第一行就出现的本地变量是一般就是参数

除非有意写成def add(a,b); s = 3

这个方法除了返回"a" "b"还会返回"s"

=end

end

在RM中使用gem和bundle

其实很多东西跟普通ruby使用gem和bundle是一样的，但是RM没有自带rubygems，因此有两种思路，一种是给RM嵌入一个rubygems，一种是利用另一个Ruby(也就是实际运行bundle的ruby.exe等)来发现require需要的文件和依赖关系。

下面以第二种思路为例，例子任务是使用iniparse这个gem来帮助RM里面读Game.ini。

注意：RPGMaker系列本身不支持载入.so文件，因此能用到的gems只能是纯Ruby形式的源文件组成的代码，这个文章与这个限制无关，也无意绕过这个限制。只是提供了另一种管理脚本插件的思路：借用rubygems。产生的gem目录也可以最后放在加密的rgssad(rgss2a/rgss3a)文件里面并用某种手段当做本地目录读出来，从而保证了加密保护游戏程序的语义，当然这里也不讨论对gem目录的加密和解包具体方法。

1. 首先像正常使用bundle一样在Game.exe目录下建立一个gemfile:

source 'http://ruby.taobao.org'

gem 'iniparse'

source 'http://ruby.taobao.org'

gem 'iniparse'

2. 用bundle在工程目录安装所需的gems：

用cmd在工程目录下运行下列代码。

bundle install --path gems

1	bundle install --path gems

3. 在RM的脚本编辑器中插入一段辅助代码，让require可以自动去找到我们指定的gem：

def rb(code, args = nil)
  open "run.rb", "w" do |f|
    f.write "
      class Output
        def <<(obj)
          puts [Marshal.dump(obj)].pack('M')
        end
      end
      $output = Output.new
    "
    f.write code
  end
  system "ruby run.rb #{args}> list.txt"
  Marshal.load File.read('list.txt').unpack("M").first
end

module Kernel
 alias oreq require
 $:.concat rb %{ $output << $: }
 def require(a)
   u = rb %{$output << Gem.find_files(ARGV[0])}, a
   u.select(&FileTest.method(:file?)).any?{|req|oreq req}
 end

 Object::RUBY_VERSION = rb %{$output << RUBY_VERSION} 
 require 'rubygems'
 alias require oreq
end

def rb(code, args = nil)

open "run.rb", "w" do |f|

f.write "

class Output

def <<(obj)

puts [Marshal.dump(obj)].pack('M')

end

$output = Output.new

f.write code

end

system "ruby run.rb #{args}> list.txt"

Marshal.load File.read('list.txt').unpack("M").first

end

module Kernel

alias oreq require

$:.concat rb %{ $output << $: }

def require(a)

u = rb %{$output << Gem.find_files(ARGV[0])}, a

u.select(&FileTest.method(:file?)).any?{|req|oreq req}

end

Object::RUBY_VERSION = rb %{$output << RUBY_VERSION}

require 'rubygems'

alias require oreq

end

4. 这个时候在脚本编辑器之后可以插入使用的代码了，这里我们用的是RGSS3(RPG Maker VX Ace)的代码例子：

require 'iniparse'
msgbox IniParse.parse(File.read('Game.ini'))["Game"]["Scripts"]

1 2	require 'iniparse' msgbox IniParse.parse(File.read('Game.ini'))["Game"]["Scripts"]

5. 最后，运行的方式是在cmd下运行下列代码，你也可以建立快捷方式：

bundle exec "Game.exe console"

1	bundle exec "Game.exe console"