src/libguac/wait-fd.c

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

#include "config.h"

#ifdef ENABLE_WINSOCK
#    include <winsock2.h>
#else
#    ifdef HAVE_POLL
#        include <poll.h>
#    else
#        include <sys/select.h>
#    endif
#endif

#ifdef HAVE_POLL
int guac_wait_for_fd(int fd, int usec_timeout) {

    /* Initialize with single underlying file descriptor */
    struct pollfd fds[1] = {{
        .fd      = fd,
        .events  = POLLIN,
        .revents = 0
    }};

    /* No timeout if usec_timeout is negative */
    if (usec_timeout < 0)
        return poll(fds, 1, -1);

    /* Handle timeout if specified, rounding up to poll()'s granularity */
    return poll(fds, 1, (usec_timeout + 999) / 1000);

}
#else
int guac_wait_for_fd(int fd, int usec_timeout) {

    fd_set fds;

    /* Initialize fd_set with single underlying file descriptor */
    FD_ZERO(&fds);
    FD_SET(fd, &fds);

    /* No timeout if usec_timeout is negative */
    if (usec_timeout < 0)
        return select(fd + 1, &fds, NULL, NULL, NULL); 

    /* Handle timeout if specified */
    struct timeval timeout = {
        .tv_sec  = usec_timeout / 1000000,
        .tv_usec = usec_timeout % 1000000
    };

    return select(fd + 1, &fds, NULL, NULL, &timeout);

}
#endif