summaryrefslogtreecommitdiff
path: root/ext/dl/lib/dl/win32.rb
diff options
context:
space:
mode:
Diffstat (limited to 'ext/dl/lib/dl/win32.rb')
-rw-r--r--ext/dl/lib/dl/win32.rb31
1 files changed, 17 insertions, 14 deletions
diff --git a/ext/dl/lib/dl/win32.rb b/ext/dl/lib/dl/win32.rb
index b507be5fde2..f0ea2a91e12 100644
--- a/ext/dl/lib/dl/win32.rb
+++ b/ext/dl/lib/dl/win32.rb
@@ -3,24 +3,27 @@
require 'dl'
class Win32API
- LIBRARY = {}
+ DLL = {}
- attr_reader :val, :args
-
- def initialize(lib, func, args, ret)
- LIBRARY[lib] ||= DL.dlopen(lib)
- ty = (ret + args).tr('V','0')
- @sym = LIBRARY[lib].sym(func, ty)
- @__dll__ = LIBRARY[lib].to_i
- @__dllname__ = lib
- @__proc__ = @sym.to_i
- @val = nil
- @args = []
+ def initialize(dllname, func, import, export = "0")
+ prototype = (export + import.to_s).tr("VPpNnLlIi", "0SSI")
+ handle = DLL[dllname] ||= DL::Handle.new(dllname)
+ begin
+ @sym = handle.sym(func, prototype)
+ rescue RuntimeError
+ @sym = handle.sym(func + "A", prototype)
+ end
end
def call(*args)
- @val,@args = @sym.call(*args)
- return @val
+ import = @sym.proto[1..-1] || ""
+ args.each_with_index do |x, i|
+ args[i] = nil if x == 0 and import[i] == ?S
+ args[i], = [x].pack("I").unpack("i") if import[i] == ?I
+ end
+ ret, = @sym.call(*args)
+ return ret || 0
end
+
alias Call call
end