add more restrictions to aalloc

Author: minad

test.c

@@ -55,7 +55,7 @@ static void random_test(tlsf* t, size_t spacelen, const size_t cap) {
                 align = 0;
             else
                 len = align * (((size_t)rand() % (cap / align)) + 1);
-            p[i] = tlsf_aalloc(t, align, len);
+            p[i] = !align || !len ? tlsf_malloc(t, len) : tlsf_aalloc(t, align, len);
             if (align)
                 assert(!((size_t)p[i] % align));
         }

tlsf.c

@@ -423,10 +423,9 @@ TLSF_API void* tlsf_malloc(tlsf* t, size_t size) {
 TLSF_API void* tlsf_aalloc(tlsf* t, size_t align, size_t size) {
     size_t adjust = adjust_size(size, ALIGN_SIZE);
 
-    if (UNLIKELY(align &&
-                 ((align & (align - 1)) || // align is not a power of two
-                  (size & (align - 1)) || // size is not a multiple of align
-                  adjust > TLSF_MAX_SIZE - align - sizeof (tlsf_block)))) // size is too large
+    if (UNLIKELY(!size ||
+                 ((align | size) & (align - 1)) || // align!=2**x, size!=n*align
+                 adjust > TLSF_MAX_SIZE - align - sizeof (tlsf_block))) // size is too large
         return 0;
 
     if (align <= ALIGN_SIZE)