fix #1260

ergochat · Sep 16, 2020 · 8c5cf1c · 8c5cf1c
1 parent c4725a2
commit 8c5cf1c
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 32 deletions.
diff --git a/irc/handlers.go b/irc/handlers.go
@@ -2194,15 +2194,7 @@ func npcHandler(server *Server, client *Client, msg ircmsg.IrcMessage, rb *Respo
 	fakeSource := msg.Params[1]
 	message := msg.Params[2:]
 
-	_, err := CasefoldName(fakeSource)
-	if err != nil {
-		client.Send(nil, client.server.name, ERR_CANNOTSENDRP, target, client.t("Fake source must be a valid nickname"))
-		return false
-	}
-
-	sourceString := fmt.Sprintf(npcNickMask, fakeSource, client.nick)
-
-	sendRoleplayMessage(server, client, sourceString, target, false, message, rb)
+	sendRoleplayMessage(server, client, fakeSource, target, false, false, message, rb)
 
 	return false
 }
@@ -2212,15 +2204,8 @@ func npcaHandler(server *Server, client *Client, msg ircmsg.IrcMessage, rb *Resp
 	target := msg.Params[0]
 	fakeSource := msg.Params[1]
 	message := msg.Params[2:]
-	sourceString := fmt.Sprintf(npcNickMask, fakeSource, client.nick)
-
-	_, err := CasefoldName(fakeSource)
-	if err != nil {
-		client.Send(nil, client.server.name, ERR_CANNOTSENDRP, target, client.t("Fake source must be a valid nickname"))
-		return false
-	}
 
-	sendRoleplayMessage(server, client, sourceString, target, true, message, rb)
+	sendRoleplayMessage(server, client, fakeSource, target, false, true, message, rb)
 
 	return false
 }
@@ -2623,9 +2608,8 @@ func sanickHandler(server *Server, client *Client, msg ircmsg.IrcMessage, rb *Re
 func sceneHandler(server *Server, client *Client, msg ircmsg.IrcMessage, rb *ResponseBuffer) bool {
 	target := msg.Params[0]
 	message := msg.Params[1:]
-	sourceString := fmt.Sprintf(sceneNickMask, client.nick)
 
-	sendRoleplayMessage(server, client, sourceString, target, false, message, rb)
+	sendRoleplayMessage(server, client, "", target, true, false, message, rb)
 
 	return false
 }

diff --git a/irc/roleplay.go b/irc/roleplay.go
@@ -4,7 +4,8 @@
 package irc
 
 import (
-	"bytes"
+	"fmt"
+	"strings"
 
 	"github.com/oragono/oragono/irc/history"
 	"github.com/oragono/oragono/irc/modes"
@@ -16,7 +17,7 @@ const (
 	sceneNickMask = "=Scene=!%s@npc.fakeuser.invalid"
 )
 
-func sendRoleplayMessage(server *Server, client *Client, source string, targetString string, isAction bool, messageParts []string, rb *ResponseBuffer) {
+func sendRoleplayMessage(server *Server, client *Client, source string, targetString string, isScene, isAction bool, messageParts []string, rb *ResponseBuffer) {
 	config := server.Config()
 	if !config.Roleplay.Enabled {
 		rb.Add(nil, client.server.name, ERR_CANNOTSENDRP, targetString, client.t("Roleplaying has been disabled by the server administrators"))
@@ -26,20 +27,27 @@ func sendRoleplayMessage(server *Server, client *Client, source string, targetSt
 		rb.Add(nil, client.server.name, ERR_CANNOTSENDRP, targetString, client.t("Insufficient privileges"))
 		return
 	}
-	cfSource, cfSourceErr := CasefoldName(source)
-	skelSource, skelErr := Skeleton(source)
-	if cfSourceErr != nil || skelErr != nil ||
-		restrictedCasefoldedNicks.Has(cfSource) || restrictedSkeletons.Has(skelSource) {
-		rb.Add(nil, client.server.name, ERR_CANNOTSENDRP, targetString, client.t("Invalid roleplay name"))
-		return
+
+	var sourceMask string
+	if isScene {
+		sourceMask = fmt.Sprintf(sceneNickMask, client.Nick())
+	} else {
+		cfSource, cfSourceErr := CasefoldName(source)
+		skelSource, skelErr := Skeleton(source)
+		if cfSourceErr != nil || skelErr != nil ||
+			restrictedCasefoldedNicks.Has(cfSource) || restrictedSkeletons.Has(skelSource) {
+			rb.Add(nil, client.server.name, ERR_CANNOTSENDRP, targetString, client.t("Invalid roleplay name"))
+			return
+		}
+		sourceMask = fmt.Sprintf(npcNickMask, source, client.Nick())
 	}
 
 	// block attempts to send CTCP messages to Tor clients
 	if len(messageParts) > 0 && len(messageParts[0]) > 0 && messageParts[0][0] == '\x01' {
 		return
 	}
 
-	var buf bytes.Buffer
+	var buf strings.Builder
 	if isAction {
 		buf.WriteString("\x01ACTION ")
 	}
@@ -87,17 +95,17 @@ func sendRoleplayMessage(server *Server, client *Client, source string, targetSt
 				// of roleplay commands, so send them a copy whether they have echo-message
 				// or not
 				if rb.session == session {
-					rb.AddSplitMessageFromClient(source, "", nil, "PRIVMSG", targetString, splitMessage)
+					rb.AddSplitMessageFromClient(sourceMask, "", nil, "PRIVMSG", targetString, splitMessage)
 				} else {
-					session.sendSplitMsgFromClientInternal(false, source, "*", nil, "PRIVMSG", targetString, splitMessage)
+					session.sendSplitMsgFromClientInternal(false, sourceMask, "*", nil, "PRIVMSG", targetString, splitMessage)
 				}
 			}
 		}
 
 		channel.AddHistoryItem(history.Item{
 			Type:    history.Privmsg,
 			Message: splitMessage,
-			Nick:    source,
+			Nick:    sourceMask,
 		}, client.Account())
 	} else {
 		target, err := CasefoldName(targetString)
@@ -115,7 +123,7 @@ func sendRoleplayMessage(server *Server, client *Client, source string, targetSt
 		cnick := client.Nick()
 		tnick := user.Nick()
 		for _, session := range user.Sessions() {
-			session.sendSplitMsgFromClientInternal(false, source, "*", nil, "PRIVMSG", tnick, splitMessage)
+			session.sendSplitMsgFromClientInternal(false, sourceMask, "*", nil, "PRIVMSG", tnick, splitMessage)
 		}
 		if away, awayMessage := user.Away(); away {
 			//TODO(dan): possibly implement cooldown of away notifications to users