File: mprobe.cpp

package info (click to toggle)
llvm-toolchain-16 1%3A16.0.6-15~deb11u2
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 1,634,820 kB
  • sloc: cpp: 6,179,261; ansic: 1,216,205; asm: 741,319; python: 196,614; objc: 75,325; f90: 49,640; lisp: 32,396; pascal: 12,286; sh: 9,394; perl: 7,442; ml: 5,494; awk: 3,523; makefile: 2,723; javascript: 1,206; xml: 886; fortran: 581; cs: 573
file content (42 lines) | stat: -rw-r--r-- 870 bytes parent folder | download | duplicates (17)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
// RUN: %clangxx %s -o %t && %run %t 2>&1 | FileCheck %s
// UNSUPPORTED: android, ubsan

#include <stdio.h>
#include <stdlib.h>
#if defined(__GLIBC_PREREQ) && __GLIBC_PREREQ(2, 2)
#include <mcheck.h>
#else
#define MCHECK_OK 0
extern "C" int mcheck(void (*abortfunc)(int mstatus));
extern "C" int mcheck_pedantic(void (*abortfunc)(int mstatus));
extern "C" int mprobe(void *ptr);
#endif

void check_heap() {
  void *p = malloc(1000);
  int res = mprobe(p);
  if (res == MCHECK_OK)
    printf("Success!\n");
  free(p);
}

int main(int argc, char *argv[]) {
  void *p;
  if (mcheck(NULL) != 0) {
    fprintf(stderr, "mcheck() failed\n");
    exit(EXIT_FAILURE);
  }

  check_heap();
  // CHECK: Success!

  if (mcheck_pedantic(NULL) != 0) {
    fprintf(stderr, "mcheck_pedantic() failed\n");
    exit(EXIT_FAILURE);
  }

  check_heap();
  // CHECK: Success!

  return 0;
}