--- a/environment/mapper.go Fri Oct 13 16:45:41 2017 +0000
+++ b/environment/mapper.go Fri Oct 13 17:04:10 2017 +0000
@@ -27,6 +27,10 @@
func (e *envMapper) Map(name string) string {
for _, item := range e.env {
if parts := strings.SplitN(item, "=", 2); parts[0] == name {
@@ -53,13 +57,18 @@
func Mapper(str string, env []string) (string, error) {
next := os.Expand(last, mapper.Map)
prev := map[string]struct{}{}
+ for i := 0; i < MaxExpandWidth; i++ { if _, ok := prev[next]; ok {
- return "", fmt.Errorf("infinite environment mapping loop while expanding '%s'", next)
@@ -67,7 +76,7 @@
+ return "", fmt.Errorf("infinite environment mapping loop while expanding '%s'", orig) // SliceMapper calls Mapper for each item in a slice and returns a new slice.
--- a/environment/mapper_test.go Fri Oct 13 16:45:41 2017 +0000
+++ b/environment/mapper_test.go Fri Oct 13 17:04:10 2017 +0000
@@ -75,7 +75,12 @@
Expect(result).To(Equal("123-ohhai!-cba"))
-func (e *environmentSuite) TestInfiniteExpansion(t sweet.T) {
+func (e *environmentSuite) TestInfiniteDepthExpansion(t sweet.T) { _, err := Mapper("${FOO}", []string{"FOO=$BAR", "BAR=$FOO"})
- Expect(err).To(MatchError("infinite environment mapping loop while expanding '$BAR'"))
+ Expect(err).To(MatchError("infinite environment mapping loop while expanding '${FOO}'")) +func (e *environmentSuite) TestInfiniteWidthExpansion(t sweet.T) { + _, err := Mapper("${FOO}", []string{"FOO=FOO${FOO}"}) + Expect(err).To(MatchError("infinite environment mapping loop while expanding '${FOO}'"))