aboutsummaryrefslogtreecommitdiff
path: root/src/worker.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/worker.c')
-rw-r--r--src/worker.c49
1 files changed, 49 insertions, 0 deletions
diff --git a/src/worker.c b/src/worker.c
new file mode 100644
index 0000000..d99884a
--- /dev/null
+++ b/src/worker.c
@@ -0,0 +1,49 @@
+#include "worker.h"
+
+#include <stdlib.h>
+
+
+static void def_free_func (prne_worker_sched_req_t *wsr) {
+ free(wsr->pollfd_arr);
+ wsr->pollfd_arr = NULL;
+ wsr->pollfd_arr_size = 0;
+}
+
+static bool def_alloc_func (prne_worker_sched_req_t *wsr, const size_t ny_size) {
+ if (ny_size == 0) {
+ def_free_func(wsr);
+ }
+ else if (ny_size != wsr->pollfd_arr_size) {
+ void *ny_buf = realloc(wsr->pollfd_arr, ny_size * sizeof(struct pollfd));
+
+ if (ny_buf == NULL) {
+ return false;
+ }
+ wsr->pollfd_arr = (struct pollfd*)ny_buf;
+ wsr->pollfd_arr_size = ny_size;
+ }
+
+ return true;
+}
+
+static prne_worker_sched_req_mem_func_t def_mem_func = { def_alloc_func, def_free_func, NULL };
+
+
+bool prne_init_worker_sched_req (prne_worker_sched_req_t *wsr, prne_worker_sched_req_mem_func_t *in_mem_func) {
+ prne_worker_sched_req_t ret;
+
+ ret.pollfd_arr_size = 0;
+ ret.pollfd_arr = NULL;
+ ret.timeout.tv_sec = 0;
+ ret.timeout.tv_nsec = 0;
+ ret.mem_func = *(in_mem_func != NULL ? in_mem_func : &def_mem_func);
+ ret.flags = PRNE_WORKER_SCHED_FLAG_NONE;
+ ret.pollfd_ready = false;
+
+ if (!ret.mem_func.alloc(&ret, 0)) {
+ return false;
+ }
+
+ *wsr = ret;
+ return true;
+}