Commit | Line | Data |
---|---|---|
96518518 PM |
1 | /* |
2 | * Copyright (c) 2008 Patrick McHardy <kaber@trash.net> | |
9370761c | 3 | * Copyright (c) 2012 Pablo Neira Ayuso <pablo@netfilter.org> |
96518518 PM |
4 | * |
5 | * This program is free software; you can redistribute it and/or modify | |
6 | * it under the terms of the GNU General Public License version 2 as | |
7 | * published by the Free Software Foundation. | |
8 | */ | |
9 | ||
10 | #include <linux/module.h> | |
11 | #include <linux/init.h> | |
12 | #include <linux/list.h> | |
13 | #include <linux/skbuff.h> | |
14 | #include <linux/netlink.h> | |
15 | #include <linux/netfilter.h> | |
16 | #include <linux/netfilter_ipv4.h> | |
17 | #include <linux/netfilter/nfnetlink.h> | |
18 | #include <linux/netfilter/nf_tables.h> | |
19 | #include <net/netfilter/nf_tables.h> | |
0ca743a5 | 20 | #include <net/netfilter/nf_tables_ipv4.h> |
96518518 PM |
21 | #include <net/route.h> |
22 | #include <net/ip.h> | |
23 | ||
24 | static unsigned int nf_route_table_hook(const struct nf_hook_ops *ops, | |
25 | struct sk_buff *skb, | |
238e54c9 | 26 | const struct nf_hook_state *state) |
96518518 PM |
27 | { |
28 | unsigned int ret; | |
0ca743a5 | 29 | struct nft_pktinfo pkt; |
96518518 PM |
30 | u32 mark; |
31 | __be32 saddr, daddr; | |
32 | u_int8_t tos; | |
33 | const struct iphdr *iph; | |
34 | ||
35 | /* root is playing with raw sockets. */ | |
36 | if (skb->len < sizeof(struct iphdr) || | |
37 | ip_hdrlen(skb) < sizeof(struct iphdr)) | |
38 | return NF_ACCEPT; | |
39 | ||
238e54c9 | 40 | nft_set_pktinfo_ipv4(&pkt, ops, skb, state->in, state->out); |
0ca743a5 | 41 | |
96518518 PM |
42 | mark = skb->mark; |
43 | iph = ip_hdr(skb); | |
44 | saddr = iph->saddr; | |
45 | daddr = iph->daddr; | |
46 | tos = iph->tos; | |
47 | ||
3876d22d | 48 | ret = nft_do_chain(&pkt, ops); |
96518518 PM |
49 | if (ret != NF_DROP && ret != NF_QUEUE) { |
50 | iph = ip_hdr(skb); | |
51 | ||
52 | if (iph->saddr != saddr || | |
53 | iph->daddr != daddr || | |
54 | skb->mark != mark || | |
55 | iph->tos != tos) | |
56 | if (ip_route_me_harder(skb, RTN_UNSPEC)) | |
57 | ret = NF_DROP; | |
58 | } | |
59 | return ret; | |
60 | } | |
61 | ||
2a37d755 | 62 | static const struct nf_chain_type nft_chain_route_ipv4 = { |
9370761c PNA |
63 | .name = "route", |
64 | .type = NFT_CHAIN_T_ROUTE, | |
fa2c1de0 PM |
65 | .family = NFPROTO_IPV4, |
66 | .owner = THIS_MODULE, | |
9370761c | 67 | .hook_mask = (1 << NF_INET_LOCAL_OUT), |
fa2c1de0 | 68 | .hooks = { |
9370761c | 69 | [NF_INET_LOCAL_OUT] = nf_route_table_hook, |
96518518 | 70 | }, |
96518518 PM |
71 | }; |
72 | ||
9370761c | 73 | static int __init nft_chain_route_init(void) |
96518518 | 74 | { |
9370761c | 75 | return nft_register_chain_type(&nft_chain_route_ipv4); |
96518518 PM |
76 | } |
77 | ||
9370761c | 78 | static void __exit nft_chain_route_exit(void) |
96518518 | 79 | { |
9370761c | 80 | nft_unregister_chain_type(&nft_chain_route_ipv4); |
96518518 PM |
81 | } |
82 | ||
9370761c PNA |
83 | module_init(nft_chain_route_init); |
84 | module_exit(nft_chain_route_exit); | |
96518518 PM |
85 | |
86 | MODULE_LICENSE("GPL"); | |
87 | MODULE_AUTHOR("Patrick McHardy <kaber@trash.net>"); | |
9370761c | 88 | MODULE_ALIAS_NFT_CHAIN(AF_INET, "route"); |