c09d833 · typeable/wai

55 ( accept, SockAddr

56 )

57import qualified Network.Socket.ByteString as Sock

58−import Control.Exception (bracket, finally, Exception, SomeException, catch)

58+import Control.Exception (bracket, finally, Exception, SomeException, catch, throwIO)

59import Control.Concurrent (forkIO)

60import Data.Maybe (fromMaybe)

61

170 -> [ByteString]

171 -> SockAddr

172 -> E.Iteratee S.ByteString IO (E.Enumeratee S.ByteString S.ByteString IO a, Request)

173−parseRequest' _ [] _ = E.throwError $ NotEnoughLines []

173+parseRequest' _ [] _ = throwError $ NotEnoughLines []

174parseRequest' port (firstLine:otherLines) remoteHost' = do

175 (method, rpath', gets, httpversion) <- parseFirst firstLine

176 let rpath =

217 (method, query, http') <-

218 case pieces of

219 [x, y, z] -> return (x, y, z)

220− _ -> E.throwError $ BadFirstLine $ B.unpack s

220+ _ -> throwError $ BadFirstLine $ B.unpack s

221 let (hfirst, hsecond) = B.splitAt 5 http'

222 if (hfirst == "HTTP/")

223 then

224 let (rpath, qstring) = B.break (== '?') query

225 in return (method, rpath, qstring, hsecond)

226− else E.throwError NonHttp

226+ else throwError NonHttp

227{-# INLINE parseFirst #-} -- FIXME is this inline necessary? the function is only called from one place and not exported

228

229httpBuilder, spaceBuilder, newlineBuilder, transferEncodingBuilder

341#else

342 mbs <- liftIO $ timeout readTimeout $ Sock.recv socket len

343 case mbs of

344− Nothing -> E.throwError SocketTimeout

344+ Nothing -> throwError SocketTimeout

345 Just bs -> go bs

346#endif

347 where

360 -> ([ByteString] -> [ByteString])

361 -> E.Iteratee S.ByteString IO [ByteString]

362takeUntilBlank count _

363− | count > maxHeaders = E.throwError TooManyHeaders

363+ | count > maxHeaders = throwError TooManyHeaders

364takeUntilBlank count front = do

365 l <- takeLineMax 0 id

366 if B.null l

373takeLineMax len front = do

374 mbs <- EL.head

375 case mbs of

376− Nothing -> E.throwError IncompleteHeaders

376+ Nothing -> throwError IncompleteHeaders

377 Just bs -> do

378 let (x, y) = S.breakByte 10 bs

379 x' = if S.length x > 0 && S.last x == 13

382 let len' = len + B.length x

383 case () of

384 ()

385− | len' > maxHeaderLength -> E.throwError OverLargeHeader

385+ | len' > maxHeaderLength -> throwError OverLargeHeader

386 | B.null y -> takeLineMax len' $ front . (:) x

387 | otherwise -> do

388 E.yield () $ E.Chunks [B.drop 1 y]

389 return $ B.concat $ front [x']

390+

391+throwError = liftIO . throwIO