1 files changed, 150 insertions, 0 deletions
diff --git a/src/os.c b/src/os.c
new file mode 100644
index 0000000..addcdac
--- /dev/null
+++ b/src/os.c
@@ -0,0 +1,150 @@
+/*
+ * Copyright 2019-2021 OARC, Inc.
+ * Copyright 2017-2018 Akamai Technologies
+ * Copyright 2006-2016 Nominum, Inc.
+ * All rights reserved.
+ *
+ * Licensed 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"
+
+#include "os.h"
+
+#include "log.h"
+#include "util.h"
+
+#include <errno.h>
+#include <signal.h>
+#include <inttypes.h>
+#include <stdbool.h>
+#include <stdlib.h>
+#include <string.h>
+#include <poll.h>
+
+void perf_os_blocksignal(int sig, bool block)
+{
+    sigset_t sset;
+    int      op;
+
+    op = block ? SIG_BLOCK : SIG_UNBLOCK;
+
+    if (sigemptyset(&sset) < 0 || sigaddset(&sset, sig) < 0 || pthread_sigmask(op, &sset, NULL) < 0) {
+        char __s[256];
+        perf_log_fatal("pthread_sigmask: %s", perf_strerror_r(errno, __s, sizeof(__s)));
+    }
+}
+
+void perf_os_handlesignal(int sig, void (*handler)(int))
+{
+    struct sigaction sa;
+
+    memset(&sa, 0, sizeof(sa));
+    sa.sa_handler = handler;
+
+    if (sigfillset(&sa.sa_mask) < 0 || sigaction(sig, &sa, NULL) < 0) {
+        char __s[256];
+        perf_log_fatal("sigaction: %s", perf_strerror_r(errno, __s, sizeof(__s)));
+    }
+}
+
+perf_result_t
+perf_os_waituntilreadable(struct perf_net_socket* sock, int pipe_fd, int64_t timeout)
+{
+    return perf_os_waituntilanyreadable(sock, 1, pipe_fd, timeout);
+}
+
+perf_result_t
+perf_os_waituntilanyreadable(struct perf_net_socket* socks, unsigned int nfds, int pipe_fd,
+    int64_t timeout)
+{
+    struct pollfd fds[nfds + 1];
+    size_t        i;
+    int           to, n;
+
+    for (i = 0; i < nfds; i++) {
+        if (socks[i].have_more)
+            return (PERF_R_SUCCESS);
+
+        fds[i].fd     = socks[i].fd;
+        fds[i].events = POLLIN;
+    }
+
+    fds[nfds].fd     = pipe_fd;
+    fds[nfds].events = POLLIN;
+
+    if (timeout < 0) {
+        to = -1;
+    } else {
+        to = timeout / 1000;
+        if (timeout && !to) {
+            to = 1;
+        }
+    }
+
+    n = poll(fds, nfds + 1, to);
+    if (n < 0) {
+        if (errno != EINTR) {
+            char __s[256];
+            perf_log_fatal("select(): %s", perf_strerror_r(errno, __s, sizeof(__s)));
+        }
+        return (PERF_R_CANCELED);
+    } else if (n == 0) {
+        return (PERF_R_TIMEDOUT);
+    } else if (fds[nfds].revents & POLLIN) {
+        return (PERF_R_CANCELED);
+    } else {
+        return (PERF_R_SUCCESS);
+    }
+}
+
+perf_result_t
+perf_os_waituntilanywritable(struct perf_net_socket* socks, unsigned int nfds, int pipe_fd,
+    int64_t timeout)
+{
+    struct pollfd fds[nfds + 1];
+    size_t        i;
+    int           to, n;
+
+    for (i = 0; i < nfds; i++) {
+        fds[i].fd     = socks[i].fd;
+        fds[i].events = POLLOUT;
+    }
+
+    fds[nfds].fd     = pipe_fd;
+    fds[nfds].events = POLLIN;
+
+    if (timeout < 0) {
+        to = -1;
+    } else {
+        to = timeout / 1000;
+        if (timeout && !to) {
+            to = 1;
+        }
+    }
+
+    n = poll(fds, nfds + 1, to);
+    if (n < 0) {
+        if (errno != EINTR) {
+            char __s[256];
+            perf_log_fatal("select(): %s", perf_strerror_r(errno, __s, sizeof(__s)));
+        }
+        return (PERF_R_CANCELED);
+    } else if (n == 0) {
+        return (PERF_R_TIMEDOUT);
+    } else if (fds[nfds].revents & POLLIN) {
+        return (PERF_R_CANCELED);
+    } else {
+        return (PERF_R_SUCCESS);
+    }
+}