generated from TrueCloudLab/basic
2d4a9fc6dc
* Pass RequestTarget instead only namespace * Refactor unit-tests and dependencies Signed-off-by: Airat Arifullin <aarifullin@yadro.com>
108 lines
2.8 KiB
Go
108 lines
2.8 KiB
Go
package engine
|
|
|
|
import (
|
|
"git.frostfs.info/TrueCloudLab/policy-engine/pkg/chain"
|
|
"git.frostfs.info/TrueCloudLab/policy-engine/pkg/resource"
|
|
)
|
|
|
|
type defaultChainRouter struct {
|
|
morph MorphRuleChainStorage
|
|
|
|
local LocalOverrideStorage
|
|
}
|
|
|
|
func NewDefaultChainRouter(morph MorphRuleChainStorage) ChainRouter {
|
|
return &defaultChainRouter{
|
|
morph: morph,
|
|
}
|
|
}
|
|
|
|
func NewDefaultChainRouterWithLocalOverrides(morph MorphRuleChainStorage, local LocalOverrideStorage) ChainRouter {
|
|
return &defaultChainRouter{
|
|
morph: morph,
|
|
local: local,
|
|
}
|
|
}
|
|
|
|
func (dr *defaultChainRouter) IsAllowed(name chain.Name, rt RequestTarget, r resource.Request) (status chain.Status, ruleFound bool, err error) {
|
|
status, ruleFound, err = dr.checkLocal(name, rt, r)
|
|
if err != nil {
|
|
return chain.NoRuleFound, false, err
|
|
} else if ruleFound {
|
|
// The local overrides have the highest priority and thus
|
|
// morph rules are not considered if a local one is found.
|
|
return
|
|
}
|
|
|
|
status, ruleFound, err = dr.checkMorph(name, rt, r)
|
|
return
|
|
}
|
|
|
|
func (dr *defaultChainRouter) checkLocal(name chain.Name, rt RequestTarget, r resource.Request) (status chain.Status, ruleFound bool, err error) {
|
|
if dr.local == nil {
|
|
return
|
|
}
|
|
var ruleFounds []bool
|
|
for _, target := range rt.Targets() {
|
|
status, ruleFound, err = dr.matchLocalOverrides(name, target, r)
|
|
if err != nil || ruleFound && status != chain.Allow {
|
|
return
|
|
}
|
|
ruleFounds = append(ruleFounds, ruleFound)
|
|
}
|
|
|
|
status = chain.NoRuleFound
|
|
for _, ruleFound = range ruleFounds {
|
|
if ruleFound {
|
|
status = chain.Allow
|
|
break
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func (dr *defaultChainRouter) checkMorph(name chain.Name, rt RequestTarget, r resource.Request) (status chain.Status, ruleFound bool, err error) {
|
|
var ruleFounds []bool
|
|
for _, target := range rt.Targets() {
|
|
status, ruleFound, err = dr.matchMorphRuleChains(name, target, r)
|
|
if err != nil || ruleFound && status != chain.Allow {
|
|
return
|
|
}
|
|
ruleFounds = append(ruleFounds, ruleFound)
|
|
}
|
|
|
|
status = chain.NoRuleFound
|
|
for _, ruleFound = range ruleFounds {
|
|
if ruleFound {
|
|
status = chain.Allow
|
|
break
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func (dr *defaultChainRouter) matchLocalOverrides(name chain.Name, target Target, r resource.Request) (status chain.Status, ruleFound bool, err error) {
|
|
localOverrides, err := dr.local.ListOverrides(name, target)
|
|
if err != nil {
|
|
return
|
|
}
|
|
for _, c := range localOverrides {
|
|
if status, ruleFound = c.Match(r); ruleFound && status != chain.Allow {
|
|
return
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func (dr *defaultChainRouter) matchMorphRuleChains(name chain.Name, target Target, r resource.Request) (status chain.Status, ruleFound bool, err error) {
|
|
namespaceChains, err := dr.morph.ListMorphRuleChains(name, target)
|
|
if err != nil {
|
|
return
|
|
}
|
|
for _, c := range namespaceChains {
|
|
if status, ruleFound = c.Match(r); ruleFound {
|
|
return
|
|
}
|
|
}
|
|
return
|
|
}
|