fixed heap-use-after-free in solidgen

[csRadar.git] / csrMem.h

diff --git a/csrMem.h b/csrMem.h
index b2be8e8bcb576d9c68dd3a5895da8b3ae172703b..25913be651236abce298516b4f570cbc5828fb1d 100644 (file)
--- a/csrMem.h
+++ b/csrMem.h
@@ -50,6 +50,7 @@ void csr_sb_inc( void *arr, u32 amt )
        raw[1] += amt;
 }
 
+__attribute__((warn_unused_result))
 void *csr_sb_reserve( void *arr, u32 amt, u32 esize )
 {
        u32 cap = arr? csr_sb_cap( arr ): 0;
@@ -83,6 +84,14 @@ void csr_sb_free( void *arr )
        }
 }
 
+void csr_sb_clear( void *arr )
+{
+       if( arr )
+       {
+               ((u32 *)csr_sb_raw( arr ))[1] = 0;
+       }
+}
+
 // djb2 - Dan Bernstein
 unsigned long djb2( unsigned char const *str )
 {