{-# LANGUAGE BangPatterns           #-}
{-# LANGUAGE CPP                    #-}
{-# LANGUAGE DeriveDataTypeable     #-}
{-# LANGUAGE FlexibleContexts       #-}
{-# LANGUAGE FlexibleInstances      #-}
{-# LANGUAGE FunctionalDependencies #-}
{-# LANGUAGE MagicHash              #-}
{-# LANGUAGE MultiParamTypeClasses  #-}
{-# LANGUAGE OverloadedStrings      #-}
{-# LANGUAGE RankNTypes             #-}

#if __GLASGOW_HASKELL__ < 900
-- Bump up from the default 1.5, otherwise our decoder fast path is no good.
-- We went over the threshold when we switched to using ST.
--
-- However, this flag is not supported on GHC 9.0 and later and eye-balling the
-- Core suggests that the new inlining heuristics don't require it.
{-# OPTIONS_GHC -funfolding-keeness-factor=2.0 #-}
#endif

-- |
-- Module      : Codec.CBOR.Read
-- Copyright   : (c) Duncan Coutts 2015-2017
-- License     : BSD3-style (see LICENSE.txt)
--
-- Maintainer  : duncan@community.haskell.org
-- Stability   : experimental
-- Portability : non-portable (GHC extensions)
--
-- Tools for reading values in a CBOR-encoded format
-- back into ordinary values.
--
module Codec.CBOR.Read
  ( deserialiseFromBytes         -- :: Decoder a -> ByteString -> Either String (ByteString, a)
  , deserialiseFromBytesWithSize -- :: Decoder a -> ByteString -> Either String (ByteString, ByteOffset, a)
  , deserialiseIncremental       -- :: Decoder a -> ST s (IDecode s a)
  , DeserialiseFailure(..)
  , IDecode(..)
  , ByteOffset
  ) where

#include "cbor.h"

#if !MIN_VERSION_base(4,8,0)
import           Control.Applicative
#endif
import           GHC.Int

import           Control.DeepSeq
import           Control.Monad (ap)
import           Control.Monad.ST
import           Data.Array.IArray
import           Data.Array.Unboxed
import qualified Data.Array.Base as A
import           Data.Monoid
import           Data.Bits
import           Data.ByteString                (ByteString)
import qualified Data.ByteString                as BS
import qualified Data.ByteString.Unsafe         as BS
import qualified Data.ByteString.Lazy           as LBS
import qualified Data.ByteString.Lazy.Internal  as LBS
import qualified Data.Text          as T
import qualified Data.Text.Encoding as T
import           Data.Word
import           GHC.Word
#if defined(ARCH_32bit)
import           GHC.IntWord64
#endif
import           GHC.Exts
import           GHC.Float (float2Double)
import           Data.Typeable
import           Control.Exception

-- We do all numeric conversions explicitly to be careful about overflows.
import           Prelude hiding (fromIntegral)

import qualified Codec.CBOR.ByteArray as BA
import           Codec.CBOR.Decoding hiding (DecodeAction(Done, Fail))
import           Codec.CBOR.Decoding (DecodeAction)
import qualified Codec.CBOR.Decoding as D
import           Codec.CBOR.Magic

--------------------------------------------------------------------------------

-- | An exception type that may be returned (by pure functions) or
-- thrown (by IO actions) that fail to deserialise a given input.
--
-- @since 0.2.0.0
data DeserialiseFailure = DeserialiseFailure ByteOffset String
  deriving (DeserialiseFailure -> DeserialiseFailure -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: DeserialiseFailure -> DeserialiseFailure -> Bool
$c/= :: DeserialiseFailure -> DeserialiseFailure -> Bool
== :: DeserialiseFailure -> DeserialiseFailure -> Bool
$c== :: DeserialiseFailure -> DeserialiseFailure -> Bool
Eq, Int -> DeserialiseFailure -> ShowS
[DeserialiseFailure] -> ShowS
DeserialiseFailure -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [DeserialiseFailure] -> ShowS
$cshowList :: [DeserialiseFailure] -> ShowS
show :: DeserialiseFailure -> String
$cshow :: DeserialiseFailure -> String
showsPrec :: Int -> DeserialiseFailure -> ShowS
$cshowsPrec :: Int -> DeserialiseFailure -> ShowS
Show, Typeable)

instance NFData DeserialiseFailure where
  rnf :: DeserialiseFailure -> ()
rnf (DeserialiseFailure ByteOffset
offset String
msg) = forall a. NFData a => a -> ()
rnf ByteOffset
offset seq :: forall a b. a -> b -> b
`seq` forall a. NFData a => a -> ()
rnf String
msg seq :: forall a b. a -> b -> b
`seq` ()

instance Exception DeserialiseFailure where
#if MIN_VERSION_base(4,8,0)
    displayException :: DeserialiseFailure -> String
displayException (DeserialiseFailure ByteOffset
off String
msg) =
      String
"Codec.CBOR: deserialising failed at offset "
           forall a. [a] -> [a] -> [a]
++ forall a. Show a => a -> String
show ByteOffset
off forall a. [a] -> [a] -> [a]
++ String
" : " forall a. [a] -> [a] -> [a]
++ String
msg
#endif

-- | An Incremental decoder, used to represent the result of
-- attempting to run a decoder over a given input, and return a value
-- of type @a@.
data IDecode s a
  = -- | The decoder has consumed the available input and needs more
    -- to continue. Provide 'Just' if more input is available and
    -- 'Nothing' otherwise, and you will get a new 'IDecode'.
    Partial (Maybe BS.ByteString -> ST s (IDecode s a))

    -- | The decoder has successfully finished. Except for the output
    -- value you also get any unused input as well as the number of
    -- bytes consumed.
  | Done !BS.ByteString {-# UNPACK #-} !ByteOffset a

    -- | The decoder ran into an error. The decoder either used
    -- 'fail' or was not provided enough input. Contains any
    -- unconsumed input, the number of bytes consumed, and a
    -- 'DeserialiseFailure' exception describing the reason why the
    -- failure occurred.
  | Fail !BS.ByteString {-# UNPACK #-} !ByteOffset DeserialiseFailure

-- | Given a 'Decoder' and some 'LBS.ByteString' representing
-- an encoded CBOR value, return 'Either' the decoded CBOR value
-- or an error. In addition to the decoded value return any remaining input
-- content.
--
-- @since 0.2.0.0
deserialiseFromBytes :: (forall s. Decoder s a)
                     -> LBS.ByteString
                     -> Either DeserialiseFailure (LBS.ByteString, a)
deserialiseFromBytes :: forall a.
(forall s. Decoder s a)
-> ByteString -> Either DeserialiseFailure (ByteString, a)
deserialiseFromBytes forall s. Decoder s a
d ByteString
lbs =
    forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap forall {a} {b} {b}. (a, b, b) -> (a, b)
f forall a b. (a -> b) -> a -> b
$ forall a.
(forall s. ST s (IDecode s a))
-> ByteString
-> Either DeserialiseFailure (ByteString, ByteOffset, a)
runIDecode (forall s a. Decoder s a -> ST s (IDecode s a)
deserialiseIncremental forall s. Decoder s a
d) ByteString
lbs
  where f :: (a, b, b) -> (a, b)
f (a
rest, b
_, b
x) = (a
rest, b
x)

-- | Given a 'Decoder' and some 'LBS.ByteString' representing
-- an encoded CBOR value, return 'Either' the decoded CBOR value
-- or an error. In addition to the decoded value return any remaining input
-- content and the number of bytes consumed.
--
-- @since 0.2.0.0
deserialiseFromBytesWithSize :: (forall s. Decoder s a)
                             -> LBS.ByteString
                             -> Either DeserialiseFailure (LBS.ByteString, ByteOffset, a)
deserialiseFromBytesWithSize :: forall a.
(forall s. Decoder s a)
-> ByteString
-> Either DeserialiseFailure (ByteString, ByteOffset, a)
deserialiseFromBytesWithSize forall s. Decoder s a
d ByteString
lbs =
    forall a.
(forall s. ST s (IDecode s a))
-> ByteString
-> Either DeserialiseFailure (ByteString, ByteOffset, a)
runIDecode (forall s a. Decoder s a -> ST s (IDecode s a)
deserialiseIncremental forall s. Decoder s a
d) ByteString
lbs

runIDecode :: (forall s. ST s (IDecode s a))
           -> LBS.ByteString
           -> Either DeserialiseFailure (LBS.ByteString, ByteOffset, a)
runIDecode :: forall a.
(forall s. ST s (IDecode s a))
-> ByteString
-> Either DeserialiseFailure (ByteString, ByteOffset, a)
runIDecode forall s. ST s (IDecode s a)
d ByteString
lbs =
    forall a. (forall s. ST s a) -> a
runST (forall s a.
ByteString
-> IDecode s a
-> ST s (Either DeserialiseFailure (ByteString, ByteOffset, a))
go ByteString
lbs forall (m :: * -> *) a b. Monad m => (a -> m b) -> m a -> m b
=<< forall s. ST s (IDecode s a)
d)
  where
    go :: LBS.ByteString
       -> IDecode s a
       -> ST s (Either DeserialiseFailure (LBS.ByteString, ByteOffset, a))
    go :: forall s a.
ByteString
-> IDecode s a
-> ST s (Either DeserialiseFailure (ByteString, ByteOffset, a))
go  ByteString
_                  (Fail ByteString
_ ByteOffset
_ DeserialiseFailure
err)  = forall (m :: * -> *) a. Monad m => a -> m a
return (forall a b. a -> Either a b
Left DeserialiseFailure
err)
    go  ByteString
lbs'               (Done ByteString
bs ByteOffset
off a
x) = let rest :: ByteString
rest
                                                   | ByteString -> Bool
BS.null ByteString
bs = ByteString
lbs'
                                                   | Bool
otherwise  = ByteString -> ByteString -> ByteString
LBS.Chunk ByteString
bs ByteString
lbs'
                                             in forall (m :: * -> *) a. Monad m => a -> m a
return (forall a b. b -> Either a b
Right (ByteString
rest, ByteOffset
off, a
x))
    go  ByteString
LBS.Empty          (Partial  Maybe ByteString -> ST s (IDecode s a)
k)    = Maybe ByteString -> ST s (IDecode s a)
k forall a. Maybe a
Nothing   forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a.
ByteString
-> IDecode s a
-> ST s (Either DeserialiseFailure (ByteString, ByteOffset, a))
go ByteString
LBS.Empty
    go (LBS.Chunk ByteString
bs ByteString
lbs') (Partial  Maybe ByteString -> ST s (IDecode s a)
k)    = Maybe ByteString -> ST s (IDecode s a)
k (forall a. a -> Maybe a
Just ByteString
bs) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a.
ByteString
-> IDecode s a
-> ST s (Either DeserialiseFailure (ByteString, ByteOffset, a))
go ByteString
lbs'

-- | Run a 'Decoder' incrementally, returning a continuation
-- representing the result of the incremental decode.
--
-- @since 0.2.0.0
deserialiseIncremental :: Decoder s a -> ST s (IDecode s a)
deserialiseIncremental :: forall s a. Decoder s a -> ST s (IDecode s a)
deserialiseIncremental Decoder s a
decoder = do
    DecodeAction s a
da <- forall s a. Decoder s a -> ST s (DecodeAction s a)
getDecodeAction Decoder s a
decoder
    forall s a.
IncrementalDecoder s (ByteString, ByteOffset, a)
-> ST s (IDecode s a)
runIncrementalDecoder (forall s a.
DecodeAction s a
-> IncrementalDecoder s (ByteString, ByteOffset, a)
runDecodeAction DecodeAction s a
da)

----------------------------------------------
-- A monad for building incremental decoders
--

newtype IncrementalDecoder s a = IncrementalDecoder {
       forall s a.
IncrementalDecoder s a
-> forall r. (a -> ST s (IDecode s r)) -> ST s (IDecode s r)
unIncrementalDecoder ::
         forall r. (a -> ST s (IDecode s r)) -> ST s (IDecode s r)
     }

instance Functor (IncrementalDecoder s) where
    fmap :: forall a b.
(a -> b) -> IncrementalDecoder s a -> IncrementalDecoder s b
fmap a -> b
f IncrementalDecoder s a
a = IncrementalDecoder s a
a forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall (m :: * -> *) a. Monad m => a -> m a
return forall b c a. (b -> c) -> (a -> b) -> a -> c
. a -> b
f

instance Applicative (IncrementalDecoder s) where
    pure :: forall a. a -> IncrementalDecoder s a
pure a
x = forall s a.
(forall r. (a -> ST s (IDecode s r)) -> ST s (IDecode s r))
-> IncrementalDecoder s a
IncrementalDecoder forall a b. (a -> b) -> a -> b
$ \a -> ST s (IDecode s r)
k -> a -> ST s (IDecode s r)
k a
x
    <*> :: forall a b.
IncrementalDecoder s (a -> b)
-> IncrementalDecoder s a -> IncrementalDecoder s b
(<*>) = forall (m :: * -> *) a b. Monad m => m (a -> b) -> m a -> m b
ap

instance Monad (IncrementalDecoder s) where
    return :: forall a. a -> IncrementalDecoder s a
return = forall (f :: * -> *) a. Applicative f => a -> f a
pure

    {-# INLINE (>>=) #-}
    IncrementalDecoder s a
m >>= :: forall a b.
IncrementalDecoder s a
-> (a -> IncrementalDecoder s b) -> IncrementalDecoder s b
>>= a -> IncrementalDecoder s b
f = forall s a.
(forall r. (a -> ST s (IDecode s r)) -> ST s (IDecode s r))
-> IncrementalDecoder s a
IncrementalDecoder forall a b. (a -> b) -> a -> b
$ \b -> ST s (IDecode s r)
k ->
                forall s a.
IncrementalDecoder s a
-> forall r. (a -> ST s (IDecode s r)) -> ST s (IDecode s r)
unIncrementalDecoder IncrementalDecoder s a
m forall a b. (a -> b) -> a -> b
$ \a
x ->
                  forall s a.
IncrementalDecoder s a
-> forall r. (a -> ST s (IDecode s r)) -> ST s (IDecode s r)
unIncrementalDecoder (a -> IncrementalDecoder s b
f a
x) b -> ST s (IDecode s r)
k

runIncrementalDecoder :: IncrementalDecoder s (ByteString, ByteOffset, a)
                      -> ST s (IDecode s a)
runIncrementalDecoder :: forall s a.
IncrementalDecoder s (ByteString, ByteOffset, a)
-> ST s (IDecode s a)
runIncrementalDecoder (IncrementalDecoder forall r.
((ByteString, ByteOffset, a) -> ST s (IDecode s r))
-> ST s (IDecode s r)
f) =
  forall r.
((ByteString, ByteOffset, a) -> ST s (IDecode s r))
-> ST s (IDecode s r)
f (\(ByteString
trailing, ByteOffset
off, a
x) -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$ forall s a. ByteString -> ByteOffset -> a -> IDecode s a
Done ByteString
trailing ByteOffset
off a
x)

decodeFail :: ByteString -> ByteOffset -> String -> IncrementalDecoder s a
decodeFail :: forall s a.
ByteString -> ByteOffset -> String -> IncrementalDecoder s a
decodeFail ByteString
trailing ByteOffset
off String
msg = forall s a.
(forall r. (a -> ST s (IDecode s r)) -> ST s (IDecode s r))
-> IncrementalDecoder s a
IncrementalDecoder forall a b. (a -> b) -> a -> b
$ \a -> ST s (IDecode s r)
_ -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$ forall s a.
ByteString -> ByteOffset -> DeserialiseFailure -> IDecode s a
Fail ByteString
trailing ByteOffset
off DeserialiseFailure
exn
  where exn :: DeserialiseFailure
exn = ByteOffset -> String -> DeserialiseFailure
DeserialiseFailure ByteOffset
off String
msg

needChunk :: IncrementalDecoder s (Maybe ByteString)
needChunk :: forall s. IncrementalDecoder s (Maybe ByteString)
needChunk = forall s a.
(forall r. (a -> ST s (IDecode s r)) -> ST s (IDecode s r))
-> IncrementalDecoder s a
IncrementalDecoder forall a b. (a -> b) -> a -> b
$ \Maybe ByteString -> ST s (IDecode s r)
k -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$ forall s a. (Maybe ByteString -> ST s (IDecode s a)) -> IDecode s a
Partial forall a b. (a -> b) -> a -> b
$ \Maybe ByteString
mbs -> Maybe ByteString -> ST s (IDecode s r)
k Maybe ByteString
mbs

lift :: ST s a -> IncrementalDecoder s a
lift :: forall s a. ST s a -> IncrementalDecoder s a
lift ST s a
action = forall s a.
(forall r. (a -> ST s (IDecode s r)) -> ST s (IDecode s r))
-> IncrementalDecoder s a
IncrementalDecoder (\a -> ST s (IDecode s r)
k -> ST s a
action forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= a -> ST s (IDecode s r)
k)

--------------------------------------------
-- The main decoder
--

-- The top level entry point
runDecodeAction :: DecodeAction s a
                -> IncrementalDecoder s (ByteString, ByteOffset, a)
runDecodeAction :: forall s a.
DecodeAction s a
-> IncrementalDecoder s (ByteString, ByteOffset, a)
runDecodeAction (D.Fail String
msg)        = forall s a.
ByteString -> ByteOffset -> String -> IncrementalDecoder s a
decodeFail ByteString
BS.empty ByteOffset
0 String
msg
runDecodeAction (D.Done a
x)          = forall (m :: * -> *) a. Monad m => a -> m a
return (ByteString
BS.empty, ByteOffset
0, a
x)
runDecodeAction (D.PeekAvailable Int# -> ST s (DecodeAction s a)
k) = forall s a. ST s a -> IncrementalDecoder s a
lift (Int# -> ST s (DecodeAction s a)
k Int#
0#) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a.
DecodeAction s a
-> IncrementalDecoder s (ByteString, ByteOffset, a)
runDecodeAction
runDecodeAction DecodeAction s a
da = do
    Maybe ByteString
mbs <- forall s. IncrementalDecoder s (Maybe ByteString)
needChunk
    case Maybe ByteString
mbs of
      Maybe ByteString
Nothing -> forall s a.
ByteString -> ByteOffset -> String -> IncrementalDecoder s a
decodeFail ByteString
BS.empty ByteOffset
0 String
"end of input"
      Just ByteString
bs -> forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow DecodeAction s a
da ByteString
bs ByteOffset
0

-- The decoder is split into a fast path and a slow path. The fast path is
-- used for a single input chunk. It decodes as far as it can, reading only
-- whole tokens that fit within the input chunk. When it cannot read any
-- further it returns control to the slow path. The slow path fixes up all the
-- complicated corner cases with tokens that span chunk boundaries, gets more
-- input and then goes back into the fast path.
--
-- The idea is that chunks are usually large, and we can use simpler and
-- faster code if we don't make it deal with the general case of tokens that
-- span chunk boundaries.

-- These are all the ways in which the fast path can finish, and return
-- control to the slow path. In particular there are three different cases
-- of tokens spanning a chunk boundary.
--
data SlowPath s a
   = FastDone                      {-# UNPACK #-} !ByteString a
   | SlowConsumeTokenBytes         {-# UNPACK #-} !ByteString (ByteString   -> ST s (DecodeAction s a)) {-# UNPACK #-} !Int
   | SlowConsumeTokenByteArray     {-# UNPACK #-} !ByteString (BA.ByteArray -> ST s (DecodeAction s a)) {-# UNPACK #-} !Int
   | SlowConsumeTokenString        {-# UNPACK #-} !ByteString (T.Text       -> ST s (DecodeAction s a)) {-# UNPACK #-} !Int
   | SlowConsumeTokenUtf8ByteArray {-# UNPACK #-} !ByteString (BA.ByteArray -> ST s (DecodeAction s a)) {-# UNPACK #-} !Int
#if defined(ARCH_32bit)
   | SlowPeekByteOffset            {-# UNPACK #-} !ByteString (Int64#       -> ST s (DecodeAction s a))
#else
   | SlowPeekByteOffset            {-# UNPACK #-} !ByteString (Int#         -> ST s (DecodeAction s a))
#endif
   | SlowDecodeAction              {-# UNPACK #-} !ByteString (DecodeAction s a)
   | SlowFail                      {-# UNPACK #-} !ByteString String


-- The main fast path. The fast path itself is actually split into two parts
-- the main version 'go_fast' and a version used when we are near the end of
-- the chunk, 'go_fast_end'.
--
-- This version can then do fewer tests when we're not near the end of the
-- chunk, in particular we just check if there's enough input buffer space
-- left for the largest possible fixed-size token (8+1 bytes).
--
go_fast :: ByteString -> DecodeAction s a -> ST s (SlowPath s a)

go_fast :: forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast !ByteString
bs DecodeAction s a
da | ByteString -> Int
BS.length ByteString
bs forall a. Ord a => a -> a -> Bool
< Int
9 = forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeWord Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (W# Word#
w#) -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeWord8 Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (W# Word#
w#) ->
        case Word# -> Word# -> Int#
gtWord# Word#
w# Word#
0xff## of
          Int#
0#                  -> Word# -> ST s (DecodeAction s a)
k Word#
w#  forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                   -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeWord16 Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (W# Word#
w#) ->
        case Word# -> Word# -> Int#
gtWord# Word#
w# Word#
0xffff## of
          Int#
0#                  -> Word# -> ST s (DecodeAction s a)
k Word#
w#  forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                   -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeWord32 Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (W# Word#
w#) ->
#if defined(ARCH_32bit)
                                 k w# >>= go_fast (BS.unsafeDrop sz bs)
#else
        case Word# -> Word# -> Int#
gtWord# Word#
w# Word#
0xffffffff## of
          Int#
0#                  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                   -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
#endif

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeNegWord Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeNegWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (W# Word#
w#) -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeInt Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (I# Int#
n#) -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeInt8 Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (I# Int#
n#) ->
        case (Int#
n# Int# -> Int# -> Int#
># Int#
0x7f#) Int# -> Int# -> Int#
`orI#` (Int#
n# Int# -> Int# -> Int#
<# Int#
-0x80#) of
          Int#
0#                  -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                   -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeInt16 Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (I# Int#
n#) ->
        case (Int#
n# Int# -> Int# -> Int#
># Int#
0x7fff#) Int# -> Int# -> Int#
`orI#` (Int#
n# Int# -> Int# -> Int#
<# Int#
-0x8000#) of
          Int#
0#                  -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                   -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeInt32 Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (I# Int#
n#) ->
#if defined(ARCH_32bit)
                                 k n# >>= go_fast (BS.unsafeDrop sz bs)
#else
        case (Int#
n# Int# -> Int# -> Int#
># Int#
0x7fffffff#) Int# -> Int# -> Int#
`orI#` (Int#
n# Int# -> Int# -> Int#
<# Int#
-0x80000000#) of
          Int#
0#                  -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                   -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
#endif

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeListLen Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeListLen (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (I# Int#
n#) -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeMapLen Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeMapLen (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (I# Int#
n#) -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeTag Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeTag (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (W# Word#
w#) -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeWordCanonical Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz w :: Word
w@(W# Word#
w#)
        | Int -> Word -> Bool
isWordCanonical Int
sz Word
w  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise             -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeWord8Canonical Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz w :: Word
w@(W# Word#
w#) ->
        case Word# -> Word# -> Int#
gtWord# Word#
w# Word#
0xff## of
          Int#
0# | Int -> Word -> Bool
isWordCanonical Int
sz Word
w  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                          -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeWord16Canonical Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz w :: Word
w@(W# Word#
w#) ->
        case Word# -> Word# -> Int#
gtWord# Word#
w# Word#
0xffff## of
          Int#
0# | Int -> Word -> Bool
isWordCanonical Int
sz Word
w  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                          -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeWord32Canonical Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz w :: Word
w@(W# Word#
w#) ->
        case Word# -> Int#
w_out_of_range Word#
w# of
          Int#
0# | Int -> Word -> Bool
isWordCanonical Int
sz Word
w  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                          -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
  where
    w_out_of_range :: Word# -> Int#
    w_out_of_range :: Word# -> Int#
w_out_of_range Word#
_w# =
#if defined(ARCH_32bit)
      0#
#else
      Word# -> Word# -> Int#
gtWord# Word#
_w# Word#
0xffffffff##
#endif

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeNegWordCanonical Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeNegWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz w :: Word
w@(W# Word#
w#)
        | Int -> Word -> Bool
isWordCanonical Int
sz Word
w  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise             -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeIntCanonical Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz n :: Int
n@(I# Int#
n#)
        | Int -> Int -> Bool
isIntCanonical Int
sz Int
n  -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise            -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeInt8Canonical Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz n :: Int
n@(I# Int#
n#) ->
        case (Int#
n# Int# -> Int# -> Int#
># Int#
0x7f#) Int# -> Int# -> Int#
`orI#` (Int#
n# Int# -> Int# -> Int#
<# Int#
-0x80#) of
          Int#
0# | Int -> Int -> Bool
isIntCanonical Int
sz Int
n  -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                         -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeInt16Canonical Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz n :: Int
n@(I# Int#
n#) ->
        case (Int#
n# Int# -> Int# -> Int#
># Int#
0x7fff#) Int# -> Int# -> Int#
`orI#` (Int#
n# Int# -> Int# -> Int#
<# Int#
-0x8000#) of
          Int#
0# | Int -> Int -> Bool
isIntCanonical Int
sz Int
n  -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                         -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeInt32Canonical Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz n :: Int
n@(I# Int#
n#) ->
        case Int# -> Int#
n_out_of_range Int#
n# of
          Int#
0# | Int -> Int -> Bool
isIntCanonical Int
sz Int
n  -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                         -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
  where
    n_out_of_range :: Int# -> Int#
    n_out_of_range :: Int# -> Int#
n_out_of_range Int#
_n# =
#if defined(ARCH_32bit)
      0#
#else
      (Int#
_n# Int# -> Int# -> Int#
># Int#
0x7fffffff#) Int# -> Int# -> Int#
`orI#` (Int#
_n# Int# -> Int# -> Int#
<# Int#
-0x80000000#)
#endif

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeListLenCanonical Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeListLen (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz n :: Int
n@(I# Int#
n#)
          -- List length can't be negative, cast it to Word.
        | Int -> Word -> Bool
isWordCanonical Int
sz (Int -> Word
intToWord Int
n)  -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise                         -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeMapLenCanonical Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeMapLen (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz n :: Int
n@(I# Int#
n#)
          -- Map length can't be negative, cast it to Word.
        | Int -> Word -> Bool
isWordCanonical Int
sz (Int -> Word
intToWord Int
n) -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise                        -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeTagCanonical Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeTag (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz w :: Word
w@(W# Word#
w#)
        | Int -> Word -> Bool
isWordCanonical Int
sz Word
w  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise             -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

#if defined(ARCH_32bit)
go_fast !bs da@(ConsumeWord64 k) =
  case tryConsumeWord64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> go_fast_end bs da
    DecodedToken sz (W64# w#) -> k w# >>= go_fast (BS.unsafeDrop sz bs)

go_fast !bs da@(ConsumeNegWord64 k) =
  case tryConsumeNegWord64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> go_fast_end bs da
    DecodedToken sz (W64# w#) -> k w# >>= go_fast (BS.unsafeDrop sz bs)

go_fast !bs da@(ConsumeInt64 k) =
  case tryConsumeInt64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> go_fast_end bs da
    DecodedToken sz (I64# i#) -> k i# >>= go_fast (BS.unsafeDrop sz bs)

go_fast !bs da@(ConsumeListLen64 k) =
  case tryConsumeListLen64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> go_fast_end bs da
    DecodedToken sz (I64# i#) -> k i# >>= go_fast (BS.unsafeDrop sz bs)

go_fast !bs da@(ConsumeMapLen64 k) =
  case tryConsumeMapLen64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> go_fast_end bs da
    DecodedToken sz (I64# i#) -> k i# >>= go_fast (BS.unsafeDrop sz bs)

go_fast !bs da@(ConsumeTag64 k) =
  case tryConsumeTag64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> go_fast_end bs da
    DecodedToken sz (W64# w#) -> k w# >>= go_fast (BS.unsafeDrop sz bs)

go_fast !bs da@(ConsumeWord64Canonical k) =
  case tryConsumeWord64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> go_fast_end bs da
    DecodedToken sz w@(W64# w#)
      | isWord64Canonical sz w  -> k w# >>= go_fast (BS.unsafeDrop sz bs)
      | otherwise               -> go_fast_end bs da

go_fast !bs da@(ConsumeNegWord64Canonical k) =
  case tryConsumeNegWord64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> go_fast_end bs da
    DecodedToken sz w@(W64# w#)
      | isWord64Canonical sz w  -> k w# >>= go_fast (BS.unsafeDrop sz bs)
      | otherwise               -> go_fast_end bs da

go_fast !bs da@(ConsumeInt64Canonical k) =
  case tryConsumeInt64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> go_fast_end bs da
    DecodedToken sz i@(I64# i#)
      | isInt64Canonical sz i  -> k i# >>= go_fast (BS.unsafeDrop sz bs)
      | otherwise              -> go_fast_end bs da

go_fast !bs da@(ConsumeListLen64Canonical k) =
  case tryConsumeListLen64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> go_fast_end bs da
    DecodedToken sz i@(I64# i#)
        -- List length can't be negative, cast it to Word64#.
      | isWord64Canonical sz (int64ToWord64 i) -> k i# >>= go_fast (BS.unsafeDrop sz bs)
      | otherwise                              -> go_fast_end bs da

go_fast !bs da@(ConsumeMapLen64Canonical k) =
  case tryConsumeMapLen64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> go_fast_end bs da
    DecodedToken sz i@(I64# i#)
        -- Map length can't be negative, cast it to Word64#.
      | isWord64Canonical sz (int64ToWord64 i) -> k i# >>= go_fast (BS.unsafeDrop sz bs)
      | otherwise                              -> go_fast_end bs da

go_fast !bs da@(ConsumeTag64Canonical k) =
  case tryConsumeTag64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> go_fast_end bs da
    DecodedToken sz w@(W64# w#)
      | isWord64Canonical sz w  -> k w# >>= go_fast (BS.unsafeDrop sz bs)
      | otherwise               -> go_fast_end bs da
#endif

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeInteger Integer -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (BigIntToken Integer)
tryConsumeInteger (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
sz (BigIntToken Bool
_ Integer
n) -> Integer -> ST s (DecodeAction s a)
k Integer
n forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
      DecodedToken (BigIntToken Integer)
_                                 -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeFloat Float# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Float
tryConsumeFloat (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Float
DecodeFailure     -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (F# Float#
f#) -> Float# -> ST s (DecodeAction s a)
k Float#
f# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeDouble Double# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Double
tryConsumeDouble (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Double
DecodeFailure     -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (D# Double#
f#) -> Double# -> ST s (DecodeAction s a)
k Double#
f# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeBytes ByteString -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeBytes (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken (LongToken ByteString)
DecodeFailure                   -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (Fits Bool
_ ByteString
bstr)   -> ByteString -> ST s (DecodeAction s a)
k ByteString
bstr forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
      DecodedToken Int
sz (TooLong Bool
_ Int
len) -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteString -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenBytes
                                                   (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) ByteString -> ST s (DecodeAction s a)
k Int
len

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeByteArray ByteArray -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeBytes (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken (LongToken ByteString)
DecodeFailure                 -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (Fits Bool
_ ByteString
str)  -> ByteArray -> ST s (DecodeAction s a)
k (ByteString -> ByteArray
BA.fromByteString ByteString
str) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
      DecodedToken Int
sz (TooLong Bool
_ Int
len) -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteArray -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenByteArray
                                                   (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) ByteArray -> ST s (DecodeAction s a)
k Int
len

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeString Text -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeString (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken (LongToken ByteString)
DecodeFailure                   -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (Fits Bool
_ ByteString
str)    -> case ByteString -> Either UnicodeException Text
T.decodeUtf8' ByteString
str of
        Right Text
t -> Text -> ST s (DecodeAction s a)
k Text
t forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        Left UnicodeException
_e -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"invalid UTF8"
      DecodedToken Int
sz (TooLong Bool
_ Int
len) -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (Text -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenString
                                                   (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) Text -> ST s (DecodeAction s a)
k Int
len

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeUtf8ByteArray ByteArray -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeString (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken (LongToken ByteString)
DecodeFailure                   -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (Fits Bool
_ ByteString
str)    -> ByteArray -> ST s (DecodeAction s a)
k (ByteString -> ByteArray
BA.fromByteString ByteString
str)
                                         forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
      DecodedToken Int
sz (TooLong Bool
_ Int
len) -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteArray -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenUtf8ByteArray
                                                   (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) ByteArray -> ST s (DecodeAction s a)
k Int
len

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeBool Bool -> ST s (DecodeAction s a)
k) =
    case Word8 -> DecodedToken Bool
tryConsumeBool (ByteString -> Word8
BS.unsafeHead ByteString
bs) of
      DecodedToken Bool
DecodeFailure     -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz Bool
b -> Bool -> ST s (DecodeAction s a)
k Bool
b forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeSimple Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeSimple (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (W# Word#
w#) -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeIntegerCanonical Integer -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (BigIntToken Integer)
tryConsumeInteger (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
sz (BigIntToken Bool
True Integer
n) -> Integer -> ST s (DecodeAction s a)
k Integer
n forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
      DecodedToken (BigIntToken Integer)
_                                    -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da


go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeFloat16Canonical Float# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Float
tryConsumeFloat (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Float
DecodeFailure     -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz f :: Float
f@(F# Float#
f#)
        | Int -> ByteString -> Float -> Bool
isFloat16Canonical Int
sz ByteString
bs Float
f -> Float# -> ST s (DecodeAction s a)
k Float#
f# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise                  -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeFloatCanonical Float# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Float
tryConsumeFloat (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Float
DecodeFailure     -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz f :: Float
f@(F# Float#
f#)
        | Int -> ByteString -> Float -> Bool
isFloatCanonical Int
sz ByteString
bs Float
f -> Float# -> ST s (DecodeAction s a)
k Float#
f# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise                -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeDoubleCanonical Double# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Double
tryConsumeDouble (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Double
DecodeFailure     -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz f :: Double
f@(D# Double#
f#)
        | Int -> ByteString -> Double -> Bool
isDoubleCanonical Int
sz ByteString
bs Double
f -> Double# -> ST s (DecodeAction s a)
k Double#
f# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise                 -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeBytesCanonical ByteString -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeBytes (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
sz (Fits    Bool
True ByteString
bstr) -> ByteString -> ST s (DecodeAction s a)
k ByteString
bstr forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
      DecodedToken Int
sz (TooLong Bool
True Int
len)  ->
        forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteString -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenBytes (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) ByteString -> ST s (DecodeAction s a)
k Int
len
      DecodedToken (LongToken ByteString)
_                                   -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeByteArrayCanonical ByteArray -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeBytes (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
sz (Fits Bool
True ByteString
str)    ->
        ByteArray -> ST s (DecodeAction s a)
k (ByteString -> ByteArray
BA.fromByteString ByteString
str) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
      DecodedToken Int
sz (TooLong Bool
True Int
len) ->
        forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteArray -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenByteArray (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) ByteArray -> ST s (DecodeAction s a)
k Int
len
      DecodedToken (LongToken ByteString)
_                                  -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeStringCanonical Text -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeString (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
sz (Fits Bool
True ByteString
str)    -> case ByteString -> Either UnicodeException Text
T.decodeUtf8' ByteString
str of
        Right Text
t -> Text -> ST s (DecodeAction s a)
k Text
t forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        Left UnicodeException
_e -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"invalid UTF8"
      DecodedToken Int
sz (TooLong Bool
True Int
len) ->
        forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (Text -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenString (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) Text -> ST s (DecodeAction s a)
k Int
len
      DecodedToken (LongToken ByteString)
_                                  -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeUtf8ByteArrayCanonical ByteArray -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeString (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
sz (Fits Bool
True ByteString
str)    ->
        ByteArray -> ST s (DecodeAction s a)
k (ByteString -> ByteArray
BA.fromByteString ByteString
str) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
      DecodedToken Int
sz (TooLong Bool
True Int
len) ->
        forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteArray -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenUtf8ByteArray (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) ByteArray -> ST s (DecodeAction s a)
k Int
len
      DecodedToken (LongToken ByteString)
_                                  -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeSimpleCanonical Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeSimple (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (W# Word#
w#)
        | Int -> Word# -> Bool
isSimpleCanonical Int
sz Word#
w# -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise               -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeBytesIndef ST s (DecodeAction s a)
k) =
    case Word8 -> DecodedToken ()
tryConsumeBytesIndef (ByteString -> Word8
BS.unsafeHead ByteString
bs) of
      DecodedToken ()
DecodeFailure     -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz ()
_ -> ST s (DecodeAction s a)
k forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeStringIndef ST s (DecodeAction s a)
k) =
    case Word8 -> DecodedToken ()
tryConsumeStringIndef (ByteString -> Word8
BS.unsafeHead ByteString
bs) of
      DecodedToken ()
DecodeFailure     -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz ()
_ -> ST s (DecodeAction s a)
k forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeListLenIndef ST s (DecodeAction s a)
k) =
    case Word8 -> DecodedToken ()
tryConsumeListLenIndef (ByteString -> Word8
BS.unsafeHead ByteString
bs) of
      DecodedToken ()
DecodeFailure     -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz ()
_ -> ST s (DecodeAction s a)
k forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeMapLenIndef ST s (DecodeAction s a)
k) =
    case Word8 -> DecodedToken ()
tryConsumeMapLenIndef (ByteString -> Word8
BS.unsafeHead ByteString
bs) of
      DecodedToken ()
DecodeFailure     -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz ()
_ -> ST s (DecodeAction s a)
k forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeNull ST s (DecodeAction s a)
k) =
    case Word8 -> DecodedToken ()
tryConsumeNull (ByteString -> Word8
BS.unsafeHead ByteString
bs) of
      DecodedToken ()
DecodeFailure     -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz ()
_ -> ST s (DecodeAction s a)
k forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeListLenOrIndef Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeListLenOrIndef (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (I# Int#
n#) -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs da :: DecodeAction s a
da@(ConsumeMapLenOrIndef Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeMapLenOrIndef (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
      DecodedToken Int
sz (I# Int#
n#) -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs (ConsumeBreakOr Bool -> ST s (DecodeAction s a)
k) =
    case Word8 -> DecodedToken ()
tryConsumeBreakOr (ByteString -> Word8
BS.unsafeHead ByteString
bs) of
      DecodedToken ()
DecodeFailure     -> Bool -> ST s (DecodeAction s a)
k Bool
False forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast ByteString
bs
      DecodedToken Int
sz ()
_ -> Bool -> ST s (DecodeAction s a)
k Bool
True forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast !ByteString
bs (PeekTokenType TokenType -> ST s (DecodeAction s a)
k) =
    let !hdr :: Word8
hdr  = ByteString -> Word8
BS.unsafeHead ByteString
bs
        !tkty :: TokenType
tkty = Array Word8 TokenType
decodeTokenTypeTable forall (a :: * -> * -> *) e i.
(IArray a e, Ix i) =>
a i e -> Int -> e
`A.unsafeAt` Word8 -> Int
word8ToInt Word8
hdr
    in TokenType -> ST s (DecodeAction s a)
k TokenType
tkty forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast ByteString
bs

go_fast !ByteString
bs (PeekAvailable Int# -> ST s (DecodeAction s a)
k) = Int# -> ST s (DecodeAction s a)
k (case ByteString -> Int
BS.length ByteString
bs of I# Int#
len# -> Int#
len#) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast ByteString
bs

go_fast !ByteString
bs da :: DecodeAction s a
da@PeekByteOffset{} = forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
go_fast !ByteString
bs da :: DecodeAction s a
da@D.Fail{} = forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da
go_fast !ByteString
bs da :: DecodeAction s a
da@D.Done{} = forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs DecodeAction s a
da


-- This variant of the fast path has to do a few more checks because we're
-- near the end of the chunk. The guarantee we provide here is that we will
-- decode any tokens where the whole token fits within the input buffer. So
-- if we return with input buffer space still unconsumed (and we're not done
-- or failed) then there's one remaining token that spans the end of the
-- input chunk (the slow path fixup code relies on this guarantee).
--
go_fast_end :: ByteString -> DecodeAction s a -> ST s (SlowPath s a)

-- these three cases don't need any input

go_fast_end :: forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end !ByteString
bs (D.Fail String
msg)      = forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
msg
go_fast_end !ByteString
bs (D.Done a
x)        = forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> a -> SlowPath s a
FastDone ByteString
bs a
x
go_fast_end !ByteString
bs (PeekAvailable Int# -> ST s (DecodeAction s a)
k) = Int# -> ST s (DecodeAction s a)
k (case ByteString -> Int
BS.length ByteString
bs of I# Int#
len# -> Int#
len#) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs

go_fast_end !ByteString
bs (PeekByteOffset Int# -> ST s (DecodeAction s a)
k) = forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString -> (Int# -> ST s (DecodeAction s a)) -> SlowPath s a
SlowPeekByteOffset ByteString
bs Int# -> ST s (DecodeAction s a)
k

-- the next two cases only need the 1 byte token header
go_fast_end !ByteString
bs DecodeAction s a
da | ByteString -> Bool
BS.null ByteString
bs = forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> DecodeAction s a -> SlowPath s a
SlowDecodeAction ByteString
bs DecodeAction s a
da

go_fast_end !ByteString
bs (ConsumeBreakOr Bool -> ST s (DecodeAction s a)
k) =
    case Word8 -> DecodedToken ()
tryConsumeBreakOr (ByteString -> Word8
BS.unsafeHead ByteString
bs) of
      DecodedToken ()
DecodeFailure     -> Bool -> ST s (DecodeAction s a)
k Bool
False forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs
      DecodedToken Int
sz ()
_ -> Bool -> ST s (DecodeAction s a)
k Bool
True  forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (PeekTokenType TokenType -> ST s (DecodeAction s a)
k) =
    let !hdr :: Word8
hdr  = ByteString -> Word8
BS.unsafeHead ByteString
bs
        !tkty :: TokenType
tkty = Array Word8 TokenType
decodeTokenTypeTable forall (a :: * -> * -> *) e i.
(IArray a e, Ix i) =>
a i e -> Int -> e
`A.unsafeAt` Word8 -> Int
word8ToInt Word8
hdr
    in TokenType -> ST s (DecodeAction s a)
k TokenType
tkty forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs

-- all the remaining cases have to decode the current token

go_fast_end !ByteString
bs DecodeAction s a
da
    | let !hdr :: Word8
hdr = ByteString -> Word8
BS.unsafeHead ByteString
bs
    , ByteString -> Int
BS.length ByteString
bs forall a. Ord a => a -> a -> Bool
< Word8 -> Int
tokenSize Word8
hdr
    = forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> DecodeAction s a -> SlowPath s a
SlowDecodeAction ByteString
bs DecodeAction s a
da

go_fast_end !ByteString
bs (ConsumeWord Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected word"
      DecodedToken Int
sz (W# Word#
w#) -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeWord8 Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected word8"
      DecodedToken Int
sz (W# Word#
w#) ->
        case Word# -> Word# -> Int#
gtWord# Word#
w# Word#
0xff## of
          Int#
0#                  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                   -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected word8"

go_fast_end !ByteString
bs (ConsumeWord16 Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected word16"
      DecodedToken Int
sz (W# Word#
w#) ->
        case Word# -> Word# -> Int#
gtWord# Word#
w# Word#
0xffff## of
          Int#
0#                  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                   -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected word16"

go_fast_end !ByteString
bs (ConsumeWord32 Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected word32"
      DecodedToken Int
sz (W# Word#
w#) ->
#if defined(ARCH_32bit)
                                 k w# >>= go_fast_end (BS.unsafeDrop sz bs)
#else
        case Word# -> Word# -> Int#
gtWord# Word#
w# Word#
0xffffffff## of
          Int#
0#                  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                   -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected word32"
#endif

go_fast_end !ByteString
bs (ConsumeNegWord Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeNegWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected negative int"
      DecodedToken Int
sz (W# Word#
w#) -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeInt Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected int"
      DecodedToken Int
sz (I# Int#
n#) -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeInt8 Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected int8"
      DecodedToken Int
sz (I# Int#
n#) ->
        case (Int#
n# Int# -> Int# -> Int#
># Int#
0x7f#) Int# -> Int# -> Int#
`orI#` (Int#
n# Int# -> Int# -> Int#
<# Int#
-0x80#) of
          Int#
0#                  -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                   -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected int8"

go_fast_end !ByteString
bs (ConsumeInt16 Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected int16"
      DecodedToken Int
sz (I# Int#
n#) ->
        case (Int#
n# Int# -> Int# -> Int#
># Int#
0x7fff#) Int# -> Int# -> Int#
`orI#` (Int#
n# Int# -> Int# -> Int#
<# Int#
-0x8000#) of
          Int#
0#                  -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                   -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected int16"

go_fast_end !ByteString
bs (ConsumeInt32 Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected int32"
      DecodedToken Int
sz (I# Int#
n#) ->
#if defined(ARCH_32bit)
                                 k n# >>= go_fast_end (BS.unsafeDrop sz bs)
#else
        case (Int#
n# Int# -> Int# -> Int#
># Int#
0x7fffffff#) Int# -> Int# -> Int#
`orI#` (Int#
n# Int# -> Int# -> Int#
<# Int#
-0x80000000#) of
          Int#
0#                  -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Int#
_                   -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected int32"
#endif

go_fast_end !ByteString
bs (ConsumeListLen Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeListLen (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected list len"
      DecodedToken Int
sz (I# Int#
n#) -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeMapLen Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeMapLen (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected map len"
      DecodedToken Int
sz (I# Int#
n#) -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeTag Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeTag (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected tag"
      DecodedToken Int
sz (W# Word#
w#) -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeWordCanonical Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected word"
      DecodedToken Int
sz w :: Word
w@(W# Word#
w#)
        | Int -> Word -> Bool
isWordCanonical Int
sz Word
w  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise             -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical word"

go_fast_end !ByteString
bs (ConsumeWord8Canonical Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected word8"
      DecodedToken Int
sz w :: Word
w@(W# Word#
w#) -> case Word# -> Word# -> Int#
gtWord# Word#
w# Word#
0xff## of
          Int#
0# | Int -> Word -> Bool
isWordCanonical Int
sz Word
w  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
             | Bool
otherwise             -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical word8"
          Int#
_                          -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected word8"

go_fast_end !ByteString
bs (ConsumeWord16Canonical Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected word16"
      DecodedToken Int
sz w :: Word
w@(W# Word#
w#) -> case Word# -> Word# -> Int#
gtWord# Word#
w# Word#
0xffff## of
        Int#
0# | Int -> Word -> Bool
isWordCanonical Int
sz Word
w  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
           | Bool
otherwise             -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical word16"
        Int#
_                          -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected word16"

go_fast_end !ByteString
bs (ConsumeWord32Canonical Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected word32"
      DecodedToken Int
sz w :: Word
w@(W# Word#
w#) -> case Word# -> Int#
w_out_of_range Word#
w# of
        Int#
0# | Int -> Word -> Bool
isWordCanonical Int
sz Word
w  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
           | Bool
otherwise             -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical word32"
        Int#
_                          -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected word32"
  where
    w_out_of_range :: Word# -> Int#
    w_out_of_range :: Word# -> Int#
w_out_of_range Word#
_w# =
#if defined(ARCH_32bit)
      0#
#else
      Word# -> Word# -> Int#
gtWord# Word#
_w# Word#
0xffffffff##
#endif

go_fast_end !ByteString
bs (ConsumeNegWordCanonical Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeNegWord (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected negative int"
      DecodedToken Int
sz w :: Word
w@(W# Word#
w#)
        | Int -> Word -> Bool
isWordCanonical Int
sz Word
w  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise             -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical negative int"

go_fast_end !ByteString
bs (ConsumeIntCanonical Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected int"
      DecodedToken Int
sz n :: Int
n@(I# Int#
n#)
        | Int -> Int -> Bool
isIntCanonical Int
sz Int
n  -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise            -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical int"

go_fast_end !ByteString
bs (ConsumeInt8Canonical Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected int8"
      DecodedToken Int
sz n :: Int
n@(I# Int#
n#) ->
        case (Int#
n# Int# -> Int# -> Int#
># Int#
0x7f#) Int# -> Int# -> Int#
`orI#` (Int#
n# Int# -> Int# -> Int#
<# Int#
-0x80#) of
          Int#
0# | Int -> Int -> Bool
isIntCanonical Int
sz Int
n  -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
             | Bool
otherwise            -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical int8"
          Int#
_                         -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected int8"

go_fast_end !ByteString
bs (ConsumeInt16Canonical Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected int16"
      DecodedToken Int
sz n :: Int
n@(I# Int#
n#) ->
        case (Int#
n# Int# -> Int# -> Int#
># Int#
0x7fff#) Int# -> Int# -> Int#
`orI#` (Int#
n# Int# -> Int# -> Int#
<# Int#
-0x8000#) of
          Int#
0# | Int -> Int -> Bool
isIntCanonical Int
sz Int
n  -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
             | Bool
otherwise            -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical int16"
          Int#
_                         -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected int16"

go_fast_end !ByteString
bs (ConsumeInt32Canonical Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeInt (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected int32"
      DecodedToken Int
sz n :: Int
n@(I# Int#
n#) ->
        case Int# -> Int#
n_out_of_range Int#
n# of
          Int#
0# | Int -> Int -> Bool
isIntCanonical Int
sz Int
n  -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
             | Bool
otherwise            -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical int32"
          Int#
_                         -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected int32"
  where
    n_out_of_range :: Int# -> Int#
    n_out_of_range :: Int# -> Int#
n_out_of_range Int#
_n# =
#if defined(ARCH_32bit)
      0#
#else
      (Int#
_n# Int# -> Int# -> Int#
># Int#
0x7fffffff#) Int# -> Int# -> Int#
`orI#` (Int#
_n# Int# -> Int# -> Int#
<# Int#
-0x80000000#)
#endif

go_fast_end !ByteString
bs (ConsumeListLenCanonical Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeListLen (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected list len"
      DecodedToken Int
sz n :: Int
n@(I# Int#
n#)
          -- List length can't be negative, cast it to Word#.
        | Int -> Word -> Bool
isWordCanonical Int
sz (Int -> Word
intToWord Int
n) -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise                        -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical list len"

go_fast_end !ByteString
bs (ConsumeMapLenCanonical Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeMapLen (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected map len"
      DecodedToken Int
sz n :: Int
n@(I# Int#
n#)
          -- Map length can't be negative, cast it to Word#.
        | Int -> Word -> Bool
isWordCanonical Int
sz (Int -> Word
intToWord Int
n) -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise                        -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical map len"

go_fast_end !ByteString
bs (ConsumeTagCanonical Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeTag (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected tag"
      DecodedToken Int
sz w :: Word
w@(W# Word#
w#)
        | Int -> Word -> Bool
isWordCanonical Int
sz Word
w  -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise             -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical tag"

#if defined(ARCH_32bit)
go_fast_end !bs (ConsumeWord64 k) =
  case tryConsumeWord64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> return $! SlowFail bs "expected word64"
    DecodedToken sz (W64# w#) -> k w# >>= go_fast_end (BS.unsafeDrop sz bs)

go_fast_end !bs (ConsumeNegWord64 k) =
  case tryConsumeNegWord64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> return $! SlowFail bs "expected negative int"
    DecodedToken sz (W64# w#) -> k w# >>= go_fast_end (BS.unsafeDrop sz bs)

go_fast_end !bs (ConsumeInt64 k) =
  case tryConsumeInt64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> return $! SlowFail bs "expected int64"
    DecodedToken sz (I64# i#) -> k i# >>= go_fast_end (BS.unsafeDrop sz bs)

go_fast_end !bs (ConsumeListLen64 k) =
  case tryConsumeListLen64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> return $! SlowFail bs "expected list len 64"
    DecodedToken sz (I64# i#) -> k i# >>= go_fast_end (BS.unsafeDrop sz bs)

go_fast_end !bs (ConsumeMapLen64 k) =
  case tryConsumeMapLen64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> return $! SlowFail bs "expected map len 64"
    DecodedToken sz (I64# i#) -> k i# >>= go_fast_end (BS.unsafeDrop sz bs)

go_fast_end !bs (ConsumeTag64 k) =
  case tryConsumeTag64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> return $! SlowFail bs "expected tag64"
    DecodedToken sz (W64# w#) -> k w# >>= go_fast_end (BS.unsafeDrop sz bs)

go_fast_end !bs (ConsumeWord64Canonical k) =
  case tryConsumeWord64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> return $! SlowFail bs "expected word64"
    DecodedToken sz w@(W64# w#)
      | isWord64Canonical sz w  -> k w# >>= go_fast_end (BS.unsafeDrop sz bs)
      | otherwise               -> return $! SlowFail bs "non-canonical word64"

go_fast_end !bs (ConsumeNegWord64Canonical k) =
  case tryConsumeNegWord64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> return $! SlowFail bs "expected negative int"
    DecodedToken sz w@(W64# w#)
      | isWord64Canonical sz w  -> k w# >>= go_fast_end (BS.unsafeDrop sz bs)
      | otherwise               -> return $! SlowFail bs "non-canonical negative int"

go_fast_end !bs (ConsumeInt64Canonical k) =
  case tryConsumeInt64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> return $! SlowFail bs "expected int64"
    DecodedToken sz i@(I64# i#)
      | isInt64Canonical sz i  -> k i# >>= go_fast_end (BS.unsafeDrop sz bs)
      | otherwise              -> return $! SlowFail bs "non-canonical int64"

go_fast_end !bs (ConsumeListLen64Canonical k) =
  case tryConsumeListLen64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> return $! SlowFail bs "expected list len 64"
    DecodedToken sz i@(I64# i#)
        -- List length can't be negative, cast it to Word64#.
      | isWord64Canonical sz (int64ToWord64 i) ->
          k i# >>= go_fast_end (BS.unsafeDrop sz bs)
      | otherwise ->
          return $! SlowFail bs "non-canonical list len 64"

go_fast_end !bs (ConsumeMapLen64Canonical k) =
  case tryConsumeMapLen64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> return $! SlowFail bs "expected map len 64"
    DecodedToken sz i@(I64# i#)
        -- Map length can't be negative, cast it to Word64#.
      | isWord64Canonical sz (int64ToWord64 i) ->
          k i# >>= go_fast_end (BS.unsafeDrop sz bs)
      | otherwise ->
          return $! SlowFail bs "non-canonical map len 64"

go_fast_end !bs (ConsumeTag64Canonical k) =
  case tryConsumeTag64 (BS.unsafeHead bs) bs of
    DecodeFailure             -> return $! SlowFail bs "expected tag64"
    DecodedToken sz w@(W64# w#)
      | isWord64Canonical sz w  -> k w# >>= go_fast_end (BS.unsafeDrop sz bs)
      | otherwise               -> return $! SlowFail bs "non-canonical tag64"

#endif

go_fast_end !ByteString
bs (ConsumeInteger Integer -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (BigIntToken Integer)
tryConsumeInteger (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken (BigIntToken Integer)
DecodeFailure                         -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected integer"
      DecodedToken Int
sz (BigIntToken Bool
_ Integer
n)     -> Integer -> ST s (DecodeAction s a)
k Integer
n forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
      DecodedToken Int
sz (BigUIntNeedBody Bool
_ Int
len) -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteString -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenBytes (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) (forall s a.
(Integer -> ST s (DecodeAction s a))
-> ByteString -> ST s (DecodeAction s a)
adjustContBigUIntNeedBody Integer -> ST s (DecodeAction s a)
k) Int
len
      DecodedToken Int
sz (BigNIntNeedBody Bool
_ Int
len) -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteString -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenBytes (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) (forall s a.
(Integer -> ST s (DecodeAction s a))
-> ByteString -> ST s (DecodeAction s a)
adjustContBigNIntNeedBody Integer -> ST s (DecodeAction s a)
k) Int
len
      DecodedToken Int
sz  BigIntToken Integer
BigUIntNeedHeader    -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> DecodeAction s a -> SlowPath s a
SlowDecodeAction      (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) (forall s a.
(Integer -> ST s (DecodeAction s a)) -> DecodeAction s a
adjustContBigUIntNeedHeader Integer -> ST s (DecodeAction s a)
k)
      DecodedToken Int
sz  BigIntToken Integer
BigNIntNeedHeader    -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> DecodeAction s a -> SlowPath s a
SlowDecodeAction      (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) (forall s a.
(Integer -> ST s (DecodeAction s a)) -> DecodeAction s a
adjustContBigNIntNeedHeader Integer -> ST s (DecodeAction s a)
k)

go_fast_end !ByteString
bs (ConsumeFloat Float# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Float
tryConsumeFloat (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Float
DecodeFailure     -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected float"
      DecodedToken Int
sz (F# Float#
f#) -> Float# -> ST s (DecodeAction s a)
k Float#
f# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeDouble Double# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Double
tryConsumeDouble (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Double
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected double"
      DecodedToken Int
sz (D# Double#
f#) -> Double# -> ST s (DecodeAction s a)
k Double#
f# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeBytes ByteString -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeBytes (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken (LongToken ByteString)
DecodeFailure                   -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected bytes"
      DecodedToken Int
sz (Fits Bool
_ ByteString
bstr)   -> ByteString -> ST s (DecodeAction s a)
k ByteString
bstr forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
      DecodedToken Int
sz (TooLong Bool
_ Int
len) -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteString -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenBytes
                                                   (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) ByteString -> ST s (DecodeAction s a)
k Int
len

go_fast_end !ByteString
bs (ConsumeByteArray ByteArray -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeBytes (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken (LongToken ByteString)
DecodeFailure                   -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected string"
      DecodedToken Int
sz (Fits Bool
_ ByteString
str)    -> (ByteArray -> ST s (DecodeAction s a)
k forall a b. (a -> b) -> a -> b
$! ByteString -> ByteArray
BA.fromByteString ByteString
str)
                                         forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
      DecodedToken Int
sz (TooLong Bool
_ Int
len) -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteArray -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenByteArray
                                                   (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) ByteArray -> ST s (DecodeAction s a)
k Int
len

go_fast_end !ByteString
bs (ConsumeString Text -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeString (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken (LongToken ByteString)
DecodeFailure                   -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected string"
      DecodedToken Int
sz (Fits Bool
_ ByteString
str)    -> case ByteString -> Either UnicodeException Text
T.decodeUtf8' ByteString
str of
        Right Text
t -> Text -> ST s (DecodeAction s a)
k Text
t forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        Left UnicodeException
_e -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"invalid UTF8"
      DecodedToken Int
sz (TooLong Bool
_ Int
len) -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (Text -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenString
                                                   (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) Text -> ST s (DecodeAction s a)
k Int
len

go_fast_end !ByteString
bs (ConsumeUtf8ByteArray ByteArray -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeString (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken (LongToken ByteString)
DecodeFailure                   -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected string"
      DecodedToken Int
sz (Fits Bool
_ ByteString
str)    -> (ByteArray -> ST s (DecodeAction s a)
k forall a b. (a -> b) -> a -> b
$! ByteString -> ByteArray
BA.fromByteString ByteString
str)
                                         forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
      DecodedToken Int
sz (TooLong Bool
_ Int
len) -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteArray -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenUtf8ByteArray
                                                   (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) ByteArray -> ST s (DecodeAction s a)
k Int
len

go_fast_end !ByteString
bs (ConsumeBool Bool -> ST s (DecodeAction s a)
k) =
    case Word8 -> DecodedToken Bool
tryConsumeBool (ByteString -> Word8
BS.unsafeHead ByteString
bs) of
      DecodedToken Bool
DecodeFailure     -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected bool"
      DecodedToken Int
sz Bool
b -> Bool -> ST s (DecodeAction s a)
k Bool
b forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeSimple Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeSimple (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected simple"
      DecodedToken Int
sz (W# Word#
w#) -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeIntegerCanonical Integer -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (BigIntToken Integer)
tryConsumeInteger (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken (BigIntToken Integer)
DecodeFailure                         -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected integer"
      DecodedToken Int
sz (BigIntToken Bool
True Integer
n)  -> Integer -> ST s (DecodeAction s a)
k Integer
n forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
      DecodedToken Int
sz (BigUIntNeedBody Bool
True Int
len) -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteString -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenBytes
        (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) (forall s a.
(Integer -> ST s (DecodeAction s a))
-> ByteString -> ST s (DecodeAction s a)
adjustContCanonicalBigUIntNeedBody Integer -> ST s (DecodeAction s a)
k) Int
len
      DecodedToken Int
sz (BigNIntNeedBody Bool
True Int
len) -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteString -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenBytes
        (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) (forall s a.
(Integer -> ST s (DecodeAction s a))
-> ByteString -> ST s (DecodeAction s a)
adjustContCanonicalBigNIntNeedBody Integer -> ST s (DecodeAction s a)
k) Int
len
      DecodedToken Int
sz  BigIntToken Integer
BigUIntNeedHeader -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> DecodeAction s a -> SlowPath s a
SlowDecodeAction
        (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) (forall s a.
(Integer -> ST s (DecodeAction s a)) -> DecodeAction s a
adjustContCanonicalBigUIntNeedHeader Integer -> ST s (DecodeAction s a)
k)
      DecodedToken Int
sz  BigIntToken Integer
BigNIntNeedHeader -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> DecodeAction s a -> SlowPath s a
SlowDecodeAction
        (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) (forall s a.
(Integer -> ST s (DecodeAction s a)) -> DecodeAction s a
adjustContCanonicalBigNIntNeedHeader Integer -> ST s (DecodeAction s a)
k)
      DecodedToken (BigIntToken Integer)
_ -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical integer"

go_fast_end !ByteString
bs (ConsumeFloat16Canonical Float# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Float
tryConsumeFloat (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Float
DecodeFailure     -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected float"
      DecodedToken Int
sz f :: Float
f@(F# Float#
f#)
        | Int -> ByteString -> Float -> Bool
isFloat16Canonical Int
sz ByteString
bs Float
f -> Float# -> ST s (DecodeAction s a)
k Float#
f# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise                  -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical float16"

go_fast_end !ByteString
bs (ConsumeFloatCanonical Float# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Float
tryConsumeFloat (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Float
DecodeFailure     -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected float"
      DecodedToken Int
sz f :: Float
f@(F# Float#
f#)
        | Int -> ByteString -> Float -> Bool
isFloatCanonical Int
sz ByteString
bs Float
f -> Float# -> ST s (DecodeAction s a)
k Float#
f# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise                -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical float"

go_fast_end !ByteString
bs (ConsumeDoubleCanonical Double# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Double
tryConsumeDouble (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Double
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected double"
      DecodedToken Int
sz f :: Double
f@(D# Double#
f#)
        | Int -> ByteString -> Double -> Bool
isDoubleCanonical Int
sz ByteString
bs Double
f -> Double# -> ST s (DecodeAction s a)
k Double#
f# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise                 -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical double"

go_fast_end !ByteString
bs (ConsumeBytesCanonical ByteString -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeBytes (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken (LongToken ByteString)
DecodeFailure         -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected bytes"
      DecodedToken Int
sz LongToken ByteString
token -> case LongToken ByteString
token of
        Fits Bool
True ByteString
bstr   -> ByteString -> ST s (DecodeAction s a)
k ByteString
bstr forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        TooLong Bool
True Int
len -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteString -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenBytes (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) ByteString -> ST s (DecodeAction s a)
k Int
len
        LongToken ByteString
_                -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical length prefix"

go_fast_end !ByteString
bs (ConsumeByteArrayCanonical ByteArray -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeBytes (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken (LongToken ByteString)
DecodeFailure         -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected string"
      DecodedToken Int
sz LongToken ByteString
token -> case LongToken ByteString
token of
        Fits Bool
True ByteString
str    ->
          (ByteArray -> ST s (DecodeAction s a)
k forall a b. (a -> b) -> a -> b
$! ByteString -> ByteArray
BA.fromByteString ByteString
str) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        TooLong Bool
True Int
len ->
           forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteArray -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenByteArray (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) ByteArray -> ST s (DecodeAction s a)
k Int
len
        LongToken ByteString
_                -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical length prefix"

go_fast_end !ByteString
bs (ConsumeStringCanonical Text -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeString (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken (LongToken ByteString)
DecodeFailure         -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected string"
      DecodedToken Int
sz LongToken ByteString
token -> case LongToken ByteString
token of
        Fits Bool
True ByteString
str    -> case ByteString -> Either UnicodeException Text
T.decodeUtf8' ByteString
str of
          Right Text
t -> Text -> ST s (DecodeAction s a)
k Text
t forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
          Left UnicodeException
_e -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"invalid UTF8"
        TooLong Bool
True Int
len -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (Text -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenString (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) Text -> ST s (DecodeAction s a)
k Int
len
        LongToken ByteString
_                -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical length prefix"

go_fast_end !ByteString
bs (ConsumeUtf8ByteArrayCanonical ByteArray -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeString (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken (LongToken ByteString)
DecodeFailure                 -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected string"
      DecodedToken Int
sz LongToken ByteString
token -> case LongToken ByteString
token of
        Fits Bool
True ByteString
str    ->
          (ByteArray -> ST s (DecodeAction s a)
k forall a b. (a -> b) -> a -> b
$! ByteString -> ByteArray
BA.fromByteString ByteString
str) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        TooLong Bool
True Int
len ->
          forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a.
ByteString
-> (ByteArray -> ST s (DecodeAction s a)) -> Int -> SlowPath s a
SlowConsumeTokenUtf8ByteArray (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs) ByteArray -> ST s (DecodeAction s a)
k Int
len
        LongToken ByteString
_                ->
          forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical length prefix"

go_fast_end !ByteString
bs (ConsumeSimpleCanonical Word# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Word
tryConsumeSimple (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Word
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected simple"
      DecodedToken Int
sz (W# Word#
w#)
        | Int -> Word# -> Bool
isSimpleCanonical Int
sz Word#
w# -> Word# -> ST s (DecodeAction s a)
k Word#
w# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)
        | Bool
otherwise               -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"non-canonical simple"

go_fast_end !ByteString
bs (ConsumeBytesIndef ST s (DecodeAction s a)
k) =
    case Word8 -> DecodedToken ()
tryConsumeBytesIndef (ByteString -> Word8
BS.unsafeHead ByteString
bs) of
      DecodedToken ()
DecodeFailure     -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected bytes start"
      DecodedToken Int
sz ()
_ -> ST s (DecodeAction s a)
k forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeStringIndef ST s (DecodeAction s a)
k) =
    case Word8 -> DecodedToken ()
tryConsumeStringIndef (ByteString -> Word8
BS.unsafeHead ByteString
bs) of
      DecodedToken ()
DecodeFailure     -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected string start"
      DecodedToken Int
sz ()
_ -> ST s (DecodeAction s a)
k forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeListLenIndef ST s (DecodeAction s a)
k) =
    case Word8 -> DecodedToken ()
tryConsumeListLenIndef (ByteString -> Word8
BS.unsafeHead ByteString
bs) of
      DecodedToken ()
DecodeFailure     -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected list start"
      DecodedToken Int
sz ()
_ -> ST s (DecodeAction s a)
k forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeMapLenIndef ST s (DecodeAction s a)
k) =
    case Word8 -> DecodedToken ()
tryConsumeMapLenIndef (ByteString -> Word8
BS.unsafeHead ByteString
bs) of
      DecodedToken ()
DecodeFailure     -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected map start"
      DecodedToken Int
sz ()
_ -> ST s (DecodeAction s a)
k forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeNull ST s (DecodeAction s a)
k) =
    case Word8 -> DecodedToken ()
tryConsumeNull (ByteString -> Word8
BS.unsafeHead ByteString
bs) of
      DecodedToken ()
DecodeFailure     -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected null"
      DecodedToken Int
sz ()
_ -> ST s (DecodeAction s a)
k forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeListLenOrIndef Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeListLenOrIndef (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected list len or indef"
      DecodedToken Int
sz (I# Int#
n#) -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)

go_fast_end !ByteString
bs (ConsumeMapLenOrIndef Int# -> ST s (DecodeAction s a)
k) =
    case Word8 -> ByteString -> DecodedToken Int
tryConsumeMapLenOrIndef (ByteString -> Word8
BS.unsafeHead ByteString
bs) ByteString
bs of
      DecodedToken Int
DecodeFailure           -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall s a. ByteString -> String -> SlowPath s a
SlowFail ByteString
bs String
"expected map len or indef"
      DecodedToken Int
sz (I# Int#
n#) -> Int# -> ST s (DecodeAction s a)
k Int#
n# forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end (Int -> ByteString -> ByteString
BS.unsafeDrop Int
sz ByteString
bs)


-- The slow path starts off by running the fast path on the current chunk
-- then looking at where it finished, fixing up the chunk boundary issues,
-- getting more input and going around again.
--
-- The offset here is the offset after of all data consumed so far,
-- so not including the current chunk.
--
go_slow :: DecodeAction s a -> ByteString -> ByteOffset
        -> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow :: forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow DecodeAction s a
da ByteString
bs !ByteOffset
offset = do
  SlowPath s a
slowpath <- forall s a. ST s a -> IncrementalDecoder s a
lift forall a b. (a -> b) -> a -> b
$ forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast ByteString
bs DecodeAction s a
da
  case SlowPath s a
slowpath of
    FastDone ByteString
bs' a
x -> forall (m :: * -> *) a. Monad m => a -> m a
return (ByteString
bs', ByteOffset
offset', a
x)
      where
        !offset' :: ByteOffset
offset' = ByteOffset
offset forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 (ByteString -> Int
BS.length ByteString
bs forall a. Num a => a -> a -> a
- ByteString -> Int
BS.length ByteString
bs')

    SlowConsumeTokenBytes ByteString
bs' ByteString -> ST s (DecodeAction s a)
k Int
len -> do
      (ByteString
bstr, ByteString
bs'') <- forall s.
Int
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteString)
getTokenVarLen Int
len ByteString
bs' ByteOffset
offset'
      forall s a. ST s a -> IncrementalDecoder s a
lift (ByteString -> ST s (DecodeAction s a)
k ByteString
bstr) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \DecodeAction s a
daz -> forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow DecodeAction s a
daz ByteString
bs'' (ByteOffset
offset' forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 Int
len)
      where
        !offset' :: ByteOffset
offset' = ByteOffset
offset forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 (ByteString -> Int
BS.length ByteString
bs forall a. Num a => a -> a -> a
- ByteString -> Int
BS.length ByteString
bs')

    SlowConsumeTokenByteArray ByteString
bs' ByteArray -> ST s (DecodeAction s a)
k Int
len -> do
      (ByteString
bstr, ByteString
bs'') <- forall s.
Int
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteString)
getTokenVarLen Int
len ByteString
bs' ByteOffset
offset'
      let !str :: ByteArray
str = ByteString -> ByteArray
BA.fromByteString ByteString
bstr
      forall s a. ST s a -> IncrementalDecoder s a
lift (ByteArray -> ST s (DecodeAction s a)
k ByteArray
str) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \DecodeAction s a
daz -> forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow DecodeAction s a
daz ByteString
bs'' (ByteOffset
offset' forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 Int
len)
      where
        !offset' :: ByteOffset
offset' = ByteOffset
offset forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 (ByteString -> Int
BS.length ByteString
bs forall a. Num a => a -> a -> a
- ByteString -> Int
BS.length ByteString
bs')

    SlowConsumeTokenString ByteString
bs' Text -> ST s (DecodeAction s a)
k Int
len -> do
      (ByteString
bstr, ByteString
bs'') <- forall s.
Int
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteString)
getTokenVarLen Int
len ByteString
bs' ByteOffset
offset'
      case ByteString -> Either UnicodeException Text
T.decodeUtf8' ByteString
bstr of
        Right Text
str -> forall s a. ST s a -> IncrementalDecoder s a
lift (Text -> ST s (DecodeAction s a)
k Text
str) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \DecodeAction s a
daz ->
                     forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow DecodeAction s a
daz ByteString
bs'' (ByteOffset
offset' forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 Int
len)
        Left UnicodeException
_e   -> forall s a.
ByteString -> ByteOffset -> String -> IncrementalDecoder s a
decodeFail ByteString
bs' ByteOffset
offset' String
"invalid UTF8"
      where
        !offset' :: ByteOffset
offset' = ByteOffset
offset forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 (ByteString -> Int
BS.length ByteString
bs forall a. Num a => a -> a -> a
- ByteString -> Int
BS.length ByteString
bs')

    SlowConsumeTokenUtf8ByteArray ByteString
bs' ByteArray -> ST s (DecodeAction s a)
k Int
len -> do
      (ByteString
bstr, ByteString
bs'') <- forall s.
Int
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteString)
getTokenVarLen Int
len ByteString
bs' ByteOffset
offset'
      let !str :: ByteArray
str = ByteString -> ByteArray
BA.fromByteString ByteString
bstr
      forall s a. ST s a -> IncrementalDecoder s a
lift (ByteArray -> ST s (DecodeAction s a)
k ByteArray
str) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \DecodeAction s a
daz -> forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow DecodeAction s a
daz ByteString
bs'' (ByteOffset
offset' forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 Int
len)
      where
        !offset' :: ByteOffset
offset' = ByteOffset
offset forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 (ByteString -> Int
BS.length ByteString
bs forall a. Num a => a -> a -> a
- ByteString -> Int
BS.length ByteString
bs')

    -- we didn't have enough input in the buffer
    SlowDecodeAction ByteString
bs' DecodeAction s a
da' | ByteString -> Bool
BS.null ByteString
bs' -> do
      -- in this case we're exactly out of input
      -- so we can get more input and carry on
      Maybe ByteString
mbs <- forall s. IncrementalDecoder s (Maybe ByteString)
needChunk
      case Maybe ByteString
mbs of
        Maybe ByteString
Nothing   -> forall s a.
ByteString -> ByteOffset -> String -> IncrementalDecoder s a
decodeFail ByteString
bs' ByteOffset
offset' String
"end of input"
        Just ByteString
bs'' -> forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow DecodeAction s a
da' ByteString
bs'' ByteOffset
offset'
      where
        !offset' :: ByteOffset
offset' = ByteOffset
offset forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 (ByteString -> Int
BS.length ByteString
bs forall a. Num a => a -> a -> a
- ByteString -> Int
BS.length ByteString
bs')

    SlowDecodeAction ByteString
bs' DecodeAction s a
da' ->
      -- of course we should only end up here when we really are out of
      -- input, otherwise go_fast_end could have continued
      forall a. (?callStack::CallStack) => Bool -> a -> a
assert (ByteString -> Int
BS.length ByteString
bs' forall a. Ord a => a -> a -> Bool
< Word8 -> Int
tokenSize ((?callStack::CallStack) => ByteString -> Word8
BS.head ByteString
bs')) forall a b. (a -> b) -> a -> b
$
      forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow_fixup DecodeAction s a
da' ByteString
bs' ByteOffset
offset'
      where
        !offset' :: ByteOffset
offset' = ByteOffset
offset forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 (ByteString -> Int
BS.length ByteString
bs forall a. Num a => a -> a -> a
- ByteString -> Int
BS.length ByteString
bs')

    SlowPeekByteOffset ByteString
bs' Int# -> ST s (DecodeAction s a)
k ->
      forall s a. ST s a -> IncrementalDecoder s a
lift
#if MIN_VERSION_base(4,17,0)
        (k (int64ToInt# off#))
#else
        (Int# -> ST s (DecodeAction s a)
k Int#
off#)

#endif
        forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \DecodeAction s a
daz -> forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow DecodeAction s a
daz ByteString
bs' ByteOffset
offset'
      where
        !offset' :: ByteOffset
offset'@(I64# Int#
off#) = ByteOffset
offset forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 (ByteString -> Int
BS.length ByteString
bs forall a. Num a => a -> a -> a
- ByteString -> Int
BS.length ByteString
bs')

    SlowFail ByteString
bs' String
msg -> forall s a.
ByteString -> ByteOffset -> String -> IncrementalDecoder s a
decodeFail ByteString
bs' ByteOffset
offset' String
msg
      where
        !offset' :: ByteOffset
offset' = ByteOffset
offset forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 (ByteString -> Int
BS.length ByteString
bs forall a. Num a => a -> a -> a
- ByteString -> Int
BS.length ByteString
bs')

-- The complicated case is when a token spans a chunk boundary.
--
-- Our goal is to get enough input so that go_fast_end can consume exactly one
-- token without need for further fixups.
--
go_slow_fixup :: DecodeAction s a -> ByteString -> ByteOffset
              -> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow_fixup :: forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow_fixup DecodeAction s a
da !ByteString
bs !ByteOffset
offset = do
    let !hdr :: Word8
hdr = (?callStack::CallStack) => ByteString -> Word8
BS.head ByteString
bs
        !sz :: Int
sz  = Word8 -> Int
tokenSize Word8
hdr
    Maybe ByteString
mbs <- forall s. IncrementalDecoder s (Maybe ByteString)
needChunk
    case Maybe ByteString
mbs of
      Maybe ByteString
Nothing -> forall s a.
ByteString -> ByteOffset -> String -> IncrementalDecoder s a
decodeFail ByteString
bs ByteOffset
offset String
"end of input"

      Just ByteString
bs'
          -- We have enough input now, try reading one final token
        | ByteString -> Int
BS.length ByteString
bs forall a. Num a => a -> a -> a
+ ByteString -> Int
BS.length ByteString
bs' forall a. Ord a => a -> a -> Bool
>= Int
sz
       -> forall s a.
DecodeAction s a
-> Int
-> ByteString
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow_overlapped DecodeAction s a
da Int
sz ByteString
bs ByteString
bs' ByteOffset
offset

          -- We still don't have enough input, get more
        | Bool
otherwise
       -> forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow_fixup DecodeAction s a
da (ByteString
bs forall a. Semigroup a => a -> a -> a
<> ByteString
bs') ByteOffset
offset

-- We've now got more input, but we have one token that spanned the old and
-- new input buffers, so we have to decode that one before carrying on
go_slow_overlapped :: DecodeAction s a -> Int -> ByteString -> ByteString
                   -> ByteOffset
                   -> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow_overlapped :: forall s a.
DecodeAction s a
-> Int
-> ByteString
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow_overlapped DecodeAction s a
da Int
sz ByteString
bs_cur ByteString
bs_next !ByteOffset
offset =

    -- we have:
    --   sz            the size of the pending input token
    --   bs_cur        the tail end of the previous input buffer
    --   bs_next       the next input chunk

    -- we know the old buffer is too small, but the combo is enough
    forall a. (?callStack::CallStack) => Bool -> a -> a
assert (ByteString -> Int
BS.length ByteString
bs_cur forall a. Ord a => a -> a -> Bool
< Int
sz) forall a b. (a -> b) -> a -> b
$
    forall a. (?callStack::CallStack) => Bool -> a -> a
assert (ByteString -> Int
BS.length ByteString
bs_cur forall a. Num a => a -> a -> a
+ ByteString -> Int
BS.length ByteString
bs_next forall a. Ord a => a -> a -> Bool
>= Int
sz) forall a b. (a -> b) -> a -> b
$

    -- we make:
    --   bs_tok        a buffer containing only the pending input token
    --   bs'           the tail of the next input chunk,
    --                   which will become the next input buffer

    let bs_tok :: ByteString
bs_tok   = ByteString
bs_cur forall a. Semigroup a => a -> a -> a
<> Int -> ByteString -> ByteString
BS.unsafeTake (Int
sz forall a. Num a => a -> a -> a
- ByteString -> Int
BS.length ByteString
bs_cur) ByteString
bs_next
        bs' :: ByteString
bs'      =           Int -> ByteString -> ByteString
BS.unsafeDrop (Int
sz forall a. Num a => a -> a -> a
- ByteString -> Int
BS.length ByteString
bs_cur) ByteString
bs_next
        offset' :: ByteOffset
offset'  = ByteOffset
offset forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 Int
sz in

    -- so the token chunk should be exactly the right size
    forall a. (?callStack::CallStack) => Bool -> a -> a
assert (ByteString -> Int
BS.length ByteString
bs_tok forall a. Eq a => a -> a -> Bool
== Int
sz) forall a b. (a -> b) -> a -> b
$
    -- and overall we shouldn't loose any input
    forall a. (?callStack::CallStack) => Bool -> a -> a
assert (ByteString -> Int
BS.length ByteString
bs_cur forall a. Num a => a -> a -> a
+ ByteString -> Int
BS.length ByteString
bs_next forall a. Eq a => a -> a -> Bool
== Int
sz forall a. Num a => a -> a -> a
+ ByteString -> Int
BS.length ByteString
bs') forall a b. (a -> b) -> a -> b
$ do

    -- so now we can run the fast path to consume just this one token
    SlowPath s a
slowpath <- forall s a. ST s a -> IncrementalDecoder s a
lift forall a b. (a -> b) -> a -> b
$ forall s a. ByteString -> DecodeAction s a -> ST s (SlowPath s a)
go_fast_end ByteString
bs_tok DecodeAction s a
da
    case SlowPath s a
slowpath of

      -- typically we'll fall out of the fast path having
      -- consumed exactly one token, now with no trailing data
      SlowDecodeAction ByteString
bs_empty DecodeAction s a
da' ->
        forall a. (?callStack::CallStack) => Bool -> a -> a
assert (ByteString -> Bool
BS.null ByteString
bs_empty) forall a b. (a -> b) -> a -> b
$
        forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow DecodeAction s a
da' ByteString
bs' ByteOffset
offset'

      -- but the other possibilities can happen too
      FastDone ByteString
bs_empty a
x ->
        forall a. (?callStack::CallStack) => Bool -> a -> a
assert (ByteString -> Bool
BS.null ByteString
bs_empty) forall a b. (a -> b) -> a -> b
$
        forall (m :: * -> *) a. Monad m => a -> m a
return (ByteString
bs', ByteOffset
offset', a
x)

      SlowConsumeTokenBytes ByteString
bs_empty ByteString -> ST s (DecodeAction s a)
k Int
len ->
        forall a. (?callStack::CallStack) => Bool -> a -> a
assert (ByteString -> Bool
BS.null ByteString
bs_empty) forall a b. (a -> b) -> a -> b
$ do
        (ByteString
bstr, ByteString
bs'') <- forall s.
ByteString
-> ByteOffset
-> Int
-> IncrementalDecoder s (ByteString, ByteString)
getTokenShortOrVarLen ByteString
bs' ByteOffset
offset' Int
len
        forall s a. ST s a -> IncrementalDecoder s a
lift (ByteString -> ST s (DecodeAction s a)
k ByteString
bstr) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \DecodeAction s a
daz -> forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow DecodeAction s a
daz ByteString
bs'' (ByteOffset
offset' forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 Int
len)

      SlowConsumeTokenByteArray ByteString
bs_empty ByteArray -> ST s (DecodeAction s a)
k Int
len ->
        forall a. (?callStack::CallStack) => Bool -> a -> a
assert (ByteString -> Bool
BS.null ByteString
bs_empty) forall a b. (a -> b) -> a -> b
$ do
        (ByteString
bstr, ByteString
bs'') <- forall s.
ByteString
-> ByteOffset
-> Int
-> IncrementalDecoder s (ByteString, ByteString)
getTokenShortOrVarLen ByteString
bs' ByteOffset
offset' Int
len
        let !ba :: ByteArray
ba = ByteString -> ByteArray
BA.fromByteString ByteString
bstr
        forall s a. ST s a -> IncrementalDecoder s a
lift (ByteArray -> ST s (DecodeAction s a)
k ByteArray
ba) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \DecodeAction s a
daz -> forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow DecodeAction s a
daz ByteString
bs'' (ByteOffset
offset' forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 Int
len)

      SlowConsumeTokenString ByteString
bs_empty Text -> ST s (DecodeAction s a)
k Int
len ->
        forall a. (?callStack::CallStack) => Bool -> a -> a
assert (ByteString -> Bool
BS.null ByteString
bs_empty) forall a b. (a -> b) -> a -> b
$ do
        (ByteString
bstr, ByteString
bs'') <- forall s.
ByteString
-> ByteOffset
-> Int
-> IncrementalDecoder s (ByteString, ByteString)
getTokenShortOrVarLen ByteString
bs' ByteOffset
offset' Int
len
        case ByteString -> Either UnicodeException Text
T.decodeUtf8' ByteString
bstr of
          Right Text
str -> forall s a. ST s a -> IncrementalDecoder s a
lift (Text -> ST s (DecodeAction s a)
k Text
str) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \DecodeAction s a
daz ->
                       forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow DecodeAction s a
daz ByteString
bs'' (ByteOffset
offset' forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 Int
len)
          Left UnicodeException
_e   -> forall s a.
ByteString -> ByteOffset -> String -> IncrementalDecoder s a
decodeFail ByteString
bs' ByteOffset
offset' String
"invalid UTF8"

      SlowConsumeTokenUtf8ByteArray ByteString
bs_empty ByteArray -> ST s (DecodeAction s a)
k Int
len ->
        forall a. (?callStack::CallStack) => Bool -> a -> a
assert (ByteString -> Bool
BS.null ByteString
bs_empty) forall a b. (a -> b) -> a -> b
$ do
        (ByteString
bstr, ByteString
bs'') <- forall s.
ByteString
-> ByteOffset
-> Int
-> IncrementalDecoder s (ByteString, ByteString)
getTokenShortOrVarLen ByteString
bs' ByteOffset
offset' Int
len
        let !ba :: ByteArray
ba = ByteString -> ByteArray
BA.fromByteString ByteString
bstr
        forall s a. ST s a -> IncrementalDecoder s a
lift (ByteArray -> ST s (DecodeAction s a)
k ByteArray
ba) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \DecodeAction s a
daz -> forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow DecodeAction s a
daz ByteString
bs'' (ByteOffset
offset' forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 Int
len)

      SlowPeekByteOffset ByteString
bs_empty Int# -> ST s (DecodeAction s a)
k ->
        forall a. (?callStack::CallStack) => Bool -> a -> a
assert (ByteString -> Bool
BS.null ByteString
bs_empty) forall a b. (a -> b) -> a -> b
$ do
        forall s a. ST s a -> IncrementalDecoder s a
lift
#if MIN_VERSION_base(4,17,0)
          (k (int64ToInt# off#))
#else
          (Int# -> ST s (DecodeAction s a)
k Int#
off#)
#endif
          forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \DecodeAction s a
daz -> forall s a.
DecodeAction s a
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteOffset, a)
go_slow DecodeAction s a
daz ByteString
bs' ByteOffset
offset'
        where
          !(I64# Int#
off#) = ByteOffset
offset'

      SlowFail ByteString
bs_unconsumed String
msg ->
        forall s a.
ByteString -> ByteOffset -> String -> IncrementalDecoder s a
decodeFail (ByteString
bs_unconsumed forall a. Semigroup a => a -> a -> a
<> ByteString
bs') ByteOffset
offset'' String
msg
        where
          !offset'' :: ByteOffset
offset'' = ByteOffset
offset forall a. Num a => a -> a -> a
+ Int -> ByteOffset
intToInt64 (Int
sz forall a. Num a => a -> a -> a
- ByteString -> Int
BS.length ByteString
bs_unconsumed)
  where
    {-# INLINE getTokenShortOrVarLen #-}
    getTokenShortOrVarLen :: BS.ByteString
                          -> ByteOffset
                          -> Int
                          -> IncrementalDecoder s (ByteString, ByteString)
    getTokenShortOrVarLen :: forall s.
ByteString
-> ByteOffset
-> Int
-> IncrementalDecoder s (ByteString, ByteString)
getTokenShortOrVarLen ByteString
bs' ByteOffset
offset' Int
len
      | ByteString -> Int
BS.length ByteString
bs' forall a. Ord a => a -> a -> Bool
< Int
len = forall s.
Int
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteString)
getTokenVarLen Int
len ByteString
bs' ByteOffset
offset'
      | Bool
otherwise           = let !bstr :: ByteString
bstr = Int -> ByteString -> ByteString
BS.take Int
len ByteString
bs'
                                  !bs'' :: ByteString
bs'' = Int -> ByteString -> ByteString
BS.drop Int
len ByteString
bs'
                               in forall (m :: * -> *) a. Monad m => a -> m a
return (ByteString
bstr, ByteString
bs'')


-- TODO FIXME: we can do slightly better here. If we're returning a
-- lazy string (String, lazy Text, lazy ByteString) then we don't have
-- to strictify here and if we're returning a strict string perhaps we
-- can still stream the utf8 validation/converstion

-- TODO FIXME: also consider sharing or not sharing here, and possibly
-- rechunking.

getTokenVarLen :: Int -> ByteString -> ByteOffset
               -> IncrementalDecoder s (ByteString, ByteString)
getTokenVarLen :: forall s.
Int
-> ByteString
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteString)
getTokenVarLen Int
len ByteString
bs ByteOffset
offset =
    forall a. (?callStack::CallStack) => Bool -> a -> a
assert (Int
len forall a. Ord a => a -> a -> Bool
> ByteString -> Int
BS.length ByteString
bs) forall a b. (a -> b) -> a -> b
$ do
    Maybe ByteString
mbs <- forall s. IncrementalDecoder s (Maybe ByteString)
needChunk
    case Maybe ByteString
mbs of
      Maybe ByteString
Nothing -> forall s a.
ByteString -> ByteOffset -> String -> IncrementalDecoder s a
decodeFail ByteString
BS.empty ByteOffset
offset String
"end of input"
      Just ByteString
bs'
        | let n :: Int
n = Int
len forall a. Num a => a -> a -> a
- ByteString -> Int
BS.length ByteString
bs
        , ByteString -> Int
BS.length ByteString
bs' forall a. Ord a => a -> a -> Bool
>= Int
n ->
            let !tok :: ByteString
tok = ByteString
bs forall a. Semigroup a => a -> a -> a
<> Int -> ByteString -> ByteString
BS.unsafeTake Int
n ByteString
bs'
             in forall (m :: * -> *) a. Monad m => a -> m a
return (ByteString
tok, Int -> ByteString -> ByteString
BS.drop Int
n ByteString
bs')

        | Bool
otherwise -> forall s.
[ByteString]
-> Int
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteString)
getTokenVarLenSlow
                         [ByteString
bs',ByteString
bs]
                         (Int
len forall a. Num a => a -> a -> a
- (ByteString -> Int
BS.length ByteString
bs forall a. Num a => a -> a -> a
+ ByteString -> Int
BS.length ByteString
bs'))
                         ByteOffset
offset

getTokenVarLenSlow :: [ByteString] -> Int -> ByteOffset
                   -> IncrementalDecoder s (ByteString, ByteString)
getTokenVarLenSlow :: forall s.
[ByteString]
-> Int
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteString)
getTokenVarLenSlow [ByteString]
bss Int
n ByteOffset
offset = do
    Maybe ByteString
mbs <- forall s. IncrementalDecoder s (Maybe ByteString)
needChunk
    case Maybe ByteString
mbs of
      Maybe ByteString
Nothing -> forall s a.
ByteString -> ByteOffset -> String -> IncrementalDecoder s a
decodeFail ByteString
BS.empty ByteOffset
offset String
"end of input"
      Just ByteString
bs
        | ByteString -> Int
BS.length ByteString
bs forall a. Ord a => a -> a -> Bool
>= Int
n ->
            let !tok :: ByteString
tok = [ByteString] -> ByteString
BS.concat (forall a. [a] -> [a]
reverse (Int -> ByteString -> ByteString
BS.unsafeTake Int
n ByteString
bs forall a. a -> [a] -> [a]
: [ByteString]
bss))
             in forall (m :: * -> *) a. Monad m => a -> m a
return (ByteString
tok, Int -> ByteString -> ByteString
BS.drop Int
n ByteString
bs)
        | Bool
otherwise -> forall s.
[ByteString]
-> Int
-> ByteOffset
-> IncrementalDecoder s (ByteString, ByteString)
getTokenVarLenSlow (ByteString
bsforall a. a -> [a] -> [a]
:[ByteString]
bss) (Int
n forall a. Num a => a -> a -> a
- ByteString -> Int
BS.length ByteString
bs) ByteOffset
offset



tokenSize :: Word8 -> Int
tokenSize :: Word8 -> Int
tokenSize Word8
hdr =
    Word8 -> Int
word8ToInt forall a b. (a -> b) -> a -> b
$
      UArray Word8 Word8
decodeTableSz forall (a :: * -> * -> *) e i.
(IArray a e, Ix i) =>
a i e -> Int -> e
`A.unsafeAt` (Word8 -> Int
word8ToInt Word8
hdr forall a. Bits a => a -> a -> a
.&. Int
0x1f)

decodeTableSz :: UArray Word8 Word8
decodeTableSz :: UArray Word8 Word8
decodeTableSz =
  forall (a :: * -> * -> *) e i.
(IArray a e, Ix i) =>
(i, i) -> [(i, e)] -> a i e
array (Word8
0, Word8
0x1f) forall a b. (a -> b) -> a -> b
$
      [ (Word8 -> Word8 -> Word8
encodeHeader Word8
0 Word8
n, Word8
1) | Word8
n <- [Word8
0..Word8
0x1f] ]
   forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
0 Word8
n, Word8
s) | (Word8
n, Word8
s) <- forall a b. [a] -> [b] -> [(a, b)]
zip [Word8
24..Word8
27] [Word8
2,Word8
3,Word8
5,Word8
9] ]

decodeTokenTypeTable :: Array Word8 TokenType
decodeTokenTypeTable :: Array Word8 TokenType
decodeTokenTypeTable =
  forall (a :: * -> * -> *) e i.
(IArray a e, Ix i) =>
(i, i) -> [(i, e)] -> a i e
array (forall a. Bounded a => a
minBound, forall a. Bounded a => a
maxBound) forall a b. (a -> b) -> a -> b
$
    [ (Word8 -> Word8 -> Word8
encodeHeader Word8
0 Word8
n,  TokenType
TypeUInt) | Word8
n <-  [Word8
0..Word8
26] ]
 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
0 Word8
27, TokenType
TypeUInt64)
    , (Word8 -> Word8 -> Word8
encodeHeader Word8
0 Word8
31, TokenType
TypeInvalid) ]

 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
1 Word8
n,  TokenType
TypeNInt) | Word8
n <-  [Word8
0..Word8
26] ]
 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
1 Word8
27, TokenType
TypeNInt64)
    , (Word8 -> Word8 -> Word8
encodeHeader Word8
1 Word8
31, TokenType
TypeInvalid) ]

 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
2 Word8
n,  TokenType
TypeBytes) | Word8
n <-  [Word8
0..Word8
27] ]
 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
2 Word8
31, TokenType
TypeBytesIndef) ]

 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
3 Word8
n,  TokenType
TypeString) | Word8
n <-  [Word8
0..Word8
27] ]
 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
3 Word8
31, TokenType
TypeStringIndef) ]

 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
4 Word8
n,  TokenType
TypeListLen) | Word8
n <-  [Word8
0..Word8
26] ]
 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
4 Word8
27, TokenType
TypeListLen64)
    , (Word8 -> Word8 -> Word8
encodeHeader Word8
4 Word8
31, TokenType
TypeListLenIndef) ]

 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
5 Word8
n,  TokenType
TypeMapLen) | Word8
n <-  [Word8
0..Word8
26] ]
 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
5 Word8
27, TokenType
TypeMapLen64)
    , (Word8 -> Word8 -> Word8
encodeHeader Word8
5 Word8
31, TokenType
TypeMapLenIndef) ]

 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
6 Word8
n,  TokenType
TypeTag) | Word8
n <- Word8
0forall a. a -> [a] -> [a]
:Word8
1forall a. a -> [a] -> [a]
:[Word8
4..Word8
26] ]
 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
6 Word8
2,  TokenType
TypeInteger)
    , (Word8 -> Word8 -> Word8
encodeHeader Word8
6 Word8
3,  TokenType
TypeInteger)
    , (Word8 -> Word8 -> Word8
encodeHeader Word8
6 Word8
27, TokenType
TypeTag64)
    , (Word8 -> Word8 -> Word8
encodeHeader Word8
6 Word8
31, TokenType
TypeInvalid) ]

 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
7 Word8
n,  TokenType
TypeSimple) | Word8
n <-  [Word8
0..Word8
19] ]
 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
7 Word8
20, TokenType
TypeBool)
    , (Word8 -> Word8 -> Word8
encodeHeader Word8
7 Word8
21, TokenType
TypeBool)
    , (Word8 -> Word8 -> Word8
encodeHeader Word8
7 Word8
22, TokenType
TypeNull)
    , (Word8 -> Word8 -> Word8
encodeHeader Word8
7 Word8
23, TokenType
TypeSimple)
    , (Word8 -> Word8 -> Word8
encodeHeader Word8
7 Word8
24, TokenType
TypeSimple)
    , (Word8 -> Word8 -> Word8
encodeHeader Word8
7 Word8
25, TokenType
TypeFloat16)
    , (Word8 -> Word8 -> Word8
encodeHeader Word8
7 Word8
26, TokenType
TypeFloat32)
    , (Word8 -> Word8 -> Word8
encodeHeader Word8
7 Word8
27, TokenType
TypeFloat64)
    , (Word8 -> Word8 -> Word8
encodeHeader Word8
7 Word8
31, TokenType
TypeBreak) ]

 forall a. [a] -> [a] -> [a]
++ [ (Word8 -> Word8 -> Word8
encodeHeader Word8
mt Word8
n, TokenType
TypeInvalid) | Word8
mt <- [Word8
0..Word8
7], Word8
n <- [Word8
28..Word8
30] ]

encodeHeader :: Word8 -> Word8 -> Word8
encodeHeader :: Word8 -> Word8 -> Word8
encodeHeader Word8
mt Word8
ai = Word8
mt forall a. Bits a => a -> Int -> a
`shiftL` Int
5 forall a. Bits a => a -> a -> a
.|. Word8
ai

data DecodedToken a = DecodedToken !Int !a | DecodeFailure
  deriving Int -> DecodedToken a -> ShowS
forall a. Show a => Int -> DecodedToken a -> ShowS
forall a. Show a => [DecodedToken a] -> ShowS
forall a. Show a => DecodedToken a -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [DecodedToken a] -> ShowS
$cshowList :: forall a. Show a => [DecodedToken a] -> ShowS
show :: DecodedToken a -> String
$cshow :: forall a. Show a => DecodedToken a -> String
showsPrec :: Int -> DecodedToken a -> ShowS
$cshowsPrec :: forall a. Show a => Int -> DecodedToken a -> ShowS
Show
-- TODO add classification for DecodeFailure

-- | Note that canonicity information is calculated lazily. This way we don't
-- need to concern ourselves with two distinct paths, while according to
-- benchmarks it doesn't affect performance in the non-canonical case.
data LongToken a = Fits Bool {- canonical? -} !a
                 | TooLong Bool {- canonical? -} !Int
  deriving Int -> LongToken a -> ShowS
forall a. Show a => Int -> LongToken a -> ShowS
forall a. Show a => [LongToken a] -> ShowS
forall a. Show a => LongToken a -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [LongToken a] -> ShowS
$cshowList :: forall a. Show a => [LongToken a] -> ShowS
show :: LongToken a -> String
$cshow :: forall a. Show a => LongToken a -> String
showsPrec :: Int -> LongToken a -> ShowS
$cshowsPrec :: forall a. Show a => Int -> LongToken a -> ShowS
Show

-- Canoncal NaN floats:
--
-- In these float/double canonical tests we check NaNs are canonical too.
-- There are lots of bit values representing NaN, for each of the flat types.
-- The rule from CBOR RFC 7049, section 3.9 is that the canonical NaN is the
-- CBOR term f97e00 which is the canonical half-float representation. We do
-- this by testing for the size being 3 (since tryConsumeFloat/Double only
-- return 3 when the header byte is 0xf9) and the 16 bytes being 0x7e00.

{-# INLINE isFloat16Canonical #-}
isFloat16Canonical :: Int -> BS.ByteString -> Float -> Bool
isFloat16Canonical :: Int -> ByteString -> Float -> Bool
isFloat16Canonical Int
sz ByteString
bs Float
f
  | Int
sz forall a. Eq a => a -> a -> Bool
/= Int
3   = Bool
False
  | forall a. RealFloat a => a -> Bool
isNaN Float
f   = ByteString -> Word16
eatTailWord16 ByteString
bs forall a. Eq a => a -> a -> Bool
== Word16
0x7e00
  | Bool
otherwise = Bool
True

{-# INLINE isFloatCanonical #-}
isFloatCanonical :: Int -> BS.ByteString -> Float -> Bool
isFloatCanonical :: Int -> ByteString -> Float -> Bool
isFloatCanonical Int
sz ByteString
bs Float
f
  | forall a. RealFloat a => a -> Bool
isNaN Float
f   = Int
sz forall a. Eq a => a -> a -> Bool
== Int
3 Bool -> Bool -> Bool
&& ByteString -> Word16
eatTailWord16 ByteString
bs forall a. Eq a => a -> a -> Bool
== Word16
0x7e00
  | Bool
otherwise = Int
sz forall a. Eq a => a -> a -> Bool
== Int
5

{-# INLINE isDoubleCanonical #-}
isDoubleCanonical :: Int -> BS.ByteString -> Double -> Bool
isDoubleCanonical :: Int -> ByteString -> Double -> Bool
isDoubleCanonical Int
sz ByteString
bs Double
f
  | forall a. RealFloat a => a -> Bool
isNaN Double
f   = Int
sz forall a. Eq a => a -> a -> Bool
== Int
3 Bool -> Bool -> Bool
&& ByteString -> Word16
eatTailWord16 ByteString
bs forall a. Eq a => a -> a -> Bool
== Word16
0x7e00
  | Bool
otherwise = Int
sz forall a. Eq a => a -> a -> Bool
== Int
9

{-# INLINE isWordCanonical #-}
isWordCanonical :: Int -> Word -> Bool
isWordCanonical :: Int -> Word -> Bool
isWordCanonical Int
sz !Word
w
  | Int
sz forall a. Eq a => a -> a -> Bool
== Int
2   = Word
w forall a. Ord a => a -> a -> Bool
> Word
0x17
  | Int
sz forall a. Eq a => a -> a -> Bool
== Int
3   = Word
w forall a. Ord a => a -> a -> Bool
> Word
0xff
  | Int
sz forall a. Eq a => a -> a -> Bool
== Int
5   = Word
w forall a. Ord a => a -> a -> Bool
> Word
0xffff
  | Int
sz forall a. Eq a => a -> a -> Bool
== Int
9   = Word
w forall a. Ord a => a -> a -> Bool
> Word
0xffffffff
  | Bool
otherwise = Bool
True

{-# INLINE isIntCanonical #-}
isIntCanonical :: Int -> Int -> Bool
isIntCanonical :: Int -> Int -> Bool
isIntCanonical Int
sz Int
i
  | Int
i forall a. Ord a => a -> a -> Bool
< Int
0     = Int -> Word -> Bool
isWordCanonical Int
sz (forall a. Bits a => a -> a
complement Word
w)
  | Bool
otherwise = Int -> Word -> Bool
isWordCanonical Int
sz       Word
w
  where
    w :: Word
w = Int -> Word
intToWord Int
i

#if defined(ARCH_32bit)
{-# INLINE isWord64Canonical #-}
isWord64Canonical :: Int -> Word64 -> Bool
isWord64Canonical sz w
  | sz == 2   = w > 0x17)
  | sz == 3   = w > 0xff)
  | sz == 5   = w > 0xffff)
  | sz == 9   = w > 0xffffffff)
  | otherwise = True

{-# INLINE isInt64Canonical #-}
isInt64Canonical :: Int -> Int64# -> Bool
isInt64Canonical sz i#
  | isTrue# (i# `ltInt64#` intToInt64# 0#) = isWord64Canonical sz (not64# w#)
  | otherwise                              = isWord64Canonical sz         w#
  where
    w# = int64ToWord64# i#
#endif

{-# INLINE isSimpleCanonical #-}
isSimpleCanonical :: Int -> Word# -> Bool
isSimpleCanonical :: Int -> Word# -> Bool
isSimpleCanonical Int
2 Word#
w# = Int# -> Bool
isTrue# (Word#
w# Word# -> Word# -> Int#
`gtWord#` Word#
0x17##)
isSimpleCanonical Int
_ Word#
_  = Bool
True -- only size 1 and 2 are possible here


-- TODO FIXME: check with 7.10 and file ticket:
-- a case analysis against 0x00 .. 0xff :: Word8 turns into a huge chain
-- of >= tests. It could use a jump table, or at least it could use a binary
-- division. Whereas for Int or Word it does the right thing.

{-# INLINE tryConsumeWord #-}
tryConsumeWord :: Word8 -> ByteString -> DecodedToken Word
tryConsumeWord :: Word8 -> ByteString -> DecodedToken Word
tryConsumeWord Word8
hdr !ByteString
bs = case Word8 -> Word
word8ToWord Word8
hdr of
  -- Positive integers (type 0)
  Word
0x00 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
0
  Word
0x01 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
1
  Word
0x02 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
2
  Word
0x03 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
3
  Word
0x04 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
4
  Word
0x05 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
5
  Word
0x06 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
6
  Word
0x07 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
7
  Word
0x08 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
8
  Word
0x09 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
9
  Word
0x0a -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
10
  Word
0x0b -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
11
  Word
0x0c -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
12
  Word
0x0d -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
13
  Word
0x0e -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
14
  Word
0x0f -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
15
  Word
0x10 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
16
  Word
0x11 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
17
  Word
0x12 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
18
  Word
0x13 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
19
  Word
0x14 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
20
  Word
0x15 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
21
  Word
0x16 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
22
  Word
0x17 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
23
  Word
0x18 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
2 forall a b. (a -> b) -> a -> b
$! Word8 -> Word
word8ToWord  (ByteString -> Word8
eatTailWord8 ByteString
bs)
  Word
0x19 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
3 forall a b. (a -> b) -> a -> b
$! Word16 -> Word
word16ToWord (ByteString -> Word16
eatTailWord16 ByteString
bs)
  Word
0x1a -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
5 forall a b. (a -> b) -> a -> b
$! Word32 -> Word
word32ToWord (ByteString -> Word32
eatTailWord32 ByteString
bs)
#if defined(ARCH_64bit)
  Word
0x1b -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
9 forall a b. (a -> b) -> a -> b
$! Word64 -> Word
word64ToWord (ByteString -> Word64
eatTailWord64 ByteString
bs)
#else
  0x1b -> case word64ToWord (eatTailWord64 bs) of
            Just n  -> DecodedToken 9 n
            Nothing -> DecodeFailure
#endif
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeNegWord #-}
tryConsumeNegWord :: Word8 -> ByteString -> DecodedToken Word
tryConsumeNegWord :: Word8 -> ByteString -> DecodedToken Word
tryConsumeNegWord Word8
hdr !ByteString
bs = case Word8 -> Word
word8ToWord Word8
hdr of
  -- Positive integers (type 0)
  Word
0x20 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
0
  Word
0x21 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
1
  Word
0x22 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
2
  Word
0x23 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
3
  Word
0x24 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
4
  Word
0x25 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
5
  Word
0x26 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
6
  Word
0x27 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
7
  Word
0x28 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
8
  Word
0x29 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
9
  Word
0x2a -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
10
  Word
0x2b -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
11
  Word
0x2c -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
12
  Word
0x2d -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
13
  Word
0x2e -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
14
  Word
0x2f -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
15
  Word
0x30 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
16
  Word
0x31 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
17
  Word
0x32 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
18
  Word
0x33 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
19
  Word
0x34 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
20
  Word
0x35 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
21
  Word
0x36 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
22
  Word
0x37 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
23
  Word
0x38 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
2 forall a b. (a -> b) -> a -> b
$! (Word8 -> Word
word8ToWord  (ByteString -> Word8
eatTailWord8 ByteString
bs))
  Word
0x39 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
3 forall a b. (a -> b) -> a -> b
$! (Word16 -> Word
word16ToWord (ByteString -> Word16
eatTailWord16 ByteString
bs))
  Word
0x3a -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
5 forall a b. (a -> b) -> a -> b
$! (Word32 -> Word
word32ToWord (ByteString -> Word32
eatTailWord32 ByteString
bs))
#if defined(ARCH_64bit)
  Word
0x3b -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
9 forall a b. (a -> b) -> a -> b
$! (Word64 -> Word
word64ToWord (ByteString -> Word64
eatTailWord64 ByteString
bs))
#else
  0x3b -> case word64ToWord (eatTailWord64 bs) of
            Just n  -> DecodedToken 9 n
            Nothing -> DecodeFailure
#endif
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeInt #-}
tryConsumeInt :: Word8 -> ByteString -> DecodedToken Int
tryConsumeInt :: Word8 -> ByteString -> DecodedToken Int
tryConsumeInt Word8
hdr !ByteString
bs = case Word8 -> Word
word8ToWord Word8
hdr of
  -- Positive integers (type 0)
  Word
0x00 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
0
  Word
0x01 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
1
  Word
0x02 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
2
  Word
0x03 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
3
  Word
0x04 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
4
  Word
0x05 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
5
  Word
0x06 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
6
  Word
0x07 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
7
  Word
0x08 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
8
  Word
0x09 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
9
  Word
0x0a -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
10
  Word
0x0b -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
11
  Word
0x0c -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
12
  Word
0x0d -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
13
  Word
0x0e -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
14
  Word
0x0f -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
15
  Word
0x10 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
16
  Word
0x11 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
17
  Word
0x12 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
18
  Word
0x13 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
19
  Word
0x14 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
20
  Word
0x15 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
21
  Word
0x16 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
22
  Word
0x17 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
23
  Word
0x18 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
2 forall a b. (a -> b) -> a -> b
$! (Word8 -> Int
word8ToInt  (ByteString -> Word8
eatTailWord8 ByteString
bs))
  Word
0x19 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
3 forall a b. (a -> b) -> a -> b
$! (Word16 -> Int
word16ToInt (ByteString -> Word16
eatTailWord16 ByteString
bs))
#if defined(ARCH_64bit)
  Word
0x1a -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
5 forall a b. (a -> b) -> a -> b
$! (Word32 -> Int
word32ToInt (ByteString -> Word32
eatTailWord32 ByteString
bs))
#else
  0x1a -> case word32ToInt (eatTailWord32 bs) of
            Just n  -> DecodedToken 5 n
            Nothing -> DecodeFailure
#endif
  Word
0x1b -> case Word64 -> Maybe Int
word64ToInt (ByteString -> Word64
eatTailWord64 ByteString
bs) of
            Just Int
n  -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
9 Int
n
            Maybe Int
Nothing -> forall a. DecodedToken a
DecodeFailure

  -- Negative integers (type 1)
  Word
0x20 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
1)
  Word
0x21 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
2)
  Word
0x22 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
3)
  Word
0x23 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
4)
  Word
0x24 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
5)
  Word
0x25 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
6)
  Word
0x26 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
7)
  Word
0x27 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
8)
  Word
0x28 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
9)
  Word
0x29 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
10)
  Word
0x2a -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
11)
  Word
0x2b -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
12)
  Word
0x2c -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
13)
  Word
0x2d -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
14)
  Word
0x2e -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
15)
  Word
0x2f -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
16)
  Word
0x30 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
17)
  Word
0x31 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
18)
  Word
0x32 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
19)
  Word
0x33 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
20)
  Word
0x34 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
21)
  Word
0x35 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
22)
  Word
0x36 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
23)
  Word
0x37 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
24)
  Word
0x38 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
2 forall a b. (a -> b) -> a -> b
$! (-Int
1 forall a. Num a => a -> a -> a
- Word8 -> Int
word8ToInt  (ByteString -> Word8
eatTailWord8 ByteString
bs))
  Word
0x39 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
3 forall a b. (a -> b) -> a -> b
$! (-Int
1 forall a. Num a => a -> a -> a
- Word16 -> Int
word16ToInt (ByteString -> Word16
eatTailWord16 ByteString
bs))
#if defined(ARCH_64bit)
  Word
0x3a -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
5 forall a b. (a -> b) -> a -> b
$! (-Int
1 forall a. Num a => a -> a -> a
- Word32 -> Int
word32ToInt (ByteString -> Word32
eatTailWord32 ByteString
bs))
#else
  0x3a -> case word32ToInt (eatTailWord32 bs) of
            Just n  -> DecodedToken 5 (-1 - n)
            Nothing -> DecodeFailure
#endif
  Word
0x3b -> case Word64 -> Maybe Int
word64ToInt (ByteString -> Word64
eatTailWord64 ByteString
bs) of
            Just Int
n  -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
9 (-Int
1 forall a. Num a => a -> a -> a
- Int
n)
            Maybe Int
Nothing -> forall a. DecodedToken a
DecodeFailure
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeInteger #-}
tryConsumeInteger :: Word8 -> ByteString -> DecodedToken (BigIntToken Integer)
tryConsumeInteger :: Word8 -> ByteString -> DecodedToken (BigIntToken Integer)
tryConsumeInteger Word8
hdr !ByteString
bs = case Word8 -> Word
word8ToWord Word8
hdr of

  -- Positive integers (type 0)
  Word
0x00 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
0)
  Word
0x01 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
1)
  Word
0x02 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
2)
  Word
0x03 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
3)
  Word
0x04 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
4)
  Word
0x05 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
5)
  Word
0x06 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
6)
  Word
0x07 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
7)
  Word
0x08 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
8)
  Word
0x09 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
9)
  Word
0x0a -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
10)
  Word
0x0b -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
11)
  Word
0x0c -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
12)
  Word
0x0d -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
13)
  Word
0x0e -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
14)
  Word
0x0f -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
15)
  Word
0x10 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
16)
  Word
0x11 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
17)
  Word
0x12 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
18)
  Word
0x13 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
19)
  Word
0x14 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
20)
  Word
0x15 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
21)
  Word
0x16 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
22)
  Word
0x17 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True Integer
23)

  Word
0x18 -> let !w :: Word8
w = ByteString -> Word8
eatTailWord8 ByteString
bs
              sz :: Int
sz = Int
2
          in forall a. Int -> a -> DecodedToken a
DecodedToken Int
sz (forall a. Bool -> Integer -> BigIntToken a
BigIntToken (Int -> Word -> Bool
isWordCanonical Int
sz (Word8 -> Word
word8ToWord Word8
w))    forall a b. (a -> b) -> a -> b
$! forall a. Integral a => a -> Integer
toInteger Word8
w)
  Word
0x19 -> let !w :: Word16
w = ByteString -> Word16
eatTailWord16 ByteString
bs
              sz :: Int
sz = Int
3
          in forall a. Int -> a -> DecodedToken a
DecodedToken Int
sz (forall a. Bool -> Integer -> BigIntToken a
BigIntToken (Int -> Word -> Bool
isWordCanonical Int
sz (Word16 -> Word
word16ToWord Word16
w))   forall a b. (a -> b) -> a -> b
$! forall a. Integral a => a -> Integer
toInteger Word16
w)
  Word
0x1a -> let !w :: Word32
w = ByteString -> Word32
eatTailWord32 ByteString
bs
              sz :: Int
sz = Int
5
          in forall a. Int -> a -> DecodedToken a
DecodedToken Int
sz (forall a. Bool -> Integer -> BigIntToken a
BigIntToken (Int -> Word -> Bool
isWordCanonical Int
sz (Word32 -> Word
word32ToWord Word32
w))   forall a b. (a -> b) -> a -> b
$! forall a. Integral a => a -> Integer
toInteger Word32
w)
  Word
0x1b -> let !w :: Word64
w = ByteString -> Word64
eatTailWord64 ByteString
bs
              sz :: Int
sz = Int
9
#if defined(ARCH_32bit)
          in DecodedToken sz (BigIntToken (isWord64Canonical sz (word64ToWord w)) $! toInteger w)
#else
          in forall a. Int -> a -> DecodedToken a
DecodedToken Int
sz (forall a. Bool -> Integer -> BigIntToken a
BigIntToken (Int -> Word -> Bool
isWordCanonical Int
sz (Word64 -> Word
word64ToWord Word64
w))   forall a b. (a -> b) -> a -> b
$! forall a. Integral a => a -> Integer
toInteger Word64
w)
#endif

  -- Negative integers (type 1)
  Word
0x20 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
1))
  Word
0x21 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
2))
  Word
0x22 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
3))
  Word
0x23 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
4))
  Word
0x24 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
5))
  Word
0x25 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
6))
  Word
0x26 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
7))
  Word
0x27 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
8))
  Word
0x28 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
9))
  Word
0x29 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
10))
  Word
0x2a -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
11))
  Word
0x2b -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
12))
  Word
0x2c -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
13))
  Word
0x2d -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
14))
  Word
0x2e -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
15))
  Word
0x2f -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
16))
  Word
0x30 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
17))
  Word
0x31 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
18))
  Word
0x32 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
19))
  Word
0x33 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
20))
  Word
0x34 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
21))
  Word
0x35 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
22))
  Word
0x36 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
23))
  Word
0x37 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (forall a. Bool -> Integer -> BigIntToken a
BigIntToken Bool
True (-Integer
24))
  Word
0x38 -> let !w :: Word8
w = ByteString -> Word8
eatTailWord8 ByteString
bs
              sz :: Int
sz = Int
2
          in forall a. Int -> a -> DecodedToken a
DecodedToken Int
sz (forall a. Bool -> Integer -> BigIntToken a
BigIntToken (Int -> Word -> Bool
isWordCanonical Int
sz (Word8 -> Word
word8ToWord Word8
w))    forall a b. (a -> b) -> a -> b
$! (-Integer
1 forall a. Num a => a -> a -> a
- forall a. Integral a => a -> Integer
toInteger Word8
w))
  Word
0x39 -> let !w :: Word16
w = ByteString -> Word16
eatTailWord16 ByteString
bs
              sz :: Int
sz = Int
3
          in forall a. Int -> a -> DecodedToken a
DecodedToken Int
sz (forall a. Bool -> Integer -> BigIntToken a
BigIntToken (Int -> Word -> Bool
isWordCanonical Int
sz (Word16 -> Word
word16ToWord Word16
w))   forall a b. (a -> b) -> a -> b
$! (-Integer
1 forall a. Num a => a -> a -> a
- forall a. Integral a => a -> Integer
toInteger Word16
w))
  Word
0x3a -> let !w :: Word32
w = ByteString -> Word32
eatTailWord32 ByteString
bs
              sz :: Int
sz = Int
5
          in forall a. Int -> a -> DecodedToken a
DecodedToken Int
sz (forall a. Bool -> Integer -> BigIntToken a
BigIntToken (Int -> Word -> Bool
isWordCanonical Int
sz (Word32 -> Word
word32ToWord Word32
w))   forall a b. (a -> b) -> a -> b
$! (-Integer
1 forall a. Num a => a -> a -> a
- forall a. Integral a => a -> Integer
toInteger Word32
w))
  Word
0x3b -> let !w :: Word64
w = ByteString -> Word64
eatTailWord64 ByteString
bs
              sz :: Int
sz = Int
9
#if defined(ARCH_32bit)
          in DecodedToken sz (BigIntToken (isWord64Canonical sz (word64ToWord w)) $! (-1 - toInteger w))
#else
          in forall a. Int -> a -> DecodedToken a
DecodedToken Int
sz (forall a. Bool -> Integer -> BigIntToken a
BigIntToken (Int -> Word -> Bool
isWordCanonical Int
sz (Word64 -> Word
word64ToWord Word64
w))   forall a b. (a -> b) -> a -> b
$! (-Integer
1 forall a. Num a => a -> a -> a
- forall a. Integral a => a -> Integer
toInteger Word64
w))
#endif

  Word
0xc2 -> forall a. ByteString -> DecodedToken (BigIntToken a)
readBigUInt ByteString
bs
  Word
0xc3 -> forall a. ByteString -> DecodedToken (BigIntToken a)
readBigNInt ByteString
bs

  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeBytes #-}
tryConsumeBytes :: Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeBytes :: Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeBytes Word8
hdr !ByteString
bs = case Word8 -> Word
word8ToWord Word8
hdr of

  -- Bytes (type 2)
  Word
0x40 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
0 ByteString
bs
  Word
0x41 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
1 ByteString
bs
  Word
0x42 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
2 ByteString
bs
  Word
0x43 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
3 ByteString
bs
  Word
0x44 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
4 ByteString
bs
  Word
0x45 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
5 ByteString
bs
  Word
0x46 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
6 ByteString
bs
  Word
0x47 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
7 ByteString
bs
  Word
0x48 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
8 ByteString
bs
  Word
0x49 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
9 ByteString
bs
  Word
0x4a -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
10 ByteString
bs
  Word
0x4b -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
11 ByteString
bs
  Word
0x4c -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
12 ByteString
bs
  Word
0x4d -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
13 ByteString
bs
  Word
0x4e -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
14 ByteString
bs
  Word
0x4f -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
15 ByteString
bs
  Word
0x50 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
16 ByteString
bs
  Word
0x51 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
17 ByteString
bs
  Word
0x52 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
18 ByteString
bs
  Word
0x53 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
19 ByteString
bs
  Word
0x54 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
20 ByteString
bs
  Word
0x55 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
21 ByteString
bs
  Word
0x56 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
22 ByteString
bs
  Word
0x57 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
23 ByteString
bs
  Word
0x58 -> ByteString -> DecodedToken (LongToken ByteString)
readBytes8  ByteString
bs
  Word
0x59 -> ByteString -> DecodedToken (LongToken ByteString)
readBytes16 ByteString
bs
  Word
0x5a -> ByteString -> DecodedToken (LongToken ByteString)
readBytes32 ByteString
bs
  Word
0x5b -> ByteString -> DecodedToken (LongToken ByteString)
readBytes64 ByteString
bs
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeString #-}
tryConsumeString :: Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeString :: Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeString Word8
hdr !ByteString
bs = case Word8 -> Word
word8ToWord Word8
hdr of

  -- Strings (type 3)
  Word
0x60 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
0 ByteString
bs
  Word
0x61 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
1 ByteString
bs
  Word
0x62 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
2 ByteString
bs
  Word
0x63 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
3 ByteString
bs
  Word
0x64 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
4 ByteString
bs
  Word
0x65 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
5 ByteString
bs
  Word
0x66 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
6 ByteString
bs
  Word
0x67 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
7 ByteString
bs
  Word
0x68 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
8 ByteString
bs
  Word
0x69 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
9 ByteString
bs
  Word
0x6a -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
10 ByteString
bs
  Word
0x6b -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
11 ByteString
bs
  Word
0x6c -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
12 ByteString
bs
  Word
0x6d -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
13 ByteString
bs
  Word
0x6e -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
14 ByteString
bs
  Word
0x6f -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
15 ByteString
bs
  Word
0x70 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
16 ByteString
bs
  Word
0x71 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
17 ByteString
bs
  Word
0x72 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
18 ByteString
bs
  Word
0x73 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
19 ByteString
bs
  Word
0x74 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
20 ByteString
bs
  Word
0x75 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
21 ByteString
bs
  Word
0x76 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
22 ByteString
bs
  Word
0x77 -> Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
23 ByteString
bs
  Word
0x78 -> ByteString -> DecodedToken (LongToken ByteString)
readBytes8  ByteString
bs
  Word
0x79 -> ByteString -> DecodedToken (LongToken ByteString)
readBytes16 ByteString
bs
  Word
0x7a -> ByteString -> DecodedToken (LongToken ByteString)
readBytes32 ByteString
bs
  Word
0x7b -> ByteString -> DecodedToken (LongToken ByteString)
readBytes64 ByteString
bs
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeListLen #-}
tryConsumeListLen :: Word8 -> ByteString -> DecodedToken Int
tryConsumeListLen :: Word8 -> ByteString -> DecodedToken Int
tryConsumeListLen Word8
hdr !ByteString
bs = case Word8 -> Word
word8ToWord Word8
hdr of
  -- List structures (type 4)
  Word
0x80 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
0
  Word
0x81 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
1
  Word
0x82 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
2
  Word
0x83 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
3
  Word
0x84 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
4
  Word
0x85 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
5
  Word
0x86 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
6
  Word
0x87 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
7
  Word
0x88 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
8
  Word
0x89 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
9
  Word
0x8a -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
10
  Word
0x8b -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
11
  Word
0x8c -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
12
  Word
0x8d -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
13
  Word
0x8e -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
14
  Word
0x8f -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
15
  Word
0x90 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
16
  Word
0x91 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
17
  Word
0x92 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
18
  Word
0x93 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
19
  Word
0x94 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
20
  Word
0x95 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
21
  Word
0x96 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
22
  Word
0x97 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
23
  Word
0x98 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
2 (Word8 -> Int
word8ToInt  (ByteString -> Word8
eatTailWord8 ByteString
bs))
  Word
0x99 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
3 (Word16 -> Int
word16ToInt (ByteString -> Word16
eatTailWord16 ByteString
bs))
#if defined(ARCH_64bit)
  Word
0x9a -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
5 (Word32 -> Int
word32ToInt (ByteString -> Word32
eatTailWord32 ByteString
bs))
#else
  0x9a -> case word32ToInt (eatTailWord32 bs) of
            Just n  -> DecodedToken 5 n
            Nothing -> DecodeFailure
#endif
  Word
0x9b -> case Word64 -> Maybe Int
word64ToInt (ByteString -> Word64
eatTailWord64 ByteString
bs) of
            Just Int
n  -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
9 Int
n
            Maybe Int
Nothing -> forall a. DecodedToken a
DecodeFailure
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeMapLen #-}
tryConsumeMapLen :: Word8 -> ByteString -> DecodedToken Int
tryConsumeMapLen :: Word8 -> ByteString -> DecodedToken Int
tryConsumeMapLen Word8
hdr !ByteString
bs = case Word8 -> Word
word8ToWord Word8
hdr of
  -- Map structures (type 5)
  Word
0xa0 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
0
  Word
0xa1 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
1
  Word
0xa2 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
2
  Word
0xa3 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
3
  Word
0xa4 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
4
  Word
0xa5 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
5
  Word
0xa6 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
6
  Word
0xa7 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
7
  Word
0xa8 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
8
  Word
0xa9 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
9
  Word
0xaa -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
10
  Word
0xab -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
11
  Word
0xac -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
12
  Word
0xad -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
13
  Word
0xae -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
14
  Word
0xaf -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
15
  Word
0xb0 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
16
  Word
0xb1 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
17
  Word
0xb2 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
18
  Word
0xb3 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
19
  Word
0xb4 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
20
  Word
0xb5 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
21
  Word
0xb6 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
22
  Word
0xb7 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
23
  Word
0xb8 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
2 forall a b. (a -> b) -> a -> b
$! (Word8 -> Int
word8ToInt  (ByteString -> Word8
eatTailWord8 ByteString
bs))
  Word
0xb9 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
3 forall a b. (a -> b) -> a -> b
$! (Word16 -> Int
word16ToInt (ByteString -> Word16
eatTailWord16 ByteString
bs))
#if defined(ARCH_64bit)
  Word
0xba -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
5 forall a b. (a -> b) -> a -> b
$! (Word32 -> Int
word32ToInt (ByteString -> Word32
eatTailWord32 ByteString
bs))
#else
  0xba -> case word32ToInt (eatTailWord32 bs) of
            Just n  -> DecodedToken 5 n
            Nothing -> DecodeFailure
#endif
  Word
0xbb -> case Word64 -> Maybe Int
word64ToInt (ByteString -> Word64
eatTailWord64 ByteString
bs) of
            Just Int
n  -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
9 Int
n
            Maybe Int
Nothing -> forall a. DecodedToken a
DecodeFailure
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeListLenIndef #-}
tryConsumeListLenIndef :: Word8 -> DecodedToken ()
tryConsumeListLenIndef :: Word8 -> DecodedToken ()
tryConsumeListLenIndef Word8
hdr = case Word8 -> Word
word8ToWord Word8
hdr of
  Word
0x9f -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 ()
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeMapLenIndef #-}
tryConsumeMapLenIndef :: Word8 -> DecodedToken ()
tryConsumeMapLenIndef :: Word8 -> DecodedToken ()
tryConsumeMapLenIndef Word8
hdr = case Word8 -> Word
word8ToWord Word8
hdr of
  Word
0xbf -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 ()
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeListLenOrIndef #-}
tryConsumeListLenOrIndef :: Word8 -> ByteString -> DecodedToken Int
tryConsumeListLenOrIndef :: Word8 -> ByteString -> DecodedToken Int
tryConsumeListLenOrIndef Word8
hdr !ByteString
bs = case Word8 -> Word
word8ToWord Word8
hdr of

  -- List structures (type 4)
  Word
0x80 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
0
  Word
0x81 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
1
  Word
0x82 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
2
  Word
0x83 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
3
  Word
0x84 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
4
  Word
0x85 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
5
  Word
0x86 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
6
  Word
0x87 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
7
  Word
0x88 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
8
  Word
0x89 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
9
  Word
0x8a -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
10
  Word
0x8b -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
11
  Word
0x8c -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
12
  Word
0x8d -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
13
  Word
0x8e -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
14
  Word
0x8f -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
15
  Word
0x90 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
16
  Word
0x91 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
17
  Word
0x92 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
18
  Word
0x93 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
19
  Word
0x94 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
20
  Word
0x95 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
21
  Word
0x96 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
22
  Word
0x97 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
23
  Word
0x98 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
2 forall a b. (a -> b) -> a -> b
$! (Word8 -> Int
word8ToInt  (ByteString -> Word8
eatTailWord8 ByteString
bs))
  Word
0x99 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
3 forall a b. (a -> b) -> a -> b
$! (Word16 -> Int
word16ToInt (ByteString -> Word16
eatTailWord16 ByteString
bs))
#if defined(ARCH_64bit)
  Word
0x9a -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
5 forall a b. (a -> b) -> a -> b
$! (Word32 -> Int
word32ToInt (ByteString -> Word32
eatTailWord32 ByteString
bs))
#else
  0x9a -> case word32ToInt (eatTailWord32 bs) of
            Just n  -> DecodedToken 5 n
            Nothing -> DecodeFailure
#endif
  Word
0x9b -> case Word64 -> Maybe Int
word64ToInt (ByteString -> Word64
eatTailWord64 ByteString
bs) of
            Just Int
n  -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
9 Int
n
            Maybe Int
Nothing -> forall a. DecodedToken a
DecodeFailure
  Word
0x9f -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
1) -- indefinite length
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeMapLenOrIndef #-}
tryConsumeMapLenOrIndef :: Word8 -> ByteString -> DecodedToken Int
tryConsumeMapLenOrIndef :: Word8 -> ByteString -> DecodedToken Int
tryConsumeMapLenOrIndef Word8
hdr !ByteString
bs = case Word8 -> Word
word8ToWord Word8
hdr of

  -- Map structures (type 5)
  Word
0xa0 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
0
  Word
0xa1 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
1
  Word
0xa2 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
2
  Word
0xa3 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
3
  Word
0xa4 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
4
  Word
0xa5 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
5
  Word
0xa6 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
6
  Word
0xa7 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
7
  Word
0xa8 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
8
  Word
0xa9 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
9
  Word
0xaa -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
10
  Word
0xab -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
11
  Word
0xac -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
12
  Word
0xad -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
13
  Word
0xae -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
14
  Word
0xaf -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
15
  Word
0xb0 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
16
  Word
0xb1 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
17
  Word
0xb2 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
18
  Word
0xb3 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
19
  Word
0xb4 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
20
  Word
0xb5 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
21
  Word
0xb6 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
22
  Word
0xb7 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Int
23
  Word
0xb8 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
2 forall a b. (a -> b) -> a -> b
$! (Word8 -> Int
word8ToInt  (ByteString -> Word8
eatTailWord8 ByteString
bs))
  Word
0xb9 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
3 forall a b. (a -> b) -> a -> b
$! (Word16 -> Int
word16ToInt (ByteString -> Word16
eatTailWord16 ByteString
bs))
#if defined(ARCH_64bit)
  Word
0xba -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
5 forall a b. (a -> b) -> a -> b
$! (Word32 -> Int
word32ToInt (ByteString -> Word32
eatTailWord32 ByteString
bs))
#else
  0xba -> case word32ToInt (eatTailWord32 bs) of
            Just n  -> DecodedToken 5 n
            Nothing -> DecodeFailure
#endif
  Word
0xbb -> case Word64 -> Maybe Int
word64ToInt (ByteString -> Word64
eatTailWord64 ByteString
bs) of
            Just Int
n  -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
9 Int
n
            Maybe Int
Nothing -> forall a. DecodedToken a
DecodeFailure
  Word
0xbf -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 (-Int
1) -- indefinite length
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeTag #-}
tryConsumeTag :: Word8 -> ByteString -> DecodedToken Word
tryConsumeTag :: Word8 -> ByteString -> DecodedToken Word
tryConsumeTag Word8
hdr !ByteString
bs = case Word8 -> Word
word8ToWord Word8
hdr of

  -- Tagged values (type 6)
  Word
0xc0 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
0
  Word
0xc1 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
1
  Word
0xc2 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
2
  Word
0xc3 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
3
  Word
0xc4 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
4
  Word
0xc5 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
5
  Word
0xc6 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
6
  Word
0xc7 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
7
  Word
0xc8 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
8
  Word
0xc9 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
9
  Word
0xca -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
10
  Word
0xcb -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
11
  Word
0xcc -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
12
  Word
0xcd -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
13
  Word
0xce -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
14
  Word
0xcf -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
15
  Word
0xd0 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
16
  Word
0xd1 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
17
  Word
0xd2 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
18
  Word
0xd3 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
19
  Word
0xd4 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
20
  Word
0xd5 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
21
  Word
0xd6 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
22
  Word
0xd7 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
23
  Word
0xd8 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
2 forall a b. (a -> b) -> a -> b
$! (Word8 -> Word
word8ToWord  (ByteString -> Word8
eatTailWord8 ByteString
bs))
  Word
0xd9 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
3 forall a b. (a -> b) -> a -> b
$! (Word16 -> Word
word16ToWord (ByteString -> Word16
eatTailWord16 ByteString
bs))
  Word
0xda -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
5 forall a b. (a -> b) -> a -> b
$! (Word32 -> Word
word32ToWord (ByteString -> Word32
eatTailWord32 ByteString
bs))
#if defined(ARCH_64bit)
  Word
0xdb -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
9 forall a b. (a -> b) -> a -> b
$! (Word64 -> Word
word64ToWord (ByteString -> Word64
eatTailWord64 ByteString
bs))
#else
  0xdb -> case word64ToWord (eatTailWord64 bs) of
            Just n  -> DecodedToken 9 n
            Nothing -> DecodeFailure
#endif
  Word
_    -> forall a. DecodedToken a
DecodeFailure

--
-- 64-on-32 bit code paths
--

#if defined(ARCH_32bit)
tryConsumeWord64 :: Word8 -> ByteString -> DecodedToken Word64
tryConsumeWord64 hdr !bs = case word8ToWord hdr of
  -- Positive integers (type 0)
  0x00 -> DecodedToken 1 0
  0x01 -> DecodedToken 1 1
  0x02 -> DecodedToken 1 2
  0x03 -> DecodedToken 1 3
  0x04 -> DecodedToken 1 4
  0x05 -> DecodedToken 1 5
  0x06 -> DecodedToken 1 6
  0x07 -> DecodedToken 1 7
  0x08 -> DecodedToken 1 8
  0x09 -> DecodedToken 1 9
  0x0a -> DecodedToken 1 10
  0x0b -> DecodedToken 1 11
  0x0c -> DecodedToken 1 12
  0x0d -> DecodedToken 1 13
  0x0e -> DecodedToken 1 14
  0x0f -> DecodedToken 1 15
  0x10 -> DecodedToken 1 16
  0x11 -> DecodedToken 1 17
  0x12 -> DecodedToken 1 18
  0x13 -> DecodedToken 1 19
  0x14 -> DecodedToken 1 20
  0x15 -> DecodedToken 1 21
  0x16 -> DecodedToken 1 22
  0x17 -> DecodedToken 1 23
  0x18 -> DecodedToken 2 $! (word8ToWord64  (eatTailWord8  bs))
  0x19 -> DecodedToken 3 $! (word16ToWord64 (eatTailWord16 bs))
  0x1a -> DecodedToken 5 $! (word32ToWord64 (eatTailWord32 bs))
  0x1b -> DecodedToken 9 $!                 (eatTailWord64 bs)
  _    -> DecodeFailure
{-# INLINE tryConsumeWord64 #-}

tryConsumeNegWord64 :: Word8 -> ByteString -> DecodedToken Word64
tryConsumeNegWord64 hdr !bs = case word8ToWord hdr of
  -- Positive integers (type 0)
  0x20 -> DecodedToken 1 0
  0x21 -> DecodedToken 1 1
  0x22 -> DecodedToken 1 2
  0x23 -> DecodedToken 1 3
  0x24 -> DecodedToken 1 4
  0x25 -> DecodedToken 1 5
  0x26 -> DecodedToken 1 6
  0x27 -> DecodedToken 1 7
  0x28 -> DecodedToken 1 8
  0x29 -> DecodedToken 1 9
  0x2a -> DecodedToken 1 10
  0x2b -> DecodedToken 1 11
  0x2c -> DecodedToken 1 12
  0x2d -> DecodedToken 1 13
  0x2e -> DecodedToken 1 14
  0x2f -> DecodedToken 1 15
  0x30 -> DecodedToken 1 16
  0x31 -> DecodedToken 1 17
  0x32 -> DecodedToken 1 18
  0x33 -> DecodedToken 1 19
  0x34 -> DecodedToken 1 20
  0x35 -> DecodedToken 1 21
  0x36 -> DecodedToken 1 22
  0x37 -> DecodedToken 1 23
  0x38 -> DecodedToken 2 $! (word8ToWord64  (eatTailWord8  bs))
  0x39 -> DecodedToken 3 $! (word16ToWord64 (eatTailWord16 bs))
  0x3a -> DecodedToken 5 $! (word32ToWord64 (eatTailWord32 bs))
  0x3b -> DecodedToken 9 $!                 (eatTailWord64 bs)
  _    -> DecodeFailure
{-# INLINE tryConsumeNegWord64 #-}

tryConsumeInt64 :: Word8 -> ByteString -> DecodedToken Int64
tryConsumeInt64 hdr !bs = case word8ToWord hdr of
  -- Positive integers (type 0)
  0x00 -> DecodedToken 1 0
  0x01 -> DecodedToken 1 1
  0x02 -> DecodedToken 1 2
  0x03 -> DecodedToken 1 3
  0x04 -> DecodedToken 1 4
  0x05 -> DecodedToken 1 5
  0x06 -> DecodedToken 1 6
  0x07 -> DecodedToken 1 7
  0x08 -> DecodedToken 1 8
  0x09 -> DecodedToken 1 9
  0x0a -> DecodedToken 1 10
  0x0b -> DecodedToken 1 11
  0x0c -> DecodedToken 1 12
  0x0d -> DecodedToken 1 13
  0x0e -> DecodedToken 1 14
  0x0f -> DecodedToken 1 15
  0x10 -> DecodedToken 1 16
  0x11 -> DecodedToken 1 17
  0x12 -> DecodedToken 1 18
  0x13 -> DecodedToken 1 19
  0x14 -> DecodedToken 1 20
  0x15 -> DecodedToken 1 21
  0x16 -> DecodedToken 1 22
  0x17 -> DecodedToken 1 23
  0x18 -> DecodedToken 2 $! (word8ToInt64  (eatTailWord8  bs))
  0x19 -> DecodedToken 3 $! (word16ToInt64 (eatTailWord16 bs))
  0x1a -> DecodedToken 5 $! (word32ToInt64 (eatTailWord32 bs))
  0x1b -> case word64ToInt64 (eatTailWord64 bs) of
            Just n  -> DecodedToken 9 n
            Nothing -> DecodeFailure

  -- Negative integers (type 1)
  0x20 -> DecodedToken 1 (-1)
  0x21 -> DecodedToken 1 (-2)
  0x22 -> DecodedToken 1 (-3)
  0x23 -> DecodedToken 1 (-4)
  0x24 -> DecodedToken 1 (-5)
  0x25 -> DecodedToken 1 (-6)
  0x26 -> DecodedToken 1 (-7)
  0x27 -> DecodedToken 1 (-8)
  0x28 -> DecodedToken 1 (-9)
  0x29 -> DecodedToken 1 (-10)
  0x2a -> DecodedToken 1 (-11)
  0x2b -> DecodedToken 1 (-12)
  0x2c -> DecodedToken 1 (-13)
  0x2d -> DecodedToken 1 (-14)
  0x2e -> DecodedToken 1 (-15)
  0x2f -> DecodedToken 1 (-16)
  0x30 -> DecodedToken 1 (-17)
  0x31 -> DecodedToken 1 (-18)
  0x32 -> DecodedToken 1 (-19)
  0x33 -> DecodedToken 1 (-20)
  0x34 -> DecodedToken 1 (-21)
  0x35 -> DecodedToken 1 (-22)
  0x36 -> DecodedToken 1 (-23)
  0x37 -> DecodedToken 1 (-24)
  0x38 -> DecodedToken 2 $! (-1 - word8ToInt64  (eatTailWord8  bs))
  0x39 -> DecodedToken 3 $! (-1 - word16ToInt64 (eatTailWord16 bs))
  0x3a -> DecodedToken 5 $! (-1 - word32ToInt64 (eatTailWord32 bs))
  0x3b -> case word64ToInt64 (eatTailWord64 bs) of
            Just n  -> DecodedToken 9 (-1 - n)
            Nothing -> DecodeFailure
  _    -> DecodeFailure
{-# INLINE tryConsumeInt64 #-}

tryConsumeListLen64 :: Word8 -> ByteString -> DecodedToken Int64
tryConsumeListLen64 hdr !bs = case word8ToWord hdr of
  -- List structures (type 4)
  0x80 -> DecodedToken 1 0
  0x81 -> DecodedToken 1 1
  0x82 -> DecodedToken 1 2
  0x83 -> DecodedToken 1 3
  0x84 -> DecodedToken 1 4
  0x85 -> DecodedToken 1 5
  0x86 -> DecodedToken 1 6
  0x87 -> DecodedToken 1 7
  0x88 -> DecodedToken 1 8
  0x89 -> DecodedToken 1 9
  0x8a -> DecodedToken 1 10
  0x8b -> DecodedToken 1 11
  0x8c -> DecodedToken 1 12
  0x8d -> DecodedToken 1 13
  0x8e -> DecodedToken 1 14
  0x8f -> DecodedToken 1 15
  0x90 -> DecodedToken 1 16
  0x91 -> DecodedToken 1 17
  0x92 -> DecodedToken 1 18
  0x93 -> DecodedToken 1 19
  0x94 -> DecodedToken 1 20
  0x95 -> DecodedToken 1 21
  0x96 -> DecodedToken 1 22
  0x97 -> DecodedToken 1 23
  0x98 -> DecodedToken 2 $! (word8ToInt64  (eatTailWord8  bs))
  0x99 -> DecodedToken 3 $! (word16ToInt64 (eatTailWord16 bs))
  0x9a -> DecodedToken 5 $! (word32ToInt64 (eatTailWord32 bs))
  0x9b -> case word64ToInt64 (eatTailWord64 bs) of
            Just n  -> DecodedToken 9 n
            Nothing -> DecodeFailure
  _    -> DecodeFailure
{-# INLINE tryConsumeListLen64 #-}

tryConsumeMapLen64 :: Word8 -> ByteString -> DecodedToken Int64
tryConsumeMapLen64 hdr !bs = case word8ToWord hdr of
  -- Map structures (type 5)
  0xa0 -> DecodedToken 1 0
  0xa1 -> DecodedToken 1 1
  0xa2 -> DecodedToken 1 2
  0xa3 -> DecodedToken 1 3
  0xa4 -> DecodedToken 1 4
  0xa5 -> DecodedToken 1 5
  0xa6 -> DecodedToken 1 6
  0xa7 -> DecodedToken 1 7
  0xa8 -> DecodedToken 1 8
  0xa9 -> DecodedToken 1 9
  0xaa -> DecodedToken 1 10
  0xab -> DecodedToken 1 11
  0xac -> DecodedToken 1 12
  0xad -> DecodedToken 1 13
  0xae -> DecodedToken 1 14
  0xaf -> DecodedToken 1 15
  0xb0 -> DecodedToken 1 16
  0xb1 -> DecodedToken 1 17
  0xb2 -> DecodedToken 1 18
  0xb3 -> DecodedToken 1 19
  0xb4 -> DecodedToken 1 20
  0xb5 -> DecodedToken 1 21
  0xb6 -> DecodedToken 1 22
  0xb7 -> DecodedToken 1 23
  0xb8 -> DecodedToken 2 $! (word8ToInt64  (eatTailWord8  bs))
  0xb9 -> DecodedToken 3 $! (word16ToInt64 (eatTailWord16 bs))
  0xba -> DecodedToken 5 $! (word32ToInt64 (eatTailWord32 bs))
  0xbb -> case word64ToInt64 (eatTailWord64 bs) of
            Just n  -> DecodedToken 9 n
            Nothing -> DecodeFailure
  _    -> DecodeFailure
{-# INLINE tryConsumeMapLen64 #-}

tryConsumeTag64 :: Word8 -> ByteString -> DecodedToken Word64
tryConsumeTag64 hdr !bs = case word8ToWord hdr of
  -- Tagged values (type 6)
  0xc0 -> DecodedToken 1 0
  0xc1 -> DecodedToken 1 1
  0xc2 -> DecodedToken 1 2
  0xc3 -> DecodedToken 1 3
  0xc4 -> DecodedToken 1 4
  0xc5 -> DecodedToken 1 5
  0xc6 -> DecodedToken 1 6
  0xc7 -> DecodedToken 1 7
  0xc8 -> DecodedToken 1 8
  0xc9 -> DecodedToken 1 9
  0xca -> DecodedToken 1 10
  0xcb -> DecodedToken 1 11
  0xcc -> DecodedToken 1 12
  0xcd -> DecodedToken 1 13
  0xce -> DecodedToken 1 14
  0xcf -> DecodedToken 1 15
  0xd0 -> DecodedToken 1 16
  0xd1 -> DecodedToken 1 17
  0xd2 -> DecodedToken 1 18
  0xd3 -> DecodedToken 1 19
  0xd4 -> DecodedToken 1 20
  0xd5 -> DecodedToken 1 21
  0xd6 -> DecodedToken 1 22
  0xd7 -> DecodedToken 1 23
  0xd8 -> DecodedToken 2 $! (word8ToWord64  (eatTailWord8  bs))
  0xd9 -> DecodedToken 3 $! (word16ToWord64 (eatTailWord16 bs))
  0xda -> DecodedToken 5 $! (word32ToWord64 (eatTailWord32 bs))
  0xdb -> DecodedToken 9 $!                 (eatTailWord64 bs)
  _    -> DecodeFailure
{-# INLINE tryConsumeTag64 #-}
#endif

{-# INLINE tryConsumeFloat #-}
tryConsumeFloat :: Word8 -> ByteString -> DecodedToken Float
tryConsumeFloat :: Word8 -> ByteString -> DecodedToken Float
tryConsumeFloat Word8
hdr !ByteString
bs = case Word8 -> Word
word8ToWord Word8
hdr of
  Word
0xf9 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
3 forall a b. (a -> b) -> a -> b
$! (Word16 -> Float
wordToFloat16 (ByteString -> Word16
eatTailWord16 ByteString
bs))
  Word
0xfa -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
5 forall a b. (a -> b) -> a -> b
$! (Word32 -> Float
wordToFloat32 (ByteString -> Word32
eatTailWord32 ByteString
bs))
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeDouble #-}
tryConsumeDouble :: Word8 -> ByteString -> DecodedToken Double
tryConsumeDouble :: Word8 -> ByteString -> DecodedToken Double
tryConsumeDouble Word8
hdr !ByteString
bs = case Word8 -> Word
word8ToWord Word8
hdr of
  Word
0xf9 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
3 forall a b. (a -> b) -> a -> b
$! (Float -> Double
float2Double forall a b. (a -> b) -> a -> b
$ Word16 -> Float
wordToFloat16 (ByteString -> Word16
eatTailWord16 ByteString
bs))
  Word
0xfa -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
5 forall a b. (a -> b) -> a -> b
$! (Float -> Double
float2Double forall a b. (a -> b) -> a -> b
$ Word32 -> Float
wordToFloat32 (ByteString -> Word32
eatTailWord32 ByteString
bs))
  Word
0xfb -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
9 forall a b. (a -> b) -> a -> b
$!                (Word64 -> Double
wordToFloat64 (ByteString -> Word64
eatTailWord64 ByteString
bs))
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeBool #-}
tryConsumeBool :: Word8 -> DecodedToken Bool
tryConsumeBool :: Word8 -> DecodedToken Bool
tryConsumeBool Word8
hdr = case Word8 -> Word
word8ToWord Word8
hdr of
  Word
0xf4 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Bool
False
  Word
0xf5 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Bool
True
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeSimple #-}
tryConsumeSimple :: Word8 -> ByteString -> DecodedToken Word
tryConsumeSimple :: Word8 -> ByteString -> DecodedToken Word
tryConsumeSimple Word8
hdr !ByteString
bs = case Word8 -> Word
word8ToWord Word8
hdr of

  -- Simple and floats (type 7)
  Word
0xe0 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
0
  Word
0xe1 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
1
  Word
0xe2 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
2
  Word
0xe3 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
3
  Word
0xe4 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
4
  Word
0xe5 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
5
  Word
0xe6 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
6
  Word
0xe7 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
7
  Word
0xe8 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
8
  Word
0xe9 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
9
  Word
0xea -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
10
  Word
0xeb -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
11
  Word
0xec -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
12
  Word
0xed -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
13
  Word
0xee -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
14
  Word
0xef -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
15
  Word
0xf0 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
16
  Word
0xf1 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
17
  Word
0xf2 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
18
  Word
0xf3 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
19
  Word
0xf4 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
20
  Word
0xf5 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
21
  Word
0xf6 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
22
  Word
0xf7 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 Word
23
  Word
0xf8 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
2 forall a b. (a -> b) -> a -> b
$! (Word8 -> Word
word8ToWord (ByteString -> Word8
eatTailWord8 ByteString
bs))
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeBytesIndef #-}
tryConsumeBytesIndef :: Word8 -> DecodedToken ()
tryConsumeBytesIndef :: Word8 -> DecodedToken ()
tryConsumeBytesIndef Word8
hdr = case Word8 -> Word
word8ToWord Word8
hdr of
  Word
0x5f -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 ()
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeStringIndef #-}
tryConsumeStringIndef :: Word8 -> DecodedToken ()
tryConsumeStringIndef :: Word8 -> DecodedToken ()
tryConsumeStringIndef Word8
hdr = case Word8 -> Word
word8ToWord Word8
hdr of
  Word
0x7f -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 ()
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeNull #-}
tryConsumeNull :: Word8 -> DecodedToken ()
tryConsumeNull :: Word8 -> DecodedToken ()
tryConsumeNull Word8
hdr = case Word8 -> Word
word8ToWord Word8
hdr of
  Word
0xf6 -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 ()
  Word
_    -> forall a. DecodedToken a
DecodeFailure


{-# INLINE tryConsumeBreakOr #-}
tryConsumeBreakOr :: Word8 -> DecodedToken ()
tryConsumeBreakOr :: Word8 -> DecodedToken ()
tryConsumeBreakOr Word8
hdr = case Word8 -> Word
word8ToWord Word8
hdr of
  Word
0xff -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 ()
  Word
_    -> forall a. DecodedToken a
DecodeFailure

{-# INLINE readBytesSmall #-}
readBytesSmall :: Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall :: Int -> ByteString -> DecodedToken (LongToken ByteString)
readBytesSmall Int
n ByteString
bs
  -- if n <= bound then ok return it all
  | Int
n forall a. Num a => a -> a -> a
+ Int
hdrsz forall a. Ord a => a -> a -> Bool
<= ByteString -> Int
BS.length ByteString
bs
  = forall a. Int -> a -> DecodedToken a
DecodedToken (Int
nforall a. Num a => a -> a -> a
+Int
hdrsz) forall a b. (a -> b) -> a -> b
$ forall a. Bool -> a -> LongToken a
Fits Bool
True forall a b. (a -> b) -> a -> b
$
      Int -> ByteString -> ByteString
BS.unsafeTake Int
n (Int -> ByteString -> ByteString
BS.unsafeDrop Int
hdrsz ByteString
bs)

  -- if n > bound then slow path, multi-chunk
  | Bool
otherwise
  = forall a. Int -> a -> DecodedToken a
DecodedToken Int
hdrsz forall a b. (a -> b) -> a -> b
$ forall a. Bool -> Int -> LongToken a
TooLong Bool
True Int
n
  where
    hdrsz :: Int
hdrsz = Int
1

{-# INLINE readBytes8 #-}
{-# INLINE readBytes16 #-}
{-# INLINE readBytes32 #-}
{-# INLINE readBytes64 #-}
readBytes8, readBytes16, readBytes32, readBytes64
  :: ByteString -> DecodedToken (LongToken ByteString)

readBytes8 :: ByteString -> DecodedToken (LongToken ByteString)
readBytes8 ByteString
bs
  | Int
n forall a. Ord a => a -> a -> Bool
<= ByteString -> Int
BS.length ByteString
bs forall a. Num a => a -> a -> a
- Int
hdrsz
  = forall a. Int -> a -> DecodedToken a
DecodedToken (Int
nforall a. Num a => a -> a -> a
+Int
hdrsz) forall a b. (a -> b) -> a -> b
$ forall a. Bool -> a -> LongToken a
Fits Bool
lengthCanonical forall a b. (a -> b) -> a -> b
$
      Int -> ByteString -> ByteString
BS.unsafeTake Int
n (Int -> ByteString -> ByteString
BS.unsafeDrop Int
hdrsz ByteString
bs)

  -- if n > bound then slow path, multi-chunk
  | Bool
otherwise
  = forall a. Int -> a -> DecodedToken a
DecodedToken Int
hdrsz forall a b. (a -> b) -> a -> b
$ forall a. Bool -> Int -> LongToken a
TooLong Bool
lengthCanonical Int
n
  where
    hdrsz :: Int
hdrsz           = Int
2
    !n :: Int
n              = Word8 -> Int
word8ToInt (ByteString -> Word8
eatTailWord8 ByteString
bs)
    lengthCanonical :: Bool
lengthCanonical = Int -> Int -> Bool
isIntCanonical Int
hdrsz Int
n

readBytes16 :: ByteString -> DecodedToken (LongToken ByteString)
readBytes16 ByteString
bs
  | Int
n forall a. Ord a => a -> a -> Bool
<= ByteString -> Int
BS.length ByteString
bs forall a. Num a => a -> a -> a
- Int
hdrsz
  = forall a. Int -> a -> DecodedToken a
DecodedToken (Int
nforall a. Num a => a -> a -> a
+Int
hdrsz) forall a b. (a -> b) -> a -> b
$ forall a. Bool -> a -> LongToken a
Fits Bool
lengthCanonical forall a b. (a -> b) -> a -> b
$
      Int -> ByteString -> ByteString
BS.unsafeTake Int
n (Int -> ByteString -> ByteString
BS.unsafeDrop Int
hdrsz ByteString
bs)

  -- if n > bound then slow path, multi-chunk
  | Bool
otherwise
  = forall a. Int -> a -> DecodedToken a
DecodedToken Int
hdrsz forall a b. (a -> b) -> a -> b
$ forall a. Bool -> Int -> LongToken a
TooLong Bool
lengthCanonical Int
n
  where
    hdrsz :: Int
hdrsz           = Int
3
    !n :: Int
n              = Word16 -> Int
word16ToInt (ByteString -> Word16
eatTailWord16 ByteString
bs)
    lengthCanonical :: Bool
lengthCanonical = Int -> Int -> Bool
isIntCanonical Int
hdrsz Int
n

readBytes32 :: ByteString -> DecodedToken (LongToken ByteString)
readBytes32 ByteString
bs = case Word32 -> Int
word32ToInt (ByteString -> Word32
eatTailWord32 ByteString
bs) of
#if defined(ARCH_32bit)
    Just n
#else
    Int
n
#endif
      | Int
n forall a. Ord a => a -> a -> Bool
<= ByteString -> Int
BS.length ByteString
bs forall a. Num a => a -> a -> a
- Int
hdrsz
                  -> forall a. Int -> a -> DecodedToken a
DecodedToken (Int
nforall a. Num a => a -> a -> a
+Int
hdrsz) forall a b. (a -> b) -> a -> b
$ forall a. Bool -> a -> LongToken a
Fits (Int -> Int -> Bool
isIntCanonical Int
hdrsz Int
n) forall a b. (a -> b) -> a -> b
$
                       Int -> ByteString -> ByteString
BS.unsafeTake Int
n (Int -> ByteString -> ByteString
BS.unsafeDrop Int
hdrsz ByteString
bs)

      -- if n > bound then slow path, multi-chunk
      | Bool
otherwise -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
hdrsz forall a b. (a -> b) -> a -> b
$ forall a. Bool -> Int -> LongToken a
TooLong (Int -> Int -> Bool
isIntCanonical Int
hdrsz Int
n) Int
n

#if defined(ARCH_32bit)
    Nothing       -> DecodeFailure
#endif
  where
    hdrsz :: Int
hdrsz = Int
5

readBytes64 :: ByteString -> DecodedToken (LongToken ByteString)
readBytes64 ByteString
bs = case Word64 -> Maybe Int
word64ToInt (ByteString -> Word64
eatTailWord64 ByteString
bs) of
    Just Int
n
      | Int
n forall a. Ord a => a -> a -> Bool
<= ByteString -> Int
BS.length ByteString
bs forall a. Num a => a -> a -> a
- Int
hdrsz
                  -> forall a. Int -> a -> DecodedToken a
DecodedToken (Int
nforall a. Num a => a -> a -> a
+Int
hdrsz) forall a b. (a -> b) -> a -> b
$ forall a. Bool -> a -> LongToken a
Fits (Int -> Int -> Bool
isIntCanonical Int
hdrsz Int
n) forall a b. (a -> b) -> a -> b
$
                            Int -> ByteString -> ByteString
BS.unsafeTake Int
n (Int -> ByteString -> ByteString
BS.unsafeDrop Int
hdrsz ByteString
bs)

      -- if n > bound then slow path, multi-chunk
      | Bool
otherwise -> forall a. Int -> a -> DecodedToken a
DecodedToken Int
hdrsz forall a b. (a -> b) -> a -> b
$ forall a. Bool -> Int -> LongToken a
TooLong (Int -> Int -> Bool
isIntCanonical Int
hdrsz Int
n) Int
n

    Maybe Int
Nothing       -> forall a. DecodedToken a
DecodeFailure
  where
    hdrsz :: Int
hdrsz = Int
9

------------------------------------------------------------------------------
-- Reading big integers
--

-- Big ints consist of two CBOR tokens: a tag token (2 for positive, 3 for
-- negative) followed by a bytes token. Our usual invariant (for go_fast and
-- go_fast_end) only guarantees that we've got enough space to decode the
-- first token. So given that there's two tokens and the second is variable
-- length then there are several points where we can discover we're out of
-- input buffer space.
--
-- In those cases we need to break out of the fast path but we must arrange
-- things so that we can continue later once we've got more input buffer.
--
-- In particular, we might run out of space when:
--   1. trying to decode the header of the second token (bytes); or
--   2. trying to read the bytes body
--
--- The existing mechanisms we've got to drop out of the fast path are:
--   * SlowDecodeAction to re-read a whole token
--   * SlowConsumeTokenBytes to read the body of a bytes token
--
-- Of course when we resume we need to convert the bytes into an integer.
-- Rather than making new fast path return mechanisms we can reuse the
-- existing ones, so long as we're prepared to allocate new continuation
-- closures. This seems a reasonable price to pay to reduce complexity since
-- decoding a big int across an input buffer boundary ought to be rare, and
-- allocating a new continuation closure isn't that expensive.
--
-- Note that canonicity information is calculated lazily. This way we don't need
-- to concern ourselves with two distinct paths, while according to benchmarks
-- it doesn't affect performance in the non-canonical case.

data BigIntToken a = BigIntToken     Bool {- canonical? -} Integer
                   | BigUIntNeedBody Bool {- canonical? -} Int
                   | BigNIntNeedBody Bool {- canonical? -} Int
                   | BigUIntNeedHeader
                   | BigNIntNeedHeader

-- So when we have to break out because we can't read the whole bytes body
-- in one go then we need to use SlowConsumeTokenBytes but we can adjust the
-- continuation so that when we get the ByteString back we convert it to an
-- Integer before calling the original continuation.

adjustContBigUIntNeedBody, adjustContBigNIntNeedBody
  :: (Integer -> ST s (DecodeAction s a))
  -> (ByteString -> ST s (DecodeAction s a))

adjustContBigUIntNeedBody :: forall s a.
(Integer -> ST s (DecodeAction s a))
-> ByteString -> ST s (DecodeAction s a)
adjustContBigUIntNeedBody Integer -> ST s (DecodeAction s a)
k = \ByteString
bs -> Integer -> ST s (DecodeAction s a)
k forall a b. (a -> b) -> a -> b
$! ByteString -> Integer
uintegerFromBytes ByteString
bs
adjustContBigNIntNeedBody :: forall s a.
(Integer -> ST s (DecodeAction s a))
-> ByteString -> ST s (DecodeAction s a)
adjustContBigNIntNeedBody Integer -> ST s (DecodeAction s a)
k = \ByteString
bs -> Integer -> ST s (DecodeAction s a)
k forall a b. (a -> b) -> a -> b
$! ByteString -> Integer
nintegerFromBytes ByteString
bs

adjustContCanonicalBigUIntNeedBody, adjustContCanonicalBigNIntNeedBody
  :: (Integer -> ST s (DecodeAction s a))
  -> (ByteString -> ST s (DecodeAction s a))

adjustContCanonicalBigUIntNeedBody :: forall s a.
(Integer -> ST s (DecodeAction s a))
-> ByteString -> ST s (DecodeAction s a)
adjustContCanonicalBigUIntNeedBody Integer -> ST s (DecodeAction s a)
k = \ByteString
bs ->
  if ByteString -> Bool
isBigIntRepCanonical ByteString
bs
  then Integer -> ST s (DecodeAction s a)
k forall a b. (a -> b) -> a -> b
$! ByteString -> Integer
uintegerFromBytes ByteString
bs
  else forall (f :: * -> *) a. Applicative f => a -> f a
pure forall a b. (a -> b) -> a -> b
$! forall s a. String -> DecodeAction s a
D.Fail (String
"non-canonical integer")

adjustContCanonicalBigNIntNeedBody :: forall s a.
(Integer -> ST s (DecodeAction s a))
-> ByteString -> ST s (DecodeAction s a)
adjustContCanonicalBigNIntNeedBody Integer -> ST s (DecodeAction s a)
k = \ByteString
bs ->
  if ByteString -> Bool
isBigIntRepCanonical ByteString
bs
  then Integer -> ST s (DecodeAction s a)
k forall a b. (a -> b) -> a -> b
$! ByteString -> Integer
nintegerFromBytes ByteString
bs
  else forall (f :: * -> *) a. Applicative f => a -> f a
pure forall a b. (a -> b) -> a -> b
$! forall s a. String -> DecodeAction s a
D.Fail (String
"non-canonical integer")

-- And when we have to break out because we can't read the bytes token header
-- in one go then we need to use SlowDecodeAction but we have to make two
-- adjustments. When we resume we need to read a bytes token, not a big int.
-- That is we don't want to re-read the tag token. Indeed we cannot even if we
-- wanted to because the slow path code only guarantees to arrange for one
-- complete token header in the input buffer. So we must pretend that we did
-- in fact want to read a bytes token using ConsumeBytes, and then we can
-- adjust the continuation for that in the same way as above.

adjustContBigUIntNeedHeader, adjustContBigNIntNeedHeader
  :: (Integer -> ST s (DecodeAction s a))
  -> DecodeAction s a

adjustContBigUIntNeedHeader :: forall s a.
(Integer -> ST s (DecodeAction s a)) -> DecodeAction s a
adjustContBigUIntNeedHeader Integer -> ST s (DecodeAction s a)
k = forall s a.
(ByteString -> ST s (DecodeAction s a)) -> DecodeAction s a
ConsumeBytes (\ByteString
bs -> Integer -> ST s (DecodeAction s a)
k forall a b. (a -> b) -> a -> b
$! ByteString -> Integer
uintegerFromBytes ByteString
bs)
adjustContBigNIntNeedHeader :: forall s a.
(Integer -> ST s (DecodeAction s a)) -> DecodeAction s a
adjustContBigNIntNeedHeader Integer -> ST s (DecodeAction s a)
k = forall s a.
(ByteString -> ST s (DecodeAction s a)) -> DecodeAction s a
ConsumeBytes (\ByteString
bs -> Integer -> ST s (DecodeAction s a)
k forall a b. (a -> b) -> a -> b
$! ByteString -> Integer
nintegerFromBytes ByteString
bs)

adjustContCanonicalBigUIntNeedHeader, adjustContCanonicalBigNIntNeedHeader
  :: (Integer -> ST s (DecodeAction s a))
  -> DecodeAction s a

adjustContCanonicalBigUIntNeedHeader :: forall s a.
(Integer -> ST s (DecodeAction s a)) -> DecodeAction s a
adjustContCanonicalBigUIntNeedHeader Integer -> ST s (DecodeAction s a)
k = forall s a.
(ByteString -> ST s (DecodeAction s a)) -> DecodeAction s a
ConsumeBytesCanonical forall a b. (a -> b) -> a -> b
$ \ByteString
bs ->
  if ByteString -> Bool
isBigIntRepCanonical ByteString
bs
  then Integer -> ST s (DecodeAction s a)
k forall a b. (a -> b) -> a -> b
$! ByteString -> Integer
uintegerFromBytes ByteString
bs
  else forall (f :: * -> *) a. Applicative f => a -> f a
pure forall a b. (a -> b) -> a -> b
$! forall s a. String -> DecodeAction s a
D.Fail (String
"non-canonical integer")

adjustContCanonicalBigNIntNeedHeader :: forall s a.
(Integer -> ST s (DecodeAction s a)) -> DecodeAction s a
adjustContCanonicalBigNIntNeedHeader Integer -> ST s (DecodeAction s a)
k = forall s a.
(ByteString -> ST s (DecodeAction s a)) -> DecodeAction s a
ConsumeBytesCanonical forall a b. (a -> b) -> a -> b
$ \ByteString
bs ->
  if ByteString -> Bool
isBigIntRepCanonical ByteString
bs
  then Integer -> ST s (DecodeAction s a)
k forall a b. (a -> b) -> a -> b
$! ByteString -> Integer
nintegerFromBytes ByteString
bs
  else forall (f :: * -> *) a. Applicative f => a -> f a
pure forall a b. (a -> b) -> a -> b
$! forall s a. String -> DecodeAction s a
D.Fail (String
"non-canonical integer")

-- So finally when reading the input buffer we check if we have enough space
-- to read the header of the bytes token and then try to read the bytes body,
-- using the appropriate break-out codes as above.

{-# INLINE readBigUInt #-}
readBigUInt :: ByteString -> DecodedToken (BigIntToken a)
readBigUInt :: forall a. ByteString -> DecodedToken (BigIntToken a)
readBigUInt ByteString
bs
    | let bs' :: ByteString
bs' = ByteString -> ByteString
BS.unsafeTail ByteString
bs
    , Bool -> Bool
not (ByteString -> Bool
BS.null ByteString
bs')
    , let !hdr :: Word8
hdr = ByteString -> Word8
BS.unsafeHead ByteString
bs'
    , ByteString -> Int
BS.length ByteString
bs' forall a. Ord a => a -> a -> Bool
>= Word8 -> Int
tokenSize Word8
hdr
    = case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeBytes Word8
hdr ByteString
bs' of
        DecodedToken (LongToken ByteString)
DecodeFailure                           -> forall a. DecodedToken a
DecodeFailure
        DecodedToken Int
sz (Fits Bool
canonical ByteString
bstr)   -> forall a. Int -> a -> DecodedToken a
DecodedToken (Int
1forall a. Num a => a -> a -> a
+Int
sz)
          (forall a. Bool -> Integer -> BigIntToken a
BigIntToken (Bool
canonical Bool -> Bool -> Bool
&& ByteString -> Bool
isBigIntRepCanonical ByteString
bstr)
                       (ByteString -> Integer
uintegerFromBytes ByteString
bstr))
        DecodedToken Int
sz (TooLong Bool
canonical Int
len) ->
          forall a. Int -> a -> DecodedToken a
DecodedToken (Int
1forall a. Num a => a -> a -> a
+Int
sz) (forall a. Bool -> Int -> BigIntToken a
BigUIntNeedBody Bool
canonical Int
len)

    | Bool
otherwise
    = forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 forall a. BigIntToken a
BigUIntNeedHeader

{-# INLINE readBigNInt #-}
readBigNInt :: ByteString -> DecodedToken (BigIntToken a)
readBigNInt :: forall a. ByteString -> DecodedToken (BigIntToken a)
readBigNInt ByteString
bs
    | let bs' :: ByteString
bs' = ByteString -> ByteString
BS.unsafeTail ByteString
bs
    , Bool -> Bool
not (ByteString -> Bool
BS.null ByteString
bs')
    , let !hdr :: Word8
hdr = ByteString -> Word8
BS.unsafeHead ByteString
bs'
    , ByteString -> Int
BS.length ByteString
bs' forall a. Ord a => a -> a -> Bool
>= Word8 -> Int
tokenSize Word8
hdr
    = case Word8 -> ByteString -> DecodedToken (LongToken ByteString)
tryConsumeBytes Word8
hdr ByteString
bs' of
        DecodedToken (LongToken ByteString)
DecodeFailure                           -> forall a. DecodedToken a
DecodeFailure
        DecodedToken Int
sz (Fits Bool
canonical ByteString
bstr)   -> forall a. Int -> a -> DecodedToken a
DecodedToken (Int
1forall a. Num a => a -> a -> a
+Int
sz)
          (forall a. Bool -> Integer -> BigIntToken a
BigIntToken (Bool
canonical Bool -> Bool -> Bool
&& ByteString -> Bool
isBigIntRepCanonical ByteString
bstr)
                       (ByteString -> Integer
nintegerFromBytes ByteString
bstr))
        DecodedToken Int
sz (TooLong Bool
canonical Int
len) ->
          forall a. Int -> a -> DecodedToken a
DecodedToken (Int
1forall a. Num a => a -> a -> a
+Int
sz) (forall a. Bool -> Int -> BigIntToken a
BigNIntNeedBody Bool
canonical Int
len)

    | Bool
otherwise
    = forall a. Int -> a -> DecodedToken a
DecodedToken Int
1 forall a. BigIntToken a
BigNIntNeedHeader

-- Binary representation of a big integer is canonical if it's at least 9 bytes
-- long (as for smaller values the canonical representation is the same one as
-- for Int) and the leading byte is not zero (meaning that it's the smallest
-- representation for the number in question).
isBigIntRepCanonical :: ByteString -> Bool
isBigIntRepCanonical :: ByteString -> Bool
isBigIntRepCanonical ByteString
bstr = ByteString -> Int
BS.length ByteString
bstr forall a. Ord a => a -> a -> Bool
> Int
8 Bool -> Bool -> Bool
&& ByteString -> Word8
BS.unsafeHead ByteString
bstr forall a. Eq a => a -> a -> Bool
/= Word8
0x00