xref: /qemu/util/oslib-win32.c (revision 727385c4)
1 /*
2  * os-win32.c
3  *
4  * Copyright (c) 2003-2008 Fabrice Bellard
5  * Copyright (c) 2010-2016 Red Hat, Inc.
6  *
7  * QEMU library functions for win32 which are shared between QEMU and
8  * the QEMU tools.
9  *
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:
16  *
17  * The above copyright notice and this permission notice shall be included in
18  * all copies or substantial portions of the Software.
19  *
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.
27  *
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.
31  */
32 
33 #include "qemu/osdep.h"
34 #include <windows.h>
35 #include "qemu-common.h"
36 #include "qapi/error.h"
37 #include "qemu/main-loop.h"
38 #include "trace.h"
39 #include "qemu/sockets.h"
40 #include "qemu/cutils.h"
41 #include "qemu/error-report.h"
42 #include <malloc.h>
43 
44 /* this must come after including "trace.h" */
45 #include <shlobj.h>
46 
47 void *qemu_oom_check(void *ptr)
48 {
49     if (ptr == NULL) {
50         fprintf(stderr, "Failed to allocate memory: %lu\n", GetLastError());
51         abort();
52     }
53     return ptr;
54 }
55 
56 void *qemu_try_memalign(size_t alignment, size_t size)
57 {
58     void *ptr;
59 
60     g_assert(size != 0);
61     if (alignment < sizeof(void *)) {
62         alignment = sizeof(void *);
63     } else {
64         g_assert(is_power_of_2(alignment));
65     }
66     ptr = _aligned_malloc(size, alignment);
67     trace_qemu_memalign(alignment, size, ptr);
68     return ptr;
69 }
70 
71 void *qemu_memalign(size_t alignment, size_t size)
72 {
73     return qemu_oom_check(qemu_try_memalign(alignment, size));
74 }
75 
76 static int get_allocation_granularity(void)
77 {
78     SYSTEM_INFO system_info;
79 
80     GetSystemInfo(&system_info);
81     return system_info.dwAllocationGranularity;
82 }
83 
84 void *qemu_anon_ram_alloc(size_t size, uint64_t *align, bool shared,
85                           bool noreserve)
86 {
87     void *ptr;
88 
89     if (noreserve) {
90         /*
91          * We need a MEM_COMMIT before accessing any memory in a MEM_RESERVE
92          * area; we cannot easily mimic POSIX MAP_NORESERVE semantics.
93          */
94         error_report("Skipping reservation of swap space is not supported.");
95         return NULL;
96     }
97 
98     ptr = VirtualAlloc(NULL, size, MEM_COMMIT, PAGE_READWRITE);
99     trace_qemu_anon_ram_alloc(size, ptr);
100 
101     if (ptr && align) {
102         *align = MAX(get_allocation_granularity(), getpagesize());
103     }
104     return ptr;
105 }
106 
107 void qemu_vfree(void *ptr)
108 {
109     trace_qemu_vfree(ptr);
110     _aligned_free(ptr);
111 }
112 
113 void qemu_anon_ram_free(void *ptr, size_t size)
114 {
115     trace_qemu_anon_ram_free(ptr, size);
116     if (ptr) {
117         VirtualFree(ptr, 0, MEM_RELEASE);
118     }
119 }
120 
121 #ifndef _POSIX_THREAD_SAFE_FUNCTIONS
122 /* FIXME: add proper locking */
123 struct tm *gmtime_r(const time_t *timep, struct tm *result)
124 {
125     struct tm *p = gmtime(timep);
126     memset(result, 0, sizeof(*result));
127     if (p) {
128         *result = *p;
129         p = result;
130     }
131     return p;
132 }
133 
134 /* FIXME: add proper locking */
135 struct tm *localtime_r(const time_t *timep, struct tm *result)
136 {
137     struct tm *p = localtime(timep);
138     memset(result, 0, sizeof(*result));
139     if (p) {
140         *result = *p;
141         p = result;
142     }
143     return p;
144 }
145 #endif /* _POSIX_THREAD_SAFE_FUNCTIONS */
146 
147 static int socket_error(void)
148 {
149     switch (WSAGetLastError()) {
150     case 0:
151         return 0;
152     case WSAEINTR:
153         return EINTR;
154     case WSAEINVAL:
155         return EINVAL;
156     case WSA_INVALID_HANDLE:
157         return EBADF;
158     case WSA_NOT_ENOUGH_MEMORY:
159         return ENOMEM;
160     case WSA_INVALID_PARAMETER:
161         return EINVAL;
162     case WSAENAMETOOLONG:
163         return ENAMETOOLONG;
164     case WSAENOTEMPTY:
165         return ENOTEMPTY;
166     case WSAEWOULDBLOCK:
167          /* not using EWOULDBLOCK as we don't want code to have
168           * to check both EWOULDBLOCK and EAGAIN */
169         return EAGAIN;
170     case WSAEINPROGRESS:
171         return EINPROGRESS;
172     case WSAEALREADY:
173         return EALREADY;
174     case WSAENOTSOCK:
175         return ENOTSOCK;
176     case WSAEDESTADDRREQ:
177         return EDESTADDRREQ;
178     case WSAEMSGSIZE:
179         return EMSGSIZE;
180     case WSAEPROTOTYPE:
181         return EPROTOTYPE;
182     case WSAENOPROTOOPT:
183         return ENOPROTOOPT;
184     case WSAEPROTONOSUPPORT:
185         return EPROTONOSUPPORT;
186     case WSAEOPNOTSUPP:
187         return EOPNOTSUPP;
188     case WSAEAFNOSUPPORT:
189         return EAFNOSUPPORT;
190     case WSAEADDRINUSE:
191         return EADDRINUSE;
192     case WSAEADDRNOTAVAIL:
193         return EADDRNOTAVAIL;
194     case WSAENETDOWN:
195         return ENETDOWN;
196     case WSAENETUNREACH:
197         return ENETUNREACH;
198     case WSAENETRESET:
199         return ENETRESET;
200     case WSAECONNABORTED:
201         return ECONNABORTED;
202     case WSAECONNRESET:
203         return ECONNRESET;
204     case WSAENOBUFS:
205         return ENOBUFS;
206     case WSAEISCONN:
207         return EISCONN;
208     case WSAENOTCONN:
209         return ENOTCONN;
210     case WSAETIMEDOUT:
211         return ETIMEDOUT;
212     case WSAECONNREFUSED:
213         return ECONNREFUSED;
214     case WSAELOOP:
215         return ELOOP;
216     case WSAEHOSTUNREACH:
217         return EHOSTUNREACH;
218     default:
219         return EIO;
220     }
221 }
222 
223 void qemu_set_block(int fd)
224 {
225     unsigned long opt = 0;
226     WSAEventSelect(fd, NULL, 0);
227     ioctlsocket(fd, FIONBIO, &opt);
228 }
229 
230 int qemu_try_set_nonblock(int fd)
231 {
232     unsigned long opt = 1;
233     if (ioctlsocket(fd, FIONBIO, &opt) != NO_ERROR) {
234         return -socket_error();
235     }
236     return 0;
237 }
238 
239 void qemu_set_nonblock(int fd)
240 {
241     (void)qemu_try_set_nonblock(fd);
242 }
243 
244 int socket_set_fast_reuse(int fd)
245 {
246     /* Enabling the reuse of an endpoint that was used by a socket still in
247      * TIME_WAIT state is usually performed by setting SO_REUSEADDR. On Windows
248      * fast reuse is the default and SO_REUSEADDR does strange things. So we
249      * don't have to do anything here. More info can be found at:
250      * http://msdn.microsoft.com/en-us/library/windows/desktop/ms740621.aspx */
251     return 0;
252 }
253 
254 int inet_aton(const char *cp, struct in_addr *ia)
255 {
256     uint32_t addr = inet_addr(cp);
257     if (addr == 0xffffffff) {
258         return 0;
259     }
260     ia->s_addr = addr;
261     return 1;
262 }
263 
264 void qemu_set_cloexec(int fd)
265 {
266 }
267 
268 /* Offset between 1/1/1601 and 1/1/1970 in 100 nanosec units */
269 #define _W32_FT_OFFSET (116444736000000000ULL)
270 
271 int qemu_gettimeofday(qemu_timeval *tp)
272 {
273   union {
274     unsigned long long ns100; /*time since 1 Jan 1601 in 100ns units */
275     FILETIME ft;
276   }  _now;
277 
278   if(tp) {
279       GetSystemTimeAsFileTime (&_now.ft);
280       tp->tv_usec=(long)((_now.ns100 / 10ULL) % 1000000ULL );
281       tp->tv_sec= (long)((_now.ns100 - _W32_FT_OFFSET) / 10000000ULL);
282   }
283   /* Always return 0 as per Open Group Base Specifications Issue 6.
284      Do not set errno on error.  */
285   return 0;
286 }
287 
288 int qemu_get_thread_id(void)
289 {
290     return GetCurrentThreadId();
291 }
292 
293 char *
294 qemu_get_local_state_pathname(const char *relative_pathname)
295 {
296     HRESULT result;
297     char base_path[MAX_PATH+1] = "";
298 
299     result = SHGetFolderPath(NULL, CSIDL_COMMON_APPDATA, NULL,
300                              /* SHGFP_TYPE_CURRENT */ 0, base_path);
301     if (result != S_OK) {
302         /* misconfigured environment */
303         g_critical("CSIDL_COMMON_APPDATA unavailable: %ld", (long)result);
304         abort();
305     }
306     return g_strdup_printf("%s" G_DIR_SEPARATOR_S "%s", base_path,
307                            relative_pathname);
308 }
309 
310 void qemu_set_tty_echo(int fd, bool echo)
311 {
312     HANDLE handle = (HANDLE)_get_osfhandle(fd);
313     DWORD dwMode = 0;
314 
315     if (handle == INVALID_HANDLE_VALUE) {
316         return;
317     }
318 
319     GetConsoleMode(handle, &dwMode);
320 
321     if (echo) {
322         SetConsoleMode(handle, dwMode | ENABLE_ECHO_INPUT | ENABLE_LINE_INPUT);
323     } else {
324         SetConsoleMode(handle,
325                        dwMode & ~(ENABLE_ECHO_INPUT | ENABLE_LINE_INPUT));
326     }
327 }
328 
329 static const char *exec_dir;
330 
331 void qemu_init_exec_dir(const char *argv0)
332 {
333 
334     char *p;
335     char buf[MAX_PATH];
336     DWORD len;
337 
338     if (exec_dir) {
339         return;
340     }
341 
342     len = GetModuleFileName(NULL, buf, sizeof(buf) - 1);
343     if (len == 0) {
344         return;
345     }
346 
347     buf[len] = 0;
348     p = buf + len - 1;
349     while (p != buf && *p != '\\') {
350         p--;
351     }
352     *p = 0;
353     if (access(buf, R_OK) == 0) {
354         exec_dir = g_strdup(buf);
355     } else {
356         exec_dir = CONFIG_BINDIR;
357     }
358 }
359 
360 const char *qemu_get_exec_dir(void)
361 {
362     return exec_dir;
363 }
364 
365 int getpagesize(void)
366 {
367     SYSTEM_INFO system_info;
368 
369     GetSystemInfo(&system_info);
370     return system_info.dwPageSize;
371 }
372 
373 void os_mem_prealloc(int fd, char *area, size_t memory, int smp_cpus,
374                      Error **errp)
375 {
376     int i;
377     size_t pagesize = qemu_real_host_page_size;
378 
379     memory = (memory + pagesize - 1) & -pagesize;
380     for (i = 0; i < memory / pagesize; i++) {
381         memset(area + pagesize * i, 0, 1);
382     }
383 }
384 
385 char *qemu_get_pid_name(pid_t pid)
386 {
387     /* XXX Implement me */
388     abort();
389 }
390 
391 
392 pid_t qemu_fork(Error **errp)
393 {
394     errno = ENOSYS;
395     error_setg_errno(errp, errno,
396                      "cannot fork child process");
397     return -1;
398 }
399 
400 
401 #undef connect
402 int qemu_connect_wrap(int sockfd, const struct sockaddr *addr,
403                       socklen_t addrlen)
404 {
405     int ret;
406     ret = connect(sockfd, addr, addrlen);
407     if (ret < 0) {
408         if (WSAGetLastError() == WSAEWOULDBLOCK) {
409             errno = EINPROGRESS;
410         } else {
411             errno = socket_error();
412         }
413     }
414     return ret;
415 }
416 
417 
418 #undef listen
419 int qemu_listen_wrap(int sockfd, int backlog)
420 {
421     int ret;
422     ret = listen(sockfd, backlog);
423     if (ret < 0) {
424         errno = socket_error();
425     }
426     return ret;
427 }
428 
429 
430 #undef bind
431 int qemu_bind_wrap(int sockfd, const struct sockaddr *addr,
432                    socklen_t addrlen)
433 {
434     int ret;
435     ret = bind(sockfd, addr, addrlen);
436     if (ret < 0) {
437         errno = socket_error();
438     }
439     return ret;
440 }
441 
442 
443 #undef socket
444 int qemu_socket_wrap(int domain, int type, int protocol)
445 {
446     int ret;
447     ret = socket(domain, type, protocol);
448     if (ret < 0) {
449         errno = socket_error();
450     }
451     return ret;
452 }
453 
454 
455 #undef accept
456 int qemu_accept_wrap(int sockfd, struct sockaddr *addr,
457                      socklen_t *addrlen)
458 {
459     int ret;
460     ret = accept(sockfd, addr, addrlen);
461     if (ret < 0) {
462         errno = socket_error();
463     }
464     return ret;
465 }
466 
467 
468 #undef shutdown
469 int qemu_shutdown_wrap(int sockfd, int how)
470 {
471     int ret;
472     ret = shutdown(sockfd, how);
473     if (ret < 0) {
474         errno = socket_error();
475     }
476     return ret;
477 }
478 
479 
480 #undef ioctlsocket
481 int qemu_ioctlsocket_wrap(int fd, int req, void *val)
482 {
483     int ret;
484     ret = ioctlsocket(fd, req, val);
485     if (ret < 0) {
486         errno = socket_error();
487     }
488     return ret;
489 }
490 
491 
492 #undef closesocket
493 int qemu_closesocket_wrap(int fd)
494 {
495     int ret;
496     ret = closesocket(fd);
497     if (ret < 0) {
498         errno = socket_error();
499     }
500     return ret;
501 }
502 
503 
504 #undef getsockopt
505 int qemu_getsockopt_wrap(int sockfd, int level, int optname,
506                          void *optval, socklen_t *optlen)
507 {
508     int ret;
509     ret = getsockopt(sockfd, level, optname, optval, optlen);
510     if (ret < 0) {
511         errno = socket_error();
512     }
513     return ret;
514 }
515 
516 
517 #undef setsockopt
518 int qemu_setsockopt_wrap(int sockfd, int level, int optname,
519                          const void *optval, socklen_t optlen)
520 {
521     int ret;
522     ret = setsockopt(sockfd, level, optname, optval, optlen);
523     if (ret < 0) {
524         errno = socket_error();
525     }
526     return ret;
527 }
528 
529 
530 #undef getpeername
531 int qemu_getpeername_wrap(int sockfd, struct sockaddr *addr,
532                           socklen_t *addrlen)
533 {
534     int ret;
535     ret = getpeername(sockfd, addr, addrlen);
536     if (ret < 0) {
537         errno = socket_error();
538     }
539     return ret;
540 }
541 
542 
543 #undef getsockname
544 int qemu_getsockname_wrap(int sockfd, struct sockaddr *addr,
545                           socklen_t *addrlen)
546 {
547     int ret;
548     ret = getsockname(sockfd, addr, addrlen);
549     if (ret < 0) {
550         errno = socket_error();
551     }
552     return ret;
553 }
554 
555 
556 #undef send
557 ssize_t qemu_send_wrap(int sockfd, const void *buf, size_t len, int flags)
558 {
559     int ret;
560     ret = send(sockfd, buf, len, flags);
561     if (ret < 0) {
562         errno = socket_error();
563     }
564     return ret;
565 }
566 
567 
568 #undef sendto
569 ssize_t qemu_sendto_wrap(int sockfd, const void *buf, size_t len, int flags,
570                          const struct sockaddr *addr, socklen_t addrlen)
571 {
572     int ret;
573     ret = sendto(sockfd, buf, len, flags, addr, addrlen);
574     if (ret < 0) {
575         errno = socket_error();
576     }
577     return ret;
578 }
579 
580 
581 #undef recv
582 ssize_t qemu_recv_wrap(int sockfd, void *buf, size_t len, int flags)
583 {
584     int ret;
585     ret = recv(sockfd, buf, len, flags);
586     if (ret < 0) {
587         errno = socket_error();
588     }
589     return ret;
590 }
591 
592 
593 #undef recvfrom
594 ssize_t qemu_recvfrom_wrap(int sockfd, void *buf, size_t len, int flags,
595                            struct sockaddr *addr, socklen_t *addrlen)
596 {
597     int ret;
598     ret = recvfrom(sockfd, buf, len, flags, addr, addrlen);
599     if (ret < 0) {
600         errno = socket_error();
601     }
602     return ret;
603 }
604 
605 bool qemu_write_pidfile(const char *filename, Error **errp)
606 {
607     char buffer[128];
608     int len;
609     HANDLE file;
610     OVERLAPPED overlap;
611     BOOL ret;
612     memset(&overlap, 0, sizeof(overlap));
613 
614     file = CreateFile(filename, GENERIC_WRITE, FILE_SHARE_READ, NULL,
615                       OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
616 
617     if (file == INVALID_HANDLE_VALUE) {
618         error_setg(errp, "Failed to create PID file");
619         return false;
620     }
621     len = snprintf(buffer, sizeof(buffer), FMT_pid "\n", (pid_t)getpid());
622     ret = WriteFile(file, (LPCVOID)buffer, (DWORD)len,
623                     NULL, &overlap);
624     CloseHandle(file);
625     if (ret == 0) {
626         error_setg(errp, "Failed to write PID file");
627         return false;
628     }
629     return true;
630 }
631 
632 char *qemu_get_host_name(Error **errp)
633 {
634     wchar_t tmp[MAX_COMPUTERNAME_LENGTH + 1];
635     DWORD size = G_N_ELEMENTS(tmp);
636 
637     if (GetComputerNameW(tmp, &size) == 0) {
638         error_setg_win32(errp, GetLastError(), "failed close handle");
639         return NULL;
640     }
641 
642     return g_utf16_to_utf8(tmp, size, NULL, NULL, NULL);
643 }
644 
645 size_t qemu_get_host_physmem(void)
646 {
647     MEMORYSTATUSEX statex;
648     statex.dwLength = sizeof(statex);
649 
650     if (GlobalMemoryStatusEx(&statex)) {
651         return statex.ullTotalPhys;
652     }
653     return 0;
654 }
655