improve whitelisting

pull/13/head
Simon Let 7 years ago
parent 6b499ee11d
commit d5e2d2764b
  1. 8
      sanitize-history/resh-sanitize-history.go

@ -339,7 +339,7 @@ func (s *sanitizer) sanitizeCmdToken(token string) (string, error) {
// NOTE: do not sanitize single letter tokens // NOTE: do not sanitize single letter tokens
return token, nil return token, nil
} }
if s.whitelist[strings.ToLower(token)] == true { if s.isInWhitelist(token) == true {
return token, nil return token, nil
} }
@ -376,7 +376,7 @@ func (s *sanitizer) sanitizeToken(token string) string {
// NOTE: do not sanitize single letter tokens // NOTE: do not sanitize single letter tokens
return token return token
} }
if s.whitelist[token] { if s.isInWhitelist(token) {
return token return token
} }
return s.hashToken(token) return s.hashToken(token)
@ -414,3 +414,7 @@ func (s *sanitizer) trimHash(hash string) string {
} }
return hash[:length] return hash[:length]
} }
func (s *sanitizer) isInWhitelist(token string) bool {
return s.whitelist[strings.ToLower(token)] == true
}
Loading…
Cancel
Save