/*
* Copyright (c) 2019-2022 David Timber <dxdt@dev.snart.me>
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
#include <errno.h>

#include "util_rt.h"
#include "pth.h"


void prne_init_worker (prne_worker_t *w) {
	prne_memzero(w, sizeof(prne_worker_t));
}

void prne_free_worker (prne_worker_t *w) {
	if (w == NULL) {
		return;
	}

	if (w->ctx != NULL) {
		prne_assert(w->free_ctx != NULL);
		w->free_ctx(w->ctx);
	}
	if (w->pth != NULL) {
		pth_abort(w->pth);
	}
	pth_attr_destroy(w->attr);

	prne_memzero(w, sizeof(prne_worker_t));
}

void prne_fin_worker (prne_worker_t *w) {
	if (w->fin != NULL) {
		w->fin(w->ctx);
	}
}

int prne_pth_poll (
	struct pollfd *pfd,
	const nfds_t nfs,
	const int timeout,
	pth_event_t ev)
{
	struct pollfd my_pfd[nfs];
	nfds_t p;
	int ret;

	p = 0;
	for (nfds_t i = 0; i < nfs; i += 1) {
		if (pfd[i].fd >= FD_SETSIZE) {
			errno = EINVAL;
			return -1;
		}
		if (0 <= pfd[i].fd) {
			my_pfd[p].fd = pfd[i].fd;
			my_pfd[p].events = pfd[i].events;
			p += 1;
		}
	}
	if (p == 0) {
		return 0;
	}

	ret = pth_poll_ev(my_pfd, p, timeout, ev);
	if (ret >= 0) {
		p = 0;
		for (nfds_t i = 0; i < nfs; i += 1) {
			if (0 <= pfd[i].fd) {
				pfd[i].revents = my_pfd[p].revents;
				p += 1;
			}
			else {
				pfd[i].revents = 0;
			}
		}
	}

	return ret;
}

void prne_pth_cv_notify (pth_mutex_t *lock, pth_cond_t *cond, bool broadcast) {
	prne_dbgtrap(pth_mutex_acquire(lock, FALSE, NULL));
	prne_dbgtrap(pth_cond_notify(cond, broadcast));
	pth_mutex_release(lock);
}

pth_time_t prne_pth_tstimeout (const struct timespec ts) {
	return pth_timeout(ts.tv_sec, ts.tv_nsec / 1000);
}

void prne_pth_reset_timer (pth_event_t *ev, const struct timespec *ts) {
	pth_event_free(*ev, FALSE);
	if (ts != NULL) {
		*ev = pth_event(
			PTH_EVENT_TIME,
			prne_pth_tstimeout(*ts));
		prne_assert(*ev != NULL);
	}
}