在RMVA使用Fiddle代替win32api

lifrtam

1. #===============================================================
   
2. # Vx ace  Fiddle
   
3. #
   
4. #===============================================================
   
5. module Fiddle
   
6.   
   
7.   TYPE_VOID   = DL::TYPE_VOID
   
8.   TYPE_INT    = DL::TYPE_INT
   
9.   TYPE_LONG   = DL::TYPE_LONG
   
10.   TYPE_VOIDP  = DL::TYPE_VOIDP
    
11.   TYPE_FLOAT  = DL::TYPE_FLOAT
    
12.   TYPE_DOUBLE = DL::TYPE_DOUBLE
    
13.   TYPE_CHAR   = DL::TYPE_CHAR
    
14.   TYPE_SHORT  = DL::TYPE_SHORT
    
15.   TYPE_UCHAR  = -TYPE_CHAR
    
16.   TYPE_USHORT = -TYPE_SHORT
    
17.   TYPE_UINT   = -TYPE_INT
    
18.   TYPE_ULONG  = -TYPE_LONG
    
19.   #调用约定
    
20.   DEFAULT = :stdcall
    
21. 
    
22.   #========== 函数封装 ==========
    
23.   class Function
    
24.     def initialize(ptr, args, ret, abi = DEFAULT)
    
25.       @ptr  = ptr
    
26.       @args = args
    
27.       @ret  = ret
    
28.       @calltype = case abi
    
29.                   when :stdcall, :STDCALL, 1 then :stdcall
    
30.                   when :cdecl,   :CDECL,   0 then :cdecl
    
31.                   else :stdcall
    
32.                   end
    
33.       @cfunc = DL::CFunc.new(ptr, ret, 'fiddle_func', @calltype)
    
34.     end
    
35. 
    
36.     def call(*args)
    
37.       temp_ptrs = []
    
38.       converted = args.zip(@args).map do |val, type|
    
39.         if type == TYPE_VOIDP
    
40.           if val.nil?
    
41.             0
    
42.           elsif val.is_a?(String)
    
43.             #转为 UTF-16LE（Unicode）
    
44.             utf16 = val.encode("UTF-16LE") + "\0\0".encode("UTF-16LE")
    
45.             bin = utf16.force_encoding("ASCII-8BIT")
    
46.             ptr = DL::CPtr[bin]
    
47.             temp_ptrs << ptr
    
48.             ptr.to_i
    
49.           elsif val.is_a?(DL::CPtr)
    
50.             val.to_i
    
51.           else
    
52.             val
    
53.           end
    
54.         else
    
55.           val
    
56.         end
    
57.       end
    
58.       @cfunc.call(converted)
    
59.     end
    
60.   end
    
61. 
    
62.   Pointer = DL::CPtr
    
63. 
    
64.   #========== 签名解析器=========================
    
65.   module CParser
    
66.    
    
67.     def self.parse_signature(signature, tymap = nil)
    
68.       tymap ||= {}
    
69.       signature = signature.gsub(/\s+/, " ").strip
    
70.       case signature
    
71.       when /^([\w@\*\s]+)\(([\w\*\s\,\[\]]*)\)$/
    
72.         ret_part = $1
    
73.         args_part = $2.strip
    
74.         ret_tokens = ret_part.split(/\s+/)
    
75.         args_list = args_part.split(/\s*,\s*/)
    
76.         func_name = ret_tokens.pop
    
77.         if func_name =~ /^\*/
    
78.           func_name.gsub!(/^\*+/, "")
    
79.           ret_tokens.push("*")
    
80.         end
    
81.         ret_type = join_types(ret_tokens)
    
82.         [func_name, parse_ctype(ret_type, tymap), args_list.map { |a| parse_ctype(a.strip, tymap) }]
    
83.       else
    
84.         raise "无法解析函数原型: #{signature}"
    
85.       end
    
86.     end
    
87. 
    
88.     #解析单个类型字符串
    
89.     def self.parse_ctype(ty, tymap = nil)
    
90.       tymap ||= {}
    
91.       case ty
    
92.       when Array
    
93.         [parse_ctype(ty[0], tymap), ty[1]]
    
94.       when "void"
    
95.         TYPE_VOID
    
96.       when "char"
    
97.         TYPE_CHAR
    
98.       when "unsigned char"
    
99.         -TYPE_CHAR
    
100.       when "short"
     
101.         TYPE_SHORT
     
102.       when "unsigned short"
     
103.         -TYPE_SHORT
     
104.       when "int"
     
105.         TYPE_INT
     
106.       when "unsigned int", "uint"
     
107.         -TYPE_INT
     
108.       when "long"
     
109.         TYPE_LONG
     
110.       when "unsigned long"
     
111.         -TYPE_LONG
     
112.       when "long long"
     
113.         if defined?(TYPE_LONG_LONG)
     
114.           TYPE_LONG_LONG
     
115.         else
     
116.           raise "不支持 long long 类型"
     
117.         end
     
118.       when "unsigned long long"
     
119.         if defined?(TYPE_LONG_LONG)
     
120.           -TYPE_LONG_LONG
     
121.         else
     
122.           raise "不支持 unsigned long long 类型"
     
123.         end
     
124.       when "float"
     
125.         TYPE_FLOAT
     
126.       when "double"
     
127.         TYPE_DOUBLE
     
128.       when /\*/, /\[\s*\]/
     
129.         TYPE_VOIDP
     
130.       else
     
131.         if tymap[ty]
     
132.           parse_ctype(tymap[ty], tymap)
     
133.         else
     
134.           raise "未知类型: #{ty}"
     
135.         end
     
136.       end
     
137.     end
     
138. 
     
139.    
     
140.     def self.join_types(tokens)
     
141.       tokens.join(" ")
     
142.     end
     
143.   end
     
144. 
     
145.   class Handle
     
146.     def initialize(lib)
     
147.       @handle = DL.dlopen(lib)
     
148.     end
     
149. 
     
150.     def [](func)
     
151.       @handle[func]
     
152.     end
     
153. 
     
154.    
     
155.     def extern(signature, call_type = DEFAULT)
     
156.       func_name, ret_type, arg_types = CParser.parse_signature(signature)
     
157.       ptr = self[func_name]
     
158.       if ptr.nil? || ptr == 0
     
159.         raise "找不到函数: #{func_name}"
     
160.       end
     
161.       Function.new(ptr, arg_types, ret_type, call_type)
     
162.     end
     
163.   end
     
164. end
     
165. 
     
166. #===============================================================
     
167. # 实际上是DL伪装Fiddle而已。
     
168. #===============================================================
     
169. user32 = Fiddle::Handle.new('user32')
     
170. msgbox_ptr = user32['MessageBoxW']
     
171. msgbox = Fiddle::Function.new(msgbox_ptr,
     
172.   [Fiddle::TYPE_LONG, Fiddle::TYPE_VOIDP, Fiddle::TYPE_VOIDP, Fiddle::TYPE_LONG],
     
173.   Fiddle::TYPE_LONG)  #使用 DEFAULT (stdcall)
     
174. 
     
175. msgbox.call(0, "我自由了！", "FFI", 0)
     
176. 
     
177. #===============================================================
     
178. user32 = Fiddle::Handle.new('user32')
     
179. msgbox = user32.extern('int MessageBoxW(long, void*, void*, long)')
     
180. 
     
181. msgbox.call(0, "我再也不用win32api了！", "FFI", 0)
     
182. 
     
183. 
     

复制代码

lifrtam

其实是把DL当作Fiddle来用