Fix more allocations found by tests

Author: tannewt

extmod/modasyncio.c

@@ -179,7 +179,8 @@ mp_obj_t mp_asyncio_context = MP_OBJ_NULL;
 
 static mp_obj_t task_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *args) {
     mp_arg_check_num(n_args, n_kw, 1, 2, false);
-    mp_obj_task_t *self = m_new_obj(mp_obj_task_t);
+    // CIRCUITPY-CHANGE: Task holds onto core and data so collect it.
+    mp_obj_task_t *self = m_malloc_with_collect(sizeof(mp_obj_task_t));
     self->pairheap.base.type = type;
     mp_pairheap_init_node(task_lt, &self->pairheap);
     self->coro = args[0];

extmod/vfs.c

@@ -237,7 +237,8 @@ mp_obj_t mp_vfs_mount(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args
     }
 
     // create new object
-    mp_vfs_mount_t *vfs = m_new_obj(mp_vfs_mount_t);
+    // CIRCUITPY-CHANGE: Collect the mount object because it references others
+    mp_vfs_mount_t *vfs = m_malloc_with_collect(sizeof(mp_vfs_mount_t));
     vfs->str = mnt_str;
     vfs->len = mnt_len;
     vfs->obj = vfs_obj;

ports/unix/alloc.c

@@ -58,7 +58,8 @@ void mp_unix_alloc_exec(size_t min_size, void **ptr, size_t *size) {
     }
 
     // add new link to the list of mmap'd regions
-    mmap_region_t *rg = m_new_obj(mmap_region_t);
+    // CIRCUITPY-CHANGE: Collect the mmap region because it points to others.
+    mmap_region_t *rg = m_malloc_with_collect(sizeof(mmap_region_t));
     rg->ptr = *ptr;
     rg->len = min_size;
     rg->next = MP_STATE_VM(mmap_region_head);

py/map.c

@@ -87,7 +87,7 @@ static size_t get_hash_alloc_greater_or_equal_to(size_t x) {
 /* map                                                                        */
 
 // CIRCUITPY-CHANGE: Helper for allocating tables of elements
-#define malloc_table(num) m_malloc_helper(sizeof(mp_map_elem_t) * (num), M_MALLOC_COLLECT)
+#define malloc_table(num) m_malloc_helper(sizeof(mp_map_elem_t) * (num), M_MALLOC_COLLECT | M_MALLOC_RAISE_ERROR | M_MALLOC_ENSURE_ZEROED)
 
 void mp_map_init(mp_map_t *map, size_t n) {
     if (n == 0) {

py/pystack.h

@@ -82,7 +82,8 @@ static inline void mp_local_free(void *ptr) {
 }
 
 static inline void *mp_nonlocal_alloc(size_t n_bytes) {
-    return m_new(uint8_t, n_bytes);
+    // CIRCUITPY-CHANGE: Collect the allocated memory because it holds function arguments.
+    return m_malloc_with_collect(n_bytes);
 }
 
 static inline void *mp_nonlocal_realloc(void *ptr, size_t old_n_bytes, size_t new_n_bytes) {