puppet 3.6.2-x86-mingw32 → 3.7.0-x86-mingw32

data/lib/puppet/util/windows/api_types.rb

@@ -0,0 +1,255 @@
+require 'ffi'
+require 'puppet/util/windows/string'
+
+module Puppet::Util::Windows::APITypes
+  module ::FFI
+    WIN32_FALSE = 0
+
+    # standard Win32 error codes
+    ERROR_SUCCESS = 0
+  end
+
+  module ::FFI::Library
+    # Wrapper method for attach_function + private
+    def attach_function_private(*args)
+      attach_function(*args)
+      private args[0]
+    end
+  end
+
+  class ::FFI::Pointer
+    NULL_HANDLE = 0
+    NULL_TERMINATOR_WCHAR = 0
+
+    def self.from_string_to_wide_string(str, &block)
+      str = Puppet::Util::Windows::String.wide_string(str)
+      FFI::MemoryPointer.new(:byte, str.bytesize) do |ptr|
+        # uchar here is synonymous with byte
+        ptr.put_array_of_uchar(0, str.bytes.to_a)
+
+        yield ptr
+      end
+
+      # ptr has already had free called, so nothing to return
+      nil
+    end
+
+    def read_win32_bool
+      # BOOL is always a 32-bit integer in Win32
+      # some Win32 APIs return 1 for true, while others are non-0
+      read_int32 != FFI::WIN32_FALSE
+    end
+
+    alias_method :read_dword, :read_uint32
+    alias_method :read_win32_ulong, :read_uint32
+
+    alias_method :read_hresult, :read_int32
+
+    def read_handle
+      type_size == 4 ? read_uint32 : read_uint64
+    end
+
+    alias_method :read_wchar, :read_uint16
+    alias_method :read_word,  :read_uint16
+
+    def read_wide_string(char_length)
+      # char_length is number of wide chars (typically excluding NULLs), *not* bytes
+      str = get_bytes(0, char_length * 2).force_encoding('UTF-16LE')
+      str.encode(Encoding.default_external)
+    end
+
+    def read_arbitrary_wide_string_up_to(max_char_length = 512)
+      # max_char_length is number of wide chars (typically excluding NULLs), *not* bytes
+      # use a pointer to read one UTF-16LE char (2 bytes) at a time
+      wchar_ptr = FFI::Pointer.new(:wchar, address)
+
+      # now iterate 2 bytes at a time until an offset lower than max_char_length is found
+      0.upto(max_char_length - 1) do |i|
+        if wchar_ptr[i].read_wchar == NULL_TERMINATOR_WCHAR
+          return read_wide_string(i)
+        end
+      end
+
+      read_wide_string(max_char_length)
+    end
+
+    def read_win32_local_pointer(&block)
+      ptr = nil
+      begin
+        ptr = read_pointer
+        yield ptr
+      ensure
+        if ptr && ! ptr.null?
+          if FFI::WIN32::LocalFree(ptr.address) != FFI::Pointer::NULL_HANDLE
+            Puppet.debug "LocalFree memory leak"
+          end
+        end
+      end
+
+      # ptr has already had LocalFree called, so nothing to return
+      nil
+    end
+
+    def read_com_memory_pointer(&block)
+      ptr = nil
+      begin
+        ptr = read_pointer
+        yield ptr
+      ensure
+        FFI::WIN32::CoTaskMemFree(ptr) if ptr && ! ptr.null?
+      end
+
+      # ptr has already had CoTaskMemFree called, so nothing to return
+      nil
+    end
+
+
+    alias_method :write_dword, :write_uint32
+    alias_method :write_word, :write_uint16
+  end
+
+  # FFI Types
+  # https://github.com/ffi/ffi/wiki/Types
+
+  # Windows - Common Data Types
+  # http://msdn.microsoft.com/en-us/library/cc230309.aspx
+
+  # Windows Data Types
+  # http://msdn.microsoft.com/en-us/library/windows/desktop/aa383751(v=vs.85).aspx
+
+  FFI.typedef :uint16, :word
+  FFI.typedef :uint32, :dword
+  # uintptr_t is defined in an FFI conf as platform specific, either
+  # ulong_long on x64 or just ulong on x86
+  FFI.typedef :uintptr_t, :handle
+  FFI.typedef :uintptr_t, :hwnd
+
+  # buffer_inout is similar to pointer (platform specific), but optimized for buffers
+  FFI.typedef :buffer_inout, :lpwstr
+  # buffer_in is similar to pointer (platform specific), but optimized for CONST read only buffers
+  FFI.typedef :buffer_in, :lpcwstr
+  FFI.typedef :buffer_in, :lpcolestr
+
+  # string is also similar to pointer, but should be used for const char *
+  # NOTE that this is not wide, useful only for A suffixed functions
+  FFI.typedef :string, :lpcstr
+
+  # pointer in FFI is platform specific
+  # NOTE: for API calls with reserved lpvoid parameters, pass a FFI::Pointer::NULL
+  FFI.typedef :pointer, :lpcvoid
+  FFI.typedef :pointer, :lpvoid
+  FFI.typedef :pointer, :lpword
+  FFI.typedef :pointer, :lpdword
+  FFI.typedef :pointer, :pdword
+  FFI.typedef :pointer, :phandle
+  FFI.typedef :pointer, :ulong_ptr
+  FFI.typedef :pointer, :pbool
+  FFI.typedef :pointer, :lpunknown
+
+  # any time LONG / ULONG is in a win32 API definition DO NOT USE platform specific width
+  # which is what FFI uses by default
+  # instead create new aliases for these very special cases
+  # NOTE: not a good idea to redefine FFI :ulong since other typedefs may rely on it
+  FFI.typedef :uint32, :win32_ulong
+  FFI.typedef :int32, :win32_long
+  # FFI bool can be only 1 byte at times,
+  # Win32 BOOL is a signed int, and is always 4 bytes, even on x64
+  # http://blogs.msdn.com/b/oldnewthing/archive/2011/03/28/10146459.aspx
+  FFI.typedef :int32, :win32_bool
+
+  # Same as a LONG, a 32-bit signed integer
+  FFI.typedef :int32, :hresult
+
+  # NOTE: FFI already defines (u)short as a 16-bit (un)signed like this:
+  # FFI.typedef :uint16, :ushort
+  # FFI.typedef :int16, :short
+
+  # 8 bits per byte
+  FFI.typedef :uchar, :byte
+  FFI.typedef :uint16, :wchar
+
+  module ::FFI::WIN32
+    extend ::FFI::Library
+
+    # http://msdn.microsoft.com/en-us/library/windows/desktop/aa373931(v=vs.85).aspx
+    # typedef struct _GUID {
+    #   DWORD Data1;
+    #   WORD  Data2;
+    #   WORD  Data3;
+    #   BYTE  Data4[8];
+    # } GUID;
+    class GUID < FFI::Struct
+      layout :Data1, :dword,
+             :Data2, :word,
+             :Data3, :word,
+             :Data4, [:byte, 8]
+
+      def self.[](s)
+        raise 'Bad GUID format.' unless s =~ /^[0-9a-f]{8}-([0-9a-f]{4}-){3}[0-9a-f]{12}$/i
+
+        new.tap do |guid|
+          guid[:Data1] = s[0, 8].to_i(16)
+          guid[:Data2] = s[9, 4].to_i(16)
+          guid[:Data3] = s[14, 4].to_i(16)
+          guid[:Data4][0] = s[19, 2].to_i(16)
+          guid[:Data4][1] = s[21, 2].to_i(16)
+          s[24, 12].split('').each_slice(2).with_index do |a, i|
+            guid[:Data4][i + 2] = a.join('').to_i(16)
+          end
+        end
+      end
+
+      def ==(other) Windows.memcmp(other, self, size) == 0 end
+    end
+
+    # http://msdn.microsoft.com/en-us/library/windows/desktop/ms724950(v=vs.85).aspx
+    # typedef struct _SYSTEMTIME {
+    #   WORD wYear;
+    #   WORD wMonth;
+    #   WORD wDayOfWeek;
+    #   WORD wDay;
+    #   WORD wHour;
+    #   WORD wMinute;
+    #   WORD wSecond;
+    #   WORD wMilliseconds;
+    # } SYSTEMTIME, *PSYSTEMTIME;
+    class SYSTEMTIME < FFI::Struct
+      layout :wYear, :word,
+             :wMonth, :word,
+             :wDayOfWeek, :word,
+             :wDay, :word,
+             :wHour, :word,
+             :wMinute, :word,
+             :wSecond, :word,
+             :wMilliseconds, :word
+
+      def to_local_time
+        Time.local(self[:wYear], self[:wMonth], self[:wDay],
+          self[:wHour], self[:wMinute], self[:wSecond], self[:wMilliseconds] * 1000)
+      end
+    end
+
+    ffi_convention :stdcall
+
+    # http://msdn.microsoft.com/en-us/library/windows/desktop/aa366730(v=vs.85).aspx
+    # HLOCAL WINAPI LocalFree(
+    #   _In_  HLOCAL hMem
+    # );
+    ffi_lib :kernel32
+    attach_function :LocalFree, [:handle], :handle
+
+    # http://msdn.microsoft.com/en-us/library/windows/desktop/ms724211(v=vs.85).aspx
+    # BOOL WINAPI CloseHandle(
+    #   _In_  HANDLE hObject
+    # );
+    ffi_lib :kernel32
+    attach_function_private :CloseHandle, [:handle], :win32_bool
+
+    # http://msdn.microsoft.com/en-us/library/windows/desktop/ms680722(v=vs.85).aspx
+    # void CoTaskMemFree(
+    #   _In_opt_  LPVOID pv
+    # );
+    ffi_lib :ole32
+    attach_function :CoTaskMemFree, [:lpvoid], :void
+  end
+end

data/lib/puppet/util/windows/com.rb

@@ -0,0 +1,224 @@
+require 'ffi'
+
+module Puppet::Util::Windows::COM
+  extend FFI::Library
+
+  ffi_convention :stdcall
+
+  S_OK = 0
+  S_FALSE = 1
+
+  def SUCCEEDED(hr) hr >= 0 end
+  def FAILED(hr) hr < 0 end
+
+  module_function :SUCCEEDED, :FAILED
+
+  def raise_if_hresult_failed(name, *args)
+    failed = FAILED(result = send(name, *args)) and raise "#{name} failed (hresult #{format('%#08x', result)})."
+
+    result
+  ensure
+    yield failed if block_given?
+  end
+
+  module_function :raise_if_hresult_failed
+
+  CLSCTX_INPROC_SERVER = 0x1
+  CLSCTX_INPROC_HANDLER = 0x2
+  CLSCTX_LOCAL_SERVER = 0x4
+  CLSCTX_INPROC_SERVER16 = 0x8
+  CLSCTX_REMOTE_SERVER = 0x10
+  CLSCTX_INPROC_HANDLER16 = 0x20
+  CLSCTX_RESERVED1 = 0x40
+  CLSCTX_RESERVED2 = 0x80
+  CLSCTX_RESERVED3 = 0x100
+  CLSCTX_RESERVED4 = 0x200
+  CLSCTX_NO_CODE_DOWNLOAD = 0x400
+  CLSCTX_RESERVED5 = 0x800
+  CLSCTX_NO_CUSTOM_MARSHAL = 0x1000
+  CLSCTX_ENABLE_CODE_DOWNLOAD = 0x2000
+  CLSCTX_NO_FAILURE_LOG = 0x4000
+  CLSCTX_DISABLE_AAA = 0x8000
+  CLSCTX_ENABLE_AAA = 0x10000
+  CLSCTX_FROM_DEFAULT_CONTEXT = 0x20000
+  CLSCTX_ACTIVATE_32_BIT_SERVER = 0x40000
+  CLSCTX_ACTIVATE_64_BIT_SERVER = 0x80000
+  CLSCTX_ENABLE_CLOAKING = 0x100000
+  CLSCTX_PS_DLL = -0x80000000
+  CLSCTX_INPROC = CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER
+  CLSCTX_ALL = CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER | CLSCTX_LOCAL_SERVER | CLSCTX_REMOTE_SERVER
+  CLSCTX_SERVER = CLSCTX_INPROC_SERVER | CLSCTX_LOCAL_SERVER | CLSCTX_REMOTE_SERVER
+
+  # http://msdn.microsoft.com/en-us/library/windows/desktop/ms686615(v=vs.85).aspx
+  # HRESULT CoCreateInstance(
+  #   _In_   REFCLSID rclsid,
+  #   _In_   LPUNKNOWN pUnkOuter,
+  #   _In_   DWORD dwClsContext,
+  #   _In_   REFIID riid,
+  #   _Out_  LPVOID *ppv
+  # );
+  ffi_lib :ole32
+  attach_function_private :CoCreateInstance,
+    [:pointer, :lpunknown, :dword, :pointer, :lpvoid], :hresult
+
+  # code modified from Unknownr project https://github.com/rpeev/Unknownr
+  # licensed under MIT
+  module Interface
+    def self.[](*args)
+      spec, iid, *ifaces = args.reverse
+
+      spec.each { |name, signature| signature[0].unshift(:pointer) }
+
+      Class.new(FFI::Struct) do
+        const_set(:IID, iid)
+
+        vtable = Class.new(FFI::Struct) do
+          vtable_hash = Hash[(ifaces.map { |iface| iface::VTBL::SPEC.to_a } << spec.to_a).flatten(1)]
+          const_set(:SPEC, vtable_hash)
+
+          layout \
+            *self::SPEC.map { |name, signature| [name, callback(*signature)] }.flatten
+        end
+
+        const_set(:VTBL, vtable)
+
+        layout \
+          :lpVtbl, :pointer
+      end
+    end
+  end
+
+  module Helpers
+    def QueryInstance(klass)
+      instance = nil
+
+      FFI::MemoryPointer.new(:pointer) do |ppv|
+        QueryInterface(klass::IID, ppv)
+
+        instance = klass.new(ppv.read_pointer)
+      end
+
+      begin
+        yield instance
+        return self
+      ensure
+        instance.Release
+      end if block_given?
+
+      instance
+    end
+
+    def UseInstance(klass, name, *args)
+      instance = nil
+
+      FFI::MemoryPointer.new(:pointer) do |ppv|
+        send(name, *args, ppv)
+
+        yield instance = klass.new(ppv.read_pointer)
+      end
+
+      self
+    ensure
+      instance.Release if instance && ! instance.null?
+    end
+  end
+
+  module Instance
+    def self.[](iface)
+      Class.new(iface) do
+        send(:include, Helpers)
+
+        def initialize(pointer)
+          self.pointer = pointer
+
+          @vtbl = self.class::VTBL.new(self[:lpVtbl])
+        end
+
+        attr_reader :vtbl
+
+        self::VTBL.members.each do |name|
+          define_method(name) do |*args|
+            if Puppet::Util::Windows::COM.FAILED(result = @vtbl[name].call(self, *args))
+              raise Puppet::Util::Windows::Error.new("Failed to call #{self}::#{name} with HRESULT: #{result}.", result)
+            end
+            result
+          end
+        end
+
+        layout \
+          :lpVtbl, :pointer
+      end
+    end
+  end
+
+  module Factory
+    def self.[](iface, clsid)
+      Class.new(iface) do
+        send(:include, Helpers)
+
+        const_set(:CLSID, clsid)
+
+        def initialize(opts = {})
+          @opts = opts
+
+          @opts[:clsctx] ||= CLSCTX_INPROC_SERVER
+
+          FFI::MemoryPointer.new(:pointer) do |ppv|
+            hr = Puppet::Util::Windows::COM.CoCreateInstance(self.class::CLSID, FFI::Pointer::NULL, @opts[:clsctx], self.class::IID, ppv)
+            if Puppet::Util::Windows::COM.FAILED(hr)
+              raise "CoCreateInstance failed (#{self.class})."
+            end
+
+            self.pointer = ppv.read_pointer
+          end
+
+          @vtbl = self.class::VTBL.new(self[:lpVtbl])
+        end
+
+        attr_reader :vtbl
+
+        self::VTBL.members.each do |name|
+          define_method(name) do |*args|
+            if Puppet::Util::Windows::COM.FAILED(result = @vtbl[name].call(self, *args))
+              raise Puppet::Util::Windows::Error.new("Failed to call #{self}::#{name} with HRESULT: #{result}.", result)
+            end
+            result
+          end
+        end
+
+        layout \
+          :lpVtbl, :pointer
+      end
+    end
+  end
+
+  IUnknown = Interface[
+    FFI::WIN32::GUID['00000000-0000-0000-C000-000000000046'],
+
+    QueryInterface: [[:pointer, :pointer], :hresult],
+    AddRef: [[], :win32_ulong],
+    Release: [[], :win32_ulong]
+  ]
+
+  Unknown = Instance[IUnknown]
+
+  # http://msdn.microsoft.com/en-us/library/windows/desktop/ms678543(v=vs.85).aspx
+  # HRESULT CoInitialize(
+  #   _In_opt_  LPVOID pvReserved
+  # );
+  ffi_lib :ole32
+  attach_function_private :CoInitialize, [:lpvoid], :hresult
+
+  # http://msdn.microsoft.com/en-us/library/windows/desktop/ms688715(v=vs.85).aspx
+  # void CoUninitialize(void);
+  ffi_lib :ole32
+  attach_function_private :CoUninitialize, [], :void
+
+  def InitializeCom
+    raise_if_hresult_failed(:CoInitialize, FFI::Pointer::NULL)
+
+    at_exit { CoUninitialize() }
+  end
+
+  module_function :InitializeCom
+end