[linux-2.6-block.git] / net / netfilter / nf_conntrack_proto_gre.c

/*
 * ip_conntrack_proto_gre.c - Version 3.0
 *
 * Connection tracking protocol helper module for GRE.
 *
 * GRE is a generic encapsulation protocol, which is generally not very
 * suited for NAT, as it has no protocol-specific part as port numbers.
 *
 * It has an optional key field, which may help us distinguishing two
 * connections between the same two hosts.
 *
 * GRE is defined in RFC 1701 and RFC 1702, as well as RFC 2784
 *
 * PPTP is built on top of a modified version of GRE, and has a mandatory
 * field called "CallID", which serves us for the same purpose as the key
 * field in plain GRE.
 *
 * Documentation about PPTP can be found in RFC 2637
 *
 * (C) 2000-2005 by Harald Welte <laforge@gnumonks.org>
 *
 * Development of this code funded by Astaro AG (http://www.astaro.com/)
 *
 * (C) 2006-2012 Patrick McHardy <kaber@trash.net>
 */

#include <linux/module.h>
#include <linux/types.h>
#include <linux/timer.h>
#include <linux/list.h>
#include <linux/seq_file.h>
#include <linux/in.h>
#include <linux/netdevice.h>
#include <linux/skbuff.h>
#include <linux/slab.h>
#include <net/dst.h>
#include <net/net_namespace.h>
#include <net/netns/generic.h>
#include <net/netfilter/nf_conntrack_l4proto.h>
#include <net/netfilter/nf_conntrack_helper.h>
#include <net/netfilter/nf_conntrack_core.h>
#include <linux/netfilter/nf_conntrack_proto_gre.h>
#include <linux/netfilter/nf_conntrack_pptp.h>

enum grep_conntrack {
	GRE_CT_UNREPLIED,
	GRE_CT_REPLIED,
	GRE_CT_MAX
};

static unsigned int gre_timeouts[GRE_CT_MAX] = {
	[GRE_CT_UNREPLIED]	= 30*HZ,
	[GRE_CT_REPLIED]	= 180*HZ,
};

static int proto_gre_net_id __read_mostly;
struct netns_proto_gre {
	struct nf_proto_net	nf;
	rwlock_t		keymap_lock;
	struct list_head	keymap_list;
	unsigned int		gre_timeouts[GRE_CT_MAX];
};

static inline struct netns_proto_gre *gre_pernet(struct net *net)
{
	return net_generic(net, proto_gre_net_id);
}

static void nf_ct_gre_keymap_flush(struct net *net)
{
	struct netns_proto_gre *net_gre = gre_pernet(net);
	struct nf_ct_gre_keymap *km, *tmp;

	write_lock_bh(&net_gre->keymap_lock);
	list_for_each_entry_safe(km, tmp, &net_gre->keymap_list, list) {
		list_del(&km->list);
		kfree(km);
	}
	write_unlock_bh(&net_gre->keymap_lock);
}

static inline int gre_key_cmpfn(const struct nf_ct_gre_keymap *km,
				const struct nf_conntrack_tuple *t)
{
	return km->tuple.src.l3num == t->src.l3num &&
	       !memcmp(&km->tuple.src.u3, &t->src.u3, sizeof(t->src.u3)) &&
	       !memcmp(&km->tuple.dst.u3, &t->dst.u3, sizeof(t->dst.u3)) &&
	       km->tuple.dst.protonum == t->dst.protonum &&
	       km->tuple.dst.u.all == t->dst.u.all;
}

/* look up the source key for a given tuple */
static __be16 gre_keymap_lookup(struct net *net, struct nf_conntrack_tuple *t)
{
	struct netns_proto_gre *net_gre = gre_pernet(net);
	struct nf_ct_gre_keymap *km;
	__be16 key = 0;

	read_lock_bh(&net_gre->keymap_lock);
	list_for_each_entry(km, &net_gre->keymap_list, list) {
		if (gre_key_cmpfn(km, t)) {
			key = km->tuple.src.u.gre.key;
			break;
		}
	}
	read_unlock_bh(&net_gre->keymap_lock);

	pr_debug("lookup src key 0x%x for ", key);
	nf_ct_dump_tuple(t);

	return key;
}

/* add a single keymap entry, associate with specified master ct */
int nf_ct_gre_keymap_add(struct nf_conn *ct, enum ip_conntrack_dir dir,
			 struct nf_conntrack_tuple *t)
{
	struct net *net = nf_ct_net(ct);
	struct netns_proto_gre *net_gre = gre_pernet(net);
	struct nf_ct_pptp_master *ct_pptp_info = nfct_help_data(ct);
	struct nf_ct_gre_keymap **kmp, *km;

	kmp = &ct_pptp_info->keymap[dir];
	if (*kmp) {
		/* check whether it's a retransmission */
		read_lock_bh(&net_gre->keymap_lock);
		list_for_each_entry(km, &net_gre->keymap_list, list) {
			if (gre_key_cmpfn(km, t) && km == *kmp) {
				read_unlock_bh(&net_gre->keymap_lock);
				return 0;
			}
		}
		read_unlock_bh(&net_gre->keymap_lock);
		pr_debug("trying to override keymap_%s for ct %p\n",
			 dir == IP_CT_DIR_REPLY ? "reply" : "orig", ct);
		return -EEXIST;
	}

	km = kmalloc(sizeof(*km), GFP_ATOMIC);
	if (!km)
		return -ENOMEM;
	memcpy(&km->tuple, t, sizeof(*t));
	*kmp = km;

	pr_debug("adding new entry %p: ", km);
	nf_ct_dump_tuple(&km->tuple);

	write_lock_bh(&net_gre->keymap_lock);
	list_add_tail(&km->list, &net_gre->keymap_list);
	write_unlock_bh(&net_gre->keymap_lock);

	return 0;
}
EXPORT_SYMBOL_GPL(nf_ct_gre_keymap_add);

/* destroy the keymap entries associated with specified master ct */
void nf_ct_gre_keymap_destroy(struct nf_conn *ct)
{
	struct net *net = nf_ct_net(ct);
	struct netns_proto_gre *net_gre = gre_pernet(net);
	struct nf_ct_pptp_master *ct_pptp_info = nfct_help_data(ct);
	enum ip_conntrack_dir dir;

	pr_debug("entering for ct %p\n", ct);

	write_lock_bh(&net_gre->keymap_lock);
	for (dir = IP_CT_DIR_ORIGINAL; dir < IP_CT_DIR_MAX; dir++) {
		if (ct_pptp_info->keymap[dir]) {
			pr_debug("removing %p from list\n",
				 ct_pptp_info->keymap[dir]);
			list_del(&ct_pptp_info->keymap[dir]->list);
			kfree(ct_pptp_info->keymap[dir]);
			ct_pptp_info->keymap[dir] = NULL;
		}
	}
	write_unlock_bh(&net_gre->keymap_lock);
}
EXPORT_SYMBOL_GPL(nf_ct_gre_keymap_destroy);

/* PUBLIC CONNTRACK PROTO HELPER FUNCTIONS */

/* invert gre part of tuple */
static bool gre_invert_tuple(struct nf_conntrack_tuple *tuple,
			     const struct nf_conntrack_tuple *orig)
{
	tuple->dst.u.gre.key = orig->src.u.gre.key;
	tuple->src.u.gre.key = orig->dst.u.gre.key;
	return true;
}

/* gre hdr info to tuple */
static bool gre_pkt_to_tuple(const struct sk_buff *skb, unsigned int dataoff,
			     struct nf_conntrack_tuple *tuple)
{
	struct net *net = dev_net(skb->dev ? skb->dev : skb_dst(skb)->dev);
	const struct gre_hdr_pptp *pgrehdr;
	struct gre_hdr_pptp _pgrehdr;
	__be16 srckey;
	const struct gre_hdr *grehdr;
	struct gre_hdr _grehdr;

	/* first only delinearize old RFC1701 GRE header */
	grehdr = skb_header_pointer(skb, dataoff, sizeof(_grehdr), &_grehdr);
	if (!grehdr || grehdr->version != GRE_VERSION_PPTP) {
		/* try to behave like "nf_conntrack_proto_generic" */
		tuple->src.u.all = 0;
		tuple->dst.u.all = 0;
		return true;
	}

	/* PPTP header is variable length, only need up to the call_id field */
	pgrehdr = skb_header_pointer(skb, dataoff, 8, &_pgrehdr);
	if (!pgrehdr)
		return true;

	if (ntohs(grehdr->protocol) != GRE_PROTOCOL_PPTP) {
		pr_debug("GRE_VERSION_PPTP but unknown proto\n");
		return false;
	}

	tuple->dst.u.gre.key = pgrehdr->call_id;
	srckey = gre_keymap_lookup(net, tuple);
	tuple->src.u.gre.key = srckey;

	return true;
}

/* print gre part of tuple */
static void gre_print_tuple(struct seq_file *s,
			    const struct nf_conntrack_tuple *tuple)
{
	seq_printf(s, "srckey=0x%x dstkey=0x%x ",
		   ntohs(tuple->src.u.gre.key),
		   ntohs(tuple->dst.u.gre.key));
}

/* print private data for conntrack */
static void gre_print_conntrack(struct seq_file *s, struct nf_conn *ct)
{
	seq_printf(s, "timeout=%u, stream_timeout=%u ",
		   (ct->proto.gre.timeout / HZ),
		   (ct->proto.gre.stream_timeout / HZ));
}

static unsigned int *gre_get_timeouts(struct net *net)
{
	return gre_pernet(net)->gre_timeouts;
}

/* Returns verdict for packet, and may modify conntrack */
static int gre_packet(struct nf_conn *ct,
		      const struct sk_buff *skb,
		      unsigned int dataoff,
		      enum ip_conntrack_info ctinfo,
		      u_int8_t pf,
		      unsigned int hooknum,
		      unsigned int *timeouts)
{
	/* If we've seen traffic both ways, this is a GRE connection.
	 * Extend timeout. */
	if (ct->status & IPS_SEEN_REPLY) {
		nf_ct_refresh_acct(ct, ctinfo, skb,
				   ct->proto.gre.stream_timeout);
		/* Also, more likely to be important, and not a probe. */
		if (!test_and_set_bit(IPS_ASSURED_BIT, &ct->status))
			nf_conntrack_event_cache(IPCT_ASSURED, ct);
	} else
		nf_ct_refresh_acct(ct, ctinfo, skb,
				   ct->proto.gre.timeout);

	return NF_ACCEPT;
}

/* Called when a new connection for this protocol found. */
static bool gre_new(struct nf_conn *ct, const struct sk_buff *skb,
		    unsigned int dataoff, unsigned int *timeouts)
{
	pr_debug(": ");
	nf_ct_dump_tuple(&ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple);

	/* initialize to sane value.  Ideally a conntrack helper
	 * (e.g. in case of pptp) is increasing them */
	ct->proto.gre.stream_timeout = timeouts[GRE_CT_REPLIED];
	ct->proto.gre.timeout = timeouts[GRE_CT_UNREPLIED];

	return true;
}

/* Called when a conntrack entry has already been removed from the hashes
 * and is about to be deleted from memory */
static void gre_destroy(struct nf_conn *ct)
{
	struct nf_conn *master = ct->master;
	pr_debug(" entering\n");

	if (!master)
		pr_debug("no master !?!\n");
	else
		nf_ct_gre_keymap_destroy(master);
}

#if IS_ENABLED(CONFIG_NF_CT_NETLINK_TIMEOUT)

#include <linux/netfilter/nfnetlink.h>
#include <linux/netfilter/nfnetlink_cttimeout.h>

static int gre_timeout_nlattr_to_obj(struct nlattr *tb[],
				     struct net *net, void *data)
{
	unsigned int *timeouts = data;
	struct netns_proto_gre *net_gre = gre_pernet(net);

	/* set default timeouts for GRE. */
	timeouts[GRE_CT_UNREPLIED] = net_gre->gre_timeouts[GRE_CT_UNREPLIED];
	timeouts[GRE_CT_REPLIED] = net_gre->gre_timeouts[GRE_CT_REPLIED];

	if (tb[CTA_TIMEOUT_GRE_UNREPLIED]) {
		timeouts[GRE_CT_UNREPLIED] =
			ntohl(nla_get_be32(tb[CTA_TIMEOUT_GRE_UNREPLIED])) * HZ;
	}
	if (tb[CTA_TIMEOUT_GRE_REPLIED]) {
		timeouts[GRE_CT_REPLIED] =
			ntohl(nla_get_be32(tb[CTA_TIMEOUT_GRE_REPLIED])) * HZ;
	}
	return 0;
}

static int
gre_timeout_obj_to_nlattr(struct sk_buff *skb, const void *data)
{
	const unsigned int *timeouts = data;

	if (nla_put_be32(skb, CTA_TIMEOUT_GRE_UNREPLIED,
			 htonl(timeouts[GRE_CT_UNREPLIED] / HZ)) ||
	    nla_put_be32(skb, CTA_TIMEOUT_GRE_REPLIED,
			 htonl(timeouts[GRE_CT_REPLIED] / HZ)))
		goto nla_put_failure;
	return 0;

nla_put_failure:
	return -ENOSPC;
}

static const struct nla_policy
gre_timeout_nla_policy[CTA_TIMEOUT_GRE_MAX+1] = {
	[CTA_TIMEOUT_GRE_UNREPLIED]	= { .type = NLA_U32 },
	[CTA_TIMEOUT_GRE_REPLIED]	= { .type = NLA_U32 },
};
#endif /* CONFIG_NF_CT_NETLINK_TIMEOUT */

static int gre_init_net(struct net *net, u_int16_t proto)
{
	struct netns_proto_gre *net_gre = gre_pernet(net);
	int i;

	rwlock_init(&net_gre->keymap_lock);
	INIT_LIST_HEAD(&net_gre->keymap_list);
	for (i = 0; i < GRE_CT_MAX; i++)
		net_gre->gre_timeouts[i] = gre_timeouts[i];

	return 0;
}

/* protocol helper struct */
static struct nf_conntrack_l4proto nf_conntrack_l4proto_gre4 __read_mostly = {
	.l3proto	 = AF_INET,
	.l4proto	 = IPPROTO_GRE,
	.name		 = "gre",
	.pkt_to_tuple	 = gre_pkt_to_tuple,
	.invert_tuple	 = gre_invert_tuple,
	.print_tuple	 = gre_print_tuple,
	.print_conntrack = gre_print_conntrack,
	.get_timeouts    = gre_get_timeouts,
	.packet		 = gre_packet,
	.new		 = gre_new,
	.destroy	 = gre_destroy,
	.me 		 = THIS_MODULE,
#if IS_ENABLED(CONFIG_NF_CT_NETLINK)
	.tuple_to_nlattr = nf_ct_port_tuple_to_nlattr,
	.nlattr_tuple_size = nf_ct_port_nlattr_tuple_size,
	.nlattr_to_tuple = nf_ct_port_nlattr_to_tuple,
	.nla_policy	 = nf_ct_port_nla_policy,
#endif
#if IS_ENABLED(CONFIG_NF_CT_NETLINK_TIMEOUT)
	.ctnl_timeout    = {
		.nlattr_to_obj	= gre_timeout_nlattr_to_obj,
		.obj_to_nlattr	= gre_timeout_obj_to_nlattr,
		.nlattr_max	= CTA_TIMEOUT_GRE_MAX,
		.obj_size	= sizeof(unsigned int) * GRE_CT_MAX,
		.nla_policy	= gre_timeout_nla_policy,
	},
#endif /* CONFIG_NF_CT_NETLINK_TIMEOUT */
	.net_id		= &proto_gre_net_id,
	.init_net	= gre_init_net,
};

static int proto_gre_net_init(struct net *net)
{
	int ret = 0;
	ret = nf_ct_l4proto_pernet_register(net, &nf_conntrack_l4proto_gre4);
	if (ret < 0)
		pr_err("nf_conntrack_gre4: pernet registration failed.\n");
	return ret;
}

static void proto_gre_net_exit(struct net *net)
{
	nf_ct_l4proto_pernet_unregister(net, &nf_conntrack_l4proto_gre4);
	nf_ct_gre_keymap_flush(net);
}

static struct pernet_operations proto_gre_net_ops = {
	.init = proto_gre_net_init,
	.exit = proto_gre_net_exit,
	.id   = &proto_gre_net_id,
	.size = sizeof(struct netns_proto_gre),
};

static int __init nf_ct_proto_gre_init(void)
{
	int ret;

	ret = register_pernet_subsys(&proto_gre_net_ops);
	if (ret < 0)
		goto out_pernet;

	ret = nf_ct_l4proto_register(&nf_conntrack_l4proto_gre4);
	if (ret < 0)
		goto out_gre4;

	return 0;
out_gre4:
	unregister_pernet_subsys(&proto_gre_net_ops);
out_pernet:
	return ret;
}

static void __exit nf_ct_proto_gre_fini(void)
{
	nf_ct_l4proto_unregister(&nf_conntrack_l4proto_gre4);
	unregister_pernet_subsys(&proto_gre_net_ops);
}

module_init(nf_ct_proto_gre_init);
module_exit(nf_ct_proto_gre_fini);

MODULE_LICENSE("GPL");
Commit	Line	Data
f09943fe PM	1	/*
	2	* ip_conntrack_proto_gre.c - Version 3.0
	3	*
	4	* Connection tracking protocol helper module for GRE.
	5	*
	6	* GRE is a generic encapsulation protocol, which is generally not very
	7	* suited for NAT, as it has no protocol-specific part as port numbers.
	8	*
	9	* It has an optional key field, which may help us distinguishing two
	10	* connections between the same two hosts.
	11	*
	12	* GRE is defined in RFC 1701 and RFC 1702, as well as RFC 2784
	13	*
	14	* PPTP is built on top of a modified version of GRE, and has a mandatory
	15	* field called "CallID", which serves us for the same purpose as the key
	16	* field in plain GRE.
	17	*
	18	* Documentation about PPTP can be found in RFC 2637
	19	*
	20	* (C) 2000-2005 by Harald Welte <laforge@gnumonks.org>
	21	*
	22	* Development of this code funded by Astaro AG (http://www.astaro.com/)
	23	*
f229f6ce	24	* (C) 2006-2012 Patrick McHardy <kaber@trash.net>
f09943fe PM	25	*/
	26
	27	#include <linux/module.h>
	28	#include <linux/types.h>
	29	#include <linux/timer.h>
	30	#include <linux/list.h>
	31	#include <linux/seq_file.h>
	32	#include <linux/in.h>
3bb0d1c0	33	#include <linux/netdevice.h>
f09943fe	34	#include <linux/skbuff.h>
5a0e3ad6	35	#include <linux/slab.h>
3bb0d1c0 AD	36	#include <net/dst.h>
	37	#include <net/net_namespace.h>
	38	#include <net/netns/generic.h>
f09943fe PM	39	#include <net/netfilter/nf_conntrack_l4proto.h>
	40	#include <net/netfilter/nf_conntrack_helper.h>
	41	#include <net/netfilter/nf_conntrack_core.h>
	42	#include <linux/netfilter/nf_conntrack_proto_gre.h>
	43	#include <linux/netfilter/nf_conntrack_pptp.h>
	44
b888341c PNA	45	enum grep_conntrack {
	46	GRE_CT_UNREPLIED,
	47	GRE_CT_REPLIED,
	48	GRE_CT_MAX
	49	};
	50
	51	static unsigned int gre_timeouts[GRE_CT_MAX] = {
	52	[GRE_CT_UNREPLIED] = 30*HZ,
	53	[GRE_CT_REPLIED] = 180*HZ,
	54	};
f09943fe	55
f99189b1	56	static int proto_gre_net_id __read_mostly;
3bb0d1c0	57	struct netns_proto_gre {
4f71d80f	58	struct nf_proto_net nf;
3bb0d1c0 AD	59	rwlock_t keymap_lock;
3bb0d1c0 AD	60	struct list_head keymap_list;
4f71d80f	61	unsigned int gre_timeouts[GRE_CT_MAX];
3bb0d1c0	62	};
f09943fe	63
4f71d80f G	64	static inline struct netns_proto_gre gre_pernet(struct net net)
	65	{
	66	return net_generic(net, proto_gre_net_id);
	67	}
	68
8142b227	69	static void nf_ct_gre_keymap_flush(struct net *net)
f09943fe	70	{
4f71d80f	71	struct netns_proto_gre *net_gre = gre_pernet(net);
51807e91	72	struct nf_ct_gre_keymap km, tmp;
f09943fe	73
3bb0d1c0 AD	74	write_lock_bh(&net_gre->keymap_lock);
3bb0d1c0 AD	75	list_for_each_entry_safe(km, tmp, &net_gre->keymap_list, list) {
51807e91 AD	76	list_del(&km->list);
51807e91 AD	77	kfree(km);
f09943fe	78	}
3bb0d1c0	79	write_unlock_bh(&net_gre->keymap_lock);
f09943fe	80	}
f09943fe PM	81
	82	static inline int gre_key_cmpfn(const struct nf_ct_gre_keymap *km,
	83	const struct nf_conntrack_tuple *t)
	84	{
	85	return km->tuple.src.l3num == t->src.l3num &&
	86	!memcmp(&km->tuple.src.u3, &t->src.u3, sizeof(t->src.u3)) &&
	87	!memcmp(&km->tuple.dst.u3, &t->dst.u3, sizeof(t->dst.u3)) &&
	88	km->tuple.dst.protonum == t->dst.protonum &&
	89	km->tuple.dst.u.all == t->dst.u.all;
	90	}
	91
	92	/* look up the source key for a given tuple */
3bb0d1c0	93	static __be16 gre_keymap_lookup(struct net net, struct nf_conntrack_tuple t)
f09943fe	94	{
4f71d80f	95	struct netns_proto_gre *net_gre = gre_pernet(net);
f09943fe PM	96	struct nf_ct_gre_keymap *km;
	97	__be16 key = 0;
	98
3bb0d1c0 AD	99	read_lock_bh(&net_gre->keymap_lock);
3bb0d1c0 AD	100	list_for_each_entry(km, &net_gre->keymap_list, list) {
f09943fe PM	101	if (gre_key_cmpfn(km, t)) {
	102	key = km->tuple.src.u.gre.key;
	103	break;
	104	}
	105	}
3bb0d1c0	106	read_unlock_bh(&net_gre->keymap_lock);
f09943fe	107
0d53778e	108	pr_debug("lookup src key 0x%x for ", key);
3c9fba65	109	nf_ct_dump_tuple(t);
f09943fe PM	110
	111	return key;
	112	}
	113
	114	/* add a single keymap entry, associate with specified master ct */
	115	int nf_ct_gre_keymap_add(struct nf_conn *ct, enum ip_conntrack_dir dir,
	116	struct nf_conntrack_tuple *t)
	117	{
3bb0d1c0	118	struct net *net = nf_ct_net(ct);
4f71d80f	119	struct netns_proto_gre *net_gre = gre_pernet(net);
1afc5679	120	struct nf_ct_pptp_master *ct_pptp_info = nfct_help_data(ct);
f09943fe PM	121	struct nf_ct_gre_keymap *kmp, km;
f09943fe PM	122
1afc5679	123	kmp = &ct_pptp_info->keymap[dir];
f09943fe PM	124	if (*kmp) {
f09943fe PM	125	/* check whether it's a retransmission */
3bb0d1c0 AD	126	read_lock_bh(&net_gre->keymap_lock);
3bb0d1c0 AD	127	list_for_each_entry(km, &net_gre->keymap_list, list) {
887464a4	128	if (gre_key_cmpfn(km, t) && km == *kmp) {
3bb0d1c0	129	read_unlock_bh(&net_gre->keymap_lock);
f09943fe	130	return 0;
887464a4	131	}
f09943fe	132	}
3bb0d1c0	133	read_unlock_bh(&net_gre->keymap_lock);
0d53778e PM	134	pr_debug("trying to override keymap_%s for ct %p\n",
0d53778e PM	135	dir == IP_CT_DIR_REPLY ? "reply" : "orig", ct);
f09943fe PM	136	return -EEXIST;
	137	}
	138
	139	km = kmalloc(sizeof(*km), GFP_ATOMIC);
	140	if (!km)
	141	return -ENOMEM;
	142	memcpy(&km->tuple, t, sizeof(*t));
	143	*kmp = km;
	144
0d53778e	145	pr_debug("adding new entry %p: ", km);
3c9fba65	146	nf_ct_dump_tuple(&km->tuple);
f09943fe	147
3bb0d1c0 AD	148	write_lock_bh(&net_gre->keymap_lock);
	149	list_add_tail(&km->list, &net_gre->keymap_list);
	150	write_unlock_bh(&net_gre->keymap_lock);
f09943fe PM	151
	152	return 0;
	153	}
	154	EXPORT_SYMBOL_GPL(nf_ct_gre_keymap_add);
	155
	156	/* destroy the keymap entries associated with specified master ct */
	157	void nf_ct_gre_keymap_destroy(struct nf_conn *ct)
	158	{
3bb0d1c0	159	struct net *net = nf_ct_net(ct);
4f71d80f	160	struct netns_proto_gre *net_gre = gre_pernet(net);
1afc5679	161	struct nf_ct_pptp_master *ct_pptp_info = nfct_help_data(ct);
f09943fe PM	162	enum ip_conntrack_dir dir;
f09943fe PM	163
0d53778e	164	pr_debug("entering for ct %p\n", ct);
f09943fe	165
3bb0d1c0	166	write_lock_bh(&net_gre->keymap_lock);
f09943fe	167	for (dir = IP_CT_DIR_ORIGINAL; dir < IP_CT_DIR_MAX; dir++) {
1afc5679	168	if (ct_pptp_info->keymap[dir]) {
0d53778e	169	pr_debug("removing %p from list\n",
1afc5679 PNA	170	ct_pptp_info->keymap[dir]);
	171	list_del(&ct_pptp_info->keymap[dir]->list);
	172	kfree(ct_pptp_info->keymap[dir]);
	173	ct_pptp_info->keymap[dir] = NULL;
f09943fe PM	174	}
f09943fe PM	175	}
3bb0d1c0	176	write_unlock_bh(&net_gre->keymap_lock);
f09943fe PM	177	}
	178	EXPORT_SYMBOL_GPL(nf_ct_gre_keymap_destroy);
	179
	180	/* PUBLIC CONNTRACK PROTO HELPER FUNCTIONS */
	181
	182	/* invert gre part of tuple */
09f263cd JE	183	static bool gre_invert_tuple(struct nf_conntrack_tuple *tuple,
09f263cd JE	184	const struct nf_conntrack_tuple *orig)
f09943fe PM	185	{
	186	tuple->dst.u.gre.key = orig->src.u.gre.key;
	187	tuple->src.u.gre.key = orig->dst.u.gre.key;
09f263cd	188	return true;
f09943fe PM	189	}
	190
	191	/* gre hdr info to tuple */
09f263cd JE	192	static bool gre_pkt_to_tuple(const struct sk_buff *skb, unsigned int dataoff,
09f263cd JE	193	struct nf_conntrack_tuple *tuple)
f09943fe	194	{
adf30907	195	struct net *net = dev_net(skb->dev ? skb->dev : skb_dst(skb)->dev);
dc35dc5a JE	196	const struct gre_hdr_pptp *pgrehdr;
dc35dc5a JE	197	struct gre_hdr_pptp _pgrehdr;
f09943fe	198	__be16 srckey;
dc35dc5a JE	199	const struct gre_hdr *grehdr;
dc35dc5a JE	200	struct gre_hdr _grehdr;
f09943fe PM	201
	202	/* first only delinearize old RFC1701 GRE header */
	203	grehdr = skb_header_pointer(skb, dataoff, sizeof(_grehdr), &_grehdr);
	204	if (!grehdr \|\| grehdr->version != GRE_VERSION_PPTP) {
	205	/* try to behave like "nf_conntrack_proto_generic" */
	206	tuple->src.u.all = 0;
	207	tuple->dst.u.all = 0;
09f263cd	208	return true;
f09943fe PM	209	}
	210
	211	/* PPTP header is variable length, only need up to the call_id field */
	212	pgrehdr = skb_header_pointer(skb, dataoff, 8, &_pgrehdr);
	213	if (!pgrehdr)
09f263cd	214	return true;
f09943fe PM	215
f09943fe PM	216	if (ntohs(grehdr->protocol) != GRE_PROTOCOL_PPTP) {
0d53778e	217	pr_debug("GRE_VERSION_PPTP but unknown proto\n");
09f263cd	218	return false;
f09943fe PM	219	}
	220
	221	tuple->dst.u.gre.key = pgrehdr->call_id;
3bb0d1c0	222	srckey = gre_keymap_lookup(net, tuple);
f09943fe PM	223	tuple->src.u.gre.key = srckey;
f09943fe PM	224
09f263cd	225	return true;
f09943fe PM	226	}
	227
	228	/* print gre part of tuple */
824f1fbe JP	229	static void gre_print_tuple(struct seq_file *s,
824f1fbe JP	230	const struct nf_conntrack_tuple *tuple)
f09943fe	231	{
824f1fbe JP	232	seq_printf(s, "srckey=0x%x dstkey=0x%x ",
	233	ntohs(tuple->src.u.gre.key),
	234	ntohs(tuple->dst.u.gre.key));
f09943fe PM	235	}
	236
	237	/* print private data for conntrack */
37246a58	238	static void gre_print_conntrack(struct seq_file s, struct nf_conn ct)
f09943fe	239	{
37246a58 SRRH	240	seq_printf(s, "timeout=%u, stream_timeout=%u ",
	241	(ct->proto.gre.timeout / HZ),
	242	(ct->proto.gre.stream_timeout / HZ));
f09943fe PM	243	}
f09943fe PM	244
2c8503f5 PNA	245	static unsigned int gre_get_timeouts(struct net net)
2c8503f5 PNA	246	{
4f71d80f	247	return gre_pernet(net)->gre_timeouts;
2c8503f5 PNA	248	}
2c8503f5 PNA	249
f09943fe PM	250	/* Returns verdict for packet, and may modify conntrack */
	251	static int gre_packet(struct nf_conn *ct,
	252	const struct sk_buff *skb,
	253	unsigned int dataoff,
	254	enum ip_conntrack_info ctinfo,
76108cea	255	u_int8_t pf,
2c8503f5 PNA	256	unsigned int hooknum,
2c8503f5 PNA	257	unsigned int *timeouts)
f09943fe PM	258	{
	259	/* If we've seen traffic both ways, this is a GRE connection.
	260	* Extend timeout. */
	261	if (ct->status & IPS_SEEN_REPLY) {
	262	nf_ct_refresh_acct(ct, ctinfo, skb,
	263	ct->proto.gre.stream_timeout);
	264	/* Also, more likely to be important, and not a probe. */
98d9ae84 FW	265	if (!test_and_set_bit(IPS_ASSURED_BIT, &ct->status))
98d9ae84 FW	266	nf_conntrack_event_cache(IPCT_ASSURED, ct);
f09943fe PM	267	} else
	268	nf_ct_refresh_acct(ct, ctinfo, skb,
	269	ct->proto.gre.timeout);
	270
	271	return NF_ACCEPT;
	272	}
	273
	274	/* Called when a new connection for this protocol found. */
09f263cd	275	static bool gre_new(struct nf_conn ct, const struct sk_buff skb,
2c8503f5	276	unsigned int dataoff, unsigned int *timeouts)
f09943fe	277	{
0d53778e	278	pr_debug(": ");
3c9fba65	279	nf_ct_dump_tuple(&ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple);
f09943fe PM	280
	281	/* initialize to sane value. Ideally a conntrack helper
	282	* (e.g. in case of pptp) is increasing them */
2c8503f5 PNA	283	ct->proto.gre.stream_timeout = timeouts[GRE_CT_REPLIED];
2c8503f5 PNA	284	ct->proto.gre.timeout = timeouts[GRE_CT_UNREPLIED];
f09943fe	285
09f263cd	286	return true;
f09943fe PM	287	}
	288
	289	/* Called when a conntrack entry has already been removed from the hashes
	290	* and is about to be deleted from memory */
	291	static void gre_destroy(struct nf_conn *ct)
	292	{
	293	struct nf_conn *master = ct->master;
0d53778e	294	pr_debug(" entering\n");
f09943fe PM	295
f09943fe PM	296	if (!master)
0d53778e	297	pr_debug("no master !?!\n");
f09943fe PM	298	else
	299	nf_ct_gre_keymap_destroy(master);
	300	}
	301
50978462 PNA	302	#if IS_ENABLED(CONFIG_NF_CT_NETLINK_TIMEOUT)
	303
	304	#include <linux/netfilter/nfnetlink.h>
	305	#include <linux/netfilter/nfnetlink_cttimeout.h>
	306
8264deb8 G	307	static int gre_timeout_nlattr_to_obj(struct nlattr *tb[],
8264deb8 G	308	struct net net, void data)
50978462 PNA	309	{
50978462 PNA	310	unsigned int *timeouts = data;
8264deb8	311	struct netns_proto_gre *net_gre = gre_pernet(net);
50978462 PNA	312
50978462 PNA	313	/* set default timeouts for GRE. */
8264deb8 G	314	timeouts[GRE_CT_UNREPLIED] = net_gre->gre_timeouts[GRE_CT_UNREPLIED];
8264deb8 G	315	timeouts[GRE_CT_REPLIED] = net_gre->gre_timeouts[GRE_CT_REPLIED];
50978462 PNA	316
	317	if (tb[CTA_TIMEOUT_GRE_UNREPLIED]) {
	318	timeouts[GRE_CT_UNREPLIED] =
	319	ntohl(nla_get_be32(tb[CTA_TIMEOUT_GRE_UNREPLIED])) * HZ;
	320	}
	321	if (tb[CTA_TIMEOUT_GRE_REPLIED]) {
	322	timeouts[GRE_CT_REPLIED] =
	323	ntohl(nla_get_be32(tb[CTA_TIMEOUT_GRE_REPLIED])) * HZ;
	324	}
	325	return 0;
	326	}
	327
	328	static int
	329	gre_timeout_obj_to_nlattr(struct sk_buff skb, const void data)
	330	{
	331	const unsigned int *timeouts = data;
	332
242ddfc0 DM	333	if (nla_put_be32(skb, CTA_TIMEOUT_GRE_UNREPLIED,
	334	htonl(timeouts[GRE_CT_UNREPLIED] / HZ)) \|\|
	335	nla_put_be32(skb, CTA_TIMEOUT_GRE_REPLIED,
	336	htonl(timeouts[GRE_CT_REPLIED] / HZ)))
	337	goto nla_put_failure;
50978462 PNA	338	return 0;
	339
	340	nla_put_failure:
	341	return -ENOSPC;
	342	}
	343
	344	static const struct nla_policy
	345	gre_timeout_nla_policy[CTA_TIMEOUT_GRE_MAX+1] = {
	346	[CTA_TIMEOUT_GRE_UNREPLIED] = { .type = NLA_U32 },
	347	[CTA_TIMEOUT_GRE_REPLIED] = { .type = NLA_U32 },
	348	};
	349	#endif /* CONFIG_NF_CT_NETLINK_TIMEOUT */
	350
f1caad27	351	static int gre_init_net(struct net *net, u_int16_t proto)
4f71d80f G	352	{
	353	struct netns_proto_gre *net_gre = gre_pernet(net);
	354	int i;
	355
	356	rwlock_init(&net_gre->keymap_lock);
	357	INIT_LIST_HEAD(&net_gre->keymap_list);
	358	for (i = 0; i < GRE_CT_MAX; i++)
	359	net_gre->gre_timeouts[i] = gre_timeouts[i];
	360
	361	return 0;
	362	}
	363
f09943fe	364	/* protocol helper struct */
61075af5	365	static struct nf_conntrack_l4proto nf_conntrack_l4proto_gre4 __read_mostly = {
f09943fe PM	366	.l3proto = AF_INET,
	367	.l4proto = IPPROTO_GRE,
	368	.name = "gre",
	369	.pkt_to_tuple = gre_pkt_to_tuple,
	370	.invert_tuple = gre_invert_tuple,
	371	.print_tuple = gre_print_tuple,
	372	.print_conntrack = gre_print_conntrack,
2c8503f5	373	.get_timeouts = gre_get_timeouts,
f09943fe PM	374	.packet = gre_packet,
	375	.new = gre_new,
	376	.destroy = gre_destroy,
	377	.me = THIS_MODULE,
c0cd1156	378	#if IS_ENABLED(CONFIG_NF_CT_NETLINK)
fdf70832	379	.tuple_to_nlattr = nf_ct_port_tuple_to_nlattr,
a400c30e	380	.nlattr_tuple_size = nf_ct_port_nlattr_tuple_size,
fdf70832	381	.nlattr_to_tuple = nf_ct_port_nlattr_to_tuple,
f73e924c	382	.nla_policy = nf_ct_port_nla_policy,
f09943fe	383	#endif
50978462 PNA	384	#if IS_ENABLED(CONFIG_NF_CT_NETLINK_TIMEOUT)
	385	.ctnl_timeout = {
	386	.nlattr_to_obj = gre_timeout_nlattr_to_obj,
	387	.obj_to_nlattr = gre_timeout_obj_to_nlattr,
	388	.nlattr_max = CTA_TIMEOUT_GRE_MAX,
	389	.obj_size = sizeof(unsigned int) * GRE_CT_MAX,
	390	.nla_policy = gre_timeout_nla_policy,
	391	},
	392	#endif /* CONFIG_NF_CT_NETLINK_TIMEOUT */
4f71d80f G	393	.net_id = &proto_gre_net_id,
4f71d80f G	394	.init_net = gre_init_net,
f09943fe PM	395	};
f09943fe PM	396
3bb0d1c0 AD	397	static int proto_gre_net_init(struct net *net)
3bb0d1c0 AD	398	{
4f71d80f	399	int ret = 0;
c296bb4d	400	ret = nf_ct_l4proto_pernet_register(net, &nf_conntrack_l4proto_gre4);
4f71d80f	401	if (ret < 0)
c296bb4d	402	pr_err("nf_conntrack_gre4: pernet registration failed.\n");
4f71d80f	403	return ret;
3bb0d1c0 AD	404	}
	405
	406	static void proto_gre_net_exit(struct net *net)
	407	{
c296bb4d	408	nf_ct_l4proto_pernet_unregister(net, &nf_conntrack_l4proto_gre4);
3bb0d1c0	409	nf_ct_gre_keymap_flush(net);
3bb0d1c0 AD	410	}
	411
	412	static struct pernet_operations proto_gre_net_ops = {
	413	.init = proto_gre_net_init,
	414	.exit = proto_gre_net_exit,
e8d02885 EB	415	.id = &proto_gre_net_id,
e8d02885 EB	416	.size = sizeof(struct netns_proto_gre),
3bb0d1c0 AD	417	};
3bb0d1c0 AD	418
f09943fe PM	419	static int __init nf_ct_proto_gre_init(void)
f09943fe PM	420	{
c296bb4d G	421	int ret;
c296bb4d G	422
c296bb4d G	423	ret = register_pernet_subsys(&proto_gre_net_ops);
	424	if (ret < 0)
	425	goto out_pernet;
	426
0d98da5d G	427	ret = nf_ct_l4proto_register(&nf_conntrack_l4proto_gre4);
	428	if (ret < 0)
	429	goto out_gre4;
	430
c296bb4d	431	return 0;
c296bb4d	432	out_gre4:
0d98da5d G	433	unregister_pernet_subsys(&proto_gre_net_ops);
0d98da5d G	434	out_pernet:
c296bb4d	435	return ret;
f09943fe PM	436	}
f09943fe PM	437
56bc0f96	438	static void __exit nf_ct_proto_gre_fini(void)
f09943fe	439	{
c296bb4d	440	nf_ct_l4proto_unregister(&nf_conntrack_l4proto_gre4);
e8d02885	441	unregister_pernet_subsys(&proto_gre_net_ops);
f09943fe PM	442	}
	443
	444	module_init(nf_ct_proto_gre_init);
	445	module_exit(nf_ct_proto_gre_fini);
	446
	447	MODULE_LICENSE("GPL");