fix a bug in SetQuitMessage

Quit messages aren't supposed to clobber each other under any
circumstances, but actually only the default "Connection closed"
message was non-clobbering, all others would clobber.
This commit is contained in:
Shivaram Lingamneni 2019-05-22 20:29:19 -04:00
parent 174115deb6
commit 99de53264e
1 changed files with 5 additions and 7 deletions

View File

@ -121,16 +121,14 @@ type Session struct {
// sets the session quit message, if there isn't one already
func (sd *Session) SetQuitMessage(message string) (set bool) {
if message == "" {
message = "Connection closed"
}
if sd.quitMessage == "" {
sd.quitMessage = "Connection closed"
sd.quitMessage = message
return true
} else {
return false
}
} else {
sd.quitMessage = message
return true
}
}
// set the negotiated message length based on session capabilities