1f987e1bdSBrian Somers /*- 2e83aaae3SBrian Somers * Copyright (c) 2001 Charles Mott <[email protected]> 3f987e1bdSBrian Somers * All rights reserved. 4f987e1bdSBrian Somers * 5f987e1bdSBrian Somers * Redistribution and use in source and binary forms, with or without 6f987e1bdSBrian Somers * modification, are permitted provided that the following conditions 7f987e1bdSBrian Somers * are met: 8f987e1bdSBrian Somers * 1. Redistributions of source code must retain the above copyright 9f987e1bdSBrian Somers * notice, this list of conditions and the following disclaimer. 10f987e1bdSBrian Somers * 2. Redistributions in binary form must reproduce the above copyright 11f987e1bdSBrian Somers * notice, this list of conditions and the following disclaimer in the 12f987e1bdSBrian Somers * documentation and/or other materials provided with the distribution. 13f987e1bdSBrian Somers * 14f987e1bdSBrian Somers * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15f987e1bdSBrian Somers * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16f987e1bdSBrian Somers * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17f987e1bdSBrian Somers * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18f987e1bdSBrian Somers * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19f987e1bdSBrian Somers * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20f987e1bdSBrian Somers * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21f987e1bdSBrian Somers * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22f987e1bdSBrian Somers * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23f987e1bdSBrian Somers * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24f987e1bdSBrian Somers * SUCH DAMAGE. 25f987e1bdSBrian Somers */ 26f987e1bdSBrian Somers 27e2505aa6SMatthew Dillon #include <sys/cdefs.h> 28e2505aa6SMatthew Dillon __FBSDID("$FreeBSD$"); 29e2505aa6SMatthew Dillon 307d96f4efSBrian Somers /* file: alias_proxy.c 317d96f4efSBrian Somers 327d96f4efSBrian Somers This file encapsulates special operations related to transparent 337d96f4efSBrian Somers proxy redirection. This is where packets with a particular destination, 347d96f4efSBrian Somers usually tcp port 80, are redirected to a proxy server. 357d96f4efSBrian Somers 367d96f4efSBrian Somers When packets are proxied, the destination address and port are 377d96f4efSBrian Somers modified. In certain cases, it is necessary to somehow encode 387d96f4efSBrian Somers the original address/port info into the packet. Two methods are 397d96f4efSBrian Somers presently supported: addition of a [DEST addr port] string at the 409d5abbddSJens Schweikhardt beginning of a tcp stream, or inclusion of an optional field 417d96f4efSBrian Somers in the IP header. 427d96f4efSBrian Somers 437d96f4efSBrian Somers There is one public API function: 447d96f4efSBrian Somers 457d96f4efSBrian Somers PacketAliasProxyRule() -- Adds and deletes proxy 467d96f4efSBrian Somers rules. 477d96f4efSBrian Somers 487d96f4efSBrian Somers Rules are stored in a linear linked list, so lookup efficiency 497d96f4efSBrian Somers won't be too good for large lists. 507d96f4efSBrian Somers 517d96f4efSBrian Somers 527d96f4efSBrian Somers Initial development: April, 1998 (cjm) 537d96f4efSBrian Somers */ 547d96f4efSBrian Somers 557d96f4efSBrian Somers 567d96f4efSBrian Somers /* System includes */ 57c649a2e0SGleb Smirnoff #ifdef _KERNEL 58c649a2e0SGleb Smirnoff #include <sys/param.h> 59c649a2e0SGleb Smirnoff #include <sys/ctype.h> 60c649a2e0SGleb Smirnoff #include <sys/libkern.h> 61c649a2e0SGleb Smirnoff #include <sys/kernel.h> 62c649a2e0SGleb Smirnoff #include <sys/malloc.h> 63c649a2e0SGleb Smirnoff #include <sys/limits.h> 64c649a2e0SGleb Smirnoff #else 65c649a2e0SGleb Smirnoff #include <sys/types.h> 66c649a2e0SGleb Smirnoff #include <sys/socket.h> 677d96f4efSBrian Somers #include <ctype.h> 687d96f4efSBrian Somers #include <stdio.h> 697d96f4efSBrian Somers #include <stdlib.h> 707d96f4efSBrian Somers #include <string.h> 717d96f4efSBrian Somers #include <netdb.h> 72c649a2e0SGleb Smirnoff #include <arpa/inet.h> 73c649a2e0SGleb Smirnoff #endif 747d96f4efSBrian Somers 757d96f4efSBrian Somers /* BSD IPV4 includes */ 767d96f4efSBrian Somers #include <netinet/in_systm.h> 777d96f4efSBrian Somers #include <netinet/in.h> 787d96f4efSBrian Somers #include <netinet/ip.h> 797d96f4efSBrian Somers #include <netinet/tcp.h> 807d96f4efSBrian Somers 81c649a2e0SGleb Smirnoff #ifdef _KERNEL 82c649a2e0SGleb Smirnoff #include <netinet/libalias/alias.h> 8375bc2620SGleb Smirnoff #include <netinet/libalias/alias_local.h> 84c649a2e0SGleb Smirnoff #else 857d96f4efSBrian Somers #include "alias.h" /* Public API functions for libalias */ 8675bc2620SGleb Smirnoff #include "alias_local.h" /* Functions used by alias*.c */ 87c649a2e0SGleb Smirnoff #endif 887d96f4efSBrian Somers 897d96f4efSBrian Somers /* 907d96f4efSBrian Somers Data structures 917d96f4efSBrian Somers */ 927d96f4efSBrian Somers 937d96f4efSBrian Somers /* 947d96f4efSBrian Somers * A linked list of arbitrary length, based on struct proxy_entry is 957d96f4efSBrian Somers * used to store proxy rules. 967d96f4efSBrian Somers */ 97f0f93429SDag-Erling Smørgrav struct proxy_entry { 985e289f9eSPoul-Henning Kamp struct libalias *la; 997d96f4efSBrian Somers #define PROXY_TYPE_ENCODE_NONE 1 1007d96f4efSBrian Somers #define PROXY_TYPE_ENCODE_TCPSTREAM 2 1017d96f4efSBrian Somers #define PROXY_TYPE_ENCODE_IPHDR 3 1027d96f4efSBrian Somers int rule_index; 1037d96f4efSBrian Somers int proxy_type; 1047d96f4efSBrian Somers u_char proto; 1057d96f4efSBrian Somers u_short proxy_port; 1067d96f4efSBrian Somers u_short server_port; 1077d96f4efSBrian Somers 1087d96f4efSBrian Somers struct in_addr server_addr; 1097d96f4efSBrian Somers 1107d96f4efSBrian Somers struct in_addr src_addr; 1117d96f4efSBrian Somers struct in_addr src_mask; 1127d96f4efSBrian Somers 1137d96f4efSBrian Somers struct in_addr dst_addr; 1147d96f4efSBrian Somers struct in_addr dst_mask; 1157d96f4efSBrian Somers 1167d96f4efSBrian Somers struct proxy_entry *next; 1177d96f4efSBrian Somers struct proxy_entry *last; 1187d96f4efSBrian Somers }; 1197d96f4efSBrian Somers 1207d96f4efSBrian Somers 1217d96f4efSBrian Somers 1227d96f4efSBrian Somers /* 1237d96f4efSBrian Somers File scope variables 1247d96f4efSBrian Somers */ 1257d96f4efSBrian Somers 1267d96f4efSBrian Somers 1277d96f4efSBrian Somers 1287d96f4efSBrian Somers /* Local (static) functions: 1297d96f4efSBrian Somers 1307d96f4efSBrian Somers IpMask() -- Utility function for creating IP 1317d96f4efSBrian Somers masks from integer (1-32) specification. 1327d96f4efSBrian Somers IpAddr() -- Utility function for converting string 1337d96f4efSBrian Somers to IP address 1347d96f4efSBrian Somers IpPort() -- Utility function for converting string 1357d96f4efSBrian Somers to port number 1367d96f4efSBrian Somers RuleAdd() -- Adds an element to the rule list. 1377d96f4efSBrian Somers RuleDelete() -- Removes an element from the rule list. 1387d96f4efSBrian Somers RuleNumberDelete() -- Removes all elements from the rule list 1397d96f4efSBrian Somers having a certain rule number. 1407d96f4efSBrian Somers ProxyEncodeTcpStream() -- Adds [DEST x.x.x.x xxxx] to the beginning 1417d96f4efSBrian Somers of a TCP stream. 1427d96f4efSBrian Somers ProxyEncodeIpHeader() -- Adds an IP option indicating the true 1437d96f4efSBrian Somers destination of a proxied IP packet 1447d96f4efSBrian Somers */ 1457d96f4efSBrian Somers 146e9d5db28SGleb Smirnoff #ifdef _KERNEL /* XXX: can it be moved to libkern? */ 147e9d5db28SGleb Smirnoff static int inet_aton(const char *cp, struct in_addr *addr); 148e9d5db28SGleb Smirnoff #endif 1497d96f4efSBrian Somers static int IpMask(int, struct in_addr *); 1507d96f4efSBrian Somers static int IpAddr(char *, struct in_addr *); 1517d96f4efSBrian Somers static int IpPort(char *, int, int *); 1525e289f9eSPoul-Henning Kamp static void RuleAdd(struct libalias *la, struct proxy_entry *); 1537d96f4efSBrian Somers static void RuleDelete(struct proxy_entry *); 1545e289f9eSPoul-Henning Kamp static int RuleNumberDelete(struct libalias *la, int); 1557d96f4efSBrian Somers static void ProxyEncodeTcpStream(struct alias_link *, struct ip *, int); 1567d96f4efSBrian Somers static void ProxyEncodeIpHeader(struct ip *, int); 1577d96f4efSBrian Somers 158e9d5db28SGleb Smirnoff #ifdef _KERNEL 159e9d5db28SGleb Smirnoff static int 160e9d5db28SGleb Smirnoff inet_aton(cp, addr) 161e9d5db28SGleb Smirnoff const char *cp; 162e9d5db28SGleb Smirnoff struct in_addr *addr; 163e9d5db28SGleb Smirnoff { 164e9d5db28SGleb Smirnoff u_long parts[4]; 165e9d5db28SGleb Smirnoff in_addr_t val; 166e9d5db28SGleb Smirnoff const char *c; 167e9d5db28SGleb Smirnoff char *endptr; 168e9d5db28SGleb Smirnoff int gotend, n; 169e9d5db28SGleb Smirnoff 170e9d5db28SGleb Smirnoff c = (const char *)cp; 171e9d5db28SGleb Smirnoff n = 0; 172e9d5db28SGleb Smirnoff /* 173e9d5db28SGleb Smirnoff * Run through the string, grabbing numbers until 174e9d5db28SGleb Smirnoff * the end of the string, or some error 175e9d5db28SGleb Smirnoff */ 176e9d5db28SGleb Smirnoff gotend = 0; 177e9d5db28SGleb Smirnoff while (!gotend) { 178cbfbc555SGleb Smirnoff unsigned long l; 179e9d5db28SGleb Smirnoff 180cbfbc555SGleb Smirnoff l = strtoul(c, &endptr, 0); 181cbfbc555SGleb Smirnoff 182cbfbc555SGleb Smirnoff if (l == ULONG_MAX || l == 0) 183e9d5db28SGleb Smirnoff return (0); 184e9d5db28SGleb Smirnoff 185cbfbc555SGleb Smirnoff val = (in_addr_t)l; 186e9d5db28SGleb Smirnoff /* 187e9d5db28SGleb Smirnoff * If the whole string is invalid, endptr will equal 188e9d5db28SGleb Smirnoff * c.. this way we can make sure someone hasn't 189e9d5db28SGleb Smirnoff * gone '.12' or something which would get past 190e9d5db28SGleb Smirnoff * the next check. 191e9d5db28SGleb Smirnoff */ 192e9d5db28SGleb Smirnoff if (endptr == c) 193e9d5db28SGleb Smirnoff return (0); 194e9d5db28SGleb Smirnoff parts[n] = val; 195e9d5db28SGleb Smirnoff c = endptr; 196e9d5db28SGleb Smirnoff 197e9d5db28SGleb Smirnoff /* Check the next character past the previous number's end */ 198e9d5db28SGleb Smirnoff switch (*c) { 199e9d5db28SGleb Smirnoff case '.' : 200e9d5db28SGleb Smirnoff /* Make sure we only do 3 dots .. */ 201e9d5db28SGleb Smirnoff if (n == 3) /* Whoops. Quit. */ 202e9d5db28SGleb Smirnoff return (0); 203e9d5db28SGleb Smirnoff n++; 204e9d5db28SGleb Smirnoff c++; 205e9d5db28SGleb Smirnoff break; 206e9d5db28SGleb Smirnoff 207e9d5db28SGleb Smirnoff case '\0': 208e9d5db28SGleb Smirnoff gotend = 1; 209e9d5db28SGleb Smirnoff break; 210e9d5db28SGleb Smirnoff 211e9d5db28SGleb Smirnoff default: 212e9d5db28SGleb Smirnoff if (isspace((unsigned char)*c)) { 213e9d5db28SGleb Smirnoff gotend = 1; 214e9d5db28SGleb Smirnoff break; 215e9d5db28SGleb Smirnoff } else 216e9d5db28SGleb Smirnoff return (0); /* Invalid character, so fail */ 217e9d5db28SGleb Smirnoff } 218e9d5db28SGleb Smirnoff 219e9d5db28SGleb Smirnoff } 220e9d5db28SGleb Smirnoff 221e9d5db28SGleb Smirnoff /* 222e9d5db28SGleb Smirnoff * Concoct the address according to 223e9d5db28SGleb Smirnoff * the number of parts specified. 224e9d5db28SGleb Smirnoff */ 225e9d5db28SGleb Smirnoff 226e9d5db28SGleb Smirnoff switch (n) { 227e9d5db28SGleb Smirnoff case 0: /* a -- 32 bits */ 228e9d5db28SGleb Smirnoff /* 229e9d5db28SGleb Smirnoff * Nothing is necessary here. Overflow checking was 230e9d5db28SGleb Smirnoff * already done in strtoul(). 231e9d5db28SGleb Smirnoff */ 232e9d5db28SGleb Smirnoff break; 233e9d5db28SGleb Smirnoff case 1: /* a.b -- 8.24 bits */ 234e9d5db28SGleb Smirnoff if (val > 0xffffff || parts[0] > 0xff) 235e9d5db28SGleb Smirnoff return (0); 236e9d5db28SGleb Smirnoff val |= parts[0] << 24; 237e9d5db28SGleb Smirnoff break; 238e9d5db28SGleb Smirnoff 239e9d5db28SGleb Smirnoff case 2: /* a.b.c -- 8.8.16 bits */ 240e9d5db28SGleb Smirnoff if (val > 0xffff || parts[0] > 0xff || parts[1] > 0xff) 241e9d5db28SGleb Smirnoff return (0); 242e9d5db28SGleb Smirnoff val |= (parts[0] << 24) | (parts[1] << 16); 243e9d5db28SGleb Smirnoff break; 244e9d5db28SGleb Smirnoff 245e9d5db28SGleb Smirnoff case 3: /* a.b.c.d -- 8.8.8.8 bits */ 246e9d5db28SGleb Smirnoff if (val > 0xff || parts[0] > 0xff || parts[1] > 0xff || 247e9d5db28SGleb Smirnoff parts[2] > 0xff) 248e9d5db28SGleb Smirnoff return (0); 249e9d5db28SGleb Smirnoff val |= (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8); 250e9d5db28SGleb Smirnoff break; 251e9d5db28SGleb Smirnoff } 252e9d5db28SGleb Smirnoff 253e9d5db28SGleb Smirnoff if (addr != NULL) 254e9d5db28SGleb Smirnoff addr->s_addr = htonl(val); 255e9d5db28SGleb Smirnoff return (1); 256e9d5db28SGleb Smirnoff } 257e9d5db28SGleb Smirnoff #endif 258e9d5db28SGleb Smirnoff 2597d96f4efSBrian Somers static int 2607d96f4efSBrian Somers IpMask(int nbits, struct in_addr *mask) 2617d96f4efSBrian Somers { 2627d96f4efSBrian Somers int i; 2637d96f4efSBrian Somers u_int imask; 2647d96f4efSBrian Somers 2657d96f4efSBrian Somers if (nbits < 0 || nbits > 32) 266ffcb611aSDag-Erling Smørgrav return (-1); 2677d96f4efSBrian Somers 2687d96f4efSBrian Somers imask = 0; 2697d96f4efSBrian Somers for (i = 0; i < nbits; i++) 2707d96f4efSBrian Somers imask = (imask >> 1) + 0x80000000; 2717d96f4efSBrian Somers mask->s_addr = htonl(imask); 2727d96f4efSBrian Somers 273ffcb611aSDag-Erling Smørgrav return (0); 2747d96f4efSBrian Somers } 2757d96f4efSBrian Somers 2767d96f4efSBrian Somers static int 2777d96f4efSBrian Somers IpAddr(char *s, struct in_addr *addr) 2787d96f4efSBrian Somers { 2797d96f4efSBrian Somers if (inet_aton(s, addr) == 0) 280ffcb611aSDag-Erling Smørgrav return (-1); 2817d96f4efSBrian Somers else 282ffcb611aSDag-Erling Smørgrav return (0); 2837d96f4efSBrian Somers } 2847d96f4efSBrian Somers 2857d96f4efSBrian Somers static int 2867d96f4efSBrian Somers IpPort(char *s, int proto, int *port) 2877d96f4efSBrian Somers { 2887d96f4efSBrian Somers int n; 2897d96f4efSBrian Somers 2907d96f4efSBrian Somers n = sscanf(s, "%d", port); 291e9d5db28SGleb Smirnoff if (n != 1) 292e9d5db28SGleb Smirnoff #ifndef _KERNEL /* XXX: we accept only numeric ports in kernel */ 293e9d5db28SGleb Smirnoff { 2947d96f4efSBrian Somers struct servent *se; 2957d96f4efSBrian Somers 2967d96f4efSBrian Somers if (proto == IPPROTO_TCP) 2977d96f4efSBrian Somers se = getservbyname(s, "tcp"); 2987d96f4efSBrian Somers else if (proto == IPPROTO_UDP) 2997d96f4efSBrian Somers se = getservbyname(s, "udp"); 3007d96f4efSBrian Somers else 301ffcb611aSDag-Erling Smørgrav return (-1); 3027d96f4efSBrian Somers 3037d96f4efSBrian Somers if (se == NULL) 304ffcb611aSDag-Erling Smørgrav return (-1); 3057d96f4efSBrian Somers 3067d96f4efSBrian Somers *port = (u_int) ntohs(se->s_port); 3077d96f4efSBrian Somers } 308e9d5db28SGleb Smirnoff #else 309e9d5db28SGleb Smirnoff return (-1); 310e9d5db28SGleb Smirnoff #endif 311ffcb611aSDag-Erling Smørgrav return (0); 3127d96f4efSBrian Somers } 3137d96f4efSBrian Somers 3147d96f4efSBrian Somers void 3155e289f9eSPoul-Henning Kamp RuleAdd(struct libalias *la, struct proxy_entry *entry) 3167d96f4efSBrian Somers { 3177d96f4efSBrian Somers int rule_index; 3187d96f4efSBrian Somers struct proxy_entry *ptr; 3197d96f4efSBrian Somers struct proxy_entry *ptr_last; 3207d96f4efSBrian Somers 321f0f93429SDag-Erling Smørgrav if (la->proxyList == NULL) { 3225e289f9eSPoul-Henning Kamp la->proxyList = entry; 3237d96f4efSBrian Somers entry->last = NULL; 3247d96f4efSBrian Somers entry->next = NULL; 3257d96f4efSBrian Somers return; 3267d96f4efSBrian Somers } 3275e289f9eSPoul-Henning Kamp entry->la = la; 3287d96f4efSBrian Somers 3297d96f4efSBrian Somers rule_index = entry->rule_index; 3305e289f9eSPoul-Henning Kamp ptr = la->proxyList; 3317d96f4efSBrian Somers ptr_last = NULL; 332f0f93429SDag-Erling Smørgrav while (ptr != NULL) { 333f0f93429SDag-Erling Smørgrav if (ptr->rule_index >= rule_index) { 334f0f93429SDag-Erling Smørgrav if (ptr_last == NULL) { 3355e289f9eSPoul-Henning Kamp entry->next = la->proxyList; 3367d96f4efSBrian Somers entry->last = NULL; 3375e289f9eSPoul-Henning Kamp la->proxyList->last = entry; 3385e289f9eSPoul-Henning Kamp la->proxyList = entry; 3397d96f4efSBrian Somers return; 3407d96f4efSBrian Somers } 3417d96f4efSBrian Somers ptr_last->next = entry; 3427d96f4efSBrian Somers ptr->last = entry; 3437d96f4efSBrian Somers entry->last = ptr->last; 3447d96f4efSBrian Somers entry->next = ptr; 3457d96f4efSBrian Somers return; 3467d96f4efSBrian Somers } 3477d96f4efSBrian Somers ptr_last = ptr; 3487d96f4efSBrian Somers ptr = ptr->next; 3497d96f4efSBrian Somers } 3507d96f4efSBrian Somers 3517d96f4efSBrian Somers ptr_last->next = entry; 3527d96f4efSBrian Somers entry->last = ptr_last; 3537d96f4efSBrian Somers entry->next = NULL; 3547d96f4efSBrian Somers } 3557d96f4efSBrian Somers 3567d96f4efSBrian Somers static void 3577d96f4efSBrian Somers RuleDelete(struct proxy_entry *entry) 3587d96f4efSBrian Somers { 3595e289f9eSPoul-Henning Kamp struct libalias *la; 3605e289f9eSPoul-Henning Kamp 3615e289f9eSPoul-Henning Kamp la = entry->la; 3627d96f4efSBrian Somers if (entry->last != NULL) 3637d96f4efSBrian Somers entry->last->next = entry->next; 3647d96f4efSBrian Somers else 3655e289f9eSPoul-Henning Kamp la->proxyList = entry->next; 3667d96f4efSBrian Somers 3677d96f4efSBrian Somers if (entry->next != NULL) 3687d96f4efSBrian Somers entry->next->last = entry->last; 3697d96f4efSBrian Somers 3707d96f4efSBrian Somers free(entry); 3717d96f4efSBrian Somers } 3727d96f4efSBrian Somers 3737d96f4efSBrian Somers static int 3745e289f9eSPoul-Henning Kamp RuleNumberDelete(struct libalias *la, int rule_index) 3757d96f4efSBrian Somers { 3767d96f4efSBrian Somers int err; 3777d96f4efSBrian Somers struct proxy_entry *ptr; 3787d96f4efSBrian Somers 3797d96f4efSBrian Somers err = -1; 3805e289f9eSPoul-Henning Kamp ptr = la->proxyList; 381f0f93429SDag-Erling Smørgrav while (ptr != NULL) { 3827d96f4efSBrian Somers struct proxy_entry *ptr_next; 3837d96f4efSBrian Somers 3847d96f4efSBrian Somers ptr_next = ptr->next; 385f0f93429SDag-Erling Smørgrav if (ptr->rule_index == rule_index) { 3867d96f4efSBrian Somers err = 0; 3877d96f4efSBrian Somers RuleDelete(ptr); 3887d96f4efSBrian Somers } 3897d96f4efSBrian Somers ptr = ptr_next; 3907d96f4efSBrian Somers } 3917d96f4efSBrian Somers 392ffcb611aSDag-Erling Smørgrav return (err); 3937d96f4efSBrian Somers } 3947d96f4efSBrian Somers 3957d96f4efSBrian Somers static void 396ed01a582SDag-Erling Smørgrav ProxyEncodeTcpStream(struct alias_link *lnk, 3977d96f4efSBrian Somers struct ip *pip, 3987d96f4efSBrian Somers int maxpacketsize) 3997d96f4efSBrian Somers { 4007d96f4efSBrian Somers int slen; 4017d96f4efSBrian Somers char buffer[40]; 4027d96f4efSBrian Somers struct tcphdr *tc; 4037d96f4efSBrian Somers 4047d96f4efSBrian Somers /* Compute pointer to tcp header */ 4059fa0fd26SDag-Erling Smørgrav tc = (struct tcphdr *)ip_next(pip); 4067d96f4efSBrian Somers 4077d96f4efSBrian Somers /* Don't modify if once already modified */ 4087d96f4efSBrian Somers 409ed01a582SDag-Erling Smørgrav if (GetAckModified(lnk)) 4107d96f4efSBrian Somers return; 4117d96f4efSBrian Somers 4127d96f4efSBrian Somers /* Translate destination address and port to string form */ 4137d96f4efSBrian Somers snprintf(buffer, sizeof(buffer) - 2, "[DEST %s %d]", 414ed01a582SDag-Erling Smørgrav inet_ntoa(GetProxyAddress(lnk)), (u_int) ntohs(GetProxyPort(lnk))); 4157d96f4efSBrian Somers 4167d96f4efSBrian Somers /* Pad string out to a multiple of two in length */ 4177d96f4efSBrian Somers slen = strlen(buffer); 418f0f93429SDag-Erling Smørgrav switch (slen % 2) { 4197d96f4efSBrian Somers case 0: 4207d96f4efSBrian Somers strcat(buffer, " \n"); 4217d96f4efSBrian Somers slen += 2; 4227d96f4efSBrian Somers break; 4237d96f4efSBrian Somers case 1: 4247d96f4efSBrian Somers strcat(buffer, "\n"); 4257d96f4efSBrian Somers slen += 1; 4267d96f4efSBrian Somers } 4277d96f4efSBrian Somers 4287d96f4efSBrian Somers /* Check for packet overflow */ 429ed01a582SDag-Erling Smørgrav if ((int)(ntohs(pip->ip_len) + strlen(buffer)) > maxpacketsize) 4307d96f4efSBrian Somers return; 4317d96f4efSBrian Somers 4327d96f4efSBrian Somers /* Shift existing TCP data and insert destination string */ 4337d96f4efSBrian Somers { 4347d96f4efSBrian Somers int dlen; 4357d96f4efSBrian Somers int hlen; 4367d96f4efSBrian Somers u_char *p; 4377d96f4efSBrian Somers 4387d96f4efSBrian Somers hlen = (pip->ip_hl + tc->th_off) << 2; 4397d96f4efSBrian Somers dlen = ntohs(pip->ip_len) - hlen; 4407d96f4efSBrian Somers 4417d96f4efSBrian Somers /* Modify first packet that has data in it */ 4427d96f4efSBrian Somers 4437d96f4efSBrian Somers if (dlen == 0) 4447d96f4efSBrian Somers return; 4457d96f4efSBrian Somers 4467d96f4efSBrian Somers p = (char *)pip; 4477d96f4efSBrian Somers p += hlen; 4487d96f4efSBrian Somers 449c8d3ca72SGleb Smirnoff bcopy(p, p + slen, dlen); 4507d96f4efSBrian Somers memcpy(p, buffer, slen); 4517d96f4efSBrian Somers } 4527d96f4efSBrian Somers 4537d96f4efSBrian Somers /* Save information about modfied sequence number */ 4547d96f4efSBrian Somers { 4557d96f4efSBrian Somers int delta; 4567d96f4efSBrian Somers 457ed01a582SDag-Erling Smørgrav SetAckModified(lnk); 458ed01a582SDag-Erling Smørgrav delta = GetDeltaSeqOut(pip, lnk); 459ed01a582SDag-Erling Smørgrav AddSeq(pip, lnk, delta + slen); 4607d96f4efSBrian Somers } 4617d96f4efSBrian Somers 4627d96f4efSBrian Somers /* Update IP header packet length and checksum */ 4637d96f4efSBrian Somers { 4647d96f4efSBrian Somers int accumulate; 4657d96f4efSBrian Somers 4667d96f4efSBrian Somers accumulate = pip->ip_len; 4677d96f4efSBrian Somers pip->ip_len = htons(ntohs(pip->ip_len) + slen); 4687d96f4efSBrian Somers accumulate -= pip->ip_len; 4697d96f4efSBrian Somers 4707d96f4efSBrian Somers ADJUST_CHECKSUM(accumulate, pip->ip_sum); 4717d96f4efSBrian Somers } 4727d96f4efSBrian Somers 4737d96f4efSBrian Somers /* Update TCP checksum, Use TcpChecksum since so many things have 4747d96f4efSBrian Somers already changed. */ 4757d96f4efSBrian Somers 4767d96f4efSBrian Somers tc->th_sum = 0; 477*59dde15eSGleb Smirnoff #ifdef _KERNEL 478*59dde15eSGleb Smirnoff tc->th_x2 = 1; 479*59dde15eSGleb Smirnoff #else 4807d96f4efSBrian Somers tc->th_sum = TcpChecksum(pip); 481*59dde15eSGleb Smirnoff #endif 4827d96f4efSBrian Somers } 4837d96f4efSBrian Somers 4847d96f4efSBrian Somers static void 4857d96f4efSBrian Somers ProxyEncodeIpHeader(struct ip *pip, 4867d96f4efSBrian Somers int maxpacketsize) 4877d96f4efSBrian Somers { 4887d96f4efSBrian Somers #define OPTION_LEN_BYTES 8 4897d96f4efSBrian Somers #define OPTION_LEN_INT16 4 4907d96f4efSBrian Somers #define OPTION_LEN_INT32 2 4917d96f4efSBrian Somers u_char option[OPTION_LEN_BYTES]; 4927d96f4efSBrian Somers 4936293e003SGleb Smirnoff #ifdef LIBALIAS_DEBUG 4947d96f4efSBrian Somers fprintf(stdout, " ip cksum 1 = %x\n", (u_int) IpChecksum(pip)); 4957d96f4efSBrian Somers fprintf(stdout, "tcp cksum 1 = %x\n", (u_int) TcpChecksum(pip)); 4964c32f5d2SBrian Somers #endif 4977d96f4efSBrian Somers 498ed01a582SDag-Erling Smørgrav (void)maxpacketsize; 499ed01a582SDag-Erling Smørgrav 5007d96f4efSBrian Somers /* Check to see that there is room to add an IP option */ 5017d96f4efSBrian Somers if (pip->ip_hl > (0x0f - OPTION_LEN_INT32)) 5027d96f4efSBrian Somers return; 5037d96f4efSBrian Somers 5047d96f4efSBrian Somers /* Build option and copy into packet */ 5057d96f4efSBrian Somers { 5067d96f4efSBrian Somers u_char *ptr; 5077d96f4efSBrian Somers struct tcphdr *tc; 5087d96f4efSBrian Somers 5097d96f4efSBrian Somers ptr = (u_char *) pip; 5107d96f4efSBrian Somers ptr += 20; 5117d96f4efSBrian Somers memcpy(ptr + OPTION_LEN_BYTES, ptr, ntohs(pip->ip_len) - 20); 5127d96f4efSBrian Somers 5137d96f4efSBrian Somers option[0] = 0x64; /* class: 3 (reserved), option 4 */ 5147d96f4efSBrian Somers option[1] = OPTION_LEN_BYTES; 5157d96f4efSBrian Somers 5167d96f4efSBrian Somers memcpy(&option[2], (u_char *) & pip->ip_dst, 4); 5177d96f4efSBrian Somers 5189fa0fd26SDag-Erling Smørgrav tc = (struct tcphdr *)ip_next(pip); 5197d96f4efSBrian Somers memcpy(&option[6], (u_char *) & tc->th_sport, 2); 5207d96f4efSBrian Somers 5217d96f4efSBrian Somers memcpy(ptr, option, 8); 5227d96f4efSBrian Somers } 5237d96f4efSBrian Somers 5247d96f4efSBrian Somers /* Update checksum, header length and packet length */ 5257d96f4efSBrian Somers { 5267d96f4efSBrian Somers int i; 5277d96f4efSBrian Somers int accumulate; 5287d96f4efSBrian Somers u_short *sptr; 5297d96f4efSBrian Somers 5307d96f4efSBrian Somers sptr = (u_short *) option; 5317d96f4efSBrian Somers accumulate = 0; 5327d96f4efSBrian Somers for (i = 0; i < OPTION_LEN_INT16; i++) 5337d96f4efSBrian Somers accumulate -= *(sptr++); 5347d96f4efSBrian Somers 5357d96f4efSBrian Somers sptr = (u_short *) pip; 5367d96f4efSBrian Somers accumulate += *sptr; 5377d96f4efSBrian Somers pip->ip_hl += OPTION_LEN_INT32; 5387d96f4efSBrian Somers accumulate -= *sptr; 5397d96f4efSBrian Somers 5407d96f4efSBrian Somers accumulate += pip->ip_len; 5417d96f4efSBrian Somers pip->ip_len = htons(ntohs(pip->ip_len) + OPTION_LEN_BYTES); 5427d96f4efSBrian Somers accumulate -= pip->ip_len; 5437d96f4efSBrian Somers 5447d96f4efSBrian Somers ADJUST_CHECKSUM(accumulate, pip->ip_sum); 5457d96f4efSBrian Somers } 5467d96f4efSBrian Somers #undef OPTION_LEN_BYTES 5477d96f4efSBrian Somers #undef OPTION_LEN_INT16 5487d96f4efSBrian Somers #undef OPTION_LEN_INT32 5496293e003SGleb Smirnoff #ifdef LIBALIAS_DEBUG 5507d96f4efSBrian Somers fprintf(stdout, " ip cksum 2 = %x\n", (u_int) IpChecksum(pip)); 5517d96f4efSBrian Somers fprintf(stdout, "tcp cksum 2 = %x\n", (u_int) TcpChecksum(pip)); 5524c32f5d2SBrian Somers #endif 5537d96f4efSBrian Somers } 5547d96f4efSBrian Somers 5557d96f4efSBrian Somers 5567d96f4efSBrian Somers /* Functions by other packet alias source files 5577d96f4efSBrian Somers 5587d96f4efSBrian Somers ProxyCheck() -- Checks whether an outgoing packet should 5597d96f4efSBrian Somers be proxied. 5607d96f4efSBrian Somers ProxyModify() -- Encodes the original destination address/port 5617d96f4efSBrian Somers for a packet which is to be redirected to 5627d96f4efSBrian Somers a proxy server. 5637d96f4efSBrian Somers */ 5647d96f4efSBrian Somers 5657d96f4efSBrian Somers int 5665e289f9eSPoul-Henning Kamp ProxyCheck(struct libalias *la, struct ip *pip, 5677d96f4efSBrian Somers struct in_addr *proxy_server_addr, 5687d96f4efSBrian Somers u_short * proxy_server_port) 5697d96f4efSBrian Somers { 5707d96f4efSBrian Somers u_short dst_port; 5717d96f4efSBrian Somers struct in_addr src_addr; 5727d96f4efSBrian Somers struct in_addr dst_addr; 5737d96f4efSBrian Somers struct proxy_entry *ptr; 5747d96f4efSBrian Somers 5757d96f4efSBrian Somers src_addr = pip->ip_src; 5767d96f4efSBrian Somers dst_addr = pip->ip_dst; 5779fa0fd26SDag-Erling Smørgrav dst_port = ((struct tcphdr *)ip_next(pip)) 5787d96f4efSBrian Somers ->th_dport; 5797d96f4efSBrian Somers 5805e289f9eSPoul-Henning Kamp ptr = la->proxyList; 581f0f93429SDag-Erling Smørgrav while (ptr != NULL) { 5827d96f4efSBrian Somers u_short proxy_port; 5837d96f4efSBrian Somers 5847d96f4efSBrian Somers proxy_port = ptr->proxy_port; 5857d96f4efSBrian Somers if ((dst_port == proxy_port || proxy_port == 0) 5867d96f4efSBrian Somers && pip->ip_p == ptr->proto 587f0f93429SDag-Erling Smørgrav && src_addr.s_addr != ptr->server_addr.s_addr) { 5887d96f4efSBrian Somers struct in_addr src_addr_masked; 5897d96f4efSBrian Somers struct in_addr dst_addr_masked; 5907d96f4efSBrian Somers 5917d96f4efSBrian Somers src_addr_masked.s_addr = src_addr.s_addr & ptr->src_mask.s_addr; 5927d96f4efSBrian Somers dst_addr_masked.s_addr = dst_addr.s_addr & ptr->dst_mask.s_addr; 5937d96f4efSBrian Somers 5947d96f4efSBrian Somers if ((src_addr_masked.s_addr == ptr->src_addr.s_addr) 595f0f93429SDag-Erling Smørgrav && (dst_addr_masked.s_addr == ptr->dst_addr.s_addr)) { 5967d96f4efSBrian Somers if ((*proxy_server_port = ptr->server_port) == 0) 5977d96f4efSBrian Somers *proxy_server_port = dst_port; 5987d96f4efSBrian Somers *proxy_server_addr = ptr->server_addr; 599ffcb611aSDag-Erling Smørgrav return (ptr->proxy_type); 6007d96f4efSBrian Somers } 6017d96f4efSBrian Somers } 6027d96f4efSBrian Somers ptr = ptr->next; 6037d96f4efSBrian Somers } 6047d96f4efSBrian Somers 605ffcb611aSDag-Erling Smørgrav return (0); 6067d96f4efSBrian Somers } 6077d96f4efSBrian Somers 6087d96f4efSBrian Somers void 609ed01a582SDag-Erling Smørgrav ProxyModify(struct libalias *la, struct alias_link *lnk, 6107d96f4efSBrian Somers struct ip *pip, 6117d96f4efSBrian Somers int maxpacketsize, 6127d96f4efSBrian Somers int proxy_type) 6137d96f4efSBrian Somers { 614ed01a582SDag-Erling Smørgrav 615ed01a582SDag-Erling Smørgrav (void)la; 616ed01a582SDag-Erling Smørgrav 617f0f93429SDag-Erling Smørgrav switch (proxy_type) { 6187d96f4efSBrian Somers case PROXY_TYPE_ENCODE_IPHDR: 6197d96f4efSBrian Somers ProxyEncodeIpHeader(pip, maxpacketsize); 6207d96f4efSBrian Somers break; 6217d96f4efSBrian Somers 6227d96f4efSBrian Somers case PROXY_TYPE_ENCODE_TCPSTREAM: 623ed01a582SDag-Erling Smørgrav ProxyEncodeTcpStream(lnk, pip, maxpacketsize); 6247d96f4efSBrian Somers break; 6257d96f4efSBrian Somers } 6267d96f4efSBrian Somers } 6277d96f4efSBrian Somers 6287d96f4efSBrian Somers 6297d96f4efSBrian Somers /* 6307d96f4efSBrian Somers Public API functions 6317d96f4efSBrian Somers */ 6327d96f4efSBrian Somers 6337d96f4efSBrian Somers int 6345e289f9eSPoul-Henning Kamp LibAliasProxyRule(struct libalias *la, const char *cmd) 6357d96f4efSBrian Somers { 6367d96f4efSBrian Somers /* 6377d96f4efSBrian Somers * This function takes command strings of the form: 6387d96f4efSBrian Somers * 6397d96f4efSBrian Somers * server <addr>[:<port>] 6407d96f4efSBrian Somers * [port <port>] 6417d96f4efSBrian Somers * [rule n] 6427d96f4efSBrian Somers * [proto tcp|udp] 6437d96f4efSBrian Somers * [src <addr>[/n]] 6447d96f4efSBrian Somers * [dst <addr>[/n]] 6457d96f4efSBrian Somers * [type encode_tcp_stream|encode_ip_hdr|no_encode] 6467d96f4efSBrian Somers * 6477d96f4efSBrian Somers * delete <rule number> 6487d96f4efSBrian Somers * 6497d96f4efSBrian Somers * Subfields can be in arbitrary order. Port numbers and addresses 6507d96f4efSBrian Somers * must be in either numeric or symbolic form. An optional rule number 6517d96f4efSBrian Somers * is used to control the order in which rules are searched. If two 6527d96f4efSBrian Somers * rules have the same number, then search order cannot be guaranteed, 6537d96f4efSBrian Somers * and the rules should be disjoint. If no rule number is specified, 6547d96f4efSBrian Somers * then 0 is used, and group 0 rules are always checked before any 6557d96f4efSBrian Somers * others. 6567d96f4efSBrian Somers */ 6577d96f4efSBrian Somers int i, n, len; 6587d96f4efSBrian Somers int cmd_len; 6597d96f4efSBrian Somers int token_count; 6607d96f4efSBrian Somers int state; 6617d96f4efSBrian Somers char *token; 6627d96f4efSBrian Somers char buffer[256]; 6637d96f4efSBrian Somers char str_port[sizeof(buffer)]; 6647d96f4efSBrian Somers char str_server_port[sizeof(buffer)]; 665d9e630b5SAndrey A. Chernov char *res = buffer; 6667d96f4efSBrian Somers 6677d96f4efSBrian Somers int rule_index; 6687d96f4efSBrian Somers int proto; 6697d96f4efSBrian Somers int proxy_type; 6707d96f4efSBrian Somers int proxy_port; 6717d96f4efSBrian Somers int server_port; 6727d96f4efSBrian Somers struct in_addr server_addr; 6737d96f4efSBrian Somers struct in_addr src_addr, src_mask; 6747d96f4efSBrian Somers struct in_addr dst_addr, dst_mask; 6757d96f4efSBrian Somers struct proxy_entry *proxy_entry; 6767d96f4efSBrian Somers 6777d96f4efSBrian Somers /* Copy command line into a buffer */ 6784834b77dSBrian Somers cmd += strspn(cmd, " \t"); 6797d96f4efSBrian Somers cmd_len = strlen(cmd); 680ed01a582SDag-Erling Smørgrav if (cmd_len > (int)(sizeof(buffer) - 1)) 681ffcb611aSDag-Erling Smørgrav return (-1); 6827d96f4efSBrian Somers strcpy(buffer, cmd); 6837d96f4efSBrian Somers 6847d96f4efSBrian Somers /* Convert to lower case */ 6857d96f4efSBrian Somers len = strlen(buffer); 6867d96f4efSBrian Somers for (i = 0; i < len; i++) 687d9e630b5SAndrey A. Chernov buffer[i] = tolower((unsigned char)buffer[i]); 6887d96f4efSBrian Somers 6897d96f4efSBrian Somers /* Set default proxy type */ 6907d96f4efSBrian Somers 6917d96f4efSBrian Somers /* Set up default values */ 6927d96f4efSBrian Somers rule_index = 0; 6937d96f4efSBrian Somers proxy_type = PROXY_TYPE_ENCODE_NONE; 6947d96f4efSBrian Somers proto = IPPROTO_TCP; 6957d96f4efSBrian Somers proxy_port = 0; 6967d96f4efSBrian Somers server_addr.s_addr = 0; 6977d96f4efSBrian Somers server_port = 0; 6987d96f4efSBrian Somers src_addr.s_addr = 0; 6997d96f4efSBrian Somers IpMask(0, &src_mask); 7007d96f4efSBrian Somers dst_addr.s_addr = 0; 7017d96f4efSBrian Somers IpMask(0, &dst_mask); 7027d96f4efSBrian Somers 7037d96f4efSBrian Somers str_port[0] = 0; 7047d96f4efSBrian Somers str_server_port[0] = 0; 7057d96f4efSBrian Somers 7067d96f4efSBrian Somers /* Parse command string with state machine */ 7077d96f4efSBrian Somers #define STATE_READ_KEYWORD 0 7087d96f4efSBrian Somers #define STATE_READ_TYPE 1 7097d96f4efSBrian Somers #define STATE_READ_PORT 2 7107d96f4efSBrian Somers #define STATE_READ_SERVER 3 7117d96f4efSBrian Somers #define STATE_READ_RULE 4 7127d96f4efSBrian Somers #define STATE_READ_DELETE 5 7137d96f4efSBrian Somers #define STATE_READ_PROTO 6 7147d96f4efSBrian Somers #define STATE_READ_SRC 7 7157d96f4efSBrian Somers #define STATE_READ_DST 8 7167d96f4efSBrian Somers state = STATE_READ_KEYWORD; 717d9e630b5SAndrey A. Chernov token = strsep(&res, " \t"); 7187d96f4efSBrian Somers token_count = 0; 719f0f93429SDag-Erling Smørgrav while (token != NULL) { 7207d96f4efSBrian Somers token_count++; 721f0f93429SDag-Erling Smørgrav switch (state) { 7227d96f4efSBrian Somers case STATE_READ_KEYWORD: 7237d96f4efSBrian Somers if (strcmp(token, "type") == 0) 7247d96f4efSBrian Somers state = STATE_READ_TYPE; 7257d96f4efSBrian Somers else if (strcmp(token, "port") == 0) 7267d96f4efSBrian Somers state = STATE_READ_PORT; 7277d96f4efSBrian Somers else if (strcmp(token, "server") == 0) 7287d96f4efSBrian Somers state = STATE_READ_SERVER; 7297d96f4efSBrian Somers else if (strcmp(token, "rule") == 0) 7307d96f4efSBrian Somers state = STATE_READ_RULE; 7317d96f4efSBrian Somers else if (strcmp(token, "delete") == 0) 7327d96f4efSBrian Somers state = STATE_READ_DELETE; 7337d96f4efSBrian Somers else if (strcmp(token, "proto") == 0) 7347d96f4efSBrian Somers state = STATE_READ_PROTO; 7357d96f4efSBrian Somers else if (strcmp(token, "src") == 0) 7367d96f4efSBrian Somers state = STATE_READ_SRC; 7377d96f4efSBrian Somers else if (strcmp(token, "dst") == 0) 7387d96f4efSBrian Somers state = STATE_READ_DST; 7397d96f4efSBrian Somers else 740ffcb611aSDag-Erling Smørgrav return (-1); 7417d96f4efSBrian Somers break; 7427d96f4efSBrian Somers 7437d96f4efSBrian Somers case STATE_READ_TYPE: 7447d96f4efSBrian Somers if (strcmp(token, "encode_ip_hdr") == 0) 7457d96f4efSBrian Somers proxy_type = PROXY_TYPE_ENCODE_IPHDR; 7467d96f4efSBrian Somers else if (strcmp(token, "encode_tcp_stream") == 0) 7477d96f4efSBrian Somers proxy_type = PROXY_TYPE_ENCODE_TCPSTREAM; 7487d96f4efSBrian Somers else if (strcmp(token, "no_encode") == 0) 7497d96f4efSBrian Somers proxy_type = PROXY_TYPE_ENCODE_NONE; 7507d96f4efSBrian Somers else 751ffcb611aSDag-Erling Smørgrav return (-1); 7527d96f4efSBrian Somers state = STATE_READ_KEYWORD; 7537d96f4efSBrian Somers break; 7547d96f4efSBrian Somers 7557d96f4efSBrian Somers case STATE_READ_PORT: 7567d96f4efSBrian Somers strcpy(str_port, token); 7577d96f4efSBrian Somers state = STATE_READ_KEYWORD; 7587d96f4efSBrian Somers break; 7597d96f4efSBrian Somers 7607d96f4efSBrian Somers case STATE_READ_SERVER: 7617d96f4efSBrian Somers { 7627d96f4efSBrian Somers int err; 7637d96f4efSBrian Somers char *p; 7647d96f4efSBrian Somers char s[sizeof(buffer)]; 7657d96f4efSBrian Somers 7667d96f4efSBrian Somers p = token; 7677d96f4efSBrian Somers while (*p != ':' && *p != 0) 7687d96f4efSBrian Somers p++; 7697d96f4efSBrian Somers 770f0f93429SDag-Erling Smørgrav if (*p != ':') { 7717d96f4efSBrian Somers err = IpAddr(token, &server_addr); 7727d96f4efSBrian Somers if (err) 773ffcb611aSDag-Erling Smørgrav return (-1); 774f0f93429SDag-Erling Smørgrav } else { 7757d96f4efSBrian Somers *p = ' '; 7767d96f4efSBrian Somers 7777d96f4efSBrian Somers n = sscanf(token, "%s %s", s, str_server_port); 7787d96f4efSBrian Somers if (n != 2) 779ffcb611aSDag-Erling Smørgrav return (-1); 7807d96f4efSBrian Somers 7817d96f4efSBrian Somers err = IpAddr(s, &server_addr); 7827d96f4efSBrian Somers if (err) 783ffcb611aSDag-Erling Smørgrav return (-1); 7847d96f4efSBrian Somers } 7857d96f4efSBrian Somers } 7867d96f4efSBrian Somers state = STATE_READ_KEYWORD; 7877d96f4efSBrian Somers break; 7887d96f4efSBrian Somers 7897d96f4efSBrian Somers case STATE_READ_RULE: 7907d96f4efSBrian Somers n = sscanf(token, "%d", &rule_index); 7917d96f4efSBrian Somers if (n != 1 || rule_index < 0) 792ffcb611aSDag-Erling Smørgrav return (-1); 7937d96f4efSBrian Somers state = STATE_READ_KEYWORD; 7947d96f4efSBrian Somers break; 7957d96f4efSBrian Somers 7967d96f4efSBrian Somers case STATE_READ_DELETE: 7977d96f4efSBrian Somers { 7987d96f4efSBrian Somers int err; 7997d96f4efSBrian Somers int rule_to_delete; 8007d96f4efSBrian Somers 8017d96f4efSBrian Somers if (token_count != 2) 802ffcb611aSDag-Erling Smørgrav return (-1); 8037d96f4efSBrian Somers 8047d96f4efSBrian Somers n = sscanf(token, "%d", &rule_to_delete); 8057d96f4efSBrian Somers if (n != 1) 806ffcb611aSDag-Erling Smørgrav return (-1); 8075e289f9eSPoul-Henning Kamp err = RuleNumberDelete(la, rule_to_delete); 8087d96f4efSBrian Somers if (err) 809ffcb611aSDag-Erling Smørgrav return (-1); 810ffcb611aSDag-Erling Smørgrav return (0); 8117d96f4efSBrian Somers } 8127d96f4efSBrian Somers 8137d96f4efSBrian Somers case STATE_READ_PROTO: 8147d96f4efSBrian Somers if (strcmp(token, "tcp") == 0) 8157d96f4efSBrian Somers proto = IPPROTO_TCP; 8167d96f4efSBrian Somers else if (strcmp(token, "udp") == 0) 8177d96f4efSBrian Somers proto = IPPROTO_UDP; 8187d96f4efSBrian Somers else 819ffcb611aSDag-Erling Smørgrav return (-1); 8207d96f4efSBrian Somers state = STATE_READ_KEYWORD; 8217d96f4efSBrian Somers break; 8227d96f4efSBrian Somers 8237d96f4efSBrian Somers case STATE_READ_SRC: 8247d96f4efSBrian Somers case STATE_READ_DST: 8257d96f4efSBrian Somers { 8267d96f4efSBrian Somers int err; 8277d96f4efSBrian Somers char *p; 8287d96f4efSBrian Somers struct in_addr mask; 8297d96f4efSBrian Somers struct in_addr addr; 8307d96f4efSBrian Somers 8317d96f4efSBrian Somers p = token; 8327d96f4efSBrian Somers while (*p != '/' && *p != 0) 8337d96f4efSBrian Somers p++; 8347d96f4efSBrian Somers 835f0f93429SDag-Erling Smørgrav if (*p != '/') { 8367d96f4efSBrian Somers IpMask(32, &mask); 8377d96f4efSBrian Somers err = IpAddr(token, &addr); 8387d96f4efSBrian Somers if (err) 839ffcb611aSDag-Erling Smørgrav return (-1); 840f0f93429SDag-Erling Smørgrav } else { 8417d96f4efSBrian Somers int nbits; 8427d96f4efSBrian Somers char s[sizeof(buffer)]; 8437d96f4efSBrian Somers 8447d96f4efSBrian Somers *p = ' '; 8457d96f4efSBrian Somers n = sscanf(token, "%s %d", s, &nbits); 8467d96f4efSBrian Somers if (n != 2) 847ffcb611aSDag-Erling Smørgrav return (-1); 8487d96f4efSBrian Somers 8497d96f4efSBrian Somers err = IpAddr(s, &addr); 8507d96f4efSBrian Somers if (err) 851ffcb611aSDag-Erling Smørgrav return (-1); 8527d96f4efSBrian Somers 8537d96f4efSBrian Somers err = IpMask(nbits, &mask); 8547d96f4efSBrian Somers if (err) 855ffcb611aSDag-Erling Smørgrav return (-1); 8567d96f4efSBrian Somers } 8577d96f4efSBrian Somers 858f0f93429SDag-Erling Smørgrav if (state == STATE_READ_SRC) { 8597d96f4efSBrian Somers src_addr = addr; 8607d96f4efSBrian Somers src_mask = mask; 861f0f93429SDag-Erling Smørgrav } else { 8627d96f4efSBrian Somers dst_addr = addr; 8637d96f4efSBrian Somers dst_mask = mask; 8647d96f4efSBrian Somers } 8657d96f4efSBrian Somers } 8667d96f4efSBrian Somers state = STATE_READ_KEYWORD; 8677d96f4efSBrian Somers break; 8687d96f4efSBrian Somers 8697d96f4efSBrian Somers default: 870ffcb611aSDag-Erling Smørgrav return (-1); 8717d96f4efSBrian Somers break; 8727d96f4efSBrian Somers } 8737d96f4efSBrian Somers 874d9e630b5SAndrey A. Chernov do { 875d9e630b5SAndrey A. Chernov token = strsep(&res, " \t"); 876d9e630b5SAndrey A. Chernov } while (token != NULL && !*token); 8777d96f4efSBrian Somers } 8787d96f4efSBrian Somers #undef STATE_READ_KEYWORD 8797d96f4efSBrian Somers #undef STATE_READ_TYPE 8807d96f4efSBrian Somers #undef STATE_READ_PORT 8817d96f4efSBrian Somers #undef STATE_READ_SERVER 8827d96f4efSBrian Somers #undef STATE_READ_RULE 8837d96f4efSBrian Somers #undef STATE_READ_DELETE 8847d96f4efSBrian Somers #undef STATE_READ_PROTO 8857d96f4efSBrian Somers #undef STATE_READ_SRC 8867d96f4efSBrian Somers #undef STATE_READ_DST 8877d96f4efSBrian Somers 8887d96f4efSBrian Somers /* Convert port strings to numbers. This needs to be done after 8897d96f4efSBrian Somers the string is parsed, because the prototype might not be designated 8907d96f4efSBrian Somers before the ports (which might be symbolic entries in /etc/services) */ 8917d96f4efSBrian Somers 892f0f93429SDag-Erling Smørgrav if (strlen(str_port) != 0) { 8937d96f4efSBrian Somers int err; 8947d96f4efSBrian Somers 8957d96f4efSBrian Somers err = IpPort(str_port, proto, &proxy_port); 8967d96f4efSBrian Somers if (err) 897ffcb611aSDag-Erling Smørgrav return (-1); 898f0f93429SDag-Erling Smørgrav } else { 8997d96f4efSBrian Somers proxy_port = 0; 9007d96f4efSBrian Somers } 9017d96f4efSBrian Somers 902f0f93429SDag-Erling Smørgrav if (strlen(str_server_port) != 0) { 9037d96f4efSBrian Somers int err; 9047d96f4efSBrian Somers 9057d96f4efSBrian Somers err = IpPort(str_server_port, proto, &server_port); 9067d96f4efSBrian Somers if (err) 907ffcb611aSDag-Erling Smørgrav return (-1); 908f0f93429SDag-Erling Smørgrav } else { 9097d96f4efSBrian Somers server_port = 0; 9107d96f4efSBrian Somers } 9117d96f4efSBrian Somers 9127d96f4efSBrian Somers /* Check that at least the server address has been defined */ 9137d96f4efSBrian Somers if (server_addr.s_addr == 0) 914ffcb611aSDag-Erling Smørgrav return (-1); 9157d96f4efSBrian Somers 9167d96f4efSBrian Somers /* Add to linked list */ 9177d96f4efSBrian Somers proxy_entry = malloc(sizeof(struct proxy_entry)); 9187d96f4efSBrian Somers if (proxy_entry == NULL) 919ffcb611aSDag-Erling Smørgrav return (-1); 9207d96f4efSBrian Somers 9217d96f4efSBrian Somers proxy_entry->proxy_type = proxy_type; 9227d96f4efSBrian Somers proxy_entry->rule_index = rule_index; 9237d96f4efSBrian Somers proxy_entry->proto = proto; 9247d96f4efSBrian Somers proxy_entry->proxy_port = htons(proxy_port); 9257d96f4efSBrian Somers proxy_entry->server_port = htons(server_port); 9267d96f4efSBrian Somers proxy_entry->server_addr = server_addr; 9277d96f4efSBrian Somers proxy_entry->src_addr.s_addr = src_addr.s_addr & src_mask.s_addr; 9287d96f4efSBrian Somers proxy_entry->dst_addr.s_addr = dst_addr.s_addr & dst_mask.s_addr; 9297d96f4efSBrian Somers proxy_entry->src_mask = src_mask; 9307d96f4efSBrian Somers proxy_entry->dst_mask = dst_mask; 9317d96f4efSBrian Somers 9325e289f9eSPoul-Henning Kamp RuleAdd(la, proxy_entry); 9337d96f4efSBrian Somers 934ffcb611aSDag-Erling Smørgrav return (0); 9357d96f4efSBrian Somers } 936