/*
 * Copyright (C) 2014-2016
 *     Olivier Gayot <ogayot@baylibre.com>
 *     Bartosz Golaszewski <bgolaszewski@baylibre.com>
 *     Olivier Gayot <olivier.gayot@sigexec.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <sys/select.h>

#include "misc.h"

#define SELECT_INIT(FDSET, SOCK) \
    do { \
        FD_ZERO(&(FDSET)); \
        FD_SET((SOCK), &(FDSET)); \
    } while (0)

#define SELECT_LOOP(RETVAL, CALL, TV) \
    do { \
        do { \
            (RETVAL) = (CALL); \
            if ((((RETVAL) < 0) && (errno != EINTR)) || (RETVAL > 0)) { \
                break; \
            } \
        } while ((TV)->tv_sec || (TV)->tv_usec); \
    } while(0)

int caod_fd_wrready(int sock, struct timeval *tv)
{
    fd_set wr_set;
    int r;

    SELECT_INIT(wr_set, sock);
    SELECT_LOOP(r, select(sock+1, NULL, &wr_set, NULL, tv), tv);

    return r;
}

int caod_fd_rdready(int sock, struct timeval *tv)
{
    fd_set rd_set;
    int r;

    SELECT_INIT(rd_set, sock);
    SELECT_LOOP(r, select(sock+1, &rd_set, NULL, NULL, tv), tv);

    return r;
}