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 29 #include "qemu/osdep.h" 30 #include <windows.h> 31 #include "qapi/error.h" 32 #include "qemu/main-loop.h" 33 #include "trace.h" 34 #include "qemu/sockets.h" 35 #include "qemu/cutils.h" 36 #include "qemu/error-report.h" 37 #include <malloc.h> 38 39 static int get_allocation_granularity(void) 40 { 41 SYSTEM_INFO system_info; 42 43 GetSystemInfo(&system_info); 44 return system_info.dwAllocationGranularity; 45 } 46 47 void *qemu_anon_ram_alloc(size_t size, uint64_t *align, bool shared, 48 bool noreserve) 49 { 50 void *ptr; 51 52 if (noreserve) { 53 /* 54 * We need a MEM_COMMIT before accessing any memory in a MEM_RESERVE 55 * area; we cannot easily mimic POSIX MAP_NORESERVE semantics. 56 */ 57 error_report("Skipping reservation of swap space is not supported."); 58 return NULL; 59 } 60 61 ptr = VirtualAlloc(NULL, size, MEM_COMMIT, PAGE_READWRITE); 62 trace_qemu_anon_ram_alloc(size, ptr); 63 64 if (ptr && align) { 65 *align = MAX(get_allocation_granularity(), getpagesize()); 66 } 67 return ptr; 68 } 69 70 void qemu_anon_ram_free(void *ptr, size_t size) 71 { 72 trace_qemu_anon_ram_free(ptr, size); 73 if (ptr) { 74 VirtualFree(ptr, 0, MEM_RELEASE); 75 } 76 } 77 78 #ifndef _POSIX_THREAD_SAFE_FUNCTIONS 79 /* FIXME: add proper locking */ 80 struct tm *gmtime_r(const time_t *timep, struct tm *result) 81 { 82 struct tm *p = gmtime(timep); 83 memset(result, 0, sizeof(*result)); 84 if (p) { 85 *result = *p; 86 p = result; 87 } 88 return p; 89 } 90 91 /* FIXME: add proper locking */ 92 struct tm *localtime_r(const time_t *timep, struct tm *result) 93 { 94 struct tm *p = localtime(timep); 95 memset(result, 0, sizeof(*result)); 96 if (p) { 97 *result = *p; 98 p = result; 99 } 100 return p; 101 } 102 #endif /* _POSIX_THREAD_SAFE_FUNCTIONS */ 103 104 static int socket_error(void) 105 { 106 switch (WSAGetLastError()) { 107 case 0: 108 return 0; 109 case WSAEINTR: 110 return EINTR; 111 case WSAEINVAL: 112 return EINVAL; 113 case WSA_INVALID_HANDLE: 114 return EBADF; 115 case WSA_NOT_ENOUGH_MEMORY: 116 return ENOMEM; 117 case WSA_INVALID_PARAMETER: 118 return EINVAL; 119 case WSAENAMETOOLONG: 120 return ENAMETOOLONG; 121 case WSAENOTEMPTY: 122 return ENOTEMPTY; 123 case WSAEWOULDBLOCK: 124 /* not using EWOULDBLOCK as we don't want code to have 125 * to check both EWOULDBLOCK and EAGAIN */ 126 return EAGAIN; 127 case WSAEINPROGRESS: 128 return EINPROGRESS; 129 case WSAEALREADY: 130 return EALREADY; 131 case WSAENOTSOCK: 132 return ENOTSOCK; 133 case WSAEDESTADDRREQ: 134 return EDESTADDRREQ; 135 case WSAEMSGSIZE: 136 return EMSGSIZE; 137 case WSAEPROTOTYPE: 138 return EPROTOTYPE; 139 case WSAENOPROTOOPT: 140 return ENOPROTOOPT; 141 case WSAEPROTONOSUPPORT: 142 return EPROTONOSUPPORT; 143 case WSAEOPNOTSUPP: 144 return EOPNOTSUPP; 145 case WSAEAFNOSUPPORT: 146 return EAFNOSUPPORT; 147 case WSAEADDRINUSE: 148 return EADDRINUSE; 149 case WSAEADDRNOTAVAIL: 150 return EADDRNOTAVAIL; 151 case WSAENETDOWN: 152 return ENETDOWN; 153 case WSAENETUNREACH: 154 return ENETUNREACH; 155 case WSAENETRESET: 156 return ENETRESET; 157 case WSAECONNABORTED: 158 return ECONNABORTED; 159 case WSAECONNRESET: 160 return ECONNRESET; 161 case WSAENOBUFS: 162 return ENOBUFS; 163 case WSAEISCONN: 164 return EISCONN; 165 case WSAENOTCONN: 166 return ENOTCONN; 167 case WSAETIMEDOUT: 168 return ETIMEDOUT; 169 case WSAECONNREFUSED: 170 return ECONNREFUSED; 171 case WSAELOOP: 172 return ELOOP; 173 case WSAEHOSTUNREACH: 174 return EHOSTUNREACH; 175 default: 176 return EIO; 177 } 178 } 179 180 void qemu_socket_set_block(int fd) 181 { 182 unsigned long opt = 0; 183 qemu_socket_unselect(fd, NULL); 184 ioctlsocket(fd, FIONBIO, &opt); 185 } 186 187 int qemu_socket_try_set_nonblock(int fd) 188 { 189 unsigned long opt = 1; 190 if (ioctlsocket(fd, FIONBIO, &opt) != NO_ERROR) { 191 return -socket_error(); 192 } 193 return 0; 194 } 195 196 void qemu_socket_set_nonblock(int fd) 197 { 198 (void)qemu_socket_try_set_nonblock(fd); 199 } 200 201 int socket_set_fast_reuse(int fd) 202 { 203 /* Enabling the reuse of an endpoint that was used by a socket still in 204 * TIME_WAIT state is usually performed by setting SO_REUSEADDR. On Windows 205 * fast reuse is the default and SO_REUSEADDR does strange things. So we 206 * don't have to do anything here. More info can be found at: 207 * http://msdn.microsoft.com/en-us/library/windows/desktop/ms740621.aspx */ 208 return 0; 209 } 210 211 int inet_aton(const char *cp, struct in_addr *ia) 212 { 213 uint32_t addr = inet_addr(cp); 214 if (addr == 0xffffffff) { 215 return 0; 216 } 217 ia->s_addr = addr; 218 return 1; 219 } 220 221 void qemu_set_cloexec(int fd) 222 { 223 } 224 225 int qemu_get_thread_id(void) 226 { 227 return GetCurrentThreadId(); 228 } 229 230 char * 231 qemu_get_local_state_dir(void) 232 { 233 const char * const *data_dirs = g_get_system_data_dirs(); 234 235 g_assert(data_dirs && data_dirs[0]); 236 237 return g_strdup(data_dirs[0]); 238 } 239 240 void qemu_set_tty_echo(int fd, bool echo) 241 { 242 HANDLE handle = (HANDLE)_get_osfhandle(fd); 243 DWORD dwMode = 0; 244 245 if (handle == INVALID_HANDLE_VALUE) { 246 return; 247 } 248 249 GetConsoleMode(handle, &dwMode); 250 251 if (echo) { 252 SetConsoleMode(handle, dwMode | ENABLE_ECHO_INPUT | ENABLE_LINE_INPUT); 253 } else { 254 SetConsoleMode(handle, 255 dwMode & ~(ENABLE_ECHO_INPUT | ENABLE_LINE_INPUT)); 256 } 257 } 258 259 int getpagesize(void) 260 { 261 SYSTEM_INFO system_info; 262 263 GetSystemInfo(&system_info); 264 return system_info.dwPageSize; 265 } 266 267 void qemu_prealloc_mem(int fd, char *area, size_t sz, int max_threads, 268 ThreadContext *tc, Error **errp) 269 { 270 int i; 271 size_t pagesize = qemu_real_host_page_size(); 272 273 sz = (sz + pagesize - 1) & -pagesize; 274 for (i = 0; i < sz / pagesize; i++) { 275 memset(area + pagesize * i, 0, 1); 276 } 277 } 278 279 char *qemu_get_pid_name(pid_t pid) 280 { 281 /* XXX Implement me */ 282 abort(); 283 } 284 285 286 bool qemu_socket_select(int sockfd, WSAEVENT hEventObject, 287 long lNetworkEvents, Error **errp) 288 { 289 SOCKET s = _get_osfhandle(sockfd); 290 291 if (errp == NULL) { 292 errp = &error_warn; 293 } 294 295 if (s == INVALID_SOCKET) { 296 error_setg(errp, "invalid socket fd=%d", sockfd); 297 return false; 298 } 299 300 if (WSAEventSelect(s, hEventObject, lNetworkEvents) != 0) { 301 error_setg_win32(errp, WSAGetLastError(), "failed to WSAEventSelect()"); 302 return false; 303 } 304 305 return true; 306 } 307 308 bool qemu_socket_unselect(int sockfd, Error **errp) 309 { 310 return qemu_socket_select(sockfd, NULL, 0, errp); 311 } 312 313 #undef connect 314 int qemu_connect_wrap(int sockfd, const struct sockaddr *addr, 315 socklen_t addrlen) 316 { 317 int ret; 318 SOCKET s = _get_osfhandle(sockfd); 319 320 if (s == INVALID_SOCKET) { 321 return -1; 322 } 323 324 ret = connect(s, addr, addrlen); 325 if (ret < 0) { 326 if (WSAGetLastError() == WSAEWOULDBLOCK) { 327 errno = EINPROGRESS; 328 } else { 329 errno = socket_error(); 330 } 331 } 332 return ret; 333 } 334 335 336 #undef listen 337 int qemu_listen_wrap(int sockfd, int backlog) 338 { 339 int ret; 340 SOCKET s = _get_osfhandle(sockfd); 341 342 if (s == INVALID_SOCKET) { 343 return -1; 344 } 345 346 ret = listen(s, backlog); 347 if (ret < 0) { 348 errno = socket_error(); 349 } 350 return ret; 351 } 352 353 354 #undef bind 355 int qemu_bind_wrap(int sockfd, const struct sockaddr *addr, 356 socklen_t addrlen) 357 { 358 int ret; 359 SOCKET s = _get_osfhandle(sockfd); 360 361 if (s == INVALID_SOCKET) { 362 return -1; 363 } 364 365 ret = bind(s, addr, addrlen); 366 if (ret < 0) { 367 errno = socket_error(); 368 } 369 return ret; 370 } 371 372 373 #undef closesocket 374 int qemu_closesocket_wrap(int fd) 375 { 376 int ret; 377 DWORD flags = 0; 378 SOCKET s = _get_osfhandle(fd); 379 380 if (s == INVALID_SOCKET) { 381 return -1; 382 } 383 384 /* 385 * If we were to just call _close on the descriptor, it would close the 386 * HANDLE, but it wouldn't free any of the resources associated to the 387 * SOCKET, and we can't call _close after calling closesocket, because 388 * closesocket has already closed the HANDLE, and _close would attempt to 389 * close the HANDLE again, resulting in a double free. We can however 390 * protect the HANDLE from actually being closed long enough to close the 391 * file descriptor, then close the socket itself. 392 */ 393 if (!GetHandleInformation((HANDLE)s, &flags)) { 394 errno = EACCES; 395 return -1; 396 } 397 398 if (!SetHandleInformation((HANDLE)s, HANDLE_FLAG_PROTECT_FROM_CLOSE, HANDLE_FLAG_PROTECT_FROM_CLOSE)) { 399 errno = EACCES; 400 return -1; 401 } 402 403 ret = close(fd); 404 405 if (!SetHandleInformation((HANDLE)s, flags, flags)) { 406 errno = EACCES; 407 return -1; 408 } 409 410 /* 411 * close() returns EBADF since we PROTECT_FROM_CLOSE the underlying handle, 412 * but the FD is actually freed 413 */ 414 if (ret < 0 && errno != EBADF) { 415 return ret; 416 } 417 418 ret = closesocket(s); 419 if (ret < 0) { 420 errno = socket_error(); 421 } 422 423 return ret; 424 } 425 426 427 #undef socket 428 int qemu_socket_wrap(int domain, int type, int protocol) 429 { 430 SOCKET s; 431 int fd; 432 433 s = socket(domain, type, protocol); 434 if (s == -1) { 435 errno = socket_error(); 436 return -1; 437 } 438 439 fd = _open_osfhandle(s, _O_BINARY); 440 if (fd < 0) { 441 closesocket(s); 442 /* _open_osfhandle may not set errno, and closesocket() may override it */ 443 errno = ENOMEM; 444 } 445 446 return fd; 447 } 448 449 450 #undef accept 451 int qemu_accept_wrap(int sockfd, struct sockaddr *addr, 452 socklen_t *addrlen) 453 { 454 int fd; 455 SOCKET s = _get_osfhandle(sockfd); 456 457 if (s == INVALID_SOCKET) { 458 return -1; 459 } 460 461 s = accept(s, addr, addrlen); 462 if (s == -1) { 463 errno = socket_error(); 464 return -1; 465 } 466 467 fd = _open_osfhandle(s, _O_BINARY); 468 if (fd < 0) { 469 closesocket(s); 470 /* _open_osfhandle may not set errno, and closesocket() may override it */ 471 errno = ENOMEM; 472 } 473 474 return fd; 475 } 476 477 478 #undef shutdown 479 int qemu_shutdown_wrap(int sockfd, int how) 480 { 481 int ret; 482 SOCKET s = _get_osfhandle(sockfd); 483 484 if (s == INVALID_SOCKET) { 485 return -1; 486 } 487 488 ret = shutdown(s, how); 489 if (ret < 0) { 490 errno = socket_error(); 491 } 492 return ret; 493 } 494 495 496 #undef ioctlsocket 497 int qemu_ioctlsocket_wrap(int fd, int req, void *val) 498 { 499 int ret; 500 SOCKET s = _get_osfhandle(fd); 501 502 if (s == INVALID_SOCKET) { 503 return -1; 504 } 505 506 ret = ioctlsocket(s, req, val); 507 if (ret < 0) { 508 errno = socket_error(); 509 } 510 return ret; 511 } 512 513 514 #undef getsockopt 515 int qemu_getsockopt_wrap(int sockfd, int level, int optname, 516 void *optval, socklen_t *optlen) 517 { 518 int ret; 519 SOCKET s = _get_osfhandle(sockfd); 520 521 if (s == INVALID_SOCKET) { 522 return -1; 523 } 524 525 ret = getsockopt(s, level, optname, optval, optlen); 526 if (ret < 0) { 527 errno = socket_error(); 528 } 529 return ret; 530 } 531 532 533 #undef setsockopt 534 int qemu_setsockopt_wrap(int sockfd, int level, int optname, 535 const void *optval, socklen_t optlen) 536 { 537 int ret; 538 SOCKET s = _get_osfhandle(sockfd); 539 540 if (s == INVALID_SOCKET) { 541 return -1; 542 } 543 544 ret = setsockopt(s, level, optname, optval, optlen); 545 if (ret < 0) { 546 errno = socket_error(); 547 } 548 return ret; 549 } 550 551 552 #undef getpeername 553 int qemu_getpeername_wrap(int sockfd, struct sockaddr *addr, 554 socklen_t *addrlen) 555 { 556 int ret; 557 SOCKET s = _get_osfhandle(sockfd); 558 559 if (s == INVALID_SOCKET) { 560 return -1; 561 } 562 563 ret = getpeername(s, addr, addrlen); 564 if (ret < 0) { 565 errno = socket_error(); 566 } 567 return ret; 568 } 569 570 571 #undef getsockname 572 int qemu_getsockname_wrap(int sockfd, struct sockaddr *addr, 573 socklen_t *addrlen) 574 { 575 int ret; 576 SOCKET s = _get_osfhandle(sockfd); 577 578 if (s == INVALID_SOCKET) { 579 return -1; 580 } 581 582 ret = getsockname(s, addr, addrlen); 583 if (ret < 0) { 584 errno = socket_error(); 585 } 586 return ret; 587 } 588 589 590 #undef send 591 ssize_t qemu_send_wrap(int sockfd, const void *buf, size_t len, int flags) 592 { 593 int ret; 594 SOCKET s = _get_osfhandle(sockfd); 595 596 if (s == INVALID_SOCKET) { 597 return -1; 598 } 599 600 ret = send(s, buf, len, flags); 601 if (ret < 0) { 602 errno = socket_error(); 603 } 604 return ret; 605 } 606 607 608 #undef sendto 609 ssize_t qemu_sendto_wrap(int sockfd, const void *buf, size_t len, int flags, 610 const struct sockaddr *addr, socklen_t addrlen) 611 { 612 int ret; 613 SOCKET s = _get_osfhandle(sockfd); 614 615 if (s == INVALID_SOCKET) { 616 return -1; 617 } 618 619 ret = sendto(s, buf, len, flags, addr, addrlen); 620 if (ret < 0) { 621 errno = socket_error(); 622 } 623 return ret; 624 } 625 626 627 #undef recv 628 ssize_t qemu_recv_wrap(int sockfd, void *buf, size_t len, int flags) 629 { 630 int ret; 631 SOCKET s = _get_osfhandle(sockfd); 632 633 if (s == INVALID_SOCKET) { 634 return -1; 635 } 636 637 ret = recv(s, buf, len, flags); 638 if (ret < 0) { 639 errno = socket_error(); 640 } 641 return ret; 642 } 643 644 645 #undef recvfrom 646 ssize_t qemu_recvfrom_wrap(int sockfd, void *buf, size_t len, int flags, 647 struct sockaddr *addr, socklen_t *addrlen) 648 { 649 int ret; 650 SOCKET s = _get_osfhandle(sockfd); 651 652 if (s == INVALID_SOCKET) { 653 return -1; 654 } 655 656 ret = recvfrom(s, buf, len, flags, addr, addrlen); 657 if (ret < 0) { 658 errno = socket_error(); 659 } 660 return ret; 661 } 662 663 bool qemu_write_pidfile(const char *filename, Error **errp) 664 { 665 char buffer[128]; 666 int len; 667 HANDLE file; 668 OVERLAPPED overlap; 669 BOOL ret; 670 memset(&overlap, 0, sizeof(overlap)); 671 672 file = CreateFile(filename, GENERIC_WRITE, FILE_SHARE_READ, NULL, 673 OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); 674 675 if (file == INVALID_HANDLE_VALUE) { 676 error_setg(errp, "Failed to create PID file"); 677 return false; 678 } 679 len = snprintf(buffer, sizeof(buffer), FMT_pid "\n", (pid_t)getpid()); 680 ret = WriteFile(file, (LPCVOID)buffer, (DWORD)len, 681 NULL, &overlap); 682 CloseHandle(file); 683 if (ret == 0) { 684 error_setg(errp, "Failed to write PID file"); 685 return false; 686 } 687 return true; 688 } 689 690 size_t qemu_get_host_physmem(void) 691 { 692 MEMORYSTATUSEX statex; 693 statex.dwLength = sizeof(statex); 694 695 if (GlobalMemoryStatusEx(&statex)) { 696 return statex.ullTotalPhys; 697 } 698 return 0; 699 } 700 701 int qemu_msync(void *addr, size_t length, int fd) 702 { 703 /** 704 * Perform the sync based on the file descriptor 705 * The sync range will most probably be wider than the one 706 * requested - but it will still get the job done 707 */ 708 return qemu_fdatasync(fd); 709 } 710