(b64) Use base64 encoding in networks' bitarrays

[ganeti-local] / src / Ganeti / Utils.hs

diff --git a/src/Ganeti/Utils.hs b/src/Ganeti/Utils.hs
index 026db21..c937f9f 100644 (file)
--- a/src/Ganeti/Utils.hs
+++ b/src/Ganeti/Utils.hs
@@ -58,6 +58,8 @@ module Ganeti.Utils
   , splitEithers
   , recombineEithers
   , setOwnerAndGroupFromNames
+  , b64StringToBitString
+  , bitStringToB64String
   ) where
 
 import Data.Char (toUpper, isAlphaNum, isDigit, isSpace)
@@ -77,6 +79,13 @@ import System.Exit
 import System.Posix.Files
 import System.Time
 
+import qualified Data.ByteString as BS
+import Data.ByteString.Base64 (decodeLenient, encode)
+import qualified Data.ByteString.Char8 as BSC
+import Data.Word (Word8)
+import Data.Char (intToDigit, digitToInt)
+import Numeric (showIntAtBase, readInt)
+
 -- * Debug functions
 
 -- | To be used only for debugging, breaks referential integrity.
@@ -458,3 +467,34 @@ setOwnerAndGroupFromNames filename daemon dGroup = do
   let uid = fst ents M.! daemon
   let gid = snd ents M.! dGroup
   setOwnerAndGroup filename uid gid
+
+type BitString = String
+
+-- | Base 64 encoded String to BitString
+wordsToBitString :: [Word8] -> BitString
+wordsToBitString =
+    concatMap (padBits . wordToBits)
+  where
+    wordToBits = flip (showIntAtBase 2 intToDigit) ""
+    padBits bs = replicate (8 - length bs) '0' ++ bs
+
+decodeB64String :: String -> [Word8]
+decodeB64String = BS.unpack . decodeLenient . BSC.pack
+
+b64StringToBitString :: String -> BitString
+b64StringToBitString = wordsToBitString . decodeB64String
+
+-- | A BitString to Base 64 encoded String
+bitStringToWords :: BitString -> [Word8]
+bitStringToWords [] = []
+bitStringToWords bs =
+    bitStringToWord c : bitStringToWords rest
+  where
+    bitStringToWord = fst . head . readInt 2 (const True) digitToInt
+    (c, rest) = splitAt 8 bs
+
+encodeB64String :: [Word8] -> String
+encodeB64String = BSC.unpack . encode . BS.pack
+
+bitStringToB64String :: BitString -> String
+bitStringToB64String  = encodeB64String . bitStringToWords