6e51f3a4d0
On some servers SASL auth would fail because we didn't fully comply with the spec. Multiple caps can be returned in a `CAP ACK`, and sometimes servers add a trailing space. Parse these responses more correctly.
64 lines
1.7 KiB
Go
64 lines
1.7 KiB
Go
package irc
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
type SASLResult struct {
|
|
Failed bool
|
|
Err error
|
|
}
|
|
|
|
// Check if a space-separated list of arguments contains a value.
|
|
func listContains(list string, value string) bool {
|
|
for _, arg_name := range strings.Split(strings.TrimSpace(list), " ") {
|
|
if arg_name == value {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (irc *Connection) setupSASLCallbacks(result chan<- *SASLResult) {
|
|
irc.AddCallback("CAP", func(e *Event) {
|
|
if len(e.Arguments) == 3 {
|
|
if e.Arguments[1] == "LS" {
|
|
if !listContains(e.Arguments[2], "sasl") {
|
|
result <- &SASLResult{true, errors.New("no SASL capability " + e.Arguments[2])}
|
|
}
|
|
}
|
|
if e.Arguments[1] == "ACK" && listContains(e.Arguments[2], "sasl") {
|
|
if irc.SASLMech != "PLAIN" {
|
|
result <- &SASLResult{true, errors.New("only PLAIN is supported")}
|
|
}
|
|
irc.SendRaw("AUTHENTICATE " + irc.SASLMech)
|
|
}
|
|
}
|
|
})
|
|
irc.AddCallback("AUTHENTICATE", func(e *Event) {
|
|
str := base64.StdEncoding.EncodeToString([]byte(fmt.Sprintf("%s\x00%s\x00%s", irc.SASLLogin, irc.SASLLogin, irc.SASLPassword)))
|
|
irc.SendRaw("AUTHENTICATE " + str)
|
|
})
|
|
irc.AddCallback("901", func(e *Event) {
|
|
irc.SendRaw("CAP END")
|
|
irc.SendRaw("QUIT")
|
|
result <- &SASLResult{true, errors.New(e.Arguments[1])}
|
|
})
|
|
irc.AddCallback("902", func(e *Event) {
|
|
irc.SendRaw("CAP END")
|
|
irc.SendRaw("QUIT")
|
|
result <- &SASLResult{true, errors.New(e.Arguments[1])}
|
|
})
|
|
irc.AddCallback("903", func(e *Event) {
|
|
result <- &SASLResult{false, nil}
|
|
})
|
|
irc.AddCallback("904", func(e *Event) {
|
|
irc.SendRaw("CAP END")
|
|
irc.SendRaw("QUIT")
|
|
result <- &SASLResult{true, errors.New(e.Arguments[1])}
|
|
})
|
|
}
|