Skip to content

Commit

Permalink
agent: Disallow 0.0.0.0 as advertise or advertise-wan address
Browse files Browse the repository at this point in the history
Fixes #2961
  • Loading branch information
magiconair committed May 9, 2017
1 parent 73a31b9 commit e365ef1
Show file tree
Hide file tree
Showing 2 changed files with 41 additions and 0 deletions.
10 changes: 10 additions & 0 deletions command/agent/command.go
Original file line number Diff line number Diff line change
Expand Up @@ -361,6 +361,16 @@ func (c *Command) readConfig() *Config {
return nil
}

if config.AdvertiseAddr == "0.0.0.0" {
c.UI.Error("Advertise address cannot be 0.0.0.0")
return nil
}

if config.AdvertiseAddrWan == "0.0.0.0" {
c.UI.Error("Advertise WAN address cannot be 0.0.0.0")
return nil
}

// Compile all the watches
for _, params := range config.Watches {
// Parse the watches, excluding the handler
Expand Down
31 changes: 31 additions & 0 deletions command/agent/command_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ import (
"io/ioutil"
"log"
"os"
"os/exec"
"path/filepath"
"reflect"
"strings"
Expand Down Expand Up @@ -50,6 +51,36 @@ func TestValidDatacenter(t *testing.T) {
}
}

// TestConfigFail should test command line flags that lead to an immediate error.
func TestConfigFail(t *testing.T) {
tests := []struct {
args []string
out string
}{
{
args: []string{"agent", "-server", "-data-dir", "foo", "-advertise", "0.0.0.0"},
out: "==> Advertise address cannot be 0.0.0.0\n",
},
{
args: []string{"agent", "-server", "-data-dir", "foo", "-advertise-wan", "0.0.0.0"},
out: "==> Advertise WAN address cannot be 0.0.0.0\n",
},
}

for _, tt := range tests {
t.Run(strings.Join(tt.args, " "), func(t *testing.T) {
cmd := exec.Command("consul", tt.args...)
b, err := cmd.CombinedOutput()
if got, want := err, "exit status 1"; got == nil || got.Error() != want {
t.Fatalf("got err %q want %q", got, want)
}
if got, want := string(b), tt.out; got != want {
t.Fatalf("got %q want %q", got, want)
}
})
}
}

func TestRetryJoin(t *testing.T) {
dir, agent := makeAgent(t, nextConfig())
defer os.RemoveAll(dir)
Expand Down

0 comments on commit e365ef1

Please sign in to comment.