Honor cnfServerV4Addr and cnfServerV6Addr.

[Lucu.git] / Network / HTTP / Lucu / Httpd.hs

diff --git a/Network/HTTP/Lucu/Httpd.hs b/Network/HTTP/Lucu/Httpd.hs
index 1d0f2b8a5c626d3aa8175c7d07927b88fdeff1cc..d9f28f110e5f1c668a3fc9f539b9f2c4981e3959 100644 (file)
--- a/Network/HTTP/Lucu/Httpd.hs
+++ b/Network/HTTP/Lucu/Httpd.hs
@@ -6,13 +6,16 @@ module Network.HTTP.Lucu.Httpd
     where
 
 import           Control.Concurrent
-import           Network hiding (accept)
+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           Network.HTTP.Lucu.SocketLike
+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
@@ -53,20 +56,83 @@ 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 so       <- listenOn (sslServerPort scnf)
-                     _loopTID <- forkIO $ httpLoop (sslContext scnf, so)
-                     return ()
+               -> 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 ()
          
-         httpLoop =<< listenOn (cnfServerPort cnf)
+         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
-      httpLoop :: SocketLike s => s -> IO ()
-      httpLoop so
-          = do (h, addr)  <- accept 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 addr tQueue
+               readerTID  <- forkIO $ requestReader cnf tree fbs h port addr tQueue
                _writerTID <- forkIO $ responseWriter cnf h tQueue readerTID
-               httpLoop so
+               httpLoop port so
+
+      waitForever :: IO ()
+      waitForever = forever (threadDelay 1000000)