1 files changed, 65 insertions, 0 deletions
diff --git a/misc.c b/misc.c
new file mode 100644
index 0000000..9570e44
--- /dev/null
+++ b/misc.c
@@ -0,0 +1,65 @@
+/*
+ * 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;
+}