{-# LANGUAGE MultiWayIf #-}
{-# LANGUAGE TypeApplications #-}
{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE RankNTypes #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# OPTIONS_HADDOCK hide #-}
module Codec.Archive.Tar.Pack (
pack,
packAndCheck,
packFileEntry,
packDirectoryEntry,
packSymlinkEntry,
longLinkEntry,
getDirectoryContentsRecursive,
) where
import Codec.Archive.Tar.LongNames
import Codec.Archive.Tar.Types
import Control.Monad (join, when, forM, (>=>))
import qualified Data.ByteString as B
import qualified Data.ByteString.Lazy as BL
import Data.Foldable
import System.FilePath
( (</>) )
import qualified System.FilePath as FilePath.Native
( addTrailingPathSeparator, hasTrailingPathSeparator, splitDirectories )
import System.Directory
( listDirectory, doesDirectoryExist, getModificationTime
, pathIsSymbolicLink, getSymbolicLinkTarget
, Permissions(..), getPermissions, getFileSize )
import Data.Time.Clock
( UTCTime )
import Data.Time.Clock.POSIX
( utcTimeToPOSIXSeconds )
import System.IO
( IOMode(ReadMode), openBinaryFile, hFileSize )
import System.IO.Unsafe (unsafeInterleaveIO)
import Control.Exception (throwIO, SomeException)
import Codec.Archive.Tar.Check.Internal (checkEntrySecurity)
pack
:: FilePath
-> [FilePath]
-> IO [Entry]
pack :: FilePath -> [FilePath] -> IO [Entry]
pack = (GenEntry FilePath FilePath -> Maybe SomeException)
-> FilePath -> [FilePath] -> IO [Entry]
packAndCheck (Maybe SomeException
-> GenEntry FilePath FilePath -> Maybe SomeException
forall a b. a -> b -> a
const Maybe SomeException
forall a. Maybe a
Nothing)
packAndCheck
:: (GenEntry FilePath FilePath -> Maybe SomeException)
-> FilePath
-> [FilePath]
-> IO [Entry]
packAndCheck :: (GenEntry FilePath FilePath -> Maybe SomeException)
-> FilePath -> [FilePath] -> IO [Entry]
packAndCheck GenEntry FilePath FilePath -> Maybe SomeException
secCB FilePath
baseDir [FilePath]
relpaths = do
paths <- FilePath -> [FilePath] -> IO [FilePath]
preparePaths FilePath
baseDir [FilePath]
relpaths
entries <- packPaths baseDir paths
traverse_ (maybe (pure ()) throwIO . secCB) entries
pure $ concatMap encodeLongNames entries
preparePaths :: FilePath -> [FilePath] -> IO [FilePath]
preparePaths :: FilePath -> [FilePath] -> IO [FilePath]
preparePaths FilePath
baseDir = ([[FilePath]] -> [FilePath]) -> IO [[FilePath]] -> IO [FilePath]
forall a b. (a -> b) -> IO a -> IO b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[FilePath]] -> [FilePath]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
concat (IO [[FilePath]] -> IO [FilePath])
-> ([FilePath] -> IO [[FilePath]]) -> [FilePath] -> IO [FilePath]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. [IO [FilePath]] -> IO [[FilePath]]
forall a. [IO a] -> IO [a]
interleave ([IO [FilePath]] -> IO [[FilePath]])
-> ([FilePath] -> [IO [FilePath]]) -> [FilePath] -> IO [[FilePath]]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (FilePath -> IO [FilePath]) -> [FilePath] -> [IO [FilePath]]
forall a b. (a -> b) -> [a] -> [b]
map FilePath -> IO [FilePath]
go
where
go :: FilePath -> IO [FilePath]
go FilePath
relpath = do
let abspath :: FilePath
abspath = FilePath
baseDir FilePath -> FilePath -> FilePath
</> FilePath
relpath
isDir <- FilePath -> IO Bool
doesDirectoryExist FilePath
abspath
isSymlink <- pathIsSymbolicLink abspath
if isDir && not isSymlink then do
entries <- getDirectoryContentsRecursive abspath
let entries' = (FilePath -> FilePath) -> [FilePath] -> [FilePath]
forall a b. (a -> b) -> [a] -> [b]
map (FilePath
relpath FilePath -> FilePath -> FilePath
</>) [FilePath]
entries
return $ if null relpath
then entries'
else FilePath.Native.addTrailingPathSeparator relpath : entries'
else return [relpath]
packPaths
:: FilePath
-> [FilePath]
-> IO [GenEntry FilePath FilePath]
packPaths :: FilePath -> [FilePath] -> IO [GenEntry FilePath FilePath]
packPaths FilePath
baseDir [FilePath]
paths = [IO (GenEntry FilePath FilePath)]
-> IO [GenEntry FilePath FilePath]
forall a. [IO a] -> IO [a]
interleave ([IO (GenEntry FilePath FilePath)]
-> IO [GenEntry FilePath FilePath])
-> [IO (GenEntry FilePath FilePath)]
-> IO [GenEntry FilePath FilePath]
forall a b. (a -> b) -> a -> b
$ ((FilePath -> IO (GenEntry FilePath FilePath))
-> [FilePath] -> [IO (GenEntry FilePath FilePath)])
-> [FilePath]
-> (FilePath -> IO (GenEntry FilePath FilePath))
-> [IO (GenEntry FilePath FilePath)]
forall a b c. (a -> b -> c) -> b -> a -> c
flip (FilePath -> IO (GenEntry FilePath FilePath))
-> [FilePath] -> [IO (GenEntry FilePath FilePath)]
forall a b. (a -> b) -> [a] -> [b]
map [FilePath]
paths ((FilePath -> IO (GenEntry FilePath FilePath))
-> [IO (GenEntry FilePath FilePath)])
-> (FilePath -> IO (GenEntry FilePath FilePath))
-> [IO (GenEntry FilePath FilePath)]
forall a b. (a -> b) -> a -> b
$ \FilePath
relpath -> do
let isDir :: Bool
isDir = FilePath -> Bool
FilePath.Native.hasTrailingPathSeparator FilePath
abspath
abspath :: FilePath
abspath = FilePath
baseDir FilePath -> FilePath -> FilePath
</> FilePath
relpath
isSymlink <- FilePath -> IO Bool
pathIsSymbolicLink FilePath
abspath
let mkEntry
| Bool
isSymlink = FilePath -> tarPath -> IO (GenEntry tarPath FilePath)
forall tarPath.
FilePath -> tarPath -> IO (GenEntry tarPath FilePath)
packSymlinkEntry
| Bool
isDir = FilePath -> tarPath -> IO (GenEntry tarPath FilePath)
forall tarPath linkTarget.
FilePath -> tarPath -> IO (GenEntry tarPath linkTarget)
packDirectoryEntry
| Bool
otherwise = FilePath -> tarPath -> IO (GenEntry tarPath FilePath)
forall tarPath linkTarget.
FilePath -> tarPath -> IO (GenEntry tarPath linkTarget)
packFileEntry
mkEntry abspath relpath
interleave :: [IO a] -> IO [a]
interleave :: forall a. [IO a] -> IO [a]
interleave = IO [a] -> IO [a]
forall a. IO a -> IO a
unsafeInterleaveIO (IO [a] -> IO [a]) -> ([IO a] -> IO [a]) -> [IO a] -> IO [a]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. [IO a] -> IO [a]
forall a. [IO a] -> IO [a]
go
where
go :: [IO a] -> IO [a]
go [] = [a] -> IO [a]
forall a. a -> IO a
forall (m :: * -> *) a. Monad m => a -> m a
return []
go (IO a
x:[IO a]
xs) = do
x' <- IO a
x
xs' <- interleave xs
return (x':xs')
packFileEntry
:: FilePath
-> tarPath
-> IO (GenEntry tarPath linkTarget)
packFileEntry :: forall tarPath linkTarget.
FilePath -> tarPath -> IO (GenEntry tarPath linkTarget)
packFileEntry FilePath
filepath tarPath
tarpath = do
mtime <- FilePath -> IO EpochTime
getModTime FilePath
filepath
perms <- getPermissions filepath
approxSize <- getFileSize filepath
(content, size) <- if approxSize < 131072
then do
cnt <- B.readFile filepath
pure (BL.fromStrict cnt, fromIntegral $ B.length cnt)
else do
hndl <- openBinaryFile filepath ReadMode
sz <- hFileSize hndl
cnt <- BL.hGetContents hndl
pure (cnt, fromInteger sz)
pure (simpleEntry tarpath (NormalFile content size))
{ entryPermissions =
if executable perms then executableFilePermissions else ordinaryFilePermissions
, entryTime = mtime
}
packDirectoryEntry
:: FilePath
-> tarPath
-> IO (GenEntry tarPath linkTarget)
packDirectoryEntry :: forall tarPath linkTarget.
FilePath -> tarPath -> IO (GenEntry tarPath linkTarget)
packDirectoryEntry FilePath
filepath tarPath
tarpath = do
mtime <- FilePath -> IO EpochTime
getModTime FilePath
filepath
return (directoryEntry tarpath) {
entryTime = mtime
}
packSymlinkEntry
:: FilePath
-> tarPath
-> IO (GenEntry tarPath FilePath)
packSymlinkEntry :: forall tarPath.
FilePath -> tarPath -> IO (GenEntry tarPath FilePath)
packSymlinkEntry FilePath
filepath tarPath
tarpath = do
linkTarget <- FilePath -> IO FilePath
getSymbolicLinkTarget FilePath
filepath
pure $ symlinkEntry tarpath linkTarget
getDirectoryContentsRecursive :: FilePath -> IO [FilePath]
getDirectoryContentsRecursive :: FilePath -> IO [FilePath]
getDirectoryContentsRecursive FilePath
dir0 =
([FilePath] -> [FilePath]) -> IO [FilePath] -> IO [FilePath]
forall a b. (a -> b) -> IO a -> IO b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (Int -> [FilePath] -> [FilePath]
forall a. Int -> [a] -> [a]
drop Int
1) (FilePath -> [FilePath] -> IO [FilePath]
recurseDirectories FilePath
dir0 [FilePath
""])
recurseDirectories :: FilePath -> [FilePath] -> IO [FilePath]
recurseDirectories :: FilePath -> [FilePath] -> IO [FilePath]
recurseDirectories FilePath
_ [] = [FilePath] -> IO [FilePath]
forall a. a -> IO a
forall (m :: * -> *) a. Monad m => a -> m a
return []
recurseDirectories FilePath
base (FilePath
dir:[FilePath]
dirs) = IO [FilePath] -> IO [FilePath]
forall a. IO a -> IO a
unsafeInterleaveIO (IO [FilePath] -> IO [FilePath]) -> IO [FilePath] -> IO [FilePath]
forall a b. (a -> b) -> a -> b
$ do
(files, dirs') <- [FilePath]
-> [FilePath] -> [FilePath] -> IO ([FilePath], [FilePath])
collect [] [] ([FilePath] -> IO ([FilePath], [FilePath]))
-> IO [FilePath] -> IO ([FilePath], [FilePath])
forall (m :: * -> *) a b. Monad m => (a -> m b) -> m a -> m b
=<< FilePath -> IO [FilePath]
listDirectory (FilePath
base FilePath -> FilePath -> FilePath
</> FilePath
dir)
files' <- recurseDirectories base (dirs' ++ dirs)
return (dir : files ++ files')
where
collect :: [FilePath]
-> [FilePath] -> [FilePath] -> IO ([FilePath], [FilePath])
collect [FilePath]
files [FilePath]
dirs' [] = ([FilePath], [FilePath]) -> IO ([FilePath], [FilePath])
forall a. a -> IO a
forall (m :: * -> *) a. Monad m => a -> m a
return ([FilePath] -> [FilePath]
forall a. [a] -> [a]
reverse [FilePath]
files, [FilePath] -> [FilePath]
forall a. [a] -> [a]
reverse [FilePath]
dirs')
collect [FilePath]
files [FilePath]
dirs' (FilePath
entry:[FilePath]
entries) = do
let dirEntry :: FilePath
dirEntry = FilePath
dir FilePath -> FilePath -> FilePath
</> FilePath
entry
dirEntry' :: FilePath
dirEntry' = FilePath -> FilePath
FilePath.Native.addTrailingPathSeparator FilePath
dirEntry
isDirectory <- FilePath -> IO Bool
doesDirectoryExist (FilePath
base FilePath -> FilePath -> FilePath
</> FilePath
dirEntry)
isSymlink <- pathIsSymbolicLink (base </> dirEntry)
if isDirectory && not isSymlink
then collect files (dirEntry':dirs') entries
else collect (dirEntry:files) dirs' entries
getModTime :: FilePath -> IO EpochTime
getModTime :: FilePath -> IO EpochTime
getModTime FilePath
path = do
t <- FilePath -> IO UTCTime
getModificationTime FilePath
path
return . floor . utcTimeToPOSIXSeconds $ t