mirror of https://github.com/status-im/consul.git
Adds complete ACL support for listing events.
This commit is contained in:
parent
babb0a1235
commit
ededf330ba
|
@ -83,6 +83,14 @@ func (s *HTTPServer) EventList(resp http.ResponseWriter, req *http.Request) (int
|
||||||
return nil, nil
|
return nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Fetch the ACL token, if any.
|
||||||
|
var token string
|
||||||
|
s.parseToken(req, &token)
|
||||||
|
acl, err := s.agent.resolveToken(token)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
// Look for a name filter
|
// Look for a name filter
|
||||||
var nameFilter string
|
var nameFilter string
|
||||||
if filt := req.URL.Query().Get("name"); filt != "" {
|
if filt := req.URL.Query().Get("name"); filt != "" {
|
||||||
|
@ -126,7 +134,20 @@ RUN_QUERY:
|
||||||
// Get the recent events
|
// Get the recent events
|
||||||
events := s.agent.UserEvents()
|
events := s.agent.UserEvents()
|
||||||
|
|
||||||
// Filter the events if necessary
|
// Filter the events using the ACL, if present
|
||||||
|
if acl != nil {
|
||||||
|
for i := 0; i < len(events); i++ {
|
||||||
|
name := events[i].Name
|
||||||
|
if acl.EventRead(name) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
s.agent.logger.Printf("[DEBUG] agent: dropping event %q from result due to ACLs", name)
|
||||||
|
events = append(events[:i], events[i+1:]...)
|
||||||
|
i--
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Filter the events if requested
|
||||||
if nameFilter != "" {
|
if nameFilter != "" {
|
||||||
for i := 0; i < len(events); i++ {
|
for i := 0; i < len(events); i++ {
|
||||||
if events[i].Name != nameFilter {
|
if events[i].Name != nameFilter {
|
||||||
|
|
|
@ -5,6 +5,7 @@ import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httptest"
|
"net/http/httptest"
|
||||||
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
@ -192,6 +193,71 @@ func TestEventList_Filter(t *testing.T) {
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestEventList_ACLFilter(t *testing.T) {
|
||||||
|
dir, srv := makeHTTPServerWithACLs(t)
|
||||||
|
defer os.RemoveAll(dir)
|
||||||
|
defer srv.Shutdown()
|
||||||
|
defer srv.agent.Shutdown()
|
||||||
|
|
||||||
|
// Fire an event.
|
||||||
|
p := &UserEvent{Name: "foo"}
|
||||||
|
if err := srv.agent.UserEvent("dc1", "root", p); err != nil {
|
||||||
|
t.Fatalf("err: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Try no token.
|
||||||
|
{
|
||||||
|
testutil.WaitForResult(func() (bool, error) {
|
||||||
|
req, err := http.NewRequest("GET", "/v1/event/list", nil)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
resp := httptest.NewRecorder()
|
||||||
|
obj, err := srv.EventList(resp, req)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
list, ok := obj.([]*UserEvent)
|
||||||
|
if !ok {
|
||||||
|
return false, fmt.Errorf("bad: %#v", obj)
|
||||||
|
}
|
||||||
|
if len(list) != 0 {
|
||||||
|
return false, fmt.Errorf("bad: %#v", list)
|
||||||
|
}
|
||||||
|
return true, nil
|
||||||
|
}, func(err error) {
|
||||||
|
t.Fatalf("err: %v", err)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// Try the root token.
|
||||||
|
{
|
||||||
|
testutil.WaitForResult(func() (bool, error) {
|
||||||
|
req, err := http.NewRequest("GET", "/v1/event/list?token=root", nil)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
resp := httptest.NewRecorder()
|
||||||
|
obj, err := srv.EventList(resp, req)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
list, ok := obj.([]*UserEvent)
|
||||||
|
if !ok {
|
||||||
|
return false, fmt.Errorf("bad: %#v", obj)
|
||||||
|
}
|
||||||
|
if len(list) != 1 || list[0].Name != "foo" {
|
||||||
|
return false, fmt.Errorf("bad: %#v", list)
|
||||||
|
}
|
||||||
|
return true, nil
|
||||||
|
}, func(err error) {
|
||||||
|
t.Fatalf("err: %v", err)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestEventList_Blocking(t *testing.T) {
|
func TestEventList_Blocking(t *testing.T) {
|
||||||
httpTest(t, func(srv *HTTPServer) {
|
httpTest(t, func(srv *HTTPServer) {
|
||||||
p := &UserEvent{Name: "test"}
|
p := &UserEvent{Name: "test"}
|
||||||
|
|
Loading…
Reference in New Issue