Enforce group membership checks in the SAML server

saml/saml.go

@@ -124,6 +124,20 @@ func (c *Config) GetSSOGroups(serviceProviderID string) []string {
 	return sp.SSOGroups
 }
 
+func (c *Config) GetAllSSOGroups() []string {
+	tmp := make(map[string]struct{})
+	for _, sp := range c.serviceProviderMap {
+		for _, group := range sp.SSOGroups {
+			tmp[group] = struct{}{}
+		}
+	}
+	var out []string
+	for group := range tmp {
+		out = append(out, group)
+	}
+	return out
+}
+
 // Read users from a YAML-encoded file, in a format surprisingly
 // compatible with git.autistici.org/id/auth/server.
 //
@@ -311,7 +325,7 @@ func NewSAMLIDP(config *Config) (http.Handler, error) {
 	h := idp.Handler()
 
 	root := mux.NewRouter()
-	root.PathPrefix(ssoURL.Path).Handler(w.Wrap(h, svc, nil))
+	root.PathPrefix(ssoURL.Path).Handler(w.Wrap(h, svc, config.GetAllSSOGroups()))
 	root.Handle(metadataURL.Path, h)
 	return root, nil
 }