project-base/server-skynet-source-3rd-jemalloc
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fallback to unbuffered printing if OOM

Browse Source
This commit is contained in:
Yinan Zhang 2020-01-13 14:23:10 -08:00
parent cd6e908241
commit f81341a48b
4 changed files with 39 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/jemalloc/internal/buf_writer.h
View File

@ -23,7 +23,9 @@ buf_write_init(buf_write_arg_t *arg, void (*write_cb)(void *, const char *),
void *cbopaque, char *buf, size_t buf_len) { void *cbopaque, char *buf, size_t buf_len) {
arg->write_cb = write_cb; arg->write_cb = write_cb;
arg->cbopaque = cbopaque; arg->cbopaque = cbopaque;
assert(buf != NULL);
arg->buf = buf; arg->buf = buf;
assert(buf_len >= 2);
arg->buf_size = buf_len - 1; /* Accommodating '\0' at the end. */ arg->buf_size = buf_len - 1; /* Accommodating '\0' at the end. */
arg->buf_end = 0; arg->buf_end = 0;
} }

16
src/jemalloc.c
View File

@ -3716,12 +3716,16 @@ je_malloc_stats_print(void (*write_cb)(void *, const char *), void *cbopaque,
char *buf = (char *)iallocztm(tsdn, STATS_PRINT_BUFSIZE, char *buf = (char *)iallocztm(tsdn, STATS_PRINT_BUFSIZE,
sz_size2index(STATS_PRINT_BUFSIZE), false, NULL, true, sz_size2index(STATS_PRINT_BUFSIZE), false, NULL, true,
arena_get(TSDN_NULL, 0, true), true); arena_get(TSDN_NULL, 0, true), true);
buf_write_arg_t buf_arg; if (buf == NULL) {
buf_write_init(&buf_arg, write_cb, cbopaque, buf, stats_print(write_cb, cbopaque, opts);
STATS_PRINT_BUFSIZE); } else {
stats_print(buf_write_cb, &buf_arg, opts); buf_write_arg_t buf_arg;
buf_write_flush(&buf_arg); buf_write_init(&buf_arg, write_cb, cbopaque, buf,
idalloctm(tsdn, buf, NULL, NULL, true, true); STATS_PRINT_BUFSIZE);
stats_print(buf_write_cb, &buf_arg, opts);
buf_write_flush(&buf_arg);
idalloctm(tsdn, buf, NULL, NULL, true, true);
}
} }
check_entry_exit_locking(tsdn); check_entry_exit_locking(tsdn);

21
src/prof_log.c
View File

@ -633,12 +633,15 @@ prof_log_stop(tsdn_t *tsdn) {
sz_size2index(PROF_LOG_STOP_BUFSIZE), false, NULL, true, sz_size2index(PROF_LOG_STOP_BUFSIZE), false, NULL, true,
arena_get(TSDN_NULL, 0, true), true); arena_get(TSDN_NULL, 0, true), true);
buf_write_arg_t buf_arg; buf_write_arg_t buf_arg;
buf_write_init(&buf_arg, prof_emitter_write_cb, &arg, buf, if (buf == NULL) {
PROF_LOG_STOP_BUFSIZE); emitter_init(&emitter, emitter_output_json_compact,
prof_emitter_write_cb, &arg);
/* Emit to json. */ } else {
emitter_init(&emitter, emitter_output_json_compact, buf_write_cb, buf_write_init(&buf_arg, prof_emitter_write_cb, &arg, buf,
&buf_arg); PROF_LOG_STOP_BUFSIZE);
emitter_init(&emitter, emitter_output_json_compact,
buf_write_cb, &buf_arg);
}
emitter_begin(&emitter); emitter_begin(&emitter);
prof_log_emit_metadata(&emitter); prof_log_emit_metadata(&emitter);
@ -647,8 +650,10 @@ prof_log_stop(tsdn_t *tsdn) {
prof_log_emit_allocs(tsd, &emitter); prof_log_emit_allocs(tsd, &emitter);
emitter_end(&emitter); emitter_end(&emitter);
buf_write_flush(&buf_arg); if (buf != NULL) {
idalloctm(tsdn, buf, NULL, NULL, true, true); buf_write_flush(&buf_arg);
idalloctm(tsdn, buf, NULL, NULL, true, true);
}
/* Reset global state. */ /* Reset global state. */
if (log_tables_initialized) { if (log_tables_initialized) {

21
src/prof_recent.c
View File

@ -465,12 +465,17 @@ prof_recent_alloc_dump(tsd_t *tsd, void (*write_cb)(void *, const char *),
char *buf = (char *)iallocztm(tsd_tsdn(tsd), PROF_RECENT_PRINT_BUFSIZE, char *buf = (char *)iallocztm(tsd_tsdn(tsd), PROF_RECENT_PRINT_BUFSIZE,
sz_size2index(PROF_RECENT_PRINT_BUFSIZE), false, NULL, true, sz_size2index(PROF_RECENT_PRINT_BUFSIZE), false, NULL, true,
arena_get(tsd_tsdn(tsd), 0, false), true); arena_get(tsd_tsdn(tsd), 0, false), true);
buf_write_arg_t buf_arg;
buf_write_init(&buf_arg, write_cb, cbopaque, buf,
PROF_RECENT_PRINT_BUFSIZE);
emitter_t emitter; emitter_t emitter;
emitter_init(&emitter, emitter_output_json_compact, buf_write_cb, buf_write_arg_t buf_arg;
&buf_arg); if (buf == NULL) {
emitter_init(&emitter, emitter_output_json_compact, write_cb,
cbopaque);
} else {
buf_write_init(&buf_arg, write_cb, cbopaque, buf,
PROF_RECENT_PRINT_BUFSIZE);
emitter_init(&emitter, emitter_output_json_compact,
buf_write_cb, &buf_arg);
}
emitter_begin(&emitter); emitter_begin(&emitter);
malloc_mutex_lock(tsd_tsdn(tsd), &prof_recent_alloc_mtx); malloc_mutex_lock(tsd_tsdn(tsd), &prof_recent_alloc_mtx);
@ -530,8 +535,10 @@ prof_recent_alloc_dump(tsd_t *tsd, void (*write_cb)(void *, const char *),
malloc_mutex_unlock(tsd_tsdn(tsd), &prof_recent_alloc_mtx); malloc_mutex_unlock(tsd_tsdn(tsd), &prof_recent_alloc_mtx);
emitter_end(&emitter); emitter_end(&emitter);
buf_write_flush(&buf_arg); if (buf != NULL) {
idalloctm(tsd_tsdn(tsd), buf, NULL, NULL, true, true); buf_write_flush(&buf_arg);
idalloctm(tsd_tsdn(tsd), buf, NULL, NULL, true, true);
}
} }
#undef PROF_RECENT_PRINT_BUFSIZE #undef PROF_RECENT_PRINT_BUFSIZE