// Copyright (c) 2015 Ryan Prichard
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to
// deal in the Software without restriction, including without limitation the
// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
// sell copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
// IN THE SOFTWARE.

#include "Util.h"

#include <assert.h>
#include <errno.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>

#include "../shared/DebugClient.h"

// Write the entire buffer, restarting it as necessary.
bool writeAll(int fd, const void *buffer, size_t size) {
    size_t written = 0;
    while (written < size) {
        int ret = write(fd,
                        reinterpret_cast<const char*>(buffer) + written,
                        size - written);
        if (ret == -1 && errno == EINTR) {
            continue;
        }
        if (ret <= 0) {
            trace("write failed: "
                "fd=%d errno=%d size=%u written=%d ret=%d",
                fd,
                errno,
                static_cast<unsigned int>(size),
                static_cast<unsigned int>(written),
                ret);
            return false;
        }
        assert(static_cast<size_t>(ret) <= size - written);
        written += ret;
    }
    assert(written == size);
    return true;
}

bool writeStr(int fd, const char *str) {
    return writeAll(fd, str, strlen(str));
}

void selectWrapper(const char *diagName, int nfds, fd_set *readfds) {
    int ret = select(nfds, readfds, NULL, NULL, NULL);
    if (ret < 0) {
        if (errno == EINTR) {
            FD_ZERO(readfds);
            return;
        }
#ifdef WINPTY_TARGET_MSYS1
        // The select system call sometimes fails with EAGAIN instead of EINTR.
        // This apparantly only happens with the old Cygwin fork "MSYS" used in
        // the mingw.org project.  select is not supposed to fail with EAGAIN,
        // and EAGAIN does not make much sense as an error code.  (The whole
        // point of select is to block.)
        if (errno == EAGAIN) {
            trace("%s select returned EAGAIN: interpreting like EINTR",
                diagName);
            FD_ZERO(readfds);
            return;
        }
#endif
        fprintf(stderr, "Internal error: %s select failed: "
            "error %d", diagName, errno);
        abort();
    }
}