| /* | |
| * os-posix-lib.c | |
| * | |
| * Copyright (c) 2003-2008 Fabrice Bellard | |
| * Copyright (c) 2010 Red Hat, Inc. | |
| * | |
| * QEMU library functions on POSIX which are shared between QEMU and | |
| * the QEMU tools. | |
| * | |
| * Permission is hereby granted, free of charge, to any person obtaining a copy | |
| * of this software and associated documentation files (the "Software"), to deal | |
| * in the Software without restriction, including without limitation the rights | |
| * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |
| * copies of the Software, and to permit persons to whom the Software is | |
| * furnished to do so, subject to the following conditions: | |
| * | |
| * The above copyright notice and this permission notice shall be included in | |
| * all copies or substantial portions of the Software. | |
| * | |
| * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
| * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
| * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL | |
| * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |
| * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
| * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | |
| * THE SOFTWARE. | |
| */ | |
| #include "qemu/osdep.h" | |
| #include <termios.h> | |
| #include <glib/gprintf.h> | |
| #include "system/system.h" | |
| #include "trace.h" | |
| #include "qapi/error.h" | |
| #include "qemu/error-report.h" | |
| #include "qemu/madvise.h" | |
| #include "qemu/sockets.h" | |
| #include "qemu/thread.h" | |
| #include <libgen.h> | |
| #include "qemu/cutils.h" | |
| #include "qemu/units.h" | |
| #include "qemu/thread-context.h" | |
| #include "qemu/main-loop.h" | |
| #ifdef CONFIG_LINUX | |
| #include <sys/syscall.h> | |
| #endif | |
| #ifdef __FreeBSD__ | |
| #include <sys/thr.h> | |
| #include <sys/user.h> | |
| #include <libutil.h> | |
| #endif | |
| #ifdef __NetBSD__ | |
| #include <lwp.h> | |
| #endif | |
| #include "qemu/memalign.h" | |
| #include "qemu/mmap-alloc.h" | |
| #define MAX_MEM_PREALLOC_THREAD_COUNT 32 | |
| struct MemsetThread; | |
| static QLIST_HEAD(, MemsetContext) memset_contexts = | |
| QLIST_HEAD_INITIALIZER(memset_contexts); | |
| typedef struct MemsetContext { | |
| bool all_threads_created; | |
| bool any_thread_failed; | |
| struct MemsetThread *threads; | |
| int num_threads; | |
| QLIST_ENTRY(MemsetContext) next; | |
| } MemsetContext; | |
| struct MemsetThread { | |
| char *addr; | |
| size_t numpages; | |
| size_t hpagesize; | |
| QemuThread pgthread; | |
| sigjmp_buf env; | |
| MemsetContext *context; | |
| }; | |
| typedef struct MemsetThread MemsetThread; | |
| /* used by sigbus_handler() */ | |
| static MemsetContext *sigbus_memset_context; | |
| struct sigaction sigbus_oldact; | |
| static QemuMutex sigbus_mutex; | |
| static QemuMutex page_mutex; | |
| static QemuCond page_cond; | |
| int qemu_get_thread_id(void) | |
| { | |
| #if defined(__linux__) | |
| return syscall(SYS_gettid); | |
| #elif defined(__FreeBSD__) | |
| /* thread id is up to INT_MAX */ | |
| long tid; | |
| thr_self(&tid); | |
| return (int)tid; | |
| #elif defined(__NetBSD__) | |
| return _lwp_self(); | |
| #elif defined(__OpenBSD__) | |
| return getthrid(); | |
| #else | |
| return getpid(); | |
| #endif | |
| } | |
| int qemu_kill_thread(int tid, int sig) | |
| { | |
| #if defined(__linux__) | |
| return syscall(__NR_tgkill, getpid(), tid, sig); | |
| #elif defined(__FreeBSD__) | |
| return thr_kill2(getpid(), tid, sig); | |
| #elif defined(__NetBSD__) | |
| return _lwp_kill(tid, sig); | |
| #elif defined(__OpenBSD__) | |
| return thrkill(tid, sig, NULL); | |
| #else | |
| return kill(tid, sig); | |
| #endif | |
| } | |
| int qemu_daemon(int nochdir, int noclose) | |
| { | |
| return daemon(nochdir, noclose); | |
| } | |
| bool qemu_write_pidfile(const char *path, Error **errp) | |
| { | |
| int fd; | |
| char pidstr[32]; | |
| while (1) { | |
| struct stat a, b; | |
| struct flock lock = { | |
| .l_type = F_WRLCK, | |
| .l_whence = SEEK_SET, | |
| .l_len = 0, | |
| }; | |
| fd = qemu_create(path, O_WRONLY, S_IRUSR | S_IWUSR, errp); | |
| if (fd == -1) { | |
| return false; | |
| } | |
| if (fstat(fd, &b) < 0) { | |
| error_setg_errno(errp, errno, "Cannot stat file"); | |
| goto fail_close; | |
| } | |
| if (fcntl(fd, F_SETLK, &lock)) { | |
| error_setg_errno(errp, errno, "Cannot lock pid file"); | |
| goto fail_close; | |
| } | |
| /* | |
| * Now make sure the path we locked is the same one that now | |
| * exists on the filesystem. | |
| */ | |
| if (stat(path, &a) < 0) { | |
| /* | |
| * PID file disappeared, someone else must be racing with | |
| * us, so try again. | |
| */ | |
| close(fd); | |
| continue; | |
| } | |
| if (a.st_ino == b.st_ino) { | |
| break; | |
| } | |
| /* | |
| * PID file was recreated, someone else must be racing with | |
| * us, so try again. | |
| */ | |
| close(fd); | |
| } | |
| if (ftruncate(fd, 0) < 0) { | |
| error_setg_errno(errp, errno, "Failed to truncate pid file"); | |
| goto fail_unlink; | |
| } | |
| snprintf(pidstr, sizeof(pidstr), FMT_pid "\n", getpid()); | |
| if (qemu_write_full(fd, pidstr, strlen(pidstr)) != strlen(pidstr)) { | |
| error_setg(errp, "Failed to write pid file"); | |
| goto fail_unlink; | |
| } | |
| return true; | |
| fail_unlink: | |
| unlink(path); | |
| fail_close: | |
| close(fd); | |
| return false; | |
| } | |
| /* alloc shared memory pages */ | |
| void *qemu_anon_ram_alloc(size_t size, uint64_t *alignment, bool shared, | |
| bool noreserve) | |
| { | |
| const uint32_t qemu_map_flags = (shared ? QEMU_MAP_SHARED : 0) | | |
| (noreserve ? QEMU_MAP_NORESERVE : 0); | |
| size_t align = QEMU_VMALLOC_ALIGN; | |
| #ifndef EMSCRIPTEN | |
| void *ptr = qemu_ram_mmap(-1, size, align, qemu_map_flags, 0); | |
| if (ptr == MAP_FAILED) { | |
| return NULL; | |
| } | |
| #else | |
| /* | |
| * qemu_ram_mmap is not implemented for Emscripten. Use qemu_memalign | |
| * for the anonymous allocation. noreserve is ignored as there is no swap | |
| * space on Emscripten, and shared is ignored as there is no other | |
| * processes on Emscripten. | |
| */ | |
| void *ptr = qemu_memalign(align, size); | |
| #endif | |
| if (alignment) { | |
| *alignment = align; | |
| } | |
| trace_qemu_anon_ram_alloc(size, ptr); | |
| return ptr; | |
| } | |
| void qemu_anon_ram_free(void *ptr, size_t size) | |
| { | |
| trace_qemu_anon_ram_free(ptr, size); | |
| #ifndef EMSCRIPTEN | |
| qemu_ram_munmap(-1, ptr, size); | |
| #else | |
| /* | |
| * qemu_ram_munmap is not implemented for Emscripten and qemu_memalign | |
| * was used for the allocation. Use the corresponding freeing function | |
| * here. | |
| */ | |
| qemu_vfree(ptr); | |
| #endif | |
| } | |
| bool qemu_set_blocking(int fd, bool block, Error **errp) | |
| { | |
| g_autoptr(GError) err = NULL; | |
| if (!g_unix_set_fd_nonblocking(fd, !block, &err)) { | |
| error_setg_errno(errp, errno, | |
| "Can't set file descriptor %d %s: %s", fd, | |
| block ? "blocking" : "non-blocking", | |
| err->message); | |
| return false; | |
| } | |
| return true; | |
| } | |
| int socket_set_fast_reuse(int fd) | |
| { | |
| int val = 1, ret; | |
| ret = setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, | |
| (const char *)&val, sizeof(val)); | |
| assert(ret == 0); | |
| return ret; | |
| } | |
| void qemu_set_cloexec(int fd) | |
| { | |
| int f; | |
| f = fcntl(fd, F_GETFD); | |
| assert(f != -1); | |
| f = fcntl(fd, F_SETFD, f | FD_CLOEXEC); | |
| assert(f != -1); | |
| } | |
| int qemu_socketpair(int domain, int type, int protocol, int sv[2]) | |
| { | |
| int ret; | |
| #ifdef SOCK_CLOEXEC | |
| ret = socketpair(domain, type | SOCK_CLOEXEC, protocol, sv); | |
| if (ret != -1 || errno != EINVAL) { | |
| return ret; | |
| } | |
| #endif | |
| ret = socketpair(domain, type, protocol, sv); | |
| if (ret == 0) { | |
| qemu_set_cloexec(sv[0]); | |
| qemu_set_cloexec(sv[1]); | |
| } | |
| return ret; | |
| } | |
| void qemu_clear_cloexec(int fd) | |
| { | |
| int f; | |
| f = fcntl(fd, F_GETFD); | |
| assert(f != -1); | |
| f = fcntl(fd, F_SETFD, f & ~FD_CLOEXEC); | |
| assert(f != -1); | |
| } | |
| char * | |
| qemu_get_local_state_dir(void) | |
| { | |
| return get_relocated_path(CONFIG_QEMU_LOCALSTATEDIR); | |
| } | |
| void qemu_set_tty_echo(int fd, bool echo) | |
| { | |
| struct termios tty; | |
| tcgetattr(fd, &tty); | |
| if (echo) { | |
| tty.c_lflag |= ECHO | ECHONL | ICANON | IEXTEN; | |
| } else { | |
| tty.c_lflag &= ~(ECHO | ECHONL | ICANON | IEXTEN); | |
| } | |
| tcsetattr(fd, TCSANOW, &tty); | |
| } | |
| #ifdef CONFIG_LINUX | |
| static void sigbus_handler(int signal, siginfo_t *siginfo, void *ctx) | |
| #else /* CONFIG_LINUX */ | |
| static void sigbus_handler(int signal) | |
| #endif /* CONFIG_LINUX */ | |
| { | |
| int i; | |
| if (sigbus_memset_context) { | |
| for (i = 0; i < sigbus_memset_context->num_threads; i++) { | |
| MemsetThread *thread = &sigbus_memset_context->threads[i]; | |
| if (qemu_thread_is_self(&thread->pgthread)) { | |
| siglongjmp(thread->env, 1); | |
| } | |
| } | |
| } | |
| #ifdef CONFIG_LINUX | |
| /* | |
| * We assume that the MCE SIGBUS handler could have been registered. We | |
| * should never receive BUS_MCEERR_AO on any of our threads, but only on | |
| * the main thread registered for PR_MCE_KILL_EARLY. Further, we should not | |
| * receive BUS_MCEERR_AR triggered by action of other threads on one of | |
| * our threads. So, no need to check for unrelated SIGBUS when seeing one | |
| * for our threads. | |
| * | |
| * We will forward to the MCE handler, which will either handle the SIGBUS | |
| * or reinstall the default SIGBUS handler and reraise the SIGBUS. The | |
| * default SIGBUS handler will crash the process, so we don't care. | |
| */ | |
| if (sigbus_oldact.sa_flags & SA_SIGINFO) { | |
| sigbus_oldact.sa_sigaction(signal, siginfo, ctx); | |
| return; | |
| } | |
| #endif /* CONFIG_LINUX */ | |
| warn_report("qemu_prealloc_mem: unrelated SIGBUS detected and ignored"); | |
| } | |
| static void *do_touch_pages(void *arg) | |
| { | |
| MemsetThread *memset_args = (MemsetThread *)arg; | |
| sigset_t set, oldset; | |
| int ret = 0; | |
| /* | |
| * On Linux, the page faults from the loop below can cause mmap_sem | |
| * contention with allocation of the thread stacks. Do not start | |
| * clearing until all threads have been created. | |
| */ | |
| qemu_mutex_lock(&page_mutex); | |
| while (!memset_args->context->all_threads_created) { | |
| qemu_cond_wait(&page_cond, &page_mutex); | |
| } | |
| qemu_mutex_unlock(&page_mutex); | |
| /* unblock SIGBUS */ | |
| sigemptyset(&set); | |
| sigaddset(&set, SIGBUS); | |
| pthread_sigmask(SIG_UNBLOCK, &set, &oldset); | |
| if (sigsetjmp(memset_args->env, 1)) { | |
| ret = -EFAULT; | |
| } else { | |
| char *addr = memset_args->addr; | |
| size_t numpages = memset_args->numpages; | |
| size_t hpagesize = memset_args->hpagesize; | |
| size_t i; | |
| for (i = 0; i < numpages; i++) { | |
| /* | |
| * Read & write back the same value, so we don't | |
| * corrupt existing user/app data that might be | |
| * stored. | |
| * | |
| * 'volatile' to stop compiler optimizing this away | |
| * to a no-op | |
| */ | |
| *(volatile char *)addr = *addr; | |
| addr += hpagesize; | |
| } | |
| } | |
| pthread_sigmask(SIG_SETMASK, &oldset, NULL); | |
| return (void *)(uintptr_t)ret; | |
| } | |
| static void *do_madv_populate_write_pages(void *arg) | |
| { | |
| MemsetThread *memset_args = (MemsetThread *)arg; | |
| const size_t size = memset_args->numpages * memset_args->hpagesize; | |
| char * const addr = memset_args->addr; | |
| int ret = 0; | |
| /* See do_touch_pages(). */ | |
| qemu_mutex_lock(&page_mutex); | |
| while (!memset_args->context->all_threads_created) { | |
| qemu_cond_wait(&page_cond, &page_mutex); | |
| } | |
| qemu_mutex_unlock(&page_mutex); | |
| if (size && qemu_madvise(addr, size, QEMU_MADV_POPULATE_WRITE)) { | |
| ret = -errno; | |
| } | |
| return (void *)(uintptr_t)ret; | |
| } | |
| static inline int get_memset_num_threads(size_t hpagesize, size_t numpages, | |
| int max_threads) | |
| { | |
| long host_procs = sysconf(_SC_NPROCESSORS_ONLN); | |
| int ret = 1; | |
| if (host_procs > 0) { | |
| ret = MIN(MIN(host_procs, MAX_MEM_PREALLOC_THREAD_COUNT), max_threads); | |
| } | |
| /* Especially with gigantic pages, don't create more threads than pages. */ | |
| ret = MIN(ret, numpages); | |
| /* Don't start threads to prealloc comparatively little memory. */ | |
| ret = MIN(ret, MAX(1, hpagesize * numpages / (64 * MiB))); | |
| /* In case sysconf() fails, we fall back to single threaded */ | |
| return ret; | |
| } | |
| static int wait_and_free_mem_prealloc_context(MemsetContext *context) | |
| { | |
| int i, ret = 0, tmp; | |
| for (i = 0; i < context->num_threads; i++) { | |
| tmp = (uintptr_t)qemu_thread_join(&context->threads[i].pgthread); | |
| if (tmp) { | |
| ret = tmp; | |
| } | |
| } | |
| g_free(context->threads); | |
| g_free(context); | |
| return ret; | |
| } | |
| static int touch_all_pages(char *area, size_t hpagesize, size_t numpages, | |
| int max_threads, ThreadContext *tc, bool async, | |
| bool use_madv_populate_write) | |
| { | |
| static gsize initialized = 0; | |
| MemsetContext *context = g_malloc0(sizeof(MemsetContext)); | |
| size_t numpages_per_thread, leftover; | |
| void *(*touch_fn)(void *); | |
| int ret, i = 0; | |
| char *addr = area; | |
| /* | |
| * Asynchronous preallocation is only allowed when using MADV_POPULATE_WRITE | |
| * and prealloc context for thread placement. | |
| */ | |
| if (!use_madv_populate_write || !tc) { | |
| async = false; | |
| } | |
| context->num_threads = | |
| get_memset_num_threads(hpagesize, numpages, max_threads); | |
| if (g_once_init_enter(&initialized)) { | |
| qemu_mutex_init(&page_mutex); | |
| qemu_cond_init(&page_cond); | |
| g_once_init_leave(&initialized, 1); | |
| } | |
| if (use_madv_populate_write) { | |
| /* | |
| * Avoid creating a single thread for MADV_POPULATE_WRITE when | |
| * preallocating synchronously. | |
| */ | |
| if (context->num_threads == 1 && !async) { | |
| ret = 0; | |
| if (qemu_madvise(area, hpagesize * numpages, | |
| QEMU_MADV_POPULATE_WRITE)) { | |
| ret = -errno; | |
| } | |
| g_free(context); | |
| return ret; | |
| } | |
| touch_fn = do_madv_populate_write_pages; | |
| } else { | |
| touch_fn = do_touch_pages; | |
| } | |
| context->threads = g_new0(MemsetThread, context->num_threads); | |
| numpages_per_thread = numpages / context->num_threads; | |
| leftover = numpages % context->num_threads; | |
| for (i = 0; i < context->num_threads; i++) { | |
| context->threads[i].addr = addr; | |
| context->threads[i].numpages = numpages_per_thread + (i < leftover); | |
| context->threads[i].hpagesize = hpagesize; | |
| context->threads[i].context = context; | |
| if (tc) { | |
| thread_context_create_thread(tc, &context->threads[i].pgthread, | |
| "touch_pages", | |
| touch_fn, &context->threads[i], | |
| QEMU_THREAD_JOINABLE); | |
| } else { | |
| qemu_thread_create(&context->threads[i].pgthread, "touch_pages", | |
| touch_fn, &context->threads[i], | |
| QEMU_THREAD_JOINABLE); | |
| } | |
| addr += context->threads[i].numpages * hpagesize; | |
| } | |
| if (async) { | |
| /* | |
| * async requests currently require the BQL. Add it to the list and kick | |
| * preallocation off during qemu_finish_async_prealloc_mem(). | |
| */ | |
| assert(bql_locked()); | |
| QLIST_INSERT_HEAD(&memset_contexts, context, next); | |
| return 0; | |
| } | |
| if (!use_madv_populate_write) { | |
| sigbus_memset_context = context; | |
| } | |
| qemu_mutex_lock(&page_mutex); | |
| context->all_threads_created = true; | |
| qemu_cond_broadcast(&page_cond); | |
| qemu_mutex_unlock(&page_mutex); | |
| ret = wait_and_free_mem_prealloc_context(context); | |
| if (!use_madv_populate_write) { | |
| sigbus_memset_context = NULL; | |
| } | |
| return ret; | |
| } | |
| bool qemu_finish_async_prealloc_mem(Error **errp) | |
| { | |
| int ret = 0, tmp; | |
| MemsetContext *context, *next_context; | |
| /* Waiting for preallocation requires the BQL. */ | |
| assert(bql_locked()); | |
| if (QLIST_EMPTY(&memset_contexts)) { | |
| return true; | |
| } | |
| qemu_mutex_lock(&page_mutex); | |
| QLIST_FOREACH(context, &memset_contexts, next) { | |
| context->all_threads_created = true; | |
| } | |
| qemu_cond_broadcast(&page_cond); | |
| qemu_mutex_unlock(&page_mutex); | |
| QLIST_FOREACH_SAFE(context, &memset_contexts, next, next_context) { | |
| QLIST_REMOVE(context, next); | |
| tmp = wait_and_free_mem_prealloc_context(context); | |
| if (tmp) { | |
| ret = tmp; | |
| } | |
| } | |
| if (ret) { | |
| error_setg_errno(errp, -ret, | |
| "qemu_prealloc_mem: preallocating memory failed"); | |
| return false; | |
| } | |
| return true; | |
| } | |
| static bool madv_populate_write_possible(char *area, size_t pagesize) | |
| { | |
| return !qemu_madvise(area, pagesize, QEMU_MADV_POPULATE_WRITE) || | |
| errno != EINVAL; | |
| } | |
| bool qemu_prealloc_mem(int fd, char *area, size_t sz, int max_threads, | |
| ThreadContext *tc, bool async, Error **errp) | |
| { | |
| static gsize initialized; | |
| int ret; | |
| #ifndef EMSCRIPTEN | |
| size_t hpagesize = qemu_fd_getpagesize(fd); | |
| #else | |
| /* | |
| * mmap-alloc.c is excluded from Emscripten build, so qemu_fd_getpagesize | |
| * is unavailable. Fallback to the lower level implementation. | |
| */ | |
| size_t hpagesize = qemu_real_host_page_size(); | |
| #endif | |
| size_t numpages = DIV_ROUND_UP(sz, hpagesize); | |
| bool use_madv_populate_write; | |
| struct sigaction act; | |
| bool rv = true; | |
| /* | |
| * Sense on every invocation, as MADV_POPULATE_WRITE cannot be used for | |
| * some special mappings, such as mapping /dev/mem. | |
| */ | |
| use_madv_populate_write = madv_populate_write_possible(area, hpagesize); | |
| if (!use_madv_populate_write) { | |
| if (g_once_init_enter(&initialized)) { | |
| qemu_mutex_init(&sigbus_mutex); | |
| g_once_init_leave(&initialized, 1); | |
| } | |
| qemu_mutex_lock(&sigbus_mutex); | |
| memset(&act, 0, sizeof(act)); | |
| #ifdef CONFIG_LINUX | |
| act.sa_sigaction = &sigbus_handler; | |
| act.sa_flags = SA_SIGINFO; | |
| #else /* CONFIG_LINUX */ | |
| act.sa_handler = &sigbus_handler; | |
| act.sa_flags = 0; | |
| #endif /* CONFIG_LINUX */ | |
| ret = sigaction(SIGBUS, &act, &sigbus_oldact); | |
| if (ret) { | |
| qemu_mutex_unlock(&sigbus_mutex); | |
| error_setg_errno(errp, errno, | |
| "qemu_prealloc_mem: failed to install signal handler"); | |
| return false; | |
| } | |
| } | |
| /* touch pages simultaneously */ | |
| ret = touch_all_pages(area, hpagesize, numpages, max_threads, tc, async, | |
| use_madv_populate_write); | |
| if (ret) { | |
| error_setg_errno(errp, -ret, | |
| "qemu_prealloc_mem: preallocating memory failed"); | |
| rv = false; | |
| } | |
| if (!use_madv_populate_write) { | |
| ret = sigaction(SIGBUS, &sigbus_oldact, NULL); | |
| if (ret) { | |
| /* Terminate QEMU since it can't recover from error */ | |
| perror("qemu_prealloc_mem: failed to reinstall signal handler"); | |
| exit(1); | |
| } | |
| qemu_mutex_unlock(&sigbus_mutex); | |
| } | |
| return rv; | |
| } | |
| char *qemu_get_pid_name(pid_t pid) | |
| { | |
| char *name = NULL; | |
| #if defined(__FreeBSD__) | |
| /* BSDs don't have /proc, but they provide a nice substitute */ | |
| struct kinfo_proc *proc = kinfo_getproc(pid); | |
| if (proc) { | |
| name = g_strdup(proc->ki_comm); | |
| free(proc); | |
| } | |
| #else | |
| /* Assume a system with reasonable procfs */ | |
| char *pid_path; | |
| size_t len; | |
| pid_path = g_strdup_printf("/proc/%d/cmdline", pid); | |
| g_file_get_contents(pid_path, &name, &len, NULL); | |
| g_free(pid_path); | |
| #endif | |
| return name; | |
| } | |
| void *qemu_alloc_stack(size_t *sz) | |
| { | |
| void *ptr; | |
| int flags; | |
| #ifdef CONFIG_DEBUG_STACK_USAGE | |
| void *ptr2; | |
| #endif | |
| size_t pagesz = qemu_real_host_page_size(); | |
| #ifdef _SC_THREAD_STACK_MIN | |
| /* avoid stacks smaller than _SC_THREAD_STACK_MIN */ | |
| long min_stack_sz = sysconf(_SC_THREAD_STACK_MIN); | |
| *sz = MAX(MAX(min_stack_sz, 0), *sz); | |
| #endif | |
| /* adjust stack size to a multiple of the page size */ | |
| *sz = ROUND_UP(*sz, pagesz); | |
| /* allocate one extra page for the guard page */ | |
| *sz += pagesz; | |
| flags = MAP_PRIVATE | MAP_ANONYMOUS; | |
| #if defined(MAP_STACK) && defined(__OpenBSD__) | |
| /* Only enable MAP_STACK on OpenBSD. Other OS's such as | |
| * Linux/FreeBSD/NetBSD have a flag with the same name | |
| * but have differing functionality. OpenBSD will SEGV | |
| * if it spots execution with a stack pointer pointing | |
| * at memory that was not allocated with MAP_STACK. | |
| */ | |
| flags |= MAP_STACK; | |
| #endif | |
| ptr = mmap(NULL, *sz, PROT_READ | PROT_WRITE, flags, -1, 0); | |
| if (ptr == MAP_FAILED) { | |
| perror("failed to allocate memory for stack"); | |
| abort(); | |
| } | |
| /* Stack grows down -- guard page at the bottom. */ | |
| if (mprotect(ptr, pagesz, PROT_NONE) != 0) { | |
| perror("failed to set up stack guard page"); | |
| abort(); | |
| } | |
| #ifdef CONFIG_DEBUG_STACK_USAGE | |
| for (ptr2 = ptr + pagesz; ptr2 < ptr + *sz; ptr2 += sizeof(uint32_t)) { | |
| *(uint32_t *)ptr2 = 0xdeadbeaf; | |
| } | |
| #endif | |
| return ptr; | |
| } | |
| #ifdef CONFIG_DEBUG_STACK_USAGE | |
| static __thread unsigned int max_stack_usage; | |
| #endif | |
| void qemu_free_stack(void *stack, size_t sz) | |
| { | |
| #ifdef CONFIG_DEBUG_STACK_USAGE | |
| unsigned int usage; | |
| void *ptr; | |
| for (ptr = stack + qemu_real_host_page_size(); ptr < stack + sz; | |
| ptr += sizeof(uint32_t)) { | |
| if (*(uint32_t *)ptr != 0xdeadbeaf) { | |
| break; | |
| } | |
| } | |
| usage = sz - (uintptr_t) (ptr - stack); | |
| if (usage > max_stack_usage) { | |
| error_report("thread %d max stack usage increased from %u to %u", | |
| qemu_get_thread_id(), max_stack_usage, usage); | |
| max_stack_usage = usage; | |
| } | |
| #endif | |
| munmap(stack, sz); | |
| } | |
| /* | |
| * Disable CFI checks. | |
| * We are going to call a signal handler directly. Such handler may or may not | |
| * have been defined in our binary, so there's no guarantee that the pointer | |
| * used to set the handler is a cfi-valid pointer. Since the handlers are | |
| * stored in kernel memory, changing the handler to an attacker-defined | |
| * function requires being able to call a sigaction() syscall, | |
| * which is not as easy as overwriting a pointer in memory. | |
| */ | |
| QEMU_DISABLE_CFI | |
| void sigaction_invoke(struct sigaction *action, | |
| struct qemu_signalfd_siginfo *info) | |
| { | |
| siginfo_t si = {}; | |
| si.si_signo = info->ssi_signo; | |
| si.si_errno = info->ssi_errno; | |
| si.si_code = info->ssi_code; | |
| /* Convert the minimal set of fields defined by POSIX. | |
| * Positive si_code values are reserved for kernel-generated | |
| * signals, where the valid siginfo fields are determined by | |
| * the signal number. But according to POSIX, it is unspecified | |
| * whether SI_USER and SI_QUEUE have values less than or equal to | |
| * zero. | |
| */ | |
| if (info->ssi_code == SI_USER || info->ssi_code == SI_QUEUE || | |
| info->ssi_code <= 0) { | |
| /* SIGTERM, etc. */ | |
| si.si_pid = info->ssi_pid; | |
| si.si_uid = info->ssi_uid; | |
| } else if (info->ssi_signo == SIGILL || info->ssi_signo == SIGFPE || | |
| info->ssi_signo == SIGSEGV || info->ssi_signo == SIGBUS) { | |
| si.si_addr = (void *)(uintptr_t)info->ssi_addr; | |
| } else if (info->ssi_signo == SIGCHLD) { | |
| si.si_pid = info->ssi_pid; | |
| si.si_status = info->ssi_status; | |
| si.si_uid = info->ssi_uid; | |
| } | |
| action->sa_sigaction(info->ssi_signo, &si, NULL); | |
| } | |
| size_t qemu_get_host_physmem(void) | |
| { | |
| #ifdef _SC_PHYS_PAGES | |
| long pages = sysconf(_SC_PHYS_PAGES); | |
| if (pages > 0) { | |
| if (pages > SIZE_MAX / qemu_real_host_page_size()) { | |
| return SIZE_MAX; | |
| } else { | |
| return pages * qemu_real_host_page_size(); | |
| } | |
| } | |
| #endif | |
| return 0; | |
| } | |
| int qemu_msync(void *addr, size_t length, int fd) | |
| { | |
| size_t align_mask = ~(qemu_real_host_page_size() - 1); | |
| /** | |
| * There are no strict reqs as per the length of mapping | |
| * to be synced. Still the length needs to follow the address | |
| * alignment changes. Additionally - round the size to the multiple | |
| * of PAGE_SIZE | |
| */ | |
| length += ((uintptr_t)addr & (qemu_real_host_page_size() - 1)); | |
| length = (length + ~align_mask) & align_mask; | |
| addr = (void *)((uintptr_t)addr & align_mask); | |
| return msync(addr, length, MS_SYNC); | |
| } | |
| static bool qemu_close_all_open_fd_proc(const int *skip, unsigned int nskip) | |
| { | |
| struct dirent *de; | |
| int fd, dfd; | |
| DIR *dir; | |
| unsigned int skip_start = 0, skip_end = nskip; | |
| dir = opendir("/proc/self/fd"); | |
| if (!dir) { | |
| /* If /proc is not mounted, there is nothing that can be done. */ | |
| return false; | |
| } | |
| /* Avoid closing the directory. */ | |
| dfd = dirfd(dir); | |
| for (de = readdir(dir); de; de = readdir(dir)) { | |
| bool close_fd = true; | |
| if (de->d_name[0] == '.') { | |
| continue; | |
| } | |
| fd = atoi(de->d_name); | |
| if (fd == dfd) { | |
| continue; | |
| } | |
| for (unsigned int i = skip_start; i < skip_end; i++) { | |
| if (fd < skip[i]) { | |
| /* We are below the next skipped fd, break */ | |
| break; | |
| } else if (fd == skip[i]) { | |
| close_fd = false; | |
| /* Restrict the range as we found fds matching start/end */ | |
| if (i == skip_start) { | |
| skip_start++; | |
| } else if (i == skip_end) { | |
| skip_end--; | |
| } | |
| break; | |
| } | |
| } | |
| if (close_fd) { | |
| close(fd); | |
| } | |
| } | |
| closedir(dir); | |
| return true; | |
| } | |
| static bool qemu_close_all_open_fd_close_range(const int *skip, | |
| unsigned int nskip, | |
| int open_max) | |
| { | |
| #ifdef CONFIG_CLOSE_RANGE | |
| int max_fd = open_max - 1; | |
| int first = 0, last; | |
| unsigned int cur_skip = 0; | |
| int ret; | |
| do { | |
| /* Find the start boundary of the range to close */ | |
| while (cur_skip < nskip && first == skip[cur_skip]) { | |
| cur_skip++; | |
| first++; | |
| } | |
| /* Find the upper boundary of the range to close */ | |
| last = max_fd; | |
| if (cur_skip < nskip) { | |
| last = skip[cur_skip] - 1; | |
| last = MIN(last, max_fd); | |
| } | |
| /* With the adjustments to the range, we might be done. */ | |
| if (first > last) { | |
| break; | |
| } | |
| ret = close_range(first, last, 0); | |
| if (ret < 0) { | |
| return false; | |
| } | |
| first = last + 1; | |
| } while (last < max_fd); | |
| return true; | |
| #else | |
| return false; | |
| #endif | |
| } | |
| static void qemu_close_all_open_fd_fallback(const int *skip, unsigned int nskip, | |
| int open_max) | |
| { | |
| unsigned int cur_skip = 0; | |
| /* Fallback */ | |
| for (int i = 0; i < open_max; i++) { | |
| if (cur_skip < nskip && i == skip[cur_skip]) { | |
| cur_skip++; | |
| continue; | |
| } | |
| close(i); | |
| } | |
| } | |
| /* | |
| * Close all open file descriptors. | |
| */ | |
| void qemu_close_all_open_fd(const int *skip, unsigned int nskip) | |
| { | |
| int open_max = sysconf(_SC_OPEN_MAX); | |
| assert(skip != NULL || nskip == 0); | |
| if (!qemu_close_all_open_fd_close_range(skip, nskip, open_max) && | |
| !qemu_close_all_open_fd_proc(skip, nskip)) { | |
| qemu_close_all_open_fd_fallback(skip, nskip, open_max); | |
| } | |
| } | |
| int qemu_shm_alloc(size_t size, Error **errp) | |
| { | |
| g_autoptr(GString) shm_name = g_string_new(NULL); | |
| int fd, oflag, cur_sequence; | |
| static int sequence; | |
| mode_t mode; | |
| cur_sequence = qatomic_fetch_inc(&sequence); | |
| /* | |
| * Let's use `mode = 0` because we don't want other processes to open our | |
| * memory unless we share the file descriptor with them. | |
| */ | |
| mode = 0; | |
| oflag = O_RDWR | O_CREAT | O_EXCL; | |
| /* | |
| * Some operating systems allow creating anonymous POSIX shared memory | |
| * objects (e.g. FreeBSD provides the SHM_ANON constant), but this is not | |
| * defined by POSIX, so let's create a unique name. | |
| * | |
| * From Linux's shm_open(3) man-page: | |
| * For portable use, a shared memory object should be identified | |
| * by a name of the form /somename;" | |
| */ | |
| g_string_printf(shm_name, "/qemu-" FMT_pid "-shm-%d", getpid(), | |
| cur_sequence); | |
| fd = shm_open(shm_name->str, oflag, mode); | |
| if (fd < 0) { | |
| error_setg_errno(errp, errno, | |
| "failed to create POSIX shared memory"); | |
| return -1; | |
| } | |
| /* | |
| * We have the file descriptor, so we no longer need to expose the | |
| * POSIX shared memory object. However it will remain allocated as long as | |
| * there are file descriptors pointing to it. | |
| */ | |
| shm_unlink(shm_name->str); | |
| if (ftruncate(fd, size) == -1) { | |
| error_setg_errno(errp, errno, | |
| "failed to resize POSIX shared memory to %zu", size); | |
| close(fd); | |
| return -1; | |
| } | |
| return fd; | |
| } |