migration/rdma: Plug memory leaks in qemu_rdma_registration_stop()
[qemu/armbru.git] / util / oslib-win32.c
blobe9b14ab17847edcbe237c41f81914a120d506288
1 /*
2 * os-win32.c
4 * Copyright (c) 2003-2008 Fabrice Bellard
5 * Copyright (c) 2010-2016 Red Hat, Inc.
7 * QEMU library functions for win32 which are shared between QEMU and
8 * the QEMU tools.
10 * Permission is hereby granted, free of charge, to any person obtaining a copy
11 * of this software and associated documentation files (the "Software"), to deal
12 * in the Software without restriction, including without limitation the rights
13 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
14 * copies of the Software, and to permit persons to whom the Software is
15 * furnished to do so, subject to the following conditions:
17 * The above copyright notice and this permission notice shall be included in
18 * all copies or substantial portions of the Software.
20 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
21 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
22 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
23 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
24 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
25 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
26 * THE SOFTWARE.
28 * The implementation of g_poll (functions poll_rest, g_poll) at the end of
29 * this file are based on code from GNOME glib-2 and use a different license,
30 * see the license comment there.
33 #include "qemu/osdep.h"
34 #include <windows.h>
35 #include "qemu-common.h"
36 #include "qapi/error.h"
37 #include "sysemu/sysemu.h"
38 #include "qemu/main-loop.h"
39 #include "trace.h"
40 #include "qemu/sockets.h"
41 #include "qemu/cutils.h"
43 /* this must come after including "trace.h" */
44 #include <shlobj.h>
46 void *qemu_oom_check(void *ptr)
48 if (ptr == NULL) {
49 fprintf(stderr, "Failed to allocate memory: %lu\n", GetLastError());
50 abort();
52 return ptr;
55 void *qemu_try_memalign(size_t alignment, size_t size)
57 void *ptr;
59 if (!size) {
60 abort();
62 ptr = VirtualAlloc(NULL, size, MEM_COMMIT, PAGE_READWRITE);
63 trace_qemu_memalign(alignment, size, ptr);
64 return ptr;
67 void *qemu_memalign(size_t alignment, size_t size)
69 return qemu_oom_check(qemu_try_memalign(alignment, size));
72 static int get_allocation_granularity(void)
74 SYSTEM_INFO system_info;
76 GetSystemInfo(&system_info);
77 return system_info.dwAllocationGranularity;
80 void *qemu_anon_ram_alloc(size_t size, uint64_t *align, bool shared)
82 void *ptr;
84 ptr = VirtualAlloc(NULL, size, MEM_COMMIT, PAGE_READWRITE);
85 trace_qemu_anon_ram_alloc(size, ptr);
87 if (ptr && align) {
88 *align = MAX(get_allocation_granularity(), getpagesize());
90 return ptr;
93 void qemu_vfree(void *ptr)
95 trace_qemu_vfree(ptr);
96 if (ptr) {
97 VirtualFree(ptr, 0, MEM_RELEASE);
101 void qemu_anon_ram_free(void *ptr, size_t size)
103 trace_qemu_anon_ram_free(ptr, size);
104 if (ptr) {
105 VirtualFree(ptr, 0, MEM_RELEASE);
109 #ifndef CONFIG_LOCALTIME_R
110 /* FIXME: add proper locking */
111 struct tm *gmtime_r(const time_t *timep, struct tm *result)
113 struct tm *p = gmtime(timep);
114 memset(result, 0, sizeof(*result));
115 if (p) {
116 *result = *p;
117 p = result;
119 return p;
122 /* FIXME: add proper locking */
123 struct tm *localtime_r(const time_t *timep, struct tm *result)
125 struct tm *p = localtime(timep);
126 memset(result, 0, sizeof(*result));
127 if (p) {
128 *result = *p;
129 p = result;
131 return p;
133 #endif /* CONFIG_LOCALTIME_R */
135 void qemu_set_block(int fd)
137 unsigned long opt = 0;
138 WSAEventSelect(fd, NULL, 0);
139 ioctlsocket(fd, FIONBIO, &opt);
142 void qemu_set_nonblock(int fd)
144 unsigned long opt = 1;
145 ioctlsocket(fd, FIONBIO, &opt);
146 qemu_fd_register(fd);
149 int socket_set_fast_reuse(int fd)
151 /* Enabling the reuse of an endpoint that was used by a socket still in
152 * TIME_WAIT state is usually performed by setting SO_REUSEADDR. On Windows
153 * fast reuse is the default and SO_REUSEADDR does strange things. So we
154 * don't have to do anything here. More info can be found at:
155 * http://msdn.microsoft.com/en-us/library/windows/desktop/ms740621.aspx */
156 return 0;
160 static int socket_error(void)
162 switch (WSAGetLastError()) {
163 case 0:
164 return 0;
165 case WSAEINTR:
166 return EINTR;
167 case WSAEINVAL:
168 return EINVAL;
169 case WSA_INVALID_HANDLE:
170 return EBADF;
171 case WSA_NOT_ENOUGH_MEMORY:
172 return ENOMEM;
173 case WSA_INVALID_PARAMETER:
174 return EINVAL;
175 case WSAENAMETOOLONG:
176 return ENAMETOOLONG;
177 case WSAENOTEMPTY:
178 return ENOTEMPTY;
179 case WSAEWOULDBLOCK:
180 /* not using EWOULDBLOCK as we don't want code to have
181 * to check both EWOULDBLOCK and EAGAIN */
182 return EAGAIN;
183 case WSAEINPROGRESS:
184 return EINPROGRESS;
185 case WSAEALREADY:
186 return EALREADY;
187 case WSAENOTSOCK:
188 return ENOTSOCK;
189 case WSAEDESTADDRREQ:
190 return EDESTADDRREQ;
191 case WSAEMSGSIZE:
192 return EMSGSIZE;
193 case WSAEPROTOTYPE:
194 return EPROTOTYPE;
195 case WSAENOPROTOOPT:
196 return ENOPROTOOPT;
197 case WSAEPROTONOSUPPORT:
198 return EPROTONOSUPPORT;
199 case WSAEOPNOTSUPP:
200 return EOPNOTSUPP;
201 case WSAEAFNOSUPPORT:
202 return EAFNOSUPPORT;
203 case WSAEADDRINUSE:
204 return EADDRINUSE;
205 case WSAEADDRNOTAVAIL:
206 return EADDRNOTAVAIL;
207 case WSAENETDOWN:
208 return ENETDOWN;
209 case WSAENETUNREACH:
210 return ENETUNREACH;
211 case WSAENETRESET:
212 return ENETRESET;
213 case WSAECONNABORTED:
214 return ECONNABORTED;
215 case WSAECONNRESET:
216 return ECONNRESET;
217 case WSAENOBUFS:
218 return ENOBUFS;
219 case WSAEISCONN:
220 return EISCONN;
221 case WSAENOTCONN:
222 return ENOTCONN;
223 case WSAETIMEDOUT:
224 return ETIMEDOUT;
225 case WSAECONNREFUSED:
226 return ECONNREFUSED;
227 case WSAELOOP:
228 return ELOOP;
229 case WSAEHOSTUNREACH:
230 return EHOSTUNREACH;
231 default:
232 return EIO;
236 int inet_aton(const char *cp, struct in_addr *ia)
238 uint32_t addr = inet_addr(cp);
239 if (addr == 0xffffffff) {
240 return 0;
242 ia->s_addr = addr;
243 return 1;
246 void qemu_set_cloexec(int fd)
250 /* Offset between 1/1/1601 and 1/1/1970 in 100 nanosec units */
251 #define _W32_FT_OFFSET (116444736000000000ULL)
253 int qemu_gettimeofday(qemu_timeval *tp)
255 union {
256 unsigned long long ns100; /*time since 1 Jan 1601 in 100ns units */
257 FILETIME ft;
258 } _now;
260 if(tp) {
261 GetSystemTimeAsFileTime (&_now.ft);
262 tp->tv_usec=(long)((_now.ns100 / 10ULL) % 1000000ULL );
263 tp->tv_sec= (long)((_now.ns100 - _W32_FT_OFFSET) / 10000000ULL);
265 /* Always return 0 as per Open Group Base Specifications Issue 6.
266 Do not set errno on error. */
267 return 0;
270 int qemu_get_thread_id(void)
272 return GetCurrentThreadId();
275 char *
276 qemu_get_local_state_pathname(const char *relative_pathname)
278 HRESULT result;
279 char base_path[MAX_PATH+1] = "";
281 result = SHGetFolderPath(NULL, CSIDL_COMMON_APPDATA, NULL,
282 /* SHGFP_TYPE_CURRENT */ 0, base_path);
283 if (result != S_OK) {
284 /* misconfigured environment */
285 g_critical("CSIDL_COMMON_APPDATA unavailable: %ld", (long)result);
286 abort();
288 return g_strdup_printf("%s" G_DIR_SEPARATOR_S "%s", base_path,
289 relative_pathname);
292 void qemu_set_tty_echo(int fd, bool echo)
294 HANDLE handle = (HANDLE)_get_osfhandle(fd);
295 DWORD dwMode = 0;
297 if (handle == INVALID_HANDLE_VALUE) {
298 return;
301 GetConsoleMode(handle, &dwMode);
303 if (echo) {
304 SetConsoleMode(handle, dwMode | ENABLE_ECHO_INPUT | ENABLE_LINE_INPUT);
305 } else {
306 SetConsoleMode(handle,
307 dwMode & ~(ENABLE_ECHO_INPUT | ENABLE_LINE_INPUT));
311 static char exec_dir[PATH_MAX];
313 void qemu_init_exec_dir(const char *argv0)
316 char *p;
317 char buf[MAX_PATH];
318 DWORD len;
320 len = GetModuleFileName(NULL, buf, sizeof(buf) - 1);
321 if (len == 0) {
322 return;
325 buf[len] = 0;
326 p = buf + len - 1;
327 while (p != buf && *p != '\\') {
328 p--;
330 *p = 0;
331 if (access(buf, R_OK) == 0) {
332 pstrcpy(exec_dir, sizeof(exec_dir), buf);
336 char *qemu_get_exec_dir(void)
338 return g_strdup(exec_dir);
341 #if !GLIB_CHECK_VERSION(2, 50, 0)
343 * The original implementation of g_poll from glib has a problem on Windows
344 * when using timeouts < 10 ms.
346 * Whenever g_poll is called with timeout < 10 ms, it does a quick poll instead
347 * of wait. This causes significant performance degradation of QEMU.
349 * The following code is a copy of the original code from glib/gpoll.c
350 * (glib commit 20f4d1820b8d4d0fc4447188e33efffd6d4a88d8 from 2014-02-19).
351 * Some debug code was removed and the code was reformatted.
352 * All other code modifications are marked with 'QEMU'.
356 * gpoll.c: poll(2) abstraction
357 * Copyright 1998 Owen Taylor
358 * Copyright 2008 Red Hat, Inc.
360 * This library is free software; you can redistribute it and/or
361 * modify it under the terms of the GNU Lesser General Public
362 * License as published by the Free Software Foundation; either
363 * version 2 of the License, or (at your option) any later version.
365 * This library is distributed in the hope that it will be useful,
366 * but WITHOUT ANY WARRANTY; without even the implied warranty of
367 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
368 * Lesser General Public License for more details.
370 * You should have received a copy of the GNU Lesser General Public
371 * License along with this library; if not, see <http://www.gnu.org/licenses/>.
374 static int poll_rest(gboolean poll_msgs, HANDLE *handles, gint nhandles,
375 GPollFD *fds, guint nfds, gint timeout)
377 DWORD ready;
378 GPollFD *f;
379 int recursed_result;
381 if (poll_msgs) {
382 /* Wait for either messages or handles
383 * -> Use MsgWaitForMultipleObjectsEx
385 ready = MsgWaitForMultipleObjectsEx(nhandles, handles, timeout,
386 QS_ALLINPUT, MWMO_ALERTABLE);
388 if (ready == WAIT_FAILED) {
389 gchar *emsg = g_win32_error_message(GetLastError());
390 g_warning("MsgWaitForMultipleObjectsEx failed: %s", emsg);
391 g_free(emsg);
393 } else if (nhandles == 0) {
394 /* No handles to wait for, just the timeout */
395 if (timeout == INFINITE) {
396 ready = WAIT_FAILED;
397 } else {
398 SleepEx(timeout, TRUE);
399 ready = WAIT_TIMEOUT;
401 } else {
402 /* Wait for just handles
403 * -> Use WaitForMultipleObjectsEx
405 ready =
406 WaitForMultipleObjectsEx(nhandles, handles, FALSE, timeout, TRUE);
407 if (ready == WAIT_FAILED) {
408 gchar *emsg = g_win32_error_message(GetLastError());
409 g_warning("WaitForMultipleObjectsEx failed: %s", emsg);
410 g_free(emsg);
414 if (ready == WAIT_FAILED) {
415 return -1;
416 } else if (ready == WAIT_TIMEOUT || ready == WAIT_IO_COMPLETION) {
417 return 0;
418 } else if (poll_msgs && ready == WAIT_OBJECT_0 + nhandles) {
419 for (f = fds; f < &fds[nfds]; ++f) {
420 if (f->fd == G_WIN32_MSG_HANDLE && f->events & G_IO_IN) {
421 f->revents |= G_IO_IN;
425 /* If we have a timeout, or no handles to poll, be satisfied
426 * with just noticing we have messages waiting.
428 if (timeout != 0 || nhandles == 0) {
429 return 1;
432 /* If no timeout and handles to poll, recurse to poll them,
433 * too.
435 recursed_result = poll_rest(FALSE, handles, nhandles, fds, nfds, 0);
436 return (recursed_result == -1) ? -1 : 1 + recursed_result;
437 } else if (/* QEMU: removed the following unneeded statement which causes
438 * a compiler warning: ready >= WAIT_OBJECT_0 && */
439 ready < WAIT_OBJECT_0 + nhandles) {
440 for (f = fds; f < &fds[nfds]; ++f) {
441 if ((HANDLE) f->fd == handles[ready - WAIT_OBJECT_0]) {
442 f->revents = f->events;
446 /* If no timeout and polling several handles, recurse to poll
447 * the rest of them.
449 if (timeout == 0 && nhandles > 1) {
450 /* Remove the handle that fired */
451 int i;
452 for (i = ready - WAIT_OBJECT_0 + 1; i < nhandles; i++) {
453 handles[i-1] = handles[i];
455 nhandles--;
456 recursed_result = poll_rest(FALSE, handles, nhandles, fds, nfds, 0);
457 return (recursed_result == -1) ? -1 : 1 + recursed_result;
459 return 1;
462 return 0;
465 gint g_poll(GPollFD *fds, guint nfds, gint timeout)
467 HANDLE handles[MAXIMUM_WAIT_OBJECTS];
468 gboolean poll_msgs = FALSE;
469 GPollFD *f;
470 gint nhandles = 0;
471 int retval;
473 for (f = fds; f < &fds[nfds]; ++f) {
474 if (f->fd == G_WIN32_MSG_HANDLE && (f->events & G_IO_IN)) {
475 poll_msgs = TRUE;
476 } else if (f->fd > 0) {
477 /* Don't add the same handle several times into the array, as
478 * docs say that is not allowed, even if it actually does seem
479 * to work.
481 gint i;
483 for (i = 0; i < nhandles; i++) {
484 if (handles[i] == (HANDLE) f->fd) {
485 break;
489 if (i == nhandles) {
490 if (nhandles == MAXIMUM_WAIT_OBJECTS) {
491 g_warning("Too many handles to wait for!\n");
492 break;
493 } else {
494 handles[nhandles++] = (HANDLE) f->fd;
500 for (f = fds; f < &fds[nfds]; ++f) {
501 f->revents = 0;
504 if (timeout == -1) {
505 timeout = INFINITE;
508 /* Polling for several things? */
509 if (nhandles > 1 || (nhandles > 0 && poll_msgs)) {
510 /* First check if one or several of them are immediately
511 * available
513 retval = poll_rest(poll_msgs, handles, nhandles, fds, nfds, 0);
515 /* If not, and we have a significant timeout, poll again with
516 * timeout then. Note that this will return indication for only
517 * one event, or only for messages. We ignore timeouts less than
518 * ten milliseconds as they are mostly pointless on Windows, the
519 * MsgWaitForMultipleObjectsEx() call will timeout right away
520 * anyway.
522 * Modification for QEMU: replaced timeout >= 10 by timeout > 0.
524 if (retval == 0 && (timeout == INFINITE || timeout > 0)) {
525 retval = poll_rest(poll_msgs, handles, nhandles,
526 fds, nfds, timeout);
528 } else {
529 /* Just polling for one thing, so no need to check first if
530 * available immediately
532 retval = poll_rest(poll_msgs, handles, nhandles, fds, nfds, timeout);
535 if (retval == -1) {
536 for (f = fds; f < &fds[nfds]; ++f) {
537 f->revents = 0;
541 return retval;
543 #endif
545 int getpagesize(void)
547 SYSTEM_INFO system_info;
549 GetSystemInfo(&system_info);
550 return system_info.dwPageSize;
553 void os_mem_prealloc(int fd, char *area, size_t memory, int smp_cpus,
554 Error **errp)
556 int i;
557 size_t pagesize = qemu_real_host_page_size;
559 memory = (memory + pagesize - 1) & -pagesize;
560 for (i = 0; i < memory / pagesize; i++) {
561 memset(area + pagesize * i, 0, 1);
565 char *qemu_get_pid_name(pid_t pid)
567 /* XXX Implement me */
568 abort();
572 pid_t qemu_fork(Error **errp)
574 errno = ENOSYS;
575 error_setg_errno(errp, errno,
576 "cannot fork child process");
577 return -1;
581 #undef connect
582 int qemu_connect_wrap(int sockfd, const struct sockaddr *addr,
583 socklen_t addrlen)
585 int ret;
586 ret = connect(sockfd, addr, addrlen);
587 if (ret < 0) {
588 if (WSAGetLastError() == WSAEWOULDBLOCK) {
589 errno = EINPROGRESS;
590 } else {
591 errno = socket_error();
594 return ret;
598 #undef listen
599 int qemu_listen_wrap(int sockfd, int backlog)
601 int ret;
602 ret = listen(sockfd, backlog);
603 if (ret < 0) {
604 errno = socket_error();
606 return ret;
610 #undef bind
611 int qemu_bind_wrap(int sockfd, const struct sockaddr *addr,
612 socklen_t addrlen)
614 int ret;
615 ret = bind(sockfd, addr, addrlen);
616 if (ret < 0) {
617 errno = socket_error();
619 return ret;
623 #undef socket
624 int qemu_socket_wrap(int domain, int type, int protocol)
626 int ret;
627 ret = socket(domain, type, protocol);
628 if (ret < 0) {
629 errno = socket_error();
631 return ret;
635 #undef accept
636 int qemu_accept_wrap(int sockfd, struct sockaddr *addr,
637 socklen_t *addrlen)
639 int ret;
640 ret = accept(sockfd, addr, addrlen);
641 if (ret < 0) {
642 errno = socket_error();
644 return ret;
648 #undef shutdown
649 int qemu_shutdown_wrap(int sockfd, int how)
651 int ret;
652 ret = shutdown(sockfd, how);
653 if (ret < 0) {
654 errno = socket_error();
656 return ret;
660 #undef ioctlsocket
661 int qemu_ioctlsocket_wrap(int fd, int req, void *val)
663 int ret;
664 ret = ioctlsocket(fd, req, val);
665 if (ret < 0) {
666 errno = socket_error();
668 return ret;
672 #undef closesocket
673 int qemu_closesocket_wrap(int fd)
675 int ret;
676 ret = closesocket(fd);
677 if (ret < 0) {
678 errno = socket_error();
680 return ret;
684 #undef getsockopt
685 int qemu_getsockopt_wrap(int sockfd, int level, int optname,
686 void *optval, socklen_t *optlen)
688 int ret;
689 ret = getsockopt(sockfd, level, optname, optval, optlen);
690 if (ret < 0) {
691 errno = socket_error();
693 return ret;
697 #undef setsockopt
698 int qemu_setsockopt_wrap(int sockfd, int level, int optname,
699 const void *optval, socklen_t optlen)
701 int ret;
702 ret = setsockopt(sockfd, level, optname, optval, optlen);
703 if (ret < 0) {
704 errno = socket_error();
706 return ret;
710 #undef getpeername
711 int qemu_getpeername_wrap(int sockfd, struct sockaddr *addr,
712 socklen_t *addrlen)
714 int ret;
715 ret = getpeername(sockfd, addr, addrlen);
716 if (ret < 0) {
717 errno = socket_error();
719 return ret;
723 #undef getsockname
724 int qemu_getsockname_wrap(int sockfd, struct sockaddr *addr,
725 socklen_t *addrlen)
727 int ret;
728 ret = getsockname(sockfd, addr, addrlen);
729 if (ret < 0) {
730 errno = socket_error();
732 return ret;
736 #undef send
737 ssize_t qemu_send_wrap(int sockfd, const void *buf, size_t len, int flags)
739 int ret;
740 ret = send(sockfd, buf, len, flags);
741 if (ret < 0) {
742 errno = socket_error();
744 return ret;
748 #undef sendto
749 ssize_t qemu_sendto_wrap(int sockfd, const void *buf, size_t len, int flags,
750 const struct sockaddr *addr, socklen_t addrlen)
752 int ret;
753 ret = sendto(sockfd, buf, len, flags, addr, addrlen);
754 if (ret < 0) {
755 errno = socket_error();
757 return ret;
761 #undef recv
762 ssize_t qemu_recv_wrap(int sockfd, void *buf, size_t len, int flags)
764 int ret;
765 ret = recv(sockfd, buf, len, flags);
766 if (ret < 0) {
767 errno = socket_error();
769 return ret;
773 #undef recvfrom
774 ssize_t qemu_recvfrom_wrap(int sockfd, void *buf, size_t len, int flags,
775 struct sockaddr *addr, socklen_t *addrlen)
777 int ret;
778 ret = recvfrom(sockfd, buf, len, flags, addr, addrlen);
779 if (ret < 0) {
780 errno = socket_error();
782 return ret;
785 bool qemu_write_pidfile(const char *filename, Error **errp)
787 char buffer[128];
788 int len;
789 HANDLE file;
790 OVERLAPPED overlap;
791 BOOL ret;
792 memset(&overlap, 0, sizeof(overlap));
794 file = CreateFile(filename, GENERIC_WRITE, FILE_SHARE_READ, NULL,
795 OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
797 if (file == INVALID_HANDLE_VALUE) {
798 error_setg(errp, "Failed to create PID file");
799 return false;
801 len = snprintf(buffer, sizeof(buffer), FMT_pid "\n", (pid_t)getpid());
802 ret = WriteFile(file, (LPCVOID)buffer, (DWORD)len,
803 NULL, &overlap);
804 CloseHandle(file);
805 if (ret == 0) {
806 error_setg(errp, "Failed to write PID file");
807 return false;
809 return true;