Blame axfer/waiter.c

Packit 229ac0
// SPDX-License-Identifier: GPL-2.0
Packit 229ac0
//
Packit 229ac0
// waiter.c - I/O event waiter.
Packit 229ac0
//
Packit 229ac0
// Copyright (c) 2018 Takashi Sakamoto <o-takashi@sakamocchi.jp>
Packit 229ac0
//
Packit 229ac0
// Licensed under the terms of the GNU General Public License, version 2.
Packit 229ac0
Packit 229ac0
#include "waiter.h"
Packit 229ac0
Packit 229ac0
#include <stdlib.h>
Packit 229ac0
#include <string.h>
Packit 229ac0
#include <errno.h>
Packit 229ac0
Packit 229ac0
#include "misc.h"
Packit 229ac0
Packit 229ac0
static const char *const waiter_type_labels[] = {
Packit 229ac0
	[WAITER_TYPE_DEFAULT] = "default",
Packit 229ac0
	[WAITER_TYPE_POLL] = "poll",
Packit 229ac0
	[WAITER_TYPE_SELECT] = "select",
Packit 229ac0
	[WAITER_TYPE_EPOLL] = "epoll",
Packit 229ac0
};
Packit 229ac0
Packit 229ac0
enum waiter_type waiter_type_from_label(const char *label)
Packit 229ac0
{
Packit 229ac0
	int i;
Packit 229ac0
Packit 229ac0
	for (i = 0; i < ARRAY_SIZE(waiter_type_labels); ++i) {
Packit 229ac0
		if (!strcmp(waiter_type_labels[i], label))
Packit 229ac0
			return i;
Packit 229ac0
	}
Packit 229ac0
Packit 229ac0
	return WAITER_TYPE_DEFAULT;
Packit 229ac0
}
Packit 229ac0
Packit 229ac0
const char *waiter_label_from_type(enum waiter_type type)
Packit 229ac0
{
Packit 229ac0
	return waiter_type_labels[type];
Packit 229ac0
}
Packit 229ac0
Packit 229ac0
int waiter_context_init(struct waiter_context *waiter,
Packit 229ac0
			enum waiter_type type, unsigned int pfd_count)
Packit 229ac0
{
Packit 229ac0
	struct {
Packit 229ac0
		enum waiter_type type;
Packit 229ac0
		const struct waiter_data *waiter;
Packit 229ac0
	} entries[] = {
Packit 229ac0
		{WAITER_TYPE_POLL,	&waiter_poll},
Packit 229ac0
		{WAITER_TYPE_SELECT,	&waiter_select},
Packit 229ac0
		{WAITER_TYPE_EPOLL,	&waiter_epoll},
Packit 229ac0
	};
Packit 229ac0
	int i;
Packit 229ac0
Packit 229ac0
	if (pfd_count == 0)
Packit 229ac0
		return -EINVAL;
Packit 229ac0
Packit 229ac0
	for (i = 0; i < ARRAY_SIZE(entries); ++i) {
Packit 229ac0
		if (entries[i].type == type)
Packit 229ac0
			break;
Packit 229ac0
	}
Packit 229ac0
	if (i == ARRAY_SIZE(entries))
Packit 229ac0
		return -EINVAL;
Packit 229ac0
Packit 229ac0
	waiter->private_data = malloc(entries[i].waiter->private_size);
Packit 229ac0
	if (waiter->private_data == NULL)
Packit 229ac0
		return -ENOMEM;
Packit 229ac0
	memset(waiter->private_data, 0, entries[i].waiter->private_size);
Packit 229ac0
Packit 229ac0
	waiter->type = type;
Packit 229ac0
	waiter->ops = &entries[i].waiter->ops;
Packit 229ac0
Packit 229ac0
	waiter->pfds = calloc(pfd_count, sizeof(*waiter->pfds));
Packit 229ac0
	if (waiter->pfds == NULL)
Packit 229ac0
		return -ENOMEM;
Packit 229ac0
	waiter->pfd_count = pfd_count;
Packit 229ac0
Packit 229ac0
	return 0;
Packit 229ac0
}
Packit 229ac0
Packit 229ac0
int waiter_context_prepare(struct waiter_context *waiter)
Packit 229ac0
{
Packit 229ac0
	return waiter->ops->prepare(waiter);
Packit 229ac0
}
Packit 229ac0
Packit 229ac0
int waiter_context_wait_event(struct waiter_context *waiter,
Packit 229ac0
				int timeout_msec)
Packit 229ac0
{
Packit 229ac0
	return waiter->ops->wait_event(waiter, timeout_msec);
Packit 229ac0
}
Packit 229ac0
Packit 229ac0
void waiter_context_release(struct waiter_context *waiter)
Packit 229ac0
{
Packit 229ac0
	waiter->ops->release(waiter);
Packit 229ac0
}
Packit 229ac0
Packit 229ac0
void waiter_context_destroy(struct waiter_context *waiter)
Packit 229ac0
{
Packit 229ac0
	if (waiter->pfds)
Packit 229ac0
		free(waiter->pfds);
Packit 229ac0
	waiter->pfd_count = 0;
Packit 229ac0
	if (waiter->private_data)
Packit 229ac0
		free(waiter->private_data);
Packit 229ac0
	waiter->private_data = NULL;
Packit 229ac0
}