import Control.Exception
import Control.Monad
import qualified Data.ByteString.Lazy as B
+import Data.Char
import Data.Dynamic
import qualified Data.Map as M
import Data.Time.Clock.POSIX
encodings :: M.Map String String
encodings = M.fromList [
+ (".bz2", "bzip2"),
(".gz", "gzip"),
- (".bz2", "bzip2")
+ (".z", "compress")
]
types :: M.Map String String
types = M.fromList [
- (".html", "text/html")
+ (".avi", "video/x-msvideo"),
+ (".css", "text/css"),
+ (".doc", "application/msword"),
+ (".gif", "image/gif"),
+ (".htm", "text/html"),
+ (".html", "text/html"),
+ (".ico", "image/vnd.microsoft.icon"),
+ (".il", "application/octet-stream"),
+ (".jar", "application/java-archive"),
+ (".jpeg", "image/jpeg"),
+ (".jpg", "image/jpeg"),
+ (".js", "application/x-javascript"),
+ (".mid", "audio/midi"),
+ (".midi", "audio/midi"),
+ (".mov", "video/quicktime"),
+ (".mp3", "audio/mpeg"),
+ (".mpeg", "video/mpeg"),
+ (".mpg", "video/mpeg"),
+ (".pdf", "application/pdf"),
+ (".png", "image/png"),
+ (".ppt", "application/vnd.ms-powerpoint"),
+ (".ps", "application/postscript"),
+ (".svg", "image/svg+xml"),
+ (".swf", "application/x-shockwave-flash"),
+ (".tar", "application/x-tar"),
+ (".tgz", "application/x-gzip"),
+ (".tif", "image/tiff"),
+ (".tiff", "image/tiff"),
+ (".wav", "audio/x-wav"),
+ (".wmv", "video/x-ms-wmv"),
+ (".xaml", "application/xaml+xml"),
+ (".xap", "application/x-silverlight-app"),
+ (".xhtml", "application/xhtml+xml"),
+ (".xls", "application/vnd.ms-excel"),
+ (".xml", "text/xml"),
+ (".xsl", "text/xml"),
+ (".zip", "application/zip")
]
throwExceptionCGI :: Exception e => e -> CGI a
checkExtension :: FilePath -> CGI ()
checkExtension file = do
let (base, ext) = splitExtension file
- ext' <- case M.lookup ext encodings of
+ ext' <- case M.lookup (map toLower ext) encodings of
Nothing -> return ext
Just e -> do
setHeader "Content-Encoding" e
return $ takeExtension base
- case M.lookup ext' types of
+ case M.lookup (map toLower ext') types of
Nothing -> throwExceptionCGI Forbidden
Just t -> setHeader "Content-Type" t