ip_raudio_pxy.c revision 1.2.4.3 1 1.2.4.3 yamt /* $NetBSD: ip_raudio_pxy.c,v 1.2.4.3 2012/10/30 17:22:20 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_raudio_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
11 1.2.4.2 yamt #include <sys/cdefs.h>
12 1.2.4.3 yamt __KERNEL_RCSID(1, "$NetBSD: ip_raudio_pxy.c,v 1.2.4.3 2012/10/30 17:22:20 yamt Exp $");
13 1.2.4.2 yamt
14 1.2.4.2 yamt #define IPF_RAUDIO_PROXY
15 1.2.4.2 yamt
16 1.2.4.2 yamt
17 1.2.4.2 yamt void ipf_p_raudio_main_load(void);
18 1.2.4.2 yamt void ipf_p_raudio_main_unload(void);
19 1.2.4.2 yamt int ipf_p_raudio_new(void *, fr_info_t *, ap_session_t *, nat_t *);
20 1.2.4.2 yamt int ipf_p_raudio_in(void *, fr_info_t *, ap_session_t *, nat_t *);
21 1.2.4.2 yamt int ipf_p_raudio_out(void *, fr_info_t *, ap_session_t *, nat_t *);
22 1.2.4.2 yamt
23 1.2.4.2 yamt static frentry_t raudiofr;
24 1.2.4.2 yamt
25 1.2.4.2 yamt int raudio_proxy_init = 0;
26 1.2.4.2 yamt
27 1.2.4.2 yamt
28 1.2.4.2 yamt /*
29 1.2.4.2 yamt * Real Audio application proxy initialization.
30 1.2.4.2 yamt */
31 1.2.4.2 yamt void
32 1.2.4.2 yamt ipf_p_raudio_main_load(void)
33 1.2.4.2 yamt {
34 1.2.4.2 yamt bzero((char *)&raudiofr, sizeof(raudiofr));
35 1.2.4.2 yamt raudiofr.fr_ref = 1;
36 1.2.4.2 yamt raudiofr.fr_flags = FR_INQUE|FR_PASS|FR_QUICK|FR_KEEPSTATE;
37 1.2.4.2 yamt MUTEX_INIT(&raudiofr.fr_lock, "Real Audio proxy rule lock");
38 1.2.4.2 yamt raudio_proxy_init = 1;
39 1.2.4.2 yamt }
40 1.2.4.2 yamt
41 1.2.4.2 yamt
42 1.2.4.2 yamt void
43 1.2.4.2 yamt ipf_p_raudio_main_unload(void)
44 1.2.4.2 yamt {
45 1.2.4.2 yamt if (raudio_proxy_init == 1) {
46 1.2.4.2 yamt MUTEX_DESTROY(&raudiofr.fr_lock);
47 1.2.4.2 yamt raudio_proxy_init = 0;
48 1.2.4.2 yamt }
49 1.2.4.2 yamt }
50 1.2.4.2 yamt
51 1.2.4.2 yamt
52 1.2.4.2 yamt /*
53 1.2.4.2 yamt * Setup for a new proxy to handle Real Audio.
54 1.2.4.2 yamt */
55 1.2.4.2 yamt int
56 1.2.4.2 yamt ipf_p_raudio_new(void *arg, fr_info_t *fin, ap_session_t *aps, nat_t *nat)
57 1.2.4.2 yamt {
58 1.2.4.2 yamt raudio_t *rap;
59 1.2.4.2 yamt
60 1.2.4.3 yamt nat = nat; /* LINT */
61 1.2.4.3 yamt
62 1.2.4.3 yamt if (fin->fin_v != 4)
63 1.2.4.3 yamt return -1;
64 1.2.4.3 yamt
65 1.2.4.2 yamt KMALLOCS(aps->aps_data, void *, sizeof(raudio_t));
66 1.2.4.2 yamt if (aps->aps_data == NULL)
67 1.2.4.2 yamt return -1;
68 1.2.4.2 yamt
69 1.2.4.2 yamt bzero(aps->aps_data, sizeof(raudio_t));
70 1.2.4.2 yamt rap = aps->aps_data;
71 1.2.4.2 yamt aps->aps_psiz = sizeof(raudio_t);
72 1.2.4.2 yamt rap->rap_mode = RAP_M_TCP; /* default is for TCP */
73 1.2.4.2 yamt return 0;
74 1.2.4.2 yamt }
75 1.2.4.2 yamt
76 1.2.4.2 yamt
77 1.2.4.2 yamt
78 1.2.4.2 yamt int
79 1.2.4.2 yamt ipf_p_raudio_out(void *arg, fr_info_t *fin, ap_session_t *aps, nat_t *nat)
80 1.2.4.2 yamt {
81 1.2.4.2 yamt raudio_t *rap = aps->aps_data;
82 1.2.4.2 yamt unsigned char membuf[512 + 1], *s;
83 1.2.4.2 yamt u_short id = 0;
84 1.2.4.2 yamt tcphdr_t *tcp;
85 1.2.4.2 yamt int off, dlen;
86 1.2.4.2 yamt int len = 0;
87 1.2.4.2 yamt mb_t *m;
88 1.2.4.2 yamt
89 1.2.4.2 yamt nat = nat; /* LINT */
90 1.2.4.2 yamt
91 1.2.4.2 yamt /*
92 1.2.4.2 yamt * If we've already processed the start messages, then nothing left
93 1.2.4.2 yamt * for the proxy to do.
94 1.2.4.2 yamt */
95 1.2.4.2 yamt if (rap->rap_eos == 1)
96 1.2.4.2 yamt return 0;
97 1.2.4.2 yamt
98 1.2.4.2 yamt m = fin->fin_m;
99 1.2.4.2 yamt tcp = (tcphdr_t *)fin->fin_dp;
100 1.2.4.2 yamt off = (char *)tcp - (char *)fin->fin_ip;
101 1.2.4.2 yamt off += (TCP_OFF(tcp) << 2) + fin->fin_ipoff;
102 1.2.4.2 yamt
103 1.2.4.2 yamt #ifdef __sgi
104 1.2.4.2 yamt dlen = fin->fin_plen - off;
105 1.2.4.2 yamt #else
106 1.2.4.2 yamt dlen = MSGDSIZE(m) - off;
107 1.2.4.2 yamt #endif
108 1.2.4.2 yamt if (dlen <= 0)
109 1.2.4.2 yamt return 0;
110 1.2.4.2 yamt
111 1.2.4.2 yamt if (dlen > sizeof(membuf))
112 1.2.4.2 yamt dlen = sizeof(membuf);
113 1.2.4.2 yamt
114 1.2.4.2 yamt bzero((char *)membuf, sizeof(membuf));
115 1.2.4.2 yamt COPYDATA(m, off, dlen, (char *)membuf);
116 1.2.4.2 yamt /*
117 1.2.4.2 yamt * In all the startup parsing, ensure that we don't go outside
118 1.2.4.2 yamt * the packet buffer boundary.
119 1.2.4.2 yamt */
120 1.2.4.2 yamt /*
121 1.2.4.2 yamt * Look for the start of connection "PNA" string if not seen yet.
122 1.2.4.2 yamt */
123 1.2.4.2 yamt if (rap->rap_seenpna == 0) {
124 1.2.4.2 yamt s = (u_char *)memstr("PNA", (char *)membuf, 3, dlen);
125 1.2.4.2 yamt if (s == NULL)
126 1.2.4.2 yamt return 0;
127 1.2.4.2 yamt s += 3;
128 1.2.4.2 yamt rap->rap_seenpna = 1;
129 1.2.4.2 yamt } else
130 1.2.4.2 yamt s = membuf;
131 1.2.4.2 yamt
132 1.2.4.2 yamt /*
133 1.2.4.2 yamt * Directly after the PNA will be the version number of this
134 1.2.4.2 yamt * connection.
135 1.2.4.2 yamt */
136 1.2.4.2 yamt if (rap->rap_seenpna == 1 && rap->rap_seenver == 0) {
137 1.2.4.2 yamt if ((s + 1) - membuf < dlen) {
138 1.2.4.2 yamt rap->rap_version = (*s << 8) | *(s + 1);
139 1.2.4.2 yamt s += 2;
140 1.2.4.2 yamt rap->rap_seenver = 1;
141 1.2.4.2 yamt } else
142 1.2.4.2 yamt return 0;
143 1.2.4.2 yamt }
144 1.2.4.2 yamt
145 1.2.4.2 yamt /*
146 1.2.4.2 yamt * Now that we've been past the PNA and version number, we're into the
147 1.2.4.2 yamt * startup messages block. This ends when a message with an ID of 0.
148 1.2.4.2 yamt */
149 1.2.4.2 yamt while ((rap->rap_eos == 0) && ((s + 1) - membuf < dlen)) {
150 1.2.4.2 yamt if (rap->rap_gotid == 0) {
151 1.2.4.2 yamt id = (*s << 8) | *(s + 1);
152 1.2.4.2 yamt s += 2;
153 1.2.4.2 yamt rap->rap_gotid = 1;
154 1.2.4.2 yamt if (id == RA_ID_END) {
155 1.2.4.2 yamt rap->rap_eos = 1;
156 1.2.4.2 yamt break;
157 1.2.4.2 yamt }
158 1.2.4.2 yamt } else if (rap->rap_gotlen == 0) {
159 1.2.4.2 yamt len = (*s << 8) | *(s + 1);
160 1.2.4.2 yamt s += 2;
161 1.2.4.2 yamt rap->rap_gotlen = 1;
162 1.2.4.2 yamt }
163 1.2.4.2 yamt
164 1.2.4.2 yamt if (rap->rap_gotid == 1 && rap->rap_gotlen == 1) {
165 1.2.4.2 yamt if (id == RA_ID_UDP) {
166 1.2.4.2 yamt rap->rap_mode &= ~RAP_M_TCP;
167 1.2.4.2 yamt rap->rap_mode |= RAP_M_UDP;
168 1.2.4.2 yamt rap->rap_plport = (*s << 8) | *(s + 1);
169 1.2.4.2 yamt } else if (id == RA_ID_ROBUST) {
170 1.2.4.2 yamt rap->rap_mode |= RAP_M_ROBUST;
171 1.2.4.2 yamt rap->rap_prport = (*s << 8) | *(s + 1);
172 1.2.4.2 yamt }
173 1.2.4.2 yamt s += len;
174 1.2.4.2 yamt rap->rap_gotlen = 0;
175 1.2.4.2 yamt rap->rap_gotid = 0;
176 1.2.4.2 yamt }
177 1.2.4.2 yamt }
178 1.2.4.2 yamt return 0;
179 1.2.4.2 yamt }
180 1.2.4.2 yamt
181 1.2.4.2 yamt
182 1.2.4.2 yamt int
183 1.2.4.2 yamt ipf_p_raudio_in(void *arg, fr_info_t *fin, ap_session_t *aps, nat_t *nat)
184 1.2.4.2 yamt {
185 1.2.4.2 yamt unsigned char membuf[IPF_MAXPORTLEN + 1], *s;
186 1.2.4.2 yamt tcphdr_t *tcp, tcph, *tcp2 = &tcph;
187 1.2.4.2 yamt raudio_t *rap = aps->aps_data;
188 1.2.4.2 yamt ipf_main_softc_t *softc;
189 1.2.4.2 yamt ipf_nat_softc_t *softn;
190 1.2.4.2 yamt struct in_addr swa, swb;
191 1.2.4.2 yamt int off, dlen, slen;
192 1.2.4.2 yamt int a1, a2, a3, a4;
193 1.2.4.2 yamt u_short sp, dp;
194 1.2.4.2 yamt fr_info_t fi;
195 1.2.4.2 yamt tcp_seq seq;
196 1.2.4.2 yamt nat_t *nat2;
197 1.2.4.2 yamt u_char swp;
198 1.2.4.2 yamt ip_t *ip;
199 1.2.4.2 yamt mb_t *m;
200 1.2.4.2 yamt
201 1.2.4.2 yamt softc = fin->fin_main_soft;
202 1.2.4.2 yamt softn = softc->ipf_nat_soft;
203 1.2.4.2 yamt /*
204 1.2.4.2 yamt * Wait until we've seen the end of the start messages and even then
205 1.2.4.2 yamt * only proceed further if we're using UDP. If they want to use TCP
206 1.2.4.2 yamt * then data is sent back on the same channel that is already open.
207 1.2.4.2 yamt */
208 1.2.4.2 yamt if (rap->rap_sdone != 0)
209 1.2.4.2 yamt return 0;
210 1.2.4.2 yamt
211 1.2.4.2 yamt m = fin->fin_m;
212 1.2.4.2 yamt tcp = (tcphdr_t *)fin->fin_dp;
213 1.2.4.2 yamt off = (char *)tcp - (char *)fin->fin_ip;
214 1.2.4.2 yamt off += (TCP_OFF(tcp) << 2) + fin->fin_ipoff;
215 1.2.4.2 yamt
216 1.2.4.2 yamt #ifdef __sgi
217 1.2.4.2 yamt dlen = fin->fin_plen - off;
218 1.2.4.2 yamt #else
219 1.2.4.2 yamt dlen = MSGDSIZE(m) - off;
220 1.2.4.2 yamt #endif
221 1.2.4.2 yamt if (dlen <= 0)
222 1.2.4.2 yamt return 0;
223 1.2.4.2 yamt
224 1.2.4.2 yamt if (dlen > sizeof(membuf))
225 1.2.4.2 yamt dlen = sizeof(membuf);
226 1.2.4.2 yamt
227 1.2.4.2 yamt bzero((char *)membuf, sizeof(membuf));
228 1.2.4.2 yamt COPYDATA(m, off, dlen, (char *)membuf);
229 1.2.4.2 yamt
230 1.2.4.2 yamt seq = ntohl(tcp->th_seq);
231 1.2.4.2 yamt /*
232 1.2.4.2 yamt * Check to see if the data in this packet is of interest to us.
233 1.2.4.2 yamt * We only care for the first 19 bytes coming back from the server.
234 1.2.4.2 yamt */
235 1.2.4.2 yamt if (rap->rap_sseq == 0) {
236 1.2.4.2 yamt s = (u_char *)memstr("PNA", (char *)membuf, 3, dlen);
237 1.2.4.2 yamt if (s == NULL)
238 1.2.4.2 yamt return 0;
239 1.2.4.2 yamt a1 = s - membuf;
240 1.2.4.2 yamt dlen -= a1;
241 1.2.4.2 yamt a1 = 0;
242 1.2.4.2 yamt rap->rap_sseq = seq;
243 1.2.4.2 yamt a2 = MIN(dlen, sizeof(rap->rap_svr));
244 1.2.4.2 yamt } else if (seq <= rap->rap_sseq + sizeof(rap->rap_svr)) {
245 1.2.4.2 yamt /*
246 1.2.4.2 yamt * seq # which is the start of data and from that the offset
247 1.2.4.2 yamt * into the buffer array.
248 1.2.4.2 yamt */
249 1.2.4.2 yamt a1 = seq - rap->rap_sseq;
250 1.2.4.2 yamt a2 = MIN(dlen, sizeof(rap->rap_svr));
251 1.2.4.2 yamt a2 -= a1;
252 1.2.4.2 yamt s = membuf;
253 1.2.4.2 yamt } else
254 1.2.4.2 yamt return 0;
255 1.2.4.2 yamt
256 1.2.4.2 yamt for (a3 = a1, a4 = a2; (a4 > 0) && (a3 < 19) && (a3 >= 0); a4--,a3++) {
257 1.2.4.2 yamt rap->rap_sbf |= (1 << a3);
258 1.2.4.2 yamt rap->rap_svr[a3] = *s++;
259 1.2.4.2 yamt }
260 1.2.4.2 yamt
261 1.2.4.2 yamt if ((rap->rap_sbf != 0x7ffff) || (!rap->rap_eos)) /* 19 bits */
262 1.2.4.2 yamt return 0;
263 1.2.4.2 yamt rap->rap_sdone = 1;
264 1.2.4.2 yamt
265 1.2.4.2 yamt s = (u_char *)rap->rap_svr + 11;
266 1.2.4.2 yamt if (((*s << 8) | *(s + 1)) == RA_ID_ROBUST) {
267 1.2.4.2 yamt s += 2;
268 1.2.4.2 yamt rap->rap_srport = (*s << 8) | *(s + 1);
269 1.2.4.2 yamt }
270 1.2.4.2 yamt
271 1.2.4.2 yamt ip = fin->fin_ip;
272 1.2.4.2 yamt swp = ip->ip_p;
273 1.2.4.2 yamt swa = ip->ip_src;
274 1.2.4.2 yamt swb = ip->ip_dst;
275 1.2.4.2 yamt
276 1.2.4.2 yamt ip->ip_p = IPPROTO_UDP;
277 1.2.4.2 yamt ip->ip_src = nat->nat_ndstip;
278 1.2.4.2 yamt ip->ip_dst = nat->nat_odstip;
279 1.2.4.2 yamt
280 1.2.4.2 yamt bcopy((char *)fin, (char *)&fi, sizeof(fi));
281 1.2.4.2 yamt bzero((char *)tcp2, sizeof(*tcp2));
282 1.2.4.2 yamt TCP_OFF_A(tcp2, 5);
283 1.2.4.2 yamt fi.fin_flx |= FI_IGNORE;
284 1.2.4.2 yamt fi.fin_dp = (char *)tcp2;
285 1.2.4.2 yamt fi.fin_fr = &raudiofr;
286 1.2.4.2 yamt fi.fin_dlen = sizeof(*tcp2);
287 1.2.4.2 yamt fi.fin_plen = fi.fin_hlen + sizeof(*tcp2);
288 1.2.4.2 yamt tcp2->th_win = htons(8192);
289 1.2.4.2 yamt slen = ip->ip_len;
290 1.2.4.2 yamt ip->ip_len = htons(fin->fin_hlen + sizeof(*tcp));
291 1.2.4.2 yamt
292 1.2.4.2 yamt if (((rap->rap_mode & RAP_M_UDP_ROBUST) == RAP_M_UDP_ROBUST) &&
293 1.2.4.2 yamt (rap->rap_srport != 0)) {
294 1.2.4.2 yamt dp = rap->rap_srport;
295 1.2.4.2 yamt sp = rap->rap_prport;
296 1.2.4.2 yamt tcp2->th_sport = htons(sp);
297 1.2.4.2 yamt tcp2->th_dport = htons(dp);
298 1.2.4.2 yamt fi.fin_data[0] = dp;
299 1.2.4.2 yamt fi.fin_data[1] = sp;
300 1.2.4.2 yamt fi.fin_out = 0;
301 1.2.4.2 yamt MUTEX_ENTER(&softn->ipf_nat_new);
302 1.2.4.2 yamt nat2 = ipf_nat_add(&fi, nat->nat_ptr, NULL,
303 1.2.4.2 yamt NAT_SLAVE|IPN_UDP | (sp ? 0 : SI_W_SPORT),
304 1.2.4.2 yamt NAT_OUTBOUND);
305 1.2.4.2 yamt MUTEX_EXIT(&softn->ipf_nat_new);
306 1.2.4.2 yamt if (nat2 != NULL) {
307 1.2.4.2 yamt (void) ipf_nat_proto(&fi, nat2, IPN_UDP);
308 1.2.4.2 yamt MUTEX_ENTER(&nat2->nat_lock);
309 1.2.4.2 yamt ipf_nat_update(&fi, nat2);
310 1.2.4.2 yamt MUTEX_EXIT(&nat2->nat_lock);
311 1.2.4.2 yamt
312 1.2.4.2 yamt (void) ipf_state_add(softc, &fi, NULL,
313 1.2.4.2 yamt (sp ? 0 : SI_W_SPORT));
314 1.2.4.2 yamt }
315 1.2.4.2 yamt }
316 1.2.4.2 yamt
317 1.2.4.2 yamt if ((rap->rap_mode & RAP_M_UDP) == RAP_M_UDP) {
318 1.2.4.2 yamt sp = rap->rap_plport;
319 1.2.4.2 yamt tcp2->th_sport = htons(sp);
320 1.2.4.2 yamt tcp2->th_dport = 0; /* XXX - don't specify remote port */
321 1.2.4.2 yamt fi.fin_data[0] = sp;
322 1.2.4.2 yamt fi.fin_data[1] = 0;
323 1.2.4.2 yamt fi.fin_out = 1;
324 1.2.4.2 yamt MUTEX_ENTER(&softn->ipf_nat_new);
325 1.2.4.2 yamt nat2 = ipf_nat_add(&fi, nat->nat_ptr, NULL,
326 1.2.4.2 yamt NAT_SLAVE|IPN_UDP|SI_W_DPORT,
327 1.2.4.2 yamt NAT_OUTBOUND);
328 1.2.4.2 yamt MUTEX_EXIT(&softn->ipf_nat_new);
329 1.2.4.2 yamt if (nat2 != NULL) {
330 1.2.4.2 yamt (void) ipf_nat_proto(&fi, nat2, IPN_UDP);
331 1.2.4.2 yamt MUTEX_ENTER(&nat2->nat_lock);
332 1.2.4.2 yamt ipf_nat_update(&fi, nat2);
333 1.2.4.2 yamt MUTEX_EXIT(&nat2->nat_lock);
334 1.2.4.2 yamt
335 1.2.4.2 yamt (void) ipf_state_add(softc, &fi, NULL, SI_W_DPORT);
336 1.2.4.2 yamt }
337 1.2.4.2 yamt }
338 1.2.4.2 yamt
339 1.2.4.2 yamt ip->ip_p = swp;
340 1.2.4.2 yamt ip->ip_len = slen;
341 1.2.4.2 yamt ip->ip_src = swa;
342 1.2.4.2 yamt ip->ip_dst = swb;
343 1.2.4.2 yamt return 0;
344 1.2.4.2 yamt }
345