Merge branch 'lukeshu/fast'

1 files changed, 17 insertions, 4 deletions

diff --git a/lib/btrfs/btrfsitem/item_freespacebitmap.go b/lib/btrfs/btrfsitem/item_freespacebitmap.go
index ad46204..ebc00e4 100644
--- a/lib/btrfs/btrfsitem/item_freespacebitmap.go
+++ b/lib/btrfs/btrfsitem/item_freespacebitmap.go
@@ -1,4 +1,4 @@
-// Copyright (C) 2022  Luke Shumaker <lukeshu@lukeshu.com>
+// Copyright (C) 2022-2023  Luke Shumaker <lukeshu@lukeshu.com>
 //
 // SPDX-License-Identifier: GPL-2.0-or-later
 
@@ -6,13 +6,26 @@ package btrfsitem
 
 // key.objectid = object ID of the FreeSpaceInfo (logical_addr)
 // key.offset = offset of the FreeSpaceInfo (size)
-type FreeSpaceBitmap []byte // FREE_SPACE_BITMAP=200
+type FreeSpaceBitmap struct { // complex FREE_SPACE_BITMAP=200
+	Bitmap []byte
+}
+
+func (o *FreeSpaceBitmap) Free() {
+	bytePool.Put(o.Bitmap)
+	*o = FreeSpaceBitmap{}
+	freeSpaceBitmapPool.Put(o)
+}
+
+func (o FreeSpaceBitmap) Clone() FreeSpaceBitmap {
+	o.Bitmap = cloneBytes(o.Bitmap)
+	return o
+}
 
 func (o *FreeSpaceBitmap) UnmarshalBinary(dat []byte) (int, error) {
-	*o = dat
+	o.Bitmap = cloneBytes(dat)
 	return len(dat), nil
 }
 
 func (o FreeSpaceBitmap) MarshalBinary() ([]byte, error) {
-	return []byte(o), nil
+	return append([]byte(nil), o.Bitmap...), nil
 }