diff --git a/cert_checker.go b/cert_checker.go index 30a265a..629720f 100644 --- a/cert_checker.go +++ b/cert_checker.go @@ -65,7 +65,7 @@ func (c *ServerCheck) Check() error { var f stanza.StreamFeatures packet, err := stanza.NextPacket(decoder) if err != nil { - err = fmt.Errorf("stream open decode features: %s", err) + err = fmt.Errorf("stream open decode features: %w", err) return err } @@ -86,7 +86,7 @@ func (c *ServerCheck) Check() error { var k stanza.TLSProceed if err = decoder.DecodeElement(&k, nil); err != nil { - return fmt.Errorf("expecting starttls proceed: %s", err) + return fmt.Errorf("expecting starttls proceed: %w", err) } var tlsConfig tls.Config diff --git a/stanza/message.go b/stanza/message.go index 35e44c1..a6364b0 100644 --- a/stanza/message.go +++ b/stanza/message.go @@ -8,16 +8,22 @@ import ( // ============================================================================ // Message Packet +// LocalizedString is a string node with a language attribute. +type LocalizedString struct { + Content string `xml:",chardata"` + Lang string `xml:"http://www.w3.org/XML/1998/namespace lang,attr,omitempty"` +} + // Message implements RFC 6120 - A.5 Client Namespace (a part) type Message struct { XMLName xml.Name `xml:"message"` Attrs - Subject string `xml:"subject,omitempty"` - Body string `xml:"body,omitempty"` - Thread string `xml:"thread,omitempty"` - Error Err `xml:"error,omitempty"` - Extensions []MsgExtension `xml:",omitempty"` + Subject []LocalizedString `xml:"subject,omitempty"` + Body []LocalizedString `xml:"body,omitempty"` + Thread string `xml:"thread,omitempty"` + Error Err `xml:"error,omitempty"` + Extensions []MsgExtension `xml:",omitempty"` } func (Message) Name() string { diff --git a/stanza/packet.go b/stanza/packet.go index 567796d..00f76ba 100644 --- a/stanza/packet.go +++ b/stanza/packet.go @@ -10,7 +10,7 @@ type Attrs struct { Id string `xml:"id,attr,omitempty"` From string `xml:"from,attr,omitempty"` To string `xml:"to,attr,omitempty"` - Lang string `xml:"lang,attr,omitempty"` + Lang string `xml:"http://www.w3.org/XML/1998/namespace lang,attr,omitempty"` } type packetFormatter interface { diff --git a/stanza/parser.go b/stanza/parser.go index b5f11cf..9017c97 100644 --- a/stanza/parser.go +++ b/stanza/parser.go @@ -91,7 +91,7 @@ func NextXmppToken(p *xml.Decoder) (xml.Token, error) { return xml.StartElement{}, errors.New("connection closed") } if err != nil { - return xml.StartElement{}, fmt.Errorf("NextStart %s", err) + return xml.StartElement{}, fmt.Errorf("NextStart: %w", err) } switch t := t.(type) { case xml.StartElement: @@ -112,7 +112,7 @@ func NextStart(p *xml.Decoder) (xml.StartElement, error) { return xml.StartElement{}, errors.New("connection closed") } if err != nil { - return xml.StartElement{}, fmt.Errorf("NextStart %s", err) + return xml.StartElement{}, fmt.Errorf("NextStart: %w", err) } switch t := t.(type) { case xml.StartElement: diff --git a/stream_manager.go b/stream_manager.go index 7bfb42c..c09f183 100644 --- a/stream_manager.go +++ b/stream_manager.go @@ -113,22 +113,27 @@ func (sm *StreamManager) Stop() { } func (sm *StreamManager) connect() error { - if sm.client != nil { - if c, ok := sm.client.(*Client); ok { - if c.CurrentState.getState() == StateDisconnected { - sm.Metrics = initMetrics() - err := c.Connect() - if err != nil { - return err - } - if sm.PostConnect != nil { - sm.PostConnect(sm.client) - } - return nil - } + if sm.client == nil { + return errors.New("client is not set") + } + + if c, ok := sm.client.(*Client); ok { + if c.CurrentState.getState() != StateDisconnected { + return errors.New("client is not disconnected") } } - return errors.New("client is not disconnected") + + sm.Metrics = initMetrics() + + if err := sm.client.Connect(); err != nil { + return err + } + + if sm.PostConnect != nil { + sm.PostConnect(sm.client) + } + + return nil } // resume manages the reconnection loop and apply the define backoff to avoid overloading the server. diff --git a/tcp_server_mock.go b/tcp_server_mock.go index d189e3a..ea726f9 100644 --- a/tcp_server_mock.go +++ b/tcp_server_mock.go @@ -155,7 +155,7 @@ func respondToIQ(t *testing.T, sc *ServerConn) { mResp, err := xml.Marshal(iqResp) _, err = fmt.Fprintln(sc.connection, string(mResp)) if err != nil { - t.Errorf("Could not send response stanza : %s", err) + t.Errorf("Could not send response stanza : %w", err) } return } @@ -175,15 +175,15 @@ func discardPresence(t *testing.T, sc *ServerConn) { if err != nil { if netErr, ok := err.(net.Error); ok && netErr.Timeout() { - t.Errorf("read timeout: %s", err) + t.Errorf("read timeout: %w", err) } else { - t.Errorf("read error: %s", err) + t.Errorf("read error: %w", err) } } err = xml.Unmarshal(recvBuf, &presenceStz) if err != nil { - t.Errorf("Expected presence but this happened : %s", err.Error()) + t.Errorf("Expected presence but this happened : %w", err) } } @@ -223,7 +223,7 @@ func sendStreamFeatures(t *testing.T, sc *ServerConn) { </mechanisms> </stream:features>` if _, err := fmt.Fprintln(sc.connection, features); err != nil { - t.Errorf("cannot send stream feature: %s", err) + t.Errorf("cannot send stream feature: %w", err) } } @@ -231,7 +231,7 @@ func sendStreamFeatures(t *testing.T, sc *ServerConn) { func readAuth(t *testing.T, decoder *xml.Decoder) string { se, err := stanza.NextStart(decoder) if err != nil { - t.Errorf("cannot read auth: %s", err) + t.Errorf("cannot read auth: %w", err) return "" } @@ -239,7 +239,7 @@ func readAuth(t *testing.T, decoder *xml.Decoder) string { nv = &stanza.SASLAuth{} // Decode element into pointer storage if err = decoder.DecodeElement(nv, &se); err != nil { - t.Errorf("cannot decode auth: %s", err) + t.Errorf("cannot decode auth: %w", err) return "" } @@ -256,7 +256,7 @@ func sendBindFeature(t *testing.T, sc *ServerConn) { <bind xmlns='urn:ietf:params:xml:ns:xmpp-bind'/> </stream:features>` if _, err := fmt.Fprintln(sc.connection, features); err != nil { - t.Errorf("cannot send stream feature: %s", err) + t.Errorf("cannot send stream feature: %w", err) } } @@ -267,21 +267,21 @@ func sendRFC3921Feature(t *testing.T, sc *ServerConn) { <session xmlns='urn:ietf:params:xml:ns:xmpp-session'/> </stream:features>` if _, err := fmt.Fprintln(sc.connection, features); err != nil { - t.Errorf("cannot send stream feature: %s", err) + t.Errorf("cannot send stream feature: %w", err) } } func bind(t *testing.T, sc *ServerConn) { se, err := stanza.NextStart(sc.decoder) if err != nil { - t.Errorf("cannot read bind: %s", err) + t.Errorf("cannot read bind: %w", err) return } iq := &stanza.IQ{} // Decode element into pointer storage if err = sc.decoder.DecodeElement(&iq, &se); err != nil { - t.Errorf("cannot decode bind iq: %s", err) + t.Errorf("cannot decode bind iq: %w", err) return } @@ -300,14 +300,14 @@ func bind(t *testing.T, sc *ServerConn) { func session(t *testing.T, sc *ServerConn) { se, err := stanza.NextStart(sc.decoder) if err != nil { - t.Errorf("cannot read session: %s", err) + t.Errorf("cannot read session: %w", err) return } iq := &stanza.IQ{} // Decode element into pointer storage if err = sc.decoder.DecodeElement(&iq, &se); err != nil { - t.Errorf("cannot decode session iq: %s", err) + t.Errorf("cannot decode session iq: %w", err) return } diff --git a/websocket_transport.go b/websocket_transport.go index 7631fc8..19cf95f 100644 --- a/websocket_transport.go +++ b/websocket_transport.go @@ -168,7 +168,7 @@ func (t *WebsocketTransport) cleanup(code websocket.StatusCode) error { t.queue = nil } if t.wsConn != nil { - err = t.wsConn.Close(websocket.StatusGoingAway, "Done") + err = t.wsConn.Close(code, "Done") t.wsConn = nil } if t.closeFunc != nil {