mirror of
https://github.com/idris-lang/Idris2.git
synced 2024-12-21 02:31:50 +03:00
449 lines
15 KiB
Idris
449 lines
15 KiB
Idris
module Data.Buffer
|
|
|
|
import Data.List
|
|
|
|
%default total
|
|
|
|
-- Reading and writing binary buffers. Note that this primitives are unsafe,
|
|
-- in that they don't check that buffer locations are within bounds.
|
|
-- We really need a safe wrapper!
|
|
-- They are used in the Idris compiler itself for reading/writing checked
|
|
-- files.
|
|
|
|
-- This is known to the compiler, so maybe ought to be moved to Builtin
|
|
export
|
|
data Buffer : Type where [external]
|
|
|
|
%foreign "scheme:blodwen-buffer-size"
|
|
"RefC:getBufferSize"
|
|
"node:lambda:b => b.length"
|
|
prim__bufferSize : Buffer -> Int
|
|
|
|
export %inline
|
|
rawSize : HasIO io => Buffer -> io Int
|
|
rawSize buf = pure (prim__bufferSize buf)
|
|
|
|
%foreign "scheme:blodwen-new-buffer"
|
|
"RefC:newBuffer"
|
|
"node:lambda:s=>Buffer.alloc(s)"
|
|
prim__newBuffer : Int -> PrimIO Buffer
|
|
|
|
export
|
|
newBuffer : HasIO io => Int -> io (Maybe Buffer)
|
|
newBuffer size
|
|
= if size >= 0
|
|
then do buf <- primIO (prim__newBuffer size)
|
|
pure $ Just buf
|
|
else pure Nothing
|
|
-- if prim__nullAnyPtr buf /= 0
|
|
-- then pure Nothing
|
|
-- else pure $ Just $ MkBuffer buf size 0
|
|
|
|
|
|
------------------------------------------------------------------------
|
|
-- BitsN primitives
|
|
|
|
-- There is no endianness indication (LE/BE) for UInt8 since it is a single byte
|
|
|
|
-- TODO: remove me when we remove the deprecated `setByte` in a future release
|
|
%foreign "scheme:blodwen-buffer-setbyte"
|
|
"RefC:setBufferUInt8"
|
|
"node:lambda:(buf,offset,value)=>buf.writeUInt8(value, offset)"
|
|
prim__setByte : Buffer -> (offset : Int) -> (val : Int) -> PrimIO ()
|
|
|
|
%foreign "scheme:blodwen-buffer-setbyte"
|
|
"RefC:setBufferUInt8"
|
|
"node:lambda:(buf,offset,value)=>buf.writeUInt8(value, offset)"
|
|
prim__setBits8 : Buffer -> (offset : Int) -> (val: Bits8) -> PrimIO ()
|
|
|
|
-- Assumes val is in the range 0-255
|
|
||| Use `setBits8` instead, as its value is correctly limited.
|
|
%deprecate
|
|
export %inline
|
|
setByte : HasIO io => Buffer -> (offset : Int) -> (val : Int) -> io ()
|
|
setByte buf offset val
|
|
= primIO (prim__setByte buf offset val)
|
|
|
|
export %inline
|
|
setBits8 : HasIO io => Buffer -> (offset : Int) -> (val : Bits8) -> io ()
|
|
setBits8 buf offset val
|
|
= primIO (prim__setBits8 buf offset val)
|
|
|
|
-- TODO: remove me when we remove the deprecated `getByte` in a future release
|
|
%foreign "scheme:blodwen-buffer-getbyte"
|
|
"RefC:getBufferByte"
|
|
"node:lambda:(buf,offset)=>buf.readUInt8(offset)"
|
|
prim__getByte : Buffer -> (offset : Int) -> PrimIO Int
|
|
|
|
%foreign "scheme:blodwen-buffer-getbyte"
|
|
"RefC:getBufferUInt8"
|
|
"node:lambda:(buf,offset)=>buf.readUInt8(offset)"
|
|
prim__getBits8 : Buffer -> (offset : Int) -> PrimIO Bits8
|
|
|
|
||| Use `getBits8` instead, as its value is correctly limited.
|
|
%deprecate
|
|
export %inline
|
|
getByte : HasIO io => Buffer -> (offset : Int) -> io Int
|
|
getByte buf offset
|
|
= primIO (prim__getByte buf offset)
|
|
|
|
export %inline
|
|
getBits8 : HasIO io => Buffer -> (offset : Int) -> io Bits8
|
|
getBits8 buf offset
|
|
= primIO (prim__getBits8 buf offset)
|
|
|
|
%foreign "scheme:blodwen-buffer-setbits16"
|
|
"RefC:setBufferUInt16LE"
|
|
"node:lambda:(buf,offset,value)=>buf.writeUInt16LE(value, offset)"
|
|
prim__setBits16 : Buffer -> (offset : Int) -> (value : Bits16) -> PrimIO ()
|
|
|
|
export %inline
|
|
setBits16 : HasIO io => Buffer -> (offset : Int) -> (val : Bits16) -> io ()
|
|
setBits16 buf offset val
|
|
= primIO (prim__setBits16 buf offset val)
|
|
|
|
%foreign "scheme:blodwen-buffer-getbits16"
|
|
"RefC:getBufferUInt16LE"
|
|
"node:lambda:(buf,offset)=>buf.readUInt16LE(offset)"
|
|
prim__getBits16 : Buffer -> (offset : Int) -> PrimIO Bits16
|
|
|
|
export %inline
|
|
getBits16 : HasIO io => Buffer -> (offset : Int) -> io Bits16
|
|
getBits16 buf offset
|
|
= primIO (prim__getBits16 buf offset)
|
|
|
|
%foreign "scheme:blodwen-buffer-setbits32"
|
|
"RefC:setBufferUInt32LE"
|
|
"node:lambda:(buf,offset,value)=>buf.writeUInt32LE(value, offset)"
|
|
prim__setBits32 : Buffer -> (offset : Int) -> (value : Bits32) -> PrimIO ()
|
|
|
|
export %inline
|
|
setBits32 : HasIO io => Buffer -> (offset : Int) -> (val : Bits32) -> io ()
|
|
setBits32 buf offset val
|
|
= primIO (prim__setBits32 buf offset val)
|
|
|
|
%foreign "scheme:blodwen-buffer-getbits32"
|
|
"RefC:getBufferUInt32LE"
|
|
"node:lambda:(buf,offset)=>buf.readUInt32LE(offset)"
|
|
prim__getBits32 : Buffer -> Int -> PrimIO Bits32
|
|
|
|
export %inline
|
|
getBits32 : HasIO io => Buffer -> (offset : Int) -> io Bits32
|
|
getBits32 buf offset
|
|
= primIO (prim__getBits32 buf offset)
|
|
|
|
%foreign "scheme:blodwen-buffer-setbits64"
|
|
"RefC:setBufferUInt64LE"
|
|
"node:lambda:(buf,offset,value)=>buf.writeBigUInt64LE(value, offset)"
|
|
prim__setBits64 : Buffer -> Int -> Bits64 -> PrimIO ()
|
|
|
|
export %inline
|
|
setBits64 : HasIO io => Buffer -> (offset : Int) -> (val : Bits64) -> io ()
|
|
setBits64 buf offset val
|
|
= primIO (prim__setBits64 buf offset val)
|
|
|
|
%foreign "scheme:blodwen-buffer-getbits64"
|
|
"RefC:getBufferUInt64LE"
|
|
"node:lambda:(buf,offset)=>buf.readBigUInt64LE(offset)"
|
|
prim__getBits64 : Buffer -> (offset : Int) -> PrimIO Bits64
|
|
|
|
export %inline
|
|
getBits64 : HasIO io => Buffer -> (offset : Int) -> io Bits64
|
|
getBits64 buf offset
|
|
= primIO (prim__getBits64 buf offset)
|
|
|
|
------------------------------------------------------------------------
|
|
-- IntN primitives
|
|
|
|
-- There is no endianness indication (LE/BE) for Int8 since it is a single byte
|
|
|
|
%foreign "scheme:blodwen-buffer-setint8"
|
|
prim__setInt8 : Buffer -> (offset : Int) -> (val : Int8) -> PrimIO ()
|
|
|
|
export %inline
|
|
setInt8 : HasIO io => Buffer -> (offset : Int) -> (val : Int8) -> io ()
|
|
setInt8 buf offset val
|
|
= primIO (prim__setInt8 buf offset val)
|
|
|
|
%foreign "scheme:blodwen-buffer-getint8"
|
|
prim__getInt8 : Buffer -> (offset : Int) -> PrimIO Int8
|
|
|
|
export %inline
|
|
getInt8 : HasIO io => Buffer -> (offset : Int) -> io Int8
|
|
getInt8 buf offset
|
|
= primIO (prim__getInt8 buf offset)
|
|
|
|
%foreign "scheme:blodwen-buffer-setint16"
|
|
"RefC:setBufferInt16LE"
|
|
"node:lambda:(buf,offset,value)=>buf.writeInt16LE(value, offset)"
|
|
prim__setInt16 : Buffer -> (offset : Int) -> (val : Int16) -> PrimIO ()
|
|
|
|
export %inline
|
|
setInt16 : HasIO io => Buffer -> (offset : Int) -> (val : Int16) -> io ()
|
|
setInt16 buf offset val
|
|
= primIO (prim__setInt16 buf offset val)
|
|
|
|
%foreign "scheme:blodwen-buffer-getint16"
|
|
prim__getInt16 : Buffer -> (offset : Int) -> PrimIO Int16
|
|
|
|
export %inline
|
|
getInt16 : HasIO io => Buffer -> (offset : Int) -> io Int16
|
|
getInt16 buf offset
|
|
= primIO (prim__getInt16 buf offset)
|
|
|
|
|
|
%foreign "scheme:blodwen-buffer-setint32"
|
|
"RefC:setBufferInt32LE"
|
|
"node:lambda:(buf,offset,value)=>buf.writeInt32LE(value, offset)"
|
|
prim__setInt32 : Buffer -> (offset : Int) -> (val : Int32) -> PrimIO ()
|
|
|
|
export %inline
|
|
setInt32 : HasIO io => Buffer -> (offset : Int) -> (val : Int32) -> io ()
|
|
setInt32 buf offset val
|
|
= primIO (prim__setInt32 buf offset val)
|
|
|
|
%foreign "scheme:blodwen-buffer-getint32"
|
|
"RefC:getBufferInt32LE"
|
|
"node:lambda:(buf,offset)=>buf.readInt32LE(offset)"
|
|
prim__getInt32 : Buffer -> (offset : Int) -> PrimIO Int32
|
|
|
|
export %inline
|
|
getInt32 : HasIO io => Buffer -> (offset : Int) -> io Int32
|
|
getInt32 buf offset
|
|
= primIO (prim__getInt32 buf offset)
|
|
|
|
%foreign "scheme:blodwen-buffer-setint64"
|
|
prim__setInt64 : Buffer -> (offset : Int) -> (val : Int64) -> PrimIO ()
|
|
|
|
export %inline
|
|
setInt64 : HasIO io => Buffer -> (offset : Int) -> (val : Int64) -> io ()
|
|
setInt64 buf offset val
|
|
= primIO (prim__setInt64 buf offset val)
|
|
|
|
%foreign "scheme:blodwen-buffer-getint64"
|
|
prim__getInt64 : Buffer -> (offset : Int) -> PrimIO Int64
|
|
|
|
export %inline
|
|
getInt64 : HasIO io => Buffer -> (offset : Int) -> io Int64
|
|
getInt64 buf offset
|
|
= primIO (prim__getInt64 buf offset)
|
|
|
|
------------------------------------------------------------------------
|
|
-- Int (backend-dependent: 64 on scheme, refc, and 32 on js)
|
|
|
|
%foreign "scheme:blodwen-buffer-setint"
|
|
"RefC:setBufferInt64LE"
|
|
"node:lambda:(buf,offset,value)=>buf.writeInt32LE(value, offset)"
|
|
prim__setInt : Buffer -> (offset : Int) -> (val : Int) -> PrimIO ()
|
|
|
|
export %inline
|
|
setInt : HasIO io => Buffer -> (offset : Int) -> (val : Int) -> io ()
|
|
setInt buf offset val
|
|
= primIO (prim__setInt buf offset val)
|
|
|
|
%foreign "scheme:blodwen-buffer-getint"
|
|
"RefC:getBufferInt64LE"
|
|
"node:lambda:(buf,offset)=>buf.readInt32LE(offset)"
|
|
prim__getInt : Buffer -> (offset : Int) -> PrimIO Int
|
|
|
|
export %inline
|
|
getInt : HasIO io => Buffer -> (offset : Int) -> io Int
|
|
getInt buf offset
|
|
= primIO (prim__getInt buf offset)
|
|
|
|
------------------------------------------------------------------------
|
|
-- Double
|
|
|
|
%foreign "scheme:blodwen-buffer-setdouble"
|
|
"RefC:setBufferDouble"
|
|
"node:lambda:(buf,offset,value)=>buf.writeDoubleLE(value, offset)"
|
|
prim__setDouble : Buffer -> (offset : Int) -> (val : Double) -> PrimIO ()
|
|
|
|
export %inline
|
|
setDouble : HasIO io => Buffer -> (offset : Int) -> (val : Double) -> io ()
|
|
setDouble buf offset val
|
|
= primIO (prim__setDouble buf offset val)
|
|
|
|
%foreign "scheme:blodwen-buffer-getdouble"
|
|
"RefC:getBufferDouble"
|
|
"node:lambda:(buf,offset)=>buf.readDoubleLE(offset)"
|
|
prim__getDouble : Buffer -> (offset : Int) -> PrimIO Double
|
|
|
|
export %inline
|
|
getDouble : HasIO io => Buffer -> (offset : Int) -> io Double
|
|
getDouble buf offset
|
|
= primIO (prim__getDouble buf offset)
|
|
|
|
------------------------------------------------------------------------
|
|
-- Bool
|
|
|
|
export
|
|
setBool : HasIO io => Buffer -> (offset : Int) -> (val : Bool) -> io ()
|
|
setBool buf offset val = setBits8 buf offset (ifThenElse val 0 1)
|
|
|
|
export
|
|
getBool : HasIO io => Buffer -> (offset : Int) -> io Bool
|
|
getBool buf offset = (0 ==) <$> getBits8 buf offset
|
|
|
|
------------------------------------------------------------------------
|
|
-- Arbitrary precision nums
|
|
|
|
||| setNat returns the end offset
|
|
export
|
|
setNat : HasIO io => Buffer -> (offset : Int) -> (val : Nat) -> io Int
|
|
setNat buf offset val
|
|
= do let limbs = toLimbs (cast val)
|
|
let len = foldl (const . (1+)) 0 limbs -- tail recursive length
|
|
setInt64 buf offset len
|
|
setLimbs (offset + 8) limbs
|
|
|
|
where
|
|
|
|
toLimbs : Integer -> List Bits32
|
|
toLimbs 0 = []
|
|
toLimbs (-1) = [-1]
|
|
toLimbs x = fromInteger (prim__and_Integer x 0xffffffff) ::
|
|
toLimbs (assert_smaller x (prim__shr_Integer x 32))
|
|
|
|
setLimbs : (offset : Int) -> List Bits32 -> io Int
|
|
setLimbs offset [] = pure offset
|
|
setLimbs offset (limb :: limbs)
|
|
= do setBits32 buf offset limb
|
|
setLimbs (offset + 4) limbs
|
|
|
|
||| getNat returns the end offset
|
|
export
|
|
getNat : HasIO io => Buffer -> (offset : Int) -> io (Int, Nat)
|
|
getNat buf offset
|
|
= do len <- getInt64 buf offset
|
|
when (len < 0) $ assert_total $ idris_crash "corrupt Nat"
|
|
limbs <- getLimbs [<] (offset + 8) len
|
|
pure (offset + 8 + 4 * cast len, cast $ fromLimbs limbs)
|
|
|
|
where
|
|
|
|
fromLimbs : List Bits32 -> Integer
|
|
fromLimbs [] = 0
|
|
fromLimbs (x :: xs) = cast x + prim__shl_Integer (fromLimbs xs) 32
|
|
|
|
getLimbs : SnocList Bits32 -> (offset : Int) -> (len : Int64) -> io (List Bits32)
|
|
getLimbs acc offset 0 = pure (acc <>> [])
|
|
getLimbs acc offset len
|
|
= do limb <- getBits32 buf offset
|
|
getLimbs (acc :< limb) (offset + 4) (assert_smaller len (len -1))
|
|
|
|
||| setInteger returns the end offset
|
|
export
|
|
setInteger : HasIO io => Buffer -> (offset : Int) -> (val : Integer) -> io Int
|
|
setInteger buf offset val = if val < 0
|
|
then do setBool buf offset True
|
|
setNat buf (offset + 1) (cast (- val))
|
|
else do setBool buf offset False
|
|
setNat buf (offset + 1) (cast val)
|
|
|
|
||| getInteger returns the end offset
|
|
export
|
|
getInteger : HasIO io => Buffer -> (offset : Int) -> io (Int, Integer)
|
|
getInteger buf offset
|
|
= do b <- getBool buf offset
|
|
map (ifThenElse b negate id . cast) <$> getNat buf (offset + 1)
|
|
|
|
------------------------------------------------------------------------
|
|
-- String
|
|
|
|
-- Get the length of a string in bytes, rather than characters
|
|
export
|
|
%foreign "scheme:blodwen-stringbytelen"
|
|
"C:strlen, libc 6"
|
|
"javascript:lambda:(string)=>new TextEncoder().encode(string).length"
|
|
stringByteLength : String -> Int
|
|
|
|
%foreign "scheme:blodwen-buffer-setstring"
|
|
"RefC:setBufferString"
|
|
"node:lambda:(buf,offset,value)=>buf.write(value, offset,buf.length - offset, 'utf-8')"
|
|
prim__setString : Buffer -> (offset : Int) -> (val : String) -> PrimIO ()
|
|
|
|
export %inline
|
|
setString : HasIO io => Buffer -> (offset : Int) -> (val : String) -> io ()
|
|
setString buf offset val
|
|
= primIO (prim__setString buf offset val)
|
|
|
|
%foreign "scheme:blodwen-buffer-getstring"
|
|
"RefC:getBufferString"
|
|
"node:lambda:(buf,offset,len)=>buf.slice(offset, offset+len).toString('utf-8')"
|
|
prim__getString : Buffer -> (offset : Int) -> (len : Int) -> PrimIO String
|
|
|
|
export %inline
|
|
getString : HasIO io => Buffer -> (offset : Int) -> (len : Int) -> io String
|
|
getString buf offset len
|
|
= primIO (prim__getString buf offset len)
|
|
|
|
export
|
|
covering
|
|
bufferData : HasIO io => Buffer -> io (List Int)
|
|
bufferData buf
|
|
= do len <- rawSize buf
|
|
unpackTo [] len
|
|
where
|
|
covering
|
|
unpackTo : List Int -> Int -> io (List Int)
|
|
unpackTo acc 0 = pure acc
|
|
unpackTo acc offset
|
|
= do val <- getByte buf (offset - 1)
|
|
unpackTo (val :: acc) (offset - 1)
|
|
|
|
|
|
%foreign "scheme:blodwen-buffer-copydata"
|
|
"RefC:copyBuffer"
|
|
"node:lambda:(b1,o1,length,b2,o2)=>b1.copy(b2,o2,o1,o1+length)"
|
|
prim__copyData : (src : Buffer) -> (srcOffset, len : Int) ->
|
|
(dst : Buffer) -> (dstOffset : Int) -> PrimIO ()
|
|
|
|
export
|
|
copyData : HasIO io => Buffer -> (srcOffset, len : Int) ->
|
|
(dst : Buffer) -> (dstOffset : Int) -> io ()
|
|
copyData src start len dest offset
|
|
= primIO (prim__copyData src start len dest offset)
|
|
|
|
export
|
|
resizeBuffer : HasIO io => Buffer -> Int -> io (Maybe Buffer)
|
|
resizeBuffer old newsize
|
|
= do Just buf <- newBuffer newsize
|
|
| Nothing => pure Nothing
|
|
-- If the new buffer is smaller than the old one, just copy what
|
|
-- fits
|
|
oldsize <- rawSize old
|
|
let len = if newsize < oldsize then newsize else oldsize
|
|
copyData old 0 len buf 0
|
|
pure (Just buf)
|
|
|
|
||| Create a buffer containing the concatenated content from a
|
|
||| list of buffers.
|
|
export
|
|
concatBuffers : HasIO io => List Buffer -> io (Maybe Buffer)
|
|
concatBuffers xs
|
|
= do let sizes = map prim__bufferSize xs
|
|
let (totalSize, revCumulative) = foldl scanSize (0,[]) sizes
|
|
let cumulative = reverse revCumulative
|
|
Just buf <- newBuffer totalSize
|
|
| Nothing => pure Nothing
|
|
traverse_ (\(b, size, watermark) => copyData b 0 size buf watermark) (zip3 xs sizes cumulative)
|
|
pure (Just buf)
|
|
where
|
|
scanSize : (Int, List Int) -> Int -> (Int, List Int)
|
|
scanSize (s, cs) x = (s+x, s::cs)
|
|
|
|
||| Split a buffer into two at a position.
|
|
export
|
|
splitBuffer : HasIO io => Buffer -> Int -> io (Maybe (Buffer, Buffer))
|
|
splitBuffer buf pos = do size <- rawSize buf
|
|
if pos > 0 && pos < size
|
|
then do Just first <- newBuffer pos
|
|
| Nothing => pure Nothing
|
|
Just second <- newBuffer (size - pos)
|
|
| Nothing => pure Nothing
|
|
copyData buf 0 pos first 0
|
|
copyData buf pos (size-pos) second 0
|
|
pure $ Just (first, second)
|
|
else pure Nothing
|