+{-# LANGUAGE
+ CPP
+ , UnicodeSyntax
+ #-}
-- |The entry point of Lucu httpd.
module Network.HTTP.Lucu.Httpd
- ( FallbackHandler
- , runHttpd
+ ( runHttpd
)
where
-
-import Control.Concurrent
-import Network
-import qualified Network.Socket as So
-import Network.HTTP.Lucu.Config
-import Network.HTTP.Lucu.Interaction
-import Network.HTTP.Lucu.RequestReader
-import Network.HTTP.Lucu.Resource.Tree
-import Network.HTTP.Lucu.ResponseWriter
-import qualified OpenSSL.Session as SSL
-import System.IO
-import System.Posix.Signals
+import Control.Concurrent
+import Control.Exception
+import Control.Monad
+import Control.Monad.Unicode
+import Data.Maybe
+import Network.BSD
+import Network.Socket
+import Network.HTTP.Lucu.Config
+import Network.HTTP.Lucu.Dispatcher
+import Network.HTTP.Lucu.Interaction
+import Network.HTTP.Lucu.RequestReader
+import Network.HTTP.Lucu.ResponseWriter
+import Network.HTTP.Lucu.SocketLike as SL
+import Prelude.Unicode
-- |This is the entry point of Lucu httpd. It listens to a socket and
--- waits for clients. Computation of 'runHttpd' never stops by itself
--- so the only way to stop it is to raise an exception in the thread
--- computing it.
---
--- Note that 'runHttpd' automatically makes SIGPIPE be ignored by
--- computing @'System.Posix.Signals.installHandler'
--- 'System.Posix.Signals.sigPIPE' 'System.Posix.Signals.Ignore'
--- 'Prelude.Nothing'@. This can hardly cause a problem but it may do.
+-- waits for clients. 'runHttpd' never stops by itself so the only way
+-- to stop it is to raise an exception in the thread running it.
--
-- Example:
--
--- > module Main where
--- > import Network.HTTP.Lucu
--- >
--- > main :: IO ()
--- > main = let config = defaultConfig
--- > resources = mkResTree [ ([], helloWorld) ]
--- > in
--- > runHttpd config resourcees []
--- >
--- > helloWorld :: ResourceDef
--- > helloWorld = ResourceDef {
--- > resUsesNativeThread = False
--- > , resIsGreedy = False
--- > , resGet
--- > = Just $ do setContentType $ read "text/plain"
--- > output "Hello, world!"
--- > , resHead = Nothing
--- > , resPost = Nothing
--- > , resPut = Nothing
--- > , resDelete = Nothing
--- > }
-runHttpd :: Config -> ResTree -> [FallbackHandler] -> IO ()
-runHttpd cnf tree fbs
- = withSocketsDo $
- do _ <- installHandler sigPIPE Ignore Nothing
-
- case cnfSSLConfig cnf of
- Nothing
- -> return ()
- Just scnf
- -> do so <- listenOn (sslServerPort scnf)
- _loopTID <- forkIO $ httpsLoop (sslContext scnf) so
- return ()
-
- httpLoop =<< listenOn (cnfServerPort cnf)
+-- @
+-- {-\# LANGUAGE OverloadedStrings \#-}
+-- {-\# LANGUAGE QuasiQuotes \#-}
+-- module Main where
+-- import qualified "Data.Collections" as C
+-- import "Network"
+-- import "Network.HTTP.Lucu"
+--
+-- main :: 'IO' ()
+-- main = let config = 'defaultConfig'
+-- tree :: 'ResourceTree'
+-- tree = C.fromList [ ([], 'nonGreedy' helloWorld) ]
+-- in
+-- 'Network.withSocketsDo' '.' 'runHttpd' config '$' 'resourceMap' tree
+--
+-- helloWorld :: 'Network.HTTP.Lucu.Resource'
+-- helloWorld = C.fromList
+-- [ ( 'Network.HTTP.Lucu.GET'
+-- , do 'Network.HTTP.Lucu.setContentType' ['Network.HTTP.Lucu.mimeType'| text/plain |]
+-- 'Network.HTTP.Lucu.putChunk' \"Hello, world!\"
+-- }
+-- @
+runHttpd ∷ HostMapper α ⇒ Config → α → IO ()
+runHttpd cnf hm
+ = do let launchers
+ = catMaybes
+ [ do addr ← cnfServerV4Addr cnf
+ return ( launchListener =≪ listenOn AF_INET addr (cnfServerPort cnf)
+ )
+ , do addr ← cnfServerV6Addr cnf
+ return ( launchListener =≪ listenOn AF_INET6 addr (cnfServerPort cnf)
+ )
+#if defined(HAVE_SSL)
+ , do scnf ← cnfSSLConfig cnf
+ addr ← cnfServerV4Addr cnf
+ return ( do so ← listenOn AF_INET addr (sslServerPort scnf)
+ launchListener (sslContext scnf, so)
+ )
+ , do scnf ← cnfSSLConfig cnf
+ addr ← cnfServerV6Addr cnf
+ return ( do so ← listenOn AF_INET6 addr (sslServerPort scnf)
+ launchListener (sslContext scnf, so)
+ )
+#endif
+ ]
+ sequence_ launchers
+ waitForever
where
- httpLoop :: Socket -> IO ()
- httpLoop so
- = do (h, addr) <- acceptHTTP so
- tQueue <- newInteractionQueue
- readerTID <- forkIO $ requestReader cnf tree fbs h addr tQueue
- _writerTID <- forkIO $ responseWriter cnf h tQueue readerTID
- httpLoop so
+ launchListener ∷ SocketLike s ⇒ s → IO ()
+ launchListener so
+ = do p ← SL.socketPort so
+ -- FIXME: Don't throw away the thread ID as we can't
+ -- kill it later then.
+ void ∘ forkIO $ httpLoop p so
- httpsLoop :: SSL.SSLContext -> Socket -> IO ()
- httpsLoop ctx so
- = do (ssl, addr) <- acceptHTTPS ctx so
- tQueue <- newInteractionQueue
- readerTID <- forkIO $ requestReader cnf tree fbs ssl addr tQueue
- _writerTID <- forkIO $ responseWriter cnf ssl tQueue readerTID
- httpsLoop ctx so
+ listenOn ∷ Family → HostName → ServiceName → IO Socket
+ listenOn fam host srv
+ = do proto ← getProtocolNumber "tcp"
+ let hints = defaultHints {
+ addrFlags = [AI_PASSIVE]
+ , addrFamily = fam
+ , addrSocketType = Stream
+ , addrProtocol = proto
+ }
+ addrs ← getAddrInfo (Just hints) (Just host) (Just srv)
+ let addr = head addrs
+ bracketOnError
+ (socket (addrFamily addr)
+ (addrSocketType addr)
+ (addrProtocol addr))
+ sClose
+ (\ sock →
+ do setSocketOption sock ReuseAddr 1
+ bindSocket sock (addrAddress addr)
+ listen sock maxListenQueue
+ return sock
+ )
- acceptHTTP :: Socket -> IO (Handle, So.SockAddr)
- acceptHTTP soSelf
- = do (soPeer, addr) <- So.accept soSelf
- hPeer <- So.socketToHandle soPeer ReadWriteMode
- return (hPeer, addr)
+ httpLoop ∷ SocketLike s ⇒ PortNumber → s → IO ()
+ httpLoop port so
+ = do (h, addr) ← SL.accept so
+ tQueue ← mkInteractionQueue
+ readerTID ← forkIO $ requestReader cnf hm h port addr tQueue
+ _writerTID ← forkIO $ responseWriter cnf h tQueue readerTID
+ httpLoop port so
- acceptHTTPS :: SSL.SSLContext -> Socket -> IO (SSL.SSL, So.SockAddr)
- acceptHTTPS ctx so
- = do (so', addr) <- So.accept so
- ssl <- SSL.connection ctx so'
- SSL.accept ssl
- return (ssl, addr)
+ waitForever ∷ IO ()
+ waitForever = forever $ threadDelay 1000000
projects / Lucu.git / blobdiff