Add the plugin code.

Author: hsluoyz

authz.go

@@ -0,0 +1,66 @@
+package authz
+
+import (
+	"github.com/mholt/caddy/caddyhttp/httpserver"
+	"github.com/mholt/caddy"
+	"net/http"
+	"github.com/casbin/casbin"
+)
+
+// Authorizer is a middleware for filtering clients based on their ip or country's ISO code.
+type Authorizer struct {
+	Next   httpserver.Handler
+	Enforcer *casbin.Enforcer
+}
+
+// Init initializes the plugin
+func init() {
+	caddy.RegisterPlugin("authz", caddy.Plugin{
+		ServerType: "http",
+		Action:     Setup,
+	})
+}
+
+// Setup parses the ipfilter configuration and returns the middleware handler.
+func Setup(c *caddy.Controller) error {
+	e := casbin.NewEnforcer("authz_model.conf", "authz_policy.csv")
+
+	// Create new middleware
+	newMiddleWare := func(next httpserver.Handler) httpserver.Handler {
+		return &Authorizer{
+			Next:   next,
+			Enforcer: e,
+		}
+	}
+	// Add middleware
+	cfg := httpserver.GetConfig(c)
+	cfg.AddMiddleware(newMiddleWare)
+
+	return nil
+}
+
+// ServeHTTP serves the request.
+func (a Authorizer) ServeHTTP(w http.ResponseWriter, r *http.Request) (int, error) {
+	if !a.CheckPermission(r) {
+		w.WriteHeader(403)
+		return http.StatusForbidden, nil
+	} else {
+		return a.Next.ServeHTTP(w, r)
+	}
+}
+
+// GetUserName gets the user name from the request.
+// Currently, only HTTP basic authentication is supported
+func (a *Authorizer) GetUserName(r *http.Request) string {
+	username, _, _ := r.BasicAuth()
+	return username
+}
+
+// CheckPermission checks the user/method/path combination from the request.
+// Returns true (permission granted) or false (permission forbidden)
+func (a *Authorizer) CheckPermission(r *http.Request) bool {
+	user := a.GetUserName(r)
+	method := r.Method
+	path := r.URL.Path
+	return a.Enforcer.Enforce(user, path, method)
+}

authz_model.conf

@@ -0,0 +1,14 @@
+[request_definition]
+r = sub, obj, act
+
+[policy_definition]
+p = sub, obj, act
+
+[role_definition]
+g = _, _
+
+[policy_effect]
+e = some(where (p.eft == allow))
+
+[matchers]
+m = g(r.sub, p.sub) && keyMatch(r.obj, p.obj) && (r.act == p.act || p.act == "*")

authz_policy.csv

@@ -0,0 +1,7 @@
+p, alice, /dataset1/*, GET
+p, alice, /dataset1/resource1, POST
+p, bob, /dataset2/resource1, *
+p, bob, /dataset2/resource2, GET
+p, bob, /dataset2/folder1/*, POST
+p, dataset1_admin, /dataset1/*, *
+g, cathy, dataset1_admin

authz_test.go

@@ -0,0 +1,90 @@
+package authz
+
+import (
+	"testing"
+	"github.com/mholt/caddy/caddyhttp/httpserver"
+	"net/http/httptest"
+	"net/http"
+	"github.com/casbin/casbin"
+)
+
+func testRequest(t *testing.T, handler Authorizer, user string, path string, method string, code int) {
+	r, _ := http.NewRequest(method, path, nil)
+	r.SetBasicAuth(user, "123")
+	w := httptest.NewRecorder()
+	handler.ServeHTTP(w, r)
+
+	if w.Code != code {
+		t.Errorf("%s, %s, %s: %d, supposed to be %d", user, path, method, w.Code, code)
+	}
+}
+
+func TestBasic(t *testing.T) {
+	e := casbin.NewEnforcer("authz_model.conf", "authz_policy.csv")
+
+	handler := Authorizer{
+		Next: httpserver.HandlerFunc(func(w http.ResponseWriter, r *http.Request) (int, error) {
+			return http.StatusOK, nil
+		}),
+		Enforcer: e,
+	}
+
+	testRequest(t, handler, "alice", "/dataset1/resource1", "GET", 200)
+	testRequest(t, handler, "alice", "/dataset1/resource1", "POST", 200)
+	testRequest(t, handler, "alice", "/dataset1/resource2", "GET", 200)
+	testRequest(t, handler, "alice", "/dataset1/resource2", "POST", 403)
+}
+
+func TestPathWildcard(t *testing.T) {
+	e := casbin.NewEnforcer("authz_model.conf", "authz_policy.csv")
+
+	handler := Authorizer{
+		Next: httpserver.HandlerFunc(func(w http.ResponseWriter, r *http.Request) (int, error) {
+			return http.StatusOK, nil
+		}),
+		Enforcer: e,
+	}
+
+	testRequest(t, handler, "bob", "/dataset2/resource1", "GET", 200)
+	testRequest(t, handler, "bob", "/dataset2/resource1", "POST", 200)
+	testRequest(t, handler, "bob", "/dataset2/resource1", "DELETE", 200)
+	testRequest(t, handler, "bob", "/dataset2/resource2", "GET", 200)
+	testRequest(t, handler, "bob", "/dataset2/resource2", "POST", 403)
+	testRequest(t, handler, "bob", "/dataset2/resource2", "DELETE", 403)
+
+	testRequest(t, handler, "bob", "/dataset2/folder1/item1", "GET", 403)
+	testRequest(t, handler, "bob", "/dataset2/folder1/item1", "POST", 200)
+	testRequest(t, handler, "bob", "/dataset2/folder1/item1", "DELETE", 403)
+	testRequest(t, handler, "bob", "/dataset2/folder1/item2", "GET", 403)
+	testRequest(t, handler, "bob", "/dataset2/folder1/item2", "POST", 200)
+	testRequest(t, handler, "bob", "/dataset2/folder1/item2", "DELETE", 403)
+}
+
+func TestRBAC(t *testing.T) {
+	e := casbin.NewEnforcer("authz_model.conf", "authz_policy.csv")
+
+	handler := Authorizer{
+		Next: httpserver.HandlerFunc(func(w http.ResponseWriter, r *http.Request) (int, error) {
+			return http.StatusOK, nil
+		}),
+		Enforcer: e,
+	}
+
+	// cathy can access all /dataset1/* resources via all methods because it has the dataset1_admin role.
+	testRequest(t, handler, "cathy", "/dataset1/item", "GET", 200)
+	testRequest(t, handler, "cathy", "/dataset1/item", "POST", 200)
+	testRequest(t, handler, "cathy", "/dataset1/item", "DELETE", 200)
+	testRequest(t, handler, "cathy", "/dataset2/item", "GET", 403)
+	testRequest(t, handler, "cathy", "/dataset2/item", "POST", 403)
+	testRequest(t, handler, "cathy", "/dataset2/item", "DELETE", 403)
+
+	// delete all roles on user cathy, so cathy cannot access any resources now.
+	e.DeleteRolesForUser("cathy")
+
+	testRequest(t, handler, "cathy", "/dataset1/item", "GET", 403)
+	testRequest(t, handler, "cathy", "/dataset1/item", "POST", 403)
+	testRequest(t, handler, "cathy", "/dataset1/item", "DELETE", 403)
+	testRequest(t, handler, "cathy", "/dataset2/item", "GET", 403)
+	testRequest(t, handler, "cathy", "/dataset2/item", "POST", 403)
+	testRequest(t, handler, "cathy", "/dataset2/item", "DELETE", 403)
+}