[web] Make Canvaskit's malloc more useful

Author: mdebbar

lib/web_ui/lib/src/engine/canvaskit/canvaskit_api.dart

@@ -1273,34 +1273,46 @@ Float32List toSkColorStops(List<double>? colorStops) {
   return skColorStops;
 }
 
-@JS('Float32Array')
-external _NativeFloat32ArrayType get _nativeFloat32ArrayType;
-
 @JS()
 @staticInterop
-class _NativeFloat32ArrayType {}
+abstract class _ArrayType {}
+
+@JS('Float32Array')
+external _ArrayType get _nativeFloat32ArrayType;
+
+@JS('Uint32Array')
+external _ArrayType get _nativeUint32ArrayType;
 
 @JS('window.flutterCanvasKit.Malloc')
-external SkFloat32List _mallocFloat32List(
-  _NativeFloat32ArrayType float32ListType,
-  int size,
-);
+external Object _mallocList(_ArrayType arrayType, int size);
 
 /// Allocates a [Float32List] backed by WASM memory, managed by
 /// a [SkFloat32List].
 ///
-/// To free the allocated array use [freeFloat32List].
+/// To free the allocated array use [freeList].
 SkFloat32List mallocFloat32List(int size) {
-  return _mallocFloat32List(_nativeFloat32ArrayType, size);
+  return _mallocList(_nativeFloat32ArrayType, size) as SkFloat32List;
+}
+
+/// Allocates a [Uint32List] backed by WASM memory, managed by
+/// a [SkUint32List].
+///
+/// To free the allocated array use [freeList].
+SkUint32List mallocUint32List(int size) {
+  return _mallocList(_nativeUint32ArrayType, size) as SkUint32List;
 }
 
-/// Frees the WASM memory occupied by a [SkFloat32List].
+/// Frees the WASM memory occupied by a [SkFloat32List] or [SkUint32List].
 ///
 /// The [list] is no longer usable after calling this function.
 ///
 /// Use this function to free lists owned by the engine.
 @JS('window.flutterCanvasKit.Free')
-external void freeFloat32List(SkFloat32List list);
+external void freeList(_SkList list);
+
+@JS()
+@staticInterop
+abstract class _SkList {}
 
 /// Wraps a [Float32List] backed by WASM memory.
 ///
@@ -1309,7 +1321,7 @@ external void freeFloat32List(SkFloat32List list);
 /// that's attached to the current WASM memory block.
 @JS()
 @staticInterop
-class SkFloat32List {}
+class SkFloat32List extends _SkList {}
 
 extension SkFloat32ListExtension on SkFloat32List {
   /// Returns the [Float32List] object backed by WASM memory.
@@ -1322,6 +1334,26 @@ extension SkFloat32ListExtension on SkFloat32List {
   external Float32List toTypedArray();
 }
 
+/// Wraps a [Uint32List] backed by WASM memory.
+///
+/// This wrapper is necessary because the raw [Uint32List] will get detached
+/// when WASM grows its memory. Call [toTypedArray] to get a new instance
+/// that's attached to the current WASM memory block.
+@JS()
+@staticInterop
+class SkUint32List extends _SkList {}
+
+extension SkUint32ListExtension on SkUint32List {
+  /// Returns the [Uint32List] object backed by WASM memory.
+  ///
+  /// Do not reuse the returned list across multiple WASM function/method
+  /// invocations that may lead to WASM memory to grow. When WASM memory
+  /// grows the [Uint32List] object becomes "detached" and is no longer
+  /// usable. Instead, call this method every time you need to read from
+  /// or write to the list.
+  external Uint32List toTypedArray();
+}
+
 /// Writes [color] information into the given [skColor] buffer.
 Float32List _populateSkColor(SkFloat32List skColor, ui.Color color) {
   final Float32List array = skColor.toTypedArray();
@@ -1590,7 +1622,7 @@ Float32List toOuterSkRect(ui.RRect rrect) {
 /// Uses `CanvasKit.Malloc` to allocate storage for the points in the WASM
 /// memory to avoid unnecessary copying. Unless CanvasKit takes ownership of
 /// the list the returned list must be explicitly freed using
-/// [freeMallocedFloat32List].
+/// [freeList].
 SkFloat32List toMallocedSkPoints(List<ui.Offset> points) {
   final int len = points.length;
   final SkFloat32List skPoints = mallocFloat32List(len * 2);

lib/web_ui/lib/src/engine/canvaskit/canvaskit_canvas.dart

@@ -326,7 +326,7 @@ class CanvasKitCanvas implements ui.Canvas {
       pointMode,
       skPoints.toTypedArray(),
     );
-    freeFloat32List(skPoints);
+    freeList(skPoints);
   }
 
   @override

lib/web_ui/lib/src/engine/canvaskit/path.dart

@@ -89,7 +89,7 @@ class CkPath extends ManagedSkiaObject<SkPath> implements ui.Path {
     assert(points != null);
     final SkFloat32List encodedPoints = toMallocedSkPoints(points);
     skiaObject.addPoly(encodedPoints.toTypedArray(), close);
-    freeFloat32List(encodedPoints);
+    freeList(encodedPoints);
   }
 
   @override

lib/web_ui/test/canvaskit/canvaskit_api_test.dart

@@ -812,7 +812,7 @@ void _pathTests() {
       ui.Offset(10, 10),
     ]);
     path.addPoly(encodedPoints.toTypedArray(), true);
-    freeFloat32List(encodedPoints);
+    freeList(encodedPoints);
   });
 
   test('addRRect', () {