mirror of
https://github.com/gravitational/teleport.git
synced 2026-11-02 23:31:08 +08:00
a-palchikov
b6f4d6fc71
* Compute warnings when mapping traits to roles * Log warnings for case-insensitive traits to role matches. Updates https://github.com/gravitational/teleport/issues/6016. Co-authored-by: Andrew Lytvynov <andrew@goteleport.com>
93 lines
3.2 KiB
Go
93 lines
3.2 KiB
Go
package utils
|
|
|
|
import (
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/gravitational/trace"
|
|
)
|
|
|
|
// ContainsExpansion returns true if value contains
|
|
// expansion syntax, e.g. $1 or ${10}
|
|
func ContainsExpansion(val string) bool {
|
|
return reExpansion.FindAllStringIndex(val, -1) != nil
|
|
}
|
|
|
|
// GlobToRegexp replaces glob-style standalone wildcard values
|
|
// with real .* regexp-friendly values, does not modify regexp-compatible values,
|
|
// quotes non-wildcard values
|
|
func GlobToRegexp(in string) string {
|
|
return replaceWildcard.ReplaceAllString(regexp.QuoteMeta(in), "(.*)")
|
|
}
|
|
|
|
// ReplaceRegexp replaces value in string, accepts regular expression and simplified
|
|
// wildcard syntax, it has several important differeneces with standard lib
|
|
// regexp replacer:
|
|
// * Wildcard globs '*' are treated as regular expression .* expression
|
|
// * Expression is treated as regular expression if it starts with ^ and ends with $
|
|
// * Full match is expected, partial replacements ignored
|
|
// * If there is no match, returns a NotFound error
|
|
func ReplaceRegexp(expression string, replaceWith string, input string) (string, error) {
|
|
return ReplaceRegexpWithConfig(expression, replaceWith, input, RegexpConfig{})
|
|
}
|
|
|
|
// ReplaceRegexpWithConfig behaves exactly like ReplaceRegexp but its behavior
|
|
// can be customized
|
|
func ReplaceRegexpWithConfig(expression string, replaceWith string, input string, config RegexpConfig) (string, error) {
|
|
if !strings.HasPrefix(expression, "^") || !strings.HasSuffix(expression, "$") {
|
|
// replace glob-style wildcards with regexp wildcards
|
|
// for plain strings, and quote all characters that could
|
|
// be interpreted in regular expression
|
|
expression = "^" + GlobToRegexp(expression) + "$"
|
|
}
|
|
if config.IgnoreCase {
|
|
expression = "(?i)" + expression
|
|
}
|
|
expr, err := regexp.Compile(expression)
|
|
if err != nil {
|
|
return "", trace.BadParameter(err.Error())
|
|
}
|
|
// if there is no match, return NotFound error
|
|
index := expr.FindAllStringIndex(input, -1)
|
|
if len(index) == 0 {
|
|
return "", trace.NotFound("no match found")
|
|
}
|
|
return expr.ReplaceAllString(input, replaceWith), nil
|
|
}
|
|
|
|
// RegexpConfig defines the configuration of the regular expression matcher
|
|
type RegexpConfig struct {
|
|
// IgnoreCase specifies whether matching is case-insensitive
|
|
IgnoreCase bool
|
|
}
|
|
|
|
// SliceMatchesRegex checks if input matches any of the expressions. The
|
|
// match is always evaluated as a regex either an exact match or regexp.
|
|
func SliceMatchesRegex(input string, expressions []string) (bool, error) {
|
|
for _, expression := range expressions {
|
|
if !strings.HasPrefix(expression, "^") || !strings.HasSuffix(expression, "$") {
|
|
// replace glob-style wildcards with regexp wildcards
|
|
// for plain strings, and quote all characters that could
|
|
// be interpreted in regular expression
|
|
expression = "^" + GlobToRegexp(expression) + "$"
|
|
}
|
|
|
|
expr, err := regexp.Compile(expression)
|
|
if err != nil {
|
|
return false, trace.BadParameter(err.Error())
|
|
}
|
|
|
|
// Since the expression is always surrounded by ^ and $ this is an exact
|
|
// match for either a a plain string (for example ^hello$) or for a regexp
|
|
// (for example ^hel*o$).
|
|
if expr.MatchString(input) {
|
|
return true, nil
|
|
}
|
|
}
|
|
|
|
return false, nil
|
|
}
|
|
|
|
var replaceWildcard = regexp.MustCompile(`(\\\*)`)
|
|
var reExpansion = regexp.MustCompile(`\$[^\$]+`)
|