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