1 /* Copyright Joyent, Inc. and other Node contributors. All rights reserved. 2 * Permission is hereby granted, free of charge, to any person obtaining a copy 3 * of this software and associated documentation files (the "Software"), to 4 * deal in the Software without restriction, including without limitation the 5 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or 6 * sell copies of the Software, and to permit persons to whom the Software is 7 * furnished to do so, subject to the following conditions: 8 * 9 * The above copyright notice and this permission notice shall be included in 10 * all copies or substantial portions of the Software. 11 * 12 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 13 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 14 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 15 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 16 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING 17 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS 18 * IN THE SOFTWARE. 19 */ 20 21 #include "uv.h" 22 #include "internal.h" 23 24 #include <stddef.h> /* NULL */ 25 #include <stdio.h> /* printf */ 26 #include <stdlib.h> 27 #include <string.h> /* strerror */ 28 #include <errno.h> 29 #include <assert.h> 30 #include <unistd.h> 31 #include <sys/types.h> 32 #include <sys/stat.h> 33 #include <fcntl.h> /* O_CLOEXEC */ 34 #include <sys/ioctl.h> 35 #include <sys/socket.h> 36 #include <sys/un.h> 37 #include <netinet/in.h> 38 #include <arpa/inet.h> 39 #include <limits.h> /* INT_MAX, PATH_MAX, IOV_MAX */ 40 #include <sys/uio.h> /* writev */ 41 #include <sys/resource.h> /* getrusage */ 42 #include <pwd.h> 43 #include <sys/utsname.h> 44 #include <sys/time.h> 45 46 #ifdef __sun 47 # include <sys/filio.h> 48 # include <sys/types.h> 49 # include <sys/wait.h> 50 #endif 51 52 #if defined(__APPLE__) 53 # include <sys/filio.h> 54 # endif /* defined(__APPLE__) */ 55 56 57 #if defined(__APPLE__) && !TARGET_OS_IPHONE 58 # include <crt_externs.h> 59 # include <mach-o/dyld.h> /* _NSGetExecutablePath */ 60 # define environ (*_NSGetEnviron()) 61 #else /* defined(__APPLE__) && !TARGET_OS_IPHONE */ 62 extern char** environ; 63 #endif /* !(defined(__APPLE__) && !TARGET_OS_IPHONE) */ 64 65 66 #if defined(__DragonFly__) || \ 67 defined(__FreeBSD__) || \ 68 defined(__FreeBSD_kernel__) || \ 69 defined(__NetBSD__) || \ 70 defined(__OpenBSD__) 71 # include <sys/sysctl.h> 72 # include <sys/filio.h> 73 # include <sys/wait.h> 74 # if defined(__FreeBSD__) 75 # define uv__accept4 accept4 76 # endif 77 # if defined(__NetBSD__) 78 # define uv__accept4(a, b, c, d) paccept((a), (b), (c), NULL, (d)) 79 # endif 80 #endif 81 82 #if defined(__MVS__) 83 #include <sys/ioctl.h> 84 #endif 85 86 #if defined(__linux__) 87 # include <sched.h> 88 # include <sys/syscall.h> 89 # define uv__accept4 accept4 90 #endif 91 92 #if defined(__linux__) && defined(__SANITIZE_THREAD__) && defined(__clang__) 93 # include <sanitizer/linux_syscall_hooks.h> 94 #endif 95 96 static int uv__run_pending(uv_loop_t* loop); 97 98 /* Verify that uv_buf_t is ABI-compatible with struct iovec. */ 99 STATIC_ASSERT(sizeof(uv_buf_t) == sizeof(struct iovec)); 100 STATIC_ASSERT(sizeof(((uv_buf_t*) 0)->base) == 101 sizeof(((struct iovec*) 0)->iov_base)); 102 STATIC_ASSERT(sizeof(((uv_buf_t*) 0)->len) == 103 sizeof(((struct iovec*) 0)->iov_len)); 104 STATIC_ASSERT(offsetof(uv_buf_t, base) == offsetof(struct iovec, iov_base)); 105 STATIC_ASSERT(offsetof(uv_buf_t, len) == offsetof(struct iovec, iov_len)); 106 107 uv_hrtime(void)108 uint64_t uv_hrtime(void) { 109 return uv__hrtime(UV_CLOCK_PRECISE); 110 } 111 112 uv_close(uv_handle_t * handle,uv_close_cb close_cb)113 void uv_close(uv_handle_t* handle, uv_close_cb close_cb) { 114 assert(!uv__is_closing(handle)); 115 116 handle->flags |= UV_HANDLE_CLOSING; 117 handle->close_cb = close_cb; 118 119 switch (handle->type) { 120 case UV_NAMED_PIPE: 121 uv__pipe_close((uv_pipe_t*)handle); 122 break; 123 124 case UV_TTY: 125 uv__stream_close((uv_stream_t*)handle); 126 break; 127 128 case UV_TCP: 129 uv__tcp_close((uv_tcp_t*)handle); 130 break; 131 132 case UV_UDP: 133 uv__udp_close((uv_udp_t*)handle); 134 break; 135 136 case UV_PREPARE: 137 uv__prepare_close((uv_prepare_t*)handle); 138 break; 139 140 case UV_CHECK: 141 uv__check_close((uv_check_t*)handle); 142 break; 143 144 case UV_IDLE: 145 uv__idle_close((uv_idle_t*)handle); 146 break; 147 148 case UV_ASYNC: 149 uv__async_close((uv_async_t*)handle); 150 break; 151 152 case UV_TIMER: 153 uv__timer_close((uv_timer_t*)handle); 154 break; 155 156 case UV_PROCESS: 157 uv__process_close((uv_process_t*)handle); 158 break; 159 160 case UV_FS_EVENT: 161 uv__fs_event_close((uv_fs_event_t*)handle); 162 break; 163 164 case UV_POLL: 165 uv__poll_close((uv_poll_t*)handle); 166 break; 167 168 case UV_FS_POLL: 169 uv__fs_poll_close((uv_fs_poll_t*)handle); 170 /* Poll handles use file system requests, and one of them may still be 171 * running. The poll code will call uv__make_close_pending() for us. */ 172 return; 173 174 case UV_SIGNAL: 175 uv__signal_close((uv_signal_t*) handle); 176 break; 177 178 default: 179 assert(0); 180 } 181 182 uv__make_close_pending(handle); 183 } 184 uv__socket_sockopt(uv_handle_t * handle,int optname,int * value)185 int uv__socket_sockopt(uv_handle_t* handle, int optname, int* value) { 186 int r; 187 int fd; 188 socklen_t len; 189 190 if (handle == NULL || value == NULL) 191 return UV_EINVAL; 192 193 if (handle->type == UV_TCP || handle->type == UV_NAMED_PIPE) 194 fd = uv__stream_fd((uv_stream_t*) handle); 195 else if (handle->type == UV_UDP) 196 fd = ((uv_udp_t *) handle)->io_watcher.fd; 197 else 198 return UV_ENOTSUP; 199 200 len = sizeof(*value); 201 202 if (*value == 0) 203 r = getsockopt(fd, SOL_SOCKET, optname, value, &len); 204 else 205 r = setsockopt(fd, SOL_SOCKET, optname, (const void*) value, len); 206 207 if (r < 0) 208 return UV__ERR(errno); 209 210 return 0; 211 } 212 uv__make_close_pending(uv_handle_t * handle)213 void uv__make_close_pending(uv_handle_t* handle) { 214 assert(handle->flags & UV_HANDLE_CLOSING); 215 assert(!(handle->flags & UV_HANDLE_CLOSED)); 216 handle->next_closing = handle->loop->closing_handles; 217 handle->loop->closing_handles = handle; 218 } 219 uv__getiovmax(void)220 int uv__getiovmax(void) { 221 #if defined(IOV_MAX) 222 return IOV_MAX; 223 #elif defined(_SC_IOV_MAX) 224 static int iovmax_cached = -1; 225 int iovmax; 226 227 iovmax = uv__load_relaxed(&iovmax_cached); 228 if (iovmax != -1) 229 return iovmax; 230 231 /* On some embedded devices (arm-linux-uclibc based ip camera), 232 * sysconf(_SC_IOV_MAX) can not get the correct value. The return 233 * value is -1 and the errno is EINPROGRESS. Degrade the value to 1. 234 */ 235 iovmax = sysconf(_SC_IOV_MAX); 236 if (iovmax == -1) 237 iovmax = 1; 238 239 uv__store_relaxed(&iovmax_cached, iovmax); 240 241 return iovmax; 242 #else 243 return 1024; 244 #endif 245 } 246 247 uv__finish_close(uv_handle_t * handle)248 static void uv__finish_close(uv_handle_t* handle) { 249 uv_signal_t* sh; 250 251 /* Note: while the handle is in the UV_HANDLE_CLOSING state now, it's still 252 * possible for it to be active in the sense that uv__is_active() returns 253 * true. 254 * 255 * A good example is when the user calls uv_shutdown(), immediately followed 256 * by uv_close(). The handle is considered active at this point because the 257 * completion of the shutdown req is still pending. 258 */ 259 assert(handle->flags & UV_HANDLE_CLOSING); 260 assert(!(handle->flags & UV_HANDLE_CLOSED)); 261 handle->flags |= UV_HANDLE_CLOSED; 262 263 switch (handle->type) { 264 case UV_PREPARE: 265 case UV_CHECK: 266 case UV_IDLE: 267 case UV_ASYNC: 268 case UV_TIMER: 269 case UV_PROCESS: 270 case UV_FS_EVENT: 271 case UV_FS_POLL: 272 case UV_POLL: 273 break; 274 275 case UV_SIGNAL: 276 /* If there are any caught signals "trapped" in the signal pipe, 277 * we can't call the close callback yet. Reinserting the handle 278 * into the closing queue makes the event loop spin but that's 279 * okay because we only need to deliver the pending events. 280 */ 281 sh = (uv_signal_t*) handle; 282 if (sh->caught_signals > sh->dispatched_signals) { 283 handle->flags ^= UV_HANDLE_CLOSED; 284 uv__make_close_pending(handle); /* Back into the queue. */ 285 return; 286 } 287 break; 288 289 case UV_NAMED_PIPE: 290 case UV_TCP: 291 case UV_TTY: 292 uv__stream_destroy((uv_stream_t*)handle); 293 break; 294 295 case UV_UDP: 296 uv__udp_finish_close((uv_udp_t*)handle); 297 break; 298 299 default: 300 assert(0); 301 break; 302 } 303 304 uv__handle_unref(handle); 305 QUEUE_REMOVE(&handle->handle_queue); 306 307 if (handle->close_cb) { 308 handle->close_cb(handle); 309 } 310 } 311 312 uv__run_closing_handles(uv_loop_t * loop)313 static void uv__run_closing_handles(uv_loop_t* loop) { 314 uv_handle_t* p; 315 uv_handle_t* q; 316 317 p = loop->closing_handles; 318 loop->closing_handles = NULL; 319 320 while (p) { 321 q = p->next_closing; 322 uv__finish_close(p); 323 p = q; 324 } 325 } 326 327 uv_is_closing(const uv_handle_t * handle)328 int uv_is_closing(const uv_handle_t* handle) { 329 return uv__is_closing(handle); 330 } 331 332 uv_backend_fd(const uv_loop_t * loop)333 int uv_backend_fd(const uv_loop_t* loop) { 334 return loop->backend_fd; 335 } 336 337 uv__loop_alive(const uv_loop_t * loop)338 static int uv__loop_alive(const uv_loop_t* loop) { 339 return uv__has_active_handles(loop) || 340 uv__has_active_reqs(loop) || 341 !QUEUE_EMPTY(&loop->pending_queue) || 342 loop->closing_handles != NULL; 343 } 344 345 uv__backend_timeout(const uv_loop_t * loop)346 static int uv__backend_timeout(const uv_loop_t* loop) { 347 if (loop->stop_flag == 0 && 348 /* uv__loop_alive(loop) && */ 349 (uv__has_active_handles(loop) || uv__has_active_reqs(loop)) && 350 QUEUE_EMPTY(&loop->pending_queue) && 351 QUEUE_EMPTY(&loop->idle_handles) && 352 loop->closing_handles == NULL) 353 return uv__next_timeout(loop); 354 return 0; 355 } 356 357 uv_backend_timeout(const uv_loop_t * loop)358 int uv_backend_timeout(const uv_loop_t* loop) { 359 if (QUEUE_EMPTY(&loop->watcher_queue)) 360 return uv__backend_timeout(loop); 361 /* Need to call uv_run to update the backend fd state. */ 362 return 0; 363 } 364 365 uv_loop_alive(const uv_loop_t * loop)366 int uv_loop_alive(const uv_loop_t* loop) { 367 return uv__loop_alive(loop); 368 } 369 370 uv_run(uv_loop_t * loop,uv_run_mode mode)371 int uv_run(uv_loop_t* loop, uv_run_mode mode) { 372 int timeout; 373 int r; 374 int ran_pending; 375 376 r = uv__loop_alive(loop); 377 if (!r) 378 uv__update_time(loop); 379 380 while (r != 0 && loop->stop_flag == 0) { 381 uv__update_time(loop); 382 uv__run_timers(loop); 383 ran_pending = uv__run_pending(loop); 384 uv__run_idle(loop); 385 uv__run_prepare(loop); 386 387 timeout = 0; 388 if ((mode == UV_RUN_ONCE && !ran_pending) || mode == UV_RUN_DEFAULT) 389 timeout = uv__backend_timeout(loop); 390 391 uv__io_poll(loop, timeout); 392 393 /* Run one final update on the provider_idle_time in case uv__io_poll 394 * returned because the timeout expired, but no events were received. This 395 * call will be ignored if the provider_entry_time was either never set (if 396 * the timeout == 0) or was already updated b/c an event was received. 397 */ 398 uv__metrics_update_idle_time(loop); 399 400 uv__run_check(loop); 401 uv__run_closing_handles(loop); 402 403 if (mode == UV_RUN_ONCE) { 404 /* UV_RUN_ONCE implies forward progress: at least one callback must have 405 * been invoked when it returns. uv__io_poll() can return without doing 406 * I/O (meaning: no callbacks) when its timeout expires - which means we 407 * have pending timers that satisfy the forward progress constraint. 408 * 409 * UV_RUN_NOWAIT makes no guarantees about progress so it's omitted from 410 * the check. 411 */ 412 uv__update_time(loop); 413 uv__run_timers(loop); 414 } 415 416 r = uv__loop_alive(loop); 417 if (mode == UV_RUN_ONCE || mode == UV_RUN_NOWAIT) 418 break; 419 } 420 421 /* The if statement lets gcc compile it to a conditional store. Avoids 422 * dirtying a cache line. 423 */ 424 if (loop->stop_flag != 0) 425 loop->stop_flag = 0; 426 427 return r; 428 } 429 430 uv_update_time(uv_loop_t * loop)431 void uv_update_time(uv_loop_t* loop) { 432 uv__update_time(loop); 433 } 434 435 uv_is_active(const uv_handle_t * handle)436 int uv_is_active(const uv_handle_t* handle) { 437 return uv__is_active(handle); 438 } 439 440 441 /* Open a socket in non-blocking close-on-exec mode, atomically if possible. */ uv__socket(int domain,int type,int protocol)442 int uv__socket(int domain, int type, int protocol) { 443 int sockfd; 444 int err; 445 446 #if defined(SOCK_NONBLOCK) && defined(SOCK_CLOEXEC) 447 sockfd = socket(domain, type | SOCK_NONBLOCK | SOCK_CLOEXEC, protocol); 448 if (sockfd != -1) 449 return sockfd; 450 451 if (errno != EINVAL) 452 return UV__ERR(errno); 453 #endif 454 455 sockfd = socket(domain, type, protocol); 456 if (sockfd == -1) 457 return UV__ERR(errno); 458 459 err = uv__nonblock(sockfd, 1); 460 if (err == 0) 461 err = uv__cloexec(sockfd, 1); 462 463 if (err) { 464 uv__close(sockfd); 465 return err; 466 } 467 468 #if defined(SO_NOSIGPIPE) 469 { 470 int on = 1; 471 setsockopt(sockfd, SOL_SOCKET, SO_NOSIGPIPE, &on, sizeof(on)); 472 } 473 #endif 474 475 return sockfd; 476 } 477 478 /* get a file pointer to a file in read-only and close-on-exec mode */ uv__open_file(const char * path)479 FILE* uv__open_file(const char* path) { 480 int fd; 481 FILE* fp; 482 483 fd = uv__open_cloexec(path, O_RDONLY); 484 if (fd < 0) 485 return NULL; 486 487 fp = fdopen(fd, "r"); 488 if (fp == NULL) 489 uv__close(fd); 490 491 return fp; 492 } 493 494 uv__accept(int sockfd)495 int uv__accept(int sockfd) { 496 int peerfd; 497 int err; 498 499 (void) &err; 500 assert(sockfd >= 0); 501 502 do 503 #ifdef uv__accept4 504 peerfd = uv__accept4(sockfd, NULL, NULL, SOCK_NONBLOCK|SOCK_CLOEXEC); 505 #else 506 peerfd = accept(sockfd, NULL, NULL); 507 #endif 508 while (peerfd == -1 && errno == EINTR); 509 510 if (peerfd == -1) 511 return UV__ERR(errno); 512 513 #ifndef uv__accept4 514 err = uv__cloexec(peerfd, 1); 515 if (err == 0) 516 err = uv__nonblock(peerfd, 1); 517 518 if (err != 0) { 519 uv__close(peerfd); 520 return err; 521 } 522 #endif 523 524 return peerfd; 525 } 526 527 528 /* close() on macos has the "interesting" quirk that it fails with EINTR 529 * without closing the file descriptor when a thread is in the cancel state. 530 * That's why libuv calls close$NOCANCEL() instead. 531 * 532 * glibc on linux has a similar issue: close() is a cancellation point and 533 * will unwind the thread when it's in the cancel state. Work around that 534 * by making the system call directly. Musl libc is unaffected. 535 */ uv__close_nocancel(int fd)536 int uv__close_nocancel(int fd) { 537 #if defined(__APPLE__) 538 #pragma GCC diagnostic push 539 #pragma GCC diagnostic ignored "-Wdollar-in-identifier-extension" 540 #if defined(__LP64__) || TARGET_OS_IPHONE 541 extern int close$NOCANCEL(int); 542 return close$NOCANCEL(fd); 543 #else 544 extern int close$NOCANCEL$UNIX2003(int); 545 return close$NOCANCEL$UNIX2003(fd); 546 #endif 547 #pragma GCC diagnostic pop 548 #elif defined(__linux__) && defined(__SANITIZE_THREAD__) && defined(__clang__) 549 long rc; 550 __sanitizer_syscall_pre_close(fd); 551 rc = syscall(SYS_close, fd); 552 __sanitizer_syscall_post_close(rc, fd); 553 return rc; 554 #elif defined(__linux__) && !defined(__SANITIZE_THREAD__) 555 return syscall(SYS_close, fd); 556 #else 557 return close(fd); 558 #endif 559 } 560 561 uv__close_nocheckstdio(int fd)562 int uv__close_nocheckstdio(int fd) { 563 int saved_errno; 564 int rc; 565 566 assert(fd > -1); /* Catch uninitialized io_watcher.fd bugs. */ 567 568 saved_errno = errno; 569 rc = uv__close_nocancel(fd); 570 if (rc == -1) { 571 rc = UV__ERR(errno); 572 if (rc == UV_EINTR || rc == UV__ERR(EINPROGRESS)) 573 rc = 0; /* The close is in progress, not an error. */ 574 errno = saved_errno; 575 } 576 577 return rc; 578 } 579 580 uv__close(int fd)581 int uv__close(int fd) { 582 assert(fd > STDERR_FILENO); /* Catch stdio close bugs. */ 583 #if defined(__MVS__) 584 SAVE_ERRNO(epoll_file_close(fd)); 585 #endif 586 return uv__close_nocheckstdio(fd); 587 } 588 589 #if UV__NONBLOCK_IS_IOCTL uv__nonblock_ioctl(int fd,int set)590 int uv__nonblock_ioctl(int fd, int set) { 591 int r; 592 593 do 594 r = ioctl(fd, FIONBIO, &set); 595 while (r == -1 && errno == EINTR); 596 597 if (r) 598 return UV__ERR(errno); 599 600 return 0; 601 } 602 #endif 603 604 uv__nonblock_fcntl(int fd,int set)605 int uv__nonblock_fcntl(int fd, int set) { 606 int flags; 607 int r; 608 609 do 610 r = fcntl(fd, F_GETFL); 611 while (r == -1 && errno == EINTR); 612 613 if (r == -1) 614 return UV__ERR(errno); 615 616 /* Bail out now if already set/clear. */ 617 if (!!(r & O_NONBLOCK) == !!set) 618 return 0; 619 620 if (set) 621 flags = r | O_NONBLOCK; 622 else 623 flags = r & ~O_NONBLOCK; 624 625 do 626 r = fcntl(fd, F_SETFL, flags); 627 while (r == -1 && errno == EINTR); 628 629 if (r) 630 return UV__ERR(errno); 631 632 return 0; 633 } 634 635 uv__cloexec(int fd,int set)636 int uv__cloexec(int fd, int set) { 637 int flags; 638 int r; 639 640 flags = 0; 641 if (set) 642 flags = FD_CLOEXEC; 643 644 do 645 r = fcntl(fd, F_SETFD, flags); 646 while (r == -1 && errno == EINTR); 647 648 if (r) 649 return UV__ERR(errno); 650 651 return 0; 652 } 653 654 uv__recvmsg(int fd,struct msghdr * msg,int flags)655 ssize_t uv__recvmsg(int fd, struct msghdr* msg, int flags) { 656 struct cmsghdr* cmsg; 657 ssize_t rc; 658 int* pfd; 659 int* end; 660 #if defined(__linux__) 661 static int no_msg_cmsg_cloexec; 662 if (0 == uv__load_relaxed(&no_msg_cmsg_cloexec)) { 663 rc = recvmsg(fd, msg, flags | 0x40000000); /* MSG_CMSG_CLOEXEC */ 664 if (rc != -1) 665 return rc; 666 if (errno != EINVAL) 667 return UV__ERR(errno); 668 rc = recvmsg(fd, msg, flags); 669 if (rc == -1) 670 return UV__ERR(errno); 671 uv__store_relaxed(&no_msg_cmsg_cloexec, 1); 672 } else { 673 rc = recvmsg(fd, msg, flags); 674 } 675 #else 676 rc = recvmsg(fd, msg, flags); 677 #endif 678 if (rc == -1) 679 return UV__ERR(errno); 680 if (msg->msg_controllen == 0) 681 return rc; 682 for (cmsg = CMSG_FIRSTHDR(msg); cmsg != NULL; cmsg = CMSG_NXTHDR(msg, cmsg)) 683 if (cmsg->cmsg_type == SCM_RIGHTS) 684 for (pfd = (int*) CMSG_DATA(cmsg), 685 end = (int*) ((char*) cmsg + cmsg->cmsg_len); 686 pfd < end; 687 pfd += 1) 688 uv__cloexec(*pfd, 1); 689 return rc; 690 } 691 692 uv_cwd(char * buffer,size_t * size)693 int uv_cwd(char* buffer, size_t* size) { 694 char scratch[1 + UV__PATH_MAX]; 695 696 if (buffer == NULL || size == NULL) 697 return UV_EINVAL; 698 699 /* Try to read directly into the user's buffer first... */ 700 if (getcwd(buffer, *size) != NULL) 701 goto fixup; 702 703 if (errno != ERANGE) 704 return UV__ERR(errno); 705 706 /* ...or into scratch space if the user's buffer is too small 707 * so we can report how much space to provide on the next try. 708 */ 709 if (getcwd(scratch, sizeof(scratch)) == NULL) 710 return UV__ERR(errno); 711 712 buffer = scratch; 713 714 fixup: 715 716 *size = strlen(buffer); 717 718 if (*size > 1 && buffer[*size - 1] == '/') { 719 *size -= 1; 720 buffer[*size] = '\0'; 721 } 722 723 if (buffer == scratch) { 724 *size += 1; 725 return UV_ENOBUFS; 726 } 727 728 return 0; 729 } 730 731 uv_chdir(const char * dir)732 int uv_chdir(const char* dir) { 733 if (chdir(dir)) 734 return UV__ERR(errno); 735 736 return 0; 737 } 738 739 uv_disable_stdio_inheritance(void)740 void uv_disable_stdio_inheritance(void) { 741 int fd; 742 743 /* Set the CLOEXEC flag on all open descriptors. Unconditionally try the 744 * first 16 file descriptors. After that, bail out after the first error. 745 */ 746 for (fd = 0; ; fd++) 747 if (uv__cloexec(fd, 1) && fd > 15) 748 break; 749 } 750 751 uv_fileno(const uv_handle_t * handle,uv_os_fd_t * fd)752 int uv_fileno(const uv_handle_t* handle, uv_os_fd_t* fd) { 753 int fd_out; 754 755 switch (handle->type) { 756 case UV_TCP: 757 case UV_NAMED_PIPE: 758 case UV_TTY: 759 fd_out = uv__stream_fd((uv_stream_t*) handle); 760 break; 761 762 case UV_UDP: 763 fd_out = ((uv_udp_t *) handle)->io_watcher.fd; 764 break; 765 766 case UV_POLL: 767 fd_out = ((uv_poll_t *) handle)->io_watcher.fd; 768 break; 769 770 default: 771 return UV_EINVAL; 772 } 773 774 if (uv__is_closing(handle) || fd_out == -1) 775 return UV_EBADF; 776 777 *fd = fd_out; 778 return 0; 779 } 780 781 uv__run_pending(uv_loop_t * loop)782 static int uv__run_pending(uv_loop_t* loop) { 783 QUEUE* q; 784 QUEUE pq; 785 uv__io_t* w; 786 787 if (QUEUE_EMPTY(&loop->pending_queue)) 788 return 0; 789 790 QUEUE_MOVE(&loop->pending_queue, &pq); 791 792 while (!QUEUE_EMPTY(&pq)) { 793 q = QUEUE_HEAD(&pq); 794 QUEUE_REMOVE(q); 795 QUEUE_INIT(q); 796 w = QUEUE_DATA(q, uv__io_t, pending_queue); 797 w->cb(loop, w, POLLOUT); 798 } 799 800 return 1; 801 } 802 803 next_power_of_two(unsigned int val)804 static unsigned int next_power_of_two(unsigned int val) { 805 val -= 1; 806 val |= val >> 1; 807 val |= val >> 2; 808 val |= val >> 4; 809 val |= val >> 8; 810 val |= val >> 16; 811 val += 1; 812 return val; 813 } 814 maybe_resize(uv_loop_t * loop,unsigned int len)815 static void maybe_resize(uv_loop_t* loop, unsigned int len) { 816 uv__io_t** watchers; 817 void* fake_watcher_list; 818 void* fake_watcher_count; 819 unsigned int nwatchers; 820 unsigned int i; 821 822 if (len <= loop->nwatchers) 823 return; 824 825 /* Preserve fake watcher list and count at the end of the watchers */ 826 if (loop->watchers != NULL) { 827 fake_watcher_list = loop->watchers[loop->nwatchers]; 828 fake_watcher_count = loop->watchers[loop->nwatchers + 1]; 829 } else { 830 fake_watcher_list = NULL; 831 fake_watcher_count = NULL; 832 } 833 834 nwatchers = next_power_of_two(len + 2) - 2; 835 watchers = uv__reallocf(loop->watchers, 836 (nwatchers + 2) * sizeof(loop->watchers[0])); 837 838 if (watchers == NULL) 839 abort(); 840 for (i = loop->nwatchers; i < nwatchers; i++) 841 watchers[i] = NULL; 842 watchers[nwatchers] = fake_watcher_list; 843 watchers[nwatchers + 1] = fake_watcher_count; 844 845 loop->watchers = watchers; 846 loop->nwatchers = nwatchers; 847 } 848 849 uv__io_init(uv__io_t * w,uv__io_cb cb,int fd)850 void uv__io_init(uv__io_t* w, uv__io_cb cb, int fd) { 851 assert(cb != NULL); 852 assert(fd >= -1); 853 QUEUE_INIT(&w->pending_queue); 854 QUEUE_INIT(&w->watcher_queue); 855 w->cb = cb; 856 w->fd = fd; 857 w->events = 0; 858 w->pevents = 0; 859 860 #if defined(UV_HAVE_KQUEUE) 861 w->rcount = 0; 862 w->wcount = 0; 863 #endif /* defined(UV_HAVE_KQUEUE) */ 864 } 865 866 uv__io_start(uv_loop_t * loop,uv__io_t * w,unsigned int events)867 void uv__io_start(uv_loop_t* loop, uv__io_t* w, unsigned int events) { 868 assert(0 == (events & ~(POLLIN | POLLOUT | UV__POLLRDHUP | UV__POLLPRI))); 869 assert(0 != events); 870 assert(w->fd >= 0); 871 assert(w->fd < INT_MAX); 872 873 w->pevents |= events; 874 maybe_resize(loop, w->fd + 1); 875 876 #if !defined(__sun) 877 /* The event ports backend needs to rearm all file descriptors on each and 878 * every tick of the event loop but the other backends allow us to 879 * short-circuit here if the event mask is unchanged. 880 */ 881 if (w->events == w->pevents) 882 return; 883 #endif 884 885 if (QUEUE_EMPTY(&w->watcher_queue)) 886 QUEUE_INSERT_TAIL(&loop->watcher_queue, &w->watcher_queue); 887 888 if (loop->watchers[w->fd] == NULL) { 889 loop->watchers[w->fd] = w; 890 loop->nfds++; 891 } 892 } 893 894 uv__io_stop(uv_loop_t * loop,uv__io_t * w,unsigned int events)895 void uv__io_stop(uv_loop_t* loop, uv__io_t* w, unsigned int events) { 896 assert(0 == (events & ~(POLLIN | POLLOUT | UV__POLLRDHUP | UV__POLLPRI))); 897 assert(0 != events); 898 899 if (w->fd == -1) 900 return; 901 902 assert(w->fd >= 0); 903 904 /* Happens when uv__io_stop() is called on a handle that was never started. */ 905 if ((unsigned) w->fd >= loop->nwatchers) 906 return; 907 908 w->pevents &= ~events; 909 910 if (w->pevents == 0) { 911 QUEUE_REMOVE(&w->watcher_queue); 912 QUEUE_INIT(&w->watcher_queue); 913 w->events = 0; 914 915 if (w == loop->watchers[w->fd]) { 916 assert(loop->nfds > 0); 917 loop->watchers[w->fd] = NULL; 918 loop->nfds--; 919 } 920 } 921 else if (QUEUE_EMPTY(&w->watcher_queue)) 922 QUEUE_INSERT_TAIL(&loop->watcher_queue, &w->watcher_queue); 923 } 924 925 uv__io_close(uv_loop_t * loop,uv__io_t * w)926 void uv__io_close(uv_loop_t* loop, uv__io_t* w) { 927 uv__io_stop(loop, w, POLLIN | POLLOUT | UV__POLLRDHUP | UV__POLLPRI); 928 QUEUE_REMOVE(&w->pending_queue); 929 930 /* Remove stale events for this file descriptor */ 931 if (w->fd != -1) 932 uv__platform_invalidate_fd(loop, w->fd); 933 } 934 935 uv__io_feed(uv_loop_t * loop,uv__io_t * w)936 void uv__io_feed(uv_loop_t* loop, uv__io_t* w) { 937 if (QUEUE_EMPTY(&w->pending_queue)) 938 QUEUE_INSERT_TAIL(&loop->pending_queue, &w->pending_queue); 939 } 940 941 uv__io_active(const uv__io_t * w,unsigned int events)942 int uv__io_active(const uv__io_t* w, unsigned int events) { 943 assert(0 == (events & ~(POLLIN | POLLOUT | UV__POLLRDHUP | UV__POLLPRI))); 944 assert(0 != events); 945 return 0 != (w->pevents & events); 946 } 947 948 uv__fd_exists(uv_loop_t * loop,int fd)949 int uv__fd_exists(uv_loop_t* loop, int fd) { 950 return (unsigned) fd < loop->nwatchers && loop->watchers[fd] != NULL; 951 } 952 953 uv_getrusage(uv_rusage_t * rusage)954 int uv_getrusage(uv_rusage_t* rusage) { 955 struct rusage usage; 956 957 if (getrusage(RUSAGE_SELF, &usage)) 958 return UV__ERR(errno); 959 960 rusage->ru_utime.tv_sec = usage.ru_utime.tv_sec; 961 rusage->ru_utime.tv_usec = usage.ru_utime.tv_usec; 962 963 rusage->ru_stime.tv_sec = usage.ru_stime.tv_sec; 964 rusage->ru_stime.tv_usec = usage.ru_stime.tv_usec; 965 966 #if !defined(__MVS__) && !defined(__HAIKU__) 967 rusage->ru_maxrss = usage.ru_maxrss; 968 rusage->ru_ixrss = usage.ru_ixrss; 969 rusage->ru_idrss = usage.ru_idrss; 970 rusage->ru_isrss = usage.ru_isrss; 971 rusage->ru_minflt = usage.ru_minflt; 972 rusage->ru_majflt = usage.ru_majflt; 973 rusage->ru_nswap = usage.ru_nswap; 974 rusage->ru_inblock = usage.ru_inblock; 975 rusage->ru_oublock = usage.ru_oublock; 976 rusage->ru_msgsnd = usage.ru_msgsnd; 977 rusage->ru_msgrcv = usage.ru_msgrcv; 978 rusage->ru_nsignals = usage.ru_nsignals; 979 rusage->ru_nvcsw = usage.ru_nvcsw; 980 rusage->ru_nivcsw = usage.ru_nivcsw; 981 #endif 982 983 return 0; 984 } 985 986 uv__open_cloexec(const char * path,int flags)987 int uv__open_cloexec(const char* path, int flags) { 988 #if defined(O_CLOEXEC) 989 int fd; 990 991 fd = open(path, flags | O_CLOEXEC); 992 if (fd == -1) 993 return UV__ERR(errno); 994 995 return fd; 996 #else /* O_CLOEXEC */ 997 int err; 998 int fd; 999 1000 fd = open(path, flags); 1001 if (fd == -1) 1002 return UV__ERR(errno); 1003 1004 err = uv__cloexec(fd, 1); 1005 if (err) { 1006 uv__close(fd); 1007 return err; 1008 } 1009 1010 return fd; 1011 #endif /* O_CLOEXEC */ 1012 } 1013 1014 uv__slurp(const char * filename,char * buf,size_t len)1015 int uv__slurp(const char* filename, char* buf, size_t len) { 1016 ssize_t n; 1017 int fd; 1018 1019 assert(len > 0); 1020 1021 fd = uv__open_cloexec(filename, O_RDONLY); 1022 if (fd < 0) 1023 return fd; 1024 1025 do 1026 n = read(fd, buf, len - 1); 1027 while (n == -1 && errno == EINTR); 1028 1029 if (uv__close_nocheckstdio(fd)) 1030 abort(); 1031 1032 if (n < 0) 1033 return UV__ERR(errno); 1034 1035 buf[n] = '\0'; 1036 1037 return 0; 1038 } 1039 1040 uv__dup2_cloexec(int oldfd,int newfd)1041 int uv__dup2_cloexec(int oldfd, int newfd) { 1042 #if defined(__FreeBSD__) || defined(__NetBSD__) || defined(__linux__) 1043 int r; 1044 1045 r = dup3(oldfd, newfd, O_CLOEXEC); 1046 if (r == -1) 1047 return UV__ERR(errno); 1048 1049 return r; 1050 #else 1051 int err; 1052 int r; 1053 1054 r = dup2(oldfd, newfd); /* Never retry. */ 1055 if (r == -1) 1056 return UV__ERR(errno); 1057 1058 err = uv__cloexec(newfd, 1); 1059 if (err != 0) { 1060 uv__close(newfd); 1061 return err; 1062 } 1063 1064 return r; 1065 #endif 1066 } 1067 1068 uv_os_homedir(char * buffer,size_t * size)1069 int uv_os_homedir(char* buffer, size_t* size) { 1070 uv_passwd_t pwd; 1071 size_t len; 1072 int r; 1073 1074 /* Check if the HOME environment variable is set first. The task of 1075 performing input validation on buffer and size is taken care of by 1076 uv_os_getenv(). */ 1077 r = uv_os_getenv("HOME", buffer, size); 1078 1079 if (r != UV_ENOENT) 1080 return r; 1081 1082 /* HOME is not set, so call uv__getpwuid_r() */ 1083 r = uv__getpwuid_r(&pwd); 1084 1085 if (r != 0) { 1086 return r; 1087 } 1088 1089 len = strlen(pwd.homedir); 1090 1091 if (len >= *size) { 1092 *size = len + 1; 1093 uv_os_free_passwd(&pwd); 1094 return UV_ENOBUFS; 1095 } 1096 1097 memcpy(buffer, pwd.homedir, len + 1); 1098 *size = len; 1099 uv_os_free_passwd(&pwd); 1100 1101 return 0; 1102 } 1103 1104 uv_os_tmpdir(char * buffer,size_t * size)1105 int uv_os_tmpdir(char* buffer, size_t* size) { 1106 const char* buf; 1107 size_t len; 1108 1109 if (buffer == NULL || size == NULL || *size == 0) 1110 return UV_EINVAL; 1111 1112 #define CHECK_ENV_VAR(name) \ 1113 do { \ 1114 buf = getenv(name); \ 1115 if (buf != NULL) \ 1116 goto return_buffer; \ 1117 } \ 1118 while (0) 1119 1120 /* Check the TMPDIR, TMP, TEMP, and TEMPDIR environment variables in order */ 1121 CHECK_ENV_VAR("TMPDIR"); 1122 CHECK_ENV_VAR("TMP"); 1123 CHECK_ENV_VAR("TEMP"); 1124 CHECK_ENV_VAR("TEMPDIR"); 1125 1126 #undef CHECK_ENV_VAR 1127 1128 /* No temp environment variables defined */ 1129 #if defined(__ANDROID__) 1130 buf = "/data/local/tmp"; 1131 #else 1132 buf = "/tmp"; 1133 #endif 1134 1135 return_buffer: 1136 len = strlen(buf); 1137 1138 if (len >= *size) { 1139 *size = len + 1; 1140 return UV_ENOBUFS; 1141 } 1142 1143 /* The returned directory should not have a trailing slash. */ 1144 if (len > 1 && buf[len - 1] == '/') { 1145 len--; 1146 } 1147 1148 memcpy(buffer, buf, len + 1); 1149 buffer[len] = '\0'; 1150 *size = len; 1151 1152 return 0; 1153 } 1154 1155 uv__getpwuid_r(uv_passwd_t * pwd)1156 int uv__getpwuid_r(uv_passwd_t* pwd) { 1157 struct passwd pw; 1158 struct passwd* result; 1159 char* buf; 1160 uid_t uid; 1161 size_t bufsize; 1162 size_t name_size; 1163 size_t homedir_size; 1164 size_t shell_size; 1165 long initsize; 1166 int r; 1167 1168 if (pwd == NULL) 1169 return UV_EINVAL; 1170 1171 initsize = sysconf(_SC_GETPW_R_SIZE_MAX); 1172 1173 if (initsize <= 0) 1174 bufsize = 4096; 1175 else 1176 bufsize = (size_t) initsize; 1177 1178 uid = geteuid(); 1179 buf = NULL; 1180 1181 for (;;) { 1182 uv__free(buf); 1183 buf = uv__malloc(bufsize); 1184 1185 if (buf == NULL) 1186 return UV_ENOMEM; 1187 1188 do 1189 r = getpwuid_r(uid, &pw, buf, bufsize, &result); 1190 while (r == EINTR); 1191 1192 if (r != ERANGE) 1193 break; 1194 1195 bufsize *= 2; 1196 } 1197 1198 if (r != 0) { 1199 uv__free(buf); 1200 return UV__ERR(r); 1201 } 1202 1203 if (result == NULL) { 1204 uv__free(buf); 1205 return UV_ENOENT; 1206 } 1207 1208 /* Allocate memory for the username, shell, and home directory */ 1209 name_size = strlen(pw.pw_name) + 1; 1210 homedir_size = strlen(pw.pw_dir) + 1; 1211 shell_size = strlen(pw.pw_shell) + 1; 1212 pwd->username = uv__malloc(name_size + homedir_size + shell_size); 1213 1214 if (pwd->username == NULL) { 1215 uv__free(buf); 1216 return UV_ENOMEM; 1217 } 1218 1219 /* Copy the username */ 1220 memcpy(pwd->username, pw.pw_name, name_size); 1221 1222 /* Copy the home directory */ 1223 pwd->homedir = pwd->username + name_size; 1224 memcpy(pwd->homedir, pw.pw_dir, homedir_size); 1225 1226 /* Copy the shell */ 1227 pwd->shell = pwd->homedir + homedir_size; 1228 memcpy(pwd->shell, pw.pw_shell, shell_size); 1229 1230 /* Copy the uid and gid */ 1231 pwd->uid = pw.pw_uid; 1232 pwd->gid = pw.pw_gid; 1233 1234 uv__free(buf); 1235 1236 return 0; 1237 } 1238 1239 uv_os_free_passwd(uv_passwd_t * pwd)1240 void uv_os_free_passwd(uv_passwd_t* pwd) { 1241 if (pwd == NULL) 1242 return; 1243 1244 /* 1245 The memory for name, shell, and homedir are allocated in a single 1246 uv__malloc() call. The base of the pointer is stored in pwd->username, so 1247 that is the field that needs to be freed. 1248 */ 1249 uv__free(pwd->username); 1250 pwd->username = NULL; 1251 pwd->shell = NULL; 1252 pwd->homedir = NULL; 1253 } 1254 1255 uv_os_get_passwd(uv_passwd_t * pwd)1256 int uv_os_get_passwd(uv_passwd_t* pwd) { 1257 return uv__getpwuid_r(pwd); 1258 } 1259 1260 uv_translate_sys_error(int sys_errno)1261 int uv_translate_sys_error(int sys_errno) { 1262 /* If < 0 then it's already a libuv error. */ 1263 return sys_errno <= 0 ? sys_errno : -sys_errno; 1264 } 1265 1266 uv_os_environ(uv_env_item_t ** envitems,int * count)1267 int uv_os_environ(uv_env_item_t** envitems, int* count) { 1268 int i, j, cnt; 1269 uv_env_item_t* envitem; 1270 1271 *envitems = NULL; 1272 *count = 0; 1273 1274 for (i = 0; environ[i] != NULL; i++); 1275 1276 *envitems = uv__calloc(i, sizeof(**envitems)); 1277 1278 if (*envitems == NULL) 1279 return UV_ENOMEM; 1280 1281 for (j = 0, cnt = 0; j < i; j++) { 1282 char* buf; 1283 char* ptr; 1284 1285 if (environ[j] == NULL) 1286 break; 1287 1288 buf = uv__strdup(environ[j]); 1289 if (buf == NULL) 1290 goto fail; 1291 1292 ptr = strchr(buf, '='); 1293 if (ptr == NULL) { 1294 uv__free(buf); 1295 continue; 1296 } 1297 1298 *ptr = '\0'; 1299 1300 envitem = &(*envitems)[cnt]; 1301 envitem->name = buf; 1302 envitem->value = ptr + 1; 1303 1304 cnt++; 1305 } 1306 1307 *count = cnt; 1308 return 0; 1309 1310 fail: 1311 for (i = 0; i < cnt; i++) { 1312 envitem = &(*envitems)[cnt]; 1313 uv__free(envitem->name); 1314 } 1315 uv__free(*envitems); 1316 1317 *envitems = NULL; 1318 *count = 0; 1319 return UV_ENOMEM; 1320 } 1321 1322 uv_os_getenv(const char * name,char * buffer,size_t * size)1323 int uv_os_getenv(const char* name, char* buffer, size_t* size) { 1324 char* var; 1325 size_t len; 1326 1327 if (name == NULL || buffer == NULL || size == NULL || *size == 0) 1328 return UV_EINVAL; 1329 1330 var = getenv(name); 1331 1332 if (var == NULL) 1333 return UV_ENOENT; 1334 1335 len = strlen(var); 1336 1337 if (len >= *size) { 1338 *size = len + 1; 1339 return UV_ENOBUFS; 1340 } 1341 1342 memcpy(buffer, var, len + 1); 1343 *size = len; 1344 1345 return 0; 1346 } 1347 1348 uv_os_setenv(const char * name,const char * value)1349 int uv_os_setenv(const char* name, const char* value) { 1350 if (name == NULL || value == NULL) 1351 return UV_EINVAL; 1352 1353 if (setenv(name, value, 1) != 0) 1354 return UV__ERR(errno); 1355 1356 return 0; 1357 } 1358 1359 uv_os_unsetenv(const char * name)1360 int uv_os_unsetenv(const char* name) { 1361 if (name == NULL) 1362 return UV_EINVAL; 1363 1364 if (unsetenv(name) != 0) 1365 return UV__ERR(errno); 1366 1367 return 0; 1368 } 1369 1370 uv_os_gethostname(char * buffer,size_t * size)1371 int uv_os_gethostname(char* buffer, size_t* size) { 1372 /* 1373 On some platforms, if the input buffer is not large enough, gethostname() 1374 succeeds, but truncates the result. libuv can detect this and return ENOBUFS 1375 instead by creating a large enough buffer and comparing the hostname length 1376 to the size input. 1377 */ 1378 char buf[UV_MAXHOSTNAMESIZE]; 1379 size_t len; 1380 1381 if (buffer == NULL || size == NULL || *size == 0) 1382 return UV_EINVAL; 1383 1384 if (gethostname(buf, sizeof(buf)) != 0) 1385 return UV__ERR(errno); 1386 1387 buf[sizeof(buf) - 1] = '\0'; /* Null terminate, just to be safe. */ 1388 len = strlen(buf); 1389 1390 if (len >= *size) { 1391 *size = len + 1; 1392 return UV_ENOBUFS; 1393 } 1394 1395 memcpy(buffer, buf, len + 1); 1396 *size = len; 1397 return 0; 1398 } 1399 1400 uv_get_osfhandle(int fd)1401 uv_os_fd_t uv_get_osfhandle(int fd) { 1402 return fd; 1403 } 1404 uv_open_osfhandle(uv_os_fd_t os_fd)1405 int uv_open_osfhandle(uv_os_fd_t os_fd) { 1406 return os_fd; 1407 } 1408 uv_os_getpid(void)1409 uv_pid_t uv_os_getpid(void) { 1410 return getpid(); 1411 } 1412 1413 uv_os_getppid(void)1414 uv_pid_t uv_os_getppid(void) { 1415 return getppid(); 1416 } 1417 1418 uv_os_getpriority(uv_pid_t pid,int * priority)1419 int uv_os_getpriority(uv_pid_t pid, int* priority) { 1420 int r; 1421 1422 if (priority == NULL) 1423 return UV_EINVAL; 1424 1425 errno = 0; 1426 r = getpriority(PRIO_PROCESS, (int) pid); 1427 1428 if (r == -1 && errno != 0) 1429 return UV__ERR(errno); 1430 1431 *priority = r; 1432 return 0; 1433 } 1434 1435 uv_os_setpriority(uv_pid_t pid,int priority)1436 int uv_os_setpriority(uv_pid_t pid, int priority) { 1437 if (priority < UV_PRIORITY_HIGHEST || priority > UV_PRIORITY_LOW) 1438 return UV_EINVAL; 1439 1440 if (setpriority(PRIO_PROCESS, (int) pid, priority) != 0) 1441 return UV__ERR(errno); 1442 1443 return 0; 1444 } 1445 1446 uv_os_uname(uv_utsname_t * buffer)1447 int uv_os_uname(uv_utsname_t* buffer) { 1448 struct utsname buf; 1449 int r; 1450 1451 if (buffer == NULL) 1452 return UV_EINVAL; 1453 1454 if (uname(&buf) == -1) { 1455 r = UV__ERR(errno); 1456 goto error; 1457 } 1458 1459 r = uv__strscpy(buffer->sysname, buf.sysname, sizeof(buffer->sysname)); 1460 if (r == UV_E2BIG) 1461 goto error; 1462 1463 #ifdef _AIX 1464 r = snprintf(buffer->release, 1465 sizeof(buffer->release), 1466 "%s.%s", 1467 buf.version, 1468 buf.release); 1469 if (r >= sizeof(buffer->release)) { 1470 r = UV_E2BIG; 1471 goto error; 1472 } 1473 #else 1474 r = uv__strscpy(buffer->release, buf.release, sizeof(buffer->release)); 1475 if (r == UV_E2BIG) 1476 goto error; 1477 #endif 1478 1479 r = uv__strscpy(buffer->version, buf.version, sizeof(buffer->version)); 1480 if (r == UV_E2BIG) 1481 goto error; 1482 1483 #if defined(_AIX) || defined(__PASE__) 1484 r = uv__strscpy(buffer->machine, "ppc64", sizeof(buffer->machine)); 1485 #else 1486 r = uv__strscpy(buffer->machine, buf.machine, sizeof(buffer->machine)); 1487 #endif 1488 1489 if (r == UV_E2BIG) 1490 goto error; 1491 1492 return 0; 1493 1494 error: 1495 buffer->sysname[0] = '\0'; 1496 buffer->release[0] = '\0'; 1497 buffer->version[0] = '\0'; 1498 buffer->machine[0] = '\0'; 1499 return r; 1500 } 1501 uv__getsockpeername(const uv_handle_t * handle,uv__peersockfunc func,struct sockaddr * name,int * namelen)1502 int uv__getsockpeername(const uv_handle_t* handle, 1503 uv__peersockfunc func, 1504 struct sockaddr* name, 1505 int* namelen) { 1506 socklen_t socklen; 1507 uv_os_fd_t fd; 1508 int r; 1509 1510 r = uv_fileno(handle, &fd); 1511 if (r < 0) 1512 return r; 1513 1514 /* sizeof(socklen_t) != sizeof(int) on some systems. */ 1515 socklen = (socklen_t) *namelen; 1516 1517 if (func(fd, name, &socklen)) 1518 return UV__ERR(errno); 1519 1520 *namelen = (int) socklen; 1521 return 0; 1522 } 1523 uv_gettimeofday(uv_timeval64_t * tv)1524 int uv_gettimeofday(uv_timeval64_t* tv) { 1525 struct timeval time; 1526 1527 if (tv == NULL) 1528 return UV_EINVAL; 1529 1530 if (gettimeofday(&time, NULL) != 0) 1531 return UV__ERR(errno); 1532 1533 tv->tv_sec = (int64_t) time.tv_sec; 1534 tv->tv_usec = (int32_t) time.tv_usec; 1535 return 0; 1536 } 1537 uv_sleep(unsigned int msec)1538 void uv_sleep(unsigned int msec) { 1539 struct timespec timeout; 1540 int rc; 1541 1542 timeout.tv_sec = msec / 1000; 1543 timeout.tv_nsec = (msec % 1000) * 1000 * 1000; 1544 1545 do 1546 rc = nanosleep(&timeout, &timeout); 1547 while (rc == -1 && errno == EINTR); 1548 1549 assert(rc == 0); 1550 } 1551 uv__search_path(const char * prog,char * buf,size_t * buflen)1552 int uv__search_path(const char* prog, char* buf, size_t* buflen) { 1553 char abspath[UV__PATH_MAX]; 1554 size_t abspath_size; 1555 char trypath[UV__PATH_MAX]; 1556 char* cloned_path; 1557 char* path_env; 1558 char* token; 1559 1560 if (buf == NULL || buflen == NULL || *buflen == 0) 1561 return UV_EINVAL; 1562 1563 /* 1564 * Possibilities for prog: 1565 * i) an absolute path such as: /home/user/myprojects/nodejs/node 1566 * ii) a relative path such as: ./node or ../myprojects/nodejs/node 1567 * iii) a bare filename such as "node", after exporting PATH variable 1568 * to its location. 1569 */ 1570 1571 /* Case i) and ii) absolute or relative paths */ 1572 if (strchr(prog, '/') != NULL) { 1573 if (realpath(prog, abspath) != abspath) 1574 return UV__ERR(errno); 1575 1576 abspath_size = strlen(abspath); 1577 1578 *buflen -= 1; 1579 if (*buflen > abspath_size) 1580 *buflen = abspath_size; 1581 1582 memcpy(buf, abspath, *buflen); 1583 buf[*buflen] = '\0'; 1584 1585 return 0; 1586 } 1587 1588 /* Case iii). Search PATH environment variable */ 1589 cloned_path = NULL; 1590 token = NULL; 1591 path_env = getenv("PATH"); 1592 1593 if (path_env == NULL) 1594 return UV_EINVAL; 1595 1596 cloned_path = uv__strdup(path_env); 1597 if (cloned_path == NULL) 1598 return UV_ENOMEM; 1599 1600 token = strtok(cloned_path, ":"); 1601 while (token != NULL) { 1602 snprintf(trypath, sizeof(trypath) - 1, "%s/%s", token, prog); 1603 if (realpath(trypath, abspath) == abspath) { 1604 /* Check the match is executable */ 1605 if (access(abspath, X_OK) == 0) { 1606 abspath_size = strlen(abspath); 1607 1608 *buflen -= 1; 1609 if (*buflen > abspath_size) 1610 *buflen = abspath_size; 1611 1612 memcpy(buf, abspath, *buflen); 1613 buf[*buflen] = '\0'; 1614 1615 uv__free(cloned_path); 1616 return 0; 1617 } 1618 } 1619 token = strtok(NULL, ":"); 1620 } 1621 uv__free(cloned_path); 1622 1623 /* Out of tokens (path entries), and no match found */ 1624 return UV_EINVAL; 1625 } 1626 1627 uv_available_parallelism(void)1628 unsigned int uv_available_parallelism(void) { 1629 #ifdef __linux__ 1630 cpu_set_t set; 1631 long rc; 1632 1633 memset(&set, 0, sizeof(set)); 1634 1635 /* sysconf(_SC_NPROCESSORS_ONLN) in musl calls sched_getaffinity() but in 1636 * glibc it's... complicated... so for consistency try sched_getaffinity() 1637 * before falling back to sysconf(_SC_NPROCESSORS_ONLN). 1638 */ 1639 if (0 == sched_getaffinity(0, sizeof(set), &set)) 1640 rc = CPU_COUNT(&set); 1641 else 1642 rc = sysconf(_SC_NPROCESSORS_ONLN); 1643 1644 if (rc < 1) 1645 rc = 1; 1646 1647 return (unsigned) rc; 1648 #elif defined(__MVS__) 1649 return 1; /* TODO(bnoordhuis) Read from CSD_NUMBER_ONLINE_CPUS? */ 1650 #else /* __linux__ */ 1651 long rc; 1652 1653 rc = sysconf(_SC_NPROCESSORS_ONLN); 1654 if (rc < 1) 1655 rc = 1; 1656 1657 return (unsigned) rc; 1658 #endif /* __linux__ */ 1659 } 1660