uhh whats a submodule i have never heard of that bro

go-xmpp/xmpp_transport.go

@@ -0,0 +1,163 @@
+package xmpp
+
+import (
+	"bufio"
+	"crypto/tls"
+	"encoding/xml"
+	"errors"
+	"fmt"
+	"io"
+	"net"
+	"time"
+
+	"gosrc.io/xmpp/stanza"
+)
+
+// XMPPTransport implements the XMPP native TCP transport
+// The decoder is expected to be initialized after connecting to a server.
+type XMPPTransport struct {
+	openStatement string
+	Config        TransportConfiguration
+	TLSConfig     *tls.Config
+	decoder       *xml.Decoder
+	conn          net.Conn
+	readWriter    io.ReadWriter
+	logFile       io.Writer
+	isSecure      bool
+	// Used to close TCP connection when a stream close message is received from the server
+	closeChan chan stanza.StreamClosePacket
+}
+
+var componentStreamOpen = fmt.Sprintf("<?xml version='1.0'?><stream:stream to='%%s' xmlns='%s' xmlns:stream='%s'>", stanza.NSComponent, stanza.NSStream)
+
+var clientStreamOpen = fmt.Sprintf("<?xml version='1.0'?><stream:stream to='%%s' xmlns='%s' xmlns:stream='%s' version='1.0'>", stanza.NSClient, stanza.NSStream)
+
+func (t *XMPPTransport) Connect() (string, error) {
+	var err error
+
+	// Since we're starting a new connection, reset the encryption status
+	t.isSecure = false
+
+	t.conn, err = net.DialTimeout("tcp", t.Config.Address, time.Duration(t.Config.ConnectTimeout)*time.Second)
+	if err != nil {
+		return "", NewConnError(err, true)
+	}
+
+	t.closeChan = make(chan stanza.StreamClosePacket, 1)
+	t.readWriter = newStreamLogger(t.conn, t.logFile)
+	t.decoder = xml.NewDecoder(bufio.NewReaderSize(t.readWriter, maxPacketSize))
+	t.decoder.CharsetReader = t.Config.CharsetReader
+	return t.StartStream()
+}
+
+func (t *XMPPTransport) StartStream() (string, error) {
+	if _, err := fmt.Fprintf(t, t.openStatement, t.Config.Domain); err != nil {
+		t.Close()
+		return "", NewConnError(err, true)
+	}
+
+	sessionID, err := stanza.InitStream(t.GetDecoder())
+	if err != nil {
+		t.Close()
+		return "", NewConnError(err, false)
+	}
+	return sessionID, nil
+}
+
+func (t *XMPPTransport) DoesStartTLS() bool {
+	return true
+}
+
+func (t *XMPPTransport) GetDomain() string {
+	return t.Config.Domain
+}
+
+func (t *XMPPTransport) GetDecoder() *xml.Decoder {
+	return t.decoder
+}
+
+func (t *XMPPTransport) IsSecure() bool {
+	return t.isSecure
+}
+
+func (t *XMPPTransport) StartTLS() error {
+	if t.Config.TLSConfig == nil {
+		t.TLSConfig = &tls.Config{}
+	} else {
+		t.TLSConfig = t.Config.TLSConfig.Clone()
+	}
+
+	if t.TLSConfig.ServerName == "" {
+		t.TLSConfig.ServerName = t.Config.Domain
+	}
+	tlsConn := tls.Client(t.conn, t.TLSConfig)
+	// We convert existing connection to TLS
+	if err := tlsConn.Handshake(); err != nil {
+		return err
+	}
+
+	t.isSecure = false
+	t.conn = tlsConn
+	t.readWriter = newStreamLogger(tlsConn, t.logFile)
+	t.decoder = xml.NewDecoder(bufio.NewReaderSize(t.readWriter, maxPacketSize))
+	t.decoder.CharsetReader = t.Config.CharsetReader
+
+	if !t.TLSConfig.InsecureSkipVerify {
+		if err := tlsConn.VerifyHostname(t.Config.Domain); err != nil {
+			return err
+		}
+	}
+
+	t.isSecure = true
+	return nil
+}
+
+func (t *XMPPTransport) Ping() error {
+	n, err := t.conn.Write([]byte("\n"))
+	if err != nil {
+		return err
+	}
+	if n != 1 {
+		return errors.New("could not write ping")
+	}
+	return nil
+}
+
+func (t *XMPPTransport) Read(p []byte) (n int, err error) {
+	if t.readWriter == nil {
+		return 0, errors.New("cannot read: not connected, no readwriter")
+	}
+	return t.readWriter.Read(p)
+}
+
+func (t *XMPPTransport) Write(p []byte) (n int, err error) {
+	if t.readWriter == nil {
+		return 0, errors.New("cannot write: not connected, no readwriter")
+	}
+	return t.readWriter.Write(p)
+}
+
+func (t *XMPPTransport) Close() error {
+	if t.readWriter != nil {
+		_, _ = t.readWriter.Write([]byte(stanza.StreamClose))
+	}
+
+	// Try to wait for the stream close tag from the server. After a timeout, disconnect anyway.
+	select {
+	case <-t.closeChan:
+	case <-time.After(time.Duration(t.Config.ConnectTimeout) * time.Second):
+	}
+
+	if t.conn != nil {
+		return t.conn.Close()
+	}
+	return nil
+}
+
+func (t *XMPPTransport) LogTraffic(logFile io.Writer) {
+	t.logFile = logFile
+}
+
+func (t *XMPPTransport) ReceivedStreamClose() {
+	t.closeChan <- stanza.StreamClosePacket{}
+}