hardened_malloc/util.c

45 lines
1.1 KiB
C
Raw Normal View History

2018-10-03 16:55:25 -04:00
#include <errno.h>
2018-08-21 15:23:22 -04:00
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#ifdef __ANDROID__
#include <async_safe/log.h>
int mallopt(int param, int value);
#define M_BIONIC_RESTORE_DEFAULT_SIGABRT_HANDLER (-1003)
#endif
2018-08-21 15:23:22 -04:00
#include "util.h"
#ifndef __ANDROID__
2018-10-03 16:55:25 -04:00
static int write_full(int fd, const char *buf, size_t length) {
do {
ssize_t bytes_written = write(fd, buf, length);
if (bytes_written == -1) {
if (errno == EINTR) {
continue;
}
return -1;
}
buf += bytes_written;
length -= bytes_written;
} while (length);
return 0;
}
#endif
2018-10-03 16:55:25 -04:00
2018-08-21 15:23:22 -04:00
COLD noreturn void fatal_error(const char *s) {
#ifdef __ANDROID__
mallopt(M_BIONIC_RESTORE_DEFAULT_SIGABRT_HANDLER, 0);
async_safe_fatal("hardened_malloc: fatal allocator error: %s", s);
#else
2018-10-03 16:22:28 -04:00
const char *prefix = "fatal allocator error: ";
2018-10-03 16:55:25 -04:00
(void)(write_full(STDERR_FILENO, prefix, strlen(prefix)) != -1 &&
write_full(STDERR_FILENO, s, strlen(s)) != -1 &&
write_full(STDERR_FILENO, "\n", 1));
2018-08-21 15:23:22 -04:00
abort();
#endif
2018-08-21 15:23:22 -04:00
}