1 /*	$NetBSD: ip_netbios_pxy.c,v 1.3 2012/07/22 14:27:51 darrenr Exp $	*/
2 
3 /*
4  * Simple netbios-dgm transparent proxy for in-kernel use.
5  * For use with the NAT code.
6  * Id: ip_netbios_pxy.c,v 1.1.1.2 2012/07/22 13:45:30 darrenr Exp
7  */
8 
9 /*-
10  * Copyright (c) 2002-2003 Paul J. Ledbetter III
11  * All rights reserved.
12  *
13  * Redistribution and use in source and binary forms, with or without
14  * modification, are permitted provided that the following conditions
15  * are met:
16  * 1. Redistributions of source code must retain the above copyright
17  *    notice, this list of conditions and the following disclaimer.
18  * 2. Redistributions in binary form must reproduce the above copyright
19  *    notice, this list of conditions and the following disclaimer in the
20  *    documentation and/or other materials provided with the distribution.
21  *
22  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32  * SUCH DAMAGE.
33  *
34  * Id: ip_netbios_pxy.c,v 1.1.1.2 2012/07/22 13:45:30 darrenr Exp
35  */
36 
37 #include <sys/cdefs.h>
38 __KERNEL_RCSID(1, "$NetBSD: ip_netbios_pxy.c,v 1.3 2012/07/22 14:27:51 darrenr Exp $");
39 
40 #define	IPF_NETBIOS_PROXY
41 
42 void ipf_p_netbios_main_load(void);
43 void ipf_p_netbios_main_unload(void);
44 int ipf_p_netbios_out(void *, fr_info_t *, ap_session_t *, nat_t *);
45 
46 static	frentry_t	netbiosfr;
47 
48 int	netbios_proxy_init = 0;
49 
50 /*
51  * Initialize local structures.
52  */
53 void
ipf_p_netbios_main_load(void)54 ipf_p_netbios_main_load(void)
55 {
56 	bzero((char *)&netbiosfr, sizeof(netbiosfr));
57 	netbiosfr.fr_ref = 1;
58 	netbiosfr.fr_flags = FR_INQUE|FR_PASS|FR_QUICK|FR_KEEPSTATE;
59 	MUTEX_INIT(&netbiosfr.fr_lock, "NETBIOS proxy rule lock");
60 	netbios_proxy_init = 1;
61 }
62 
63 
64 void
ipf_p_netbios_main_unload(void)65 ipf_p_netbios_main_unload(void)
66 {
67 	if (netbios_proxy_init == 1) {
68 		MUTEX_DESTROY(&netbiosfr.fr_lock);
69 		netbios_proxy_init = 0;
70 	}
71 }
72 
73 
74 int
ipf_p_netbios_out(void * arg,fr_info_t * fin,ap_session_t * aps,nat_t * nat)75 ipf_p_netbios_out(void *arg, fr_info_t *fin, ap_session_t *aps, nat_t *nat)
76 {
77 	char dgmbuf[6];
78 	int off, dlen;
79 	udphdr_t *udp;
80 	ip_t *ip;
81 	mb_t *m;
82 
83 	aps = aps;	/* LINT */
84 	nat = nat;	/* LINT */
85 
86 	m = fin->fin_m;
87 	dlen = fin->fin_dlen - sizeof(*udp);
88 	/*
89 	 * no net bios datagram could possibly be shorter than this
90 	 */
91 	if (dlen < 11)
92 		return 0;
93 
94 	ip = fin->fin_ip;
95 	udp = (udphdr_t *)fin->fin_dp;
96 	off = (char *)udp - (char *)ip + sizeof(*udp) + fin->fin_ipoff;
97 
98 	/*
99 	 * move past the
100 	 *	ip header;
101 	 *	udp header;
102 	 *	4 bytes into the net bios dgm header.
103 	 *  According to rfc1002, this should be the exact location of
104 	 *  the source address/port
105 	 */
106 	off += 4;
107 
108 	/* Copy NATed source Address/port*/
109 	dgmbuf[0] = (char)((ip->ip_src.s_addr     ) &0xFF);
110 	dgmbuf[1] = (char)((ip->ip_src.s_addr >> 8) &0xFF);
111 	dgmbuf[2] = (char)((ip->ip_src.s_addr >> 16)&0xFF);
112 	dgmbuf[3] = (char)((ip->ip_src.s_addr >> 24)&0xFF);
113 
114 	dgmbuf[4] = (char)((udp->uh_sport     )&0xFF);
115 	dgmbuf[5] = (char)((udp->uh_sport >> 8)&0xFF);
116 
117 	/* replace data in packet */
118 	COPYBACK(m, off, sizeof(dgmbuf), dgmbuf);
119 
120 	return 0;
121 }
122