// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause

/*
 * Helpers for handling qevents.
 */

#include <stdio.h>
#include <string.h>

#include "tc_qevent.h"
#include "utils.h"

void qevents_init(struct qevent_util *qevents)
{
	if (!qevents)
		return;

	for (; qevents->id; qevents++)
		memset(qevents->data, 0, qevents->data_size);
}

int qevent_parse(struct qevent_util *qevents, int *p_argc, char ***p_argv)
{
	char **argv = *p_argv;
	int argc = *p_argc;
	const char *name = *argv;
	int err;

	if (!qevents)
		goto out;

	for (; qevents->id; qevents++) {
		if (strcmp(name, qevents->id) == 0) {
			NEXT_ARG();
			err = qevents->parse_qevent(qevents, &argc, &argv);
			if (err)
				return err;

			*p_argc = argc;
			*p_argv = argv;
			return 0;
		}
	}

out:
	fprintf(stderr, "Unknown qevent `%s'\n", name);
	return -1;
}

int qevents_read(struct qevent_util *qevents, struct rtattr **tb)
{
	int err;

	if (!qevents)
		return 0;

	for (; qevents->id; qevents++) {
		if (tb[qevents->attr]) {
			err = qevents->read_qevent(qevents, tb);
			if (err)
				return err;
		}
	}

	return 0;
}

void qevents_print(struct qevent_util *qevents, FILE *f)
{
	int first = true;

	if (!qevents)
		return;

	for (; qevents->id; qevents++) {
		struct qevent_base *qeb = qevents->data;

		if (qeb->block_idx) {
			if (first) {
				open_json_array(PRINT_JSON, "qevents");
				first = false;
			}

			open_json_object(NULL);
			print_string(PRINT_ANY, "kind", "qevent %s", qevents->id);
			qevents->print_qevent(qevents, f);
			print_string(PRINT_FP, NULL, "%s", " ");
			close_json_object();
		}
	}

	if (!first)
		close_json_array(PRINT_ANY, "");
}

bool qevents_have_block(struct qevent_util *qevents, __u32 block_idx)
{
	if (!qevents)
		return false;

	for (; qevents->id; qevents++) {
		struct qevent_base *qeb = qevents->data;

		if (qeb->block_idx == block_idx)
			return true;
	}

	return false;
}

int qevents_dump(struct qevent_util *qevents, struct nlmsghdr *n)
{
	int err;

	if (!qevents)
		return 0;

	for (; qevents->id; qevents++) {
		struct qevent_base *qeb = qevents->data;

		if (qeb->block_idx) {
			err = qevents->dump_qevent(qevents, n);
			if (err)
				return err;
		}
	}

	return 0;
}

static int parse_block_idx(const char *arg, struct qevent_base *qeb)
{
	if (qeb->block_idx) {
		fprintf(stderr, "Qevent block index already specified\n");
		return -1;
	}

	if (get_unsigned(&qeb->block_idx, arg, 10) || !qeb->block_idx) {
		fprintf(stderr, "Illegal qevent block index\n");
		return -1;
	}

	return 0;
}

static int read_block_idx(struct rtattr *attr, struct qevent_base *qeb)
{
	if (qeb->block_idx) {
		fprintf(stderr, "Qevent block index already specified\n");
		return -1;
	}

	qeb->block_idx = rta_getattr_u32(attr);
	if (!qeb->block_idx) {
		fprintf(stderr, "Illegal qevent block index\n");
		return -1;
	}

	return 0;
}

static void print_block_idx(FILE *f, __u32 block_idx)
{
	print_uint(PRINT_ANY, "block", " block %u", block_idx);
}

int qevent_parse_plain(struct qevent_util *qu, int *p_argc, char ***p_argv)
{
	struct qevent_plain *qe = qu->data;
	char **argv = *p_argv;
	int argc = *p_argc;

	if (qe->base.block_idx) {
		fprintf(stderr, "Duplicate qevent\n");
		return -1;
	}

	while (argc > 0) {
		if (strcmp(*argv, "block") == 0) {
			NEXT_ARG();
			if (parse_block_idx(*argv, &qe->base))
				return -1;
		} else {
			break;
		}
		NEXT_ARG_FWD();
	}

	if (!qe->base.block_idx) {
		fprintf(stderr, "Unspecified qevent block index\n");
		return -1;
	}

	*p_argc = argc;
	*p_argv = argv;
	return 0;
}

int qevent_read_plain(struct qevent_util *qu, struct rtattr **tb)
{
	struct qevent_plain *qe = qu->data;

	return read_block_idx(tb[qu->attr], &qe->base);
}

void qevent_print_plain(struct qevent_util *qu, FILE *f)
{
	struct qevent_plain *qe = qu->data;

	print_block_idx(f, qe->base.block_idx);
}

int qevent_dump_plain(struct qevent_util *qu, struct nlmsghdr *n)
{
	struct qevent_plain *qe = qu->data;

	return addattr32(n, 1024, qu->attr, qe->base.block_idx);
}