include/linux/netfilter_ipv4/ip_conntrack.h

   1 #ifndef _IP_CONNTRACK_H
   2 #define _IP_CONNTRACK_H
   3
   4 #include <linux/netfilter/nf_conntrack_common.h>
   5
   6 #ifdef __KERNEL__
   7 #include <linux/config.h>
   8 #include <linux/netfilter_ipv4/ip_conntrack_tuple.h>
   9 #include <linux/bitops.h>
  10 #include <linux/compiler.h>
  11 #include <asm/atomic.h>
  12
  13 #include <linux/netfilter_ipv4/ip_conntrack_tcp.h>
  14 #include <linux/netfilter_ipv4/ip_conntrack_icmp.h>
  15 #include <linux/netfilter_ipv4/ip_conntrack_proto_gre.h>
  16 #include <linux/netfilter_ipv4/ip_conntrack_sctp.h>
  17
  18 /* per conntrack: protocol private data */
  19 union ip_conntrack_proto {
  20         /* insert conntrack proto private data here */
  21         struct ip_ct_gre gre;
  22         struct ip_ct_sctp sctp;
  23         struct ip_ct_tcp tcp;
  24         struct ip_ct_icmp icmp;
  25 };
  26
  27 union ip_conntrack_expect_proto {
  28         /* insert expect proto private data here */
  29 };
  30
  31 /* Add protocol helper include file here */
  32 #include <linux/netfilter_ipv4/ip_conntrack_pptp.h>
  33 #include <linux/netfilter_ipv4/ip_conntrack_amanda.h>
  34 #include <linux/netfilter_ipv4/ip_conntrack_ftp.h>
  35 #include <linux/netfilter_ipv4/ip_conntrack_irc.h>
  36
  37 /* per conntrack: application helper private data */
  38 union ip_conntrack_help {
  39         /* insert conntrack helper private data (master) here */
  40         struct ip_ct_pptp_master ct_pptp_info;
  41         struct ip_ct_ftp_master ct_ftp_info;
  42         struct ip_ct_irc_master ct_irc_info;
  43 };
  44
  45 #ifdef CONFIG_IP_NF_NAT_NEEDED
  46 #include <linux/netfilter_ipv4/ip_nat.h>
  47 #include <linux/netfilter_ipv4/ip_nat_pptp.h>
  48
  49 /* per conntrack: nat application helper private data */
  50 union ip_conntrack_nat_help {
  51         /* insert nat helper private data here */
  52         struct ip_nat_pptp nat_pptp_info;
  53 };
  54 #endif
  55
  56 #include <linux/types.h>
  57 #include <linux/skbuff.h>
  58
  59 #ifdef CONFIG_NETFILTER_DEBUG
  60 #define IP_NF_ASSERT(x)                                                 \
  61 do {                                                                    \
  62         if (!(x))                                                       \
  63                 /* Wooah!  I'm tripping my conntrack in a frenzy of     \
  64                    netplay... */                                        \
  65                 printk("NF_IP_ASSERT: %s:%i(%s)\n",                     \
  66                        __FILE__, __LINE__, __FUNCTION__);               \
  67 } while(0)
  68 #else
  69 #define IP_NF_ASSERT(x)
  70 #endif
  71
  72 struct ip_conntrack_helper;
  73
  74 struct ip_conntrack
  75 {
  76         /* Usage count in here is 1 for hash table/destruct timer, 1 per skb,
  77            plus 1 for any connection(s) we are `master' for */
  78         struct nf_conntrack ct_general;
  79
  80         /* Have we seen traffic both ways yet? (bitset) */
  81         unsigned long status;
  82
  83         /* Timer function; drops refcnt when it goes off. */
  84         struct timer_list timeout;
  85
  86 #ifdef CONFIG_IP_NF_CT_ACCT
  87         /* Accounting Information (same cache line as other written members) */
  88         struct ip_conntrack_counter counters[IP_CT_DIR_MAX];
  89 #endif
  90         /* If we were expected by an expectation, this will be it */
  91         struct ip_conntrack *master;
  92
  93         /* Current number of expected connections */
  94         unsigned int expecting;
  95
  96         /* Unique ID that identifies this conntrack*/
  97         unsigned int id;
  98
  99         /* Helper, if any. */
 100         struct ip_conntrack_helper *helper;
 101
 102         /* Storage reserved for other modules: */
 103         union ip_conntrack_proto proto;
 104
 105         union ip_conntrack_help help;
 106
 107 #ifdef CONFIG_IP_NF_NAT_NEEDED
 108         struct {
 109                 struct ip_nat_info info;
 110                 union ip_conntrack_nat_help help;
 111 #if defined(CONFIG_IP_NF_TARGET_MASQUERADE) || \
 112         defined(CONFIG_IP_NF_TARGET_MASQUERADE_MODULE)
 113                 int masq_index;
 114 #endif
 115         } nat;
 116 #endif /* CONFIG_IP_NF_NAT_NEEDED */
 117
 118 #if defined(CONFIG_IP_NF_CONNTRACK_MARK)
 119         u_int32_t mark;
 120 #endif
 121
 122         /* Traversed often, so hopefully in different cacheline to top */
 123         /* These are my tuples; original and reply */
 124         struct ip_conntrack_tuple_hash tuplehash[IP_CT_DIR_MAX];
 125 };
 126
 127 struct ip_conntrack_expect
 128 {
 129         /* Internal linked list (global expectation list) */
 130         struct list_head list;
 131
 132         /* We expect this tuple, with the following mask */
 133         struct ip_conntrack_tuple tuple, mask;
 134
 135         /* Function to call after setup and insertion */
 136         void (*expectfn)(struct ip_conntrack *new,
 137                          struct ip_conntrack_expect *this);
 138
 139         /* The conntrack of the master connection */
 140         struct ip_conntrack *master;
 141
 142         /* Timer function; deletes the expectation. */
 143         struct timer_list timeout;
 144
 145         /* Usage count. */
 146         atomic_t use;
 147
 148         /* Unique ID */
 149         unsigned int id;
 150
 151         /* Flags */
 152         unsigned int flags;
 153
 154 #ifdef CONFIG_IP_NF_NAT_NEEDED
 155         /* This is the original per-proto part, used to map the
 156          * expected connection the way the recipient expects. */
 157         union ip_conntrack_manip_proto saved_proto;
 158         /* Direction relative to the master connection. */
 159         enum ip_conntrack_dir dir;
 160 #endif
 161 };
 162
 163 #define IP_CT_EXPECT_PERMANENT  0x1
 164
 165 static inline struct ip_conntrack *
 166 tuplehash_to_ctrack(const struct ip_conntrack_tuple_hash *hash)
 167 {
 168         return container_of(hash, struct ip_conntrack,
 169                             tuplehash[hash->tuple.dst.dir]);
 170 }
 171
 172 /* get master conntrack via master expectation */
 173 #define master_ct(conntr) (conntr->master)
 174
 175 /* Alter reply tuple (maybe alter helper). */
 176 extern void
 177 ip_conntrack_alter_reply(struct ip_conntrack *conntrack,
 178                          const struct ip_conntrack_tuple *newreply);
 179
 180 /* Is this tuple taken? (ignoring any belonging to the given
 181    conntrack). */
 182 extern int
 183 ip_conntrack_tuple_taken(const struct ip_conntrack_tuple *tuple,
 184                          const struct ip_conntrack *ignored_conntrack);
 185
 186 /* Return conntrack_info and tuple hash for given skb. */
 187 static inline struct ip_conntrack *
 188 ip_conntrack_get(const struct sk_buff *skb, enum ip_conntrack_info *ctinfo)
 189 {
 190         *ctinfo = skb->nfctinfo;
 191         return (struct ip_conntrack *)skb->nfct;
 192 }
 193
 194 /* decrement reference count on a conntrack */
 195 static inline void
 196 ip_conntrack_put(struct ip_conntrack *ct)
 197 {
 198         IP_NF_ASSERT(ct);
 199         nf_conntrack_put(&ct->ct_general);
 200 }
 201
 202 extern int invert_tuplepr(struct ip_conntrack_tuple *inverse,
 203                           const struct ip_conntrack_tuple *orig);
 204
 205 extern void __ip_ct_refresh_acct(struct ip_conntrack *ct,
 206                                  enum ip_conntrack_info ctinfo,
 207                                  const struct sk_buff *skb,
 208                                  unsigned long extra_jiffies,
 209                                  int do_acct);
 210
 211 /* Refresh conntrack for this many jiffies and do accounting */
 212 static inline void ip_ct_refresh_acct(struct ip_conntrack *ct,
 213                                       enum ip_conntrack_info ctinfo,
 214                                       const struct sk_buff *skb,
 215                                       unsigned long extra_jiffies)
 216 {
 217         __ip_ct_refresh_acct(ct, ctinfo, skb, extra_jiffies, 1);
 218 }
 219
 220 /* Refresh conntrack for this many jiffies */
 221 static inline void ip_ct_refresh(struct ip_conntrack *ct,
 222                                  const struct sk_buff *skb,
 223                                  unsigned long extra_jiffies)
 224 {
 225         __ip_ct_refresh_acct(ct, 0, skb, extra_jiffies, 0);
 226 }
 227
 228 /* These are for NAT.  Icky. */
 229 /* Update TCP window tracking data when NAT mangles the packet */
 230 extern void ip_conntrack_tcp_update(struct sk_buff *skb,
 231                                     struct ip_conntrack *conntrack,
 232                                     enum ip_conntrack_dir dir);
 233
 234 /* Call me when a conntrack is destroyed. */
 235 extern void (*ip_conntrack_destroyed)(struct ip_conntrack *conntrack);
 236
 237 /* Fake conntrack entry for untracked connections */
 238 extern struct ip_conntrack ip_conntrack_untracked;
 239
 240 /* Returns new sk_buff, or NULL */
 241 struct sk_buff *
 242 ip_ct_gather_frags(struct sk_buff *skb, u_int32_t user);
 243
 244 /* Iterate over all conntracks: if iter returns true, it's deleted. */
 245 extern void
 246 ip_ct_iterate_cleanup(int (*iter)(struct ip_conntrack *i, void *data),
 247                       void *data);
 248
 249 extern struct ip_conntrack_helper *
 250 __ip_conntrack_helper_find_byname(const char *);
 251 extern struct ip_conntrack_helper *
 252 ip_conntrack_helper_find_get(const struct ip_conntrack_tuple *tuple);
 253 extern void ip_conntrack_helper_put(struct ip_conntrack_helper *helper);
 254
 255 extern struct ip_conntrack_protocol *
 256 __ip_conntrack_proto_find(u_int8_t protocol);
 257 extern struct ip_conntrack_protocol *
 258 ip_conntrack_proto_find_get(u_int8_t protocol);
 259 extern void ip_conntrack_proto_put(struct ip_conntrack_protocol *proto);
 260
 261 extern void ip_ct_remove_expectations(struct ip_conntrack *ct);
 262
 263 extern struct ip_conntrack *ip_conntrack_alloc(struct ip_conntrack_tuple *,
 264                                                struct ip_conntrack_tuple *);
 265
 266 extern void ip_conntrack_free(struct ip_conntrack *ct);
 267
 268 extern void ip_conntrack_hash_insert(struct ip_conntrack *ct);
 269
 270 extern struct ip_conntrack_expect *
 271 __ip_conntrack_expect_find(const struct ip_conntrack_tuple *tuple);
 272
 273 extern struct ip_conntrack_expect *
 274 ip_conntrack_expect_find(const struct ip_conntrack_tuple *tuple);
 275
 276 extern struct ip_conntrack_tuple_hash *
 277 __ip_conntrack_find(const struct ip_conntrack_tuple *tuple,
 278                     const struct ip_conntrack *ignored_conntrack);
 279
 280 extern void ip_conntrack_flush(void);
 281
 282 /* It's confirmed if it is, or has been in the hash table. */
 283 static inline int is_confirmed(struct ip_conntrack *ct)
 284 {
 285         return test_bit(IPS_CONFIRMED_BIT, &ct->status);
 286 }
 287
 288 static inline int is_dying(struct ip_conntrack *ct)
 289 {
 290         return test_bit(IPS_DYING_BIT, &ct->status);
 291 }
 292
 293 extern unsigned int ip_conntrack_htable_size;
 294
 295 #define CONNTRACK_STAT_INC(count) (__get_cpu_var(ip_conntrack_stat).count++)
 296
 297 #ifdef CONFIG_IP_NF_CONNTRACK_EVENTS
 298 #include <linux/notifier.h>
 299 #include <linux/interrupt.h>
 300
 301 struct ip_conntrack_ecache {
 302         struct ip_conntrack *ct;
 303         unsigned int events;
 304 };
 305 DECLARE_PER_CPU(struct ip_conntrack_ecache, ip_conntrack_ecache);
 306
 307 #define CONNTRACK_ECACHE(x)     (__get_cpu_var(ip_conntrack_ecache).x)
 308
 309 extern struct notifier_block *ip_conntrack_chain;
 310 extern struct notifier_block *ip_conntrack_expect_chain;
 311
 312 static inline int ip_conntrack_register_notifier(struct notifier_block *nb)
 313 {
 314         return notifier_chain_register(&ip_conntrack_chain, nb);
 315 }
 316
 317 static inline int ip_conntrack_unregister_notifier(struct notifier_block *nb)
 318 {
 319         return notifier_chain_unregister(&ip_conntrack_chain, nb);
 320 }
 321
 322 static inline int
 323 ip_conntrack_expect_register_notifier(struct notifier_block *nb)
 324 {
 325         return notifier_chain_register(&ip_conntrack_expect_chain, nb);
 326 }
 327
 328 static inline int
 329 ip_conntrack_expect_unregister_notifier(struct notifier_block *nb)
 330 {
 331         return notifier_chain_unregister(&ip_conntrack_expect_chain, nb);
 332 }
 333
 334 extern void ip_ct_deliver_cached_events(const struct ip_conntrack *ct);
 335 extern void __ip_ct_event_cache_init(struct ip_conntrack *ct);
 336
 337 static inline void
 338 ip_conntrack_event_cache(enum ip_conntrack_events event,
 339                          const struct sk_buff *skb)
 340 {
 341         struct ip_conntrack *ct = (struct ip_conntrack *)skb->nfct;
 342         struct ip_conntrack_ecache *ecache;
 343
 344         local_bh_disable();
 345         ecache = &__get_cpu_var(ip_conntrack_ecache);
 346         if (ct != ecache->ct)
 347                 __ip_ct_event_cache_init(ct);
 348         ecache->events |= event;
 349         local_bh_enable();
 350 }
 351
 352 static inline void ip_conntrack_event(enum ip_conntrack_events event,
 353                                       struct ip_conntrack *ct)
 354 {
 355         if (is_confirmed(ct) && !is_dying(ct))
 356                 notifier_call_chain(&ip_conntrack_chain, event, ct);
 357 }
 358
 359 static inline void
 360 ip_conntrack_expect_event(enum ip_conntrack_expect_events event,
 361                           struct ip_conntrack_expect *exp)
 362 {
 363         notifier_call_chain(&ip_conntrack_expect_chain, event, exp);
 364 }
 365 #else /* CONFIG_IP_NF_CONNTRACK_EVENTS */
 366 static inline void ip_conntrack_event_cache(enum ip_conntrack_events event,
 367                                             const struct sk_buff *skb) {}
 368 static inline void ip_conntrack_event(enum ip_conntrack_events event,
 369                                       struct ip_conntrack *ct) {}
 370 static inline void ip_ct_deliver_cached_events(const struct ip_conntrack *ct) {}
 371 static inline void
 372 ip_conntrack_expect_event(enum ip_conntrack_expect_events event,
 373                           struct ip_conntrack_expect *exp) {}
 374 #endif /* CONFIG_IP_NF_CONNTRACK_EVENTS */
 375
 376 #ifdef CONFIG_IP_NF_NAT_NEEDED
 377 static inline int ip_nat_initialized(struct ip_conntrack *conntrack,
 378                                      enum ip_nat_manip_type manip)
 379 {
 380         if (manip == IP_NAT_MANIP_SRC)
 381                 return test_bit(IPS_SRC_NAT_DONE_BIT, &conntrack->status);
 382         return test_bit(IPS_DST_NAT_DONE_BIT, &conntrack->status);
 383 }
 384 #endif /* CONFIG_IP_NF_NAT_NEEDED */
 385
 386 #endif /* __KERNEL__ */
 387 #endif /* _IP_CONNTRACK_H */