2 - Copyright (C) 2015-2016 Ramakrishnan Muthukrishnan <ram@rkrishnan.org>
4 - This file is part of FuncTorrent.
6 - FuncTorrent is free software; you can redistribute it and/or modify
7 - it under the terms of the GNU General Public License as published by
8 - the Free Software Foundation; either version 3 of the License, or
9 - (at your option) any later version.
11 - FuncTorrent is distributed in the hope that it will be useful,
12 - but WITHOUT ANY WARRANTY; without even the implied warranty of
13 - MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 - GNU General Public License for more details.
16 - You should have received a copy of the GNU General Public License
17 - along with FuncTorrent; if not, see <http://www.gnu.org/licenses/>
20 module FuncTorrent.Utils
36 import Prelude hiding (writeFile, take)
38 import qualified Crypto.Hash.SHA1 as SHA1 (hash)
39 import Control.Exception.Base (IOException, try)
40 import Data.ByteString (ByteString, writeFile, hPut, hGet, take)
41 import qualified Data.ByteString.Base16 as B16 (encode)
42 import qualified Data.ByteString.Char8 as BC
43 import Data.List (intercalate)
44 import System.IO (Handle, hSeek, SeekMode(..))
45 import System.Directory (doesFileExist)
50 splitN :: Int -> BC.ByteString -> [BC.ByteString]
51 splitN n bs | BC.null bs = []
52 | otherwise = BC.take n bs : splitN n (BC.drop n bs)
54 splitNum :: Integer -> Integer -> [Integer]
55 splitNum n d | n == 0 = []
57 | otherwise = d : splitNum (n - d) d
59 createDummyFile :: FilePath -> Int -> IO (Either IOException ())
60 createDummyFile path size = do
61 dfe <- doesFileExist path
64 try $ writeFile path (BC.replicate size '\0')
68 -- write into a file at a specific offet
69 writeFileAtOffset :: Handle -> Integer -> ByteString -> IO ()
70 writeFileAtOffset h offset block = do
71 hSeek h AbsoluteSeek offset
74 readFileAtOffset :: Handle -> Integer -> Integer -> IO ByteString
75 readFileAtOffset h offset len = do
76 hSeek h AbsoluteSeek offset
77 hGet h (fromInteger len)
79 verifyHash :: ByteString -> ByteString -> Bool
80 verifyHash bs pieceHash =
81 take 20 (SHA1.hash bs) == pieceHash
83 toPort :: ByteString -> Port
84 toPort = read . ("0x" ++) . BC.unpack . B16.encode
86 toIP :: ByteString -> IP
87 toIP = Data.List.intercalate "." .
88 map (show . toInt . ("0x" ++) . BC.unpack) .
91 toInt :: String -> Integer
94 getHostname :: String -> String
95 getHostname url = takeWhile (/= ':') $ drop 2 $ dropWhile (/= '/') url
97 getPort :: String -> Port
98 getPort url = read $ takeWhile (/= '/') $ drop 1 $ dropWhile (/= ':') $ drop 2 $ dropWhile (/= '/') url