68 lines
2.3 KiB
Diff
68 lines
2.3 KiB
Diff
|
diff --git a/src/OVAL/probes/independent/yamlfilecontent_probe.c b/src/OVAL/probes/independent/yamlfilecontent_probe.c
|
||
|
|
index 6f18abf83..17741a240 100644
|
||
|
|
--- a/src/OVAL/probes/independent/yamlfilecontent_probe.c
|
||
|
|
+++ b/src/OVAL/probes/independent/yamlfilecontent_probe.c
|
||
|
|
@@ -206,6 +206,7 @@ static int yaml_path_query(const char *filepath, const char *yaml_path_cstr, str
|
||
|
|
yaml_event_type_t event_type;
|
||
|
|
bool sequence = false;
|
||
|
|
bool mapping = false;
|
||
|
|
+ bool fake_mapping = false;
|
||
|
|
int index = 0;
|
||
|
|
char *key = strdup("#");
|
||
|
|
|
||
|
|
@@ -224,21 +225,39 @@ static int yaml_path_query(const char *filepath, const char *yaml_path_cstr, str
|
||
|
|
|
||
|
|
if (sequence) {
|
||
|
|
if (event_type == YAML_SEQUENCE_END_EVENT) {
|
||
|
|
- sequence = false;
|
||
|
|
+ if (fake_mapping) {
|
||
|
|
+ fake_mapping = false;
|
||
|
|
+ if (record && record->itemcount > 0) {
|
||
|
|
+ oscap_list_add(values, record);
|
||
|
|
+ } else {
|
||
|
|
+ // Do not collect empty records
|
||
|
|
+ oscap_htable_free0(record);
|
||
|
|
+ }
|
||
|
|
+ record = NULL;
|
||
|
|
+ } else {
|
||
|
|
+ sequence = false;
|
||
|
|
+ }
|
||
|
|
} else if (event_type == YAML_SEQUENCE_START_EVENT) {
|
||
|
|
- result_error("YAML path '%s' points to a multi-dimensional structure (sequence containing another sequence)", yaml_path_cstr);
|
||
|
|
- goto cleanup;
|
||
|
|
+ if (mapping || fake_mapping) {
|
||
|
|
+ result_error("YAML path '%s' points to a multi-dimensional structure (a map or a sequence containing other sequences)", yaml_path_cstr);
|
||
|
|
+ goto cleanup;
|
||
|
|
+ } else {
|
||
|
|
+ fake_mapping = true;
|
||
|
|
+ record = oscap_htable_new();
|
||
|
|
+ }
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
if (event_type == YAML_SEQUENCE_START_EVENT) {
|
||
|
|
sequence = true;
|
||
|
|
+ if (mapping)
|
||
|
|
+ index++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (mapping) {
|
||
|
|
if (event_type == YAML_MAPPING_END_EVENT) {
|
||
|
|
mapping = false;
|
||
|
|
- if (record->itemcount > 0) {
|
||
|
|
+ if (record && record->itemcount > 0) {
|
||
|
|
oscap_list_add(values, record);
|
||
|
|
} else {
|
||
|
|
// Do not collect empty records
|
||
|
|
@@ -255,6 +274,10 @@ static int yaml_path_query(const char *filepath, const char *yaml_path_cstr, str
|
||
|
|
result_error("YAML path '%s' points to an invalid structure (map containing another map)", yaml_path_cstr);
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
+ if (fake_mapping) {
|
||
|
|
+ result_error("YAML path '%s' points to a multi-dimensional structure (two-dimensional sequence containing a map)", yaml_path_cstr);
|
||
|
|
+ goto cleanup;
|
||
|
|
+ }
|
||
|
|
mapping = true;
|
||
|
|
sequence = false;
|
||
|
|
index = 0;
|