000caf77f7b8dfc5b21d5ddea5d9d52d4438b693
[openafs.git] / src / rx / rx_user.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 /* rx_user.c contains routines specific to the user space UNIX implementation of rx */
11
12 #include <afsconfig.h>
13 #include <afs/param.h>
14
15 RCSID
16     ("$Header$");
17
18 # include <sys/types.h>
19 # include <errno.h>
20 # include <signal.h>
21 # include <string.h>
22 #ifdef AFS_NT40_ENV
23 # include <WINNT/syscfg.h>
24 #else
25 # include <sys/socket.h>
26 # include <sys/file.h>
27 # include <netdb.h>
28 # include <sys/stat.h>
29 # include <netinet/in.h>
30 # include <sys/time.h>
31 # include <net/if.h>
32 # include <sys/ioctl.h>
33 #endif
34 # include <fcntl.h>
35 #if !defined(AFS_AIX_ENV) && !defined(AFS_NT40_ENV) 
36 # include <sys/syscall.h>
37 #endif
38 #include <afs/afs_args.h>
39 #include <afs/afsutil.h>
40
41 #ifndef IPPORT_USERRESERVED
42 /* If in.h doesn't define this, define it anyway.  Unfortunately, defining
43    this doesn't put the code into the kernel to restrict kernel assigned
44    port numbers to numbers below IPPORT_USERRESERVED...  */
45 #define IPPORT_USERRESERVED 5000
46 # endif
47
48 #ifndef AFS_NT40_ENV
49 # include <sys/time.h>
50 #endif
51 # include "rx_internal.h"
52 # include "rx.h"
53 # include "rx_globals.h"
54
55 #ifdef AFS_PTHREAD_ENV
56 #include <assert.h>
57
58 /*
59  * The rx_if_init_mutex mutex protects the following global variables:
60  * Inited
61  */
62
63 afs_kmutex_t rx_if_init_mutex;
64 #define LOCK_IF_INIT MUTEX_ENTER(&rx_if_init_mutex)
65 #define UNLOCK_IF_INIT MUTEX_EXIT(&rx_if_init_mutex)
66
67 /*
68  * The rx_if_mutex mutex protects the following global variables:
69  * myNetFlags
70  * myNetMTUs
71  * myNetMasks
72  */
73
74 afs_kmutex_t rx_if_mutex;
75 #define LOCK_IF MUTEX_ENTER(&rx_if_mutex)
76 #define UNLOCK_IF MUTEX_EXIT(&rx_if_mutex)
77 #else
78 #define LOCK_IF_INIT
79 #define UNLOCK_IF_INIT
80 #define LOCK_IF
81 #define UNLOCK_IF
82 #endif /* AFS_PTHREAD_ENV */
83
84
85 /*
86  * Make a socket for receiving/sending IP packets.  Set it into non-blocking
87  * and large buffering modes.  If port isn't specified, the kernel will pick
88  * one.  Returns the socket (>= 0) on success.  Returns OSI_NULLSOCKET on
89  * failure. Port must be in network byte order. 
90  */
91 osi_socket
92 rxi_GetHostUDPSocket(u_int ahost, u_short port)
93 {
94     int binds, code = 0;
95     osi_socket socketFd = OSI_NULLSOCKET;
96     struct sockaddr_in taddr;
97     char *name = "rxi_GetUDPSocket: ";
98 #ifdef AFS_LINUX22_ENV
99 #if defined(ADAPT_PMTU)
100     int pmtu=IP_PMTUDISC_WANT;
101     int recverr=1;
102 #else
103     int pmtu=IP_PMTUDISC_DONT;
104 #endif
105 #endif
106 #if defined(HAVE_LINUX_ERRQUEUE_H) && defined(ADAPT_PMTU)
107 #include <linux/types.h>
108 #include <linux/errqueue.h>
109 #ifndef IP_MTU
110 #define IP_MTU 14
111 #endif
112 #endif
113
114 #if !defined(AFS_NT40_ENV) 
115     if (ntohs(port) >= IPPORT_RESERVED && ntohs(port) < IPPORT_USERRESERVED) {
116 /*      (osi_Msg "%s*WARNING* port number %d is not a reserved port number.  Use port numbers above %d\n", name, port, IPPORT_USERRESERVED);
117 */ ;
118     }
119     if (ntohs(port) > 0 && ntohs(port) < IPPORT_RESERVED && geteuid() != 0) {
120         (osi_Msg
121          "%sport number %d is a reserved port number which may only be used by root.  Use port numbers above %d\n",
122          name, ntohs(port), IPPORT_USERRESERVED);
123         goto error;
124     }
125 #endif
126     socketFd = socket(AF_INET, SOCK_DGRAM, 0);
127
128     if (socketFd < 0) {
129         perror("socket");
130         goto error;
131     }
132
133 #ifdef AFS_NT40_ENV
134     rxi_xmit_init(socketFd);
135 #endif /* AFS_NT40_ENV */
136
137     taddr.sin_addr.s_addr = ahost;
138     taddr.sin_family = AF_INET;
139     taddr.sin_port = (u_short) port;
140 #ifdef STRUCT_SOCKADDR_HAS_SA_LEN
141     taddr.sin_len = sizeof(struct sockaddr_in);
142 #endif
143 #define MAX_RX_BINDS 10
144     for (binds = 0; binds < MAX_RX_BINDS; binds++) {
145         if (binds)
146             rxi_Delay(10);
147         code = bind(socketFd, (struct sockaddr *)&taddr, sizeof(taddr));
148         if (!code)
149             break;
150     }
151     if (code) {
152         perror("bind");
153         (osi_Msg "%sbind failed\n", name);
154         goto error;
155     }
156 #if !defined(AFS_NT40_ENV) 
157     /*
158      * Set close-on-exec on rx socket 
159      */
160     fcntl(socketFd, F_SETFD, 1);
161 #endif
162
163     /* Use one of three different ways of getting a socket buffer expanded to
164      * a reasonable size.
165      */
166     {
167         int greedy = 0;
168         int len1, len2;
169
170         len1 = 32766;
171         len2 = rx_UdpBufSize;
172
173         /* find the size closest to rx_UdpBufSize that will be accepted */
174         while (!greedy && len2 > len1) {
175             greedy =
176                 (setsockopt
177                   (socketFd, SOL_SOCKET, SO_RCVBUF, (char *)&len2,
178                    sizeof(len2)) >= 0);
179             if (!greedy)
180                 len2 /= 2;
181         }
182
183         /* but do not let it get smaller than 32K */ 
184         if (len2 < len1)
185             len2 = len1;
186
187         if (len1 < len2)
188             len1 = len2;
189
190
191         greedy =
192             (setsockopt
193              (socketFd, SOL_SOCKET, SO_SNDBUF, (char *)&len1,
194               sizeof(len1)) >= 0)
195             &&
196             (setsockopt
197              (socketFd, SOL_SOCKET, SO_RCVBUF, (char *)&len2,
198               sizeof(len2)) >= 0);
199         if (!greedy)
200             (osi_Msg "%s*WARNING* Unable to increase buffering on socket\n",
201              name);
202         if (rx_stats_active) {
203             rx_AtomicSwap(&rx_stats.socketGreedy, greedy, rx_stats_mutex);
204         }
205     }
206
207 #ifdef AFS_LINUX22_ENV
208     setsockopt(socketFd, SOL_IP, IP_MTU_DISCOVER, &pmtu, sizeof(pmtu));
209 #if defined(ADAPT_PMTU)
210     setsockopt(socketFd, SOL_IP, IP_RECVERR, &recverr, sizeof(recverr));
211 #endif
212 #endif
213     if (rxi_Listen(socketFd) < 0) {
214         goto error;
215     }
216
217     return socketFd;
218
219   error:
220 #ifdef AFS_NT40_ENV
221     if (socketFd >= 0)
222         closesocket(socketFd);
223 #else
224     if (socketFd >= 0)
225         close(socketFd);
226 #endif
227
228     return OSI_NULLSOCKET;
229 }
230
231 osi_socket
232 rxi_GetUDPSocket(u_short port)
233 {
234     return rxi_GetHostUDPSocket(htonl(INADDR_ANY), port);
235 }
236
237 void
238 osi_Panic(char *msg, ...)
239 {
240     va_list ap;
241     va_start(ap, msg);
242     (osi_Msg "Fatal Rx error: ");
243     (osi_VMsg msg, ap);
244     va_end(ap);
245     fflush(stderr);
246     fflush(stdout);
247     afs_abort();
248 }
249
250 /*
251  * osi_AssertFailU() -- used by the osi_Assert() macro.
252  */
253
254 void
255 osi_AssertFailU(const char *expr, const char *file, int line)
256 {
257     osi_Panic("assertion failed: %s, file: %s, line: %d\n", expr,
258               file, line);
259 }
260
261 #if defined(AFS_AIX32_ENV) && !defined(KERNEL)
262 #ifndef osi_Alloc
263 static const char memZero;
264 void *
265 osi_Alloc(afs_int32 x)
266 {
267     /* 
268      * 0-length allocs may return NULL ptr from malloc, so we special-case
269      * things so that NULL returned iff an error occurred 
270      */
271     if (x == 0)
272         return (void *)&memZero;
273     return(malloc(x));
274 }
275
276 void
277 osi_Free(void *x, afs_int32 size)
278 {
279     if (x == &memZero)
280         return;
281     free(x);
282 }
283 #endif
284 #endif /* defined(AFS_AIX32_ENV) && !defined(KERNEL) */
285
286 #define ADDRSPERSITE    16
287
288
289 static afs_uint32 rxi_NetAddrs[ADDRSPERSITE];   /* host order */
290 static int myNetMTUs[ADDRSPERSITE];
291 static int myNetMasks[ADDRSPERSITE];
292 static int myNetFlags[ADDRSPERSITE];
293 static u_int rxi_numNetAddrs;
294 static int Inited = 0;
295
296 #if defined(AFS_NT40_ENV) 
297 int
298 rxi_getaddr(void)
299 {
300     /* The IP address list can change so we must query for it */
301     rx_GetIFInfo();
302
303     /* we don't want to use the loopback adapter which is first */
304     /* this is a bad bad hack */
305     if (rxi_numNetAddrs > 1)
306         return htonl(rxi_NetAddrs[1]);  
307     else if (rxi_numNetAddrs > 0)
308         return htonl(rxi_NetAddrs[0]);
309     else
310         return 0;
311 }
312
313 /* 
314 ** return number of addresses 
315 ** and the addresses themselves in the buffer
316 ** maxSize - max number of interfaces to return.
317 */
318 int
319 rx_getAllAddr(afs_uint32 * buffer, int maxSize)
320 {
321     int count = 0, offset = 0;
322
323     /* The IP address list can change so we must query for it */
324     rx_GetIFInfo();
325
326 #ifdef AFS_DJGPP_ENV
327     /* we don't want to use the loopback adapter which is first */
328     /* this is a bad bad hack.
329      * and doesn't hold true on Windows.
330      */
331     if ( rxi_numNetAddrs > 1 )
332         offset = 1;
333 #endif /* AFS_DJGPP_ENV */
334
335     for (count = 0; offset < rxi_numNetAddrs && maxSize > 0;
336          count++, offset++, maxSize--)
337         buffer[count] = htonl(rxi_NetAddrs[offset]);
338
339     return count;
340 }
341
342 /* this function returns the total number of interface addresses
343  * the buffer has to be passed in by the caller. It also returns
344  * the matching interface mask and mtu.  All values are returned
345  * in network byte order.
346  */
347 int
348 rx_getAllAddrMaskMtu(afs_uint32 addrBuffer[], afs_uint32 maskBuffer[],
349                      afs_uint32 mtuBuffer[], int maxSize)
350 {
351     int count = 0, offset = 0;
352
353     /* The IP address list can change so we must query for it */
354     rx_GetIFInfo();
355
356 #ifdef AFS_DJGPP_ENV
357     /* we don't want to use the loopback adapter which is first */
358     /* this is a bad bad hack.
359      * and doesn't hold true on Windows.
360      */
361     if ( rxi_numNetAddrs > 1 )
362         offset = 1;
363 #endif /* AFS_DJGPP_ENV */
364
365     for (count = 0; 
366          offset < rxi_numNetAddrs && maxSize > 0;
367          count++, offset++, maxSize--) {
368         addrBuffer[count] = htonl(rxi_NetAddrs[offset]);
369         maskBuffer[count] = htonl(myNetMasks[offset]);
370         mtuBuffer[count]  = htonl(myNetMTUs[offset]);
371     }
372     return count;
373 }
374 #endif
375
376 #ifdef AFS_NT40_ENV
377 extern int rxinit_status;
378 void 
379 rxi_InitMorePackets(void) {
380     int npackets, ncbufs;
381
382     ncbufs = (rx_maxJumboRecvSize - RX_FIRSTBUFFERSIZE);
383     if (ncbufs > 0) {
384         ncbufs = ncbufs / RX_CBUFFERSIZE;
385         npackets = rx_initSendWindow - 1;
386         rxi_MorePackets(npackets * (ncbufs + 1));
387     }
388 }
389 void
390 rx_GetIFInfo(void)
391 {
392     u_int maxsize;
393     u_int rxsize;
394     afs_uint32 i;
395
396     LOCK_IF_INIT;
397     if (Inited) {
398         if (Inited < 2 && rxinit_status == 0) {
399             /* We couldn't initialize more packets earlier.
400              * Do it now. */
401             rxi_InitMorePackets();
402             Inited = 2;
403         }
404         UNLOCK_IF_INIT;
405         return;
406     }
407     Inited = 1;
408     UNLOCK_IF_INIT;
409
410     LOCK_IF;
411     rxi_numNetAddrs = ADDRSPERSITE;
412     (void)syscfg_GetIFInfo(&rxi_numNetAddrs, rxi_NetAddrs,
413                            myNetMasks, myNetMTUs, myNetFlags);
414
415     for (i = 0; i < rxi_numNetAddrs; i++) {
416         rxsize = rxi_AdjustIfMTU(myNetMTUs[i] - RX_IPUDP_SIZE);
417         maxsize =
418             rxi_nRecvFrags * rxsize + (rxi_nRecvFrags - 1) * UDP_HDR_SIZE;
419         maxsize = rxi_AdjustMaxMTU(rxsize, maxsize);
420         if (rx_maxReceiveSize < maxsize) {
421             rx_maxReceiveSize = MIN(RX_MAX_PACKET_SIZE, maxsize);
422             rx_maxReceiveSize =
423                 MIN(rx_maxReceiveSize, rx_maxReceiveSizeUser);
424         }
425
426     }
427     UNLOCK_IF;
428
429     /*
430      * If rxinit_status is still set, rx_InitHost() has yet to be called
431      * and we therefore do not have any mutex locks initialized.  As a
432      * result we cannot call rxi_MorePackets() without crashing.
433      */
434     if (rxinit_status)
435         return;
436
437     rxi_InitMorePackets();
438 }
439 #endif
440
441 static afs_uint32
442 fudge_netmask(afs_uint32 addr)
443 {
444     afs_uint32 msk;
445
446     if (IN_CLASSA(addr))
447         msk = IN_CLASSA_NET;
448     else if (IN_CLASSB(addr))
449         msk = IN_CLASSB_NET;
450     else if (IN_CLASSC(addr))
451         msk = IN_CLASSC_NET;
452     else
453         msk = 0;
454
455     return msk;
456 }
457
458
459
460 #if !defined(AFS_AIX_ENV) && !defined(AFS_NT40_ENV) && !defined(AFS_LINUX20_ENV) 
461 int
462 rxi_syscall(a3, a4, a5)
463      afs_uint32 a3, a4;
464      void *a5;
465 {
466     afs_uint32 rcode;
467     void (*old) ();
468
469     old = (void (*)())signal(SIGSYS, SIG_IGN);
470
471 #if defined(AFS_SGI_ENV)
472     rcode = afs_syscall(a3, a4, a5);
473 #else
474     rcode = syscall(AFS_SYSCALL, 28 /* AFSCALL_CALL */ , a3, a4, a5);
475 #endif /* AFS_SGI_ENV */
476
477     signal(SIGSYS, old);
478
479     return rcode;
480 }
481 #endif /* AFS_AIX_ENV */
482
483 #ifndef AFS_NT40_ENV
484 void
485 rx_GetIFInfo(void)
486 {
487     int s;
488     int i, j, len, res;
489     struct ifconf ifc;
490     struct ifreq ifs[ADDRSPERSITE];
491     struct ifreq *ifr;
492 #ifdef  AFS_AIX41_ENV
493     char buf[BUFSIZ], *cp, *cplim;
494 #endif
495     struct sockaddr_in *a;
496
497     LOCK_IF_INIT;
498     if (Inited) {
499         UNLOCK_IF_INIT;
500         return;
501     }
502     Inited = 1;
503     UNLOCK_IF_INIT;
504     LOCK_IF;
505     rxi_numNetAddrs = 0;
506     memset(rxi_NetAddrs, 0, sizeof(rxi_NetAddrs));
507     memset(myNetFlags, 0, sizeof(myNetFlags));
508     memset(myNetMTUs, 0, sizeof(myNetMTUs));
509     memset(myNetMasks, 0, sizeof(myNetMasks));
510     UNLOCK_IF;
511     s = socket(AF_INET, SOCK_DGRAM, 0);
512     if (s < 0)
513         return;
514
515 #ifdef  AFS_AIX41_ENV
516     ifc.ifc_len = sizeof(buf);
517     ifc.ifc_buf = buf;
518     ifr = ifc.ifc_req;
519 #else
520     ifc.ifc_len = sizeof(ifs);
521     ifc.ifc_buf = (caddr_t) & ifs[0];
522     memset(&ifs[0], 0, sizeof(ifs));
523 #endif
524     res = ioctl(s, SIOCGIFCONF, &ifc);
525     if (res < 0) {
526         /* fputs(stderr, "ioctl error IFCONF\n"); */
527         close(s);
528         return;
529     }
530
531     LOCK_IF;
532 #ifdef  AFS_AIX41_ENV
533 #define size(p) MAX((p).sa_len, sizeof(p))
534     cplim = buf + ifc.ifc_len;  /*skip over if's with big ifr_addr's */
535     for (cp = buf; cp < cplim;
536          cp += sizeof(ifr->ifr_name) + MAX(a->sin_len, sizeof(*a))) {
537         if (rxi_numNetAddrs >= ADDRSPERSITE)
538             break;
539
540         ifr = (struct ifreq *)cp;
541 #else
542     len = ifc.ifc_len / sizeof(struct ifreq);
543     if (len > ADDRSPERSITE)
544         len = ADDRSPERSITE;
545
546     for (i = 0; i < len; ++i) {
547         ifr = &ifs[i];
548         res = ioctl(s, SIOCGIFADDR, ifr);
549 #endif
550         if (res < 0) {
551             /* fputs(stderr, "ioctl error IFADDR\n");
552              * perror(ifr->ifr_name);   */
553             continue;
554         }
555         a = (struct sockaddr_in *)&ifr->ifr_addr;
556         if (a->sin_family != AF_INET)
557             continue;
558         rxi_NetAddrs[rxi_numNetAddrs] = ntohl(a->sin_addr.s_addr);
559         if (rxi_NetAddrs[rxi_numNetAddrs] == 0x7f000001) {
560             /* we don't really care about "localhost" */
561             continue;
562         }
563         for (j = 0; j < rxi_numNetAddrs; j++) {
564             if (rxi_NetAddrs[j] == rxi_NetAddrs[rxi_numNetAddrs])
565                 break;
566         }
567         if (j < rxi_numNetAddrs)
568             continue;
569
570         /* fprintf(stderr, "if %s addr=%x\n", ifr->ifr_name,
571          * rxi_NetAddrs[rxi_numNetAddrs]); */
572
573 #ifdef SIOCGIFFLAGS
574         res = ioctl(s, SIOCGIFFLAGS, ifr);
575         if (res == 0) {
576             myNetFlags[rxi_numNetAddrs] = ifr->ifr_flags;
577 #ifdef IFF_LOOPBACK
578             /* Handle aliased loopbacks as well. */
579             if (ifr->ifr_flags & IFF_LOOPBACK)
580                 continue;
581 #endif
582             /* fprintf(stderr, "if %s flags=%x\n", 
583              * ifr->ifr_name, ifr->ifr_flags); */
584         } else {                /*
585                                  * fputs(stderr, "ioctl error IFFLAGS\n");
586                                  * perror(ifr->ifr_name); */
587         }
588 #endif /* SIOCGIFFLAGS */
589
590 #if !defined(AFS_AIX_ENV)  && !defined(AFS_LINUX20_ENV)
591         /* this won't run on an AIX system w/o a cache manager */
592         rxi_syscallp = rxi_syscall;
593 #endif
594
595         /* If I refer to kernel extensions that aren't loaded on AIX, the 
596          * program refuses to load and run, so I simply can't include the 
597          * following code.  Fortunately, AIX is the one operating system in
598          * which the subsequent ioctl works reliably. */
599         if (rxi_syscallp) {
600             if ((*rxi_syscallp) (20 /*AFSOP_GETMTU */ ,
601                                  htonl(rxi_NetAddrs[rxi_numNetAddrs]),
602                                  &(myNetMTUs[rxi_numNetAddrs]))) {
603                 /* fputs(stderr, "syscall error GETMTU\n");
604                  * perror(ifr->ifr_name); */
605                 myNetMTUs[rxi_numNetAddrs] = 0;
606             }
607             if ((*rxi_syscallp) (42 /*AFSOP_GETMASK */ ,
608                                  htonl(rxi_NetAddrs[rxi_numNetAddrs]),
609                                  &(myNetMasks[rxi_numNetAddrs]))) {
610                 /* fputs(stderr, "syscall error GETMASK\n");
611                  * perror(ifr->ifr_name); */
612                 myNetMasks[rxi_numNetAddrs] = 0;
613             } else
614                 myNetMasks[rxi_numNetAddrs] =
615                     ntohl(myNetMasks[rxi_numNetAddrs]);
616             /* fprintf(stderr, "if %s mask=0x%x\n", 
617              * ifr->ifr_name, myNetMasks[rxi_numNetAddrs]); */
618         }
619
620         if (myNetMTUs[rxi_numNetAddrs] == 0) {
621             myNetMTUs[rxi_numNetAddrs] = OLD_MAX_PACKET_SIZE + RX_IPUDP_SIZE;
622 #ifdef SIOCGIFMTU
623             res = ioctl(s, SIOCGIFMTU, ifr);
624             if ((res == 0) && (ifr->ifr_metric > 128)) {        /* sanity check */
625                 myNetMTUs[rxi_numNetAddrs] = ifr->ifr_metric;
626                 /* fprintf(stderr, "if %s mtu=%d\n", 
627                  * ifr->ifr_name, ifr->ifr_metric); */
628             } else {
629                 /* fputs(stderr, "ioctl error IFMTU\n");
630                  * perror(ifr->ifr_name); */
631             }
632 #endif
633         }
634
635         if (myNetMasks[rxi_numNetAddrs] == 0) {
636             myNetMasks[rxi_numNetAddrs] =
637                 fudge_netmask(rxi_NetAddrs[rxi_numNetAddrs]);
638 #ifdef SIOCGIFNETMASK
639             res = ioctl(s, SIOCGIFNETMASK, ifr);
640             if ((res == 0)) {
641                 a = (struct sockaddr_in *)&ifr->ifr_addr;
642                 myNetMasks[rxi_numNetAddrs] = ntohl(a->sin_addr.s_addr);
643                 /* fprintf(stderr, "if %s subnetmask=0x%x\n", 
644                  * ifr->ifr_name, myNetMasks[rxi_numNetAddrs]); */
645             } else {
646                 /* fputs(stderr, "ioctl error IFMASK\n");
647                  * perror(ifr->ifr_name); */
648             }
649 #endif
650         }
651
652         if (rxi_NetAddrs[rxi_numNetAddrs] != 0x7f000001) {      /* ignore lo0 */
653             int maxsize;
654             maxsize =
655                 rxi_nRecvFrags * (myNetMTUs[rxi_numNetAddrs] - RX_IP_SIZE);
656             maxsize -= UDP_HDR_SIZE;    /* only the first frag has a UDP hdr */
657             if (rx_maxReceiveSize < maxsize)
658                 rx_maxReceiveSize = MIN(RX_MAX_PACKET_SIZE, maxsize);
659             ++rxi_numNetAddrs;
660         }
661     }
662     UNLOCK_IF;
663     close(s);
664
665     /* have to allocate at least enough to allow a single packet to reach its
666      * maximum size, so ReadPacket will work.  Allocate enough for a couple
667      * of packets to do so, for good measure */
668     {
669         int npackets, ncbufs;
670
671         rx_maxJumboRecvSize =
672             RX_HEADER_SIZE + rxi_nDgramPackets * RX_JUMBOBUFFERSIZE +
673             (rxi_nDgramPackets - 1) * RX_JUMBOHEADERSIZE;
674         rx_maxJumboRecvSize = MAX(rx_maxJumboRecvSize, rx_maxReceiveSize);
675         ncbufs = (rx_maxJumboRecvSize - RX_FIRSTBUFFERSIZE);
676         if (ncbufs > 0) {
677             ncbufs = ncbufs / RX_CBUFFERSIZE;
678             npackets = rx_initSendWindow - 1;
679             rxi_MorePackets(npackets * (ncbufs + 1));
680         }
681     }
682 }
683 #endif /* AFS_NT40_ENV */
684
685 /* Called from rxi_FindPeer, when initializing a clear rx_peer structure,
686  * to get interesting information.
687  * Curiously enough, the rx_peerHashTable_lock currently protects the
688  * Inited variable (and hence rx_GetIFInfo). When the fs suite uses
689  * pthreads, this issue will need to be revisited.
690  */
691
692 void
693 rxi_InitPeerParams(struct rx_peer *pp)
694 {
695     afs_uint32 ppaddr;
696     u_short rxmtu;
697     int ix;
698 #if defined(ADAPT_PMTU) && defined(IP_MTU)
699     int sock;
700     struct sockaddr_in addr;
701 #endif
702
703     LOCK_IF_INIT;
704     if (!Inited) {
705         UNLOCK_IF_INIT;
706         /*
707          * there's a race here since more than one thread could call
708          * rx_GetIFInfo.  The race stops in rx_GetIFInfo.
709          */
710         rx_GetIFInfo();
711     } else {
712         UNLOCK_IF_INIT;
713     }
714
715 #ifdef ADAPT_MTU
716     /* try to second-guess IP, and identify which link is most likely to
717      * be used for traffic to/from this host. */
718     ppaddr = ntohl(pp->host);
719     
720     pp->ifMTU = 0;
721     pp->timeout.sec = 2;
722     pp->rateFlag = 2;         /* start timing after two full packets */
723     /* I don't initialize these, because I presume they are bzero'd... 
724      * pp->burstSize pp->burst pp->burstWait.sec pp->burstWait.usec
725      * pp->timeout.usec */
726     
727     LOCK_IF;
728     for (ix = 0; ix < rxi_numNetAddrs; ++ix) {
729         if ((rxi_NetAddrs[ix] & myNetMasks[ix]) == (ppaddr & myNetMasks[ix])) {
730 #ifdef IFF_POINTOPOINT
731             if (myNetFlags[ix] & IFF_POINTOPOINT)
732                 pp->timeout.sec = 4;
733 #endif /* IFF_POINTOPOINT */
734             rxmtu = myNetMTUs[ix] - RX_IPUDP_SIZE;
735             if (rxmtu < RX_MIN_PACKET_SIZE)
736                 rxmtu = RX_MIN_PACKET_SIZE;
737             if (pp->ifMTU < rxmtu)
738                 pp->ifMTU = MIN(rx_MyMaxSendSize, rxmtu);
739         }
740     }
741     UNLOCK_IF;
742     if (!pp->ifMTU) {         /* not local */
743         pp->timeout.sec = 3;
744         pp->ifMTU = MIN(rx_MyMaxSendSize, RX_REMOTE_PACKET_SIZE);
745     }
746 #else /* ADAPT_MTU */
747     pp->rateFlag = 2;           /* start timing after two full packets */
748     pp->timeout.sec = 2;
749     pp->ifMTU = MIN(rx_MyMaxSendSize, OLD_MAX_PACKET_SIZE);
750 #endif /* ADAPT_MTU */
751 #if defined(ADAPT_PMTU) && defined(IP_MTU)
752     sock=socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP);
753     if (sock >= 0) {
754       addr.sin_family = AF_INET;
755       addr.sin_addr.s_addr = pp->host;
756       addr.sin_port = pp->port;
757       if (connect(sock, (struct sockaddr *)&addr, sizeof(addr)) == 0) {
758         int mtu=0;
759         socklen_t s = sizeof(mtu);
760         if (getsockopt(sock, SOL_IP, IP_MTU, &mtu, &s)== 0) {
761           pp->ifMTU = MIN(mtu - RX_IPUDP_SIZE, pp->ifMTU);
762         }
763       }
764       close(sock);
765     }
766 #endif
767     pp->ifMTU = rxi_AdjustIfMTU(pp->ifMTU);
768     pp->maxMTU = OLD_MAX_PACKET_SIZE;   /* for compatibility with old guys */
769     pp->natMTU = MIN((int)pp->ifMTU, OLD_MAX_PACKET_SIZE);
770     pp->maxDgramPackets =
771         MIN(rxi_nDgramPackets,
772             rxi_AdjustDgramPackets(rxi_nSendFrags, pp->ifMTU));
773     pp->ifDgramPackets =
774         MIN(rxi_nDgramPackets,
775             rxi_AdjustDgramPackets(rxi_nSendFrags, pp->ifMTU));
776     pp->maxDgramPackets = 1;
777     /* Initialize slow start parameters */
778     pp->MTU = MIN(pp->natMTU, pp->maxMTU);
779     pp->cwind = 1;
780     pp->nDgramPackets = 1;
781     pp->congestSeq = 0;
782 }
783
784 /* Don't expose jumobgram internals. */
785 void
786 rx_SetNoJumbo(void)
787 {
788     rx_maxReceiveSize = OLD_MAX_PACKET_SIZE;
789     rxi_nSendFrags = rxi_nRecvFrags = 1;
790 }
791
792 /* Override max MTU.  If rx_SetNoJumbo is called, it must be 
793    called before calling rx_SetMaxMTU since SetNoJumbo clobbers rx_maxReceiveSize */
794 void
795 rx_SetMaxMTU(int mtu)
796 {
797     rx_MyMaxSendSize = rx_maxReceiveSizeUser = rx_maxReceiveSize = mtu;
798 }
799
800 #if defined(HAVE_LINUX_ERRQUEUE_H) && defined(ADAPT_PMTU)
801 int
802 rxi_HandleSocketError(int socket)
803 {
804     struct msghdr msg;
805     struct cmsghdr *cmsg;
806     struct sock_extended_err *err;
807     struct sockaddr_in addr;
808     struct sockaddr *offender;
809     char controlmsgbuf[256];
810     int ret=0;
811     int code;
812
813     msg.msg_name = &addr;
814     msg.msg_namelen = sizeof(addr);
815     msg.msg_iov = NULL;
816     msg.msg_iovlen = 0;
817     msg.msg_control = controlmsgbuf;
818     msg.msg_controllen = 256;
819     msg.msg_flags = 0;
820     code = recvmsg(socket, &msg, MSG_ERRQUEUE|MSG_DONTWAIT|MSG_TRUNC);
821
822     if (code < 0 || !(msg.msg_flags & MSG_ERRQUEUE))
823         goto out;
824
825     for (cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg)) {
826        if ((char *)cmsg - controlmsgbuf > msg.msg_controllen - CMSG_SPACE(0) ||
827            (char *)cmsg - controlmsgbuf > msg.msg_controllen - CMSG_SPACE(cmsg->cmsg_len) ||
828            cmsg->cmsg_len == 0) {
829            cmsg = 0;
830            break;
831         }
832         if (cmsg->cmsg_level == SOL_IP && cmsg->cmsg_type == IP_RECVERR)
833             break;
834     }
835     if (!cmsg)
836         goto out;
837     ret=1;
838     err =(struct sock_extended_err *) CMSG_DATA(cmsg);
839     
840     if (err->ee_errno == EMSGSIZE && err->ee_info >= 68) {
841         rxi_SetPeerMtu(addr.sin_addr.s_addr, addr.sin_port,
842                        err->ee_info - RX_IPUDP_SIZE);
843     }
844     /* other DEST_UNREACH's and TIME_EXCEEDED should be dealt with too */
845     
846 out:
847     return ret;
848 }
849 #endif