kernel tree reorganization stage 1: Major cvs repository work (not logged as
[dragonfly.git] / sys / netproto / ncp / ncp_sock.c
1 /*
2  * Copyright (c) 1999, Boris Popov
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *    This product includes software developed by Boris Popov.
16  * 4. Neither the name of the author nor the names of any co-contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
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  * $FreeBSD: src/sys/netncp/ncp_sock.c,v 1.2 1999/10/12 10:36:59 bp Exp $
33  * $DragonFly: src/sys/netproto/ncp/ncp_sock.c,v 1.7 2003/08/07 21:54:35 dillon Exp $
34  *
35  * Low level socket routines
36  */
37
38 #include "opt_inet.h"
39 #include "opt_ipx.h"
40
41 #if !defined(INET) && !defined(IPX)
42 #error "NCP requeires either INET of IPX protocol family"
43 #endif
44
45 #include <sys/param.h>
46 #include <sys/errno.h>
47 #include <sys/malloc.h>
48 #include <sys/systm.h>
49 #include <sys/proc.h>
50 #include <sys/socket.h>
51 #include <sys/socketvar.h>
52 #include <sys/protosw.h>
53 #include <sys/kernel.h>
54 #include <sys/uio.h>
55 #include <sys/syslog.h>
56 #include <sys/mbuf.h>
57 #include <net/route.h>
58
59 #ifdef IPX
60 #include <netproto/ipx/ipx.h>
61 #include <netproto/ipx/ipx_pcb.h>
62 #endif
63
64 #include "ncp.h"
65 #include "ncp_conn.h"
66 #include "ncp_sock.h"
67 #include "ncp_subr.h"
68 #include "ncp_rq.h"
69
70 #ifdef IPX
71 #define ipx_setnullnet(x) ((x).x_net.s_net[0]=0); ((x).x_net.s_net[1]=0);
72 #define ipx_setnullhost(x) ((x).x_host.s_host[0] = 0); \
73         ((x).x_host.s_host[1] = 0); ((x).x_host.s_host[2] = 0);
74 #endif
75
76 /*int ncp_poll(struct socket *so, int events);*/
77 /*static int ncp_getsockname(struct socket *so, caddr_t asa, int *alen);*/
78 static int ncp_soconnect(struct socket *so,struct sockaddr *target, struct thread *p);
79
80
81 /* This will need only if native IP used, or (unlikely) NCP will be
82  * implemented on the socket level
83  */
84 static int
85 ncp_soconnect(struct socket *so,struct sockaddr *target, struct thread *td) {
86         int error,s;
87
88         error = soconnect(so, (struct sockaddr*)target, td);
89         if (error)
90                 return error;
91         /*
92          * Wait for the connection to complete. Cribbed from the
93          * connect system call but with the wait timing out so
94          * that interruptible mounts don't hang here for a long time.
95          */
96         error = EIO;
97         s = splnet();
98         while ((so->so_state & SS_ISCONNECTING) && so->so_error == 0) {
99                 (void) tsleep((caddr_t)&so->so_timeo, 0, "ncpcon", 2 * hz);
100                 if ((so->so_state & SS_ISCONNECTING) &&
101                     so->so_error == 0 /*&& rep &&*/) {
102                         so->so_state &= ~SS_ISCONNECTING;
103                         splx(s);
104                         goto bad;
105                 }
106         }
107         if (so->so_error) {
108                 error = so->so_error;
109                 so->so_error = 0;
110                 splx(s);
111                 goto bad;
112         }
113                 splx(s);
114         error=0;
115 bad:
116         return error;
117 }
118 #ifdef notyet
119 static int
120 ncp_getsockname(struct socket *so, caddr_t asa, int *alen) {
121         struct sockaddr *sa;
122         int len=0, error;
123
124         sa = 0;
125         error = (*so->so_proto->pr_usrreqs->pru_sockaddr)(so, &sa);
126         if (error==0) {
127                 if (sa) {
128                         len = min(len, sa->sa_len);
129                         bcopy(sa, (caddr_t)asa, (u_int)len);
130                 }
131                 *alen=len;
132         }
133         if (sa)
134                 FREE(sa, M_SONAME);
135         return (error);
136 }
137 #endif
138 int ncp_sock_recv(struct socket *so, struct mbuf **mp, int *rlen)
139 {
140         struct uio auio;
141         struct thread *td = curthread; /* XXX */
142         int error,flags,len;
143         
144         auio.uio_resid = len = 1000000;
145         auio.uio_td = td;
146         flags = MSG_DONTWAIT;
147
148 /*      error = so->so_proto->pr_usrreqs->pru_soreceive(so, 0, &auio,
149             (struct mbuf **)0, (struct mbuf **)0, &flags);*/
150         error = so->so_proto->pr_usrreqs->pru_soreceive(so, 0, &auio,
151             mp, (struct mbuf **)0, &flags);
152         *rlen = len - auio.uio_resid;
153 /*      if (!error) {
154             *rlen=iov.iov_len;
155         } else
156             *rlen=0;*/
157 #ifdef NCP_SOCKET_DEBUG
158         if (error)
159                 printf("ncp_recv: err=%d\n", error);
160 #endif
161         return (error);
162 }
163
164 int
165 ncp_sock_send(struct socket *so, struct mbuf *top, struct ncp_rq *rqp)
166 {
167         struct thread *td = curthread; /* XXX */
168         struct sockaddr *to = 0;
169         struct ncp_conn *conn = rqp->conn;
170         struct mbuf *m;
171         int error, flags=0;
172         int sendwait;
173
174         for(;;) {
175                 m = m_copym(top, 0, M_COPYALL, M_WAIT);
176 /*              NCPDDEBUG(m);*/
177                 error = so->so_proto->pr_usrreqs->pru_sosend(so, to, 0, m, 0, flags, td);
178                 if (error == 0 || error == EINTR || error == ENETDOWN)
179                         break;
180                 if (rqp->rexmit == 0) break;
181                 rqp->rexmit--;
182                 tsleep(&sendwait, 0, "ncprsn", conn->li.timeout * hz);
183                 error = ncp_chkintr(conn, td);
184                 if (error == EINTR) break;
185         }
186         if (error) {
187                 log(LOG_INFO, "ncp_send: error %d for server %s", error, conn->li.server);
188         }
189         return error;
190 }
191
192 int
193 ncp_poll(struct socket *so, int events){
194     struct thread *td = curthread; /* XXX */
195     struct ucred *cred=NULL;
196     return so->so_proto->pr_usrreqs->pru_sopoll(so, events, cred, td);
197 }
198
199 int
200 ncp_sock_rselect(struct socket *so, struct thread *td,
201         struct timeval *tv, int events)
202 {
203         struct timeval atv,rtv,ttv;
204         struct proc *p = td->td_proc;
205         int s,timo,error=0;
206
207         KKASSERT(p);
208
209         if (tv) {
210                 atv=*tv;
211                 if (itimerfix(&atv)) {
212                         error = EINVAL;
213                         goto done;
214                 }
215                 getmicrouptime(&rtv);
216                 timevaladd(&atv, &rtv);
217         }
218         timo = 0;
219 retry:
220         p->p_flag |= P_SELECT;
221         error = ncp_poll(so, events);
222         if (error) {
223                 error = 0;
224                 goto done;
225         }
226         if (tv) {
227                 getmicrouptime(&rtv);
228                 if (timevalcmp(&rtv, &atv, >=))
229                         goto done;
230                 ttv=atv;
231                 timevalsub(&ttv, &rtv);
232                 timo = tvtohz(&ttv);
233         }
234         s = splhigh();
235         if ((p->p_flag & P_SELECT) == 0) {
236                 splx(s);
237                 goto retry;
238         }
239         p->p_flag &= ~P_SELECT;
240         error = tsleep((caddr_t)&selwait, 0, "ncpslt", timo);
241         splx(s);
242 done:
243         p->p_flag &= ~P_SELECT;
244         if (error == ERESTART) {
245 /*              printf("Signal: %x", CURSIG(p));*/
246                 error = 0;
247         }
248         return (error);
249 }
250
251 #ifdef IPX
252 /* 
253  * Connect to specified server via IPX
254  */
255 int
256 ncp_sock_connect_ipx(struct ncp_conn *conn) {
257         struct sockaddr_ipx sipx;
258         struct ipxpcb *npcb;
259         struct thread *td = conn->td;
260         int addrlen, error, count;
261
262         sipx.sipx_port = htons(0);
263
264         for (count = 0;;count++) {
265                 if (count > (IPXPORT_WELLKNOWN-IPXPORT_RESERVED)*2) {
266                         error = EADDRINUSE;
267                         goto bad;
268                 }
269                 conn->ncp_so = conn->wdg_so = NULL;
270                 checkbad(socreate(AF_IPX, &conn->ncp_so, SOCK_DGRAM, 0, td));
271                 if (conn->li.opt & NCP_OPT_WDOG)
272                         checkbad(socreate(AF_IPX, &conn->wdg_so, SOCK_DGRAM,0,td));
273                 addrlen = sizeof(sipx);
274                 sipx.sipx_family = AF_IPX;
275                 ipx_setnullnet(sipx.sipx_addr);
276                 ipx_setnullhost(sipx.sipx_addr);
277                 sipx.sipx_len = addrlen;
278                 error = sobind(conn->ncp_so, (struct sockaddr *)&sipx, td);
279                 if (error == 0) {
280                         if ((conn->li.opt & NCP_OPT_WDOG) == 0)
281                                 break;
282                         sipx.sipx_addr = sotoipxpcb(conn->ncp_so)->ipxp_laddr;
283                         sipx.sipx_port = htons(ntohs(sipx.sipx_port) + 1);
284                         ipx_setnullnet(sipx.sipx_addr);
285                         ipx_setnullhost(sipx.sipx_addr);
286                         error = sobind(conn->wdg_so, (struct sockaddr *)&sipx, td);
287                 }
288                 if (!error) break;
289                 if (error != EADDRINUSE) goto bad;
290                 sipx.sipx_port = htons((ntohs(sipx.sipx_port)+4) & 0xfff8);
291                 soclose(conn->ncp_so);
292                 if (conn->wdg_so)
293                         soclose(conn->wdg_so);
294         }
295         npcb = sotoipxpcb(conn->ncp_so);
296         npcb->ipxp_dpt = IPXPROTO_NCP;
297         /* IPXrouted must be running, i.e. route must be presented */
298         conn->li.ipxaddr.sipx_len = sizeof(struct sockaddr_ipx);
299         checkbad(ncp_soconnect(conn->ncp_so, &conn->li.saddr, td));
300         if (conn->wdg_so) {
301                 sotoipxpcb(conn->wdg_so)->ipxp_laddr.x_net = npcb->ipxp_laddr.x_net;
302                 sotoipxpcb(conn->wdg_so)->ipxp_laddr.x_host= npcb->ipxp_laddr.x_host;
303         }
304         if (!error) {
305                 conn->flags |= NCPFL_SOCONN;
306         }
307 #ifdef NCPBURST
308         if (ncp_burst_enabled) {
309                 checkbad(socreate(AF_IPX, &conn->bc_so, SOCK_DGRAM, 0, td));
310                 bzero(&sipx, sizeof(sipx));
311                 sipx.sipx_len = sizeof(sipx);
312                 checkbad(sobind(conn->bc_so, (struct sockaddr *)&sipx, td));
313                 checkbad(ncp_soconnect(conn->bc_so, &conn->li.saddr, td));
314         }
315 #endif
316         if (!error) {
317                 conn->flags |= NCPFL_SOCONN;
318                 ncp_sock_checksum(conn, 0);
319         }
320         return error;
321 bad:
322         ncp_sock_disconnect(conn);
323         return (error);
324 }
325
326 int
327 ncp_sock_checksum(struct ncp_conn *conn, int enable) {
328
329 #ifdef SO_IPX_CHECKSUM
330         if (enable) {
331                 sotoipxpcb(conn->ncp_so)->ipxp_flags |= IPXP_CHECKSUM;
332         } else {
333                 sotoipxpcb(conn->ncp_so)->ipxp_flags &= ~IPXP_CHECKSUM;
334         }
335 #endif
336         return 0;
337 }
338 #endif
339
340 #ifdef INET
341 /* 
342  * Connect to specified server via IP
343  */
344 int
345 ncp_sock_connect_in(struct ncp_conn *conn) {
346         struct sockaddr_in sin;
347         struct thread *td = conn->td;
348         int addrlen=sizeof(sin), error;
349
350         conn->flags = 0;
351         bzero(&sin,addrlen);
352         conn->ncp_so = conn->wdg_so = NULL;
353         checkbad(socreate(AF_INET, &conn->ncp_so, SOCK_DGRAM, IPPROTO_UDP, td));
354         sin.sin_family = AF_INET;
355         sin.sin_len = addrlen;
356         checkbad(sobind(conn->ncp_so, (struct sockaddr *)&sin, td));
357         checkbad(ncp_soconnect(conn->ncp_so,(struct sockaddr*)&conn->li.addr, td));
358         if  (!error)
359                 conn->flags |= NCPFL_SOCONN;
360         return error;
361 bad:
362         ncp_sock_disconnect(conn);
363         return (error);
364 }
365 #endif
366
367
368 /*
369  * Connection expected to be locked
370  */
371 int
372 ncp_sock_disconnect(struct ncp_conn *conn) {
373         struct socket *so;
374         conn->flags &= ~(NCPFL_SOCONN | NCPFL_ATTACHED | NCPFL_LOGGED);
375         if (conn->ncp_so) {
376                 so = conn->ncp_so;
377                 conn->ncp_so = (struct socket *)0;
378                 soshutdown(so, 2);
379                 soclose(so);
380         }
381         if (conn->wdg_so) {
382                 so = conn->wdg_so;
383                 conn->wdg_so = (struct socket *)0;
384                 soshutdown(so, 2);
385                 soclose(so);
386         }
387 #ifdef NCPBURST
388         if (conn->bc_so) {
389                 so = conn->bc_so;
390                 conn->bc_so = (struct socket *)NULL;
391                 soshutdown(so, 2);
392                 soclose(so);
393         }
394 #endif
395         return 0;
396 }
397
398 #ifdef IPX
399 static void
400 ncp_watchdog(struct ncp_conn *conn) {
401         char *buf;
402         struct mbuf *m;
403         int error, len, flags;
404         struct socket *so;
405         struct sockaddr *sa;
406         struct uio auio;
407
408         sa = NULL;
409         while (conn->wdg_so) { /* not a loop */
410                 so = conn->wdg_so;
411                 auio.uio_resid = len = 1000000;
412                 auio.uio_td = curthread;
413                 flags = MSG_DONTWAIT;
414                 error = so->so_proto->pr_usrreqs->pru_soreceive(so, 
415                     (struct sockaddr**)&sa, &auio, &m, (struct mbuf**)0, &flags);
416                 if (error) break;
417                 len -= auio.uio_resid;
418                 NCPSDEBUG("got watch dog %d\n",len);
419                 if (len != 2) break;
420                 buf = mtod(m, char*);
421                 if (buf[1] != '?') break;
422                 buf[1] = 'Y';
423                 error = so->so_proto->pr_usrreqs->pru_sosend(so, (struct sockaddr*)sa, 0, m, 0, 0, curthread);
424                 NCPSDEBUG("send watch dog %d\n",error);
425                 break;
426         }
427         if (sa) FREE(sa, M_SONAME);
428         return;
429 }
430 #endif /* IPX */
431
432 void
433 ncp_check_conn(struct ncp_conn *conn) {
434         int s;
435
436         if (conn == NULL || !(conn->flags & NCPFL_ATTACHED))
437                 return;
438         s = splnet();
439         ncp_check_rq(conn);
440         splx(s);
441 #ifdef IPX
442         ncp_watchdog(conn);
443 #endif
444 }