llvm/compiler-rt/test/tsan/Darwin/malloc_size.mm

// Test that malloc_zone_from_ptr returns a valid zone for a 0-sized allocation.
// Test that malloc_size does not crash for an invalid pointer.

// RUN: %clang_tsan %s -o %t -framework Foundation
// RUN: %run %t 2>&1 | FileCheck %s

#include <malloc/malloc.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/mman.h>

int some_global;

void describe_zone(void *p) {
  malloc_zone_t *z = malloc_zone_from_ptr(p);
  if (z) {
    fprintf(stderr, "zone = %p\n", z);
  }	else {
  	fprintf(stderr, "zone = no zone\n");
  }
}

int main() {
  void *p;
  size_t s;

  p = malloc(0x40);
  s = malloc_size(p);
  fprintf(stderr, "size = 0x%zx\n", s);
  // CHECK: size = 0x40
  describe_zone(p);
  // CHECK: zone = 0x{{[0-9a-f]+}}

  p = malloc(0);
  s = malloc_size(p);
  fprintf(stderr, "size = 0x%zx\n", s);
  // CHECK: size = 0x1
  describe_zone(p);
  // CHECK: zone = 0x{{[0-9a-f]+}}

  p = &some_global;
  s = malloc_size(p);
  fprintf(stderr, "size = 0x%zx\n", s);
  // CHECK: size = 0x0
  describe_zone(p);
  // CHECK: zone = no zone

  p = mmap(0, 0x1000, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, -1, 0);
  if (!p) {
  	fprintf(stderr, "mmap failed\n");
  	exit(1);
  }
  s = malloc_size(p);
  fprintf(stderr, "size = 0x%zx\n", s);
  // CHECK: size = 0x0
  describe_zone(p);
  // CHECK: zone = no zone

  p = (void *)0x42;  // invalid pointer
  s = malloc_size(p);
  fprintf(stderr, "size = 0x%zx\n", s);
  // CHECK: size = 0x0
  describe_zone(p);
  // CHECK: zone = no zone

  return 0;
}