diff --git a/internal/shell/shell.go b/internal/shell/shell.go index 0d2df7b11abc6c8261a77f46b7eb4c93e9c63e70..58e63fc245bfda708c87f9b39895705d3ee5d344 100644 --- a/internal/shell/shell.go +++ b/internal/shell/shell.go @@ -157,16 +157,17 @@ func (s *Shell) SetBlockFuncs(blockFuncs []BlockFunc) { // CommandsBlocker creates a BlockFunc that blocks exact command matches func CommandsBlocker(bannedCommands []string) BlockFunc { - bannedSet := make(map[string]bool) + bannedSet := make(map[string]struct{}) for _, cmd := range bannedCommands { - bannedSet[cmd] = true + bannedSet[cmd] = struct{}{} } return func(args []string) bool { if len(args) == 0 { return false } - return bannedSet[args[0]] + _, ok := bannedSet[args[0]] + return ok } }