mirror of
https://github.com/ergochat/ergo.git
synced 2024-11-29 15:40:02 +01:00
fix #1005
This commit is contained in:
parent
a46c0eed9f
commit
2779fe7c10
@ -175,7 +175,7 @@ func (s *Session) EndMultilineBatch(label string) (batch MultilineBatch, err err
|
|||||||
}
|
}
|
||||||
s.deferredFakelagCount = fakelagBill
|
s.deferredFakelagCount = fakelagBill
|
||||||
|
|
||||||
if batch.label == "" || batch.label != label || batch.message.LenLines() == 0 {
|
if batch.label == "" || batch.label != label || !batch.message.ValidMultiline() {
|
||||||
err = errInvalidMultilineBatch
|
err = errInvalidMultilineBatch
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@ -1357,9 +1357,14 @@ func (session *Session) sendSplitMsgFromClientInternal(blocking bool, nickmask,
|
|||||||
session.SendRawMessage(msg, blocking)
|
session.SendRawMessage(msg, blocking)
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
for i, messagePair := range message.Split {
|
msgidSent := false // send msgid on the first nonblank line
|
||||||
|
for _, messagePair := range message.Split {
|
||||||
|
if len(messagePair.Message) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
var msgid string
|
var msgid string
|
||||||
if i == 0 {
|
if !msgidSent {
|
||||||
|
msgidSent = true
|
||||||
msgid = message.Msgid
|
msgid = message.Msgid
|
||||||
}
|
}
|
||||||
session.sendFromClientInternal(blocking, message.Time, msgid, nickmask, accountName, tags, command, target, messagePair.Message)
|
session.sendFromClientInternal(blocking, message.Time, msgid, nickmask, accountName, tags, command, target, messagePair.Message)
|
||||||
|
@ -1793,33 +1793,35 @@ func nickHandler(server *Server, client *Client, msg ircmsg.IrcMessage, rb *Resp
|
|||||||
|
|
||||||
// helper to store a batched PRIVMSG in the session object
|
// helper to store a batched PRIVMSG in the session object
|
||||||
func absorbBatchedMessage(server *Server, client *Client, msg ircmsg.IrcMessage, batchTag string, histType history.ItemType, rb *ResponseBuffer) {
|
func absorbBatchedMessage(server *Server, client *Client, msg ircmsg.IrcMessage, batchTag string, histType history.ItemType, rb *ResponseBuffer) {
|
||||||
|
var errorCode, errorMessage string
|
||||||
|
defer func() {
|
||||||
|
if errorCode != "" {
|
||||||
|
if histType != history.Notice {
|
||||||
|
rb.Add(nil, server.name, "FAIL", "BATCH", errorCode, errorMessage)
|
||||||
|
}
|
||||||
|
rb.session.EndMultilineBatch("")
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
if batchTag != rb.session.batch.label {
|
if batchTag != rb.session.batch.label {
|
||||||
if histType != history.Notice {
|
errorCode, errorMessage = "MULTILINE_INVALID", client.t("Incorrect batch tag sent")
|
||||||
rb.Add(nil, server.name, "FAIL", "BATCH", "MULTILINE_INVALID", client.t("Incorrect batch tag sent"))
|
|
||||||
}
|
|
||||||
rb.session.EndMultilineBatch("")
|
|
||||||
return
|
return
|
||||||
} else if len(msg.Params) < 2 || msg.Params[1] == "" {
|
} else if len(msg.Params) < 2 {
|
||||||
if histType != history.Notice {
|
errorCode, errorMessage = "MULTILINE_INVALID", client.t("Invalid multiline batch")
|
||||||
rb.Add(nil, server.name, "FAIL", "BATCH", "MULTILINE_INVALID", client.t("Invalid multiline batch"))
|
|
||||||
}
|
|
||||||
rb.session.EndMultilineBatch("")
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
rb.session.batch.command = msg.Command
|
rb.session.batch.command = msg.Command
|
||||||
isConcat, _ := msg.GetTag(caps.MultilineConcatTag)
|
isConcat, _ := msg.GetTag(caps.MultilineConcatTag)
|
||||||
|
if isConcat && len(msg.Params[1]) == 0 {
|
||||||
|
errorCode, errorMessage = "MULTILINE_INVALID", client.t("Cannot send a blank line with the multiline concat tag")
|
||||||
|
return
|
||||||
|
}
|
||||||
rb.session.batch.message.Append(msg.Params[1], isConcat)
|
rb.session.batch.message.Append(msg.Params[1], isConcat)
|
||||||
config := server.Config()
|
config := server.Config()
|
||||||
if config.Limits.Multiline.MaxBytes < rb.session.batch.message.LenBytes() {
|
if config.Limits.Multiline.MaxBytes < rb.session.batch.message.LenBytes() {
|
||||||
if histType != history.Notice {
|
errorCode, errorMessage = "MULTILINE_MAX_BYTES", strconv.Itoa(config.Limits.Multiline.MaxBytes)
|
||||||
rb.Add(nil, server.name, "FAIL", "BATCH", "MULTILINE_MAX_BYTES", strconv.Itoa(config.Limits.Multiline.MaxBytes))
|
|
||||||
}
|
|
||||||
rb.session.EndMultilineBatch("")
|
|
||||||
} else if config.Limits.Multiline.MaxLines != 0 && config.Limits.Multiline.MaxLines < rb.session.batch.message.LenLines() {
|
} else if config.Limits.Multiline.MaxLines != 0 && config.Limits.Multiline.MaxLines < rb.session.batch.message.LenLines() {
|
||||||
if histType != history.Notice {
|
errorCode, errorMessage = "MULTILINE_MAX_LINES", strconv.Itoa(config.Limits.Multiline.MaxLines)
|
||||||
rb.Add(nil, server.name, "FAIL", "BATCH", "MULTILINE_MAX_LINES", strconv.Itoa(config.Limits.Multiline.MaxLines))
|
|
||||||
}
|
|
||||||
rb.session.EndMultilineBatch("")
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -73,10 +73,24 @@ func (sm *SplitMessage) LenBytes() (result int) {
|
|||||||
}
|
}
|
||||||
for i := 0; i < len(sm.Split); i++ {
|
for i := 0; i < len(sm.Split); i++ {
|
||||||
result += len(sm.Split[i].Message)
|
result += len(sm.Split[i].Message)
|
||||||
|
// bill for the joining newline if necessary
|
||||||
|
if i != 0 && !sm.Split[i].Concat {
|
||||||
|
result += 1
|
||||||
|
}
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (sm *SplitMessage) ValidMultiline() bool {
|
||||||
|
// must contain at least one nonblank line
|
||||||
|
for i := 0; i < len(sm.Split); i++ {
|
||||||
|
if len(sm.Split[i].Message) != 0 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
func (sm *SplitMessage) IsRestrictedCTCPMessage() bool {
|
func (sm *SplitMessage) IsRestrictedCTCPMessage() bool {
|
||||||
if IsRestrictedCTCPMessage(sm.Message) {
|
if IsRestrictedCTCPMessage(sm.Message) {
|
||||||
return true
|
return true
|
||||||
|
Loading…
Reference in New Issue
Block a user