sys/netinet/libalias/alias_local.h

   1 /*-
   2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
   3  *
   4  * Copyright (c) 2001 Charles Mott <cm@linktel.net>
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  26  * SUCH DAMAGE.
  27  *
  28  * $FreeBSD$
  29  */
  30
  31 /*
  32  * Alias_local.h contains the function prototypes for alias.c,
  33  * alias_db.c, alias_util.c and alias_ftp.c, alias_irc.c (as well
  34  * as any future add-ons).  It also includes macros, globals and
  35  * struct definitions shared by more than one alias*.c file.
  36  *
  37  * This include file is intended to be used only within the aliasing
  38  * software.  Outside world interfaces are defined in alias.h
  39  *
  40  * This software is placed into the public domain with no restrictions
  41  * on its distribution.
  42  *
  43  * Initial version:  August, 1996  (cjm)
  44  *
  45  * <updated several times by original author and Eivind Eklund>
  46  */
  47
  48 #ifndef _ALIAS_LOCAL_H_
  49 #define _ALIAS_LOCAL_H_
  50
  51 #include <sys/types.h>
  52 #include <sys/sysctl.h>
  53
  54 #ifdef _KERNEL
  55 #include <sys/malloc.h>
  56 #include <sys/param.h>
  57 #include <sys/lock.h>
  58 #include <sys/mutex.h>
  59
  60 /* XXX: LibAliasSetTarget() uses this constant. */
  61 #define INADDR_NONE     0xffffffff
  62
  63 #include <netinet/libalias/alias_sctp.h>
  64 #else
  65 #include "alias_sctp.h"
  66 #endif
  67
  68 /* Sizes of input and output link tables */
  69 #define LINK_TABLE_OUT_SIZE        4001
  70 #define LINK_TABLE_IN_SIZE         4001
  71
  72 #define GET_ALIAS_PORT          -1
  73 #define GET_ALIAS_ID            GET_ALIAS_PORT
  74
  75 #ifdef _KERNEL
  76 #define INET_NTOA_BUF(buf) (buf)
  77 #else
  78 #define INET_NTOA_BUF(buf) (buf), sizeof(buf)
  79 #endif
  80
  81 struct proxy_entry;
  82
  83 struct libalias {
  84         LIST_ENTRY(libalias) instancelist;
  85
  86         int             packetAliasMode;        /* Mode flags                      */
  87         /* - documented in alias.h  */
  88
  89         struct in_addr  aliasAddress;   /* Address written onto source     */
  90         /* field of IP packet.           */
  91
  92         struct in_addr  targetAddress;  /* IP address incoming packets     */
  93         /* are sent to if no aliasing    */
  94         /* link already exists           */
  95
  96         struct in_addr  nullAddress;    /* Used as a dummy parameter for   */
  97         /* some function calls           */
  98
  99                         LIST_HEAD     (, alias_link) linkTableOut[LINK_TABLE_OUT_SIZE];
 100         /* Lookup table of pointers to     */
 101         /* chains of link records. Each  */
 102
 103                         LIST_HEAD     (, alias_link) linkTableIn[LINK_TABLE_IN_SIZE];
 104         /* link record is doubly indexed */
 105         /* into input and output lookup  */
 106         /* tables.                       */
 107
 108         /* Link statistics                 */
 109         int             icmpLinkCount;
 110         int             udpLinkCount;
 111         int             tcpLinkCount;
 112         int             pptpLinkCount;
 113         int             protoLinkCount;
 114         int             fragmentIdLinkCount;
 115         int             fragmentPtrLinkCount;
 116         int             sockCount;
 117
 118         int             cleanupIndex;   /* Index to chain of link table    */
 119         /* being inspected for old links   */
 120
 121         int             timeStamp;      /* System time in seconds for      */
 122         /* current packet                  */
 123
 124         int             lastCleanupTime;        /* Last time
 125                                                  * IncrementalCleanup()  */
 126         /* was called                      */
 127
 128         int             deleteAllLinks; /* If equal to zero, DeleteLink()  */
 129         /* will not remove permanent links */
 130
 131         /* log descriptor        */
 132 #ifdef  _KERNEL
 133         char           *logDesc;
 134 #else
 135         FILE           *logDesc;
 136 #endif
 137         /* statistics monitoring */
 138
 139         int             newDefaultLink; /* Indicates if a new aliasing     */
 140         /* link has been created after a   */
 141         /* call to PacketAliasIn/Out().    */
 142
 143 #ifndef NO_FW_PUNCH
 144         int             fireWallFD;     /* File descriptor to be able to   */
 145         /* control firewall.  Opened by    */
 146         /* PacketAliasSetMode on first     */
 147         /* setting the PKT_ALIAS_PUNCH_FW  */
 148         /* flag.                           */
 149         int             fireWallBaseNum;        /* The first firewall entry
 150                                                  * free for our use */
 151         int             fireWallNumNums;        /* How many entries can we
 152                                                  * use? */
 153         int             fireWallActiveNum;      /* Which entry did we last
 154                                                  * use? */
 155         char           *fireWallField;  /* bool array for entries */
 156 #endif
 157
 158         unsigned int    skinnyPort;     /* TCP port used by the Skinny     */
 159         /* protocol.                       */
 160
 161         struct proxy_entry *proxyList;
 162
 163         struct in_addr  true_addr;      /* in network byte order. */
 164         u_short         true_port;      /* in host byte order. */
 165
 166         /* Port ranges for aliasing. */
 167         u_short         aliasPortLower;
 168         u_short         aliasPortLength;
 169
 170         /*
 171          * sctp code support
 172          */
 173
 174         /* counts associations that have progressed to UP and not yet removed */
 175         int             sctpLinkCount;
 176 #ifdef  _KERNEL
 177         /* timing queue for keeping track of association timeouts */
 178         struct sctp_nat_timer sctpNatTimer;
 179
 180         /* size of hash table used in this instance */
 181         u_int sctpNatTableSize;
 182
 183 /*
 184  * local look up table sorted by l_vtag/l_port
 185  */
 186         LIST_HEAD(sctpNatTableL, sctp_nat_assoc) *sctpTableLocal;
 187 /*
 188  * global look up table sorted by g_vtag/g_port
 189  */
 190         LIST_HEAD(sctpNatTableG, sctp_nat_assoc) *sctpTableGlobal;
 191
 192         /*
 193          * avoid races in libalias: every public function has to use it.
 194          */
 195         struct mtx mutex;
 196 #endif
 197 };
 198
 199 /* Macros */
 200
 201 #ifdef _KERNEL
 202 #define LIBALIAS_LOCK_INIT(l) \
 203         mtx_init(&l->mutex, "per-instance libalias mutex", NULL, MTX_DEF)
 204 #define LIBALIAS_LOCK_ASSERT(l) mtx_assert(&l->mutex, MA_OWNED)
 205 #define LIBALIAS_LOCK(l) mtx_lock(&l->mutex)
 206 #define LIBALIAS_UNLOCK(l) mtx_unlock(&l->mutex)
 207 #define LIBALIAS_LOCK_DESTROY(l)        mtx_destroy(&l->mutex)
 208 #else
 209 #define LIBALIAS_LOCK_INIT(l)
 210 #define LIBALIAS_LOCK_ASSERT(l)
 211 #define LIBALIAS_LOCK(l)
 212 #define LIBALIAS_UNLOCK(l)
 213 #define LIBALIAS_LOCK_DESTROY(l)
 214 #endif
 215
 216 /*
 217  * The following macro is used to update an
 218  * internet checksum.  "delta" is a 32-bit
 219  * accumulation of all the changes to the
 220  * checksum (adding in new 16-bit words and
 221  * subtracting out old words), and "cksum"
 222  * is the checksum value to be updated.
 223  */
 224 #define ADJUST_CHECKSUM(acc, cksum) \
 225         do { \
 226                 acc += cksum; \
 227                 if (acc < 0) { \
 228                         acc = -acc; \
 229                         acc = (acc >> 16) + (acc & 0xffff); \
 230                         acc += acc >> 16; \
 231                         cksum = (u_short) ~acc; \
 232                 } else { \
 233                         acc = (acc >> 16) + (acc & 0xffff); \
 234                         acc += acc >> 16; \
 235                         cksum = (u_short) acc; \
 236                 } \
 237         } while (0)
 238
 239
 240 /* Prototypes */
 241
 242 /*
 243  * SctpFunction prototypes
 244  *
 245  */
 246 void AliasSctpInit(struct libalias *la);
 247 void AliasSctpTerm(struct libalias *la);
 248 int SctpAlias(struct libalias *la, struct ip *ip, int direction);
 249
 250 /*
 251  * We do not calculate TCP checksums when libalias is a kernel
 252  * module, since it has no idea about checksum offloading.
 253  * If TCP data has changed, then we just set checksum to zero,
 254  * and caller must recalculate it himself.
 255  * In case if libalias will edit UDP data, the same approach
 256  * should be used.
 257  */
 258 #ifndef _KERNEL
 259 u_short         IpChecksum(struct ip *_pip);
 260 u_short         TcpChecksum(struct ip *_pip);
 261 #endif
 262 void
 263 DifferentialChecksum(u_short * _cksum, void * _new, void * _old, int _n);
 264
 265 /* Internal data access */
 266 struct alias_link *
 267 AddLink(struct libalias *la, struct in_addr src_addr, struct in_addr dst_addr,
 268     struct in_addr alias_addr, u_short src_port, u_short dst_port,
 269     int alias_param, int link_type);
 270 struct alias_link *
 271 FindIcmpIn(struct libalias *la, struct in_addr _dst_addr, struct in_addr _alias_addr,
 272     u_short _id_alias, int _create);
 273 struct alias_link *
 274 FindIcmpOut(struct libalias *la, struct in_addr _src_addr, struct in_addr _dst_addr,
 275     u_short _id, int _create);
 276 struct alias_link *
 277 FindFragmentIn1(struct libalias *la, struct in_addr _dst_addr, struct in_addr _alias_addr,
 278     u_short _ip_id);
 279 struct alias_link *
 280 FindFragmentIn2(struct libalias *la, struct in_addr _dst_addr, struct in_addr _alias_addr,
 281     u_short _ip_id);
 282 struct alias_link *
 283                 AddFragmentPtrLink(struct libalias *la, struct in_addr _dst_addr, u_short _ip_id);
 284 struct alias_link *
 285                 FindFragmentPtr(struct libalias *la, struct in_addr _dst_addr, u_short _ip_id);
 286 struct alias_link *
 287 FindProtoIn(struct libalias *la, struct in_addr _dst_addr, struct in_addr _alias_addr,
 288     u_char _proto);
 289 struct alias_link *
 290 FindProtoOut(struct libalias *la, struct in_addr _src_addr, struct in_addr _dst_addr,
 291     u_char _proto);
 292 struct alias_link *
 293 FindUdpTcpIn(struct libalias *la, struct in_addr _dst_addr, struct in_addr _alias_addr,
 294     u_short _dst_port, u_short _alias_port, u_char _proto, int _create);
 295 struct alias_link *
 296 FindUdpTcpOut(struct libalias *la, struct in_addr _src_addr, struct in_addr _dst_addr,
 297     u_short _src_port, u_short _dst_port, u_char _proto, int _create);
 298 struct alias_link *
 299 AddPptp(struct libalias *la, struct in_addr _src_addr, struct in_addr _dst_addr,
 300     struct in_addr _alias_addr, u_int16_t _src_call_id);
 301 struct alias_link *
 302 FindPptpOutByCallId(struct libalias *la, struct in_addr _src_addr,
 303     struct in_addr _dst_addr, u_int16_t _src_call_id);
 304 struct alias_link *
 305 FindPptpInByCallId(struct libalias *la, struct in_addr _dst_addr,
 306     struct in_addr _alias_addr, u_int16_t _dst_call_id);
 307 struct alias_link *
 308 FindPptpOutByPeerCallId(struct libalias *la, struct in_addr _src_addr,
 309     struct in_addr _dst_addr, u_int16_t _dst_call_id);
 310 struct alias_link *
 311 FindPptpInByPeerCallId(struct libalias *la, struct in_addr _dst_addr,
 312     struct in_addr _alias_addr, u_int16_t _alias_call_id);
 313 struct alias_link *
 314 FindRtspOut(struct libalias *la, struct in_addr _src_addr, struct in_addr _dst_addr,
 315     u_short _src_port, u_short _alias_port, u_char _proto);
 316 struct in_addr
 317                 FindOriginalAddress(struct libalias *la, struct in_addr _alias_addr);
 318 struct in_addr
 319                 FindAliasAddress(struct libalias *la, struct in_addr _original_addr);
 320 struct in_addr
 321 FindSctpRedirectAddress(struct libalias *la,  struct sctp_nat_msg *sm);
 322
 323 /* External data access/modification */
 324 int
 325 FindNewPortGroup(struct libalias *la, struct in_addr _dst_addr, struct in_addr _alias_addr,
 326     u_short _src_port, u_short _dst_port, u_short _port_count,
 327     u_char _proto, u_char _align);
 328 void            GetFragmentAddr(struct alias_link *_lnk, struct in_addr *_src_addr);
 329 void            SetFragmentAddr(struct alias_link *_lnk, struct in_addr _src_addr);
 330 void            GetFragmentPtr(struct alias_link *_lnk, char **_fptr);
 331 void            SetFragmentPtr(struct alias_link *_lnk, char *fptr);
 332 void            SetStateIn(struct alias_link *_lnk, int _state);
 333 void            SetStateOut(struct alias_link *_lnk, int _state);
 334 int             GetStateIn (struct alias_link *_lnk);
 335 int             GetStateOut(struct alias_link *_lnk);
 336 struct in_addr
 337                 GetOriginalAddress(struct alias_link *_lnk);
 338 struct in_addr
 339                 GetDestAddress(struct alias_link *_lnk);
 340 struct in_addr
 341                 GetAliasAddress(struct alias_link *_lnk);
 342 struct in_addr
 343                 GetDefaultAliasAddress(struct libalias *la);
 344 void            SetDefaultAliasAddress(struct libalias *la, struct in_addr _alias_addr);
 345 u_short         GetOriginalPort(struct alias_link *_lnk);
 346 u_short         GetAliasPort(struct alias_link *_lnk);
 347 struct in_addr
 348                 GetProxyAddress(struct alias_link *_lnk);
 349 void            SetProxyAddress(struct alias_link *_lnk, struct in_addr _addr);
 350 u_short         GetProxyPort(struct alias_link *_lnk);
 351 void            SetProxyPort(struct alias_link *_lnk, u_short _port);
 352 void            SetAckModified(struct alias_link *_lnk);
 353 int             GetAckModified(struct alias_link *_lnk);
 354 int             GetDeltaAckIn(u_long, struct alias_link *_lnk);
 355 int             GetDeltaSeqOut(u_long, struct alias_link *lnk);
 356 void            AddSeq(struct alias_link *lnk, int delta, u_int ip_hl,
 357                     u_short ip_len, u_long th_seq, u_int th_off);
 358 void            SetExpire (struct alias_link *_lnk, int _expire);
 359 void            ClearCheckNewLink(struct libalias *la);
 360 void            SetProtocolFlags(struct alias_link *_lnk, int _pflags);
 361 int             GetProtocolFlags(struct alias_link *_lnk);
 362 void            SetDestCallId(struct alias_link *_lnk, u_int16_t _cid);
 363
 364 #ifndef NO_FW_PUNCH
 365 void            PunchFWHole(struct alias_link *_lnk);
 366
 367 #endif
 368
 369 /* Housekeeping function */
 370 void            HouseKeeping(struct libalias *);
 371
 372 /* Tcp specific routines */
 373 /* lint -save -library Suppress flexelint warnings */
 374
 375 /* Transparent proxy routines */
 376 int
 377 ProxyCheck(struct libalias *la, struct in_addr *proxy_server_addr,
 378     u_short * proxy_server_port, struct in_addr src_addr,
 379     struct in_addr dst_addr, u_short dst_port, u_char ip_p);
 380 void
 381 ProxyModify(struct libalias *la, struct alias_link *_lnk, struct ip *_pip,
 382     int _maxpacketsize, int _proxy_type);
 383
 384 enum alias_tcp_state {
 385         ALIAS_TCP_STATE_NOT_CONNECTED,
 386         ALIAS_TCP_STATE_CONNECTED,
 387         ALIAS_TCP_STATE_DISCONNECTED
 388 };
 389
 390 #if defined(_NETINET_IP_H_)
 391 static __inline void *
 392 ip_next(struct ip *iphdr)
 393 {
 394         char *p = (char *)iphdr;
 395         return (&p[iphdr->ip_hl * 4]);
 396 }
 397 #endif
 398
 399 #if defined(_NETINET_TCP_H_)
 400 static __inline void *
 401 tcp_next(struct tcphdr *tcphdr)
 402 {
 403         char *p = (char *)tcphdr;
 404         return (&p[tcphdr->th_off * 4]);
 405 }
 406 #endif
 407
 408 #if defined(_NETINET_UDP_H_)
 409 static __inline void *
 410 udp_next(struct udphdr *udphdr)
 411 {
 412         return ((void *)(udphdr + 1));
 413 }
 414 #endif
 415
 416 #endif                          /* !_ALIAS_LOCAL_H_ */