| /* |
| * QEMU low level functions |
| * |
| * Copyright (c) 2003 Fabrice Bellard |
| * |
| * 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 <stdlib.h> |
| #include <stdio.h> |
| #include <stdarg.h> |
| #include <string.h> |
| #include <errno.h> |
| #include <unistd.h> |
| #include <fcntl.h> |
| |
| /* Needed early for CONFIG_BSD etc. */ |
| #include "config-host.h" |
| |
| #ifdef CONFIG_SOLARIS |
| #include <sys/types.h> |
| #include <sys/statvfs.h> |
| #endif |
| |
| #ifdef CONFIG_EVENTFD |
| #include <sys/eventfd.h> |
| #endif |
| |
| #ifdef _WIN32 |
| #include <windows.h> |
| #elif defined(CONFIG_BSD) |
| #include <stdlib.h> |
| #else |
| #include <malloc.h> |
| #endif |
| |
| #ifdef CONFIG_ANDROID |
| #ifdef WIN32 |
| #include <winsock2.h> |
| #include <stdint.h> |
| typedef int32_t socklen_t; |
| #else |
| #include <sys/socket.h> |
| #endif |
| #endif /* CONFIG_ANDROID */ |
| |
| #include "qemu-common.h" |
| #include "sysemu.h" |
| #include "qemu_socket.h" |
| |
| #if !defined(_POSIX_C_SOURCE) || defined(_WIN32) || defined(__sun__) || defined(__APPLE__) |
| static void *oom_check(void *ptr) |
| { |
| if (ptr == NULL) { |
| #if defined(_WIN32) |
| fprintf(stderr, "Failed to allocate memory: %lu\n", GetLastError()); |
| #else |
| fprintf(stderr, "Failed to allocate memory: %s\n", strerror(errno)); |
| #endif |
| abort(); |
| } |
| return ptr; |
| } |
| #endif |
| |
| #if defined(_WIN32) |
| void *qemu_memalign(size_t alignment, size_t size) |
| { |
| if (!size) { |
| abort(); |
| } |
| return oom_check(VirtualAlloc(NULL, size, MEM_COMMIT, PAGE_READWRITE)); |
| } |
| |
| void *qemu_vmalloc(size_t size) |
| { |
| /* FIXME: this is not exactly optimal solution since VirtualAlloc |
| has 64Kb granularity, but at least it guarantees us that the |
| memory is page aligned. */ |
| if (!size) { |
| abort(); |
| } |
| return oom_check(VirtualAlloc(NULL, size, MEM_COMMIT, PAGE_READWRITE)); |
| } |
| |
| void qemu_vfree(void *ptr) |
| { |
| VirtualFree(ptr, 0, MEM_RELEASE); |
| } |
| |
| #else |
| |
| void *qemu_memalign(size_t alignment, size_t size) |
| { |
| #if defined(_POSIX_C_SOURCE) && !defined(__sun__) && !defined(__APPLE__) |
| int ret; |
| void *ptr; |
| ret = posix_memalign(&ptr, alignment, size); |
| if (ret != 0) { |
| fprintf(stderr, "Failed to allocate %zu B: %s\n", |
| size, strerror(ret)); |
| abort(); |
| } |
| return ptr; |
| #elif defined(CONFIG_BSD) |
| return oom_check(valloc(size)); |
| #else |
| return oom_check(memalign(alignment, size)); |
| #endif |
| } |
| |
| /* alloc shared memory pages */ |
| void *qemu_vmalloc(size_t size) |
| { |
| return qemu_memalign(getpagesize(), size); |
| } |
| |
| void qemu_vfree(void *ptr) |
| { |
| free(ptr); |
| } |
| |
| #endif |
| |
| int qemu_create_pidfile(const char *filename) |
| { |
| char buffer[128]; |
| int len; |
| #ifndef _WIN32 |
| int fd; |
| |
| fd = qemu_open(filename, O_RDWR | O_CREAT, 0600); |
| if (fd == -1) |
| return -1; |
| |
| if (lockf(fd, F_TLOCK, 0) == -1) |
| return -1; |
| |
| len = snprintf(buffer, sizeof(buffer), "%ld\n", (long)getpid()); |
| if (write(fd, buffer, len) != len) |
| return -1; |
| #else |
| HANDLE file; |
| OVERLAPPED overlap; |
| BOOL ret; |
| memset(&overlap, 0, sizeof(overlap)); |
| |
| file = CreateFile(filename, GENERIC_WRITE, FILE_SHARE_READ, NULL, |
| OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); |
| |
| if (file == INVALID_HANDLE_VALUE) |
| return -1; |
| |
| len = snprintf(buffer, sizeof(buffer), "%ld\n", (long)getpid()); |
| ret = WriteFileEx(file, (LPCVOID)buffer, (DWORD)len, |
| &overlap, NULL); |
| if (ret == 0) |
| return -1; |
| #endif |
| return 0; |
| } |
| |
| #ifdef _WIN32 |
| |
| /* mingw32 needs ffs for compilations without optimization. */ |
| int ffs(int i) |
| { |
| /* Use gcc's builtin ffs. */ |
| return __builtin_ffs(i); |
| } |
| |
| /* Offset between 1/1/1601 and 1/1/1970 in 100 nanosec units */ |
| #define _W32_FT_OFFSET (116444736000000000ULL) |
| |
| int qemu_gettimeofday(qemu_timeval *tp) |
| { |
| union { |
| unsigned long long ns100; /*time since 1 Jan 1601 in 100ns units */ |
| FILETIME ft; |
| } _now; |
| |
| if(tp) |
| { |
| GetSystemTimeAsFileTime (&_now.ft); |
| tp->tv_usec=(long)((_now.ns100 / 10ULL) % 1000000ULL ); |
| tp->tv_sec= (long)((_now.ns100 - _W32_FT_OFFSET) / 10000000ULL); |
| } |
| /* Always return 0 as per Open Group Base Specifications Issue 6. |
| Do not set errno on error. */ |
| return 0; |
| } |
| #endif /* _WIN32 */ |
| |
| |
| #ifdef _WIN32 |
| #ifndef CONFIG_ANDROID |
| void socket_set_nonblock(int fd) |
| { |
| unsigned long opt = 1; |
| ioctlsocket(fd, FIONBIO, &opt); |
| } |
| #endif |
| |
| int inet_aton(const char *cp, struct in_addr *ia) |
| { |
| uint32_t addr = inet_addr(cp); |
| if (addr == 0xffffffff) |
| return 0; |
| ia->s_addr = addr; |
| return 1; |
| } |
| |
| void qemu_set_cloexec(int fd) |
| { |
| } |
| |
| #else |
| |
| #ifndef CONFIG_ANDROID |
| void socket_set_nonblock(int fd) |
| { |
| int f; |
| f = fcntl(fd, F_GETFL); |
| fcntl(fd, F_SETFL, f | O_NONBLOCK); |
| } |
| #endif |
| |
| void qemu_set_cloexec(int fd) |
| { |
| int f; |
| f = fcntl(fd, F_GETFD); |
| fcntl(fd, F_SETFD, f | FD_CLOEXEC); |
| } |
| |
| #endif |
| |
| /* |
| * Opens a file with FD_CLOEXEC set |
| */ |
| int qemu_open(const char *name, int flags, ...) |
| { |
| int ret; |
| int mode = 0; |
| |
| if (flags & O_CREAT) { |
| va_list ap; |
| |
| va_start(ap, flags); |
| mode = va_arg(ap, int); |
| va_end(ap); |
| } |
| |
| #ifdef O_CLOEXEC |
| ret = open(name, flags | O_CLOEXEC, mode); |
| #else |
| ret = open(name, flags, mode); |
| if (ret >= 0) { |
| qemu_set_cloexec(ret); |
| } |
| #endif |
| |
| return ret; |
| } |
| |
| /* |
| * A variant of write(2) which handles partial write. |
| * |
| * Return the number of bytes transferred. |
| * Set errno if fewer than `count' bytes are written. |
| * |
| * This function don't work with non-blocking fd's. |
| * Any of the possibilities with non-bloking fd's is bad: |
| * - return a short write (then name is wrong) |
| * - busy wait adding (errno == EAGAIN) to the loop |
| */ |
| ssize_t qemu_write_full(int fd, const void *buf, size_t count) |
| { |
| ssize_t ret = 0; |
| ssize_t total = 0; |
| |
| while (count) { |
| ret = write(fd, buf, count); |
| if (ret < 0) { |
| if (errno == EINTR) |
| continue; |
| break; |
| } |
| |
| count -= ret; |
| buf += ret; |
| total += ret; |
| } |
| |
| return total; |
| } |
| |
| #ifndef _WIN32 |
| /* |
| * Creates an eventfd that looks like a pipe and has EFD_CLOEXEC set. |
| */ |
| int qemu_eventfd(int fds[2]) |
| { |
| #ifdef CONFIG_EVENTFD |
| int ret; |
| |
| ret = eventfd(0, 0); |
| if (ret >= 0) { |
| fds[0] = ret; |
| qemu_set_cloexec(ret); |
| if ((fds[1] = dup(ret)) == -1) { |
| close(ret); |
| return -1; |
| } |
| qemu_set_cloexec(fds[1]); |
| return 0; |
| } |
| |
| if (errno != ENOSYS) { |
| return -1; |
| } |
| #endif |
| |
| return qemu_pipe(fds); |
| } |
| |
| /* |
| * Creates a pipe with FD_CLOEXEC set on both file descriptors |
| */ |
| int qemu_pipe(int pipefd[2]) |
| { |
| int ret; |
| |
| #ifdef CONFIG_PIPE2 |
| ret = pipe2(pipefd, O_CLOEXEC); |
| if (ret != -1 || errno != ENOSYS) { |
| return ret; |
| } |
| #endif |
| ret = pipe(pipefd); |
| if (ret == 0) { |
| qemu_set_cloexec(pipefd[0]); |
| qemu_set_cloexec(pipefd[1]); |
| } |
| |
| return ret; |
| } |
| #endif |
| |
| /* |
| * Opens a socket with FD_CLOEXEC set |
| */ |
| int qemu_socket(int domain, int type, int protocol) |
| { |
| int ret; |
| |
| #ifdef SOCK_CLOEXEC |
| ret = socket(domain, type | SOCK_CLOEXEC, protocol); |
| if (ret != -1 || errno != EINVAL) { |
| return ret; |
| } |
| #endif |
| ret = socket(domain, type, protocol); |
| if (ret >= 0) { |
| qemu_set_cloexec(ret); |
| } |
| |
| return ret; |
| } |
| |
| /* |
| * Accept a connection and set FD_CLOEXEC |
| */ |
| int qemu_accept(int s, struct sockaddr *addr, socklen_t *addrlen) |
| { |
| int ret; |
| |
| #ifdef CONFIG_ACCEPT4 |
| ret = accept4(s, addr, addrlen, SOCK_CLOEXEC); |
| if (ret != -1 || errno != ENOSYS) { |
| return ret; |
| } |
| #endif |
| ret = accept(s, addr, addrlen); |
| if (ret >= 0) { |
| qemu_set_cloexec(ret); |
| } |
| |
| return ret; |
| } |
| |
| #ifdef WIN32 |
| int asprintf( char **, char *, ... ); |
| int vasprintf( char **, char *, va_list ); |
| |
| int vasprintf( char **sptr, char *fmt, va_list argv ) |
| { |
| int wanted = vsnprintf( *sptr = NULL, 0, fmt, argv ); |
| if( (wanted > 0) && ((*sptr = malloc( 1 + wanted )) != NULL) ) |
| return vsprintf( *sptr, fmt, argv ); |
| |
| return wanted; |
| } |
| |
| int asprintf( char **sptr, char *fmt, ... ) |
| { |
| int retval; |
| va_list argv; |
| va_start( argv, fmt ); |
| retval = vasprintf( sptr, fmt, argv ); |
| va_end( argv ); |
| return retval; |
| } |
| #endif |