net/netfilter/nf_conntrack_proto_sctp.c

   1 /*
   2  * Connection tracking protocol helper module for SCTP.
   3  *
   4  * Copyright (c) 2004 Kiran Kumar Immidi <immidi_kiran@yahoo.com>
   5  * Copyright (c) 2004-2012 Patrick McHardy <kaber@trash.net>
   6  *
   7  * SCTP is defined in RFC 2960. References to various sections in this code
   8  * are to this RFC.
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License version 2 as
  12  * published by the Free Software Foundation.
  13  */
  14
  15 #include <linux/types.h>
  16 #include <linux/timer.h>
  17 #include <linux/netfilter.h>
  18 #include <linux/in.h>
  19 #include <linux/ip.h>
  20 #include <linux/sctp.h>
  21 #include <linux/string.h>
  22 #include <linux/seq_file.h>
  23 #include <linux/spinlock.h>
  24 #include <linux/interrupt.h>
  25 #include <net/sctp/checksum.h>
  26
  27 #include <net/netfilter/nf_log.h>
  28 #include <net/netfilter/nf_conntrack.h>
  29 #include <net/netfilter/nf_conntrack_l4proto.h>
  30 #include <net/netfilter/nf_conntrack_ecache.h>
  31
  32 /* FIXME: Examine ipfilter's timeouts and conntrack transitions more
  33    closely.  They're more complex. --RR
  34
  35    And so for me for SCTP :D -Kiran */
  36
  37 static const char *const sctp_conntrack_names[] = {
  38         "NONE",
  39         "CLOSED",
  40         "COOKIE_WAIT",
  41         "COOKIE_ECHOED",
  42         "ESTABLISHED",
  43         "SHUTDOWN_SENT",
  44         "SHUTDOWN_RECD",
  45         "SHUTDOWN_ACK_SENT",
  46         "HEARTBEAT_SENT",
  47         "HEARTBEAT_ACKED",
  48 };
  49
  50 #define SECS  * HZ
  51 #define MINS  * 60 SECS
  52 #define HOURS * 60 MINS
  53 #define DAYS  * 24 HOURS
  54
  55 static unsigned int sctp_timeouts[SCTP_CONNTRACK_MAX] __read_mostly = {
  56         [SCTP_CONNTRACK_CLOSED]                 = 10 SECS,
  57         [SCTP_CONNTRACK_COOKIE_WAIT]            = 3 SECS,
  58         [SCTP_CONNTRACK_COOKIE_ECHOED]          = 3 SECS,
  59         [SCTP_CONNTRACK_ESTABLISHED]            = 5 DAYS,
  60         [SCTP_CONNTRACK_SHUTDOWN_SENT]          = 3 SECS,
  61         [SCTP_CONNTRACK_SHUTDOWN_RECD]          = 3 SECS,
  62         [SCTP_CONNTRACK_SHUTDOWN_ACK_SENT]      = 3 SECS,
  63         [SCTP_CONNTRACK_HEARTBEAT_SENT]         = 30 SECS,
  64         [SCTP_CONNTRACK_HEARTBEAT_ACKED]        = 210 SECS,
  65 };
  66
  67 #define sNO SCTP_CONNTRACK_NONE
  68 #define sCL SCTP_CONNTRACK_CLOSED
  69 #define sCW SCTP_CONNTRACK_COOKIE_WAIT
  70 #define sCE SCTP_CONNTRACK_COOKIE_ECHOED
  71 #define sES SCTP_CONNTRACK_ESTABLISHED
  72 #define sSS SCTP_CONNTRACK_SHUTDOWN_SENT
  73 #define sSR SCTP_CONNTRACK_SHUTDOWN_RECD
  74 #define sSA SCTP_CONNTRACK_SHUTDOWN_ACK_SENT
  75 #define sHS SCTP_CONNTRACK_HEARTBEAT_SENT
  76 #define sHA SCTP_CONNTRACK_HEARTBEAT_ACKED
  77 #define sIV SCTP_CONNTRACK_MAX
  78
  79 /*
  80         These are the descriptions of the states:
  81
  82 NOTE: These state names are tantalizingly similar to the states of an
  83 SCTP endpoint. But the interpretation of the states is a little different,
  84 considering that these are the states of the connection and not of an end
  85 point. Please note the subtleties. -Kiran
  86
  87 NONE              - Nothing so far.
  88 COOKIE WAIT       - We have seen an INIT chunk in the original direction, or also
  89                     an INIT_ACK chunk in the reply direction.
  90 COOKIE ECHOED     - We have seen a COOKIE_ECHO chunk in the original direction.
  91 ESTABLISHED       - We have seen a COOKIE_ACK in the reply direction.
  92 SHUTDOWN_SENT     - We have seen a SHUTDOWN chunk in the original direction.
  93 SHUTDOWN_RECD     - We have seen a SHUTDOWN chunk in the reply directoin.
  94 SHUTDOWN_ACK_SENT - We have seen a SHUTDOWN_ACK chunk in the direction opposite
  95                     to that of the SHUTDOWN chunk.
  96 CLOSED            - We have seen a SHUTDOWN_COMPLETE chunk in the direction of
  97                     the SHUTDOWN chunk. Connection is closed.
  98 HEARTBEAT_SENT    - We have seen a HEARTBEAT in a new flow.
  99 HEARTBEAT_ACKED   - We have seen a HEARTBEAT-ACK in the direction opposite to
 100                     that of the HEARTBEAT chunk. Secondary connection is
 101                     established.
 102 */
 103
 104 /* TODO
 105  - I have assumed that the first INIT is in the original direction.
 106  This messes things when an INIT comes in the reply direction in CLOSED
 107  state.
 108  - Check the error type in the reply dir before transitioning from
 109 cookie echoed to closed.
 110  - Sec 5.2.4 of RFC 2960
 111  - Full Multi Homing support.
 112 */
 113
 114 /* SCTP conntrack state transitions */
 115 static const u8 sctp_conntracks[2][11][SCTP_CONNTRACK_MAX] = {
 116         {
 117 /*      ORIGINAL        */
 118 /*                  sNO, sCL, sCW, sCE, sES, sSS, sSR, sSA, sHS, sHA */
 119 /* init         */ {sCW, sCW, sCW, sCE, sES, sCL, sCL, sSA, sCW, sHA},
 120 /* init_ack     */ {sCL, sCL, sCW, sCE, sES, sSS, sSR, sSA, sCL, sHA},
 121 /* abort        */ {sCL, sCL, sCL, sCL, sCL, sCL, sCL, sCL, sCL, sCL},
 122 /* shutdown     */ {sCL, sCL, sCW, sCE, sSS, sSS, sSR, sSA, sCL, sSS},
 123 /* shutdown_ack */ {sSA, sCL, sCW, sCE, sES, sSA, sSA, sSA, sSA, sHA},
 124 /* error        */ {sCL, sCL, sCW, sCE, sES, sSS, sSR, sSA, sCL, sHA},/* Can't have Stale cookie*/
 125 /* cookie_echo  */ {sCL, sCL, sCE, sCE, sES, sSS, sSR, sSA, sCL, sHA},/* 5.2.4 - Big TODO */
 126 /* cookie_ack   */ {sCL, sCL, sCW, sCE, sES, sSS, sSR, sSA, sCL, sHA},/* Can't come in orig dir */
 127 /* shutdown_comp*/ {sCL, sCL, sCW, sCE, sES, sSS, sSR, sCL, sCL, sHA},
 128 /* heartbeat    */ {sHS, sCL, sCW, sCE, sES, sSS, sSR, sSA, sHS, sHA},
 129 /* heartbeat_ack*/ {sCL, sCL, sCW, sCE, sES, sSS, sSR, sSA, sHS, sHA}
 130         },
 131         {
 132 /*      REPLY   */
 133 /*                  sNO, sCL, sCW, sCE, sES, sSS, sSR, sSA, sHS, sHA */
 134 /* init         */ {sIV, sCL, sCW, sCE, sES, sSS, sSR, sSA, sIV, sHA},/* INIT in sCL Big TODO */
 135 /* init_ack     */ {sIV, sCL, sCW, sCE, sES, sSS, sSR, sSA, sIV, sHA},
 136 /* abort        */ {sIV, sCL, sCL, sCL, sCL, sCL, sCL, sCL, sIV, sCL},
 137 /* shutdown     */ {sIV, sCL, sCW, sCE, sSR, sSS, sSR, sSA, sIV, sSR},
 138 /* shutdown_ack */ {sIV, sCL, sCW, sCE, sES, sSA, sSA, sSA, sIV, sHA},
 139 /* error        */ {sIV, sCL, sCW, sCL, sES, sSS, sSR, sSA, sIV, sHA},
 140 /* cookie_echo  */ {sIV, sCL, sCW, sCE, sES, sSS, sSR, sSA, sIV, sHA},/* Can't come in reply dir */
 141 /* cookie_ack   */ {sIV, sCL, sCW, sES, sES, sSS, sSR, sSA, sIV, sHA},
 142 /* shutdown_comp*/ {sIV, sCL, sCW, sCE, sES, sSS, sSR, sCL, sIV, sHA},
 143 /* heartbeat    */ {sIV, sCL, sCW, sCE, sES, sSS, sSR, sSA, sHS, sHA},
 144 /* heartbeat_ack*/ {sIV, sCL, sCW, sCE, sES, sSS, sSR, sSA, sHA, sHA}
 145         }
 146 };
 147
 148 static inline struct nf_sctp_net *sctp_pernet(struct net *net)
 149 {
 150         return &net->ct.nf_ct_proto.sctp;
 151 }
 152
 153 static bool sctp_pkt_to_tuple(const struct sk_buff *skb, unsigned int dataoff,
 154                               struct net *net, struct nf_conntrack_tuple *tuple)
 155 {
 156         const struct sctphdr *hp;
 157         struct sctphdr _hdr;
 158
 159         /* Actually only need first 4 bytes to get ports. */
 160         hp = skb_header_pointer(skb, dataoff, 4, &_hdr);
 161         if (hp == NULL)
 162                 return false;
 163
 164         tuple->src.u.sctp.port = hp->source;
 165         tuple->dst.u.sctp.port = hp->dest;
 166         return true;
 167 }
 168
 169 static bool sctp_invert_tuple(struct nf_conntrack_tuple *tuple,
 170                               const struct nf_conntrack_tuple *orig)
 171 {
 172         tuple->src.u.sctp.port = orig->dst.u.sctp.port;
 173         tuple->dst.u.sctp.port = orig->src.u.sctp.port;
 174         return true;
 175 }
 176
 177 #ifdef CONFIG_NF_CONNTRACK_PROCFS
 178 /* Print out the private part of the conntrack. */
 179 static void sctp_print_conntrack(struct seq_file *s, struct nf_conn *ct)
 180 {
 181         seq_printf(s, "%s ", sctp_conntrack_names[ct->proto.sctp.state]);
 182 }
 183 #endif
 184
 185 #define for_each_sctp_chunk(skb, sch, _sch, offset, dataoff, count)     \
 186 for ((offset) = (dataoff) + sizeof(struct sctphdr), (count) = 0;        \
 187         (offset) < (skb)->len &&                                        \
 188         ((sch) = skb_header_pointer((skb), (offset), sizeof(_sch), &(_sch)));   \
 189         (offset) += (ntohs((sch)->length) + 3) & ~3, (count)++)
 190
 191 /* Some validity checks to make sure the chunks are fine */
 192 static int do_basic_checks(struct nf_conn *ct,
 193                            const struct sk_buff *skb,
 194                            unsigned int dataoff,
 195                            unsigned long *map)
 196 {
 197         u_int32_t offset, count;
 198         struct sctp_chunkhdr _sch, *sch;
 199         int flag;
 200
 201         flag = 0;
 202
 203         for_each_sctp_chunk (skb, sch, _sch, offset, dataoff, count) {
 204                 pr_debug("Chunk Num: %d  Type: %d\n", count, sch->type);
 205
 206                 if (sch->type == SCTP_CID_INIT ||
 207                     sch->type == SCTP_CID_INIT_ACK ||
 208                     sch->type == SCTP_CID_SHUTDOWN_COMPLETE)
 209                         flag = 1;
 210
 211                 /*
 212                  * Cookie Ack/Echo chunks not the first OR
 213                  * Init / Init Ack / Shutdown compl chunks not the only chunks
 214                  * OR zero-length.
 215                  */
 216                 if (((sch->type == SCTP_CID_COOKIE_ACK ||
 217                       sch->type == SCTP_CID_COOKIE_ECHO ||
 218                       flag) &&
 219                      count != 0) || !sch->length) {
 220                         pr_debug("Basic checks failed\n");
 221                         return 1;
 222                 }
 223
 224                 if (map)
 225                         set_bit(sch->type, map);
 226         }
 227
 228         pr_debug("Basic checks passed\n");
 229         return count == 0;
 230 }
 231
 232 static int sctp_new_state(enum ip_conntrack_dir dir,
 233                           enum sctp_conntrack cur_state,
 234                           int chunk_type)
 235 {
 236         int i;
 237
 238         pr_debug("Chunk type: %d\n", chunk_type);
 239
 240         switch (chunk_type) {
 241         case SCTP_CID_INIT:
 242                 pr_debug("SCTP_CID_INIT\n");
 243                 i = 0;
 244                 break;
 245         case SCTP_CID_INIT_ACK:
 246                 pr_debug("SCTP_CID_INIT_ACK\n");
 247                 i = 1;
 248                 break;
 249         case SCTP_CID_ABORT:
 250                 pr_debug("SCTP_CID_ABORT\n");
 251                 i = 2;
 252                 break;
 253         case SCTP_CID_SHUTDOWN:
 254                 pr_debug("SCTP_CID_SHUTDOWN\n");
 255                 i = 3;
 256                 break;
 257         case SCTP_CID_SHUTDOWN_ACK:
 258                 pr_debug("SCTP_CID_SHUTDOWN_ACK\n");
 259                 i = 4;
 260                 break;
 261         case SCTP_CID_ERROR:
 262                 pr_debug("SCTP_CID_ERROR\n");
 263                 i = 5;
 264                 break;
 265         case SCTP_CID_COOKIE_ECHO:
 266                 pr_debug("SCTP_CID_COOKIE_ECHO\n");
 267                 i = 6;
 268                 break;
 269         case SCTP_CID_COOKIE_ACK:
 270                 pr_debug("SCTP_CID_COOKIE_ACK\n");
 271                 i = 7;
 272                 break;
 273         case SCTP_CID_SHUTDOWN_COMPLETE:
 274                 pr_debug("SCTP_CID_SHUTDOWN_COMPLETE\n");
 275                 i = 8;
 276                 break;
 277         case SCTP_CID_HEARTBEAT:
 278                 pr_debug("SCTP_CID_HEARTBEAT");
 279                 i = 9;
 280                 break;
 281         case SCTP_CID_HEARTBEAT_ACK:
 282                 pr_debug("SCTP_CID_HEARTBEAT_ACK");
 283                 i = 10;
 284                 break;
 285         default:
 286                 /* Other chunks like DATA or SACK do not change the state */
 287                 pr_debug("Unknown chunk type, Will stay in %s\n",
 288                          sctp_conntrack_names[cur_state]);
 289                 return cur_state;
 290         }
 291
 292         pr_debug("dir: %d   cur_state: %s  chunk_type: %d  new_state: %s\n",
 293                  dir, sctp_conntrack_names[cur_state], chunk_type,
 294                  sctp_conntrack_names[sctp_conntracks[dir][i][cur_state]]);
 295
 296         return sctp_conntracks[dir][i][cur_state];
 297 }
 298
 299 static unsigned int *sctp_get_timeouts(struct net *net)
 300 {
 301         return sctp_pernet(net)->timeouts;
 302 }
 303
 304 /* Returns verdict for packet, or -NF_ACCEPT for invalid. */
 305 static int sctp_packet(struct nf_conn *ct,
 306                        const struct sk_buff *skb,
 307                        unsigned int dataoff,
 308                        enum ip_conntrack_info ctinfo,
 309                        u_int8_t pf,
 310                        unsigned int *timeouts)
 311 {
 312         enum sctp_conntrack new_state, old_state;
 313         enum ip_conntrack_dir dir = CTINFO2DIR(ctinfo);
 314         const struct sctphdr *sh;
 315         struct sctphdr _sctph;
 316         const struct sctp_chunkhdr *sch;
 317         struct sctp_chunkhdr _sch;
 318         u_int32_t offset, count;
 319         unsigned long map[256 / sizeof(unsigned long)] = { 0 };
 320
 321         sh = skb_header_pointer(skb, dataoff, sizeof(_sctph), &_sctph);
 322         if (sh == NULL)
 323                 goto out;
 324
 325         if (do_basic_checks(ct, skb, dataoff, map) != 0)
 326                 goto out;
 327
 328         /* Check the verification tag (Sec 8.5) */
 329         if (!test_bit(SCTP_CID_INIT, map) &&
 330             !test_bit(SCTP_CID_SHUTDOWN_COMPLETE, map) &&
 331             !test_bit(SCTP_CID_COOKIE_ECHO, map) &&
 332             !test_bit(SCTP_CID_ABORT, map) &&
 333             !test_bit(SCTP_CID_SHUTDOWN_ACK, map) &&
 334             !test_bit(SCTP_CID_HEARTBEAT, map) &&
 335             !test_bit(SCTP_CID_HEARTBEAT_ACK, map) &&
 336             sh->vtag != ct->proto.sctp.vtag[dir]) {
 337                 pr_debug("Verification tag check failed\n");
 338                 goto out;
 339         }
 340
 341         old_state = new_state = SCTP_CONNTRACK_NONE;
 342         spin_lock_bh(&ct->lock);
 343         for_each_sctp_chunk (skb, sch, _sch, offset, dataoff, count) {
 344                 /* Special cases of Verification tag check (Sec 8.5.1) */
 345                 if (sch->type == SCTP_CID_INIT) {
 346                         /* (A) vtag MUST be zero */
 347                         if (sh->vtag != 0)
 348                                 goto out_unlock;
 349                 } else if (sch->type == SCTP_CID_ABORT) {
 350                         /* (B) vtag MUST match own vtag if T flag is unset OR
 351                          * MUST match peer's vtag if T flag is set
 352                          */
 353                         if ((!(sch->flags & SCTP_CHUNK_FLAG_T) &&
 354                              sh->vtag != ct->proto.sctp.vtag[dir]) ||
 355                             ((sch->flags & SCTP_CHUNK_FLAG_T) &&
 356                              sh->vtag != ct->proto.sctp.vtag[!dir]))
 357                                 goto out_unlock;
 358                 } else if (sch->type == SCTP_CID_SHUTDOWN_COMPLETE) {
 359                         /* (C) vtag MUST match own vtag if T flag is unset OR
 360                          * MUST match peer's vtag if T flag is set
 361                          */
 362                         if ((!(sch->flags & SCTP_CHUNK_FLAG_T) &&
 363                              sh->vtag != ct->proto.sctp.vtag[dir]) ||
 364                             ((sch->flags & SCTP_CHUNK_FLAG_T) &&
 365                              sh->vtag != ct->proto.sctp.vtag[!dir]))
 366                                 goto out_unlock;
 367                 } else if (sch->type == SCTP_CID_COOKIE_ECHO) {
 368                         /* (D) vtag must be same as init_vtag as found in INIT_ACK */
 369                         if (sh->vtag != ct->proto.sctp.vtag[dir])
 370                                 goto out_unlock;
 371                 } else if (sch->type == SCTP_CID_HEARTBEAT ||
 372                            sch->type == SCTP_CID_HEARTBEAT_ACK) {
 373                         if (ct->proto.sctp.vtag[dir] == 0) {
 374                                 pr_debug("Setting vtag %x for dir %d\n",
 375                                          sh->vtag, dir);
 376                                 ct->proto.sctp.vtag[dir] = sh->vtag;
 377                         } else if (sh->vtag != ct->proto.sctp.vtag[dir]) {
 378                                 pr_debug("Verification tag check failed\n");
 379                                 goto out_unlock;
 380                         }
 381                 }
 382
 383                 old_state = ct->proto.sctp.state;
 384                 new_state = sctp_new_state(dir, old_state, sch->type);
 385
 386                 /* Invalid */
 387                 if (new_state == SCTP_CONNTRACK_MAX) {
 388                         pr_debug("nf_conntrack_sctp: Invalid dir=%i ctype=%u "
 389                                  "conntrack=%u\n",
 390                                  dir, sch->type, old_state);
 391                         goto out_unlock;
 392                 }
 393
 394                 /* If it is an INIT or an INIT ACK note down the vtag */
 395                 if (sch->type == SCTP_CID_INIT ||
 396                     sch->type == SCTP_CID_INIT_ACK) {
 397                         struct sctp_inithdr _inithdr, *ih;
 398
 399                         ih = skb_header_pointer(skb, offset + sizeof(_sch),
 400                                                 sizeof(_inithdr), &_inithdr);
 401                         if (ih == NULL)
 402                                 goto out_unlock;
 403                         pr_debug("Setting vtag %x for dir %d\n",
 404                                  ih->init_tag, !dir);
 405                         ct->proto.sctp.vtag[!dir] = ih->init_tag;
 406                 }
 407
 408                 ct->proto.sctp.state = new_state;
 409                 if (old_state != new_state)
 410                         nf_conntrack_event_cache(IPCT_PROTOINFO, ct);
 411         }
 412         spin_unlock_bh(&ct->lock);
 413
 414         nf_ct_refresh_acct(ct, ctinfo, skb, timeouts[new_state]);
 415
 416         if (old_state == SCTP_CONNTRACK_COOKIE_ECHOED &&
 417             dir == IP_CT_DIR_REPLY &&
 418             new_state == SCTP_CONNTRACK_ESTABLISHED) {
 419                 pr_debug("Setting assured bit\n");
 420                 set_bit(IPS_ASSURED_BIT, &ct->status);
 421                 nf_conntrack_event_cache(IPCT_ASSURED, ct);
 422         }
 423
 424         return NF_ACCEPT;
 425
 426 out_unlock:
 427         spin_unlock_bh(&ct->lock);
 428 out:
 429         return -NF_ACCEPT;
 430 }
 431
 432 /* Called when a new connection for this protocol found. */
 433 static bool sctp_new(struct nf_conn *ct, const struct sk_buff *skb,
 434                      unsigned int dataoff, unsigned int *timeouts)
 435 {
 436         enum sctp_conntrack new_state;
 437         const struct sctphdr *sh;
 438         struct sctphdr _sctph;
 439         const struct sctp_chunkhdr *sch;
 440         struct sctp_chunkhdr _sch;
 441         u_int32_t offset, count;
 442         unsigned long map[256 / sizeof(unsigned long)] = { 0 };
 443
 444         sh = skb_header_pointer(skb, dataoff, sizeof(_sctph), &_sctph);
 445         if (sh == NULL)
 446                 return false;
 447
 448         if (do_basic_checks(ct, skb, dataoff, map) != 0)
 449                 return false;
 450
 451         /* If an OOTB packet has any of these chunks discard (Sec 8.4) */
 452         if (test_bit(SCTP_CID_ABORT, map) ||
 453             test_bit(SCTP_CID_SHUTDOWN_COMPLETE, map) ||
 454             test_bit(SCTP_CID_COOKIE_ACK, map))
 455                 return false;
 456
 457         memset(&ct->proto.sctp, 0, sizeof(ct->proto.sctp));
 458         new_state = SCTP_CONNTRACK_MAX;
 459         for_each_sctp_chunk (skb, sch, _sch, offset, dataoff, count) {
 460                 /* Don't need lock here: this conntrack not in circulation yet */
 461                 new_state = sctp_new_state(IP_CT_DIR_ORIGINAL,
 462                                            SCTP_CONNTRACK_NONE, sch->type);
 463
 464                 /* Invalid: delete conntrack */
 465                 if (new_state == SCTP_CONNTRACK_NONE ||
 466                     new_state == SCTP_CONNTRACK_MAX) {
 467                         pr_debug("nf_conntrack_sctp: invalid new deleting.\n");
 468                         return false;
 469                 }
 470
 471                 /* Copy the vtag into the state info */
 472                 if (sch->type == SCTP_CID_INIT) {
 473                         struct sctp_inithdr _inithdr, *ih;
 474                         /* Sec 8.5.1 (A) */
 475                         if (sh->vtag)
 476                                 return false;
 477
 478                         ih = skb_header_pointer(skb, offset + sizeof(_sch),
 479                                                 sizeof(_inithdr), &_inithdr);
 480                         if (!ih)
 481                                 return false;
 482
 483                         pr_debug("Setting vtag %x for new conn\n",
 484                                  ih->init_tag);
 485
 486                         ct->proto.sctp.vtag[IP_CT_DIR_REPLY] = ih->init_tag;
 487                 } else if (sch->type == SCTP_CID_HEARTBEAT) {
 488                         pr_debug("Setting vtag %x for secondary conntrack\n",
 489                                  sh->vtag);
 490                         ct->proto.sctp.vtag[IP_CT_DIR_ORIGINAL] = sh->vtag;
 491                 }
 492                 /* If it is a shutdown ack OOTB packet, we expect a return
 493                    shutdown complete, otherwise an ABORT Sec 8.4 (5) and (8) */
 494                 else {
 495                         pr_debug("Setting vtag %x for new conn OOTB\n",
 496                                  sh->vtag);
 497                         ct->proto.sctp.vtag[IP_CT_DIR_REPLY] = sh->vtag;
 498                 }
 499
 500                 ct->proto.sctp.state = new_state;
 501         }
 502
 503         return true;
 504 }
 505
 506 static int sctp_error(struct net *net, struct nf_conn *tpl, struct sk_buff *skb,
 507                       unsigned int dataoff,
 508                       u8 pf, unsigned int hooknum)
 509 {
 510         const struct sctphdr *sh;
 511         const char *logmsg;
 512
 513         if (skb->len < dataoff + sizeof(struct sctphdr)) {
 514                 logmsg = "nf_ct_sctp: short packet ";
 515                 goto out_invalid;
 516         }
 517         if (net->ct.sysctl_checksum && hooknum == NF_INET_PRE_ROUTING &&
 518             skb->ip_summed == CHECKSUM_NONE) {
 519                 if (!skb_make_writable(skb, dataoff + sizeof(struct sctphdr))) {
 520                         logmsg = "nf_ct_sctp: failed to read header ";
 521                         goto out_invalid;
 522                 }
 523                 sh = (const struct sctphdr *)(skb->data + dataoff);
 524                 if (sh->checksum != sctp_compute_cksum(skb, dataoff)) {
 525                         logmsg = "nf_ct_sctp: bad CRC ";
 526                         goto out_invalid;
 527                 }
 528                 skb->ip_summed = CHECKSUM_UNNECESSARY;
 529         }
 530         return NF_ACCEPT;
 531 out_invalid:
 532         if (LOG_INVALID(net, IPPROTO_SCTP))
 533                 nf_log_packet(net, pf, 0, skb, NULL, NULL, NULL, "%s", logmsg);
 534         return -NF_ACCEPT;
 535 }
 536
 537 static bool sctp_can_early_drop(const struct nf_conn *ct)
 538 {
 539         switch (ct->proto.sctp.state) {
 540         case SCTP_CONNTRACK_SHUTDOWN_SENT:
 541         case SCTP_CONNTRACK_SHUTDOWN_RECD:
 542         case SCTP_CONNTRACK_SHUTDOWN_ACK_SENT:
 543                 return true;
 544         default:
 545                 break;
 546         }
 547
 548         return false;
 549 }
 550
 551 #if IS_ENABLED(CONFIG_NF_CT_NETLINK)
 552
 553 #include <linux/netfilter/nfnetlink.h>
 554 #include <linux/netfilter/nfnetlink_conntrack.h>
 555
 556 static int sctp_to_nlattr(struct sk_buff *skb, struct nlattr *nla,
 557                           struct nf_conn *ct)
 558 {
 559         struct nlattr *nest_parms;
 560
 561         spin_lock_bh(&ct->lock);
 562         nest_parms = nla_nest_start(skb, CTA_PROTOINFO_SCTP | NLA_F_NESTED);
 563         if (!nest_parms)
 564                 goto nla_put_failure;
 565
 566         if (nla_put_u8(skb, CTA_PROTOINFO_SCTP_STATE, ct->proto.sctp.state) ||
 567             nla_put_be32(skb, CTA_PROTOINFO_SCTP_VTAG_ORIGINAL,
 568                          ct->proto.sctp.vtag[IP_CT_DIR_ORIGINAL]) ||
 569             nla_put_be32(skb, CTA_PROTOINFO_SCTP_VTAG_REPLY,
 570                          ct->proto.sctp.vtag[IP_CT_DIR_REPLY]))
 571                 goto nla_put_failure;
 572
 573         spin_unlock_bh(&ct->lock);
 574
 575         nla_nest_end(skb, nest_parms);
 576
 577         return 0;
 578
 579 nla_put_failure:
 580         spin_unlock_bh(&ct->lock);
 581         return -1;
 582 }
 583
 584 static const struct nla_policy sctp_nla_policy[CTA_PROTOINFO_SCTP_MAX+1] = {
 585         [CTA_PROTOINFO_SCTP_STATE]          = { .type = NLA_U8 },
 586         [CTA_PROTOINFO_SCTP_VTAG_ORIGINAL]  = { .type = NLA_U32 },
 587         [CTA_PROTOINFO_SCTP_VTAG_REPLY]     = { .type = NLA_U32 },
 588 };
 589
 590 static int nlattr_to_sctp(struct nlattr *cda[], struct nf_conn *ct)
 591 {
 592         struct nlattr *attr = cda[CTA_PROTOINFO_SCTP];
 593         struct nlattr *tb[CTA_PROTOINFO_SCTP_MAX+1];
 594         int err;
 595
 596         /* updates may not contain the internal protocol info, skip parsing */
 597         if (!attr)
 598                 return 0;
 599
 600         err = nla_parse_nested(tb, CTA_PROTOINFO_SCTP_MAX, attr,
 601                                sctp_nla_policy, NULL);
 602         if (err < 0)
 603                 return err;
 604
 605         if (!tb[CTA_PROTOINFO_SCTP_STATE] ||
 606             !tb[CTA_PROTOINFO_SCTP_VTAG_ORIGINAL] ||
 607             !tb[CTA_PROTOINFO_SCTP_VTAG_REPLY])
 608                 return -EINVAL;
 609
 610         spin_lock_bh(&ct->lock);
 611         ct->proto.sctp.state = nla_get_u8(tb[CTA_PROTOINFO_SCTP_STATE]);
 612         ct->proto.sctp.vtag[IP_CT_DIR_ORIGINAL] =
 613                 nla_get_be32(tb[CTA_PROTOINFO_SCTP_VTAG_ORIGINAL]);
 614         ct->proto.sctp.vtag[IP_CT_DIR_REPLY] =
 615                 nla_get_be32(tb[CTA_PROTOINFO_SCTP_VTAG_REPLY]);
 616         spin_unlock_bh(&ct->lock);
 617
 618         return 0;
 619 }
 620
 621 static int sctp_nlattr_size(void)
 622 {
 623         return nla_total_size(0)        /* CTA_PROTOINFO_SCTP */
 624                 + nla_policy_len(sctp_nla_policy, CTA_PROTOINFO_SCTP_MAX + 1);
 625 }
 626 #endif
 627
 628 #if IS_ENABLED(CONFIG_NF_CT_NETLINK_TIMEOUT)
 629
 630 #include <linux/netfilter/nfnetlink.h>
 631 #include <linux/netfilter/nfnetlink_cttimeout.h>
 632
 633 static int sctp_timeout_nlattr_to_obj(struct nlattr *tb[],
 634                                       struct net *net, void *data)
 635 {
 636         unsigned int *timeouts = data;
 637         struct nf_sctp_net *sn = sctp_pernet(net);
 638         int i;
 639
 640         /* set default SCTP timeouts. */
 641         for (i=0; i<SCTP_CONNTRACK_MAX; i++)
 642                 timeouts[i] = sn->timeouts[i];
 643
 644         /* there's a 1:1 mapping between attributes and protocol states. */
 645         for (i=CTA_TIMEOUT_SCTP_UNSPEC+1; i<CTA_TIMEOUT_SCTP_MAX+1; i++) {
 646                 if (tb[i]) {
 647                         timeouts[i] = ntohl(nla_get_be32(tb[i])) * HZ;
 648                 }
 649         }
 650         return 0;
 651 }
 652
 653 static int
 654 sctp_timeout_obj_to_nlattr(struct sk_buff *skb, const void *data)
 655 {
 656         const unsigned int *timeouts = data;
 657         int i;
 658
 659         for (i=CTA_TIMEOUT_SCTP_UNSPEC+1; i<CTA_TIMEOUT_SCTP_MAX+1; i++) {
 660                 if (nla_put_be32(skb, i, htonl(timeouts[i] / HZ)))
 661                         goto nla_put_failure;
 662         }
 663         return 0;
 664
 665 nla_put_failure:
 666         return -ENOSPC;
 667 }
 668
 669 static const struct nla_policy
 670 sctp_timeout_nla_policy[CTA_TIMEOUT_SCTP_MAX+1] = {
 671         [CTA_TIMEOUT_SCTP_CLOSED]               = { .type = NLA_U32 },
 672         [CTA_TIMEOUT_SCTP_COOKIE_WAIT]          = { .type = NLA_U32 },
 673         [CTA_TIMEOUT_SCTP_COOKIE_ECHOED]        = { .type = NLA_U32 },
 674         [CTA_TIMEOUT_SCTP_ESTABLISHED]          = { .type = NLA_U32 },
 675         [CTA_TIMEOUT_SCTP_SHUTDOWN_SENT]        = { .type = NLA_U32 },
 676         [CTA_TIMEOUT_SCTP_SHUTDOWN_RECD]        = { .type = NLA_U32 },
 677         [CTA_TIMEOUT_SCTP_SHUTDOWN_ACK_SENT]    = { .type = NLA_U32 },
 678         [CTA_TIMEOUT_SCTP_HEARTBEAT_SENT]       = { .type = NLA_U32 },
 679         [CTA_TIMEOUT_SCTP_HEARTBEAT_ACKED]      = { .type = NLA_U32 },
 680 };
 681 #endif /* CONFIG_NF_CT_NETLINK_TIMEOUT */
 682
 683
 684 #ifdef CONFIG_SYSCTL
 685 static struct ctl_table sctp_sysctl_table[] = {
 686         {
 687                 .procname       = "nf_conntrack_sctp_timeout_closed",
 688                 .maxlen         = sizeof(unsigned int),
 689                 .mode           = 0644,
 690                 .proc_handler   = proc_dointvec_jiffies,
 691         },
 692         {
 693                 .procname       = "nf_conntrack_sctp_timeout_cookie_wait",
 694                 .maxlen         = sizeof(unsigned int),
 695                 .mode           = 0644,
 696                 .proc_handler   = proc_dointvec_jiffies,
 697         },
 698         {
 699                 .procname       = "nf_conntrack_sctp_timeout_cookie_echoed",
 700                 .maxlen         = sizeof(unsigned int),
 701                 .mode           = 0644,
 702                 .proc_handler   = proc_dointvec_jiffies,
 703         },
 704         {
 705                 .procname       = "nf_conntrack_sctp_timeout_established",
 706                 .maxlen         = sizeof(unsigned int),
 707                 .mode           = 0644,
 708                 .proc_handler   = proc_dointvec_jiffies,
 709         },
 710         {
 711                 .procname       = "nf_conntrack_sctp_timeout_shutdown_sent",
 712                 .maxlen         = sizeof(unsigned int),
 713                 .mode           = 0644,
 714                 .proc_handler   = proc_dointvec_jiffies,
 715         },
 716         {
 717                 .procname       = "nf_conntrack_sctp_timeout_shutdown_recd",
 718                 .maxlen         = sizeof(unsigned int),
 719                 .mode           = 0644,
 720                 .proc_handler   = proc_dointvec_jiffies,
 721         },
 722         {
 723                 .procname       = "nf_conntrack_sctp_timeout_shutdown_ack_sent",
 724                 .maxlen         = sizeof(unsigned int),
 725                 .mode           = 0644,
 726                 .proc_handler   = proc_dointvec_jiffies,
 727         },
 728         {
 729                 .procname       = "nf_conntrack_sctp_timeout_heartbeat_sent",
 730                 .maxlen         = sizeof(unsigned int),
 731                 .mode           = 0644,
 732                 .proc_handler   = proc_dointvec_jiffies,
 733         },
 734         {
 735                 .procname       = "nf_conntrack_sctp_timeout_heartbeat_acked",
 736                 .maxlen         = sizeof(unsigned int),
 737                 .mode           = 0644,
 738                 .proc_handler   = proc_dointvec_jiffies,
 739         },
 740         { }
 741 };
 742 #endif
 743
 744 static int sctp_kmemdup_sysctl_table(struct nf_proto_net *pn,
 745                                      struct nf_sctp_net *sn)
 746 {
 747 #ifdef CONFIG_SYSCTL
 748         if (pn->ctl_table)
 749                 return 0;
 750
 751         pn->ctl_table = kmemdup(sctp_sysctl_table,
 752                                 sizeof(sctp_sysctl_table),
 753                                 GFP_KERNEL);
 754         if (!pn->ctl_table)
 755                 return -ENOMEM;
 756
 757         pn->ctl_table[0].data = &sn->timeouts[SCTP_CONNTRACK_CLOSED];
 758         pn->ctl_table[1].data = &sn->timeouts[SCTP_CONNTRACK_COOKIE_WAIT];
 759         pn->ctl_table[2].data = &sn->timeouts[SCTP_CONNTRACK_COOKIE_ECHOED];
 760         pn->ctl_table[3].data = &sn->timeouts[SCTP_CONNTRACK_ESTABLISHED];
 761         pn->ctl_table[4].data = &sn->timeouts[SCTP_CONNTRACK_SHUTDOWN_SENT];
 762         pn->ctl_table[5].data = &sn->timeouts[SCTP_CONNTRACK_SHUTDOWN_RECD];
 763         pn->ctl_table[6].data = &sn->timeouts[SCTP_CONNTRACK_SHUTDOWN_ACK_SENT];
 764         pn->ctl_table[7].data = &sn->timeouts[SCTP_CONNTRACK_HEARTBEAT_SENT];
 765         pn->ctl_table[8].data = &sn->timeouts[SCTP_CONNTRACK_HEARTBEAT_ACKED];
 766 #endif
 767         return 0;
 768 }
 769
 770 static int sctp_init_net(struct net *net, u_int16_t proto)
 771 {
 772         struct nf_sctp_net *sn = sctp_pernet(net);
 773         struct nf_proto_net *pn = &sn->pn;
 774
 775         if (!pn->users) {
 776                 int i;
 777
 778                 for (i = 0; i < SCTP_CONNTRACK_MAX; i++)
 779                         sn->timeouts[i] = sctp_timeouts[i];
 780         }
 781
 782         return sctp_kmemdup_sysctl_table(pn, sn);
 783 }
 784
 785 static struct nf_proto_net *sctp_get_net_proto(struct net *net)
 786 {
 787         return &net->ct.nf_ct_proto.sctp.pn;
 788 }
 789
 790 struct nf_conntrack_l4proto nf_conntrack_l4proto_sctp4 __read_mostly = {
 791         .l3proto                = PF_INET,
 792         .l4proto                = IPPROTO_SCTP,
 793         .pkt_to_tuple           = sctp_pkt_to_tuple,
 794         .invert_tuple           = sctp_invert_tuple,
 795 #ifdef CONFIG_NF_CONNTRACK_PROCFS
 796         .print_conntrack        = sctp_print_conntrack,
 797 #endif
 798         .packet                 = sctp_packet,
 799         .get_timeouts           = sctp_get_timeouts,
 800         .new                    = sctp_new,
 801         .error                  = sctp_error,
 802         .can_early_drop         = sctp_can_early_drop,
 803         .me                     = THIS_MODULE,
 804 #if IS_ENABLED(CONFIG_NF_CT_NETLINK)
 805         .to_nlattr              = sctp_to_nlattr,
 806         .nlattr_size            = sctp_nlattr_size,
 807         .from_nlattr            = nlattr_to_sctp,
 808         .tuple_to_nlattr        = nf_ct_port_tuple_to_nlattr,
 809         .nlattr_tuple_size      = nf_ct_port_nlattr_tuple_size,
 810         .nlattr_to_tuple        = nf_ct_port_nlattr_to_tuple,
 811         .nla_policy             = nf_ct_port_nla_policy,
 812 #endif
 813 #if IS_ENABLED(CONFIG_NF_CT_NETLINK_TIMEOUT)
 814         .ctnl_timeout           = {
 815                 .nlattr_to_obj  = sctp_timeout_nlattr_to_obj,
 816                 .obj_to_nlattr  = sctp_timeout_obj_to_nlattr,
 817                 .nlattr_max     = CTA_TIMEOUT_SCTP_MAX,
 818                 .obj_size       = sizeof(unsigned int) * SCTP_CONNTRACK_MAX,
 819                 .nla_policy     = sctp_timeout_nla_policy,
 820         },
 821 #endif /* CONFIG_NF_CT_NETLINK_TIMEOUT */
 822         .init_net               = sctp_init_net,
 823         .get_net_proto          = sctp_get_net_proto,
 824 };
 825 EXPORT_SYMBOL_GPL(nf_conntrack_l4proto_sctp4);
 826
 827 struct nf_conntrack_l4proto nf_conntrack_l4proto_sctp6 __read_mostly = {
 828         .l3proto                = PF_INET6,
 829         .l4proto                = IPPROTO_SCTP,
 830         .pkt_to_tuple           = sctp_pkt_to_tuple,
 831         .invert_tuple           = sctp_invert_tuple,
 832 #ifdef CONFIG_NF_CONNTRACK_PROCFS
 833         .print_conntrack        = sctp_print_conntrack,
 834 #endif
 835         .packet                 = sctp_packet,
 836         .get_timeouts           = sctp_get_timeouts,
 837         .new                    = sctp_new,
 838         .error                  = sctp_error,
 839         .can_early_drop         = sctp_can_early_drop,
 840         .me                     = THIS_MODULE,
 841 #if IS_ENABLED(CONFIG_NF_CT_NETLINK)
 842         .to_nlattr              = sctp_to_nlattr,
 843         .nlattr_size            = sctp_nlattr_size,
 844         .from_nlattr            = nlattr_to_sctp,
 845         .tuple_to_nlattr        = nf_ct_port_tuple_to_nlattr,
 846         .nlattr_tuple_size      = nf_ct_port_nlattr_tuple_size,
 847         .nlattr_to_tuple        = nf_ct_port_nlattr_to_tuple,
 848         .nla_policy             = nf_ct_port_nla_policy,
 849 #if IS_ENABLED(CONFIG_NF_CT_NETLINK_TIMEOUT)
 850         .ctnl_timeout           = {
 851                 .nlattr_to_obj  = sctp_timeout_nlattr_to_obj,
 852                 .obj_to_nlattr  = sctp_timeout_obj_to_nlattr,
 853                 .nlattr_max     = CTA_TIMEOUT_SCTP_MAX,
 854                 .obj_size       = sizeof(unsigned int) * SCTP_CONNTRACK_MAX,
 855                 .nla_policy     = sctp_timeout_nla_policy,
 856         },
 857 #endif /* CONFIG_NF_CT_NETLINK_TIMEOUT */
 858 #endif
 859         .init_net               = sctp_init_net,
 860         .get_net_proto          = sctp_get_net_proto,
 861 };
 862 EXPORT_SYMBOL_GPL(nf_conntrack_l4proto_sctp6);