0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-12-25 06:53:58 +01:00
construct/ircd/allocator_je.cc

187 lines
3.9 KiB
C++

// Matrix Construct
//
// Copyright (C) Matrix Construct Developers, Authors & Contributors
// Copyright (C) 2016-2018 Jason Volk <jason@zemos.net>
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice is present in all copies. The
// full license for this software is available in the LICENSE file.
#include <RB_INC_JEMALLOC_H
#if defined(IRCD_ALLOCATOR_USE_JEMALLOC) && defined(HAVE_JEMALLOC_H)
#define IRCD_ALLOCATOR_JEMALLOC
#endif
namespace ircd::allocator::je
{
static std::function<void (std::ostream &, const string_view &)> stats_callback;
static void stats_handler(void *, const char *) noexcept;
extern info::versions malloc_version_api;
extern info::versions malloc_version_abi;
}
#if defined(IRCD_ALLOCATOR_USE_JEMALLOC)
const char *
__attribute__((weak))
malloc_conf
{
"narenas:1"
",tcache:false"
",metadata_thp:always"
};
#endif
decltype(ircd::allocator::je::malloc_version_api)
ircd::allocator::je::malloc_version_api
{
"jemalloc", info::versions::API, 0,
#ifdef HAVE_JEMALLOC_H
{
JEMALLOC_VERSION_MAJOR,
JEMALLOC_VERSION_MINOR,
JEMALLOC_VERSION_BUGFIX
},
JEMALLOC_VERSION
#endif
};
decltype(ircd::allocator::je::malloc_version_abi)
ircd::allocator::je::malloc_version_abi
{
"jemalloc", info::versions::ABI, 0, {0, 0, 0}, []
(info::versions &v, const mutable_buffer &buf)
{
#ifdef IRCD_ALLOCATOR_JEMALLOC
const string_view val
{
*reinterpret_cast<const char *const *>(data(allocator::get("version", buf)))
};
if(!val)
return;
strlcpy(buf, val);
const string_view semantic(split(val, '-').first);
v.semantic[0] = lex_cast<ulong>(token(semantic, '.', 0));
v.semantic[1] = lex_cast<ulong>(token(semantic, '.', 1));
v.semantic[2] = lex_cast<ulong>(token(semantic, '.', 2));
#endif
}
};
decltype(ircd::allocator::je::available)
ircd::allocator::je::available
{
#if defined(IRCD_ALLOCATOR_JEMALLOC)
mods::ldso::has("jemalloc")
#endif
};
#if defined(IRCD_ALLOCATOR_JEMALLOC)
bool
ircd::allocator::trim(const size_t &pad)
noexcept
{
return false;
}
#endif
#if defined(IRCD_ALLOCATOR_JEMALLOC)
ircd::string_view
ircd::allocator::get(const string_view &key_,
const mutable_buffer &buf)
{
thread_local char key[128];
strlcpy(key, key_);
size_t len(size(buf));
syscall(::mallctl, key, data(buf), &len, nullptr, 0UL);
return string_view
{
data(buf), std::min(len, size(buf))
};
}
#endif
#if defined(IRCD_ALLOCATOR_JEMALLOC)
ircd::string_view
ircd::allocator::set(const string_view &key_,
const string_view &val,
const mutable_buffer &cur)
{
thread_local char key[128];
strlcpy(key, key_);
size_t curlen(size(cur));
syscall(::mallctl, key, data(cur), &curlen, const_cast<char *>(data(val)), size(val));
return string_view
{
data(cur), std::min(curlen, size(cur))
};
}
#endif
void
ircd::allocator::je::stats_handler(void *const ptr,
const char *const msg)
noexcept try
{
auto &out
{
*reinterpret_cast<std::stringstream *>(ptr)
};
stats_callback(out, msg);
}
catch(const std::bad_function_call &)
{
assert(0);
return;
}
#if defined(IRCD_ALLOCATOR_JEMALLOC)
ircd::string_view
ircd::allocator::info(const mutable_buffer &buf,
const string_view &opts_)
{
std::stringstream out;
pubsetbuf(out, buf);
je::stats_callback = []
(auto &out, const string_view &msg)
{
out << msg;
};
thread_local char opts_buf[64];
const char *const opts
{
opts_?
data(strlcpy(opts_buf, opts_)):
""
};
malloc_stats_print(je::stats_handler, &out, opts);
out << std::endl;
return view(out, buf);
}
#endif
#if defined(IRCD_ALLOCATOR_JEMALLOC)
void
ircd::allocator::scope::hook_init()
noexcept
{
}
#endif
#if defined(IRCD_ALLOCATOR_JEMALLOC)
void
ircd::allocator::scope::hook_fini()
noexcept
{
}
#endif