mirror of
https://github.com/golang/go
synced 2024-11-22 00:14:42 -07:00
ebd27d62fd
cut the slowest tests down from a few seconds to under half a second. R=r DELTA=21 (6 added, 1 deleted, 14 changed) OCL=35509 CL=35519
86 lines
1.9 KiB
Go
86 lines
1.9 KiB
Go
// $G $D/$F.go && $L $F.$A && ./$A.out
|
|
|
|
// Copyright 2009 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.
|
|
|
|
// Random malloc test.
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag";
|
|
"malloc";
|
|
"rand";
|
|
"unsafe";
|
|
)
|
|
|
|
var chatty = flag.Bool("v", false, "chatty");
|
|
|
|
var footprint uint64;
|
|
var allocated uint64;
|
|
func bigger() {
|
|
if f := malloc.GetStats().Sys; footprint < f {
|
|
footprint = f;
|
|
if *chatty {
|
|
println("Footprint", footprint, " for ", allocated);
|
|
}
|
|
if footprint > 1e9 {
|
|
panicln("too big");
|
|
}
|
|
}
|
|
}
|
|
|
|
// Prime the data structures by allocating one of
|
|
// each block in order. After this, there should be
|
|
// little reason to ask for more memory from the OS.
|
|
func prime() {
|
|
for i := 0; i < 16; i++ {
|
|
b := malloc.Alloc(1<<uint(i));
|
|
malloc.Free(b);
|
|
}
|
|
for i := uintptr(0); i < 256; i++ {
|
|
b := malloc.Alloc(i<<12);
|
|
malloc.Free(b);
|
|
}
|
|
}
|
|
|
|
func memset(b *byte, c byte, n uintptr) {
|
|
np := uintptr(n);
|
|
for i := uintptr(0); i < np; i++ {
|
|
*(*byte)(unsafe.Pointer(uintptr(unsafe.Pointer(b))+i)) = c;
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse();
|
|
// prime();
|
|
var blocks [1] struct { base *byte; siz uintptr; };
|
|
for i := 0; i < 1<<10; i++ {
|
|
if i%(1<<10) == 0 && *chatty {
|
|
println(i);
|
|
}
|
|
b := rand.Int() % len(blocks);
|
|
if blocks[b].base != nil {
|
|
// println("Free", blocks[b].siz, blocks[b].base);
|
|
malloc.Free(blocks[b].base);
|
|
blocks[b].base = nil;
|
|
allocated -= uint64(blocks[b].siz);
|
|
continue
|
|
}
|
|
siz := uintptr(rand.Int() >> (11 + rand.Uint32() % 20));
|
|
base := malloc.Alloc(siz);
|
|
// ptr := uintptr(syscall.BytePtr(base))+uintptr(siz/2);
|
|
// obj, size, ref, ok := allocator.find(ptr);
|
|
// if obj != base || *ref != 0 || !ok {
|
|
// panicln("find", siz, obj, ref, ok);
|
|
// }
|
|
blocks[b].base = base;
|
|
blocks[b].siz = siz;
|
|
allocated += uint64(siz);
|
|
// println("Alloc", siz, base);
|
|
memset(base, 0xbb, siz);
|
|
bigger();
|
|
}
|
|
}
|