1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
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!")
}
}
|
