util/oslib-win32.c

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