Introduce new xmpp/sasl.go

1 files changed, 50 insertions, 0 deletions

diff --git a/xmpp/sasl.go b/xmpp/sasl.go
new file mode 100644
index 0000000..cbc1f11
--- /dev/null
+++ b/xmpp/sasl.go
@@ -0,0 +1,50 @@
+package xmpp
+
+import (
+	"encoding/xml"
+	"encoding/base64"
+	"errors"
+)
+
+func sendSaslAuth(b []xml.Token, c *Conn) error {
+	mechanisms := make([]string, 0)
+	for i, v := range(b) {
+		switch token := v.(type) {
+		case xml.StartElement:
+			expected := xml.Name{"urn:ietf:params:xml:ns:xmpp-sasl", "mechanism"}
+			if token.Name == expected {
+				if i >= (len(b)-1) { continue }
+				switch payload := b[i+1].(type) {
+				case xml.CharData:
+					mechanisms = append(mechanisms, string(payload))
+				}
+			}
+		}
+	}
+
+	for _, v := range(mechanisms) {
+		if v == "PLAIN" {
+			start := xml.StartElement{
+				xml.Name{"urn:ietf:params:xml:ns:xmpp-sasl", "auth"},
+				[]xml.Attr{
+					xml.Attr{xml.Name{"", "mechanism"}, "PLAIN"},
+				},
+			}
+
+			data := []byte("\x00" + username(c.jid) + "\x00" + c.pwd)
+			dst := make([]byte, base64.StdEncoding.EncodedLen(len(data)))
+			base64.StdEncoding.Encode(dst, data)
+			payload := xml.CharData(dst)
+
+			end := start.End()
+
+			c.enc.encodeNow(start)
+			c.enc.encode(payload)
+			c.enc.encodeNow(end)
+
+			return nil
+		}
+	}
+
+	return errors.New("No compatible SASL mechanism given")
+}