/* ebt_pkttype

 *

 * Authors:

 * Bart De Schuymer <bdschuym@pandora.be>

 *

 * April, 2003

 */

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <getopt.h>

#include <netdb.h>

#include <xtables.h>

#include <linux/if_packet.h>

#include <linux/netfilter_bridge/ebt_pkttype.h>

static const char *classes[] = {

	"host",

	"broadcast",

	"multicast",

	"otherhost",

	"outgoing",

	"loopback",

	"fastroute",

};

static const struct option brpkttype_opts[] =

{

	{ "pkttype-type"        , required_argument, 0, '1' },

	{ 0 }

};

static void brpkttype_print_help(void)

{

	printf(

"pkttype options:\n"

"--pkttype-type    [!] type: class the packet belongs to\n"

"Possible values: broadcast, multicast, host, otherhost, or any other byte value (which would be pretty useless).\n");

}

static int brpkttype_parse(int c, char **argv, int invert, unsigned int *flags,

			   const void *entry, struct xt_entry_match **match)

{

	struct ebt_pkttype_info *ptinfo = (struct ebt_pkttype_info *)(*match)->data;

	char *end;

	long int i;

	switch (c) {

	case '1':

		if (invert)

			ptinfo->invert = 1;

		i = strtol(optarg, &end, 16);

		if (*end != '\0') {

			for (i = 0; i < ARRAY_SIZE(classes); i++) {

				if (!strcasecmp(optarg, classes[i]))

					break;

			}

			if (i >= ARRAY_SIZE(classes))

				xtables_error(PARAMETER_PROBLEM, "Could not parse class '%s'", optarg);

		}

		if (i < 0 || i > 255)

			xtables_error(PARAMETER_PROBLEM, "Problem with specified pkttype class");

		ptinfo->pkt_type = (uint8_t)i;

		break;

	default:

		return 0;

	}

	return 1;

}

static void brpkttype_print(const void *ip, const struct xt_entry_match *match, int numeric)

{

	struct ebt_pkttype_info *pt = (struct ebt_pkttype_info *)match->data;

	if (pt->invert)

		printf("! ");

	printf("--pkttype-type ");

	if (pt->pkt_type < ARRAY_SIZE(classes))

		printf("%s ", classes[pt->pkt_type]);

	else

		printf("%d ", pt->pkt_type);

}

static int brpkttype_xlate(struct xt_xlate *xl,

			  const struct xt_xlate_mt_params *params)

{

	const struct ebt_pkttype_info *info = (const void*)params->match->data;

	xt_xlate_add(xl, "meta pkttype %s", info->invert ? "!= " : "");

	if (info->pkt_type < 3)

		xt_xlate_add(xl, "%s ", classes[info->pkt_type]);

	else if (info->pkt_type == 3)

		xt_xlate_add(xl, "other ");

	else

		xt_xlate_add(xl, "%d ", info->pkt_type);

	return 1;

}

static struct xtables_match brpkttype_match = {

	.name		= "pkttype",

	.version	= XTABLES_VERSION,

	.family		= NFPROTO_BRIDGE,

	.size		= XT_ALIGN(sizeof(struct ebt_pkttype_info)),

	.userspacesize	= XT_ALIGN(sizeof(struct ebt_pkttype_info)),

	.help		= brpkttype_print_help,

	.parse		= brpkttype_parse,

	.print		= brpkttype_print,

	.xlate		= brpkttype_xlate,

	.extra_opts	= brpkttype_opts,

};

void _init(void)

{

	xtables_register_match(&brpkttype_match);

}