mirror of git://gcc.gnu.org/git/gcc.git
				
				
				
			
		
			
				
	
	
		
			100 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			100 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Go
		
	
	
	
// 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.
 | 
						|
 | 
						|
// Fixed-size object allocator. Returned memory is not zeroed.
 | 
						|
//
 | 
						|
// See malloc.go for overview.
 | 
						|
 | 
						|
package runtime
 | 
						|
 | 
						|
import "unsafe"
 | 
						|
 | 
						|
// FixAlloc is a simple free-list allocator for fixed size objects.
 | 
						|
// Malloc uses a FixAlloc wrapped around sysAlloc to manages its
 | 
						|
// MCache and MSpan objects.
 | 
						|
//
 | 
						|
// Memory returned by fixalloc.alloc is zeroed by default, but the
 | 
						|
// caller may take responsibility for zeroing allocations by setting
 | 
						|
// the zero flag to false. This is only safe if the memory never
 | 
						|
// contains heap pointers.
 | 
						|
//
 | 
						|
// The caller is responsible for locking around FixAlloc calls.
 | 
						|
// Callers can keep state in the object but the first word is
 | 
						|
// smashed by freeing and reallocating.
 | 
						|
//
 | 
						|
// Consider marking fixalloc'd types go:notinheap.
 | 
						|
type fixalloc struct {
 | 
						|
	size   uintptr
 | 
						|
	first  func(arg, p unsafe.Pointer) // called first time p is returned
 | 
						|
	arg    unsafe.Pointer
 | 
						|
	list   *mlink
 | 
						|
	chunk  uintptr // use uintptr instead of unsafe.Pointer to avoid write barriers
 | 
						|
	nchunk uint32
 | 
						|
	inuse  uintptr // in-use bytes now
 | 
						|
	stat   *uint64
 | 
						|
	zero   bool // zero allocations
 | 
						|
}
 | 
						|
 | 
						|
// A generic linked list of blocks.  (Typically the block is bigger than sizeof(MLink).)
 | 
						|
// Since assignments to mlink.next will result in a write barrier being performed
 | 
						|
// this cannot be used by some of the internal GC structures. For example when
 | 
						|
// the sweeper is placing an unmarked object on the free list it does not want the
 | 
						|
// write barrier to be called since that could result in the object being reachable.
 | 
						|
//
 | 
						|
//go:notinheap
 | 
						|
type mlink struct {
 | 
						|
	next *mlink
 | 
						|
}
 | 
						|
 | 
						|
// Initialize f to allocate objects of the given size,
 | 
						|
// using the allocator to obtain chunks of memory.
 | 
						|
func (f *fixalloc) init(size uintptr, first func(arg, p unsafe.Pointer), arg unsafe.Pointer, stat *uint64) {
 | 
						|
	f.size = size
 | 
						|
	f.first = first
 | 
						|
	f.arg = arg
 | 
						|
	f.list = nil
 | 
						|
	f.chunk = 0
 | 
						|
	f.nchunk = 0
 | 
						|
	f.inuse = 0
 | 
						|
	f.stat = stat
 | 
						|
	f.zero = true
 | 
						|
}
 | 
						|
 | 
						|
func (f *fixalloc) alloc() unsafe.Pointer {
 | 
						|
	if f.size == 0 {
 | 
						|
		print("runtime: use of FixAlloc_Alloc before FixAlloc_Init\n")
 | 
						|
		throw("runtime: internal error")
 | 
						|
	}
 | 
						|
 | 
						|
	if f.list != nil {
 | 
						|
		v := unsafe.Pointer(f.list)
 | 
						|
		f.list = f.list.next
 | 
						|
		f.inuse += f.size
 | 
						|
		if f.zero {
 | 
						|
			memclrNoHeapPointers(v, f.size)
 | 
						|
		}
 | 
						|
		return v
 | 
						|
	}
 | 
						|
	if uintptr(f.nchunk) < f.size {
 | 
						|
		f.chunk = uintptr(persistentalloc(_FixAllocChunk, 0, f.stat))
 | 
						|
		f.nchunk = _FixAllocChunk
 | 
						|
	}
 | 
						|
 | 
						|
	v := unsafe.Pointer(f.chunk)
 | 
						|
	if f.first != nil {
 | 
						|
		f.first(f.arg, v)
 | 
						|
	}
 | 
						|
	f.chunk = f.chunk + f.size
 | 
						|
	f.nchunk -= uint32(f.size)
 | 
						|
	f.inuse += f.size
 | 
						|
	return v
 | 
						|
}
 | 
						|
 | 
						|
func (f *fixalloc) free(p unsafe.Pointer) {
 | 
						|
	f.inuse -= f.size
 | 
						|
	v := (*mlink)(p)
 | 
						|
	v.next = f.list
 | 
						|
	f.list = v
 | 
						|
}
 |