X-Git-Url: http://git.cielonegro.org/gitweb.cgi?a=blobdiff_plain;f=Network%2FHTTP%2FLucu%2FHttpd.hs;h=883a9a651f4999a893811670ebb114badc5e4bad;hb=874e6a4cc1229d29f1d902f36482cf0f78e30c9f;hp=cbbb517abf2d98effe2dc806d1ebef7904fc204b;hpb=ba87ca171f0a058f501655ffa6f6e845f2f800a6;p=Lucu.git

diff --git a/Network/HTTP/Lucu/Httpd.hs b/Network/HTTP/Lucu/Httpd.hs
index cbbb517..883a9a6 100644
--- a/Network/HTTP/Lucu/Httpd.hs
+++ b/Network/HTTP/Lucu/Httpd.hs
@@ -1,32 +1,34 @@
--- | The entry point of Lucu httpd.
+{-# LANGUAGE
+    CPP
+  , UnicodeSyntax
+  #-}
+-- |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           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           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.Interaction
+import Network.HTTP.Lucu.RequestReader
+import Network.HTTP.Lucu.Resource.Tree
+import Network.HTTP.Lucu.ResponseWriter
+import Network.HTTP.Lucu.SocketLike as SL
 
 -- |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:
 --
+-- > {-# LANGUAGE OverloadedStrings #-}
 -- > module Main where
 -- > import Network.HTTP.Lucu
 -- > 
@@ -34,42 +36,76 @@ import           System.Posix.Signals
 -- > 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"
--- >                              output "Hello, world!"
--- >              , resHead   = Nothing
--- >              , resPost   = Nothing
--- >              , resPut    = Nothing
--- >              , resDelete = Nothing
+-- > helloWorld = emptyResource {
+-- >                resGet
+-- >                  = Just $ do setContentType $ parseMIMEType "text/plain"
+-- >                              putChunk "Hello, world!"
 -- >              }
-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 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
-      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
+      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. [1]
+               void $ forkIO $ httpLoop p 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     ← mkInteractionQueue
+               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)