Home | History | Annotate | Line # | Download | only in netinet
ip_rcmd_pxy.c revision 1.2.4.4
      1  1.2.4.4  yamt /*	$NetBSD: ip_rcmd_pxy.c,v 1.2.4.4 2014/05/22 11:40:58 yamt Exp $	*/
      2  1.2.4.2  yamt 
      3  1.2.4.2  yamt /*
      4  1.2.4.3  yamt  * Copyright (C) 2012 by Darren Reed.
      5  1.2.4.2  yamt  *
      6  1.2.4.2  yamt  * See the IPFILTER.LICENCE file for details on licencing.
      7  1.2.4.2  yamt  *
      8  1.2.4.3  yamt  * Id: ip_rcmd_pxy.c,v 1.1.1.2 2012/07/22 13:45:33 darrenr Exp
      9  1.2.4.2  yamt  *
     10  1.2.4.2  yamt  * Simple RCMD transparent proxy for in-kernel use.  For use with the NAT
     11  1.2.4.2  yamt  * code.
     12  1.2.4.2  yamt  */
     13  1.2.4.2  yamt 
     14  1.2.4.2  yamt #include <sys/cdefs.h>
     15  1.2.4.4  yamt __KERNEL_RCSID(1, "$NetBSD: ip_rcmd_pxy.c,v 1.2.4.4 2014/05/22 11:40:58 yamt Exp $");
     16  1.2.4.2  yamt 
     17  1.2.4.2  yamt #define	IPF_RCMD_PROXY
     18  1.2.4.2  yamt 
     19  1.2.4.3  yamt typedef struct rcmdinfo {
     20  1.2.4.3  yamt 	u_32_t	rcmd_port;	/* Port number seen */
     21  1.2.4.3  yamt 	u_32_t	rcmd_portseq;	/* Sequence number where port is first seen */
     22  1.2.4.3  yamt 	ipnat_t	*rcmd_rule;	/* Template rule for back connection */
     23  1.2.4.3  yamt } rcmdinfo_t;
     24  1.2.4.3  yamt 
     25  1.2.4.2  yamt void ipf_p_rcmd_main_load(void);
     26  1.2.4.2  yamt void ipf_p_rcmd_main_unload(void);
     27  1.2.4.2  yamt 
     28  1.2.4.2  yamt int ipf_p_rcmd_init(void);
     29  1.2.4.2  yamt void ipf_p_rcmd_fini(void);
     30  1.2.4.2  yamt void ipf_p_rcmd_del(ipf_main_softc_t *, ap_session_t *);
     31  1.2.4.2  yamt int ipf_p_rcmd_new(void *, fr_info_t *, ap_session_t *, nat_t *);
     32  1.2.4.2  yamt int ipf_p_rcmd_out(void *, fr_info_t *, ap_session_t *, nat_t *);
     33  1.2.4.2  yamt int ipf_p_rcmd_in(void *, fr_info_t *, ap_session_t *, nat_t *);
     34  1.2.4.2  yamt u_short ipf_rcmd_atoi(char *);
     35  1.2.4.2  yamt int ipf_p_rcmd_portmsg(fr_info_t *, ap_session_t *, nat_t *);
     36  1.2.4.2  yamt 
     37  1.2.4.2  yamt static	frentry_t	rcmdfr;
     38  1.2.4.2  yamt 
     39  1.2.4.2  yamt static	int		rcmd_proxy_init = 0;
     40  1.2.4.2  yamt 
     41  1.2.4.2  yamt 
     42  1.2.4.2  yamt /*
     43  1.2.4.2  yamt  * RCMD application proxy initialization.
     44  1.2.4.2  yamt  */
     45  1.2.4.2  yamt void
     46  1.2.4.2  yamt ipf_p_rcmd_main_load(void)
     47  1.2.4.2  yamt {
     48  1.2.4.2  yamt 	bzero((char *)&rcmdfr, sizeof(rcmdfr));
     49  1.2.4.2  yamt 	rcmdfr.fr_ref = 1;
     50  1.2.4.2  yamt 	rcmdfr.fr_flags = FR_INQUE|FR_PASS|FR_QUICK|FR_KEEPSTATE;
     51  1.2.4.2  yamt 	MUTEX_INIT(&rcmdfr.fr_lock, "RCMD proxy rule lock");
     52  1.2.4.2  yamt 	rcmd_proxy_init = 1;
     53  1.2.4.2  yamt }
     54  1.2.4.2  yamt 
     55  1.2.4.2  yamt 
     56  1.2.4.2  yamt void
     57  1.2.4.2  yamt ipf_p_rcmd_main_unload(void)
     58  1.2.4.2  yamt {
     59  1.2.4.2  yamt 	if (rcmd_proxy_init == 1) {
     60  1.2.4.2  yamt 		MUTEX_DESTROY(&rcmdfr.fr_lock);
     61  1.2.4.2  yamt 		rcmd_proxy_init = 0;
     62  1.2.4.2  yamt 	}
     63  1.2.4.2  yamt }
     64  1.2.4.2  yamt 
     65  1.2.4.2  yamt 
     66  1.2.4.2  yamt /*
     67  1.2.4.2  yamt  * Setup for a new RCMD proxy.
     68  1.2.4.2  yamt  */
     69  1.2.4.2  yamt int
     70  1.2.4.2  yamt ipf_p_rcmd_new(void *arg, fr_info_t *fin, ap_session_t *aps, nat_t *nat)
     71  1.2.4.2  yamt {
     72  1.2.4.2  yamt 	tcphdr_t *tcp = (tcphdr_t *)fin->fin_dp;
     73  1.2.4.2  yamt 	rcmdinfo_t *rc;
     74  1.2.4.2  yamt 	ipnat_t *ipn;
     75  1.2.4.2  yamt 
     76  1.2.4.2  yamt 	fin = fin;	/* LINT */
     77  1.2.4.2  yamt 
     78  1.2.4.3  yamt 	KMALLOC(rc, rcmdinfo_t *);
     79  1.2.4.2  yamt 	if (rc == NULL) {
     80  1.2.4.2  yamt 		printf("ipf_p_rcmd_new:KMALLOCS(%zu) failed\n", sizeof(*rc));
     81  1.2.4.2  yamt 		return -1;
     82  1.2.4.2  yamt 	}
     83  1.2.4.3  yamt 	aps->aps_sport = tcp->th_sport;
     84  1.2.4.3  yamt 	aps->aps_dport = tcp->th_dport;
     85  1.2.4.3  yamt 
     86  1.2.4.3  yamt 	ipn = ipf_proxy_rule_rev(nat);
     87  1.2.4.3  yamt 	if (ipn == NULL) {
     88  1.2.4.3  yamt 		KFREE(rc);
     89  1.2.4.3  yamt 		return -1;
     90  1.2.4.3  yamt 	}
     91  1.2.4.2  yamt 
     92  1.2.4.2  yamt 	aps->aps_data = rc;
     93  1.2.4.3  yamt 	aps->aps_psiz = sizeof(*rc);
     94  1.2.4.2  yamt 	bzero((char *)rc, sizeof(*rc));
     95  1.2.4.2  yamt 
     96  1.2.4.3  yamt 	rc->rcmd_rule = ipn;
     97  1.2.4.2  yamt 
     98  1.2.4.2  yamt 	return 0;
     99  1.2.4.2  yamt }
    100  1.2.4.2  yamt 
    101  1.2.4.2  yamt 
    102  1.2.4.2  yamt void
    103  1.2.4.2  yamt ipf_p_rcmd_del(ipf_main_softc_t *softc, ap_session_t *aps)
    104  1.2.4.2  yamt {
    105  1.2.4.2  yamt 	rcmdinfo_t *rci;
    106  1.2.4.2  yamt 
    107  1.2.4.2  yamt 	rci = aps->aps_data;
    108  1.2.4.2  yamt 	if (rci != NULL) {
    109  1.2.4.3  yamt 		rci->rcmd_rule->in_flags |= IPN_DELETE;
    110  1.2.4.3  yamt 		ipf_nat_rule_deref(softc, &rci->rcmd_rule);
    111  1.2.4.2  yamt 	}
    112  1.2.4.2  yamt }
    113  1.2.4.2  yamt 
    114  1.2.4.2  yamt 
    115  1.2.4.2  yamt /*
    116  1.2.4.2  yamt  * ipf_rcmd_atoi - implement a simple version of atoi
    117  1.2.4.2  yamt  */
    118  1.2.4.2  yamt u_short
    119  1.2.4.2  yamt ipf_rcmd_atoi(char *ptr)
    120  1.2.4.2  yamt {
    121  1.2.4.2  yamt 	char *s = ptr, c;
    122  1.2.4.2  yamt 	u_short i = 0;
    123  1.2.4.2  yamt 
    124  1.2.4.2  yamt 	while (((c = *s++) != '\0') && ISDIGIT(c)) {
    125  1.2.4.2  yamt 		i *= 10;
    126  1.2.4.2  yamt 		i += c - '0';
    127  1.2.4.2  yamt 	}
    128  1.2.4.2  yamt 	return i;
    129  1.2.4.2  yamt }
    130  1.2.4.2  yamt 
    131  1.2.4.2  yamt 
    132  1.2.4.2  yamt int
    133  1.2.4.2  yamt ipf_p_rcmd_portmsg(fr_info_t *fin, ap_session_t *aps, nat_t *nat)
    134  1.2.4.2  yamt {
    135  1.2.4.2  yamt 	tcphdr_t *tcp, tcph, *tcp2 = &tcph;
    136  1.2.4.2  yamt 	int off, dlen, nflags, direction;
    137  1.2.4.2  yamt 	ipf_main_softc_t *softc;
    138  1.2.4.3  yamt 	ipf_nat_softc_t *softn;
    139  1.2.4.2  yamt 	char portbuf[8], *s;
    140  1.2.4.2  yamt 	rcmdinfo_t *rc;
    141  1.2.4.2  yamt 	fr_info_t fi;
    142  1.2.4.2  yamt 	u_short sp;
    143  1.2.4.2  yamt 	nat_t *nat2;
    144  1.2.4.3  yamt #ifdef USE_INET6
    145  1.2.4.3  yamt 	ip6_t *ip6;
    146  1.2.4.3  yamt #endif
    147  1.2.4.3  yamt 	int tcpsz;
    148  1.2.4.3  yamt 	int slen = 0;
    149  1.2.4.2  yamt 	ip_t *ip;
    150  1.2.4.2  yamt 	mb_t *m;
    151  1.2.4.2  yamt 
    152  1.2.4.2  yamt 	tcp = (tcphdr_t *)fin->fin_dp;
    153  1.2.4.2  yamt 
    154  1.2.4.2  yamt 	m = fin->fin_m;
    155  1.2.4.2  yamt 	ip = fin->fin_ip;
    156  1.2.4.3  yamt 	tcpsz = TCP_OFF(tcp) << 2;
    157  1.2.4.3  yamt #ifdef USE_INET6
    158  1.2.4.3  yamt 	ip6 = (ip6_t *)fin->fin_ip;
    159  1.2.4.2  yamt #endif
    160  1.2.4.3  yamt 	softc = fin->fin_main_soft;
    161  1.2.4.3  yamt 	softn = softc->ipf_nat_soft;
    162  1.2.4.3  yamt 	off = (char *)tcp - (char *)ip + tcpsz + fin->fin_ipoff;
    163  1.2.4.3  yamt 
    164  1.2.4.3  yamt 	dlen = fin->fin_dlen - tcpsz;
    165  1.2.4.2  yamt 	if (dlen <= 0)
    166  1.2.4.2  yamt 		return 0;
    167  1.2.4.2  yamt 
    168  1.2.4.2  yamt 	rc = (rcmdinfo_t *)aps->aps_data;
    169  1.2.4.2  yamt 	if ((rc->rcmd_portseq != 0) &&
    170  1.2.4.2  yamt 	    (tcp->th_seq != rc->rcmd_portseq))
    171  1.2.4.2  yamt 		return 0;
    172  1.2.4.2  yamt 
    173  1.2.4.2  yamt 	bzero(portbuf, sizeof(portbuf));
    174  1.2.4.2  yamt 	COPYDATA(m, off, MIN(sizeof(portbuf), dlen), portbuf);
    175  1.2.4.2  yamt 
    176  1.2.4.2  yamt 	portbuf[sizeof(portbuf) - 1] = '\0';
    177  1.2.4.2  yamt 	s = portbuf;
    178  1.2.4.2  yamt 	sp = ipf_rcmd_atoi(s);
    179  1.2.4.2  yamt 	if (sp == 0) {
    180  1.2.4.2  yamt #ifdef IP_RCMD_PROXY_DEBUG
    181  1.2.4.2  yamt 		printf("ipf_p_rcmd_portmsg:sp == 0 dlen %d [%s]\n",
    182  1.2.4.2  yamt 		       dlen, portbuf);
    183  1.2.4.2  yamt #endif
    184  1.2.4.2  yamt 		return 0;
    185  1.2.4.2  yamt 	}
    186  1.2.4.2  yamt 
    187  1.2.4.2  yamt 	if (rc->rcmd_port != 0 && sp != rc->rcmd_port) {
    188  1.2.4.2  yamt #ifdef IP_RCMD_PROXY_DEBUG
    189  1.2.4.2  yamt 		printf("ipf_p_rcmd_portmsg:sp(%d) != rcmd_port(%d)\n",
    190  1.2.4.2  yamt 		       sp, rc->rcmd_port);
    191  1.2.4.2  yamt #endif
    192  1.2.4.2  yamt 		return 0;
    193  1.2.4.2  yamt 	}
    194  1.2.4.2  yamt 
    195  1.2.4.2  yamt 	rc->rcmd_port = sp;
    196  1.2.4.2  yamt 	rc->rcmd_portseq = tcp->th_seq;
    197  1.2.4.2  yamt 
    198  1.2.4.2  yamt 	/*
    199  1.2.4.2  yamt 	 * Initialise the packet info structure so we can search the NAT
    200  1.2.4.2  yamt 	 * table to see if there already is soemthing present that matches
    201  1.2.4.2  yamt 	 * up with what we want to add.
    202  1.2.4.2  yamt 	 */
    203  1.2.4.2  yamt 	bcopy((char *)fin, (char *)&fi, sizeof(fi));
    204  1.2.4.2  yamt 	fi.fin_flx |= FI_IGNORE;
    205  1.2.4.2  yamt 	fi.fin_data[0] = 0;
    206  1.2.4.2  yamt 	fi.fin_data[1] = sp;
    207  1.2.4.3  yamt 	fi.fin_src6 = nat->nat_ndst6;
    208  1.2.4.3  yamt 	fi.fin_dst6 = nat->nat_nsrc6;
    209  1.2.4.2  yamt 
    210  1.2.4.3  yamt 	if (nat->nat_v[0] == 6) {
    211  1.2.4.3  yamt #ifdef USE_INET6
    212  1.2.4.3  yamt 		if (nat->nat_dir == NAT_OUTBOUND) {
    213  1.2.4.3  yamt 			nat2 = ipf_nat6_outlookup(&fi, NAT_SEARCH|IPN_TCP,
    214  1.2.4.3  yamt 						  nat->nat_pr[1],
    215  1.2.4.3  yamt 						  &nat->nat_osrc6.in6,
    216  1.2.4.3  yamt 						  &nat->nat_odst6.in6);
    217  1.2.4.3  yamt 		} else {
    218  1.2.4.3  yamt 			nat2 = ipf_nat6_inlookup(&fi, NAT_SEARCH|IPN_TCP,
    219  1.2.4.3  yamt 						 nat->nat_pr[0],
    220  1.2.4.3  yamt 						 &nat->nat_osrc6.in6,
    221  1.2.4.3  yamt 						 &nat->nat_odst6.in6);
    222  1.2.4.3  yamt 		}
    223  1.2.4.3  yamt #else
    224  1.2.4.3  yamt 		nat2 = (void *)-1;
    225  1.2.4.3  yamt #endif
    226  1.2.4.2  yamt 	} else {
    227  1.2.4.3  yamt 		if (nat->nat_dir == NAT_OUTBOUND) {
    228  1.2.4.3  yamt 			nat2 = ipf_nat_outlookup(&fi, NAT_SEARCH|IPN_TCP,
    229  1.2.4.3  yamt 						 nat->nat_pr[1],
    230  1.2.4.3  yamt 						 nat->nat_osrcip,
    231  1.2.4.3  yamt 						 nat->nat_odstip);
    232  1.2.4.3  yamt 		} else {
    233  1.2.4.3  yamt 			nat2 = ipf_nat_inlookup(&fi, NAT_SEARCH|IPN_TCP,
    234  1.2.4.3  yamt 						nat->nat_pr[0],
    235  1.2.4.3  yamt 						nat->nat_osrcip,
    236  1.2.4.3  yamt 						nat->nat_odstip);
    237  1.2.4.3  yamt 		}
    238  1.2.4.2  yamt 	}
    239  1.2.4.3  yamt 	if (nat2 != NULL)
    240  1.2.4.3  yamt 		return APR_ERR(1);
    241  1.2.4.2  yamt 
    242  1.2.4.3  yamt 	/*
    243  1.2.4.3  yamt 	 * Add skeleton NAT entry for connection which will come
    244  1.2.4.3  yamt 	 * back the other way.
    245  1.2.4.3  yamt 	 */
    246  1.2.4.2  yamt 
    247  1.2.4.3  yamt 	if (nat->nat_v[0] == 6) {
    248  1.2.4.3  yamt #ifdef USE_INET6
    249  1.2.4.3  yamt 		slen = ip6->ip6_plen;
    250  1.2.4.3  yamt 		ip6->ip6_plen = htons(sizeof(*tcp));
    251  1.2.4.3  yamt #endif
    252  1.2.4.3  yamt 	} else {
    253  1.2.4.2  yamt 		slen = ip->ip_len;
    254  1.2.4.2  yamt 		ip->ip_len = htons(fin->fin_hlen + sizeof(*tcp));
    255  1.2.4.3  yamt 	}
    256  1.2.4.2  yamt 
    257  1.2.4.3  yamt 	/*
    258  1.2.4.3  yamt 	 * Fill out the fake TCP header with a few fields that ipfilter
    259  1.2.4.3  yamt 	 * considers to be important.
    260  1.2.4.3  yamt 	 */
    261  1.2.4.3  yamt 	bzero((char *)tcp2, sizeof(*tcp2));
    262  1.2.4.3  yamt 	tcp2->th_win = htons(8192);
    263  1.2.4.3  yamt 	TCP_OFF_A(tcp2, 5);
    264  1.2.4.3  yamt 	tcp2->th_flags = TH_SYN;
    265  1.2.4.3  yamt 
    266  1.2.4.3  yamt 	fi.fin_dp = (char *)tcp2;
    267  1.2.4.3  yamt 	fi.fin_fr = &rcmdfr;
    268  1.2.4.3  yamt 	fi.fin_dlen = sizeof(*tcp2);
    269  1.2.4.3  yamt 	fi.fin_plen = fi.fin_hlen + sizeof(*tcp2);
    270  1.2.4.3  yamt 	fi.fin_flx &= FI_LOWTTL|FI_FRAG|FI_TCPUDP|FI_OPTIONS|FI_IGNORE;
    271  1.2.4.2  yamt 
    272  1.2.4.3  yamt 	if (nat->nat_dir == NAT_OUTBOUND) {
    273  1.2.4.3  yamt 		fi.fin_out = 0;
    274  1.2.4.3  yamt 		direction = NAT_INBOUND;
    275  1.2.4.3  yamt 	} else {
    276  1.2.4.3  yamt 		fi.fin_out = 1;
    277  1.2.4.3  yamt 		direction = NAT_OUTBOUND;
    278  1.2.4.3  yamt 	}
    279  1.2.4.3  yamt 	nflags = SI_W_SPORT|NAT_SLAVE|IPN_TCP;
    280  1.2.4.2  yamt 
    281  1.2.4.3  yamt 	MUTEX_ENTER(&softn->ipf_nat_new);
    282  1.2.4.3  yamt 	if (fin->fin_v == 4)
    283  1.2.4.3  yamt 		nat2 = ipf_nat_add(&fi, rc->rcmd_rule, NULL, nflags,
    284  1.2.4.2  yamt 				   direction);
    285  1.2.4.3  yamt #ifdef USE_INET6
    286  1.2.4.3  yamt 	else
    287  1.2.4.3  yamt 		nat2 = ipf_nat6_add(&fi, rc->rcmd_rule, NULL, nflags,
    288  1.2.4.3  yamt 				    direction);
    289  1.2.4.3  yamt #endif
    290  1.2.4.3  yamt 	MUTEX_EXIT(&softn->ipf_nat_new);
    291  1.2.4.2  yamt 
    292  1.2.4.3  yamt 	if (nat2 != NULL) {
    293  1.2.4.3  yamt 		(void) ipf_nat_proto(&fi, nat2, IPN_TCP);
    294  1.2.4.3  yamt 		MUTEX_ENTER(&nat2->nat_lock);
    295  1.2.4.3  yamt 		ipf_nat_update(&fi, nat2);
    296  1.2.4.3  yamt 		MUTEX_EXIT(&nat2->nat_lock);
    297  1.2.4.3  yamt 		fi.fin_ifp = NULL;
    298  1.2.4.3  yamt 		if (nat2->nat_dir == NAT_INBOUND)
    299  1.2.4.3  yamt 			fi.fin_dst6 = nat->nat_osrc6;
    300  1.2.4.3  yamt 		(void) ipf_state_add(softc, &fi, NULL, SI_W_SPORT);
    301  1.2.4.3  yamt 	}
    302  1.2.4.3  yamt 	if (nat->nat_v[0] == 6) {
    303  1.2.4.3  yamt #ifdef USE_INET6
    304  1.2.4.3  yamt 		ip6->ip6_plen = slen;
    305  1.2.4.3  yamt #endif
    306  1.2.4.3  yamt 	} else {
    307  1.2.4.2  yamt 		ip->ip_len = slen;
    308  1.2.4.2  yamt 	}
    309  1.2.4.3  yamt 	if (nat2 == NULL)
    310  1.2.4.3  yamt 		return APR_ERR(1);
    311  1.2.4.2  yamt 	return 0;
    312  1.2.4.2  yamt }
    313  1.2.4.2  yamt 
    314  1.2.4.2  yamt 
    315  1.2.4.2  yamt int
    316  1.2.4.2  yamt ipf_p_rcmd_out(void *arg, fr_info_t *fin, ap_session_t *aps, nat_t *nat)
    317  1.2.4.2  yamt {
    318  1.2.4.2  yamt 	if (nat->nat_dir == NAT_OUTBOUND)
    319  1.2.4.2  yamt 		return ipf_p_rcmd_portmsg(fin, aps, nat);
    320  1.2.4.2  yamt 	return 0;
    321  1.2.4.2  yamt }
    322  1.2.4.2  yamt 
    323  1.2.4.2  yamt 
    324  1.2.4.2  yamt int
    325  1.2.4.2  yamt ipf_p_rcmd_in(void *arg, fr_info_t *fin, ap_session_t *aps, nat_t *nat)
    326  1.2.4.2  yamt {
    327  1.2.4.2  yamt 	if (nat->nat_dir == NAT_INBOUND)
    328  1.2.4.2  yamt 		return ipf_p_rcmd_portmsg(fin, aps, nat);
    329  1.2.4.2  yamt 	return 0;
    330  1.2.4.2  yamt }
    331