1f987e1bdSBrian Somers /*- 2e83aaae3SBrian Somers * Copyright (c) 2001 Charles Mott <cm@linktel.net> 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) { 178e9d5db28SGleb Smirnoff val = strtoul(c, &endptr, 0); 179e9d5db28SGleb Smirnoff 180e9d5db28SGleb Smirnoff if (val == ULONG_MAX || val == 0) 181e9d5db28SGleb Smirnoff return (0); 182e9d5db28SGleb Smirnoff 183e9d5db28SGleb Smirnoff /* 184e9d5db28SGleb Smirnoff * If the whole string is invalid, endptr will equal 185e9d5db28SGleb Smirnoff * c.. this way we can make sure someone hasn't 186e9d5db28SGleb Smirnoff * gone '.12' or something which would get past 187e9d5db28SGleb Smirnoff * the next check. 188e9d5db28SGleb Smirnoff */ 189e9d5db28SGleb Smirnoff if (endptr == c) 190e9d5db28SGleb Smirnoff return (0); 191e9d5db28SGleb Smirnoff parts[n] = val; 192e9d5db28SGleb Smirnoff c = endptr; 193e9d5db28SGleb Smirnoff 194e9d5db28SGleb Smirnoff /* Check the next character past the previous number's end */ 195e9d5db28SGleb Smirnoff switch (*c) { 196e9d5db28SGleb Smirnoff case '.' : 197e9d5db28SGleb Smirnoff /* Make sure we only do 3 dots .. */ 198e9d5db28SGleb Smirnoff if (n == 3) /* Whoops. Quit. */ 199e9d5db28SGleb Smirnoff return (0); 200e9d5db28SGleb Smirnoff n++; 201e9d5db28SGleb Smirnoff c++; 202e9d5db28SGleb Smirnoff break; 203e9d5db28SGleb Smirnoff 204e9d5db28SGleb Smirnoff case '\0': 205e9d5db28SGleb Smirnoff gotend = 1; 206e9d5db28SGleb Smirnoff break; 207e9d5db28SGleb Smirnoff 208e9d5db28SGleb Smirnoff default: 209e9d5db28SGleb Smirnoff if (isspace((unsigned char)*c)) { 210e9d5db28SGleb Smirnoff gotend = 1; 211e9d5db28SGleb Smirnoff break; 212e9d5db28SGleb Smirnoff } else 213e9d5db28SGleb Smirnoff return (0); /* Invalid character, so fail */ 214e9d5db28SGleb Smirnoff } 215e9d5db28SGleb Smirnoff 216e9d5db28SGleb Smirnoff } 217e9d5db28SGleb Smirnoff 218e9d5db28SGleb Smirnoff /* 219e9d5db28SGleb Smirnoff * Concoct the address according to 220e9d5db28SGleb Smirnoff * the number of parts specified. 221e9d5db28SGleb Smirnoff */ 222e9d5db28SGleb Smirnoff 223e9d5db28SGleb Smirnoff switch (n) { 224e9d5db28SGleb Smirnoff case 0: /* a -- 32 bits */ 225e9d5db28SGleb Smirnoff /* 226e9d5db28SGleb Smirnoff * Nothing is necessary here. Overflow checking was 227e9d5db28SGleb Smirnoff * already done in strtoul(). 228e9d5db28SGleb Smirnoff */ 229e9d5db28SGleb Smirnoff break; 230e9d5db28SGleb Smirnoff case 1: /* a.b -- 8.24 bits */ 231e9d5db28SGleb Smirnoff if (val > 0xffffff || parts[0] > 0xff) 232e9d5db28SGleb Smirnoff return (0); 233e9d5db28SGleb Smirnoff val |= parts[0] << 24; 234e9d5db28SGleb Smirnoff break; 235e9d5db28SGleb Smirnoff 236e9d5db28SGleb Smirnoff case 2: /* a.b.c -- 8.8.16 bits */ 237e9d5db28SGleb Smirnoff if (val > 0xffff || parts[0] > 0xff || parts[1] > 0xff) 238e9d5db28SGleb Smirnoff return (0); 239e9d5db28SGleb Smirnoff val |= (parts[0] << 24) | (parts[1] << 16); 240e9d5db28SGleb Smirnoff break; 241e9d5db28SGleb Smirnoff 242e9d5db28SGleb Smirnoff case 3: /* a.b.c.d -- 8.8.8.8 bits */ 243e9d5db28SGleb Smirnoff if (val > 0xff || parts[0] > 0xff || parts[1] > 0xff || 244e9d5db28SGleb Smirnoff parts[2] > 0xff) 245e9d5db28SGleb Smirnoff return (0); 246e9d5db28SGleb Smirnoff val |= (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8); 247e9d5db28SGleb Smirnoff break; 248e9d5db28SGleb Smirnoff } 249e9d5db28SGleb Smirnoff 250e9d5db28SGleb Smirnoff if (addr != NULL) 251e9d5db28SGleb Smirnoff addr->s_addr = htonl(val); 252e9d5db28SGleb Smirnoff return (1); 253e9d5db28SGleb Smirnoff } 254e9d5db28SGleb Smirnoff #endif 255e9d5db28SGleb Smirnoff 2567d96f4efSBrian Somers static int 2577d96f4efSBrian Somers IpMask(int nbits, struct in_addr *mask) 2587d96f4efSBrian Somers { 2597d96f4efSBrian Somers int i; 2607d96f4efSBrian Somers u_int imask; 2617d96f4efSBrian Somers 2627d96f4efSBrian Somers if (nbits < 0 || nbits > 32) 263ffcb611aSDag-Erling Smørgrav return (-1); 2647d96f4efSBrian Somers 2657d96f4efSBrian Somers imask = 0; 2667d96f4efSBrian Somers for (i = 0; i < nbits; i++) 2677d96f4efSBrian Somers imask = (imask >> 1) + 0x80000000; 2687d96f4efSBrian Somers mask->s_addr = htonl(imask); 2697d96f4efSBrian Somers 270ffcb611aSDag-Erling Smørgrav return (0); 2717d96f4efSBrian Somers } 2727d96f4efSBrian Somers 2737d96f4efSBrian Somers static int 2747d96f4efSBrian Somers IpAddr(char *s, struct in_addr *addr) 2757d96f4efSBrian Somers { 2767d96f4efSBrian Somers if (inet_aton(s, addr) == 0) 277ffcb611aSDag-Erling Smørgrav return (-1); 2787d96f4efSBrian Somers else 279ffcb611aSDag-Erling Smørgrav return (0); 2807d96f4efSBrian Somers } 2817d96f4efSBrian Somers 2827d96f4efSBrian Somers static int 2837d96f4efSBrian Somers IpPort(char *s, int proto, int *port) 2847d96f4efSBrian Somers { 2857d96f4efSBrian Somers int n; 2867d96f4efSBrian Somers 2877d96f4efSBrian Somers n = sscanf(s, "%d", port); 288e9d5db28SGleb Smirnoff if (n != 1) 289e9d5db28SGleb Smirnoff #ifndef _KERNEL /* XXX: we accept only numeric ports in kernel */ 290e9d5db28SGleb Smirnoff { 2917d96f4efSBrian Somers struct servent *se; 2927d96f4efSBrian Somers 2937d96f4efSBrian Somers if (proto == IPPROTO_TCP) 2947d96f4efSBrian Somers se = getservbyname(s, "tcp"); 2957d96f4efSBrian Somers else if (proto == IPPROTO_UDP) 2967d96f4efSBrian Somers se = getservbyname(s, "udp"); 2977d96f4efSBrian Somers else 298ffcb611aSDag-Erling Smørgrav return (-1); 2997d96f4efSBrian Somers 3007d96f4efSBrian Somers if (se == NULL) 301ffcb611aSDag-Erling Smørgrav return (-1); 3027d96f4efSBrian Somers 3037d96f4efSBrian Somers *port = (u_int) ntohs(se->s_port); 3047d96f4efSBrian Somers } 305e9d5db28SGleb Smirnoff #else 306e9d5db28SGleb Smirnoff return (-1); 307e9d5db28SGleb Smirnoff #endif 308ffcb611aSDag-Erling Smørgrav return (0); 3097d96f4efSBrian Somers } 3107d96f4efSBrian Somers 3117d96f4efSBrian Somers void 3125e289f9eSPoul-Henning Kamp RuleAdd(struct libalias *la, struct proxy_entry *entry) 3137d96f4efSBrian Somers { 3147d96f4efSBrian Somers int rule_index; 3157d96f4efSBrian Somers struct proxy_entry *ptr; 3167d96f4efSBrian Somers struct proxy_entry *ptr_last; 3177d96f4efSBrian Somers 318f0f93429SDag-Erling Smørgrav if (la->proxyList == NULL) { 3195e289f9eSPoul-Henning Kamp la->proxyList = entry; 3207d96f4efSBrian Somers entry->last = NULL; 3217d96f4efSBrian Somers entry->next = NULL; 3227d96f4efSBrian Somers return; 3237d96f4efSBrian Somers } 3245e289f9eSPoul-Henning Kamp entry->la = la; 3257d96f4efSBrian Somers 3267d96f4efSBrian Somers rule_index = entry->rule_index; 3275e289f9eSPoul-Henning Kamp ptr = la->proxyList; 3287d96f4efSBrian Somers ptr_last = NULL; 329f0f93429SDag-Erling Smørgrav while (ptr != NULL) { 330f0f93429SDag-Erling Smørgrav if (ptr->rule_index >= rule_index) { 331f0f93429SDag-Erling Smørgrav if (ptr_last == NULL) { 3325e289f9eSPoul-Henning Kamp entry->next = la->proxyList; 3337d96f4efSBrian Somers entry->last = NULL; 3345e289f9eSPoul-Henning Kamp la->proxyList->last = entry; 3355e289f9eSPoul-Henning Kamp la->proxyList = entry; 3367d96f4efSBrian Somers return; 3377d96f4efSBrian Somers } 3387d96f4efSBrian Somers ptr_last->next = entry; 3397d96f4efSBrian Somers ptr->last = entry; 3407d96f4efSBrian Somers entry->last = ptr->last; 3417d96f4efSBrian Somers entry->next = ptr; 3427d96f4efSBrian Somers return; 3437d96f4efSBrian Somers } 3447d96f4efSBrian Somers ptr_last = ptr; 3457d96f4efSBrian Somers ptr = ptr->next; 3467d96f4efSBrian Somers } 3477d96f4efSBrian Somers 3487d96f4efSBrian Somers ptr_last->next = entry; 3497d96f4efSBrian Somers entry->last = ptr_last; 3507d96f4efSBrian Somers entry->next = NULL; 3517d96f4efSBrian Somers } 3527d96f4efSBrian Somers 3537d96f4efSBrian Somers static void 3547d96f4efSBrian Somers RuleDelete(struct proxy_entry *entry) 3557d96f4efSBrian Somers { 3565e289f9eSPoul-Henning Kamp struct libalias *la; 3575e289f9eSPoul-Henning Kamp 3585e289f9eSPoul-Henning Kamp la = entry->la; 3597d96f4efSBrian Somers if (entry->last != NULL) 3607d96f4efSBrian Somers entry->last->next = entry->next; 3617d96f4efSBrian Somers else 3625e289f9eSPoul-Henning Kamp la->proxyList = entry->next; 3637d96f4efSBrian Somers 3647d96f4efSBrian Somers if (entry->next != NULL) 3657d96f4efSBrian Somers entry->next->last = entry->last; 3667d96f4efSBrian Somers 3677d96f4efSBrian Somers free(entry); 3687d96f4efSBrian Somers } 3697d96f4efSBrian Somers 3707d96f4efSBrian Somers static int 3715e289f9eSPoul-Henning Kamp RuleNumberDelete(struct libalias *la, int rule_index) 3727d96f4efSBrian Somers { 3737d96f4efSBrian Somers int err; 3747d96f4efSBrian Somers struct proxy_entry *ptr; 3757d96f4efSBrian Somers 3767d96f4efSBrian Somers err = -1; 3775e289f9eSPoul-Henning Kamp ptr = la->proxyList; 378f0f93429SDag-Erling Smørgrav while (ptr != NULL) { 3797d96f4efSBrian Somers struct proxy_entry *ptr_next; 3807d96f4efSBrian Somers 3817d96f4efSBrian Somers ptr_next = ptr->next; 382f0f93429SDag-Erling Smørgrav if (ptr->rule_index == rule_index) { 3837d96f4efSBrian Somers err = 0; 3847d96f4efSBrian Somers RuleDelete(ptr); 3857d96f4efSBrian Somers } 3867d96f4efSBrian Somers ptr = ptr_next; 3877d96f4efSBrian Somers } 3887d96f4efSBrian Somers 389ffcb611aSDag-Erling Smørgrav return (err); 3907d96f4efSBrian Somers } 3917d96f4efSBrian Somers 3927d96f4efSBrian Somers static void 393ed01a582SDag-Erling Smørgrav ProxyEncodeTcpStream(struct alias_link *lnk, 3947d96f4efSBrian Somers struct ip *pip, 3957d96f4efSBrian Somers int maxpacketsize) 3967d96f4efSBrian Somers { 3977d96f4efSBrian Somers int slen; 3987d96f4efSBrian Somers char buffer[40]; 3997d96f4efSBrian Somers struct tcphdr *tc; 4007d96f4efSBrian Somers 4017d96f4efSBrian Somers /* Compute pointer to tcp header */ 4029fa0fd26SDag-Erling Smørgrav tc = (struct tcphdr *)ip_next(pip); 4037d96f4efSBrian Somers 4047d96f4efSBrian Somers /* Don't modify if once already modified */ 4057d96f4efSBrian Somers 406ed01a582SDag-Erling Smørgrav if (GetAckModified(lnk)) 4077d96f4efSBrian Somers return; 4087d96f4efSBrian Somers 4097d96f4efSBrian Somers /* Translate destination address and port to string form */ 4107d96f4efSBrian Somers snprintf(buffer, sizeof(buffer) - 2, "[DEST %s %d]", 411ed01a582SDag-Erling Smørgrav inet_ntoa(GetProxyAddress(lnk)), (u_int) ntohs(GetProxyPort(lnk))); 4127d96f4efSBrian Somers 4137d96f4efSBrian Somers /* Pad string out to a multiple of two in length */ 4147d96f4efSBrian Somers slen = strlen(buffer); 415f0f93429SDag-Erling Smørgrav switch (slen % 2) { 4167d96f4efSBrian Somers case 0: 4177d96f4efSBrian Somers strcat(buffer, " \n"); 4187d96f4efSBrian Somers slen += 2; 4197d96f4efSBrian Somers break; 4207d96f4efSBrian Somers case 1: 4217d96f4efSBrian Somers strcat(buffer, "\n"); 4227d96f4efSBrian Somers slen += 1; 4237d96f4efSBrian Somers } 4247d96f4efSBrian Somers 4257d96f4efSBrian Somers /* Check for packet overflow */ 426ed01a582SDag-Erling Smørgrav if ((int)(ntohs(pip->ip_len) + strlen(buffer)) > maxpacketsize) 4277d96f4efSBrian Somers return; 4287d96f4efSBrian Somers 4297d96f4efSBrian Somers /* Shift existing TCP data and insert destination string */ 4307d96f4efSBrian Somers { 4317d96f4efSBrian Somers int dlen; 4327d96f4efSBrian Somers int hlen; 4337d96f4efSBrian Somers u_char *p; 4347d96f4efSBrian Somers 4357d96f4efSBrian Somers hlen = (pip->ip_hl + tc->th_off) << 2; 4367d96f4efSBrian Somers dlen = ntohs(pip->ip_len) - hlen; 4377d96f4efSBrian Somers 4387d96f4efSBrian Somers /* Modify first packet that has data in it */ 4397d96f4efSBrian Somers 4407d96f4efSBrian Somers if (dlen == 0) 4417d96f4efSBrian Somers return; 4427d96f4efSBrian Somers 4437d96f4efSBrian Somers p = (char *)pip; 4447d96f4efSBrian Somers p += hlen; 4457d96f4efSBrian Somers 446c8d3ca72SGleb Smirnoff bcopy(p, p + slen, dlen); 4477d96f4efSBrian Somers memcpy(p, buffer, slen); 4487d96f4efSBrian Somers } 4497d96f4efSBrian Somers 4507d96f4efSBrian Somers /* Save information about modfied sequence number */ 4517d96f4efSBrian Somers { 4527d96f4efSBrian Somers int delta; 4537d96f4efSBrian Somers 454ed01a582SDag-Erling Smørgrav SetAckModified(lnk); 455ed01a582SDag-Erling Smørgrav delta = GetDeltaSeqOut(pip, lnk); 456ed01a582SDag-Erling Smørgrav AddSeq(pip, lnk, delta + slen); 4577d96f4efSBrian Somers } 4587d96f4efSBrian Somers 4597d96f4efSBrian Somers /* Update IP header packet length and checksum */ 4607d96f4efSBrian Somers { 4617d96f4efSBrian Somers int accumulate; 4627d96f4efSBrian Somers 4637d96f4efSBrian Somers accumulate = pip->ip_len; 4647d96f4efSBrian Somers pip->ip_len = htons(ntohs(pip->ip_len) + slen); 4657d96f4efSBrian Somers accumulate -= pip->ip_len; 4667d96f4efSBrian Somers 4677d96f4efSBrian Somers ADJUST_CHECKSUM(accumulate, pip->ip_sum); 4687d96f4efSBrian Somers } 4697d96f4efSBrian Somers 4707d96f4efSBrian Somers /* Update TCP checksum, Use TcpChecksum since so many things have 4717d96f4efSBrian Somers already changed. */ 4727d96f4efSBrian Somers 4737d96f4efSBrian Somers tc->th_sum = 0; 4747d96f4efSBrian Somers tc->th_sum = TcpChecksum(pip); 4757d96f4efSBrian Somers } 4767d96f4efSBrian Somers 4777d96f4efSBrian Somers static void 4787d96f4efSBrian Somers ProxyEncodeIpHeader(struct ip *pip, 4797d96f4efSBrian Somers int maxpacketsize) 4807d96f4efSBrian Somers { 4817d96f4efSBrian Somers #define OPTION_LEN_BYTES 8 4827d96f4efSBrian Somers #define OPTION_LEN_INT16 4 4837d96f4efSBrian Somers #define OPTION_LEN_INT32 2 4847d96f4efSBrian Somers u_char option[OPTION_LEN_BYTES]; 4857d96f4efSBrian Somers 4864c32f5d2SBrian Somers #ifdef DEBUG 4877d96f4efSBrian Somers fprintf(stdout, " ip cksum 1 = %x\n", (u_int) IpChecksum(pip)); 4887d96f4efSBrian Somers fprintf(stdout, "tcp cksum 1 = %x\n", (u_int) TcpChecksum(pip)); 4894c32f5d2SBrian Somers #endif 4907d96f4efSBrian Somers 491ed01a582SDag-Erling Smørgrav (void)maxpacketsize; 492ed01a582SDag-Erling Smørgrav 4937d96f4efSBrian Somers /* Check to see that there is room to add an IP option */ 4947d96f4efSBrian Somers if (pip->ip_hl > (0x0f - OPTION_LEN_INT32)) 4957d96f4efSBrian Somers return; 4967d96f4efSBrian Somers 4977d96f4efSBrian Somers /* Build option and copy into packet */ 4987d96f4efSBrian Somers { 4997d96f4efSBrian Somers u_char *ptr; 5007d96f4efSBrian Somers struct tcphdr *tc; 5017d96f4efSBrian Somers 5027d96f4efSBrian Somers ptr = (u_char *) pip; 5037d96f4efSBrian Somers ptr += 20; 5047d96f4efSBrian Somers memcpy(ptr + OPTION_LEN_BYTES, ptr, ntohs(pip->ip_len) - 20); 5057d96f4efSBrian Somers 5067d96f4efSBrian Somers option[0] = 0x64; /* class: 3 (reserved), option 4 */ 5077d96f4efSBrian Somers option[1] = OPTION_LEN_BYTES; 5087d96f4efSBrian Somers 5097d96f4efSBrian Somers memcpy(&option[2], (u_char *) & pip->ip_dst, 4); 5107d96f4efSBrian Somers 5119fa0fd26SDag-Erling Smørgrav tc = (struct tcphdr *)ip_next(pip); 5127d96f4efSBrian Somers memcpy(&option[6], (u_char *) & tc->th_sport, 2); 5137d96f4efSBrian Somers 5147d96f4efSBrian Somers memcpy(ptr, option, 8); 5157d96f4efSBrian Somers } 5167d96f4efSBrian Somers 5177d96f4efSBrian Somers /* Update checksum, header length and packet length */ 5187d96f4efSBrian Somers { 5197d96f4efSBrian Somers int i; 5207d96f4efSBrian Somers int accumulate; 5217d96f4efSBrian Somers u_short *sptr; 5227d96f4efSBrian Somers 5237d96f4efSBrian Somers sptr = (u_short *) option; 5247d96f4efSBrian Somers accumulate = 0; 5257d96f4efSBrian Somers for (i = 0; i < OPTION_LEN_INT16; i++) 5267d96f4efSBrian Somers accumulate -= *(sptr++); 5277d96f4efSBrian Somers 5287d96f4efSBrian Somers sptr = (u_short *) pip; 5297d96f4efSBrian Somers accumulate += *sptr; 5307d96f4efSBrian Somers pip->ip_hl += OPTION_LEN_INT32; 5317d96f4efSBrian Somers accumulate -= *sptr; 5327d96f4efSBrian Somers 5337d96f4efSBrian Somers accumulate += pip->ip_len; 5347d96f4efSBrian Somers pip->ip_len = htons(ntohs(pip->ip_len) + OPTION_LEN_BYTES); 5357d96f4efSBrian Somers accumulate -= pip->ip_len; 5367d96f4efSBrian Somers 5377d96f4efSBrian Somers ADJUST_CHECKSUM(accumulate, pip->ip_sum); 5387d96f4efSBrian Somers } 5397d96f4efSBrian Somers #undef OPTION_LEN_BYTES 5407d96f4efSBrian Somers #undef OPTION_LEN_INT16 5417d96f4efSBrian Somers #undef OPTION_LEN_INT32 5424c32f5d2SBrian Somers #ifdef DEBUG 5437d96f4efSBrian Somers fprintf(stdout, " ip cksum 2 = %x\n", (u_int) IpChecksum(pip)); 5447d96f4efSBrian Somers fprintf(stdout, "tcp cksum 2 = %x\n", (u_int) TcpChecksum(pip)); 5454c32f5d2SBrian Somers #endif 5467d96f4efSBrian Somers } 5477d96f4efSBrian Somers 5487d96f4efSBrian Somers 5497d96f4efSBrian Somers /* Functions by other packet alias source files 5507d96f4efSBrian Somers 5517d96f4efSBrian Somers ProxyCheck() -- Checks whether an outgoing packet should 5527d96f4efSBrian Somers be proxied. 5537d96f4efSBrian Somers ProxyModify() -- Encodes the original destination address/port 5547d96f4efSBrian Somers for a packet which is to be redirected to 5557d96f4efSBrian Somers a proxy server. 5567d96f4efSBrian Somers */ 5577d96f4efSBrian Somers 5587d96f4efSBrian Somers int 5595e289f9eSPoul-Henning Kamp ProxyCheck(struct libalias *la, struct ip *pip, 5607d96f4efSBrian Somers struct in_addr *proxy_server_addr, 5617d96f4efSBrian Somers u_short * proxy_server_port) 5627d96f4efSBrian Somers { 5637d96f4efSBrian Somers u_short dst_port; 5647d96f4efSBrian Somers struct in_addr src_addr; 5657d96f4efSBrian Somers struct in_addr dst_addr; 5667d96f4efSBrian Somers struct proxy_entry *ptr; 5677d96f4efSBrian Somers 5687d96f4efSBrian Somers src_addr = pip->ip_src; 5697d96f4efSBrian Somers dst_addr = pip->ip_dst; 5709fa0fd26SDag-Erling Smørgrav dst_port = ((struct tcphdr *)ip_next(pip)) 5717d96f4efSBrian Somers ->th_dport; 5727d96f4efSBrian Somers 5735e289f9eSPoul-Henning Kamp ptr = la->proxyList; 574f0f93429SDag-Erling Smørgrav while (ptr != NULL) { 5757d96f4efSBrian Somers u_short proxy_port; 5767d96f4efSBrian Somers 5777d96f4efSBrian Somers proxy_port = ptr->proxy_port; 5787d96f4efSBrian Somers if ((dst_port == proxy_port || proxy_port == 0) 5797d96f4efSBrian Somers && pip->ip_p == ptr->proto 580f0f93429SDag-Erling Smørgrav && src_addr.s_addr != ptr->server_addr.s_addr) { 5817d96f4efSBrian Somers struct in_addr src_addr_masked; 5827d96f4efSBrian Somers struct in_addr dst_addr_masked; 5837d96f4efSBrian Somers 5847d96f4efSBrian Somers src_addr_masked.s_addr = src_addr.s_addr & ptr->src_mask.s_addr; 5857d96f4efSBrian Somers dst_addr_masked.s_addr = dst_addr.s_addr & ptr->dst_mask.s_addr; 5867d96f4efSBrian Somers 5877d96f4efSBrian Somers if ((src_addr_masked.s_addr == ptr->src_addr.s_addr) 588f0f93429SDag-Erling Smørgrav && (dst_addr_masked.s_addr == ptr->dst_addr.s_addr)) { 5897d96f4efSBrian Somers if ((*proxy_server_port = ptr->server_port) == 0) 5907d96f4efSBrian Somers *proxy_server_port = dst_port; 5917d96f4efSBrian Somers *proxy_server_addr = ptr->server_addr; 592ffcb611aSDag-Erling Smørgrav return (ptr->proxy_type); 5937d96f4efSBrian Somers } 5947d96f4efSBrian Somers } 5957d96f4efSBrian Somers ptr = ptr->next; 5967d96f4efSBrian Somers } 5977d96f4efSBrian Somers 598ffcb611aSDag-Erling Smørgrav return (0); 5997d96f4efSBrian Somers } 6007d96f4efSBrian Somers 6017d96f4efSBrian Somers void 602ed01a582SDag-Erling Smørgrav ProxyModify(struct libalias *la, struct alias_link *lnk, 6037d96f4efSBrian Somers struct ip *pip, 6047d96f4efSBrian Somers int maxpacketsize, 6057d96f4efSBrian Somers int proxy_type) 6067d96f4efSBrian Somers { 607ed01a582SDag-Erling Smørgrav 608ed01a582SDag-Erling Smørgrav (void)la; 609ed01a582SDag-Erling Smørgrav 610f0f93429SDag-Erling Smørgrav switch (proxy_type) { 6117d96f4efSBrian Somers case PROXY_TYPE_ENCODE_IPHDR: 6127d96f4efSBrian Somers ProxyEncodeIpHeader(pip, maxpacketsize); 6137d96f4efSBrian Somers break; 6147d96f4efSBrian Somers 6157d96f4efSBrian Somers case PROXY_TYPE_ENCODE_TCPSTREAM: 616ed01a582SDag-Erling Smørgrav ProxyEncodeTcpStream(lnk, pip, maxpacketsize); 6177d96f4efSBrian Somers break; 6187d96f4efSBrian Somers } 6197d96f4efSBrian Somers } 6207d96f4efSBrian Somers 6217d96f4efSBrian Somers 6227d96f4efSBrian Somers /* 6237d96f4efSBrian Somers Public API functions 6247d96f4efSBrian Somers */ 6257d96f4efSBrian Somers 6267d96f4efSBrian Somers int 6275e289f9eSPoul-Henning Kamp LibAliasProxyRule(struct libalias *la, const char *cmd) 6287d96f4efSBrian Somers { 6297d96f4efSBrian Somers /* 6307d96f4efSBrian Somers * This function takes command strings of the form: 6317d96f4efSBrian Somers * 6327d96f4efSBrian Somers * server <addr>[:<port>] 6337d96f4efSBrian Somers * [port <port>] 6347d96f4efSBrian Somers * [rule n] 6357d96f4efSBrian Somers * [proto tcp|udp] 6367d96f4efSBrian Somers * [src <addr>[/n]] 6377d96f4efSBrian Somers * [dst <addr>[/n]] 6387d96f4efSBrian Somers * [type encode_tcp_stream|encode_ip_hdr|no_encode] 6397d96f4efSBrian Somers * 6407d96f4efSBrian Somers * delete <rule number> 6417d96f4efSBrian Somers * 6427d96f4efSBrian Somers * Subfields can be in arbitrary order. Port numbers and addresses 6437d96f4efSBrian Somers * must be in either numeric or symbolic form. An optional rule number 6447d96f4efSBrian Somers * is used to control the order in which rules are searched. If two 6457d96f4efSBrian Somers * rules have the same number, then search order cannot be guaranteed, 6467d96f4efSBrian Somers * and the rules should be disjoint. If no rule number is specified, 6477d96f4efSBrian Somers * then 0 is used, and group 0 rules are always checked before any 6487d96f4efSBrian Somers * others. 6497d96f4efSBrian Somers */ 6507d96f4efSBrian Somers int i, n, len; 6517d96f4efSBrian Somers int cmd_len; 6527d96f4efSBrian Somers int token_count; 6537d96f4efSBrian Somers int state; 6547d96f4efSBrian Somers char *token; 6557d96f4efSBrian Somers char buffer[256]; 6567d96f4efSBrian Somers char str_port[sizeof(buffer)]; 6577d96f4efSBrian Somers char str_server_port[sizeof(buffer)]; 658d9e630b5SAndrey A. Chernov char *res = buffer; 6597d96f4efSBrian Somers 6607d96f4efSBrian Somers int rule_index; 6617d96f4efSBrian Somers int proto; 6627d96f4efSBrian Somers int proxy_type; 6637d96f4efSBrian Somers int proxy_port; 6647d96f4efSBrian Somers int server_port; 6657d96f4efSBrian Somers struct in_addr server_addr; 6667d96f4efSBrian Somers struct in_addr src_addr, src_mask; 6677d96f4efSBrian Somers struct in_addr dst_addr, dst_mask; 6687d96f4efSBrian Somers struct proxy_entry *proxy_entry; 6697d96f4efSBrian Somers 6707d96f4efSBrian Somers /* Copy command line into a buffer */ 6714834b77dSBrian Somers cmd += strspn(cmd, " \t"); 6727d96f4efSBrian Somers cmd_len = strlen(cmd); 673ed01a582SDag-Erling Smørgrav if (cmd_len > (int)(sizeof(buffer) - 1)) 674ffcb611aSDag-Erling Smørgrav return (-1); 6757d96f4efSBrian Somers strcpy(buffer, cmd); 6767d96f4efSBrian Somers 6777d96f4efSBrian Somers /* Convert to lower case */ 6787d96f4efSBrian Somers len = strlen(buffer); 6797d96f4efSBrian Somers for (i = 0; i < len; i++) 680d9e630b5SAndrey A. Chernov buffer[i] = tolower((unsigned char)buffer[i]); 6817d96f4efSBrian Somers 6827d96f4efSBrian Somers /* Set default proxy type */ 6837d96f4efSBrian Somers 6847d96f4efSBrian Somers /* Set up default values */ 6857d96f4efSBrian Somers rule_index = 0; 6867d96f4efSBrian Somers proxy_type = PROXY_TYPE_ENCODE_NONE; 6877d96f4efSBrian Somers proto = IPPROTO_TCP; 6887d96f4efSBrian Somers proxy_port = 0; 6897d96f4efSBrian Somers server_addr.s_addr = 0; 6907d96f4efSBrian Somers server_port = 0; 6917d96f4efSBrian Somers src_addr.s_addr = 0; 6927d96f4efSBrian Somers IpMask(0, &src_mask); 6937d96f4efSBrian Somers dst_addr.s_addr = 0; 6947d96f4efSBrian Somers IpMask(0, &dst_mask); 6957d96f4efSBrian Somers 6967d96f4efSBrian Somers str_port[0] = 0; 6977d96f4efSBrian Somers str_server_port[0] = 0; 6987d96f4efSBrian Somers 6997d96f4efSBrian Somers /* Parse command string with state machine */ 7007d96f4efSBrian Somers #define STATE_READ_KEYWORD 0 7017d96f4efSBrian Somers #define STATE_READ_TYPE 1 7027d96f4efSBrian Somers #define STATE_READ_PORT 2 7037d96f4efSBrian Somers #define STATE_READ_SERVER 3 7047d96f4efSBrian Somers #define STATE_READ_RULE 4 7057d96f4efSBrian Somers #define STATE_READ_DELETE 5 7067d96f4efSBrian Somers #define STATE_READ_PROTO 6 7077d96f4efSBrian Somers #define STATE_READ_SRC 7 7087d96f4efSBrian Somers #define STATE_READ_DST 8 7097d96f4efSBrian Somers state = STATE_READ_KEYWORD; 710d9e630b5SAndrey A. Chernov token = strsep(&res, " \t"); 7117d96f4efSBrian Somers token_count = 0; 712f0f93429SDag-Erling Smørgrav while (token != NULL) { 7137d96f4efSBrian Somers token_count++; 714f0f93429SDag-Erling Smørgrav switch (state) { 7157d96f4efSBrian Somers case STATE_READ_KEYWORD: 7167d96f4efSBrian Somers if (strcmp(token, "type") == 0) 7177d96f4efSBrian Somers state = STATE_READ_TYPE; 7187d96f4efSBrian Somers else if (strcmp(token, "port") == 0) 7197d96f4efSBrian Somers state = STATE_READ_PORT; 7207d96f4efSBrian Somers else if (strcmp(token, "server") == 0) 7217d96f4efSBrian Somers state = STATE_READ_SERVER; 7227d96f4efSBrian Somers else if (strcmp(token, "rule") == 0) 7237d96f4efSBrian Somers state = STATE_READ_RULE; 7247d96f4efSBrian Somers else if (strcmp(token, "delete") == 0) 7257d96f4efSBrian Somers state = STATE_READ_DELETE; 7267d96f4efSBrian Somers else if (strcmp(token, "proto") == 0) 7277d96f4efSBrian Somers state = STATE_READ_PROTO; 7287d96f4efSBrian Somers else if (strcmp(token, "src") == 0) 7297d96f4efSBrian Somers state = STATE_READ_SRC; 7307d96f4efSBrian Somers else if (strcmp(token, "dst") == 0) 7317d96f4efSBrian Somers state = STATE_READ_DST; 7327d96f4efSBrian Somers else 733ffcb611aSDag-Erling Smørgrav return (-1); 7347d96f4efSBrian Somers break; 7357d96f4efSBrian Somers 7367d96f4efSBrian Somers case STATE_READ_TYPE: 7377d96f4efSBrian Somers if (strcmp(token, "encode_ip_hdr") == 0) 7387d96f4efSBrian Somers proxy_type = PROXY_TYPE_ENCODE_IPHDR; 7397d96f4efSBrian Somers else if (strcmp(token, "encode_tcp_stream") == 0) 7407d96f4efSBrian Somers proxy_type = PROXY_TYPE_ENCODE_TCPSTREAM; 7417d96f4efSBrian Somers else if (strcmp(token, "no_encode") == 0) 7427d96f4efSBrian Somers proxy_type = PROXY_TYPE_ENCODE_NONE; 7437d96f4efSBrian Somers else 744ffcb611aSDag-Erling Smørgrav return (-1); 7457d96f4efSBrian Somers state = STATE_READ_KEYWORD; 7467d96f4efSBrian Somers break; 7477d96f4efSBrian Somers 7487d96f4efSBrian Somers case STATE_READ_PORT: 7497d96f4efSBrian Somers strcpy(str_port, token); 7507d96f4efSBrian Somers state = STATE_READ_KEYWORD; 7517d96f4efSBrian Somers break; 7527d96f4efSBrian Somers 7537d96f4efSBrian Somers case STATE_READ_SERVER: 7547d96f4efSBrian Somers { 7557d96f4efSBrian Somers int err; 7567d96f4efSBrian Somers char *p; 7577d96f4efSBrian Somers char s[sizeof(buffer)]; 7587d96f4efSBrian Somers 7597d96f4efSBrian Somers p = token; 7607d96f4efSBrian Somers while (*p != ':' && *p != 0) 7617d96f4efSBrian Somers p++; 7627d96f4efSBrian Somers 763f0f93429SDag-Erling Smørgrav if (*p != ':') { 7647d96f4efSBrian Somers err = IpAddr(token, &server_addr); 7657d96f4efSBrian Somers if (err) 766ffcb611aSDag-Erling Smørgrav return (-1); 767f0f93429SDag-Erling Smørgrav } else { 7687d96f4efSBrian Somers *p = ' '; 7697d96f4efSBrian Somers 7707d96f4efSBrian Somers n = sscanf(token, "%s %s", s, str_server_port); 7717d96f4efSBrian Somers if (n != 2) 772ffcb611aSDag-Erling Smørgrav return (-1); 7737d96f4efSBrian Somers 7747d96f4efSBrian Somers err = IpAddr(s, &server_addr); 7757d96f4efSBrian Somers if (err) 776ffcb611aSDag-Erling Smørgrav return (-1); 7777d96f4efSBrian Somers } 7787d96f4efSBrian Somers } 7797d96f4efSBrian Somers state = STATE_READ_KEYWORD; 7807d96f4efSBrian Somers break; 7817d96f4efSBrian Somers 7827d96f4efSBrian Somers case STATE_READ_RULE: 7837d96f4efSBrian Somers n = sscanf(token, "%d", &rule_index); 7847d96f4efSBrian Somers if (n != 1 || rule_index < 0) 785ffcb611aSDag-Erling Smørgrav return (-1); 7867d96f4efSBrian Somers state = STATE_READ_KEYWORD; 7877d96f4efSBrian Somers break; 7887d96f4efSBrian Somers 7897d96f4efSBrian Somers case STATE_READ_DELETE: 7907d96f4efSBrian Somers { 7917d96f4efSBrian Somers int err; 7927d96f4efSBrian Somers int rule_to_delete; 7937d96f4efSBrian Somers 7947d96f4efSBrian Somers if (token_count != 2) 795ffcb611aSDag-Erling Smørgrav return (-1); 7967d96f4efSBrian Somers 7977d96f4efSBrian Somers n = sscanf(token, "%d", &rule_to_delete); 7987d96f4efSBrian Somers if (n != 1) 799ffcb611aSDag-Erling Smørgrav return (-1); 8005e289f9eSPoul-Henning Kamp err = RuleNumberDelete(la, rule_to_delete); 8017d96f4efSBrian Somers if (err) 802ffcb611aSDag-Erling Smørgrav return (-1); 803ffcb611aSDag-Erling Smørgrav return (0); 8047d96f4efSBrian Somers } 8057d96f4efSBrian Somers 8067d96f4efSBrian Somers case STATE_READ_PROTO: 8077d96f4efSBrian Somers if (strcmp(token, "tcp") == 0) 8087d96f4efSBrian Somers proto = IPPROTO_TCP; 8097d96f4efSBrian Somers else if (strcmp(token, "udp") == 0) 8107d96f4efSBrian Somers proto = IPPROTO_UDP; 8117d96f4efSBrian Somers else 812ffcb611aSDag-Erling Smørgrav return (-1); 8137d96f4efSBrian Somers state = STATE_READ_KEYWORD; 8147d96f4efSBrian Somers break; 8157d96f4efSBrian Somers 8167d96f4efSBrian Somers case STATE_READ_SRC: 8177d96f4efSBrian Somers case STATE_READ_DST: 8187d96f4efSBrian Somers { 8197d96f4efSBrian Somers int err; 8207d96f4efSBrian Somers char *p; 8217d96f4efSBrian Somers struct in_addr mask; 8227d96f4efSBrian Somers struct in_addr addr; 8237d96f4efSBrian Somers 8247d96f4efSBrian Somers p = token; 8257d96f4efSBrian Somers while (*p != '/' && *p != 0) 8267d96f4efSBrian Somers p++; 8277d96f4efSBrian Somers 828f0f93429SDag-Erling Smørgrav if (*p != '/') { 8297d96f4efSBrian Somers IpMask(32, &mask); 8307d96f4efSBrian Somers err = IpAddr(token, &addr); 8317d96f4efSBrian Somers if (err) 832ffcb611aSDag-Erling Smørgrav return (-1); 833f0f93429SDag-Erling Smørgrav } else { 8347d96f4efSBrian Somers int nbits; 8357d96f4efSBrian Somers char s[sizeof(buffer)]; 8367d96f4efSBrian Somers 8377d96f4efSBrian Somers *p = ' '; 8387d96f4efSBrian Somers n = sscanf(token, "%s %d", s, &nbits); 8397d96f4efSBrian Somers if (n != 2) 840ffcb611aSDag-Erling Smørgrav return (-1); 8417d96f4efSBrian Somers 8427d96f4efSBrian Somers err = IpAddr(s, &addr); 8437d96f4efSBrian Somers if (err) 844ffcb611aSDag-Erling Smørgrav return (-1); 8457d96f4efSBrian Somers 8467d96f4efSBrian Somers err = IpMask(nbits, &mask); 8477d96f4efSBrian Somers if (err) 848ffcb611aSDag-Erling Smørgrav return (-1); 8497d96f4efSBrian Somers } 8507d96f4efSBrian Somers 851f0f93429SDag-Erling Smørgrav if (state == STATE_READ_SRC) { 8527d96f4efSBrian Somers src_addr = addr; 8537d96f4efSBrian Somers src_mask = mask; 854f0f93429SDag-Erling Smørgrav } else { 8557d96f4efSBrian Somers dst_addr = addr; 8567d96f4efSBrian Somers dst_mask = mask; 8577d96f4efSBrian Somers } 8587d96f4efSBrian Somers } 8597d96f4efSBrian Somers state = STATE_READ_KEYWORD; 8607d96f4efSBrian Somers break; 8617d96f4efSBrian Somers 8627d96f4efSBrian Somers default: 863ffcb611aSDag-Erling Smørgrav return (-1); 8647d96f4efSBrian Somers break; 8657d96f4efSBrian Somers } 8667d96f4efSBrian Somers 867d9e630b5SAndrey A. Chernov do { 868d9e630b5SAndrey A. Chernov token = strsep(&res, " \t"); 869d9e630b5SAndrey A. Chernov } while (token != NULL && !*token); 8707d96f4efSBrian Somers } 8717d96f4efSBrian Somers #undef STATE_READ_KEYWORD 8727d96f4efSBrian Somers #undef STATE_READ_TYPE 8737d96f4efSBrian Somers #undef STATE_READ_PORT 8747d96f4efSBrian Somers #undef STATE_READ_SERVER 8757d96f4efSBrian Somers #undef STATE_READ_RULE 8767d96f4efSBrian Somers #undef STATE_READ_DELETE 8777d96f4efSBrian Somers #undef STATE_READ_PROTO 8787d96f4efSBrian Somers #undef STATE_READ_SRC 8797d96f4efSBrian Somers #undef STATE_READ_DST 8807d96f4efSBrian Somers 8817d96f4efSBrian Somers /* Convert port strings to numbers. This needs to be done after 8827d96f4efSBrian Somers the string is parsed, because the prototype might not be designated 8837d96f4efSBrian Somers before the ports (which might be symbolic entries in /etc/services) */ 8847d96f4efSBrian Somers 885f0f93429SDag-Erling Smørgrav if (strlen(str_port) != 0) { 8867d96f4efSBrian Somers int err; 8877d96f4efSBrian Somers 8887d96f4efSBrian Somers err = IpPort(str_port, proto, &proxy_port); 8897d96f4efSBrian Somers if (err) 890ffcb611aSDag-Erling Smørgrav return (-1); 891f0f93429SDag-Erling Smørgrav } else { 8927d96f4efSBrian Somers proxy_port = 0; 8937d96f4efSBrian Somers } 8947d96f4efSBrian Somers 895f0f93429SDag-Erling Smørgrav if (strlen(str_server_port) != 0) { 8967d96f4efSBrian Somers int err; 8977d96f4efSBrian Somers 8987d96f4efSBrian Somers err = IpPort(str_server_port, proto, &server_port); 8997d96f4efSBrian Somers if (err) 900ffcb611aSDag-Erling Smørgrav return (-1); 901f0f93429SDag-Erling Smørgrav } else { 9027d96f4efSBrian Somers server_port = 0; 9037d96f4efSBrian Somers } 9047d96f4efSBrian Somers 9057d96f4efSBrian Somers /* Check that at least the server address has been defined */ 9067d96f4efSBrian Somers if (server_addr.s_addr == 0) 907ffcb611aSDag-Erling Smørgrav return (-1); 9087d96f4efSBrian Somers 9097d96f4efSBrian Somers /* Add to linked list */ 9107d96f4efSBrian Somers proxy_entry = malloc(sizeof(struct proxy_entry)); 9117d96f4efSBrian Somers if (proxy_entry == NULL) 912ffcb611aSDag-Erling Smørgrav return (-1); 9137d96f4efSBrian Somers 9147d96f4efSBrian Somers proxy_entry->proxy_type = proxy_type; 9157d96f4efSBrian Somers proxy_entry->rule_index = rule_index; 9167d96f4efSBrian Somers proxy_entry->proto = proto; 9177d96f4efSBrian Somers proxy_entry->proxy_port = htons(proxy_port); 9187d96f4efSBrian Somers proxy_entry->server_port = htons(server_port); 9197d96f4efSBrian Somers proxy_entry->server_addr = server_addr; 9207d96f4efSBrian Somers proxy_entry->src_addr.s_addr = src_addr.s_addr & src_mask.s_addr; 9217d96f4efSBrian Somers proxy_entry->dst_addr.s_addr = dst_addr.s_addr & dst_mask.s_addr; 9227d96f4efSBrian Somers proxy_entry->src_mask = src_mask; 9237d96f4efSBrian Somers proxy_entry->dst_mask = dst_mask; 9247d96f4efSBrian Somers 9255e289f9eSPoul-Henning Kamp RuleAdd(la, proxy_entry); 9267d96f4efSBrian Somers 927ffcb611aSDag-Erling Smørgrav return (0); 9287d96f4efSBrian Somers } 929