sys/contrib/pf/net/pf_mtag.h

   1 /*      $FreeBSD$       */
   2 /*
   3  * Copyright (c) 2001 Daniel Hartmeier
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  *
  10  *    - Redistributions of source code must retain the above copyright
  11  *      notice, this list of conditions and the following disclaimer.
  12  *    - Redistributions in binary form must reproduce the above
  13  *      copyright notice, this list of conditions and the following
  14  *      disclaimer in the documentation and/or other materials provided
  15  *      with the distribution.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  18  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  19  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  20  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  21  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  22  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
  23  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  24  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  25  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  26  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  27  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  28  * POSSIBILITY OF SUCH DAMAGE.
  29  *
  30  */
  31
  32 #ifndef _NET_PF_MTAG_H_
  33 #define _NET_PF_MTAG_H_
  34
  35 #ifdef _KERNEL
  36
  37 #define PF_TAG_GENERATED                0x01
  38 #define PF_TAG_FRAGCACHE                0x02
  39 #define PF_TAG_TRANSLATE_LOCALHOST      0x04
  40
  41 struct pf_mtag {
  42         void            *hdr;           /* saved hdr pos in mbuf, for ECN */
  43         u_int            rtableid;      /* alternate routing table id */
  44         u_int32_t        qid;           /* queue id */
  45         u_int16_t        tag;           /* tag id */
  46         u_int8_t         flags;
  47         u_int8_t         routed;
  48         sa_family_t      af;            /* for ECN */
  49 };
  50
  51 static __inline struct pf_mtag *pf_find_mtag(struct mbuf *);
  52 static __inline struct pf_mtag *pf_get_mtag(struct mbuf *);
  53
  54 static __inline struct pf_mtag *
  55 pf_find_mtag(struct mbuf *m)
  56 {
  57         struct m_tag    *mtag;
  58
  59         if ((mtag = m_tag_find(m, PACKET_TAG_PF, NULL)) == NULL)
  60                 return (NULL);
  61
  62         return ((struct pf_mtag *)(mtag + 1));
  63 }
  64
  65 static __inline struct pf_mtag *
  66 pf_get_mtag(struct mbuf *m)
  67 {
  68         struct m_tag    *mtag;
  69
  70         if ((mtag = m_tag_find(m, PACKET_TAG_PF, NULL)) == NULL) {
  71                 mtag = m_tag_get(PACKET_TAG_PF, sizeof(struct pf_mtag),
  72                     M_NOWAIT);
  73                 if (mtag == NULL)
  74                         return (NULL);
  75                 bzero(mtag + 1, sizeof(struct pf_mtag));
  76                 m_tag_prepend(m, mtag);
  77         }
  78
  79         return ((struct pf_mtag *)(mtag + 1));
  80 }
  81 #endif /* _KERNEL */
  82 #endif /* _NET_PF_MTAG_H_ */