1 files changed, 27 insertions, 21 deletions

diff --git a/Data/OpenPGP/Internal.hs b/Data/OpenPGP/Internal.hs
index 24330f1..175a62e 100644
--- a/Data/OpenPGP/Internal.hs
+++ b/Data/OpenPGP/Internal.hs
@@ -1,17 +1,20 @@
-{-# LANGUAGE BangPatterns #-}
+{-# LANGUAGE BangPatterns, MagicHash #-}
 module Data.OpenPGP.Internal where
 
 import           Data.Bits
-import qualified Data.ByteString          as BS
-import qualified Data.ByteString.Internal as BS
+import qualified Data.ByteString           as BS
+import qualified Data.ByteString.Internal  as BS
 import           Data.Word
 import           Foreign.ForeignPtr
 import           Foreign.Ptr
 import           Foreign.Storable
+import           GHC.Exts
+import           GHC.Integer.GMP.Internals
 import           System.Endian
 import           System.IO.Unsafe
 
 
+
 decode_s2k_count :: Word8 -> Word32
 decode_s2k_count c =  (16 + (fromIntegral c .&. 15)) `shiftL`
         ((fromIntegral c `shiftR` 4) + 6)
@@ -28,6 +31,7 @@ encode_s2k_count iterations
                 | count < 32 = (count, c)
                 | otherwise = encode_s2k_count' (count `shiftR` 1) (c+1)
 
+{-
 getBigNum :: BS.ByteString -> Integer
 getBigNum bytes = unsafeDupablePerformIO $
     let (fptr,offset,len) = BS.toForeignPtr bytes
@@ -49,23 +53,25 @@ getBigNum bytes = unsafeDupablePerformIO $
                     go2 (p `plusPtr` 1) a'
         a <- go p 0
         go2 (castPtr e :: Ptr Word8) a
+-}
+getBigNum :: BS.ByteString -> Integer
+getBigNum bytes = unsafeDupablePerformIO $
+    let (fptr,offset,len) = BS.toForeignPtr bytes
+    in withForeignPtr fptr $ \ptr -> do
+        let Ptr addr = ptr `plusPtr` offset :: Ptr Word64
+            I# n = len
+        importIntegerFromAddr addr (int2Word# n) 1#
 
-{-
-unchunk :: L.ByteString -> BS.ByteString
-unchunk b = L.toStrict $ foldr reappend L.empty $ L.toChunks b
-
-reappend :: BS.ByteString -> BS.ByteString -> Maybe BS.ByteString
-reappend a b =
-    let (ap,ao,al) = BS.toForeignPtr a
-        (bp,bo,bl) = BS.toForeignPtr b
-    in if ap == bp && ao+al == bo
-        then Just $ BS.PS ap ao (al+bl)
-        else Nothing
+putBigNum :: Integer -> (Word16, BS.ByteString)
+putBigNum 0 = (1,BS.singleton 0)
+putBigNum i = ( (fromIntegral (BS.length bytes) - 1) * 8 + sigBit
+            , bytes )
+ where
+    sigBit = fromIntegral $ 8 - countLeadingZeros (BS.index bytes 0)
+    bytes = integerToBS i
 
-reconsChunk :: S.ByteString -> L.ByteString -> L.ByteString
-reconsChunk b bs = case L.toChunks bs of
-    (c:cs) -> case reappend b c of
-        Just x  -> L.fromChunks (x:cs)
-        Nothing -> L.fromChunks (b:c:cs)
-    _      -> L.fromChunks [b]
--}
+integerToBS :: Integer -> BS.ByteString
+integerToBS i = BS.unsafeCreate (I# (word2Int# (sizeInBaseInteger i 256#))) $ \ptr -> do
+        let Ptr addr = ptr
+        cnt <- exportIntegerToAddr i addr 1#
+        return ()