net/ipv4/netfilter/ip_nat_snmp_basic.c

   1 /*
   2  * ip_nat_snmp_basic.c
   3  *
   4  * Basic SNMP Application Layer Gateway
   5  *
   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.
   9  *
  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.
  13  *
  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.
  17  *
  18  * Currently, only SNMPv1 and SNMPv2 are supported.
  19  *
  20  * More information on ALG and associated issues can be found in
  21  * RFC 2962
  22  *
  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.
  25  *
  26  * Copyright (c) 2000 RP Internet (www.rpi.net.au).
  27  *
  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
  39  *
  40  * Author: James Morris <jmorris@intercode.com.au>
  41  *
  42  * Updates:
  43  * 2000-08-06: Convert to new helper API (Harald Welte).
  44  *
  45  */
  46 #include <linux/config.h>
  47 #include <linux/module.h>
  48 #include <linux/types.h>
  49 #include <linux/kernel.h>
  50 #include <linux/netfilter_ipv4.h>
  51 #include <linux/netfilter_ipv4/ip_nat.h>
  52 #include <linux/netfilter_ipv4/ip_nat_helper.h>
  53 #include <linux/brlock.h>
  54 #include <linux/types.h>
  55 #include <linux/ip.h>
  56 #include <net/checksum.h>
  57 #include <net/udp.h>
  58 #include <asm/uaccess.h>
  59
  60
  61
  62 #define SNMP_PORT 161
  63 #define SNMP_TRAP_PORT 162
  64 #define NOCT1(n) (u_int8_t )((n) & 0xff)
  65
  66 static int debug = 0;
  67 static spinlock_t snmp_lock = SPIN_LOCK_UNLOCKED;
  68
  69 /*
  70  * Application layer address mapping mimics the NAT mapping, but
  71  * only for the first octet in this case (a more flexible system
  72  * can be implemented if needed).
  73  */
  74 struct oct1_map
  75 {
  76         u_int8_t from;
  77         u_int8_t to;
  78 };
  79
  80
  81 /*****************************************************************************
  82  *
  83  * Basic ASN.1 decoding routines (gxsnmp author Dirk Wisse)
  84  *
  85  *****************************************************************************/
  86
  87 /* Class */
  88 #define ASN1_UNI        0       /* Universal */
  89 #define ASN1_APL        1       /* Application */
  90 #define ASN1_CTX        2       /* Context */
  91 #define ASN1_PRV        3       /* Private */
  92
  93 /* Tag */
  94 #define ASN1_EOC        0       /* End Of Contents */
  95 #define ASN1_BOL        1       /* Boolean */
  96 #define ASN1_INT        2       /* Integer */
  97 #define ASN1_BTS        3       /* Bit String */
  98 #define ASN1_OTS        4       /* Octet String */
  99 #define ASN1_NUL        5       /* Null */
 100 #define ASN1_OJI        6       /* Object Identifier  */
 101 #define ASN1_OJD        7       /* Object Description */
 102 #define ASN1_EXT        8       /* External */
 103 #define ASN1_SEQ        16      /* Sequence */
 104 #define ASN1_SET        17      /* Set */
 105 #define ASN1_NUMSTR     18      /* Numerical String */
 106 #define ASN1_PRNSTR     19      /* Printable String */
 107 #define ASN1_TEXSTR     20      /* Teletext String */
 108 #define ASN1_VIDSTR     21      /* Video String */
 109 #define ASN1_IA5STR     22      /* IA5 String */
 110 #define ASN1_UNITIM     23      /* Universal Time */
 111 #define ASN1_GENTIM     24      /* General Time */
 112 #define ASN1_GRASTR     25      /* Graphical String */
 113 #define ASN1_VISSTR     26      /* Visible String */
 114 #define ASN1_GENSTR     27      /* General String */
 115
 116 /* Primitive / Constructed methods*/
 117 #define ASN1_PRI        0       /* Primitive */
 118 #define ASN1_CON        1       /* Constructed */
 119
 120 /*
 121  * Error codes.
 122  */
 123 #define ASN1_ERR_NOERROR                0
 124 #define ASN1_ERR_DEC_EMPTY              2
 125 #define ASN1_ERR_DEC_EOC_MISMATCH       3
 126 #define ASN1_ERR_DEC_LENGTH_MISMATCH    4
 127 #define ASN1_ERR_DEC_BADVALUE           5
 128
 129 /*
 130  * ASN.1 context.
 131  */
 132 struct asn1_ctx
 133 {
 134         int error;                      /* Error condition */
 135         unsigned char *pointer;         /* Octet just to be decoded */
 136         unsigned char *begin;           /* First octet */
 137         unsigned char *end;             /* Octet after last octet */
 138 };
 139
 140 /*
 141  * Octet string (not null terminated)
 142  */
 143 struct asn1_octstr
 144 {
 145         unsigned char *data;
 146         unsigned int len;
 147 };
 148
 149 static void asn1_open(struct asn1_ctx *ctx,
 150                       unsigned char *buf,
 151                       unsigned int len)
 152 {
 153         ctx->begin = buf;
 154         ctx->end = buf + len;
 155         ctx->pointer = buf;
 156         ctx->error = ASN1_ERR_NOERROR;
 157 }
 158
 159 static unsigned char asn1_octet_decode(struct asn1_ctx *ctx, unsigned char *ch)
 160 {
 161         if (ctx->pointer >= ctx->end) {
 162                 ctx->error = ASN1_ERR_DEC_EMPTY;
 163                 return 0;
 164         }
 165         *ch = *(ctx->pointer)++;
 166         return 1;
 167 }
 168
 169 static unsigned char asn1_tag_decode(struct asn1_ctx *ctx, unsigned int *tag)
 170 {
 171         unsigned char ch;
 172
 173         *tag = 0;
 174
 175         do
 176         {
 177                 if (!asn1_octet_decode(ctx, &ch))
 178                         return 0;
 179                 *tag <<= 7;
 180                 *tag |= ch & 0x7F;
 181         } while ((ch & 0x80) == 0x80);
 182         return 1;
 183 }
 184
 185 static unsigned char asn1_id_decode(struct asn1_ctx *ctx,
 186                                     unsigned int *cls,
 187                                     unsigned int *con,
 188                                     unsigned int *tag)
 189 {
 190         unsigned char ch;
 191
 192         if (!asn1_octet_decode(ctx, &ch))
 193                 return 0;
 194
 195         *cls = (ch & 0xC0) >> 6;
 196         *con = (ch & 0x20) >> 5;
 197         *tag = (ch & 0x1F);
 198
 199         if (*tag == 0x1F) {
 200                 if (!asn1_tag_decode(ctx, tag))
 201                         return 0;
 202         }
 203         return 1;
 204 }
 205
 206 static unsigned char asn1_length_decode(struct asn1_ctx *ctx,
 207                                         unsigned int *def,
 208                                         unsigned int *len)
 209 {
 210         unsigned char ch, cnt;
 211
 212         if (!asn1_octet_decode(ctx, &ch))
 213                 return 0;
 214
 215         if (ch == 0x80)
 216                 *def = 0;
 217         else {
 218                 *def = 1;
 219
 220                 if (ch < 0x80)
 221                         *len = ch;
 222                 else {
 223                         cnt = (unsigned char) (ch & 0x7F);
 224                         *len = 0;
 225
 226                         while (cnt > 0) {
 227                                 if (!asn1_octet_decode(ctx, &ch))
 228                                         return 0;
 229                                 *len <<= 8;
 230                                 *len |= ch;
 231                                 cnt--;
 232                         }
 233                 }
 234         }
 235
 236         /* don't trust len bigger than ctx buffer */
 237         if (*len > ctx->end - ctx->pointer)
 238                 return 0;
 239
 240         return 1;
 241 }
 242
 243 static unsigned char asn1_header_decode(struct asn1_ctx *ctx,
 244                                         unsigned char **eoc,
 245                                         unsigned int *cls,
 246                                         unsigned int *con,
 247                                         unsigned int *tag)
 248 {
 249         unsigned int def, len;
 250
 251         if (!asn1_id_decode(ctx, cls, con, tag))
 252                 return 0;
 253
 254         if (!asn1_length_decode(ctx, &def, &len))
 255                 return 0;
 256
 257         /* primitive shall be definite, indefinite shall be constructed */
 258         if (*con == ASN1_PRI && !def)
 259                 return 0;
 260
 261         if (def)
 262                 *eoc = ctx->pointer + len;
 263         else
 264                 *eoc = 0;
 265         return 1;
 266 }
 267
 268 static unsigned char asn1_eoc_decode(struct asn1_ctx *ctx, unsigned char *eoc)
 269 {
 270         unsigned char ch;
 271
 272         if (eoc == 0) {
 273                 if (!asn1_octet_decode(ctx, &ch))
 274                         return 0;
 275
 276                 if (ch != 0x00) {
 277                         ctx->error = ASN1_ERR_DEC_EOC_MISMATCH;
 278                         return 0;
 279                 }
 280
 281                 if (!asn1_octet_decode(ctx, &ch))
 282                         return 0;
 283
 284                 if (ch != 0x00) {
 285                         ctx->error = ASN1_ERR_DEC_EOC_MISMATCH;
 286                         return 0;
 287                 }
 288                 return 1;
 289         } else {
 290                 if (ctx->pointer != eoc) {
 291                         ctx->error = ASN1_ERR_DEC_LENGTH_MISMATCH;
 292                         return 0;
 293                 }
 294                 return 1;
 295         }
 296 }
 297
 298 static unsigned char asn1_null_decode(struct asn1_ctx *ctx, unsigned char *eoc)
 299 {
 300         ctx->pointer = eoc;
 301         return 1;
 302 }
 303
 304 static unsigned char asn1_long_decode(struct asn1_ctx *ctx,
 305                                       unsigned char *eoc,
 306                                       long *integer)
 307 {
 308         unsigned char ch;
 309         unsigned int  len;
 310
 311         if (!asn1_octet_decode(ctx, &ch))
 312                 return 0;
 313
 314         *integer = (signed char) ch;
 315         len = 1;
 316
 317         while (ctx->pointer < eoc) {
 318                 if (++len > sizeof (long)) {
 319                         ctx->error = ASN1_ERR_DEC_BADVALUE;
 320                         return 0;
 321                 }
 322
 323                 if (!asn1_octet_decode(ctx, &ch))
 324                         return 0;
 325
 326                 *integer <<= 8;
 327                 *integer |= ch;
 328         }
 329         return 1;
 330 }
 331
 332 static unsigned char asn1_uint_decode(struct asn1_ctx *ctx,
 333                                       unsigned char *eoc,
 334                                       unsigned int *integer)
 335 {
 336         unsigned char ch;
 337         unsigned int  len;
 338
 339         if (!asn1_octet_decode(ctx, &ch))
 340                 return 0;
 341
 342         *integer = ch;
 343         if (ch == 0) len = 0;
 344         else len = 1;
 345
 346         while (ctx->pointer < eoc) {
 347                 if (++len > sizeof (unsigned int)) {
 348                         ctx->error = ASN1_ERR_DEC_BADVALUE;
 349                         return 0;
 350                 }
 351
 352                 if (!asn1_octet_decode(ctx, &ch))
 353                         return 0;
 354
 355                 *integer <<= 8;
 356                 *integer |= ch;
 357         }
 358         return 1;
 359 }
 360
 361 static unsigned char asn1_ulong_decode(struct asn1_ctx *ctx,
 362                                        unsigned char *eoc,
 363                                        unsigned long *integer)
 364 {
 365         unsigned char ch;
 366         unsigned int  len;
 367
 368         if (!asn1_octet_decode(ctx, &ch))
 369                 return 0;
 370
 371         *integer = ch;
 372         if (ch == 0) len = 0;
 373         else len = 1;
 374
 375         while (ctx->pointer < eoc) {
 376                 if (++len > sizeof (unsigned long)) {
 377                         ctx->error = ASN1_ERR_DEC_BADVALUE;
 378                         return 0;
 379                 }
 380
 381                 if (!asn1_octet_decode(ctx, &ch))
 382                         return 0;
 383
 384                 *integer <<= 8;
 385                 *integer |= ch;
 386         }
 387         return 1;
 388 }
 389
 390 static unsigned char asn1_octets_decode(struct asn1_ctx *ctx,
 391                                         unsigned char *eoc,
 392                                         unsigned char **octets,
 393                                         unsigned int *len)
 394 {
 395         unsigned char *ptr;
 396
 397         *len = 0;
 398
 399         *octets = kmalloc(eoc - ctx->pointer, GFP_ATOMIC);
 400         if (*octets == NULL) {
 401                 if (net_ratelimit())
 402                         printk("OOM in bsalg (%d)\n", __LINE__);
 403                 return 0;
 404         }
 405
 406         ptr = *octets;
 407         while (ctx->pointer < eoc) {
 408                 if (!asn1_octet_decode(ctx, (unsigned char *)ptr++)) {
 409                         kfree(*octets);
 410                         *octets = NULL;
 411                         return 0;
 412                 }
 413                 (*len)++;
 414         }
 415         return 1;
 416 }
 417
 418 static unsigned char asn1_subid_decode(struct asn1_ctx *ctx,
 419                                        unsigned long *subid)
 420 {
 421         unsigned char ch;
 422
 423         *subid = 0;
 424
 425         do {
 426                 if (!asn1_octet_decode(ctx, &ch))
 427                         return 0;
 428
 429                 *subid <<= 7;
 430                 *subid |= ch & 0x7F;
 431         } while ((ch & 0x80) == 0x80);
 432         return 1;
 433 }
 434
 435 static unsigned char asn1_oid_decode(struct asn1_ctx *ctx,
 436                                      unsigned char *eoc,
 437                                      unsigned long **oid,
 438                                      unsigned int *len)
 439 {
 440         unsigned long subid;
 441         unsigned int  size;
 442         unsigned long *optr;
 443
 444         size = eoc - ctx->pointer + 1;
 445
 446         /* first subid actually encodes first two subids */
 447         if (size < 2 || size > ULONG_MAX/sizeof(unsigned long))
 448                 return 0;
 449
 450         *oid = kmalloc(size * sizeof(unsigned long), GFP_ATOMIC);
 451         if (*oid == NULL) {
 452                 if (net_ratelimit())
 453                         printk("OOM in bsalg (%d)\n", __LINE__);
 454                 return 0;
 455         }
 456
 457         optr = *oid;
 458
 459         if (!asn1_subid_decode(ctx, &subid)) {
 460                 kfree(*oid);
 461                 *oid = NULL;
 462                 return 0;
 463         }
 464
 465         if (subid < 40) {
 466                 optr [0] = 0;
 467                 optr [1] = subid;
 468         } else if (subid < 80) {
 469                 optr [0] = 1;
 470                 optr [1] = subid - 40;
 471         } else {
 472                 optr [0] = 2;
 473                 optr [1] = subid - 80;
 474         }
 475
 476         *len = 2;
 477         optr += 2;
 478
 479         while (ctx->pointer < eoc) {
 480                 if (++(*len) > size) {
 481                         ctx->error = ASN1_ERR_DEC_BADVALUE;
 482                         kfree(*oid);
 483                         *oid = NULL;
 484                         return 0;
 485                 }
 486
 487                 if (!asn1_subid_decode(ctx, optr++)) {
 488                         kfree(*oid);
 489                         *oid = NULL;
 490                         return 0;
 491                 }
 492         }
 493         return 1;
 494 }
 495
 496 /*****************************************************************************
 497  *
 498  * SNMP decoding routines (gxsnmp author Dirk Wisse)
 499  *
 500  *****************************************************************************/
 501
 502 /* SNMP Versions */
 503 #define SNMP_V1                         0
 504 #define SNMP_V2C                        1
 505 #define SNMP_V2                         2
 506 #define SNMP_V3                         3
 507
 508 /* Default Sizes */
 509 #define SNMP_SIZE_COMM                  256
 510 #define SNMP_SIZE_OBJECTID              128
 511 #define SNMP_SIZE_BUFCHR                256
 512 #define SNMP_SIZE_BUFINT                128
 513 #define SNMP_SIZE_SMALLOBJECTID         16
 514
 515 /* Requests */
 516 #define SNMP_PDU_GET                    0
 517 #define SNMP_PDU_NEXT                   1
 518 #define SNMP_PDU_RESPONSE               2
 519 #define SNMP_PDU_SET                    3
 520 #define SNMP_PDU_TRAP1                  4
 521 #define SNMP_PDU_BULK                   5
 522 #define SNMP_PDU_INFORM                 6
 523 #define SNMP_PDU_TRAP2                  7
 524
 525 /* Errors */
 526 #define SNMP_NOERROR                    0
 527 #define SNMP_TOOBIG                     1
 528 #define SNMP_NOSUCHNAME                 2
 529 #define SNMP_BADVALUE                   3
 530 #define SNMP_READONLY                   4
 531 #define SNMP_GENERROR                   5
 532 #define SNMP_NOACCESS                   6
 533 #define SNMP_WRONGTYPE                  7
 534 #define SNMP_WRONGLENGTH                8
 535 #define SNMP_WRONGENCODING              9
 536 #define SNMP_WRONGVALUE                 10
 537 #define SNMP_NOCREATION                 11
 538 #define SNMP_INCONSISTENTVALUE          12
 539 #define SNMP_RESOURCEUNAVAILABLE        13
 540 #define SNMP_COMMITFAILED               14
 541 #define SNMP_UNDOFAILED                 15
 542 #define SNMP_AUTHORIZATIONERROR         16
 543 #define SNMP_NOTWRITABLE                17
 544 #define SNMP_INCONSISTENTNAME           18
 545
 546 /* General SNMP V1 Traps */
 547 #define SNMP_TRAP_COLDSTART             0
 548 #define SNMP_TRAP_WARMSTART             1
 549 #define SNMP_TRAP_LINKDOWN              2
 550 #define SNMP_TRAP_LINKUP                3
 551 #define SNMP_TRAP_AUTFAILURE            4
 552 #define SNMP_TRAP_EQPNEIGHBORLOSS       5
 553 #define SNMP_TRAP_ENTSPECIFIC           6
 554
 555 /* SNMPv1 Types */
 556 #define SNMP_NULL                0
 557 #define SNMP_INTEGER             1    /* l  */
 558 #define SNMP_OCTETSTR            2    /* c  */
 559 #define SNMP_DISPLAYSTR          2    /* c  */
 560 #define SNMP_OBJECTID            3    /* ul */
 561 #define SNMP_IPADDR              4    /* uc */
 562 #define SNMP_COUNTER             5    /* ul */
 563 #define SNMP_GAUGE               6    /* ul */
 564 #define SNMP_TIMETICKS           7    /* ul */
 565 #define SNMP_OPAQUE              8    /* c  */
 566
 567 /* Additional SNMPv2 Types */
 568 #define SNMP_UINTEGER            5    /* ul */
 569 #define SNMP_BITSTR              9    /* uc */
 570 #define SNMP_NSAP               10    /* uc */
 571 #define SNMP_COUNTER64          11    /* ul */
 572 #define SNMP_NOSUCHOBJECT       12
 573 #define SNMP_NOSUCHINSTANCE     13
 574 #define SNMP_ENDOFMIBVIEW       14
 575
 576 union snmp_syntax
 577 {
 578         unsigned char uc[0];    /* 8 bit unsigned */
 579         char c[0];              /* 8 bit signed */
 580         unsigned long ul[0];    /* 32 bit unsigned */
 581         long l[0];              /* 32 bit signed */
 582 };
 583
 584 struct snmp_object
 585 {
 586         unsigned long *id;
 587         unsigned int id_len;
 588         unsigned short type;
 589         unsigned int syntax_len;
 590         union snmp_syntax syntax;
 591 };
 592
 593 struct snmp_request
 594 {
 595         unsigned long id;
 596         unsigned int error_status;
 597         unsigned int error_index;
 598 };
 599
 600 struct snmp_v1_trap
 601 {
 602         unsigned long *id;
 603         unsigned int id_len;
 604         unsigned long ip_address;       /* pointer  */
 605         unsigned int general;
 606         unsigned int specific;
 607         unsigned long time;
 608 };
 609
 610 /* SNMP types */
 611 #define SNMP_IPA    0
 612 #define SNMP_CNT    1
 613 #define SNMP_GGE    2
 614 #define SNMP_TIT    3
 615 #define SNMP_OPQ    4
 616 #define SNMP_C64    6
 617
 618 /* SNMP errors */
 619 #define SERR_NSO    0
 620 #define SERR_NSI    1
 621 #define SERR_EOM    2
 622
 623 static void inline mangle_address(unsigned char *begin,
 624                                   unsigned char *addr,
 625                                   const struct oct1_map *map,
 626                                   u_int16_t *check);
 627 struct snmp_cnv
 628 {
 629         unsigned int class;
 630         unsigned int tag;
 631         int syntax;
 632 };
 633
 634 static struct snmp_cnv snmp_conv [] =
 635 {
 636         {ASN1_UNI, ASN1_NUL, SNMP_NULL},
 637         {ASN1_UNI, ASN1_INT, SNMP_INTEGER},
 638         {ASN1_UNI, ASN1_OTS, SNMP_OCTETSTR},
 639         {ASN1_UNI, ASN1_OTS, SNMP_DISPLAYSTR},
 640         {ASN1_UNI, ASN1_OJI, SNMP_OBJECTID},
 641         {ASN1_APL, SNMP_IPA, SNMP_IPADDR},
 642         {ASN1_APL, SNMP_CNT, SNMP_COUNTER},     /* Counter32 */
 643         {ASN1_APL, SNMP_GGE, SNMP_GAUGE},       /* Gauge32 == Unsigned32  */
 644         {ASN1_APL, SNMP_TIT, SNMP_TIMETICKS},
 645         {ASN1_APL, SNMP_OPQ, SNMP_OPAQUE},
 646
 647         /* SNMPv2 data types and errors */
 648         {ASN1_UNI, ASN1_BTS, SNMP_BITSTR},
 649         {ASN1_APL, SNMP_C64, SNMP_COUNTER64},
 650         {ASN1_CTX, SERR_NSO, SNMP_NOSUCHOBJECT},
 651         {ASN1_CTX, SERR_NSI, SNMP_NOSUCHINSTANCE},
 652         {ASN1_CTX, SERR_EOM, SNMP_ENDOFMIBVIEW},
 653         {0,       0,       -1}
 654 };
 655
 656 static unsigned char snmp_tag_cls2syntax(unsigned int tag,
 657                                          unsigned int cls,
 658                                          unsigned short *syntax)
 659 {
 660         struct snmp_cnv *cnv;
 661
 662         cnv = snmp_conv;
 663
 664         while (cnv->syntax != -1) {
 665                 if (cnv->tag == tag && cnv->class == cls) {
 666                         *syntax = cnv->syntax;
 667                         return 1;
 668                 }
 669                 cnv++;
 670         }
 671         return 0;
 672 }
 673
 674 static unsigned char snmp_object_decode(struct asn1_ctx *ctx,
 675                                         struct snmp_object **obj)
 676 {
 677         unsigned int cls, con, tag, len, idlen;
 678         unsigned short type;
 679         unsigned char *eoc, *end, *p;
 680         unsigned long *lp, *id;
 681         unsigned long ul;
 682         long  l;
 683
 684         *obj = NULL;
 685         id = NULL;
 686
 687         if (!asn1_header_decode(ctx, &eoc, &cls, &con, &tag))
 688                 return 0;
 689
 690         if (cls != ASN1_UNI || con != ASN1_CON || tag != ASN1_SEQ)
 691                 return 0;
 692
 693         if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
 694                 return 0;
 695
 696         if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_OJI)
 697                 return 0;
 698
 699         if (!asn1_oid_decode(ctx, end, &id, &idlen))
 700                 return 0;
 701
 702         if (!asn1_header_decode(ctx, &end, &cls, &con, &tag)) {
 703                 kfree(id);
 704                 return 0;
 705         }
 706
 707         if (con != ASN1_PRI) {
 708                 kfree(id);
 709                 return 0;
 710         }
 711
 712         if (!snmp_tag_cls2syntax(tag, cls, &type)) {
 713                 kfree(id);
 714                 return 0;
 715         }
 716
 717         switch (type) {
 718                 case SNMP_INTEGER:
 719                         len = sizeof(long);
 720                         if (!asn1_long_decode(ctx, end, &l)) {
 721                                 kfree(id);
 722                                 return 0;
 723                         }
 724                         *obj = kmalloc(sizeof(struct snmp_object) + len,
 725                                        GFP_ATOMIC);
 726                         if (*obj == NULL) {
 727                                 kfree(id);
 728                                 if (net_ratelimit())
 729                                         printk("OOM in bsalg (%d)\n", __LINE__);
 730                                 return 0;
 731                         }
 732                         (*obj)->syntax.l[0] = l;
 733                         break;
 734                 case SNMP_OCTETSTR:
 735                 case SNMP_OPAQUE:
 736                         if (!asn1_octets_decode(ctx, end, &p, &len)) {
 737                                 kfree(id);
 738                                 return 0;
 739                         }
 740                         *obj = kmalloc(sizeof(struct snmp_object) + len,
 741                                        GFP_ATOMIC);
 742                         if (*obj == NULL) {
 743                                 kfree(id);
 744                                 if (net_ratelimit())
 745                                         printk("OOM in bsalg (%d)\n", __LINE__);
 746                                 return 0;
 747                         }
 748                         memcpy((*obj)->syntax.c, p, len);
 749                         kfree(p);
 750                         break;
 751                 case SNMP_NULL:
 752                 case SNMP_NOSUCHOBJECT:
 753                 case SNMP_NOSUCHINSTANCE:
 754                 case SNMP_ENDOFMIBVIEW:
 755                         len = 0;
 756                         *obj = kmalloc(sizeof(struct snmp_object), GFP_ATOMIC);
 757                         if (*obj == NULL) {
 758                                 kfree(id);
 759                                 if (net_ratelimit())
 760                                         printk("OOM in bsalg (%d)\n", __LINE__);
 761                                 return 0;
 762                         }
 763                         if (!asn1_null_decode(ctx, end)) {
 764                                 kfree(id);
 765                                 kfree(*obj);
 766                                 *obj = NULL;
 767                                 return 0;
 768                         }
 769                         break;
 770                 case SNMP_OBJECTID:
 771                         if (!asn1_oid_decode(ctx, end, (unsigned long **)&lp, &len)) {
 772                                 kfree(id);
 773                                 return 0;
 774                         }
 775                         len *= sizeof(unsigned long);
 776                         *obj = kmalloc(sizeof(struct snmp_object) + len, GFP_ATOMIC);
 777                         if (*obj == NULL) {
 778                                 kfree(id);
 779                                 if (net_ratelimit())
 780                                         printk("OOM in bsalg (%d)\n", __LINE__);
 781                                 return 0;
 782                         }
 783                         memcpy((*obj)->syntax.ul, lp, len);
 784                         kfree(lp);
 785                         break;
 786                 case SNMP_IPADDR:
 787                         if (!asn1_octets_decode(ctx, end, &p, &len)) {
 788                                 kfree(id);
 789                                 return 0;
 790                         }
 791                         if (len != 4) {
 792                                 kfree(p);
 793                                 kfree(id);
 794                                 return 0;
 795                         }
 796                         *obj = kmalloc(sizeof(struct snmp_object) + len, GFP_ATOMIC);
 797                         if (*obj == NULL) {
 798                                 kfree(p);
 799                                 kfree(id);
 800                                 if (net_ratelimit())
 801                                         printk("OOM in bsalg (%d)\n", __LINE__);
 802                                 return 0;
 803                         }
 804                         memcpy((*obj)->syntax.uc, p, len);
 805                         kfree(p);
 806                         break;
 807                 case SNMP_COUNTER:
 808                 case SNMP_GAUGE:
 809                 case SNMP_TIMETICKS:
 810                         len = sizeof(unsigned long);
 811                         if (!asn1_ulong_decode(ctx, end, &ul)) {
 812                                 kfree(id);
 813                                 return 0;
 814                         }
 815                         *obj = kmalloc(sizeof(struct snmp_object) + len, GFP_ATOMIC);
 816                         if (*obj == NULL) {
 817                                 kfree(id);
 818                                 if (net_ratelimit())
 819                                         printk("OOM in bsalg (%d)\n", __LINE__);
 820                                 return 0;
 821                         }
 822                         (*obj)->syntax.ul[0] = ul;
 823                         break;
 824                 default:
 825                         kfree(id);
 826                         return 0;
 827         }
 828
 829         (*obj)->syntax_len = len;
 830         (*obj)->type = type;
 831         (*obj)->id = id;
 832         (*obj)->id_len = idlen;
 833
 834         if (!asn1_eoc_decode(ctx, eoc)) {
 835                 kfree(id);
 836                 kfree(*obj);
 837                 *obj = NULL;
 838                 return 0;
 839         }
 840         return 1;
 841 }
 842
 843 static unsigned char snmp_request_decode(struct asn1_ctx *ctx,
 844                                          struct snmp_request *request)
 845 {
 846         unsigned int cls, con, tag;
 847         unsigned char *end;
 848
 849         if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
 850                 return 0;
 851
 852         if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_INT)
 853                 return 0;
 854
 855         if (!asn1_ulong_decode(ctx, end, &request->id))
 856                 return 0;
 857
 858         if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
 859                 return 0;
 860
 861         if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_INT)
 862                 return 0;
 863
 864         if (!asn1_uint_decode(ctx, end, &request->error_status))
 865                 return 0;
 866
 867         if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
 868                 return 0;
 869
 870         if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_INT)
 871                 return 0;
 872
 873         if (!asn1_uint_decode(ctx, end, &request->error_index))
 874                 return 0;
 875
 876         return 1;
 877 }
 878
 879 /*
 880  * Fast checksum update for possibly oddly-aligned UDP byte, from the
 881  * code example in the draft.
 882  */
 883 static void fast_csum(unsigned char *csum,
 884                       const unsigned char *optr,
 885                       const unsigned char *nptr,
 886                       int odd)
 887 {
 888         long x, old, new;
 889
 890         x = csum[0] * 256 + csum[1];
 891
 892         x =~ x & 0xFFFF;
 893
 894         if (odd) old = optr[0] * 256;
 895         else old = optr[0];
 896
 897         x -= old & 0xFFFF;
 898         if (x <= 0) {
 899                 x--;
 900                 x &= 0xFFFF;
 901         }
 902
 903         if (odd) new = nptr[0] * 256;
 904         else new = nptr[0];
 905
 906         x += new & 0xFFFF;
 907         if (x & 0x10000) {
 908                 x++;
 909                 x &= 0xFFFF;
 910         }
 911
 912         x =~ x & 0xFFFF;
 913         csum[0] = x / 256;
 914         csum[1] = x & 0xFF;
 915 }
 916
 917 /*
 918  * Mangle IP address.
 919  *      - begin points to the start of the snmp messgae
 920  *      - addr points to the start of the address
 921  */
 922 static void inline mangle_address(unsigned char *begin,
 923                                   unsigned char *addr,
 924                                   const struct oct1_map *map,
 925                                   u_int16_t *check)
 926 {
 927         if (map->from == NOCT1(*addr)) {
 928                 u_int32_t old;
 929
 930                 if (debug)
 931                         memcpy(&old, (unsigned char *)addr, sizeof(old));
 932
 933                 *addr = map->to;
 934
 935                 /* Update UDP checksum if being used */
 936                 if (*check) {
 937                         unsigned char odd = !((addr - begin) % 2);
 938
 939                         fast_csum((unsigned char *)check,
 940                                   &map->from, &map->to, odd);
 941
 942                 }
 943
 944                 if (debug)
 945                         printk(KERN_DEBUG "bsalg: mapped %u.%u.%u.%u to "
 946                                "%u.%u.%u.%u\n", NIPQUAD(old), NIPQUAD(*addr));
 947         }
 948 }
 949
 950 static unsigned char snmp_trap_decode(struct asn1_ctx *ctx,
 951                                       struct snmp_v1_trap *trap,
 952                                       const struct oct1_map *map,
 953                                       u_int16_t *check)
 954 {
 955         unsigned int cls, con, tag, len;
 956         unsigned char *end;
 957
 958         if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
 959                 return 0;
 960
 961         if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_OJI)
 962                 return 0;
 963
 964         if (!asn1_oid_decode(ctx, end, &trap->id, &trap->id_len))
 965                 return 0;
 966
 967         if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
 968                 goto err_id_free;
 969
 970         if (!((cls == ASN1_APL && con == ASN1_PRI && tag == SNMP_IPA) ||
 971               (cls == ASN1_UNI && con == ASN1_PRI && tag == ASN1_OTS)))
 972                 goto err_id_free;
 973
 974         if (!asn1_octets_decode(ctx, end, (unsigned char **)&trap->ip_address, &len))
 975                 goto err_id_free;
 976
 977         /* IPv4 only */
 978         if (len != 4)
 979                 goto err_addr_free;
 980
 981         mangle_address(ctx->begin, ctx->pointer - 4, map, check);
 982
 983         if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
 984                 goto err_addr_free;
 985
 986         if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_INT)
 987                 goto err_addr_free;;
 988
 989         if (!asn1_uint_decode(ctx, end, &trap->general))
 990                 goto err_addr_free;;
 991
 992         if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
 993                 goto err_addr_free;
 994
 995         if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_INT)
 996                 goto err_addr_free;
 997
 998         if (!asn1_uint_decode(ctx, end, &trap->specific))
 999                 goto err_addr_free;
1000
1001         if (!asn1_header_decode(ctx, &end, &cls, &con, &tag))
1002                 goto err_addr_free;
1003
1004         if (!((cls == ASN1_APL && con == ASN1_PRI && tag == SNMP_TIT) ||
1005               (cls == ASN1_UNI && con == ASN1_PRI && tag == ASN1_INT)))
1006                 goto err_addr_free;
1007
1008         if (!asn1_ulong_decode(ctx, end, &trap->time))
1009                 goto err_addr_free;
1010
1011         return 1;
1012
1013 err_addr_free:
1014         kfree((unsigned long *)trap->ip_address);
1015
1016 err_id_free:
1017         kfree(trap->id);
1018
1019         return 0;
1020 }
1021
1022 /*****************************************************************************
1023  *
1024  * Misc. routines
1025  *
1026  *****************************************************************************/
1027
1028 static void hex_dump(unsigned char *buf, size_t len)
1029 {
1030         size_t i;
1031
1032         for (i = 0; i < len; i++) {
1033                 if (i && !(i % 16))
1034                         printk("\n");
1035                 printk("%02x ", *(buf + i));
1036         }
1037         printk("\n");
1038 }
1039
1040 /*
1041  * Parse and mangle SNMP message according to mapping.
1042  * (And this is the fucking 'basic' method).
1043  */
1044 static int snmp_parse_mangle(unsigned char *msg,
1045                              u_int16_t len,
1046                              const struct oct1_map *map,
1047                              u_int16_t *check)
1048 {
1049         unsigned char *eoc, *end;
1050         unsigned int cls, con, tag, vers, pdutype;
1051         struct asn1_ctx ctx;
1052         struct asn1_octstr comm;
1053         struct snmp_object **obj;
1054
1055         if (debug > 1)
1056                 hex_dump(msg, len);
1057
1058         asn1_open(&ctx, msg, len);
1059
1060         /*
1061          * Start of SNMP message.
1062          */
1063         if (!asn1_header_decode(&ctx, &eoc, &cls, &con, &tag))
1064                 return 0;
1065         if (cls != ASN1_UNI || con != ASN1_CON || tag != ASN1_SEQ)
1066                 return 0;
1067
1068         /*
1069          * Version 1 or 2 handled.
1070          */
1071         if (!asn1_header_decode(&ctx, &end, &cls, &con, &tag))
1072                 return 0;
1073         if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_INT)
1074                 return 0;
1075         if (!asn1_uint_decode (&ctx, end, &vers))
1076                 return 0;
1077         if (debug > 1)
1078                 printk(KERN_DEBUG "bsalg: snmp version: %u\n", vers + 1);
1079         if (vers > 1)
1080                 return 1;
1081
1082         /*
1083          * Community.
1084          */
1085         if (!asn1_header_decode (&ctx, &end, &cls, &con, &tag))
1086                 return 0;
1087         if (cls != ASN1_UNI || con != ASN1_PRI || tag != ASN1_OTS)
1088                 return 0;
1089         if (!asn1_octets_decode(&ctx, end, &comm.data, &comm.len))
1090                 return 0;
1091         if (debug > 1) {
1092                 unsigned int i;
1093
1094                 printk(KERN_DEBUG "bsalg: community: ");
1095                 for (i = 0; i < comm.len; i++)
1096                         printk("%c", comm.data[i]);
1097                 printk("\n");
1098         }
1099         kfree(comm.data);
1100
1101         /*
1102          * PDU type
1103          */
1104         if (!asn1_header_decode(&ctx, &eoc, &cls, &con, &pdutype))
1105                 return 0;
1106         if (cls != ASN1_CTX || con != ASN1_CON)
1107                 return 0;
1108         if (debug > 1) {
1109                 unsigned char *pdus[] = {
1110                         [SNMP_PDU_GET] = "get",
1111                         [SNMP_PDU_NEXT] = "get-next",
1112                         [SNMP_PDU_RESPONSE] = "response",
1113                         [SNMP_PDU_SET] = "set",
1114                         [SNMP_PDU_TRAP1] = "trapv1",
1115                         [SNMP_PDU_BULK] = "bulk",
1116                         [SNMP_PDU_INFORM] = "inform",
1117                         [SNMP_PDU_TRAP2] = "trapv2"
1118                 };
1119
1120                 if (pdutype > SNMP_PDU_TRAP2)
1121                         printk(KERN_DEBUG "bsalg: bad pdu type %u\n", pdutype);
1122                 else
1123                         printk(KERN_DEBUG "bsalg: pdu: %s\n", pdus[pdutype]);
1124         }
1125         if (pdutype != SNMP_PDU_RESPONSE &&
1126             pdutype != SNMP_PDU_TRAP1 && pdutype != SNMP_PDU_TRAP2)
1127                 return 1;
1128
1129         /*
1130          * Request header or v1 trap
1131          */
1132         if (pdutype == SNMP_PDU_TRAP1) {
1133                 struct snmp_v1_trap trap;
1134                 unsigned char ret = snmp_trap_decode(&ctx, &trap, map, check);
1135
1136                 if (ret) {
1137                         kfree(trap.id);
1138                         kfree((unsigned long *)trap.ip_address);
1139                 } else
1140                         return ret;
1141
1142         } else {
1143                 struct snmp_request req;
1144
1145                 if (!snmp_request_decode(&ctx, &req))
1146                         return 0;
1147
1148                 if (debug > 1)
1149                         printk(KERN_DEBUG "bsalg: request: id=0x%lx error_status=%u "
1150                         "error_index=%u\n", req.id, req.error_status,
1151                         req.error_index);
1152         }
1153
1154         /*
1155          * Loop through objects, look for IP addresses to mangle.
1156          */
1157         if (!asn1_header_decode(&ctx, &eoc, &cls, &con, &tag))
1158                 return 0;
1159
1160         if (cls != ASN1_UNI || con != ASN1_CON || tag != ASN1_SEQ)
1161                 return 0;
1162
1163         obj = kmalloc(sizeof(struct snmp_object), GFP_ATOMIC);
1164         if (obj == NULL) {
1165                 if (net_ratelimit())
1166                         printk(KERN_WARNING "OOM in bsalg(%d)\n", __LINE__);
1167                 return 0;
1168         }
1169
1170         while (!asn1_eoc_decode(&ctx, eoc)) {
1171                 unsigned int i;
1172
1173                 if (!snmp_object_decode(&ctx, obj)) {
1174                         if (*obj) {
1175                                 if ((*obj)->id)
1176                                         kfree((*obj)->id);
1177                                 kfree(*obj);
1178                         }
1179                         kfree(obj);
1180                         return 0;
1181                 }
1182
1183                 if (debug > 1) {
1184                         printk(KERN_DEBUG "bsalg: object: ");
1185                         for (i = 0; i < (*obj)->id_len; i++) {
1186                                 if (i > 0)
1187                                         printk(".");
1188                                 printk("%lu", (*obj)->id[i]);
1189                         }
1190                         printk(": type=%u\n", (*obj)->type);
1191
1192                 }
1193
1194                 if ((*obj)->type == SNMP_IPADDR)
1195                         mangle_address(ctx.begin, ctx.pointer - 4 , map, check);
1196
1197                 kfree((*obj)->id);
1198                 kfree(*obj);
1199         }
1200         kfree(obj);
1201
1202         if (!asn1_eoc_decode(&ctx, eoc))
1203                 return 0;
1204
1205         return 1;
1206 }
1207
1208 /*****************************************************************************
1209  *
1210  * NAT routines.
1211  *
1212  *****************************************************************************/
1213
1214 /*
1215  * SNMP translation routine.
1216  */
1217 static int snmp_translate(struct ip_conntrack *ct,
1218                           struct ip_nat_info *info,
1219                           enum ip_conntrack_info ctinfo,
1220                           unsigned int hooknum,
1221                           struct sk_buff **pskb)
1222 {
1223         struct iphdr *iph = (*pskb)->nh.iph;
1224         struct udphdr *udph = (struct udphdr *)((u_int32_t *)iph + iph->ihl);
1225         u_int16_t udplen = ntohs(udph->len);
1226         u_int16_t paylen = udplen - sizeof(struct udphdr);
1227         int dir = CTINFO2DIR(ctinfo);
1228         struct oct1_map map;
1229
1230         /*
1231          * Determine mappping for application layer addresses based
1232          * on NAT manipulations for the packet.
1233          */
1234         if (dir == IP_CT_DIR_ORIGINAL) {
1235                 /* SNAT traps */
1236                 map.from = NOCT1(ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.src.ip);
1237                 map.to = NOCT1(ct->tuplehash[IP_CT_DIR_REPLY].tuple.dst.ip);
1238         } else {
1239                 /* DNAT replies */
1240                 map.from = NOCT1(ct->tuplehash[IP_CT_DIR_REPLY].tuple.src.ip);
1241                 map.to = NOCT1(ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.dst.ip);
1242         }
1243
1244         if (map.from == map.to)
1245                 return NF_ACCEPT;
1246
1247         if (!snmp_parse_mangle((unsigned char *)udph + sizeof(struct udphdr),
1248                                paylen, &map, &udph->check)) {
1249                 printk(KERN_WARNING "bsalg: parser failed\n");
1250                 return NF_DROP;
1251         }
1252         return NF_ACCEPT;
1253 }
1254
1255 /*
1256  * NAT helper function, packets arrive here from NAT code.
1257  */
1258 static unsigned int nat_help(struct ip_conntrack *ct,
1259                              struct ip_conntrack_expect *exp,
1260                              struct ip_nat_info *info,
1261                              enum ip_conntrack_info ctinfo,
1262                              unsigned int hooknum,
1263                              struct sk_buff **pskb)
1264 {
1265         int dir = CTINFO2DIR(ctinfo);
1266         struct iphdr *iph = (*pskb)->nh.iph;
1267         struct udphdr *udph = (struct udphdr *)((u_int32_t *)iph + iph->ihl);
1268
1269         spin_lock_bh(&snmp_lock);
1270
1271         /*
1272          * Translate snmp replies on pre-routing (DNAT) and snmp traps
1273          * on post routing (SNAT).
1274          */
1275         if (!((dir == IP_CT_DIR_REPLY && hooknum == NF_IP_PRE_ROUTING &&
1276                         udph->source == ntohs(SNMP_PORT)) ||
1277               (dir == IP_CT_DIR_ORIGINAL && hooknum == NF_IP_POST_ROUTING &&
1278                         udph->dest == ntohs(SNMP_TRAP_PORT)))) {
1279                 spin_unlock_bh(&snmp_lock);
1280                 return NF_ACCEPT;
1281         }
1282
1283         if (debug > 1) {
1284                 printk(KERN_DEBUG "bsalg: dir=%s hook=%d manip=%s len=%d "
1285                        "src=%u.%u.%u.%u:%u dst=%u.%u.%u.%u:%u "
1286                        "osrc=%u.%u.%u.%u odst=%u.%u.%u.%u "
1287                        "rsrc=%u.%u.%u.%u rdst=%u.%u.%u.%u "
1288                        "\n",
1289                        dir == IP_CT_DIR_REPLY ? "reply" : "orig", hooknum,
1290                        HOOK2MANIP(hooknum) == IP_NAT_MANIP_SRC ? "snat" :
1291                        "dnat", (*pskb)->len,
1292                        NIPQUAD(iph->saddr), ntohs(udph->source),
1293                        NIPQUAD(iph->daddr), ntohs(udph->dest),
1294                        NIPQUAD(ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.src.ip),
1295                        NIPQUAD(ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.dst.ip),
1296                        NIPQUAD(ct->tuplehash[IP_CT_DIR_REPLY].tuple.src.ip),
1297                        NIPQUAD(ct->tuplehash[IP_CT_DIR_REPLY].tuple.dst.ip));
1298         }
1299
1300         /*
1301          * Make sure the packet length is ok.  So far, we were only guaranteed
1302          * to have a valid length IP header plus 8 bytes, which means we have
1303          * enough room for a UDP header.  Just verify the UDP length field so we
1304          * can mess around with the payload.
1305          */
1306          if (ntohs(udph->len) == (*pskb)->len - (iph->ihl << 2)) {
1307                 int ret = snmp_translate(ct, info, ctinfo, hooknum, pskb);
1308                 spin_unlock_bh(&snmp_lock);
1309                 return ret;
1310         }
1311
1312         if (net_ratelimit())
1313                 printk(KERN_WARNING "bsalg: dropping malformed packet "
1314                        "src=%u.%u.%u.%u dst=%u.%u.%u.%u\n",
1315                        NIPQUAD(iph->saddr), NIPQUAD(iph->daddr));
1316         spin_unlock_bh(&snmp_lock);
1317         return NF_DROP;
1318 }
1319
1320 static struct ip_nat_helper snmp = {
1321         { NULL, NULL },
1322         "snmp",
1323         IP_NAT_HELPER_F_STANDALONE,
1324         THIS_MODULE,
1325         { { 0, { .udp = { __constant_htons(SNMP_PORT) } } },
1326           { 0, { 0 }, IPPROTO_UDP } },
1327         { { 0, { .udp = { 0xFFFF } } },
1328           { 0, { 0 }, 0xFFFF } },
1329         nat_help, NULL };
1330
1331 static struct ip_nat_helper snmp_trap = {
1332         { NULL, NULL },
1333         "snmp_trap",
1334         IP_NAT_HELPER_F_STANDALONE,
1335         THIS_MODULE,
1336         { { 0, { .udp = { __constant_htons(SNMP_TRAP_PORT) } } },
1337           { 0, { 0 }, IPPROTO_UDP } },
1338         { { 0, { .udp = { 0xFFFF } } },
1339           { 0, { 0 }, 0xFFFF } },
1340         nat_help, NULL };
1341
1342 /*****************************************************************************
1343  *
1344  * Module stuff.
1345  *
1346  *****************************************************************************/
1347
1348 static int __init init(void)
1349 {
1350         int ret = 0;
1351
1352         ret = ip_nat_helper_register(&snmp);
1353         if (ret < 0)
1354                 return ret;
1355         ret = ip_nat_helper_register(&snmp_trap);
1356         if (ret < 0) {
1357                 ip_nat_helper_unregister(&snmp);
1358                 return ret;
1359         }
1360         return ret;
1361 }
1362
1363 static void __exit fini(void)
1364 {
1365         ip_nat_helper_unregister(&snmp);
1366         ip_nat_helper_unregister(&snmp_trap);
1367         br_write_lock_bh(BR_NETPROTO_LOCK);
1368         br_write_unlock_bh(BR_NETPROTO_LOCK);
1369 }
1370
1371 module_init(init);
1372 module_exit(fini);
1373
1374 MODULE_PARM(debug, "i");
1375 MODULE_DESCRIPTION("Basic SNMP Application Layer Gateway");
1376 MODULE_LICENSE("GPL");