Linux-2.6.12-rc2
[linux-2.6/linux-acpi-2.6/ibm-acpi-2.6.git] / net / ipv4 / netfilter / ip_nat_snmp_basic.c
blob2a48b6e635aef35193fb3b70babe78d17c03519d
1 /*
2 * ip_nat_snmp_basic.c
4 * Basic SNMP Application Layer Gateway
6 * This IP NAT module is intended for use with SNMP network
7 * discovery and monitoring applications where target networks use
8 * conflicting private address realms.
10 * Static NAT is used to remap the networks from the view of the network
11 * management system at the IP layer, and this module remaps some application
12 * layer addresses to match.
14 * The simplest form of ALG is performed, where only tagged IP addresses
15 * are modified. The module does not need to be MIB aware and only scans
16 * messages at the ASN.1/BER level.
18 * Currently, only SNMPv1 and SNMPv2 are supported.
20 * More information on ALG and associated issues can be found in
21 * RFC 2962
23 * The ASB.1/BER parsing code is derived from the gxsnmp package by Gregory
24 * McLean & Jochen Friedrich, stripped down for use in the kernel.
26 * Copyright (c) 2000 RP Internet (www.rpi.net.au).
28 * This program is free software; you can redistribute it and/or modify
29 * it under the terms of the GNU General Public License as published by
30 * the Free Software Foundation; either version 2 of the License, or
31 * (at your option) any later version.
32 * This program is distributed in the hope that it will be useful,
33 * but WITHOUT ANY WARRANTY; without even the implied warranty of
34 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
35 * GNU General Public License for more details.
36 * You should have received a copy of the GNU General Public License
37 * along with this program; if not, write to the Free Software
38 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
40 * Author: James Morris <jmorris@intercode.com.au>
42 * Updates:
43 * 2000-08-06: Convert to new helper API (Harald Welte).
46 #include <linux/config.h>
47 #include <linux/module.h>
48 #include <linux/types.h>
49 #include <linux/kernel.h>
50 #include <linux/moduleparam.h>
51 #include <linux/netfilter_ipv4.h>
52 #include <linux/netfilter_ipv4/ip_nat.h>
53 #include <linux/netfilter_ipv4/ip_conntrack_helper.h>
54 #include <linux/netfilter_ipv4/ip_nat_helper.h>
55 #include <linux/ip.h>
56 #include <net/checksum.h>
57 #include <net/udp.h>
58 #include <asm/uaccess.h>
60 MODULE_LICENSE("GPL");
61 MODULE_AUTHOR("James Morris <jmorris@intercode.com.au>");
62 MODULE_DESCRIPTION("Basic SNMP Application Layer Gateway");
64 #define SNMP_PORT 161
65 #define SNMP_TRAP_PORT 162
66 #define NOCT1(n) (u_int8_t )((n) & 0xff)
68 static int debug;
69 static DEFINE_SPINLOCK(snmp_lock);
71 /*
72 * Application layer address mapping mimics the NAT mapping, but
73 * only for the first octet in this case (a more flexible system
74 * can be implemented if needed).
76 struct oct1_map
78 u_int8_t from;
79 u_int8_t to;
83 /*****************************************************************************
85 * Basic ASN.1 decoding routines (gxsnmp author Dirk Wisse)
87 *****************************************************************************/
89 /* Class */
90 #define ASN1_UNI 0 /* Universal */
91 #define ASN1_APL 1 /* Application */
92 #define ASN1_CTX 2 /* Context */
93 #define ASN1_PRV 3 /* Private */
95 /* Tag */
96 #define ASN1_EOC 0 /* End Of Contents */
97 #define ASN1_BOL 1 /* Boolean */
98 #define ASN1_INT 2 /* Integer */
99 #define ASN1_BTS 3 /* Bit String */
100 #define ASN1_OTS 4 /* Octet String */
101 #define ASN1_NUL 5 /* Null */
102 #define ASN1_OJI 6 /* Object Identifier */
103 #define ASN1_OJD 7 /* Object Description */
104 #define ASN1_EXT 8 /* External */
105 #define ASN1_SEQ 16 /* Sequence */
106 #define ASN1_SET 17 /* Set */
107 #define ASN1_NUMSTR 18 /* Numerical String */
108 #define ASN1_PRNSTR 19 /* Printable String */
109 #define ASN1_TEXSTR 20 /* Teletext String */
110 #define ASN1_VIDSTR 21 /* Video String */
111 #define ASN1_IA5STR 22 /* IA5 String */
112 #define ASN1_UNITIM 23 /* Universal Time */
113 #define ASN1_GENTIM 24 /* General Time */
114 #define ASN1_GRASTR 25 /* Graphical String */
115 #define ASN1_VISSTR 26 /* Visible String */
116 #define ASN1_GENSTR 27 /* General String */
118 /* Primitive / Constructed methods*/
119 #define ASN1_PRI 0 /* Primitive */
120 #define ASN1_CON 1 /* Constructed */
123 * Error codes.
125 #define ASN1_ERR_NOERROR 0
126 #define ASN1_ERR_DEC_EMPTY 2
127 #define ASN1_ERR_DEC_EOC_MISMATCH 3
128 #define ASN1_ERR_DEC_LENGTH_MISMATCH 4
129 #define ASN1_ERR_DEC_BADVALUE 5
132 * ASN.1 context.
134 struct asn1_ctx
136 int error; /* Error condition */
137 unsigned char *pointer; /* Octet just to be decoded */
138 unsigned char *begin; /* First octet */
139 unsigned char *end; /* Octet after last octet */
143 * Octet string (not null terminated)
145 struct asn1_octstr
147 unsigned char *data;
148 unsigned int len;
151 static void asn1_open(struct asn1_ctx *ctx,
152 unsigned char *buf,
153 unsigned int len)
155 ctx->begin = buf;
156 ctx->end = buf + len;
157 ctx->pointer = buf;
158 ctx->error = ASN1_ERR_NOERROR;
161 static unsigned char asn1_octet_decode(struct asn1_ctx *ctx, unsigned char *ch)
163 if (ctx->pointer >= ctx->end) {
164 ctx->error = ASN1_ERR_DEC_EMPTY;
165 return 0;
167 *ch = *(ctx->pointer)++;
168 return 1;
171 static unsigned char asn1_tag_decode(struct asn1_ctx *ctx, unsigned int *tag)
173 unsigned char ch;
175 *tag = 0;
179 if (!asn1_octet_decode(ctx, &ch))
180 return 0;
181 *tag <<= 7;
182 *tag |= ch & 0x7F;
183 } while ((ch & 0x80) == 0x80);
184 return 1;
187 static unsigned char asn1_id_decode(struct asn1_ctx *ctx,
188 unsigned int *cls,
189 unsigned int *con,
190 unsigned int *tag)
192 unsigned char ch;
194 if (!asn1_octet_decode(ctx, &ch))
195 return 0;
197 *cls = (ch & 0xC0) >> 6;
198 *con = (ch & 0x20) >> 5;
199 *tag = (ch & 0x1F);
201 if (*tag == 0x1F) {
202 if (!asn1_tag_decode(ctx, tag))
203 return 0;
205 return 1;
208 static unsigned char asn1_length_decode(struct asn1_ctx *ctx,
209 unsigned int *def,
210 unsigned int *len)
212 unsigned char ch, cnt;
214 if (!asn1_octet_decode(ctx, &ch))
215 return 0;
217 if (ch == 0x80)
218 *def = 0;
219 else {
220 *def = 1;
222 if (ch < 0x80)
223 *len = ch;
224 else {
225 cnt = (unsigned char) (ch & 0x7F);
226 *len = 0;
228 while (cnt > 0) {
229 if (!asn1_octet_decode(ctx, &ch))
230 return 0;
231 *len <<= 8;
232 *len |= ch;
233 cnt--;
237 return 1;
240 static unsigned char asn1_header_decode(struct asn1_ctx *ctx,
241 unsigned char **eoc,
242 unsigned int *cls,
243 unsigned int *con,
244 unsigned int *tag)
246 unsigned int def, len;
248 if (!asn1_id_decode(ctx, cls, con, tag))
249 return 0;
251 if (!asn1_length_decode(ctx, &def, &len))
252 return 0;
254 if (def)
255 *eoc = ctx->pointer + len;
256 else
257 *eoc = NULL;
258 return 1;
261 static unsigned char asn1_eoc_decode(struct asn1_ctx *ctx, unsigned char *eoc)
263 unsigned char ch;
265 if (eoc == 0) {
266 if (!asn1_octet_decode(ctx, &ch))
267 return 0;
269 if (ch != 0x00) {
270 ctx->error = ASN1_ERR_DEC_EOC_MISMATCH;
271 return 0;
274 if (!asn1_octet_decode(ctx, &ch))
275 return 0;
277 if (ch != 0x00) {
278 ctx->error = ASN1_ERR_DEC_EOC_MISMATCH;
279 return 0;
281 return 1;
282 } else {
283 if (ctx->pointer != eoc) {
284 ctx->error = ASN1_ERR_DEC_LENGTH_MISMATCH;
285 return 0;
287 return 1;
291 static unsigned char asn1_null_decode(struct asn1_ctx *ctx, unsigned char *eoc)
293 ctx->pointer = eoc;
294 return 1;
297 static unsigned char asn1_long_decode(struct asn1_ctx *ctx,
298 unsigned char *eoc,
299 long *integer)
301 unsigned char ch;
302 unsigned int len;
304 if (!asn1_octet_decode(ctx, &ch))
305 return 0;
307 *integer = (signed char) ch;
308 len = 1;
310 while (ctx->pointer < eoc) {
311 if (++len > sizeof (long)) {
312 ctx->error = ASN1_ERR_DEC_BADVALUE;
313 return 0;
316 if (!asn1_octet_decode(ctx, &ch))
317 return 0;
319 *integer <<= 8;
320 *integer |= ch;
322 return 1;
325 static unsigned char asn1_uint_decode(struct asn1_ctx *ctx,
326 unsigned char *eoc,
327 unsigned int *integer)
329 unsigned char ch;
330 unsigned int len;
332 if (!asn1_octet_decode(ctx, &ch))
333 return 0;
335 *integer = ch;
336 if (ch == 0) len = 0;
337 else len = 1;
339 while (ctx->pointer < eoc) {
340 if (++len > sizeof (unsigned int)) {
341 ctx->error = ASN1_ERR_DEC_BADVALUE;
342 return 0;
345 if (!asn1_octet_decode(ctx, &ch))
346 return 0;
348 *integer <<= 8;
349 *integer |= ch;
351 return 1;
354 static unsigned char asn1_ulong_decode(struct asn1_ctx *ctx,
355 unsigned char *eoc,
356 unsigned long *integer)
358 unsigned char ch;
359 unsigned int len;
361 if (!asn1_octet_decode(ctx, &ch))
362 return 0;
364 *integer = ch;
365 if (ch == 0) len = 0;
366 else len = 1;
368 while (ctx->pointer < eoc) {
369 if (++len > sizeof (unsigned long)) {
370 ctx->error = ASN1_ERR_DEC_BADVALUE;
371 return 0;
374 if (!asn1_octet_decode(ctx, &ch))
375 return 0;
377 *integer <<= 8;
378 *integer |= ch;
380 return 1;
383 static unsigned char asn1_octets_decode(struct asn1_ctx *ctx,
384 unsigned char *eoc,
385 unsigned char **octets,
386 unsigned int *len)
388 unsigned char *ptr;
390 *len = 0;
392 *octets = kmalloc(eoc - ctx->pointer, GFP_ATOMIC);
393 if (*octets == NULL) {
394 if (net_ratelimit())
395 printk("OOM in bsalg (%d)\n", __LINE__);
396 return 0;
399 ptr = *octets;
400 while (ctx->pointer < eoc) {
401 if (!asn1_octet_decode(ctx, (unsigned char *)ptr++)) {
402 kfree(*octets);
403 *octets = NULL;
404 return 0;
406 (*len)++;
408 return 1;
411 static unsigned char asn1_subid_decode(struct asn1_ctx *ctx,
412 unsigned long *subid)
414 unsigned char ch;
416 *subid = 0;
418 do {
419 if (!asn1_octet_decode(ctx, &ch))
420 return 0;
422 *subid <<= 7;
423 *subid |= ch & 0x7F;
424 } while ((ch & 0x80) == 0x80);
425 return 1;
428 static unsigned char asn1_oid_decode(struct asn1_ctx *ctx,
429 unsigned char *eoc,
430 unsigned long **oid,
431 unsigned int *len)
433 unsigned long subid;
434 unsigned int size;
435 unsigned long *optr;
437 size = eoc - ctx->pointer + 1;
438 *oid = kmalloc(size * sizeof(unsigned long), GFP_ATOMIC);
439 if (*oid == NULL) {
440 if (net_ratelimit())
441 printk("OOM in bsalg (%d)\n", __LINE__);
442 return 0;
445 optr = *oid;
447 if (!asn1_subid_decode(ctx, &subid)) {
448 kfree(*oid);
449 *oid = NULL;
450 return 0;
453 if (subid < 40) {
454 optr [0] = 0;
455 optr [1] = subid;
456 } else if (subid < 80) {
457 optr [0] = 1;
458 optr [1] = subid - 40;
459 } else {
460 optr [0] = 2;
461 optr [1] = subid - 80;
464 *len = 2;
465 optr += 2;
467 while (ctx->pointer < eoc) {
468 if (++(*len) > size) {
469 ctx->error = ASN1_ERR_DEC_BADVALUE;
470 kfree(*oid);
471 *oid = NULL;
472 return 0;
475 if (!asn1_subid_decode(ctx, optr++)) {
476 kfree(*oid);
477 *oid = NULL;
478 return 0;
481 return 1;
484 /*****************************************************************************
486 * SNMP decoding routines (gxsnmp author Dirk Wisse)
488 *****************************************************************************/
490 /* SNMP Versions */
491 #define SNMP_V1 0
492 #define SNMP_V2C 1
493 #define SNMP_V2 2
494 #define SNMP_V3 3
496 /* Default Sizes */
497 #define SNMP_SIZE_COMM 256
498 #define SNMP_SIZE_OBJECTID 128
499 #define SNMP_SIZE_BUFCHR 256
500 #define SNMP_SIZE_BUFINT 128
501 #define SNMP_SIZE_SMALLOBJECTID 16
503 /* Requests */
504 #define SNMP_PDU_GET 0
505 #define SNMP_PDU_NEXT 1
506 #define SNMP_PDU_RESPONSE 2
507 #define SNMP_PDU_SET 3
508 #define SNMP_PDU_TRAP1 4
509 #define SNMP_PDU_BULK 5
510 #define SNMP_PDU_INFORM 6
511 #define SNMP_PDU_TRAP2 7
513 /* Errors */
514 #define SNMP_NOERROR 0
515 #define SNMP_TOOBIG 1
516 #define SNMP_NOSUCHNAME 2
517 #define SNMP_BADVALUE 3
518 #define SNMP_READONLY 4
519 #define SNMP_GENERROR 5
520 #define SNMP_NOACCESS 6
521 #define SNMP_WRONGTYPE 7
522 #define SNMP_WRONGLENGTH 8
523 #define SNMP_WRONGENCODING 9
524 #define SNMP_WRONGVALUE 10
525 #define SNMP_NOCREATION 11
526 #define SNMP_INCONSISTENTVALUE 12
527 #define SNMP_RESOURCEUNAVAILABLE 13
528 #define SNMP_COMMITFAILED 14
529 #define SNMP_UNDOFAILED 15
530 #define SNMP_AUTHORIZATIONERROR 16
531 #define SNMP_NOTWRITABLE 17
532 #define SNMP_INCONSISTENTNAME 18
534 /* General SNMP V1 Traps */
535 #define SNMP_TRAP_COLDSTART 0
536 #define SNMP_TRAP_WARMSTART 1
537 #define SNMP_TRAP_LINKDOWN 2
538 #define SNMP_TRAP_LINKUP 3
539 #define SNMP_TRAP_AUTFAILURE 4
540 #define SNMP_TRAP_EQPNEIGHBORLOSS 5
541 #define SNMP_TRAP_ENTSPECIFIC 6
543 /* SNMPv1 Types */
544 #define SNMP_NULL 0
545 #define SNMP_INTEGER 1 /* l */
546 #define SNMP_OCTETSTR 2 /* c */
547 #define SNMP_DISPLAYSTR 2 /* c */
548 #define SNMP_OBJECTID 3 /* ul */
549 #define SNMP_IPADDR 4 /* uc */
550 #define SNMP_COUNTER 5 /* ul */
551 #define SNMP_GAUGE 6 /* ul */
552 #define SNMP_TIMETICKS 7 /* ul */
553 #define SNMP_OPAQUE 8 /* c */
555 /* Additional SNMPv2 Types */
556 #define SNMP_UINTEGER 5 /* ul */
557 #define SNMP_BITSTR 9 /* uc */
558 #define SNMP_NSAP 10 /* uc */
559 #define SNMP_COUNTER64 11 /* ul */
560 #define SNMP_NOSUCHOBJECT 12
561 #define SNMP_NOSUCHINSTANCE 13
562 #define SNMP_ENDOFMIBVIEW 14
564 union snmp_syntax
566 unsigned char uc[0]; /* 8 bit unsigned */
567 char c[0]; /* 8 bit signed */
568 unsigned long ul[0]; /* 32 bit unsigned */
569 long l[0]; /* 32 bit signed */
572 struct snmp_object
574 unsigned long *id;
575 unsigned int id_len;
576 unsigned short type;
577 unsigned int syntax_len;
578 union snmp_syntax syntax;
581 struct snmp_request
583 unsigned long id;
584 unsigned int error_status;
585 unsigned int error_index;
588 struct snmp_v1_trap
590 unsigned long *id;
591 unsigned int id_len;
592 unsigned long ip_address; /* pointer */
593 unsigned int general;
594 unsigned int specific;
595 unsigned long time;
598 /* SNMP types */
599 #define SNMP_IPA 0
600 #define SNMP_CNT 1
601 #define SNMP_GGE 2
602 #define SNMP_TIT 3
603 #define SNMP_OPQ 4
604 #define SNMP_C64 6
606 /* SNMP errors */
607 #define SERR_NSO 0
608 #define SERR_NSI 1
609 #define SERR_EOM 2
611 static inline void mangle_address(unsigned char *begin,
612 unsigned char *addr,
613 const struct oct1_map *map,
614 u_int16_t *check);
615 struct snmp_cnv
617 unsigned int class;
618 unsigned int tag;
619 int syntax;
622 static struct snmp_cnv snmp_conv [] =
624 {ASN1_UNI, ASN1_NUL, SNMP_NULL},
625 {ASN1_UNI, ASN1_INT, SNMP_INTEGER},
626 {ASN1_UNI, ASN1_OTS, SNMP_OCTETSTR},
627 {ASN1_UNI, ASN1_OTS, SNMP_DISPLAYSTR},
628 {ASN1_UNI, ASN1_OJI, SNMP_OBJECTID},
629 {ASN1_APL, SNMP_IPA, SNMP_IPADDR},
630 {ASN1_APL, SNMP_CNT, SNMP_COUNTER}, /* Counter32 */
631 {ASN1_APL, SNMP_GGE, SNMP_GAUGE}, /* Gauge32 == Unsigned32 */
632 {ASN1_APL, SNMP_TIT, SNMP_TIMETICKS},
633 {ASN1_APL, SNMP_OPQ, SNMP_OPAQUE},
635 /* SNMPv2 data types and errors */
636 {ASN1_UNI, ASN1_BTS, SNMP_BITSTR},
637 {ASN1_APL, SNMP_C64, SNMP_COUNTER64},
638 {ASN1_CTX, SERR_NSO, SNMP_NOSUCHOBJECT},
639 {ASN1_CTX, SERR_NSI, SNMP_NOSUCHINSTANCE},
640 {ASN1_CTX, SERR_EOM, SNMP_ENDOFMIBVIEW},
641 {0, 0, -1}
644 static unsigned char snmp_tag_cls2syntax(unsigned int tag,
645 unsigned int cls,
646 unsigned short *syntax)
648 struct snmp_cnv *cnv;
650 cnv = snmp_conv;
652 while (cnv->syntax != -1) {
653 if (cnv->tag == tag && cnv->class == cls) {
654 *syntax = cnv->syntax;
655 return 1;
657 cnv++;
659 return 0;
662 static unsigned char snmp_object_decode(struct asn1_ctx *ctx,
663 struct snmp_object **obj)
665 unsigned int cls, con, tag, len, idlen;
666 unsigned short type;
667 unsigned char *eoc, *end, *p;
668 unsigned long *lp, *id;
669 unsigned long ul;
670 long l;
672 *obj = NULL;
673 id = NULL;
675 if (!asn1_header_decode(ctx, &eoc, &cls, &con, &tag))
676 return 0;
678 if (cls != ASN1_UNI || con != ASN1_CON || tag != ASN1_SEQ)
679 return 0;
681 if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
682 return 0;
684 if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_OJI)
685 return 0;
687 if (!asn1_oid_decode(ctx, end, &id, &idlen))
688 return 0;
690 if (!asn1_header_decode(ctx, &end, &cls, &con, &tag)) {
691 kfree(id);
692 return 0;
695 if (con != ASN1_PRI) {
696 kfree(id);
697 return 0;
700 if (!snmp_tag_cls2syntax(tag, cls, &type)) {
701 kfree(id);
702 return 0;
705 switch (type) {
706 case SNMP_INTEGER:
707 len = sizeof(long);
708 if (!asn1_long_decode(ctx, end, &l)) {
709 kfree(id);
710 return 0;
712 *obj = kmalloc(sizeof(struct snmp_object) + len,
713 GFP_ATOMIC);
714 if (*obj == NULL) {
715 kfree(id);
716 if (net_ratelimit())
717 printk("OOM in bsalg (%d)\n", __LINE__);
718 return 0;
720 (*obj)->syntax.l[0] = l;
721 break;
722 case SNMP_OCTETSTR:
723 case SNMP_OPAQUE:
724 if (!asn1_octets_decode(ctx, end, &p, &len)) {
725 kfree(id);
726 return 0;
728 *obj = kmalloc(sizeof(struct snmp_object) + len,
729 GFP_ATOMIC);
730 if (*obj == NULL) {
731 kfree(id);
732 if (net_ratelimit())
733 printk("OOM in bsalg (%d)\n", __LINE__);
734 return 0;
736 memcpy((*obj)->syntax.c, p, len);
737 kfree(p);
738 break;
739 case SNMP_NULL:
740 case SNMP_NOSUCHOBJECT:
741 case SNMP_NOSUCHINSTANCE:
742 case SNMP_ENDOFMIBVIEW:
743 len = 0;
744 *obj = kmalloc(sizeof(struct snmp_object), GFP_ATOMIC);
745 if (*obj == NULL) {
746 kfree(id);
747 if (net_ratelimit())
748 printk("OOM in bsalg (%d)\n", __LINE__);
749 return 0;
751 if (!asn1_null_decode(ctx, end)) {
752 kfree(id);
753 kfree(*obj);
754 *obj = NULL;
755 return 0;
757 break;
758 case SNMP_OBJECTID:
759 if (!asn1_oid_decode(ctx, end, (unsigned long **)&lp, &len)) {
760 kfree(id);
761 return 0;
763 len *= sizeof(unsigned long);
764 *obj = kmalloc(sizeof(struct snmp_object) + len, GFP_ATOMIC);
765 if (*obj == NULL) {
766 kfree(id);
767 if (net_ratelimit())
768 printk("OOM in bsalg (%d)\n", __LINE__);
769 return 0;
771 memcpy((*obj)->syntax.ul, lp, len);
772 kfree(lp);
773 break;
774 case SNMP_IPADDR:
775 if (!asn1_octets_decode(ctx, end, &p, &len)) {
776 kfree(id);
777 return 0;
779 if (len != 4) {
780 kfree(p);
781 kfree(id);
782 return 0;
784 *obj = kmalloc(sizeof(struct snmp_object) + len, GFP_ATOMIC);
785 if (*obj == NULL) {
786 kfree(p);
787 kfree(id);
788 if (net_ratelimit())
789 printk("OOM in bsalg (%d)\n", __LINE__);
790 return 0;
792 memcpy((*obj)->syntax.uc, p, len);
793 kfree(p);
794 break;
795 case SNMP_COUNTER:
796 case SNMP_GAUGE:
797 case SNMP_TIMETICKS:
798 len = sizeof(unsigned long);
799 if (!asn1_ulong_decode(ctx, end, &ul)) {
800 kfree(id);
801 return 0;
803 *obj = kmalloc(sizeof(struct snmp_object) + len, GFP_ATOMIC);
804 if (*obj == NULL) {
805 kfree(id);
806 if (net_ratelimit())
807 printk("OOM in bsalg (%d)\n", __LINE__);
808 return 0;
810 (*obj)->syntax.ul[0] = ul;
811 break;
812 default:
813 kfree(id);
814 return 0;
817 (*obj)->syntax_len = len;
818 (*obj)->type = type;
819 (*obj)->id = id;
820 (*obj)->id_len = idlen;
822 if (!asn1_eoc_decode(ctx, eoc)) {
823 kfree(id);
824 kfree(*obj);
825 *obj = NULL;
826 return 0;
828 return 1;
831 static unsigned char snmp_request_decode(struct asn1_ctx *ctx,
832 struct snmp_request *request)
834 unsigned int cls, con, tag;
835 unsigned char *end;
837 if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
838 return 0;
840 if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_INT)
841 return 0;
843 if (!asn1_ulong_decode(ctx, end, &request->id))
844 return 0;
846 if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
847 return 0;
849 if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_INT)
850 return 0;
852 if (!asn1_uint_decode(ctx, end, &request->error_status))
853 return 0;
855 if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
856 return 0;
858 if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_INT)
859 return 0;
861 if (!asn1_uint_decode(ctx, end, &request->error_index))
862 return 0;
864 return 1;
868 * Fast checksum update for possibly oddly-aligned UDP byte, from the
869 * code example in the draft.
871 static void fast_csum(unsigned char *csum,
872 const unsigned char *optr,
873 const unsigned char *nptr,
874 int odd)
876 long x, old, new;
878 x = csum[0] * 256 + csum[1];
880 x =~ x & 0xFFFF;
882 if (odd) old = optr[0] * 256;
883 else old = optr[0];
885 x -= old & 0xFFFF;
886 if (x <= 0) {
887 x--;
888 x &= 0xFFFF;
891 if (odd) new = nptr[0] * 256;
892 else new = nptr[0];
894 x += new & 0xFFFF;
895 if (x & 0x10000) {
896 x++;
897 x &= 0xFFFF;
900 x =~ x & 0xFFFF;
901 csum[0] = x / 256;
902 csum[1] = x & 0xFF;
906 * Mangle IP address.
907 * - begin points to the start of the snmp messgae
908 * - addr points to the start of the address
910 static inline void mangle_address(unsigned char *begin,
911 unsigned char *addr,
912 const struct oct1_map *map,
913 u_int16_t *check)
915 if (map->from == NOCT1(*addr)) {
916 u_int32_t old;
918 if (debug)
919 memcpy(&old, (unsigned char *)addr, sizeof(old));
921 *addr = map->to;
923 /* Update UDP checksum if being used */
924 if (*check) {
925 unsigned char odd = !((addr - begin) % 2);
927 fast_csum((unsigned char *)check,
928 &map->from, &map->to, odd);
932 if (debug)
933 printk(KERN_DEBUG "bsalg: mapped %u.%u.%u.%u to "
934 "%u.%u.%u.%u\n", NIPQUAD(old), NIPQUAD(*addr));
938 static unsigned char snmp_trap_decode(struct asn1_ctx *ctx,
939 struct snmp_v1_trap *trap,
940 const struct oct1_map *map,
941 u_int16_t *check)
943 unsigned int cls, con, tag, len;
944 unsigned char *end;
946 if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
947 return 0;
949 if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_OJI)
950 return 0;
952 if (!asn1_oid_decode(ctx, end, &trap->id, &trap->id_len))
953 return 0;
955 if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
956 goto err_id_free;
958 if (!((cls == ASN1_APL && con == ASN1_PRI && tag == SNMP_IPA) ||
959 (cls == ASN1_UNI && con == ASN1_PRI && tag == ASN1_OTS)))
960 goto err_id_free;
962 if (!asn1_octets_decode(ctx, end, (unsigned char **)&trap->ip_address, &len))
963 goto err_id_free;
965 /* IPv4 only */
966 if (len != 4)
967 goto err_addr_free;
969 mangle_address(ctx->begin, ctx->pointer - 4, map, check);
971 if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
972 goto err_addr_free;
974 if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_INT)
975 goto err_addr_free;
977 if (!asn1_uint_decode(ctx, end, &trap->general))
978 goto err_addr_free;
980 if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
981 goto err_addr_free;
983 if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_INT)
984 goto err_addr_free;
986 if (!asn1_uint_decode(ctx, end, &trap->specific))
987 goto err_addr_free;
989 if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
990 goto err_addr_free;
992 if (!((cls == ASN1_APL && con == ASN1_PRI && tag == SNMP_TIT) ||
993 (cls == ASN1_UNI && con == ASN1_PRI && tag == ASN1_INT)))
994 goto err_addr_free;
996 if (!asn1_ulong_decode(ctx, end, &trap->time))
997 goto err_addr_free;
999 return 1;
1001 err_id_free:
1002 kfree(trap->id);
1004 err_addr_free:
1005 kfree((unsigned long *)trap->ip_address);
1007 return 0;
1010 /*****************************************************************************
1012 * Misc. routines
1014 *****************************************************************************/
1016 static void hex_dump(unsigned char *buf, size_t len)
1018 size_t i;
1020 for (i = 0; i < len; i++) {
1021 if (i && !(i % 16))
1022 printk("\n");
1023 printk("%02x ", *(buf + i));
1025 printk("\n");
1029 * Parse and mangle SNMP message according to mapping.
1030 * (And this is the fucking 'basic' method).
1032 static int snmp_parse_mangle(unsigned char *msg,
1033 u_int16_t len,
1034 const struct oct1_map *map,
1035 u_int16_t *check)
1037 unsigned char *eoc, *end;
1038 unsigned int cls, con, tag, vers, pdutype;
1039 struct asn1_ctx ctx;
1040 struct asn1_octstr comm;
1041 struct snmp_object **obj;
1043 if (debug > 1)
1044 hex_dump(msg, len);
1046 asn1_open(&ctx, msg, len);
1049 * Start of SNMP message.
1051 if (!asn1_header_decode(&ctx, &eoc, &cls, &con, &tag))
1052 return 0;
1053 if (cls != ASN1_UNI || con != ASN1_CON || tag != ASN1_SEQ)
1054 return 0;
1057 * Version 1 or 2 handled.
1059 if (!asn1_header_decode(&ctx, &end, &cls, &con, &tag))
1060 return 0;
1061 if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_INT)
1062 return 0;
1063 if (!asn1_uint_decode (&ctx, end, &vers))
1064 return 0;
1065 if (debug > 1)
1066 printk(KERN_DEBUG "bsalg: snmp version: %u\n", vers + 1);
1067 if (vers > 1)
1068 return 1;
1071 * Community.
1073 if (!asn1_header_decode (&ctx, &end, &cls, &con, &tag))
1074 return 0;
1075 if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_OTS)
1076 return 0;
1077 if (!asn1_octets_decode(&ctx, end, &comm.data, &comm.len))
1078 return 0;
1079 if (debug > 1) {
1080 unsigned int i;
1082 printk(KERN_DEBUG "bsalg: community: ");
1083 for (i = 0; i < comm.len; i++)
1084 printk("%c", comm.data[i]);
1085 printk("\n");
1087 kfree(comm.data);
1090 * PDU type
1092 if (!asn1_header_decode(&ctx, &eoc, &cls, &con, &pdutype))
1093 return 0;
1094 if (cls != ASN1_CTX || con != ASN1_CON)
1095 return 0;
1096 if (debug > 1) {
1097 unsigned char *pdus[] = {
1098 [SNMP_PDU_GET] = "get",
1099 [SNMP_PDU_NEXT] = "get-next",
1100 [SNMP_PDU_RESPONSE] = "response",
1101 [SNMP_PDU_SET] = "set",
1102 [SNMP_PDU_TRAP1] = "trapv1",
1103 [SNMP_PDU_BULK] = "bulk",
1104 [SNMP_PDU_INFORM] = "inform",
1105 [SNMP_PDU_TRAP2] = "trapv2"
1108 if (pdutype > SNMP_PDU_TRAP2)
1109 printk(KERN_DEBUG "bsalg: bad pdu type %u\n", pdutype);
1110 else
1111 printk(KERN_DEBUG "bsalg: pdu: %s\n", pdus[pdutype]);
1113 if (pdutype != SNMP_PDU_RESPONSE &&
1114 pdutype != SNMP_PDU_TRAP1 && pdutype != SNMP_PDU_TRAP2)
1115 return 1;
1118 * Request header or v1 trap
1120 if (pdutype == SNMP_PDU_TRAP1) {
1121 struct snmp_v1_trap trap;
1122 unsigned char ret = snmp_trap_decode(&ctx, &trap, map, check);
1124 /* Discard trap allocations regardless */
1125 kfree(trap.id);
1126 kfree((unsigned long *)trap.ip_address);
1128 if (!ret)
1129 return ret;
1131 } else {
1132 struct snmp_request req;
1134 if (!snmp_request_decode(&ctx, &req))
1135 return 0;
1137 if (debug > 1)
1138 printk(KERN_DEBUG "bsalg: request: id=0x%lx error_status=%u "
1139 "error_index=%u\n", req.id, req.error_status,
1140 req.error_index);
1144 * Loop through objects, look for IP addresses to mangle.
1146 if (!asn1_header_decode(&ctx, &eoc, &cls, &con, &tag))
1147 return 0;
1149 if (cls != ASN1_UNI || con != ASN1_CON || tag != ASN1_SEQ)
1150 return 0;
1152 obj = kmalloc(sizeof(struct snmp_object), GFP_ATOMIC);
1153 if (obj == NULL) {
1154 if (net_ratelimit())
1155 printk(KERN_WARNING "OOM in bsalg(%d)\n", __LINE__);
1156 return 0;
1159 while (!asn1_eoc_decode(&ctx, eoc)) {
1160 unsigned int i;
1162 if (!snmp_object_decode(&ctx, obj)) {
1163 if (*obj) {
1164 if ((*obj)->id)
1165 kfree((*obj)->id);
1166 kfree(*obj);
1168 kfree(obj);
1169 return 0;
1172 if (debug > 1) {
1173 printk(KERN_DEBUG "bsalg: object: ");
1174 for (i = 0; i < (*obj)->id_len; i++) {
1175 if (i > 0)
1176 printk(".");
1177 printk("%lu", (*obj)->id[i]);
1179 printk(": type=%u\n", (*obj)->type);
1183 if ((*obj)->type == SNMP_IPADDR)
1184 mangle_address(ctx.begin, ctx.pointer - 4 , map, check);
1186 kfree((*obj)->id);
1187 kfree(*obj);
1189 kfree(obj);
1191 if (!asn1_eoc_decode(&ctx, eoc))
1192 return 0;
1194 return 1;
1197 /*****************************************************************************
1199 * NAT routines.
1201 *****************************************************************************/
1204 * SNMP translation routine.
1206 static int snmp_translate(struct ip_conntrack *ct,
1207 enum ip_conntrack_info ctinfo,
1208 struct sk_buff **pskb)
1210 struct iphdr *iph = (*pskb)->nh.iph;
1211 struct udphdr *udph = (struct udphdr *)((u_int32_t *)iph + iph->ihl);
1212 u_int16_t udplen = ntohs(udph->len);
1213 u_int16_t paylen = udplen - sizeof(struct udphdr);
1214 int dir = CTINFO2DIR(ctinfo);
1215 struct oct1_map map;
1218 * Determine mappping for application layer addresses based
1219 * on NAT manipulations for the packet.
1221 if (dir == IP_CT_DIR_ORIGINAL) {
1222 /* SNAT traps */
1223 map.from = NOCT1(ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.src.ip);
1224 map.to = NOCT1(ct->tuplehash[IP_CT_DIR_REPLY].tuple.dst.ip);
1225 } else {
1226 /* DNAT replies */
1227 map.from = NOCT1(ct->tuplehash[IP_CT_DIR_REPLY].tuple.src.ip);
1228 map.to = NOCT1(ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.dst.ip);
1231 if (map.from == map.to)
1232 return NF_ACCEPT;
1234 if (!snmp_parse_mangle((unsigned char *)udph + sizeof(struct udphdr),
1235 paylen, &map, &udph->check)) {
1236 if (net_ratelimit())
1237 printk(KERN_WARNING "bsalg: parser failed\n");
1238 return NF_DROP;
1240 return NF_ACCEPT;
1243 /* We don't actually set up expectations, just adjust internal IP
1244 * addresses if this is being NATted */
1245 static int help(struct sk_buff **pskb,
1246 struct ip_conntrack *ct,
1247 enum ip_conntrack_info ctinfo)
1249 int dir = CTINFO2DIR(ctinfo);
1250 unsigned int ret;
1251 struct iphdr *iph = (*pskb)->nh.iph;
1252 struct udphdr *udph = (struct udphdr *)((u_int32_t *)iph + iph->ihl);
1254 /* SNMP replies and originating SNMP traps get mangled */
1255 if (udph->source == ntohs(SNMP_PORT) && dir != IP_CT_DIR_REPLY)
1256 return NF_ACCEPT;
1257 if (udph->dest == ntohs(SNMP_TRAP_PORT) && dir != IP_CT_DIR_ORIGINAL)
1258 return NF_ACCEPT;
1260 /* No NAT? */
1261 if (!(ct->status & IPS_NAT_MASK))
1262 return NF_ACCEPT;
1265 * Make sure the packet length is ok. So far, we were only guaranteed
1266 * to have a valid length IP header plus 8 bytes, which means we have
1267 * enough room for a UDP header. Just verify the UDP length field so we
1268 * can mess around with the payload.
1270 if (ntohs(udph->len) != (*pskb)->len - (iph->ihl << 2)) {
1271 if (net_ratelimit())
1272 printk(KERN_WARNING "SNMP: dropping malformed packet "
1273 "src=%u.%u.%u.%u dst=%u.%u.%u.%u\n",
1274 NIPQUAD(iph->saddr), NIPQUAD(iph->daddr));
1275 return NF_DROP;
1278 if (!skb_ip_make_writable(pskb, (*pskb)->len))
1279 return NF_DROP;
1281 spin_lock_bh(&snmp_lock);
1282 ret = snmp_translate(ct, ctinfo, pskb);
1283 spin_unlock_bh(&snmp_lock);
1284 return ret;
1287 static struct ip_conntrack_helper snmp_helper = {
1288 .max_expected = 0,
1289 .timeout = 180,
1290 .me = THIS_MODULE,
1291 .help = help,
1292 .name = "snmp",
1294 .tuple = { .src = { .u = { __constant_htons(SNMP_PORT) } },
1295 .dst = { .protonum = IPPROTO_UDP },
1297 .mask = { .src = { .u = { 0xFFFF } },
1298 .dst = { .protonum = 0xFF },
1302 static struct ip_conntrack_helper snmp_trap_helper = {
1303 .max_expected = 0,
1304 .timeout = 180,
1305 .me = THIS_MODULE,
1306 .help = help,
1307 .name = "snmp_trap",
1309 .tuple = { .src = { .u = { __constant_htons(SNMP_TRAP_PORT) } },
1310 .dst = { .protonum = IPPROTO_UDP },
1312 .mask = { .src = { .u = { 0xFFFF } },
1313 .dst = { .protonum = 0xFF },
1317 /*****************************************************************************
1319 * Module stuff.
1321 *****************************************************************************/
1323 static int __init init(void)
1325 int ret = 0;
1327 ret = ip_conntrack_helper_register(&snmp_helper);
1328 if (ret < 0)
1329 return ret;
1330 ret = ip_conntrack_helper_register(&snmp_trap_helper);
1331 if (ret < 0) {
1332 ip_conntrack_helper_unregister(&snmp_helper);
1333 return ret;
1335 return ret;
1338 static void __exit fini(void)
1340 ip_conntrack_helper_unregister(&snmp_helper);
1341 ip_conntrack_helper_unregister(&snmp_trap_helper);
1344 module_init(init);
1345 module_exit(fini);
1347 module_param(debug, bool, 0600);