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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
|
{-# LANGUAGE OverloadedStrings #-}
import Control.Monad.Trans.Resource (runResourceT)
import Control.Monad.Trans (lift)
import Control.Monad.IO.Class (liftIO)
import Control.Monad.Fix (fix)
import Control.Monad
import Control.Concurrent (forkIO)
import Control.Concurrent.STM
-- import Control.Concurrent.STM.TChan
import Network.Socket
import XMPPTypes (withPort)
import Text.Printf
import System.Posix.Signals
import Data.ByteString (ByteString)
import qualified Data.ByteString.Char8 as Strict8
-- import qualified Data.ByteString.Lazy.Char8 as Lazy8
import Data.Conduit
import qualified Data.Conduit.List as CL
import qualified Data.Conduit.Binary as CB
import qualified Text.XML.Stream.Render as XML
import qualified Text.XML.Stream.Parse as XML
import Data.XML.Types as XML
import Data.Maybe (catMaybes)
import Data.Monoid ( (<>) )
import qualified Control.Concurrent.STM.UpdateStream as Slotted
import ControlMaybe
import NestingXML
import Server
wlog s = putStrLn s
where _ = s :: String
wlogb s = Strict8.putStrLn s
control sv = atomically . putTMVar (serverCommand sv)
-- Note: This function ignores name space qualification
elementAttrs expected (EventBeginElement name attrs)
| nameLocalName name==expected
= return attrs
elementAttrs _ _ = mzero
getStreamName (EventBeginElement name _) = name
xmlStream :: ReadCommand -> WriteCommand -> ( Source IO XML.Event
, Sink XML.Event IO () )
xmlStream conread conwrite = (xsrc,xsnk)
where
xsrc = src $= XML.parseBytes XML.def
xsnk = XML.renderBytes XML.def =$ snk
src = do
v <- lift conread
maybe (return ()) -- lift . wlog $ "conread: Nothing")
(\v -> yield v >> src)
v
snk = awaitForever $ liftIO . conwrite
type FlagCommand = STM Bool
type ReadCommand = IO (Maybe ByteString)
type WriteCommand = ByteString -> IO Bool
data Stanza
= UnrecognizedStanza { stanzaChan :: TChan (Maybe XML.Event) }
prettyPrint prefix xs =
liftIO $
CL.sourceList xs
$= XML.renderBytes (XML.def { XML.rsPretty=True })
=$= CB.lines
$$ CL.mapM_ (wlogb . (prefix <>))
xmppInbound :: ConnectionKey -> FlagCommand
-> Source IO XML.Event
-> TChan Stanza
-> Sink XML.Event IO ()
xmppInbound k pingflag src stanzas = doNestingXML $ do
withXML $ \begindoc -> do
when (begindoc==EventBeginDocument) $ do
-- liftIO . wlog $ "begin-doc"
withXML $ \xml -> do
withJust (elementAttrs "stream" xml) $ \stream_attrs -> do
-- liftIO . wlog $ "stream: " ++ show (getStreamName xml)
-- liftIO . wlog $ "stream atributes: " ++ show stream_attrs
fix $ \loop -> do
-- liftIO . wlog $ "waiting for stanza."
chan <- liftIO $ atomically newTChan
whenJust nextElement $ \stanza -> do
stanza_lvl <- nesting
liftIO . atomically $ writeTChan chan (Just stanza)
-- liftIO . wlog $ "stanza: "++show stanza
liftIO . atomically $ writeTChan stanzas $
UnrecognizedStanza chan
doUntilCloser stanza_lvl $ \xml -> do
liftIO . atomically $ writeTChan chan (Just xml)
-- liftIO . wlog $ "-stanza: " ++ show xml
liftIO . atomically $ writeTChan chan Nothing
loop
chanContents :: TChan x -> IO [x]
chanContents ch = do
x <- atomically $ do
bempty <- isEmptyTChan ch
if bempty
then return Nothing
else fmap Just $ readTChan ch
maybe (return [])
(\x -> do
xs <- chanContents ch
return (x:xs))
x
isEventBeginElement (EventBeginElement {}) = True
isEventBeginElement _ = False
isEventEndElement (EventEndElement {}) = True
isEventEndElement _ = False
forkConnection :: ConnectionKey
-> FlagCommand
-> Source IO XML.Event
-> Sink XML.Event IO ()
-> TChan Stanza
-> IO (TChan Stanza)
forkConnection k pingflag src snk stanzas = do
rdone <- atomically newEmptyTMVar
forkIO $ do
-- src $$ awaitForever (lift . putStrLn . takeWhile (/=' ') . show)
src $$ xmppInbound k pingflag src stanzas
atomically $ putTMVar rdone ()
wlog $ "end reader fork: " ++ show k
slots <- atomically $ Slotted.new isEventBeginElement isEventEndElement
let _ = slots :: Slotted.UpdateStream () XML.Event
let slot_src = do
what <- lift . atomically $ orElse
(Slotted.pull slots >>= \x -> return $ do
yield x
slot_src)
(readTMVar rdone >> return (return ()))
what
forkIO $ do slot_src $$ snk
wlog $ "end post-queue fork: " ++ show k
output <- atomically newTChan
forkIO $ do
fix $ \loop -> do
what <- atomically $ foldr1 orElse
[readTChan output >>= \stanza -> return $ do
let xchan = stanzaChan stanza
fix $ \inner -> do
what <- atomically $ orElse
(readTChan xchan >>= \mxml -> return $ do
case mxml of
Just xml -> do
atomically $ Slotted.push slots Nothing xml
inner
Nothing -> return ())
(readTMVar rdone >> return (return ()))
what
loop
,do pingflag >>= check
return $ do
wlog $ "TODO: send ping"
loop
,readTMVar rdone >> return (return ())
]
what
wlog $ "end pre-queue fork: " ++ show k
return output
monitor sv params = do
chan <- return $ serverEvent sv
stanzas <- atomically newTChan
fix $ \loop -> do
action <- atomically $ foldr1 orElse
[ readTChan chan >>= \(k,e) -> return $ do
case e of
Connection pingflag conread conwrite -> do
let (xsrc,xsnk) = xmlStream conread conwrite
forkConnection k pingflag xsrc xsnk stanzas
wlog $ tomsg k "Connection"
EOF -> wlog $ tomsg k "EOF"
HalfConnection In -> do
wlog $ tomsg k "ReadOnly"
control sv (Connect (callBackAddress k) params)
HalfConnection Out -> wlog $ tomsg k "WriteOnly"
RequiresPing -> wlog $ tomsg k "RequiresPing"
_ -> return ()
, readTChan stanzas >>= \stanza -> return $ do
xs <- chanContents (stanzaChan stanza)
wlog ""
prettyPrint "STANZA: " (catMaybes xs)
]
action
loop
where
tomsg k str = printf "%12s %s" str (show k)
where
_ = str :: String
data ConnectionKey
= PeerKey { callBackAddress :: SockAddr }
| ClientKey { localAddress :: SockAddr }
deriving (Show, Ord, Eq)
peerKey (sock,addr) = do
peer <- getPeerName sock
return $ PeerKey (peer `withPort` fromIntegral peerport)
clientKey (sock,addr) = return $ ClientKey addr
peerport = 5269
clientport = 5222
main = runResourceT $ do
sv <- server
lift $ do
peer_params <- return (connectionDefaults peerKey)
{ pingInterval = 5000
, timeout = 5000
, duplex = False }
client_params <- return $ connectionDefaults clientKey
forkIO $ monitor sv peer_params
control sv (Listen peerport peer_params)
-- control sv (Listen clientport client_params)
-- atomically $ newEmptyTMVar >>= readTMVar -- Wait for control-c
quitVar <- newEmptyTMVarIO
installHandler sigTERM (CatchOnce (atomically $ putTMVar quitVar True)) Nothing
installHandler sigINT (CatchOnce (atomically $ putTMVar quitVar True)) Nothing
quitMessage <- atomically $ takeTMVar quitVar
wlog "goodbye."
return ()
|