/*
 * Copyright (c) 2008-2012 Patrick McHardy <kaber@trash.net>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 * Development of this code funded by Astaro AG (http://www.astaro.com/)
 */

#include <netlink/netfilter/nfnl.h>
#include <netlink/netfilter/netfilter.h>
#include <netlink/netfilter/nft_table.h>
#include <netlink/netfilter/nft_chain.h>
#include <netlink/netfilter/nft_rule.h>
#include <netlink/netfilter/nft_expr.h>
#include <netlink/netfilter/nft_data.h>
#include <netlink/netfilter/nft_setelem.h>
#include <netlink/netfilter/nft_set.h>
#include <linux/netfilter/nf_tables.h>

#include <nftables.h>
#include <netlink.h>
#include <expression.h>
#include <gmputil.h>
#include <utils.h>
#include <erec.h>

static struct nl_sock *nf_sock;

static void __init netlink_open_sock(void)
{
	nlmsg_set_default_size(65536);
	nf_sock = nl_socket_alloc();
	if (nf_sock == NULL)
		memory_allocation_error();

	nfnl_connect(nf_sock);
	nl_socket_set_nonblocking(nf_sock);
}

static void __exit netlink_close_sock(void)
{
	nl_socket_free(nf_sock);
}

static void netlink_set_callback(nl_recvmsg_msg_cb_t func, void *arg)
{
	nl_socket_modify_cb(nf_sock, NL_CB_VALID, NL_CB_CUSTOM, func, arg);
}

void netlink_dump_object(struct nl_object *obj)
{
#ifdef DEBUG
	struct nl_dump_params params = {
		.dp_fd		= stdout,
		.dp_type	= NL_DUMP_DETAILS,
	};

	if (!(debug_level & DEBUG_NETLINK))
		return;
	nl_object_dump(obj, &params);
#endif
}

static int netlink_io_error(struct netlink_ctx *ctx, const struct location *loc,
			    const char *fmt, ...)
{
	struct error_record *erec;
	va_list ap;

	if (loc == NULL)
		loc = &internal_location;

	va_start(ap, fmt);
	erec = erec_vcreate(EREC_ERROR, loc, fmt, ap);
	va_end(ap);
	erec_queue(erec, ctx->msgs);
	return -1;
}

struct nfnl_nft_table *alloc_nft_table(const struct handle *h)
{
	struct nfnl_nft_table *nlt;

	nlt = nfnl_nft_table_alloc();
	if (nlt == NULL)
		memory_allocation_error();
	nfnl_nft_table_set_family(nlt, h->family);
	if (h->table != NULL)
		nfnl_nft_table_set_name(nlt, h->table, strlen(h->table) + 1);
	return nlt;
}

struct nfnl_nft_chain *alloc_nft_chain(const struct handle *h)
{
	struct nfnl_nft_chain *nlc;

	nlc = nfnl_nft_chain_alloc();
	if (nlc == NULL)
		memory_allocation_error();
	nfnl_nft_chain_set_family(nlc, h->family);
	nfnl_nft_chain_set_table(nlc, h->table, strlen(h->table) + 1);
	if (h->handle != 0)
		nfnl_nft_chain_set_handle(nlc, h->handle);
	if (h->chain != NULL)
		nfnl_nft_chain_set_name(nlc, h->chain, strlen(h->chain) + 1);
	return nlc;
}

struct nfnl_nft_rule *alloc_nft_rule(const struct handle *h)
{
	struct nfnl_nft_rule *nlr;

	nlr = nfnl_nft_rule_alloc();
	if (nlr == NULL)
		memory_allocation_error();
	nfnl_nft_rule_set_family(nlr, h->family);
	nfnl_nft_rule_set_table(nlr, h->table, strlen(h->table) + 1);
	if (h->chain != NULL)
		nfnl_nft_rule_set_chain(nlr, h->chain, strlen(h->chain) + 1);
	if (h->handle)
		nfnl_nft_rule_set_handle(nlr, h->handle);
	return nlr;
}

struct nfnl_nft_expr *alloc_nft_expr(int (*init)(struct nfnl_nft_expr *))
{
	struct nfnl_nft_expr *nle;

	nle = nfnl_nft_expr_alloc();
	if (nle == NULL || init(nle) != 0)
		memory_allocation_error();
	return nle;
}

struct nfnl_nft_set *alloc_nft_set(const struct handle *h)
{
	struct nfnl_nft_set *nls;

	nls = nfnl_nft_set_alloc();
	if (nls == NULL)
		memory_allocation_error();
	nfnl_nft_set_set_family(nls, h->family);
	nfnl_nft_set_set_table(nls, h->table, strlen(h->table) + 1);
	if (h->set != NULL)
		nfnl_nft_set_set_name(nls, h->set, strlen(h->set) + 1);
	return nls;
}

static struct nfnl_nft_setelem *alloc_nft_setelem(const struct expr *expr)
{
	struct nfnl_nft_setelem *nlse;

	nlse = nfnl_nft_setelem_alloc();
	if (nlse == NULL)
		memory_allocation_error();

	if (expr->ops->type == EXPR_VALUE || expr->flags & EXPR_F_INTERVAL_END)
		nfnl_nft_setelem_set_key(nlse, netlink_gen_data(expr));
	else {
		assert(expr->ops->type == EXPR_MAPPING);
		nfnl_nft_setelem_set_key(nlse, netlink_gen_data(expr->left));
		nfnl_nft_setelem_set_data(nlse, netlink_gen_data(expr->right));
	}

	if (expr->flags & EXPR_F_INTERVAL_END)
		nfnl_nft_setelem_set_flags(nlse, NFT_SET_ELEM_INTERVAL_END);

	return nlse;
}

struct nfnl_nft_data *alloc_nft_data(const void *data, unsigned int len)
{
	struct nfnl_nft_data *nld;

	assert(len > 0);
	nld = nfnl_nft_data_alloc(data, len);
	if (nld == NULL)
		memory_allocation_error();
	return nld;
}

struct nfnl_nft_data *netlink_gen_raw_data(const mpz_t value,
					   enum byteorder byteorder,
					   unsigned int len)
{
	unsigned char data[len];

	mpz_export_data(data, value, byteorder, len);
	return alloc_nft_data(data, len);
}

static struct nfnl_nft_data *netlink_gen_concat_data(const struct expr *expr)
{
	const struct expr *i;
	unsigned int len, offset;

	len = 0;
	list_for_each_entry(i, &expr->expressions, list)
		len += i->len;

	if (1) {
		unsigned char data[len / BITS_PER_BYTE];

		offset = 0;
		list_for_each_entry(i, &expr->expressions, list) {
			assert(i->ops->type == EXPR_VALUE);
			mpz_export_data(data + offset, i->value, i->byteorder,
					i->len / BITS_PER_BYTE);
			offset += i->len / BITS_PER_BYTE;
		}
	
		return alloc_nft_data(data, len / BITS_PER_BYTE);
	}
}

static struct nfnl_nft_data *netlink_gen_constant_data(const struct expr *expr)
{
	assert(expr->ops->type == EXPR_VALUE);
	return netlink_gen_raw_data(expr->value, expr->byteorder,
				    div_round_up(expr->len, BITS_PER_BYTE));
}

static struct nfnl_nft_data *netlink_gen_verdict(const struct expr *expr)
{
	struct nfnl_nft_data *verdict;

	verdict = nfnl_nft_verdict_alloc();
	nfnl_nft_verdict_set_verdict(verdict, expr->verdict);

	switch (expr->verdict) {
	case NFT_JUMP:
	case NFT_GOTO:
		nfnl_nft_verdict_set_chain(verdict, expr->chain);
		break;
	}

	return verdict;
}

struct nfnl_nft_data *netlink_gen_data(const struct expr *expr)
{
	switch (expr->ops->type) {
	case EXPR_VALUE:
		return netlink_gen_constant_data(expr);
	case EXPR_CONCAT:
		return netlink_gen_concat_data(expr);
	case EXPR_VERDICT:
		return netlink_gen_verdict(expr);
	default:
		BUG("invalid data expression type %s\n", expr->ops->name);
	}
}

struct expr *netlink_alloc_value(const struct location *loc,
				 const struct nfnl_nft_data *nld)
{
	return constant_expr_alloc(loc, &invalid_type, BYTEORDER_INVALID,
				   nfnl_nft_data_get_size(nld) * BITS_PER_BYTE,
				   nfnl_nft_data_get(nld));
}

static struct expr *netlink_alloc_verdict(const struct location *loc,
					  const struct nfnl_nft_data *nld)
{
	unsigned int code;
	char *chain;

	code = nfnl_nft_verdict_get_verdict(nld);
	switch (code) {
	case NFT_JUMP:
	case NFT_GOTO:
		chain = xstrdup(nfnl_nft_verdict_get_chain(nld));
		break;
	default:
		chain = NULL;
		break;
	}

	return verdict_expr_alloc(loc, code, chain);
}

struct expr *netlink_alloc_data(const struct location *loc,
				const struct nfnl_nft_data *nld,
				enum nft_registers dreg)
{
	switch (dreg) {
	case NFT_REG_VERDICT:
		return netlink_alloc_verdict(loc, nld);
	default:
		return netlink_alloc_value(loc, nld);
	}
}

int netlink_add_rule(struct netlink_ctx *ctx, const struct handle *h,
		     const struct rule *rule, uint32_t flags)
{
	struct nfnl_nft_rule *nlr;
	int err;

	nlr = alloc_nft_rule(&rule->handle);
	err = netlink_linearize_rule(ctx, nlr, rule);
	if (err == 0) {
		err = nfnl_nft_rule_add(nf_sock, nlr, flags | NLM_F_EXCL);
		if (err < 0)
			netlink_io_error(ctx, &rule->location,
					 "Could not add rule: %s", nl_geterror(err));
	}
	nfnl_nft_rule_put(nlr);
	return err;
}

int netlink_delete_rule(struct netlink_ctx *ctx, const struct handle *h,
			const struct location *loc)
{
	struct nfnl_nft_rule *nlr;
	int err;

	nlr = alloc_nft_rule(h);
	err = nfnl_nft_rule_delete(nf_sock, nlr, 0);
	nfnl_nft_rule_put(nlr);

	if (err < 0)
		netlink_io_error(ctx, loc, "Could not delete rule: %s",
				 nl_geterror(err));
	return err;
}

static void list_rule_cb(struct nl_object *obj, void *arg)
{
	const struct nfnl_nft_rule *nlr = (struct nfnl_nft_rule *)obj;
	struct netlink_ctx *ctx = arg;
	struct rule *rule;

	netlink_dump_object(obj);
	if (!nfnl_nft_rule_test_family(nlr) ||
	    !nfnl_nft_rule_test_table(nlr) ||
	    !nfnl_nft_rule_test_chain(nlr) ||
	    !nfnl_nft_rule_test_handle(nlr)) {
		netlink_io_error(ctx, NULL, "Incomplete rule received");
		return;
	}

	rule = netlink_delinearize_rule(ctx, obj);
	list_add_tail(&rule->list, &ctx->list);
}

static int netlink_list_rules(struct netlink_ctx *ctx, const struct handle *h,
			      const struct location *loc)
{
	struct nl_cache *rule_cache;
	struct nfnl_nft_rule *nlr;
	int err;

	err = nfnl_nft_rule_alloc_cache(nf_sock, &rule_cache);
	if (err < 0)
		return netlink_io_error(ctx, loc,
					"Could not receive rules from kernel: %s",
					nl_geterror(err));

	nlr = alloc_nft_rule(h);
	nl_cache_foreach_filter(rule_cache, OBJ_CAST(nlr), list_rule_cb, ctx);
	nfnl_nft_rule_put(nlr);
	nl_cache_free(rule_cache);
	return 0;
}

static int netlink_get_rule_cb(struct nl_msg *msg, void *arg)
{
	return nl_msg_parse(msg, list_rule_cb, arg);
}

int netlink_get_rule(struct netlink_ctx *ctx, const struct handle *h,
		     const struct location *loc)
{
	struct nfnl_nft_rule *nlr;
	int err;

	nlr = alloc_nft_rule(h);
	netlink_set_callback(netlink_get_rule_cb, ctx);
	err = nfnl_nft_rule_query(nf_sock, nlr, 0);
	if (err == 0)
		err = nl_recvmsgs_default(nf_sock);
	nfnl_nft_rule_put(nlr);

	if (err < 0)
		return netlink_io_error(ctx, loc,
					"Could not receive rule from kernel: %s",
					nl_geterror(err));
	return err;
}

static void flush_rule_cb(struct nl_object *obj, void *arg)
{
	struct netlink_ctx *ctx = arg;
	int err;

	netlink_dump_object(obj);
	err = nfnl_nft_rule_delete(nf_sock, (struct nfnl_nft_rule *)obj, 0);
	if (err < 0)
		netlink_io_error(ctx, NULL, "Could not delete rule: %s",
				 nl_geterror(err));
}

static int netlink_flush_rules(struct netlink_ctx *ctx, const struct handle *h,
			       const struct location *loc)
{
	struct nl_cache *rule_cache;
	struct nfnl_nft_rule *nlr;
	int err;

	err = nfnl_nft_rule_alloc_cache(nf_sock, &rule_cache);
	if (err < 0)
		return netlink_io_error(ctx, loc,
					"Could not receive rules from kernel: %s",
					nl_geterror(err));

	nlr = alloc_nft_rule(h);
	nl_cache_foreach_filter(rule_cache, OBJ_CAST(nlr), flush_rule_cb, ctx);
	nfnl_nft_rule_put(nlr);
	nl_cache_free(rule_cache);
	return 0;
}

int netlink_add_chain(struct netlink_ctx *ctx, const struct handle *h,
		      const struct location *loc, const struct chain *chain)
{
	struct nfnl_nft_chain *nlc;
	int err;

	nlc = alloc_nft_chain(h);
	if (chain != NULL && chain->flags & CHAIN_F_BASECHAIN) {
		nfnl_nft_chain_set_hooknum(nlc, chain->hooknum);
		nfnl_nft_chain_set_priority(nlc, chain->priority);
	}
	netlink_dump_object(OBJ_CAST(nlc));
	err = nfnl_nft_chain_add(nf_sock, nlc, NLM_F_EXCL);
	nfnl_nft_chain_put(nlc);

	if (err < 0)
		netlink_io_error(ctx, loc, "Could not add chain: %s",
				 nl_geterror(err));
	return err;
}

int netlink_rename_chain(struct netlink_ctx *ctx, const struct handle *h,
			 const struct location *loc, const char *name)
{
	struct nfnl_nft_chain *nlc;
	int err;

	nlc = alloc_nft_chain(h);
	nfnl_nft_chain_set_name(nlc, name, strlen(name) + 1);
	netlink_dump_object(OBJ_CAST(nlc));
	err = nfnl_nft_chain_add(nf_sock, nlc, 0);
	nfnl_nft_chain_put(nlc);

	if (err < 0)
		netlink_io_error(ctx, loc, "Could not rename chain: %s",
				 nl_geterror(err));
	return err;
}

int netlink_delete_chain(struct netlink_ctx *ctx, const struct handle *h,
			 const struct location *loc)
{
	struct nfnl_nft_chain *nlc;
	int err;

	nlc = alloc_nft_chain(h);
	netlink_dump_object(OBJ_CAST(nlc));
	err = nfnl_nft_chain_delete(nf_sock, nlc, 0);
	nfnl_nft_chain_put(nlc);

	if (err < 0)
		netlink_io_error(ctx, loc, "Could not delete chain: %s",
				 nl_geterror(err));
	return err;
}

static void list_chain_cb(struct nl_object *obj, void *arg)
{
	struct nfnl_nft_chain *nlc = (struct nfnl_nft_chain *)obj;
	struct netlink_ctx *ctx = arg;
	struct chain *chain;

	netlink_dump_object(obj);
	if (!nfnl_nft_chain_test_family(nlc) ||
	    !nfnl_nft_chain_test_table(nlc) ||
	    !nfnl_nft_chain_test_name(nlc)) {
		netlink_io_error(ctx, NULL, "Incomplete chain received");
		return;
	}

	chain = chain_alloc(nfnl_nft_chain_get_name(nlc));
	chain->handle.family = nfnl_nft_chain_get_family(nlc);
	chain->handle.table  = xstrdup(nfnl_nft_chain_get_table(nlc));
	chain->handle.handle = nfnl_nft_chain_get_handle(nlc);
	chain->hooknum       = nfnl_nft_chain_get_hooknum(nlc);
	chain->priority      = nfnl_nft_chain_get_priority(nlc);
	list_add_tail(&chain->list, &ctx->list);
}

int netlink_list_chains(struct netlink_ctx *ctx, const struct handle *h,
			const struct location *loc)
{
	struct nl_cache *chain_cache;
	struct nfnl_nft_chain *nlc;
	struct chain *chain;
	int err;

	err = nfnl_nft_chain_alloc_cache(nf_sock, &chain_cache);
	if (err < 0)
		return netlink_io_error(ctx, loc,
					"Could not receive chains from kernel: %s",
					nl_geterror(err));

	nlc = alloc_nft_chain(h);
	nl_cache_foreach_filter(chain_cache, OBJ_CAST(nlc), list_chain_cb, ctx);
	nfnl_nft_chain_put(nlc);
	nl_cache_free(chain_cache);

	/* Caller wants all existing chains */
	if (h->chain == NULL)
		return 0;

	/* Check if this chain exists, otherwise return an error */
	list_for_each_entry(chain, &ctx->list, list) {
		if (strcmp(chain->handle.chain, h->chain) == 0)
			return 0;
	}

	return netlink_io_error(ctx, NULL,
				"Could not find chain `%s' in table `%s': %s",
				h->chain, h->table,
				nl_geterror(NLE_OBJ_NOTFOUND));
}

static int netlink_get_chain_cb(struct nl_msg *msg, void *arg)
{
	return nl_msg_parse(msg, list_chain_cb, arg);
}

int netlink_get_chain(struct netlink_ctx *ctx, const struct handle *h,
		      const struct location *loc)
{
	struct nfnl_nft_chain *nlc;
	int err;

	nlc = alloc_nft_chain(h);
	netlink_set_callback(netlink_get_chain_cb, ctx);
	err = nfnl_nft_chain_query(nf_sock, nlc, 0);
	if (err == 0)
		err = nl_recvmsgs_default(nf_sock);
	netlink_set_callback(NULL, NULL);
	nfnl_nft_chain_put(nlc);

	if (err < 0)
		return netlink_io_error(ctx, loc,
					"Could not receive chain from kernel: %s",
					nl_geterror(err));
	return err;
}

int netlink_list_chain(struct netlink_ctx *ctx, const struct handle *h,
		       const struct location *loc)
{
	return netlink_list_rules(ctx, h, loc);
}

int netlink_flush_chain(struct netlink_ctx *ctx, const struct handle *h,
			const struct location *loc)
{
	return netlink_flush_rules(ctx, h, loc);
}

int netlink_add_table(struct netlink_ctx *ctx, const struct handle *h,
		      const struct location *loc, const struct table *table)
{
	struct nfnl_nft_table *nlt;
	int err;

	nlt = alloc_nft_table(h);
	err = nfnl_nft_table_add(nf_sock, nlt, NLM_F_EXCL);
	nfnl_nft_table_put(nlt);

	if (err < 0)
		netlink_io_error(ctx, loc, "Could not add table: %s",
				 nl_geterror(err));
	return err;
}

int netlink_delete_table(struct netlink_ctx *ctx, const struct handle *h,
			 const struct location *loc)
{
	struct nfnl_nft_table *nlt;
	int err;

	nlt = alloc_nft_table(h);
	err = nfnl_nft_table_delete(nf_sock, nlt, 0);
	nfnl_nft_table_put(nlt);

	if (err < 0)
		netlink_io_error(ctx, loc, "Could not delete table: %s",
				 nl_geterror(err));
	return err;
}

static void list_table_cb(struct nl_object *obj, void *arg)
{
	struct nfnl_nft_table *nlt = (struct nfnl_nft_table *)obj;
	struct netlink_ctx *ctx = arg;
	struct table *table;

	netlink_dump_object(obj);
	if (!nfnl_nft_table_test_family(nlt) ||
	    !nfnl_nft_table_test_name(nlt)) {
		netlink_io_error(ctx, NULL, "Incomplete table received");
		return;
	}

	table = table_alloc();
	table->handle.family = nfnl_nft_table_get_family(nlt);
	table->handle.table  = xstrdup(nfnl_nft_table_get_name(nlt));
	list_add_tail(&table->list, &ctx->list);
}

int netlink_list_tables(struct netlink_ctx *ctx, const struct handle *h,
			const struct location *loc)
{
	struct nl_cache *table_cache;
	struct nfnl_nft_table *nlt;
	int err;

	err = nfnl_nft_table_alloc_cache(nf_sock, &table_cache);
	if (err < 0)
		return netlink_io_error(ctx, loc,
					"Could not receive tables from kernel: %s",
					nl_geterror(err));

	nlt = alloc_nft_table(h);
	nl_cache_foreach_filter(table_cache, OBJ_CAST(nlt), list_table_cb, ctx);
	nfnl_nft_table_put(nlt);
	nl_cache_free(table_cache);
	return 0;
}

static int netlink_get_table_cb(struct nl_msg *msg, void *arg)
{
	return nl_msg_parse(msg, list_table_cb, arg);
}

int netlink_get_table(struct netlink_ctx *ctx, const struct handle *h,
		      const struct location *loc)
{
	struct nfnl_nft_table *nlt;
	int err;

	nlt = alloc_nft_table(h);
	netlink_set_callback(netlink_get_table_cb, ctx);
	err = nfnl_nft_table_query(nf_sock, nlt, 0);
	if (err == 0)
		err = nl_recvmsgs_default(nf_sock);
	nfnl_nft_table_put(nlt);

	if (err < 0)
		return netlink_io_error(ctx, loc,
					"Could not receive table from kernel: %s",
					nl_geterror(err));
	return err;
}


int netlink_list_table(struct netlink_ctx *ctx, const struct handle *h,
		       const struct location *loc)
{
	return netlink_list_rules(ctx, h, loc);
}

int netlink_flush_table(struct netlink_ctx *ctx, const struct handle *h,
			const struct location *loc)
{
	return netlink_flush_rules(ctx, h, loc);
}

static enum nft_data_types dtype_map_to_kernel(const struct datatype *dtype)
{
	switch (dtype->type) {
	case TYPE_VERDICT:
		return NFT_DATA_VERDICT;
	default:
		return dtype->type;
	}
}

static const struct datatype *dtype_map_from_kernel(enum nft_data_types type)
{
	switch (type) {
	case NFT_DATA_VERDICT:
		return &verdict_type;
	default:
		return datatype_lookup(type);
	}
}

static void add_set_cb(struct nl_object *obj, void *arg)
{
	struct nfnl_nft_set *nls = (struct nfnl_nft_set *)obj;
	struct netlink_ctx *ctx = arg;
	struct set *set = ctx->set;

	netlink_dump_object(OBJ_CAST(nls));
	set->handle.set = xstrdup(nfnl_nft_set_get_name(nls));
}

static int netlink_add_set_cb(struct nl_msg *msg, void *arg)
{
	return nl_msg_parse(msg, add_set_cb, arg);
}

int netlink_add_set(struct netlink_ctx *ctx, const struct handle *h,
		    struct set *set)
{
	struct nfnl_nft_set *nls;
	int err;

	nls = alloc_nft_set(h);
	nfnl_nft_set_set_flags(nls, set->flags);
	nfnl_nft_set_set_keytype(nls, dtype_map_to_kernel(set->keytype));
	nfnl_nft_set_set_keylen(nls, set->keylen / BITS_PER_BYTE);
	if (set->flags & NFT_SET_MAP) {
		nfnl_nft_set_set_datatype(nls, dtype_map_to_kernel(set->datatype));
		nfnl_nft_set_set_datalen(nls, set->datalen / BITS_PER_BYTE);
	}
	netlink_dump_object(OBJ_CAST(nls));

	ctx->set = set;
	netlink_set_callback(netlink_add_set_cb, ctx);
	err = nfnl_nft_set_add(nf_sock, nls, NLM_F_EXCL | NLM_F_ECHO);
	if (err == 0)
		err = nl_recvmsgs_default(nf_sock);
	netlink_set_callback(NULL, NULL);
	nfnl_nft_set_put(nls);
	ctx->set = NULL;

	if (err < 0)
		netlink_io_error(ctx, NULL, "Could not add set: %s",
				 nl_geterror(err));
	return err;
}

int netlink_delete_set(struct netlink_ctx *ctx, const struct handle *h,
		       const struct location *loc)
{
	struct nfnl_nft_set *nls;
	int err;

	nls = alloc_nft_set(h);
	err = nfnl_nft_set_delete(nf_sock, nls, 0);
	nfnl_nft_set_put(nls);

	if (err < 0)
		netlink_io_error(ctx, loc, "Could not delete set: %s",
				 nl_geterror(err));
	return err;
}

static void list_set_cb(struct nl_object *obj, void *arg)
{
	struct nfnl_nft_set *nls = (struct nfnl_nft_set *)obj;
	struct netlink_ctx *ctx = arg;
	const struct datatype *keytype, *datatype;
	uint32_t flags;
	struct set *set;

	netlink_dump_object(obj);
	if (!nfnl_nft_set_test_family(nls) ||
	    !nfnl_nft_set_test_table(nls) ||
	    !nfnl_nft_set_test_name(nls) ||
	    !nfnl_nft_set_test_keytype(nls) ||
	    !nfnl_nft_set_test_keylen(nls)) {
		netlink_io_error(ctx, NULL, "Incomplete set received");
		return;
	}

	keytype = dtype_map_from_kernel(nfnl_nft_set_get_keytype(nls));
	if (keytype == NULL) {
		netlink_io_error(ctx, NULL, "Unknown data type in set key %u",
				 nfnl_nft_set_get_keytype(nls));
		return;
	}

	flags = nfnl_nft_set_get_flags(nls);
	if (flags & NFT_SET_MAP) {
		datatype = dtype_map_from_kernel(nfnl_nft_set_get_datatype(nls));
		if (datatype == NULL) {
			netlink_io_error(ctx, NULL, "Unknown data type in set key %u",
					 nfnl_nft_set_get_datatype(nls));
			return;
		}
	} else
		datatype = NULL;

	set = set_alloc(&internal_location);
	set->handle.family = nfnl_nft_set_get_family(nls);
	set->handle.table  = xstrdup(nfnl_nft_set_get_table(nls));
	set->handle.set    = xstrdup(nfnl_nft_set_get_name(nls));
	set->keytype       = keytype;
	set->keylen        = nfnl_nft_set_get_keylen(nls) * BITS_PER_BYTE;
	set->flags         = flags;
	set->datatype      = datatype;
	set->datalen       = nfnl_nft_set_get_datalen(nls) * BITS_PER_BYTE;
	list_add_tail(&set->list, &ctx->list);
}

int netlink_list_sets(struct netlink_ctx *ctx, const struct handle *h,
		      const struct location *loc)
{
	struct nl_cache *set_cache;
	int err;

	err = nfnl_nft_set_alloc_cache(nf_sock, h->family, h->table, &set_cache);
	if (err < 0)
		return netlink_io_error(ctx, loc,
					"Could not receive sets from kernel: %s",
					nl_geterror(err));

	nl_cache_foreach(set_cache, list_set_cb, ctx);
	nl_cache_free(set_cache);
	return 0;
}

static int netlink_get_set_cb(struct nl_msg *msg, void *arg)
{
	return nl_msg_parse(msg, list_set_cb, arg);
}

int netlink_get_set(struct netlink_ctx *ctx, const struct handle *h,
		    const struct location *loc)
{
	struct nfnl_nft_set *nls;
	int err;

	nls = alloc_nft_set(h);
	netlink_dump_object(OBJ_CAST(nls));
	netlink_set_callback(netlink_get_set_cb, ctx);
	err = nfnl_nft_set_query(nf_sock, nls, 0);
	if (err == 0)
		err = nl_recvmsgs_default(nf_sock);
	netlink_set_callback(NULL, NULL);

	nfnl_nft_set_put(nls);

	if (err < 0)
		return netlink_io_error(ctx, loc,
					"Could not receive set from kernel: %s",
					nl_geterror(err));
	return err;
}

static int alloc_setelem_cache(const struct expr *set, struct nl_cache **res)
{
	struct nfnl_nft_setelem *nlse;
	struct nl_cache *elements;
	const struct expr *expr;
	int err;

	err = nl_cache_alloc_name("netfilter/nft_setelem", &elements);
	if (err < 0)
		return err;
	list_for_each_entry(expr, &set->expressions, list) {
		nlse = alloc_nft_setelem(expr);
		netlink_dump_object(OBJ_CAST(nlse));
		nl_cache_add(elements, OBJ_CAST(nlse));
	}
	*res = elements;
	return 0;
}

int netlink_add_setelems(struct netlink_ctx *ctx, const struct handle *h,
			 const struct expr *expr)
{
	struct nfnl_nft_set *nls;
	struct nl_cache *elements;
	int err;

	nls = alloc_nft_set(h);
	netlink_dump_object(OBJ_CAST(nls));

	err = alloc_setelem_cache(expr, &elements);
	if (err < 0)
		goto out;
	err = nfnl_nft_setelem_add(nf_sock, nls, elements, 0);
	if (err < 0)
		goto out;
	err = nl_recvmsgs_default(nf_sock);
out:
	nfnl_nft_set_put(nls);
	if (err < 0)
		netlink_io_error(ctx, &expr->location,
				 "Could not add set elements: %s",
				 nl_geterror(err));
	return err;
}

int netlink_delete_setelems(struct netlink_ctx *ctx, const struct handle *h,
			    const struct expr *expr)
{
	struct nfnl_nft_set *nls;
	struct nl_cache *elements;
	int err;

	nls = alloc_nft_set(h);
	err = alloc_setelem_cache(expr, &elements);
	if (err < 0)
		goto out;
	err = nfnl_nft_setelem_delete(nf_sock, nls, elements, 0);
	if (err < 0)
		goto out;
	err = nl_recvmsgs_default(nf_sock);
out:
	nfnl_nft_set_put(nls);
	if (err < 0)
		netlink_io_error(ctx, &expr->location,
				 "Could not delete set elements: %s",
				 nl_geterror(err));
	return err;
}

static void list_setelem_cb(struct nl_object *obj, void *arg)
{
	struct nfnl_nft_setelem *nlse = nl_object_priv(obj);
	struct nfnl_nft_data *nld;
	struct netlink_ctx *ctx = arg;
	struct set *set = ctx->set;
	struct expr *expr, *data;
	uint32_t flags;

	netlink_dump_object(obj);
	if (!nfnl_nft_setelem_test_key(nlse)) {
		netlink_io_error(ctx, NULL, "Incomplete set element received");
		return;
	}

	nld   = nfnl_nft_setelem_get_key(nlse);
	flags = nfnl_nft_setelem_get_flags(nlse);

	expr = netlink_alloc_value(&internal_location, nld);
	expr->dtype	= set->keytype;
	expr->byteorder	= set->keytype->byteorder;
	if (expr->byteorder == BYTEORDER_HOST_ENDIAN)
		mpz_switch_byteorder(expr->value, expr->len / BITS_PER_BYTE);

	if (flags & NFT_SET_ELEM_INTERVAL_END)
		expr->flags |= EXPR_F_INTERVAL_END;
	else if (nfnl_nft_setelem_test_data(nlse)) {
		nld = nfnl_nft_setelem_get_data(nlse);

		data = netlink_alloc_data(&internal_location, nld,
					  set->datatype->type == TYPE_VERDICT ?
					  NFT_REG_VERDICT : NFT_REG_1);
		data->dtype = set->datatype;

		expr = mapping_expr_alloc(&internal_location, expr, data);
	}

	compound_expr_add(set->init, expr);
}

extern void interval_map_decompose(struct expr *set);

int netlink_get_setelems(struct netlink_ctx *ctx, const struct handle *h,
			 const struct location *loc, struct set *set)
{
	struct nl_cache *elements;
	struct nfnl_nft_set *nls;
	int err;

	nls = alloc_nft_set(h);
	netlink_dump_object(OBJ_CAST(nls));

	err = nfnl_nft_setelem_alloc_cache(nf_sock, nls, &elements);
	if (err < 0)
		goto out;
	err = nl_recvmsgs_default(nf_sock);
	if (err < 0)
		goto out;

	ctx->set = set;
	set->init = set_expr_alloc(loc);
	nl_cache_foreach(elements, list_setelem_cb, ctx);
	nl_cache_free(elements);
	ctx->set = NULL;

	if (set->flags & NFT_SET_INTERVAL)
		interval_map_decompose(set->init);
out:
	nfnl_nft_set_put(nls);
	if (err < 0)
		netlink_io_error(ctx, loc, "Could not receive set elements: %s",
				 nl_geterror(err));
	return err;
}