support for custom allocator in ByteBuffer

base.use

@@ -2,6 +2,7 @@ Name: Base
 Description: Base library that complements the built-in SDK.
 SourcePath: source/base
 Imports: Order
+Imports: Allocator
 Imports: ByteBuffer
 Imports: DateTime
 Imports: TimeSpan

source/base/Allocator.ooc

@@ -0,0 +1,18 @@
+/* This file is part of magic-sdk, an sdk for the open source programming language magic.
+ *
+ * Copyright (C) 2016 magic-lang
+ *
+ * This software may be modified and distributed under the terms
+ * of the MIT license.  See the LICENSE file for details.
+ */
+
+AbstractAllocator: abstract class {
+	allocate: abstract func (size: SizeT) -> Void*
+	deallocate: abstract func (pointer: Void*)
+}
+
+MallocAllocator: class extends AbstractAllocator {
+	init: func
+	allocate: override func (size: SizeT) -> Void* { malloc(size) }
+	deallocate: override func (pointer: Void*) { memfree(pointer) }
+}

source/base/ByteBuffer.ooc

@@ -7,6 +7,7 @@
  */
 
 use collections
+import Allocator
 import ReferenceCounter
 import Debug
 
@@ -15,21 +16,24 @@ ByteBuffer: class {
 	_size: Int
 	_referenceCount: ReferenceCounter
 	_ownsMemory: Bool
+	_allocator: AbstractAllocator = null
 	pointer ::= this _pointer
 	size ::= this _size
 	referenceCount ::= this _referenceCount
 
-	init: func (=_pointer, =_size, ownsMemory := false) {
+	init: func (=_pointer, =_size, ownsMemory := false, allocator := null as AbstractAllocator) {
 		this _referenceCount = ReferenceCounter new(this)
 		this _ownsMemory = ownsMemory
+		this _allocator = allocator ?? MallocAllocator new() as AbstractAllocator
 	}
 	free: override func {
 		if (this _referenceCount != null)
 			this _referenceCount free()
 		this _referenceCount = null
 		if (this _ownsMemory)
-			memfree(this _pointer)
+			this _allocator deallocate(this _pointer)
 		this _pointer = null
+		this _allocator free()
 		super()
 	}
 	zero: func ~whole { this memset(0) }
@@ -50,7 +54,7 @@ ByteBuffer: class {
 	copyTo: func (other: This, start: Int, destination: Int, length: Int) {
 		memcpy(other pointer + destination, this pointer + start, length)
 	}
-	new: static func ~size (size: Int) -> This { _RecyclableByteBuffer new(size) }
+	new: static func ~size (size: Int, allocator: AbstractAllocator = null) -> This { _RecyclableByteBuffer new(size, allocator) }
 	new: static func ~recover (pointer: Byte*, size: Int, recover: Func (This) -> Bool) -> This {
 		_RecoverableByteBuffer new(pointer, size, recover)
 	}
@@ -86,7 +90,7 @@ _RecoverableByteBuffer: class extends ByteBuffer {
 }
 
 _RecyclableByteBuffer: class extends ByteBuffer {
-	init: func (pointer: Byte*, size: Int) { super(pointer, size, true) }
+	init: func (pointer: Byte*, size: Int, allocator: AbstractAllocator = null) { super(pointer, size, true, allocator) }
 	_forceFree: func {
 		this _size = 0
 		this free()
@@ -111,7 +115,7 @@ _RecyclableByteBuffer: class extends ByteBuffer {
 	_smallRecycleBin := static VectorList<This> new()
 	_mediumRecycleBin := static VectorList<This> new()
 	_largeRecycleBin := static VectorList<This> new()
-	new: static func ~fromSize (size: Int) -> This {
+	new: static func ~fromSize (size: Int, allocator: AbstractAllocator = null) -> This {
 		buffer: This = null
 		bin := This _getBin(size)
 		This _lock lock()
@@ -123,7 +127,8 @@ _RecyclableByteBuffer: class extends ByteBuffer {
 			}
 		This _lock unlock()
 		version(debugByteBuffer) { if (buffer == null) Debug print("No RecyclableByteBuffer available in the bin; allocating a new one") }
-		buffer == null ? This new(malloc(size), size) : buffer
+		allocator = allocator ?? MallocAllocator new() as AbstractAllocator
+		buffer == null ? This new(allocator allocate(size) as Byte*, size, allocator) : buffer
 	}
 	_getBin: static func (size: Int) -> VectorList<This> {
 		size < 10000 ? This _smallRecycleBin : (size < 100000 ? This _mediumRecycleBin : This _largeRecycleBin)

test/base/ByteBufferTest.ooc

@@ -9,6 +9,18 @@
 use base
 use unit
 
+CustomAllocator: class extends AbstractAllocator {
+	_allocCount := 0
+	init: func
+	allocate: override func (size: SizeT) -> Void* {
+		this _allocCount += 1
+		malloc(size)
+	}
+	deallocate: override func (pointer: Void*) {
+		memfree(pointer)
+	}
+}
+
 ByteBufferTest: class extends Fixture {
 	init: func {
 		super("ByteBuffer")
@@ -78,6 +90,14 @@ ByteBufferTest: class extends Fixture {
 			expect(buffer pointer[63] as Int, is equal to(63))
 			buffer referenceCount decrease()
 		})
+		this add("custom alloc", This _testCustomAlloc)
+	}
+	_testCustomAlloc: static func {
+		allocator := CustomAllocator new()
+		expect(allocator _allocCount, is equal to(0))
+		buffer := ByteBuffer new(128, allocator)
+		expect(allocator _allocCount, is equal to(1))
+		buffer free()
 	}
 }