node_modules/node-pty/deps/winpty/src/unix-adapter/Util.cc

   1 // Copyright (c) 2015 Ryan Prichard
   2 //
   3 // Permission is hereby granted, free of charge, to any person obtaining a copy
   4 // of this software and associated documentation files (the "Software"), to
   5 // deal in the Software without restriction, including without limitation the
   6 // rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
   7 // sell copies of the Software, and to permit persons to whom the Software is
   8 // furnished to do so, subject to the following conditions:
   9 //
  10 // The above copyright notice and this permission notice shall be included in
  11 // all copies or substantial portions of the Software.
  12 //
  13 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  14 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  15 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  16 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  17 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  18 // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  19 // IN THE SOFTWARE.
  20
  21 #include "Util.h"
  22
  23 #include <assert.h>
  24 #include <errno.h>
  25 #include <stdio.h>
  26 #include <string.h>
  27 #include <unistd.h>
  28
  29 #include "../shared/DebugClient.h"
  30
  31 // Write the entire buffer, restarting it as necessary.
  32 bool writeAll(int fd, const void *buffer, size_t size) {
  33     size_t written = 0;
  34     while (written < size) {
  35         int ret = write(fd,
  36                         reinterpret_cast<const char*>(buffer) + written,
  37                         size - written);
  38         if (ret == -1 && errno == EINTR) {
  39             continue;
  40         }
  41         if (ret <= 0) {
  42             trace("write failed: "
  43                 "fd=%d errno=%d size=%u written=%d ret=%d",
  44                 fd,
  45                 errno,
  46                 static_cast<unsigned int>(size),
  47                 static_cast<unsigned int>(written),
  48                 ret);
  49             return false;
  50         }
  51         assert(static_cast<size_t>(ret) <= size - written);
  52         written += ret;
  53     }
  54     assert(written == size);
  55     return true;
  56 }
  57
  58 bool writeStr(int fd, const char *str) {
  59     return writeAll(fd, str, strlen(str));
  60 }
  61
  62 void selectWrapper(const char *diagName, int nfds, fd_set *readfds) {
  63     int ret = select(nfds, readfds, NULL, NULL, NULL);
  64     if (ret < 0) {
  65         if (errno == EINTR) {
  66             FD_ZERO(readfds);
  67             return;
  68         }
  69 #ifdef WINPTY_TARGET_MSYS1
  70         // The select system call sometimes fails with EAGAIN instead of EINTR.
  71         // This apparantly only happens with the old Cygwin fork "MSYS" used in
  72         // the mingw.org project.  select is not supposed to fail with EAGAIN,
  73         // and EAGAIN does not make much sense as an error code.  (The whole
  74         // point of select is to block.)
  75         if (errno == EAGAIN) {
  76             trace("%s select returned EAGAIN: interpreting like EINTR",
  77                 diagName);
  78             FD_ZERO(readfds);
  79             return;
  80         }
  81 #endif
  82         fprintf(stderr, "Internal error: %s select failed: "
  83             "error %d", diagName, errno);
  84         abort();
  85     }
  86 }