--- | The entry point of Lucu httpd.
+-- |The entry point of Lucu httpd.
module Network.HTTP.Lucu.Httpd
- ( runHttpd
+ ( FallbackHandler
+ , runHttpd
)
where
import Control.Concurrent
-import Network
-import qualified Network.Socket as So
+import Control.Exception
+import Control.Monad
+import Network.BSD
+import Network.Socket
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 System.IO
+import Network.HTTP.Lucu.SocketLike as SL
import System.Posix.Signals
--- | This is the entry point of Lucu httpd. It listens to a socket and
+-- |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 @installHandler sigPIPE Ignore Nothing@. This can hardly
--- cause a problem but it may do.
+-- 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.
--
-- Example:
--
-- > main = let config = defaultConfig
-- > resources = mkResTree [ ([], helloWorld) ]
-- > in
--- > runHttpd config resourcees
+-- > runHttpd config resourcees []
-- >
-- > helloWorld :: ResourceDef
-- > helloWorld = ResourceDef {
-- > resUsesNativeThread = False
-- > , resIsGreedy = False
-- > , resGet
--- > = Just $ do setContentType $ "text" </> "plain"
+-- > = Just $ do setContentType $ read "text/plain"
-- > output "Hello, world!"
-- > , resHead = Nothing
-- > , resPost = Nothing
-- > , resPut = Nothing
-- > , resDelete = Nothing
-- > }
-runHttpd :: Config -> ResTree -> IO ()
-runHttpd cnf tree
- = cnf `seq` tree `seq`
- withSocketsDo $
- do installHandler sigPIPE Ignore Nothing
- so <- listenOn (cnfServerPort cnf)
- loop so
+runHttpd :: Config -> ResTree -> [FallbackHandler] -> IO ()
+runHttpd cnf tree fbs
+ = withSocketsDo $
+ do _ <- installHandler sigPIPE Ignore Nothing
+
+ -- FIXME: TERRIBLE CODE. NEED MAJOR REWRITE.
+ case cnfSSLConfig cnf of
+ Nothing
+ -> return ()
+ Just scnf
+ -> do case cnfServerV4Addr cnf of
+ Nothing
+ -> return ()
+ Just v4addr
+ -> do so <- listenOn AF_INET v4addr (sslServerPort scnf)
+ p <- socketPort so
+ -- FIXME: Don't throw away the thread
+ -- ID as we can't kill it later
+ -- then. [1]
+ _ <- forkIO $ httpLoop p (sslContext scnf, so)
+ return ()
+
+ case cnfServerV6Addr cnf of
+ Nothing
+ -> return ()
+ Just v6addr
+ -> do so <- listenOn AF_INET6 v6addr (sslServerPort scnf)
+ p <- socketPort so
+ -- FIXME: [1]
+ _ <- forkIO $ httpLoop p (sslContext scnf, so)
+ return ()
+
+ case cnfServerV4Addr cnf of
+ Nothing
+ -> return ()
+ Just v4addr
+ -> do so <- listenOn AF_INET v4addr (cnfServerPort cnf)
+ p <- socketPort so
+ -- FIXME: [1]
+ _ <- forkIO $ httpLoop p so
+ return ()
+
+ case cnfServerV6Addr cnf of
+ Nothing
+ -> return ()
+ Just v6addr
+ -> do so <- listenOn AF_INET6 v6addr (cnfServerPort cnf)
+ p <- socketPort so
+ -- FIXME: [1]
+ _ <- forkIO $ httpLoop p so
+ return ()
+
+ waitForever
where
- loop :: Socket -> IO ()
- loop so
- -- 本當は Network.accept を使ひたいが、このアクションは勝手に
- -- リモートのIPを逆引きするので、使へない。
- = so `seq`
- do (h, addr) <- accept' so
- tQueue <- newInteractionQueue
- readerTID <- forkIO $ requestReader cnf tree h addr tQueue
- writerTID <- forkIO $ responseWriter cnf h tQueue readerTID
- loop 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
+ )
+
+ httpLoop :: SocketLike s => PortNumber -> s -> IO ()
+ httpLoop port so
+ = do (h, addr) <- SL.accept so
+ tQueue <- newInteractionQueue
+ readerTID <- forkIO $ requestReader cnf tree fbs h port addr tQueue
+ _writerTID <- forkIO $ responseWriter cnf h tQueue readerTID
+ httpLoop port so
- accept' :: Socket -> IO (Handle, So.SockAddr)
- accept' soSelf
- = soSelf `seq`
- do (soPeer, addr) <- So.accept soSelf
- hPeer <- So.socketToHandle soPeer ReadWriteMode
- return (hPeer, addr)
+ waitForever :: IO ()
+ waitForever = forever (threadDelay 1000000)
projects / Lucu.git / blobdiff