mirror of https://github.com/status-im/consul.git
Merge pull request #655 from hashicorp/f-exec-acl
agent: pass locally configured token during remote exec
This commit is contained in:
commit
a95b5a63bb
|
@ -242,6 +242,7 @@ func (a *Agent) remoteExecGetSpec(event *remoteExecEvent, spec *remoteExecSpec)
|
||||||
AllowStale: true, // Stale read for scale! Retry on failure.
|
AllowStale: true, // Stale read for scale! Retry on failure.
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
get.Token = a.config.ACLToken
|
||||||
var out structs.IndexedDirEntries
|
var out structs.IndexedDirEntries
|
||||||
QUERY:
|
QUERY:
|
||||||
if err := a.RPC("KVS.Get", &get, &out); err != nil {
|
if err := a.RPC("KVS.Get", &get, &out); err != nil {
|
||||||
|
@ -308,6 +309,7 @@ func (a *Agent) remoteExecWriteKey(event *remoteExecEvent, suffix string, val []
|
||||||
Session: event.Session,
|
Session: event.Session,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
write.Token = a.config.ACLToken
|
||||||
var success bool
|
var success bool
|
||||||
if err := a.RPC("KVS.Apply", &write, &success); err != nil {
|
if err := a.RPC("KVS.Apply", &write, &success); err != nil {
|
||||||
return err
|
return err
|
||||||
|
|
|
@ -84,7 +84,20 @@ func TestRexecWriter(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestRemoteExecGetSpec(t *testing.T) {
|
func TestRemoteExecGetSpec(t *testing.T) {
|
||||||
dir, agent := makeAgent(t, nextConfig())
|
config := nextConfig()
|
||||||
|
testRemoteExecGetSpec(t, config)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRemoteExecGetSpec_ACLToken(t *testing.T) {
|
||||||
|
config := nextConfig()
|
||||||
|
config.ACLDatacenter = "dc1"
|
||||||
|
config.ACLToken = "root"
|
||||||
|
config.ACLDefaultPolicy = "deny"
|
||||||
|
testRemoteExecGetSpec(t, config)
|
||||||
|
}
|
||||||
|
|
||||||
|
func testRemoteExecGetSpec(t *testing.T, c *Config) {
|
||||||
|
dir, agent := makeAgent(t, c)
|
||||||
defer os.RemoveAll(dir)
|
defer os.RemoveAll(dir)
|
||||||
defer agent.Shutdown()
|
defer agent.Shutdown()
|
||||||
testutil.WaitForLeader(t, agent.RPC, "dc1")
|
testutil.WaitForLeader(t, agent.RPC, "dc1")
|
||||||
|
@ -117,7 +130,20 @@ func TestRemoteExecGetSpec(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestRemoteExecWrites(t *testing.T) {
|
func TestRemoteExecWrites(t *testing.T) {
|
||||||
dir, agent := makeAgent(t, nextConfig())
|
config := nextConfig()
|
||||||
|
testRemoteExecWrites(t, config)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRemoteExecWrites_ACLToken(t *testing.T) {
|
||||||
|
config := nextConfig()
|
||||||
|
config.ACLDatacenter = "dc1"
|
||||||
|
config.ACLToken = "root"
|
||||||
|
config.ACLDefaultPolicy = "deny"
|
||||||
|
testRemoteExecWrites(t, config)
|
||||||
|
}
|
||||||
|
|
||||||
|
func testRemoteExecWrites(t *testing.T, c *Config) {
|
||||||
|
dir, agent := makeAgent(t, c)
|
||||||
defer os.RemoveAll(dir)
|
defer os.RemoveAll(dir)
|
||||||
defer agent.Shutdown()
|
defer agent.Shutdown()
|
||||||
testutil.WaitForLeader(t, agent.RPC, "dc1")
|
testutil.WaitForLeader(t, agent.RPC, "dc1")
|
||||||
|
@ -275,6 +301,7 @@ func setKV(t *testing.T, agent *Agent, key string, val []byte) {
|
||||||
Value: val,
|
Value: val,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
write.Token = agent.config.ACLToken
|
||||||
var success bool
|
var success bool
|
||||||
if err := agent.RPC("KVS.Apply", &write, &success); err != nil {
|
if err := agent.RPC("KVS.Apply", &write, &success); err != nil {
|
||||||
t.Fatalf("err: %v", err)
|
t.Fatalf("err: %v", err)
|
||||||
|
@ -286,6 +313,7 @@ func getKV(t *testing.T, agent *Agent, key string) *structs.DirEntry {
|
||||||
Datacenter: agent.config.Datacenter,
|
Datacenter: agent.config.Datacenter,
|
||||||
Key: key,
|
Key: key,
|
||||||
}
|
}
|
||||||
|
req.Token = agent.config.ACLToken
|
||||||
var out structs.IndexedDirEntries
|
var out structs.IndexedDirEntries
|
||||||
if err := agent.RPC("KVS.Get", &req, &out); err != nil {
|
if err := agent.RPC("KVS.Get", &req, &out); err != nil {
|
||||||
t.Fatalf("err: %v", err)
|
t.Fatalf("err: %v", err)
|
||||||
|
|
Loading…
Reference in New Issue