mirror of
https://github.com/golang/go.git
synced 2025-12-08 06:10:04 +00:00
runtime: add new page allocator core
This change adds a new bitmap-based allocator to the runtime with tests. It does not yet integrate the page allocator into the runtime and thus this change is almost purely additive. Updates #35112. Change-Id: Ic3d024c28abee8be8797d3918116a80f901cc2bf Reviewed-on: https://go-review.googlesource.com/c/go/+/190622 Run-TryBot: Michael Knyszek <mknyszek@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Austin Clements <austin@google.com>
This commit is contained in:
Michael Anthony Knyszek
Michael Knyszek
parent
05aa4a7b74
commit
39e8cb0faa
6 changed files with 1716 additions and 2 deletions
658
src/runtime/mpagealloc_test.go
Normal file
658
src/runtime/mpagealloc_test.go
Normal file
|
|
@ -0,0 +1,658 @@
|
|||
// Copyright 2019 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package runtime_test
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
. "runtime"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func checkPageAlloc(t *testing.T, want, got *PageAlloc) {
|
||||
// Ensure start and end are correct.
|
||||
wantStart, wantEnd := want.Bounds()
|
||||
gotStart, gotEnd := got.Bounds()
|
||||
if gotStart != wantStart {
|
||||
t.Fatalf("start values not equal: got %d, want %d", gotStart, wantStart)
|
||||
}
|
||||
if gotEnd != wantEnd {
|
||||
t.Fatalf("end values not equal: got %d, want %d", gotEnd, wantEnd)
|
||||
}
|
||||
|
||||
for i := gotStart; i < gotEnd; i++ {
|
||||
// Check the bitmaps.
|
||||
if !checkPallocBits(t, got.PallocBits(i), want.PallocBits(i)) {
|
||||
t.Logf("in chunk %d", i)
|
||||
}
|
||||
}
|
||||
// TODO(mknyszek): Verify summaries too?
|
||||
}
|
||||
|
||||
func TestPageAllocAlloc(t *testing.T) {
|
||||
type hit struct {
|
||||
npages, base uintptr
|
||||
}
|
||||
tests := map[string]struct {
|
||||
before map[ChunkIdx][]BitRange
|
||||
after map[ChunkIdx][]BitRange
|
||||
hits []hit
|
||||
}{
|
||||
"AllFree1": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
},
|
||||
hits: []hit{
|
||||
{1, PageBase(BaseChunkIdx, 0)},
|
||||
{1, PageBase(BaseChunkIdx, 1)},
|
||||
{1, PageBase(BaseChunkIdx, 2)},
|
||||
{1, PageBase(BaseChunkIdx, 3)},
|
||||
{1, PageBase(BaseChunkIdx, 4)},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, 5}},
|
||||
},
|
||||
},
|
||||
"ManyArena1": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 2: {{0, PallocChunkPages - 1}},
|
||||
},
|
||||
hits: []hit{
|
||||
{1, PageBase(BaseChunkIdx+2, PallocChunkPages-1)},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 2: {{0, PallocChunkPages}},
|
||||
},
|
||||
},
|
||||
"NotContiguous1": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 0xff: {{0, 0}},
|
||||
},
|
||||
hits: []hit{
|
||||
{1, PageBase(BaseChunkIdx+0xff, 0)},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 0xff: {{0, 1}},
|
||||
},
|
||||
},
|
||||
"AllFree2": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
},
|
||||
hits: []hit{
|
||||
{2, PageBase(BaseChunkIdx, 0)},
|
||||
{2, PageBase(BaseChunkIdx, 2)},
|
||||
{2, PageBase(BaseChunkIdx, 4)},
|
||||
{2, PageBase(BaseChunkIdx, 6)},
|
||||
{2, PageBase(BaseChunkIdx, 8)},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, 10}},
|
||||
},
|
||||
},
|
||||
"Straddle2": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages - 1}},
|
||||
BaseChunkIdx + 1: {{1, PallocChunkPages - 1}},
|
||||
},
|
||||
hits: []hit{
|
||||
{2, PageBase(BaseChunkIdx, PallocChunkPages-1)},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages}},
|
||||
},
|
||||
},
|
||||
"AllFree5": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
},
|
||||
hits: []hit{
|
||||
{5, PageBase(BaseChunkIdx, 0)},
|
||||
{5, PageBase(BaseChunkIdx, 5)},
|
||||
{5, PageBase(BaseChunkIdx, 10)},
|
||||
{5, PageBase(BaseChunkIdx, 15)},
|
||||
{5, PageBase(BaseChunkIdx, 20)},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, 25}},
|
||||
},
|
||||
},
|
||||
"AllFree64": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
},
|
||||
hits: []hit{
|
||||
{64, PageBase(BaseChunkIdx, 0)},
|
||||
{64, PageBase(BaseChunkIdx, 64)},
|
||||
{64, PageBase(BaseChunkIdx, 128)},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, 192}},
|
||||
},
|
||||
},
|
||||
"AllFree65": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
},
|
||||
hits: []hit{
|
||||
{65, PageBase(BaseChunkIdx, 0)},
|
||||
{65, PageBase(BaseChunkIdx, 65)},
|
||||
{65, PageBase(BaseChunkIdx, 130)},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, 195}},
|
||||
},
|
||||
},
|
||||
// TODO(mknyszek): Add tests close to the chunk size.
|
||||
"ExhaustPallocChunkPages-3": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
},
|
||||
hits: []hit{
|
||||
{PallocChunkPages - 3, PageBase(BaseChunkIdx, 0)},
|
||||
{PallocChunkPages - 3, 0},
|
||||
{1, PageBase(BaseChunkIdx, PallocChunkPages-3)},
|
||||
{2, PageBase(BaseChunkIdx, PallocChunkPages-2)},
|
||||
{1, 0},
|
||||
{PallocChunkPages - 3, 0},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
},
|
||||
},
|
||||
"AllFreePallocChunkPages": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
},
|
||||
hits: []hit{
|
||||
{PallocChunkPages, PageBase(BaseChunkIdx, 0)},
|
||||
{PallocChunkPages, 0},
|
||||
{1, 0},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
},
|
||||
},
|
||||
"StraddlePallocChunkPages": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages / 2}},
|
||||
BaseChunkIdx + 1: {{PallocChunkPages / 2, PallocChunkPages / 2}},
|
||||
},
|
||||
hits: []hit{
|
||||
{PallocChunkPages, PageBase(BaseChunkIdx, PallocChunkPages/2)},
|
||||
{PallocChunkPages, 0},
|
||||
{1, 0},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages}},
|
||||
},
|
||||
},
|
||||
"StraddlePallocChunkPages+1": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages / 2}},
|
||||
BaseChunkIdx + 1: {},
|
||||
},
|
||||
hits: []hit{
|
||||
{PallocChunkPages + 1, PageBase(BaseChunkIdx, PallocChunkPages/2)},
|
||||
{PallocChunkPages, 0},
|
||||
{1, PageBase(BaseChunkIdx+1, PallocChunkPages/2+1)},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages/2 + 2}},
|
||||
},
|
||||
},
|
||||
"AllFreePallocChunkPages*2": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
BaseChunkIdx + 1: {},
|
||||
},
|
||||
hits: []hit{
|
||||
{PallocChunkPages * 2, PageBase(BaseChunkIdx, 0)},
|
||||
{PallocChunkPages * 2, 0},
|
||||
{1, 0},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages}},
|
||||
},
|
||||
},
|
||||
"NotContiguousPallocChunkPages*2": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
BaseChunkIdx + 0x100: {},
|
||||
BaseChunkIdx + 0x101: {},
|
||||
},
|
||||
hits: []hit{
|
||||
{PallocChunkPages * 2, PageBase(BaseChunkIdx+0x100, 0)},
|
||||
{21, PageBase(BaseChunkIdx, 0)},
|
||||
{1, PageBase(BaseChunkIdx, 21)},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, 22}},
|
||||
BaseChunkIdx + 0x100: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 0x101: {{0, PallocChunkPages}},
|
||||
},
|
||||
},
|
||||
"StraddlePallocChunkPages*2": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages / 2}},
|
||||
BaseChunkIdx + 1: {},
|
||||
BaseChunkIdx + 2: {{PallocChunkPages / 2, PallocChunkPages / 2}},
|
||||
},
|
||||
hits: []hit{
|
||||
{PallocChunkPages * 2, PageBase(BaseChunkIdx, PallocChunkPages/2)},
|
||||
{PallocChunkPages * 2, 0},
|
||||
{1, 0},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 2: {{0, PallocChunkPages}},
|
||||
},
|
||||
},
|
||||
"StraddlePallocChunkPages*5/4": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages * 3 / 4}},
|
||||
BaseChunkIdx + 2: {{0, PallocChunkPages * 3 / 4}},
|
||||
BaseChunkIdx + 3: {{0, 0}},
|
||||
},
|
||||
hits: []hit{
|
||||
{PallocChunkPages * 5 / 4, PageBase(BaseChunkIdx+2, PallocChunkPages*3/4)},
|
||||
{PallocChunkPages * 5 / 4, 0},
|
||||
{1, PageBase(BaseChunkIdx+1, PallocChunkPages*3/4)},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages*3/4 + 1}},
|
||||
BaseChunkIdx + 2: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 3: {{0, PallocChunkPages}},
|
||||
},
|
||||
},
|
||||
"AllFreePallocChunkPages*7+5": {
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
BaseChunkIdx + 1: {},
|
||||
BaseChunkIdx + 2: {},
|
||||
BaseChunkIdx + 3: {},
|
||||
BaseChunkIdx + 4: {},
|
||||
BaseChunkIdx + 5: {},
|
||||
BaseChunkIdx + 6: {},
|
||||
BaseChunkIdx + 7: {},
|
||||
},
|
||||
hits: []hit{
|
||||
{PallocChunkPages*7 + 5, PageBase(BaseChunkIdx, 0)},
|
||||
{PallocChunkPages*7 + 5, 0},
|
||||
{1, PageBase(BaseChunkIdx+7, 5)},
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 2: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 3: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 4: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 5: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 6: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 7: {{0, 6}},
|
||||
},
|
||||
},
|
||||
}
|
||||
for name, v := range tests {
|
||||
v := v
|
||||
t.Run(name, func(t *testing.T) {
|
||||
b := NewPageAlloc(v.before)
|
||||
defer FreePageAlloc(b)
|
||||
|
||||
for iter, i := range v.hits {
|
||||
if a := b.Alloc(i.npages); a != i.base {
|
||||
t.Fatalf("bad alloc #%d: want 0x%x, got 0x%x", iter+1, i.base, a)
|
||||
}
|
||||
}
|
||||
want := NewPageAlloc(v.after)
|
||||
defer FreePageAlloc(want)
|
||||
|
||||
checkPageAlloc(t, want, b)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestPageAllocExhaust(t *testing.T) {
|
||||
for _, npages := range []uintptr{1, 2, 3, 4, 5, 8, 16, 64, 1024, 1025, 2048, 2049} {
|
||||
npages := npages
|
||||
t.Run(fmt.Sprintf("%d", npages), func(t *testing.T) {
|
||||
// Construct b.
|
||||
bDesc := make(map[ChunkIdx][]BitRange)
|
||||
for i := ChunkIdx(0); i < 4; i++ {
|
||||
bDesc[BaseChunkIdx+i] = []BitRange{}
|
||||
}
|
||||
b := NewPageAlloc(bDesc)
|
||||
defer FreePageAlloc(b)
|
||||
|
||||
// Allocate into b with npages until we've exhausted the heap.
|
||||
nAlloc := (PallocChunkPages * 4) / int(npages)
|
||||
for i := 0; i < nAlloc; i++ {
|
||||
addr := PageBase(BaseChunkIdx, uint(i)*uint(npages))
|
||||
if a := b.Alloc(npages); a != addr {
|
||||
t.Fatalf("bad alloc #%d: want 0x%x, got 0x%x", i+1, addr, a)
|
||||
}
|
||||
}
|
||||
|
||||
// Check to make sure the next allocation fails.
|
||||
if a := b.Alloc(npages); a != 0 {
|
||||
t.Fatalf("bad alloc #%d: want 0, got 0x%x", nAlloc, a)
|
||||
}
|
||||
|
||||
// Construct what we want the heap to look like now.
|
||||
allocPages := nAlloc * int(npages)
|
||||
wantDesc := make(map[ChunkIdx][]BitRange)
|
||||
for i := ChunkIdx(0); i < 4; i++ {
|
||||
if allocPages >= PallocChunkPages {
|
||||
wantDesc[BaseChunkIdx+i] = []BitRange{{0, PallocChunkPages}}
|
||||
allocPages -= PallocChunkPages
|
||||
} else if allocPages > 0 {
|
||||
wantDesc[BaseChunkIdx+i] = []BitRange{{0, uint(allocPages)}}
|
||||
allocPages = 0
|
||||
} else {
|
||||
wantDesc[BaseChunkIdx+i] = []BitRange{}
|
||||
}
|
||||
}
|
||||
want := NewPageAlloc(wantDesc)
|
||||
defer FreePageAlloc(want)
|
||||
|
||||
// Check to make sure the heap b matches what we want.
|
||||
checkPageAlloc(t, want, b)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestPageAllocFree(t *testing.T) {
|
||||
tests := map[string]struct {
|
||||
before map[ChunkIdx][]BitRange
|
||||
after map[ChunkIdx][]BitRange
|
||||
npages uintptr
|
||||
frees []uintptr
|
||||
}{
|
||||
"Free1": {
|
||||
npages: 1,
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
},
|
||||
frees: []uintptr{
|
||||
PageBase(BaseChunkIdx, 0),
|
||||
PageBase(BaseChunkIdx, 1),
|
||||
PageBase(BaseChunkIdx, 2),
|
||||
PageBase(BaseChunkIdx, 3),
|
||||
PageBase(BaseChunkIdx, 4),
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{5, PallocChunkPages - 5}},
|
||||
},
|
||||
},
|
||||
"ManyArena1": {
|
||||
npages: 1,
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 2: {{0, PallocChunkPages}},
|
||||
},
|
||||
frees: []uintptr{
|
||||
PageBase(BaseChunkIdx, PallocChunkPages/2),
|
||||
PageBase(BaseChunkIdx+1, 0),
|
||||
PageBase(BaseChunkIdx+2, PallocChunkPages-1),
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages / 2}, {PallocChunkPages/2 + 1, PallocChunkPages/2 - 1}},
|
||||
BaseChunkIdx + 1: {{1, PallocChunkPages - 1}},
|
||||
BaseChunkIdx + 2: {{0, PallocChunkPages - 1}},
|
||||
},
|
||||
},
|
||||
"Free2": {
|
||||
npages: 2,
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
},
|
||||
frees: []uintptr{
|
||||
PageBase(BaseChunkIdx, 0),
|
||||
PageBase(BaseChunkIdx, 2),
|
||||
PageBase(BaseChunkIdx, 4),
|
||||
PageBase(BaseChunkIdx, 6),
|
||||
PageBase(BaseChunkIdx, 8),
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{10, PallocChunkPages - 10}},
|
||||
},
|
||||
},
|
||||
"Straddle2": {
|
||||
npages: 2,
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{PallocChunkPages - 1, 1}},
|
||||
BaseChunkIdx + 1: {{0, 1}},
|
||||
},
|
||||
frees: []uintptr{
|
||||
PageBase(BaseChunkIdx, PallocChunkPages-1),
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
BaseChunkIdx + 1: {},
|
||||
},
|
||||
},
|
||||
"Free5": {
|
||||
npages: 5,
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
},
|
||||
frees: []uintptr{
|
||||
PageBase(BaseChunkIdx, 0),
|
||||
PageBase(BaseChunkIdx, 5),
|
||||
PageBase(BaseChunkIdx, 10),
|
||||
PageBase(BaseChunkIdx, 15),
|
||||
PageBase(BaseChunkIdx, 20),
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{25, PallocChunkPages - 25}},
|
||||
},
|
||||
},
|
||||
"Free64": {
|
||||
npages: 64,
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
},
|
||||
frees: []uintptr{
|
||||
PageBase(BaseChunkIdx, 0),
|
||||
PageBase(BaseChunkIdx, 64),
|
||||
PageBase(BaseChunkIdx, 128),
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{192, PallocChunkPages - 192}},
|
||||
},
|
||||
},
|
||||
"Free65": {
|
||||
npages: 65,
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
},
|
||||
frees: []uintptr{
|
||||
PageBase(BaseChunkIdx, 0),
|
||||
PageBase(BaseChunkIdx, 65),
|
||||
PageBase(BaseChunkIdx, 130),
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{195, PallocChunkPages - 195}},
|
||||
},
|
||||
},
|
||||
"FreePallocChunkPages": {
|
||||
npages: PallocChunkPages,
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
},
|
||||
frees: []uintptr{
|
||||
PageBase(BaseChunkIdx, 0),
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
},
|
||||
},
|
||||
"StraddlePallocChunkPages": {
|
||||
npages: PallocChunkPages,
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{PallocChunkPages / 2, PallocChunkPages / 2}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages / 2}},
|
||||
},
|
||||
frees: []uintptr{
|
||||
PageBase(BaseChunkIdx, PallocChunkPages/2),
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
BaseChunkIdx + 1: {},
|
||||
},
|
||||
},
|
||||
"StraddlePallocChunkPages+1": {
|
||||
npages: PallocChunkPages + 1,
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages}},
|
||||
},
|
||||
frees: []uintptr{
|
||||
PageBase(BaseChunkIdx, PallocChunkPages/2),
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages / 2}},
|
||||
BaseChunkIdx + 1: {{PallocChunkPages/2 + 1, PallocChunkPages/2 - 1}},
|
||||
},
|
||||
},
|
||||
"FreePallocChunkPages*2": {
|
||||
npages: PallocChunkPages * 2,
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages}},
|
||||
},
|
||||
frees: []uintptr{
|
||||
PageBase(BaseChunkIdx, 0),
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
BaseChunkIdx + 1: {},
|
||||
},
|
||||
},
|
||||
"StraddlePallocChunkPages*2": {
|
||||
npages: PallocChunkPages * 2,
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 2: {{0, PallocChunkPages}},
|
||||
},
|
||||
frees: []uintptr{
|
||||
PageBase(BaseChunkIdx, PallocChunkPages/2),
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages / 2}},
|
||||
BaseChunkIdx + 1: {},
|
||||
BaseChunkIdx + 2: {{PallocChunkPages / 2, PallocChunkPages / 2}},
|
||||
},
|
||||
},
|
||||
"AllFreePallocChunkPages*7+5": {
|
||||
npages: PallocChunkPages*7 + 5,
|
||||
before: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 1: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 2: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 3: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 4: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 5: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 6: {{0, PallocChunkPages}},
|
||||
BaseChunkIdx + 7: {{0, PallocChunkPages}},
|
||||
},
|
||||
frees: []uintptr{
|
||||
PageBase(BaseChunkIdx, 0),
|
||||
},
|
||||
after: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
BaseChunkIdx + 1: {},
|
||||
BaseChunkIdx + 2: {},
|
||||
BaseChunkIdx + 3: {},
|
||||
BaseChunkIdx + 4: {},
|
||||
BaseChunkIdx + 5: {},
|
||||
BaseChunkIdx + 6: {},
|
||||
BaseChunkIdx + 7: {{5, PallocChunkPages - 5}},
|
||||
},
|
||||
},
|
||||
}
|
||||
for name, v := range tests {
|
||||
v := v
|
||||
t.Run(name, func(t *testing.T) {
|
||||
b := NewPageAlloc(v.before)
|
||||
defer FreePageAlloc(b)
|
||||
for _, addr := range v.frees {
|
||||
b.Free(addr, v.npages)
|
||||
}
|
||||
|
||||
want := NewPageAlloc(v.after)
|
||||
defer FreePageAlloc(want)
|
||||
checkPageAlloc(t, want, b)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestPageAllocAllocAndFree(t *testing.T) {
|
||||
type hit struct {
|
||||
alloc bool
|
||||
npages uintptr
|
||||
base uintptr
|
||||
}
|
||||
tests := map[string]struct {
|
||||
init map[ChunkIdx][]BitRange
|
||||
hits []hit
|
||||
}{
|
||||
// TODO(mknyszek): Write more tests here.
|
||||
"Chunks8": {
|
||||
init: map[ChunkIdx][]BitRange{
|
||||
BaseChunkIdx: {},
|
||||
BaseChunkIdx + 1: {},
|
||||
BaseChunkIdx + 2: {},
|
||||
BaseChunkIdx + 3: {},
|
||||
BaseChunkIdx + 4: {},
|
||||
BaseChunkIdx + 5: {},
|
||||
BaseChunkIdx + 6: {},
|
||||
BaseChunkIdx + 7: {},
|
||||
},
|
||||
hits: []hit{
|
||||
{true, PallocChunkPages * 8, PageBase(BaseChunkIdx, 0)},
|
||||
{false, PallocChunkPages * 8, PageBase(BaseChunkIdx, 0)},
|
||||
{true, PallocChunkPages * 8, PageBase(BaseChunkIdx, 0)},
|
||||
{false, PallocChunkPages * 8, PageBase(BaseChunkIdx, 0)},
|
||||
{true, PallocChunkPages * 8, PageBase(BaseChunkIdx, 0)},
|
||||
{false, PallocChunkPages * 8, PageBase(BaseChunkIdx, 0)},
|
||||
{true, 1, PageBase(BaseChunkIdx, 0)},
|
||||
{false, 1, PageBase(BaseChunkIdx, 0)},
|
||||
{true, PallocChunkPages * 8, PageBase(BaseChunkIdx, 0)},
|
||||
},
|
||||
},
|
||||
}
|
||||
for name, v := range tests {
|
||||
v := v
|
||||
t.Run(name, func(t *testing.T) {
|
||||
b := NewPageAlloc(v.init)
|
||||
defer FreePageAlloc(b)
|
||||
|
||||
for iter, i := range v.hits {
|
||||
if i.alloc {
|
||||
if a := b.Alloc(i.npages); a != i.base {
|
||||
t.Fatalf("bad alloc #%d: want 0x%x, got 0x%x", iter+1, i.base, a)
|
||||
}
|
||||
} else {
|
||||
b.Free(i.base, i.npages)
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue