Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 1 | /* Shared library add-on to iptables to add byte tracking support. */ |
| 2 | #include <stdio.h> |
| 3 | #include <netdb.h> |
| 4 | #include <string.h> |
| 5 | #include <stdlib.h> |
| 6 | #include <getopt.h> |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 7 | #include <xtables.h> |
Patrick McHardy | 40d5475 | 2007-04-18 07:00:36 +0000 | [diff] [blame] | 8 | #include <linux/netfilter/nf_conntrack_common.h> |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 9 | #include <linux/netfilter/xt_connbytes.h> |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 10 | |
Jan Engelhardt | 181dead | 2007-10-04 16:27:07 +0000 | [diff] [blame] | 11 | static void connbytes_help(void) |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 12 | { |
| 13 | printf( |
Jan Engelhardt | 8b7c64d | 2008-04-15 11:48:25 +0200 | [diff] [blame] | 14 | "connbytes match options:\n" |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 15 | " [!] --connbytes from:[to]\n" |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 16 | " --connbytes-dir [original, reply, both]\n" |
Jan Engelhardt | 8b7c64d | 2008-04-15 11:48:25 +0200 | [diff] [blame] | 17 | " --connbytes-mode [packets, bytes, avgpkt]\n"); |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 18 | } |
| 19 | |
Jan Engelhardt | 181dead | 2007-10-04 16:27:07 +0000 | [diff] [blame] | 20 | static const struct option connbytes_opts[] = { |
Patrick McHardy | 500f483 | 2007-09-08 15:59:04 +0000 | [diff] [blame] | 21 | { "connbytes", 1, NULL, '1' }, |
| 22 | { "connbytes-dir", 1, NULL, '2' }, |
| 23 | { "connbytes-mode", 1, NULL, '3' }, |
Max Kellermann | 9ee386a | 2008-01-29 13:48:05 +0000 | [diff] [blame] | 24 | { .name = NULL } |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 25 | }; |
| 26 | |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 27 | static void |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 28 | parse_range(const char *arg, struct xt_connbytes_info *si) |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 29 | { |
| 30 | char *colon,*p; |
| 31 | |
Harald Welte | 7dc57e2 | 2004-11-18 22:59:36 +0000 | [diff] [blame] | 32 | si->count.from = strtoul(arg,&colon,10); |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 33 | if (*colon != ':') |
| 34 | exit_error(PARAMETER_PROBLEM, "Bad range `%s'", arg); |
Harald Welte | 7dc57e2 | 2004-11-18 22:59:36 +0000 | [diff] [blame] | 35 | si->count.to = strtoul(colon+1,&p,10); |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 36 | if (p == colon+1) { |
| 37 | /* second number omited */ |
Harald Welte | 7dc57e2 | 2004-11-18 22:59:36 +0000 | [diff] [blame] | 38 | si->count.to = 0xffffffff; |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 39 | } |
Harald Welte | 7dc57e2 | 2004-11-18 22:59:36 +0000 | [diff] [blame] | 40 | if (si->count.from > si->count.to) |
| 41 | exit_error(PARAMETER_PROBLEM, "%llu should be less than %llu", |
Patrick McHardy | c329d6a | 2007-09-05 14:19:23 +0000 | [diff] [blame] | 42 | (unsigned long long)si->count.from, |
| 43 | (unsigned long long)si->count.to); |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 44 | } |
| 45 | |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 46 | static int |
Jan Engelhardt | 181dead | 2007-10-04 16:27:07 +0000 | [diff] [blame] | 47 | connbytes_parse(int c, char **argv, int invert, unsigned int *flags, |
| 48 | const void *entry, struct xt_entry_match **match) |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 49 | { |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 50 | struct xt_connbytes_info *sinfo = (struct xt_connbytes_info *)(*match)->data; |
Martin Josefsson | 1146088 | 2004-05-08 14:02:36 +0000 | [diff] [blame] | 51 | unsigned long i; |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 52 | |
| 53 | switch (c) { |
| 54 | case '1': |
Harald Welte | 7dc57e2 | 2004-11-18 22:59:36 +0000 | [diff] [blame] | 55 | if (check_inverse(optarg, &invert, &optind, 0)) |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 56 | optind++; |
| 57 | |
| 58 | parse_range(argv[optind-1], sinfo); |
| 59 | if (invert) { |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 60 | i = sinfo->count.from; |
Harald Welte | 7dc57e2 | 2004-11-18 22:59:36 +0000 | [diff] [blame] | 61 | sinfo->count.from = sinfo->count.to; |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 62 | sinfo->count.to = i; |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 63 | } |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 64 | *flags |= 1; |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 65 | break; |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 66 | case '2': |
| 67 | if (!strcmp(optarg, "original")) |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 68 | sinfo->direction = XT_CONNBYTES_DIR_ORIGINAL; |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 69 | else if (!strcmp(optarg, "reply")) |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 70 | sinfo->direction = XT_CONNBYTES_DIR_REPLY; |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 71 | else if (!strcmp(optarg, "both")) |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 72 | sinfo->direction = XT_CONNBYTES_DIR_BOTH; |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 73 | else |
| 74 | exit_error(PARAMETER_PROBLEM, |
| 75 | "Unknown --connbytes-dir `%s'", optarg); |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 76 | |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 77 | *flags |= 2; |
| 78 | break; |
| 79 | case '3': |
Harald Welte | 7dc57e2 | 2004-11-18 22:59:36 +0000 | [diff] [blame] | 80 | if (!strcmp(optarg, "packets")) |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 81 | sinfo->what = XT_CONNBYTES_PKTS; |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 82 | else if (!strcmp(optarg, "bytes")) |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 83 | sinfo->what = XT_CONNBYTES_BYTES; |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 84 | else if (!strcmp(optarg, "avgpkt")) |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 85 | sinfo->what = XT_CONNBYTES_AVGPKT; |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 86 | else |
| 87 | exit_error(PARAMETER_PROBLEM, |
| 88 | "Unknown --connbytes-mode `%s'", optarg); |
| 89 | *flags |= 4; |
Piotrek Kaczmarek | 1c0f236 | 2005-04-24 16:19:51 +0000 | [diff] [blame] | 90 | break; |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 91 | default: |
| 92 | return 0; |
| 93 | } |
| 94 | |
| 95 | return 1; |
| 96 | } |
| 97 | |
Jan Engelhardt | 181dead | 2007-10-04 16:27:07 +0000 | [diff] [blame] | 98 | static void connbytes_check(unsigned int flags) |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 99 | { |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 100 | if (flags != 7) |
| 101 | exit_error(PARAMETER_PROBLEM, "You must specify `--connbytes'" |
Harald Welte | 402c311 | 2005-12-05 12:08:03 +0000 | [diff] [blame] | 102 | "`--connbytes-dir' and `--connbytes-mode'"); |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 103 | } |
| 104 | |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 105 | static void print_mode(struct xt_connbytes_info *sinfo) |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 106 | { |
| 107 | switch (sinfo->what) { |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 108 | case XT_CONNBYTES_PKTS: |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 109 | fputs("packets ", stdout); |
| 110 | break; |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 111 | case XT_CONNBYTES_BYTES: |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 112 | fputs("bytes ", stdout); |
| 113 | break; |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 114 | case XT_CONNBYTES_AVGPKT: |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 115 | fputs("avgpkt ", stdout); |
| 116 | break; |
Harald Welte | 7dc57e2 | 2004-11-18 22:59:36 +0000 | [diff] [blame] | 117 | default: |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 118 | fputs("unknown ", stdout); |
Harald Welte | 7dc57e2 | 2004-11-18 22:59:36 +0000 | [diff] [blame] | 119 | break; |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 120 | } |
| 121 | } |
| 122 | |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 123 | static void print_direction(struct xt_connbytes_info *sinfo) |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 124 | { |
| 125 | switch (sinfo->direction) { |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 126 | case XT_CONNBYTES_DIR_ORIGINAL: |
Harald Welte | 7dc57e2 | 2004-11-18 22:59:36 +0000 | [diff] [blame] | 127 | fputs("original ", stdout); |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 128 | break; |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 129 | case XT_CONNBYTES_DIR_REPLY: |
Harald Welte | 7dc57e2 | 2004-11-18 22:59:36 +0000 | [diff] [blame] | 130 | fputs("reply ", stdout); |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 131 | break; |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 132 | case XT_CONNBYTES_DIR_BOTH: |
Harald Welte | 7dc57e2 | 2004-11-18 22:59:36 +0000 | [diff] [blame] | 133 | fputs("both ", stdout); |
| 134 | break; |
| 135 | default: |
| 136 | fputs("unknown ", stdout); |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 137 | break; |
| 138 | } |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 139 | } |
| 140 | |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 141 | static void |
Jan Engelhardt | 181dead | 2007-10-04 16:27:07 +0000 | [diff] [blame] | 142 | connbytes_print(const void *ip, const struct xt_entry_match *match, int numeric) |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 143 | { |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 144 | struct xt_connbytes_info *sinfo = (struct xt_connbytes_info *)match->data; |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 145 | |
Harald Welte | 7dc57e2 | 2004-11-18 22:59:36 +0000 | [diff] [blame] | 146 | if (sinfo->count.from > sinfo->count.to) |
Patrick McHardy | c329d6a | 2007-09-05 14:19:23 +0000 | [diff] [blame] | 147 | printf("connbytes ! %llu:%llu ", |
| 148 | (unsigned long long)sinfo->count.to, |
| 149 | (unsigned long long)sinfo->count.from); |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 150 | else |
Patrick McHardy | c329d6a | 2007-09-05 14:19:23 +0000 | [diff] [blame] | 151 | printf("connbytes %llu:%llu ", |
| 152 | (unsigned long long)sinfo->count.from, |
| 153 | (unsigned long long)sinfo->count.to); |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 154 | |
| 155 | fputs("connbytes mode ", stdout); |
| 156 | print_mode(sinfo); |
| 157 | |
| 158 | fputs("connbytes direction ", stdout); |
| 159 | print_direction(sinfo); |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 160 | } |
| 161 | |
Jan Engelhardt | 181dead | 2007-10-04 16:27:07 +0000 | [diff] [blame] | 162 | static void connbytes_save(const void *ip, const struct xt_entry_match *match) |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 163 | { |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 164 | struct xt_connbytes_info *sinfo = (struct xt_connbytes_info *)match->data; |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 165 | |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 166 | if (sinfo->count.from > sinfo->count.to) |
Patrick McHardy | c329d6a | 2007-09-05 14:19:23 +0000 | [diff] [blame] | 167 | printf("! --connbytes %llu:%llu ", |
| 168 | (unsigned long long)sinfo->count.to, |
| 169 | (unsigned long long)sinfo->count.from); |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 170 | else |
Patrick McHardy | c329d6a | 2007-09-05 14:19:23 +0000 | [diff] [blame] | 171 | printf("--connbytes %llu:%llu ", |
| 172 | (unsigned long long)sinfo->count.from, |
| 173 | (unsigned long long)sinfo->count.to); |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 174 | |
| 175 | fputs("--connbytes-mode ", stdout); |
| 176 | print_mode(sinfo); |
| 177 | |
Harald Welte | 402c311 | 2005-12-05 12:08:03 +0000 | [diff] [blame] | 178 | fputs("--connbytes-dir ", stdout); |
Harald Welte | 93f4a3d | 2004-11-18 22:50:01 +0000 | [diff] [blame] | 179 | print_direction(sinfo); |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 180 | } |
| 181 | |
Jan Engelhardt | 181dead | 2007-10-04 16:27:07 +0000 | [diff] [blame] | 182 | static struct xtables_match connbytes_match = { |
Jan Engelhardt | 03d9948 | 2008-11-18 12:27:54 +0100 | [diff] [blame] | 183 | .family = NFPROTO_IPV4, |
Pablo Neira | 8caee8b | 2004-12-28 13:11:59 +0000 | [diff] [blame] | 184 | .name = "connbytes", |
Jan Engelhardt | 8b7c64d | 2008-04-15 11:48:25 +0200 | [diff] [blame] | 185 | .version = XTABLES_VERSION, |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 186 | .size = XT_ALIGN(sizeof(struct xt_connbytes_info)), |
| 187 | .userspacesize = XT_ALIGN(sizeof(struct xt_connbytes_info)), |
Jan Engelhardt | 181dead | 2007-10-04 16:27:07 +0000 | [diff] [blame] | 188 | .help = connbytes_help, |
| 189 | .parse = connbytes_parse, |
| 190 | .final_check = connbytes_check, |
| 191 | .print = connbytes_print, |
| 192 | .save = connbytes_save, |
| 193 | .extra_opts = connbytes_opts, |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 194 | }; |
| 195 | |
Jan Engelhardt | 181dead | 2007-10-04 16:27:07 +0000 | [diff] [blame] | 196 | static struct xtables_match connbytes_match6 = { |
Jan Engelhardt | 03d9948 | 2008-11-18 12:27:54 +0100 | [diff] [blame] | 197 | .family = NFPROTO_IPV6, |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 198 | .name = "connbytes", |
Jan Engelhardt | 8b7c64d | 2008-04-15 11:48:25 +0200 | [diff] [blame] | 199 | .version = XTABLES_VERSION, |
Yasuyuki KOZAKAI | 6aac500 | 2007-08-04 08:25:43 +0000 | [diff] [blame] | 200 | .size = XT_ALIGN(sizeof(struct xt_connbytes_info)), |
| 201 | .userspacesize = XT_ALIGN(sizeof(struct xt_connbytes_info)), |
Jan Engelhardt | 181dead | 2007-10-04 16:27:07 +0000 | [diff] [blame] | 202 | .help = connbytes_help, |
| 203 | .parse = connbytes_parse, |
| 204 | .final_check = connbytes_check, |
| 205 | .print = connbytes_print, |
| 206 | .save = connbytes_save, |
| 207 | .extra_opts = connbytes_opts, |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 208 | }; |
| 209 | |
| 210 | void _init(void) |
| 211 | { |
Jan Engelhardt | 181dead | 2007-10-04 16:27:07 +0000 | [diff] [blame] | 212 | xtables_register_match(&connbytes_match); |
| 213 | xtables_register_match(&connbytes_match6); |
Martin Devera | 766113a | 2003-06-19 12:23:37 +0000 | [diff] [blame] | 214 | } |