check-jemalloc-prof.py

gistfile1.txt

"""
  Check if libjemalloc was built with `--enable-prof` (and thus has
  memory profiling capabilities) or not.
  Author: Ori Livneh
  No copyright restriction
  Original Source: https://gist.github.com/atdt/736137113e577795e5fc
  
  Able to specify the path to libjemalloc as the first argument.
"""
import ctypes
import ctypes.util
import errno
import os
import sys

jemalloc_so = ""
if len(sys.argv) > 1:
    jemalloc_so = sys.argv[1]
else:
    jemalloc_so = ctypes.util.find_library('jemalloc')

if jemalloc_so is None:
    sys.exit('Could not find libjemalloc! Is it installed?')
jemalloc = ctypes.CDLL(jemalloc_so, use_errno=True)

has_prof = ctypes.c_bool(False)
has_prof_len = ctypes.c_size_t(ctypes.sizeof(has_prof))
rv = jemalloc.mallctl('config.prof', ctypes.byref(has_prof),
                      ctypes.byref(has_prof_len), None, None)
if rv != 0:
    err = ctypes.get_errno()
    raise OSError(err, os.strerror(err))

if has_prof:
    print('OK: jemalloc was built with `--enable-prof`.')
    sys.exit(0)
else:
    print('NOT OK: jemalloc was not built with `--enable-prof`.')
    sys.exit(1)