Contents

class Specinfra::Command::Windows::Base::RegistryKey < Specinfra::Command::Windows::Base
  class << self
    REGISTRY_KEY_TYPES = {
      :type_string       => 'String',
      :type_binary       => 'Binary',
      :type_dword        => 'DWord',
      :type_qword        => 'QWord',
      :type_multistring  => 'MultiString',
      :type_expandstring => 'ExpandString'
    }

    def check_exists(key_name)
      cmd = "(Get-Item 'Registry::#{key_name}') -ne $null"
      Backend::PowerShell::Command.new { exec cmd }
    end

    def check_has_property(key_name, key_property)
      cmd = "(Get-Item 'Registry::#{key_name}').GetValueKind('#{key_property[:name]}') -eq '#{get_key_type([key_property[:type]])}'"
      Backend::PowerShell::Command.new { exec cmd }
    end

    def check_has_value(key_name, key_property)
      value = convert_key_property_value key_property
      cmd = "(Compare-Object (Get-Item 'Registry::#{key_name}').GetValue('#{key_property[:name]}') #{value}) -eq $null"
      Backend::PowerShell::Command.new { exec cmd }
    end

    private
    def do_not_convert?(key_type)
      key_type.to_s =~ /_converted/i
    end

    def get_key_type(key_type)
       REGISTRY_KEY_TYPES[key_type.to_s.gsub("_converted").to_sym]
    end

    def convert_key_property_value property
      return property if do_not_convert? property[:type]
      case property[:type]
      when :type_binary
        byte_array = [property[:value]].pack('H*').bytes.to_a
        "([byte[]] #{byte_array.join(',')})"
      when :type_dword
        [property[:value].rjust(8, '0').scan(/[0-9a-f]{2}/i).reverse.join].pack("H*").unpack("l").first
      when :type_qword
        property[:value].hex
      else
        string_array = property[:value].split("\n").map {|s| "'#{s}'"}.reduce {|acc, s| "#{acc},#{s}"}
        "@(#{string_array})"
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
specinfra-2.35.0	lib/specinfra/command/windows/base/registry_key.rb
specinfra-2.34.9	lib/specinfra/command/windows/base/registry_key.rb