The spans do now contain the size of a span, no longer the end offset.

[cc65] / libsrc / common / _heapblocksize.s

diff --git a/libsrc/common/_heapblocksize.s b/libsrc/common/_heapblocksize.s
index 8db62e0b910ee7a3d6d2369d4e92a0e9c9584f4f..2be32fef4301c85d6e7f2875339725d7ca73bd3f 100644 (file)
--- a/libsrc/common/_heapblocksize.s
+++ b/libsrc/common/_heapblocksize.s
@@ -6,32 +6,68 @@
 ; Return the size of an allocated block.
 ;
 
-       .importzp       ptr1
+       .importzp       ptr1, ptr2
                .export         __heapblocksize
 
         .include        "_heap.inc"
 
+        .macpack        generic
+        .macpack        cpu
+
 ;-----------------------------------------------------------------------------
 ; Code
 
 __heapblocksize:
 
-; Decrement the block pointer so it points to the admin data
+; Below the user data is a pointer that points to the start of the real
+; (raw) memory block. The first word of this block is the size. To access
+; the raw block pointer, we will decrement the high byte of the pointer,
+; the pointer is then at offset 254/255.
 
-        sec
-        sbc     #HEAP_ADMIN_SPACE       ; Assume it's less than 256
-        bcs     L1
+        sta     ptr1
         dex
-L1:     sta     ptr1
         stx     ptr1+1
+        ldy     #$FE
+        lda     (ptr1),y
+        sta     ptr2            ; Place the raw block pointer into ptr2
+        iny
+        lda     (ptr1),y
+        sta     ptr2+2
 
-; Load the size from the given block
+; Load the size from the raw block
 
-        ldy     #1
-        lda     (ptr1),y
+        ldy     #usedblock::size+1
+        lda     (ptr2),y
         tax
+.if (.cpu .bitand CPU_ISET_65SC02)
+        lda     (ptr2)
+.else
         dey
-        lda     (ptr1),y
+        lda     (ptr2),y
+.endif
+
+; Correct the raw block size so that is shows the user visible portion. To
+; do that, we must decrease the size by the amount of unused memory, which is
+; the difference between the user space pointer and the raw memory block
+; pointer. Since we have decremented the user space pointer by 256, we will
+; have to correct the result.
+;
+;       return size - (ptr1 + 256 - ptr2)
+;       return size - ptr1 - 256 + ptr2
+
+        dex                     ; - 256
+        add     ptr2
+        pha
+        txa
+        adc     ptr2+1
+        tax
+        pla
+        sub     ptr1
+        pha
+        txa
+        sbc     ptr1+1
+        tax
+        pla
 
 ; Done