7fc6b1b259
The buffered writer adopts a signature identical to `write_cb`, so that it can be plugged into anywhere `write_cb` appears.
128 lines
3.8 KiB
C
128 lines
3.8 KiB
C
#ifndef JEMALLOC_INTERNAL_MALLOC_IO_H
|
|
#define JEMALLOC_INTERNAL_MALLOC_IO_H
|
|
|
|
#ifdef _WIN32
|
|
# ifdef _WIN64
|
|
# define FMT64_PREFIX "ll"
|
|
# define FMTPTR_PREFIX "ll"
|
|
# else
|
|
# define FMT64_PREFIX "ll"
|
|
# define FMTPTR_PREFIX ""
|
|
# endif
|
|
# define FMTd32 "d"
|
|
# define FMTu32 "u"
|
|
# define FMTx32 "x"
|
|
# define FMTd64 FMT64_PREFIX "d"
|
|
# define FMTu64 FMT64_PREFIX "u"
|
|
# define FMTx64 FMT64_PREFIX "x"
|
|
# define FMTdPTR FMTPTR_PREFIX "d"
|
|
# define FMTuPTR FMTPTR_PREFIX "u"
|
|
# define FMTxPTR FMTPTR_PREFIX "x"
|
|
#else
|
|
# include <inttypes.h>
|
|
# define FMTd32 PRId32
|
|
# define FMTu32 PRIu32
|
|
# define FMTx32 PRIx32
|
|
# define FMTd64 PRId64
|
|
# define FMTu64 PRIu64
|
|
# define FMTx64 PRIx64
|
|
# define FMTdPTR PRIdPTR
|
|
# define FMTuPTR PRIuPTR
|
|
# define FMTxPTR PRIxPTR
|
|
#endif
|
|
|
|
/* Size of stack-allocated buffer passed to buferror(). */
|
|
#define BUFERROR_BUF 64
|
|
|
|
/*
|
|
* Size of stack-allocated buffer used by malloc_{,v,vc}printf(). This must be
|
|
* large enough for all possible uses within jemalloc.
|
|
*/
|
|
#define MALLOC_PRINTF_BUFSIZE 4096
|
|
|
|
int buferror(int err, char *buf, size_t buflen);
|
|
uintmax_t malloc_strtoumax(const char *restrict nptr, char **restrict endptr,
|
|
int base);
|
|
void malloc_write(const char *s);
|
|
|
|
/*
|
|
* malloc_vsnprintf() supports a subset of snprintf(3) that avoids floating
|
|
* point math.
|
|
*/
|
|
size_t malloc_vsnprintf(char *str, size_t size, const char *format,
|
|
va_list ap);
|
|
size_t malloc_snprintf(char *str, size_t size, const char *format, ...)
|
|
JEMALLOC_FORMAT_PRINTF(3, 4);
|
|
/*
|
|
* The caller can set write_cb to null to choose to print with the
|
|
* je_malloc_message hook.
|
|
*/
|
|
void malloc_vcprintf(void (*write_cb)(void *, const char *), void *cbopaque,
|
|
const char *format, va_list ap);
|
|
void malloc_cprintf(void (*write_cb)(void *, const char *), void *cbopaque,
|
|
const char *format, ...) JEMALLOC_FORMAT_PRINTF(3, 4);
|
|
void malloc_printf(const char *format, ...) JEMALLOC_FORMAT_PRINTF(1, 2);
|
|
|
|
static inline ssize_t
|
|
malloc_write_fd(int fd, const void *buf, size_t count) {
|
|
#if defined(JEMALLOC_USE_SYSCALL) && defined(SYS_write)
|
|
/*
|
|
* Use syscall(2) rather than write(2) when possible in order to avoid
|
|
* the possibility of memory allocation within libc. This is necessary
|
|
* on FreeBSD; most operating systems do not have this problem though.
|
|
*
|
|
* syscall() returns long or int, depending on platform, so capture the
|
|
* result in the widest plausible type to avoid compiler warnings.
|
|
*/
|
|
long result = syscall(SYS_write, fd, buf, count);
|
|
#else
|
|
ssize_t result = (ssize_t)write(fd, buf,
|
|
#ifdef _WIN32
|
|
(unsigned int)
|
|
#endif
|
|
count);
|
|
#endif
|
|
return (ssize_t)result;
|
|
}
|
|
|
|
static inline ssize_t
|
|
malloc_read_fd(int fd, void *buf, size_t count) {
|
|
#if defined(JEMALLOC_USE_SYSCALL) && defined(SYS_read)
|
|
long result = syscall(SYS_read, fd, buf, count);
|
|
#else
|
|
ssize_t result = read(fd, buf,
|
|
#ifdef _WIN32
|
|
(unsigned int)
|
|
#endif
|
|
count);
|
|
#endif
|
|
return (ssize_t)result;
|
|
}
|
|
|
|
/******************************************************************************/
|
|
|
|
/*
|
|
* The rest is buffered writing utility.
|
|
*
|
|
* The only difference when using the buffered writer is that cbopaque is
|
|
* passed to write_cb only when the buffer is flushed. It would make a
|
|
* difference if cbopaque points to something that's changing for each write_cb
|
|
* call, or something that affects write_cb in a way dependent on the content
|
|
* of the output string. However, the most typical usage case in practice is
|
|
* that cbopaque points to some "option like" content for the write_cb, so it
|
|
* doesn't matter.
|
|
*/
|
|
|
|
typedef struct {
|
|
void (*write_cb)(void *, const char *);
|
|
void *cbopaque;
|
|
char *buf;
|
|
size_t buf_size; /* must be one less than the capacity of buf array */
|
|
size_t buf_end;
|
|
} buf_writer_arg_t;
|
|
|
|
void buf_writer_flush(buf_writer_arg_t *arg);
|
|
void buffered_write_cb(void *buf_writer_arg, const char *s);
|
|
|
|
#endif /* JEMALLOC_INTERNAL_MALLOC_IO_H */
|