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