1 diff -Nurb linux-2.6.22-521/include/linux/netfilter/xt_MARK.h linux-2.6.22-522/include/linux/netfilter/xt_MARK.h
2 --- linux-2.6.22-521/include/linux/netfilter/xt_MARK.h 2007-07-08 19:32:17.000000000 -0400
3 +++ linux-2.6.22-522/include/linux/netfilter/xt_MARK.h 2008-09-17 17:59:53.000000000 -0400
11 struct xt_mark_target_info_v1 {
12 diff -Nurb linux-2.6.22-521/include/linux/netfilter/xt_SETXID.h linux-2.6.22-522/include/linux/netfilter/xt_SETXID.h
13 --- linux-2.6.22-521/include/linux/netfilter/xt_SETXID.h 1969-12-31 19:00:00.000000000 -0500
14 +++ linux-2.6.22-522/include/linux/netfilter/xt_SETXID.h 2008-09-17 17:59:53.000000000 -0400
16 +#ifndef _XT_SETXID_H_target
17 +#define _XT_SETXID_H_target
24 +struct xt_setxid_target_info_v1 {
29 +#endif /*_XT_SETXID_H_target*/
30 diff -Nurb linux-2.6.22-521/include/linux/netfilter_ipv4/ipt_MARK.h linux-2.6.22-522/include/linux/netfilter_ipv4/ipt_MARK.h
31 --- linux-2.6.22-521/include/linux/netfilter_ipv4/ipt_MARK.h 2007-07-08 19:32:17.000000000 -0400
32 +++ linux-2.6.22-522/include/linux/netfilter_ipv4/ipt_MARK.h 2008-09-17 17:59:53.000000000 -0400
34 #define IPT_MARK_SET XT_MARK_SET
35 #define IPT_MARK_AND XT_MARK_AND
36 #define IPT_MARK_OR XT_MARK_OR
37 +#define IPT_MARK_COPYXID XT_MARK_COPYXID
39 #define ipt_mark_target_info_v1 xt_mark_target_info_v1
41 diff -Nurb linux-2.6.22-521/include/linux/netfilter_ipv4/ipt_SETXID.h linux-2.6.22-522/include/linux/netfilter_ipv4/ipt_SETXID.h
42 --- linux-2.6.22-521/include/linux/netfilter_ipv4/ipt_SETXID.h 1969-12-31 19:00:00.000000000 -0500
43 +++ linux-2.6.22-522/include/linux/netfilter_ipv4/ipt_SETXID.h 2008-09-17 17:59:53.000000000 -0400
45 +#ifndef _IPT_SETXID_H_target
46 +#define _IPT_SETXID_H_target
48 +/* Backwards compatibility for old userspace */
50 +#include <linux/netfilter/xt_SETXID.h>
53 +#define IPT_SET_PACKET_XID XT_SET_PACKET_XID
55 +#define ipt_setxid_target_info_v1 xt_setxid_target_info_v1
57 +#endif /*_IPT_SETXID_H_target*/
58 diff -Nurb linux-2.6.22-521/include/net/netfilter/nf_conntrack.h linux-2.6.22-522/include/net/netfilter/nf_conntrack.h
59 --- linux-2.6.22-521/include/net/netfilter/nf_conntrack.h 2007-07-08 19:32:17.000000000 -0400
60 +++ linux-2.6.22-522/include/net/netfilter/nf_conntrack.h 2008-09-17 17:59:53.000000000 -0400
62 /* Storage reserved for other modules: */
63 union nf_conntrack_proto proto;
65 + /* PLANETLAB. VNET-specific */
66 + int xid[IP_CT_DIR_MAX];
68 /* features dynamically at the end: helper, nat (both optional) */
71 diff -Nurb linux-2.6.22-521/net/netfilter/Kconfig linux-2.6.22-522/net/netfilter/Kconfig
72 --- linux-2.6.22-521/net/netfilter/Kconfig 2007-07-08 19:32:17.000000000 -0400
73 +++ linux-2.6.22-522/net/netfilter/Kconfig 2008-09-17 17:59:53.000000000 -0400
76 To compile it as a module, choose M here. If unsure, say N.
78 +config NETFILTER_XT_TARGET_SETXID
79 + tristate '"SETXID" target support'
80 + depends on NETFILTER_XTABLES
82 + This option adds a `SETXID' target, which allows you to alter the
85 config NETFILTER_XT_MATCH_COMMENT
86 tristate '"comment" match support'
87 depends on NETFILTER_XTABLES
88 diff -Nurb linux-2.6.22-521/net/netfilter/Makefile linux-2.6.22-522/net/netfilter/Makefile
89 --- linux-2.6.22-521/net/netfilter/Makefile 2007-07-08 19:32:17.000000000 -0400
90 +++ linux-2.6.22-522/net/netfilter/Makefile 2008-09-17 17:59:53.000000000 -0400
92 obj-$(CONFIG_NETFILTER_XTABLES) += x_tables.o xt_tcpudp.o
95 +obj-$(CONFIG_NETFILTER_XT_TARGET_SETXID) += xt_SETXID.o
96 obj-$(CONFIG_NETFILTER_XT_TARGET_CLASSIFY) += xt_CLASSIFY.o
97 obj-$(CONFIG_NETFILTER_XT_TARGET_CONNMARK) += xt_CONNMARK.o
98 obj-$(CONFIG_NETFILTER_XT_TARGET_DSCP) += xt_DSCP.o
99 diff -Nurb linux-2.6.22-521/net/netfilter/nf_conntrack_core.c linux-2.6.22-522/net/netfilter/nf_conntrack_core.c
100 --- linux-2.6.22-521/net/netfilter/nf_conntrack_core.c 2007-07-08 19:32:17.000000000 -0400
101 +++ linux-2.6.22-522/net/netfilter/nf_conntrack_core.c 2008-09-17 17:59:53.000000000 -0400
104 /* Overload tuple linked list to put us in unconfirmed list. */
105 list_add(&conntrack->tuplehash[IP_CT_DIR_ORIGINAL].list, &unconfirmed);
106 + conntrack->xid[IP_CT_DIR_ORIGINAL] = -1;
107 + conntrack->xid[IP_CT_DIR_REPLY] = -1;
109 write_unlock_bh(&nf_conntrack_lock);
111 diff -Nurb linux-2.6.22-521/net/netfilter/xt_MARK.c linux-2.6.22-522/net/netfilter/xt_MARK.c
112 --- linux-2.6.22-521/net/netfilter/xt_MARK.c 2007-07-08 19:32:17.000000000 -0400
113 +++ linux-2.6.22-522/net/netfilter/xt_MARK.c 2008-09-17 18:29:52.000000000 -0400
115 * This program is free software; you can redistribute it and/or modify
116 * it under the terms of the GNU General Public License version 2 as
117 * published by the Free Software Foundation.
121 #include <linux/module.h>
122 +#include <linux/version.h>
123 #include <linux/skbuff.h>
124 #include <linux/ip.h>
125 +#include <net/udp.h>
126 #include <net/checksum.h>
127 +#include <net/route.h>
128 +#include <net/inet_hashtables.h>
130 +#include <net/netfilter/nf_conntrack.h>
131 #include <linux/netfilter/x_tables.h>
132 #include <linux/netfilter/xt_MARK.h>
135 MODULE_ALIAS("ipt_MARK");
136 MODULE_ALIAS("ip6t_MARK");
138 +#define PEERCRED_SET(x) ((x!=0) && (x!=(unsigned int)-1))
140 +static inline u_int16_t
141 +get_dst_port(struct nf_conntrack_tuple *tuple)
143 + switch (tuple->dst.protonum) {
145 + /* XXX Truncate 32-bit GRE key to 16 bits */
146 +#if LINUX_VERSION_CODE >= KERNEL_VERSION(2,6,11)
147 + return tuple->dst.u.gre.key;
149 + return htons(ntohl(tuple->dst.u.gre.key));
152 + /* Bind on ICMP echo ID */
153 + return tuple->src.u.icmp.id;
155 + return tuple->dst.u.tcp.port;
157 + return tuple->dst.u.udp.port;
159 + return tuple->dst.u.all;
163 +static inline u_int16_t
164 +get_src_port(struct nf_conntrack_tuple *tuple)
166 + switch (tuple->dst.protonum) {
168 + /* XXX Truncate 32-bit GRE key to 16 bits */
169 + return htons(ntohl(tuple->src.u.gre.key));
171 + /* Bind on ICMP echo ID */
172 + return tuple->src.u.icmp.id;
174 + return tuple->src.u.tcp.port;
176 + return tuple->src.u.udp.port;
178 + return tuple->src.u.all;
183 target_v0(struct sk_buff **pskb,
184 const struct net_device *in,
189 +extern DEFINE_PER_CPU(int, sknid_elevator);
191 +static struct sock *__udp4_lib_lookup(__be32 saddr, __be16 sport,
192 + __be32 daddr, __be16 dport,
193 + int dif, struct hlist_head udptable[])
195 + struct sock *sk, *result = NULL;
196 + struct hlist_node *node;
197 + unsigned short hnum = ntohs(dport);
200 + read_lock(&udp_hash_lock);
202 + sk_for_each(sk, node, &udptable[hnum & (UDP_HTABLE_SIZE - 1)]) {
203 + struct inet_sock *inet = inet_sk(sk);
205 + if (sk->sk_hash == hnum && !ipv6_only_sock(sk)) {
206 + int score = (sk->sk_family == PF_INET ? 1 : 0);
208 + if (inet->rcv_saddr) {
209 + if (inet->rcv_saddr != daddr)
213 + /* block non nx_info ips */
214 + if (!v4_addr_in_nx_info(sk->sk_nx_info,
215 + daddr, NXA_MASK_BIND))
219 + if (inet->daddr != saddr)
224 + if (inet->dport != sport)
228 + if (sk->sk_bound_dev_if) {
229 + if (sk->sk_bound_dev_if != dif)
236 + } else if (score > badness) {
245 + read_unlock(&udp_hash_lock);
249 +#define related(ct) (ct==(IP_CT_IS_REPLY + IP_CT_RELATED))
252 target_v1(struct sk_buff **pskb,
253 const struct net_device *in,
255 const void *targinfo)
257 const struct xt_mark_target_info_v1 *markinfo = targinfo;
259 + enum ip_conntrack_info ctinfo;
260 + struct sock *connection_sk;
262 + struct nf_conn *ct;
263 + extern struct inet_hashinfo tcp_hashinfo;
264 + enum ip_conntrack_dir dir;
268 + u_int16_t proto, src_port;
274 switch (markinfo->mode) {
276 @@ -58,13 +183,126 @@
278 mark = (*pskb)->mark | markinfo->mark;
281 + case XT_MARK_COPYXID:
282 + dif = ((struct rtable *)(*pskb)->dst)->rt_iif;
284 + ct = nf_ct_get((*pskb), &ctinfo);
288 + dir = CTINFO2DIR(ctinfo);
289 + src_ip = ct->tuplehash[dir].tuple.src.u3.ip;
290 + dst_ip = ct->tuplehash[dir].tuple.dst.u3.ip;
291 + src_port = get_src_port(&ct->tuplehash[dir].tuple);
292 + proto = ct->tuplehash[dir].tuple.dst.protonum;
294 + ip = ct->tuplehash[dir].tuple.dst.u3.ip;
295 + port = get_dst_port(&ct->tuplehash[dir].tuple);
298 + if ((*pskb)->mark>0) /* The packet is marked, it's going out */
300 + ct->xid[0]=(*pskb)->mark;
303 + if (ct->xid[0] > 0) {
307 + else if (proto == 17) {
309 + if (!(*pskb)->mark) {
310 + sk = __udp4_lib_lookup(src_ip, src_port, ip, port,
313 + if (sk && hooknum==NF_IP_LOCAL_IN) {
322 + if ((*pskb)->mark>0) /* The packet is marked, it's going out */
324 + ct->xid[0]=(*pskb)->mark;
327 + else if (proto == 6) /* TCP */{
328 + int sockettype=0; /* Established socket */
329 + /* Looks for an established socket or a listening socket corresponding to the 4-tuple, in
330 + * that order. The order is important for Codemux connections to be handled properly */
332 + connection_sk = inet_lookup_established(&tcp_hashinfo, src_ip, src_port, ip, port, dif);
334 + if (!connection_sk) {
335 + connection_sk = inet_lookup_listener(&tcp_hashinfo, ip, port, dif);
336 + sockettype=1; /* Listening socket */
339 + if (connection_sk) {
340 + /* The peercred is not set. We set it if the other side has an xid. */
341 + if (!PEERCRED_SET(connection_sk->sk_peercred.uid)
342 + && ct->xid[!dir]>0 && (sockettype==0)) {
343 + connection_sk->sk_peercred.gid = connection_sk->sk_peercred.uid = ct->xid[!dir];
346 + /* The peercred is set, and is not equal to the XID of 'the other side' */
347 + else if (PEERCRED_SET(connection_sk->sk_peercred.uid) && (connection_sk->sk_peercred.uid != ct->xid[!dir]) && (sockettype==0)) {
348 + mark = connection_sk->sk_peercred.uid;
351 + /* Has this connection already been tagged? */
352 + if (ct->xid[dir] < 1) {
353 + /* No - let's tag it */
354 + ct->xid[dir]=connection_sk->sk_nid;
358 + if (mark==-1 && (ct->xid[dir]!= 0))
359 + mark = ct->xid[dir];
361 + if (connection_sk->sk_state == TCP_TIME_WAIT) {
362 + inet_twsk_put(inet_twsk(connection_sk));
366 + sock_put(connection_sk);
369 + /* All else failed. Is this a connection over raw sockets? That explains
370 + * why we couldn't get anything out of skb->sk, or look up a "real" connection.*/
371 + if (ct->xid[dir]<1) {
372 + if ((*pskb)->skb_tag) {
373 + ct->xid[dir]=(*pskb)->skb_tag;
377 + /* Covers CoDemux case */
378 + if (mark < 1 && (ct->xid[dir]>0)) {
379 + mark = ct->xid[dir];
382 + if (mark < 1 && (ct->xid[!dir]>0)) {
383 + mark = ct->xid[!dir];
389 (*pskb)->mark = mark;
392 + curtag=&__get_cpu_var(sknid_elevator);
393 + if (mark > 0 && *curtag==-2 && hooknum==NF_IP_LOCAL_IN)
402 checkentry_v0(const char *tablename,
406 if (markinfo->mode != XT_MARK_SET
407 && markinfo->mode != XT_MARK_AND
408 - && markinfo->mode != XT_MARK_OR) {
409 + && markinfo->mode != XT_MARK_OR
410 + && markinfo->mode != XT_MARK_COPYXID) {
411 printk(KERN_WARNING "MARK: unknown mode %u\n",
414 diff -Nurb linux-2.6.22-521/net/netfilter/xt_SETXID.c linux-2.6.22-522/net/netfilter/xt_SETXID.c
415 --- linux-2.6.22-521/net/netfilter/xt_SETXID.c 1969-12-31 19:00:00.000000000 -0500
416 +++ linux-2.6.22-522/net/netfilter/xt_SETXID.c 2008-09-17 17:59:53.000000000 -0400
418 +#include <linux/module.h>
419 +#include <linux/skbuff.h>
420 +#include <linux/ip.h>
421 +#include <net/checksum.h>
422 +#include <linux/vs_network.h>
424 +#include <linux/netfilter/x_tables.h>
425 +#include <linux/netfilter/xt_SETXID.h>
427 +MODULE_LICENSE("GPL");
429 +MODULE_DESCRIPTION("");
430 +MODULE_ALIAS("ipt_SETXID");
433 +target_v1(struct sk_buff **pskb,
434 + const struct net_device *in,
435 + const struct net_device *out,
436 + unsigned int hooknum,
437 + const struct xt_target *target,
438 + const void *targinfo)
440 + const struct xt_setxid_target_info_v1 *setxidinfo = targinfo;
442 + switch (setxidinfo->mode) {
443 + case XT_SET_PACKET_XID:
444 + (*pskb)->skb_tag = setxidinfo->mark;
447 + return XT_CONTINUE;
452 +checkentry_v1(const char *tablename,
454 + const struct xt_target *target,
456 + unsigned int hook_mask)
458 + struct xt_setxid_target_info_v1 *setxidinfo = targinfo;
460 + if (setxidinfo->mode != XT_SET_PACKET_XID) {
461 + printk(KERN_WARNING "SETXID: unknown mode %u\n",
469 +static struct xt_target xt_setxid_target[] = {
474 + .checkentry = checkentry_v1,
475 + .target = target_v1,
476 + .targetsize = sizeof(struct xt_setxid_target_info_v1),
482 +static int __init init(void)
486 + err = xt_register_targets(xt_setxid_target, ARRAY_SIZE(xt_setxid_target));
490 +static void __exit fini(void)
492 + xt_unregister_targets(xt_setxid_target, ARRAY_SIZE(xt_setxid_target));