pax_global_header00006660000000000000000000000064145710552650014524gustar00rootroot0000000000000052 comment=c051d69509c8593fd877754bda99997d7865daef go-xmpp-0.1.1/000077500000000000000000000000001457105526500131125ustar00rootroot00000000000000go-xmpp-0.1.1/.travis.yml000066400000000000000000000000571457105526500152250ustar00rootroot00000000000000language: go go: - tip script: - go test go-xmpp-0.1.1/LICENSE000066400000000000000000000027071457105526500141250ustar00rootroot00000000000000Copyright (c) 2009 The Go Authors. All rights reserved. Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: * Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. * Neither the name of Google Inc. nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. go-xmpp-0.1.1/README.md000066400000000000000000000002031457105526500143640ustar00rootroot00000000000000go-xmpp ======= go xmpp library (original was written by russ cox ) [Documentation](https://godoc.org/github.com/xmppo/go-xmpp) go-xmpp-0.1.1/_example/000077500000000000000000000000001457105526500147045ustar00rootroot00000000000000go-xmpp-0.1.1/_example/example-gui.go000066400000000000000000000045031457105526500174520ustar00rootroot00000000000000package main import ( "crypto/tls" "log" "os" "strings" "github.com/mattn/go-gtk/gtk" "github.com/xmppo/go-xmpp" ) func main() { gtk.Init(&os.Args) window := gtk.NewWindow(gtk.WINDOW_TOPLEVEL) window.SetTitle("GoTalk") window.Connect("destroy", func() { gtk.MainQuit() }) vbox := gtk.NewVBox(false, 1) scrolledwin := gtk.NewScrolledWindow(nil, nil) textview := gtk.NewTextView() textview.SetEditable(false) textview.SetCursorVisible(false) scrolledwin.Add(textview) vbox.Add(scrolledwin) buffer := textview.GetBuffer() entry := gtk.NewEntry() vbox.PackEnd(entry, false, false, 0) window.Add(vbox) window.SetSizeRequest(300, 400) window.ShowAll() dialog := gtk.NewDialog() dialog.SetTitle(window.GetTitle()) sgroup := gtk.NewSizeGroup(gtk.SIZE_GROUP_HORIZONTAL) hbox := gtk.NewHBox(false, 1) dialog.GetVBox().Add(hbox) label := gtk.NewLabel("username:") sgroup.AddWidget(label) hbox.Add(label) username := gtk.NewEntry() hbox.Add(username) hbox = gtk.NewHBox(false, 1) dialog.GetVBox().Add(hbox) label = gtk.NewLabel("password:") sgroup.AddWidget(label) hbox.Add(label) password := gtk.NewEntry() password.SetVisibility(false) hbox.Add(password) dialog.AddButton(gtk.STOCK_OK, gtk.RESPONSE_OK) dialog.AddButton(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL) dialog.SetDefaultResponse(gtk.RESPONSE_OK) dialog.SetTransientFor(window) dialog.ShowAll() res := dialog.Run() username_ := username.GetText() password_ := password.GetText() dialog.Destroy() if res != gtk.RESPONSE_OK { os.Exit(0) } xmpp.DefaultConfig = tls.Config{ ServerName: "talk.google.com", InsecureSkipVerify: false, } talk, err := xmpp.NewClient("talk.google.com:443", username_, password_, false) if err != nil { log.Fatal(err) } entry.Connect("activate", func() { text := entry.GetText() tokens := strings.SplitN(text, " ", 2) if len(tokens) == 2 { func() { defer recover() talk.Send(xmpp.Chat{Remote: tokens[0], Type: "chat", Text: tokens[1]}) entry.SetText("") }() } }) go func() { for { func() { defer recover() chat, err := talk.Recv() if err != nil { log.Fatal(err) } var iter gtk.TextIter buffer.GetStartIter(&iter) if msg, ok := chat.(xmpp.Chat); ok { buffer.Insert(&iter, msg.Remote+": "+msg.Text+"\n") } }() } }() gtk.Main() } go-xmpp-0.1.1/_example/example.go000066400000000000000000000041651457105526500166740ustar00rootroot00000000000000package main import ( "bufio" "crypto/tls" "flag" "fmt" "log" "os" "strings" "github.com/xmppo/go-xmpp" ) var ( server = flag.String("server", "talk.google.com:443", "server") username = flag.String("username", "", "username") password = flag.String("password", "", "password") status = flag.String("status", "xa", "status") statusMessage = flag.String("status-msg", "I for one welcome our new codebot overlords.", "status message") notls = flag.Bool("notls", false, "No TLS") debug = flag.Bool("debug", false, "debug output") session = flag.Bool("session", false, "use server session") ) func serverName(host string) string { return strings.Split(host, ":")[0] } func main() { flag.Usage = func() { fmt.Fprintf(os.Stderr, "usage: example [options]\n") flag.PrintDefaults() os.Exit(2) } flag.Parse() if *username == "" || *password == "" { if *debug && *username == "" && *password == "" { fmt.Fprintf(os.Stderr, "no username or password were given; attempting ANONYMOUS auth\n") } else if *username != "" || *password != "" { flag.Usage() } } if !*notls { xmpp.DefaultConfig = tls.Config{ ServerName: serverName(*server), InsecureSkipVerify: false, } } var talk *xmpp.Client var err error options := xmpp.Options{ Host: *server, User: *username, Password: *password, NoTLS: *notls, Debug: *debug, Session: *session, Status: *status, StatusMessage: *statusMessage, } talk, err = options.NewClient() if err != nil { log.Fatal(err) } go func() { for { chat, err := talk.Recv() if err != nil { log.Fatal(err) } switch v := chat.(type) { case xmpp.Chat: fmt.Println(v.Remote, v.Text) case xmpp.Presence: fmt.Println(v.From, v.Show) } } }() for { in := bufio.NewReader(os.Stdin) line, err := in.ReadString('\n') if err != nil { continue } line = strings.TrimRight(line, "\n") tokens := strings.SplitN(line, " ", 2) if len(tokens) == 2 { talk.Send(xmpp.Chat{Remote: tokens[0], Type: "chat", Text: tokens[1]}) } } } go-xmpp-0.1.1/go.mod000066400000000000000000000001571457105526500142230ustar00rootroot00000000000000module github.com/xmppo/go-xmpp go 1.21.5 require ( golang.org/x/crypto v0.20.0 golang.org/x/net v0.21.0 ) go-xmpp-0.1.1/go.sum000066400000000000000000000004701457105526500142460ustar00rootroot00000000000000golang.org/x/crypto v0.20.0 h1:jmAMJJZXr5KiCw05dfYK9QnqaqKLYXijU23lsEdcQqg= golang.org/x/crypto v0.20.0/go.mod h1:Xwo95rrVNIoSMx9wa1JroENMToLWn3RNVrTBpLHgZPQ= golang.org/x/net v0.21.0 h1:AQyQV4dYCvJ7vGmJyKki9+PBdyvhkSd8EIx/qb0AYv4= golang.org/x/net v0.21.0/go.mod h1:bIjVDfnllIU7BJ2DNgfnXvpSvtn8VRwhlsaeUTyUS44= go-xmpp-0.1.1/xmpp.go000066400000000000000000001265701457105526500144400ustar00rootroot00000000000000// Copyright 2011 The Go Authors. All rights reserved. // Use of this source code is governed by a BSD-style // license that can be found in the LICENSE file. // TODO(rsc): // More precise error handling. // Presence functionality. // TODO(mattn): // Add proxy authentication. // Package xmpp implements a simple Google Talk client // using the XMPP protocol described in RFC 3920 and RFC 3921. package xmpp import ( "bufio" "bytes" "crypto/hmac" "crypto/rand" "crypto/sha1" "crypto/sha256" "crypto/sha512" "crypto/tls" "crypto/x509" "encoding/base64" "encoding/binary" "encoding/xml" "errors" "fmt" "hash" "io" "math/big" "net" "net/http" "net/url" "os" "regexp" "slices" "strconv" "strings" "time" "golang.org/x/crypto/pbkdf2" "golang.org/x/net/proxy" ) const ( nsStream = "http://etherx.jabber.org/streams" nsTLS = "urn:ietf:params:xml:ns:xmpp-tls" nsSASL = "urn:ietf:params:xml:ns:xmpp-sasl" nsBind = "urn:ietf:params:xml:ns:xmpp-bind" nsSASLCB = "urn:xmpp:sasl-cb:0" nsClient = "jabber:client" nsSession = "urn:ietf:params:xml:ns:xmpp-session" ) // Default TLS configuration options var DefaultConfig = &tls.Config{} // DebugWriter is the writer used to write debugging output to. var DebugWriter io.Writer = os.Stderr // Cookie is a unique XMPP session identifier type Cookie uint64 func getCookie() Cookie { var buf [8]byte if _, err := rand.Reader.Read(buf[:]); err != nil { panic("Failed to read random bytes: " + err.Error()) } return Cookie(binary.LittleEndian.Uint64(buf[:])) } // Client holds XMPP connection options type Client struct { conn net.Conn // connection to server jid string // Jabber ID for our connection domain string p *xml.Decoder stanzaWriter io.Writer Mechanism string } func (c *Client) JID() string { return c.jid } func containsIgnoreCase(s, substr string) bool { s, substr = strings.ToUpper(s), strings.ToUpper(substr) return strings.Contains(s, substr) } func connect(host, user, passwd string, timeout time.Duration) (net.Conn, error) { addr := host if strings.TrimSpace(host) == "" { a := strings.SplitN(user, "@", 2) if len(a) == 2 { addr = a[1] } } a := strings.SplitN(host, ":", 2) if len(a) == 1 { addr += ":5222" } http_proxy := os.Getenv("HTTP_PROXY") if http_proxy == "" { http_proxy = os.Getenv("http_proxy") } // test for no proxy, takes a comma separated list with substrings to match if http_proxy != "" { noproxy := os.Getenv("NO_PROXY") if noproxy == "" { noproxy = os.Getenv("no_proxy") } if noproxy != "" { nplist := strings.Split(noproxy, ",") for _, s := range nplist { if containsIgnoreCase(addr, s) { http_proxy = "" break } } } } socks5Target, socks5 := strings.CutPrefix(http_proxy, "socks5://") if http_proxy != "" && !socks5 { url, err := url.Parse(http_proxy) if err == nil { addr = url.Host } } var c net.Conn var err error if socks5 { dialer, err := proxy.SOCKS5("tcp", socks5Target, nil, nil) if err != nil { return nil, err } c, err = dialer.Dial("tcp", addr) if err != nil { return nil, err } } else { c, err = net.DialTimeout("tcp", addr, timeout) if err != nil { return nil, err } } if http_proxy != "" && !socks5 { fmt.Fprintf(c, "CONNECT %s HTTP/1.1\r\n", host) fmt.Fprintf(c, "Host: %s\r\n", host) fmt.Fprintf(c, "\r\n") br := bufio.NewReader(c) req, _ := http.NewRequest("CONNECT", host, nil) resp, err := http.ReadResponse(br, req) if err != nil { return nil, err } if resp.StatusCode != 200 { f := strings.SplitN(resp.Status, " ", 2) return nil, errors.New(f[1]) } } return c, nil } // Options are used to specify additional options for new clients, such as a Resource. type Options struct { // Host specifies what host to connect to, as either "hostname" or "hostname:port" // If host is not specified, the DNS SRV should be used to find the host from the domainpart of the JID. // Default the port to 5222. Host string // User specifies what user to authenticate to the remote server. User string // Password supplies the password to use for authentication with the remote server. Password string // DialTimeout is the time limit for establishing a connection. A // DialTimeout of zero means no timeout. DialTimeout time.Duration // Resource specifies an XMPP client resource, like "bot", instead of accepting one // from the server. Use "" to let the server generate one for your client. Resource string // OAuthScope provides go-xmpp the required scope for OAuth2 authentication. OAuthScope string // OAuthToken provides go-xmpp with the required OAuth2 token used to authenticate OAuthToken string // OAuthXmlNs provides go-xmpp with the required namespaced used for OAuth2 authentication. This is // provided to the server as the xmlns:auth attribute of the OAuth2 authentication request. OAuthXmlNs string // TLS Config TLSConfig *tls.Config // InsecureAllowUnencryptedAuth permits authentication over a TCP connection that has not been promoted to // TLS by STARTTLS; this could leak authentication information over the network, or permit man in the middle // attacks. InsecureAllowUnencryptedAuth bool // NoTLS directs go-xmpp to not use TLS initially to contact the server; instead, a plain old unencrypted // TCP connection should be used. (Can be combined with StartTLS to support STARTTLS-based servers.) NoTLS bool // StartTLS directs go-xmpp to STARTTLS if the server supports it; go-xmpp will automatically STARTTLS // if the server requires it regardless of this option. StartTLS bool // Debug output Debug bool // Use server sessions Session bool // Presence Status Status string // Status message StatusMessage string // Auth mechanism to use Mechanism string } // NewClient establishes a new Client connection based on a set of Options. func (o Options) NewClient() (*Client, error) { host := o.Host if strings.TrimSpace(host) == "" { a := strings.SplitN(o.User, "@", 2) if len(a) == 2 { if _, addrs, err := net.LookupSRV("xmpp-client", "tcp", a[1]); err == nil { if len(addrs) > 0 { // default to first record host = fmt.Sprintf("%s:%d", addrs[0].Target, addrs[0].Port) defP := addrs[0].Priority for _, adr := range addrs { if adr.Priority < defP { host = fmt.Sprintf("%s:%d", adr.Target, adr.Port) defP = adr.Priority } } } else { host = a[1] } } else { host = a[1] } } } c, err := connect(host, o.User, o.Password, o.DialTimeout) if err != nil { return nil, err } if strings.LastIndex(host, ":") > 0 { host = host[:strings.LastIndex(host, ":")] } client := new(Client) if o.NoTLS { client.conn = c } else { var tlsconn *tls.Conn if o.TLSConfig != nil { tlsconn = tls.Client(c, o.TLSConfig) host = o.TLSConfig.ServerName } else { newconfig := DefaultConfig.Clone() newconfig.ServerName = host tlsconn = tls.Client(c, newconfig) } if err = tlsconn.Handshake(); err != nil { return nil, err } insecureSkipVerify := DefaultConfig.InsecureSkipVerify if o.TLSConfig != nil { insecureSkipVerify = o.TLSConfig.InsecureSkipVerify } if !insecureSkipVerify { if err = tlsconn.VerifyHostname(host); err != nil { return nil, err } } client.conn = tlsconn } if err := client.init(&o); err != nil { client.Close() return nil, err } return client, nil } // NewClient creates a new connection to a host given as "hostname" or "hostname:port". // If host is not specified, the DNS SRV should be used to find the host from the domainpart of the JID. // Default the port to 5222. func NewClient(host, user, passwd string, debug bool) (*Client, error) { opts := Options{ Host: host, User: user, Password: passwd, Debug: debug, Session: false, } return opts.NewClient() } // NewClientNoTLS creates a new client without TLS func NewClientNoTLS(host, user, passwd string, debug bool) (*Client, error) { opts := Options{ Host: host, User: user, Password: passwd, NoTLS: true, Debug: debug, Session: false, } return opts.NewClient() } // Close closes the XMPP connection func (c *Client) Close() error { if c.conn != (*tls.Conn)(nil) { fmt.Fprintf(c.stanzaWriter, "\n") // Wait for the server also closing the stream. for { select { case <-time.After(10 * time.Second): break default: ee, err := nextEnd(c.p) // If the server already closed the stream it is // likely to receive an error when trying to parse // the stream. Therefore the connection is also closed // if an error is received. if err != nil { return c.conn.Close() } if ee.Name.Local == "stream" { return c.conn.Close() } } } } return nil } func cnonce() string { randSize := big.NewInt(0) randSize.Lsh(big.NewInt(1), 64) cn, err := rand.Int(rand.Reader, randSize) if err != nil { return "" } return fmt.Sprintf("%016x", cn) } func (c *Client) init(o *Options) error { var domain string var user string a := strings.SplitN(o.User, "@", 2) // Check if User is not empty. Otherwise, we'll be attempting ANONYMOUS with Host domain. switch { case len(o.User) > 0: if len(a) != 2 { return errors.New("xmpp: invalid username (want user@domain): " + o.User) } user = a[0] domain = a[1] case strings.Contains(o.Host, ":"): domain = strings.SplitN(o.Host, ":", 2)[0] default: domain = o.Host } // Declare intent to be a jabber client and gather stream features. f, err := c.startStream(o, domain) if err != nil { return err } // If the server requires we STARTTLS, attempt to do so. if f, err = c.startTLSIfRequired(f, o, domain); err != nil { return err } var mechanism, channelBinding, clientFirstMessage, clientFinalMessageBare, authMessage string var serverSignature, keyingMaterial []byte var scramPlus, ok, tlsConnOK, tls13, serverEndPoint bool var cbsSlice []string var tlsConn *tls.Conn if o.User == "" && o.Password == "" { foundAnonymous := false for _, m := range f.Mechanisms.Mechanism { if m == "ANONYMOUS" { fmt.Fprintf(c.stanzaWriter, "\n", nsSASL) foundAnonymous = true break } } if !foundAnonymous { return fmt.Errorf("ANONYMOUS authentication is not an option and username and password were not specified") } } else { // Even digest forms of authentication are unsafe if we do not know that the host // we are talking to is the actual server, and not a man in the middle playing // proxy. if !c.IsEncrypted() && !o.InsecureAllowUnencryptedAuth { return errors.New("refusing to authenticate over unencrypted TCP connection") } tlsConn, ok = c.conn.(*tls.Conn) if ok { tlsConnOK = true } mechanism = "" if o.Mechanism != "" { if slices.Contains(f.Mechanisms.Mechanism, o.Mechanism) { mechanism = o.Mechanism } } else { switch { case slices.Contains(f.Mechanisms.Mechanism, "SCRAM-SHA-512-PLUS") && tlsConnOK: mechanism = "SCRAM-SHA-512-PLUS" case slices.Contains(f.Mechanisms.Mechanism, "SCRAM-SHA-256-PLUS") && tlsConnOK: mechanism = "SCRAM-SHA-256-PLUS" case slices.Contains(f.Mechanisms.Mechanism, "SCRAM-SHA-1-PLUS") && tlsConnOK: mechanism = "SCRAM-SHA-1-PLUS" case slices.Contains(f.Mechanisms.Mechanism, "SCRAM-SHA-512"): mechanism = "SCRAM-SHA-512" case slices.Contains(f.Mechanisms.Mechanism, "SCRAM-SHA-256"): mechanism = "SCRAM-SHA-256" case slices.Contains(f.Mechanisms.Mechanism, "SCRAM-SHA-1"): mechanism = "SCRAM-SHA-1" case slices.Contains(f.Mechanisms.Mechanism, "X-OAUTH2"): mechanism = "X-OAUTH2" case slices.Contains(f.Mechanisms.Mechanism, "PLAIN") && tlsConnOK: mechanism = "PLAIN" } } if strings.HasPrefix(mechanism, "SCRAM-SHA") { if strings.HasSuffix(mechanism, "PLUS") { scramPlus = true } if scramPlus { for _, cbs := range f.ChannelBindings.ChannelBinding { cbsSlice = append(cbsSlice, cbs.Type) } tlsState := tlsConn.ConnectionState() switch tlsState.Version { case tls.VersionTLS13: tls13 = true if slices.Contains(cbsSlice, "tls-server-end-point") && !slices.Contains(cbsSlice, "tls-exporter") { serverEndPoint = true } else { keyingMaterial, err = tlsState.ExportKeyingMaterial("EXPORTER-Channel-Binding", nil, 32) if err != nil { return err } } case tls.VersionTLS10, tls.VersionTLS11, tls.VersionTLS12: if slices.Contains(cbsSlice, "tls-server-end-point") && !slices.Contains(cbsSlice, "tls-unique") { serverEndPoint = true } else { keyingMaterial = tlsState.TLSUnique } default: return errors.New(mechanism + ": unknown TLS version") } if serverEndPoint { var h hash.Hash // This material is not necessary for `tls-server-end-point` binding, but it is required to check that // the TLS connection was not renegotiated. This function will fail if that's the case (see // https://pkg.go.dev/crypto/tls#ConnectionState.ExportKeyingMaterial _, err = tlsState.ExportKeyingMaterial("EXPORTER-Channel-Binding", nil, 32) if err != nil { return err } switch tlsState.PeerCertificates[0].SignatureAlgorithm { case x509.SHA1WithRSA, x509.SHA256WithRSA, x509.ECDSAWithSHA1, x509.ECDSAWithSHA256, x509.SHA256WithRSAPSS: h = sha256.New() case x509.SHA384WithRSA, x509.ECDSAWithSHA384, x509.SHA384WithRSAPSS: h = sha512.New384() case x509.SHA512WithRSA, x509.ECDSAWithSHA512, x509.SHA512WithRSAPSS: h = sha512.New() } h.Write(tlsState.PeerCertificates[0].Raw) keyingMaterial = h.Sum(nil) h.Reset() } if len(keyingMaterial) == 0 { return errors.New(mechanism + ": no keying material") } switch { case tls13 && !serverEndPoint: channelBinding = base64.StdEncoding.EncodeToString(append([]byte("p=tls-exporter,,"), keyingMaterial[:]...)) case serverEndPoint: channelBinding = base64.StdEncoding.EncodeToString(append([]byte("p=tls-server-end-point,,"), keyingMaterial[:]...)) default: channelBinding = base64.StdEncoding.EncodeToString(append([]byte("p=tls-unique,,"), keyingMaterial[:]...)) } } var shaNewFn func() hash.Hash switch mechanism { case "SCRAM-SHA-512", "SCRAM-SHA-512-PLUS": shaNewFn = sha512.New case "SCRAM-SHA-256", "SCRAM-SHA-256-PLUS": shaNewFn = sha256.New case "SCRAM-SHA-1", "SCRAM-SHA-1-PLUS": shaNewFn = sha1.New default: return errors.New("unsupported auth mechanism") } clientNonce := cnonce() if scramPlus { switch { case tls13 && !serverEndPoint: clientFirstMessage = "p=tls-exporter,,n=" + user + ",r=" + clientNonce case serverEndPoint: clientFirstMessage = "p=tls-server-end-point,,n=" + user + ",r=" + clientNonce default: clientFirstMessage = "p=tls-unique,,n=" + user + ",r=" + clientNonce } } else { clientFirstMessage = "n,,n=" + user + ",r=" + clientNonce } fmt.Fprintf(c.stanzaWriter, "%s\n", nsSASL, mechanism, base64.StdEncoding.EncodeToString([]byte(clientFirstMessage))) var sfm string if err = c.p.DecodeElement(&sfm, nil); err != nil { return errors.New("unmarshal : " + err.Error()) } b, err := base64.StdEncoding.DecodeString(string(sfm)) if err != nil { return err } var serverNonce, dgProtect string var salt []byte var iterations int for _, serverReply := range strings.Split(string(b), ",") { switch { case strings.HasPrefix(serverReply, "r="): serverNonce = strings.SplitN(serverReply, "=", 2)[1] if !strings.HasPrefix(serverNonce, clientNonce) { return errors.New("SCRAM: server nonce didn't start with client nonce") } case strings.HasPrefix(serverReply, "s="): salt, err = base64.StdEncoding.DecodeString(strings.SplitN(serverReply, "=", 2)[1]) if err != nil { return err } if string(salt) == "" { return errors.New("SCRAM: server sent empty salt") } case strings.HasPrefix(serverReply, "i="): iterations, err = strconv.Atoi(strings.SplitN(serverReply, "=", 2)[1]) if err != nil { return err } case strings.HasPrefix(serverReply, "d="): serverDgProtectHash := strings.SplitN(serverReply, "=", 2)[1] slices.Sort(f.Mechanisms.Mechanism) for _, mech := range f.Mechanisms.Mechanism { if dgProtect == "" { dgProtect = mech } else { dgProtect = dgProtect + "," + mech } } dgProtect = dgProtect + "|" slices.Sort(cbsSlice) for i, cb := range cbsSlice { if i == 0 { dgProtect = dgProtect + cb } else { dgProtect = dgProtect + "," + cb } } dgh := shaNewFn() dgh.Write([]byte(dgProtect)) dHash := dgh.Sum(nil) dHashb64 := base64.StdEncoding.EncodeToString(dHash) if dHashb64 != serverDgProtectHash { return errors.New("SCRAM: downgrade protection hash mismatch") } dgh.Reset() case strings.HasPrefix(serverReply, "m="): return errors.New("SCRAM: server sent reserved 'm' attribute.") } } if scramPlus { clientFinalMessageBare = "c=" + channelBinding + ",r=" + serverNonce } else { clientFinalMessageBare = "c=biws,r=" + serverNonce } saltedPassword := pbkdf2.Key([]byte(o.Password), salt, iterations, shaNewFn().Size(), shaNewFn) h := hmac.New(shaNewFn, saltedPassword) _, err = h.Write([]byte("Client Key")) if err != nil { return err } clientKey := h.Sum(nil) h.Reset() var storedKey []byte switch mechanism { case "SCRAM-SHA-512", "SCRAM-SHA-512-PLUS": storedKey512 := sha512.Sum512(clientKey) storedKey = storedKey512[:] case "SCRAM-SHA-256", "SCRAM-SH-256-PLUS": storedKey256 := sha256.Sum256(clientKey) storedKey = storedKey256[:] case "SCRAM-SHA-1", "SCRAM-SHA-1-PLUS": storedKey1 := sha1.Sum(clientKey) storedKey = storedKey1[:] } _, err = h.Write([]byte("Server Key")) if err != nil { return err } serverFirstMessage, err := base64.StdEncoding.DecodeString(sfm) if err != nil { return err } authMessage = strings.SplitAfter(clientFirstMessage, ",,")[1] + "," + string(serverFirstMessage) + "," + clientFinalMessageBare h = hmac.New(shaNewFn, storedKey[:]) _, err = h.Write([]byte(authMessage)) if err != nil { return err } clientSignature := h.Sum(nil) h.Reset() if len(clientKey) != len(clientSignature) { return errors.New("SCRAM: client key and signature length mismatch") } clientProof := make([]byte, len(clientKey)) for i := range clientKey { clientProof[i] = clientKey[i] ^ clientSignature[i] } h = hmac.New(shaNewFn, saltedPassword) _, err = h.Write([]byte("Server Key")) if err != nil { return err } serverKey := h.Sum(nil) h.Reset() h = hmac.New(shaNewFn, serverKey) _, err = h.Write([]byte(authMessage)) if err != nil { return err } serverSignature = h.Sum(nil) if string(serverSignature) == "" { return errors.New("SCRAM: calculated an empty server signature") } clientFinalMessage := base64.StdEncoding.EncodeToString([]byte(clientFinalMessageBare + ",p=" + base64.StdEncoding.EncodeToString(clientProof))) fmt.Fprintf(c.stanzaWriter, "%s\n", nsSASL, clientFinalMessage) } if mechanism == "X-OAUTH2" && o.OAuthToken != "" && o.OAuthScope != "" { // Oauth authentication: send base64-encoded \x00 user \x00 token. raw := "\x00" + user + "\x00" + o.OAuthToken enc := make([]byte, base64.StdEncoding.EncodedLen(len(raw))) base64.StdEncoding.Encode(enc, []byte(raw)) fmt.Fprintf(c.stanzaWriter, "%s\n", nsSASL, o.OAuthXmlNs, enc) } if mechanism == "PLAIN" { // Plain authentication: send base64-encoded \x00 user \x00 password. raw := "\x00" + user + "\x00" + o.Password enc := make([]byte, base64.StdEncoding.EncodedLen(len(raw))) base64.StdEncoding.Encode(enc, []byte(raw)) fmt.Fprintf(c.conn, "%s\n", nsSASL, enc) } } if mechanism == "" { return fmt.Errorf("no viable authentication method available: %v", f.Mechanisms.Mechanism) } // Next message should be either success or failure. name, val, err := next(c.p) if err != nil { return err } switch v := val.(type) { case *saslSuccess: if strings.HasPrefix(mechanism, "SCRAM-SHA") { successMsg, err := base64.StdEncoding.DecodeString(v.Text) if err != nil { return err } if !strings.HasPrefix(string(successMsg), "v=") { return errors.New("server sent unexpected content in SCRAM success message") } serverSignatureReply := strings.SplitN(string(successMsg), "v=", 2)[1] serverSignatureRemote, err := base64.StdEncoding.DecodeString(serverSignatureReply) if err != nil { return err } if string(serverSignature) != string(serverSignatureRemote) { return errors.New("SCRAM: server signature mismatch") } c.Mechanism = mechanism } case *saslFailure: errorMessage := v.Text if errorMessage == "" { // v.Any is type of sub-element in failure, // which gives a description of what failed if there was no text element errorMessage = v.Any.Local } return errors.New("auth failure: " + errorMessage) default: return errors.New("expected or , got <" + name.Local + "> in " + name.Space) } // Now that we're authenticated, we're supposed to start the stream over again. // Declare intent to be a jabber client. if f, err = c.startStream(o, domain); err != nil { return err } // Generate a unique cookie cookie := getCookie() // Send IQ message asking to bind to the local user name. if o.Resource == "" { fmt.Fprintf(c.stanzaWriter, "\n", cookie, nsBind) } else { fmt.Fprintf(c.stanzaWriter, "%s\n", cookie, nsBind, o.Resource) } var iq clientIQ if err = c.p.DecodeElement(&iq, nil); err != nil { return errors.New("unmarshal : " + err.Error()) } if &iq.Bind == nil { return errors.New(" result missing ") } c.jid = iq.Bind.Jid // our local id c.domain = domain if o.Session { // if server support session, open it fmt.Fprintf(c.stanzaWriter, "\n", xmlEscape(domain), cookie, nsSession) } // We're connected and can now receive and send messages. fmt.Fprintf(c.stanzaWriter, "%s%s\n", o.Status, o.StatusMessage) return nil } // startTlsIfRequired examines the server's stream features and, if STARTTLS is required or supported, performs the TLS handshake. // f will be updated if the handshake completes, as the new stream's features are typically different from the original. func (c *Client) startTLSIfRequired(f *streamFeatures, o *Options, domain string) (*streamFeatures, error) { // whether we start tls is a matter of opinion: the server's and the user's. switch { case f.StartTLS == nil: // the server does not support STARTTLS return f, nil case !o.StartTLS && f.StartTLS.Required == nil: return f, nil case f.StartTLS.Required != nil: // the server requires STARTTLS. case !o.StartTLS: // the user wants STARTTLS and the server supports it. } var err error fmt.Fprintf(c.stanzaWriter, "\n") var k tlsProceed if err = c.p.DecodeElement(&k, nil); err != nil { return f, errors.New("unmarshal : " + err.Error()) } tc := o.TLSConfig if tc == nil { tc = DefaultConfig.Clone() // TODO(scott): we should consider using the server's address or reverse lookup tc.ServerName = domain } t := tls.Client(c.conn, tc) if err = t.Handshake(); err != nil { return f, errors.New("starttls handshake: " + err.Error()) } c.conn = t // restart our declaration of XMPP stream intentions. tf, err := c.startStream(o, domain) if err != nil { return f, err } return tf, nil } // startStream will start a new XML decoder for the connection, signal the start of a stream to the server and verify that the server has // also started the stream; if o.Debug is true, startStream will tee decoded XML data to stderr. The features advertised by the server // will be returned. func (c *Client) startStream(o *Options, domain string) (*streamFeatures, error) { if o.Debug { c.p = xml.NewDecoder(tee{c.conn, DebugWriter}) c.stanzaWriter = io.MultiWriter(c.conn, DebugWriter) } else { c.p = xml.NewDecoder(c.conn) c.stanzaWriter = c.conn } _, err := fmt.Fprintf(c.stanzaWriter, ""+ "\n", xmlEscape(domain), nsClient, nsStream) if err != nil { return nil, err } // We expect the server to start a . se, err := nextStart(c.p) if err != nil { return nil, err } if se.Name.Space != nsStream || se.Name.Local != "stream" { return nil, fmt.Errorf("expected but got <%v> in %v", se.Name.Local, se.Name.Space) } // Now we're in the stream and can use Unmarshal. // Next message should be to tell us authentication options. // See section 4.6 in RFC 3920. f := new(streamFeatures) if err = c.p.DecodeElement(f, nil); err != nil { return f, errors.New("unmarshal : " + err.Error()) } return f, nil } // IsEncrypted will return true if the client is connected using a TLS transport, either because it used. // TLS to connect from the outset, or because it successfully used STARTTLS to promote a TCP connection to TLS. func (c *Client) IsEncrypted() bool { _, ok := c.conn.(*tls.Conn) return ok } // Chat is an incoming or outgoing XMPP chat message. type Chat struct { Remote string Type string Text string Subject string Thread string Ooburl string Oobdesc string Lang string Roster Roster Other []string OtherElem []XMLElement Stamp time.Time } type Roster []Contact type Contact struct { Remote string Name string Group []string } // Presence is an XMPP presence notification. type Presence struct { From string To string Type string Show string Status string } type IQ struct { ID string From string To string Type string Query []byte } // Recv waits to receive the next XMPP stanza. // Return type is either a presence notification or a chat message. func (c *Client) Recv() (stanza interface{}, err error) { for { _, val, err := next(c.p) if err != nil { return Chat{}, err } switch v := val.(type) { case *clientMessage: if v.Event.XMLNS == XMPPNS_PUBSUB_EVENT { // Handle Pubsub notifications switch v.Event.Items.Node { case XMPPNS_AVATAR_PEP_METADATA: if len(v.Event.Items.Items) == 0 { return AvatarMetadata{}, errors.New("No avatar metadata items available") } return handleAvatarMetadata(v.Event.Items.Items[0].Body, v.From) // I am not sure whether this can even happen. // XEP-0084 only specifies a subscription to // the metadata node. /*case XMPPNS_AVATAR_PEP_DATA: return handleAvatarData(v.Event.Items.Items[0].Body, v.From, v.Event.Items.Items[0].ID)*/ default: return pubsubClientToReturn(v.Event), nil } } stamp, _ := time.Parse( "2006-01-02T15:04:05Z", v.Delay.Stamp, ) chat := Chat{ Remote: v.From, Type: v.Type, Text: v.Body, Subject: v.Subject, Thread: v.Thread, Other: v.OtherStrings(), OtherElem: v.Other, Stamp: stamp, Lang: v.Lang, } return chat, nil case *clientQuery: var r Roster for _, item := range v.Item { r = append(r, Contact{item.Jid, item.Name, item.Group}) } return Chat{Type: "roster", Roster: r}, nil case *clientPresence: return Presence{v.From, v.To, v.Type, v.Show, v.Status}, nil case *clientIQ: switch { case v.Query.XMLName.Space == "urn:xmpp:ping": // TODO check more strictly err := c.SendResultPing(v.ID, v.From) if err != nil { return Chat{}, err } fallthrough case v.Type == "error": switch v.ID { case "sub1": // Pubsub subscription failed var errs []clientPubsubError err := xml.Unmarshal([]byte(v.Error.InnerXML), &errs) if err != nil { return PubsubSubscription{}, err } var errsStr []string for _, e := range errs { errsStr = append(errsStr, e.XMLName.Local) } return PubsubSubscription{ Errors: errsStr, }, nil default: res, err := xml.Marshal(v.Query) if err != nil { return Chat{}, err } return IQ{ ID: v.ID, From: v.From, To: v.To, Type: v.Type, Query: res, }, nil } case v.Type == "result": switch v.ID { case "sub1": if v.Query.XMLName.Local == "pubsub" { // Subscription or unsubscription was successful var sub clientPubsubSubscription err := xml.Unmarshal([]byte(v.Query.InnerXML), &sub) if err != nil { return PubsubSubscription{}, err } return PubsubSubscription{ SubID: sub.SubID, JID: sub.JID, Node: sub.Node, Errors: nil, }, nil } case "unsub1": if v.Query.XMLName.Local == "pubsub" { var sub clientPubsubSubscription err := xml.Unmarshal([]byte(v.Query.InnerXML), &sub) if err != nil { return PubsubUnsubscription{}, err } return PubsubUnsubscription{ SubID: sub.SubID, JID: v.From, Node: sub.Node, Errors: nil, }, nil } else { // Unsubscribing MAY contain a pubsub element. But it does // not have to return PubsubUnsubscription{ SubID: "", JID: v.From, Node: "", Errors: nil, }, nil } case "info1": if v.Query.XMLName.Space == XMPPNS_DISCO_ITEMS { var itemsQuery clientDiscoItemsQuery err := xml.Unmarshal(v.InnerXML, &itemsQuery) if err != nil { return []DiscoItem{}, err } return DiscoItems{ Jid: v.From, Items: clientDiscoItemsToReturn(itemsQuery.Items), }, nil } case "info3": if v.Query.XMLName.Space == XMPPNS_DISCO_INFO { var disco clientDiscoQuery err := xml.Unmarshal(v.InnerXML, &disco) if err != nil { return DiscoResult{}, err } return DiscoResult{ Features: clientFeaturesToReturn(disco.Features), Identities: clientIdentitiesToReturn(disco.Identities), }, nil } case "items1", "items3": if v.Query.XMLName.Local == "pubsub" { var p clientPubsubItems err := xml.Unmarshal([]byte(v.Query.InnerXML), &p) if err != nil { return PubsubItems{}, err } switch p.Node { case XMPPNS_AVATAR_PEP_DATA: if len(p.Items) == 0 { return AvatarData{}, errors.New("No avatar data items available") } return handleAvatarData(p.Items[0].Body, v.From, p.Items[0].ID) case XMPPNS_AVATAR_PEP_METADATA: if len(p.Items) == 0 { return AvatarMetadata{}, errors.New("No avatar metadata items available") } return handleAvatarMetadata(p.Items[0].Body, v.From) default: return PubsubItems{ p.Node, pubsubItemsToReturn(p.Items), }, nil } } // Note: XEP-0084 states that metadata and data // should be fetched with an id of retrieve1. // Since we already have PubSub implemented, we // can just use items1 and items3 to do the same // as an Avatar node is just a PEP (PubSub) node. /*case "retrieve1": var p clientPubsubItems err := xml.Unmarshal([]byte(v.Query.InnerXML), &p) if err != nil { return PubsubItems{}, err } switch p.Node { case XMPPNS_AVATAR_PEP_DATA: return handleAvatarData(p.Items[0].Body, v.From, p.Items[0].ID) case XMPPNS_AVATAR_PEP_METADATA: return handleAvatarMetadata(p.Items[0].Body, v }*/ default: res, err := xml.Marshal(v.Query) if err != nil { return Chat{}, err } return IQ{ ID: v.ID, From: v.From, To: v.To, Type: v.Type, Query: res, }, nil } case v.Query.XMLName.Local == "": return IQ{ID: v.ID, From: v.From, To: v.To, Type: v.Type}, nil default: res, err := xml.Marshal(v.Query) if err != nil { return Chat{}, err } return IQ{ ID: v.ID, From: v.From, To: v.To, Type: v.Type, Query: res, }, nil } } } } // Send sends the message wrapped inside an XMPP message stanza body. func (c *Client) Send(chat Chat) (n int, err error) { var subtext, thdtext, oobtext string if chat.Subject != `` { subtext = `` + xmlEscape(chat.Subject) + `` } if chat.Thread != `` { thdtext = `` + xmlEscape(chat.Thread) + `` } if chat.Ooburl != `` { oobtext = `` + xmlEscape(chat.Ooburl) + `` if chat.Oobdesc != `` { oobtext += `` + xmlEscape(chat.Oobdesc) + `` } oobtext += `` } stanza := "" + subtext + "%s" + oobtext + thdtext + "\n" chat.Text = validUTF8(chat.Text) return fmt.Fprintf(c.stanzaWriter, stanza, xmlEscape(chat.Remote), xmlEscape(chat.Type), cnonce(), xmlEscape(chat.Text)) } // SendOOB sends OOB data wrapped inside an XMPP message stanza, without actual body. func (c *Client) SendOOB(chat Chat) (n int, err error) { var thdtext, oobtext string if chat.Thread != `` { thdtext = `` + xmlEscape(chat.Thread) + `` } if chat.Ooburl != `` { oobtext = `` + xmlEscape(chat.Ooburl) + `` if chat.Oobdesc != `` { oobtext += `` + xmlEscape(chat.Oobdesc) + `` } oobtext += `` } return fmt.Fprintf(c.stanzaWriter, ""+oobtext+thdtext+"\n", xmlEscape(chat.Remote), xmlEscape(chat.Type), cnonce()) } // SendOrg sends the original text without being wrapped in an XMPP message stanza. func (c *Client) SendOrg(org string) (n int, err error) { return fmt.Fprint(c.stanzaWriter, org+"\n") } // SendPresence sends Presence wrapped inside XMPP presence stanza. func (c *Client) SendPresence(presence Presence) (n int, err error) { // Forge opening presence tag var buf string = "" // TODO: there may be optional tag "priority", but former presence type does not take this into account // so either we must follow std, change type xmpp.Presence and break backward compatibility // or leave it as-is and potentially break client software if presence.Show != "" { // https://www.ietf.org/rfc/rfc3921.txt 2.2.2.1, show can be only // away, chat, dnd, xa switch presence.Show { case "away", "chat", "dnd", "xa": buf = buf + fmt.Sprintf("%s", xmlEscape(presence.Show)) } } if presence.Status != "" { buf = buf + fmt.Sprintf("%s", xmlEscape(presence.Status)) } buf = buf + "" return fmt.Fprint(c.stanzaWriter, buf) } // SendKeepAlive sends a "whitespace keepalive" as described in chapter 4.6.1 of RFC6120. func (c *Client) SendKeepAlive() (n int, err error) { return fmt.Fprintf(c.conn, " ") } // SendHtml sends the message as HTML as defined by XEP-0071 func (c *Client) SendHtml(chat Chat) (n int, err error) { return fmt.Fprintf(c.stanzaWriter, ""+ "%s"+ "%s\n", xmlEscape(chat.Remote), xmlEscape(chat.Type), xmlEscape(chat.Text), chat.Text) } // Roster asks for the chat roster. func (c *Client) Roster() error { fmt.Fprintf(c.stanzaWriter, "\n", xmlEscape(c.jid)) return nil } // RFC 3920 C.1 Streams name space type streamFeatures struct { XMLName xml.Name `xml:"http://etherx.jabber.org/streams features"` StartTLS *tlsStartTLS Mechanisms saslMechanisms ChannelBindings saslChannelBindings Bind bindBind Session bool } type streamError struct { XMLName xml.Name `xml:"http://etherx.jabber.org/streams error"` Any xml.Name Text string } // RFC 3920 C.3 TLS name space type tlsStartTLS struct { XMLName xml.Name `xml:"urn:ietf:params:xml:ns:xmpp-tls starttls"` Required *string `xml:"required"` } type tlsProceed struct { XMLName xml.Name `xml:"urn:ietf:params:xml:ns:xmpp-tls proceed"` } type tlsFailure struct { XMLName xml.Name `xml:"urn:ietf:params:xml:ns:xmpp-tls failure"` } // RFC 3920 C.4 SASL name space type saslMechanisms struct { XMLName xml.Name `xml:"urn:ietf:params:xml:ns:xmpp-sasl mechanisms"` Mechanism []string `xml:"mechanism"` } type saslChannelBindings struct { XMLName xml.Name `xml:"sasl-channel-binding"` Text string `xml:",chardata"` Xmlns string `xml:"xmlns,attr"` ChannelBinding []struct { Text string `xml:",chardata"` Type string `xml:"type,attr"` } `xml:"channel-binding"` } type saslAbort struct { XMLName xml.Name `xml:"urn:ietf:params:xml:ns:xmpp-sasl abort"` } type saslSuccess struct { XMLName xml.Name `xml:"urn:ietf:params:xml:ns:xmpp-sasl success"` Text string `xml:",chardata"` } type saslFailure struct { XMLName xml.Name `xml:"urn:ietf:params:xml:ns:xmpp-sasl failure"` Any xml.Name `xml:",any"` Text string `xml:"text"` } // RFC 3920 C.5 Resource binding name space type bindBind struct { XMLName xml.Name `xml:"urn:ietf:params:xml:ns:xmpp-bind bind"` Resource string Jid string `xml:"jid"` } // RFC 3921 B.1 jabber:client type clientMessage struct { XMLName xml.Name `xml:"jabber:client message"` From string `xml:"from,attr"` ID string `xml:"id,attr"` To string `xml:"to,attr"` Type string `xml:"type,attr"` // chat, error, groupchat, headline, or normal Lang string `xml:"lang,attr"` // These should technically be []clientText, but string is much more convenient. Subject string `xml:"subject"` Body string `xml:"body"` Thread string `xml:"thread"` // Pubsub Event clientPubsubEvent `xml:"event"` // Any hasn't matched element Other []XMLElement `xml:",any"` Delay Delay `xml:"delay"` } func (m *clientMessage) OtherStrings() []string { a := make([]string, len(m.Other)) for i, e := range m.Other { a[i] = e.String() } return a } type XMLElement struct { XMLName xml.Name Attr []xml.Attr `xml:",any,attr"` // Save the attributes of the xml element InnerXML string `xml:",innerxml"` } func (e *XMLElement) String() string { r := bytes.NewReader([]byte(e.InnerXML)) d := xml.NewDecoder(r) var buf bytes.Buffer for { tok, err := d.Token() if err != nil { break } switch v := tok.(type) { case xml.StartElement: err = d.Skip() case xml.CharData: _, err = buf.Write(v) } if err != nil { break } } return buf.String() } type Delay struct { Stamp string `xml:"stamp,attr"` } type clientPresence struct { XMLName xml.Name `xml:"jabber:client presence"` From string `xml:"from,attr"` ID string `xml:"id,attr"` To string `xml:"to,attr"` Type string `xml:"type,attr"` // error, probe, subscribe, subscribed, unavailable, unsubscribe, unsubscribed Lang string `xml:"lang,attr"` Show string `xml:"show"` // away, chat, dnd, xa Status string `xml:"status"` // sb []clientText Priority string `xml:"priority,attr"` Error *clientError } type clientIQ struct { // info/query XMLName xml.Name `xml:"jabber:client iq"` From string `xml:"from,attr"` ID string `xml:"id,attr"` To string `xml:"to,attr"` Type string `xml:"type,attr"` // error, get, result, set Query XMLElement `xml:",any"` Error clientError Bind bindBind InnerXML []byte `xml:",innerxml"` } type clientError struct { XMLName xml.Name `xml:"jabber:client error"` Code string `xml:",attr"` Type string `xml:"type,attr"` Any xml.Name InnerXML []byte `xml:",innerxml"` Text string } type clientQuery struct { Item []rosterItem } type rosterItem struct { XMLName xml.Name `xml:"jabber:iq:roster item"` Jid string `xml:",attr"` Name string `xml:",attr"` Subscription string `xml:",attr"` Group []string } // Scan XML token stream to find next StartElement. func nextStart(p *xml.Decoder) (xml.StartElement, error) { for { t, err := p.Token() if err != nil || t == nil { return xml.StartElement{}, err } switch t := t.(type) { case xml.StartElement: return t, nil } } } // Scan XML token stream to find next EndElement func nextEnd(p *xml.Decoder) (xml.EndElement, error) { p.Strict = false for { t, err := p.Token() if err != nil || t == nil { return xml.EndElement{}, err } switch t := t.(type) { case xml.EndElement: return t, nil } } } // Scan XML token stream for next element and save into val. // If val == nil, allocate new element based on proto map. // Either way, return val. func next(p *xml.Decoder) (xml.Name, interface{}, error) { // Read start element to find out what type we want. se, err := nextStart(p) if err != nil { return xml.Name{}, nil, err } // Put it in an interface and allocate one. var nv interface{} switch se.Name.Space + " " + se.Name.Local { case nsStream + " features": nv = &streamFeatures{} case nsStream + " error": nv = &streamError{} case nsTLS + " starttls": nv = &tlsStartTLS{} case nsTLS + " proceed": nv = &tlsProceed{} case nsTLS + " failure": nv = &tlsFailure{} case nsSASL + " mechanisms": nv = &saslMechanisms{} case nsSASL + " challenge": nv = "" case nsSASL + " response": nv = "" case nsSASL + " abort": nv = &saslAbort{} case nsSASL + " success": nv = &saslSuccess{} case nsSASL + " failure": nv = &saslFailure{} case nsSASLCB + " sasl-channel-binding": nv = &saslChannelBindings{} case nsBind + " bind": nv = &bindBind{} case nsClient + " message": nv = &clientMessage{} case nsClient + " presence": nv = &clientPresence{} case nsClient + " iq": nv = &clientIQ{} case nsClient + " error": nv = &clientError{} default: return xml.Name{}, nil, errors.New("unexpected XMPP message " + se.Name.Space + " <" + se.Name.Local + "/>") } // Unmarshal into that storage. if err = p.DecodeElement(nv, &se); err != nil { return xml.Name{}, nil, err } return se.Name, nv, err } func xmlEscape(s string) string { var b bytes.Buffer xml.Escape(&b, []byte(s)) return b.String() } type tee struct { r io.Reader w io.Writer } func (t tee) Read(p []byte) (n int, err error) { n, err = t.r.Read(p) if n > 0 { t.w.Write(p[0:n]) t.w.Write([]byte("\n")) } return } func validUTF8(s string) string { // Remove invalid code points. s = strings.ToValidUTF8(s, "�") reg := regexp.MustCompile(`[\x{0000}-\x{0008}\x{000B}\x{000C}\x{000E}-\x{001F}]`) s = reg.ReplaceAllString(s, "�") return s } go-xmpp-0.1.1/xmpp_avatar.go000066400000000000000000000052241457105526500157660ustar00rootroot00000000000000package xmpp import ( "crypto/sha1" "encoding/base64" "encoding/hex" "encoding/xml" "errors" "strconv" ) const ( XMPPNS_AVATAR_PEP_DATA = "urn:xmpp:avatar:data" XMPPNS_AVATAR_PEP_METADATA = "urn:xmpp:avatar:metadata" ) type clientAvatarData struct { XMLName xml.Name `xml:"data"` Data []byte `xml:",innerxml"` } type clientAvatarInfo struct { XMLName xml.Name `xml:"info"` Bytes string `xml:"bytes,attr"` Width string `xml:"width,attr"` Height string `xml:"height,attr"` ID string `xml:"id,attr"` Type string `xml:"type,attr"` URL string `xml:"url,attr"` } type clientAvatarMetadata struct { XMLName xml.Name `xml:"metadata"` XMLNS string `xml:"xmlns,attr"` Info clientAvatarInfo `xml:"info"` } type AvatarData struct { Data []byte From string } type AvatarMetadata struct { From string Bytes int Width int Height int ID string Type string URL string } func handleAvatarData(itemsBody []byte, from, id string) (AvatarData, error) { var data clientAvatarData err := xml.Unmarshal(itemsBody, &data) if err != nil { return AvatarData{}, err } // Base64-decode the avatar data to check its SHA1 hash dataRaw, err := base64.StdEncoding.DecodeString( string(data.Data)) if err != nil { return AvatarData{}, err } hash := sha1.Sum(dataRaw) hashStr := hex.EncodeToString(hash[:]) if hashStr != id { return AvatarData{}, errors.New("SHA1 hashes do not match") } return AvatarData{ Data: dataRaw, From: from, }, nil } func handleAvatarMetadata(body []byte, from string) (AvatarMetadata, error) { var meta clientAvatarMetadata err := xml.Unmarshal(body, &meta) if err != nil { return AvatarMetadata{}, err } return AvatarMetadata{ From: from, Bytes: atoiw(meta.Info.Bytes), Width: atoiw(meta.Info.Width), Height: atoiw(meta.Info.Height), ID: meta.Info.ID, Type: meta.Info.Type, URL: meta.Info.URL, }, nil } // A wrapper for atoi which just returns -1 if an error occurs func atoiw(str string) int { i, err := strconv.Atoi(str) if err != nil { return -1 } return i } func (c *Client) AvatarSubscribeMetadata(jid string) { c.PubsubSubscribeNode(XMPPNS_AVATAR_PEP_METADATA, jid) } func (c *Client) AvatarUnsubscribeMetadata(jid string) { c.PubsubUnsubscribeNode(XMPPNS_AVATAR_PEP_METADATA, jid) } func (c *Client) AvatarRequestData(jid string) { c.PubsubRequestLastItems(XMPPNS_AVATAR_PEP_DATA, jid) } func (c *Client) AvatarRequestDataByID(jid, id string) { c.PubsubRequestItem(XMPPNS_AVATAR_PEP_DATA, jid, id) } func (c *Client) AvatarRequestMetadata(jid string) { c.PubsubRequestLastItems(XMPPNS_AVATAR_PEP_METADATA, jid) } go-xmpp-0.1.1/xmpp_disco.go000066400000000000000000000035701457105526500156130ustar00rootroot00000000000000package xmpp import ( "encoding/xml" ) const ( XMPPNS_DISCO_ITEMS = "http://jabber.org/protocol/disco#items" XMPPNS_DISCO_INFO = "http://jabber.org/protocol/disco#info" ) type clientDiscoFeature struct { XMLName xml.Name `xml:"feature"` Var string `xml:"var,attr"` } type clientDiscoIdentity struct { XMLName xml.Name `xml:"identity"` Category string `xml:"category,attr"` Type string `xml:"type,attr"` Name string `xml:"name,attr"` } type clientDiscoQuery struct { XMLName xml.Name `xml:"query"` Features []clientDiscoFeature `xml:"feature"` Identities []clientDiscoIdentity `xml:"identity"` } type clientDiscoItem struct { XMLName xml.Name `xml:"item"` Jid string `xml:"jid,attr"` Node string `xml:"node,attr"` Name string `xml:"name,attr"` } type clientDiscoItemsQuery struct { XMLName xml.Name `xml:"query"` Items []clientDiscoItem `xml:"item"` } type DiscoIdentity struct { Category string Type string Name string } type DiscoItem struct { Jid string Name string Node string } type DiscoResult struct { Features []string Identities []DiscoIdentity } type DiscoItems struct { Jid string Items []DiscoItem } func clientFeaturesToReturn(features []clientDiscoFeature) []string { var ret []string for _, feature := range features { ret = append(ret, feature.Var) } return ret } func clientIdentitiesToReturn(identities []clientDiscoIdentity) []DiscoIdentity { var ret []DiscoIdentity for _, id := range identities { ret = append(ret, DiscoIdentity{ Category: id.Category, Type: id.Type, Name: id.Name, }) } return ret } func clientDiscoItemsToReturn(items []clientDiscoItem) []DiscoItem { var ret []DiscoItem for _, item := range items { ret = append(ret, DiscoItem{ Jid: item.Jid, Name: item.Name, Node: item.Node, }) } return ret } go-xmpp-0.1.1/xmpp_information_query.go000066400000000000000000000033141457105526500202600ustar00rootroot00000000000000package xmpp import ( "fmt" "strconv" ) const ( IQTypeGet = "get" IQTypeSet = "set" IQTypeResult = "result" ) func (c *Client) Discovery() (string, error) { // use getCookie for a pseudo random id. reqID := strconv.FormatUint(uint64(getCookie()), 10) return c.RawInformationQuery(c.jid, c.domain, reqID, IQTypeGet, XMPPNS_DISCO_ITEMS, "") } // Discover information about a node func (c *Client) DiscoverNodeInfo(node string) (string, error) { query := fmt.Sprintf("", XMPPNS_DISCO_INFO, node) return c.RawInformation(c.jid, c.domain, "info3", IQTypeGet, query) } // Discover items that the server exposes func (c *Client) DiscoverServerItems() (string, error) { return c.DiscoverEntityItems(c.domain) } // Discover items that an entity exposes func (c *Client) DiscoverEntityItems(jid string) (string, error) { query := fmt.Sprintf("", XMPPNS_DISCO_ITEMS) return c.RawInformation(c.jid, jid, "info1", IQTypeGet, query) } // RawInformationQuery sends an information query request to the server. func (c *Client) RawInformationQuery(from, to, id, iqType, requestNamespace, body string) (string, error) { const xmlIQ = "%s\n" _, err := fmt.Fprintf(c.stanzaWriter, xmlIQ, xmlEscape(from), xmlEscape(to), id, iqType, requestNamespace, body) return id, err } // rawInformation send a IQ request with the payload body to the server func (c *Client) RawInformation(from, to, id, iqType, body string) (string, error) { const xmlIQ = "%s\n" _, err := fmt.Fprintf(c.stanzaWriter, xmlIQ, xmlEscape(from), xmlEscape(to), id, iqType, body) return id, err } go-xmpp-0.1.1/xmpp_muc.go000066400000000000000000000102521457105526500152710ustar00rootroot00000000000000// Copyright 2013 Flo Lauber . All rights reserved. // Use of this source code is governed by a BSD-style // license that can be found in the LICENSE file. // TODO(flo): // - support password protected MUC rooms // - cleanup signatures of join/leave functions package xmpp import ( "errors" "fmt" "time" ) const ( nsMUC = "http://jabber.org/protocol/muc" nsMUCUser = "http://jabber.org/protocol/muc#user" NoHistory = 0 CharHistory = 1 StanzaHistory = 2 SecondsHistory = 3 SinceHistory = 4 ) // Send sends room topic wrapped inside an XMPP message stanza body. func (c *Client) SendTopic(chat Chat) (n int, err error) { return fmt.Fprintf(c.stanzaWriter, ""+"%s\n", xmlEscape(chat.Remote), xmlEscape(chat.Type), xmlEscape(chat.Text)) } func (c *Client) JoinMUCNoHistory(jid, nick string) (n int, err error) { if nick == "" { nick = c.jid } return fmt.Fprintf(c.stanzaWriter, "\n"+ ""+ "\n"+ "", xmlEscape(jid), xmlEscape(nick), nsMUC) } // xep-0045 7.2 func (c *Client) JoinMUC(jid, nick string, history_type, history int, history_date *time.Time) (n int, err error) { if nick == "" { nick = c.jid } switch history_type { case NoHistory: return fmt.Fprintf(c.stanzaWriter, ""+ ""+ "\n", xmlEscape(jid), xmlEscape(nick), nsMUC) case CharHistory: return fmt.Fprintf(c.stanzaWriter, ""+ "\n"+ ""+ "\n", xmlEscape(jid), xmlEscape(nick), nsMUC, history) case StanzaHistory: return fmt.Fprintf(c.stanzaWriter, ""+ ""+ ""+ "\n", xmlEscape(jid), xmlEscape(nick), nsMUC, history) case SecondsHistory: return fmt.Fprintf(c.stanzaWriter, ""+ ""+ ""+ "\n", xmlEscape(jid), xmlEscape(nick), nsMUC, history) case SinceHistory: if history_date != nil { return fmt.Fprintf(c.stanzaWriter, ""+ ""+ ""+ "\n", xmlEscape(jid), xmlEscape(nick), nsMUC, history_date.Format(time.RFC3339)) } } return 0, errors.New("Unknown history option") } // xep-0045 7.2.6 func (c *Client) JoinProtectedMUC(jid, nick string, password string, history_type, history int, history_date *time.Time) (n int, err error) { if nick == "" { nick = c.jid } switch history_type { case NoHistory: return fmt.Fprintf(c.stanzaWriter, ""+ ""+ "%s"+ ""+ "\n", xmlEscape(jid), xmlEscape(nick), nsMUC, xmlEscape(password)) case CharHistory: return fmt.Fprintf(c.stanzaWriter, ""+ ""+ "%s"+ ""+ "\n", xmlEscape(jid), xmlEscape(nick), nsMUC, xmlEscape(password), history) case StanzaHistory: return fmt.Fprintf(c.stanzaWriter, ""+ ""+ "%s"+ ""+ "\n", xmlEscape(jid), xmlEscape(nick), nsMUC, xmlEscape(password), history) case SecondsHistory: return fmt.Fprintf(c.stanzaWriter, ""+ ""+ "%s"+ ""+ "\n", xmlEscape(jid), xmlEscape(nick), nsMUC, xmlEscape(password), history) case SinceHistory: if history_date != nil { return fmt.Fprintf(c.stanzaWriter, ""+ ""+ "%s"+ ""+ "\n", xmlEscape(jid), xmlEscape(nick), nsMUC, xmlEscape(password), history_date.Format(time.RFC3339)) } } return 0, errors.New("Unknown history option") } // xep-0045 7.14 func (c *Client) LeaveMUC(jid string) (n int, err error) { return fmt.Fprintf(c.stanzaWriter, "\n", c.jid, xmlEscape(jid)) } go-xmpp-0.1.1/xmpp_ping.go000066400000000000000000000014351457105526500154450ustar00rootroot00000000000000package xmpp import ( "fmt" ) func (c *Client) PingC2S(jid, server string) error { if jid == "" { jid = c.jid } if server == "" { server = c.domain } _, err := fmt.Fprintf(c.stanzaWriter, ""+ ""+ "\n", xmlEscape(jid), xmlEscape(server)) return err } func (c *Client) PingS2S(fromServer, toServer string) error { _, err := fmt.Fprintf(c.stanzaWriter, ""+ ""+ "\n", xmlEscape(fromServer), xmlEscape(toServer)) return err } func (c *Client) SendResultPing(id, toServer string) error { _, err := fmt.Fprintf(c.stanzaWriter, "\n", xmlEscape(toServer), xmlEscape(id)) return err } go-xmpp-0.1.1/xmpp_pubsub.go000066400000000000000000000054051457105526500160110ustar00rootroot00000000000000package xmpp import ( "encoding/xml" "fmt" ) const ( XMPPNS_PUBSUB = "http://jabber.org/protocol/pubsub" XMPPNS_PUBSUB_EVENT = "http://jabber.org/protocol/pubsub#event" ) type clientPubsubItem struct { XMLName xml.Name `xml:"item"` ID string `xml:"id,attr"` Body []byte `xml:",innerxml"` } type clientPubsubItems struct { XMLName xml.Name `xml:"items"` Node string `xml:"node,attr"` Items []clientPubsubItem `xml:"item"` } type clientPubsubEvent struct { XMLName xml.Name `xml:"event"` XMLNS string `xml:"xmlns,attr"` Items clientPubsubItems `xml:"items"` } type clientPubsubError struct { XMLName xml.Name } type clientPubsubSubscription struct { XMLName xml.Name `xml:"subscription"` Node string `xml:"node,attr"` JID string `xml:"jid,attr"` SubID string `xml:"subid,attr"` } type PubsubEvent struct { Node string Items []PubsubItem } type PubsubSubscription struct { SubID string JID string Node string Errors []string } type PubsubUnsubscription PubsubSubscription type PubsubItem struct { ID string InnerXML []byte } type PubsubItems struct { Node string Items []PubsubItem } // Converts []clientPubsubItem to []PubsubItem func pubsubItemsToReturn(items []clientPubsubItem) []PubsubItem { var tmp []PubsubItem for _, i := range items { tmp = append(tmp, PubsubItem{ ID: i.ID, InnerXML: i.Body, }) } return tmp } func pubsubClientToReturn(event clientPubsubEvent) PubsubEvent { return PubsubEvent{ Node: event.Items.Node, Items: pubsubItemsToReturn(event.Items.Items), } } func pubsubStanza(body string) string { return fmt.Sprintf("%s", XMPPNS_PUBSUB, body) } func pubsubSubscriptionStanza(node, jid string) string { body := fmt.Sprintf("", xmlEscape(node), xmlEscape(jid)) return pubsubStanza(body) } func pubsubUnsubscriptionStanza(node, jid string) string { body := fmt.Sprintf("", xmlEscape(node), xmlEscape(jid)) return pubsubStanza(body) } func (c *Client) PubsubSubscribeNode(node, jid string) { c.RawInformation(c.jid, jid, "sub1", "set", pubsubSubscriptionStanza(node, c.jid)) } func (c *Client) PubsubUnsubscribeNode(node, jid string) { c.RawInformation(c.jid, jid, "unsub1", "set", pubsubUnsubscriptionStanza(node, c.jid)) } func (c *Client) PubsubRequestLastItems(node, jid string) { body := fmt.Sprintf("", node) c.RawInformation(c.jid, jid, "items1", "get", pubsubStanza(body)) } func (c *Client) PubsubRequestItem(node, jid, id string) { body := fmt.Sprintf("", node, id) c.RawInformation(c.jid, jid, "items3", "get", pubsubStanza(body)) } go-xmpp-0.1.1/xmpp_subscription.go000066400000000000000000000011321457105526500172260ustar00rootroot00000000000000package xmpp import ( "fmt" ) func (c *Client) ApproveSubscription(jid string) { fmt.Fprintf(c.stanzaWriter, "\n", xmlEscape(jid)) } func (c *Client) RevokeSubscription(jid string) { fmt.Fprintf(c.stanzaWriter, "\n", xmlEscape(jid)) } func (c *Client) RetrieveSubscription(jid string) { fmt.Fprintf(c.conn, "\n", xmlEscape(jid)) } func (c *Client) RequestSubscription(jid string) { fmt.Fprintf(c.stanzaWriter, "\n", xmlEscape(jid)) } go-xmpp-0.1.1/xmpp_test.go000066400000000000000000000060501457105526500154650ustar00rootroot00000000000000package xmpp import ( "bytes" "encoding/xml" "io" "net" "reflect" "strings" "testing" "time" ) type localAddr struct{} func (a *localAddr) Network() string { return "tcp" } func (addr *localAddr) String() string { return "localhost:5222" } type testConn struct { *bytes.Buffer } func tConnect(s string) net.Conn { var conn testConn conn.Buffer = bytes.NewBufferString(s) return &conn } func (*testConn) Close() error { return nil } func (*testConn) LocalAddr() net.Addr { return &localAddr{} } func (*testConn) RemoteAddr() net.Addr { return &localAddr{} } func (*testConn) SetDeadline(time.Time) error { return nil } func (*testConn) SetReadDeadline(time.Time) error { return nil } func (*testConn) SetWriteDeadline(time.Time) error { return nil } var text = strings.TrimSpace(` {"random": "<text>"} InvalidJson: JSON_PARSING_ERROR : Missing Required Field: message_id\n `) func TestStanzaError(t *testing.T) { var c Client c.conn = tConnect(text) c.p = xml.NewDecoder(c.conn) v, err := c.Recv() if err != nil { t.Fatalf("Recv() = %v", err) } chat := Chat{ Type: "error", Other: []string{ "\n\t\t{\"random\": \"\"}\n\t", "\n\t\t\n\t\t\n\t", }, OtherElem: []XMLElement{ { XMLName: xml.Name{Space: "google:mobile:data", Local: "gcm"}, Attr: []xml.Attr{{Name: xml.Name{Space: "", Local: "xmlns"}, Value: "google:mobile:data"}}, InnerXML: "\n\t\t{\"random\": \"<text>\"}\n\t", }, { XMLName: xml.Name{Space: "jabber:client", Local: "error"}, Attr: []xml.Attr{{Name: xml.Name{Space: "", Local: "code"}, Value: "400"}, {Name: xml.Name{Space: "", Local: "type"}, Value: "modify"}}, InnerXML: ` InvalidJson: JSON_PARSING_ERROR : Missing Required Field: message_id\n `, }, }, } if !reflect.DeepEqual(v, chat) { t.Errorf("Recv() = %#v; want %#v", v, chat) } } func TestEOFError(t *testing.T) { var c Client c.conn = tConnect("") c.p = xml.NewDecoder(c.conn) _, err := c.Recv() if err != io.EOF { t.Errorf("Recv() did not return io.EOF on end of input stream") } } var emptyPubSub = strings.TrimSpace(` `) func TestEmptyPubsub(t *testing.T) { var c Client c.conn = tConnect(emptyPubSub) c.p = xml.NewDecoder(c.conn) m, err := c.Recv() switch m.(type) { case AvatarData: if err == nil { t.Errorf("Expected an error to be returned") } default: t.Errorf("Recv() = %v", m) t.Errorf("Expected a return value of AvatarData") } }