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