rx: Don't clear the receive queue when out of packets
[openafs.git] / src / rx / rx.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:  Extended Remote Procedure Call */
11
12 #include <afsconfig.h>
13 #include <afs/param.h>
14
15 #ifdef KERNEL
16 # include "afs/sysincludes.h"
17 # include "afsincludes.h"
18 # ifndef UKERNEL
19 #  include "h/types.h"
20 #  include "h/time.h"
21 #  include "h/stat.h"
22 #  ifdef AFS_LINUX20_ENV
23 #   include "h/socket.h"
24 #  endif
25 #  include "netinet/in.h"
26 #  ifdef AFS_SUN5_ENV
27 #   include "netinet/ip6.h"
28 #   include "inet/common.h"
29 #   include "inet/ip.h"
30 #   include "inet/ip_ire.h"
31 #  endif
32 #  include "afs/afs_args.h"
33 #  include "afs/afs_osi.h"
34 #  ifdef RX_KERNEL_TRACE
35 #   include "rx_kcommon.h"
36 #  endif
37 #  if   defined(AFS_AIX_ENV)
38 #   include "h/systm.h"
39 #  endif
40 #  ifdef RXDEBUG
41 #   undef RXDEBUG                       /* turn off debugging */
42 #  endif /* RXDEBUG */
43 #  if defined(AFS_SGI_ENV)
44 #   include "sys/debug.h"
45 #  endif
46 # else /* !UKERNEL */
47 #  include "afs/sysincludes.h"
48 #  include "afsincludes.h"
49 # endif /* !UKERNEL */
50 # include "afs/lock.h"
51 # include "rx_kmutex.h"
52 # include "rx_kernel.h"
53 # define        AFSOP_STOP_RXCALLBACK   210     /* Stop CALLBACK process */
54 # define        AFSOP_STOP_AFS          211     /* Stop AFS process */
55 # define        AFSOP_STOP_BKG          212     /* Stop BKG process */
56 extern afs_int32 afs_termState;
57 # ifdef AFS_AIX41_ENV
58 #  include "sys/lockl.h"
59 #  include "sys/lock_def.h"
60 # endif /* AFS_AIX41_ENV */
61 # include "afs/rxgen_consts.h"
62 #else /* KERNEL */
63 # include <roken.h>
64
65 # ifdef AFS_NT40_ENV
66 #  include <afs/afsutil.h>
67 #  include <WINNT\afsreg.h>
68 # endif
69
70 # include "rx_user.h"
71 #endif /* KERNEL */
72
73 #include "rx.h"
74 #include "rx_clock.h"
75 #include "rx_queue.h"
76 #include "rx_atomic.h"
77 #include "rx_globals.h"
78 #include "rx_trace.h"
79 #include "rx_internal.h"
80 #include "rx_stats.h"
81
82 #include <afs/rxgen_consts.h>
83
84 #ifndef KERNEL
85 #ifdef AFS_PTHREAD_ENV
86 #ifndef AFS_NT40_ENV
87 int (*registerProgram) (pid_t, char *) = 0;
88 int (*swapNameProgram) (pid_t, const char *, char *) = 0;
89 #endif
90 #else
91 int (*registerProgram) (PROCESS, char *) = 0;
92 int (*swapNameProgram) (PROCESS, const char *, char *) = 0;
93 #endif
94 #endif
95
96 /* Local static routines */
97 static void rxi_DestroyConnectionNoLock(struct rx_connection *conn);
98 static void rxi_ComputeRoundTripTime(struct rx_packet *, struct rx_ackPacket *,
99                                      struct rx_call *, struct rx_peer *,
100                                      struct clock *);
101 static void rxi_Resend(struct rxevent *event, void *arg0, void *arg1,
102                        int istack);
103
104 #ifdef RX_ENABLE_LOCKS
105 static void rxi_SetAcksInTransmitQueue(struct rx_call *call);
106 #endif
107
108 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
109 struct rx_tq_debug {
110     rx_atomic_t rxi_start_aborted; /* rxi_start awoke after rxi_Send in error.*/
111     rx_atomic_t rxi_start_in_error;
112 } rx_tq_debug;
113 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
114
115 /* Constant delay time before sending an acknowledge of the last packet
116  * received.  This is to avoid sending an extra acknowledge when the
117  * client is about to make another call, anyway, or the server is
118  * about to respond.
119  *
120  * The lastAckDelay may not exceeed 400ms without causing peers to
121  * unecessarily timeout.
122  */
123 struct clock rx_lastAckDelay = {0, 400000};
124
125 /* Constant delay time before sending a soft ack when none was requested.
126  * This is to make sure we send soft acks before the sender times out,
127  * Normally we wait and send a hard ack when the receiver consumes the packet
128  *
129  * This value has been 100ms in all shipping versions of OpenAFS. Changing it
130  * will require changes to the peer's RTT calculations.
131  */
132 struct clock rx_softAckDelay = {0, 100000};
133
134 /*
135  * rxi_rpc_peer_stat_cnt counts the total number of peer stat structures
136  * currently allocated within rx.  This number is used to allocate the
137  * memory required to return the statistics when queried.
138  * Protected by the rx_rpc_stats mutex.
139  */
140
141 static unsigned int rxi_rpc_peer_stat_cnt;
142
143 /*
144  * rxi_rpc_process_stat_cnt counts the total number of local process stat
145  * structures currently allocated within rx.  The number is used to allocate
146  * the memory required to return the statistics when queried.
147  * Protected by the rx_rpc_stats mutex.
148  */
149
150 static unsigned int rxi_rpc_process_stat_cnt;
151
152 /*
153  * rxi_busyChannelError is the error to return to the application when a call
154  * channel appears busy (inferred from the receipt of RX_PACKET_TYPE_BUSY
155  * packets on the channel), and there are other call channels in the
156  * connection that are not busy. If 0, we do not return errors upon receiving
157  * busy packets; we just keep trying on the same call channel until we hit a
158  * timeout.
159  */
160 static afs_int32 rxi_busyChannelError = 0;
161
162 rx_atomic_t rx_nWaiting = RX_ATOMIC_INIT(0);
163 rx_atomic_t rx_nWaited = RX_ATOMIC_INIT(0);
164
165 #if !defined(offsetof)
166 #include <stddef.h>             /* for definition of offsetof() */
167 #endif
168
169 #ifdef RX_ENABLE_LOCKS
170 afs_kmutex_t rx_atomic_mutex;
171 #endif
172
173 /* Forward prototypes */
174 static struct rx_call * rxi_NewCall(struct rx_connection *, int);
175
176 #ifdef AFS_PTHREAD_ENV
177
178 /*
179  * Use procedural initialization of mutexes/condition variables
180  * to ease NT porting
181  */
182
183 extern afs_kmutex_t rx_quota_mutex;
184 extern afs_kmutex_t rx_pthread_mutex;
185 extern afs_kmutex_t rx_packets_mutex;
186 extern afs_kmutex_t rx_refcnt_mutex;
187 extern afs_kmutex_t des_init_mutex;
188 extern afs_kmutex_t des_random_mutex;
189 extern afs_kmutex_t rx_clock_mutex;
190 extern afs_kmutex_t rxi_connCacheMutex;
191 extern afs_kmutex_t rx_event_mutex;
192 extern afs_kmutex_t event_handler_mutex;
193 extern afs_kmutex_t listener_mutex;
194 extern afs_kmutex_t rx_if_init_mutex;
195 extern afs_kmutex_t rx_if_mutex;
196
197 extern afs_kcondvar_t rx_event_handler_cond;
198 extern afs_kcondvar_t rx_listener_cond;
199
200 static afs_kmutex_t epoch_mutex;
201 static afs_kmutex_t rx_init_mutex;
202 static afs_kmutex_t rx_debug_mutex;
203 static afs_kmutex_t rx_rpc_stats;
204
205 static void
206 rxi_InitPthread(void)
207 {
208     MUTEX_INIT(&rx_clock_mutex, "clock", MUTEX_DEFAULT, 0);
209     MUTEX_INIT(&rx_stats_mutex, "stats", MUTEX_DEFAULT, 0);
210     MUTEX_INIT(&rx_atomic_mutex, "atomic", MUTEX_DEFAULT, 0);
211     MUTEX_INIT(&rx_quota_mutex, "quota", MUTEX_DEFAULT, 0);
212     MUTEX_INIT(&rx_pthread_mutex, "pthread", MUTEX_DEFAULT, 0);
213     MUTEX_INIT(&rx_packets_mutex, "packets", MUTEX_DEFAULT, 0);
214     MUTEX_INIT(&rx_refcnt_mutex, "refcnts", MUTEX_DEFAULT, 0);
215     MUTEX_INIT(&epoch_mutex, "epoch", MUTEX_DEFAULT, 0);
216     MUTEX_INIT(&rx_init_mutex, "init", MUTEX_DEFAULT, 0);
217     MUTEX_INIT(&rx_event_mutex, "event", MUTEX_DEFAULT, 0);
218     MUTEX_INIT(&event_handler_mutex, "event handler", MUTEX_DEFAULT, 0);
219     MUTEX_INIT(&rxi_connCacheMutex, "conn cache", MUTEX_DEFAULT, 0);
220     MUTEX_INIT(&listener_mutex, "listener", MUTEX_DEFAULT, 0);
221     MUTEX_INIT(&rx_if_init_mutex, "if init", MUTEX_DEFAULT, 0);
222     MUTEX_INIT(&rx_if_mutex, "if", MUTEX_DEFAULT, 0);
223     MUTEX_INIT(&rx_debug_mutex, "debug", MUTEX_DEFAULT, 0);
224
225     CV_INIT(&rx_event_handler_cond, "evhand", CV_DEFAULT, 0);
226     CV_INIT(&rx_listener_cond, "rxlisten", CV_DEFAULT, 0);
227
228     osi_Assert(pthread_key_create(&rx_thread_id_key, NULL) == 0);
229     osi_Assert(pthread_key_create(&rx_ts_info_key, NULL) == 0);
230
231     MUTEX_INIT(&rx_rpc_stats, "rx_rpc_stats", MUTEX_DEFAULT, 0);
232     MUTEX_INIT(&rx_freePktQ_lock, "rx_freePktQ_lock", MUTEX_DEFAULT, 0);
233 #ifdef  RX_ENABLE_LOCKS
234 #ifdef RX_LOCKS_DB
235     rxdb_init();
236 #endif /* RX_LOCKS_DB */
237     MUTEX_INIT(&freeSQEList_lock, "freeSQEList lock", MUTEX_DEFAULT, 0);
238     MUTEX_INIT(&rx_freeCallQueue_lock, "rx_freeCallQueue_lock", MUTEX_DEFAULT,
239                0);
240     CV_INIT(&rx_waitingForPackets_cv, "rx_waitingForPackets_cv", CV_DEFAULT,
241             0);
242     MUTEX_INIT(&rx_peerHashTable_lock, "rx_peerHashTable_lock", MUTEX_DEFAULT,
243                0);
244     MUTEX_INIT(&rx_connHashTable_lock, "rx_connHashTable_lock", MUTEX_DEFAULT,
245                0);
246     MUTEX_INIT(&rx_serverPool_lock, "rx_serverPool_lock", MUTEX_DEFAULT, 0);
247     MUTEX_INIT(&rxi_keyCreate_lock, "rxi_keyCreate_lock", MUTEX_DEFAULT, 0);
248 #endif /* RX_ENABLE_LOCKS */
249 }
250
251 pthread_once_t rx_once_init = PTHREAD_ONCE_INIT;
252 #define INIT_PTHREAD_LOCKS osi_Assert(pthread_once(&rx_once_init, rxi_InitPthread)==0)
253 /*
254  * The rx_stats_mutex mutex protects the following global variables:
255  * rxi_lowConnRefCount
256  * rxi_lowPeerRefCount
257  * rxi_nCalls
258  * rxi_Alloccnt
259  * rxi_Allocsize
260  * rx_tq_debug
261  * rx_stats
262  */
263
264 /*
265  * The rx_quota_mutex mutex protects the following global variables:
266  * rxi_dataQuota
267  * rxi_minDeficit
268  * rxi_availProcs
269  * rxi_totalMin
270  */
271
272 /*
273  * The rx_freePktQ_lock protects the following global variables:
274  * rx_nFreePackets
275  */
276
277 /*
278  * The rx_packets_mutex mutex protects the following global variables:
279  * rx_nPackets
280  * rx_TSFPQLocalMax
281  * rx_TSFPQGlobSize
282  * rx_TSFPQMaxProcs
283  */
284
285 /*
286  * The rx_pthread_mutex mutex protects the following global variables:
287  * rxi_fcfs_thread_num
288  */
289 #else
290 #define INIT_PTHREAD_LOCKS
291 #endif
292
293
294 /* Variables for handling the minProcs implementation.  availProcs gives the
295  * number of threads available in the pool at this moment (not counting dudes
296  * executing right now).  totalMin gives the total number of procs required
297  * for handling all minProcs requests.  minDeficit is a dynamic variable
298  * tracking the # of procs required to satisfy all of the remaining minProcs
299  * demands.
300  * For fine grain locking to work, the quota check and the reservation of
301  * a server thread has to come while rxi_availProcs and rxi_minDeficit
302  * are locked. To this end, the code has been modified under #ifdef
303  * RX_ENABLE_LOCKS so that quota checks and reservation occur at the
304  * same time. A new function, ReturnToServerPool() returns the allocation.
305  *
306  * A call can be on several queue's (but only one at a time). When
307  * rxi_ResetCall wants to remove the call from a queue, it has to ensure
308  * that no one else is touching the queue. To this end, we store the address
309  * of the queue lock in the call structure (under the call lock) when we
310  * put the call on a queue, and we clear the call_queue_lock when the
311  * call is removed from a queue (once the call lock has been obtained).
312  * This allows rxi_ResetCall to safely synchronize with others wishing
313  * to manipulate the queue.
314  */
315
316 #if defined(RX_ENABLE_LOCKS)
317 static afs_kmutex_t rx_rpc_stats;
318 #endif
319
320 /* We keep a "last conn pointer" in rxi_FindConnection. The odds are
321 ** pretty good that the next packet coming in is from the same connection
322 ** as the last packet, since we're send multiple packets in a transmit window.
323 */
324 struct rx_connection *rxLastConn = 0;
325
326 #ifdef RX_ENABLE_LOCKS
327 /* The locking hierarchy for rx fine grain locking is composed of these
328  * tiers:
329  *
330  * rx_connHashTable_lock - synchronizes conn creation, rx_connHashTable access
331  * conn_call_lock - used to synchonize rx_EndCall and rx_NewCall
332  * call->lock - locks call data fields.
333  * These are independent of each other:
334  *      rx_freeCallQueue_lock
335  *      rxi_keyCreate_lock
336  * rx_serverPool_lock
337  * freeSQEList_lock
338  *
339  * serverQueueEntry->lock
340  * rx_peerHashTable_lock - locked under rx_connHashTable_lock
341  * rx_rpc_stats
342  * peer->lock - locks peer data fields.
343  * conn_data_lock - that more than one thread is not updating a conn data
344  *                  field at the same time.
345  * rx_freePktQ_lock
346  *
347  * lowest level:
348  *      multi_handle->lock
349  *      rxevent_lock
350  *      rx_packets_mutex
351  *      rx_stats_mutex
352  *      rx_refcnt_mutex
353  *      rx_atomic_mutex
354  *
355  * Do we need a lock to protect the peer field in the conn structure?
356  *      conn->peer was previously a constant for all intents and so has no
357  *      lock protecting this field. The multihomed client delta introduced
358  *      a RX code change : change the peer field in the connection structure
359  *      to that remote interface from which the last packet for this
360  *      connection was sent out. This may become an issue if further changes
361  *      are made.
362  */
363 #define SET_CALL_QUEUE_LOCK(C, L) (C)->call_queue_lock = (L)
364 #define CLEAR_CALL_QUEUE_LOCK(C) (C)->call_queue_lock = NULL
365 #ifdef RX_LOCKS_DB
366 /* rxdb_fileID is used to identify the lock location, along with line#. */
367 static int rxdb_fileID = RXDB_FILE_RX;
368 #endif /* RX_LOCKS_DB */
369 #else /* RX_ENABLE_LOCKS */
370 #define SET_CALL_QUEUE_LOCK(C, L)
371 #define CLEAR_CALL_QUEUE_LOCK(C)
372 #endif /* RX_ENABLE_LOCKS */
373 struct rx_serverQueueEntry *rx_waitForPacket = 0;
374 struct rx_serverQueueEntry *rx_waitingForPacket = 0;
375
376 /* ------------Exported Interfaces------------- */
377
378 /* This function allows rxkad to set the epoch to a suitably random number
379  * which rx_NewConnection will use in the future.  The principle purpose is to
380  * get rxnull connections to use the same epoch as the rxkad connections do, at
381  * least once the first rxkad connection is established.  This is important now
382  * that the host/port addresses aren't used in FindConnection: the uniqueness
383  * of epoch/cid matters and the start time won't do. */
384
385 #ifdef AFS_PTHREAD_ENV
386 /*
387  * This mutex protects the following global variables:
388  * rx_epoch
389  */
390
391 #define LOCK_EPOCH MUTEX_ENTER(&epoch_mutex)
392 #define UNLOCK_EPOCH MUTEX_EXIT(&epoch_mutex)
393 #else
394 #define LOCK_EPOCH
395 #define UNLOCK_EPOCH
396 #endif /* AFS_PTHREAD_ENV */
397
398 void
399 rx_SetEpoch(afs_uint32 epoch)
400 {
401     LOCK_EPOCH;
402     rx_epoch = epoch;
403     UNLOCK_EPOCH;
404 }
405
406 /* Initialize rx.  A port number may be mentioned, in which case this
407  * becomes the default port number for any service installed later.
408  * If 0 is provided for the port number, a random port will be chosen
409  * by the kernel.  Whether this will ever overlap anything in
410  * /etc/services is anybody's guess...  Returns 0 on success, -1 on
411  * error. */
412 #ifndef AFS_NT40_ENV
413 static
414 #endif
415 int rxinit_status = 1;
416 #ifdef AFS_PTHREAD_ENV
417 /*
418  * This mutex protects the following global variables:
419  * rxinit_status
420  */
421
422 #define LOCK_RX_INIT MUTEX_ENTER(&rx_init_mutex)
423 #define UNLOCK_RX_INIT MUTEX_EXIT(&rx_init_mutex)
424 #else
425 #define LOCK_RX_INIT
426 #define UNLOCK_RX_INIT
427 #endif
428
429 int
430 rx_InitHost(u_int host, u_int port)
431 {
432 #ifdef KERNEL
433     osi_timeval_t tv;
434 #else /* KERNEL */
435     struct timeval tv;
436 #endif /* KERNEL */
437     char *htable, *ptable;
438     int tmp_status;
439
440     SPLVAR;
441
442     INIT_PTHREAD_LOCKS;
443     LOCK_RX_INIT;
444     if (rxinit_status == 0) {
445         tmp_status = rxinit_status;
446         UNLOCK_RX_INIT;
447         return tmp_status;      /* Already started; return previous error code. */
448     }
449 #ifdef RXDEBUG
450     rxi_DebugInit();
451 #endif
452 #ifdef AFS_NT40_ENV
453     if (afs_winsockInit() < 0)
454         return -1;
455 #endif
456
457 #ifndef KERNEL
458     /*
459      * Initialize anything necessary to provide a non-premptive threading
460      * environment.
461      */
462     rxi_InitializeThreadSupport();
463 #endif
464
465     /* Allocate and initialize a socket for client and perhaps server
466      * connections. */
467
468     rx_socket = rxi_GetHostUDPSocket(host, (u_short) port);
469     if (rx_socket == OSI_NULLSOCKET) {
470         UNLOCK_RX_INIT;
471         return RX_ADDRINUSE;
472     }
473 #if defined(RX_ENABLE_LOCKS) && defined(KERNEL)
474 #ifdef RX_LOCKS_DB
475     rxdb_init();
476 #endif /* RX_LOCKS_DB */
477     MUTEX_INIT(&rx_stats_mutex, "rx_stats_mutex", MUTEX_DEFAULT, 0);
478     MUTEX_INIT(&rx_quota_mutex, "rx_quota_mutex", MUTEX_DEFAULT, 0);
479     MUTEX_INIT(&rx_pthread_mutex, "rx_pthread_mutex", MUTEX_DEFAULT, 0);
480     MUTEX_INIT(&rx_packets_mutex, "rx_packets_mutex", MUTEX_DEFAULT, 0);
481     MUTEX_INIT(&rx_refcnt_mutex, "rx_refcnt_mutex", MUTEX_DEFAULT, 0);
482     MUTEX_INIT(&rx_rpc_stats, "rx_rpc_stats", MUTEX_DEFAULT, 0);
483     MUTEX_INIT(&rx_freePktQ_lock, "rx_freePktQ_lock", MUTEX_DEFAULT, 0);
484     MUTEX_INIT(&freeSQEList_lock, "freeSQEList lock", MUTEX_DEFAULT, 0);
485     MUTEX_INIT(&rx_freeCallQueue_lock, "rx_freeCallQueue_lock", MUTEX_DEFAULT,
486                0);
487     CV_INIT(&rx_waitingForPackets_cv, "rx_waitingForPackets_cv", CV_DEFAULT,
488             0);
489     MUTEX_INIT(&rx_peerHashTable_lock, "rx_peerHashTable_lock", MUTEX_DEFAULT,
490                0);
491     MUTEX_INIT(&rx_connHashTable_lock, "rx_connHashTable_lock", MUTEX_DEFAULT,
492                0);
493     MUTEX_INIT(&rx_serverPool_lock, "rx_serverPool_lock", MUTEX_DEFAULT, 0);
494 #if defined(AFS_HPUX110_ENV)
495     if (!uniprocessor)
496         rx_sleepLock = alloc_spinlock(LAST_HELD_ORDER - 10, "rx_sleepLock");
497 #endif /* AFS_HPUX110_ENV */
498 #endif /* RX_ENABLE_LOCKS && KERNEL */
499
500     rxi_nCalls = 0;
501     rx_connDeadTime = 12;
502     rx_tranquil = 0;            /* reset flag */
503     rxi_ResetStatistics();
504     htable = (char *)
505         osi_Alloc(rx_hashTableSize * sizeof(struct rx_connection *));
506     PIN(htable, rx_hashTableSize * sizeof(struct rx_connection *));     /* XXXXX */
507     memset(htable, 0, rx_hashTableSize * sizeof(struct rx_connection *));
508     ptable = (char *)osi_Alloc(rx_hashTableSize * sizeof(struct rx_peer *));
509     PIN(ptable, rx_hashTableSize * sizeof(struct rx_peer *));   /* XXXXX */
510     memset(ptable, 0, rx_hashTableSize * sizeof(struct rx_peer *));
511
512     /* Malloc up a bunch of packets & buffers */
513     rx_nFreePackets = 0;
514     queue_Init(&rx_freePacketQueue);
515     rxi_NeedMorePackets = FALSE;
516     rx_nPackets = 0;    /* rx_nPackets is managed by rxi_MorePackets* */
517
518     /* enforce a minimum number of allocated packets */
519     if (rx_extraPackets < rxi_nSendFrags * rx_maxSendWindow)
520         rx_extraPackets = rxi_nSendFrags * rx_maxSendWindow;
521
522     /* allocate the initial free packet pool */
523 #ifdef RX_ENABLE_TSFPQ
524     rxi_MorePacketsTSFPQ(rx_extraPackets + RX_MAX_QUOTA + 2, RX_TS_FPQ_FLUSH_GLOBAL, 0);
525 #else /* RX_ENABLE_TSFPQ */
526     rxi_MorePackets(rx_extraPackets + RX_MAX_QUOTA + 2);        /* fudge */
527 #endif /* RX_ENABLE_TSFPQ */
528     rx_CheckPackets();
529
530     NETPRI;
531
532     clock_Init();
533
534 #if defined(AFS_NT40_ENV) && !defined(AFS_PTHREAD_ENV)
535     tv.tv_sec = clock_now.sec;
536     tv.tv_usec = clock_now.usec;
537     srand((unsigned int)tv.tv_usec);
538 #else
539     osi_GetTime(&tv);
540 #endif
541     if (port) {
542         rx_port = port;
543     } else {
544 #if defined(KERNEL) && !defined(UKERNEL)
545         /* Really, this should never happen in a real kernel */
546         rx_port = 0;
547 #else
548         struct sockaddr_in addr;
549 #ifdef AFS_NT40_ENV
550         int addrlen = sizeof(addr);
551 #else
552         socklen_t addrlen = sizeof(addr);
553 #endif
554         if (getsockname((intptr_t)rx_socket, (struct sockaddr *)&addr, &addrlen)) {
555             rx_Finalize();
556             return -1;
557         }
558         rx_port = addr.sin_port;
559 #endif
560     }
561     rx_stats.minRtt.sec = 9999999;
562 #ifdef  KERNEL
563     rx_SetEpoch(tv.tv_sec | 0x80000000);
564 #else
565     rx_SetEpoch(tv.tv_sec);     /* Start time of this package, rxkad
566                                  * will provide a randomer value. */
567 #endif
568     MUTEX_ENTER(&rx_quota_mutex);
569     rxi_dataQuota += rx_extraQuota; /* + extra pkts caller asked to rsrv */
570     MUTEX_EXIT(&rx_quota_mutex);
571     /* *Slightly* random start time for the cid.  This is just to help
572      * out with the hashing function at the peer */
573     rx_nextCid = ((tv.tv_sec ^ tv.tv_usec) << RX_CIDSHIFT);
574     rx_connHashTable = (struct rx_connection **)htable;
575     rx_peerHashTable = (struct rx_peer **)ptable;
576
577     rx_hardAckDelay.sec = 0;
578     rx_hardAckDelay.usec = 100000;      /* 100 milliseconds */
579
580     rxevent_Init(20, rxi_ReScheduleEvents);
581
582     /* Initialize various global queues */
583     queue_Init(&rx_idleServerQueue);
584     queue_Init(&rx_incomingCallQueue);
585     queue_Init(&rx_freeCallQueue);
586
587 #if defined(AFS_NT40_ENV) && !defined(KERNEL)
588     /* Initialize our list of usable IP addresses. */
589     rx_GetIFInfo();
590 #endif
591
592 #if defined(RXK_LISTENER_ENV) || !defined(KERNEL)
593     /* Start listener process (exact function is dependent on the
594      * implementation environment--kernel or user space) */
595     rxi_StartListener();
596 #endif
597
598     USERPRI;
599     tmp_status = rxinit_status = 0;
600     UNLOCK_RX_INIT;
601     return tmp_status;
602 }
603
604 int
605 rx_Init(u_int port)
606 {
607     return rx_InitHost(htonl(INADDR_ANY), port);
608 }
609
610 /* RTT Timer
611  * ---------
612  *
613  * The rxi_rto functions implement a TCP (RFC2988) style algorithm for
614  * maintaing the round trip timer.
615  *
616  */
617
618 /*!
619  * Start a new RTT timer for a given call and packet.
620  *
621  * There must be no resendEvent already listed for this call, otherwise this
622  * will leak events - intended for internal use within the RTO code only
623  *
624  * @param[in] call
625  *      the RX call to start the timer for
626  * @param[in] lastPacket
627  *      a flag indicating whether the last packet has been sent or not
628  *
629  * @pre call must be locked before calling this function
630  *
631  */
632 static_inline void
633 rxi_rto_startTimer(struct rx_call *call, int lastPacket, int istack)
634 {
635     struct clock now, retryTime;
636
637     clock_GetTime(&now);
638     retryTime = now;
639
640     clock_Add(&retryTime, &call->rto);
641
642     /* If we're sending the last packet, and we're the client, then the server
643      * may wait for an additional 400ms before returning the ACK, wait for it
644      * rather than hitting a timeout */
645     if (lastPacket && call->conn->type == RX_CLIENT_CONNECTION)
646         clock_Addmsec(&retryTime, 400);
647
648     MUTEX_ENTER(&rx_refcnt_mutex);
649     CALL_HOLD(call, RX_CALL_REFCOUNT_RESEND);
650     MUTEX_EXIT(&rx_refcnt_mutex);
651     call->resendEvent = rxevent_PostNow2(&retryTime, &now, rxi_Resend,
652                                          call, 0, istack);
653 }
654
655 /*!
656  * Cancel an RTT timer for a given call.
657  *
658  *
659  * @param[in] call
660  *      the RX call to cancel the timer for
661  *
662  * @pre call must be locked before calling this function
663  *
664  */
665
666 static_inline void
667 rxi_rto_cancel(struct rx_call *call)
668 {
669     if (!call->resendEvent)
670         return;
671
672     rxevent_Cancel(call->resendEvent, call, RX_CALL_REFCOUNT_RESEND);
673 }
674
675 /*!
676  * Tell the RTO timer that we have sent a packet.
677  *
678  * If the timer isn't already running, then start it. If the timer is running,
679  * then do nothing.
680  *
681  * @param[in] call
682  *      the RX call that the packet has been sent on
683  * @param[in] lastPacket
684  *      A flag which is true if this is the last packet for the call
685  *
686  * @pre The call must be locked before calling this function
687  *
688  */
689
690 static_inline void
691 rxi_rto_packet_sent(struct rx_call *call, int lastPacket, int istack)
692 {
693     if (call->resendEvent)
694         return;
695
696     rxi_rto_startTimer(call, lastPacket, istack);
697 }
698
699 /*!
700  * Tell the RTO timer that we have received an new ACK message
701  *
702  * This function should be called whenever a call receives an ACK that
703  * acknowledges new packets. Whatever happens, we stop the current timer.
704  * If there are unacked packets in the queue which have been sent, then
705  * we restart the timer from now. Otherwise, we leave it stopped.
706  *
707  * @param[in] call
708  *      the RX call that the ACK has been received on
709  */
710
711 static_inline void
712 rxi_rto_packet_acked(struct rx_call *call, int istack)
713 {
714     struct rx_packet *p, *nxp;
715
716     rxi_rto_cancel(call);
717
718     if (queue_IsEmpty(&call->tq))
719         return;
720
721     for (queue_Scan(&call->tq, p, nxp, rx_packet)) {
722         if (p->header.seq > call->tfirst + call->twind)
723             return;
724
725         if (!(p->flags & RX_PKTFLAG_ACKED) && p->flags & RX_PKTFLAG_SENT) {
726             rxi_rto_startTimer(call, p->header.flags & RX_LAST_PACKET, istack);
727             return;
728         }
729     }
730 }
731
732
733 /**
734  * Set an initial round trip timeout for a peer connection
735  *
736  * @param[in] secs The timeout to set in seconds
737  */
738
739 void
740 rx_rto_setPeerTimeoutSecs(struct rx_peer *peer, int secs) {
741     peer->rtt = secs * 8000;
742 }
743
744 /**
745  * Sets the error generated when a busy call channel is detected.
746  *
747  * @param[in] error The error to return for a call on a busy channel.
748  *
749  * @pre Neither rx_Init nor rx_InitHost have been called yet
750  */
751 void
752 rx_SetBusyChannelError(afs_int32 error)
753 {
754     osi_Assert(rxinit_status != 0);
755     rxi_busyChannelError = error;
756 }
757
758 /* called with unincremented nRequestsRunning to see if it is OK to start
759  * a new thread in this service.  Could be "no" for two reasons: over the
760  * max quota, or would prevent others from reaching their min quota.
761  */
762 #ifdef RX_ENABLE_LOCKS
763 /* This verion of QuotaOK reserves quota if it's ok while the
764  * rx_serverPool_lock is held.  Return quota using ReturnToServerPool().
765  */
766 static int
767 QuotaOK(struct rx_service *aservice)
768 {
769     /* check if over max quota */
770     if (aservice->nRequestsRunning >= aservice->maxProcs) {
771         return 0;
772     }
773
774     /* under min quota, we're OK */
775     /* otherwise, can use only if there are enough to allow everyone
776      * to go to their min quota after this guy starts.
777      */
778
779     MUTEX_ENTER(&rx_quota_mutex);
780     if ((aservice->nRequestsRunning < aservice->minProcs)
781         || (rxi_availProcs > rxi_minDeficit)) {
782         aservice->nRequestsRunning++;
783         /* just started call in minProcs pool, need fewer to maintain
784          * guarantee */
785         if (aservice->nRequestsRunning <= aservice->minProcs)
786             rxi_minDeficit--;
787         rxi_availProcs--;
788         MUTEX_EXIT(&rx_quota_mutex);
789         return 1;
790     }
791     MUTEX_EXIT(&rx_quota_mutex);
792
793     return 0;
794 }
795
796 static void
797 ReturnToServerPool(struct rx_service *aservice)
798 {
799     aservice->nRequestsRunning--;
800     MUTEX_ENTER(&rx_quota_mutex);
801     if (aservice->nRequestsRunning < aservice->minProcs)
802         rxi_minDeficit++;
803     rxi_availProcs++;
804     MUTEX_EXIT(&rx_quota_mutex);
805 }
806
807 #else /* RX_ENABLE_LOCKS */
808 static int
809 QuotaOK(struct rx_service *aservice)
810 {
811     int rc = 0;
812     /* under min quota, we're OK */
813     if (aservice->nRequestsRunning < aservice->minProcs)
814         return 1;
815
816     /* check if over max quota */
817     if (aservice->nRequestsRunning >= aservice->maxProcs)
818         return 0;
819
820     /* otherwise, can use only if there are enough to allow everyone
821      * to go to their min quota after this guy starts.
822      */
823     MUTEX_ENTER(&rx_quota_mutex);
824     if (rxi_availProcs > rxi_minDeficit)
825         rc = 1;
826     MUTEX_EXIT(&rx_quota_mutex);
827     return rc;
828 }
829 #endif /* RX_ENABLE_LOCKS */
830
831 #ifndef KERNEL
832 /* Called by rx_StartServer to start up lwp's to service calls.
833    NExistingProcs gives the number of procs already existing, and which
834    therefore needn't be created. */
835 static void
836 rxi_StartServerProcs(int nExistingProcs)
837 {
838     struct rx_service *service;
839     int i;
840     int maxdiff = 0;
841     int nProcs = 0;
842
843     /* For each service, reserve N processes, where N is the "minimum"
844      * number of processes that MUST be able to execute a request in parallel,
845      * at any time, for that process.  Also compute the maximum difference
846      * between any service's maximum number of processes that can run
847      * (i.e. the maximum number that ever will be run, and a guarantee
848      * that this number will run if other services aren't running), and its
849      * minimum number.  The result is the extra number of processes that
850      * we need in order to provide the latter guarantee */
851     for (i = 0; i < RX_MAX_SERVICES; i++) {
852         int diff;
853         service = rx_services[i];
854         if (service == (struct rx_service *)0)
855             break;
856         nProcs += service->minProcs;
857         diff = service->maxProcs - service->minProcs;
858         if (diff > maxdiff)
859             maxdiff = diff;
860     }
861     nProcs += maxdiff;          /* Extra processes needed to allow max number requested to run in any given service, under good conditions */
862     nProcs -= nExistingProcs;   /* Subtract the number of procs that were previously created for use as server procs */
863     for (i = 0; i < nProcs; i++) {
864         rxi_StartServerProc(rx_ServerProc, rx_stackSize);
865     }
866 }
867 #endif /* KERNEL */
868
869 #ifdef AFS_NT40_ENV
870 /* This routine is only required on Windows */
871 void
872 rx_StartClientThread(void)
873 {
874 #ifdef AFS_PTHREAD_ENV
875     pthread_t pid;
876     pid = pthread_self();
877 #endif /* AFS_PTHREAD_ENV */
878 }
879 #endif /* AFS_NT40_ENV */
880
881 /* This routine must be called if any services are exported.  If the
882  * donateMe flag is set, the calling process is donated to the server
883  * process pool */
884 void
885 rx_StartServer(int donateMe)
886 {
887     struct rx_service *service;
888     int i;
889     SPLVAR;
890     clock_NewTime();
891
892     NETPRI;
893     /* Start server processes, if necessary (exact function is dependent
894      * on the implementation environment--kernel or user space).  DonateMe
895      * will be 1 if there is 1 pre-existing proc, i.e. this one.  In this
896      * case, one less new proc will be created rx_StartServerProcs.
897      */
898     rxi_StartServerProcs(donateMe);
899
900     /* count up the # of threads in minProcs, and add set the min deficit to
901      * be that value, too.
902      */
903     for (i = 0; i < RX_MAX_SERVICES; i++) {
904         service = rx_services[i];
905         if (service == (struct rx_service *)0)
906             break;
907         MUTEX_ENTER(&rx_quota_mutex);
908         rxi_totalMin += service->minProcs;
909         /* below works even if a thread is running, since minDeficit would
910          * still have been decremented and later re-incremented.
911          */
912         rxi_minDeficit += service->minProcs;
913         MUTEX_EXIT(&rx_quota_mutex);
914     }
915
916     /* Turn on reaping of idle server connections */
917     rxi_ReapConnections(NULL, NULL, NULL);
918
919     USERPRI;
920
921     if (donateMe) {
922 #ifndef AFS_NT40_ENV
923 #ifndef KERNEL
924         char name[32];
925         static int nProcs;
926 #ifdef AFS_PTHREAD_ENV
927         pid_t pid;
928         pid = afs_pointer_to_int(pthread_self());
929 #else /* AFS_PTHREAD_ENV */
930         PROCESS pid;
931         LWP_CurrentProcess(&pid);
932 #endif /* AFS_PTHREAD_ENV */
933
934         sprintf(name, "srv_%d", ++nProcs);
935         if (registerProgram)
936             (*registerProgram) (pid, name);
937 #endif /* KERNEL */
938 #endif /* AFS_NT40_ENV */
939         rx_ServerProc(NULL);    /* Never returns */
940     }
941 #ifdef RX_ENABLE_TSFPQ
942     /* no use leaving packets around in this thread's local queue if
943      * it isn't getting donated to the server thread pool.
944      */
945     rxi_FlushLocalPacketsTSFPQ();
946 #endif /* RX_ENABLE_TSFPQ */
947     return;
948 }
949
950 /* Create a new client connection to the specified service, using the
951  * specified security object to implement the security model for this
952  * connection. */
953 struct rx_connection *
954 rx_NewConnection(afs_uint32 shost, u_short sport, u_short sservice,
955                  struct rx_securityClass *securityObject,
956                  int serviceSecurityIndex)
957 {
958     int hashindex, i;
959     afs_int32 cid;
960     struct rx_connection *conn;
961
962     SPLVAR;
963
964     clock_NewTime();
965     dpf(("rx_NewConnection(host %x, port %u, service %u, securityObject %p, "
966          "serviceSecurityIndex %d)\n",
967          ntohl(shost), ntohs(sport), sservice, securityObject,
968          serviceSecurityIndex));
969
970     /* Vasilsi said: "NETPRI protects Cid and Alloc", but can this be true in
971      * the case of kmem_alloc? */
972     conn = rxi_AllocConnection();
973 #ifdef  RX_ENABLE_LOCKS
974     MUTEX_INIT(&conn->conn_call_lock, "conn call lock", MUTEX_DEFAULT, 0);
975     MUTEX_INIT(&conn->conn_data_lock, "conn data lock", MUTEX_DEFAULT, 0);
976     CV_INIT(&conn->conn_call_cv, "conn call cv", CV_DEFAULT, 0);
977 #endif
978     NETPRI;
979     MUTEX_ENTER(&rx_connHashTable_lock);
980     cid = (rx_nextCid += RX_MAXCALLS);
981     conn->type = RX_CLIENT_CONNECTION;
982     conn->cid = cid;
983     conn->epoch = rx_epoch;
984     conn->peer = rxi_FindPeer(shost, sport, 0, 1);
985     conn->serviceId = sservice;
986     conn->securityObject = securityObject;
987     conn->securityData = (void *) 0;
988     conn->securityIndex = serviceSecurityIndex;
989     rx_SetConnDeadTime(conn, rx_connDeadTime);
990     rx_SetConnSecondsUntilNatPing(conn, 0);
991     conn->ackRate = RX_FAST_ACK_RATE;
992     conn->nSpecific = 0;
993     conn->specific = NULL;
994     conn->challengeEvent = NULL;
995     conn->delayedAbortEvent = NULL;
996     conn->abortCount = 0;
997     conn->error = 0;
998     for (i = 0; i < RX_MAXCALLS; i++) {
999         conn->twind[i] = rx_initSendWindow;
1000         conn->rwind[i] = rx_initReceiveWindow;
1001         conn->lastBusy[i] = 0;
1002     }
1003
1004     RXS_NewConnection(securityObject, conn);
1005     hashindex =
1006         CONN_HASH(shost, sport, conn->cid, conn->epoch, RX_CLIENT_CONNECTION);
1007
1008     conn->refCount++;           /* no lock required since only this thread knows... */
1009     conn->next = rx_connHashTable[hashindex];
1010     rx_connHashTable[hashindex] = conn;
1011     if (rx_stats_active)
1012         rx_atomic_inc(&rx_stats.nClientConns);
1013     MUTEX_EXIT(&rx_connHashTable_lock);
1014     USERPRI;
1015     return conn;
1016 }
1017
1018 /**
1019  * Ensure a connection's timeout values are valid.
1020  *
1021  * @param[in] conn The connection to check
1022  *
1023  * @post conn->secondUntilDead <= conn->idleDeadTime <= conn->hardDeadTime,
1024  *       unless idleDeadTime and/or hardDeadTime are not set
1025  * @internal
1026  */
1027 static void
1028 rxi_CheckConnTimeouts(struct rx_connection *conn)
1029 {
1030     /* a connection's timeouts must have the relationship
1031      * deadTime <= idleDeadTime <= hardDeadTime. Otherwise, for example, a
1032      * total loss of network to a peer may cause an idle timeout instead of a
1033      * dead timeout, simply because the idle timeout gets hit first. Also set
1034      * a minimum deadTime of 6, just to ensure it doesn't get set too low. */
1035     /* this logic is slightly complicated by the fact that
1036      * idleDeadTime/hardDeadTime may not be set at all, but it's not too bad.
1037      */
1038     conn->secondsUntilDead = MAX(conn->secondsUntilDead, 6);
1039     if (conn->idleDeadTime) {
1040         conn->idleDeadTime = MAX(conn->idleDeadTime, conn->secondsUntilDead);
1041     }
1042     if (conn->hardDeadTime) {
1043         if (conn->idleDeadTime) {
1044             conn->hardDeadTime = MAX(conn->idleDeadTime, conn->hardDeadTime);
1045         } else {
1046             conn->hardDeadTime = MAX(conn->secondsUntilDead, conn->hardDeadTime);
1047         }
1048     }
1049 }
1050
1051 void
1052 rx_SetConnDeadTime(struct rx_connection *conn, int seconds)
1053 {
1054     /* The idea is to set the dead time to a value that allows several
1055      * keepalives to be dropped without timing out the connection. */
1056     conn->secondsUntilDead = seconds;
1057     rxi_CheckConnTimeouts(conn);
1058     conn->secondsUntilPing = conn->secondsUntilDead / 6;
1059 }
1060
1061 void
1062 rx_SetConnHardDeadTime(struct rx_connection *conn, int seconds)
1063 {
1064     conn->hardDeadTime = seconds;
1065     rxi_CheckConnTimeouts(conn);
1066 }
1067
1068 void
1069 rx_SetConnIdleDeadTime(struct rx_connection *conn, int seconds)
1070 {
1071     conn->idleDeadTime = seconds;
1072     rxi_CheckConnTimeouts(conn);
1073 }
1074
1075 int rxi_lowPeerRefCount = 0;
1076 int rxi_lowConnRefCount = 0;
1077
1078 /*
1079  * Cleanup a connection that was destroyed in rxi_DestroyConnectioNoLock.
1080  * NOTE: must not be called with rx_connHashTable_lock held.
1081  */
1082 static void
1083 rxi_CleanupConnection(struct rx_connection *conn)
1084 {
1085     /* Notify the service exporter, if requested, that this connection
1086      * is being destroyed */
1087     if (conn->type == RX_SERVER_CONNECTION && conn->service->destroyConnProc)
1088         (*conn->service->destroyConnProc) (conn);
1089
1090     /* Notify the security module that this connection is being destroyed */
1091     RXS_DestroyConnection(conn->securityObject, conn);
1092
1093     /* If this is the last connection using the rx_peer struct, set its
1094      * idle time to now. rxi_ReapConnections will reap it if it's still
1095      * idle (refCount == 0) after rx_idlePeerTime (60 seconds) have passed.
1096      */
1097     MUTEX_ENTER(&rx_peerHashTable_lock);
1098     if (conn->peer->refCount < 2) {
1099         conn->peer->idleWhen = clock_Sec();
1100         if (conn->peer->refCount < 1) {
1101             conn->peer->refCount = 1;
1102             if (rx_stats_active) {
1103                 MUTEX_ENTER(&rx_stats_mutex);
1104                 rxi_lowPeerRefCount++;
1105                 MUTEX_EXIT(&rx_stats_mutex);
1106             }
1107         }
1108     }
1109     conn->peer->refCount--;
1110     MUTEX_EXIT(&rx_peerHashTable_lock);
1111
1112     if (rx_stats_active)
1113     {
1114         if (conn->type == RX_SERVER_CONNECTION)
1115             rx_atomic_dec(&rx_stats.nServerConns);
1116         else
1117             rx_atomic_dec(&rx_stats.nClientConns);
1118     }
1119 #ifndef KERNEL
1120     if (conn->specific) {
1121         int i;
1122         for (i = 0; i < conn->nSpecific; i++) {
1123             if (conn->specific[i] && rxi_keyCreate_destructor[i])
1124                 (*rxi_keyCreate_destructor[i]) (conn->specific[i]);
1125             conn->specific[i] = NULL;
1126         }
1127         free(conn->specific);
1128     }
1129     conn->specific = NULL;
1130     conn->nSpecific = 0;
1131 #endif /* !KERNEL */
1132
1133     MUTEX_DESTROY(&conn->conn_call_lock);
1134     MUTEX_DESTROY(&conn->conn_data_lock);
1135     CV_DESTROY(&conn->conn_call_cv);
1136
1137     rxi_FreeConnection(conn);
1138 }
1139
1140 /* Destroy the specified connection */
1141 void
1142 rxi_DestroyConnection(struct rx_connection *conn)
1143 {
1144     MUTEX_ENTER(&rx_connHashTable_lock);
1145     rxi_DestroyConnectionNoLock(conn);
1146     /* conn should be at the head of the cleanup list */
1147     if (conn == rx_connCleanup_list) {
1148         rx_connCleanup_list = rx_connCleanup_list->next;
1149         MUTEX_EXIT(&rx_connHashTable_lock);
1150         rxi_CleanupConnection(conn);
1151     }
1152 #ifdef RX_ENABLE_LOCKS
1153     else {
1154         MUTEX_EXIT(&rx_connHashTable_lock);
1155     }
1156 #endif /* RX_ENABLE_LOCKS */
1157 }
1158
1159 static void
1160 rxi_DestroyConnectionNoLock(struct rx_connection *conn)
1161 {
1162     struct rx_connection **conn_ptr;
1163     int havecalls = 0;
1164     struct rx_packet *packet;
1165     int i;
1166     SPLVAR;
1167
1168     clock_NewTime();
1169
1170     NETPRI;
1171     MUTEX_ENTER(&conn->conn_data_lock);
1172     MUTEX_ENTER(&rx_refcnt_mutex);
1173     if (conn->refCount > 0)
1174         conn->refCount--;
1175     else {
1176         if (rx_stats_active) {
1177             MUTEX_ENTER(&rx_stats_mutex);
1178             rxi_lowConnRefCount++;
1179             MUTEX_EXIT(&rx_stats_mutex);
1180         }
1181     }
1182
1183     if ((conn->refCount > 0) || (conn->flags & RX_CONN_BUSY)) {
1184         /* Busy; wait till the last guy before proceeding */
1185         MUTEX_EXIT(&rx_refcnt_mutex);
1186         MUTEX_EXIT(&conn->conn_data_lock);
1187         USERPRI;
1188         return;
1189     }
1190
1191     /* If the client previously called rx_NewCall, but it is still
1192      * waiting, treat this as a running call, and wait to destroy the
1193      * connection later when the call completes. */
1194     if ((conn->type == RX_CLIENT_CONNECTION)
1195         && (conn->flags & (RX_CONN_MAKECALL_WAITING|RX_CONN_MAKECALL_ACTIVE))) {
1196         conn->flags |= RX_CONN_DESTROY_ME;
1197         MUTEX_EXIT(&conn->conn_data_lock);
1198         USERPRI;
1199         return;
1200     }
1201     MUTEX_EXIT(&rx_refcnt_mutex);
1202     MUTEX_EXIT(&conn->conn_data_lock);
1203
1204     /* Check for extant references to this connection */
1205     MUTEX_ENTER(&conn->conn_call_lock);
1206     for (i = 0; i < RX_MAXCALLS; i++) {
1207         struct rx_call *call = conn->call[i];
1208         if (call) {
1209             havecalls = 1;
1210             if (conn->type == RX_CLIENT_CONNECTION) {
1211                 MUTEX_ENTER(&call->lock);
1212                 if (call->delayedAckEvent) {
1213                     /* Push the final acknowledgment out now--there
1214                      * won't be a subsequent call to acknowledge the
1215                      * last reply packets */
1216                     rxevent_Cancel(call->delayedAckEvent, call,
1217                                    RX_CALL_REFCOUNT_DELAY);
1218                     if (call->state == RX_STATE_PRECALL
1219                         || call->state == RX_STATE_ACTIVE) {
1220                         rxi_SendAck(call, 0, 0, RX_ACK_DELAY, 0);
1221                     } else {
1222                         rxi_AckAll(NULL, call, 0);
1223                     }
1224                 }
1225                 MUTEX_EXIT(&call->lock);
1226             }
1227         }
1228     }
1229     MUTEX_EXIT(&conn->conn_call_lock);
1230
1231 #ifdef RX_ENABLE_LOCKS
1232     if (!havecalls) {
1233         if (MUTEX_TRYENTER(&conn->conn_data_lock)) {
1234             MUTEX_EXIT(&conn->conn_data_lock);
1235         } else {
1236             /* Someone is accessing a packet right now. */
1237             havecalls = 1;
1238         }
1239     }
1240 #endif /* RX_ENABLE_LOCKS */
1241
1242     if (havecalls) {
1243         /* Don't destroy the connection if there are any call
1244          * structures still in use */
1245         MUTEX_ENTER(&conn->conn_data_lock);
1246         conn->flags |= RX_CONN_DESTROY_ME;
1247         MUTEX_EXIT(&conn->conn_data_lock);
1248         USERPRI;
1249         return;
1250     }
1251
1252     if (conn->natKeepAliveEvent) {
1253         rxi_NatKeepAliveOff(conn);
1254     }
1255
1256     if (conn->delayedAbortEvent) {
1257         rxevent_Cancel(conn->delayedAbortEvent, (struct rx_call *)0, 0);
1258         packet = rxi_AllocPacket(RX_PACKET_CLASS_SPECIAL);
1259         if (packet) {
1260             MUTEX_ENTER(&conn->conn_data_lock);
1261             rxi_SendConnectionAbort(conn, packet, 0, 1);
1262             MUTEX_EXIT(&conn->conn_data_lock);
1263             rxi_FreePacket(packet);
1264         }
1265     }
1266
1267     /* Remove from connection hash table before proceeding */
1268     conn_ptr =
1269         &rx_connHashTable[CONN_HASH
1270                           (peer->host, peer->port, conn->cid, conn->epoch,
1271                            conn->type)];
1272     for (; *conn_ptr; conn_ptr = &(*conn_ptr)->next) {
1273         if (*conn_ptr == conn) {
1274             *conn_ptr = conn->next;
1275             break;
1276         }
1277     }
1278     /* if the conn that we are destroying was the last connection, then we
1279      * clear rxLastConn as well */
1280     if (rxLastConn == conn)
1281         rxLastConn = 0;
1282
1283     /* Make sure the connection is completely reset before deleting it. */
1284     /* get rid of pending events that could zap us later */
1285     if (conn->challengeEvent)
1286         rxevent_Cancel(conn->challengeEvent, (struct rx_call *)0, 0);
1287     if (conn->checkReachEvent)
1288         rxevent_Cancel(conn->checkReachEvent, (struct rx_call *)0, 0);
1289     if (conn->natKeepAliveEvent)
1290         rxevent_Cancel(conn->natKeepAliveEvent, (struct rx_call *)0, 0);
1291
1292     /* Add the connection to the list of destroyed connections that
1293      * need to be cleaned up. This is necessary to avoid deadlocks
1294      * in the routines we call to inform others that this connection is
1295      * being destroyed. */
1296     conn->next = rx_connCleanup_list;
1297     rx_connCleanup_list = conn;
1298 }
1299
1300 /* Externally available version */
1301 void
1302 rx_DestroyConnection(struct rx_connection *conn)
1303 {
1304     SPLVAR;
1305
1306     NETPRI;
1307     rxi_DestroyConnection(conn);
1308     USERPRI;
1309 }
1310
1311 void
1312 rx_GetConnection(struct rx_connection *conn)
1313 {
1314     SPLVAR;
1315
1316     NETPRI;
1317     MUTEX_ENTER(&rx_refcnt_mutex);
1318     conn->refCount++;
1319     MUTEX_EXIT(&rx_refcnt_mutex);
1320     USERPRI;
1321 }
1322
1323 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
1324 /* Wait for the transmit queue to no longer be busy.
1325  * requires the call->lock to be held */
1326 void
1327 rxi_WaitforTQBusy(struct rx_call *call) {
1328     while (!call->error && (call->flags & RX_CALL_TQ_BUSY)) {
1329         call->flags |= RX_CALL_TQ_WAIT;
1330         call->tqWaiters++;
1331 #ifdef RX_ENABLE_LOCKS
1332         osirx_AssertMine(&call->lock, "rxi_WaitforTQ lock");
1333         CV_WAIT(&call->cv_tq, &call->lock);
1334 #else /* RX_ENABLE_LOCKS */
1335         osi_rxSleep(&call->tq);
1336 #endif /* RX_ENABLE_LOCKS */
1337         call->tqWaiters--;
1338         if (call->tqWaiters == 0) {
1339             call->flags &= ~RX_CALL_TQ_WAIT;
1340         }
1341     }
1342 }
1343 #endif
1344
1345 static void
1346 rxi_WakeUpTransmitQueue(struct rx_call *call)
1347 {
1348     if (call->tqWaiters || (call->flags & RX_CALL_TQ_WAIT)) {
1349         dpf(("call %"AFS_PTR_FMT" has %d waiters and flags %d\n",
1350              call, call->tqWaiters, call->flags));
1351 #ifdef RX_ENABLE_LOCKS
1352         osirx_AssertMine(&call->lock, "rxi_Start start");
1353         CV_BROADCAST(&call->cv_tq);
1354 #else /* RX_ENABLE_LOCKS */
1355         osi_rxWakeup(&call->tq);
1356 #endif /* RX_ENABLE_LOCKS */
1357     }
1358 }
1359
1360 /* Start a new rx remote procedure call, on the specified connection.
1361  * If wait is set to 1, wait for a free call channel; otherwise return
1362  * 0.  Maxtime gives the maximum number of seconds this call may take,
1363  * after rx_NewCall returns.  After this time interval, a call to any
1364  * of rx_SendData, rx_ReadData, etc. will fail with RX_CALL_TIMEOUT.
1365  * For fine grain locking, we hold the conn_call_lock in order to
1366  * to ensure that we don't get signalle after we found a call in an active
1367  * state and before we go to sleep.
1368  */
1369 struct rx_call *
1370 rx_NewCall(struct rx_connection *conn)
1371 {
1372     int i, wait, ignoreBusy = 1;
1373     struct rx_call *call;
1374     struct clock queueTime;
1375     afs_uint32 leastBusy = 0;
1376     SPLVAR;
1377
1378     clock_NewTime();
1379     dpf(("rx_NewCall(conn %"AFS_PTR_FMT")\n", conn));
1380
1381     NETPRI;
1382     clock_GetTime(&queueTime);
1383     /*
1384      * Check if there are others waiting for a new call.
1385      * If so, let them go first to avoid starving them.
1386      * This is a fairly simple scheme, and might not be
1387      * a complete solution for large numbers of waiters.
1388      *
1389      * makeCallWaiters keeps track of the number of
1390      * threads waiting to make calls and the
1391      * RX_CONN_MAKECALL_WAITING flag bit is used to
1392      * indicate that there are indeed calls waiting.
1393      * The flag is set when the waiter is incremented.
1394      * It is only cleared when makeCallWaiters is 0.
1395      * This prevents us from accidently destroying the
1396      * connection while it is potentially about to be used.
1397      */
1398     MUTEX_ENTER(&conn->conn_call_lock);
1399     MUTEX_ENTER(&conn->conn_data_lock);
1400     while (conn->flags & RX_CONN_MAKECALL_ACTIVE) {
1401         conn->flags |= RX_CONN_MAKECALL_WAITING;
1402         conn->makeCallWaiters++;
1403         MUTEX_EXIT(&conn->conn_data_lock);
1404
1405 #ifdef  RX_ENABLE_LOCKS
1406         CV_WAIT(&conn->conn_call_cv, &conn->conn_call_lock);
1407 #else
1408         osi_rxSleep(conn);
1409 #endif
1410         MUTEX_ENTER(&conn->conn_data_lock);
1411         conn->makeCallWaiters--;
1412         if (conn->makeCallWaiters == 0)
1413             conn->flags &= ~RX_CONN_MAKECALL_WAITING;
1414     }
1415
1416     /* We are now the active thread in rx_NewCall */
1417     conn->flags |= RX_CONN_MAKECALL_ACTIVE;
1418     MUTEX_EXIT(&conn->conn_data_lock);
1419
1420     for (;;) {
1421         wait = 1;
1422
1423         for (i = 0; i < RX_MAXCALLS; i++) {
1424             call = conn->call[i];
1425             if (call) {
1426                 if (!ignoreBusy && conn->lastBusy[i] != leastBusy) {
1427                     /* we're not ignoring busy call slots; only look at the
1428                      * call slot that is the "least" busy */
1429                     continue;
1430                 }
1431
1432                 if (call->state == RX_STATE_DALLY) {
1433                     MUTEX_ENTER(&call->lock);
1434                     if (call->state == RX_STATE_DALLY) {
1435                         if (ignoreBusy && conn->lastBusy[i]) {
1436                             /* if we're ignoring busy call slots, skip any ones that
1437                              * have lastBusy set */
1438                             if (leastBusy == 0 || conn->lastBusy[i] < leastBusy) {
1439                                 leastBusy = conn->lastBusy[i];
1440                             }
1441                             MUTEX_EXIT(&call->lock);
1442                             continue;
1443                         }
1444
1445                         /*
1446                          * We are setting the state to RX_STATE_RESET to
1447                          * ensure that no one else will attempt to use this
1448                          * call once we drop the conn->conn_call_lock and
1449                          * call->lock.  We must drop the conn->conn_call_lock
1450                          * before calling rxi_ResetCall because the process
1451                          * of clearing the transmit queue can block for an
1452                          * extended period of time.  If we block while holding
1453                          * the conn->conn_call_lock, then all rx_EndCall
1454                          * processing will block as well.  This has a detrimental
1455                          * effect on overall system performance.
1456                          */
1457                         call->state = RX_STATE_RESET;
1458                         MUTEX_EXIT(&conn->conn_call_lock);
1459                         MUTEX_ENTER(&rx_refcnt_mutex);
1460                         CALL_HOLD(call, RX_CALL_REFCOUNT_BEGIN);
1461                         MUTEX_EXIT(&rx_refcnt_mutex);
1462                         rxi_ResetCall(call, 0);
1463                         (*call->callNumber)++;
1464                         if (MUTEX_TRYENTER(&conn->conn_call_lock))
1465                             break;
1466
1467                         /*
1468                          * If we failed to be able to safely obtain the
1469                          * conn->conn_call_lock we will have to drop the
1470                          * call->lock to avoid a deadlock.  When the call->lock
1471                          * is released the state of the call can change.  If it
1472                          * is no longer RX_STATE_RESET then some other thread is
1473                          * using the call.
1474                          */
1475                         MUTEX_EXIT(&call->lock);
1476                         MUTEX_ENTER(&conn->conn_call_lock);
1477                         MUTEX_ENTER(&call->lock);
1478
1479                         if (call->state == RX_STATE_RESET)
1480                             break;
1481
1482                         /*
1483                          * If we get here it means that after dropping
1484                          * the conn->conn_call_lock and call->lock that
1485                          * the call is no longer ours.  If we can't find
1486                          * a free call in the remaining slots we should
1487                          * not go immediately to RX_CONN_MAKECALL_WAITING
1488                          * because by dropping the conn->conn_call_lock
1489                          * we have given up synchronization with rx_EndCall.
1490                          * Instead, cycle through one more time to see if
1491                          * we can find a call that can call our own.
1492                          */
1493                         MUTEX_ENTER(&rx_refcnt_mutex);
1494                         CALL_RELE(call, RX_CALL_REFCOUNT_BEGIN);
1495                         MUTEX_EXIT(&rx_refcnt_mutex);
1496                         wait = 0;
1497                     }
1498                     MUTEX_EXIT(&call->lock);
1499                 }
1500             } else {
1501                 if (ignoreBusy && conn->lastBusy[i]) {
1502                     /* if we're ignoring busy call slots, skip any ones that
1503                      * have lastBusy set */
1504                     if (leastBusy == 0 || conn->lastBusy[i] < leastBusy) {
1505                         leastBusy = conn->lastBusy[i];
1506                     }
1507                     continue;
1508                 }
1509
1510                 /* rxi_NewCall returns with mutex locked */
1511                 call = rxi_NewCall(conn, i);
1512                 MUTEX_ENTER(&rx_refcnt_mutex);
1513                 CALL_HOLD(call, RX_CALL_REFCOUNT_BEGIN);
1514                 MUTEX_EXIT(&rx_refcnt_mutex);
1515                 break;
1516             }
1517         }
1518         if (i < RX_MAXCALLS) {
1519             conn->lastBusy[i] = 0;
1520             break;
1521         }
1522         if (!wait)
1523             continue;
1524         if (leastBusy && ignoreBusy) {
1525             /* we didn't find a useable call slot, but we did see at least one
1526              * 'busy' slot; look again and only use a slot with the 'least
1527              * busy time */
1528             ignoreBusy = 0;
1529             continue;
1530         }
1531
1532         MUTEX_ENTER(&conn->conn_data_lock);
1533         conn->flags |= RX_CONN_MAKECALL_WAITING;
1534         conn->makeCallWaiters++;
1535         MUTEX_EXIT(&conn->conn_data_lock);
1536
1537 #ifdef  RX_ENABLE_LOCKS
1538         CV_WAIT(&conn->conn_call_cv, &conn->conn_call_lock);
1539 #else
1540         osi_rxSleep(conn);
1541 #endif
1542         MUTEX_ENTER(&conn->conn_data_lock);
1543         conn->makeCallWaiters--;
1544         if (conn->makeCallWaiters == 0)
1545             conn->flags &= ~RX_CONN_MAKECALL_WAITING;
1546         MUTEX_EXIT(&conn->conn_data_lock);
1547     }
1548     /* Client is initially in send mode */
1549     call->state = RX_STATE_ACTIVE;
1550     call->error = conn->error;
1551     if (call->error)
1552         call->mode = RX_MODE_ERROR;
1553     else
1554         call->mode = RX_MODE_SENDING;
1555
1556     /* remember start time for call in case we have hard dead time limit */
1557     call->queueTime = queueTime;
1558     clock_GetTime(&call->startTime);
1559     hzero(call->bytesSent);
1560     hzero(call->bytesRcvd);
1561
1562     /* Turn on busy protocol. */
1563     rxi_KeepAliveOn(call);
1564
1565     /* Attempt MTU discovery */
1566     rxi_GrowMTUOn(call);
1567
1568     /*
1569      * We are no longer the active thread in rx_NewCall
1570      */
1571     MUTEX_ENTER(&conn->conn_data_lock);
1572     conn->flags &= ~RX_CONN_MAKECALL_ACTIVE;
1573     MUTEX_EXIT(&conn->conn_data_lock);
1574
1575     /*
1576      * Wake up anyone else who might be giving us a chance to
1577      * run (see code above that avoids resource starvation).
1578      */
1579 #ifdef  RX_ENABLE_LOCKS
1580     CV_BROADCAST(&conn->conn_call_cv);
1581 #else
1582     osi_rxWakeup(conn);
1583 #endif
1584     MUTEX_EXIT(&conn->conn_call_lock);
1585
1586 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
1587     if (call->flags & (RX_CALL_TQ_BUSY | RX_CALL_TQ_CLEARME)) {
1588         osi_Panic("rx_NewCall call about to be used without an empty tq");
1589     }
1590 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
1591
1592     MUTEX_EXIT(&call->lock);
1593     USERPRI;
1594
1595     dpf(("rx_NewCall(call %"AFS_PTR_FMT")\n", call));
1596     return call;
1597 }
1598
1599 static int
1600 rxi_HasActiveCalls(struct rx_connection *aconn)
1601 {
1602     int i;
1603     struct rx_call *tcall;
1604     SPLVAR;
1605
1606     NETPRI;
1607     for (i = 0; i < RX_MAXCALLS; i++) {
1608         if ((tcall = aconn->call[i])) {
1609             if ((tcall->state == RX_STATE_ACTIVE)
1610                 || (tcall->state == RX_STATE_PRECALL)) {
1611                 USERPRI;
1612                 return 1;
1613             }
1614         }
1615     }
1616     USERPRI;
1617     return 0;
1618 }
1619
1620 int
1621 rxi_GetCallNumberVector(struct rx_connection *aconn,
1622                         afs_int32 * aint32s)
1623 {
1624     int i;
1625     struct rx_call *tcall;
1626     SPLVAR;
1627
1628     NETPRI;
1629     for (i = 0; i < RX_MAXCALLS; i++) {
1630         if ((tcall = aconn->call[i]) && (tcall->state == RX_STATE_DALLY))
1631             aint32s[i] = aconn->callNumber[i] + 1;
1632         else
1633             aint32s[i] = aconn->callNumber[i];
1634     }
1635     USERPRI;
1636     return 0;
1637 }
1638
1639 int
1640 rxi_SetCallNumberVector(struct rx_connection *aconn,
1641                         afs_int32 * aint32s)
1642 {
1643     int i;
1644     struct rx_call *tcall;
1645     SPLVAR;
1646
1647     NETPRI;
1648     for (i = 0; i < RX_MAXCALLS; i++) {
1649         if ((tcall = aconn->call[i]) && (tcall->state == RX_STATE_DALLY))
1650             aconn->callNumber[i] = aint32s[i] - 1;
1651         else
1652             aconn->callNumber[i] = aint32s[i];
1653     }
1654     USERPRI;
1655     return 0;
1656 }
1657
1658 /* Advertise a new service.  A service is named locally by a UDP port
1659  * number plus a 16-bit service id.  Returns (struct rx_service *) 0
1660  * on a failure.
1661  *
1662      char *serviceName;  Name for identification purposes (e.g. the
1663                          service name might be used for probing for
1664                          statistics) */
1665 struct rx_service *
1666 rx_NewServiceHost(afs_uint32 host, u_short port, u_short serviceId,
1667                   char *serviceName, struct rx_securityClass **securityObjects,
1668                   int nSecurityObjects,
1669                   afs_int32(*serviceProc) (struct rx_call * acall))
1670 {
1671     osi_socket socket = OSI_NULLSOCKET;
1672     struct rx_service *tservice;
1673     int i;
1674     SPLVAR;
1675
1676     clock_NewTime();
1677
1678     if (serviceId == 0) {
1679         (osi_Msg
1680          "rx_NewService:  service id for service %s is not non-zero.\n",
1681          serviceName);
1682         return 0;
1683     }
1684     if (port == 0) {
1685         if (rx_port == 0) {
1686             (osi_Msg
1687              "rx_NewService: A non-zero port must be specified on this call if a non-zero port was not provided at Rx initialization (service %s).\n",
1688              serviceName);
1689             return 0;
1690         }
1691         port = rx_port;
1692         socket = rx_socket;
1693     }
1694
1695     tservice = rxi_AllocService();
1696     NETPRI;
1697
1698 #ifdef  RX_ENABLE_LOCKS
1699     MUTEX_INIT(&tservice->svc_data_lock, "svc data lock", MUTEX_DEFAULT, 0);
1700 #endif
1701
1702     for (i = 0; i < RX_MAX_SERVICES; i++) {
1703         struct rx_service *service = rx_services[i];
1704         if (service) {
1705             if (port == service->servicePort && host == service->serviceHost) {
1706                 if (service->serviceId == serviceId) {
1707                     /* The identical service has already been
1708                      * installed; if the caller was intending to
1709                      * change the security classes used by this
1710                      * service, he/she loses. */
1711                     (osi_Msg
1712                      "rx_NewService: tried to install service %s with service id %d, which is already in use for service %s\n",
1713                      serviceName, serviceId, service->serviceName);
1714                     USERPRI;
1715                     rxi_FreeService(tservice);
1716                     return service;
1717                 }
1718                 /* Different service, same port: re-use the socket
1719                  * which is bound to the same port */
1720                 socket = service->socket;
1721             }
1722         } else {
1723             if (socket == OSI_NULLSOCKET) {
1724                 /* If we don't already have a socket (from another
1725                  * service on same port) get a new one */
1726                 socket = rxi_GetHostUDPSocket(host, port);
1727                 if (socket == OSI_NULLSOCKET) {
1728                     USERPRI;
1729                     rxi_FreeService(tservice);
1730                     return 0;
1731                 }
1732             }
1733             service = tservice;
1734             service->socket = socket;
1735             service->serviceHost = host;
1736             service->servicePort = port;
1737             service->serviceId = serviceId;
1738             service->serviceName = serviceName;
1739             service->nSecurityObjects = nSecurityObjects;
1740             service->securityObjects = securityObjects;
1741             service->minProcs = 0;
1742             service->maxProcs = 1;
1743             service->idleDeadTime = 60;
1744             service->idleDeadErr = 0;
1745             service->connDeadTime = rx_connDeadTime;
1746             service->executeRequestProc = serviceProc;
1747             service->checkReach = 0;
1748             service->nSpecific = 0;
1749             service->specific = NULL;
1750             rx_services[i] = service;   /* not visible until now */
1751             USERPRI;
1752             return service;
1753         }
1754     }
1755     USERPRI;
1756     rxi_FreeService(tservice);
1757     (osi_Msg "rx_NewService: cannot support > %d services\n",
1758      RX_MAX_SERVICES);
1759     return 0;
1760 }
1761
1762 /* Set configuration options for all of a service's security objects */
1763
1764 afs_int32
1765 rx_SetSecurityConfiguration(struct rx_service *service,
1766                             rx_securityConfigVariables type,
1767                             void *value)
1768 {
1769     int i;
1770     for (i = 0; i<service->nSecurityObjects; i++) {
1771         if (service->securityObjects[i]) {
1772             RXS_SetConfiguration(service->securityObjects[i], NULL, type,
1773                                  value, NULL);
1774         }
1775     }
1776     return 0;
1777 }
1778
1779 struct rx_service *
1780 rx_NewService(u_short port, u_short serviceId, char *serviceName,
1781               struct rx_securityClass **securityObjects, int nSecurityObjects,
1782               afs_int32(*serviceProc) (struct rx_call * acall))
1783 {
1784     return rx_NewServiceHost(htonl(INADDR_ANY), port, serviceId, serviceName, securityObjects, nSecurityObjects, serviceProc);
1785 }
1786
1787 /* Generic request processing loop. This routine should be called
1788  * by the implementation dependent rx_ServerProc. If socketp is
1789  * non-null, it will be set to the file descriptor that this thread
1790  * is now listening on. If socketp is null, this routine will never
1791  * returns. */
1792 void
1793 rxi_ServerProc(int threadID, struct rx_call *newcall, osi_socket * socketp)
1794 {
1795     struct rx_call *call;
1796     afs_int32 code;
1797     struct rx_service *tservice = NULL;
1798
1799     for (;;) {
1800         if (newcall) {
1801             call = newcall;
1802             newcall = NULL;
1803         } else {
1804             call = rx_GetCall(threadID, tservice, socketp);
1805             if (socketp && *socketp != OSI_NULLSOCKET) {
1806                 /* We are now a listener thread */
1807                 return;
1808             }
1809         }
1810
1811 #ifdef  KERNEL
1812         if (afs_termState == AFSOP_STOP_RXCALLBACK) {
1813 #ifdef RX_ENABLE_LOCKS
1814             AFS_GLOCK();
1815 #endif /* RX_ENABLE_LOCKS */
1816             afs_termState = AFSOP_STOP_AFS;
1817             afs_osi_Wakeup(&afs_termState);
1818 #ifdef RX_ENABLE_LOCKS
1819             AFS_GUNLOCK();
1820 #endif /* RX_ENABLE_LOCKS */
1821             return;
1822         }
1823 #endif
1824
1825         /* if server is restarting( typically smooth shutdown) then do not
1826          * allow any new calls.
1827          */
1828
1829         if (rx_tranquil && (call != NULL)) {
1830             SPLVAR;
1831
1832             NETPRI;
1833             MUTEX_ENTER(&call->lock);
1834
1835             rxi_CallError(call, RX_RESTARTING);
1836             rxi_SendCallAbort(call, (struct rx_packet *)0, 0, 0);
1837
1838             MUTEX_EXIT(&call->lock);
1839             USERPRI;
1840             continue;
1841         }
1842
1843         tservice = call->conn->service;
1844
1845         if (tservice->beforeProc)
1846             (*tservice->beforeProc) (call);
1847
1848         code = tservice->executeRequestProc(call);
1849
1850         if (tservice->afterProc)
1851             (*tservice->afterProc) (call, code);
1852
1853         rx_EndCall(call, code);
1854
1855         if (tservice->postProc)
1856             (*tservice->postProc) (code);
1857
1858         if (rx_stats_active) {
1859             MUTEX_ENTER(&rx_stats_mutex);
1860             rxi_nCalls++;
1861             MUTEX_EXIT(&rx_stats_mutex);
1862         }
1863     }
1864 }
1865
1866
1867 void
1868 rx_WakeupServerProcs(void)
1869 {
1870     struct rx_serverQueueEntry *np, *tqp;
1871     SPLVAR;
1872
1873     NETPRI;
1874     MUTEX_ENTER(&rx_serverPool_lock);
1875
1876 #ifdef RX_ENABLE_LOCKS
1877     if (rx_waitForPacket)
1878         CV_BROADCAST(&rx_waitForPacket->cv);
1879 #else /* RX_ENABLE_LOCKS */
1880     if (rx_waitForPacket)
1881         osi_rxWakeup(rx_waitForPacket);
1882 #endif /* RX_ENABLE_LOCKS */
1883     MUTEX_ENTER(&freeSQEList_lock);
1884     for (np = rx_FreeSQEList; np; np = tqp) {
1885         tqp = *(struct rx_serverQueueEntry **)np;
1886 #ifdef RX_ENABLE_LOCKS
1887         CV_BROADCAST(&np->cv);
1888 #else /* RX_ENABLE_LOCKS */
1889         osi_rxWakeup(np);
1890 #endif /* RX_ENABLE_LOCKS */
1891     }
1892     MUTEX_EXIT(&freeSQEList_lock);
1893     for (queue_Scan(&rx_idleServerQueue, np, tqp, rx_serverQueueEntry)) {
1894 #ifdef RX_ENABLE_LOCKS
1895         CV_BROADCAST(&np->cv);
1896 #else /* RX_ENABLE_LOCKS */
1897         osi_rxWakeup(np);
1898 #endif /* RX_ENABLE_LOCKS */
1899     }
1900     MUTEX_EXIT(&rx_serverPool_lock);
1901     USERPRI;
1902 }
1903
1904 /* meltdown:
1905  * One thing that seems to happen is that all the server threads get
1906  * tied up on some empty or slow call, and then a whole bunch of calls
1907  * arrive at once, using up the packet pool, so now there are more
1908  * empty calls.  The most critical resources here are server threads
1909  * and the free packet pool.  The "doreclaim" code seems to help in
1910  * general.  I think that eventually we arrive in this state: there
1911  * are lots of pending calls which do have all their packets present,
1912  * so they won't be reclaimed, are multi-packet calls, so they won't
1913  * be scheduled until later, and thus are tying up most of the free
1914  * packet pool for a very long time.
1915  * future options:
1916  * 1.  schedule multi-packet calls if all the packets are present.
1917  * Probably CPU-bound operation, useful to return packets to pool.
1918  * Do what if there is a full window, but the last packet isn't here?
1919  * 3.  preserve one thread which *only* runs "best" calls, otherwise
1920  * it sleeps and waits for that type of call.
1921  * 4.  Don't necessarily reserve a whole window for each thread.  In fact,
1922  * the current dataquota business is badly broken.  The quota isn't adjusted
1923  * to reflect how many packets are presently queued for a running call.
1924  * So, when we schedule a queued call with a full window of packets queued
1925  * up for it, that *should* free up a window full of packets for other 2d-class
1926  * calls to be able to use from the packet pool.  But it doesn't.
1927  *
1928  * NB.  Most of the time, this code doesn't run -- since idle server threads
1929  * sit on the idle server queue and are assigned by "...ReceivePacket" as soon
1930  * as a new call arrives.
1931  */
1932 /* Sleep until a call arrives.  Returns a pointer to the call, ready
1933  * for an rx_Read. */
1934 #ifdef RX_ENABLE_LOCKS
1935 struct rx_call *
1936 rx_GetCall(int tno, struct rx_service *cur_service, osi_socket * socketp)
1937 {
1938     struct rx_serverQueueEntry *sq;
1939     struct rx_call *call = (struct rx_call *)0;
1940     struct rx_service *service = NULL;
1941
1942     MUTEX_ENTER(&freeSQEList_lock);
1943
1944     if ((sq = rx_FreeSQEList)) {
1945         rx_FreeSQEList = *(struct rx_serverQueueEntry **)sq;
1946         MUTEX_EXIT(&freeSQEList_lock);
1947     } else {                    /* otherwise allocate a new one and return that */
1948         MUTEX_EXIT(&freeSQEList_lock);
1949         sq = rxi_Alloc(sizeof(struct rx_serverQueueEntry));
1950         MUTEX_INIT(&sq->lock, "server Queue lock", MUTEX_DEFAULT, 0);
1951         CV_INIT(&sq->cv, "server Queue lock", CV_DEFAULT, 0);
1952     }
1953
1954     MUTEX_ENTER(&rx_serverPool_lock);
1955     if (cur_service != NULL) {
1956         ReturnToServerPool(cur_service);
1957     }
1958     while (1) {
1959         if (queue_IsNotEmpty(&rx_incomingCallQueue)) {
1960             struct rx_call *tcall, *ncall, *choice2 = NULL;
1961
1962             /* Scan for eligible incoming calls.  A call is not eligible
1963              * if the maximum number of calls for its service type are
1964              * already executing */
1965             /* One thread will process calls FCFS (to prevent starvation),
1966              * while the other threads may run ahead looking for calls which
1967              * have all their input data available immediately.  This helps
1968              * keep threads from blocking, waiting for data from the client. */
1969             for (queue_Scan(&rx_incomingCallQueue, tcall, ncall, rx_call)) {
1970                 service = tcall->conn->service;
1971                 if (!QuotaOK(service)) {
1972                     continue;
1973                 }
1974                 MUTEX_ENTER(&rx_pthread_mutex);
1975                 if (tno == rxi_fcfs_thread_num
1976                     || !tcall->queue_item_header.next) {
1977                     MUTEX_EXIT(&rx_pthread_mutex);
1978                     /* If we're the fcfs thread , then  we'll just use
1979                      * this call. If we haven't been able to find an optimal
1980                      * choice, and we're at the end of the list, then use a
1981                      * 2d choice if one has been identified.  Otherwise... */
1982                     call = (choice2 ? choice2 : tcall);
1983                     service = call->conn->service;
1984                 } else {
1985                     MUTEX_EXIT(&rx_pthread_mutex);
1986                     if (!queue_IsEmpty(&tcall->rq)) {
1987                         struct rx_packet *rp;
1988                         rp = queue_First(&tcall->rq, rx_packet);
1989                         if (rp->header.seq == 1) {
1990                             if (!meltdown_1pkt
1991                                 || (rp->header.flags & RX_LAST_PACKET)) {
1992                                 call = tcall;
1993                             } else if (rxi_2dchoice && !choice2
1994                                        && !(tcall->flags & RX_CALL_CLEARED)
1995                                        && (tcall->rprev > rxi_HardAckRate)) {
1996                                 choice2 = tcall;
1997                             } else
1998                                 rxi_md2cnt++;
1999                         }
2000                     }
2001                 }
2002                 if (call) {
2003                     break;
2004                 } else {
2005                     ReturnToServerPool(service);
2006                 }
2007             }
2008         }
2009
2010         if (call) {
2011             queue_Remove(call);
2012             MUTEX_EXIT(&rx_serverPool_lock);
2013             MUTEX_ENTER(&call->lock);
2014
2015             if (call->flags & RX_CALL_WAIT_PROC) {
2016                 call->flags &= ~RX_CALL_WAIT_PROC;
2017                 rx_atomic_dec(&rx_nWaiting);
2018             }
2019
2020             if (call->state != RX_STATE_PRECALL || call->error) {
2021                 MUTEX_EXIT(&call->lock);
2022                 MUTEX_ENTER(&rx_serverPool_lock);
2023                 ReturnToServerPool(service);
2024                 call = NULL;
2025                 continue;
2026             }
2027
2028             if (queue_IsEmpty(&call->rq)
2029                 || queue_First(&call->rq, rx_packet)->header.seq != 1)
2030                 rxi_SendAck(call, 0, 0, RX_ACK_DELAY, 0);
2031
2032             CLEAR_CALL_QUEUE_LOCK(call);
2033             break;
2034         } else {
2035             /* If there are no eligible incoming calls, add this process
2036              * to the idle server queue, to wait for one */
2037             sq->newcall = 0;
2038             sq->tno = tno;
2039             if (socketp) {
2040                 *socketp = OSI_NULLSOCKET;
2041             }
2042             sq->socketp = socketp;
2043             queue_Append(&rx_idleServerQueue, sq);
2044 #ifndef AFS_AIX41_ENV
2045             rx_waitForPacket = sq;
2046 #else
2047             rx_waitingForPacket = sq;
2048 #endif /* AFS_AIX41_ENV */
2049             do {
2050                 CV_WAIT(&sq->cv, &rx_serverPool_lock);
2051 #ifdef  KERNEL
2052                 if (afs_termState == AFSOP_STOP_RXCALLBACK) {
2053                     MUTEX_EXIT(&rx_serverPool_lock);
2054                     return (struct rx_call *)0;
2055                 }
2056 #endif
2057             } while (!(call = sq->newcall)
2058                      && !(socketp && *socketp != OSI_NULLSOCKET));
2059             MUTEX_EXIT(&rx_serverPool_lock);
2060             if (call) {
2061                 MUTEX_ENTER(&call->lock);
2062             }
2063             break;
2064         }
2065     }
2066
2067     MUTEX_ENTER(&freeSQEList_lock);
2068     *(struct rx_serverQueueEntry **)sq = rx_FreeSQEList;
2069     rx_FreeSQEList = sq;
2070     MUTEX_EXIT(&freeSQEList_lock);
2071
2072     if (call) {
2073         clock_GetTime(&call->startTime);
2074         call->state = RX_STATE_ACTIVE;
2075         call->mode = RX_MODE_RECEIVING;
2076 #ifdef RX_KERNEL_TRACE
2077         if (ICL_SETACTIVE(afs_iclSetp)) {
2078             int glockOwner = ISAFS_GLOCK();
2079             if (!glockOwner)
2080                 AFS_GLOCK();
2081             afs_Trace3(afs_iclSetp, CM_TRACE_WASHERE, ICL_TYPE_STRING,
2082                        __FILE__, ICL_TYPE_INT32, __LINE__, ICL_TYPE_POINTER,
2083                        call);
2084             if (!glockOwner)
2085                 AFS_GUNLOCK();
2086         }
2087 #endif
2088
2089         rxi_calltrace(RX_CALL_START, call);
2090         dpf(("rx_GetCall(port=%d, service=%d) ==> call %"AFS_PTR_FMT"\n",
2091              call->conn->service->servicePort, call->conn->service->serviceId,
2092              call));
2093
2094         MUTEX_EXIT(&call->lock);
2095         MUTEX_ENTER(&rx_refcnt_mutex);
2096         CALL_HOLD(call, RX_CALL_REFCOUNT_BEGIN);
2097         MUTEX_EXIT(&rx_refcnt_mutex);
2098     } else {
2099         dpf(("rx_GetCall(socketp=%p, *socketp=0x%x)\n", socketp, *socketp));
2100     }
2101
2102     return call;
2103 }
2104 #else /* RX_ENABLE_LOCKS */
2105 struct rx_call *
2106 rx_GetCall(int tno, struct rx_service *cur_service, osi_socket * socketp)
2107 {
2108     struct rx_serverQueueEntry *sq;
2109     struct rx_call *call = (struct rx_call *)0, *choice2;
2110     struct rx_service *service = NULL;
2111     SPLVAR;
2112
2113     NETPRI;
2114     MUTEX_ENTER(&freeSQEList_lock);
2115
2116     if ((sq = rx_FreeSQEList)) {
2117         rx_FreeSQEList = *(struct rx_serverQueueEntry **)sq;
2118         MUTEX_EXIT(&freeSQEList_lock);
2119     } else {                    /* otherwise allocate a new one and return that */
2120         MUTEX_EXIT(&freeSQEList_lock);
2121         sq = rxi_Alloc(sizeof(struct rx_serverQueueEntry));
2122         MUTEX_INIT(&sq->lock, "server Queue lock", MUTEX_DEFAULT, 0);
2123         CV_INIT(&sq->cv, "server Queue lock", CV_DEFAULT, 0);
2124     }
2125     MUTEX_ENTER(&sq->lock);
2126
2127     if (cur_service != NULL) {
2128         cur_service->nRequestsRunning--;
2129         MUTEX_ENTER(&rx_quota_mutex);
2130         if (cur_service->nRequestsRunning < cur_service->minProcs)
2131             rxi_minDeficit++;
2132         rxi_availProcs++;
2133         MUTEX_EXIT(&rx_quota_mutex);
2134     }
2135     if (queue_IsNotEmpty(&rx_incomingCallQueue)) {
2136         struct rx_call *tcall, *ncall;
2137         /* Scan for eligible incoming calls.  A call is not eligible
2138          * if the maximum number of calls for its service type are
2139          * already executing */
2140         /* One thread will process calls FCFS (to prevent starvation),
2141          * while the other threads may run ahead looking for calls which
2142          * have all their input data available immediately.  This helps
2143          * keep threads from blocking, waiting for data from the client. */
2144         choice2 = (struct rx_call *)0;
2145         for (queue_Scan(&rx_incomingCallQueue, tcall, ncall, rx_call)) {
2146             service = tcall->conn->service;
2147             if (QuotaOK(service)) {
2148                 MUTEX_ENTER(&rx_pthread_mutex);
2149                 if (tno == rxi_fcfs_thread_num
2150                     || !tcall->queue_item_header.next) {
2151                     MUTEX_EXIT(&rx_pthread_mutex);
2152                     /* If we're the fcfs thread, then  we'll just use
2153                      * this call. If we haven't been able to find an optimal
2154                      * choice, and we're at the end of the list, then use a
2155                      * 2d choice if one has been identified.  Otherwise... */
2156                     call = (choice2 ? choice2 : tcall);
2157                     service = call->conn->service;
2158                 } else {
2159                     MUTEX_EXIT(&rx_pthread_mutex);
2160                     if (!queue_IsEmpty(&tcall->rq)) {
2161                         struct rx_packet *rp;
2162                         rp = queue_First(&tcall->rq, rx_packet);
2163                         if (rp->header.seq == 1
2164                             && (!meltdown_1pkt
2165                                 || (rp->header.flags & RX_LAST_PACKET))) {
2166                             call = tcall;
2167                         } else if (rxi_2dchoice && !choice2
2168                                    && !(tcall->flags & RX_CALL_CLEARED)
2169                                    && (tcall->rprev > rxi_HardAckRate)) {
2170                             choice2 = tcall;
2171                         } else
2172                             rxi_md2cnt++;
2173                     }
2174                 }
2175             }
2176             if (call)
2177                 break;
2178         }
2179     }
2180
2181     if (call) {
2182         queue_Remove(call);
2183         /* we can't schedule a call if there's no data!!! */
2184         /* send an ack if there's no data, if we're missing the
2185          * first packet, or we're missing something between first
2186          * and last -- there's a "hole" in the incoming data. */
2187         if (queue_IsEmpty(&call->rq)
2188             || queue_First(&call->rq, rx_packet)->header.seq != 1
2189             || call->rprev != queue_Last(&call->rq, rx_packet)->header.seq)
2190             rxi_SendAck(call, 0, 0, RX_ACK_DELAY, 0);
2191
2192         call->flags &= (~RX_CALL_WAIT_PROC);
2193         service->nRequestsRunning++;
2194         /* just started call in minProcs pool, need fewer to maintain
2195          * guarantee */
2196         MUTEX_ENTER(&rx_quota_mutex);
2197         if (service->nRequestsRunning <= service->minProcs)
2198             rxi_minDeficit--;
2199         rxi_availProcs--;
2200         MUTEX_EXIT(&rx_quota_mutex);
2201         rx_atomic_dec(&rx_nWaiting);
2202         /* MUTEX_EXIT(&call->lock); */
2203     } else {
2204         /* If there are no eligible incoming calls, add this process
2205          * to the idle server queue, to wait for one */
2206         sq->newcall = 0;
2207         if (socketp) {
2208             *socketp = OSI_NULLSOCKET;
2209         }
2210         sq->socketp = socketp;
2211         queue_Append(&rx_idleServerQueue, sq);
2212         do {
2213             osi_rxSleep(sq);
2214 #ifdef  KERNEL
2215             if (afs_termState == AFSOP_STOP_RXCALLBACK) {
2216                 USERPRI;
2217                 rxi_Free(sq, sizeof(struct rx_serverQueueEntry));
2218                 return (struct rx_call *)0;
2219             }
2220 #endif
2221         } while (!(call = sq->newcall)
2222                  && !(socketp && *socketp != OSI_NULLSOCKET));
2223     }
2224     MUTEX_EXIT(&sq->lock);
2225
2226     MUTEX_ENTER(&freeSQEList_lock);
2227     *(struct rx_serverQueueEntry **)sq = rx_FreeSQEList;
2228     rx_FreeSQEList = sq;
2229     MUTEX_EXIT(&freeSQEList_lock);
2230
2231     if (call) {
2232         clock_GetTime(&call->startTime);
2233         call->state = RX_STATE_ACTIVE;
2234         call->mode = RX_MODE_RECEIVING;
2235 #ifdef RX_KERNEL_TRACE
2236         if (ICL_SETACTIVE(afs_iclSetp)) {
2237             int glockOwner = ISAFS_GLOCK();
2238             if (!glockOwner)
2239                 AFS_GLOCK();
2240             afs_Trace3(afs_iclSetp, CM_TRACE_WASHERE, ICL_TYPE_STRING,
2241                        __FILE__, ICL_TYPE_INT32, __LINE__, ICL_TYPE_POINTER,
2242                        call);
2243             if (!glockOwner)
2244                 AFS_GUNLOCK();
2245         }
2246 #endif
2247
2248         rxi_calltrace(RX_CALL_START, call);
2249         dpf(("rx_GetCall(port=%d, service=%d) ==> call %p\n",
2250              call->conn->service->servicePort, call->conn->service->serviceId,
2251              call));
2252     } else {
2253         dpf(("rx_GetCall(socketp=%p, *socketp=0x%x)\n", socketp, *socketp));
2254     }
2255
2256     USERPRI;
2257
2258     return call;
2259 }
2260 #endif /* RX_ENABLE_LOCKS */
2261
2262
2263
2264 /* Establish a procedure to be called when a packet arrives for a
2265  * call.  This routine will be called at most once after each call,
2266  * and will also be called if there is an error condition on the or
2267  * the call is complete.  Used by multi rx to build a selection
2268  * function which determines which of several calls is likely to be a
2269  * good one to read from.
2270  * NOTE: the way this is currently implemented it is probably only a
2271  * good idea to (1) use it immediately after a newcall (clients only)
2272  * and (2) only use it once.  Other uses currently void your warranty
2273  */
2274 void
2275 rx_SetArrivalProc(struct rx_call *call,
2276                   void (*proc) (struct rx_call * call,
2277                                         void * mh,
2278                                         int index),
2279                   void * handle, int arg)
2280 {
2281     call->arrivalProc = proc;
2282     call->arrivalProcHandle = handle;
2283     call->arrivalProcArg = arg;
2284 }
2285
2286 /* Call is finished (possibly prematurely).  Return rc to the peer, if
2287  * appropriate, and return the final error code from the conversation
2288  * to the caller */
2289
2290 afs_int32
2291 rx_EndCall(struct rx_call *call, afs_int32 rc)
2292 {
2293     struct rx_connection *conn = call->conn;
2294     afs_int32 error;
2295     SPLVAR;
2296
2297     dpf(("rx_EndCall(call %"AFS_PTR_FMT" rc %d error %d abortCode %d)\n",
2298           call, rc, call->error, call->abortCode));
2299
2300     NETPRI;
2301     MUTEX_ENTER(&call->lock);
2302
2303     if (rc == 0 && call->error == 0) {
2304         call->abortCode = 0;
2305         call->abortCount = 0;
2306     }
2307
2308     call->arrivalProc = (void (*)())0;
2309     if (rc && call->error == 0) {
2310         rxi_CallError(call, rc);
2311         call->mode = RX_MODE_ERROR;
2312         /* Send an abort message to the peer if this error code has
2313          * only just been set.  If it was set previously, assume the
2314          * peer has already been sent the error code or will request it
2315          */
2316         rxi_SendCallAbort(call, (struct rx_packet *)0, 0, 0);
2317     }
2318     if (conn->type == RX_SERVER_CONNECTION) {
2319         /* Make sure reply or at least dummy reply is sent */
2320         if (call->mode == RX_MODE_RECEIVING) {
2321             MUTEX_EXIT(&call->lock);
2322             rxi_WriteProc(call, 0, 0);
2323             MUTEX_ENTER(&call->lock);
2324         }
2325         if (call->mode == RX_MODE_SENDING) {
2326             MUTEX_EXIT(&call->lock);
2327             rxi_FlushWrite(call);
2328             MUTEX_ENTER(&call->lock);
2329         }
2330         rxi_calltrace(RX_CALL_END, call);
2331         /* Call goes to hold state until reply packets are acknowledged */
2332         if (call->tfirst + call->nSoftAcked < call->tnext) {
2333             call->state = RX_STATE_HOLD;
2334         } else {
2335             call->state = RX_STATE_DALLY;
2336             rxi_ClearTransmitQueue(call, 0);
2337             rxi_rto_cancel(call);
2338             rxevent_Cancel(call->keepAliveEvent, call,
2339                            RX_CALL_REFCOUNT_ALIVE);
2340         }
2341     } else {                    /* Client connection */
2342         char dummy;
2343         /* Make sure server receives input packets, in the case where
2344          * no reply arguments are expected */
2345         if ((call->mode == RX_MODE_SENDING)
2346             || (call->mode == RX_MODE_RECEIVING && call->rnext == 1)) {
2347             MUTEX_EXIT(&call->lock);
2348             (void)rxi_ReadProc(call, &dummy, 1);
2349             MUTEX_ENTER(&call->lock);
2350         }
2351
2352         /* If we had an outstanding delayed ack, be nice to the server
2353          * and force-send it now.
2354          */
2355         if (call->delayedAckEvent) {
2356             rxevent_Cancel(call->delayedAckEvent, call,
2357                            RX_CALL_REFCOUNT_DELAY);
2358             call->delayedAckEvent = NULL;
2359             rxi_SendDelayedAck(NULL, call, NULL);
2360         }
2361
2362         /* We need to release the call lock since it's lower than the
2363          * conn_call_lock and we don't want to hold the conn_call_lock
2364          * over the rx_ReadProc call. The conn_call_lock needs to be held
2365          * here for the case where rx_NewCall is perusing the calls on
2366          * the connection structure. We don't want to signal until
2367          * rx_NewCall is in a stable state. Otherwise, rx_NewCall may
2368          * have checked this call, found it active and by the time it
2369          * goes to sleep, will have missed the signal.
2370          */
2371         MUTEX_EXIT(&call->lock);
2372         MUTEX_ENTER(&conn->conn_call_lock);
2373         MUTEX_ENTER(&call->lock);
2374
2375         if (!(call->flags & RX_CALL_PEER_BUSY)) {
2376             conn->lastBusy[call->channel] = 0;
2377         }
2378
2379         MUTEX_ENTER(&conn->conn_data_lock);
2380         conn->flags |= RX_CONN_BUSY;
2381         if (conn->flags & RX_CONN_MAKECALL_WAITING) {
2382             MUTEX_EXIT(&conn->conn_data_lock);
2383 #ifdef  RX_ENABLE_LOCKS
2384             CV_BROADCAST(&conn->conn_call_cv);
2385 #else
2386             osi_rxWakeup(conn);
2387 #endif
2388         }
2389 #ifdef RX_ENABLE_LOCKS
2390         else {
2391             MUTEX_EXIT(&conn->conn_data_lock);
2392         }
2393 #endif /* RX_ENABLE_LOCKS */
2394         call->state = RX_STATE_DALLY;
2395     }
2396     error = call->error;
2397
2398     /* currentPacket, nLeft, and NFree must be zeroed here, because
2399      * ResetCall cannot: ResetCall may be called at splnet(), in the
2400      * kernel version, and may interrupt the macros rx_Read or
2401      * rx_Write, which run at normal priority for efficiency. */
2402     if (call->currentPacket) {
2403 #ifdef RX_TRACK_PACKETS
2404         call->currentPacket->flags &= ~RX_PKTFLAG_CP;
2405 #endif
2406         rxi_FreePacket(call->currentPacket);
2407         call->currentPacket = (struct rx_packet *)0;
2408     }
2409
2410     call->nLeft = call->nFree = call->curlen = 0;
2411
2412     /* Free any packets from the last call to ReadvProc/WritevProc */
2413 #ifdef RXDEBUG_PACKET
2414     call->iovqc -=
2415 #endif /* RXDEBUG_PACKET */
2416         rxi_FreePackets(0, &call->iovq);
2417     MUTEX_EXIT(&call->lock);
2418
2419     MUTEX_ENTER(&rx_refcnt_mutex);
2420     CALL_RELE(call, RX_CALL_REFCOUNT_BEGIN);
2421     MUTEX_EXIT(&rx_refcnt_mutex);
2422     if (conn->type == RX_CLIENT_CONNECTION) {
2423         MUTEX_ENTER(&conn->conn_data_lock);
2424         conn->flags &= ~RX_CONN_BUSY;
2425         MUTEX_EXIT(&conn->conn_data_lock);
2426         MUTEX_EXIT(&conn->conn_call_lock);
2427     }
2428     USERPRI;
2429     /*
2430      * Map errors to the local host's errno.h format.
2431      */
2432     error = ntoh_syserr_conv(error);
2433     return error;
2434 }
2435
2436 #if !defined(KERNEL)
2437
2438 /* Call this routine when shutting down a server or client (especially
2439  * clients).  This will allow Rx to gracefully garbage collect server
2440  * connections, and reduce the number of retries that a server might
2441  * make to a dead client.
2442  * This is not quite right, since some calls may still be ongoing and
2443  * we can't lock them to destroy them. */
2444 void
2445 rx_Finalize(void)
2446 {
2447     struct rx_connection **conn_ptr, **conn_end;
2448
2449     INIT_PTHREAD_LOCKS;
2450     LOCK_RX_INIT;
2451     if (rxinit_status == 1) {
2452         UNLOCK_RX_INIT;
2453         return;                 /* Already shutdown. */
2454     }
2455     rxi_DeleteCachedConnections();
2456     if (rx_connHashTable) {
2457         MUTEX_ENTER(&rx_connHashTable_lock);
2458         for (conn_ptr = &rx_connHashTable[0], conn_end =
2459              &rx_connHashTable[rx_hashTableSize]; conn_ptr < conn_end;
2460              conn_ptr++) {
2461             struct rx_connection *conn, *next;
2462             for (conn = *conn_ptr; conn; conn = next) {
2463                 next = conn->next;
2464                 if (conn->type == RX_CLIENT_CONNECTION) {
2465                     MUTEX_ENTER(&rx_refcnt_mutex);
2466                     conn->refCount++;
2467                     MUTEX_EXIT(&rx_refcnt_mutex);
2468 #ifdef RX_ENABLE_LOCKS
2469                     rxi_DestroyConnectionNoLock(conn);
2470 #else /* RX_ENABLE_LOCKS */
2471                     rxi_DestroyConnection(conn);
2472 #endif /* RX_ENABLE_LOCKS */
2473                 }
2474             }
2475         }
2476 #ifdef RX_ENABLE_LOCKS
2477         while (rx_connCleanup_list) {
2478             struct rx_connection *conn;
2479             conn = rx_connCleanup_list;
2480             rx_connCleanup_list = rx_connCleanup_list->next;
2481             MUTEX_EXIT(&rx_connHashTable_lock);
2482             rxi_CleanupConnection(conn);
2483             MUTEX_ENTER(&rx_connHashTable_lock);
2484         }
2485         MUTEX_EXIT(&rx_connHashTable_lock);
2486 #endif /* RX_ENABLE_LOCKS */
2487     }
2488     rxi_flushtrace();
2489
2490 #ifdef AFS_NT40_ENV
2491     afs_winsockCleanup();
2492 #endif
2493
2494     rxinit_status = 1;
2495     UNLOCK_RX_INIT;
2496 }
2497 #endif
2498
2499 /* if we wakeup packet waiter too often, can get in loop with two
2500     AllocSendPackets each waking each other up (from ReclaimPacket calls) */
2501 void
2502 rxi_PacketsUnWait(void)
2503 {
2504     if (!rx_waitingForPackets) {
2505         return;
2506     }
2507 #ifdef KERNEL
2508     if (rxi_OverQuota(RX_PACKET_CLASS_SEND)) {
2509         return;                 /* still over quota */
2510     }
2511 #endif /* KERNEL */
2512     rx_waitingForPackets = 0;
2513 #ifdef  RX_ENABLE_LOCKS
2514     CV_BROADCAST(&rx_waitingForPackets_cv);
2515 #else
2516     osi_rxWakeup(&rx_waitingForPackets);
2517 #endif
2518     return;
2519 }
2520
2521
2522 /* ------------------Internal interfaces------------------------- */
2523
2524 /* Return this process's service structure for the
2525  * specified socket and service */
2526 static struct rx_service *
2527 rxi_FindService(osi_socket socket, u_short serviceId)
2528 {
2529     struct rx_service **sp;
2530     for (sp = &rx_services[0]; *sp; sp++) {
2531         if ((*sp)->serviceId == serviceId && (*sp)->socket == socket)
2532             return *sp;
2533     }
2534     return 0;
2535 }
2536
2537 #ifdef RXDEBUG_PACKET
2538 #ifdef KDUMP_RX_LOCK
2539 static struct rx_call_rx_lock *rx_allCallsp = 0;
2540 #else
2541 static struct rx_call *rx_allCallsp = 0;
2542 #endif
2543 #endif /* RXDEBUG_PACKET */
2544
2545 /* Allocate a call structure, for the indicated channel of the
2546  * supplied connection.  The mode and state of the call must be set by
2547  * the caller. Returns the call with mutex locked. */
2548 static struct rx_call *
2549 rxi_NewCall(struct rx_connection *conn, int channel)
2550 {
2551     struct rx_call *call;
2552 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
2553     struct rx_call *cp; /* Call pointer temp */
2554     struct rx_call *nxp;        /* Next call pointer, for queue_Scan */
2555 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
2556
2557     dpf(("rxi_NewCall(conn %"AFS_PTR_FMT", channel %d)\n", conn, channel));
2558
2559     /* Grab an existing call structure, or allocate a new one.
2560      * Existing call structures are assumed to have been left reset by
2561      * rxi_FreeCall */
2562     MUTEX_ENTER(&rx_freeCallQueue_lock);
2563
2564 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
2565     /*
2566      * EXCEPT that the TQ might not yet be cleared out.
2567      * Skip over those with in-use TQs.
2568      */
2569     call = NULL;
2570     for (queue_Scan(&rx_freeCallQueue, cp, nxp, rx_call)) {
2571         if (!(cp->flags & RX_CALL_TQ_BUSY)) {
2572             call = cp;
2573             break;
2574         }
2575     }
2576     if (call) {
2577 #else /* AFS_GLOBAL_RXLOCK_KERNEL */
2578     if (queue_IsNotEmpty(&rx_freeCallQueue)) {
2579         call = queue_First(&rx_freeCallQueue, rx_call);
2580 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
2581         queue_Remove(call);
2582         if (rx_stats_active)
2583             rx_atomic_dec(&rx_stats.nFreeCallStructs);
2584         MUTEX_EXIT(&rx_freeCallQueue_lock);
2585         MUTEX_ENTER(&call->lock);
2586         CLEAR_CALL_QUEUE_LOCK(call);
2587 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
2588         /* Now, if TQ wasn't cleared earlier, do it now. */
2589         rxi_WaitforTQBusy(call);
2590         if (call->flags & RX_CALL_TQ_CLEARME) {
2591             rxi_ClearTransmitQueue(call, 1);
2592             /*queue_Init(&call->tq);*/
2593         }
2594 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
2595         /* Bind the call to its connection structure */
2596         call->conn = conn;
2597         rxi_ResetCall(call, 1);
2598     } else {
2599
2600         call = rxi_Alloc(sizeof(struct rx_call));
2601 #ifdef RXDEBUG_PACKET
2602         call->allNextp = rx_allCallsp;
2603         rx_allCallsp = call;
2604         call->call_id =
2605             rx_atomic_inc_and_read(&rx_stats.nCallStructs);
2606 #else /* RXDEBUG_PACKET */
2607         rx_atomic_inc(&rx_stats.nCallStructs);
2608 #endif /* RXDEBUG_PACKET */
2609
2610         MUTEX_EXIT(&rx_freeCallQueue_lock);
2611         MUTEX_INIT(&call->lock, "call lock", MUTEX_DEFAULT, NULL);
2612         MUTEX_ENTER(&call->lock);
2613         CV_INIT(&call->cv_twind, "call twind", CV_DEFAULT, 0);
2614         CV_INIT(&call->cv_rq, "call rq", CV_DEFAULT, 0);
2615         CV_INIT(&call->cv_tq, "call tq", CV_DEFAULT, 0);
2616
2617         /* Initialize once-only items */
2618         queue_Init(&call->tq);
2619         queue_Init(&call->rq);
2620         queue_Init(&call->iovq);
2621 #ifdef RXDEBUG_PACKET
2622         call->rqc = call->tqc = call->iovqc = 0;
2623 #endif /* RXDEBUG_PACKET */
2624         /* Bind the call to its connection structure (prereq for reset) */
2625         call->conn = conn;
2626         rxi_ResetCall(call, 1);
2627     }
2628     call->channel = channel;
2629     call->callNumber = &conn->callNumber[channel];
2630     call->rwind = conn->rwind[channel];
2631     call->twind = conn->twind[channel];
2632     /* Note that the next expected call number is retained (in
2633      * conn->callNumber[i]), even if we reallocate the call structure
2634      */
2635     conn->call[channel] = call;
2636     /* if the channel's never been used (== 0), we should start at 1, otherwise
2637      * the call number is valid from the last time this channel was used */
2638     if (*call->callNumber == 0)
2639         *call->callNumber = 1;
2640
2641     return call;
2642 }
2643
2644 /* A call has been inactive long enough that so we can throw away
2645  * state, including the call structure, which is placed on the call
2646  * free list.
2647  *
2648  * call->lock amd rx_refcnt_mutex are held upon entry.
2649  * haveCTLock is set when called from rxi_ReapConnections.
2650  */
2651 static void
2652 rxi_FreeCall(struct rx_call *call, int haveCTLock)
2653 {
2654     int channel = call->channel;
2655     struct rx_connection *conn = call->conn;
2656
2657
2658     if (call->state == RX_STATE_DALLY || call->state == RX_STATE_HOLD)
2659         (*call->callNumber)++;
2660     /*
2661      * We are setting the state to RX_STATE_RESET to
2662      * ensure that no one else will attempt to use this
2663      * call once we drop the refcnt lock. We must drop
2664      * the refcnt lock before calling rxi_ResetCall
2665      * because it cannot be held across acquiring the
2666      * freepktQ lock. NewCall does the same.
2667      */
2668     call->state = RX_STATE_RESET;
2669     MUTEX_EXIT(&rx_refcnt_mutex);
2670     rxi_ResetCall(call, 0);
2671
2672     MUTEX_ENTER(&conn->conn_call_lock);
2673     if (call->conn->call[channel] == call)
2674         call->conn->call[channel] = 0;
2675     MUTEX_EXIT(&conn->conn_call_lock);
2676
2677     MUTEX_ENTER(&rx_freeCallQueue_lock);
2678     SET_CALL_QUEUE_LOCK(call, &rx_freeCallQueue_lock);
2679 #ifdef AFS_GLOBAL_RXLOCK_KERNEL
2680     /* A call may be free even though its transmit queue is still in use.
2681      * Since we search the call list from head to tail, put busy calls at
2682      * the head of the list, and idle calls at the tail.
2683      */
2684     if (call->flags & RX_CALL_TQ_BUSY)
2685         queue_Prepend(&rx_freeCallQueue, call);
2686     else
2687         queue_Append(&rx_freeCallQueue, call);
2688 #else /* AFS_GLOBAL_RXLOCK_KERNEL */
2689     queue_Append(&rx_freeCallQueue, call);
2690 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
2691     if (rx_stats_active)
2692         rx_atomic_inc(&rx_stats.nFreeCallStructs);
2693     MUTEX_EXIT(&rx_freeCallQueue_lock);
2694
2695     /* Destroy the connection if it was previously slated for
2696      * destruction, i.e. the Rx client code previously called
2697      * rx_DestroyConnection (client connections), or
2698      * rxi_ReapConnections called the same routine (server
2699      * connections).  Only do this, however, if there are no
2700      * outstanding calls. Note that for fine grain locking, there appears
2701      * to be a deadlock in that rxi_FreeCall has a call locked and
2702      * DestroyConnectionNoLock locks each call in the conn. But note a
2703      * few lines up where we have removed this call from the conn.
2704      * If someone else destroys a connection, they either have no
2705      * call lock held or are going through this section of code.
2706      */
2707     MUTEX_ENTER(&conn->conn_data_lock);
2708     if (conn->flags & RX_CONN_DESTROY_ME && !(conn->flags & RX_CONN_MAKECALL_WAITING)) {
2709         MUTEX_ENTER(&rx_refcnt_mutex);
2710         conn->refCount++;
2711         MUTEX_EXIT(&rx_refcnt_mutex);
2712         MUTEX_EXIT(&conn->conn_data_lock);
2713 #ifdef RX_ENABLE_LOCKS
2714         if (haveCTLock)
2715             rxi_DestroyConnectionNoLock(conn);
2716         else
2717             rxi_DestroyConnection(conn);
2718 #else /* RX_ENABLE_LOCKS */
2719         rxi_DestroyConnection(conn);
2720 #endif /* RX_ENABLE_LOCKS */
2721     } else {
2722         MUTEX_EXIT(&conn->conn_data_lock);
2723     }
2724     MUTEX_ENTER(&rx_refcnt_mutex);
2725 }
2726
2727 rx_atomic_t rxi_Allocsize = RX_ATOMIC_INIT(0);
2728 rx_atomic_t rxi_Alloccnt = RX_ATOMIC_INIT(0);
2729
2730 void *
2731 rxi_Alloc(size_t size)
2732 {
2733     char *p;
2734
2735     if (rx_stats_active) {
2736         rx_atomic_add(&rxi_Allocsize, (int) size);
2737         rx_atomic_inc(&rxi_Alloccnt);
2738     }
2739
2740 p = (char *)
2741 #if defined(KERNEL) && !defined(UKERNEL) && defined(AFS_FBSD80_ENV)
2742   afs_osi_Alloc_NoSleep(size);
2743 #else
2744   osi_Alloc(size);
2745 #endif
2746     if (!p)
2747         osi_Panic("rxi_Alloc error");
2748     memset(p, 0, size);
2749     return p;
2750 }
2751
2752 void
2753 rxi_Free(void *addr, size_t size)
2754 {
2755     if (rx_stats_active) {
2756         rx_atomic_sub(&rxi_Allocsize, (int) size);
2757         rx_atomic_dec(&rxi_Alloccnt);
2758     }
2759     osi_Free(addr, size);
2760 }
2761
2762 void
2763 rxi_SetPeerMtu(struct rx_peer *peer, afs_uint32 host, afs_uint32 port, int mtu)
2764 {
2765     struct rx_peer **peer_ptr = NULL, **peer_end = NULL;
2766     struct rx_peer *next = NULL;
2767     int hashIndex;
2768
2769     if (!peer) {
2770         MUTEX_ENTER(&rx_peerHashTable_lock);
2771         if (port == 0) {
2772             peer_ptr = &rx_peerHashTable[0];
2773             peer_end = &rx_peerHashTable[rx_hashTableSize];
2774             next = NULL;
2775         resume:
2776             for ( ; peer_ptr < peer_end; peer_ptr++) {
2777                 if (!peer)
2778                     peer = *peer_ptr;
2779                 for ( ; peer; peer = next) {
2780                     next = peer->next;
2781                     if (host == peer->host)
2782                         break;
2783                 }
2784             }
2785         } else {
2786             hashIndex = PEER_HASH(host, port);
2787             for (peer = rx_peerHashTable[hashIndex]; peer; peer = peer->next) {
2788                 if ((peer->host == host) && (peer->port == port))
2789                     break;
2790             }
2791         }
2792     } else {
2793         MUTEX_ENTER(&rx_peerHashTable_lock);
2794     }
2795
2796     if (peer) {
2797         peer->refCount++;
2798         MUTEX_EXIT(&rx_peerHashTable_lock);
2799
2800         MUTEX_ENTER(&peer->peer_lock);
2801         /* We don't handle dropping below min, so don't */
2802         mtu = MAX(mtu, RX_MIN_PACKET_SIZE);
2803         peer->ifMTU=MIN(mtu, peer->ifMTU);
2804         peer->natMTU = rxi_AdjustIfMTU(peer->ifMTU);
2805         /* if we tweaked this down, need to tune our peer MTU too */
2806         peer->MTU = MIN(peer->MTU, peer->natMTU);
2807         /* if we discovered a sub-1500 mtu, degrade */
2808         if (peer->ifMTU < OLD_MAX_PACKET_SIZE)
2809             peer->maxDgramPackets = 1;
2810         /* We no longer have valid peer packet information */
2811         if (peer->maxPacketSize-RX_IPUDP_SIZE > peer->ifMTU)
2812             peer->maxPacketSize = 0;
2813         MUTEX_EXIT(&peer->peer_lock);
2814
2815         MUTEX_ENTER(&rx_peerHashTable_lock);
2816         peer->refCount--;
2817         if (host && !port) {
2818             peer = next;
2819             /* pick up where we left off */
2820             goto resume;
2821         }
2822     }
2823     MUTEX_EXIT(&rx_peerHashTable_lock);
2824 }
2825
2826 /* Find the peer process represented by the supplied (host,port)
2827  * combination.  If there is no appropriate active peer structure, a
2828  * new one will be allocated and initialized
2829  * The origPeer, if set, is a pointer to a peer structure on which the
2830  * refcount will be be decremented. This is used to replace the peer
2831  * structure hanging off a connection structure */
2832 struct rx_peer *
2833 rxi_FindPeer(afs_uint32 host, u_short port,
2834              struct rx_peer *origPeer, int create)
2835 {
2836     struct rx_peer *pp;
2837     int hashIndex;
2838     hashIndex = PEER_HASH(host, port);
2839     MUTEX_ENTER(&rx_peerHashTable_lock);
2840     for (pp = rx_peerHashTable[hashIndex]; pp; pp = pp->next) {
2841         if ((pp->host == host) && (pp->port == port))
2842             break;
2843     }
2844     if (!pp) {
2845         if (create) {
2846             pp = rxi_AllocPeer();       /* This bzero's *pp */
2847             pp->host = host;    /* set here or in InitPeerParams is zero */
2848             pp->port = port;
2849             MUTEX_INIT(&pp->peer_lock, "peer_lock", MUTEX_DEFAULT, 0);
2850             queue_Init(&pp->congestionQueue);
2851             queue_Init(&pp->rpcStats);
2852             pp->next = rx_peerHashTable[hashIndex];
2853             rx_peerHashTable[hashIndex] = pp;
2854             rxi_InitPeerParams(pp);
2855             if (rx_stats_active)
2856                 rx_atomic_inc(&rx_stats.nPeerStructs);
2857         }
2858     }
2859     if (pp && create) {
2860         pp->refCount++;
2861     }
2862     if (origPeer)
2863         origPeer->refCount--;
2864     MUTEX_EXIT(&rx_peerHashTable_lock);
2865     return pp;
2866 }
2867
2868
2869 /* Find the connection at (host, port) started at epoch, and with the
2870  * given connection id.  Creates the server connection if necessary.
2871  * The type specifies whether a client connection or a server
2872  * connection is desired.  In both cases, (host, port) specify the
2873  * peer's (host, pair) pair.  Client connections are not made
2874  * automatically by this routine.  The parameter socket gives the
2875  * socket descriptor on which the packet was received.  This is used,
2876  * in the case of server connections, to check that *new* connections
2877  * come via a valid (port, serviceId).  Finally, the securityIndex
2878  * parameter must match the existing index for the connection.  If a
2879  * server connection is created, it will be created using the supplied
2880  * index, if the index is valid for this service */
2881 struct rx_connection *
2882 rxi_FindConnection(osi_socket socket, afs_uint32 host,
2883                    u_short port, u_short serviceId, afs_uint32 cid,
2884                    afs_uint32 epoch, int type, u_int securityIndex)
2885 {
2886     int hashindex, flag, i;
2887     struct rx_connection *conn;
2888     hashindex = CONN_HASH(host, port, cid, epoch, type);
2889     MUTEX_ENTER(&rx_connHashTable_lock);
2890     rxLastConn ? (conn = rxLastConn, flag = 0) : (conn =
2891                                                   rx_connHashTable[hashindex],
2892                                                   flag = 1);
2893     for (; conn;) {
2894         if ((conn->type == type) && ((cid & RX_CIDMASK) == conn->cid)
2895             && (epoch == conn->epoch)) {
2896             struct rx_peer *pp = conn->peer;
2897             if (securityIndex != conn->securityIndex) {
2898                 /* this isn't supposed to happen, but someone could forge a packet
2899                  * like this, and there seems to be some CM bug that makes this
2900                  * happen from time to time -- in which case, the fileserver
2901                  * asserts. */
2902                 MUTEX_EXIT(&rx_connHashTable_lock);
2903                 return (struct rx_connection *)0;
2904             }
2905             if (pp->host == host && pp->port == port)
2906                 break;
2907             if (type == RX_CLIENT_CONNECTION && pp->port == port)
2908                 break;
2909             /* So what happens when it's a callback connection? */
2910             if (                /*type == RX_CLIENT_CONNECTION && */
2911                    (conn->epoch & 0x80000000))
2912                 break;
2913         }
2914         if (!flag) {
2915             /* the connection rxLastConn that was used the last time is not the
2916              ** one we are looking for now. Hence, start searching in the hash */
2917             flag = 1;
2918             conn = rx_connHashTable[hashindex];
2919         } else
2920             conn = conn->next;
2921     }
2922     if (!conn) {
2923         struct rx_service *service;
2924         if (type == RX_CLIENT_CONNECTION) {
2925             MUTEX_EXIT(&rx_connHashTable_lock);
2926             return (struct rx_connection *)0;
2927         }
2928         service = rxi_FindService(socket, serviceId);
2929         if (!service || (securityIndex >= service->nSecurityObjects)
2930             || (service->securityObjects[securityIndex] == 0)) {
2931             MUTEX_EXIT(&rx_connHashTable_lock);
2932             return (struct rx_connection *)0;
2933         }
2934         conn = rxi_AllocConnection();   /* This bzero's the connection */
2935         MUTEX_INIT(&conn->conn_call_lock, "conn call lock", MUTEX_DEFAULT, 0);
2936         MUTEX_INIT(&conn->conn_data_lock, "conn data lock", MUTEX_DEFAULT, 0);
2937         CV_INIT(&conn->conn_call_cv, "conn call cv", CV_DEFAULT, 0);
2938         conn->next = rx_connHashTable[hashindex];
2939         rx_connHashTable[hashindex] = conn;
2940         conn->peer = rxi_FindPeer(host, port, 0, 1);
2941         conn->type = RX_SERVER_CONNECTION;
2942         conn->lastSendTime = clock_Sec();       /* don't GC immediately */
2943         conn->epoch = epoch;
2944         conn->cid = cid & RX_CIDMASK;
2945         /* conn->serial = conn->lastSerial = 0; */
2946         /* conn->timeout = 0; */
2947         conn->ackRate = RX_FAST_ACK_RATE;
2948         conn->service = service;
2949         conn->serviceId = serviceId;
2950         conn->securityIndex = securityIndex;
2951         conn->securityObject = service->securityObjects[securityIndex];
2952         conn->nSpecific = 0;
2953         conn->specific = NULL;
2954         rx_SetConnDeadTime(conn, service->connDeadTime);
2955         rx_SetConnIdleDeadTime(conn, service->idleDeadTime);
2956         rx_SetServerConnIdleDeadErr(conn, service->idleDeadErr);
2957         for (i = 0; i < RX_MAXCALLS; i++) {
2958             conn->twind[i] = rx_initSendWindow;
2959             conn->rwind[i] = rx_initReceiveWindow;
2960         }
2961         /* Notify security object of the new connection */
2962         RXS_NewConnection(conn->securityObject, conn);
2963         /* XXXX Connection timeout? */
2964         if (service->newConnProc)
2965             (*service->newConnProc) (conn);
2966         if (rx_stats_active)
2967             rx_atomic_inc(&rx_stats.nServerConns);
2968     }
2969
2970     MUTEX_ENTER(&rx_refcnt_mutex);
2971     conn->refCount++;
2972     MUTEX_EXIT(&rx_refcnt_mutex);
2973
2974     rxLastConn = conn;          /* store this connection as the last conn used */
2975     MUTEX_EXIT(&rx_connHashTable_lock);
2976     return conn;
2977 }
2978
2979 /**
2980  * Timeout a call on a busy call channel if appropriate.
2981  *
2982  * @param[in] call The busy call.
2983  *
2984  * @pre 'call' is marked as busy (namely,
2985  *      call->conn->lastBusy[call->channel] != 0)
2986  *
2987  * @pre call->lock is held
2988  * @pre rxi_busyChannelError is nonzero
2989  *
2990  * @note call->lock is dropped and reacquired
2991  */
2992 static void
2993 rxi_CheckBusy(struct rx_call *call)
2994 {
2995     struct rx_connection *conn = call->conn;
2996     int channel = call->channel;
2997     int freechannel = 0;
2998     int i;
2999     afs_uint32 callNumber = *call->callNumber;
3000
3001     MUTEX_EXIT(&call->lock);
3002
3003     MUTEX_ENTER(&conn->conn_call_lock);
3004
3005     /* Are there any other call slots on this conn that we should try? Look for
3006      * slots that are empty and are either non-busy, or were marked as busy
3007      * longer than conn->secondsUntilDead seconds before this call started. */
3008
3009     for (i = 0; i < RX_MAXCALLS && !freechannel; i++) {
3010         if (i == channel) {
3011             /* only look at channels that aren't us */
3012             continue;
3013         }
3014
3015         if (conn->lastBusy[i]) {
3016             /* if this channel looked busy too recently, don't look at it */
3017             if (conn->lastBusy[i] >= call->startTime.sec) {
3018                 continue;
3019             }
3020             if (call->startTime.sec - conn->lastBusy[i] < conn->secondsUntilDead) {
3021                 continue;
3022             }
3023         }
3024
3025         if (conn->call[i]) {
3026             struct rx_call *tcall = conn->call[i];
3027             MUTEX_ENTER(&tcall->lock);
3028             if (tcall->state == RX_STATE_DALLY) {
3029                 freechannel = 1;
3030             }
3031             MUTEX_EXIT(&tcall->lock);
3032         } else {
3033             freechannel = 1;
3034         }
3035     }
3036
3037     MUTEX_EXIT(&conn->conn_call_lock);
3038
3039     MUTEX_ENTER(&call->lock);
3040
3041     /* Since the call->lock and conn->conn_call_lock have been released it is
3042      * possible that (1) the call may no longer be busy and/or (2) the call may
3043      * have been reused by another waiting thread. Therefore, we must confirm
3044      * that the call state has not changed when deciding whether or not to
3045      * force this application thread to retry by forcing a Timeout error. */
3046
3047     if (freechannel && *call->callNumber == callNumber &&
3048         (call->flags & RX_CALL_PEER_BUSY)) {
3049         /* Since 'freechannel' is set, there exists another channel in this
3050          * rx_conn that the application thread might be able to use. We know
3051          * that we have the correct call since callNumber is unchanged, and we
3052          * know that the call is still busy. So, set the call error state to
3053          * rxi_busyChannelError so the application can retry the request,
3054          * presumably on a less-busy call channel. */
3055
3056         rxi_CallError(call, rxi_busyChannelError);
3057     }
3058 }
3059
3060 /* There are two packet tracing routines available for testing and monitoring
3061  * Rx.  One is called just after every packet is received and the other is
3062  * called just before every packet is sent.  Received packets, have had their
3063  * headers decoded, and packets to be sent have not yet had their headers
3064  * encoded.  Both take two parameters: a pointer to the packet and a sockaddr
3065  * containing the network address.  Both can be modified.  The return value, if
3066  * non-zero, indicates that the packet should be dropped.  */
3067
3068 int (*rx_justReceived) (struct rx_packet *, struct sockaddr_in *) = 0;
3069 int (*rx_almostSent) (struct rx_packet *, struct sockaddr_in *) = 0;
3070
3071 /* A packet has been received off the interface.  Np is the packet, socket is
3072  * the socket number it was received from (useful in determining which service
3073  * this packet corresponds to), and (host, port) reflect the host,port of the
3074  * sender.  This call returns the packet to the caller if it is finished with
3075  * it, rather than de-allocating it, just as a small performance hack */
3076
3077 struct rx_packet *
3078 rxi_ReceivePacket(struct rx_packet *np, osi_socket socket,
3079                   afs_uint32 host, u_short port, int *tnop,
3080                   struct rx_call **newcallp)
3081 {
3082     struct rx_call *call;
3083     struct rx_connection *conn;
3084     int channel;
3085     afs_uint32 currentCallNumber;
3086     int type;
3087     int skew;
3088 #ifdef RXDEBUG
3089     char *packetType;
3090 #endif
3091     struct rx_packet *tnp;
3092
3093 #ifdef RXDEBUG
3094 /* We don't print out the packet until now because (1) the time may not be
3095  * accurate enough until now in the lwp implementation (rx_Listener only gets
3096  * the time after the packet is read) and (2) from a protocol point of view,
3097  * this is the first time the packet has been seen */
3098     packetType = (np->header.type > 0 && np->header.type < RX_N_PACKET_TYPES)
3099         ? rx_packetTypes[np->header.type - 1] : "*UNKNOWN*";
3100     dpf(("R %d %s: %x.%d.%d.%d.%d.%d.%d flags %d, packet %"AFS_PTR_FMT"\n",
3101          np->header.serial, packetType, ntohl(host), ntohs(port), np->header.serviceId,
3102          np->header.epoch, np->header.cid, np->header.callNumber,
3103          np->header.seq, np->header.flags, np));
3104 #endif
3105
3106     if (np->header.type == RX_PACKET_TYPE_VERSION) {
3107         return rxi_ReceiveVersionPacket(np, socket, host, port, 1);
3108     }
3109
3110     if (np->header.type == RX_PACKET_TYPE_DEBUG) {
3111         return rxi_ReceiveDebugPacket(np, socket, host, port, 1);
3112     }
3113 #ifdef RXDEBUG
3114     /* If an input tracer function is defined, call it with the packet and
3115      * network address.  Note this function may modify its arguments. */
3116     if (rx_justReceived) {
3117         struct sockaddr_in addr;
3118         int drop;
3119         addr.sin_family = AF_INET;
3120         addr.sin_port = port;
3121         addr.sin_addr.s_addr = host;
3122 #ifdef STRUCT_SOCKADDR_HAS_SA_LEN
3123         addr.sin_len = sizeof(addr);
3124 #endif /* AFS_OSF_ENV */
3125         drop = (*rx_justReceived) (np, &addr);
3126         /* drop packet if return value is non-zero */
3127         if (drop)
3128             return np;
3129         port = addr.sin_port;   /* in case fcn changed addr */
3130         host = addr.sin_addr.s_addr;
3131     }
3132 #endif
3133
3134     /* If packet was not sent by the client, then *we* must be the client */
3135     type = ((np->header.flags & RX_CLIENT_INITIATED) != RX_CLIENT_INITIATED)
3136         ? RX_CLIENT_CONNECTION : RX_SERVER_CONNECTION;
3137
3138     /* Find the connection (or fabricate one, if we're the server & if
3139      * necessary) associated with this packet */
3140     conn =
3141         rxi_FindConnection(socket, host, port, np->header.serviceId,
3142                            np->header.cid, np->header.epoch, type,
3143                            np->header.securityIndex);
3144
3145     if (!conn) {
3146         /* If no connection found or fabricated, just ignore the packet.
3147          * (An argument could be made for sending an abort packet for
3148          * the conn) */
3149         return np;
3150     }
3151
3152     /* If the connection is in an error state, send an abort packet and ignore
3153      * the incoming packet */
3154     if (conn->error) {
3155         /* Don't respond to an abort packet--we don't want loops! */
3156         MUTEX_ENTER(&conn->conn_data_lock);
3157         if (np->header.type != RX_PACKET_TYPE_ABORT)
3158             np = rxi_SendConnectionAbort(conn, np, 1, 0);
3159         MUTEX_ENTER(&rx_refcnt_mutex);
3160         conn->refCount--;
3161         MUTEX_EXIT(&rx_refcnt_mutex);
3162         MUTEX_EXIT(&conn->conn_data_lock);
3163         return np;
3164     }
3165
3166     /* Check for connection-only requests (i.e. not call specific). */
3167     if (np->header.callNumber == 0) {
3168         switch (np->header.type) {
3169         case RX_PACKET_TYPE_ABORT: {
3170             /* What if the supplied error is zero? */
3171             afs_int32 errcode = ntohl(rx_GetInt32(np, 0));
3172             dpf(("rxi_ReceivePacket ABORT rx_GetInt32 = %d\n", errcode));
3173             rxi_ConnectionError(conn, errcode);
3174             MUTEX_ENTER(&rx_refcnt_mutex);
3175             conn->refCount--;
3176             MUTEX_EXIT(&rx_refcnt_mutex);
3177             return np;
3178         }
3179         case RX_PACKET_TYPE_CHALLENGE:
3180             tnp = rxi_ReceiveChallengePacket(conn, np, 1);
3181             MUTEX_ENTER(&rx_refcnt_mutex);
3182             conn->refCount--;
3183             MUTEX_EXIT(&rx_refcnt_mutex);
3184             return tnp;
3185         case RX_PACKET_TYPE_RESPONSE:
3186             tnp = rxi_ReceiveResponsePacket(conn, np, 1);
3187             MUTEX_ENTER(&rx_refcnt_mutex);
3188             conn->refCount--;
3189             MUTEX_EXIT(&rx_refcnt_mutex);
3190             return tnp;
3191         case RX_PACKET_TYPE_PARAMS:
3192         case RX_PACKET_TYPE_PARAMS + 1:
3193         case RX_PACKET_TYPE_PARAMS + 2:
3194             /* ignore these packet types for now */
3195             MUTEX_ENTER(&rx_refcnt_mutex);
3196             conn->refCount--;
3197             MUTEX_EXIT(&rx_refcnt_mutex);
3198             return np;
3199
3200
3201         default:
3202             /* Should not reach here, unless the peer is broken: send an
3203              * abort packet */
3204             rxi_ConnectionError(conn, RX_PROTOCOL_ERROR);
3205             MUTEX_ENTER(&conn->conn_data_lock);
3206             tnp = rxi_SendConnectionAbort(conn, np, 1, 0);
3207             MUTEX_ENTER(&rx_refcnt_mutex);
3208             conn->refCount--;
3209             MUTEX_EXIT(&rx_refcnt_mutex);
3210             MUTEX_EXIT(&conn->conn_data_lock);
3211             return tnp;
3212         }
3213     }
3214
3215     channel = np->header.cid & RX_CHANNELMASK;
3216     call = conn->call[channel];
3217
3218     if (call) {
3219         MUTEX_ENTER(&call->lock);
3220         currentCallNumber = conn->callNumber[channel];
3221     } else if (type == RX_SERVER_CONNECTION) {  /* No call allocated */
3222         MUTEX_ENTER(&conn->conn_call_lock);
3223         call = conn->call[channel];
3224         if (call) {
3225             MUTEX_ENTER(&call->lock);
3226             MUTEX_EXIT(&conn->conn_call_lock);
3227             currentCallNumber = conn->callNumber[channel];
3228         } else {
3229             call = rxi_NewCall(conn, channel);  /* returns locked call */
3230             MUTEX_EXIT(&conn->conn_call_lock);
3231             *call->callNumber = currentCallNumber = np->header.callNumber;
3232 #ifdef RXDEBUG
3233             if (np->header.callNumber == 0)
3234                 dpf(("RecPacket call 0 %d %s: %x.%u.%u.%u.%u.%u.%u flags %d, packet %"AFS_PTR_FMT" len %d\n",
3235                      np->header.serial, rx_packetTypes[np->header.type - 1], ntohl(conn->peer->host), ntohs(conn->peer->port),
3236                      np->header.serial, np->header.epoch, np->header.cid, np->header.callNumber, np->header.seq,
3237                      np->header.flags, np, np->length));
3238 #endif
3239             call->state = RX_STATE_PRECALL;
3240             clock_GetTime(&call->queueTime);
3241             hzero(call->bytesSent);
3242             hzero(call->bytesRcvd);
3243             /*
3244              * If the number of queued calls exceeds the overload
3245              * threshold then abort this call.
3246              */
3247             if ((rx_BusyThreshold > 0) &&
3248                 (rx_atomic_read(&rx_nWaiting) > rx_BusyThreshold)) {
3249                 struct rx_packet *tp;
3250
3251                 rxi_CallError(call, rx_BusyError);
3252                 tp = rxi_SendCallAbort(call, np, 1, 0);
3253                 MUTEX_EXIT(&call->lock);
3254                 MUTEX_ENTER(&rx_refcnt_mutex);
3255                 conn->refCount--;
3256                 MUTEX_EXIT(&rx_refcnt_mutex);
3257                 if (rx_stats_active)
3258                     rx_atomic_inc(&rx_stats.nBusies);
3259                 return tp;
3260             }
3261             rxi_KeepAliveOn(call);
3262         }
3263     } else {    /* RX_CLIENT_CONNECTION and No call allocated */
3264         /* This packet can't be for this call. If the new call address is
3265          * 0 then no call is running on this channel. If there is a call
3266          * then, since this is a client connection we're getting data for
3267          * it must be for the previous call.
3268          */
3269         if (rx_stats_active)
3270             rx_atomic_inc(&rx_stats.spuriousPacketsRead);
3271         MUTEX_ENTER(&rx_refcnt_mutex);
3272         conn->refCount--;
3273         MUTEX_EXIT(&rx_refcnt_mutex);
3274         return np;
3275     }
3276
3277     /* There is a non-NULL locked call at this point */
3278     if (type == RX_SERVER_CONNECTION) { /* We're the server */
3279         if (np->header.callNumber < currentCallNumber) {
3280             MUTEX_EXIT(&call->lock);
3281             if (rx_stats_active)
3282                 rx_atomic_inc(&rx_stats.spuriousPacketsRead);
3283             MUTEX_ENTER(&rx_refcnt_mutex);
3284             conn->refCount--;
3285             MUTEX_EXIT(&rx_refcnt_mutex);
3286             return np;
3287         } else if (np->header.callNumber != currentCallNumber) {
3288             /* Wait until the transmit queue is idle before deciding
3289              * whether to reset the current call. Chances are that the
3290              * call will be in ether DALLY or HOLD state once the TQ_BUSY
3291              * flag is cleared.
3292              */
3293 #ifdef AFS_GLOBAL_RXLOCK_KERNEL
3294             if (call->state == RX_STATE_ACTIVE) {
3295                 rxi_WaitforTQBusy(call);
3296                 /*
3297                  * If we entered error state while waiting,
3298                  * must call rxi_CallError to permit rxi_ResetCall
3299                  * to processed when the tqWaiter count hits zero.
3300                  */
3301                 if (call->error) {
3302                     rxi_CallError(call, call->error);
3303                     MUTEX_EXIT(&call->lock);
3304                     MUTEX_ENTER(&rx_refcnt_mutex);
3305                     conn->refCount--;
3306                     MUTEX_EXIT(&rx_refcnt_mutex);
3307                     return np;
3308                 }
3309             }
3310 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
3311             /* If the new call cannot be taken right now send a busy and set
3312              * the error condition in this call, so that it terminates as
3313              * quickly as possible */
3314             if (call->state == RX_STATE_ACTIVE) {
3315                 struct rx_packet *tp;
3316
3317                 rxi_CallError(call, RX_CALL_DEAD);
3318                 tp = rxi_SendSpecial(call, conn, np, RX_PACKET_TYPE_BUSY,
3319                                      NULL, 0, 1);
3320                 MUTEX_EXIT(&call->lock);
3321                 MUTEX_ENTER(&rx_refcnt_mutex);
3322                 conn->refCount--;
3323                 MUTEX_EXIT(&rx_refcnt_mutex);
3324                 return tp;
3325             }
3326             rxi_ResetCall(call, 0);
3327             *call->callNumber = np->header.callNumber;
3328 #ifdef RXDEBUG
3329             if (np->header.callNumber == 0)
3330                 dpf(("RecPacket call 0 %d %s: %x.%u.%u.%u.%u.%u.%u flags %d, packet %"AFS_PTR_FMT" len %d\n",
3331                       np->header.serial, rx_packetTypes[np->header.type - 1], ntohl(conn->peer->host), ntohs(conn->peer->port),
3332                       np->header.serial, np->header.epoch, np->header.cid, np->header.callNumber, np->header.seq,
3333                       np->header.flags, np, np->length));
3334 #endif
3335             call->state = RX_STATE_PRECALL;
3336             clock_GetTime(&call->queueTime);
3337             hzero(call->bytesSent);
3338             hzero(call->bytesRcvd);
3339             /*
3340              * If the number of queued calls exceeds the overload
3341              * threshold then abort this call.
3342              */
3343             if ((rx_BusyThreshold > 0) &&
3344                 (rx_atomic_read(&rx_nWaiting) > rx_BusyThreshold)) {
3345                 struct rx_packet *tp;
3346
3347                 rxi_CallError(call, rx_BusyError);
3348                 tp = rxi_SendCallAbort(call, np, 1, 0);
3349                 MUTEX_EXIT(&call->lock);
3350                 MUTEX_ENTER(&rx_refcnt_mutex);
3351                 conn->refCount--;
3352                 MUTEX_EXIT(&rx_refcnt_mutex);
3353                 if (rx_stats_active)
3354                     rx_atomic_inc(&rx_stats.nBusies);
3355                 return tp;
3356             }
3357             rxi_KeepAliveOn(call);
3358         } else {
3359             /* Continuing call; do nothing here. */
3360         }
3361     } else {                    /* we're the client */
3362         /* Ignore all incoming acknowledgements for calls in DALLY state */
3363         if ((call->state == RX_STATE_DALLY)
3364             && (np->header.type == RX_PACKET_TYPE_ACK)) {
3365             if (rx_stats_active)
3366                 rx_atomic_inc(&rx_stats.ignorePacketDally);
3367             MUTEX_EXIT(&call->lock);
3368             MUTEX_ENTER(&rx_refcnt_mutex);
3369             conn->refCount--;
3370             MUTEX_EXIT(&rx_refcnt_mutex);
3371             return np;
3372         }
3373
3374         /* Ignore anything that's not relevant to the current call.  If there
3375          * isn't a current call, then no packet is relevant. */
3376         if (np->header.callNumber != currentCallNumber) {
3377             if (rx_stats_active)
3378                 rx_atomic_inc(&rx_stats.spuriousPacketsRead);
3379             MUTEX_EXIT(&call->lock);
3380             MUTEX_ENTER(&rx_refcnt_mutex);
3381             conn->refCount--;
3382             MUTEX_EXIT(&rx_refcnt_mutex);
3383             return np;
3384         }
3385         /* If the service security object index stamped in the packet does not
3386          * match the connection's security index, ignore the packet */
3387         if (np->header.securityIndex != conn->securityIndex) {
3388             MUTEX_EXIT(&call->lock);
3389             MUTEX_ENTER(&rx_refcnt_mutex);
3390             conn->refCount--;
3391             MUTEX_EXIT(&rx_refcnt_mutex);
3392             return np;
3393         }
3394
3395         /* If we're receiving the response, then all transmit packets are
3396          * implicitly acknowledged.  Get rid of them. */
3397         if (np->header.type == RX_PACKET_TYPE_DATA) {
3398 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
3399             /* XXX Hack. Because we must release the global rx lock when
3400              * sending packets (osi_NetSend) we drop all acks while we're
3401              * traversing the tq in rxi_Start sending packets out because
3402              * packets may move to the freePacketQueue as result of being here!
3403              * So we drop these packets until we're safely out of the
3404              * traversing. Really ugly!
3405              * For fine grain RX locking, we set the acked field in the
3406              * packets and let rxi_Start remove them from the transmit queue.
3407              */
3408             if (call->flags & RX_CALL_TQ_BUSY) {
3409 #ifdef  RX_ENABLE_LOCKS
3410                 rxi_SetAcksInTransmitQueue(call);
3411 #else
3412                 MUTEX_ENTER(&rx_refcnt_mutex);
3413                 conn->refCount--;
3414                 MUTEX_EXIT(&rx_refcnt_mutex);
3415                 return np;      /* xmitting; drop packet */
3416 #endif
3417             } else {
3418                 rxi_ClearTransmitQueue(call, 0);
3419             }
3420 #else /* AFS_GLOBAL_RXLOCK_KERNEL */
3421             rxi_ClearTransmitQueue(call, 0);
3422 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
3423         } else {
3424             if (np->header.type == RX_PACKET_TYPE_ACK) {
3425                 /* now check to see if this is an ack packet acknowledging that the
3426                  * server actually *lost* some hard-acked data.  If this happens we
3427                  * ignore this packet, as it may indicate that the server restarted in
3428                  * the middle of a call.  It is also possible that this is an old ack
3429                  * packet.  We don't abort the connection in this case, because this
3430                  * *might* just be an old ack packet.  The right way to detect a server
3431                  * restart in the midst of a call is to notice that the server epoch
3432                  * changed, btw.  */
3433                 /* XXX I'm not sure this is exactly right, since tfirst **IS**
3434                  * XXX unacknowledged.  I think that this is off-by-one, but
3435                  * XXX I don't dare change it just yet, since it will
3436                  * XXX interact badly with the server-restart detection
3437                  * XXX code in receiveackpacket.  */
3438                 if (ntohl(rx_GetInt32(np, FIRSTACKOFFSET)) < call->tfirst) {
3439                     if (rx_stats_active)
3440                         rx_atomic_inc(&rx_stats.spuriousPacketsRead);
3441                     MUTEX_EXIT(&call->lock);
3442                     MUTEX_ENTER(&rx_refcnt_mutex);
3443                     conn->refCount--;
3444                     MUTEX_EXIT(&rx_refcnt_mutex);
3445                     return np;
3446                 }
3447             }
3448         }                       /* else not a data packet */
3449     }
3450
3451     osirx_AssertMine(&call->lock, "rxi_ReceivePacket middle");
3452     /* Set remote user defined status from packet */
3453     call->remoteStatus = np->header.userStatus;
3454
3455     /* Note the gap between the expected next packet and the actual
3456      * packet that arrived, when the new packet has a smaller serial number
3457      * than expected.  Rioses frequently reorder packets all by themselves,
3458      * so this will be quite important with very large window sizes.
3459      * Skew is checked against 0 here to avoid any dependence on the type of
3460      * inPacketSkew (which may be unsigned).  In C, -1 > (unsigned) 0 is always
3461      * true!
3462      * The inPacketSkew should be a smoothed running value, not just a maximum.  MTUXXX
3463      * see CalculateRoundTripTime for an example of how to keep smoothed values.
3464      * I think using a beta of 1/8 is probably appropriate.  93.04.21
3465      */
3466     MUTEX_ENTER(&conn->conn_data_lock);
3467     skew = conn->lastSerial - np->header.serial;
3468     conn->lastSerial = np->header.serial;
3469     MUTEX_EXIT(&conn->conn_data_lock);
3470     if (skew > 0) {
3471         struct rx_peer *peer;
3472         peer = conn->peer;
3473         if (skew > peer->inPacketSkew) {
3474             dpf(("*** In skew changed from %d to %d\n",
3475                   peer->inPacketSkew, skew));
3476             peer->inPacketSkew = skew;
3477         }
3478     }
3479
3480     /* Now do packet type-specific processing */
3481     switch (np->header.type) {
3482     case RX_PACKET_TYPE_DATA:
3483         np = rxi_ReceiveDataPacket(call, np, 1, socket, host, port, tnop,
3484                                    newcallp);
3485         break;
3486     case RX_PACKET_TYPE_ACK:
3487         /* Respond immediately to ack packets requesting acknowledgement
3488          * (ping packets) */
3489         if (np->header.flags & RX_REQUEST_ACK) {
3490             if (call->error)
3491                 (void)rxi_SendCallAbort(call, 0, 1, 0);
3492             else
3493                 (void)rxi_SendAck(call, 0, np->header.serial,
3494                                   RX_ACK_PING_RESPONSE, 1);
3495         }
3496         np = rxi_ReceiveAckPacket(call, np, 1);
3497         break;
3498     case RX_PACKET_TYPE_ABORT: {
3499         /* An abort packet: reset the call, passing the error up to the user. */
3500         /* What if error is zero? */
3501         /* What if the error is -1? the application will treat it as a timeout. */
3502         afs_int32 errdata = ntohl(*(afs_int32 *) rx_DataOf(np));
3503         dpf(("rxi_ReceivePacket ABORT rx_DataOf = %d\n", errdata));
3504         rxi_CallError(call, errdata);
3505         MUTEX_EXIT(&call->lock);
3506         MUTEX_ENTER(&rx_refcnt_mutex);
3507         conn->refCount--;
3508         MUTEX_EXIT(&rx_refcnt_mutex);
3509         return np;              /* xmitting; drop packet */
3510     }
3511     case RX_PACKET_TYPE_BUSY: {
3512         struct clock busyTime;
3513         clock_NewTime();
3514         clock_GetTime(&busyTime);
3515
3516         MUTEX_EXIT(&call->lock);
3517
3518         MUTEX_ENTER(&conn->conn_call_lock);
3519         MUTEX_ENTER(&call->lock);
3520         conn->lastBusy[call->channel] = busyTime.sec;
3521         call->flags |= RX_CALL_PEER_BUSY;
3522         MUTEX_EXIT(&call->lock);
3523         MUTEX_EXIT(&conn->conn_call_lock);
3524
3525         MUTEX_ENTER(&rx_refcnt_mutex);
3526         conn->refCount--;
3527         MUTEX_EXIT(&rx_refcnt_mutex);
3528         return np;
3529     }
3530
3531     case RX_PACKET_TYPE_ACKALL:
3532         /* All packets acknowledged, so we can drop all packets previously
3533          * readied for sending */
3534 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
3535         /* XXX Hack. We because we can't release the global rx lock when
3536          * sending packets (osi_NetSend) we drop all ack pkts while we're
3537          * traversing the tq in rxi_Start sending packets out because
3538          * packets may move to the freePacketQueue as result of being
3539          * here! So we drop these packets until we're safely out of the
3540          * traversing. Really ugly!
3541          * For fine grain RX locking, we set the acked field in the packets
3542          * and let rxi_Start remove the packets from the transmit queue.
3543          */
3544         if (call->flags & RX_CALL_TQ_BUSY) {
3545 #ifdef  RX_ENABLE_LOCKS
3546             rxi_SetAcksInTransmitQueue(call);
3547             break;
3548 #else /* RX_ENABLE_LOCKS */
3549             MUTEX_EXIT(&call->lock);
3550             MUTEX_ENTER(&rx_refcnt_mutex);
3551             conn->refCount--;
3552             MUTEX_EXIT(&rx_refcnt_mutex);
3553             return np;          /* xmitting; drop packet */
3554 #endif /* RX_ENABLE_LOCKS */
3555         }
3556 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
3557         rxi_ClearTransmitQueue(call, 0);
3558         break;
3559     default:
3560         /* Should not reach here, unless the peer is broken: send an abort
3561          * packet */
3562         rxi_CallError(call, RX_PROTOCOL_ERROR);
3563         np = rxi_SendCallAbort(call, np, 1, 0);
3564         break;
3565     };
3566     /* Note when this last legitimate packet was received, for keep-alive
3567      * processing.  Note, we delay getting the time until now in the hope that
3568      * the packet will be delivered to the user before any get time is required
3569      * (if not, then the time won't actually be re-evaluated here). */
3570     call->lastReceiveTime = clock_Sec();
3571     /* we've received a legit packet, so the channel is not busy */
3572     call->flags &= ~RX_CALL_PEER_BUSY;
3573     MUTEX_EXIT(&call->lock);
3574     MUTEX_ENTER(&rx_refcnt_mutex);
3575     conn->refCount--;
3576     MUTEX_EXIT(&rx_refcnt_mutex);
3577     return np;
3578 }
3579
3580 /* return true if this is an "interesting" connection from the point of view
3581     of someone trying to debug the system */
3582 int
3583 rxi_IsConnInteresting(struct rx_connection *aconn)
3584 {
3585     int i;
3586     struct rx_call *tcall;
3587
3588     if (aconn->flags & (RX_CONN_MAKECALL_WAITING | RX_CONN_DESTROY_ME))
3589         return 1;
3590
3591     for (i = 0; i < RX_MAXCALLS; i++) {
3592         tcall = aconn->call[i];
3593         if (tcall) {
3594             if ((tcall->state == RX_STATE_PRECALL)
3595                 || (tcall->state == RX_STATE_ACTIVE))
3596                 return 1;
3597             if ((tcall->mode == RX_MODE_SENDING)
3598                 || (tcall->mode == RX_MODE_RECEIVING))
3599                 return 1;
3600         }
3601     }
3602     return 0;
3603 }
3604
3605 #ifdef KERNEL
3606 /* if this is one of the last few packets AND it wouldn't be used by the
3607    receiving call to immediately satisfy a read request, then drop it on
3608    the floor, since accepting it might prevent a lock-holding thread from
3609    making progress in its reading. If a call has been cleared while in
3610    the precall state then ignore all subsequent packets until the call
3611    is assigned to a thread. */
3612
3613 static int
3614 TooLow(struct rx_packet *ap, struct rx_call *acall)
3615 {
3616     int rc = 0;
3617
3618     MUTEX_ENTER(&rx_quota_mutex);
3619     if (((ap->header.seq != 1) && (acall->flags & RX_CALL_CLEARED)
3620          && (acall->state == RX_STATE_PRECALL))
3621         || ((rx_nFreePackets < rxi_dataQuota + 2)
3622             && !((ap->header.seq < acall->rnext + rx_initSendWindow)
3623                  && (acall->flags & RX_CALL_READER_WAIT)))) {
3624         rc = 1;
3625     }
3626     MUTEX_EXIT(&rx_quota_mutex);
3627     return rc;
3628 }
3629 #endif /* KERNEL */
3630
3631 /*!
3632  * Clear the attach wait flag on a connection and proceed.
3633  *
3634  * Any processing waiting for a connection to be attached should be
3635  * unblocked. We clear the flag and do any other needed tasks.
3636  *
3637  * @param[in] conn
3638  *      the conn to unmark waiting for attach
3639  *
3640  * @pre conn's conn_data_lock must be locked before calling this function
3641  *
3642  */
3643 static void
3644 rxi_ConnClearAttachWait(struct rx_connection *conn)
3645 {
3646     /* Indicate that rxi_CheckReachEvent is no longer running by
3647      * clearing the flag.  Must be atomic under conn_data_lock to
3648      * avoid a new call slipping by: rxi_CheckConnReach holds
3649      * conn_data_lock while checking RX_CONN_ATTACHWAIT.
3650      */
3651     conn->flags &= ~RX_CONN_ATTACHWAIT;
3652     if (conn->flags & RX_CONN_NAT_PING) {
3653         conn->flags &= ~RX_CONN_NAT_PING;
3654         rxi_ScheduleNatKeepAliveEvent(conn);
3655     }
3656 }
3657
3658 static void
3659 rxi_CheckReachEvent(struct rxevent *event, void *arg1, void *arg2)
3660 {
3661     struct rx_connection *conn = arg1;
3662     struct rx_call *acall = arg2;
3663     struct rx_call *call = acall;
3664     struct clock when, now;
3665     int i, waiting;
3666
3667     MUTEX_ENTER(&conn->conn_data_lock);
3668     conn->checkReachEvent = NULL;
3669     waiting = conn->flags & RX_CONN_ATTACHWAIT;
3670     if (event) {
3671         MUTEX_ENTER(&rx_refcnt_mutex);
3672         conn->refCount--;
3673         MUTEX_EXIT(&rx_refcnt_mutex);
3674     }
3675     MUTEX_EXIT(&conn->conn_data_lock);
3676
3677     if (waiting) {
3678         if (!call) {
3679             MUTEX_ENTER(&conn->conn_call_lock);
3680             MUTEX_ENTER(&conn->conn_data_lock);
3681             for (i = 0; i < RX_MAXCALLS; i++) {
3682                 struct rx_call *tc = conn->call[i];
3683                 if (tc && tc->state == RX_STATE_PRECALL) {
3684                     call = tc;
3685                     break;
3686                 }
3687             }
3688             if (!call)
3689                 rxi_ConnClearAttachWait(conn);
3690             MUTEX_EXIT(&conn->conn_data_lock);
3691             MUTEX_EXIT(&conn->conn_call_lock);
3692         }
3693
3694         if (call) {
3695             if (call != acall)
3696                 MUTEX_ENTER(&call->lock);
3697             rxi_SendAck(call, NULL, 0, RX_ACK_PING, 0);
3698             if (call != acall)
3699                 MUTEX_EXIT(&call->lock);
3700
3701             clock_GetTime(&now);
3702             when = now;
3703             when.sec += RX_CHECKREACH_TIMEOUT;
3704             MUTEX_ENTER(&conn->conn_data_lock);
3705             if (!conn->checkReachEvent) {
3706                 MUTEX_ENTER(&rx_refcnt_mutex);
3707                 conn->refCount++;
3708                 MUTEX_EXIT(&rx_refcnt_mutex);
3709                 conn->checkReachEvent =
3710                     rxevent_PostNow(&when, &now, rxi_CheckReachEvent, conn,
3711                                     NULL);
3712             }
3713             MUTEX_EXIT(&conn->conn_data_lock);
3714         }
3715     }
3716 }
3717
3718 static int
3719 rxi_CheckConnReach(struct rx_connection *conn, struct rx_call *call)
3720 {
3721     struct rx_service *service = conn->service;
3722     struct rx_peer *peer = conn->peer;
3723     afs_uint32 now, lastReach;
3724
3725     if (service->checkReach == 0)
3726         return 0;
3727
3728     now = clock_Sec();
3729     MUTEX_ENTER(&peer->peer_lock);
3730     lastReach = peer->lastReachTime;
3731     MUTEX_EXIT(&peer->peer_lock);
3732     if (now - lastReach < RX_CHECKREACH_TTL)
3733         return 0;
3734
3735     MUTEX_ENTER(&conn->conn_data_lock);
3736     if (conn->flags & RX_CONN_ATTACHWAIT) {
3737         MUTEX_EXIT(&conn->conn_data_lock);
3738         return 1;
3739     }
3740     conn->flags |= RX_CONN_ATTACHWAIT;
3741     MUTEX_EXIT(&conn->conn_data_lock);
3742     if (!conn->checkReachEvent)
3743         rxi_CheckReachEvent(NULL, conn, call);
3744
3745     return 1;
3746 }
3747
3748 /* try to attach call, if authentication is complete */
3749 static void
3750 TryAttach(struct rx_call *acall, osi_socket socket,
3751           int *tnop, struct rx_call **newcallp,
3752           int reachOverride)
3753 {
3754     struct rx_connection *conn = acall->conn;
3755
3756     if (conn->type == RX_SERVER_CONNECTION
3757         && acall->state == RX_STATE_PRECALL) {
3758         /* Don't attach until we have any req'd. authentication. */
3759         if (RXS_CheckAuthentication(conn->securityObject, conn) == 0) {
3760             if (reachOverride || rxi_CheckConnReach(conn, acall) == 0)
3761                 rxi_AttachServerProc(acall, socket, tnop, newcallp);
3762             /* Note:  this does not necessarily succeed; there
3763              * may not any proc available
3764              */
3765         } else {
3766             rxi_ChallengeOn(acall->conn);
3767         }
3768     }
3769 }
3770
3771 /* A data packet has been received off the interface.  This packet is
3772  * appropriate to the call (the call is in the right state, etc.).  This
3773  * routine can return a packet to the caller, for re-use */
3774
3775 struct rx_packet *
3776 rxi_ReceiveDataPacket(struct rx_call *call,
3777                       struct rx_packet *np, int istack,
3778                       osi_socket socket, afs_uint32 host, u_short port,
3779                       int *tnop, struct rx_call **newcallp)
3780 {
3781     int ackNeeded = 0;          /* 0 means no, otherwise ack_reason */
3782     int newPackets = 0;
3783     int didHardAck = 0;
3784     int haveLast = 0;
3785     afs_uint32 seq;
3786     afs_uint32 serial=0, flags=0;
3787     int isFirst;
3788     struct rx_packet *tnp;
3789     struct clock when, now;
3790     if (rx_stats_active)
3791         rx_atomic_inc(&rx_stats.dataPacketsRead);
3792
3793 #ifdef KERNEL
3794     /* If there are no packet buffers, drop this new packet, unless we can find
3795      * packet buffers from inactive calls */
3796     if (!call->error
3797         && (rxi_OverQuota(RX_PACKET_CLASS_RECEIVE) || TooLow(np, call))) {
3798         MUTEX_ENTER(&rx_freePktQ_lock);
3799         rxi_NeedMorePackets = TRUE;
3800         MUTEX_EXIT(&rx_freePktQ_lock);
3801         if (rx_stats_active)
3802             rx_atomic_inc(&rx_stats.noPacketBuffersOnRead);
3803         call->rprev = np->header.serial;
3804         rxi_calltrace(RX_TRACE_DROP, call);
3805         dpf(("packet %"AFS_PTR_FMT" dropped on receipt - quota problems\n", np));
3806         /* We used to clear the receive queue here, in an attempt to free
3807          * packets. However this is unsafe if the queue has received a
3808          * soft ACK for the final packet */
3809         clock_GetTime(&now);
3810         when = now;
3811         clock_Add(&when, &rx_softAckDelay);
3812         if (!call->delayedAckEvent
3813             || clock_Gt(&call->delayedAckEvent->eventTime, &when)) {
3814             rxevent_Cancel(call->delayedAckEvent, call,
3815                            RX_CALL_REFCOUNT_DELAY);
3816             MUTEX_ENTER(&rx_refcnt_mutex);
3817             CALL_HOLD(call, RX_CALL_REFCOUNT_DELAY);
3818             MUTEX_EXIT(&rx_refcnt_mutex);
3819
3820             call->delayedAckEvent =
3821                 rxevent_PostNow(&when, &now, rxi_SendDelayedAck, call, 0);
3822         }
3823         /* we've damaged this call already, might as well do it in. */
3824         return np;
3825     }
3826 #endif /* KERNEL */
3827
3828     /*
3829      * New in AFS 3.5, if the RX_JUMBO_PACKET flag is set then this
3830      * packet is one of several packets transmitted as a single
3831      * datagram. Do not send any soft or hard acks until all packets
3832      * in a jumbogram have been processed. Send negative acks right away.
3833      */
3834     for (isFirst = 1, tnp = NULL; isFirst || tnp; isFirst = 0) {
3835         /* tnp is non-null when there are more packets in the
3836          * current jumbo gram */
3837         if (tnp) {
3838             if (np)
3839                 rxi_FreePacket(np);
3840             np = tnp;
3841         }
3842
3843         seq = np->header.seq;
3844         serial = np->header.serial;
3845         flags = np->header.flags;
3846
3847         /* If the call is in an error state, send an abort message */
3848         if (call->error)
3849             return rxi_SendCallAbort(call, np, istack, 0);
3850
3851         /* The RX_JUMBO_PACKET is set in all but the last packet in each
3852          * AFS 3.5 jumbogram. */
3853         if (flags & RX_JUMBO_PACKET) {
3854             tnp = rxi_SplitJumboPacket(np, host, port, isFirst);
3855         } else {
3856             tnp = NULL;
3857         }
3858
3859         if (np->header.spare != 0) {
3860             MUTEX_ENTER(&call->conn->conn_data_lock);
3861             call->conn->flags |= RX_CONN_USING_PACKET_CKSUM;
3862             MUTEX_EXIT(&call->conn->conn_data_lock);
3863         }
3864
3865         /* The usual case is that this is the expected next packet */
3866         if (seq == call->rnext) {
3867
3868             /* Check to make sure it is not a duplicate of one already queued */
3869             if (queue_IsNotEmpty(&call->rq)
3870                 && queue_First(&call->rq, rx_packet)->header.seq == seq) {
3871                 if (rx_stats_active)
3872                     rx_atomic_inc(&rx_stats.dupPacketsRead);
3873                 dpf(("packet %"AFS_PTR_FMT" dropped on receipt - duplicate\n", np));
3874                 rxevent_Cancel(call->delayedAckEvent, call,
3875                                RX_CALL_REFCOUNT_DELAY);
3876                 np = rxi_SendAck(call, np, serial, RX_ACK_DUPLICATE, istack);
3877                 ackNeeded = 0;
3878                 call->rprev = seq;
3879                 continue;
3880             }
3881
3882             /* It's the next packet. Stick it on the receive queue
3883              * for this call. Set newPackets to make sure we wake
3884              * the reader once all packets have been processed */
3885 #ifdef RX_TRACK_PACKETS
3886             np->flags |= RX_PKTFLAG_RQ;
3887 #endif
3888             queue_Prepend(&call->rq, np);
3889 #ifdef RXDEBUG_PACKET
3890             call->rqc++;
3891 #endif /* RXDEBUG_PACKET */
3892             call->nSoftAcks++;
3893             np = NULL;          /* We can't use this anymore */
3894             newPackets = 1;
3895
3896             /* If an ack is requested then set a flag to make sure we
3897              * send an acknowledgement for this packet */
3898             if (flags & RX_REQUEST_ACK) {
3899                 ackNeeded = RX_ACK_REQUESTED;
3900             }
3901
3902             /* Keep track of whether we have received the last packet */
3903             if (flags & RX_LAST_PACKET) {
3904                 call->flags |= RX_CALL_HAVE_LAST;
3905                 haveLast = 1;
3906             }
3907
3908             /* Check whether we have all of the packets for this call */
3909             if (call->flags & RX_CALL_HAVE_LAST) {
3910                 afs_uint32 tseq;        /* temporary sequence number */
3911                 struct rx_packet *tp;   /* Temporary packet pointer */
3912                 struct rx_packet *nxp;  /* Next pointer, for queue_Scan */
3913
3914                 for (tseq = seq, queue_Scan(&call->rq, tp, nxp, rx_packet)) {
3915                     if (tseq != tp->header.seq)
3916                         break;
3917                     if (tp->header.flags & RX_LAST_PACKET) {
3918                         call->flags |= RX_CALL_RECEIVE_DONE;
3919                         break;
3920                     }
3921                     tseq++;
3922                 }
3923             }
3924
3925             /* Provide asynchronous notification for those who want it
3926              * (e.g. multi rx) */
3927             if (call->arrivalProc) {
3928                 (*call->arrivalProc) (call, call->arrivalProcHandle,
3929                                       call->arrivalProcArg);
3930                 call->arrivalProc = (void (*)())0;
3931             }
3932
3933             /* Update last packet received */
3934             call->rprev = seq;
3935
3936             /* If there is no server process serving this call, grab
3937              * one, if available. We only need to do this once. If a
3938              * server thread is available, this thread becomes a server
3939              * thread and the server thread becomes a listener thread. */
3940             if (isFirst) {
3941                 TryAttach(call, socket, tnop, newcallp, 0);
3942             }
3943         }
3944         /* This is not the expected next packet. */
3945         else {
3946             /* Determine whether this is a new or old packet, and if it's
3947              * a new one, whether it fits into the current receive window.
3948              * Also figure out whether the packet was delivered in sequence.
3949              * We use the prev variable to determine whether the new packet
3950              * is the successor of its immediate predecessor in the
3951              * receive queue, and the missing flag to determine whether
3952              * any of this packets predecessors are missing.  */
3953
3954             afs_uint32 prev;    /* "Previous packet" sequence number */
3955             struct rx_packet *tp;       /* Temporary packet pointer */
3956             struct rx_packet *nxp;      /* Next pointer, for queue_Scan */
3957             int missing;        /* Are any predecessors missing? */
3958
3959             /* If the new packet's sequence number has been sent to the
3960              * application already, then this is a duplicate */
3961             if (seq < call->rnext) {
3962                 if (rx_stats_active)
3963                     rx_atomic_inc(&rx_stats.dupPacketsRead);
3964                 rxevent_Cancel(call->delayedAckEvent, call,
3965                                RX_CALL_REFCOUNT_DELAY);
3966                 np = rxi_SendAck(call, np, serial, RX_ACK_DUPLICATE, istack);
3967                 ackNeeded = 0;
3968                 call->rprev = seq;
3969                 continue;
3970             }
3971
3972             /* If the sequence number is greater than what can be
3973              * accomodated by the current window, then send a negative
3974              * acknowledge and drop the packet */
3975             if ((call->rnext + call->rwind) <= seq) {
3976                 rxevent_Cancel(call->delayedAckEvent, call,
3977                                RX_CALL_REFCOUNT_DELAY);
3978                 np = rxi_SendAck(call, np, serial, RX_ACK_EXCEEDS_WINDOW,
3979                                  istack);
3980                 ackNeeded = 0;
3981                 call->rprev = seq;
3982                 continue;
3983             }
3984
3985             /* Look for the packet in the queue of old received packets */
3986             for (prev = call->rnext - 1, missing =
3987                  0, queue_Scan(&call->rq, tp, nxp, rx_packet)) {
3988                 /*Check for duplicate packet */
3989                 if (seq == tp->header.seq) {
3990                     if (rx_stats_active)
3991                         rx_atomic_inc(&rx_stats.dupPacketsRead);
3992                     rxevent_Cancel(call->delayedAckEvent, call,
3993                                    RX_CALL_REFCOUNT_DELAY);
3994                     np = rxi_SendAck(call, np, serial, RX_ACK_DUPLICATE,
3995                                      istack);
3996                     ackNeeded = 0;
3997                     call->rprev = seq;
3998                     goto nextloop;
3999                 }
4000                 /* If we find a higher sequence packet, break out and
4001                  * insert the new packet here. */
4002                 if (seq < tp->header.seq)
4003                     break;
4004                 /* Check for missing packet */
4005                 if (tp->header.seq != prev + 1) {
4006                     missing = 1;
4007                 }
4008
4009                 prev = tp->header.seq;
4010             }
4011
4012             /* Keep track of whether we have received the last packet. */
4013             if (flags & RX_LAST_PACKET) {
4014                 call->flags |= RX_CALL_HAVE_LAST;
4015             }
4016
4017             /* It's within the window: add it to the the receive queue.
4018              * tp is left by the previous loop either pointing at the
4019              * packet before which to insert the new packet, or at the
4020              * queue head if the queue is empty or the packet should be
4021              * appended. */
4022 #ifdef RX_TRACK_PACKETS
4023             np->flags |= RX_PKTFLAG_RQ;
4024 #endif
4025 #ifdef RXDEBUG_PACKET
4026             call->rqc++;
4027 #endif /* RXDEBUG_PACKET */
4028             queue_InsertBefore(tp, np);
4029             call->nSoftAcks++;
4030             np = NULL;
4031
4032             /* Check whether we have all of the packets for this call */
4033             if ((call->flags & RX_CALL_HAVE_LAST)
4034                 && !(call->flags & RX_CALL_RECEIVE_DONE)) {
4035                 afs_uint32 tseq;        /* temporary sequence number */
4036
4037                 for (tseq =
4038                      call->rnext, queue_Scan(&call->rq, tp, nxp, rx_packet)) {
4039                     if (tseq != tp->header.seq)
4040                         break;
4041                     if (tp->header.flags & RX_LAST_PACKET) {
4042                         call->flags |= RX_CALL_RECEIVE_DONE;
4043                         break;
4044                     }
4045                     tseq++;
4046                 }
4047             }
4048
4049             /* We need to send an ack of the packet is out of sequence,
4050              * or if an ack was requested by the peer. */
4051             if (seq != prev + 1 || missing) {
4052                 ackNeeded = RX_ACK_OUT_OF_SEQUENCE;
4053             } else if (flags & RX_REQUEST_ACK) {
4054                 ackNeeded = RX_ACK_REQUESTED;
4055             }
4056
4057             /* Acknowledge the last packet for each call */
4058             if (flags & RX_LAST_PACKET) {
4059                 haveLast = 1;
4060             }
4061
4062             call->rprev = seq;
4063         }
4064       nextloop:;
4065     }
4066
4067     if (newPackets) {
4068         /*
4069          * If the receiver is waiting for an iovec, fill the iovec
4070          * using the data from the receive queue */
4071         if (call->flags & RX_CALL_IOVEC_WAIT) {
4072             didHardAck = rxi_FillReadVec(call, serial);
4073             /* the call may have been aborted */
4074             if (call->error) {
4075                 return NULL;
4076             }
4077             if (didHardAck) {
4078                 ackNeeded = 0;
4079             }
4080         }
4081
4082         /* Wakeup the reader if any */
4083         if ((call->flags & RX_CALL_READER_WAIT)
4084             && (!(call->flags & RX_CALL_IOVEC_WAIT) || !(call->iovNBytes)
4085                 || (call->iovNext >= call->iovMax)
4086                 || (call->flags & RX_CALL_RECEIVE_DONE))) {
4087             call->flags &= ~RX_CALL_READER_WAIT;
4088 #ifdef  RX_ENABLE_LOCKS
4089             CV_BROADCAST(&call->cv_rq);
4090 #else
4091             osi_rxWakeup(&call->rq);
4092 #endif
4093         }
4094     }
4095
4096     /*
4097      * Send an ack when requested by the peer, or once every
4098      * rxi_SoftAckRate packets until the last packet has been
4099      * received. Always send a soft ack for the last packet in
4100      * the server's reply. */
4101     if (ackNeeded) {
4102         rxevent_Cancel(call->delayedAckEvent, call, RX_CALL_REFCOUNT_DELAY);
4103         np = rxi_SendAck(call, np, serial, ackNeeded, istack);
4104     } else if (call->nSoftAcks > (u_short) rxi_SoftAckRate) {
4105         rxevent_Cancel(call->delayedAckEvent, call, RX_CALL_REFCOUNT_DELAY);
4106         np = rxi_SendAck(call, np, serial, RX_ACK_IDLE, istack);
4107     } else if (call->nSoftAcks) {
4108         clock_GetTime(&now);
4109         when = now;
4110         if (haveLast && !(flags & RX_CLIENT_INITIATED)) {
4111             clock_Add(&when, &rx_lastAckDelay);
4112         } else {
4113             clock_Add(&when, &rx_softAckDelay);
4114         }
4115         if (!call->delayedAckEvent
4116             || clock_Gt(&call->delayedAckEvent->eventTime, &when)) {
4117             rxevent_Cancel(call->delayedAckEvent, call,
4118                            RX_CALL_REFCOUNT_DELAY);
4119             MUTEX_ENTER(&rx_refcnt_mutex);
4120             CALL_HOLD(call, RX_CALL_REFCOUNT_DELAY);
4121             MUTEX_EXIT(&rx_refcnt_mutex);
4122             call->delayedAckEvent =
4123                 rxevent_PostNow(&when, &now, rxi_SendDelayedAck, call, 0);
4124         }
4125     } else if (call->flags & RX_CALL_RECEIVE_DONE) {
4126         rxevent_Cancel(call->delayedAckEvent, call, RX_CALL_REFCOUNT_DELAY);
4127     }
4128
4129     return np;
4130 }
4131
4132 #ifdef  ADAPT_WINDOW
4133 static void rxi_ComputeRate();
4134 #endif
4135
4136 static void
4137 rxi_UpdatePeerReach(struct rx_connection *conn, struct rx_call *acall)
4138 {
4139     struct rx_peer *peer = conn->peer;
4140
4141     MUTEX_ENTER(&peer->peer_lock);
4142     peer->lastReachTime = clock_Sec();
4143     MUTEX_EXIT(&peer->peer_lock);
4144
4145     MUTEX_ENTER(&conn->conn_data_lock);
4146     if (conn->flags & RX_CONN_ATTACHWAIT) {
4147         int i;
4148
4149         rxi_ConnClearAttachWait(conn);
4150         MUTEX_EXIT(&conn->conn_data_lock);
4151
4152         for (i = 0; i < RX_MAXCALLS; i++) {
4153             struct rx_call *call = conn->call[i];
4154             if (call) {
4155                 if (call != acall)
4156                     MUTEX_ENTER(&call->lock);
4157                 /* tnop can be null if newcallp is null */
4158                 TryAttach(call, (osi_socket) - 1, NULL, NULL, 1);
4159                 if (call != acall)
4160                     MUTEX_EXIT(&call->lock);
4161             }
4162         }
4163     } else
4164         MUTEX_EXIT(&conn->conn_data_lock);
4165 }
4166
4167 #if defined(RXDEBUG) && defined(AFS_NT40_ENV)
4168 static const char *
4169 rx_ack_reason(int reason)
4170 {
4171     switch (reason) {
4172     case RX_ACK_REQUESTED:
4173         return "requested";
4174     case RX_ACK_DUPLICATE:
4175         return "duplicate";
4176     case RX_ACK_OUT_OF_SEQUENCE:
4177         return "sequence";
4178     case RX_ACK_EXCEEDS_WINDOW:
4179         return "window";
4180     case RX_ACK_NOSPACE:
4181         return "nospace";
4182     case RX_ACK_PING:
4183         return "ping";
4184     case RX_ACK_PING_RESPONSE:
4185         return "response";
4186     case RX_ACK_DELAY:
4187         return "delay";
4188     case RX_ACK_IDLE:
4189         return "idle";
4190     default:
4191         return "unknown!!";
4192     }
4193 }
4194 #endif
4195
4196
4197 /* The real smarts of the whole thing.  */
4198 struct rx_packet *
4199 rxi_ReceiveAckPacket(struct rx_call *call, struct rx_packet *np,
4200                      int istack)
4201 {
4202     struct rx_ackPacket *ap;
4203     int nAcks;
4204     struct rx_packet *tp;
4205     struct rx_packet *nxp;      /* Next packet pointer for queue_Scan */
4206     struct rx_connection *conn = call->conn;
4207     struct rx_peer *peer = conn->peer;
4208     struct clock now;           /* Current time, for RTT calculations */
4209     afs_uint32 first;
4210     afs_uint32 prev;
4211     afs_uint32 serial;
4212     /* because there are CM's that are bogus, sending weird values for this. */
4213     afs_uint32 skew = 0;
4214     int nbytes;
4215     int missing;
4216     int acked;
4217     int nNacked = 0;
4218     int newAckCount = 0;
4219     int maxDgramPackets = 0;    /* Set if peer supports AFS 3.5 jumbo datagrams */
4220     int pktsize = 0;            /* Set if we need to update the peer mtu */
4221     int conn_data_locked = 0;
4222
4223     if (rx_stats_active)
4224         rx_atomic_inc(&rx_stats.ackPacketsRead);
4225     ap = (struct rx_ackPacket *)rx_DataOf(np);
4226     nbytes = rx_Contiguous(np) - (int)((ap->acks) - (u_char *) ap);
4227     if (nbytes < 0)
4228         return np;              /* truncated ack packet */
4229
4230     /* depends on ack packet struct */
4231     nAcks = MIN((unsigned)nbytes, (unsigned)ap->nAcks);
4232     first = ntohl(ap->firstPacket);
4233     prev = ntohl(ap->previousPacket);
4234     serial = ntohl(ap->serial);
4235     /* temporarily disabled -- needs to degrade over time
4236      * skew = ntohs(ap->maxSkew); */
4237
4238     /* Ignore ack packets received out of order */
4239     if (first < call->tfirst ||
4240         (first == call->tfirst && prev < call->tprev)) {
4241         return np;
4242     }
4243
4244     call->tprev = prev;
4245
4246     if (np->header.flags & RX_SLOW_START_OK) {
4247         call->flags |= RX_CALL_SLOW_START_OK;
4248     }
4249
4250     if (ap->reason == RX_ACK_PING_RESPONSE)
4251         rxi_UpdatePeerReach(conn, call);
4252
4253     if (conn->lastPacketSizeSeq) {
4254         MUTEX_ENTER(&conn->conn_data_lock);
4255         conn_data_locked = 1;
4256         if ((first > conn->lastPacketSizeSeq) && (conn->lastPacketSize)) {
4257             pktsize = conn->lastPacketSize;
4258             conn->lastPacketSize = conn->lastPacketSizeSeq = 0;
4259         }
4260     }
4261     if ((ap->reason == RX_ACK_PING_RESPONSE) && (conn->lastPingSizeSer)) {
4262         if (!conn_data_locked) {
4263             MUTEX_ENTER(&conn->conn_data_lock);
4264             conn_data_locked = 1;
4265         }
4266         if ((conn->lastPingSizeSer == serial) && (conn->lastPingSize)) {
4267             /* process mtu ping ack */
4268             pktsize = conn->lastPingSize;
4269             conn->lastPingSizeSer = conn->lastPingSize = 0;
4270         }
4271     }
4272
4273     if (conn_data_locked) {
4274         MUTEX_EXIT(&conn->conn_data_lock);
4275         conn_data_locked = 0;
4276     }
4277 #ifdef RXDEBUG
4278 #ifdef AFS_NT40_ENV
4279     if (rxdebug_active) {
4280         char msg[512];
4281         size_t len;
4282
4283         len = _snprintf(msg, sizeof(msg),
4284                         "tid[%d] RACK: reason %s serial %u previous %u seq %u skew %d first %u acks %u space %u ",
4285                          GetCurrentThreadId(), rx_ack_reason(ap->reason),
4286                          ntohl(ap->serial), ntohl(ap->previousPacket),
4287                          (unsigned int)np->header.seq, (unsigned int)skew,
4288                          ntohl(ap->firstPacket), ap->nAcks, ntohs(ap->bufferSpace) );
4289         if (nAcks) {
4290             int offset;
4291
4292             for (offset = 0; offset < nAcks && len < sizeof(msg); offset++)
4293                 msg[len++] = (ap->acks[offset] == RX_ACK_TYPE_NACK ? '-' : '*');
4294         }
4295         msg[len++]='\n';
4296         msg[len] = '\0';
4297         OutputDebugString(msg);
4298     }
4299 #else /* AFS_NT40_ENV */
4300     if (rx_Log) {
4301         fprintf(rx_Log,
4302                 "RACK: reason %x previous %u seq %u serial %u skew %d first %u",
4303                 ap->reason, ntohl(ap->previousPacket),
4304                 (unsigned int)np->header.seq, (unsigned int)serial,
4305                 (unsigned int)skew, ntohl(ap->firstPacket));
4306         if (nAcks) {
4307             int offset;
4308             for (offset = 0; offset < nAcks; offset++)
4309                 putc(ap->acks[offset] == RX_ACK_TYPE_NACK ? '-' : '*',
4310                      rx_Log);
4311         }
4312         putc('\n', rx_Log);
4313     }
4314 #endif /* AFS_NT40_ENV */
4315 #endif
4316
4317     MUTEX_ENTER(&peer->peer_lock);
4318     if (pktsize) {
4319         /*
4320          * Start somewhere. Can't assume we can send what we can receive,
4321          * but we are clearly receiving.
4322          */
4323         if (!peer->maxPacketSize)
4324             peer->maxPacketSize = RX_MIN_PACKET_SIZE+RX_IPUDP_SIZE;
4325
4326         if (pktsize > peer->maxPacketSize) {
4327             peer->maxPacketSize = pktsize;
4328             if ((pktsize-RX_IPUDP_SIZE > peer->ifMTU)) {
4329                 peer->ifMTU=pktsize-RX_IPUDP_SIZE;
4330                 peer->natMTU = rxi_AdjustIfMTU(peer->ifMTU);
4331                 rxi_ScheduleGrowMTUEvent(call, 1);
4332             }
4333         }
4334     }
4335
4336     /* Update the outgoing packet skew value to the latest value of
4337      * the peer's incoming packet skew value.  The ack packet, of
4338      * course, could arrive out of order, but that won't affect things
4339      * much */
4340     peer->outPacketSkew = skew;
4341
4342
4343     clock_GetTime(&now);
4344
4345     /* The transmit queue splits into 4 sections.
4346      *
4347      * The first section is packets which have now been acknowledged
4348      * by a window size change in the ack. These have reached the
4349      * application layer, and may be discarded. These are packets
4350      * with sequence numbers < ap->firstPacket.
4351      *
4352      * The second section is packets which have sequence numbers in
4353      * the range ap->firstPacket to ap->firstPacket + ap->nAcks. The
4354      * contents of the packet's ack array determines whether these
4355      * packets are acknowledged or not.
4356      *
4357      * The third section is packets which fall above the range
4358      * addressed in the ack packet. These have not yet been received
4359      * by the peer.
4360      *
4361      * The four section is packets which have not yet been transmitted.
4362      * These packets will have a header.serial of 0.
4363      */
4364
4365     /* First section - implicitly acknowledged packets that can be
4366      * disposed of
4367      */
4368
4369     tp = queue_First(&call->tq, rx_packet);
4370     while(!queue_IsEnd(&call->tq, tp) && tp->header.seq < first) {
4371         struct rx_packet *next;
4372
4373         next = queue_Next(tp, rx_packet);
4374         call->tfirst = tp->header.seq + 1;
4375
4376         if (!(tp->flags & RX_PKTFLAG_ACKED)) {
4377             newAckCount++;
4378             rxi_ComputeRoundTripTime(tp, ap, call, peer, &now);
4379         }
4380
4381 #ifdef ADAPT_WINDOW
4382         rxi_ComputeRate(call->conn->peer, call, p, np, ap->reason);
4383 #endif
4384
4385 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
4386         /* XXX Hack. Because we have to release the global rx lock when sending
4387          * packets (osi_NetSend) we drop all acks while we're traversing the tq
4388          * in rxi_Start sending packets out because packets may move to the
4389          * freePacketQueue as result of being here! So we drop these packets until
4390          * we're safely out of the traversing. Really ugly!
4391          * To make it even uglier, if we're using fine grain locking, we can
4392          * set the ack bits in the packets and have rxi_Start remove the packets
4393          * when it's done transmitting.
4394          */
4395         if (call->flags & RX_CALL_TQ_BUSY) {
4396 #ifdef RX_ENABLE_LOCKS
4397             tp->flags |= RX_PKTFLAG_ACKED;
4398             call->flags |= RX_CALL_TQ_SOME_ACKED;
4399 #else /* RX_ENABLE_LOCKS */
4400             break;
4401 #endif /* RX_ENABLE_LOCKS */
4402         } else
4403 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
4404         {
4405             queue_Remove(tp);
4406 #ifdef RX_TRACK_PACKETS
4407             tp->flags &= ~RX_PKTFLAG_TQ;
4408 #endif
4409 #ifdef RXDEBUG_PACKET
4410             call->tqc--;
4411 #endif /* RXDEBUG_PACKET */
4412             rxi_FreePacket(tp); /* rxi_FreePacket mustn't wake up anyone, preemptively. */
4413         }
4414         tp = next;
4415     }
4416
4417 #ifdef ADAPT_WINDOW
4418     /* Give rate detector a chance to respond to ping requests */
4419     if (ap->reason == RX_ACK_PING_RESPONSE) {
4420         rxi_ComputeRate(peer, call, 0, np, ap->reason);
4421     }
4422 #endif
4423
4424     /* N.B. we don't turn off any timers here.  They'll go away by themselves, anyway */
4425
4426     /* Second section of the queue - packets for which we are receiving
4427      * soft ACKs
4428      *
4429      * Go through the explicit acks/nacks and record the results in
4430      * the waiting packets.  These are packets that can't be released
4431      * yet, even with a positive acknowledge.  This positive
4432      * acknowledge only means the packet has been received by the
4433      * peer, not that it will be retained long enough to be sent to
4434      * the peer's upper level.  In addition, reset the transmit timers
4435      * of any missing packets (those packets that must be missing
4436      * because this packet was out of sequence) */
4437
4438     call->nSoftAcked = 0;
4439     missing = 0;
4440     while (!queue_IsEnd(&call->tq, tp) && tp->header.seq < first + nAcks) {
4441         /* Set the acknowledge flag per packet based on the
4442          * information in the ack packet. An acknowlegded packet can
4443          * be downgraded when the server has discarded a packet it
4444          * soacked previously, or when an ack packet is received
4445          * out of sequence. */
4446         if (ap->acks[tp->header.seq - first] == RX_ACK_TYPE_ACK) {
4447             if (!(tp->flags & RX_PKTFLAG_ACKED)) {
4448                 newAckCount++;
4449                 tp->flags |= RX_PKTFLAG_ACKED;
4450                 rxi_ComputeRoundTripTime(tp, ap, call, peer, &now);
4451 #ifdef ADAPT_WINDOW
4452                 rxi_ComputeRate(call->conn->peer, call, tp, np, ap->reason);
4453 #endif
4454             }
4455             if (missing) {
4456                 nNacked++;
4457             } else {
4458                 call->nSoftAcked++;
4459             }
4460         } else /* RX_ACK_TYPE_NACK */ {
4461             tp->flags &= ~RX_PKTFLAG_ACKED;
4462             missing = 1;
4463         }
4464
4465         tp = queue_Next(tp, rx_packet);
4466     }
4467
4468     /* We don't need to take any action with the 3rd or 4th section in the
4469      * queue - they're not addressed by the contents of this ACK packet.
4470      */
4471
4472     /* If the window has been extended by this acknowledge packet,
4473      * then wakeup a sender waiting in alloc for window space, or try
4474      * sending packets now, if he's been sitting on packets due to
4475      * lack of window space */
4476     if (call->tnext < (call->tfirst + call->twind)) {
4477 #ifdef  RX_ENABLE_LOCKS
4478         CV_SIGNAL(&call->cv_twind);
4479 #else
4480         if (call->flags & RX_CALL_WAIT_WINDOW_ALLOC) {
4481             call->flags &= ~RX_CALL_WAIT_WINDOW_ALLOC;
4482             osi_rxWakeup(&call->twind);
4483         }
4484 #endif
4485         if (call->flags & RX_CALL_WAIT_WINDOW_SEND) {
4486             call->flags &= ~RX_CALL_WAIT_WINDOW_SEND;
4487         }
4488     }
4489
4490     /* if the ack packet has a receivelen field hanging off it,
4491      * update our state */
4492     if (np->length >= rx_AckDataSize(ap->nAcks) + 2 * sizeof(afs_int32)) {
4493         afs_uint32 tSize;
4494
4495         /* If the ack packet has a "recommended" size that is less than
4496          * what I am using now, reduce my size to match */
4497         rx_packetread(np, rx_AckDataSize(ap->nAcks) + (int)sizeof(afs_int32),
4498                       (int)sizeof(afs_int32), &tSize);
4499         tSize = (afs_uint32) ntohl(tSize);
4500         peer->natMTU = rxi_AdjustIfMTU(MIN(tSize, peer->ifMTU));
4501
4502         /* Get the maximum packet size to send to this peer */
4503         rx_packetread(np, rx_AckDataSize(ap->nAcks), (int)sizeof(afs_int32),
4504                       &tSize);
4505         tSize = (afs_uint32) ntohl(tSize);
4506         tSize = (afs_uint32) MIN(tSize, rx_MyMaxSendSize);
4507         tSize = rxi_AdjustMaxMTU(peer->natMTU, tSize);
4508
4509         /* sanity check - peer might have restarted with different params.
4510          * If peer says "send less", dammit, send less...  Peer should never
4511          * be unable to accept packets of the size that prior AFS versions would
4512          * send without asking.  */
4513         if (peer->maxMTU != tSize) {
4514             if (peer->maxMTU > tSize) /* possible cong., maxMTU decreased */
4515                 peer->congestSeq++;
4516             peer->maxMTU = tSize;
4517             peer->MTU = MIN(tSize, peer->MTU);
4518             call->MTU = MIN(call->MTU, tSize);
4519         }
4520
4521         if (np->length == rx_AckDataSize(ap->nAcks) + 3 * sizeof(afs_int32)) {
4522             /* AFS 3.4a */
4523             rx_packetread(np,
4524                           rx_AckDataSize(ap->nAcks) + 2 * (int)sizeof(afs_int32),
4525                           (int)sizeof(afs_int32), &tSize);
4526             tSize = (afs_uint32) ntohl(tSize);  /* peer's receive window, if it's */
4527             if (tSize < call->twind) {  /* smaller than our send */
4528                 call->twind = tSize;    /* window, we must send less... */
4529                 call->ssthresh = MIN(call->twind, call->ssthresh);
4530                 call->conn->twind[call->channel] = call->twind;
4531             }
4532
4533             /* Only send jumbograms to 3.4a fileservers. 3.3a RX gets the
4534              * network MTU confused with the loopback MTU. Calculate the
4535              * maximum MTU here for use in the slow start code below.
4536              */
4537             /* Did peer restart with older RX version? */
4538             if (peer->maxDgramPackets > 1) {
4539                 peer->maxDgramPackets = 1;
4540             }
4541         } else if (np->length >=
4542                    rx_AckDataSize(ap->nAcks) + 4 * sizeof(afs_int32)) {
4543             /* AFS 3.5 */
4544             rx_packetread(np,
4545                           rx_AckDataSize(ap->nAcks) + 2 * (int)sizeof(afs_int32),
4546                           sizeof(afs_int32), &tSize);
4547             tSize = (afs_uint32) ntohl(tSize);
4548             /*
4549              * As of AFS 3.5 we set the send window to match the receive window.
4550              */
4551             if (tSize < call->twind) {
4552                 call->twind = tSize;
4553                 call->conn->twind[call->channel] = call->twind;
4554                 call->ssthresh = MIN(call->twind, call->ssthresh);
4555             } else if (tSize > call->twind) {
4556                 call->twind = tSize;
4557                 call->conn->twind[call->channel] = call->twind;
4558             }
4559
4560             /*
4561              * As of AFS 3.5, a jumbogram is more than one fixed size
4562              * packet transmitted in a single UDP datagram. If the remote
4563              * MTU is smaller than our local MTU then never send a datagram
4564              * larger than the natural MTU.
4565              */
4566             rx_packetread(np,
4567                           rx_AckDataSize(ap->nAcks) + 3 * (int)sizeof(afs_int32),
4568                           (int)sizeof(afs_int32), &tSize);
4569             maxDgramPackets = (afs_uint32) ntohl(tSize);
4570             maxDgramPackets = MIN(maxDgramPackets, rxi_nDgramPackets);
4571             maxDgramPackets =
4572                 MIN(maxDgramPackets, (int)(peer->ifDgramPackets));
4573             if (maxDgramPackets > 1) {
4574                 peer->maxDgramPackets = maxDgramPackets;
4575                 call->MTU = RX_JUMBOBUFFERSIZE + RX_HEADER_SIZE;
4576             } else {
4577                 peer->maxDgramPackets = 1;
4578                 call->MTU = peer->natMTU;
4579             }
4580         } else if (peer->maxDgramPackets > 1) {
4581             /* Restarted with lower version of RX */
4582             peer->maxDgramPackets = 1;
4583         }
4584     } else if (peer->maxDgramPackets > 1
4585                || peer->maxMTU != OLD_MAX_PACKET_SIZE) {
4586         /* Restarted with lower version of RX */
4587         peer->maxMTU = OLD_MAX_PACKET_SIZE;
4588         peer->natMTU = OLD_MAX_PACKET_SIZE;
4589         peer->MTU = OLD_MAX_PACKET_SIZE;
4590         peer->maxDgramPackets = 1;
4591         peer->nDgramPackets = 1;
4592         peer->congestSeq++;
4593         call->MTU = OLD_MAX_PACKET_SIZE;
4594     }
4595
4596     if (nNacked) {
4597         /*
4598          * Calculate how many datagrams were successfully received after
4599          * the first missing packet and adjust the negative ack counter
4600          * accordingly.
4601          */
4602         call->nAcks = 0;
4603         call->nNacks++;
4604         nNacked = (nNacked + call->nDgramPackets - 1) / call->nDgramPackets;
4605         if (call->nNacks < nNacked) {
4606             call->nNacks = nNacked;
4607         }
4608     } else {
4609         call->nAcks += newAckCount;
4610         call->nNacks = 0;
4611     }
4612
4613     /* If the packet contained new acknowledgements, rather than just
4614      * being a duplicate of one we have previously seen, then we can restart
4615      * the RTT timer
4616      */
4617     if (newAckCount > 0)
4618         rxi_rto_packet_acked(call, istack);
4619
4620     if (call->flags & RX_CALL_FAST_RECOVER) {
4621         if (newAckCount == 0) {
4622             call->cwind = MIN((int)(call->cwind + 1), rx_maxSendWindow);
4623         } else {
4624             call->flags &= ~RX_CALL_FAST_RECOVER;
4625             call->cwind = call->nextCwind;
4626             call->nextCwind = 0;
4627             call->nAcks = 0;
4628         }
4629         call->nCwindAcks = 0;
4630     } else if (nNacked && call->nNacks >= (u_short) rx_nackThreshold) {
4631         /* Three negative acks in a row trigger congestion recovery */
4632         call->flags |= RX_CALL_FAST_RECOVER;
4633         call->ssthresh = MAX(4, MIN((int)call->cwind, (int)call->twind)) >> 1;
4634         call->cwind =
4635             MIN((int)(call->ssthresh + rx_nackThreshold), rx_maxSendWindow);
4636         call->nDgramPackets = MAX(2, (int)call->nDgramPackets) >> 1;
4637         call->nextCwind = call->ssthresh;
4638         call->nAcks = 0;
4639         call->nNacks = 0;
4640         peer->MTU = call->MTU;
4641         peer->cwind = call->nextCwind;
4642         peer->nDgramPackets = call->nDgramPackets;
4643         peer->congestSeq++;
4644         call->congestSeq = peer->congestSeq;
4645
4646         /* Reset the resend times on the packets that were nacked
4647          * so we will retransmit as soon as the window permits
4648          */
4649
4650         for (acked = 0, queue_ScanBackwards(&call->tq, tp, nxp, rx_packet)) {
4651             if (acked) {
4652                 if (!(tp->flags & RX_PKTFLAG_ACKED)) {
4653                     tp->flags &= ~RX_PKTFLAG_SENT;
4654                 }
4655             } else if (tp->flags & RX_PKTFLAG_ACKED) {
4656                 acked = 1;
4657             }
4658         }
4659     } else {
4660         /* If cwind is smaller than ssthresh, then increase
4661          * the window one packet for each ack we receive (exponential
4662          * growth).
4663          * If cwind is greater than or equal to ssthresh then increase
4664          * the congestion window by one packet for each cwind acks we
4665          * receive (linear growth).  */
4666         if (call->cwind < call->ssthresh) {
4667             call->cwind =
4668                 MIN((int)call->ssthresh, (int)(call->cwind + newAckCount));
4669             call->nCwindAcks = 0;
4670         } else {
4671             call->nCwindAcks += newAckCount;
4672             if (call->nCwindAcks >= call->cwind) {
4673                 call->nCwindAcks = 0;
4674                 call->cwind = MIN((int)(call->cwind + 1), rx_maxSendWindow);
4675             }
4676         }
4677         /*
4678          * If we have received several acknowledgements in a row then
4679          * it is time to increase the size of our datagrams
4680          */
4681         if ((int)call->nAcks > rx_nDgramThreshold) {
4682             if (peer->maxDgramPackets > 1) {
4683                 if (call->nDgramPackets < peer->maxDgramPackets) {
4684                     call->nDgramPackets++;
4685                 }
4686                 call->MTU = RX_HEADER_SIZE + RX_JUMBOBUFFERSIZE;
4687             } else if (call->MTU < peer->maxMTU) {
4688                 /* don't upgrade if we can't handle it */
4689                 if ((call->nDgramPackets == 1) && (call->MTU >= peer->ifMTU))
4690                     call->MTU = peer->ifMTU;
4691                 else {
4692                     call->MTU += peer->natMTU;
4693                     call->MTU = MIN(call->MTU, peer->maxMTU);
4694                 }
4695             }
4696             call->nAcks = 0;
4697         }
4698     }
4699
4700     MUTEX_EXIT(&peer->peer_lock);       /* rxi_Start will lock peer. */
4701
4702     /* Servers need to hold the call until all response packets have
4703      * been acknowledged. Soft acks are good enough since clients
4704      * are not allowed to clear their receive queues. */
4705     if (call->state == RX_STATE_HOLD
4706         && call->tfirst + call->nSoftAcked >= call->tnext) {
4707         call->state = RX_STATE_DALLY;
4708         rxi_ClearTransmitQueue(call, 0);
4709         rxevent_Cancel(call->keepAliveEvent, call, RX_CALL_REFCOUNT_ALIVE);
4710     } else if (!queue_IsEmpty(&call->tq)) {
4711         rxi_Start(call, istack);
4712     }
4713     return np;
4714 }
4715
4716 /* Received a response to a challenge packet */
4717 struct rx_packet *
4718 rxi_ReceiveResponsePacket(struct rx_connection *conn,
4719                           struct rx_packet *np, int istack)
4720 {
4721     int error;
4722
4723     /* Ignore the packet if we're the client */
4724     if (conn->type == RX_CLIENT_CONNECTION)
4725         return np;
4726
4727     /* If already authenticated, ignore the packet (it's probably a retry) */
4728     if (RXS_CheckAuthentication(conn->securityObject, conn) == 0)
4729         return np;
4730
4731     /* Otherwise, have the security object evaluate the response packet */
4732     error = RXS_CheckResponse(conn->securityObject, conn, np);
4733     if (error) {
4734         /* If the response is invalid, reset the connection, sending
4735          * an abort to the peer */
4736 #ifndef KERNEL
4737         rxi_Delay(1);
4738 #endif
4739         rxi_ConnectionError(conn, error);
4740         MUTEX_ENTER(&conn->conn_data_lock);
4741         np = rxi_SendConnectionAbort(conn, np, istack, 0);
4742         MUTEX_EXIT(&conn->conn_data_lock);
4743         return np;
4744     } else {
4745         /* If the response is valid, any calls waiting to attach
4746          * servers can now do so */
4747         int i;
4748
4749         for (i = 0; i < RX_MAXCALLS; i++) {
4750             struct rx_call *call = conn->call[i];
4751             if (call) {
4752                 MUTEX_ENTER(&call->lock);
4753                 if (call->state == RX_STATE_PRECALL)
4754                     rxi_AttachServerProc(call, (osi_socket) - 1, NULL, NULL);
4755                 /* tnop can be null if newcallp is null */
4756                 MUTEX_EXIT(&call->lock);
4757             }
4758         }
4759
4760         /* Update the peer reachability information, just in case
4761          * some calls went into attach-wait while we were waiting
4762          * for authentication..
4763          */
4764         rxi_UpdatePeerReach(conn, NULL);
4765     }
4766     return np;
4767 }
4768
4769 /* A client has received an authentication challenge: the security
4770  * object is asked to cough up a respectable response packet to send
4771  * back to the server.  The server is responsible for retrying the
4772  * challenge if it fails to get a response. */
4773
4774 struct rx_packet *
4775 rxi_ReceiveChallengePacket(struct rx_connection *conn,
4776                            struct rx_packet *np, int istack)
4777 {
4778     int error;
4779
4780     /* Ignore the challenge if we're the server */
4781     if (conn->type == RX_SERVER_CONNECTION)
4782         return np;
4783
4784     /* Ignore the challenge if the connection is otherwise idle; someone's
4785      * trying to use us as an oracle. */
4786     if (!rxi_HasActiveCalls(conn))
4787         return np;
4788
4789     /* Send the security object the challenge packet.  It is expected to fill
4790      * in the response. */
4791     error = RXS_GetResponse(conn->securityObject, conn, np);
4792
4793     /* If the security object is unable to return a valid response, reset the
4794      * connection and send an abort to the peer.  Otherwise send the response
4795      * packet to the peer connection. */
4796     if (error) {
4797         rxi_ConnectionError(conn, error);
4798         MUTEX_ENTER(&conn->conn_data_lock);
4799         np = rxi_SendConnectionAbort(conn, np, istack, 0);
4800         MUTEX_EXIT(&conn->conn_data_lock);
4801     } else {
4802         np = rxi_SendSpecial((struct rx_call *)0, conn, np,
4803                              RX_PACKET_TYPE_RESPONSE, NULL, -1, istack);
4804     }
4805     return np;
4806 }
4807
4808
4809 /* Find an available server process to service the current request in
4810  * the given call structure.  If one isn't available, queue up this
4811  * call so it eventually gets one */
4812 void
4813 rxi_AttachServerProc(struct rx_call *call,
4814                      osi_socket socket, int *tnop,
4815                      struct rx_call **newcallp)
4816 {
4817     struct rx_serverQueueEntry *sq;
4818     struct rx_service *service = call->conn->service;
4819     int haveQuota = 0;
4820
4821     /* May already be attached */
4822     if (call->state == RX_STATE_ACTIVE)
4823         return;
4824
4825     MUTEX_ENTER(&rx_serverPool_lock);
4826
4827     haveQuota = QuotaOK(service);
4828     if ((!haveQuota) || queue_IsEmpty(&rx_idleServerQueue)) {
4829         /* If there are no processes available to service this call,
4830          * put the call on the incoming call queue (unless it's
4831          * already on the queue).
4832          */
4833 #ifdef RX_ENABLE_LOCKS
4834         if (haveQuota)
4835             ReturnToServerPool(service);
4836 #endif /* RX_ENABLE_LOCKS */
4837
4838         if (!(call->flags & RX_CALL_WAIT_PROC)) {
4839             call->flags |= RX_CALL_WAIT_PROC;
4840             rx_atomic_inc(&rx_nWaiting);
4841             rx_atomic_inc(&rx_nWaited);
4842             rxi_calltrace(RX_CALL_ARRIVAL, call);
4843             SET_CALL_QUEUE_LOCK(call, &rx_serverPool_lock);
4844             queue_Append(&rx_incomingCallQueue, call);
4845         }
4846     } else {
4847         sq = queue_Last(&rx_idleServerQueue, rx_serverQueueEntry);
4848
4849         /* If hot threads are enabled, and both newcallp and sq->socketp
4850          * are non-null, then this thread will process the call, and the
4851          * idle server thread will start listening on this threads socket.
4852          */
4853         queue_Remove(sq);
4854         if (rx_enable_hot_thread && newcallp && sq->socketp) {
4855             *newcallp = call;
4856             *tnop = sq->tno;
4857             *sq->socketp = socket;
4858             clock_GetTime(&call->startTime);
4859             MUTEX_ENTER(&rx_refcnt_mutex);
4860             CALL_HOLD(call, RX_CALL_REFCOUNT_BEGIN);
4861             MUTEX_EXIT(&rx_refcnt_mutex);
4862         } else {
4863             sq->newcall = call;
4864         }
4865         if (call->flags & RX_CALL_WAIT_PROC) {
4866             /* Conservative:  I don't think this should happen */
4867             call->flags &= ~RX_CALL_WAIT_PROC;
4868             if (queue_IsOnQueue(call)) {
4869                 queue_Remove(call);
4870
4871                 rx_atomic_dec(&rx_nWaiting);
4872             }
4873         }
4874         call->state = RX_STATE_ACTIVE;
4875         call->mode = RX_MODE_RECEIVING;
4876 #ifdef RX_KERNEL_TRACE
4877         {
4878             int glockOwner = ISAFS_GLOCK();
4879             if (!glockOwner)
4880                 AFS_GLOCK();
4881             afs_Trace3(afs_iclSetp, CM_TRACE_WASHERE, ICL_TYPE_STRING,
4882                        __FILE__, ICL_TYPE_INT32, __LINE__, ICL_TYPE_POINTER,
4883                        call);
4884             if (!glockOwner)
4885                 AFS_GUNLOCK();
4886         }
4887 #endif
4888         if (call->flags & RX_CALL_CLEARED) {
4889             /* send an ack now to start the packet flow up again */
4890             call->flags &= ~RX_CALL_CLEARED;
4891             rxi_SendAck(call, 0, 0, RX_ACK_DELAY, 0);
4892         }
4893 #ifdef  RX_ENABLE_LOCKS
4894         CV_SIGNAL(&sq->cv);
4895 #else
4896         service->nRequestsRunning++;
4897         MUTEX_ENTER(&rx_quota_mutex);
4898         if (service->nRequestsRunning <= service->minProcs)
4899             rxi_minDeficit--;
4900         rxi_availProcs--;
4901         MUTEX_EXIT(&rx_quota_mutex);
4902         osi_rxWakeup(sq);
4903 #endif
4904     }
4905     MUTEX_EXIT(&rx_serverPool_lock);
4906 }
4907
4908 /* Delay the sending of an acknowledge event for a short while, while
4909  * a new call is being prepared (in the case of a client) or a reply
4910  * is being prepared (in the case of a server).  Rather than sending
4911  * an ack packet, an ACKALL packet is sent. */
4912 void
4913 rxi_AckAll(struct rxevent *event, struct rx_call *call, char *dummy)
4914 {
4915 #ifdef RX_ENABLE_LOCKS
4916     if (event) {
4917         MUTEX_ENTER(&call->lock);
4918         call->delayedAckEvent = NULL;
4919         MUTEX_ENTER(&rx_refcnt_mutex);
4920         CALL_RELE(call, RX_CALL_REFCOUNT_ACKALL);
4921         MUTEX_EXIT(&rx_refcnt_mutex);
4922     }
4923     rxi_SendSpecial(call, call->conn, (struct rx_packet *)0,
4924                     RX_PACKET_TYPE_ACKALL, NULL, 0, 0);
4925     call->flags |= RX_CALL_ACKALL_SENT;
4926     if (event)
4927         MUTEX_EXIT(&call->lock);
4928 #else /* RX_ENABLE_LOCKS */
4929     if (event)
4930         call->delayedAckEvent = NULL;
4931     rxi_SendSpecial(call, call->conn, (struct rx_packet *)0,
4932                     RX_PACKET_TYPE_ACKALL, NULL, 0, 0);
4933     call->flags |= RX_CALL_ACKALL_SENT;
4934 #endif /* RX_ENABLE_LOCKS */
4935 }
4936
4937 void
4938 rxi_SendDelayedAck(struct rxevent *event, void *arg1, void *unused)
4939 {
4940     struct rx_call *call = arg1;
4941 #ifdef RX_ENABLE_LOCKS
4942     if (event) {
4943         MUTEX_ENTER(&call->lock);
4944         if (event == call->delayedAckEvent)
4945             call->delayedAckEvent = NULL;
4946         MUTEX_ENTER(&rx_refcnt_mutex);
4947         CALL_RELE(call, RX_CALL_REFCOUNT_DELAY);
4948         MUTEX_EXIT(&rx_refcnt_mutex);
4949     }
4950     (void)rxi_SendAck(call, 0, 0, RX_ACK_DELAY, 0);
4951     if (event)
4952         MUTEX_EXIT(&call->lock);
4953 #else /* RX_ENABLE_LOCKS */
4954     if (event)
4955         call->delayedAckEvent = NULL;
4956     (void)rxi_SendAck(call, 0, 0, RX_ACK_DELAY, 0);
4957 #endif /* RX_ENABLE_LOCKS */
4958 }
4959
4960
4961 #ifdef RX_ENABLE_LOCKS
4962 /* Set ack in all packets in transmit queue. rxi_Start will deal with
4963  * clearing them out.
4964  */
4965 static void
4966 rxi_SetAcksInTransmitQueue(struct rx_call *call)
4967 {
4968     struct rx_packet *p, *tp;
4969     int someAcked = 0;
4970
4971     for (queue_Scan(&call->tq, p, tp, rx_packet)) {
4972         p->flags |= RX_PKTFLAG_ACKED;
4973         someAcked = 1;
4974     }
4975     if (someAcked) {
4976         call->flags |= RX_CALL_TQ_CLEARME;
4977         call->flags |= RX_CALL_TQ_SOME_ACKED;
4978     }
4979
4980     rxi_rto_cancel(call);
4981
4982     call->tfirst = call->tnext;
4983     call->nSoftAcked = 0;
4984
4985     if (call->flags & RX_CALL_FAST_RECOVER) {
4986         call->flags &= ~RX_CALL_FAST_RECOVER;
4987         call->cwind = call->nextCwind;
4988         call->nextCwind = 0;
4989     }
4990
4991     CV_SIGNAL(&call->cv_twind);
4992 }
4993 #endif /* RX_ENABLE_LOCKS */
4994
4995 /* Clear out the transmit queue for the current call (all packets have
4996  * been received by peer) */
4997 void
4998 rxi_ClearTransmitQueue(struct rx_call *call, int force)
4999 {
5000 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
5001     struct rx_packet *p, *tp;
5002
5003     if (!force && (call->flags & RX_CALL_TQ_BUSY)) {
5004         int someAcked = 0;
5005         for (queue_Scan(&call->tq, p, tp, rx_packet)) {
5006             p->flags |= RX_PKTFLAG_ACKED;
5007             someAcked = 1;
5008         }
5009         if (someAcked) {
5010             call->flags |= RX_CALL_TQ_CLEARME;
5011             call->flags |= RX_CALL_TQ_SOME_ACKED;
5012         }
5013     } else {
5014 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
5015 #ifdef RXDEBUG_PACKET
5016         call->tqc -=
5017 #endif /* RXDEBUG_PACKET */
5018             rxi_FreePackets(0, &call->tq);
5019         rxi_WakeUpTransmitQueue(call);
5020 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
5021         call->flags &= ~RX_CALL_TQ_CLEARME;
5022     }
5023 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
5024
5025     rxi_rto_cancel(call);
5026     call->tfirst = call->tnext; /* implicitly acknowledge all data already sent */
5027     call->nSoftAcked = 0;
5028
5029     if (call->flags & RX_CALL_FAST_RECOVER) {
5030         call->flags &= ~RX_CALL_FAST_RECOVER;
5031         call->cwind = call->nextCwind;
5032     }
5033 #ifdef  RX_ENABLE_LOCKS
5034     CV_SIGNAL(&call->cv_twind);
5035 #else
5036     osi_rxWakeup(&call->twind);
5037 #endif
5038 }
5039
5040 void
5041 rxi_ClearReceiveQueue(struct rx_call *call)
5042 {
5043     if (queue_IsNotEmpty(&call->rq)) {
5044         u_short count;
5045
5046         count = rxi_FreePackets(0, &call->rq);
5047         rx_packetReclaims += count;
5048 #ifdef RXDEBUG_PACKET
5049         call->rqc -= count;
5050         if ( call->rqc != 0 )
5051             dpf(("rxi_ClearReceiveQueue call %"AFS_PTR_FMT" rqc %u != 0\n", call, call->rqc));
5052 #endif
5053         call->flags &= ~(RX_CALL_RECEIVE_DONE | RX_CALL_HAVE_LAST);
5054     }
5055     if (call->state == RX_STATE_PRECALL) {
5056         call->flags |= RX_CALL_CLEARED;
5057     }
5058 }
5059
5060 /* Send an abort packet for the specified call */
5061 struct rx_packet *
5062 rxi_SendCallAbort(struct rx_call *call, struct rx_packet *packet,
5063                   int istack, int force)
5064 {
5065     afs_int32 error;
5066     struct clock when, now;
5067
5068     if (!call->error)
5069         return packet;
5070
5071     /* Clients should never delay abort messages */
5072     if (rx_IsClientConn(call->conn))
5073         force = 1;
5074
5075     if (call->abortCode != call->error) {
5076         call->abortCode = call->error;
5077         call->abortCount = 0;
5078     }
5079
5080     if (force || rxi_callAbortThreshhold == 0
5081         || call->abortCount < rxi_callAbortThreshhold) {
5082         if (call->delayedAbortEvent) {
5083             rxevent_Cancel(call->delayedAbortEvent, call,
5084                            RX_CALL_REFCOUNT_ABORT);
5085         }
5086         error = htonl(call->error);
5087         call->abortCount++;
5088         packet =
5089             rxi_SendSpecial(call, call->conn, packet, RX_PACKET_TYPE_ABORT,
5090                             (char *)&error, sizeof(error), istack);
5091     } else if (!call->delayedAbortEvent) {
5092         clock_GetTime(&now);
5093         when = now;
5094         clock_Addmsec(&when, rxi_callAbortDelay);
5095         MUTEX_ENTER(&rx_refcnt_mutex);
5096         CALL_HOLD(call, RX_CALL_REFCOUNT_ABORT);
5097         MUTEX_EXIT(&rx_refcnt_mutex);
5098         call->delayedAbortEvent =
5099             rxevent_PostNow(&when, &now, rxi_SendDelayedCallAbort, call, 0);
5100     }
5101     return packet;
5102 }
5103
5104 /* Send an abort packet for the specified connection.  Packet is an
5105  * optional pointer to a packet that can be used to send the abort.
5106  * Once the number of abort messages reaches the threshhold, an
5107  * event is scheduled to send the abort. Setting the force flag
5108  * overrides sending delayed abort messages.
5109  *
5110  * NOTE: Called with conn_data_lock held. conn_data_lock is dropped
5111  *       to send the abort packet.
5112  */
5113 struct rx_packet *
5114 rxi_SendConnectionAbort(struct rx_connection *conn,
5115                         struct rx_packet *packet, int istack, int force)
5116 {
5117     afs_int32 error;
5118     struct clock when, now;
5119
5120     if (!conn->error)
5121         return packet;
5122
5123     /* Clients should never delay abort messages */
5124     if (rx_IsClientConn(conn))
5125         force = 1;
5126
5127     if (force || rxi_connAbortThreshhold == 0
5128         || conn->abortCount < rxi_connAbortThreshhold) {
5129         if (conn->delayedAbortEvent) {
5130             rxevent_Cancel(conn->delayedAbortEvent, (struct rx_call *)0, 0);
5131         }
5132         error = htonl(conn->error);
5133         conn->abortCount++;
5134         MUTEX_EXIT(&conn->conn_data_lock);
5135         packet =
5136             rxi_SendSpecial((struct rx_call *)0, conn, packet,
5137                             RX_PACKET_TYPE_ABORT, (char *)&error,
5138                             sizeof(error), istack);
5139         MUTEX_ENTER(&conn->conn_data_lock);
5140     } else if (!conn->delayedAbortEvent) {
5141         clock_GetTime(&now);
5142         when = now;
5143         clock_Addmsec(&when, rxi_connAbortDelay);
5144         conn->delayedAbortEvent =
5145             rxevent_PostNow(&when, &now, rxi_SendDelayedConnAbort, conn, 0);
5146     }
5147     return packet;
5148 }
5149
5150 /* Associate an error all of the calls owned by a connection.  Called
5151  * with error non-zero.  This is only for really fatal things, like
5152  * bad authentication responses.  The connection itself is set in
5153  * error at this point, so that future packets received will be
5154  * rejected. */
5155 void
5156 rxi_ConnectionError(struct rx_connection *conn,
5157                     afs_int32 error)
5158 {
5159     if (error) {
5160         int i;
5161
5162         dpf(("rxi_ConnectionError conn %"AFS_PTR_FMT" error %d\n", conn, error));
5163
5164         MUTEX_ENTER(&conn->conn_data_lock);
5165         if (conn->challengeEvent)
5166             rxevent_Cancel(conn->challengeEvent, (struct rx_call *)0, 0);
5167         if (conn->natKeepAliveEvent)
5168             rxevent_Cancel(conn->natKeepAliveEvent, (struct rx_call *)0, 0);
5169         if (conn->checkReachEvent) {
5170             rxevent_Cancel(conn->checkReachEvent, (struct rx_call *)0, 0);
5171             conn->checkReachEvent = 0;
5172             conn->flags &= ~(RX_CONN_ATTACHWAIT|RX_CONN_NAT_PING);
5173             MUTEX_ENTER(&rx_refcnt_mutex);
5174             conn->refCount--;
5175             MUTEX_EXIT(&rx_refcnt_mutex);
5176         }
5177         MUTEX_EXIT(&conn->conn_data_lock);
5178         for (i = 0; i < RX_MAXCALLS; i++) {
5179             struct rx_call *call = conn->call[i];
5180             if (call) {
5181                 MUTEX_ENTER(&call->lock);
5182                 rxi_CallError(call, error);
5183                 MUTEX_EXIT(&call->lock);
5184             }
5185         }
5186         conn->error = error;
5187         if (rx_stats_active)
5188             rx_atomic_inc(&rx_stats.fatalErrors);
5189     }
5190 }
5191
5192 /**
5193  * Interrupt an in-progress call with the specified error and wakeup waiters.
5194  *
5195  * @param[in] call  The call to interrupt
5196  * @param[in] error  The error code to send to the peer
5197  */
5198 void
5199 rx_InterruptCall(struct rx_call *call, afs_int32 error)
5200 {
5201     MUTEX_ENTER(&call->lock);
5202     rxi_CallError(call, error);
5203     rxi_SendCallAbort(call, NULL, 0, 1);
5204     MUTEX_EXIT(&call->lock);
5205 }
5206
5207 void
5208 rxi_CallError(struct rx_call *call, afs_int32 error)
5209 {
5210 #ifdef DEBUG
5211     osirx_AssertMine(&call->lock, "rxi_CallError");
5212 #endif
5213     dpf(("rxi_CallError call %"AFS_PTR_FMT" error %d call->error %d\n", call, error, call->error));
5214     if (call->error)
5215         error = call->error;
5216
5217 #ifdef AFS_GLOBAL_RXLOCK_KERNEL
5218     if (!((call->flags & RX_CALL_TQ_BUSY) || (call->tqWaiters > 0))) {
5219         rxi_ResetCall(call, 0);
5220     }
5221 #else
5222     rxi_ResetCall(call, 0);
5223 #endif
5224     call->error = error;
5225 }
5226
5227 /* Reset various fields in a call structure, and wakeup waiting
5228  * processes.  Some fields aren't changed: state & mode are not
5229  * touched (these must be set by the caller), and bufptr, nLeft, and
5230  * nFree are not reset, since these fields are manipulated by
5231  * unprotected macros, and may only be reset by non-interrupting code.
5232  */
5233 #ifdef ADAPT_WINDOW
5234 /* this code requires that call->conn be set properly as a pre-condition. */
5235 #endif /* ADAPT_WINDOW */
5236
5237 void
5238 rxi_ResetCall(struct rx_call *call, int newcall)
5239 {
5240     int flags;
5241     struct rx_peer *peer;
5242     struct rx_packet *packet;
5243 #ifdef DEBUG
5244     osirx_AssertMine(&call->lock, "rxi_ResetCall");
5245 #endif
5246     dpf(("rxi_ResetCall(call %"AFS_PTR_FMT", newcall %d)\n", call, newcall));
5247
5248     /* Notify anyone who is waiting for asynchronous packet arrival */
5249     if (call->arrivalProc) {
5250         (*call->arrivalProc) (call, call->arrivalProcHandle,
5251                               call->arrivalProcArg);
5252         call->arrivalProc = (void (*)())0;
5253     }
5254
5255     if (call->growMTUEvent)
5256         rxevent_Cancel(call->growMTUEvent, call,
5257                        RX_CALL_REFCOUNT_ALIVE);
5258
5259     if (call->delayedAbortEvent) {
5260         rxevent_Cancel(call->delayedAbortEvent, call, RX_CALL_REFCOUNT_ABORT);
5261         packet = rxi_AllocPacket(RX_PACKET_CLASS_SPECIAL);
5262         if (packet) {
5263             rxi_SendCallAbort(call, packet, 0, 1);
5264             rxi_FreePacket(packet);
5265         }
5266     }
5267
5268     /*
5269      * Update the peer with the congestion information in this call
5270      * so other calls on this connection can pick up where this call
5271      * left off. If the congestion sequence numbers don't match then
5272      * another call experienced a retransmission.
5273      */
5274     peer = call->conn->peer;
5275     MUTEX_ENTER(&peer->peer_lock);
5276     if (!newcall) {
5277         if (call->congestSeq == peer->congestSeq) {
5278             peer->cwind = MAX(peer->cwind, call->cwind);
5279             peer->MTU = MAX(peer->MTU, call->MTU);
5280             peer->nDgramPackets =
5281                 MAX(peer->nDgramPackets, call->nDgramPackets);
5282         }
5283     } else {
5284         call->abortCode = 0;
5285         call->abortCount = 0;
5286     }
5287     if (peer->maxDgramPackets > 1) {
5288         call->MTU = RX_HEADER_SIZE + RX_JUMBOBUFFERSIZE;
5289     } else {
5290         call->MTU = peer->MTU;
5291     }
5292     call->cwind = MIN((int)peer->cwind, (int)peer->nDgramPackets);
5293     call->ssthresh = rx_maxSendWindow;
5294     call->nDgramPackets = peer->nDgramPackets;
5295     call->congestSeq = peer->congestSeq;
5296     call->rtt = peer->rtt;
5297     call->rtt_dev = peer->rtt_dev;
5298     clock_Zero(&call->rto);
5299     clock_Addmsec(&call->rto,
5300                   MAX(((call->rtt >> 3) + call->rtt_dev), rx_minPeerTimeout) + 200);
5301     MUTEX_EXIT(&peer->peer_lock);
5302
5303     flags = call->flags;
5304 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
5305     rxi_WaitforTQBusy(call);
5306 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
5307
5308     rxi_ClearTransmitQueue(call, 1);
5309     if (call->tqWaiters || (flags & RX_CALL_TQ_WAIT)) {
5310         dpf(("rcall %"AFS_PTR_FMT" has %d waiters and flags %d\n", call, call->tqWaiters, call->flags));
5311     }
5312     call->flags = 0;
5313
5314     if ((flags & RX_CALL_PEER_BUSY)) {
5315         /* The call channel is still busy; resetting the call doesn't change
5316          * that */
5317         call->flags |= RX_CALL_PEER_BUSY;
5318     }
5319
5320     rxi_ClearReceiveQueue(call);
5321     /* why init the queue if you just emptied it? queue_Init(&call->rq); */
5322
5323
5324     call->error = 0;
5325     call->twind = call->conn->twind[call->channel];
5326     call->rwind = call->conn->rwind[call->channel];
5327     call->nSoftAcked = 0;
5328     call->nextCwind = 0;
5329     call->nAcks = 0;
5330     call->nNacks = 0;
5331     call->nCwindAcks = 0;
5332     call->nSoftAcks = 0;
5333     call->nHardAcks = 0;
5334
5335     call->tfirst = call->rnext = call->tnext = 1;
5336     call->tprev = 0;
5337     call->rprev = 0;
5338     call->lastAcked = 0;
5339     call->localStatus = call->remoteStatus = 0;
5340
5341     if (flags & RX_CALL_READER_WAIT) {
5342 #ifdef  RX_ENABLE_LOCKS
5343         CV_BROADCAST(&call->cv_rq);
5344 #else
5345         osi_rxWakeup(&call->rq);
5346 #endif
5347     }
5348     if (flags & RX_CALL_WAIT_PACKETS) {
5349         MUTEX_ENTER(&rx_freePktQ_lock);
5350         rxi_PacketsUnWait();    /* XXX */
5351         MUTEX_EXIT(&rx_freePktQ_lock);
5352     }
5353 #ifdef  RX_ENABLE_LOCKS
5354     CV_SIGNAL(&call->cv_twind);
5355 #else
5356     if (flags & RX_CALL_WAIT_WINDOW_ALLOC)
5357         osi_rxWakeup(&call->twind);
5358 #endif
5359
5360 #ifdef RX_ENABLE_LOCKS
5361     /* The following ensures that we don't mess with any queue while some
5362      * other thread might also be doing so. The call_queue_lock field is
5363      * is only modified under the call lock. If the call is in the process
5364      * of being removed from a queue, the call is not locked until the
5365      * the queue lock is dropped and only then is the call_queue_lock field
5366      * zero'd out. So it's safe to lock the queue if call_queue_lock is set.
5367      * Note that any other routine which removes a call from a queue has to
5368      * obtain the queue lock before examing the queue and removing the call.
5369      */
5370     if (call->call_queue_lock) {
5371         MUTEX_ENTER(call->call_queue_lock);
5372         if (queue_IsOnQueue(call)) {
5373             queue_Remove(call);
5374             if (flags & RX_CALL_WAIT_PROC) {
5375                 rx_atomic_dec(&rx_nWaiting);
5376             }
5377         }
5378         MUTEX_EXIT(call->call_queue_lock);
5379         CLEAR_CALL_QUEUE_LOCK(call);
5380     }
5381 #else /* RX_ENABLE_LOCKS */
5382     if (queue_IsOnQueue(call)) {
5383         queue_Remove(call);
5384         if (flags & RX_CALL_WAIT_PROC)
5385             rx_atomic_dec(&rx_nWaiting);
5386     }
5387 #endif /* RX_ENABLE_LOCKS */
5388
5389     rxi_KeepAliveOff(call);
5390     rxevent_Cancel(call->delayedAckEvent, call, RX_CALL_REFCOUNT_DELAY);
5391 }
5392
5393 /* Send an acknowledge for the indicated packet (seq,serial) of the
5394  * indicated call, for the indicated reason (reason).  This
5395  * acknowledge will specifically acknowledge receiving the packet, and
5396  * will also specify which other packets for this call have been
5397  * received.  This routine returns the packet that was used to the
5398  * caller.  The caller is responsible for freeing it or re-using it.
5399  * This acknowledgement also returns the highest sequence number
5400  * actually read out by the higher level to the sender; the sender
5401  * promises to keep around packets that have not been read by the
5402  * higher level yet (unless, of course, the sender decides to abort
5403  * the call altogether).  Any of p, seq, serial, pflags, or reason may
5404  * be set to zero without ill effect.  That is, if they are zero, they
5405  * will not convey any information.
5406  * NOW there is a trailer field, after the ack where it will safely be
5407  * ignored by mundanes, which indicates the maximum size packet this
5408  * host can swallow.  */
5409 /*
5410     struct rx_packet *optionalPacket;  use to send ack (or null)
5411     int seq;                     Sequence number of the packet we are acking
5412     int serial;                  Serial number of the packet
5413     int pflags;                  Flags field from packet header
5414     int reason;                  Reason an acknowledge was prompted
5415 */
5416
5417 struct rx_packet *
5418 rxi_SendAck(struct rx_call *call,
5419             struct rx_packet *optionalPacket, int serial, int reason,
5420             int istack)
5421 {
5422     struct rx_ackPacket *ap;
5423     struct rx_packet *rqp;
5424     struct rx_packet *nxp;      /* For queue_Scan */
5425     struct rx_packet *p;
5426     u_char offset;
5427     afs_int32 templ;
5428     afs_uint32 padbytes = 0;
5429 #ifdef RX_ENABLE_TSFPQ
5430     struct rx_ts_info_t * rx_ts_info;
5431 #endif
5432
5433     /*
5434      * Open the receive window once a thread starts reading packets
5435      */
5436     if (call->rnext > 1) {
5437         call->conn->rwind[call->channel] = call->rwind = rx_maxReceiveWindow;
5438     }
5439
5440     /* Don't attempt to grow MTU if this is a critical ping */
5441     if (reason == RX_ACK_MTU) {
5442         /* keep track of per-call attempts, if we're over max, do in small
5443          * otherwise in larger? set a size to increment by, decrease
5444          * on failure, here?
5445          */
5446         if (call->conn->peer->maxPacketSize &&
5447             (call->conn->peer->maxPacketSize < OLD_MAX_PACKET_SIZE
5448              +RX_IPUDP_SIZE))
5449             padbytes = call->conn->peer->maxPacketSize+16;
5450         else
5451             padbytes = call->conn->peer->maxMTU + 128;
5452
5453         /* do always try a minimum size ping */
5454         padbytes = MAX(padbytes, RX_MIN_PACKET_SIZE+RX_IPUDP_SIZE+4);
5455
5456         /* subtract the ack payload */
5457         padbytes -= (rx_AckDataSize(call->rwind) + 4 * sizeof(afs_int32));
5458         reason = RX_ACK_PING;
5459     }
5460
5461     call->nHardAcks = 0;
5462     call->nSoftAcks = 0;
5463     if (call->rnext > call->lastAcked)
5464         call->lastAcked = call->rnext;
5465     p = optionalPacket;
5466
5467     if (p) {
5468         rx_computelen(p, p->length);    /* reset length, you never know */
5469     } /* where that's been...         */
5470 #ifdef RX_ENABLE_TSFPQ
5471     else {
5472         RX_TS_INFO_GET(rx_ts_info);
5473         if ((p = rx_ts_info->local_special_packet)) {
5474             rx_computelen(p, p->length);
5475         } else if ((p = rxi_AllocPacket(RX_PACKET_CLASS_SPECIAL))) {
5476             rx_ts_info->local_special_packet = p;
5477         } else { /* We won't send the ack, but don't panic. */
5478             return optionalPacket;
5479         }
5480     }
5481 #else
5482     else if (!(p = rxi_AllocPacket(RX_PACKET_CLASS_SPECIAL))) {
5483         /* We won't send the ack, but don't panic. */
5484         return optionalPacket;
5485     }
5486 #endif
5487
5488     templ = padbytes +
5489         rx_AckDataSize(call->rwind) + 4 * sizeof(afs_int32) -
5490         rx_GetDataSize(p);
5491     if (templ > 0) {
5492         if (rxi_AllocDataBuf(p, templ, RX_PACKET_CLASS_SPECIAL) > 0) {
5493 #ifndef RX_ENABLE_TSFPQ
5494             if (!optionalPacket)
5495                 rxi_FreePacket(p);
5496 #endif
5497             return optionalPacket;
5498         }
5499         templ = rx_AckDataSize(call->rwind) + 2 * sizeof(afs_int32);
5500         if (rx_Contiguous(p) < templ) {
5501 #ifndef RX_ENABLE_TSFPQ
5502             if (!optionalPacket)
5503                 rxi_FreePacket(p);
5504 #endif
5505             return optionalPacket;
5506         }
5507     }
5508
5509
5510     /* MTUXXX failing to send an ack is very serious.  We should */
5511     /* try as hard as possible to send even a partial ack; it's */
5512     /* better than nothing. */
5513     ap = (struct rx_ackPacket *)rx_DataOf(p);
5514     ap->bufferSpace = htonl(0); /* Something should go here, sometime */
5515     ap->reason = reason;
5516
5517     /* The skew computation used to be bogus, I think it's better now. */
5518     /* We should start paying attention to skew.    XXX  */
5519     ap->serial = htonl(serial);
5520     ap->maxSkew = 0;            /* used to be peer->inPacketSkew */
5521
5522     /*
5523      * First packet not yet forwarded to reader. When ACKALL has been
5524      * sent the peer has been told that all received packets will be
5525      * delivered to the reader.  The value 'rnext' is used internally
5526      * to refer to the next packet in the receive queue that must be
5527      * delivered to the reader.  From the perspective of the peer it
5528      * already has so report the last sequence number plus one if there
5529      * are packets in the receive queue awaiting processing.
5530      */
5531     if ((call->flags & RX_CALL_ACKALL_SENT) &&
5532         !queue_IsEmpty(&call->rq)) {
5533         ap->firstPacket = htonl(queue_Last(&call->rq, rx_packet)->header.seq + 1);
5534     } else
5535         ap->firstPacket = htonl(call->rnext);
5536
5537     ap->previousPacket = htonl(call->rprev);    /* Previous packet received */
5538
5539     /* No fear of running out of ack packet here because there can only be at most
5540      * one window full of unacknowledged packets.  The window size must be constrained
5541      * to be less than the maximum ack size, of course.  Also, an ack should always
5542      * fit into a single packet -- it should not ever be fragmented.  */
5543     for (offset = 0, queue_Scan(&call->rq, rqp, nxp, rx_packet)) {
5544         if (!rqp || !call->rq.next
5545             || (rqp->header.seq > (call->rnext + call->rwind))) {
5546 #ifndef RX_ENABLE_TSFPQ
5547             if (!optionalPacket)
5548                 rxi_FreePacket(p);
5549 #endif
5550             rxi_CallError(call, RX_CALL_DEAD);
5551             return optionalPacket;
5552         }
5553
5554         while (rqp->header.seq > call->rnext + offset)
5555             ap->acks[offset++] = RX_ACK_TYPE_NACK;
5556         ap->acks[offset++] = RX_ACK_TYPE_ACK;
5557
5558         if ((offset > (u_char) rx_maxReceiveWindow) || (offset > call->rwind)) {
5559 #ifndef RX_ENABLE_TSFPQ
5560             if (!optionalPacket)
5561                 rxi_FreePacket(p);
5562 #endif
5563             rxi_CallError(call, RX_CALL_DEAD);
5564             return optionalPacket;
5565         }
5566     }
5567
5568     ap->nAcks = offset;
5569     p->length = rx_AckDataSize(offset) + 4 * sizeof(afs_int32);
5570
5571     /* these are new for AFS 3.3 */
5572     templ = rxi_AdjustMaxMTU(call->conn->peer->ifMTU, rx_maxReceiveSize);
5573     templ = htonl(templ);
5574     rx_packetwrite(p, rx_AckDataSize(offset), sizeof(afs_int32), &templ);
5575     templ = htonl(call->conn->peer->ifMTU);
5576     rx_packetwrite(p, rx_AckDataSize(offset) + sizeof(afs_int32),
5577                    sizeof(afs_int32), &templ);
5578
5579     /* new for AFS 3.4 */
5580     templ = htonl(call->rwind);
5581     rx_packetwrite(p, rx_AckDataSize(offset) + 2 * sizeof(afs_int32),
5582                    sizeof(afs_int32), &templ);
5583
5584     /* new for AFS 3.5 */
5585     templ = htonl(call->conn->peer->ifDgramPackets);
5586     rx_packetwrite(p, rx_AckDataSize(offset) + 3 * sizeof(afs_int32),
5587                    sizeof(afs_int32), &templ);
5588
5589     p->header.serviceId = call->conn->serviceId;
5590     p->header.cid = (call->conn->cid | call->channel);
5591     p->header.callNumber = *call->callNumber;
5592     p->header.seq = 0;
5593     p->header.securityIndex = call->conn->securityIndex;
5594     p->header.epoch = call->conn->epoch;
5595     p->header.type = RX_PACKET_TYPE_ACK;
5596     p->header.flags = RX_SLOW_START_OK;
5597     if (reason == RX_ACK_PING) {
5598         p->header.flags |= RX_REQUEST_ACK;
5599 #ifdef ADAPT_WINDOW
5600         clock_GetTime(&call->pingRequestTime);
5601 #endif
5602         if (padbytes) {
5603             p->length = padbytes +
5604                 rx_AckDataSize(call->rwind) + 4 * sizeof(afs_int32);
5605
5606             while (padbytes--)
5607                 /* not fast but we can potentially use this if truncated
5608                  * fragments are delivered to figure out the mtu.
5609                  */
5610                 rx_packetwrite(p, rx_AckDataSize(offset) + 4 *
5611                                sizeof(afs_int32), sizeof(afs_int32),
5612                                &padbytes);
5613         }
5614     }
5615     if (call->conn->type == RX_CLIENT_CONNECTION)
5616         p->header.flags |= RX_CLIENT_INITIATED;
5617
5618 #ifdef RXDEBUG
5619 #ifdef AFS_NT40_ENV
5620     if (rxdebug_active) {
5621         char msg[512];
5622         size_t len;
5623
5624         len = _snprintf(msg, sizeof(msg),
5625                         "tid[%d] SACK: reason %s serial %u previous %u seq %u first %u acks %u space %u ",
5626                          GetCurrentThreadId(), rx_ack_reason(ap->reason),
5627                          ntohl(ap->serial), ntohl(ap->previousPacket),
5628                          (unsigned int)p->header.seq, ntohl(ap->firstPacket),
5629                          ap->nAcks, ntohs(ap->bufferSpace) );
5630         if (ap->nAcks) {
5631             int offset;
5632
5633             for (offset = 0; offset < ap->nAcks && len < sizeof(msg); offset++)
5634                 msg[len++] = (ap->acks[offset] == RX_ACK_TYPE_NACK ? '-' : '*');
5635         }
5636         msg[len++]='\n';
5637         msg[len] = '\0';
5638         OutputDebugString(msg);
5639     }
5640 #else /* AFS_NT40_ENV */
5641     if (rx_Log) {
5642         fprintf(rx_Log, "SACK: reason %x previous %u seq %u first %u ",
5643                 ap->reason, ntohl(ap->previousPacket),
5644                 (unsigned int)p->header.seq, ntohl(ap->firstPacket));
5645         if (ap->nAcks) {
5646             for (offset = 0; offset < ap->nAcks; offset++)
5647                 putc(ap->acks[offset] == RX_ACK_TYPE_NACK ? '-' : '*',
5648                      rx_Log);
5649         }
5650         putc('\n', rx_Log);
5651     }
5652 #endif /* AFS_NT40_ENV */
5653 #endif
5654     {
5655         int i, nbytes = p->length;
5656
5657         for (i = 1; i < p->niovecs; i++) {      /* vec 0 is ALWAYS header */
5658             if (nbytes <= p->wirevec[i].iov_len) {
5659                 int savelen, saven;
5660
5661                 savelen = p->wirevec[i].iov_len;
5662                 saven = p->niovecs;
5663                 p->wirevec[i].iov_len = nbytes;
5664                 p->niovecs = i + 1;
5665                 rxi_Send(call, p, istack);
5666                 p->wirevec[i].iov_len = savelen;
5667                 p->niovecs = saven;
5668                 break;
5669             } else
5670                 nbytes -= p->wirevec[i].iov_len;
5671         }
5672     }
5673     if (rx_stats_active)
5674         rx_atomic_inc(&rx_stats.ackPacketsSent);
5675 #ifndef RX_ENABLE_TSFPQ
5676     if (!optionalPacket)
5677         rxi_FreePacket(p);
5678 #endif
5679     return optionalPacket;      /* Return packet for re-use by caller */
5680 }
5681
5682 struct xmitlist {
5683    struct rx_packet **list;
5684    int len;
5685    int resending;
5686 };
5687
5688 /* Send all of the packets in the list in single datagram */
5689 static void
5690 rxi_SendList(struct rx_call *call, struct xmitlist *xmit,
5691              int istack, int moreFlag)
5692 {
5693     int i;
5694     int requestAck = 0;
5695     int lastPacket = 0;
5696     struct clock now;
5697     struct rx_connection *conn = call->conn;
5698     struct rx_peer *peer = conn->peer;
5699
5700     MUTEX_ENTER(&peer->peer_lock);
5701     peer->nSent += xmit->len;
5702     if (xmit->resending)
5703         peer->reSends += xmit->len;
5704     MUTEX_EXIT(&peer->peer_lock);
5705
5706     if (rx_stats_active) {
5707         if (xmit->resending)
5708             rx_atomic_add(&rx_stats.dataPacketsReSent, xmit->len);
5709         else
5710             rx_atomic_add(&rx_stats.dataPacketsSent, xmit->len);
5711     }
5712
5713     clock_GetTime(&now);
5714
5715     if (xmit->list[xmit->len - 1]->header.flags & RX_LAST_PACKET) {
5716         lastPacket = 1;
5717     }
5718
5719     /* Set the packet flags and schedule the resend events */
5720     /* Only request an ack for the last packet in the list */
5721     for (i = 0; i < xmit->len; i++) {
5722         struct rx_packet *packet = xmit->list[i];
5723
5724         /* Record the time sent */
5725         packet->timeSent = now;
5726         packet->flags |= RX_PKTFLAG_SENT;
5727
5728         /* Ask for an ack on retransmitted packets,  on every other packet
5729          * if the peer doesn't support slow start. Ask for an ack on every
5730          * packet until the congestion window reaches the ack rate. */
5731         if (packet->header.serial) {
5732             requestAck = 1;
5733         } else {
5734             packet->firstSent = now;
5735             if (!lastPacket && (call->cwind <= (u_short) (conn->ackRate + 1)
5736                                 || (!(call->flags & RX_CALL_SLOW_START_OK)
5737                                     && (packet->header.seq & 1)))) {
5738                 requestAck = 1;
5739             }
5740         }
5741
5742         /* Tag this packet as not being the last in this group,
5743          * for the receiver's benefit */
5744         if (i < xmit->len - 1 || moreFlag) {
5745             packet->header.flags |= RX_MORE_PACKETS;
5746         }
5747     }
5748
5749     if (requestAck) {
5750         xmit->list[xmit->len - 1]->header.flags |= RX_REQUEST_ACK;
5751     }
5752
5753     /* Since we're about to send a data packet to the peer, it's
5754      * safe to nuke any scheduled end-of-packets ack */
5755     rxevent_Cancel(call->delayedAckEvent, call, RX_CALL_REFCOUNT_DELAY);
5756
5757     MUTEX_EXIT(&call->lock);
5758     MUTEX_ENTER(&rx_refcnt_mutex);
5759     CALL_HOLD(call, RX_CALL_REFCOUNT_SEND);
5760     MUTEX_EXIT(&rx_refcnt_mutex);
5761     if (xmit->len > 1) {
5762         rxi_SendPacketList(call, conn, xmit->list, xmit->len, istack);
5763     } else {
5764         rxi_SendPacket(call, conn, xmit->list[0], istack);
5765     }
5766     MUTEX_ENTER(&call->lock);
5767     MUTEX_ENTER(&rx_refcnt_mutex);
5768     CALL_RELE(call, RX_CALL_REFCOUNT_SEND);
5769     MUTEX_EXIT(&rx_refcnt_mutex);
5770
5771     /* Tell the RTO calculation engine that we have sent a packet, and
5772      * if it was the last one */
5773     rxi_rto_packet_sent(call, lastPacket, istack);
5774
5775     /* Update last send time for this call (for keep-alive
5776      * processing), and for the connection (so that we can discover
5777      * idle connections) */
5778     conn->lastSendTime = call->lastSendTime = clock_Sec();
5779     /* Let a set of retransmits trigger an idle timeout */
5780     if (!xmit->resending)
5781         call->lastSendData = call->lastSendTime;
5782 }
5783
5784 /* When sending packets we need to follow these rules:
5785  * 1. Never send more than maxDgramPackets in a jumbogram.
5786  * 2. Never send a packet with more than two iovecs in a jumbogram.
5787  * 3. Never send a retransmitted packet in a jumbogram.
5788  * 4. Never send more than cwind/4 packets in a jumbogram
5789  * We always keep the last list we should have sent so we
5790  * can set the RX_MORE_PACKETS flags correctly.
5791  */
5792
5793 static void
5794 rxi_SendXmitList(struct rx_call *call, struct rx_packet **list, int len,
5795                  int istack)
5796 {
5797     int i;
5798     int recovery;
5799     struct xmitlist working;
5800     struct xmitlist last;
5801
5802     struct rx_peer *peer = call->conn->peer;
5803     int morePackets = 0;
5804
5805     memset(&last, 0, sizeof(struct xmitlist));
5806     working.list = &list[0];
5807     working.len = 0;
5808     working.resending = 0;
5809
5810     recovery = call->flags & RX_CALL_FAST_RECOVER;
5811
5812     for (i = 0; i < len; i++) {
5813         /* Does the current packet force us to flush the current list? */
5814         if (working.len > 0
5815             && (list[i]->header.serial || (list[i]->flags & RX_PKTFLAG_ACKED)
5816                 || list[i]->length > RX_JUMBOBUFFERSIZE)) {
5817
5818             /* This sends the 'last' list and then rolls the current working
5819              * set into the 'last' one, and resets the working set */
5820
5821             if (last.len > 0) {
5822                 rxi_SendList(call, &last, istack, 1);
5823                 /* If the call enters an error state stop sending, or if
5824                  * we entered congestion recovery mode, stop sending */
5825                 if (call->error
5826                     || (!recovery && (call->flags & RX_CALL_FAST_RECOVER)))
5827                     return;
5828             }
5829             last = working;
5830             working.len = 0;
5831             working.resending = 0;
5832             working.list = &list[i];
5833         }
5834         /* Add the current packet to the list if it hasn't been acked.
5835          * Otherwise adjust the list pointer to skip the current packet.  */
5836         if (!(list[i]->flags & RX_PKTFLAG_ACKED)) {
5837             working.len++;
5838
5839             if (list[i]->header.serial)
5840                 working.resending = 1;
5841
5842             /* Do we need to flush the list? */
5843             if (working.len >= (int)peer->maxDgramPackets
5844                 || working.len >= (int)call->nDgramPackets 
5845                 || working.len >= (int)call->cwind
5846                 || list[i]->header.serial
5847                 || list[i]->length != RX_JUMBOBUFFERSIZE) {
5848                 if (last.len > 0) {
5849                     rxi_SendList(call, &last, istack, 1);
5850                     /* If the call enters an error state stop sending, or if
5851                      * we entered congestion recovery mode, stop sending */
5852                     if (call->error
5853                         || (!recovery && (call->flags & RX_CALL_FAST_RECOVER)))
5854                         return;
5855                 }
5856                 last = working;
5857                 working.len = 0;
5858                 working.resending = 0;
5859                 working.list = &list[i + 1];
5860             }
5861         } else {
5862             if (working.len != 0) {
5863                 osi_Panic("rxi_SendList error");
5864             }
5865             working.list = &list[i + 1];
5866         }
5867     }
5868
5869     /* Send the whole list when the call is in receive mode, when
5870      * the call is in eof mode, when we are in fast recovery mode,
5871      * and when we have the last packet */
5872     if ((list[len - 1]->header.flags & RX_LAST_PACKET)
5873         || call->mode == RX_MODE_RECEIVING || call->mode == RX_MODE_EOF
5874         || (call->flags & RX_CALL_FAST_RECOVER)) {
5875         /* Check for the case where the current list contains
5876          * an acked packet. Since we always send retransmissions
5877          * in a separate packet, we only need to check the first
5878          * packet in the list */
5879         if (working.len > 0 && !(working.list[0]->flags & RX_PKTFLAG_ACKED)) {
5880             morePackets = 1;
5881         }
5882         if (last.len > 0) {
5883             rxi_SendList(call, &last, istack, morePackets);
5884             /* If the call enters an error state stop sending, or if
5885              * we entered congestion recovery mode, stop sending */
5886             if (call->error
5887                 || (!recovery && (call->flags & RX_CALL_FAST_RECOVER)))
5888                 return;
5889         }
5890         if (morePackets) {
5891             rxi_SendList(call, &working, istack, 0);
5892         }
5893     } else if (last.len > 0) {
5894         rxi_SendList(call, &last, istack, 0);
5895         /* Packets which are in 'working' are not sent by this call */
5896     }
5897 }
5898
5899 static void
5900 rxi_Resend(struct rxevent *event, void *arg0, void *arg1, int istack)
5901 {
5902     struct rx_call *call = arg0;
5903     struct rx_peer *peer;
5904     struct rx_packet *p, *nxp;
5905     struct clock maxTimeout = { 60, 0 };
5906
5907     MUTEX_ENTER(&call->lock);
5908
5909     peer = call->conn->peer;
5910
5911     /* Make sure that the event pointer is removed from the call
5912      * structure, since there is no longer a per-call retransmission
5913      * event pending. */
5914     if (event == call->resendEvent) {
5915         MUTEX_ENTER(&rx_refcnt_mutex);
5916         CALL_RELE(call, RX_CALL_REFCOUNT_RESEND);
5917         MUTEX_EXIT(&rx_refcnt_mutex);
5918         call->resendEvent = NULL;
5919     }
5920
5921     if (rxi_busyChannelError && (call->flags & RX_CALL_PEER_BUSY)) {
5922         rxi_CheckBusy(call);
5923     }
5924
5925     if (queue_IsEmpty(&call->tq)) {
5926         /* Nothing to do. This means that we've been raced, and that an
5927          * ACK has come in between when we were triggered, and when we
5928          * actually got to run. */
5929         goto out;
5930     }
5931
5932     /* We're in loss recovery */
5933     call->flags |= RX_CALL_FAST_RECOVER;
5934
5935     /* Mark all of the pending packets in the queue as being lost */
5936     for (queue_Scan(&call->tq, p, nxp, rx_packet)) {
5937         if (!(p->flags & RX_PKTFLAG_ACKED))
5938             p->flags &= ~RX_PKTFLAG_SENT;
5939     }
5940
5941     /* We're resending, so we double the timeout of the call. This will be
5942      * dropped back down by the first successful ACK that we receive.
5943      *
5944      * We apply a maximum value here of 60 seconds
5945      */
5946     clock_Add(&call->rto, &call->rto);
5947     if (clock_Gt(&call->rto, &maxTimeout))
5948         call->rto = maxTimeout;
5949
5950     /* Packet loss is most likely due to congestion, so drop our window size
5951      * and start again from the beginning */
5952     if (peer->maxDgramPackets >1) {
5953         call->MTU = RX_JUMBOBUFFERSIZE + RX_HEADER_SIZE;
5954         call->MTU = MIN(peer->natMTU, peer->maxMTU);
5955     }
5956     call->ssthresh = MAX(4, MIN((int)call->cwind, (int)call->twind)) >> 1;
5957     call->nDgramPackets = 1;
5958     call->cwind = 1;
5959     call->nextCwind = 1;
5960     call->nAcks = 0;
5961     call->nNacks = 0;
5962     MUTEX_ENTER(&peer->peer_lock);
5963     peer->MTU = call->MTU;
5964     peer->cwind = call->cwind;
5965     peer->nDgramPackets = 1;
5966     peer->congestSeq++;
5967     call->congestSeq = peer->congestSeq;
5968     MUTEX_EXIT(&peer->peer_lock);
5969
5970     rxi_Start(call, istack);
5971
5972 out:
5973     MUTEX_EXIT(&call->lock);
5974 }
5975
5976 /* This routine is called when new packets are readied for
5977  * transmission and when retransmission may be necessary, or when the
5978  * transmission window or burst count are favourable.  This should be
5979  * better optimized for new packets, the usual case, now that we've
5980  * got rid of queues of send packets. XXXXXXXXXXX */
5981 void
5982 rxi_Start(struct rx_call *call, int istack)
5983 {
5984
5985     struct rx_packet *p;
5986     struct rx_packet *nxp;      /* Next pointer for queue_Scan */
5987     int nXmitPackets;
5988     int maxXmitPackets;
5989
5990     if (call->error) {
5991 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
5992         if (rx_stats_active)
5993             rx_atomic_inc(&rx_tq_debug.rxi_start_in_error);
5994 #endif
5995         return;
5996     }
5997
5998     if (queue_IsNotEmpty(&call->tq)) {  /* If we have anything to send */
5999
6000         /* Send (or resend) any packets that need it, subject to
6001          * window restrictions and congestion burst control
6002          * restrictions.  Ask for an ack on the last packet sent in
6003          * this burst.  For now, we're relying upon the window being
6004          * considerably bigger than the largest number of packets that
6005          * are typically sent at once by one initial call to
6006          * rxi_Start.  This is probably bogus (perhaps we should ask
6007          * for an ack when we're half way through the current
6008          * window?).  Also, for non file transfer applications, this
6009          * may end up asking for an ack for every packet.  Bogus. XXXX
6010          */
6011         /*
6012          * But check whether we're here recursively, and let the other guy
6013          * do the work.
6014          */
6015 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
6016         if (!(call->flags & RX_CALL_TQ_BUSY)) {
6017             call->flags |= RX_CALL_TQ_BUSY;
6018             do {
6019 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
6020             restart:
6021 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
6022                 call->flags &= ~RX_CALL_NEED_START;
6023 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
6024                 nXmitPackets = 0;
6025                 maxXmitPackets = MIN(call->twind, call->cwind);
6026                 for (queue_Scan(&call->tq, p, nxp, rx_packet)) {
6027 #ifdef RX_TRACK_PACKETS
6028                     if ((p->flags & RX_PKTFLAG_FREE)
6029                         || (!queue_IsEnd(&call->tq, nxp)
6030                             && (nxp->flags & RX_PKTFLAG_FREE))
6031                         || (p == (struct rx_packet *)&rx_freePacketQueue)
6032                         || (nxp == (struct rx_packet *)&rx_freePacketQueue)) {
6033                         osi_Panic("rxi_Start: xmit queue clobbered");
6034                     }
6035 #endif
6036                     if (p->flags & RX_PKTFLAG_ACKED) {
6037                         /* Since we may block, don't trust this */
6038                         if (rx_stats_active)
6039                             rx_atomic_inc(&rx_stats.ignoreAckedPacket);
6040                         continue;       /* Ignore this packet if it has been acknowledged */
6041                     }
6042
6043                     /* Turn off all flags except these ones, which are the same
6044                      * on each transmission */
6045                     p->header.flags &= RX_PRESET_FLAGS;
6046
6047                     if (p->header.seq >=
6048                         call->tfirst + MIN((int)call->twind,
6049                                            (int)(call->nSoftAcked +
6050                                                  call->cwind))) {
6051                         call->flags |= RX_CALL_WAIT_WINDOW_SEND;        /* Wait for transmit window */
6052                         /* Note: if we're waiting for more window space, we can
6053                          * still send retransmits; hence we don't return here, but
6054                          * break out to schedule a retransmit event */
6055                         dpf(("call %d waiting for window (seq %d, twind %d, nSoftAcked %d, cwind %d)\n",
6056                              *(call->callNumber), p->header.seq, call->twind, call->nSoftAcked,
6057                              call->cwind));
6058                         break;
6059                     }
6060
6061                     /* Transmit the packet if it needs to be sent. */
6062                     if (!(p->flags & RX_PKTFLAG_SENT)) {
6063                         if (nXmitPackets == maxXmitPackets) {
6064                             rxi_SendXmitList(call, call->xmitList,
6065                                              nXmitPackets, istack);
6066                             goto restart;
6067                         }
6068                         dpf(("call %d xmit packet %"AFS_PTR_FMT"\n",
6069                               *(call->callNumber), p));
6070                         call->xmitList[nXmitPackets++] = p;
6071                     }
6072                 }
6073
6074                 /* xmitList now hold pointers to all of the packets that are
6075                  * ready to send. Now we loop to send the packets */
6076                 if (nXmitPackets > 0) {
6077                     rxi_SendXmitList(call, call->xmitList, nXmitPackets,
6078                                      istack);
6079                 }
6080
6081 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
6082                 if (call->error) {
6083                     /* We went into the error state while sending packets. Now is
6084                      * the time to reset the call. This will also inform the using
6085                      * process that the call is in an error state.
6086                      */
6087                     if (rx_stats_active)
6088                         rx_atomic_inc(&rx_tq_debug.rxi_start_aborted);
6089                     call->flags &= ~RX_CALL_TQ_BUSY;
6090                     rxi_WakeUpTransmitQueue(call);
6091                     rxi_CallError(call, call->error);
6092                     return;
6093                 }
6094 #ifdef RX_ENABLE_LOCKS
6095                 if (call->flags & RX_CALL_TQ_SOME_ACKED) {
6096                     int missing;
6097                     call->flags &= ~RX_CALL_TQ_SOME_ACKED;
6098                     /* Some packets have received acks. If they all have, we can clear
6099                      * the transmit queue.
6100                      */
6101                     for (missing =
6102                          0, queue_Scan(&call->tq, p, nxp, rx_packet)) {
6103                         if (p->header.seq < call->tfirst
6104                             && (p->flags & RX_PKTFLAG_ACKED)) {
6105                             queue_Remove(p);
6106 #ifdef RX_TRACK_PACKETS
6107                             p->flags &= ~RX_PKTFLAG_TQ;
6108 #endif
6109 #ifdef RXDEBUG_PACKET
6110                             call->tqc--;
6111 #endif
6112                             rxi_FreePacket(p);
6113                         } else
6114                             missing = 1;
6115                     }
6116                     if (!missing)
6117                         call->flags |= RX_CALL_TQ_CLEARME;
6118                 }
6119 #endif /* RX_ENABLE_LOCKS */
6120                 if (call->flags & RX_CALL_TQ_CLEARME)
6121                     rxi_ClearTransmitQueue(call, 1);
6122             } while (call->flags & RX_CALL_NEED_START);
6123             /*
6124              * TQ references no longer protected by this flag; they must remain
6125              * protected by the global lock.
6126              */
6127             call->flags &= ~RX_CALL_TQ_BUSY;
6128             rxi_WakeUpTransmitQueue(call);
6129         } else {
6130             call->flags |= RX_CALL_NEED_START;
6131         }
6132 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
6133     } else {
6134         rxi_rto_cancel(call);
6135     }
6136 }
6137
6138 /* Also adjusts the keep alive parameters for the call, to reflect
6139  * that we have just sent a packet (so keep alives aren't sent
6140  * immediately) */
6141 void
6142 rxi_Send(struct rx_call *call, struct rx_packet *p,
6143          int istack)
6144 {
6145     struct rx_connection *conn = call->conn;
6146
6147     /* Stamp each packet with the user supplied status */
6148     p->header.userStatus = call->localStatus;
6149
6150     /* Allow the security object controlling this call's security to
6151      * make any last-minute changes to the packet */
6152     RXS_SendPacket(conn->securityObject, call, p);
6153
6154     /* Since we're about to send SOME sort of packet to the peer, it's
6155      * safe to nuke any scheduled end-of-packets ack */
6156     rxevent_Cancel(call->delayedAckEvent, call, RX_CALL_REFCOUNT_DELAY);
6157
6158     /* Actually send the packet, filling in more connection-specific fields */
6159     MUTEX_EXIT(&call->lock);
6160     MUTEX_ENTER(&rx_refcnt_mutex);
6161     CALL_HOLD(call, RX_CALL_REFCOUNT_SEND);
6162     MUTEX_EXIT(&rx_refcnt_mutex);
6163     rxi_SendPacket(call, conn, p, istack);
6164     MUTEX_ENTER(&rx_refcnt_mutex);
6165     CALL_RELE(call, RX_CALL_REFCOUNT_SEND);
6166     MUTEX_EXIT(&rx_refcnt_mutex);
6167     MUTEX_ENTER(&call->lock);
6168
6169     /* Update last send time for this call (for keep-alive
6170      * processing), and for the connection (so that we can discover
6171      * idle connections) */
6172     if ((p->header.type != RX_PACKET_TYPE_ACK) ||
6173         (((struct rx_ackPacket *)rx_DataOf(p))->reason == RX_ACK_PING) ||
6174         (p->length <= (rx_AckDataSize(call->rwind) + 4 * sizeof(afs_int32))))
6175     {
6176         conn->lastSendTime = call->lastSendTime = clock_Sec();
6177         /* Don't count keepalive ping/acks here, so idleness can be tracked. */
6178         if ((p->header.type != RX_PACKET_TYPE_ACK) ||
6179             ((((struct rx_ackPacket *)rx_DataOf(p))->reason != RX_ACK_PING) &&
6180              (((struct rx_ackPacket *)rx_DataOf(p))->reason !=
6181               RX_ACK_PING_RESPONSE)))
6182             call->lastSendData = call->lastSendTime;
6183     }
6184 }
6185
6186 /* Check if a call needs to be destroyed.  Called by keep-alive code to ensure
6187  * that things are fine.  Also called periodically to guarantee that nothing
6188  * falls through the cracks (e.g. (error + dally) connections have keepalive
6189  * turned off.  Returns 0 if conn is well, -1 otherwise.  If otherwise, call
6190  *  may be freed!
6191  * haveCTLock Set if calling from rxi_ReapConnections
6192  */
6193 #ifdef RX_ENABLE_LOCKS
6194 int
6195 rxi_CheckCall(struct rx_call *call, int haveCTLock)
6196 #else /* RX_ENABLE_LOCKS */
6197 int
6198 rxi_CheckCall(struct rx_call *call)
6199 #endif                          /* RX_ENABLE_LOCKS */
6200 {
6201     struct rx_connection *conn = call->conn;
6202     afs_uint32 now;
6203     afs_uint32 deadTime, idleDeadTime = 0, hardDeadTime = 0;
6204     afs_uint32 fudgeFactor;
6205     int cerror = 0;
6206     int newmtu = 0;
6207
6208 #ifdef AFS_GLOBAL_RXLOCK_KERNEL
6209     if (call->flags & RX_CALL_TQ_BUSY) {
6210         /* Call is active and will be reset by rxi_Start if it's
6211          * in an error state.
6212          */
6213         return 0;
6214     }
6215 #endif
6216     /* RTT + 8*MDEV, rounded up to the next second. */
6217     fudgeFactor = (((afs_uint32) call->rtt >> 3) +
6218                    ((afs_uint32) call->rtt_dev << 1) + 1023) >> 10;
6219
6220     deadTime = conn->secondsUntilDead + fudgeFactor;
6221     now = clock_Sec();
6222     /* These are computed to the second (+- 1 second).  But that's
6223      * good enough for these values, which should be a significant
6224      * number of seconds. */
6225     if (now > (call->lastReceiveTime + deadTime)) {
6226         if (call->state == RX_STATE_ACTIVE) {
6227 #ifdef ADAPT_PMTU
6228 #if defined(KERNEL) && defined(AFS_SUN5_ENV)
6229             ire_t *ire;
6230 #if defined(AFS_SUN510_ENV) && defined(GLOBAL_NETSTACKID)
6231             netstack_t *ns =  netstack_find_by_stackid(GLOBAL_NETSTACKID);
6232             ip_stack_t *ipst = ns->netstack_ip;
6233 #endif
6234             ire = ire_cache_lookup(conn->peer->host
6235 #if defined(AFS_SUN510_ENV) && defined(ALL_ZONES)
6236                                    , ALL_ZONES
6237 #if defined(AFS_SUN510_ENV) && (defined(ICL_3_ARG) || defined(GLOBAL_NETSTACKID))
6238                                    , NULL
6239 #if defined(AFS_SUN510_ENV) && defined(GLOBAL_NETSTACKID)
6240                                    , ipst
6241 #endif
6242 #endif
6243 #endif
6244                 );
6245
6246             if (ire && ire->ire_max_frag > 0)
6247                 rxi_SetPeerMtu(NULL, conn->peer->host, 0,
6248                                ire->ire_max_frag);
6249 #if defined(GLOBAL_NETSTACKID)
6250             netstack_rele(ns);
6251 #endif
6252 #endif
6253 #endif /* ADAPT_PMTU */
6254             cerror = RX_CALL_DEAD;
6255             goto mtuout;
6256         } else {
6257 #ifdef RX_ENABLE_LOCKS
6258             /* Cancel pending events */
6259             rxevent_Cancel(call->delayedAckEvent, call,
6260                            RX_CALL_REFCOUNT_DELAY);
6261             rxi_rto_cancel(call);
6262             rxevent_Cancel(call->keepAliveEvent, call,
6263                            RX_CALL_REFCOUNT_ALIVE);
6264             if (call->growMTUEvent)
6265                 rxevent_Cancel(call->growMTUEvent, call,
6266                                RX_CALL_REFCOUNT_ALIVE);
6267             MUTEX_ENTER(&rx_refcnt_mutex);
6268             if (call->refCount == 0) {
6269                 rxi_FreeCall(call, haveCTLock);
6270                 MUTEX_EXIT(&rx_refcnt_mutex);
6271                 return -2;
6272             }
6273             MUTEX_EXIT(&rx_refcnt_mutex);
6274             return -1;
6275 #else /* RX_ENABLE_LOCKS */
6276             rxi_FreeCall(call, 0);
6277             return -2;
6278 #endif /* RX_ENABLE_LOCKS */
6279         }
6280         /* Non-active calls are destroyed if they are not responding
6281          * to pings; active calls are simply flagged in error, so the
6282          * attached process can die reasonably gracefully. */
6283     }
6284
6285     if (conn->idleDeadTime) {
6286         idleDeadTime = conn->idleDeadTime + fudgeFactor;
6287     }
6288
6289     /* see if we have a non-activity timeout */
6290     if (call->startWait && idleDeadTime
6291         && ((call->startWait + idleDeadTime) < now) &&
6292         (call->flags & RX_CALL_READER_WAIT)) {
6293         if (call->state == RX_STATE_ACTIVE) {
6294             cerror = RX_CALL_TIMEOUT;
6295             goto mtuout;
6296         }
6297     }
6298     if (call->lastSendData && idleDeadTime && (conn->idleDeadErr != 0)
6299         && ((call->lastSendData + idleDeadTime) < now)) {
6300         if (call->state == RX_STATE_ACTIVE) {
6301             cerror = conn->idleDeadErr;
6302             goto mtuout;
6303         }
6304     }
6305
6306     if (conn->hardDeadTime) {
6307         hardDeadTime = conn->hardDeadTime + fudgeFactor;
6308     }
6309
6310     /* see if we have a hard timeout */
6311     if (hardDeadTime
6312         && (now > (hardDeadTime + call->startTime.sec))) {
6313         if (call->state == RX_STATE_ACTIVE)
6314             rxi_CallError(call, RX_CALL_TIMEOUT);
6315         return -1;
6316     }
6317     return 0;
6318 mtuout:
6319     if (conn->msgsizeRetryErr && cerror != RX_CALL_TIMEOUT
6320         && call->lastReceiveTime) {
6321         int oldMTU = conn->peer->ifMTU;
6322
6323         /* if we thought we could send more, perhaps things got worse */
6324         if (conn->peer->maxPacketSize > conn->lastPacketSize)
6325             /* maxpacketsize will be cleared in rxi_SetPeerMtu */
6326             newmtu = MAX(conn->peer->maxPacketSize-RX_IPUDP_SIZE,
6327                          conn->lastPacketSize-(128+RX_IPUDP_SIZE));
6328         else
6329             newmtu = conn->lastPacketSize-(128+RX_IPUDP_SIZE);
6330
6331         /* minimum capped in SetPeerMtu */
6332         rxi_SetPeerMtu(conn->peer, 0, 0, newmtu);
6333
6334         /* clean up */
6335         conn->lastPacketSize = 0;
6336
6337         /* needed so ResetCall doesn't clobber us. */
6338         call->MTU = conn->peer->ifMTU;
6339
6340         /* if we never succeeded, let the error pass out as-is */
6341         if (conn->peer->maxPacketSize && oldMTU != conn->peer->ifMTU)
6342             cerror = conn->msgsizeRetryErr;
6343
6344     }
6345     rxi_CallError(call, cerror);
6346     return -1;
6347 }
6348
6349 void
6350 rxi_NatKeepAliveEvent(struct rxevent *event, void *arg1, void *dummy)
6351 {
6352     struct rx_connection *conn = arg1;
6353     struct rx_header theader;
6354     char tbuffer[1 + sizeof(struct rx_header)];
6355     struct sockaddr_in taddr;
6356     char *tp;
6357     char a[1] = { 0 };
6358     struct iovec tmpiov[2];
6359     osi_socket socket =
6360         (conn->type ==
6361          RX_CLIENT_CONNECTION ? rx_socket : conn->service->socket);
6362
6363
6364     tp = &tbuffer[sizeof(struct rx_header)];
6365     taddr.sin_family = AF_INET;
6366     taddr.sin_port = rx_PortOf(rx_PeerOf(conn));
6367     taddr.sin_addr.s_addr = rx_HostOf(rx_PeerOf(conn));
6368 #ifdef STRUCT_SOCKADDR_HAS_SA_LEN
6369     taddr.sin_len = sizeof(struct sockaddr_in);
6370 #endif
6371     memset(&theader, 0, sizeof(theader));
6372     theader.epoch = htonl(999);
6373     theader.cid = 0;
6374     theader.callNumber = 0;
6375     theader.seq = 0;
6376     theader.serial = 0;
6377     theader.type = RX_PACKET_TYPE_VERSION;
6378     theader.flags = RX_LAST_PACKET;
6379     theader.serviceId = 0;
6380
6381     memcpy(tbuffer, &theader, sizeof(theader));
6382     memcpy(tp, &a, sizeof(a));
6383     tmpiov[0].iov_base = tbuffer;
6384     tmpiov[0].iov_len = 1 + sizeof(struct rx_header);
6385
6386     osi_NetSend(socket, &taddr, tmpiov, 1, 1 + sizeof(struct rx_header), 1);
6387
6388     MUTEX_ENTER(&conn->conn_data_lock);
6389     MUTEX_ENTER(&rx_refcnt_mutex);
6390     /* Only reschedule ourselves if the connection would not be destroyed */
6391     if (conn->refCount <= 1) {
6392         conn->natKeepAliveEvent = NULL;
6393         MUTEX_EXIT(&rx_refcnt_mutex);
6394         MUTEX_EXIT(&conn->conn_data_lock);
6395         rx_DestroyConnection(conn); /* drop the reference for this */
6396     } else {
6397         conn->refCount--; /* drop the reference for this */
6398         MUTEX_EXIT(&rx_refcnt_mutex);
6399         conn->natKeepAliveEvent = NULL;
6400         rxi_ScheduleNatKeepAliveEvent(conn);
6401         MUTEX_EXIT(&conn->conn_data_lock);
6402     }
6403 }
6404
6405 void
6406 rxi_ScheduleNatKeepAliveEvent(struct rx_connection *conn)
6407 {
6408     if (!conn->natKeepAliveEvent && conn->secondsUntilNatPing) {
6409         struct clock when, now;
6410         clock_GetTime(&now);
6411         when = now;
6412         when.sec += conn->secondsUntilNatPing;
6413         MUTEX_ENTER(&rx_refcnt_mutex);
6414         conn->refCount++; /* hold a reference for this */
6415         MUTEX_EXIT(&rx_refcnt_mutex);
6416         conn->natKeepAliveEvent =
6417             rxevent_PostNow(&when, &now, rxi_NatKeepAliveEvent, conn, 0);
6418     }
6419 }
6420
6421 void
6422 rx_SetConnSecondsUntilNatPing(struct rx_connection *conn, afs_int32 seconds)
6423 {
6424     MUTEX_ENTER(&conn->conn_data_lock);
6425     conn->secondsUntilNatPing = seconds;
6426     if (seconds != 0) {
6427         if (!(conn->flags & RX_CONN_ATTACHWAIT))
6428             rxi_ScheduleNatKeepAliveEvent(conn);
6429         else
6430             conn->flags |= RX_CONN_NAT_PING;
6431     }
6432     MUTEX_EXIT(&conn->conn_data_lock);
6433 }
6434
6435 void
6436 rxi_NatKeepAliveOn(struct rx_connection *conn)
6437 {
6438     MUTEX_ENTER(&conn->conn_data_lock);
6439     /* if it's already attached */
6440     if (!(conn->flags & RX_CONN_ATTACHWAIT))
6441         rxi_ScheduleNatKeepAliveEvent(conn);
6442     else
6443         conn->flags |= RX_CONN_NAT_PING;
6444     MUTEX_EXIT(&conn->conn_data_lock);
6445 }
6446
6447 /* When a call is in progress, this routine is called occasionally to
6448  * make sure that some traffic has arrived (or been sent to) the peer.
6449  * If nothing has arrived in a reasonable amount of time, the call is
6450  * declared dead; if nothing has been sent for a while, we send a
6451  * keep-alive packet (if we're actually trying to keep the call alive)
6452  */
6453 void
6454 rxi_KeepAliveEvent(struct rxevent *event, void *arg1, void *dummy)
6455 {
6456     struct rx_call *call = arg1;
6457     struct rx_connection *conn;
6458     afs_uint32 now;
6459
6460     MUTEX_ENTER(&rx_refcnt_mutex);
6461     CALL_RELE(call, RX_CALL_REFCOUNT_ALIVE);
6462     MUTEX_EXIT(&rx_refcnt_mutex);
6463     MUTEX_ENTER(&call->lock);
6464     if (event == call->keepAliveEvent)
6465         call->keepAliveEvent = NULL;
6466     now = clock_Sec();
6467
6468 #ifdef RX_ENABLE_LOCKS
6469     if (rxi_CheckCall(call, 0)) {
6470         MUTEX_EXIT(&call->lock);
6471         return;
6472     }
6473 #else /* RX_ENABLE_LOCKS */
6474     if (rxi_CheckCall(call))
6475         return;
6476 #endif /* RX_ENABLE_LOCKS */
6477
6478     /* Don't try to keep alive dallying calls */
6479     if (call->state == RX_STATE_DALLY) {
6480         MUTEX_EXIT(&call->lock);
6481         return;
6482     }
6483
6484     conn = call->conn;
6485     if ((now - call->lastSendTime) > conn->secondsUntilPing) {
6486         /* Don't try to send keepalives if there is unacknowledged data */
6487         /* the rexmit code should be good enough, this little hack
6488          * doesn't quite work XXX */
6489         (void)rxi_SendAck(call, NULL, 0, RX_ACK_PING, 0);
6490     }
6491     rxi_ScheduleKeepAliveEvent(call);
6492     MUTEX_EXIT(&call->lock);
6493 }
6494
6495 /* Does what's on the nameplate. */
6496 void
6497 rxi_GrowMTUEvent(struct rxevent *event, void *arg1, void *dummy)
6498 {
6499     struct rx_call *call = arg1;
6500     struct rx_connection *conn;
6501
6502     MUTEX_ENTER(&rx_refcnt_mutex);
6503     CALL_RELE(call, RX_CALL_REFCOUNT_ALIVE);
6504     MUTEX_EXIT(&rx_refcnt_mutex);
6505     MUTEX_ENTER(&call->lock);
6506
6507     if (event == call->growMTUEvent)
6508         call->growMTUEvent = NULL;
6509
6510 #ifdef RX_ENABLE_LOCKS
6511     if (rxi_CheckCall(call, 0)) {
6512         MUTEX_EXIT(&call->lock);
6513         return;
6514     }
6515 #else /* RX_ENABLE_LOCKS */
6516     if (rxi_CheckCall(call))
6517         return;
6518 #endif /* RX_ENABLE_LOCKS */
6519
6520     /* Don't bother with dallying calls */
6521     if (call->state == RX_STATE_DALLY) {
6522         MUTEX_EXIT(&call->lock);
6523         return;
6524     }
6525
6526     conn = call->conn;
6527
6528     /*
6529      * keep being scheduled, just don't do anything if we're at peak,
6530      * or we're not set up to be properly handled (idle timeout required)
6531      */
6532     if ((conn->peer->maxPacketSize != 0) &&
6533         (conn->peer->natMTU < RX_MAX_PACKET_SIZE) &&
6534         (conn->idleDeadErr))
6535         (void)rxi_SendAck(call, NULL, 0, RX_ACK_MTU, 0);
6536     rxi_ScheduleGrowMTUEvent(call, 0);
6537     MUTEX_EXIT(&call->lock);
6538 }
6539
6540 void
6541 rxi_ScheduleKeepAliveEvent(struct rx_call *call)
6542 {
6543     if (!call->keepAliveEvent) {
6544         struct clock when, now;
6545         clock_GetTime(&now);
6546         when = now;
6547         when.sec += call->conn->secondsUntilPing;
6548         MUTEX_ENTER(&rx_refcnt_mutex);
6549         CALL_HOLD(call, RX_CALL_REFCOUNT_ALIVE);
6550         MUTEX_EXIT(&rx_refcnt_mutex);
6551         call->keepAliveEvent =
6552             rxevent_PostNow(&when, &now, rxi_KeepAliveEvent, call, 0);
6553     }
6554 }
6555
6556 void
6557 rxi_ScheduleGrowMTUEvent(struct rx_call *call, int secs)
6558 {
6559     if (!call->growMTUEvent) {
6560         struct clock when, now;
6561
6562         clock_GetTime(&now);
6563         when = now;
6564         if (!secs) {
6565             if (call->conn->secondsUntilPing)
6566                 secs = (6*call->conn->secondsUntilPing)-1;
6567
6568             if (call->conn->secondsUntilDead)
6569                 secs = MIN(secs, (call->conn->secondsUntilDead-1));
6570         }
6571
6572         when.sec += secs;
6573         MUTEX_ENTER(&rx_refcnt_mutex);
6574         CALL_HOLD(call, RX_CALL_REFCOUNT_ALIVE);
6575         MUTEX_EXIT(&rx_refcnt_mutex);
6576         call->growMTUEvent =
6577             rxevent_PostNow(&when, &now, rxi_GrowMTUEvent, call, 0);
6578     }
6579 }
6580
6581 /* N.B. rxi_KeepAliveOff:  is defined earlier as a macro */
6582 void
6583 rxi_KeepAliveOn(struct rx_call *call)
6584 {
6585     /* Pretend last packet received was received now--i.e. if another
6586      * packet isn't received within the keep alive time, then the call
6587      * will die; Initialize last send time to the current time--even
6588      * if a packet hasn't been sent yet.  This will guarantee that a
6589      * keep-alive is sent within the ping time */
6590     call->lastReceiveTime = call->lastSendTime = clock_Sec();
6591     rxi_ScheduleKeepAliveEvent(call);
6592 }
6593
6594 void
6595 rxi_GrowMTUOn(struct rx_call *call)
6596 {
6597     struct rx_connection *conn = call->conn;
6598     MUTEX_ENTER(&conn->conn_data_lock);
6599     conn->lastPingSizeSer = conn->lastPingSize = 0;
6600     MUTEX_EXIT(&conn->conn_data_lock);
6601     rxi_ScheduleGrowMTUEvent(call, 1);
6602 }
6603
6604 /* This routine is called to send connection abort messages
6605  * that have been delayed to throttle looping clients. */
6606 void
6607 rxi_SendDelayedConnAbort(struct rxevent *event,
6608                          void *arg1, void *unused)
6609 {
6610     struct rx_connection *conn = arg1;
6611
6612     afs_int32 error;
6613     struct rx_packet *packet;
6614
6615     MUTEX_ENTER(&conn->conn_data_lock);
6616     conn->delayedAbortEvent = NULL;
6617     error = htonl(conn->error);
6618     conn->abortCount++;
6619     MUTEX_EXIT(&conn->conn_data_lock);
6620     packet = rxi_AllocPacket(RX_PACKET_CLASS_SPECIAL);
6621     if (packet) {
6622         packet =
6623             rxi_SendSpecial((struct rx_call *)0, conn, packet,
6624                             RX_PACKET_TYPE_ABORT, (char *)&error,
6625                             sizeof(error), 0);
6626         rxi_FreePacket(packet);
6627     }
6628 }
6629
6630 /* This routine is called to send call abort messages
6631  * that have been delayed to throttle looping clients. */
6632 void
6633 rxi_SendDelayedCallAbort(struct rxevent *event,
6634                          void *arg1, void *dummy)
6635 {
6636     struct rx_call *call = arg1;
6637
6638     afs_int32 error;
6639     struct rx_packet *packet;
6640
6641     MUTEX_ENTER(&call->lock);
6642     call->delayedAbortEvent = NULL;
6643     error = htonl(call->error);
6644     call->abortCount++;
6645     packet = rxi_AllocPacket(RX_PACKET_CLASS_SPECIAL);
6646     if (packet) {
6647         packet =
6648             rxi_SendSpecial(call, call->conn, packet, RX_PACKET_TYPE_ABORT,
6649                             (char *)&error, sizeof(error), 0);
6650         rxi_FreePacket(packet);
6651     }
6652     MUTEX_EXIT(&call->lock);
6653     MUTEX_ENTER(&rx_refcnt_mutex);
6654     CALL_RELE(call, RX_CALL_REFCOUNT_ABORT);
6655     MUTEX_EXIT(&rx_refcnt_mutex);
6656 }
6657
6658 /* This routine is called periodically (every RX_AUTH_REQUEST_TIMEOUT
6659  * seconds) to ask the client to authenticate itself.  The routine
6660  * issues a challenge to the client, which is obtained from the
6661  * security object associated with the connection */
6662 void
6663 rxi_ChallengeEvent(struct rxevent *event,
6664                    void *arg0, void *arg1, int tries)
6665 {
6666     struct rx_connection *conn = arg0;
6667
6668     conn->challengeEvent = NULL;
6669     if (RXS_CheckAuthentication(conn->securityObject, conn) != 0) {
6670         struct rx_packet *packet;
6671         struct clock when, now;
6672
6673         if (tries <= 0) {
6674             /* We've failed to authenticate for too long.
6675              * Reset any calls waiting for authentication;
6676              * they are all in RX_STATE_PRECALL.
6677              */
6678             int i;
6679
6680             MUTEX_ENTER(&conn->conn_call_lock);
6681             for (i = 0; i < RX_MAXCALLS; i++) {
6682                 struct rx_call *call = conn->call[i];
6683                 if (call) {
6684                     MUTEX_ENTER(&call->lock);
6685                     if (call->state == RX_STATE_PRECALL) {
6686                         rxi_CallError(call, RX_CALL_DEAD);
6687                         rxi_SendCallAbort(call, NULL, 0, 0);
6688                     }
6689                     MUTEX_EXIT(&call->lock);
6690                 }
6691             }
6692             MUTEX_EXIT(&conn->conn_call_lock);
6693             return;
6694         }
6695
6696         packet = rxi_AllocPacket(RX_PACKET_CLASS_SPECIAL);
6697         if (packet) {
6698             /* If there's no packet available, do this later. */
6699             RXS_GetChallenge(conn->securityObject, conn, packet);
6700             rxi_SendSpecial((struct rx_call *)0, conn, packet,
6701                             RX_PACKET_TYPE_CHALLENGE, NULL, -1, 0);
6702             rxi_FreePacket(packet);
6703         }
6704         clock_GetTime(&now);
6705         when = now;
6706         when.sec += RX_CHALLENGE_TIMEOUT;
6707         conn->challengeEvent =
6708             rxevent_PostNow2(&when, &now, rxi_ChallengeEvent, conn, 0,
6709                          (tries - 1));
6710     }
6711 }
6712
6713 /* Call this routine to start requesting the client to authenticate
6714  * itself.  This will continue until authentication is established,
6715  * the call times out, or an invalid response is returned.  The
6716  * security object associated with the connection is asked to create
6717  * the challenge at this time.  N.B.  rxi_ChallengeOff is a macro,
6718  * defined earlier. */
6719 void
6720 rxi_ChallengeOn(struct rx_connection *conn)
6721 {
6722     if (!conn->challengeEvent) {
6723         RXS_CreateChallenge(conn->securityObject, conn);
6724         rxi_ChallengeEvent(NULL, conn, 0, RX_CHALLENGE_MAXTRIES);
6725     };
6726 }
6727
6728
6729 /* rxi_ComputeRoundTripTime is called with peer locked. */
6730 /* peer may be null */
6731 static void
6732 rxi_ComputeRoundTripTime(struct rx_packet *p,
6733                          struct rx_ackPacket *ack,
6734                          struct rx_call *call,
6735                          struct rx_peer *peer,
6736                          struct clock *now)
6737 {
6738     struct clock thisRtt, *sentp;
6739     int rtt_timeout;
6740     int serial;
6741
6742     /* If the ACK is delayed, then do nothing */
6743     if (ack->reason == RX_ACK_DELAY)
6744         return;
6745
6746     /* On the wire, jumbograms are a single UDP packet. We shouldn't count
6747      * their RTT multiple times, so only include the RTT of the last packet
6748      * in a jumbogram */
6749     if (p->flags & RX_JUMBO_PACKET)
6750         return;
6751
6752     /* Use the serial number to determine which transmission the ACK is for,
6753      * and set the sent time to match this. If we have no serial number, then
6754      * only use the ACK for RTT calculations if the packet has not been
6755      * retransmitted
6756      */
6757
6758     serial = ntohl(ack->serial);
6759     if (serial) {
6760         if (serial == p->header.serial) {
6761             sentp = &p->timeSent;
6762         } else if (serial == p->firstSerial) {
6763             sentp = &p->firstSent;
6764         } else if (clock_Eq(&p->timeSent, &p->firstSent)) {
6765             sentp = &p->firstSent;
6766         } else
6767             return;
6768     } else {
6769         if (clock_Eq(&p->timeSent, &p->firstSent)) {
6770             sentp = &p->firstSent;
6771         } else
6772             return;
6773     }
6774
6775     thisRtt = *now;
6776
6777     if (clock_Lt(&thisRtt, sentp))
6778         return;                 /* somebody set the clock back, don't count this time. */
6779
6780     clock_Sub(&thisRtt, sentp);
6781     dpf(("rxi_ComputeRoundTripTime(call=%d packet=%"AFS_PTR_FMT" rttp=%d.%06d sec)\n",
6782           p->header.callNumber, p, thisRtt.sec, thisRtt.usec));
6783
6784     if (clock_IsZero(&thisRtt)) {
6785         /*
6786          * The actual round trip time is shorter than the
6787          * clock_GetTime resolution.  It is most likely 1ms or 100ns.
6788          * Since we can't tell which at the moment we will assume 1ms.
6789          */
6790         thisRtt.usec = 1000;
6791     }
6792
6793     if (rx_stats_active) {
6794         MUTEX_ENTER(&rx_stats_mutex);
6795         if (clock_Lt(&thisRtt, &rx_stats.minRtt))
6796             rx_stats.minRtt = thisRtt;
6797         if (clock_Gt(&thisRtt, &rx_stats.maxRtt)) {
6798             if (thisRtt.sec > 60) {
6799                 MUTEX_EXIT(&rx_stats_mutex);
6800                 return;         /* somebody set the clock ahead */
6801             }
6802             rx_stats.maxRtt = thisRtt;
6803         }
6804         clock_Add(&rx_stats.totalRtt, &thisRtt);
6805         rx_atomic_inc(&rx_stats.nRttSamples);
6806         MUTEX_EXIT(&rx_stats_mutex);
6807     }
6808
6809     /* better rtt calculation courtesy of UMich crew (dave,larry,peter,?) */
6810
6811     /* Apply VanJacobson round-trip estimations */
6812     if (call->rtt) {
6813         int delta;
6814
6815         /*
6816          * srtt (call->rtt) is in units of one-eighth-milliseconds.
6817          * srtt is stored as fixed point with 3 bits after the binary
6818          * point (i.e., scaled by 8). The following magic is
6819          * equivalent to the smoothing algorithm in rfc793 with an
6820          * alpha of .875 (srtt' = rtt/8 + srtt*7/8 in fixed point).
6821          * srtt'*8 = rtt + srtt*7
6822          * srtt'*8 = srtt*8 + rtt - srtt
6823          * srtt' = srtt + rtt/8 - srtt/8
6824          * srtt' = srtt + (rtt - srtt)/8
6825          */
6826
6827         delta = _8THMSEC(&thisRtt) - call->rtt;
6828         call->rtt += (delta >> 3);
6829
6830         /*
6831          * We accumulate a smoothed rtt variance (actually, a smoothed
6832          * mean difference), then set the retransmit timer to smoothed
6833          * rtt + 4 times the smoothed variance (was 2x in van's original
6834          * paper, but 4x works better for me, and apparently for him as
6835          * well).
6836          * rttvar is stored as
6837          * fixed point with 2 bits after the binary point (scaled by
6838          * 4).  The following is equivalent to rfc793 smoothing with
6839          * an alpha of .75 (rttvar' = rttvar*3/4 + |delta| / 4).
6840          *   rttvar'*4 = rttvar*3 + |delta|
6841          *   rttvar'*4 = rttvar*4 + |delta| - rttvar
6842          *   rttvar' = rttvar + |delta|/4 - rttvar/4
6843          *   rttvar' = rttvar + (|delta| - rttvar)/4
6844          * This replaces rfc793's wired-in beta.
6845          * dev*4 = dev*4 + (|actual - expected| - dev)
6846          */
6847
6848         if (delta < 0)
6849             delta = -delta;
6850
6851         delta -= (call->rtt_dev << 1);
6852         call->rtt_dev += (delta >> 3);
6853     } else {
6854         /* I don't have a stored RTT so I start with this value.  Since I'm
6855          * probably just starting a call, and will be pushing more data down
6856          * this, I expect congestion to increase rapidly.  So I fudge a
6857          * little, and I set deviance to half the rtt.  In practice,
6858          * deviance tends to approach something a little less than
6859          * half the smoothed rtt. */
6860         call->rtt = _8THMSEC(&thisRtt) + 8;
6861         call->rtt_dev = call->rtt >> 2; /* rtt/2: they're scaled differently */
6862     }
6863     /* the smoothed RTT time is RTT + 4*MDEV
6864      *
6865      * We allow a user specified minimum to be set for this, to allow clamping
6866      * at a minimum value in the same way as TCP. In addition, we have to allow
6867      * for the possibility that this packet is answered by a delayed ACK, so we
6868      * add on a fixed 200ms to account for that timer expiring.
6869      */
6870
6871     rtt_timeout = MAX(((call->rtt >> 3) + call->rtt_dev),
6872                       rx_minPeerTimeout) + 200;
6873     clock_Zero(&call->rto);
6874     clock_Addmsec(&call->rto, rtt_timeout);
6875
6876     /* Update the peer, so any new calls start with our values */
6877     peer->rtt_dev = call->rtt_dev;
6878     peer->rtt = call->rtt;
6879
6880     dpf(("rxi_ComputeRoundTripTime(call=%d packet=%"AFS_PTR_FMT" rtt=%d ms, srtt=%d ms, rtt_dev=%d ms, timeout=%d.%06d sec)\n",
6881           p->header.callNumber, p, MSEC(&thisRtt), call->rtt >> 3, call->rtt_dev >> 2, (call->rto.sec), (call->rto.usec)));
6882 }
6883
6884
6885 /* Find all server connections that have not been active for a long time, and
6886  * toss them */
6887 void
6888 rxi_ReapConnections(struct rxevent *unused, void *unused1, void *unused2)
6889 {
6890     struct clock now, when;
6891     clock_GetTime(&now);
6892
6893     /* Find server connection structures that haven't been used for
6894      * greater than rx_idleConnectionTime */
6895     {
6896         struct rx_connection **conn_ptr, **conn_end;
6897         int i, havecalls = 0;
6898         MUTEX_ENTER(&rx_connHashTable_lock);
6899         for (conn_ptr = &rx_connHashTable[0], conn_end =
6900              &rx_connHashTable[rx_hashTableSize]; conn_ptr < conn_end;
6901              conn_ptr++) {
6902             struct rx_connection *conn, *next;
6903             struct rx_call *call;
6904             int result;
6905
6906           rereap:
6907             for (conn = *conn_ptr; conn; conn = next) {
6908                 /* XXX -- Shouldn't the connection be locked? */
6909                 next = conn->next;
6910                 havecalls = 0;
6911                 for (i = 0; i < RX_MAXCALLS; i++) {
6912                     call = conn->call[i];
6913                     if (call) {
6914                         int code;
6915                         havecalls = 1;
6916                         code = MUTEX_TRYENTER(&call->lock);
6917                         if (!code)
6918                             continue;
6919 #ifdef RX_ENABLE_LOCKS
6920                         result = rxi_CheckCall(call, 1);
6921 #else /* RX_ENABLE_LOCKS */
6922                         result = rxi_CheckCall(call);
6923 #endif /* RX_ENABLE_LOCKS */
6924                         MUTEX_EXIT(&call->lock);
6925                         if (result == -2) {
6926                             /* If CheckCall freed the call, it might
6927                              * have destroyed  the connection as well,
6928                              * which screws up the linked lists.
6929                              */
6930                             goto rereap;
6931                         }
6932                     }
6933                 }
6934                 if (conn->type == RX_SERVER_CONNECTION) {
6935                     /* This only actually destroys the connection if
6936                      * there are no outstanding calls */
6937                     MUTEX_ENTER(&conn->conn_data_lock);
6938                     MUTEX_ENTER(&rx_refcnt_mutex);
6939                     if (!havecalls && !conn->refCount
6940                         && ((conn->lastSendTime + rx_idleConnectionTime) <
6941                             now.sec)) {
6942                         conn->refCount++;       /* it will be decr in rx_DestroyConn */
6943                         MUTEX_EXIT(&rx_refcnt_mutex);
6944                         MUTEX_EXIT(&conn->conn_data_lock);
6945 #ifdef RX_ENABLE_LOCKS
6946                         rxi_DestroyConnectionNoLock(conn);
6947 #else /* RX_ENABLE_LOCKS */
6948                         rxi_DestroyConnection(conn);
6949 #endif /* RX_ENABLE_LOCKS */
6950                     }
6951 #ifdef RX_ENABLE_LOCKS
6952                     else {
6953                         MUTEX_EXIT(&rx_refcnt_mutex);
6954                         MUTEX_EXIT(&conn->conn_data_lock);
6955                     }
6956 #endif /* RX_ENABLE_LOCKS */
6957                 }
6958             }
6959         }
6960 #ifdef RX_ENABLE_LOCKS
6961         while (rx_connCleanup_list) {
6962             struct rx_connection *conn;
6963             conn = rx_connCleanup_list;
6964             rx_connCleanup_list = rx_connCleanup_list->next;
6965             MUTEX_EXIT(&rx_connHashTable_lock);
6966             rxi_CleanupConnection(conn);
6967             MUTEX_ENTER(&rx_connHashTable_lock);
6968         }
6969         MUTEX_EXIT(&rx_connHashTable_lock);
6970 #endif /* RX_ENABLE_LOCKS */
6971     }
6972
6973     /* Find any peer structures that haven't been used (haven't had an
6974      * associated connection) for greater than rx_idlePeerTime */
6975     {
6976         struct rx_peer **peer_ptr, **peer_end;
6977         int code;
6978
6979         /*
6980          * Why do we need to hold the rx_peerHashTable_lock across
6981          * the incrementing of peer_ptr since the rx_peerHashTable
6982          * array is not changing?  We don't.
6983          *
6984          * By dropping the lock periodically we can permit other
6985          * activities to be performed while a rxi_ReapConnections
6986          * call is in progress.  The goal of reap connections
6987          * is to clean up quickly without causing large amounts
6988          * of contention.  Therefore, it is important that global
6989          * mutexes not be held for extended periods of time.
6990          */
6991         for (peer_ptr = &rx_peerHashTable[0], peer_end =
6992              &rx_peerHashTable[rx_hashTableSize]; peer_ptr < peer_end;
6993              peer_ptr++) {
6994             struct rx_peer *peer, *next, *prev;
6995
6996             MUTEX_ENTER(&rx_peerHashTable_lock);
6997             for (prev = peer = *peer_ptr; peer; peer = next) {
6998                 next = peer->next;
6999                 code = MUTEX_TRYENTER(&peer->peer_lock);
7000                 if ((code) && (peer->refCount == 0)
7001                     && ((peer->idleWhen + rx_idlePeerTime) < now.sec)) {
7002                     rx_interface_stat_p rpc_stat, nrpc_stat;
7003                     size_t space;
7004
7005                     /*
7006                      * now know that this peer object is one to be
7007                      * removed from the hash table.  Once it is removed
7008                      * it can't be referenced by other threads.
7009                      * Lets remove it first and decrement the struct
7010                      * nPeerStructs count.
7011                      */
7012                     if (peer == *peer_ptr) {
7013                         *peer_ptr = next;
7014                         prev = next;
7015                     } else
7016                         prev->next = next;
7017
7018                     if (rx_stats_active)
7019                         rx_atomic_dec(&rx_stats.nPeerStructs);
7020
7021                     /*
7022                      * Now if we hold references on 'prev' and 'next'
7023                      * we can safely drop the rx_peerHashTable_lock
7024                      * while we destroy this 'peer' object.
7025                      */
7026                     if (next)
7027                         next->refCount++;
7028                     if (prev)
7029                         prev->refCount++;
7030                     MUTEX_EXIT(&rx_peerHashTable_lock);
7031
7032                     MUTEX_EXIT(&peer->peer_lock);
7033                     MUTEX_DESTROY(&peer->peer_lock);
7034                     for (queue_Scan
7035                          (&peer->rpcStats, rpc_stat, nrpc_stat,
7036                           rx_interface_stat)) {
7037                         unsigned int num_funcs;
7038                         if (!rpc_stat)
7039                             break;
7040                         queue_Remove(&rpc_stat->queue_header);
7041                         queue_Remove(&rpc_stat->all_peers);
7042                         num_funcs = rpc_stat->stats[0].func_total;
7043                         space =
7044                             sizeof(rx_interface_stat_t) +
7045                             rpc_stat->stats[0].func_total *
7046                             sizeof(rx_function_entry_v1_t);
7047
7048                         rxi_Free(rpc_stat, space);
7049
7050                         MUTEX_ENTER(&rx_rpc_stats);
7051                         rxi_rpc_peer_stat_cnt -= num_funcs;
7052                         MUTEX_EXIT(&rx_rpc_stats);
7053                     }
7054                     rxi_FreePeer(peer);
7055
7056                     /*
7057                      * Regain the rx_peerHashTable_lock and
7058                      * decrement the reference count on 'prev'
7059                      * and 'next'.
7060                      */
7061                     MUTEX_ENTER(&rx_peerHashTable_lock);
7062                     if (next)
7063                         next->refCount--;
7064                     if (prev)
7065                         prev->refCount--;
7066                 } else {
7067                     if (code) {
7068                         MUTEX_EXIT(&peer->peer_lock);
7069                     }
7070                     prev = peer;
7071                 }
7072             }
7073             MUTEX_EXIT(&rx_peerHashTable_lock);
7074         }
7075     }
7076
7077     /* THIS HACK IS A TEMPORARY HACK.  The idea is that the race condition in
7078      * rxi_AllocSendPacket, if it hits, will be handled at the next conn
7079      * GC, just below.  Really, we shouldn't have to keep moving packets from
7080      * one place to another, but instead ought to always know if we can
7081      * afford to hold onto a packet in its particular use.  */
7082     MUTEX_ENTER(&rx_freePktQ_lock);
7083     if (rx_waitingForPackets) {
7084         rx_waitingForPackets = 0;
7085 #ifdef  RX_ENABLE_LOCKS
7086         CV_BROADCAST(&rx_waitingForPackets_cv);
7087 #else
7088         osi_rxWakeup(&rx_waitingForPackets);
7089 #endif
7090     }
7091     MUTEX_EXIT(&rx_freePktQ_lock);
7092
7093     when = now;
7094     when.sec += RX_REAP_TIME;   /* Check every RX_REAP_TIME seconds */
7095     rxevent_Post(&when, rxi_ReapConnections, 0, 0);
7096 }
7097
7098
7099 /* rxs_Release - This isn't strictly necessary but, since the macro name from
7100  * rx.h is sort of strange this is better.  This is called with a security
7101  * object before it is discarded.  Each connection using a security object has
7102  * its own refcount to the object so it won't actually be freed until the last
7103  * connection is destroyed.
7104  *
7105  * This is the only rxs module call.  A hold could also be written but no one
7106  * needs it. */
7107
7108 int
7109 rxs_Release(struct rx_securityClass *aobj)
7110 {
7111     return RXS_Close(aobj);
7112 }
7113
7114 #ifdef ADAPT_WINDOW
7115 #define RXRATE_PKT_OH   (RX_HEADER_SIZE + RX_IPUDP_SIZE)
7116 #define RXRATE_SMALL_PKT    (RXRATE_PKT_OH + sizeof(struct rx_ackPacket))
7117 #define RXRATE_AVG_SMALL_PKT    (RXRATE_PKT_OH + (sizeof(struct rx_ackPacket)/2))
7118 #define RXRATE_LARGE_PKT    (RXRATE_SMALL_PKT + 256)
7119
7120 /* Adjust our estimate of the transmission rate to this peer, given
7121  * that the packet p was just acked. We can adjust peer->timeout and
7122  * call->twind. Pragmatically, this is called
7123  * only with packets of maximal length.
7124  * Called with peer and call locked.
7125  */
7126
7127 static void
7128 rxi_ComputeRate(struct rx_peer *peer, struct rx_call *call,
7129                 struct rx_packet *p, struct rx_packet *ackp, u_char ackReason)
7130 {
7131     afs_int32 xferSize, xferMs;
7132     afs_int32 minTime;
7133     struct clock newTO;
7134
7135     /* Count down packets */
7136     if (peer->rateFlag > 0)
7137         peer->rateFlag--;
7138     /* Do nothing until we're enabled */
7139     if (peer->rateFlag != 0)
7140         return;
7141     if (!call->conn)
7142         return;
7143
7144     /* Count only when the ack seems legitimate */
7145     switch (ackReason) {
7146     case RX_ACK_REQUESTED:
7147         xferSize =
7148             p->length + RX_HEADER_SIZE + call->conn->securityMaxTrailerSize;
7149         xferMs = call->rtt;
7150         break;
7151
7152     case RX_ACK_PING_RESPONSE:
7153         if (p)                  /* want the response to ping-request, not data send */
7154             return;
7155         clock_GetTime(&newTO);
7156         if (clock_Gt(&newTO, &call->pingRequestTime)) {
7157             clock_Sub(&newTO, &call->pingRequestTime);
7158             xferMs = (newTO.sec * 1000) + (newTO.usec / 1000);
7159         } else {
7160             return;
7161         }
7162         xferSize = rx_AckDataSize(rx_maxSendWindow) + RX_HEADER_SIZE;
7163         break;
7164
7165     default:
7166         return;
7167     }
7168
7169     dpf(("CONG peer %lx/%u: sample (%s) size %ld, %ld ms (to %d.%06d, rtt %u, ps %u)\n",
7170           ntohl(peer->host), ntohs(peer->port), (ackReason == RX_ACK_REQUESTED ? "dataack" : "pingack"),
7171           xferSize, xferMs, peer->timeout.sec, peer->timeout.usec, peer->smRtt, peer->ifMTU));
7172
7173     /* Track only packets that are big enough. */
7174     if ((p->length + RX_HEADER_SIZE + call->conn->securityMaxTrailerSize) <
7175         peer->ifMTU)
7176         return;
7177
7178     /* absorb RTT data (in milliseconds) for these big packets */
7179     if (peer->smRtt == 0) {
7180         peer->smRtt = xferMs;
7181     } else {
7182         peer->smRtt = ((peer->smRtt * 15) + xferMs + 4) >> 4;
7183         if (!peer->smRtt)
7184             peer->smRtt = 1;
7185     }
7186
7187     if (peer->countDown) {
7188         peer->countDown--;
7189         return;
7190     }
7191     peer->countDown = 10;       /* recalculate only every so often */
7192
7193     /* In practice, we can measure only the RTT for full packets,
7194      * because of the way Rx acks the data that it receives.  (If it's
7195      * smaller than a full packet, it often gets implicitly acked
7196      * either by the call response (from a server) or by the next call
7197      * (from a client), and either case confuses transmission times
7198      * with processing times.)  Therefore, replace the above
7199      * more-sophisticated processing with a simpler version, where the
7200      * smoothed RTT is kept for full-size packets, and the time to
7201      * transmit a windowful of full-size packets is simply RTT *
7202      * windowSize. Again, we take two steps:
7203      - ensure the timeout is large enough for a single packet's RTT;
7204      - ensure that the window is small enough to fit in the desired timeout.*/
7205
7206     /* First, the timeout check. */
7207     minTime = peer->smRtt;
7208     /* Get a reasonable estimate for a timeout period */
7209     minTime += minTime;
7210     newTO.sec = minTime / 1000;
7211     newTO.usec = (minTime - (newTO.sec * 1000)) * 1000;
7212
7213     /* Increase the timeout period so that we can always do at least
7214      * one packet exchange */
7215     if (clock_Gt(&newTO, &peer->timeout)) {
7216
7217         dpf(("CONG peer %lx/%u: timeout %d.%06d ==> %ld.%06d (rtt %u)\n",
7218               ntohl(peer->host), ntohs(peer->port), peer->timeout.sec, peer->timeout.usec,
7219               newTO.sec, newTO.usec, peer->smRtt));
7220
7221         peer->timeout = newTO;
7222     }
7223
7224     /* Now, get an estimate for the transmit window size. */
7225     minTime = peer->timeout.sec * 1000 + (peer->timeout.usec / 1000);
7226     /* Now, convert to the number of full packets that could fit in a
7227      * reasonable fraction of that interval */
7228     minTime /= (peer->smRtt << 1);
7229     minTime = MAX(minTime, rx_minPeerTimeout);
7230     xferSize = minTime;         /* (make a copy) */
7231
7232     /* Now clamp the size to reasonable bounds. */
7233     if (minTime <= 1)
7234         minTime = 1;
7235     else if (minTime > rx_maxSendWindow)
7236         minTime = rx_maxSendWindow;
7237 /*    if (minTime != peer->maxWindow) {
7238       dpf(("CONG peer %lx/%u: windowsize %lu ==> %lu (to %lu.%06lu, rtt %u)\n",
7239              ntohl(peer->host), ntohs(peer->port), peer->maxWindow, minTime,
7240              peer->timeout.sec, peer->timeout.usec, peer->smRtt));
7241       peer->maxWindow = minTime;
7242         elide... call->twind = minTime;
7243     }
7244 */
7245
7246     /* Cut back on the peer timeout if it had earlier grown unreasonably.
7247      * Discern this by calculating the timeout necessary for rx_Window
7248      * packets. */
7249     if ((xferSize > rx_maxSendWindow) && (peer->timeout.sec >= 3)) {
7250         /* calculate estimate for transmission interval in milliseconds */
7251         minTime = rx_maxSendWindow * peer->smRtt;
7252         if (minTime < 1000) {
7253             dpf(("CONG peer %lx/%u: cut TO %d.%06d by 0.5 (rtt %u)\n",
7254                  ntohl(peer->host), ntohs(peer->port), peer->timeout.sec,
7255                  peer->timeout.usec, peer->smRtt));
7256
7257             newTO.sec = 0;      /* cut back on timeout by half a second */
7258             newTO.usec = 500000;
7259             clock_Sub(&peer->timeout, &newTO);
7260         }
7261     }
7262
7263     return;
7264 }                               /* end of rxi_ComputeRate */
7265 #endif /* ADAPT_WINDOW */
7266
7267
7268 void
7269 rxi_DebugInit(void)
7270 {
7271 #ifdef RXDEBUG
7272 #ifdef AFS_NT40_ENV
7273 #define TRACE_OPTION_RX_DEBUG 16
7274     HKEY parmKey;
7275     DWORD dummyLen;
7276     DWORD TraceOption;
7277     long code;
7278
7279     rxdebug_active = 0;
7280
7281     code = RegOpenKeyEx(HKEY_LOCAL_MACHINE, AFSREG_CLT_SVC_PARAM_SUBKEY,
7282                          0, KEY_QUERY_VALUE, &parmKey);
7283     if (code != ERROR_SUCCESS)
7284         return;
7285
7286     dummyLen = sizeof(TraceOption);
7287     code = RegQueryValueEx(parmKey, "TraceOption", NULL, NULL,
7288                            (BYTE *) &TraceOption, &dummyLen);
7289     if (code == ERROR_SUCCESS) {
7290         rxdebug_active = (TraceOption & TRACE_OPTION_RX_DEBUG) ? 1 : 0;
7291     }
7292     RegCloseKey (parmKey);
7293 #endif /* AFS_NT40_ENV */
7294 #endif
7295 }
7296
7297 void
7298 rx_DebugOnOff(int on)
7299 {
7300 #ifdef RXDEBUG
7301 #ifdef AFS_NT40_ENV
7302     rxdebug_active = on;
7303 #endif
7304 #endif
7305 }
7306
7307 void
7308 rx_StatsOnOff(int on)
7309 {
7310     rx_stats_active = on;
7311 }
7312
7313
7314 /* Don't call this debugging routine directly; use dpf */
7315 void
7316 rxi_DebugPrint(char *format, ...)
7317 {
7318 #ifdef RXDEBUG
7319     va_list ap;
7320 #ifdef AFS_NT40_ENV
7321     char msg[512];
7322     char tformat[256];
7323     size_t len;
7324
7325     va_start(ap, format);
7326
7327     len = _snprintf(tformat, sizeof(tformat), "tid[%d] %s", GetCurrentThreadId(), format);
7328
7329     if (len > 0) {
7330         len = _vsnprintf(msg, sizeof(msg)-2, tformat, ap);
7331         if (len > 0)
7332             OutputDebugString(msg);
7333     }
7334     va_end(ap);
7335 #else
7336     struct clock now;
7337
7338     va_start(ap, format);
7339
7340     clock_GetTime(&now);
7341     fprintf(rx_Log, " %d.%06d:", (unsigned int)now.sec,
7342             (unsigned int)now.usec);
7343     vfprintf(rx_Log, format, ap);
7344     va_end(ap);
7345 #endif
7346 #endif
7347 }
7348
7349 #ifndef KERNEL
7350 /*
7351  * This function is used to process the rx_stats structure that is local
7352  * to a process as well as an rx_stats structure received from a remote
7353  * process (via rxdebug).  Therefore, it needs to do minimal version
7354  * checking.
7355  */
7356 void
7357 rx_PrintTheseStats(FILE * file, struct rx_statistics *s, int size,
7358                    afs_int32 freePackets, char version)
7359 {
7360     int i;
7361
7362     if (size != sizeof(struct rx_statistics)) {
7363         fprintf(file,
7364                 "Unexpected size of stats structure: was %d, expected %" AFS_SIZET_FMT "\n",
7365                 size, sizeof(struct rx_statistics));
7366     }
7367
7368     fprintf(file, "rx stats: free packets %d, allocs %d, ", (int)freePackets,
7369             s->packetRequests);
7370
7371     if (version >= RX_DEBUGI_VERSION_W_NEWPACKETTYPES) {
7372         fprintf(file, "alloc-failures(rcv %u/%u,send %u/%u,ack %u)\n",
7373                 s->receivePktAllocFailures, s->receiveCbufPktAllocFailures,
7374                 s->sendPktAllocFailures, s->sendCbufPktAllocFailures,
7375                 s->specialPktAllocFailures);
7376     } else {
7377         fprintf(file, "alloc-failures(rcv %u,send %u,ack %u)\n",
7378                 s->receivePktAllocFailures, s->sendPktAllocFailures,
7379                 s->specialPktAllocFailures);
7380     }
7381
7382     fprintf(file,
7383             "   greedy %u, " "bogusReads %u (last from host %x), "
7384             "noPackets %u, " "noBuffers %u, " "selects %u, "
7385             "sendSelects %u\n", s->socketGreedy, s->bogusPacketOnRead,
7386             s->bogusHost, s->noPacketOnRead, s->noPacketBuffersOnRead,
7387             s->selects, s->sendSelects);
7388
7389     fprintf(file, "   packets read: ");
7390     for (i = 0; i < RX_N_PACKET_TYPES; i++) {
7391         fprintf(file, "%s %u ", rx_packetTypes[i], s->packetsRead[i]);
7392     }
7393     fprintf(file, "\n");
7394
7395     fprintf(file,
7396             "   other read counters: data %u, " "ack %u, " "dup %u "
7397             "spurious %u " "dally %u\n", s->dataPacketsRead,
7398             s->ackPacketsRead, s->dupPacketsRead, s->spuriousPacketsRead,
7399             s->ignorePacketDally);
7400
7401     fprintf(file, "   packets sent: ");
7402     for (i = 0; i < RX_N_PACKET_TYPES; i++) {
7403         fprintf(file, "%s %u ", rx_packetTypes[i], s->packetsSent[i]);
7404     }
7405     fprintf(file, "\n");
7406
7407     fprintf(file,
7408             "   other send counters: ack %u, " "data %u (not resends), "
7409             "resends %u, " "pushed %u, " "acked&ignored %u\n",
7410             s->ackPacketsSent, s->dataPacketsSent, s->dataPacketsReSent,
7411             s->dataPacketsPushed, s->ignoreAckedPacket);
7412
7413     fprintf(file,
7414             "   \t(these should be small) sendFailed %u, " "fatalErrors %u\n",
7415             s->netSendFailures, (int)s->fatalErrors);
7416
7417     if (s->nRttSamples) {
7418         fprintf(file, "   Average rtt is %0.3f, with %d samples\n",
7419                 clock_Float(&s->totalRtt) / s->nRttSamples, s->nRttSamples);
7420
7421         fprintf(file, "   Minimum rtt is %0.3f, maximum is %0.3f\n",
7422                 clock_Float(&s->minRtt), clock_Float(&s->maxRtt));
7423     }
7424
7425     fprintf(file,
7426             "   %d server connections, " "%d client connections, "
7427             "%d peer structs, " "%d call structs, " "%d free call structs\n",
7428             s->nServerConns, s->nClientConns, s->nPeerStructs,
7429             s->nCallStructs, s->nFreeCallStructs);
7430
7431 #if     !defined(AFS_PTHREAD_ENV) && !defined(AFS_USE_GETTIMEOFDAY)
7432     fprintf(file, "   %d clock updates\n", clock_nUpdates);
7433 #endif
7434 }
7435
7436 /* for backward compatibility */
7437 void
7438 rx_PrintStats(FILE * file)
7439 {
7440     MUTEX_ENTER(&rx_stats_mutex);
7441     rx_PrintTheseStats(file, (struct rx_statistics *) &rx_stats,
7442                        sizeof(rx_stats), rx_nFreePackets,
7443                        RX_DEBUGI_VERSION);
7444     MUTEX_EXIT(&rx_stats_mutex);
7445 }
7446
7447 void
7448 rx_PrintPeerStats(FILE * file, struct rx_peer *peer)
7449 {
7450     fprintf(file, "Peer %x.%d.  " "Burst size %d, " "burst wait %d.%06d.\n",
7451             ntohl(peer->host), (int)ntohs(peer->port), (int)peer->burstSize,
7452             (int)peer->burstWait.sec, (int)peer->burstWait.usec);
7453
7454     fprintf(file,
7455             "   Rtt %d, " "total sent %d, " "resent %d\n",
7456             peer->rtt, peer->nSent, peer->reSends);
7457
7458     fprintf(file,
7459             "   Packet size %d, " "max in packet skew %d, "
7460             "max out packet skew %d\n", peer->ifMTU, (int)peer->inPacketSkew,
7461             (int)peer->outPacketSkew);
7462 }
7463 #endif
7464
7465 #if defined(AFS_PTHREAD_ENV) && defined(RXDEBUG)
7466 /*
7467  * This mutex protects the following static variables:
7468  * counter
7469  */
7470
7471 #define LOCK_RX_DEBUG MUTEX_ENTER(&rx_debug_mutex)
7472 #define UNLOCK_RX_DEBUG MUTEX_EXIT(&rx_debug_mutex)
7473 #else
7474 #define LOCK_RX_DEBUG
7475 #define UNLOCK_RX_DEBUG
7476 #endif /* AFS_PTHREAD_ENV */
7477
7478 #if defined(RXDEBUG) || defined(MAKEDEBUGCALL)
7479 static int
7480 MakeDebugCall(osi_socket socket, afs_uint32 remoteAddr, afs_uint16 remotePort,
7481               u_char type, void *inputData, size_t inputLength,
7482               void *outputData, size_t outputLength)
7483 {
7484     static afs_int32 counter = 100;
7485     time_t waitTime, waitCount;
7486     struct rx_header theader;
7487     char tbuffer[1500];
7488     afs_int32 code;
7489     struct timeval tv_now, tv_wake, tv_delta;
7490     struct sockaddr_in taddr, faddr;
7491 #ifdef AFS_NT40_ENV
7492     int faddrLen;
7493 #else
7494     socklen_t faddrLen;
7495 #endif
7496     fd_set imask;
7497     char *tp;
7498
7499     waitTime = 1;
7500     waitCount = 5;
7501     LOCK_RX_DEBUG;
7502     counter++;
7503     UNLOCK_RX_DEBUG;
7504     tp = &tbuffer[sizeof(struct rx_header)];
7505     taddr.sin_family = AF_INET;
7506     taddr.sin_port = remotePort;
7507     taddr.sin_addr.s_addr = remoteAddr;
7508 #ifdef STRUCT_SOCKADDR_HAS_SA_LEN
7509     taddr.sin_len = sizeof(struct sockaddr_in);
7510 #endif
7511     while (1) {
7512         memset(&theader, 0, sizeof(theader));
7513         theader.epoch = htonl(999);
7514         theader.cid = 0;
7515         theader.callNumber = htonl(counter);
7516         theader.seq = 0;
7517         theader.serial = 0;
7518         theader.type = type;
7519         theader.flags = RX_CLIENT_INITIATED | RX_LAST_PACKET;
7520         theader.serviceId = 0;
7521
7522         memcpy(tbuffer, &theader, sizeof(theader));
7523         memcpy(tp, inputData, inputLength);
7524         code =
7525             sendto(socket, tbuffer, inputLength + sizeof(struct rx_header), 0,
7526                    (struct sockaddr *)&taddr, sizeof(struct sockaddr_in));
7527
7528         /* see if there's a packet available */
7529         gettimeofday(&tv_wake, NULL);
7530         tv_wake.tv_sec += waitTime;
7531         for (;;) {
7532             FD_ZERO(&imask);
7533             FD_SET(socket, &imask);
7534             tv_delta.tv_sec = tv_wake.tv_sec;
7535             tv_delta.tv_usec = tv_wake.tv_usec;
7536             gettimeofday(&tv_now, NULL);
7537
7538             if (tv_delta.tv_usec < tv_now.tv_usec) {
7539                 /* borrow */
7540                 tv_delta.tv_usec += 1000000;
7541                 tv_delta.tv_sec--;
7542             }
7543             tv_delta.tv_usec -= tv_now.tv_usec;
7544
7545             if (tv_delta.tv_sec < tv_now.tv_sec) {
7546                 /* time expired */
7547                 break;
7548             }
7549             tv_delta.tv_sec -= tv_now.tv_sec;
7550
7551 #ifdef AFS_NT40_ENV
7552             code = select(0, &imask, 0, 0, &tv_delta);
7553 #else /* AFS_NT40_ENV */
7554             code = select(socket + 1, &imask, 0, 0, &tv_delta);
7555 #endif /* AFS_NT40_ENV */
7556             if (code == 1 && FD_ISSET(socket, &imask)) {
7557                 /* now receive a packet */
7558                 faddrLen = sizeof(struct sockaddr_in);
7559                 code =
7560                     recvfrom(socket, tbuffer, sizeof(tbuffer), 0,
7561                              (struct sockaddr *)&faddr, &faddrLen);
7562
7563                 if (code > 0) {
7564                     memcpy(&theader, tbuffer, sizeof(struct rx_header));
7565                     if (counter == ntohl(theader.callNumber))
7566                         goto success;
7567                     continue;
7568                 }
7569             }
7570             break;
7571         }
7572
7573         /* see if we've timed out */
7574         if (!--waitCount) {
7575             return -1;
7576         }
7577         waitTime <<= 1;
7578     }
7579
7580  success:
7581     code -= sizeof(struct rx_header);
7582     if (code > outputLength)
7583         code = outputLength;
7584     memcpy(outputData, tp, code);
7585     return code;
7586 }
7587 #endif /* RXDEBUG */
7588
7589 afs_int32
7590 rx_GetServerDebug(osi_socket socket, afs_uint32 remoteAddr,
7591                   afs_uint16 remotePort, struct rx_debugStats * stat,
7592                   afs_uint32 * supportedValues)
7593 {
7594 #if defined(RXDEBUG) || defined(MAKEDEBUGCALL)
7595     afs_int32 rc = 0;
7596     struct rx_debugIn in;
7597
7598     *supportedValues = 0;
7599     in.type = htonl(RX_DEBUGI_GETSTATS);
7600     in.index = 0;
7601
7602     rc = MakeDebugCall(socket, remoteAddr, remotePort, RX_PACKET_TYPE_DEBUG,
7603                        &in, sizeof(in), stat, sizeof(*stat));
7604
7605     /*
7606      * If the call was successful, fixup the version and indicate
7607      * what contents of the stat structure are valid.
7608      * Also do net to host conversion of fields here.
7609      */
7610
7611     if (rc >= 0) {
7612         if (stat->version >= RX_DEBUGI_VERSION_W_SECSTATS) {
7613             *supportedValues |= RX_SERVER_DEBUG_SEC_STATS;
7614         }
7615         if (stat->version >= RX_DEBUGI_VERSION_W_GETALLCONN) {
7616             *supportedValues |= RX_SERVER_DEBUG_ALL_CONN;
7617         }
7618         if (stat->version >= RX_DEBUGI_VERSION_W_RXSTATS) {
7619             *supportedValues |= RX_SERVER_DEBUG_RX_STATS;
7620         }
7621         if (stat->version >= RX_DEBUGI_VERSION_W_WAITERS) {
7622             *supportedValues |= RX_SERVER_DEBUG_WAITER_CNT;
7623         }
7624         if (stat->version >= RX_DEBUGI_VERSION_W_IDLETHREADS) {
7625             *supportedValues |= RX_SERVER_DEBUG_IDLE_THREADS;
7626         }
7627         if (stat->version >= RX_DEBUGI_VERSION_W_NEWPACKETTYPES) {
7628             *supportedValues |= RX_SERVER_DEBUG_NEW_PACKETS;
7629         }
7630         if (stat->version >= RX_DEBUGI_VERSION_W_GETPEER) {
7631             *supportedValues |= RX_SERVER_DEBUG_ALL_PEER;
7632         }
7633         if (stat->version >= RX_DEBUGI_VERSION_W_WAITED) {
7634             *supportedValues |= RX_SERVER_DEBUG_WAITED_CNT;
7635         }
7636         if (stat->version >= RX_DEBUGI_VERSION_W_PACKETS) {
7637             *supportedValues |= RX_SERVER_DEBUG_PACKETS_CNT;
7638         }
7639         stat->nFreePackets = ntohl(stat->nFreePackets);
7640         stat->packetReclaims = ntohl(stat->packetReclaims);
7641         stat->callsExecuted = ntohl(stat->callsExecuted);
7642         stat->nWaiting = ntohl(stat->nWaiting);
7643         stat->idleThreads = ntohl(stat->idleThreads);
7644         stat->nWaited = ntohl(stat->nWaited);
7645         stat->nPackets = ntohl(stat->nPackets);
7646     }
7647 #else
7648     afs_int32 rc = -1;
7649 #endif
7650     return rc;
7651 }
7652
7653 afs_int32
7654 rx_GetServerStats(osi_socket socket, afs_uint32 remoteAddr,
7655                   afs_uint16 remotePort, struct rx_statistics * stat,
7656                   afs_uint32 * supportedValues)
7657 {
7658 #if defined(RXDEBUG) || defined(MAKEDEBUGCALL)
7659     afs_int32 rc = 0;
7660     struct rx_debugIn in;
7661     afs_int32 *lp = (afs_int32 *) stat;
7662     int i;
7663
7664     /*
7665      * supportedValues is currently unused, but added to allow future
7666      * versioning of this function.
7667      */
7668
7669     *supportedValues = 0;
7670     in.type = htonl(RX_DEBUGI_RXSTATS);
7671     in.index = 0;
7672     memset(stat, 0, sizeof(*stat));
7673
7674     rc = MakeDebugCall(socket, remoteAddr, remotePort, RX_PACKET_TYPE_DEBUG,
7675                        &in, sizeof(in), stat, sizeof(*stat));
7676
7677     if (rc >= 0) {
7678
7679         /*
7680          * Do net to host conversion here
7681          */
7682
7683         for (i = 0; i < sizeof(*stat) / sizeof(afs_int32); i++, lp++) {
7684             *lp = ntohl(*lp);
7685         }
7686     }
7687 #else
7688     afs_int32 rc = -1;
7689 #endif
7690     return rc;
7691 }
7692
7693 afs_int32
7694 rx_GetServerVersion(osi_socket socket, afs_uint32 remoteAddr,
7695                     afs_uint16 remotePort, size_t version_length,
7696                     char *version)
7697 {
7698 #if defined(RXDEBUG) || defined(MAKEDEBUGCALL)
7699     char a[1] = { 0 };
7700     return MakeDebugCall(socket, remoteAddr, remotePort,
7701                          RX_PACKET_TYPE_VERSION, a, 1, version,
7702                          version_length);
7703 #else
7704     return -1;
7705 #endif
7706 }
7707
7708 afs_int32
7709 rx_GetServerConnections(osi_socket socket, afs_uint32 remoteAddr,
7710                         afs_uint16 remotePort, afs_int32 * nextConnection,
7711                         int allConnections, afs_uint32 debugSupportedValues,
7712                         struct rx_debugConn * conn,
7713                         afs_uint32 * supportedValues)
7714 {
7715 #if defined(RXDEBUG) || defined(MAKEDEBUGCALL)
7716     afs_int32 rc = 0;
7717     struct rx_debugIn in;
7718     int i;
7719
7720     /*
7721      * supportedValues is currently unused, but added to allow future
7722      * versioning of this function.
7723      */
7724
7725     *supportedValues = 0;
7726     if (allConnections) {
7727         in.type = htonl(RX_DEBUGI_GETALLCONN);
7728     } else {
7729         in.type = htonl(RX_DEBUGI_GETCONN);
7730     }
7731     in.index = htonl(*nextConnection);
7732     memset(conn, 0, sizeof(*conn));
7733
7734     rc = MakeDebugCall(socket, remoteAddr, remotePort, RX_PACKET_TYPE_DEBUG,
7735                        &in, sizeof(in), conn, sizeof(*conn));
7736
7737     if (rc >= 0) {
7738         *nextConnection += 1;
7739
7740         /*
7741          * Convert old connection format to new structure.
7742          */
7743
7744         if (debugSupportedValues & RX_SERVER_DEBUG_OLD_CONN) {
7745             struct rx_debugConn_vL *vL = (struct rx_debugConn_vL *)conn;
7746 #define MOVEvL(a) (conn->a = vL->a)
7747
7748             /* any old or unrecognized version... */
7749             for (i = 0; i < RX_MAXCALLS; i++) {
7750                 MOVEvL(callState[i]);
7751                 MOVEvL(callMode[i]);
7752                 MOVEvL(callFlags[i]);
7753                 MOVEvL(callOther[i]);
7754             }
7755             if (debugSupportedValues & RX_SERVER_DEBUG_SEC_STATS) {
7756                 MOVEvL(secStats.type);
7757                 MOVEvL(secStats.level);
7758                 MOVEvL(secStats.flags);
7759                 MOVEvL(secStats.expires);
7760                 MOVEvL(secStats.packetsReceived);
7761                 MOVEvL(secStats.packetsSent);
7762                 MOVEvL(secStats.bytesReceived);
7763                 MOVEvL(secStats.bytesSent);
7764             }
7765         }
7766
7767         /*
7768          * Do net to host conversion here
7769          * NOTE:
7770          *    I don't convert host or port since we are most likely
7771          *    going to want these in NBO.
7772          */
7773         conn->cid = ntohl(conn->cid);
7774         conn->serial = ntohl(conn->serial);
7775         for (i = 0; i < RX_MAXCALLS; i++) {
7776             conn->callNumber[i] = ntohl(conn->callNumber[i]);
7777         }
7778         conn->error = ntohl(conn->error);
7779         conn->secStats.flags = ntohl(conn->secStats.flags);
7780         conn->secStats.expires = ntohl(conn->secStats.expires);
7781         conn->secStats.packetsReceived =
7782             ntohl(conn->secStats.packetsReceived);
7783         conn->secStats.packetsSent = ntohl(conn->secStats.packetsSent);
7784         conn->secStats.bytesReceived = ntohl(conn->secStats.bytesReceived);
7785         conn->secStats.bytesSent = ntohl(conn->secStats.bytesSent);
7786         conn->epoch = ntohl(conn->epoch);
7787         conn->natMTU = ntohl(conn->natMTU);
7788     }
7789 #else
7790     afs_int32 rc = -1;
7791 #endif
7792     return rc;
7793 }
7794
7795 afs_int32
7796 rx_GetServerPeers(osi_socket socket, afs_uint32 remoteAddr,
7797                   afs_uint16 remotePort, afs_int32 * nextPeer,
7798                   afs_uint32 debugSupportedValues, struct rx_debugPeer * peer,
7799                   afs_uint32 * supportedValues)
7800 {
7801 #if defined(RXDEBUG) || defined(MAKEDEBUGCALL)
7802     afs_int32 rc = 0;
7803     struct rx_debugIn in;
7804
7805     /*
7806      * supportedValues is currently unused, but added to allow future
7807      * versioning of this function.
7808      */
7809
7810     *supportedValues = 0;
7811     in.type = htonl(RX_DEBUGI_GETPEER);
7812     in.index = htonl(*nextPeer);
7813     memset(peer, 0, sizeof(*peer));
7814
7815     rc = MakeDebugCall(socket, remoteAddr, remotePort, RX_PACKET_TYPE_DEBUG,
7816                        &in, sizeof(in), peer, sizeof(*peer));
7817
7818     if (rc >= 0) {
7819         *nextPeer += 1;
7820
7821         /*
7822          * Do net to host conversion here
7823          * NOTE:
7824          *    I don't convert host or port since we are most likely
7825          *    going to want these in NBO.
7826          */
7827         peer->ifMTU = ntohs(peer->ifMTU);
7828         peer->idleWhen = ntohl(peer->idleWhen);
7829         peer->refCount = ntohs(peer->refCount);
7830         peer->burstWait.sec = ntohl(peer->burstWait.sec);
7831         peer->burstWait.usec = ntohl(peer->burstWait.usec);
7832         peer->rtt = ntohl(peer->rtt);
7833         peer->rtt_dev = ntohl(peer->rtt_dev);
7834         peer->timeout.sec = 0;
7835         peer->timeout.usec = 0;
7836         peer->nSent = ntohl(peer->nSent);
7837         peer->reSends = ntohl(peer->reSends);
7838         peer->inPacketSkew = ntohl(peer->inPacketSkew);
7839         peer->outPacketSkew = ntohl(peer->outPacketSkew);
7840         peer->rateFlag = ntohl(peer->rateFlag);
7841         peer->natMTU = ntohs(peer->natMTU);
7842         peer->maxMTU = ntohs(peer->maxMTU);
7843         peer->maxDgramPackets = ntohs(peer->maxDgramPackets);
7844         peer->ifDgramPackets = ntohs(peer->ifDgramPackets);
7845         peer->MTU = ntohs(peer->MTU);
7846         peer->cwind = ntohs(peer->cwind);
7847         peer->nDgramPackets = ntohs(peer->nDgramPackets);
7848         peer->congestSeq = ntohs(peer->congestSeq);
7849         peer->bytesSent.high = ntohl(peer->bytesSent.high);
7850         peer->bytesSent.low = ntohl(peer->bytesSent.low);
7851         peer->bytesReceived.high = ntohl(peer->bytesReceived.high);
7852         peer->bytesReceived.low = ntohl(peer->bytesReceived.low);
7853     }
7854 #else
7855     afs_int32 rc = -1;
7856 #endif
7857     return rc;
7858 }
7859
7860 afs_int32
7861 rx_GetLocalPeers(afs_uint32 peerHost, afs_uint16 peerPort,
7862                 struct rx_debugPeer * peerStats)
7863 {
7864         struct rx_peer *tp;
7865         afs_int32 error = 1; /* default to "did not succeed" */
7866         afs_uint32 hashValue = PEER_HASH(peerHost, peerPort);
7867
7868         MUTEX_ENTER(&rx_peerHashTable_lock);
7869         for(tp = rx_peerHashTable[hashValue];
7870               tp != NULL; tp = tp->next) {
7871                 if (tp->host == peerHost)
7872                         break;
7873         }
7874
7875         if (tp) {
7876                 tp->refCount++;
7877                 MUTEX_EXIT(&rx_peerHashTable_lock);
7878
7879                 error = 0;
7880
7881                 MUTEX_ENTER(&tp->peer_lock);
7882                 peerStats->host = tp->host;
7883                 peerStats->port = tp->port;
7884                 peerStats->ifMTU = tp->ifMTU;
7885                 peerStats->idleWhen = tp->idleWhen;
7886                 peerStats->refCount = tp->refCount;
7887                 peerStats->burstSize = tp->burstSize;
7888                 peerStats->burst = tp->burst;
7889                 peerStats->burstWait.sec = tp->burstWait.sec;
7890                 peerStats->burstWait.usec = tp->burstWait.usec;
7891                 peerStats->rtt = tp->rtt;
7892                 peerStats->rtt_dev = tp->rtt_dev;
7893                 peerStats->timeout.sec = 0;
7894                 peerStats->timeout.usec = 0;
7895                 peerStats->nSent = tp->nSent;
7896                 peerStats->reSends = tp->reSends;
7897                 peerStats->inPacketSkew = tp->inPacketSkew;
7898                 peerStats->outPacketSkew = tp->outPacketSkew;
7899                 peerStats->rateFlag = tp->rateFlag;
7900                 peerStats->natMTU = tp->natMTU;
7901                 peerStats->maxMTU = tp->maxMTU;
7902                 peerStats->maxDgramPackets = tp->maxDgramPackets;
7903                 peerStats->ifDgramPackets = tp->ifDgramPackets;
7904                 peerStats->MTU = tp->MTU;
7905                 peerStats->cwind = tp->cwind;
7906                 peerStats->nDgramPackets = tp->nDgramPackets;
7907                 peerStats->congestSeq = tp->congestSeq;
7908                 peerStats->bytesSent.high = tp->bytesSent.high;
7909                 peerStats->bytesSent.low = tp->bytesSent.low;
7910                 peerStats->bytesReceived.high = tp->bytesReceived.high;
7911                 peerStats->bytesReceived.low = tp->bytesReceived.low;
7912                 MUTEX_EXIT(&tp->peer_lock);
7913
7914                 MUTEX_ENTER(&rx_peerHashTable_lock);
7915                 tp->refCount--;
7916         }
7917         MUTEX_EXIT(&rx_peerHashTable_lock);
7918
7919         return error;
7920 }
7921
7922 void
7923 shutdown_rx(void)
7924 {
7925     struct rx_serverQueueEntry *np;
7926     int i, j;
7927 #ifndef KERNEL
7928     struct rx_call *call;
7929     struct rx_serverQueueEntry *sq;
7930 #endif /* KERNEL */
7931
7932     LOCK_RX_INIT;
7933     if (rxinit_status == 1) {
7934         UNLOCK_RX_INIT;
7935         return;                 /* Already shutdown. */
7936     }
7937 #ifndef KERNEL
7938     rx_port = 0;
7939 #ifndef AFS_PTHREAD_ENV
7940     FD_ZERO(&rx_selectMask);
7941 #endif /* AFS_PTHREAD_ENV */
7942     rxi_dataQuota = RX_MAX_QUOTA;
7943 #ifndef AFS_PTHREAD_ENV
7944     rxi_StopListener();
7945 #endif /* AFS_PTHREAD_ENV */
7946     shutdown_rxevent();
7947     rx_SetEpoch(0);
7948 #ifndef AFS_PTHREAD_ENV
7949 #ifndef AFS_USE_GETTIMEOFDAY
7950     clock_UnInit();
7951 #endif /* AFS_USE_GETTIMEOFDAY */
7952 #endif /* AFS_PTHREAD_ENV */
7953
7954     while (!queue_IsEmpty(&rx_freeCallQueue)) {
7955         call = queue_First(&rx_freeCallQueue, rx_call);
7956         queue_Remove(call);
7957         rxi_Free(call, sizeof(struct rx_call));
7958     }
7959
7960     while (!queue_IsEmpty(&rx_idleServerQueue)) {
7961         sq = queue_First(&rx_idleServerQueue, rx_serverQueueEntry);
7962         queue_Remove(sq);
7963     }
7964 #endif /* KERNEL */
7965
7966     {
7967         struct rx_peer **peer_ptr, **peer_end;
7968         for (peer_ptr = &rx_peerHashTable[0], peer_end =
7969              &rx_peerHashTable[rx_hashTableSize]; peer_ptr < peer_end;
7970              peer_ptr++) {
7971             struct rx_peer *peer, *next;
7972
7973             MUTEX_ENTER(&rx_peerHashTable_lock);
7974             for (peer = *peer_ptr; peer; peer = next) {
7975                 rx_interface_stat_p rpc_stat, nrpc_stat;
7976                 size_t space;
7977
7978                 MUTEX_ENTER(&rx_rpc_stats);
7979                 MUTEX_ENTER(&peer->peer_lock);
7980                 for (queue_Scan
7981                      (&peer->rpcStats, rpc_stat, nrpc_stat,
7982                       rx_interface_stat)) {
7983                     unsigned int num_funcs;
7984                     if (!rpc_stat)
7985                         break;
7986                     queue_Remove(&rpc_stat->queue_header);
7987                     queue_Remove(&rpc_stat->all_peers);
7988                     num_funcs = rpc_stat->stats[0].func_total;
7989                     space =
7990                         sizeof(rx_interface_stat_t) +
7991                         rpc_stat->stats[0].func_total *
7992                         sizeof(rx_function_entry_v1_t);
7993
7994                     rxi_Free(rpc_stat, space);
7995
7996                     /* rx_rpc_stats must be held */
7997                     rxi_rpc_peer_stat_cnt -= num_funcs;
7998                 }
7999                 MUTEX_EXIT(&peer->peer_lock);
8000                 MUTEX_EXIT(&rx_rpc_stats);
8001
8002                 next = peer->next;
8003                 rxi_FreePeer(peer);
8004                 if (rx_stats_active)
8005                     rx_atomic_dec(&rx_stats.nPeerStructs);
8006             }
8007             MUTEX_EXIT(&rx_peerHashTable_lock);
8008         }
8009     }
8010     for (i = 0; i < RX_MAX_SERVICES; i++) {
8011         if (rx_services[i])
8012             rxi_Free(rx_services[i], sizeof(*rx_services[i]));
8013     }
8014     for (i = 0; i < rx_hashTableSize; i++) {
8015         struct rx_connection *tc, *ntc;
8016         MUTEX_ENTER(&rx_connHashTable_lock);
8017         for (tc = rx_connHashTable[i]; tc; tc = ntc) {
8018             ntc = tc->next;
8019             for (j = 0; j < RX_MAXCALLS; j++) {
8020                 if (tc->call[j]) {
8021                     rxi_Free(tc->call[j], sizeof(*tc->call[j]));
8022                 }
8023             }
8024             rxi_Free(tc, sizeof(*tc));
8025         }
8026         MUTEX_EXIT(&rx_connHashTable_lock);
8027     }
8028
8029     MUTEX_ENTER(&freeSQEList_lock);
8030
8031     while ((np = rx_FreeSQEList)) {
8032         rx_FreeSQEList = *(struct rx_serverQueueEntry **)np;
8033         MUTEX_DESTROY(&np->lock);
8034         rxi_Free(np, sizeof(*np));
8035     }
8036
8037     MUTEX_EXIT(&freeSQEList_lock);
8038     MUTEX_DESTROY(&freeSQEList_lock);
8039     MUTEX_DESTROY(&rx_freeCallQueue_lock);
8040     MUTEX_DESTROY(&rx_connHashTable_lock);
8041     MUTEX_DESTROY(&rx_peerHashTable_lock);
8042     MUTEX_DESTROY(&rx_serverPool_lock);
8043
8044     osi_Free(rx_connHashTable,
8045              rx_hashTableSize * sizeof(struct rx_connection *));
8046     osi_Free(rx_peerHashTable, rx_hashTableSize * sizeof(struct rx_peer *));
8047
8048     UNPIN(rx_connHashTable,
8049           rx_hashTableSize * sizeof(struct rx_connection *));
8050     UNPIN(rx_peerHashTable, rx_hashTableSize * sizeof(struct rx_peer *));
8051
8052     rxi_FreeAllPackets();
8053
8054     MUTEX_ENTER(&rx_quota_mutex);
8055     rxi_dataQuota = RX_MAX_QUOTA;
8056     rxi_availProcs = rxi_totalMin = rxi_minDeficit = 0;
8057     MUTEX_EXIT(&rx_quota_mutex);
8058     rxinit_status = 1;
8059     UNLOCK_RX_INIT;
8060 }
8061
8062 #ifdef RX_ENABLE_LOCKS
8063 void
8064 osirx_AssertMine(afs_kmutex_t * lockaddr, char *msg)
8065 {
8066     if (!MUTEX_ISMINE(lockaddr))
8067         osi_Panic("Lock not held: %s", msg);
8068 }
8069 #endif /* RX_ENABLE_LOCKS */
8070
8071 #ifndef KERNEL
8072
8073 /*
8074  * Routines to implement connection specific data.
8075  */
8076
8077 int
8078 rx_KeyCreate(rx_destructor_t rtn)
8079 {
8080     int key;
8081     MUTEX_ENTER(&rxi_keyCreate_lock);
8082     key = rxi_keyCreate_counter++;
8083     rxi_keyCreate_destructor = (rx_destructor_t *)
8084         realloc((void *)rxi_keyCreate_destructor,
8085                 (key + 1) * sizeof(rx_destructor_t));
8086     rxi_keyCreate_destructor[key] = rtn;
8087     MUTEX_EXIT(&rxi_keyCreate_lock);
8088     return key;
8089 }
8090
8091 void
8092 rx_SetSpecific(struct rx_connection *conn, int key, void *ptr)
8093 {
8094     int i;
8095     MUTEX_ENTER(&conn->conn_data_lock);
8096     if (!conn->specific) {
8097         conn->specific = (void **)malloc((key + 1) * sizeof(void *));
8098         for (i = 0; i < key; i++)
8099             conn->specific[i] = NULL;
8100         conn->nSpecific = key + 1;
8101         conn->specific[key] = ptr;
8102     } else if (key >= conn->nSpecific) {
8103         conn->specific = (void **)
8104             realloc(conn->specific, (key + 1) * sizeof(void *));
8105         for (i = conn->nSpecific; i < key; i++)
8106             conn->specific[i] = NULL;
8107         conn->nSpecific = key + 1;
8108         conn->specific[key] = ptr;
8109     } else {
8110         if (conn->specific[key] && rxi_keyCreate_destructor[key])
8111             (*rxi_keyCreate_destructor[key]) (conn->specific[key]);
8112         conn->specific[key] = ptr;
8113     }
8114     MUTEX_EXIT(&conn->conn_data_lock);
8115 }
8116
8117 void
8118 rx_SetServiceSpecific(struct rx_service *svc, int key, void *ptr)
8119 {
8120     int i;
8121     MUTEX_ENTER(&svc->svc_data_lock);
8122     if (!svc->specific) {
8123         svc->specific = (void **)malloc((key + 1) * sizeof(void *));
8124         for (i = 0; i < key; i++)
8125             svc->specific[i] = NULL;
8126         svc->nSpecific = key + 1;
8127         svc->specific[key] = ptr;
8128     } else if (key >= svc->nSpecific) {
8129         svc->specific = (void **)
8130             realloc(svc->specific, (key + 1) * sizeof(void *));
8131         for (i = svc->nSpecific; i < key; i++)
8132             svc->specific[i] = NULL;
8133         svc->nSpecific = key + 1;
8134         svc->specific[key] = ptr;
8135     } else {
8136         if (svc->specific[key] && rxi_keyCreate_destructor[key])
8137             (*rxi_keyCreate_destructor[key]) (svc->specific[key]);
8138         svc->specific[key] = ptr;
8139     }
8140     MUTEX_EXIT(&svc->svc_data_lock);
8141 }
8142
8143 void *
8144 rx_GetSpecific(struct rx_connection *conn, int key)
8145 {
8146     void *ptr;
8147     MUTEX_ENTER(&conn->conn_data_lock);
8148     if (key >= conn->nSpecific)
8149         ptr = NULL;
8150     else
8151         ptr = conn->specific[key];
8152     MUTEX_EXIT(&conn->conn_data_lock);
8153     return ptr;
8154 }
8155
8156 void *
8157 rx_GetServiceSpecific(struct rx_service *svc, int key)
8158 {
8159     void *ptr;
8160     MUTEX_ENTER(&svc->svc_data_lock);
8161     if (key >= svc->nSpecific)
8162         ptr = NULL;
8163     else
8164         ptr = svc->specific[key];
8165     MUTEX_EXIT(&svc->svc_data_lock);
8166     return ptr;
8167 }
8168
8169
8170 #endif /* !KERNEL */
8171
8172 /*
8173  * processStats is a queue used to store the statistics for the local
8174  * process.  Its contents are similar to the contents of the rpcStats
8175  * queue on a rx_peer structure, but the actual data stored within
8176  * this queue contains totals across the lifetime of the process (assuming
8177  * the stats have not been reset) - unlike the per peer structures
8178  * which can come and go based upon the peer lifetime.
8179  */
8180
8181 static struct rx_queue processStats = { &processStats, &processStats };
8182
8183 /*
8184  * peerStats is a queue used to store the statistics for all peer structs.
8185  * Its contents are the union of all the peer rpcStats queues.
8186  */
8187
8188 static struct rx_queue peerStats = { &peerStats, &peerStats };
8189
8190 /*
8191  * rxi_monitor_processStats is used to turn process wide stat collection
8192  * on and off
8193  */
8194
8195 static int rxi_monitor_processStats = 0;
8196
8197 /*
8198  * rxi_monitor_peerStats is used to turn per peer stat collection on and off
8199  */
8200
8201 static int rxi_monitor_peerStats = 0;
8202
8203 /*
8204  * rxi_AddRpcStat - given all of the information for a particular rpc
8205  * call, create (if needed) and update the stat totals for the rpc.
8206  *
8207  * PARAMETERS
8208  *
8209  * IN stats - the queue of stats that will be updated with the new value
8210  *
8211  * IN rxInterface - a unique number that identifies the rpc interface
8212  *
8213  * IN currentFunc - the index of the function being invoked
8214  *
8215  * IN totalFunc - the total number of functions in this interface
8216  *
8217  * IN queueTime - the amount of time this function waited for a thread
8218  *
8219  * IN execTime - the amount of time this function invocation took to execute
8220  *
8221  * IN bytesSent - the number bytes sent by this invocation
8222  *
8223  * IN bytesRcvd - the number bytes received by this invocation
8224  *
8225  * IN isServer - if true, this invocation was made to a server
8226  *
8227  * IN remoteHost - the ip address of the remote host
8228  *
8229  * IN remotePort - the port of the remote host
8230  *
8231  * IN addToPeerList - if != 0, add newly created stat to the global peer list
8232  *
8233  * INOUT counter - if a new stats structure is allocated, the counter will
8234  * be updated with the new number of allocated stat structures
8235  *
8236  * RETURN CODES
8237  *
8238  * Returns void.
8239  */
8240
8241 static int
8242 rxi_AddRpcStat(struct rx_queue *stats, afs_uint32 rxInterface,
8243                afs_uint32 currentFunc, afs_uint32 totalFunc,
8244                struct clock *queueTime, struct clock *execTime,
8245                afs_hyper_t * bytesSent, afs_hyper_t * bytesRcvd, int isServer,
8246                afs_uint32 remoteHost, afs_uint32 remotePort,
8247                int addToPeerList, unsigned int *counter)
8248 {
8249     int rc = 0;
8250     rx_interface_stat_p rpc_stat, nrpc_stat;
8251
8252     /*
8253      * See if there's already a structure for this interface
8254      */
8255
8256     for (queue_Scan(stats, rpc_stat, nrpc_stat, rx_interface_stat)) {
8257         if ((rpc_stat->stats[0].interfaceId == rxInterface)
8258             && (rpc_stat->stats[0].remote_is_server == isServer))
8259             break;
8260     }
8261
8262     /*
8263      * Didn't find a match so allocate a new structure and add it to the
8264      * queue.
8265      */
8266
8267     if (queue_IsEnd(stats, rpc_stat) || (rpc_stat == NULL)
8268         || (rpc_stat->stats[0].interfaceId != rxInterface)
8269         || (rpc_stat->stats[0].remote_is_server != isServer)) {
8270         int i;
8271         size_t space;
8272
8273         space =
8274             sizeof(rx_interface_stat_t) +
8275             totalFunc * sizeof(rx_function_entry_v1_t);
8276
8277         rpc_stat = rxi_Alloc(space);
8278         if (rpc_stat == NULL) {
8279             rc = 1;
8280             goto fail;
8281         }
8282         *counter += totalFunc;
8283         for (i = 0; i < totalFunc; i++) {
8284             rpc_stat->stats[i].remote_peer = remoteHost;
8285             rpc_stat->stats[i].remote_port = remotePort;
8286             rpc_stat->stats[i].remote_is_server = isServer;
8287             rpc_stat->stats[i].interfaceId = rxInterface;
8288             rpc_stat->stats[i].func_total = totalFunc;
8289             rpc_stat->stats[i].func_index = i;
8290             hzero(rpc_stat->stats[i].invocations);
8291             hzero(rpc_stat->stats[i].bytes_sent);
8292             hzero(rpc_stat->stats[i].bytes_rcvd);
8293             rpc_stat->stats[i].queue_time_sum.sec = 0;
8294             rpc_stat->stats[i].queue_time_sum.usec = 0;
8295             rpc_stat->stats[i].queue_time_sum_sqr.sec = 0;
8296             rpc_stat->stats[i].queue_time_sum_sqr.usec = 0;
8297             rpc_stat->stats[i].queue_time_min.sec = 9999999;
8298             rpc_stat->stats[i].queue_time_min.usec = 9999999;
8299             rpc_stat->stats[i].queue_time_max.sec = 0;
8300             rpc_stat->stats[i].queue_time_max.usec = 0;
8301             rpc_stat->stats[i].execution_time_sum.sec = 0;
8302             rpc_stat->stats[i].execution_time_sum.usec = 0;
8303             rpc_stat->stats[i].execution_time_sum_sqr.sec = 0;
8304             rpc_stat->stats[i].execution_time_sum_sqr.usec = 0;
8305             rpc_stat->stats[i].execution_time_min.sec = 9999999;
8306             rpc_stat->stats[i].execution_time_min.usec = 9999999;
8307             rpc_stat->stats[i].execution_time_max.sec = 0;
8308             rpc_stat->stats[i].execution_time_max.usec = 0;
8309         }
8310         queue_Prepend(stats, rpc_stat);
8311         if (addToPeerList) {
8312             queue_Prepend(&peerStats, &rpc_stat->all_peers);
8313         }
8314     }
8315
8316     /*
8317      * Increment the stats for this function
8318      */
8319
8320     hadd32(rpc_stat->stats[currentFunc].invocations, 1);
8321     hadd(rpc_stat->stats[currentFunc].bytes_sent, *bytesSent);
8322     hadd(rpc_stat->stats[currentFunc].bytes_rcvd, *bytesRcvd);
8323     clock_Add(&rpc_stat->stats[currentFunc].queue_time_sum, queueTime);
8324     clock_AddSq(&rpc_stat->stats[currentFunc].queue_time_sum_sqr, queueTime);
8325     if (clock_Lt(queueTime, &rpc_stat->stats[currentFunc].queue_time_min)) {
8326         rpc_stat->stats[currentFunc].queue_time_min = *queueTime;
8327     }
8328     if (clock_Gt(queueTime, &rpc_stat->stats[currentFunc].queue_time_max)) {
8329         rpc_stat->stats[currentFunc].queue_time_max = *queueTime;
8330     }
8331     clock_Add(&rpc_stat->stats[currentFunc].execution_time_sum, execTime);
8332     clock_AddSq(&rpc_stat->stats[currentFunc].execution_time_sum_sqr,
8333                 execTime);
8334     if (clock_Lt(execTime, &rpc_stat->stats[currentFunc].execution_time_min)) {
8335         rpc_stat->stats[currentFunc].execution_time_min = *execTime;
8336     }
8337     if (clock_Gt(execTime, &rpc_stat->stats[currentFunc].execution_time_max)) {
8338         rpc_stat->stats[currentFunc].execution_time_max = *execTime;
8339     }
8340
8341   fail:
8342     return rc;
8343 }
8344
8345 /*
8346  * rx_IncrementTimeAndCount - increment the times and count for a particular
8347  * rpc function.
8348  *
8349  * PARAMETERS
8350  *
8351  * IN peer - the peer who invoked the rpc
8352  *
8353  * IN rxInterface - a unique number that identifies the rpc interface
8354  *
8355  * IN currentFunc - the index of the function being invoked
8356  *
8357  * IN totalFunc - the total number of functions in this interface
8358  *
8359  * IN queueTime - the amount of time this function waited for a thread
8360  *
8361  * IN execTime - the amount of time this function invocation took to execute
8362  *
8363  * IN bytesSent - the number bytes sent by this invocation
8364  *
8365  * IN bytesRcvd - the number bytes received by this invocation
8366  *
8367  * IN isServer - if true, this invocation was made to a server
8368  *
8369  * RETURN CODES
8370  *
8371  * Returns void.
8372  */
8373
8374 void
8375 rx_IncrementTimeAndCount(struct rx_peer *peer, afs_uint32 rxInterface,
8376                          afs_uint32 currentFunc, afs_uint32 totalFunc,
8377                          struct clock *queueTime, struct clock *execTime,
8378                          afs_hyper_t * bytesSent, afs_hyper_t * bytesRcvd,
8379                          int isServer)
8380 {
8381
8382     if (!(rxi_monitor_peerStats || rxi_monitor_processStats))
8383         return;
8384
8385     MUTEX_ENTER(&rx_rpc_stats);
8386
8387     if (rxi_monitor_peerStats) {
8388         MUTEX_ENTER(&peer->peer_lock);
8389         rxi_AddRpcStat(&peer->rpcStats, rxInterface, currentFunc, totalFunc,
8390                        queueTime, execTime, bytesSent, bytesRcvd, isServer,
8391                        peer->host, peer->port, 1, &rxi_rpc_peer_stat_cnt);
8392         MUTEX_EXIT(&peer->peer_lock);
8393     }
8394
8395     if (rxi_monitor_processStats) {
8396         rxi_AddRpcStat(&processStats, rxInterface, currentFunc, totalFunc,
8397                        queueTime, execTime, bytesSent, bytesRcvd, isServer,
8398                        0xffffffff, 0xffffffff, 0, &rxi_rpc_process_stat_cnt);
8399     }
8400
8401     MUTEX_EXIT(&rx_rpc_stats);
8402
8403 }
8404
8405 /*
8406  * rx_MarshallProcessRPCStats - marshall an array of rpc statistics
8407  *
8408  * PARAMETERS
8409  *
8410  * IN callerVersion - the rpc stat version of the caller.
8411  *
8412  * IN count - the number of entries to marshall.
8413  *
8414  * IN stats - pointer to stats to be marshalled.
8415  *
8416  * OUT ptr - Where to store the marshalled data.
8417  *
8418  * RETURN CODES
8419  *
8420  * Returns void.
8421  */
8422 void
8423 rx_MarshallProcessRPCStats(afs_uint32 callerVersion, int count,
8424                            rx_function_entry_v1_t * stats, afs_uint32 ** ptrP)
8425 {
8426     int i;
8427     afs_uint32 *ptr;
8428
8429     /*
8430      * We only support the first version
8431      */
8432     for (ptr = *ptrP, i = 0; i < count; i++, stats++) {
8433         *(ptr++) = stats->remote_peer;
8434         *(ptr++) = stats->remote_port;
8435         *(ptr++) = stats->remote_is_server;
8436         *(ptr++) = stats->interfaceId;
8437         *(ptr++) = stats->func_total;
8438         *(ptr++) = stats->func_index;
8439         *(ptr++) = hgethi(stats->invocations);
8440         *(ptr++) = hgetlo(stats->invocations);
8441         *(ptr++) = hgethi(stats->bytes_sent);
8442         *(ptr++) = hgetlo(stats->bytes_sent);
8443         *(ptr++) = hgethi(stats->bytes_rcvd);
8444         *(ptr++) = hgetlo(stats->bytes_rcvd);
8445         *(ptr++) = stats->queue_time_sum.sec;
8446         *(ptr++) = stats->queue_time_sum.usec;
8447         *(ptr++) = stats->queue_time_sum_sqr.sec;
8448         *(ptr++) = stats->queue_time_sum_sqr.usec;
8449         *(ptr++) = stats->queue_time_min.sec;
8450         *(ptr++) = stats->queue_time_min.usec;
8451         *(ptr++) = stats->queue_time_max.sec;
8452         *(ptr++) = stats->queue_time_max.usec;
8453         *(ptr++) = stats->execution_time_sum.sec;
8454         *(ptr++) = stats->execution_time_sum.usec;
8455         *(ptr++) = stats->execution_time_sum_sqr.sec;
8456         *(ptr++) = stats->execution_time_sum_sqr.usec;
8457         *(ptr++) = stats->execution_time_min.sec;
8458         *(ptr++) = stats->execution_time_min.usec;
8459         *(ptr++) = stats->execution_time_max.sec;
8460         *(ptr++) = stats->execution_time_max.usec;
8461     }
8462     *ptrP = ptr;
8463 }
8464
8465 /*
8466  * rx_RetrieveProcessRPCStats - retrieve all of the rpc statistics for
8467  * this process
8468  *
8469  * PARAMETERS
8470  *
8471  * IN callerVersion - the rpc stat version of the caller
8472  *
8473  * OUT myVersion - the rpc stat version of this function
8474  *
8475  * OUT clock_sec - local time seconds
8476  *
8477  * OUT clock_usec - local time microseconds
8478  *
8479  * OUT allocSize - the number of bytes allocated to contain stats
8480  *
8481  * OUT statCount - the number stats retrieved from this process.
8482  *
8483  * OUT stats - the actual stats retrieved from this process.
8484  *
8485  * RETURN CODES
8486  *
8487  * Returns void.  If successful, stats will != NULL.
8488  */
8489
8490 int
8491 rx_RetrieveProcessRPCStats(afs_uint32 callerVersion, afs_uint32 * myVersion,
8492                            afs_uint32 * clock_sec, afs_uint32 * clock_usec,
8493                            size_t * allocSize, afs_uint32 * statCount,
8494                            afs_uint32 ** stats)
8495 {
8496     size_t space = 0;
8497     afs_uint32 *ptr;
8498     struct clock now;
8499     int rc = 0;
8500
8501     *stats = 0;
8502     *allocSize = 0;
8503     *statCount = 0;
8504     *myVersion = RX_STATS_RETRIEVAL_VERSION;
8505
8506     /*
8507      * Check to see if stats are enabled
8508      */
8509
8510     MUTEX_ENTER(&rx_rpc_stats);
8511     if (!rxi_monitor_processStats) {
8512         MUTEX_EXIT(&rx_rpc_stats);
8513         return rc;
8514     }
8515
8516     clock_GetTime(&now);
8517     *clock_sec = now.sec;
8518     *clock_usec = now.usec;
8519
8520     /*
8521      * Allocate the space based upon the caller version
8522      *
8523      * If the client is at an older version than we are,
8524      * we return the statistic data in the older data format, but
8525      * we still return our version number so the client knows we
8526      * are maintaining more data than it can retrieve.
8527      */
8528
8529     if (callerVersion >= RX_STATS_RETRIEVAL_FIRST_EDITION) {
8530         space = rxi_rpc_process_stat_cnt * sizeof(rx_function_entry_v1_t);
8531         *statCount = rxi_rpc_process_stat_cnt;
8532     } else {
8533         /*
8534          * This can't happen yet, but in the future version changes
8535          * can be handled by adding additional code here
8536          */
8537     }
8538
8539     if (space > (size_t) 0) {
8540         *allocSize = space;
8541         ptr = *stats = rxi_Alloc(space);
8542
8543         if (ptr != NULL) {
8544             rx_interface_stat_p rpc_stat, nrpc_stat;
8545
8546
8547             for (queue_Scan
8548                  (&processStats, rpc_stat, nrpc_stat, rx_interface_stat)) {
8549                 /*
8550                  * Copy the data based upon the caller version
8551                  */
8552                 rx_MarshallProcessRPCStats(callerVersion,
8553                                            rpc_stat->stats[0].func_total,
8554                                            rpc_stat->stats, &ptr);
8555             }
8556         } else {
8557             rc = ENOMEM;
8558         }
8559     }
8560     MUTEX_EXIT(&rx_rpc_stats);
8561     return rc;
8562 }
8563
8564 /*
8565  * rx_RetrievePeerRPCStats - retrieve all of the rpc statistics for the peers
8566  *
8567  * PARAMETERS
8568  *
8569  * IN callerVersion - the rpc stat version of the caller
8570  *
8571  * OUT myVersion - the rpc stat version of this function
8572  *
8573  * OUT clock_sec - local time seconds
8574  *
8575  * OUT clock_usec - local time microseconds
8576  *
8577  * OUT allocSize - the number of bytes allocated to contain stats
8578  *
8579  * OUT statCount - the number of stats retrieved from the individual
8580  * peer structures.
8581  *
8582  * OUT stats - the actual stats retrieved from the individual peer structures.
8583  *
8584  * RETURN CODES
8585  *
8586  * Returns void.  If successful, stats will != NULL.
8587  */
8588
8589 int
8590 rx_RetrievePeerRPCStats(afs_uint32 callerVersion, afs_uint32 * myVersion,
8591                         afs_uint32 * clock_sec, afs_uint32 * clock_usec,
8592                         size_t * allocSize, afs_uint32 * statCount,
8593                         afs_uint32 ** stats)
8594 {
8595     size_t space = 0;
8596     afs_uint32 *ptr;
8597     struct clock now;
8598     int rc = 0;
8599
8600     *stats = 0;
8601     *statCount = 0;
8602     *allocSize = 0;
8603     *myVersion = RX_STATS_RETRIEVAL_VERSION;
8604
8605     /*
8606      * Check to see if stats are enabled
8607      */
8608
8609     MUTEX_ENTER(&rx_rpc_stats);
8610     if (!rxi_monitor_peerStats) {
8611         MUTEX_EXIT(&rx_rpc_stats);
8612         return rc;
8613     }
8614
8615     clock_GetTime(&now);
8616     *clock_sec = now.sec;
8617     *clock_usec = now.usec;
8618
8619     /*
8620      * Allocate the space based upon the caller version
8621      *
8622      * If the client is at an older version than we are,
8623      * we return the statistic data in the older data format, but
8624      * we still return our version number so the client knows we
8625      * are maintaining more data than it can retrieve.
8626      */
8627
8628     if (callerVersion >= RX_STATS_RETRIEVAL_FIRST_EDITION) {
8629         space = rxi_rpc_peer_stat_cnt * sizeof(rx_function_entry_v1_t);
8630         *statCount = rxi_rpc_peer_stat_cnt;
8631     } else {
8632         /*
8633          * This can't happen yet, but in the future version changes
8634          * can be handled by adding additional code here
8635          */
8636     }
8637
8638     if (space > (size_t) 0) {
8639         *allocSize = space;
8640         ptr = *stats = rxi_Alloc(space);
8641
8642         if (ptr != NULL) {
8643             rx_interface_stat_p rpc_stat, nrpc_stat;
8644             char *fix_offset;
8645
8646             for (queue_Scan
8647                  (&peerStats, rpc_stat, nrpc_stat, rx_interface_stat)) {
8648                 /*
8649                  * We have to fix the offset of rpc_stat since we are
8650                  * keeping this structure on two rx_queues.  The rx_queue
8651                  * package assumes that the rx_queue member is the first
8652                  * member of the structure.  That is, rx_queue assumes that
8653                  * any one item is only on one queue at a time.  We are
8654                  * breaking that assumption and so we have to do a little
8655                  * math to fix our pointers.
8656                  */
8657
8658                 fix_offset = (char *)rpc_stat;
8659                 fix_offset -= offsetof(rx_interface_stat_t, all_peers);
8660                 rpc_stat = (rx_interface_stat_p) fix_offset;
8661
8662                 /*
8663                  * Copy the data based upon the caller version
8664                  */
8665                 rx_MarshallProcessRPCStats(callerVersion,
8666                                            rpc_stat->stats[0].func_total,
8667                                            rpc_stat->stats, &ptr);
8668             }
8669         } else {
8670             rc = ENOMEM;
8671         }
8672     }
8673     MUTEX_EXIT(&rx_rpc_stats);
8674     return rc;
8675 }
8676
8677 /*
8678  * rx_FreeRPCStats - free memory allocated by
8679  *                   rx_RetrieveProcessRPCStats and rx_RetrievePeerRPCStats
8680  *
8681  * PARAMETERS
8682  *
8683  * IN stats - stats previously returned by rx_RetrieveProcessRPCStats or
8684  * rx_RetrievePeerRPCStats
8685  *
8686  * IN allocSize - the number of bytes in stats.
8687  *
8688  * RETURN CODES
8689  *
8690  * Returns void.
8691  */
8692
8693 void
8694 rx_FreeRPCStats(afs_uint32 * stats, size_t allocSize)
8695 {
8696     rxi_Free(stats, allocSize);
8697 }
8698
8699 /*
8700  * rx_queryProcessRPCStats - see if process rpc stat collection is
8701  * currently enabled.
8702  *
8703  * PARAMETERS
8704  *
8705  * RETURN CODES
8706  *
8707  * Returns 0 if stats are not enabled != 0 otherwise
8708  */
8709
8710 int
8711 rx_queryProcessRPCStats(void)
8712 {
8713     int rc;
8714     MUTEX_ENTER(&rx_rpc_stats);
8715     rc = rxi_monitor_processStats;
8716     MUTEX_EXIT(&rx_rpc_stats);
8717     return rc;
8718 }
8719
8720 /*
8721  * rx_queryPeerRPCStats - see if peer stat collection is currently enabled.
8722  *
8723  * PARAMETERS
8724  *
8725  * RETURN CODES
8726  *
8727  * Returns 0 if stats are not enabled != 0 otherwise
8728  */
8729
8730 int
8731 rx_queryPeerRPCStats(void)
8732 {
8733     int rc;
8734     MUTEX_ENTER(&rx_rpc_stats);
8735     rc = rxi_monitor_peerStats;
8736     MUTEX_EXIT(&rx_rpc_stats);
8737     return rc;
8738 }
8739
8740 /*
8741  * rx_enableProcessRPCStats - begin rpc stat collection for entire process
8742  *
8743  * PARAMETERS
8744  *
8745  * RETURN CODES
8746  *
8747  * Returns void.
8748  */
8749
8750 void
8751 rx_enableProcessRPCStats(void)
8752 {
8753     MUTEX_ENTER(&rx_rpc_stats);
8754     rx_enable_stats = 1;
8755     rxi_monitor_processStats = 1;
8756     MUTEX_EXIT(&rx_rpc_stats);
8757 }
8758
8759 /*
8760  * rx_enablePeerRPCStats - begin rpc stat collection per peer structure
8761  *
8762  * PARAMETERS
8763  *
8764  * RETURN CODES
8765  *
8766  * Returns void.
8767  */
8768
8769 void
8770 rx_enablePeerRPCStats(void)
8771 {
8772     MUTEX_ENTER(&rx_rpc_stats);
8773     rx_enable_stats = 1;
8774     rxi_monitor_peerStats = 1;
8775     MUTEX_EXIT(&rx_rpc_stats);
8776 }
8777
8778 /*
8779  * rx_disableProcessRPCStats - stop rpc stat collection for entire process
8780  *
8781  * PARAMETERS
8782  *
8783  * RETURN CODES
8784  *
8785  * Returns void.
8786  */
8787
8788 void
8789 rx_disableProcessRPCStats(void)
8790 {
8791     rx_interface_stat_p rpc_stat, nrpc_stat;
8792     size_t space;
8793
8794     MUTEX_ENTER(&rx_rpc_stats);
8795
8796     /*
8797      * Turn off process statistics and if peer stats is also off, turn
8798      * off everything
8799      */
8800
8801     rxi_monitor_processStats = 0;
8802     if (rxi_monitor_peerStats == 0) {
8803         rx_enable_stats = 0;
8804     }
8805
8806     for (queue_Scan(&processStats, rpc_stat, nrpc_stat, rx_interface_stat)) {
8807         unsigned int num_funcs = 0;
8808         if (!rpc_stat)
8809             break;
8810         queue_Remove(rpc_stat);
8811         num_funcs = rpc_stat->stats[0].func_total;
8812         space =
8813             sizeof(rx_interface_stat_t) +
8814             rpc_stat->stats[0].func_total * sizeof(rx_function_entry_v1_t);
8815
8816         rxi_Free(rpc_stat, space);
8817         rxi_rpc_process_stat_cnt -= num_funcs;
8818     }
8819     MUTEX_EXIT(&rx_rpc_stats);
8820 }
8821
8822 /*
8823  * rx_disablePeerRPCStats - stop rpc stat collection for peers
8824  *
8825  * PARAMETERS
8826  *
8827  * RETURN CODES
8828  *
8829  * Returns void.
8830  */
8831
8832 void
8833 rx_disablePeerRPCStats(void)
8834 {
8835     struct rx_peer **peer_ptr, **peer_end;
8836     int code;
8837
8838     /*
8839      * Turn off peer statistics and if process stats is also off, turn
8840      * off everything
8841      */
8842
8843     rxi_monitor_peerStats = 0;
8844     if (rxi_monitor_processStats == 0) {
8845         rx_enable_stats = 0;
8846     }
8847
8848     for (peer_ptr = &rx_peerHashTable[0], peer_end =
8849          &rx_peerHashTable[rx_hashTableSize]; peer_ptr < peer_end;
8850          peer_ptr++) {
8851         struct rx_peer *peer, *next, *prev;
8852
8853         MUTEX_ENTER(&rx_peerHashTable_lock);
8854         MUTEX_ENTER(&rx_rpc_stats);
8855         for (prev = peer = *peer_ptr; peer; peer = next) {
8856             next = peer->next;
8857             code = MUTEX_TRYENTER(&peer->peer_lock);
8858             if (code) {
8859                 rx_interface_stat_p rpc_stat, nrpc_stat;
8860                 size_t space;
8861
8862                 if (prev == *peer_ptr) {
8863                     *peer_ptr = next;
8864                     prev = next;
8865                 } else
8866                     prev->next = next;
8867
8868                 if (next)
8869                     next->refCount++;
8870                 if (prev)
8871                     prev->refCount++;
8872                 peer->refCount++;
8873                 MUTEX_EXIT(&rx_peerHashTable_lock);
8874
8875                 for (queue_Scan
8876                      (&peer->rpcStats, rpc_stat, nrpc_stat,
8877                       rx_interface_stat)) {
8878                     unsigned int num_funcs = 0;
8879                     if (!rpc_stat)
8880                         break;
8881                     queue_Remove(&rpc_stat->queue_header);
8882                     queue_Remove(&rpc_stat->all_peers);
8883                     num_funcs = rpc_stat->stats[0].func_total;
8884                     space =
8885                         sizeof(rx_interface_stat_t) +
8886                         rpc_stat->stats[0].func_total *
8887                         sizeof(rx_function_entry_v1_t);
8888
8889                     rxi_Free(rpc_stat, space);
8890                     rxi_rpc_peer_stat_cnt -= num_funcs;
8891                 }
8892                 MUTEX_EXIT(&peer->peer_lock);
8893
8894                 MUTEX_ENTER(&rx_peerHashTable_lock);
8895                 if (next)
8896                     next->refCount--;
8897                 if (prev)
8898                     prev->refCount--;
8899                 peer->refCount--;
8900             } else {
8901                 prev = peer;
8902             }
8903         }
8904         MUTEX_EXIT(&rx_rpc_stats);
8905         MUTEX_EXIT(&rx_peerHashTable_lock);
8906     }
8907 }
8908
8909 /*
8910  * rx_clearProcessRPCStats - clear the contents of the rpc stats according
8911  * to clearFlag
8912  *
8913  * PARAMETERS
8914  *
8915  * IN clearFlag - flag indicating which stats to clear
8916  *
8917  * RETURN CODES
8918  *
8919  * Returns void.
8920  */
8921
8922 void
8923 rx_clearProcessRPCStats(afs_uint32 clearFlag)
8924 {
8925     rx_interface_stat_p rpc_stat, nrpc_stat;
8926
8927     MUTEX_ENTER(&rx_rpc_stats);
8928
8929     for (queue_Scan(&processStats, rpc_stat, nrpc_stat, rx_interface_stat)) {
8930         unsigned int num_funcs = 0, i;
8931         num_funcs = rpc_stat->stats[0].func_total;
8932         for (i = 0; i < num_funcs; i++) {
8933             if (clearFlag & AFS_RX_STATS_CLEAR_INVOCATIONS) {
8934                 hzero(rpc_stat->stats[i].invocations);
8935             }
8936             if (clearFlag & AFS_RX_STATS_CLEAR_BYTES_SENT) {
8937                 hzero(rpc_stat->stats[i].bytes_sent);
8938             }
8939             if (clearFlag & AFS_RX_STATS_CLEAR_BYTES_RCVD) {
8940                 hzero(rpc_stat->stats[i].bytes_rcvd);
8941             }
8942             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_SUM) {
8943                 rpc_stat->stats[i].queue_time_sum.sec = 0;
8944                 rpc_stat->stats[i].queue_time_sum.usec = 0;
8945             }
8946             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_SQUARE) {
8947                 rpc_stat->stats[i].queue_time_sum_sqr.sec = 0;
8948                 rpc_stat->stats[i].queue_time_sum_sqr.usec = 0;
8949             }
8950             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_MIN) {
8951                 rpc_stat->stats[i].queue_time_min.sec = 9999999;
8952                 rpc_stat->stats[i].queue_time_min.usec = 9999999;
8953             }
8954             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_MAX) {
8955                 rpc_stat->stats[i].queue_time_max.sec = 0;
8956                 rpc_stat->stats[i].queue_time_max.usec = 0;
8957             }
8958             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_SUM) {
8959                 rpc_stat->stats[i].execution_time_sum.sec = 0;
8960                 rpc_stat->stats[i].execution_time_sum.usec = 0;
8961             }
8962             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_SQUARE) {
8963                 rpc_stat->stats[i].execution_time_sum_sqr.sec = 0;
8964                 rpc_stat->stats[i].execution_time_sum_sqr.usec = 0;
8965             }
8966             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_MIN) {
8967                 rpc_stat->stats[i].execution_time_min.sec = 9999999;
8968                 rpc_stat->stats[i].execution_time_min.usec = 9999999;
8969             }
8970             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_MAX) {
8971                 rpc_stat->stats[i].execution_time_max.sec = 0;
8972                 rpc_stat->stats[i].execution_time_max.usec = 0;
8973             }
8974         }
8975     }
8976
8977     MUTEX_EXIT(&rx_rpc_stats);
8978 }
8979
8980 /*
8981  * rx_clearPeerRPCStats - clear the contents of the rpc stats according
8982  * to clearFlag
8983  *
8984  * PARAMETERS
8985  *
8986  * IN clearFlag - flag indicating which stats to clear
8987  *
8988  * RETURN CODES
8989  *
8990  * Returns void.
8991  */
8992
8993 void
8994 rx_clearPeerRPCStats(afs_uint32 clearFlag)
8995 {
8996     rx_interface_stat_p rpc_stat, nrpc_stat;
8997
8998     MUTEX_ENTER(&rx_rpc_stats);
8999
9000     for (queue_Scan(&peerStats, rpc_stat, nrpc_stat, rx_interface_stat)) {
9001         unsigned int num_funcs = 0, i;
9002         char *fix_offset;
9003         /*
9004          * We have to fix the offset of rpc_stat since we are
9005          * keeping this structure on two rx_queues.  The rx_queue
9006          * package assumes that the rx_queue member is the first
9007          * member of the structure.  That is, rx_queue assumes that
9008          * any one item is only on one queue at a time.  We are
9009          * breaking that assumption and so we have to do a little
9010          * math to fix our pointers.
9011          */
9012
9013         fix_offset = (char *)rpc_stat;
9014         fix_offset -= offsetof(rx_interface_stat_t, all_peers);
9015         rpc_stat = (rx_interface_stat_p) fix_offset;
9016
9017         num_funcs = rpc_stat->stats[0].func_total;
9018         for (i = 0; i < num_funcs; i++) {
9019             if (clearFlag & AFS_RX_STATS_CLEAR_INVOCATIONS) {
9020                 hzero(rpc_stat->stats[i].invocations);
9021             }
9022             if (clearFlag & AFS_RX_STATS_CLEAR_BYTES_SENT) {
9023                 hzero(rpc_stat->stats[i].bytes_sent);
9024             }
9025             if (clearFlag & AFS_RX_STATS_CLEAR_BYTES_RCVD) {
9026                 hzero(rpc_stat->stats[i].bytes_rcvd);
9027             }
9028             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_SUM) {
9029                 rpc_stat->stats[i].queue_time_sum.sec = 0;
9030                 rpc_stat->stats[i].queue_time_sum.usec = 0;
9031             }
9032             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_SQUARE) {
9033                 rpc_stat->stats[i].queue_time_sum_sqr.sec = 0;
9034                 rpc_stat->stats[i].queue_time_sum_sqr.usec = 0;
9035             }
9036             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_MIN) {
9037                 rpc_stat->stats[i].queue_time_min.sec = 9999999;
9038                 rpc_stat->stats[i].queue_time_min.usec = 9999999;
9039             }
9040             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_MAX) {
9041                 rpc_stat->stats[i].queue_time_max.sec = 0;
9042                 rpc_stat->stats[i].queue_time_max.usec = 0;
9043             }
9044             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_SUM) {
9045                 rpc_stat->stats[i].execution_time_sum.sec = 0;
9046                 rpc_stat->stats[i].execution_time_sum.usec = 0;
9047             }
9048             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_SQUARE) {
9049                 rpc_stat->stats[i].execution_time_sum_sqr.sec = 0;
9050                 rpc_stat->stats[i].execution_time_sum_sqr.usec = 0;
9051             }
9052             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_MIN) {
9053                 rpc_stat->stats[i].execution_time_min.sec = 9999999;
9054                 rpc_stat->stats[i].execution_time_min.usec = 9999999;
9055             }
9056             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_MAX) {
9057                 rpc_stat->stats[i].execution_time_max.sec = 0;
9058                 rpc_stat->stats[i].execution_time_max.usec = 0;
9059             }
9060         }
9061     }
9062
9063     MUTEX_EXIT(&rx_rpc_stats);
9064 }
9065
9066 /*
9067  * rxi_rxstat_userok points to a routine that returns 1 if the caller
9068  * is authorized to enable/disable/clear RX statistics.
9069  */
9070 static int (*rxi_rxstat_userok) (struct rx_call * call) = NULL;
9071
9072 void
9073 rx_SetRxStatUserOk(int (*proc) (struct rx_call * call))
9074 {
9075     rxi_rxstat_userok = proc;
9076 }
9077
9078 int
9079 rx_RxStatUserOk(struct rx_call *call)
9080 {
9081     if (!rxi_rxstat_userok)
9082         return 0;
9083     return rxi_rxstat_userok(call);
9084 }
9085
9086 #ifdef AFS_NT40_ENV
9087 /*
9088  * DllMain() -- Entry-point function called by the DllMainCRTStartup()
9089  *     function in the MSVC runtime DLL (msvcrt.dll).
9090  *
9091  *     Note: the system serializes calls to this function.
9092  */
9093 BOOL WINAPI
9094 DllMain(HINSTANCE dllInstHandle,        /* instance handle for this DLL module */
9095         DWORD reason,                   /* reason function is being called */
9096         LPVOID reserved)                /* reserved for future use */
9097 {
9098     switch (reason) {
9099     case DLL_PROCESS_ATTACH:
9100         /* library is being attached to a process */
9101         INIT_PTHREAD_LOCKS;
9102         return TRUE;
9103
9104     case DLL_PROCESS_DETACH:
9105         return TRUE;
9106
9107     default:
9108         return FALSE;
9109     }
9110 }
9111 #endif /* AFS_NT40_ENV */
9112
9113 #ifndef KERNEL
9114 int rx_DumpCalls(FILE *outputFile, char *cookie)
9115 {
9116 #ifdef RXDEBUG_PACKET
9117 #ifdef KDUMP_RX_LOCK
9118     struct rx_call_rx_lock *c;
9119 #else
9120     struct rx_call *c;
9121 #endif
9122 #ifdef AFS_NT40_ENV
9123     int zilch;
9124     char output[2048];
9125 #define RXDPRINTF sprintf
9126 #define RXDPRINTOUT output
9127 #else
9128 #define RXDPRINTF fprintf
9129 #define RXDPRINTOUT outputFile
9130 #endif
9131
9132     RXDPRINTF(RXDPRINTOUT, "%s - Start dumping all Rx Calls - count=%u\r\n", cookie, rx_stats.nCallStructs);
9133 #ifdef AFS_NT40_ENV
9134     WriteFile(outputFile, output, (DWORD)strlen(output), &zilch, NULL);
9135 #endif
9136
9137     for (c = rx_allCallsp; c; c = c->allNextp) {
9138         u_short rqc, tqc, iovqc;
9139         struct rx_packet *p, *np;
9140
9141         MUTEX_ENTER(&c->lock);
9142         queue_Count(&c->rq, p, np, rx_packet, rqc);
9143         queue_Count(&c->tq, p, np, rx_packet, tqc);
9144         queue_Count(&c->iovq, p, np, rx_packet, iovqc);
9145
9146         RXDPRINTF(RXDPRINTOUT, "%s - call=0x%p, id=%u, state=%u, mode=%u, conn=%p, epoch=%u, cid=%u, callNum=%u, connFlags=0x%x, flags=0x%x, "
9147                 "rqc=%u,%u, tqc=%u,%u, iovqc=%u,%u, "
9148                 "lstatus=%u, rstatus=%u, error=%d, timeout=%u, "
9149                 "resendEvent=%d, timeoutEvt=%d, keepAliveEvt=%d, delayedAckEvt=%d, delayedAbortEvt=%d, abortCode=%d, abortCount=%d, "
9150                 "lastSendTime=%u, lastRecvTime=%u, lastSendData=%u"
9151 #ifdef RX_ENABLE_LOCKS
9152                 ", refCount=%u"
9153 #endif
9154 #ifdef RX_REFCOUNT_CHECK
9155                 ", refCountBegin=%u, refCountResend=%u, refCountDelay=%u, "
9156                 "refCountAlive=%u, refCountPacket=%u, refCountSend=%u, refCountAckAll=%u, refCountAbort=%u"
9157 #endif
9158                 "\r\n",
9159                 cookie, c, c->call_id, (afs_uint32)c->state, (afs_uint32)c->mode, c->conn, c->conn?c->conn->epoch:0, c->conn?c->conn->cid:0,
9160                 c->callNumber?*c->callNumber:0, c->conn?c->conn->flags:0, c->flags,
9161                 (afs_uint32)c->rqc, (afs_uint32)rqc, (afs_uint32)c->tqc, (afs_uint32)tqc, (afs_uint32)c->iovqc, (afs_uint32)iovqc,
9162                 (afs_uint32)c->localStatus, (afs_uint32)c->remoteStatus, c->error, c->timeout,
9163                 c->resendEvent?1:0, c->timeoutEvent?1:0, c->keepAliveEvent?1:0, c->delayedAckEvent?1:0, c->delayedAbortEvent?1:0,
9164                 c->abortCode, c->abortCount, c->lastSendTime, c->lastReceiveTime, c->lastSendData
9165 #ifdef RX_ENABLE_LOCKS
9166                 , (afs_uint32)c->refCount
9167 #endif
9168 #ifdef RX_REFCOUNT_CHECK
9169                 , c->refCDebug[0],c->refCDebug[1],c->refCDebug[2],c->refCDebug[3],c->refCDebug[4],c->refCDebug[5],c->refCDebug[6],c->refCDebug[7]
9170 #endif
9171                 );
9172         MUTEX_EXIT(&c->lock);
9173
9174 #ifdef AFS_NT40_ENV
9175         WriteFile(outputFile, output, (DWORD)strlen(output), &zilch, NULL);
9176 #endif
9177     }
9178     RXDPRINTF(RXDPRINTOUT, "%s - End dumping all Rx Calls\r\n", cookie);
9179 #ifdef AFS_NT40_ENV
9180     WriteFile(outputFile, output, (DWORD)strlen(output), &zilch, NULL);
9181 #endif
9182 #endif /* RXDEBUG_PACKET */
9183     return 0;
9184 }
9185 #endif