Merge pull request #3339 from hashicorp/f-acl-force-periodic
Force Periodic ACL enforcement
This commit is contained in:
commit
8c1a97765e
|
@ -6,6 +6,7 @@ import (
|
|||
|
||||
"github.com/armon/go-metrics"
|
||||
memdb "github.com/hashicorp/go-memdb"
|
||||
"github.com/hashicorp/nomad/acl"
|
||||
"github.com/hashicorp/nomad/nomad/structs"
|
||||
)
|
||||
|
||||
|
@ -21,6 +22,13 @@ func (p *Periodic) Force(args *structs.PeriodicForceRequest, reply *structs.Peri
|
|||
}
|
||||
defer metrics.MeasureSince([]string{"nomad", "periodic", "force"}, time.Now())
|
||||
|
||||
// Check for write-job permissions
|
||||
if aclObj, err := p.srv.ResolveToken(args.SecretID); err != nil {
|
||||
return err
|
||||
} else if aclObj != nil && !aclObj.AllowNsOp(args.RequestNamespace(), acl.NamespaceCapabilitySubmitJob) {
|
||||
return structs.ErrPermissionDenied
|
||||
}
|
||||
|
||||
// Validate the arguments
|
||||
if args.JobID == "" {
|
||||
return fmt.Errorf("missing job ID for evaluation")
|
||||
|
|
|
@ -5,9 +5,11 @@ import (
|
|||
|
||||
memdb "github.com/hashicorp/go-memdb"
|
||||
"github.com/hashicorp/net-rpc-msgpackrpc"
|
||||
"github.com/hashicorp/nomad/acl"
|
||||
"github.com/hashicorp/nomad/nomad/mock"
|
||||
"github.com/hashicorp/nomad/nomad/structs"
|
||||
"github.com/hashicorp/nomad/testutil"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestPeriodicEndpoint_Force(t *testing.T) {
|
||||
|
@ -60,6 +62,84 @@ func TestPeriodicEndpoint_Force(t *testing.T) {
|
|||
}
|
||||
}
|
||||
|
||||
func TestPeriodicEndpoint_Force_ACL(t *testing.T) {
|
||||
t.Parallel()
|
||||
s1, root := testACLServer(t, func(c *Config) {
|
||||
c.NumSchedulers = 0 // Prevent automatic dequeue
|
||||
})
|
||||
defer s1.Shutdown()
|
||||
state := s1.fsm.State()
|
||||
assert := assert.New(t)
|
||||
codec := rpcClient(t, s1)
|
||||
testutil.WaitForLeader(t, s1.RPC)
|
||||
|
||||
// Create and insert a periodic job.
|
||||
job := mock.PeriodicJob()
|
||||
job.Periodic.ProhibitOverlap = true // Shouldn't affect anything.
|
||||
assert.Nil(state.UpsertJob(100, job))
|
||||
_, err := s1.periodicDispatcher.Add(job)
|
||||
assert.Nil(err)
|
||||
|
||||
// Force launch it.
|
||||
req := &structs.PeriodicForceRequest{
|
||||
JobID: job.ID,
|
||||
WriteRequest: structs.WriteRequest{
|
||||
Region: "global",
|
||||
Namespace: job.Namespace,
|
||||
},
|
||||
}
|
||||
|
||||
// Try with no token and expect permission denied
|
||||
{
|
||||
var resp structs.PeriodicForceResponse
|
||||
err := msgpackrpc.CallWithCodec(codec, "Periodic.Force", req, &resp)
|
||||
assert.NotNil(err)
|
||||
assert.Contains(err.Error(), structs.ErrPermissionDenied.Error())
|
||||
}
|
||||
|
||||
// Try with an invalid token and expect permission denied
|
||||
{
|
||||
invalidToken := mock.CreatePolicyAndToken(t, state, 1003, "invalid", mock.NodePolicy(acl.PolicyWrite))
|
||||
req.SecretID = invalidToken.SecretID
|
||||
var resp structs.PeriodicForceResponse
|
||||
err := msgpackrpc.CallWithCodec(codec, "Periodic.Force", req, &resp)
|
||||
assert.NotNil(err)
|
||||
assert.Contains(err.Error(), structs.ErrPermissionDenied.Error())
|
||||
}
|
||||
|
||||
// Fetch the response with a valid token
|
||||
{
|
||||
policy := mock.NamespacePolicy(structs.DefaultNamespace, "", []string{acl.NamespaceCapabilitySubmitJob})
|
||||
token := mock.CreatePolicyAndToken(t, state, 1005, "valid", policy)
|
||||
req.SecretID = token.SecretID
|
||||
var resp structs.PeriodicForceResponse
|
||||
assert.Nil(msgpackrpc.CallWithCodec(codec, "Periodic.Force", req, &resp))
|
||||
assert.NotEqual(0, resp.Index)
|
||||
|
||||
// Lookup the evaluation
|
||||
ws := memdb.NewWatchSet()
|
||||
eval, err := state.EvalByID(ws, resp.EvalID)
|
||||
assert.Nil(err)
|
||||
assert.NotNil(eval)
|
||||
assert.Equal(eval.CreateIndex, resp.EvalCreateIndex)
|
||||
}
|
||||
|
||||
// Fetch the response with management token
|
||||
{
|
||||
req.SecretID = root.SecretID
|
||||
var resp structs.PeriodicForceResponse
|
||||
assert.Nil(msgpackrpc.CallWithCodec(codec, "Periodic.Force", req, &resp))
|
||||
assert.NotEqual(0, resp.Index)
|
||||
|
||||
// Lookup the evaluation
|
||||
ws := memdb.NewWatchSet()
|
||||
eval, err := state.EvalByID(ws, resp.EvalID)
|
||||
assert.Nil(err)
|
||||
assert.NotNil(eval)
|
||||
assert.Equal(eval.CreateIndex, resp.EvalCreateIndex)
|
||||
}
|
||||
}
|
||||
|
||||
func TestPeriodicEndpoint_Force_NonPeriodic(t *testing.T) {
|
||||
t.Parallel()
|
||||
s1 := testServer(t, func(c *Config) {
|
||||
|
|
|
@ -1533,9 +1533,9 @@ The table below shows this endpoint's support for
|
|||
[blocking queries](/api/index.html#blocking-queries) and
|
||||
[required ACLs](/api/index.html#acls).
|
||||
|
||||
| Blocking Queries | ACL Required |
|
||||
| ---------------- | ------------ |
|
||||
| `NO` | `none` |
|
||||
| Blocking Queries | ACL Required |
|
||||
| ---------------- | ---------------------- |
|
||||
| `NO` | `namespace:submit-job` |
|
||||
|
||||
### Parameters
|
||||
|
||||
|
|
Loading…
Reference in New Issue