win_reg_stat change the module parameters for standardisation (#22732)

pull/22443/merge
Jordan Borean 8 years ago committed by Matt Davis
parent 89b78cb5e8
commit f1ab879bb6

@ -19,14 +19,13 @@
$ErrorActionPreference = "Stop" $ErrorActionPreference = "Stop"
$params = Parse-Args $args -supports_check_mode $true $params = Parse-Args -arguments $args -supports_check_mode $true
$key = Get-AnsibleParam -obj $params -name "key" -type "str" -failifempty $true $path = Get-AnsibleParam -obj $params -name "path" -type "str" -failifempty $true -aliases "key"
$property = Get-AnsibleParam -obj $params -name "property" -type "str" $name = Get-AnsibleParam -obj $params -name "name" -type "str" -aliases "entry","value"
$result = @{ $result = @{
changed = $false changed = $false
win_reg_stat = @{}
} }
Function Get-NetHiveName($hive) { Function Get-NetHiveName($hive) {
@ -94,57 +93,57 @@ Function Test-RegistryProperty($hive, $path, $property) {
} }
# Will validate the key parameter to make sure it matches known format # Will validate the key parameter to make sure it matches known format
if ($key -match "^([a-zA-Z_]*):\\(.*)$") { if ($path -match "^([a-zA-Z_]*):\\(.*)$") {
$hive = $matches[1] $hive = $matches[1]
$path = $matches[2] $reg_path = $matches[2]
} else { } else {
Fail-Json $result "key does not match format 'HIVE:\KEY_PATH'" Fail-Json $result "path does not match format 'HIVE:\KEY_PATH'"
} }
# Used when getting the actual REG_EXPAND_SZ value as well as checking the hive is a known value # Used when getting the actual REG_EXPAND_SZ value as well as checking the hive is a known value
$net_hive = Get-NetHiveName -hive $hive $net_hive = Get-NetHiveName -hive $hive
if ($net_hive -eq 'unsupported') { if ($net_hive -eq 'unsupported') {
Fail-Json $result "the hive in key is '$hive'; must be 'HKCR', 'HKCC', 'HKCU', 'HKLM' or 'HKU'" Fail-Json $result "the hive in path is '$hive'; must be 'HKCR', 'HKCC', 'HKCU', 'HKLM' or 'HKU'"
} }
if (Test-Path REGISTRY::$hive\$path) { if (Test-Path REGISTRY::$hive\$reg_path) {
if ($property -eq $null) { if ($name -eq $null) {
$property_info = @{} $property_info = @{}
$properties = Get-ItemProperty REGISTRY::$hive\$path $properties = Get-ItemProperty REGISTRY::$hive\$reg_path
foreach ($property in $properties.PSObject.Properties) { foreach ($property in $properties.PSObject.Properties) {
# Powershell adds in some metadata we need to filter out # Powershell adds in some metadata we need to filter out
$real_property = Test-RegistryProperty -hive $hive -path $path -property $property.Name $real_property = Test-RegistryProperty -hive $hive -path $reg_path -property $property.Name
if ($real_property -eq $true) { if ($real_property -eq $true) {
$property_object = Get-PropertyObject -hive $hive -net_hive $net_hive -path $path -property $property.Name $property_object = Get-PropertyObject -hive $hive -net_hive $net_hive -path $reg_path -property $property.Name
$property_info.Add($property.Name, $property_object) $property_info.Add($property.Name, $property_object)
} }
} }
$sub_keys = @() $sub_keys = @()
$sub_keys_raw = Get-ChildItem REGISTRY::$hive\$path -ErrorAction SilentlyContinue $sub_keys_raw = Get-ChildItem REGISTRY::$hive\$reg_path -ErrorAction SilentlyContinue
foreach ($sub_key in $sub_keys_raw) { foreach ($sub_key in $sub_keys_raw) {
$sub_keys += $sub_key.PSChildName $sub_keys += $sub_key.PSChildName
} }
$result.win_reg_stat.exists = $true $result.exists = $true
$result.win_reg_stat.sub_keys = $sub_keys $result.sub_keys = $sub_keys
$result.win_reg_stat.properties = $property_info $result.properties = $property_info
} else { } else {
$exists = Test-RegistryProperty -hive $hive -path $path -property $property $exists = Test-RegistryProperty -hive $hive -path $reg_path -property $name
if ($exists -eq $true) { if ($exists -eq $true) {
$propertyObject = Get-PropertyObject -hive $hive -net_hive $net_hive -path $path -property $property $propertyObject = Get-PropertyObject -hive $hive -net_hive $net_hive -path $reg_path -property $name
$result.win_reg_stat.exists = $true $result.exists = $true
$result.win_reg_stat.raw_value = $propertyObject.raw_value $result.raw_value = $propertyObject.raw_value
$result.win_reg_stat.value = $propertyObject.value $result.value = $propertyObject.value
$result.win_reg_stat.type = $propertyObject.type $result.type = $propertyObject.type
} else { } else {
$result.win_reg_stat.exists = $false $result.exists = $false
} }
} }
} else { } else {
$result.win_reg_stat.exists = $false $result.exists = $false
} }
Exit-Json $result Exit-Json $result

@ -36,28 +36,28 @@ description:
- It also returns the sub keys and properties of the key specified. - It also returns the sub keys and properties of the key specified.
- If specifying a property name through I(property), it will return the information specific for that property. - If specifying a property name through I(property), it will return the information specific for that property.
options: options:
key: path:
description: description: The full registry key path including the hive to search for.
- The full registry key path including the hive to search for.
required: true required: true
property: aliases: [ key ]
name:
description: description:
- The registry property name to get information for, the return json will not include the sub_keys and properties entries for the I(key) specified. - The registry property name to get information for, the return json will not include the sub_keys and properties entries for the I(key) specified.
required: false required: false
aliases: [ entry, value, property ]
author: "Jordan Borean (@jborean93)" author: "Jordan Borean (@jborean93)"
''' '''
EXAMPLES = r''' EXAMPLES = r'''
# Obtain information about a registry key using short form # Obtain information about a registry key using short form
- win_reg_stat: - win_reg_stat:
key: HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion path: HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion
register: current_version register: current_version
# Obtain information about a registry key property # Obtain information about a registry key property
- win_reg_stat: - win_reg_stat:
key: HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion path: HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion
property: CommonFilesDir name: CommonFilesDir
register: common_files_dir register: common_files_dir
''' '''
@ -67,11 +67,6 @@ changed:
returned: always returned: always
type: boolean type: boolean
sample: True sample: True
win_reg_stat:
description: Information about the registry key or property specified.
returned: success
type: dictionary
contains:
exists: exists:
description: States whether the registry key/property exists. description: States whether the registry key/property exists.
returned: success and path/property exists returned: success and path/property exists

@ -1,4 +1,9 @@
--- ---
- name: test
win_file:
path: "{{win_output_dir}}"
state: absent
- name: make sure win output dir exists - name: make sure win output dir exists
win_file: win_file:
path: "{{win_output_dir}}" path: "{{win_output_dir}}"
@ -20,33 +25,32 @@
win_command: powershell.exe $env:windir win_command: powershell.exe $env:windir
register: win_dir_value register: win_dir_value
- name: expect failure when not passing in key option - name: expect failure when not passing in path option
win_reg_stat: win_reg_stat:
property: a name: a
register: actual register: actual
failed_when: "actual.msg != 'Missing required argument: key'" failed_when: "actual.msg != 'Missing required argument: path'"
- name: expect failure when passing in an invalid hive - name: expect failure when passing in an invalid hive
win_reg_stat: win_reg_stat:
key: ABCD:\test path: ABCD:\test
register: actual register: actual
failed_when: actual.msg != "the hive in key is 'ABCD'; must be 'HKCR', 'HKCC', 'HKCU', 'HKLM' or 'HKU'" failed_when: actual.msg != "the hive in path is 'ABCD'; must be 'HKCR', 'HKCC', 'HKCU', 'HKLM' or 'HKU'"
- name: get known nested reg key structure for testing with short hive form - name: get known nested reg key structure for testing with short hive form
win_reg_stat: win_reg_stat:
key: HKCU:\Test\nested path: HKCU:\Test\nested
register: actual_short register: actual_short
- name: get known nested reg key structure for testing with quoted yaml - name: get known nested reg key structure for testing with quoted yaml
win_reg_stat: win_reg_stat:
key: "HKCU:\\Test\\nested" path: "HKCU:\\Test\\nested"
register: actual_quoted register: actual_quoted
- name: set expected value for reg structure - name: set expected value for reg structure
set_fact: set_fact:
expected: expected:
changed: false changed: false
win_reg_stat:
exists: true exists: true
properties: properties:
binary: { raw_value: ["0x01", "0x16"], type: 'REG_BINARY', value: [1, 22] } binary: { raw_value: ["0x01", "0x16"], type: 'REG_BINARY', value: [1, 22] }
@ -67,14 +71,13 @@
- name: get known reg key with no sub keys but some properties - name: get known reg key with no sub keys but some properties
win_reg_stat: win_reg_stat:
key: HKCU:\Test\single path: HKCU:\Test\single
register: actual register: actual
- name: set expected value for reg key with no sub keys but some properties - name: set expected value for reg key with no sub keys but some properties
set_fact: set_fact:
expected: expected:
changed: false changed: false
win_reg_stat:
exists: true exists: true
properties: properties:
none: { raw_value: [], type: 'REG_NONE', value: [] } none: { raw_value: [], type: 'REG_NONE', value: [] }
@ -90,14 +93,13 @@
- name: get known reg key without sub keys and properties - name: get known reg key without sub keys and properties
win_reg_stat: win_reg_stat:
key: HKCU:\Test\nested\nest2 path: HKCU:\Test\nested\nest2
register: actual register: actual
- name: set expected value for reg key without sub keys or properties - name: set expected value for reg key without sub keys or properties
set_fact: set_fact:
expected: expected:
changed: false changed: false
win_reg_stat:
exists: true exists: true
properties: {} properties: {}
sub_keys: [] sub_keys: []
@ -110,14 +112,13 @@
- name: get non-existant reg key - name: get non-existant reg key
win_reg_stat: win_reg_stat:
key: HKCU:\Test\Thispathwillneverexist path: HKCU:\Test\Thispathwillneverexist
register: actual register: actual
- name: set expected value for non-existant reg key - name: set expected value for non-existant reg key
set_fact: set_fact:
expected: expected:
changed: false changed: false
win_reg_stat:
exists: false exists: false
- name: validate test - name: validate test
@ -127,15 +128,14 @@
- name: get string property - name: get string property
win_reg_stat: win_reg_stat:
key: HKCU:\Test\nested path: HKCU:\Test\nested
property: string name: string
register: actual register: actual
- name: set expected string property - name: set expected string property
set_fact: set_fact:
expected: expected:
changed: false changed: false
win_reg_stat:
exists: true exists: true
raw_value: 'test' raw_value: 'test'
type: 'REG_SZ' type: 'REG_SZ'
@ -148,15 +148,14 @@
- name: get expand string property - name: get expand string property
win_reg_stat: win_reg_stat:
key: HKCU:\Test\nested path: HKCU:\Test\nested
property: expand name: expand
register: actual register: actual
- name: set expected expand string property - name: set expected expand string property
set_fact: set_fact:
expected: expected:
changed: false changed: false
win_reg_stat:
exists: true exists: true
raw_value: '%windir%\dir' raw_value: '%windir%\dir'
type: 'REG_EXPAND_SZ' type: 'REG_EXPAND_SZ'
@ -169,15 +168,14 @@
- name: get multi string property - name: get multi string property
win_reg_stat: win_reg_stat:
key: HKCU:\Test\nested path: HKCU:\Test\nested
property: multi name: multi
register: actual register: actual
- name: set expected multi string property - name: set expected multi string property
set_fact: set_fact:
expected: expected:
changed: false changed: false
win_reg_stat:
exists: true exists: true
raw_value: ['a, b', 'c'] raw_value: ['a, b', 'c']
type: 'REG_MULTI_SZ' type: 'REG_MULTI_SZ'
@ -190,15 +188,14 @@
- name: get binary property - name: get binary property
win_reg_stat: win_reg_stat:
key: HKCU:\Test\nested path: HKCU:\Test\nested
property: binary name: binary
register: actual register: actual
- name: set expected binary property - name: set expected binary property
set_fact: set_fact:
expected: expected:
changed: false changed: false
win_reg_stat:
exists: true exists: true
raw_value: ["0x01", "0x16"] raw_value: ["0x01", "0x16"]
type: 'REG_BINARY' type: 'REG_BINARY'
@ -211,15 +208,14 @@
- name: get dword property - name: get dword property
win_reg_stat: win_reg_stat:
key: HKCU:\Test\nested path: HKCU:\Test\nested
property: dword name: dword
register: actual register: actual
- name: set expected dword property - name: set expected dword property
set_fact: set_fact:
expected: expected:
changed: false changed: false
win_reg_stat:
exists: true exists: true
raw_value: 1 raw_value: 1
type: 'REG_DWORD' type: 'REG_DWORD'
@ -232,15 +228,14 @@
- name: get qword property - name: get qword property
win_reg_stat: win_reg_stat:
key: HKCU:\Test\nested path: HKCU:\Test\nested
property: qword name: qword
register: actual register: actual
- name: set expected qword property - name: set expected qword property
set_fact: set_fact:
expected: expected:
changed: false changed: false
win_reg_stat:
exists: true exists: true
raw_value: 1 raw_value: 1
type: 'REG_QWORD' type: 'REG_QWORD'
@ -253,15 +248,14 @@
- name: get none property - name: get none property
win_reg_stat: win_reg_stat:
key: HKCU:\Test\single path: HKCU:\Test\single
property: none name: none
register: actual register: actual
- name: set expected none property - name: set expected none property
set_fact: set_fact:
expected: expected:
changed: false changed: false
win_reg_stat:
exists: true exists: true
raw_value: [] raw_value: []
type: 'REG_NONE' type: 'REG_NONE'
@ -274,15 +268,14 @@
- name: get none with value property - name: get none with value property
win_reg_stat: win_reg_stat:
key: HKCU:\Test\single path: HKCU:\Test\single
property: none1 name: none1
register: actual register: actual
- name: set expected none with value property - name: set expected none with value property
set_fact: set_fact:
expected: expected:
changed: false changed: false
win_reg_stat:
exists: true exists: true
raw_value: ["0x00"] raw_value: ["0x00"]
type: 'REG_NONE' type: 'REG_NONE'
@ -295,15 +288,14 @@
- name: get non-existance property - name: get non-existance property
win_reg_stat: win_reg_stat:
key: HKCU:\Test\single path: HKCU:\Test\single
property: doesnotexist name: doesnotexist
register: actual register: actual
- name: set expected non-existance property - name: set expected non-existance property
set_fact: set_fact:
expected: expected:
changed: false changed: false
win_reg_stat:
exists: false exists: false
- name: validate test - name: validate test
@ -313,5 +305,5 @@
- name: remove registry entry - name: remove registry entry
win_regedit: win_regedit:
key: HKCU:\Test path: HKCU:\Test
state: absent state: absent

Loading…
Cancel
Save