Skip to content

Commit

Permalink
hclsyntax: Fix for expression marked conditional
Browse files Browse the repository at this point in the history
Using marked values in a for expression conditional expression would
previously panic due to calling `.False()` on the result of the
expression.

This commit makes two changes:

- Unmark the conditional expression value before testing if it is false;
- Merge any marks from the conditional into the resulting marks for the
  for expression.
  • Loading branch information
alisdair committed Jan 4, 2021
1 parent 94940f5 commit fe026c3
Show file tree
Hide file tree
Showing 2 changed files with 58 additions and 2 deletions.
24 changes: 22 additions & 2 deletions hclsyntax/expression.go
Original file line number Diff line number Diff line change
Expand Up @@ -1143,7 +1143,17 @@ func (e *ForExpr) Value(ctx *hcl.EvalContext) (cty.Value, hcl.Diagnostics) {
continue
}

if include.False() {
// Extract and merge marks from the include expression into the
// main set of marks
includeUnmarked, includeMarks := include.Unmark()
if marks == nil {
marks = includeMarks
} else {
for k := range includeMarks {
marks[k] = struct{}{}
}
}
if includeUnmarked.False() {
// Skip this element
continue
}
Expand Down Expand Up @@ -1300,7 +1310,17 @@ func (e *ForExpr) Value(ctx *hcl.EvalContext) (cty.Value, hcl.Diagnostics) {
continue
}

if include.False() {
// Extract and merge marks from the include expression into the
// main set of marks
includeUnmarked, includeMarks := include.Unmark()
if marks == nil {
marks = includeMarks
} else {
for k := range includeMarks {
marks[k] = struct{}{}
}
}
if includeUnmarked.False() {
// Skip this element
continue
}
Expand Down
36 changes: 36 additions & 0 deletions hclsyntax/expression_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -933,6 +933,42 @@ upper(
}),
1,
},
{ // Sequence for loop with marked conditional expression
`[for x in things: x if x != secret]`,
&hcl.EvalContext{
Variables: map[string]cty.Value{
"things": cty.ListVal([]cty.Value{
cty.StringVal("a"),
cty.StringVal("b"),
cty.StringVal("c"),
}),
"secret": cty.StringVal("b").Mark("sensitive"),
},
},
cty.TupleVal([]cty.Value{
cty.StringVal("a"),
cty.StringVal("c"),
}).Mark("sensitive"),
0,
},
{ // Map for loop with marked conditional expression
`{ for k, v in things: k => v if k != secret }`,
&hcl.EvalContext{
Variables: map[string]cty.Value{
"things": cty.MapVal(map[string]cty.Value{
"a": cty.True,
"b": cty.False,
"c": cty.False,
}),
"secret": cty.StringVal("b").Mark("sensitive"),
},
},
cty.ObjectVal(map[string]cty.Value{
"a": cty.True,
"c": cty.False,
}).Mark("sensitive"),
0,
},
{
`[{name: "Steve"}, {name: "Ermintrude"}].*.name`,
nil,
Expand Down

0 comments on commit fe026c3

Please sign in to comment.