2 * Licensed under GNU GPL version 2 Copyright Magnus Boden <mb@ozaba.mine.nu>
5 * Thu 21 Mar 2002 Harald Welte <laforge@gnumonks.org>
10 #include <linux/module.h>
12 #include <linux/udp.h>
14 #include <linux/netfilter.h>
15 #include <linux/netfilter_ipv4/ip_tables.h>
16 #include <linux/netfilter_ipv4/ip_conntrack_helper.h>
17 #include <linux/netfilter_ipv4/ip_conntrack_tftp.h>
19 MODULE_AUTHOR("Magnus Boden <mb@ozaba.mine.nu>");
20 MODULE_DESCRIPTION("Netfilter connection tracking module for tftp");
21 MODULE_LICENSE("GPL");
24 static int ports[MAX_PORTS];
25 static int ports_c = 0;
27 MODULE_PARM(ports, "1-" __MODULE_STRING(MAX_PORTS) "i");
28 MODULE_PARM_DESC(ports, "port numbers of tftp servers");
32 #define DEBUGP(format, args...) printk(__FILE__ ":" __FUNCTION__ ": " \
35 #define DEBUGP(format, args...)
38 static int tftp_help(struct sk_buff *skb,
39 struct ip_conntrack *ct,
40 enum ip_conntrack_info ctinfo)
43 struct ip_conntrack_expect exp;
45 if (skb_copy_bits(skb, skb->nh.iph->ihl * 4 + sizeof(struct udphdr),
46 &tftph, sizeof(tftph)) != 0)
49 switch (ntohs(tftph.opcode)) {
50 /* RRQ and WRQ works the same way */
51 case TFTP_OPCODE_READ:
52 case TFTP_OPCODE_WRITE:
54 DUMP_TUPLE(&ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple);
55 DUMP_TUPLE(&ct->tuplehash[IP_CT_DIR_REPLY].tuple);
56 memset(&exp, 0, sizeof(exp));
58 exp.tuple = ct->tuplehash[IP_CT_DIR_REPLY].tuple;
59 exp.mask.src.ip = 0xffffffff;
60 exp.mask.dst.ip = 0xffffffff;
61 exp.mask.dst.u.udp.port = 0xffff;
62 exp.mask.dst.protonum = 0xffff;
66 DUMP_TUPLE(&exp.tuple);
67 DUMP_TUPLE(&exp.mask);
68 ip_conntrack_expect_related(ct, &exp);
71 DEBUGP("Unknown opcode\n");
76 static struct ip_conntrack_helper tftp[MAX_PORTS];
77 static char tftp_names[MAX_PORTS][10];
79 static void fini(void)
83 for (i = 0 ; i < ports_c; i++) {
84 DEBUGP("unregistering helper for port %d\n",
86 ip_conntrack_helper_unregister(&tftp[i]);
90 static int __init init(void)
98 for (i = 0 ; (i < MAX_PORTS) && ports[i] ; i++) {
99 /* Create helper structure */
100 memset(&tftp[i], 0, sizeof(struct ip_conntrack_helper));
102 tftp[i].tuple.dst.protonum = IPPROTO_UDP;
103 tftp[i].tuple.src.u.udp.port = htons(ports[i]);
104 tftp[i].mask.dst.protonum = 0xFFFF;
105 tftp[i].mask.src.u.udp.port = 0xFFFF;
106 tftp[i].max_expected = 1;
108 tftp[i].flags = IP_CT_HELPER_F_REUSE_EXPECT;
109 tftp[i].me = THIS_MODULE;
110 tftp[i].help = tftp_help;
112 tmpname = &tftp_names[i][0];
113 if (ports[i] == TFTP_PORT)
114 sprintf(tmpname, "tftp");
116 sprintf(tmpname, "tftp-%d", i);
117 tftp[i].name = tmpname;
119 DEBUGP("port #%d: %d\n", i, ports[i]);
121 ret=ip_conntrack_helper_register(&tftp[i]);
123 printk("ERROR registering helper for port %d\n",
133 PROVIDES_CONNTRACK(tftp);