xref: /aosp_15_r20/external/iptables/extensions/libxt_mac.c (revision a71a954618bbadd4a345637e5edcf36eec826889)
1*a71a9546SAutomerger Merge Worker #include <stdio.h>
2*a71a9546SAutomerger Merge Worker #if defined(__GLIBC__) && __GLIBC__ == 2
3*a71a9546SAutomerger Merge Worker #include <net/ethernet.h>
4*a71a9546SAutomerger Merge Worker #else
5*a71a9546SAutomerger Merge Worker #include <linux/if_ether.h>
6*a71a9546SAutomerger Merge Worker #endif
7*a71a9546SAutomerger Merge Worker #include <xtables.h>
8*a71a9546SAutomerger Merge Worker #include <linux/netfilter/xt_mac.h>
9*a71a9546SAutomerger Merge Worker 
10*a71a9546SAutomerger Merge Worker enum {
11*a71a9546SAutomerger Merge Worker 	O_MAC = 0,
12*a71a9546SAutomerger Merge Worker };
13*a71a9546SAutomerger Merge Worker 
mac_help(void)14*a71a9546SAutomerger Merge Worker static void mac_help(void)
15*a71a9546SAutomerger Merge Worker {
16*a71a9546SAutomerger Merge Worker 	printf(
17*a71a9546SAutomerger Merge Worker "mac match options:\n"
18*a71a9546SAutomerger Merge Worker "[!] --mac-source XX:XX:XX:XX:XX:XX\n"
19*a71a9546SAutomerger Merge Worker "				Match source MAC address\n");
20*a71a9546SAutomerger Merge Worker }
21*a71a9546SAutomerger Merge Worker 
22*a71a9546SAutomerger Merge Worker #define s struct xt_mac_info
23*a71a9546SAutomerger Merge Worker static const struct xt_option_entry mac_opts[] = {
24*a71a9546SAutomerger Merge Worker 	{.name = "mac-source", .id = O_MAC, .type = XTTYPE_ETHERMAC,
25*a71a9546SAutomerger Merge Worker 	 .flags = XTOPT_MAND | XTOPT_INVERT | XTOPT_PUT,
26*a71a9546SAutomerger Merge Worker 	 XTOPT_POINTER(s, srcaddr)},
27*a71a9546SAutomerger Merge Worker 	XTOPT_TABLEEND,
28*a71a9546SAutomerger Merge Worker };
29*a71a9546SAutomerger Merge Worker #undef s
30*a71a9546SAutomerger Merge Worker 
mac_parse(struct xt_option_call * cb)31*a71a9546SAutomerger Merge Worker static void mac_parse(struct xt_option_call *cb)
32*a71a9546SAutomerger Merge Worker {
33*a71a9546SAutomerger Merge Worker 	struct xt_mac_info *macinfo = cb->data;
34*a71a9546SAutomerger Merge Worker 
35*a71a9546SAutomerger Merge Worker 	xtables_option_parse(cb);
36*a71a9546SAutomerger Merge Worker 	if (cb->invert)
37*a71a9546SAutomerger Merge Worker 		macinfo->invert = 1;
38*a71a9546SAutomerger Merge Worker }
39*a71a9546SAutomerger Merge Worker 
40*a71a9546SAutomerger Merge Worker static void
mac_print(const void * ip,const struct xt_entry_match * match,int numeric)41*a71a9546SAutomerger Merge Worker mac_print(const void *ip, const struct xt_entry_match *match, int numeric)
42*a71a9546SAutomerger Merge Worker {
43*a71a9546SAutomerger Merge Worker 	const struct xt_mac_info *info = (void *)match->data;
44*a71a9546SAutomerger Merge Worker 
45*a71a9546SAutomerger Merge Worker 	printf(" MAC ");
46*a71a9546SAutomerger Merge Worker 
47*a71a9546SAutomerger Merge Worker 	if (info->invert)
48*a71a9546SAutomerger Merge Worker 		printf("! ");
49*a71a9546SAutomerger Merge Worker 
50*a71a9546SAutomerger Merge Worker 	xtables_print_mac(info->srcaddr);
51*a71a9546SAutomerger Merge Worker }
52*a71a9546SAutomerger Merge Worker 
mac_save(const void * ip,const struct xt_entry_match * match)53*a71a9546SAutomerger Merge Worker static void mac_save(const void *ip, const struct xt_entry_match *match)
54*a71a9546SAutomerger Merge Worker {
55*a71a9546SAutomerger Merge Worker 	const struct xt_mac_info *info = (void *)match->data;
56*a71a9546SAutomerger Merge Worker 
57*a71a9546SAutomerger Merge Worker 	if (info->invert)
58*a71a9546SAutomerger Merge Worker 		printf(" !");
59*a71a9546SAutomerger Merge Worker 
60*a71a9546SAutomerger Merge Worker 	printf(" --mac-source ");
61*a71a9546SAutomerger Merge Worker 	xtables_print_mac(info->srcaddr);
62*a71a9546SAutomerger Merge Worker }
63*a71a9546SAutomerger Merge Worker 
print_mac_xlate(const unsigned char * macaddress,struct xt_xlate * xl)64*a71a9546SAutomerger Merge Worker static void print_mac_xlate(const unsigned char *macaddress,
65*a71a9546SAutomerger Merge Worker 			    struct xt_xlate *xl)
66*a71a9546SAutomerger Merge Worker {
67*a71a9546SAutomerger Merge Worker 	unsigned int i;
68*a71a9546SAutomerger Merge Worker 
69*a71a9546SAutomerger Merge Worker 	xt_xlate_add(xl, "%02x", macaddress[0]);
70*a71a9546SAutomerger Merge Worker 	for (i = 1; i < ETH_ALEN; ++i)
71*a71a9546SAutomerger Merge Worker 		xt_xlate_add(xl, ":%02x", macaddress[i]);
72*a71a9546SAutomerger Merge Worker }
73*a71a9546SAutomerger Merge Worker 
mac_xlate(struct xt_xlate * xl,const struct xt_xlate_mt_params * params)74*a71a9546SAutomerger Merge Worker static int mac_xlate(struct xt_xlate *xl,
75*a71a9546SAutomerger Merge Worker 		     const struct xt_xlate_mt_params *params)
76*a71a9546SAutomerger Merge Worker {
77*a71a9546SAutomerger Merge Worker 	const struct xt_mac_info *info = (void *)params->match->data;
78*a71a9546SAutomerger Merge Worker 
79*a71a9546SAutomerger Merge Worker 	xt_xlate_add(xl, "ether saddr%s ", info->invert ? " !=" : "");
80*a71a9546SAutomerger Merge Worker 	print_mac_xlate(info->srcaddr, xl);
81*a71a9546SAutomerger Merge Worker 
82*a71a9546SAutomerger Merge Worker 	return 1;
83*a71a9546SAutomerger Merge Worker }
84*a71a9546SAutomerger Merge Worker 
85*a71a9546SAutomerger Merge Worker static struct xtables_match mac_match = {
86*a71a9546SAutomerger Merge Worker 	.family		= NFPROTO_UNSPEC,
87*a71a9546SAutomerger Merge Worker 	.name		= "mac",
88*a71a9546SAutomerger Merge Worker 	.version	= XTABLES_VERSION,
89*a71a9546SAutomerger Merge Worker 	.size		= XT_ALIGN(sizeof(struct xt_mac_info)),
90*a71a9546SAutomerger Merge Worker 	.userspacesize	= XT_ALIGN(sizeof(struct xt_mac_info)),
91*a71a9546SAutomerger Merge Worker 	.help		= mac_help,
92*a71a9546SAutomerger Merge Worker 	.x6_parse	= mac_parse,
93*a71a9546SAutomerger Merge Worker 	.print		= mac_print,
94*a71a9546SAutomerger Merge Worker 	.save		= mac_save,
95*a71a9546SAutomerger Merge Worker 	.x6_options	= mac_opts,
96*a71a9546SAutomerger Merge Worker 	.xlate		= mac_xlate,
97*a71a9546SAutomerger Merge Worker };
98*a71a9546SAutomerger Merge Worker 
_init(void)99*a71a9546SAutomerger Merge Worker void _init(void)
100*a71a9546SAutomerger Merge Worker {
101*a71a9546SAutomerger Merge Worker 	xtables_register_match(&mac_match);
102*a71a9546SAutomerger Merge Worker }
103