-
Notifications
You must be signed in to change notification settings - Fork 125
/
Registry.pm
90 lines (67 loc) · 1.99 KB
/
Registry.pm
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package FusionInventory::Agent::Task::Inventory::Input::Win32::Registry;
use strict;
use warnings;
use English qw(-no_match_vars);
use FusionInventory::Agent::Tools;
use FusionInventory::Agent::Tools::Win32;
my @hives = qw/
HKEY_CLASSES_ROOT
HKEY_CURRENT_USER
HKEY_LOCAL_MACHINE
HKEY_USERS
HKEY_CURRENT_CONFIG
HKEY_DYN_DATA
/;
sub isEnabled {
my (%params) = @_;
return $params{registry};
}
sub _getRegistryData {
my (%params) = @_;
my @data;
my @registrys = ref($params{registry}->{PARAM}) eq 'ARRAY' ?
@{$params{registry}->{PARAM}} :
($params{registry}->{PARAM});
foreach my $option (@registrys) {
my $name = $option->{NAME};
my $regkey = $option->{REGKEY};
my $regtree = $option->{REGTREE};
my $content = $option->{content};
# This should never append, err wait...
next unless $content;
$regkey =~ s{\\}{/}g;
my $value = getRegistryValue(
path => $hives[$regtree]."/".$regkey."/".$content,
logger => $params{logger}
);
if (ref($value) eq "HASH") {
foreach ( keys %$value ) {
my $n = encodeFromRegistry($_) || '';
my $v = encodeFromRegistry($value->{$_}) || '';
push @data, { section => 'REGISTRY', entry => {
NAME => $name,
REGVALUE => "$n=$v"
}
};
}
} else {
push @data, {section => 'REGISTRY', entry => {
NAME => $name,
REGVALUE => encodeFromRegistry($value)
}
};
}
}
return @data;
}
sub doInventory {
my (%params) = @_;
return unless $params{registry}->{NAME} eq 'REGISTRY';
foreach my $data (_getRegistryData(
registry => $params{registry},
logger => $params{logger}))
{
$params{inventory}->addEntry(%$data);
}
}
1;