Project

General

Profile

Feature #8601 ยป Win32API.rb

arton (Akio Tajima), 07/05/2013 03:29 AM

 
# -*- ruby -*-
# for backward compatibility
warn "Warning:#{caller[0].sub(/:in `.*'\z/, '')}: Win32API is deprecated after Ruby 1.9.1; use dl directly instead" if $VERBOSE

class Win32API
begin
require 'fiddle'
Caller = Fiddle
rescue LoadError
require 'dl'
Caller = DL
end
DLL = {}
TYPEMAP = {"0" => Caller::TYPE_VOID, "S" => Caller::TYPE_VOIDP, "I" => Caller::TYPE_LONG}
POINTER_TYPE = Caller::SIZEOF_VOIDP == Caller::SIZEOF_LONG_LONG ? 'q*' : 'l!*'

def initialize(dllname, func, import, export = "0", calltype = :stdcall)
@proto = [import].join.tr("VPpNnLlIi", "0SSI").sub(/^(.)0*$/, '\1')
handle = DLL[dllname] ||= Caller.dlopen(dllname)
unless defined? Fiddle
@func = DL::CFunc.new(handle[func], TYPEMAP[export.tr("VPpNnLlIi", "0SSI")], func, calltype)
else
temp = [import].join.each_char.map{|s|s.tr("VPpNnLlIi", "0SSI")}.map{|v|TYPEMAP[v]}
@func = Fiddle::Function.new(handle[func], temp, TYPEMAP[export.tr("VPpNnLlIi", "0SSI")], calltype == :stdcall ? Fiddle::Function::STDCALL : Fiddle::Function::DEFAULT)
end
rescue Caller::DLError => e
raise LoadError, e.message, e.backtrace
end

def call(*args)
import = @proto.split("")
args.each_with_index do |x, i|
args[i], = [x == 0 ? nil : x].pack("p").unpack(POINTER_TYPE) if import[i] == "S"
args[i], = [x].pack("I").unpack("i") if import[i] == "I"
end
unless defined? Fiddle
ret, = @func.call(args)
else
ret, = @func.call(*args)
end
return ret || 0
end

alias Call call
end
    (1-1/1)