99484f62a9ddc2e066f43a83ed51de4e97985e92
[openafs.git] / src / rx / LINUX24 / rx_knet.c
1 /*
2  * Copyright 2000, International Business Machines Corporation and others.
3  * All Rights Reserved.
4  * 
5  * This software has been released under the terms of the IBM Public
6  * License.  For details, see the LICENSE file in the top-level source
7  * directory or online at http://www.openafs.org/dl/license10.html
8  */
9
10 /*
11  * rx_knet.c - RX kernel send, receive and timer routines.
12  *
13  * Linux implementation.
14  */
15 #include <afsconfig.h>
16 #include "afs/param.h"
17
18
19 #include <linux/version.h>
20 #ifdef AFS_LINUX22_ENV
21 #include "rx/rx_kcommon.h"
22 #if defined(AFS_LINUX24_ENV)
23 #include "h/smp_lock.h"
24 #endif
25 #include <asm/uaccess.h>
26 #ifdef ADAPT_PMTU
27 #include <linux/errqueue.h>
28 #include <linux/icmp.h>
29 #endif
30
31 /* rxk_NewSocket
32  * open and bind RX socket
33  */
34 osi_socket *
35 rxk_NewSocketHost(afs_uint32 ahost, short aport)
36 {
37     struct socket *sockp;
38     struct sockaddr_in myaddr;
39     int code;
40     KERNEL_SPACE_DECL;
41 #ifdef ADAPT_PMTU
42     int pmtu = IP_PMTUDISC_WANT;
43     int do_recverr = 1;
44 #else
45     int pmtu = IP_PMTUDISC_DONT;
46 #endif
47
48     /* We need a better test for this. if you need it back, tell us
49      * how to detect it. 
50      */
51 #ifdef LINUX_KERNEL_SOCK_CREATE_V
52     code = sock_create(AF_INET, SOCK_DGRAM, IPPROTO_UDP, &sockp, 0);
53 #else
54     code = sock_create(AF_INET, SOCK_DGRAM, IPPROTO_UDP, &sockp);
55 #endif
56     if (code < 0)
57         return NULL;
58
59     /* Bind socket */
60     myaddr.sin_family = AF_INET;
61     myaddr.sin_addr.s_addr = ahost;
62     myaddr.sin_port = aport;
63     code =
64         sockp->ops->bind(sockp, (struct sockaddr *)&myaddr, sizeof(myaddr));
65
66     if (code < 0) {
67 #if defined(AFS_LINUX24_ENV)
68         printk("sock_release(rx_socket) FIXME\n");
69 #else
70         sock_release(sockp);
71 #endif
72         return NULL;
73     }
74
75     TO_USER_SPACE();
76     sockp->ops->setsockopt(sockp, SOL_IP, IP_MTU_DISCOVER, (char *)&pmtu,
77                            sizeof(pmtu));
78 #ifdef ADAPT_PMTU
79     sockp->ops->setsockopt(sockp, SOL_IP, IP_RECVERR, (char *)&do_recverr,
80                            sizeof(do_recverr));
81 #endif
82     TO_KERNEL_SPACE();
83     return (osi_socket *)sockp;
84 }
85
86 osi_socket *
87 rxk_NewSocket(short aport)
88 {
89     return rxk_NewSocketHost(htonl(INADDR_ANY), aport);
90 }
91
92 /* free socket allocated by osi_NetSocket */
93 int
94 rxk_FreeSocket(struct socket *asocket)
95 {
96     AFS_STATCNT(osi_FreeSocket);
97     return 0;
98 }
99
100 #ifdef ADAPT_PMTU
101 void
102 handle_socket_error(osi_socket so)
103 {
104     KERNEL_SPACE_DECL;
105     struct msghdr msg;
106     struct cmsghdr *cmsg;
107     struct sock_extended_err *err;
108     struct sockaddr_in addr;
109     struct sockaddr *offender;
110     char *controlmsgbuf;
111     int code;
112     struct socket *sop = (struct socket *)so;
113
114     if (!(controlmsgbuf=rxi_Alloc(256)))
115         return;
116     msg.msg_name = &addr;
117     msg.msg_namelen = sizeof(addr);
118     msg.msg_iov = NULL;
119     msg.msg_iovlen = 0;
120     msg.msg_control = controlmsgbuf;
121     msg.msg_controllen = 256;
122     msg.msg_flags = 0;
123
124     TO_USER_SPACE();
125     code = sock_recvmsg(sop, &msg, 256, MSG_ERRQUEUE|MSG_DONTWAIT|MSG_TRUNC);
126     TO_KERNEL_SPACE();
127
128     if (code < 0 || !(msg.msg_flags & MSG_ERRQUEUE))
129         goto out;
130
131     for (cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg)) {
132         if (CMSG_OK(&msg, cmsg) && cmsg->cmsg_level == SOL_IP &&
133             cmsg->cmsg_type == IP_RECVERR)
134             break;
135     }
136     if (!cmsg)
137         goto out;
138     err = CMSG_DATA(cmsg);
139     offender = SO_EE_OFFENDER(err);
140     
141     if (offender->sa_family != AF_INET)
142        goto out;
143
144     memcpy(&addr, offender, sizeof(addr));
145
146     if (err->ee_origin == SO_EE_ORIGIN_ICMP &&
147         err->ee_type == ICMP_DEST_UNREACH &&
148         err->ee_code == ICMP_FRAG_NEEDED) {
149         rxi_SetPeerMtu(ntohl(addr.sin_addr.s_addr), ntohs(addr.sin_port),
150                        err->ee_info);
151     }
152     /* other DEST_UNREACH's and TIME_EXCEEDED should be dealt with too */
153
154 out:
155     rxi_Free(controlmsgbuf, 256);
156     return;
157 }
158 #endif
159
160 /* osi_NetSend
161  *
162  * Return codes:
163  * 0 = success
164  * non-zero = failure
165  */
166 int
167 osi_NetSend(osi_socket sop, struct sockaddr_in *to, struct iovec *iovec,
168             int iovcnt, afs_int32 size, int istack)
169 {
170     KERNEL_SPACE_DECL;
171     struct msghdr msg;
172     int code;
173 #ifdef ADAPT_PMTU
174     int sockerr;
175     size_t esize;
176
177     while (1) {
178         sockerr=0;
179         esize = sizeof(sockerr);
180         TO_USER_SPACE();
181         sop->ops->getsockopt(sop, SOL_SOCKET, SO_ERROR, (char *)&sockerr,
182                            &esize);
183         TO_KERNEL_SPACE();
184         if (sockerr == 0)
185            break;
186         handle_socket_error(sop);
187     }
188 #endif
189
190     msg.msg_iovlen = iovcnt;
191     msg.msg_iov = iovec;
192     msg.msg_name = to;
193     msg.msg_namelen = sizeof(*to);
194     msg.msg_control = NULL;
195     msg.msg_controllen = 0;
196     msg.msg_flags = 0;
197
198     TO_USER_SPACE();
199     code = sock_sendmsg(sop, &msg, size);
200     TO_KERNEL_SPACE();
201     return (code < 0) ? code : 0;
202 }
203
204
205 /* osi_NetReceive
206  * OS dependent part of kernel RX listener thread.
207  *
208  * Arguments:
209  *      so      socket to receive on, typically rx_socket
210  *      from    pointer to a sockaddr_in. 
211  *      iov     array of iovecs to fill in.
212  *      iovcnt  how many iovecs there are.
213  *      lengthp IN/OUT in: total space available in iovecs. out: size of read.
214  *
215  * Return
216  * 0 if successful
217  * error code (such as EINTER) if not
218  *
219  * Environment
220  *      Note that the maximum number of iovecs is 2 + RX_MAXWVECS. This is
221  *      so we have a little space to look for packets larger than 
222  *      rx_maxReceiveSize.
223  */
224 int rxk_lastSocketError;
225 int rxk_nSocketErrors;
226 int
227 osi_NetReceive(osi_socket so, struct sockaddr_in *from, struct iovec *iov,
228                int iovcnt, int *lengthp)
229 {
230     KERNEL_SPACE_DECL;
231     struct msghdr msg;
232     int code;
233 #ifdef ADAPT_PMTU
234     int sockerr;
235     size_t esize;
236 #endif
237     struct iovec tmpvec[RX_MAXWVECS + 2];
238     struct socket *sop = (struct socket *)so;
239
240     if (iovcnt > RX_MAXWVECS + 2) {
241         osi_Panic("Too many (%d) iovecs passed to osi_NetReceive\n", iovcnt);
242     }
243 #ifdef ADAPT_PMTU
244     while (1) {
245         sockerr=0;
246         esize = sizeof(sockerr);
247         TO_USER_SPACE();
248         sop->ops->getsockopt(sop, SOL_SOCKET, SO_ERROR, (char *)&sockerr,
249                            &esize);
250         TO_KERNEL_SPACE();
251         if (sockerr == 0)
252            break;
253         handle_socket_error(so);
254     }
255 #endif
256     memcpy(tmpvec, iov, iovcnt * sizeof(struct iovec));
257     msg.msg_name = from;
258     msg.msg_iov = tmpvec;
259     msg.msg_iovlen = iovcnt;
260     msg.msg_control = NULL;
261     msg.msg_controllen = 0;
262     msg.msg_flags = 0;
263
264     TO_USER_SPACE();
265     code = sock_recvmsg(sop, &msg, *lengthp, 0);
266     TO_KERNEL_SPACE();
267
268     if (code < 0) {
269         /* Clear the error before using the socket again.
270          * Oh joy, Linux has hidden header files as well. It appears we can
271          * simply call again and have it clear itself via sock_error().
272          */
273 #ifdef AFS_LINUX22_ENV
274         flush_signals(current); /* We don't want no stinkin' signals. */
275 #else
276         current->signal = 0;    /* We don't want no stinkin' signals. */
277 #endif
278         rxk_lastSocketError = code;
279         rxk_nSocketErrors++;
280     } else {
281         *lengthp = code;
282         code = 0;
283     }
284
285     return code;
286 }
287 #ifdef EXPORTED_TASKLIST_LOCK
288 extern rwlock_t tasklist_lock __attribute__((weak));
289 #endif
290 void
291 osi_StopListener(void)
292 {
293     extern struct task_struct *rxk_ListenerTask;
294
295     while (rxk_ListenerTask) {
296         if (rxk_ListenerTask) {
297             flush_signals(rxk_ListenerTask);
298             force_sig(SIGKILL, rxk_ListenerTask);
299         }
300         if (!rxk_ListenerTask)
301             break;
302         afs_osi_Sleep(&rxk_ListenerTask);
303     }
304     sock_release(rx_socket);
305     rx_socket = NULL;
306 }
307
308 #endif /* AFS_LINUX22_ENV */