有效地将Foreign.Ptr Word8(或ByteString)转换为UArray Int Word8

时间:2018-06-12 16:09:20

标签: parsing haskell packet pcap

我正在使用Network.Pcappcap)进行网络捕获,并计划使用Net.PacketParsingnetwork-house)进行一些检查。为此,看起来我必须将数据包解析放在

Pcap.Callback :: PktHdr -> Ptr Word8 -> IO ()

Pcap.CallbackBS :: PktHdr -> ByteString -> IO ().

将数据包作为一个' Ptr Word8'或者' ByteString'。在数据包解析方面,我有:

Net.Packet.toInPack :: UArray Int Word8 -> InPacket

转到解析所需的InPacket类型。所以,给我留下的是转换' Ptr'或者' ByteString'到“UArray” - 无论是纯粹的还是在IO中。我想我可以将ByteString解压缩到[Word8],然后从那里解压缩到UArray,但似乎必须有更好的方法。

我也担心我选择的图书馆。我过去曾经使用过网络之家并发现它很不错,但它已经老了并使用了UArray,它本身看起来有点过时了。因此欢迎提出更好的起点建议。

1 个答案:

答案 0 :(得分:1)

ByteStringPtr Word8指向外部堆,而UArray位于GHC堆上,因此任何转换函数都必须复制数据。

我还没有在库中找到任何直接转换函数,但幸运的是有一个GHC原语完全符合我们的要求,称为copyAddrToByteArray#。这让我们可以用最小的开销进行转换:

{-# language MagicHash, UnboxedTuples #-}

import qualified Data.ByteString as B
import qualified Data.ByteString.Internal as B
import qualified Data.Array.Base as A

import GHC.Types
import GHC.Prim
import GHC.Magic (runRW#)
import GHC.ForeignPtr
import Data.Word

-- when using GHC 8.2.x or later:
byteStringToUArray :: B.ByteString -> A.UArray Int Word8
byteStringToUArray (B.PS (ForeignPtr addr _) (I# start) (I# len)) =
  runRW# $ \s -> case newByteArray# len s of
    (# s, marr #) -> case copyAddrToByteArray# (plusAddr# addr start) marr 0# len s of
      s -> case unsafeFreezeByteArray# marr s of
        (# _, arr #) -> A.UArray 0 (I# (len -# 1#)) (I# len) arr
{-# inline byteStringToUArray #-}

-- when using GHC 8.0.x:
byteStringToUArray :: B.ByteString -> A.UArray Int Word8
byteStringToUArray (B.PS (ForeignPtr addr _) (I# start) (I# len)) =
  case (runRW# $ \s -> case newByteArray# len s of
    (# s, marr #) -> case copyAddrToByteArray# (plusAddr# addr start) marr 0# len s of
      s -> case unsafeFreezeByteArray# marr s of
        (# s, arr #) -> (# s, A.UArray 0 (I# (len -# 1#)) (I# len) arr #)) of
    (# _, res #) -> res
{-# inline byteStringToUArray #-}

但总的来说,array过时且现在很少使用,你是对的。