summaryrefslogtreecommitdiff
path: root/xmpp/stream_pair.go
diff options
context:
space:
mode:
Diffstat (limited to 'xmpp/stream_pair.go')
-rw-r--r--xmpp/stream_pair.go82
1 files changed, 82 insertions, 0 deletions
diff --git a/xmpp/stream_pair.go b/xmpp/stream_pair.go
new file mode 100644
index 0000000..e02cca6
--- /dev/null
+++ b/xmpp/stream_pair.go
@@ -0,0 +1,82 @@
+package xmpp
+
+import (
+ "encoding/xml"
+ "log"
+)
+
+func runStreamPair(s *session) {
+ openStream(s)
+ defer closeStream(s)
+
+ for {
+ select {
+ case data := <-s.in:
+ switch data.(type) {
+ case SessionShouldDisconnect:
+ return
+ default:
+ log.Printf("Unknown data '%d'!\n", data)
+ }
+ case _ = <-s.rx:
+ // TODO route received XML token here
+ }
+ }
+}
+
+func openStream(s *session) {
+ start := xml.StartElement{
+ xml.Name{"jabber:client", "stream:stream"},
+ []xml.Attr{
+ xml.Attr{xml.Name{"", "from"}, s.jid},
+ xml.Attr{xml.Name{"", "to"}, domainpart(s.jid)},
+ xml.Attr{xml.Name{"", "version"}, "1.0"},
+ xml.Attr{xml.Name{"", "xml:lang"}, "en"},
+ xml.Attr{xml.Name{"", "xmlns:stream"}, "http://etherx.jabber.org/streams"},
+ },
+ }
+
+ s.streamEnd = start.End()
+
+ err := s.ed.encodeToken(start)
+ if err != nil {
+ log.Println("Could not encode stream start!")
+ }
+
+ syncStreams(s)
+}
+
+// syncStreams drops XML tokens from the receiving stream until an
+// xml.StartElement with the local name `stream` is received. If this function
+// is called after opening a new stream in the sending direction it is ensured
+// that both streams directions work on the same stream level and are in sync.
+// Tokens received which are not a stream StartElement are not handled but
+// logged since this should not happen.
+func syncStreams(s *session) {
+ for {
+ select {
+ case data := <-s.in:
+ switch data.(type) {
+ case SessionShouldDisconnect:
+ return
+ default:
+ log.Printf("Unhandled data '%d' during stream sync!\n", data)
+ }
+ case t := <-s.rx:
+ switch token := t.(type) {
+ case xml.StartElement:
+ if token.Name.Local == "stream" {
+ return
+ }
+ }
+ log.Printf("Unhandled XML token '%v' during stream sync!\n", t)
+ }
+ }
+}
+
+func closeStream(s *session) {
+ err := s.ed.encodeToken(s.streamEnd)
+ if err != nil {
+ log.Println("Could not encode stream end!")
+ }
+}