Fix

Author: adithyaov

core/src/Streamly/Internal/Data/MutArray/Generic.hs

@@ -226,7 +226,7 @@ data MutArray a =
         , arrEnd :: {-# UNPACK #-}!Int
           -- ^ The length of this slice.
 
-        , arrCapacity :: {-# UNPACK #-}!Int
+        , arrBound :: {-# UNPACK #-}!Int
           -- ^ first invalid index of arrContents.
         }
 
@@ -386,7 +386,7 @@ realloc n arr = do
         $ IO
         $ \s# ->
               let s1# = copyMutableArray# arrC# arrS# arr1C# arr1S# newLen# s#
-               in (# s1#, arr1 {arrEnd = newEnd, arrCapacity = n} #)
+               in (# s1#, arr1 {arrEnd = newEnd, arrBound = n} #)
 
 reallocWith ::
        MonadIO m => String -> (Int -> Int) -> Int -> MutArray a -> m (MutArray a)
@@ -420,9 +420,9 @@ reallocWith label sizer reqSize arr = do
 {-# INLINE unsafeSnoc #-}
 unsafeSnoc, snocUnsafe :: MonadIO m => MutArray a -> a -> m (MutArray a)
 unsafeSnoc arr@MutArray {..} a = do
-    assert (arrEnd < arrCapacity) (return ())
+    assert (arrEnd < arrBound) (return ())
     let arr1 = arr {arrEnd = arrEnd + 1}
-    unsafePutIndex arrEnd arr1 a
+    unsafePutIndex (length arr) arr1 a
     return arr1
 
 -- NOINLINE to move it out of the way and not pollute the instruction cache.
@@ -445,7 +445,7 @@ snocWithRealloc sizer arr x = do
 {-# INLINE snocWith #-}
 snocWith :: MonadIO m => (Int -> Int) -> MutArray a -> a -> m (MutArray a)
 snocWith sizer arr@MutArray {..} x = do
-    if arrEnd < arrCapacity
+    if arrEnd < arrBound
     then unsafeSnoc arr x
     else snocWithRealloc sizer arr x
 
@@ -476,7 +476,7 @@ snoc = snocWith (* 2)
 {-# INLINE uninit #-}
 uninit :: MonadIO m => MutArray a -> Int -> m (MutArray a)
 uninit arr@MutArray{..} len =
-    if arrEnd + len <= arrCapacity
+    if arrEnd + len <= arrBound
     then return $ arr {arrEnd = arrEnd + len}
     else realloc (length arr + len) arr
 
@@ -784,7 +784,7 @@ chunksOf n (D.Stream step state) =
                 putIndexUnsafeWith end contents x
                 let end1 = end + 1
                 return $
-                    if start + end1 >= bound
+                    if end1 >= bound
                     then D.Skip
                             (GroupYield
                                 contents start end1 bound (GroupStart s))