123 lines
3.1 KiB
Go
123 lines
3.1 KiB
Go
package client
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"encoding/xml"
|
|
"errors"
|
|
"fmt"
|
|
"net"
|
|
|
|
"dev.sum7.eu/genofire/yaja/messages"
|
|
"dev.sum7.eu/genofire/yaja/model"
|
|
)
|
|
|
|
func (client *Client) setConnection(conn net.Conn) {
|
|
client.conn = conn
|
|
client.in = xml.NewDecoder(client.conn)
|
|
client.out = xml.NewEncoder(client.conn)
|
|
}
|
|
|
|
func (client *Client) startStream() (*messages.StreamFeatures, error) {
|
|
// XMPP-Connection
|
|
_, err := fmt.Fprintf(client.conn, "<?xml version='1.0'?>\n"+
|
|
"<stream:stream to='%s' xmlns='%s'\n"+
|
|
" xmlns:stream='%s' version='1.0'>\n",
|
|
model.XMLEscape(client.JID.Domain), messages.NSClient, messages.NSStream)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
element, err := client.Read()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if element.Name.Space != messages.NSStream || element.Name.Local != "stream" {
|
|
return nil, errors.New("is not stream")
|
|
}
|
|
f := &messages.StreamFeatures{}
|
|
if err := client.ReadDecode(f); err != nil {
|
|
return nil, err
|
|
}
|
|
debug := "stream start >"
|
|
if f.StartTLS != nil {
|
|
debug += " tls"
|
|
}
|
|
debug += " mechanism("
|
|
for _, m := range f.Mechanisms.Mechanism {
|
|
debug += m
|
|
}
|
|
debug += ")"
|
|
if f.Bind != nil {
|
|
debug += " bind"
|
|
}
|
|
client.Logging.Info(debug)
|
|
return f, nil
|
|
}
|
|
|
|
func (client *Client) connect(password string) error {
|
|
if _, err := client.startStream(); err != nil {
|
|
return err
|
|
}
|
|
if err := client.Send(&messages.TLSStartTLS{}); err != nil {
|
|
return err
|
|
}
|
|
|
|
var p messages.TLSProceed
|
|
if err := client.ReadDecode(&p); err != nil {
|
|
return err
|
|
}
|
|
// Change tcp to tls
|
|
tlsconn := tls.Client(client.conn, &tls.Config{
|
|
ServerName: client.JID.Domain,
|
|
})
|
|
client.setConnection(tlsconn)
|
|
|
|
if err := tlsconn.Handshake(); err != nil {
|
|
return err
|
|
}
|
|
if err := tlsconn.VerifyHostname(client.JID.Domain); err != nil {
|
|
return err
|
|
}
|
|
if err := client.auth(password); err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err := client.startStream(); err != nil {
|
|
return err
|
|
}
|
|
// bind to resource
|
|
bind := &messages.Bind{}
|
|
if client.JID.Resource != "" {
|
|
bind.Resource = client.JID.Resource
|
|
}
|
|
if err := client.Send(&messages.IQClient{
|
|
Type: messages.IQTypeSet,
|
|
To: model.NewJID(client.JID.Domain),
|
|
Bind: bind,
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
|
|
var iq messages.IQClient
|
|
if err := client.ReadDecode(&iq); err != nil {
|
|
return err
|
|
}
|
|
if iq.Error != nil {
|
|
if iq.Error.ServiceUnavailable == nil {
|
|
return errors.New(fmt.Sprintf("recv error on iq>bind: %s[%s]: %s -> %s -> %s", iq.Error.Code, iq.Error.Type, iq.Error.Text, messages.XMLChildrenString(iq.Error.StanzaErrorGroup), messages.XMLChildrenString(iq.Error.Other)))
|
|
} else {
|
|
client.Logging.Warn("connected without setting resource with bind after auth: service-unavailable")
|
|
}
|
|
} else if iq.Bind == nil {
|
|
return errors.New("<iq> result missing <bind>")
|
|
} else if iq.Bind.JID != nil {
|
|
client.JID.Local = iq.Bind.JID.Local
|
|
client.JID.Domain = iq.Bind.JID.Domain
|
|
client.JID.Resource = iq.Bind.JID.Resource
|
|
client.Logging.Info("set resource by server bind")
|
|
} else {
|
|
return errors.New(messages.XMLChildrenString(iq.Other))
|
|
}
|
|
// set status
|
|
return client.Send(&messages.PresenceClient{Show: messages.PresenceShowXA, Status: "online"})
|
|
}
|