Only allow valid snomasks to be set

irc/modes.go

@@ -54,10 +54,16 @@ func (client *Client) applyUserModeChanges(force bool, changes modes.ModeChanges
 			}
 			var masks []sno.Mask
 			if change.Op == modes.Add || change.Op == modes.Remove {
+				var newArg string
 				for _, char := range change.Arg {
-					masks = append(masks, sno.Mask(char))
+					mask := sno.Mask(char)
+					if sno.ValidMasks[mask] {
+						masks = append(masks, mask)
+						newArg += string(char)
 					}
 				}
+				change.Arg = newArg
+			}
 			if change.Op == modes.Add {
 				client.server.snomasks.AddMasks(client, masks...)
 				applied = append(applied, change)

irc/sno/constants.go

@@ -35,4 +35,18 @@ var (
 		LocalAccounts:      "ACCOUNT",
 		LocalXline:         "XLINE",
 	}
+
+	// ValidMasks contains the snomasks that we support.
+	ValidMasks = map[Mask]bool{
+		LocalAccouncements: true,
+		LocalConnects:      true,
+		LocalChannels:      true,
+		LocalKills:         true,
+		LocalNicks:         true,
+		LocalOpers:         true,
+		LocalQuits:         true,
+		Stats:              true,
+		LocalAccounts:      true,
+		LocalXline:         true,
+	}
 )

irc/snomanager.go

@@ -27,6 +27,11 @@ func (m *SnoManager) AddMasks(client *Client, masks ...sno.Mask) {
 	defer m.sendListMutex.Unlock()
 
 	for _, mask := range masks {
+		// confirm mask is valid
+		if !sno.ValidMasks[mask] {
+			continue
+		}
+
 		currentClientList := m.sendLists[mask]
 
 		if currentClientList == nil {