diff --git a/lib/puppet/util/windows/registry.rb b/lib/puppet/util/windows/registry.rb index 9787e9d46..2b8cf7bce 100644 --- a/lib/puppet/util/windows/registry.rb +++ b/lib/puppet/util/windows/registry.rb @@ -1,308 +1,362 @@ require 'puppet/util/windows' module Puppet::Util::Windows module Registry require 'ffi' extend FFI::Library # http://msdn.microsoft.com/en-us/library/windows/desktop/aa384129(v=vs.85).aspx KEY64 = 0x100 KEY32 = 0x200 KEY_READ = 0x20019 KEY_WRITE = 0x20006 KEY_ALL_ACCESS = 0x2003f ERROR_NO_MORE_ITEMS = 259 def root(name) Win32::Registry.const_get(name) rescue NameError raise Puppet::Error, "Invalid registry key '#{name}'", $!.backtrace end def open(name, path, mode = KEY_READ | KEY64, &block) hkey = root(name) begin hkey.open(path, mode) do |subkey| return yield subkey end rescue Win32::Registry::Error => error raise Puppet::Util::Windows::Error.new("Failed to open registry key '#{hkey.keyname}\\#{path}'", error.code, error) end end def keys(key) keys = {} each_key(key) { |subkey, filetime| keys[subkey] = filetime } keys end # subkey is String which contains name of subkey. # wtime is last write time as FILETIME (64-bit integer). (see Registry.wtime2time) def each_key(key, &block) index = 0 subkey = nil + subkey_max_len, value_max_len = reg_query_info_key_max_lengths(key.hkey) + begin - subkey, filetime = reg_enum_key(key.hkey, index) + subkey, filetime = reg_enum_key(key.hkey, index, subkey_max_len) yield subkey, filetime if !subkey.nil? index += 1 end while !subkey.nil? index end def delete_key(key, subkey_name, mode = KEY64) reg_delete_key_ex(key.hkey, subkey_name, mode) end def values(key) vals = {} each_value(key) { |subkey, type, data| vals[subkey] = data } vals end def each_value(key, &block) index = 0 subkey = nil + subkey_max_len, value_max_len = reg_query_info_key_max_lengths(key.hkey) + begin - subkey, type, data = reg_enum_value(key.hkey, index) + subkey, type, data = reg_enum_value(key.hkey, index, value_max_len) yield subkey, type, data if !subkey.nil? index += 1 end while !subkey.nil? index end def delete_value(key, subkey_name) reg_delete_value(key.hkey, subkey_name) end private def reg_enum_key(hkey, index, max_key_length = Win32::Registry::Constants::MAX_KEY_LENGTH) subkey, filetime = nil, nil FFI::MemoryPointer.new(:dword) do |subkey_length_ptr| FFI::MemoryPointer.new(FFI::WIN32::FILETIME.size) do |filetime_ptr| FFI::MemoryPointer.new(:wchar, max_key_length) do |subkey_ptr| subkey_length_ptr.write_dword(max_key_length) # RegEnumKeyEx cannot be called twice to properly size the buffer result = RegEnumKeyExW(hkey, index, subkey_ptr, subkey_length_ptr, FFI::Pointer::NULL, FFI::Pointer::NULL, FFI::Pointer::NULL, filetime_ptr) break if result == ERROR_NO_MORE_ITEMS if result != FFI::ERROR_SUCCESS raise Puppet::Util::Windows::Error.new('Failed to enumerate registry keys') end filetime = FFI::WIN32::FILETIME.new(filetime_ptr) subkey_length = subkey_length_ptr.read_dword subkey = subkey_ptr.read_wide_string(subkey_length, Encoding::UTF_8) end end end [subkey, filetime] end def reg_enum_value(hkey, index, max_value_length = Win32::Registry::Constants::MAX_VALUE_LENGTH) subkey, type, data = nil, nil, nil FFI::MemoryPointer.new(:dword) do |subkey_length_ptr| FFI::MemoryPointer.new(:wchar, max_value_length) do |subkey_ptr| # RegEnumValueW cannot be called twice to properly size the buffer subkey_length_ptr.write_dword(max_value_length) result = RegEnumValueW(hkey, index, subkey_ptr, subkey_length_ptr, FFI::Pointer::NULL, FFI::Pointer::NULL, FFI::Pointer::NULL, FFI::Pointer::NULL ) break if result == ERROR_NO_MORE_ITEMS if result != FFI::ERROR_SUCCESS raise Puppet::Util::Windows::Error.new('Failed to enumerate registry values') end subkey_length = subkey_length_ptr.read_dword subkey = subkey_ptr.read_wide_string(subkey_length, Encoding::UTF_8) type, data = read(hkey, subkey_ptr) end end [subkey, type, data] end + def reg_query_info_key_max_lengths(hkey) + result = nil + + FFI::MemoryPointer.new(:dword) do |max_subkey_name_length_ptr| + FFI::MemoryPointer.new(:dword) do |max_value_name_length_ptr| + + status = RegQueryInfoKeyW(hkey, + FFI::MemoryPointer::NULL, FFI::MemoryPointer::NULL, + FFI::MemoryPointer::NULL, FFI::MemoryPointer::NULL, + max_subkey_name_length_ptr, FFI::MemoryPointer::NULL, + FFI::MemoryPointer::NULL, max_value_name_length_ptr, + FFI::MemoryPointer::NULL, FFI::MemoryPointer::NULL, + FFI::MemoryPointer::NULL + ) + + if status != FFI::ERROR_SUCCESS + raise Puppet::Util::Windows::Error.new('Failed to query registry key info') + end + + result = [ + # Unicode characters *not* including trailing NULL + max_subkey_name_length_ptr.read_dword + 1, + max_value_name_length_ptr.read_dword + 1, + ] + end + end + + result + end + # Read a registry value named name and return array of # [ type, data ]. # When name is nil, the `default' value is read. # type is value type. (see Win32::Registry::Constants module) # data is value data, its class is: # :REG_SZ, REG_EXPAND_SZ # String # :REG_MULTI_SZ # Array of String # :REG_DWORD, REG_DWORD_BIG_ENDIAN, REG_QWORD # Integer # :REG_BINARY # String (contains binary data) # # When rtype is specified, the value type must be included by # rtype array, or TypeError is raised. def read(hkey, name_ptr, *rtype) result = nil query_value_ex(hkey, name_ptr) do |type, data_ptr, byte_length| unless rtype.empty? or rtype.include?(type) raise TypeError, "Type mismatch (expect #{rtype.inspect} but #{type} present)" end string_length = 0 # buffer is raw bytes, *not* chars - less a NULL terminator string_length = (byte_length / FFI.type_size(:wchar)) - 1 if byte_length > 0 case type when Win32::Registry::REG_SZ, Win32::Registry::REG_EXPAND_SZ result = [ type, data_ptr.read_wide_string(string_length, Encoding::UTF_8) ] when Win32::Registry::REG_MULTI_SZ result = [ type, data_ptr.read_wide_string(string_length, Encoding::UTF_8).split(/\0/) ] when Win32::Registry::REG_BINARY result = [ type, data.read_bytes(0, byte_length) ] when Win32::Registry::REG_DWORD result = [ type, data_ptr.read_dword ] when Win32::Registry::REG_DWORD_BIG_ENDIAN result = [ type, data_ptr.order(:big).read_dword ] when Win32::Registry::REG_QWORD result = [ type, data_ptr.read_qword ] else raise TypeError, "Type #{type} is not supported." end end result end def query_value_ex(hkey, name_ptr, &block) FFI::MemoryPointer.new(:dword) do |type_ptr| FFI::MemoryPointer.new(:dword) do |length_ptr| result = RegQueryValueExW(hkey, name_ptr, FFI::Pointer::NULL, type_ptr, FFI::Pointer::NULL, length_ptr) FFI::MemoryPointer.new(:byte, length_ptr.read_dword) do |buffer_ptr| result = RegQueryValueExW(hkey, name_ptr, FFI::Pointer::NULL, type_ptr, buffer_ptr, length_ptr) if result != FFI::ERROR_SUCCESS raise Puppet::Util::Windows::Error.new("Failed to query registry value") end # allows caller to use FFI MemoryPointer helpers to read / shape yield [type_ptr.read_dword, buffer_ptr, length_ptr.read_dword] end end end end def reg_delete_value(hkey, name) result = 0 FFI::Pointer.from_string_to_wide_string(name) do |name_ptr| result = RegDeleteValueW(hkey, name_ptr) if result != FFI::ERROR_SUCCESS raise Puppet::Util::Windows::Error.new("Failed to delete registry value #{name}") end end result end def reg_delete_key_ex(hkey, name, regsam = KEY64) result = 0 FFI::Pointer.from_string_to_wide_string(name) do |name_ptr| result = RegDeleteKeyExW(hkey, name_ptr, regsam, 0) if result != FFI::ERROR_SUCCESS raise Puppet::Util::Windows::Error.new("Failed to delete registry key #{name}") end end result end ffi_convention :stdcall # https://msdn.microsoft.com/en-us/library/windows/desktop/ms724862(v=vs.85).aspx # LONG WINAPI RegEnumKeyEx( # _In_ HKEY hKey, # _In_ DWORD dwIndex, # _Out_ LPTSTR lpName, # _Inout_ LPDWORD lpcName, # _Reserved_ LPDWORD lpReserved, # _Inout_ LPTSTR lpClass, # _Inout_opt_ LPDWORD lpcClass, # _Out_opt_ PFILETIME lpftLastWriteTime # ); ffi_lib :advapi32 attach_function_private :RegEnumKeyExW, [:handle, :dword, :lpwstr, :lpdword, :lpdword, :lpwstr, :lpdword, :pointer], :win32_long # https://msdn.microsoft.com/en-us/library/windows/desktop/ms724865(v=vs.85).aspx # LONG WINAPI RegEnumValue( # _In_ HKEY hKey, # _In_ DWORD dwIndex, # _Out_ LPTSTR lpValueName, # _Inout_ LPDWORD lpcchValueName, # _Reserved_ LPDWORD lpReserved, # _Out_opt_ LPDWORD lpType, # _Out_opt_ LPBYTE lpData, # _Inout_opt_ LPDWORD lpcbData # ); ffi_lib :advapi32 attach_function_private :RegEnumValueW, [:handle, :dword, :lpwstr, :lpdword, :lpdword, :lpdword, :lpbyte, :lpdword], :win32_long # https://msdn.microsoft.com/en-us/library/windows/desktop/ms724911(v=vs.85).aspx # LONG WINAPI RegQueryValueExW( # _In_ HKEY hKey, # _In_opt_ LPCTSTR lpValueName, # _Reserved_ LPDWORD lpReserved, # _Out_opt_ LPDWORD lpType, # _Out_opt_ LPBYTE lpData, # _Inout_opt_ LPDWORD lpcbData # ); ffi_lib :advapi32 attach_function_private :RegQueryValueExW, [:handle, :lpcwstr, :lpdword, :lpdword, :lpbyte, :lpdword], :win32_long # LONG WINAPI RegDeleteValue( # _In_ HKEY hKey, # _In_opt_ LPCTSTR lpValueName # ); ffi_lib :advapi32 attach_function_private :RegDeleteValueW, [:handle, :lpcwstr], :win32_long # LONG WINAPI RegDeleteKeyEx( # _In_ HKEY hKey, # _In_ LPCTSTR lpSubKey, # _In_ REGSAM samDesired, # _Reserved_ DWORD Reserved # ); ffi_lib :advapi32 attach_function_private :RegDeleteKeyExW, [:handle, :lpcwstr, :win32_ulong, :dword], :win32_long + + # https://msdn.microsoft.com/en-us/library/windows/desktop/ms724902(v=vs.85).aspx + # LONG WINAPI RegQueryInfoKey( + # _In_ HKEY hKey, + # _Out_opt_ LPTSTR lpClass, + # _Inout_opt_ LPDWORD lpcClass, + # _Reserved_ LPDWORD lpReserved, + # _Out_opt_ LPDWORD lpcSubKeys, + # _Out_opt_ LPDWORD lpcMaxSubKeyLen, + # _Out_opt_ LPDWORD lpcMaxClassLen, + # _Out_opt_ LPDWORD lpcValues, + # _Out_opt_ LPDWORD lpcMaxValueNameLen, + # _Out_opt_ LPDWORD lpcMaxValueLen, + # _Out_opt_ LPDWORD lpcbSecurityDescriptor, + # _Out_opt_ PFILETIME lpftLastWriteTime + # ); + ffi_lib :advapi32 + attach_function_private :RegQueryInfoKeyW, + [:handle, :lpwstr, :lpdword, :lpdword, :lpdword, :lpdword, :lpdword, + :lpdword, :lpdword, :lpdword, :lpdword, :pointer], :win32_long end end