uwaru/go-xmpp
1
0
Fork 0
forked from jshiffer/go-xmpp
Code Pull Requests Activity

added Options to NewClient, and Resource binding to Options

Browse Source
This commit is contained in:
Scott Dunlop 2013-08-12 15:33:50 -07:00
parent 52f561e157
commit 5e57ac52f9
1 changed files with 46 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
xmpp.go
View File

@ -95,10 +95,17 @@ func connect(host, user, passwd string) (net.Conn, error) {
return c, nil return c, nil
} }
// Options are used to specify additional options for new clients, such as a Resource.
type Options struct {
// 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
}
// NewClient creates a new connection to a host given as "hostname" or "hostname:port". // 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. // 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. // Default the port to 5222.
func NewClient(host, user, passwd string) (*Client, error) { func NewClient(host, user, passwd string, opts *Options) (*Client, error) {
c, err := connect(host, user, passwd) c, err := connect(host, user, passwd)
if err != nil { if err != nil {
return nil, err return nil, err
@ -118,14 +125,14 @@ func NewClient(host, user, passwd string) (*Client, error) {
client := new(Client) client := new(Client)
client.conn = tlsconn client.conn = tlsconn
if err := client.init(user, passwd); err != nil { if err := client.init(user, passwd, opts); err != nil {
client.Close() client.Close()
return nil, err return nil, err
} }
return client, nil return client, nil
} }
func NewClientNoTLS(host, user, passwd string) (*Client, error) { func NewClientNoTLS(host, user, passwd string, opts *Options) (*Client, error) {
c, err := connect(host, user, passwd) c, err := connect(host, user, passwd)
if err != nil { if err != nil {
return nil, err return nil, err
@ -133,7 +140,7 @@ func NewClientNoTLS(host, user, passwd string) (*Client, error) {
client := new(Client) client := new(Client)
client.conn = c client.conn = c
if err := client.init(user, passwd); err != nil { if err := client.init(user, passwd, opts); err != nil {
client.Close() client.Close()
return nil, err return nil, err
} }
@ -177,7 +184,7 @@ func cnonce() string {
return fmt.Sprintf("%016x", cn) return fmt.Sprintf("%016x", cn)
} }
func (c *Client) init(user, passwd string) error { func (c *Client) init(user, passwd string, opts *Options) error {
// For debugging: the following causes the plaintext of the connection to be duplicated to stdout. // For debugging: the following causes the plaintext of the connection to be duplicated to stdout.
//c.p = xml.NewDecoder(tee{c.conn, os.Stdout}) //c.p = xml.NewDecoder(tee{c.conn, os.Stdout})
c.p = xml.NewDecoder(c.conn) c.p = xml.NewDecoder(c.conn)
@ -189,6 +196,11 @@ func (c *Client) init(user, passwd string) error {
user = a[0] user = a[0]
domain := a[1] domain := a[1]
resource := ""
if opts != nil {
resource = opts.Resource
}
// Declare intent to be a jabber client. // Declare intent to be a jabber client.
fmt.Fprintf(c.conn, "<?xml version='1.0'?>\n"+ fmt.Fprintf(c.conn, "<?xml version='1.0'?>\n"+
"<stream:stream to='%s' xmlns='%s'\n"+ "<stream:stream to='%s' xmlns='%s'\n"+
@ -241,7 +253,7 @@ func (c *Client) init(user, passwd string) error {
kv := strings.SplitN(strings.TrimSpace(token), "=", 2) kv := strings.SplitN(strings.TrimSpace(token), "=", 2)
if len(kv) == 2 { if len(kv) == 2 {
if kv[1][0] == '"' && kv[1][len(kv[1])-1] == '"' { if kv[1][0] == '"' && kv[1][len(kv[1])-1] == '"' {
kv[1] = kv[1][1:len(kv[1])-1] kv[1] = kv[1][1 : len(kv[1])-1]
} }
tokens[kv[0]] = kv[1] tokens[kv[0]] = kv[1]
} }
@ -254,7 +266,7 @@ func (c *Client) init(user, passwd string) error {
digestUri := "xmpp/" + domain digestUri := "xmpp/" + domain
nonceCount := fmt.Sprintf("%08x", 1) nonceCount := fmt.Sprintf("%08x", 1)
digest := saslDigestResponse(user, realm, passwd, nonce, cnonceStr, "AUTHENTICATE", digestUri, nonceCount) digest := saslDigestResponse(user, realm, passwd, nonce, cnonceStr, "AUTHENTICATE", digestUri, nonceCount)
message := "username="+user+", realm="+realm+", nonce="+nonce+", cnonce="+cnonceStr+", nc="+nonceCount+", qop="+qop+", digest-uri="+digestUri+", response="+digest+", charset="+charset; message := "username=" + user + ", realm=" + realm + ", nonce=" + nonce + ", cnonce=" + cnonceStr + ", nc=" + nonceCount + ", qop=" + qop + ", digest-uri=" + digestUri + ", response=" + digest + ", charset=" + charset
fmt.Fprintf(c.conn, "<response xmlns='%s'>%s</response>\n", nsSASL, base64.StdEncoding.EncodeToString([]byte(message))) fmt.Fprintf(c.conn, "<response xmlns='%s'>%s</response>\n", nsSASL, base64.StdEncoding.EncodeToString([]byte(message)))
var rspauth saslRspAuth var rspauth saslRspAuth
@ -305,7 +317,11 @@ func (c *Client) init(user, passwd string) error {
} }
// Send IQ message asking to bind to the local user name. // Send IQ message asking to bind to the local user name.
fmt.Fprintf(c.conn, "<iq type='set' id='x'><bind xmlns='%s'/></iq>\n", nsBind) if resource == "" {
fmt.Fprintf(c.conn, "<iq type='set' id='x'><bind xmlns='%s'></bind></iq>\n", nsBind)
} else {
fmt.Fprintf(c.conn, "<iq type='set' id='x'><bind xmlns='%s'><resource>%s</resource></bind></iq>\n", nsBind, resource)
}
var iq clientIQ var iq clientIQ
if err = c.p.DecodeElement(&iq, nil); err != nil { if err = c.p.DecodeElement(&iq, nil); err != nil {
return errors.New("unmarshal <iq>: " + err.Error()) return errors.New("unmarshal <iq>: " + err.Error())