mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
cf3c131e7c
We remove the existing `swift_reflection_iterateAsyncTaskAllocations` API that attempts to provide all necessary information about a tasks's allocations starting from the task. Instead, we split it into two pieces: `swift_reflection_asyncTaskSlabPointer` to get the first slab for a task, and `+swift_reflection_asyncTaskSlabAllocations` to get the allocations in a slab, and a pointer to the next slab. We also add a dummy metadata pointer to the beginning of each slab. This allows tools to identify slab allocations on the heap without needing to locate every single async task object. They can then use `swift_reflection_asyncTaskSlabAllocations` on such allocations to find out about the contents. rdar://82549631
121 lines
4.0 KiB
C++
121 lines
4.0 KiB
C++
//===--- StackAllocator.cpp - Unit tests for the StackAllocator -----------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See https://swift.org/LICENSE.txt for license information
|
|
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "../../stdlib/public/runtime/StackAllocator.h"
|
|
#include "swift/ABI/Metadata.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace swift;
|
|
|
|
static constexpr size_t slabCapacity = 256;
|
|
static constexpr size_t firstSlabBufferCapacity = 140;
|
|
static constexpr size_t fitsIntoFirstSlab = 16;
|
|
static constexpr size_t fitsIntoSlab = slabCapacity - 16;
|
|
static constexpr size_t twoFitIntoSlab = slabCapacity / 2 - 32;
|
|
static constexpr size_t exceedsSlab = slabCapacity + 16;
|
|
|
|
static Metadata SlabMetadata;
|
|
|
|
TEST(StackAllocatorTest, withPreallocatedSlab) {
|
|
|
|
char firstSlab[firstSlabBufferCapacity];
|
|
StackAllocator<slabCapacity, &SlabMetadata> allocator(
|
|
firstSlab, firstSlabBufferCapacity);
|
|
|
|
char *mem1 = (char *)allocator.alloc(fitsIntoFirstSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 0);
|
|
char *mem1a = (char *)allocator.alloc(fitsIntoFirstSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 0);
|
|
|
|
char *mem2 = (char *)allocator.alloc(exceedsSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 1);
|
|
|
|
char *mem3 = (char *)allocator.alloc(fitsIntoSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 2);
|
|
|
|
char *mem4 = (char *)allocator.alloc(fitsIntoSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 3);
|
|
|
|
allocator.dealloc(mem4);
|
|
allocator.dealloc(mem3);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 3);
|
|
|
|
char *mem5 = (char *)allocator.alloc(twoFitIntoSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 3);
|
|
char *mem6 = (char *)allocator.alloc(twoFitIntoSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 3);
|
|
char *mem7 = (char *)allocator.alloc(twoFitIntoSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 3);
|
|
|
|
allocator.dealloc(mem7);
|
|
allocator.dealloc(mem6);
|
|
allocator.dealloc(mem5);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 3);
|
|
|
|
char *mem8 = (char *)allocator.alloc(exceedsSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 2);
|
|
|
|
allocator.dealloc(mem8);
|
|
allocator.dealloc(mem2);
|
|
allocator.dealloc(mem1a);
|
|
allocator.dealloc(mem1);
|
|
}
|
|
|
|
TEST(StackAllocatorTest, withoutPreallocatedSlab) {
|
|
|
|
constexpr size_t slabCapacity = 256;
|
|
|
|
StackAllocator<slabCapacity, &SlabMetadata> allocator;
|
|
|
|
size_t fitsIntoSlab = slabCapacity - 16;
|
|
size_t twoFitIntoSlab = slabCapacity / 2 - 32;
|
|
size_t exceedsSlab = slabCapacity + 16;
|
|
|
|
char *mem1 = (char *)allocator.alloc(twoFitIntoSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 1);
|
|
char *mem1a = (char *)allocator.alloc(twoFitIntoSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 1);
|
|
|
|
char *mem2 = (char *)allocator.alloc(exceedsSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 2);
|
|
|
|
char *mem3 = (char *)allocator.alloc(fitsIntoSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 3);
|
|
|
|
char *mem4 = (char *)allocator.alloc(fitsIntoSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 4);
|
|
|
|
allocator.dealloc(mem4);
|
|
allocator.dealloc(mem3);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 4);
|
|
|
|
char *mem5 = (char *)allocator.alloc(twoFitIntoSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 4);
|
|
char *mem6 = (char *)allocator.alloc(twoFitIntoSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 4);
|
|
char *mem7 = (char *)allocator.alloc(twoFitIntoSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 4);
|
|
|
|
allocator.dealloc(mem7);
|
|
allocator.dealloc(mem6);
|
|
allocator.dealloc(mem5);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 4);
|
|
|
|
char *mem8 = (char *)allocator.alloc(exceedsSlab);
|
|
EXPECT_EQ(allocator.getNumAllocatedSlabs(), 3);
|
|
|
|
allocator.dealloc(mem8);
|
|
allocator.dealloc(mem2);
|
|
allocator.dealloc(mem1a);
|
|
allocator.dealloc(mem1);
|
|
}
|