1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <getopt.h>
#include "../include/ebtables_u.h"
#include <linux/netfilter_bridge/ebt_mark_m.h>
#define MARK '1'
static struct option opts[] =
{
{ "mark", required_argument, 0, MARK },
{ 0 }
};
static void print_help()
{
printf(
"mark option:\n"
"--mark [!] [value][/mask]: Match nfmask value (see man page)\n");
}
static void init(struct ebt_entry_match *match)
{
struct ebt_mark_m_info *markinfo = (struct ebt_mark_m_info *)match->data;
markinfo->mark = 0;
markinfo->mask = 0;
markinfo->invert = 0;
markinfo->bitmask = 0;
}
#define OPT_MARK 0x01
static int parse(int c, char **argv, int argc, const struct ebt_u_entry *entry,
unsigned int *flags, struct ebt_entry_match **match)
{
struct ebt_mark_m_info *markinfo = (struct ebt_mark_m_info *)
(*match)->data;
char *end;
switch (c) {
case MARK:
check_option(flags, MARK);
if (check_inverse(optarg))
markinfo->invert = 1;
if (optind > argc)
print_error("No mark specified");
markinfo->mark = strtoul(argv[optind - 1], &end, 0);
markinfo->bitmask = EBT_MARK_AND;
if (*end == '/') {
if (end == argv[optind - 1])
markinfo->bitmask = EBT_MARK_OR;
markinfo->mask = strtoul(end+1, &end, 0);
} else
markinfo->mask = 0xffffffff;
if ( *end != '\0' || end == argv[optind - 1])
print_error("Bad mark value '%s'", argv[optind - 1]);
break;
default:
return 0;
}
return 1;
}
static void final_check(const struct ebt_u_entry *entry,
const struct ebt_entry_match *match, const char *name,
unsigned int hookmask, unsigned int time)
{
}
static void print(const struct ebt_u_entry *entry,
const struct ebt_entry_match *match)
{
struct ebt_mark_m_info *markinfo =
(struct ebt_mark_m_info *)match->data;
printf("--mark ");
if (markinfo->invert)
printf("! ");
if (markinfo->bitmask == EBT_MARK_OR)
printf("/0x%lx ", markinfo->mask);
else if(markinfo->mask != 0xffffffff)
printf("0x%lx/0x%lx ", markinfo->mark, markinfo->mask);
else
printf("0x%lx ", markinfo->mark);
}
static int compare(const struct ebt_entry_match *m1,
const struct ebt_entry_match *m2)
{
struct ebt_mark_m_info *markinfo1 = (struct ebt_mark_m_info *)m1->data;
struct ebt_mark_m_info *markinfo2 = (struct ebt_mark_m_info *)m2->data;
if (markinfo1->invert != markinfo2->invert)
return 0;
if (markinfo1->mark != markinfo2->mark)
return 0;
if (markinfo1->mask != markinfo2->mask)
return 0;
if (markinfo1->bitmask != markinfo2->bitmask)
return 0;
return 1;
}
static struct ebt_u_match mark_match =
{
EBT_MARK_MATCH,
sizeof(struct ebt_mark_m_info),
print_help,
init,
parse,
final_check,
print,
compare,
opts
};
static void _init(void) __attribute((constructor));
static void _init(void)
{
register_match(&mark_match);
}
|
