9ce55368c872abb4e6de3bb9a2db4df343213e10
[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 = 0;
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
5569     ap->nAcks = offset;
5570     p->length = rx_AckDataSize(offset) + 4 * sizeof(afs_int32);
5571
5572     /* these are new for AFS 3.3 */
5573     templ = rxi_AdjustMaxMTU(call->conn->peer->ifMTU, rx_maxReceiveSize);
5574     templ = htonl(templ);
5575     rx_packetwrite(p, rx_AckDataSize(offset), sizeof(afs_int32), &templ);
5576     templ = htonl(call->conn->peer->ifMTU);
5577     rx_packetwrite(p, rx_AckDataSize(offset) + sizeof(afs_int32),
5578                    sizeof(afs_int32), &templ);
5579
5580     /* new for AFS 3.4 */
5581     templ = htonl(call->rwind);
5582     rx_packetwrite(p, rx_AckDataSize(offset) + 2 * sizeof(afs_int32),
5583                    sizeof(afs_int32), &templ);
5584
5585     /* new for AFS 3.5 */
5586     templ = htonl(call->conn->peer->ifDgramPackets);
5587     rx_packetwrite(p, rx_AckDataSize(offset) + 3 * sizeof(afs_int32),
5588                    sizeof(afs_int32), &templ);
5589
5590     p->header.serviceId = call->conn->serviceId;
5591     p->header.cid = (call->conn->cid | call->channel);
5592     p->header.callNumber = *call->callNumber;
5593     p->header.seq = 0;
5594     p->header.securityIndex = call->conn->securityIndex;
5595     p->header.epoch = call->conn->epoch;
5596     p->header.type = RX_PACKET_TYPE_ACK;
5597     p->header.flags = RX_SLOW_START_OK;
5598     if (reason == RX_ACK_PING) {
5599         p->header.flags |= RX_REQUEST_ACK;
5600 #ifdef ADAPT_WINDOW
5601         clock_GetTime(&call->pingRequestTime);
5602 #endif
5603         if (padbytes) {
5604             p->length = padbytes +
5605                 rx_AckDataSize(call->rwind) + 4 * sizeof(afs_int32);
5606
5607             while (padbytes--)
5608                 /* not fast but we can potentially use this if truncated
5609                  * fragments are delivered to figure out the mtu.
5610                  */
5611                 rx_packetwrite(p, rx_AckDataSize(offset) + 4 *
5612                                sizeof(afs_int32), sizeof(afs_int32),
5613                                &padbytes);
5614         }
5615     }
5616     if (call->conn->type == RX_CLIENT_CONNECTION)
5617         p->header.flags |= RX_CLIENT_INITIATED;
5618
5619 #ifdef RXDEBUG
5620 #ifdef AFS_NT40_ENV
5621     if (rxdebug_active) {
5622         char msg[512];
5623         size_t len;
5624
5625         len = _snprintf(msg, sizeof(msg),
5626                         "tid[%d] SACK: reason %s serial %u previous %u seq %u first %u acks %u space %u ",
5627                          GetCurrentThreadId(), rx_ack_reason(ap->reason),
5628                          ntohl(ap->serial), ntohl(ap->previousPacket),
5629                          (unsigned int)p->header.seq, ntohl(ap->firstPacket),
5630                          ap->nAcks, ntohs(ap->bufferSpace) );
5631         if (ap->nAcks) {
5632             int offset;
5633
5634             for (offset = 0; offset < ap->nAcks && len < sizeof(msg); offset++)
5635                 msg[len++] = (ap->acks[offset] == RX_ACK_TYPE_NACK ? '-' : '*');
5636         }
5637         msg[len++]='\n';
5638         msg[len] = '\0';
5639         OutputDebugString(msg);
5640     }
5641 #else /* AFS_NT40_ENV */
5642     if (rx_Log) {
5643         fprintf(rx_Log, "SACK: reason %x previous %u seq %u first %u ",
5644                 ap->reason, ntohl(ap->previousPacket),
5645                 (unsigned int)p->header.seq, ntohl(ap->firstPacket));
5646         if (ap->nAcks) {
5647             for (offset = 0; offset < ap->nAcks; offset++)
5648                 putc(ap->acks[offset] == RX_ACK_TYPE_NACK ? '-' : '*',
5649                      rx_Log);
5650         }
5651         putc('\n', rx_Log);
5652     }
5653 #endif /* AFS_NT40_ENV */
5654 #endif
5655     {
5656         int i, nbytes = p->length;
5657
5658         for (i = 1; i < p->niovecs; i++) {      /* vec 0 is ALWAYS header */
5659             if (nbytes <= p->wirevec[i].iov_len) {
5660                 int savelen, saven;
5661
5662                 savelen = p->wirevec[i].iov_len;
5663                 saven = p->niovecs;
5664                 p->wirevec[i].iov_len = nbytes;
5665                 p->niovecs = i + 1;
5666                 rxi_Send(call, p, istack);
5667                 p->wirevec[i].iov_len = savelen;
5668                 p->niovecs = saven;
5669                 break;
5670             } else
5671                 nbytes -= p->wirevec[i].iov_len;
5672         }
5673     }
5674     if (rx_stats_active)
5675         rx_atomic_inc(&rx_stats.ackPacketsSent);
5676 #ifndef RX_ENABLE_TSFPQ
5677     if (!optionalPacket)
5678         rxi_FreePacket(p);
5679 #endif
5680     return optionalPacket;      /* Return packet for re-use by caller */
5681 }
5682
5683 struct xmitlist {
5684    struct rx_packet **list;
5685    int len;
5686    int resending;
5687 };
5688
5689 /* Send all of the packets in the list in single datagram */
5690 static void
5691 rxi_SendList(struct rx_call *call, struct xmitlist *xmit,
5692              int istack, int moreFlag)
5693 {
5694     int i;
5695     int requestAck = 0;
5696     int lastPacket = 0;
5697     struct clock now;
5698     struct rx_connection *conn = call->conn;
5699     struct rx_peer *peer = conn->peer;
5700
5701     MUTEX_ENTER(&peer->peer_lock);
5702     peer->nSent += xmit->len;
5703     if (xmit->resending)
5704         peer->reSends += xmit->len;
5705     MUTEX_EXIT(&peer->peer_lock);
5706
5707     if (rx_stats_active) {
5708         if (xmit->resending)
5709             rx_atomic_add(&rx_stats.dataPacketsReSent, xmit->len);
5710         else
5711             rx_atomic_add(&rx_stats.dataPacketsSent, xmit->len);
5712     }
5713
5714     clock_GetTime(&now);
5715
5716     if (xmit->list[xmit->len - 1]->header.flags & RX_LAST_PACKET) {
5717         lastPacket = 1;
5718     }
5719
5720     /* Set the packet flags and schedule the resend events */
5721     /* Only request an ack for the last packet in the list */
5722     for (i = 0; i < xmit->len; i++) {
5723         struct rx_packet *packet = xmit->list[i];
5724
5725         /* Record the time sent */
5726         packet->timeSent = now;
5727         packet->flags |= RX_PKTFLAG_SENT;
5728
5729         /* Ask for an ack on retransmitted packets,  on every other packet
5730          * if the peer doesn't support slow start. Ask for an ack on every
5731          * packet until the congestion window reaches the ack rate. */
5732         if (packet->header.serial) {
5733             requestAck = 1;
5734         } else {
5735             packet->firstSent = now;
5736             if (!lastPacket && (call->cwind <= (u_short) (conn->ackRate + 1)
5737                                 || (!(call->flags & RX_CALL_SLOW_START_OK)
5738                                     && (packet->header.seq & 1)))) {
5739                 requestAck = 1;
5740             }
5741         }
5742
5743         /* Tag this packet as not being the last in this group,
5744          * for the receiver's benefit */
5745         if (i < xmit->len - 1 || moreFlag) {
5746             packet->header.flags |= RX_MORE_PACKETS;
5747         }
5748     }
5749
5750     if (requestAck) {
5751         xmit->list[xmit->len - 1]->header.flags |= RX_REQUEST_ACK;
5752     }
5753
5754     /* Since we're about to send a data packet to the peer, it's
5755      * safe to nuke any scheduled end-of-packets ack */
5756     rxevent_Cancel(call->delayedAckEvent, call, RX_CALL_REFCOUNT_DELAY);
5757
5758     MUTEX_EXIT(&call->lock);
5759     MUTEX_ENTER(&rx_refcnt_mutex);
5760     CALL_HOLD(call, RX_CALL_REFCOUNT_SEND);
5761     MUTEX_EXIT(&rx_refcnt_mutex);
5762     if (xmit->len > 1) {
5763         rxi_SendPacketList(call, conn, xmit->list, xmit->len, istack);
5764     } else {
5765         rxi_SendPacket(call, conn, xmit->list[0], istack);
5766     }
5767     MUTEX_ENTER(&call->lock);
5768     MUTEX_ENTER(&rx_refcnt_mutex);
5769     CALL_RELE(call, RX_CALL_REFCOUNT_SEND);
5770     MUTEX_EXIT(&rx_refcnt_mutex);
5771
5772     /* Tell the RTO calculation engine that we have sent a packet, and
5773      * if it was the last one */
5774     rxi_rto_packet_sent(call, lastPacket, istack);
5775
5776     /* Update last send time for this call (for keep-alive
5777      * processing), and for the connection (so that we can discover
5778      * idle connections) */
5779     conn->lastSendTime = call->lastSendTime = clock_Sec();
5780     /* Let a set of retransmits trigger an idle timeout */
5781     if (!xmit->resending)
5782         call->lastSendData = call->lastSendTime;
5783 }
5784
5785 /* When sending packets we need to follow these rules:
5786  * 1. Never send more than maxDgramPackets in a jumbogram.
5787  * 2. Never send a packet with more than two iovecs in a jumbogram.
5788  * 3. Never send a retransmitted packet in a jumbogram.
5789  * 4. Never send more than cwind/4 packets in a jumbogram
5790  * We always keep the last list we should have sent so we
5791  * can set the RX_MORE_PACKETS flags correctly.
5792  */
5793
5794 static void
5795 rxi_SendXmitList(struct rx_call *call, struct rx_packet **list, int len,
5796                  int istack)
5797 {
5798     int i;
5799     int recovery;
5800     struct xmitlist working;
5801     struct xmitlist last;
5802
5803     struct rx_peer *peer = call->conn->peer;
5804     int morePackets = 0;
5805
5806     memset(&last, 0, sizeof(struct xmitlist));
5807     working.list = &list[0];
5808     working.len = 0;
5809     working.resending = 0;
5810
5811     recovery = call->flags & RX_CALL_FAST_RECOVER;
5812
5813     for (i = 0; i < len; i++) {
5814         /* Does the current packet force us to flush the current list? */
5815         if (working.len > 0
5816             && (list[i]->header.serial || (list[i]->flags & RX_PKTFLAG_ACKED)
5817                 || list[i]->length > RX_JUMBOBUFFERSIZE)) {
5818
5819             /* This sends the 'last' list and then rolls the current working
5820              * set into the 'last' one, and resets the working set */
5821
5822             if (last.len > 0) {
5823                 rxi_SendList(call, &last, istack, 1);
5824                 /* If the call enters an error state stop sending, or if
5825                  * we entered congestion recovery mode, stop sending */
5826                 if (call->error
5827                     || (!recovery && (call->flags & RX_CALL_FAST_RECOVER)))
5828                     return;
5829             }
5830             last = working;
5831             working.len = 0;
5832             working.resending = 0;
5833             working.list = &list[i];
5834         }
5835         /* Add the current packet to the list if it hasn't been acked.
5836          * Otherwise adjust the list pointer to skip the current packet.  */
5837         if (!(list[i]->flags & RX_PKTFLAG_ACKED)) {
5838             working.len++;
5839
5840             if (list[i]->header.serial)
5841                 working.resending = 1;
5842
5843             /* Do we need to flush the list? */
5844             if (working.len >= (int)peer->maxDgramPackets
5845                 || working.len >= (int)call->nDgramPackets 
5846                 || working.len >= (int)call->cwind
5847                 || list[i]->header.serial
5848                 || list[i]->length != RX_JUMBOBUFFERSIZE) {
5849                 if (last.len > 0) {
5850                     rxi_SendList(call, &last, istack, 1);
5851                     /* If the call enters an error state stop sending, or if
5852                      * we entered congestion recovery mode, stop sending */
5853                     if (call->error
5854                         || (!recovery && (call->flags & RX_CALL_FAST_RECOVER)))
5855                         return;
5856                 }
5857                 last = working;
5858                 working.len = 0;
5859                 working.resending = 0;
5860                 working.list = &list[i + 1];
5861             }
5862         } else {
5863             if (working.len != 0) {
5864                 osi_Panic("rxi_SendList error");
5865             }
5866             working.list = &list[i + 1];
5867         }
5868     }
5869
5870     /* Send the whole list when the call is in receive mode, when
5871      * the call is in eof mode, when we are in fast recovery mode,
5872      * and when we have the last packet */
5873     if ((list[len - 1]->header.flags & RX_LAST_PACKET)
5874         || call->mode == RX_MODE_RECEIVING || call->mode == RX_MODE_EOF
5875         || (call->flags & RX_CALL_FAST_RECOVER)) {
5876         /* Check for the case where the current list contains
5877          * an acked packet. Since we always send retransmissions
5878          * in a separate packet, we only need to check the first
5879          * packet in the list */
5880         if (working.len > 0 && !(working.list[0]->flags & RX_PKTFLAG_ACKED)) {
5881             morePackets = 1;
5882         }
5883         if (last.len > 0) {
5884             rxi_SendList(call, &last, istack, morePackets);
5885             /* If the call enters an error state stop sending, or if
5886              * we entered congestion recovery mode, stop sending */
5887             if (call->error
5888                 || (!recovery && (call->flags & RX_CALL_FAST_RECOVER)))
5889                 return;
5890         }
5891         if (morePackets) {
5892             rxi_SendList(call, &working, istack, 0);
5893         }
5894     } else if (last.len > 0) {
5895         rxi_SendList(call, &last, istack, 0);
5896         /* Packets which are in 'working' are not sent by this call */
5897     }
5898 }
5899
5900 static void
5901 rxi_Resend(struct rxevent *event, void *arg0, void *arg1, int istack)
5902 {
5903     struct rx_call *call = arg0;
5904     struct rx_peer *peer;
5905     struct rx_packet *p, *nxp;
5906     struct clock maxTimeout = { 60, 0 };
5907
5908     MUTEX_ENTER(&call->lock);
5909
5910     peer = call->conn->peer;
5911
5912     /* Make sure that the event pointer is removed from the call
5913      * structure, since there is no longer a per-call retransmission
5914      * event pending. */
5915     if (event == call->resendEvent) {
5916         MUTEX_ENTER(&rx_refcnt_mutex);
5917         CALL_RELE(call, RX_CALL_REFCOUNT_RESEND);
5918         MUTEX_EXIT(&rx_refcnt_mutex);
5919         call->resendEvent = NULL;
5920     }
5921
5922     if (rxi_busyChannelError && (call->flags & RX_CALL_PEER_BUSY)) {
5923         rxi_CheckBusy(call);
5924     }
5925
5926     if (queue_IsEmpty(&call->tq)) {
5927         /* Nothing to do. This means that we've been raced, and that an
5928          * ACK has come in between when we were triggered, and when we
5929          * actually got to run. */
5930         goto out;
5931     }
5932
5933     /* We're in loss recovery */
5934     call->flags |= RX_CALL_FAST_RECOVER;
5935
5936     /* Mark all of the pending packets in the queue as being lost */
5937     for (queue_Scan(&call->tq, p, nxp, rx_packet)) {
5938         if (!(p->flags & RX_PKTFLAG_ACKED))
5939             p->flags &= ~RX_PKTFLAG_SENT;
5940     }
5941
5942     /* We're resending, so we double the timeout of the call. This will be
5943      * dropped back down by the first successful ACK that we receive.
5944      *
5945      * We apply a maximum value here of 60 seconds
5946      */
5947     clock_Add(&call->rto, &call->rto);
5948     if (clock_Gt(&call->rto, &maxTimeout))
5949         call->rto = maxTimeout;
5950
5951     /* Packet loss is most likely due to congestion, so drop our window size
5952      * and start again from the beginning */
5953     if (peer->maxDgramPackets >1) {
5954         call->MTU = RX_JUMBOBUFFERSIZE + RX_HEADER_SIZE;
5955         call->MTU = MIN(peer->natMTU, peer->maxMTU);
5956     }
5957     call->ssthresh = MAX(4, MIN((int)call->cwind, (int)call->twind)) >> 1;
5958     call->nDgramPackets = 1;
5959     call->cwind = 1;
5960     call->nextCwind = 1;
5961     call->nAcks = 0;
5962     call->nNacks = 0;
5963     MUTEX_ENTER(&peer->peer_lock);
5964     peer->MTU = call->MTU;
5965     peer->cwind = call->cwind;
5966     peer->nDgramPackets = 1;
5967     peer->congestSeq++;
5968     call->congestSeq = peer->congestSeq;
5969     MUTEX_EXIT(&peer->peer_lock);
5970
5971     rxi_Start(call, istack);
5972
5973 out:
5974     MUTEX_EXIT(&call->lock);
5975 }
5976
5977 /* This routine is called when new packets are readied for
5978  * transmission and when retransmission may be necessary, or when the
5979  * transmission window or burst count are favourable.  This should be
5980  * better optimized for new packets, the usual case, now that we've
5981  * got rid of queues of send packets. XXXXXXXXXXX */
5982 void
5983 rxi_Start(struct rx_call *call, int istack)
5984 {
5985
5986     struct rx_packet *p;
5987     struct rx_packet *nxp;      /* Next pointer for queue_Scan */
5988     int nXmitPackets;
5989     int maxXmitPackets;
5990
5991     if (call->error) {
5992 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
5993         if (rx_stats_active)
5994             rx_atomic_inc(&rx_tq_debug.rxi_start_in_error);
5995 #endif
5996         return;
5997     }
5998
5999     if (queue_IsNotEmpty(&call->tq)) {  /* If we have anything to send */
6000
6001         /* Send (or resend) any packets that need it, subject to
6002          * window restrictions and congestion burst control
6003          * restrictions.  Ask for an ack on the last packet sent in
6004          * this burst.  For now, we're relying upon the window being
6005          * considerably bigger than the largest number of packets that
6006          * are typically sent at once by one initial call to
6007          * rxi_Start.  This is probably bogus (perhaps we should ask
6008          * for an ack when we're half way through the current
6009          * window?).  Also, for non file transfer applications, this
6010          * may end up asking for an ack for every packet.  Bogus. XXXX
6011          */
6012         /*
6013          * But check whether we're here recursively, and let the other guy
6014          * do the work.
6015          */
6016 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
6017         if (!(call->flags & RX_CALL_TQ_BUSY)) {
6018             call->flags |= RX_CALL_TQ_BUSY;
6019             do {
6020 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
6021             restart:
6022 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
6023                 call->flags &= ~RX_CALL_NEED_START;
6024 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
6025                 nXmitPackets = 0;
6026                 maxXmitPackets = MIN(call->twind, call->cwind);
6027                 for (queue_Scan(&call->tq, p, nxp, rx_packet)) {
6028 #ifdef RX_TRACK_PACKETS
6029                     if ((p->flags & RX_PKTFLAG_FREE)
6030                         || (!queue_IsEnd(&call->tq, nxp)
6031                             && (nxp->flags & RX_PKTFLAG_FREE))
6032                         || (p == (struct rx_packet *)&rx_freePacketQueue)
6033                         || (nxp == (struct rx_packet *)&rx_freePacketQueue)) {
6034                         osi_Panic("rxi_Start: xmit queue clobbered");
6035                     }
6036 #endif
6037                     if (p->flags & RX_PKTFLAG_ACKED) {
6038                         /* Since we may block, don't trust this */
6039                         if (rx_stats_active)
6040                             rx_atomic_inc(&rx_stats.ignoreAckedPacket);
6041                         continue;       /* Ignore this packet if it has been acknowledged */
6042                     }
6043
6044                     /* Turn off all flags except these ones, which are the same
6045                      * on each transmission */
6046                     p->header.flags &= RX_PRESET_FLAGS;
6047
6048                     if (p->header.seq >=
6049                         call->tfirst + MIN((int)call->twind,
6050                                            (int)(call->nSoftAcked +
6051                                                  call->cwind))) {
6052                         call->flags |= RX_CALL_WAIT_WINDOW_SEND;        /* Wait for transmit window */
6053                         /* Note: if we're waiting for more window space, we can
6054                          * still send retransmits; hence we don't return here, but
6055                          * break out to schedule a retransmit event */
6056                         dpf(("call %d waiting for window (seq %d, twind %d, nSoftAcked %d, cwind %d)\n",
6057                              *(call->callNumber), p->header.seq, call->twind, call->nSoftAcked,
6058                              call->cwind));
6059                         break;
6060                     }
6061
6062                     /* Transmit the packet if it needs to be sent. */
6063                     if (!(p->flags & RX_PKTFLAG_SENT)) {
6064                         if (nXmitPackets == maxXmitPackets) {
6065                             rxi_SendXmitList(call, call->xmitList,
6066                                              nXmitPackets, istack);
6067                             goto restart;
6068                         }
6069                         dpf(("call %d xmit packet %"AFS_PTR_FMT"\n",
6070                               *(call->callNumber), p));
6071                         call->xmitList[nXmitPackets++] = p;
6072                     }
6073                 }
6074
6075                 /* xmitList now hold pointers to all of the packets that are
6076                  * ready to send. Now we loop to send the packets */
6077                 if (nXmitPackets > 0) {
6078                     rxi_SendXmitList(call, call->xmitList, nXmitPackets,
6079                                      istack);
6080                 }
6081
6082 #ifdef  AFS_GLOBAL_RXLOCK_KERNEL
6083                 if (call->error) {
6084                     /* We went into the error state while sending packets. Now is
6085                      * the time to reset the call. This will also inform the using
6086                      * process that the call is in an error state.
6087                      */
6088                     if (rx_stats_active)
6089                         rx_atomic_inc(&rx_tq_debug.rxi_start_aborted);
6090                     call->flags &= ~RX_CALL_TQ_BUSY;
6091                     rxi_WakeUpTransmitQueue(call);
6092                     rxi_CallError(call, call->error);
6093                     return;
6094                 }
6095 #ifdef RX_ENABLE_LOCKS
6096                 if (call->flags & RX_CALL_TQ_SOME_ACKED) {
6097                     int missing;
6098                     call->flags &= ~RX_CALL_TQ_SOME_ACKED;
6099                     /* Some packets have received acks. If they all have, we can clear
6100                      * the transmit queue.
6101                      */
6102                     for (missing =
6103                          0, queue_Scan(&call->tq, p, nxp, rx_packet)) {
6104                         if (p->header.seq < call->tfirst
6105                             && (p->flags & RX_PKTFLAG_ACKED)) {
6106                             queue_Remove(p);
6107 #ifdef RX_TRACK_PACKETS
6108                             p->flags &= ~RX_PKTFLAG_TQ;
6109 #endif
6110 #ifdef RXDEBUG_PACKET
6111                             call->tqc--;
6112 #endif
6113                             rxi_FreePacket(p);
6114                         } else
6115                             missing = 1;
6116                     }
6117                     if (!missing)
6118                         call->flags |= RX_CALL_TQ_CLEARME;
6119                 }
6120 #endif /* RX_ENABLE_LOCKS */
6121                 if (call->flags & RX_CALL_TQ_CLEARME)
6122                     rxi_ClearTransmitQueue(call, 1);
6123             } while (call->flags & RX_CALL_NEED_START);
6124             /*
6125              * TQ references no longer protected by this flag; they must remain
6126              * protected by the global lock.
6127              */
6128             call->flags &= ~RX_CALL_TQ_BUSY;
6129             rxi_WakeUpTransmitQueue(call);
6130         } else {
6131             call->flags |= RX_CALL_NEED_START;
6132         }
6133 #endif /* AFS_GLOBAL_RXLOCK_KERNEL */
6134     } else {
6135         rxi_rto_cancel(call);
6136     }
6137 }
6138
6139 /* Also adjusts the keep alive parameters for the call, to reflect
6140  * that we have just sent a packet (so keep alives aren't sent
6141  * immediately) */
6142 void
6143 rxi_Send(struct rx_call *call, struct rx_packet *p,
6144          int istack)
6145 {
6146     struct rx_connection *conn = call->conn;
6147
6148     /* Stamp each packet with the user supplied status */
6149     p->header.userStatus = call->localStatus;
6150
6151     /* Allow the security object controlling this call's security to
6152      * make any last-minute changes to the packet */
6153     RXS_SendPacket(conn->securityObject, call, p);
6154
6155     /* Since we're about to send SOME sort of packet to the peer, it's
6156      * safe to nuke any scheduled end-of-packets ack */
6157     rxevent_Cancel(call->delayedAckEvent, call, RX_CALL_REFCOUNT_DELAY);
6158
6159     /* Actually send the packet, filling in more connection-specific fields */
6160     MUTEX_EXIT(&call->lock);
6161     MUTEX_ENTER(&rx_refcnt_mutex);
6162     CALL_HOLD(call, RX_CALL_REFCOUNT_SEND);
6163     MUTEX_EXIT(&rx_refcnt_mutex);
6164     rxi_SendPacket(call, conn, p, istack);
6165     MUTEX_ENTER(&rx_refcnt_mutex);
6166     CALL_RELE(call, RX_CALL_REFCOUNT_SEND);
6167     MUTEX_EXIT(&rx_refcnt_mutex);
6168     MUTEX_ENTER(&call->lock);
6169
6170     /* Update last send time for this call (for keep-alive
6171      * processing), and for the connection (so that we can discover
6172      * idle connections) */
6173     if ((p->header.type != RX_PACKET_TYPE_ACK) ||
6174         (((struct rx_ackPacket *)rx_DataOf(p))->reason == RX_ACK_PING) ||
6175         (p->length <= (rx_AckDataSize(call->rwind) + 4 * sizeof(afs_int32))))
6176     {
6177         conn->lastSendTime = call->lastSendTime = clock_Sec();
6178         /* Don't count keepalive ping/acks here, so idleness can be tracked. */
6179         if ((p->header.type != RX_PACKET_TYPE_ACK) ||
6180             ((((struct rx_ackPacket *)rx_DataOf(p))->reason != RX_ACK_PING) &&
6181              (((struct rx_ackPacket *)rx_DataOf(p))->reason !=
6182               RX_ACK_PING_RESPONSE)))
6183             call->lastSendData = call->lastSendTime;
6184     }
6185 }
6186
6187 /* Check if a call needs to be destroyed.  Called by keep-alive code to ensure
6188  * that things are fine.  Also called periodically to guarantee that nothing
6189  * falls through the cracks (e.g. (error + dally) connections have keepalive
6190  * turned off.  Returns 0 if conn is well, -1 otherwise.  If otherwise, call
6191  *  may be freed!
6192  * haveCTLock Set if calling from rxi_ReapConnections
6193  */
6194 #ifdef RX_ENABLE_LOCKS
6195 int
6196 rxi_CheckCall(struct rx_call *call, int haveCTLock)
6197 #else /* RX_ENABLE_LOCKS */
6198 int
6199 rxi_CheckCall(struct rx_call *call)
6200 #endif                          /* RX_ENABLE_LOCKS */
6201 {
6202     struct rx_connection *conn = call->conn;
6203     afs_uint32 now;
6204     afs_uint32 deadTime, idleDeadTime = 0, hardDeadTime = 0;
6205     afs_uint32 fudgeFactor;
6206     int cerror = 0;
6207     int newmtu = 0;
6208
6209 #ifdef AFS_GLOBAL_RXLOCK_KERNEL
6210     if (call->flags & RX_CALL_TQ_BUSY) {
6211         /* Call is active and will be reset by rxi_Start if it's
6212          * in an error state.
6213          */
6214         return 0;
6215     }
6216 #endif
6217     /* RTT + 8*MDEV, rounded up to the next second. */
6218     fudgeFactor = (((afs_uint32) call->rtt >> 3) +
6219                    ((afs_uint32) call->rtt_dev << 1) + 1023) >> 10;
6220
6221     deadTime = conn->secondsUntilDead + fudgeFactor;
6222     now = clock_Sec();
6223     /* These are computed to the second (+- 1 second).  But that's
6224      * good enough for these values, which should be a significant
6225      * number of seconds. */
6226     if (now > (call->lastReceiveTime + deadTime)) {
6227         if (call->state == RX_STATE_ACTIVE) {
6228 #ifdef ADAPT_PMTU
6229 #if defined(KERNEL) && defined(AFS_SUN5_ENV)
6230             ire_t *ire;
6231 #if defined(AFS_SUN510_ENV) && defined(GLOBAL_NETSTACKID)
6232             netstack_t *ns =  netstack_find_by_stackid(GLOBAL_NETSTACKID);
6233             ip_stack_t *ipst = ns->netstack_ip;
6234 #endif
6235             ire = ire_cache_lookup(conn->peer->host
6236 #if defined(AFS_SUN510_ENV) && defined(ALL_ZONES)
6237                                    , ALL_ZONES
6238 #if defined(AFS_SUN510_ENV) && (defined(ICL_3_ARG) || defined(GLOBAL_NETSTACKID))
6239                                    , NULL
6240 #if defined(AFS_SUN510_ENV) && defined(GLOBAL_NETSTACKID)
6241                                    , ipst
6242 #endif
6243 #endif
6244 #endif
6245                 );
6246
6247             if (ire && ire->ire_max_frag > 0)
6248                 rxi_SetPeerMtu(NULL, conn->peer->host, 0,
6249                                ire->ire_max_frag);
6250 #if defined(GLOBAL_NETSTACKID)
6251             netstack_rele(ns);
6252 #endif
6253 #endif
6254 #endif /* ADAPT_PMTU */
6255             cerror = RX_CALL_DEAD;
6256             goto mtuout;
6257         } else {
6258 #ifdef RX_ENABLE_LOCKS
6259             /* Cancel pending events */
6260             rxevent_Cancel(call->delayedAckEvent, call,
6261                            RX_CALL_REFCOUNT_DELAY);
6262             rxi_rto_cancel(call);
6263             rxevent_Cancel(call->keepAliveEvent, call,
6264                            RX_CALL_REFCOUNT_ALIVE);
6265             if (call->growMTUEvent)
6266                 rxevent_Cancel(call->growMTUEvent, call,
6267                                RX_CALL_REFCOUNT_ALIVE);
6268             MUTEX_ENTER(&rx_refcnt_mutex);
6269             if (call->refCount == 0) {
6270                 rxi_FreeCall(call, haveCTLock);
6271                 MUTEX_EXIT(&rx_refcnt_mutex);
6272                 return -2;
6273             }
6274             MUTEX_EXIT(&rx_refcnt_mutex);
6275             return -1;
6276 #else /* RX_ENABLE_LOCKS */
6277             rxi_FreeCall(call, 0);
6278             return -2;
6279 #endif /* RX_ENABLE_LOCKS */
6280         }
6281         /* Non-active calls are destroyed if they are not responding
6282          * to pings; active calls are simply flagged in error, so the
6283          * attached process can die reasonably gracefully. */
6284     }
6285
6286     if (conn->idleDeadTime) {
6287         idleDeadTime = conn->idleDeadTime + fudgeFactor;
6288     }
6289
6290     /* see if we have a non-activity timeout */
6291     if (call->startWait && idleDeadTime
6292         && ((call->startWait + idleDeadTime) < now) &&
6293         (call->flags & RX_CALL_READER_WAIT)) {
6294         if (call->state == RX_STATE_ACTIVE) {
6295             cerror = RX_CALL_TIMEOUT;
6296             goto mtuout;
6297         }
6298     }
6299     if (call->lastSendData && idleDeadTime && (conn->idleDeadErr != 0)
6300         && ((call->lastSendData + idleDeadTime) < now)) {
6301         if (call->state == RX_STATE_ACTIVE) {
6302             cerror = conn->idleDeadErr;
6303             goto mtuout;
6304         }
6305     }
6306
6307     if (conn->hardDeadTime) {
6308         hardDeadTime = conn->hardDeadTime + fudgeFactor;
6309     }
6310
6311     /* see if we have a hard timeout */
6312     if (hardDeadTime
6313         && (now > (hardDeadTime + call->startTime.sec))) {
6314         if (call->state == RX_STATE_ACTIVE)
6315             rxi_CallError(call, RX_CALL_TIMEOUT);
6316         return -1;
6317     }
6318     return 0;
6319 mtuout:
6320     if (conn->msgsizeRetryErr && cerror != RX_CALL_TIMEOUT
6321         && call->lastReceiveTime) {
6322         int oldMTU = conn->peer->ifMTU;
6323
6324         /* if we thought we could send more, perhaps things got worse */
6325         if (conn->peer->maxPacketSize > conn->lastPacketSize)
6326             /* maxpacketsize will be cleared in rxi_SetPeerMtu */
6327             newmtu = MAX(conn->peer->maxPacketSize-RX_IPUDP_SIZE,
6328                          conn->lastPacketSize-(128+RX_IPUDP_SIZE));
6329         else
6330             newmtu = conn->lastPacketSize-(128+RX_IPUDP_SIZE);
6331
6332         /* minimum capped in SetPeerMtu */
6333         rxi_SetPeerMtu(conn->peer, 0, 0, newmtu);
6334
6335         /* clean up */
6336         conn->lastPacketSize = 0;
6337
6338         /* needed so ResetCall doesn't clobber us. */
6339         call->MTU = conn->peer->ifMTU;
6340
6341         /* if we never succeeded, let the error pass out as-is */
6342         if (conn->peer->maxPacketSize && oldMTU != conn->peer->ifMTU)
6343             cerror = conn->msgsizeRetryErr;
6344
6345     }
6346     rxi_CallError(call, cerror);
6347     return -1;
6348 }
6349
6350 void
6351 rxi_NatKeepAliveEvent(struct rxevent *event, void *arg1, void *dummy)
6352 {
6353     struct rx_connection *conn = arg1;
6354     struct rx_header theader;
6355     char tbuffer[1 + sizeof(struct rx_header)];
6356     struct sockaddr_in taddr;
6357     char *tp;
6358     char a[1] = { 0 };
6359     struct iovec tmpiov[2];
6360     osi_socket socket =
6361         (conn->type ==
6362          RX_CLIENT_CONNECTION ? rx_socket : conn->service->socket);
6363
6364
6365     tp = &tbuffer[sizeof(struct rx_header)];
6366     taddr.sin_family = AF_INET;
6367     taddr.sin_port = rx_PortOf(rx_PeerOf(conn));
6368     taddr.sin_addr.s_addr = rx_HostOf(rx_PeerOf(conn));
6369 #ifdef STRUCT_SOCKADDR_HAS_SA_LEN
6370     taddr.sin_len = sizeof(struct sockaddr_in);
6371 #endif
6372     memset(&theader, 0, sizeof(theader));
6373     theader.epoch = htonl(999);
6374     theader.cid = 0;
6375     theader.callNumber = 0;
6376     theader.seq = 0;
6377     theader.serial = 0;
6378     theader.type = RX_PACKET_TYPE_VERSION;
6379     theader.flags = RX_LAST_PACKET;
6380     theader.serviceId = 0;
6381
6382     memcpy(tbuffer, &theader, sizeof(theader));
6383     memcpy(tp, &a, sizeof(a));
6384     tmpiov[0].iov_base = tbuffer;
6385     tmpiov[0].iov_len = 1 + sizeof(struct rx_header);
6386
6387     osi_NetSend(socket, &taddr, tmpiov, 1, 1 + sizeof(struct rx_header), 1);
6388
6389     MUTEX_ENTER(&conn->conn_data_lock);
6390     MUTEX_ENTER(&rx_refcnt_mutex);
6391     /* Only reschedule ourselves if the connection would not be destroyed */
6392     if (conn->refCount <= 1) {
6393         conn->natKeepAliveEvent = NULL;
6394         MUTEX_EXIT(&rx_refcnt_mutex);
6395         MUTEX_EXIT(&conn->conn_data_lock);
6396         rx_DestroyConnection(conn); /* drop the reference for this */
6397     } else {
6398         conn->refCount--; /* drop the reference for this */
6399         MUTEX_EXIT(&rx_refcnt_mutex);
6400         conn->natKeepAliveEvent = NULL;
6401         rxi_ScheduleNatKeepAliveEvent(conn);
6402         MUTEX_EXIT(&conn->conn_data_lock);
6403     }
6404 }
6405
6406 void
6407 rxi_ScheduleNatKeepAliveEvent(struct rx_connection *conn)
6408 {
6409     if (!conn->natKeepAliveEvent && conn->secondsUntilNatPing) {
6410         struct clock when, now;
6411         clock_GetTime(&now);
6412         when = now;
6413         when.sec += conn->secondsUntilNatPing;
6414         MUTEX_ENTER(&rx_refcnt_mutex);
6415         conn->refCount++; /* hold a reference for this */
6416         MUTEX_EXIT(&rx_refcnt_mutex);
6417         conn->natKeepAliveEvent =
6418             rxevent_PostNow(&when, &now, rxi_NatKeepAliveEvent, conn, 0);
6419     }
6420 }
6421
6422 void
6423 rx_SetConnSecondsUntilNatPing(struct rx_connection *conn, afs_int32 seconds)
6424 {
6425     MUTEX_ENTER(&conn->conn_data_lock);
6426     conn->secondsUntilNatPing = seconds;
6427     if (seconds != 0) {
6428         if (!(conn->flags & RX_CONN_ATTACHWAIT))
6429             rxi_ScheduleNatKeepAliveEvent(conn);
6430         else
6431             conn->flags |= RX_CONN_NAT_PING;
6432     }
6433     MUTEX_EXIT(&conn->conn_data_lock);
6434 }
6435
6436 void
6437 rxi_NatKeepAliveOn(struct rx_connection *conn)
6438 {
6439     MUTEX_ENTER(&conn->conn_data_lock);
6440     /* if it's already attached */
6441     if (!(conn->flags & RX_CONN_ATTACHWAIT))
6442         rxi_ScheduleNatKeepAliveEvent(conn);
6443     else
6444         conn->flags |= RX_CONN_NAT_PING;
6445     MUTEX_EXIT(&conn->conn_data_lock);
6446 }
6447
6448 /* When a call is in progress, this routine is called occasionally to
6449  * make sure that some traffic has arrived (or been sent to) the peer.
6450  * If nothing has arrived in a reasonable amount of time, the call is
6451  * declared dead; if nothing has been sent for a while, we send a
6452  * keep-alive packet (if we're actually trying to keep the call alive)
6453  */
6454 void
6455 rxi_KeepAliveEvent(struct rxevent *event, void *arg1, void *dummy)
6456 {
6457     struct rx_call *call = arg1;
6458     struct rx_connection *conn;
6459     afs_uint32 now;
6460
6461     MUTEX_ENTER(&rx_refcnt_mutex);
6462     CALL_RELE(call, RX_CALL_REFCOUNT_ALIVE);
6463     MUTEX_EXIT(&rx_refcnt_mutex);
6464     MUTEX_ENTER(&call->lock);
6465     if (event == call->keepAliveEvent)
6466         call->keepAliveEvent = NULL;
6467     now = clock_Sec();
6468
6469 #ifdef RX_ENABLE_LOCKS
6470     if (rxi_CheckCall(call, 0)) {
6471         MUTEX_EXIT(&call->lock);
6472         return;
6473     }
6474 #else /* RX_ENABLE_LOCKS */
6475     if (rxi_CheckCall(call))
6476         return;
6477 #endif /* RX_ENABLE_LOCKS */
6478
6479     /* Don't try to keep alive dallying calls */
6480     if (call->state == RX_STATE_DALLY) {
6481         MUTEX_EXIT(&call->lock);
6482         return;
6483     }
6484
6485     conn = call->conn;
6486     if ((now - call->lastSendTime) > conn->secondsUntilPing) {
6487         /* Don't try to send keepalives if there is unacknowledged data */
6488         /* the rexmit code should be good enough, this little hack
6489          * doesn't quite work XXX */
6490         (void)rxi_SendAck(call, NULL, 0, RX_ACK_PING, 0);
6491     }
6492     rxi_ScheduleKeepAliveEvent(call);
6493     MUTEX_EXIT(&call->lock);
6494 }
6495
6496 /* Does what's on the nameplate. */
6497 void
6498 rxi_GrowMTUEvent(struct rxevent *event, void *arg1, void *dummy)
6499 {
6500     struct rx_call *call = arg1;
6501     struct rx_connection *conn;
6502
6503     MUTEX_ENTER(&rx_refcnt_mutex);
6504     CALL_RELE(call, RX_CALL_REFCOUNT_ALIVE);
6505     MUTEX_EXIT(&rx_refcnt_mutex);
6506     MUTEX_ENTER(&call->lock);
6507
6508     if (event == call->growMTUEvent)
6509         call->growMTUEvent = NULL;
6510
6511 #ifdef RX_ENABLE_LOCKS
6512     if (rxi_CheckCall(call, 0)) {
6513         MUTEX_EXIT(&call->lock);
6514         return;
6515     }
6516 #else /* RX_ENABLE_LOCKS */
6517     if (rxi_CheckCall(call))
6518         return;
6519 #endif /* RX_ENABLE_LOCKS */
6520
6521     /* Don't bother with dallying calls */
6522     if (call->state == RX_STATE_DALLY) {
6523         MUTEX_EXIT(&call->lock);
6524         return;
6525     }
6526
6527     conn = call->conn;
6528
6529     /*
6530      * keep being scheduled, just don't do anything if we're at peak,
6531      * or we're not set up to be properly handled (idle timeout required)
6532      */
6533     if ((conn->peer->maxPacketSize != 0) &&
6534         (conn->peer->natMTU < RX_MAX_PACKET_SIZE) &&
6535         (conn->idleDeadErr))
6536         (void)rxi_SendAck(call, NULL, 0, RX_ACK_MTU, 0);
6537     rxi_ScheduleGrowMTUEvent(call, 0);
6538     MUTEX_EXIT(&call->lock);
6539 }
6540
6541 void
6542 rxi_ScheduleKeepAliveEvent(struct rx_call *call)
6543 {
6544     if (!call->keepAliveEvent) {
6545         struct clock when, now;
6546         clock_GetTime(&now);
6547         when = now;
6548         when.sec += call->conn->secondsUntilPing;
6549         MUTEX_ENTER(&rx_refcnt_mutex);
6550         CALL_HOLD(call, RX_CALL_REFCOUNT_ALIVE);
6551         MUTEX_EXIT(&rx_refcnt_mutex);
6552         call->keepAliveEvent =
6553             rxevent_PostNow(&when, &now, rxi_KeepAliveEvent, call, 0);
6554     }
6555 }
6556
6557 void
6558 rxi_ScheduleGrowMTUEvent(struct rx_call *call, int secs)
6559 {
6560     if (!call->growMTUEvent) {
6561         struct clock when, now;
6562
6563         clock_GetTime(&now);
6564         when = now;
6565         if (!secs) {
6566             if (call->conn->secondsUntilPing)
6567                 secs = (6*call->conn->secondsUntilPing)-1;
6568
6569             if (call->conn->secondsUntilDead)
6570                 secs = MIN(secs, (call->conn->secondsUntilDead-1));
6571         }
6572
6573         when.sec += secs;
6574         MUTEX_ENTER(&rx_refcnt_mutex);
6575         CALL_HOLD(call, RX_CALL_REFCOUNT_ALIVE);
6576         MUTEX_EXIT(&rx_refcnt_mutex);
6577         call->growMTUEvent =
6578             rxevent_PostNow(&when, &now, rxi_GrowMTUEvent, call, 0);
6579     }
6580 }
6581
6582 /* N.B. rxi_KeepAliveOff:  is defined earlier as a macro */
6583 void
6584 rxi_KeepAliveOn(struct rx_call *call)
6585 {
6586     /* Pretend last packet received was received now--i.e. if another
6587      * packet isn't received within the keep alive time, then the call
6588      * will die; Initialize last send time to the current time--even
6589      * if a packet hasn't been sent yet.  This will guarantee that a
6590      * keep-alive is sent within the ping time */
6591     call->lastReceiveTime = call->lastSendTime = clock_Sec();
6592     rxi_ScheduleKeepAliveEvent(call);
6593 }
6594
6595 void
6596 rxi_GrowMTUOn(struct rx_call *call)
6597 {
6598     struct rx_connection *conn = call->conn;
6599     MUTEX_ENTER(&conn->conn_data_lock);
6600     conn->lastPingSizeSer = conn->lastPingSize = 0;
6601     MUTEX_EXIT(&conn->conn_data_lock);
6602     rxi_ScheduleGrowMTUEvent(call, 1);
6603 }
6604
6605 /* This routine is called to send connection abort messages
6606  * that have been delayed to throttle looping clients. */
6607 void
6608 rxi_SendDelayedConnAbort(struct rxevent *event,
6609                          void *arg1, void *unused)
6610 {
6611     struct rx_connection *conn = arg1;
6612
6613     afs_int32 error;
6614     struct rx_packet *packet;
6615
6616     MUTEX_ENTER(&conn->conn_data_lock);
6617     conn->delayedAbortEvent = NULL;
6618     error = htonl(conn->error);
6619     conn->abortCount++;
6620     MUTEX_EXIT(&conn->conn_data_lock);
6621     packet = rxi_AllocPacket(RX_PACKET_CLASS_SPECIAL);
6622     if (packet) {
6623         packet =
6624             rxi_SendSpecial((struct rx_call *)0, conn, packet,
6625                             RX_PACKET_TYPE_ABORT, (char *)&error,
6626                             sizeof(error), 0);
6627         rxi_FreePacket(packet);
6628     }
6629 }
6630
6631 /* This routine is called to send call abort messages
6632  * that have been delayed to throttle looping clients. */
6633 void
6634 rxi_SendDelayedCallAbort(struct rxevent *event,
6635                          void *arg1, void *dummy)
6636 {
6637     struct rx_call *call = arg1;
6638
6639     afs_int32 error;
6640     struct rx_packet *packet;
6641
6642     MUTEX_ENTER(&call->lock);
6643     call->delayedAbortEvent = NULL;
6644     error = htonl(call->error);
6645     call->abortCount++;
6646     packet = rxi_AllocPacket(RX_PACKET_CLASS_SPECIAL);
6647     if (packet) {
6648         packet =
6649             rxi_SendSpecial(call, call->conn, packet, RX_PACKET_TYPE_ABORT,
6650                             (char *)&error, sizeof(error), 0);
6651         rxi_FreePacket(packet);
6652     }
6653     MUTEX_EXIT(&call->lock);
6654     MUTEX_ENTER(&rx_refcnt_mutex);
6655     CALL_RELE(call, RX_CALL_REFCOUNT_ABORT);
6656     MUTEX_EXIT(&rx_refcnt_mutex);
6657 }
6658
6659 /* This routine is called periodically (every RX_AUTH_REQUEST_TIMEOUT
6660  * seconds) to ask the client to authenticate itself.  The routine
6661  * issues a challenge to the client, which is obtained from the
6662  * security object associated with the connection */
6663 void
6664 rxi_ChallengeEvent(struct rxevent *event,
6665                    void *arg0, void *arg1, int tries)
6666 {
6667     struct rx_connection *conn = arg0;
6668
6669     conn->challengeEvent = NULL;
6670     if (RXS_CheckAuthentication(conn->securityObject, conn) != 0) {
6671         struct rx_packet *packet;
6672         struct clock when, now;
6673
6674         if (tries <= 0) {
6675             /* We've failed to authenticate for too long.
6676              * Reset any calls waiting for authentication;
6677              * they are all in RX_STATE_PRECALL.
6678              */
6679             int i;
6680
6681             MUTEX_ENTER(&conn->conn_call_lock);
6682             for (i = 0; i < RX_MAXCALLS; i++) {
6683                 struct rx_call *call = conn->call[i];
6684                 if (call) {
6685                     MUTEX_ENTER(&call->lock);
6686                     if (call->state == RX_STATE_PRECALL) {
6687                         rxi_CallError(call, RX_CALL_DEAD);
6688                         rxi_SendCallAbort(call, NULL, 0, 0);
6689                     }
6690                     MUTEX_EXIT(&call->lock);
6691                 }
6692             }
6693             MUTEX_EXIT(&conn->conn_call_lock);
6694             return;
6695         }
6696
6697         packet = rxi_AllocPacket(RX_PACKET_CLASS_SPECIAL);
6698         if (packet) {
6699             /* If there's no packet available, do this later. */
6700             RXS_GetChallenge(conn->securityObject, conn, packet);
6701             rxi_SendSpecial((struct rx_call *)0, conn, packet,
6702                             RX_PACKET_TYPE_CHALLENGE, NULL, -1, 0);
6703             rxi_FreePacket(packet);
6704         }
6705         clock_GetTime(&now);
6706         when = now;
6707         when.sec += RX_CHALLENGE_TIMEOUT;
6708         conn->challengeEvent =
6709             rxevent_PostNow2(&when, &now, rxi_ChallengeEvent, conn, 0,
6710                          (tries - 1));
6711     }
6712 }
6713
6714 /* Call this routine to start requesting the client to authenticate
6715  * itself.  This will continue until authentication is established,
6716  * the call times out, or an invalid response is returned.  The
6717  * security object associated with the connection is asked to create
6718  * the challenge at this time.  N.B.  rxi_ChallengeOff is a macro,
6719  * defined earlier. */
6720 void
6721 rxi_ChallengeOn(struct rx_connection *conn)
6722 {
6723     if (!conn->challengeEvent) {
6724         RXS_CreateChallenge(conn->securityObject, conn);
6725         rxi_ChallengeEvent(NULL, conn, 0, RX_CHALLENGE_MAXTRIES);
6726     };
6727 }
6728
6729
6730 /* rxi_ComputeRoundTripTime is called with peer locked. */
6731 /* peer may be null */
6732 static void
6733 rxi_ComputeRoundTripTime(struct rx_packet *p,
6734                          struct rx_ackPacket *ack,
6735                          struct rx_call *call,
6736                          struct rx_peer *peer,
6737                          struct clock *now)
6738 {
6739     struct clock thisRtt, *sentp;
6740     int rtt_timeout;
6741     int serial;
6742
6743     /* If the ACK is delayed, then do nothing */
6744     if (ack->reason == RX_ACK_DELAY)
6745         return;
6746
6747     /* On the wire, jumbograms are a single UDP packet. We shouldn't count
6748      * their RTT multiple times, so only include the RTT of the last packet
6749      * in a jumbogram */
6750     if (p->flags & RX_JUMBO_PACKET)
6751         return;
6752
6753     /* Use the serial number to determine which transmission the ACK is for,
6754      * and set the sent time to match this. If we have no serial number, then
6755      * only use the ACK for RTT calculations if the packet has not been
6756      * retransmitted
6757      */
6758
6759     serial = ntohl(ack->serial);
6760     if (serial) {
6761         if (serial == p->header.serial) {
6762             sentp = &p->timeSent;
6763         } else if (serial == p->firstSerial) {
6764             sentp = &p->firstSent;
6765         } else if (clock_Eq(&p->timeSent, &p->firstSent)) {
6766             sentp = &p->firstSent;
6767         } else
6768             return;
6769     } else {
6770         if (clock_Eq(&p->timeSent, &p->firstSent)) {
6771             sentp = &p->firstSent;
6772         } else
6773             return;
6774     }
6775
6776     thisRtt = *now;
6777
6778     if (clock_Lt(&thisRtt, sentp))
6779         return;                 /* somebody set the clock back, don't count this time. */
6780
6781     clock_Sub(&thisRtt, sentp);
6782     dpf(("rxi_ComputeRoundTripTime(call=%d packet=%"AFS_PTR_FMT" rttp=%d.%06d sec)\n",
6783           p->header.callNumber, p, thisRtt.sec, thisRtt.usec));
6784
6785     if (clock_IsZero(&thisRtt)) {
6786         /*
6787          * The actual round trip time is shorter than the
6788          * clock_GetTime resolution.  It is most likely 1ms or 100ns.
6789          * Since we can't tell which at the moment we will assume 1ms.
6790          */
6791         thisRtt.usec = 1000;
6792     }
6793
6794     if (rx_stats_active) {
6795         MUTEX_ENTER(&rx_stats_mutex);
6796         if (clock_Lt(&thisRtt, &rx_stats.minRtt))
6797             rx_stats.minRtt = thisRtt;
6798         if (clock_Gt(&thisRtt, &rx_stats.maxRtt)) {
6799             if (thisRtt.sec > 60) {
6800                 MUTEX_EXIT(&rx_stats_mutex);
6801                 return;         /* somebody set the clock ahead */
6802             }
6803             rx_stats.maxRtt = thisRtt;
6804         }
6805         clock_Add(&rx_stats.totalRtt, &thisRtt);
6806         rx_atomic_inc(&rx_stats.nRttSamples);
6807         MUTEX_EXIT(&rx_stats_mutex);
6808     }
6809
6810     /* better rtt calculation courtesy of UMich crew (dave,larry,peter,?) */
6811
6812     /* Apply VanJacobson round-trip estimations */
6813     if (call->rtt) {
6814         int delta;
6815
6816         /*
6817          * srtt (call->rtt) is in units of one-eighth-milliseconds.
6818          * srtt is stored as fixed point with 3 bits after the binary
6819          * point (i.e., scaled by 8). The following magic is
6820          * equivalent to the smoothing algorithm in rfc793 with an
6821          * alpha of .875 (srtt' = rtt/8 + srtt*7/8 in fixed point).
6822          * srtt'*8 = rtt + srtt*7
6823          * srtt'*8 = srtt*8 + rtt - srtt
6824          * srtt' = srtt + rtt/8 - srtt/8
6825          * srtt' = srtt + (rtt - srtt)/8
6826          */
6827
6828         delta = _8THMSEC(&thisRtt) - call->rtt;
6829         call->rtt += (delta >> 3);
6830
6831         /*
6832          * We accumulate a smoothed rtt variance (actually, a smoothed
6833          * mean difference), then set the retransmit timer to smoothed
6834          * rtt + 4 times the smoothed variance (was 2x in van's original
6835          * paper, but 4x works better for me, and apparently for him as
6836          * well).
6837          * rttvar is stored as
6838          * fixed point with 2 bits after the binary point (scaled by
6839          * 4).  The following is equivalent to rfc793 smoothing with
6840          * an alpha of .75 (rttvar' = rttvar*3/4 + |delta| / 4).
6841          *   rttvar'*4 = rttvar*3 + |delta|
6842          *   rttvar'*4 = rttvar*4 + |delta| - rttvar
6843          *   rttvar' = rttvar + |delta|/4 - rttvar/4
6844          *   rttvar' = rttvar + (|delta| - rttvar)/4
6845          * This replaces rfc793's wired-in beta.
6846          * dev*4 = dev*4 + (|actual - expected| - dev)
6847          */
6848
6849         if (delta < 0)
6850             delta = -delta;
6851
6852         delta -= (call->rtt_dev << 1);
6853         call->rtt_dev += (delta >> 3);
6854     } else {
6855         /* I don't have a stored RTT so I start with this value.  Since I'm
6856          * probably just starting a call, and will be pushing more data down
6857          * this, I expect congestion to increase rapidly.  So I fudge a
6858          * little, and I set deviance to half the rtt.  In practice,
6859          * deviance tends to approach something a little less than
6860          * half the smoothed rtt. */
6861         call->rtt = _8THMSEC(&thisRtt) + 8;
6862         call->rtt_dev = call->rtt >> 2; /* rtt/2: they're scaled differently */
6863     }
6864     /* the smoothed RTT time is RTT + 4*MDEV
6865      *
6866      * We allow a user specified minimum to be set for this, to allow clamping
6867      * at a minimum value in the same way as TCP. In addition, we have to allow
6868      * for the possibility that this packet is answered by a delayed ACK, so we
6869      * add on a fixed 200ms to account for that timer expiring.
6870      */
6871
6872     rtt_timeout = MAX(((call->rtt >> 3) + call->rtt_dev),
6873                       rx_minPeerTimeout) + 200;
6874     clock_Zero(&call->rto);
6875     clock_Addmsec(&call->rto, rtt_timeout);
6876
6877     /* Update the peer, so any new calls start with our values */
6878     peer->rtt_dev = call->rtt_dev;
6879     peer->rtt = call->rtt;
6880
6881     dpf(("rxi_ComputeRoundTripTime(call=%d packet=%"AFS_PTR_FMT" rtt=%d ms, srtt=%d ms, rtt_dev=%d ms, timeout=%d.%06d sec)\n",
6882           p->header.callNumber, p, MSEC(&thisRtt), call->rtt >> 3, call->rtt_dev >> 2, (call->rto.sec), (call->rto.usec)));
6883 }
6884
6885
6886 /* Find all server connections that have not been active for a long time, and
6887  * toss them */
6888 void
6889 rxi_ReapConnections(struct rxevent *unused, void *unused1, void *unused2)
6890 {
6891     struct clock now, when;
6892     clock_GetTime(&now);
6893
6894     /* Find server connection structures that haven't been used for
6895      * greater than rx_idleConnectionTime */
6896     {
6897         struct rx_connection **conn_ptr, **conn_end;
6898         int i, havecalls = 0;
6899         MUTEX_ENTER(&rx_connHashTable_lock);
6900         for (conn_ptr = &rx_connHashTable[0], conn_end =
6901              &rx_connHashTable[rx_hashTableSize]; conn_ptr < conn_end;
6902              conn_ptr++) {
6903             struct rx_connection *conn, *next;
6904             struct rx_call *call;
6905             int result;
6906
6907           rereap:
6908             for (conn = *conn_ptr; conn; conn = next) {
6909                 /* XXX -- Shouldn't the connection be locked? */
6910                 next = conn->next;
6911                 havecalls = 0;
6912                 for (i = 0; i < RX_MAXCALLS; i++) {
6913                     call = conn->call[i];
6914                     if (call) {
6915                         int code;
6916                         havecalls = 1;
6917                         code = MUTEX_TRYENTER(&call->lock);
6918                         if (!code)
6919                             continue;
6920 #ifdef RX_ENABLE_LOCKS
6921                         result = rxi_CheckCall(call, 1);
6922 #else /* RX_ENABLE_LOCKS */
6923                         result = rxi_CheckCall(call);
6924 #endif /* RX_ENABLE_LOCKS */
6925                         MUTEX_EXIT(&call->lock);
6926                         if (result == -2) {
6927                             /* If CheckCall freed the call, it might
6928                              * have destroyed  the connection as well,
6929                              * which screws up the linked lists.
6930                              */
6931                             goto rereap;
6932                         }
6933                     }
6934                 }
6935                 if (conn->type == RX_SERVER_CONNECTION) {
6936                     /* This only actually destroys the connection if
6937                      * there are no outstanding calls */
6938                     MUTEX_ENTER(&conn->conn_data_lock);
6939                     MUTEX_ENTER(&rx_refcnt_mutex);
6940                     if (!havecalls && !conn->refCount
6941                         && ((conn->lastSendTime + rx_idleConnectionTime) <
6942                             now.sec)) {
6943                         conn->refCount++;       /* it will be decr in rx_DestroyConn */
6944                         MUTEX_EXIT(&rx_refcnt_mutex);
6945                         MUTEX_EXIT(&conn->conn_data_lock);
6946 #ifdef RX_ENABLE_LOCKS
6947                         rxi_DestroyConnectionNoLock(conn);
6948 #else /* RX_ENABLE_LOCKS */
6949                         rxi_DestroyConnection(conn);
6950 #endif /* RX_ENABLE_LOCKS */
6951                     }
6952 #ifdef RX_ENABLE_LOCKS
6953                     else {
6954                         MUTEX_EXIT(&rx_refcnt_mutex);
6955                         MUTEX_EXIT(&conn->conn_data_lock);
6956                     }
6957 #endif /* RX_ENABLE_LOCKS */
6958                 }
6959             }
6960         }
6961 #ifdef RX_ENABLE_LOCKS
6962         while (rx_connCleanup_list) {
6963             struct rx_connection *conn;
6964             conn = rx_connCleanup_list;
6965             rx_connCleanup_list = rx_connCleanup_list->next;
6966             MUTEX_EXIT(&rx_connHashTable_lock);
6967             rxi_CleanupConnection(conn);
6968             MUTEX_ENTER(&rx_connHashTable_lock);
6969         }
6970         MUTEX_EXIT(&rx_connHashTable_lock);
6971 #endif /* RX_ENABLE_LOCKS */
6972     }
6973
6974     /* Find any peer structures that haven't been used (haven't had an
6975      * associated connection) for greater than rx_idlePeerTime */
6976     {
6977         struct rx_peer **peer_ptr, **peer_end;
6978         int code;
6979
6980         /*
6981          * Why do we need to hold the rx_peerHashTable_lock across
6982          * the incrementing of peer_ptr since the rx_peerHashTable
6983          * array is not changing?  We don't.
6984          *
6985          * By dropping the lock periodically we can permit other
6986          * activities to be performed while a rxi_ReapConnections
6987          * call is in progress.  The goal of reap connections
6988          * is to clean up quickly without causing large amounts
6989          * of contention.  Therefore, it is important that global
6990          * mutexes not be held for extended periods of time.
6991          */
6992         for (peer_ptr = &rx_peerHashTable[0], peer_end =
6993              &rx_peerHashTable[rx_hashTableSize]; peer_ptr < peer_end;
6994              peer_ptr++) {
6995             struct rx_peer *peer, *next, *prev;
6996
6997             MUTEX_ENTER(&rx_peerHashTable_lock);
6998             for (prev = peer = *peer_ptr; peer; peer = next) {
6999                 next = peer->next;
7000                 code = MUTEX_TRYENTER(&peer->peer_lock);
7001                 if ((code) && (peer->refCount == 0)
7002                     && ((peer->idleWhen + rx_idlePeerTime) < now.sec)) {
7003                     rx_interface_stat_p rpc_stat, nrpc_stat;
7004                     size_t space;
7005
7006                     /*
7007                      * now know that this peer object is one to be
7008                      * removed from the hash table.  Once it is removed
7009                      * it can't be referenced by other threads.
7010                      * Lets remove it first and decrement the struct
7011                      * nPeerStructs count.
7012                      */
7013                     if (peer == *peer_ptr) {
7014                         *peer_ptr = next;
7015                         prev = next;
7016                     } else
7017                         prev->next = next;
7018
7019                     if (rx_stats_active)
7020                         rx_atomic_dec(&rx_stats.nPeerStructs);
7021
7022                     /*
7023                      * Now if we hold references on 'prev' and 'next'
7024                      * we can safely drop the rx_peerHashTable_lock
7025                      * while we destroy this 'peer' object.
7026                      */
7027                     if (next)
7028                         next->refCount++;
7029                     if (prev)
7030                         prev->refCount++;
7031                     MUTEX_EXIT(&rx_peerHashTable_lock);
7032
7033                     MUTEX_EXIT(&peer->peer_lock);
7034                     MUTEX_DESTROY(&peer->peer_lock);
7035                     for (queue_Scan
7036                          (&peer->rpcStats, rpc_stat, nrpc_stat,
7037                           rx_interface_stat)) {
7038                         unsigned int num_funcs;
7039                         if (!rpc_stat)
7040                             break;
7041                         queue_Remove(&rpc_stat->queue_header);
7042                         queue_Remove(&rpc_stat->all_peers);
7043                         num_funcs = rpc_stat->stats[0].func_total;
7044                         space =
7045                             sizeof(rx_interface_stat_t) +
7046                             rpc_stat->stats[0].func_total *
7047                             sizeof(rx_function_entry_v1_t);
7048
7049                         rxi_Free(rpc_stat, space);
7050
7051                         MUTEX_ENTER(&rx_rpc_stats);
7052                         rxi_rpc_peer_stat_cnt -= num_funcs;
7053                         MUTEX_EXIT(&rx_rpc_stats);
7054                     }
7055                     rxi_FreePeer(peer);
7056
7057                     /*
7058                      * Regain the rx_peerHashTable_lock and
7059                      * decrement the reference count on 'prev'
7060                      * and 'next'.
7061                      */
7062                     MUTEX_ENTER(&rx_peerHashTable_lock);
7063                     if (next)
7064                         next->refCount--;
7065                     if (prev)
7066                         prev->refCount--;
7067                 } else {
7068                     if (code) {
7069                         MUTEX_EXIT(&peer->peer_lock);
7070                     }
7071                     prev = peer;
7072                 }
7073             }
7074             MUTEX_EXIT(&rx_peerHashTable_lock);
7075         }
7076     }
7077
7078     /* THIS HACK IS A TEMPORARY HACK.  The idea is that the race condition in
7079      * rxi_AllocSendPacket, if it hits, will be handled at the next conn
7080      * GC, just below.  Really, we shouldn't have to keep moving packets from
7081      * one place to another, but instead ought to always know if we can
7082      * afford to hold onto a packet in its particular use.  */
7083     MUTEX_ENTER(&rx_freePktQ_lock);
7084     if (rx_waitingForPackets) {
7085         rx_waitingForPackets = 0;
7086 #ifdef  RX_ENABLE_LOCKS
7087         CV_BROADCAST(&rx_waitingForPackets_cv);
7088 #else
7089         osi_rxWakeup(&rx_waitingForPackets);
7090 #endif
7091     }
7092     MUTEX_EXIT(&rx_freePktQ_lock);
7093
7094     when = now;
7095     when.sec += RX_REAP_TIME;   /* Check every RX_REAP_TIME seconds */
7096     rxevent_Post(&when, rxi_ReapConnections, 0, 0);
7097 }
7098
7099
7100 /* rxs_Release - This isn't strictly necessary but, since the macro name from
7101  * rx.h is sort of strange this is better.  This is called with a security
7102  * object before it is discarded.  Each connection using a security object has
7103  * its own refcount to the object so it won't actually be freed until the last
7104  * connection is destroyed.
7105  *
7106  * This is the only rxs module call.  A hold could also be written but no one
7107  * needs it. */
7108
7109 int
7110 rxs_Release(struct rx_securityClass *aobj)
7111 {
7112     return RXS_Close(aobj);
7113 }
7114
7115 #ifdef ADAPT_WINDOW
7116 #define RXRATE_PKT_OH   (RX_HEADER_SIZE + RX_IPUDP_SIZE)
7117 #define RXRATE_SMALL_PKT    (RXRATE_PKT_OH + sizeof(struct rx_ackPacket))
7118 #define RXRATE_AVG_SMALL_PKT    (RXRATE_PKT_OH + (sizeof(struct rx_ackPacket)/2))
7119 #define RXRATE_LARGE_PKT    (RXRATE_SMALL_PKT + 256)
7120
7121 /* Adjust our estimate of the transmission rate to this peer, given
7122  * that the packet p was just acked. We can adjust peer->timeout and
7123  * call->twind. Pragmatically, this is called
7124  * only with packets of maximal length.
7125  * Called with peer and call locked.
7126  */
7127
7128 static void
7129 rxi_ComputeRate(struct rx_peer *peer, struct rx_call *call,
7130                 struct rx_packet *p, struct rx_packet *ackp, u_char ackReason)
7131 {
7132     afs_int32 xferSize, xferMs;
7133     afs_int32 minTime;
7134     struct clock newTO;
7135
7136     /* Count down packets */
7137     if (peer->rateFlag > 0)
7138         peer->rateFlag--;
7139     /* Do nothing until we're enabled */
7140     if (peer->rateFlag != 0)
7141         return;
7142     if (!call->conn)
7143         return;
7144
7145     /* Count only when the ack seems legitimate */
7146     switch (ackReason) {
7147     case RX_ACK_REQUESTED:
7148         xferSize =
7149             p->length + RX_HEADER_SIZE + call->conn->securityMaxTrailerSize;
7150         xferMs = call->rtt;
7151         break;
7152
7153     case RX_ACK_PING_RESPONSE:
7154         if (p)                  /* want the response to ping-request, not data send */
7155             return;
7156         clock_GetTime(&newTO);
7157         if (clock_Gt(&newTO, &call->pingRequestTime)) {
7158             clock_Sub(&newTO, &call->pingRequestTime);
7159             xferMs = (newTO.sec * 1000) + (newTO.usec / 1000);
7160         } else {
7161             return;
7162         }
7163         xferSize = rx_AckDataSize(rx_maxSendWindow) + RX_HEADER_SIZE;
7164         break;
7165
7166     default:
7167         return;
7168     }
7169
7170     dpf(("CONG peer %lx/%u: sample (%s) size %ld, %ld ms (to %d.%06d, rtt %u, ps %u)\n",
7171           ntohl(peer->host), ntohs(peer->port), (ackReason == RX_ACK_REQUESTED ? "dataack" : "pingack"),
7172           xferSize, xferMs, peer->timeout.sec, peer->timeout.usec, peer->smRtt, peer->ifMTU));
7173
7174     /* Track only packets that are big enough. */
7175     if ((p->length + RX_HEADER_SIZE + call->conn->securityMaxTrailerSize) <
7176         peer->ifMTU)
7177         return;
7178
7179     /* absorb RTT data (in milliseconds) for these big packets */
7180     if (peer->smRtt == 0) {
7181         peer->smRtt = xferMs;
7182     } else {
7183         peer->smRtt = ((peer->smRtt * 15) + xferMs + 4) >> 4;
7184         if (!peer->smRtt)
7185             peer->smRtt = 1;
7186     }
7187
7188     if (peer->countDown) {
7189         peer->countDown--;
7190         return;
7191     }
7192     peer->countDown = 10;       /* recalculate only every so often */
7193
7194     /* In practice, we can measure only the RTT for full packets,
7195      * because of the way Rx acks the data that it receives.  (If it's
7196      * smaller than a full packet, it often gets implicitly acked
7197      * either by the call response (from a server) or by the next call
7198      * (from a client), and either case confuses transmission times
7199      * with processing times.)  Therefore, replace the above
7200      * more-sophisticated processing with a simpler version, where the
7201      * smoothed RTT is kept for full-size packets, and the time to
7202      * transmit a windowful of full-size packets is simply RTT *
7203      * windowSize. Again, we take two steps:
7204      - ensure the timeout is large enough for a single packet's RTT;
7205      - ensure that the window is small enough to fit in the desired timeout.*/
7206
7207     /* First, the timeout check. */
7208     minTime = peer->smRtt;
7209     /* Get a reasonable estimate for a timeout period */
7210     minTime += minTime;
7211     newTO.sec = minTime / 1000;
7212     newTO.usec = (minTime - (newTO.sec * 1000)) * 1000;
7213
7214     /* Increase the timeout period so that we can always do at least
7215      * one packet exchange */
7216     if (clock_Gt(&newTO, &peer->timeout)) {
7217
7218         dpf(("CONG peer %lx/%u: timeout %d.%06d ==> %ld.%06d (rtt %u)\n",
7219               ntohl(peer->host), ntohs(peer->port), peer->timeout.sec, peer->timeout.usec,
7220               newTO.sec, newTO.usec, peer->smRtt));
7221
7222         peer->timeout = newTO;
7223     }
7224
7225     /* Now, get an estimate for the transmit window size. */
7226     minTime = peer->timeout.sec * 1000 + (peer->timeout.usec / 1000);
7227     /* Now, convert to the number of full packets that could fit in a
7228      * reasonable fraction of that interval */
7229     minTime /= (peer->smRtt << 1);
7230     minTime = MAX(minTime, rx_minPeerTimeout);
7231     xferSize = minTime;         /* (make a copy) */
7232
7233     /* Now clamp the size to reasonable bounds. */
7234     if (minTime <= 1)
7235         minTime = 1;
7236     else if (minTime > rx_maxSendWindow)
7237         minTime = rx_maxSendWindow;
7238 /*    if (minTime != peer->maxWindow) {
7239       dpf(("CONG peer %lx/%u: windowsize %lu ==> %lu (to %lu.%06lu, rtt %u)\n",
7240              ntohl(peer->host), ntohs(peer->port), peer->maxWindow, minTime,
7241              peer->timeout.sec, peer->timeout.usec, peer->smRtt));
7242       peer->maxWindow = minTime;
7243         elide... call->twind = minTime;
7244     }
7245 */
7246
7247     /* Cut back on the peer timeout if it had earlier grown unreasonably.
7248      * Discern this by calculating the timeout necessary for rx_Window
7249      * packets. */
7250     if ((xferSize > rx_maxSendWindow) && (peer->timeout.sec >= 3)) {
7251         /* calculate estimate for transmission interval in milliseconds */
7252         minTime = rx_maxSendWindow * peer->smRtt;
7253         if (minTime < 1000) {
7254             dpf(("CONG peer %lx/%u: cut TO %d.%06d by 0.5 (rtt %u)\n",
7255                  ntohl(peer->host), ntohs(peer->port), peer->timeout.sec,
7256                  peer->timeout.usec, peer->smRtt));
7257
7258             newTO.sec = 0;      /* cut back on timeout by half a second */
7259             newTO.usec = 500000;
7260             clock_Sub(&peer->timeout, &newTO);
7261         }
7262     }
7263
7264     return;
7265 }                               /* end of rxi_ComputeRate */
7266 #endif /* ADAPT_WINDOW */
7267
7268
7269 void
7270 rxi_DebugInit(void)
7271 {
7272 #ifdef RXDEBUG
7273 #ifdef AFS_NT40_ENV
7274 #define TRACE_OPTION_RX_DEBUG 16
7275     HKEY parmKey;
7276     DWORD dummyLen;
7277     DWORD TraceOption;
7278     long code;
7279
7280     rxdebug_active = 0;
7281
7282     code = RegOpenKeyEx(HKEY_LOCAL_MACHINE, AFSREG_CLT_SVC_PARAM_SUBKEY,
7283                          0, KEY_QUERY_VALUE, &parmKey);
7284     if (code != ERROR_SUCCESS)
7285         return;
7286
7287     dummyLen = sizeof(TraceOption);
7288     code = RegQueryValueEx(parmKey, "TraceOption", NULL, NULL,
7289                            (BYTE *) &TraceOption, &dummyLen);
7290     if (code == ERROR_SUCCESS) {
7291         rxdebug_active = (TraceOption & TRACE_OPTION_RX_DEBUG) ? 1 : 0;
7292     }
7293     RegCloseKey (parmKey);
7294 #endif /* AFS_NT40_ENV */
7295 #endif
7296 }
7297
7298 void
7299 rx_DebugOnOff(int on)
7300 {
7301 #ifdef RXDEBUG
7302 #ifdef AFS_NT40_ENV
7303     rxdebug_active = on;
7304 #endif
7305 #endif
7306 }
7307
7308 void
7309 rx_StatsOnOff(int on)
7310 {
7311     rx_stats_active = on;
7312 }
7313
7314
7315 /* Don't call this debugging routine directly; use dpf */
7316 void
7317 rxi_DebugPrint(char *format, ...)
7318 {
7319 #ifdef RXDEBUG
7320     va_list ap;
7321 #ifdef AFS_NT40_ENV
7322     char msg[512];
7323     char tformat[256];
7324     size_t len;
7325
7326     va_start(ap, format);
7327
7328     len = _snprintf(tformat, sizeof(tformat), "tid[%d] %s", GetCurrentThreadId(), format);
7329
7330     if (len > 0) {
7331         len = _vsnprintf(msg, sizeof(msg)-2, tformat, ap);
7332         if (len > 0)
7333             OutputDebugString(msg);
7334     }
7335     va_end(ap);
7336 #else
7337     struct clock now;
7338
7339     va_start(ap, format);
7340
7341     clock_GetTime(&now);
7342     fprintf(rx_Log, " %d.%06d:", (unsigned int)now.sec,
7343             (unsigned int)now.usec);
7344     vfprintf(rx_Log, format, ap);
7345     va_end(ap);
7346 #endif
7347 #endif
7348 }
7349
7350 #ifndef KERNEL
7351 /*
7352  * This function is used to process the rx_stats structure that is local
7353  * to a process as well as an rx_stats structure received from a remote
7354  * process (via rxdebug).  Therefore, it needs to do minimal version
7355  * checking.
7356  */
7357 void
7358 rx_PrintTheseStats(FILE * file, struct rx_statistics *s, int size,
7359                    afs_int32 freePackets, char version)
7360 {
7361     int i;
7362
7363     if (size != sizeof(struct rx_statistics)) {
7364         fprintf(file,
7365                 "Unexpected size of stats structure: was %d, expected %" AFS_SIZET_FMT "\n",
7366                 size, sizeof(struct rx_statistics));
7367     }
7368
7369     fprintf(file, "rx stats: free packets %d, allocs %d, ", (int)freePackets,
7370             s->packetRequests);
7371
7372     if (version >= RX_DEBUGI_VERSION_W_NEWPACKETTYPES) {
7373         fprintf(file, "alloc-failures(rcv %u/%u,send %u/%u,ack %u)\n",
7374                 s->receivePktAllocFailures, s->receiveCbufPktAllocFailures,
7375                 s->sendPktAllocFailures, s->sendCbufPktAllocFailures,
7376                 s->specialPktAllocFailures);
7377     } else {
7378         fprintf(file, "alloc-failures(rcv %u,send %u,ack %u)\n",
7379                 s->receivePktAllocFailures, s->sendPktAllocFailures,
7380                 s->specialPktAllocFailures);
7381     }
7382
7383     fprintf(file,
7384             "   greedy %u, " "bogusReads %u (last from host %x), "
7385             "noPackets %u, " "noBuffers %u, " "selects %u, "
7386             "sendSelects %u\n", s->socketGreedy, s->bogusPacketOnRead,
7387             s->bogusHost, s->noPacketOnRead, s->noPacketBuffersOnRead,
7388             s->selects, s->sendSelects);
7389
7390     fprintf(file, "   packets read: ");
7391     for (i = 0; i < RX_N_PACKET_TYPES; i++) {
7392         fprintf(file, "%s %u ", rx_packetTypes[i], s->packetsRead[i]);
7393     }
7394     fprintf(file, "\n");
7395
7396     fprintf(file,
7397             "   other read counters: data %u, " "ack %u, " "dup %u "
7398             "spurious %u " "dally %u\n", s->dataPacketsRead,
7399             s->ackPacketsRead, s->dupPacketsRead, s->spuriousPacketsRead,
7400             s->ignorePacketDally);
7401
7402     fprintf(file, "   packets sent: ");
7403     for (i = 0; i < RX_N_PACKET_TYPES; i++) {
7404         fprintf(file, "%s %u ", rx_packetTypes[i], s->packetsSent[i]);
7405     }
7406     fprintf(file, "\n");
7407
7408     fprintf(file,
7409             "   other send counters: ack %u, " "data %u (not resends), "
7410             "resends %u, " "pushed %u, " "acked&ignored %u\n",
7411             s->ackPacketsSent, s->dataPacketsSent, s->dataPacketsReSent,
7412             s->dataPacketsPushed, s->ignoreAckedPacket);
7413
7414     fprintf(file,
7415             "   \t(these should be small) sendFailed %u, " "fatalErrors %u\n",
7416             s->netSendFailures, (int)s->fatalErrors);
7417
7418     if (s->nRttSamples) {
7419         fprintf(file, "   Average rtt is %0.3f, with %d samples\n",
7420                 clock_Float(&s->totalRtt) / s->nRttSamples, s->nRttSamples);
7421
7422         fprintf(file, "   Minimum rtt is %0.3f, maximum is %0.3f\n",
7423                 clock_Float(&s->minRtt), clock_Float(&s->maxRtt));
7424     }
7425
7426     fprintf(file,
7427             "   %d server connections, " "%d client connections, "
7428             "%d peer structs, " "%d call structs, " "%d free call structs\n",
7429             s->nServerConns, s->nClientConns, s->nPeerStructs,
7430             s->nCallStructs, s->nFreeCallStructs);
7431
7432 #if     !defined(AFS_PTHREAD_ENV) && !defined(AFS_USE_GETTIMEOFDAY)
7433     fprintf(file, "   %d clock updates\n", clock_nUpdates);
7434 #endif
7435 }
7436
7437 /* for backward compatibility */
7438 void
7439 rx_PrintStats(FILE * file)
7440 {
7441     MUTEX_ENTER(&rx_stats_mutex);
7442     rx_PrintTheseStats(file, (struct rx_statistics *) &rx_stats,
7443                        sizeof(rx_stats), rx_nFreePackets,
7444                        RX_DEBUGI_VERSION);
7445     MUTEX_EXIT(&rx_stats_mutex);
7446 }
7447
7448 void
7449 rx_PrintPeerStats(FILE * file, struct rx_peer *peer)
7450 {
7451     fprintf(file, "Peer %x.%d.  " "Burst size %d, " "burst wait %d.%06d.\n",
7452             ntohl(peer->host), (int)ntohs(peer->port), (int)peer->burstSize,
7453             (int)peer->burstWait.sec, (int)peer->burstWait.usec);
7454
7455     fprintf(file,
7456             "   Rtt %d, " "total sent %d, " "resent %d\n",
7457             peer->rtt, peer->nSent, peer->reSends);
7458
7459     fprintf(file,
7460             "   Packet size %d, " "max in packet skew %d, "
7461             "max out packet skew %d\n", peer->ifMTU, (int)peer->inPacketSkew,
7462             (int)peer->outPacketSkew);
7463 }
7464 #endif
7465
7466 #if defined(AFS_PTHREAD_ENV) && defined(RXDEBUG)
7467 /*
7468  * This mutex protects the following static variables:
7469  * counter
7470  */
7471
7472 #define LOCK_RX_DEBUG MUTEX_ENTER(&rx_debug_mutex)
7473 #define UNLOCK_RX_DEBUG MUTEX_EXIT(&rx_debug_mutex)
7474 #else
7475 #define LOCK_RX_DEBUG
7476 #define UNLOCK_RX_DEBUG
7477 #endif /* AFS_PTHREAD_ENV */
7478
7479 #if defined(RXDEBUG) || defined(MAKEDEBUGCALL)
7480 static int
7481 MakeDebugCall(osi_socket socket, afs_uint32 remoteAddr, afs_uint16 remotePort,
7482               u_char type, void *inputData, size_t inputLength,
7483               void *outputData, size_t outputLength)
7484 {
7485     static afs_int32 counter = 100;
7486     time_t waitTime, waitCount;
7487     struct rx_header theader;
7488     char tbuffer[1500];
7489     afs_int32 code;
7490     struct timeval tv_now, tv_wake, tv_delta;
7491     struct sockaddr_in taddr, faddr;
7492 #ifdef AFS_NT40_ENV
7493     int faddrLen;
7494 #else
7495     socklen_t faddrLen;
7496 #endif
7497     fd_set imask;
7498     char *tp;
7499
7500     waitTime = 1;
7501     waitCount = 5;
7502     LOCK_RX_DEBUG;
7503     counter++;
7504     UNLOCK_RX_DEBUG;
7505     tp = &tbuffer[sizeof(struct rx_header)];
7506     taddr.sin_family = AF_INET;
7507     taddr.sin_port = remotePort;
7508     taddr.sin_addr.s_addr = remoteAddr;
7509 #ifdef STRUCT_SOCKADDR_HAS_SA_LEN
7510     taddr.sin_len = sizeof(struct sockaddr_in);
7511 #endif
7512     while (1) {
7513         memset(&theader, 0, sizeof(theader));
7514         theader.epoch = htonl(999);
7515         theader.cid = 0;
7516         theader.callNumber = htonl(counter);
7517         theader.seq = 0;
7518         theader.serial = 0;
7519         theader.type = type;
7520         theader.flags = RX_CLIENT_INITIATED | RX_LAST_PACKET;
7521         theader.serviceId = 0;
7522
7523         memcpy(tbuffer, &theader, sizeof(theader));
7524         memcpy(tp, inputData, inputLength);
7525         code =
7526             sendto(socket, tbuffer, inputLength + sizeof(struct rx_header), 0,
7527                    (struct sockaddr *)&taddr, sizeof(struct sockaddr_in));
7528
7529         /* see if there's a packet available */
7530         gettimeofday(&tv_wake, NULL);
7531         tv_wake.tv_sec += waitTime;
7532         for (;;) {
7533             FD_ZERO(&imask);
7534             FD_SET(socket, &imask);
7535             tv_delta.tv_sec = tv_wake.tv_sec;
7536             tv_delta.tv_usec = tv_wake.tv_usec;
7537             gettimeofday(&tv_now, NULL);
7538
7539             if (tv_delta.tv_usec < tv_now.tv_usec) {
7540                 /* borrow */
7541                 tv_delta.tv_usec += 1000000;
7542                 tv_delta.tv_sec--;
7543             }
7544             tv_delta.tv_usec -= tv_now.tv_usec;
7545
7546             if (tv_delta.tv_sec < tv_now.tv_sec) {
7547                 /* time expired */
7548                 break;
7549             }
7550             tv_delta.tv_sec -= tv_now.tv_sec;
7551
7552 #ifdef AFS_NT40_ENV
7553             code = select(0, &imask, 0, 0, &tv_delta);
7554 #else /* AFS_NT40_ENV */
7555             code = select(socket + 1, &imask, 0, 0, &tv_delta);
7556 #endif /* AFS_NT40_ENV */
7557             if (code == 1 && FD_ISSET(socket, &imask)) {
7558                 /* now receive a packet */
7559                 faddrLen = sizeof(struct sockaddr_in);
7560                 code =
7561                     recvfrom(socket, tbuffer, sizeof(tbuffer), 0,
7562                              (struct sockaddr *)&faddr, &faddrLen);
7563
7564                 if (code > 0) {
7565                     memcpy(&theader, tbuffer, sizeof(struct rx_header));
7566                     if (counter == ntohl(theader.callNumber))
7567                         goto success;
7568                     continue;
7569                 }
7570             }
7571             break;
7572         }
7573
7574         /* see if we've timed out */
7575         if (!--waitCount) {
7576             return -1;
7577         }
7578         waitTime <<= 1;
7579     }
7580
7581  success:
7582     code -= sizeof(struct rx_header);
7583     if (code > outputLength)
7584         code = outputLength;
7585     memcpy(outputData, tp, code);
7586     return code;
7587 }
7588 #endif /* RXDEBUG */
7589
7590 afs_int32
7591 rx_GetServerDebug(osi_socket socket, afs_uint32 remoteAddr,
7592                   afs_uint16 remotePort, struct rx_debugStats * stat,
7593                   afs_uint32 * supportedValues)
7594 {
7595 #if defined(RXDEBUG) || defined(MAKEDEBUGCALL)
7596     afs_int32 rc = 0;
7597     struct rx_debugIn in;
7598
7599     *supportedValues = 0;
7600     in.type = htonl(RX_DEBUGI_GETSTATS);
7601     in.index = 0;
7602
7603     rc = MakeDebugCall(socket, remoteAddr, remotePort, RX_PACKET_TYPE_DEBUG,
7604                        &in, sizeof(in), stat, sizeof(*stat));
7605
7606     /*
7607      * If the call was successful, fixup the version and indicate
7608      * what contents of the stat structure are valid.
7609      * Also do net to host conversion of fields here.
7610      */
7611
7612     if (rc >= 0) {
7613         if (stat->version >= RX_DEBUGI_VERSION_W_SECSTATS) {
7614             *supportedValues |= RX_SERVER_DEBUG_SEC_STATS;
7615         }
7616         if (stat->version >= RX_DEBUGI_VERSION_W_GETALLCONN) {
7617             *supportedValues |= RX_SERVER_DEBUG_ALL_CONN;
7618         }
7619         if (stat->version >= RX_DEBUGI_VERSION_W_RXSTATS) {
7620             *supportedValues |= RX_SERVER_DEBUG_RX_STATS;
7621         }
7622         if (stat->version >= RX_DEBUGI_VERSION_W_WAITERS) {
7623             *supportedValues |= RX_SERVER_DEBUG_WAITER_CNT;
7624         }
7625         if (stat->version >= RX_DEBUGI_VERSION_W_IDLETHREADS) {
7626             *supportedValues |= RX_SERVER_DEBUG_IDLE_THREADS;
7627         }
7628         if (stat->version >= RX_DEBUGI_VERSION_W_NEWPACKETTYPES) {
7629             *supportedValues |= RX_SERVER_DEBUG_NEW_PACKETS;
7630         }
7631         if (stat->version >= RX_DEBUGI_VERSION_W_GETPEER) {
7632             *supportedValues |= RX_SERVER_DEBUG_ALL_PEER;
7633         }
7634         if (stat->version >= RX_DEBUGI_VERSION_W_WAITED) {
7635             *supportedValues |= RX_SERVER_DEBUG_WAITED_CNT;
7636         }
7637         if (stat->version >= RX_DEBUGI_VERSION_W_PACKETS) {
7638             *supportedValues |= RX_SERVER_DEBUG_PACKETS_CNT;
7639         }
7640         stat->nFreePackets = ntohl(stat->nFreePackets);
7641         stat->packetReclaims = ntohl(stat->packetReclaims);
7642         stat->callsExecuted = ntohl(stat->callsExecuted);
7643         stat->nWaiting = ntohl(stat->nWaiting);
7644         stat->idleThreads = ntohl(stat->idleThreads);
7645         stat->nWaited = ntohl(stat->nWaited);
7646         stat->nPackets = ntohl(stat->nPackets);
7647     }
7648 #else
7649     afs_int32 rc = -1;
7650 #endif
7651     return rc;
7652 }
7653
7654 afs_int32
7655 rx_GetServerStats(osi_socket socket, afs_uint32 remoteAddr,
7656                   afs_uint16 remotePort, struct rx_statistics * stat,
7657                   afs_uint32 * supportedValues)
7658 {
7659 #if defined(RXDEBUG) || defined(MAKEDEBUGCALL)
7660     afs_int32 rc = 0;
7661     struct rx_debugIn in;
7662     afs_int32 *lp = (afs_int32 *) stat;
7663     int i;
7664
7665     /*
7666      * supportedValues is currently unused, but added to allow future
7667      * versioning of this function.
7668      */
7669
7670     *supportedValues = 0;
7671     in.type = htonl(RX_DEBUGI_RXSTATS);
7672     in.index = 0;
7673     memset(stat, 0, sizeof(*stat));
7674
7675     rc = MakeDebugCall(socket, remoteAddr, remotePort, RX_PACKET_TYPE_DEBUG,
7676                        &in, sizeof(in), stat, sizeof(*stat));
7677
7678     if (rc >= 0) {
7679
7680         /*
7681          * Do net to host conversion here
7682          */
7683
7684         for (i = 0; i < sizeof(*stat) / sizeof(afs_int32); i++, lp++) {
7685             *lp = ntohl(*lp);
7686         }
7687     }
7688 #else
7689     afs_int32 rc = -1;
7690 #endif
7691     return rc;
7692 }
7693
7694 afs_int32
7695 rx_GetServerVersion(osi_socket socket, afs_uint32 remoteAddr,
7696                     afs_uint16 remotePort, size_t version_length,
7697                     char *version)
7698 {
7699 #if defined(RXDEBUG) || defined(MAKEDEBUGCALL)
7700     char a[1] = { 0 };
7701     return MakeDebugCall(socket, remoteAddr, remotePort,
7702                          RX_PACKET_TYPE_VERSION, a, 1, version,
7703                          version_length);
7704 #else
7705     return -1;
7706 #endif
7707 }
7708
7709 afs_int32
7710 rx_GetServerConnections(osi_socket socket, afs_uint32 remoteAddr,
7711                         afs_uint16 remotePort, afs_int32 * nextConnection,
7712                         int allConnections, afs_uint32 debugSupportedValues,
7713                         struct rx_debugConn * conn,
7714                         afs_uint32 * supportedValues)
7715 {
7716 #if defined(RXDEBUG) || defined(MAKEDEBUGCALL)
7717     afs_int32 rc = 0;
7718     struct rx_debugIn in;
7719     int i;
7720
7721     /*
7722      * supportedValues is currently unused, but added to allow future
7723      * versioning of this function.
7724      */
7725
7726     *supportedValues = 0;
7727     if (allConnections) {
7728         in.type = htonl(RX_DEBUGI_GETALLCONN);
7729     } else {
7730         in.type = htonl(RX_DEBUGI_GETCONN);
7731     }
7732     in.index = htonl(*nextConnection);
7733     memset(conn, 0, sizeof(*conn));
7734
7735     rc = MakeDebugCall(socket, remoteAddr, remotePort, RX_PACKET_TYPE_DEBUG,
7736                        &in, sizeof(in), conn, sizeof(*conn));
7737
7738     if (rc >= 0) {
7739         *nextConnection += 1;
7740
7741         /*
7742          * Convert old connection format to new structure.
7743          */
7744
7745         if (debugSupportedValues & RX_SERVER_DEBUG_OLD_CONN) {
7746             struct rx_debugConn_vL *vL = (struct rx_debugConn_vL *)conn;
7747 #define MOVEvL(a) (conn->a = vL->a)
7748
7749             /* any old or unrecognized version... */
7750             for (i = 0; i < RX_MAXCALLS; i++) {
7751                 MOVEvL(callState[i]);
7752                 MOVEvL(callMode[i]);
7753                 MOVEvL(callFlags[i]);
7754                 MOVEvL(callOther[i]);
7755             }
7756             if (debugSupportedValues & RX_SERVER_DEBUG_SEC_STATS) {
7757                 MOVEvL(secStats.type);
7758                 MOVEvL(secStats.level);
7759                 MOVEvL(secStats.flags);
7760                 MOVEvL(secStats.expires);
7761                 MOVEvL(secStats.packetsReceived);
7762                 MOVEvL(secStats.packetsSent);
7763                 MOVEvL(secStats.bytesReceived);
7764                 MOVEvL(secStats.bytesSent);
7765             }
7766         }
7767
7768         /*
7769          * Do net to host conversion here
7770          * NOTE:
7771          *    I don't convert host or port since we are most likely
7772          *    going to want these in NBO.
7773          */
7774         conn->cid = ntohl(conn->cid);
7775         conn->serial = ntohl(conn->serial);
7776         for (i = 0; i < RX_MAXCALLS; i++) {
7777             conn->callNumber[i] = ntohl(conn->callNumber[i]);
7778         }
7779         conn->error = ntohl(conn->error);
7780         conn->secStats.flags = ntohl(conn->secStats.flags);
7781         conn->secStats.expires = ntohl(conn->secStats.expires);
7782         conn->secStats.packetsReceived =
7783             ntohl(conn->secStats.packetsReceived);
7784         conn->secStats.packetsSent = ntohl(conn->secStats.packetsSent);
7785         conn->secStats.bytesReceived = ntohl(conn->secStats.bytesReceived);
7786         conn->secStats.bytesSent = ntohl(conn->secStats.bytesSent);
7787         conn->epoch = ntohl(conn->epoch);
7788         conn->natMTU = ntohl(conn->natMTU);
7789     }
7790 #else
7791     afs_int32 rc = -1;
7792 #endif
7793     return rc;
7794 }
7795
7796 afs_int32
7797 rx_GetServerPeers(osi_socket socket, afs_uint32 remoteAddr,
7798                   afs_uint16 remotePort, afs_int32 * nextPeer,
7799                   afs_uint32 debugSupportedValues, struct rx_debugPeer * peer,
7800                   afs_uint32 * supportedValues)
7801 {
7802 #if defined(RXDEBUG) || defined(MAKEDEBUGCALL)
7803     afs_int32 rc = 0;
7804     struct rx_debugIn in;
7805
7806     /*
7807      * supportedValues is currently unused, but added to allow future
7808      * versioning of this function.
7809      */
7810
7811     *supportedValues = 0;
7812     in.type = htonl(RX_DEBUGI_GETPEER);
7813     in.index = htonl(*nextPeer);
7814     memset(peer, 0, sizeof(*peer));
7815
7816     rc = MakeDebugCall(socket, remoteAddr, remotePort, RX_PACKET_TYPE_DEBUG,
7817                        &in, sizeof(in), peer, sizeof(*peer));
7818
7819     if (rc >= 0) {
7820         *nextPeer += 1;
7821
7822         /*
7823          * Do net to host conversion here
7824          * NOTE:
7825          *    I don't convert host or port since we are most likely
7826          *    going to want these in NBO.
7827          */
7828         peer->ifMTU = ntohs(peer->ifMTU);
7829         peer->idleWhen = ntohl(peer->idleWhen);
7830         peer->refCount = ntohs(peer->refCount);
7831         peer->burstWait.sec = ntohl(peer->burstWait.sec);
7832         peer->burstWait.usec = ntohl(peer->burstWait.usec);
7833         peer->rtt = ntohl(peer->rtt);
7834         peer->rtt_dev = ntohl(peer->rtt_dev);
7835         peer->timeout.sec = 0;
7836         peer->timeout.usec = 0;
7837         peer->nSent = ntohl(peer->nSent);
7838         peer->reSends = ntohl(peer->reSends);
7839         peer->inPacketSkew = ntohl(peer->inPacketSkew);
7840         peer->outPacketSkew = ntohl(peer->outPacketSkew);
7841         peer->rateFlag = ntohl(peer->rateFlag);
7842         peer->natMTU = ntohs(peer->natMTU);
7843         peer->maxMTU = ntohs(peer->maxMTU);
7844         peer->maxDgramPackets = ntohs(peer->maxDgramPackets);
7845         peer->ifDgramPackets = ntohs(peer->ifDgramPackets);
7846         peer->MTU = ntohs(peer->MTU);
7847         peer->cwind = ntohs(peer->cwind);
7848         peer->nDgramPackets = ntohs(peer->nDgramPackets);
7849         peer->congestSeq = ntohs(peer->congestSeq);
7850         peer->bytesSent.high = ntohl(peer->bytesSent.high);
7851         peer->bytesSent.low = ntohl(peer->bytesSent.low);
7852         peer->bytesReceived.high = ntohl(peer->bytesReceived.high);
7853         peer->bytesReceived.low = ntohl(peer->bytesReceived.low);
7854     }
7855 #else
7856     afs_int32 rc = -1;
7857 #endif
7858     return rc;
7859 }
7860
7861 afs_int32
7862 rx_GetLocalPeers(afs_uint32 peerHost, afs_uint16 peerPort,
7863                 struct rx_debugPeer * peerStats)
7864 {
7865         struct rx_peer *tp;
7866         afs_int32 error = 1; /* default to "did not succeed" */
7867         afs_uint32 hashValue = PEER_HASH(peerHost, peerPort);
7868
7869         MUTEX_ENTER(&rx_peerHashTable_lock);
7870         for(tp = rx_peerHashTable[hashValue];
7871               tp != NULL; tp = tp->next) {
7872                 if (tp->host == peerHost)
7873                         break;
7874         }
7875
7876         if (tp) {
7877                 tp->refCount++;
7878                 MUTEX_EXIT(&rx_peerHashTable_lock);
7879
7880                 error = 0;
7881
7882                 MUTEX_ENTER(&tp->peer_lock);
7883                 peerStats->host = tp->host;
7884                 peerStats->port = tp->port;
7885                 peerStats->ifMTU = tp->ifMTU;
7886                 peerStats->idleWhen = tp->idleWhen;
7887                 peerStats->refCount = tp->refCount;
7888                 peerStats->burstSize = tp->burstSize;
7889                 peerStats->burst = tp->burst;
7890                 peerStats->burstWait.sec = tp->burstWait.sec;
7891                 peerStats->burstWait.usec = tp->burstWait.usec;
7892                 peerStats->rtt = tp->rtt;
7893                 peerStats->rtt_dev = tp->rtt_dev;
7894                 peerStats->timeout.sec = 0;
7895                 peerStats->timeout.usec = 0;
7896                 peerStats->nSent = tp->nSent;
7897                 peerStats->reSends = tp->reSends;
7898                 peerStats->inPacketSkew = tp->inPacketSkew;
7899                 peerStats->outPacketSkew = tp->outPacketSkew;
7900                 peerStats->rateFlag = tp->rateFlag;
7901                 peerStats->natMTU = tp->natMTU;
7902                 peerStats->maxMTU = tp->maxMTU;
7903                 peerStats->maxDgramPackets = tp->maxDgramPackets;
7904                 peerStats->ifDgramPackets = tp->ifDgramPackets;
7905                 peerStats->MTU = tp->MTU;
7906                 peerStats->cwind = tp->cwind;
7907                 peerStats->nDgramPackets = tp->nDgramPackets;
7908                 peerStats->congestSeq = tp->congestSeq;
7909                 peerStats->bytesSent.high = tp->bytesSent.high;
7910                 peerStats->bytesSent.low = tp->bytesSent.low;
7911                 peerStats->bytesReceived.high = tp->bytesReceived.high;
7912                 peerStats->bytesReceived.low = tp->bytesReceived.low;
7913                 MUTEX_EXIT(&tp->peer_lock);
7914
7915                 MUTEX_ENTER(&rx_peerHashTable_lock);
7916                 tp->refCount--;
7917         }
7918         MUTEX_EXIT(&rx_peerHashTable_lock);
7919
7920         return error;
7921 }
7922
7923 void
7924 shutdown_rx(void)
7925 {
7926     struct rx_serverQueueEntry *np;
7927     int i, j;
7928 #ifndef KERNEL
7929     struct rx_call *call;
7930     struct rx_serverQueueEntry *sq;
7931 #endif /* KERNEL */
7932
7933     LOCK_RX_INIT;
7934     if (rxinit_status == 1) {
7935         UNLOCK_RX_INIT;
7936         return;                 /* Already shutdown. */
7937     }
7938 #ifndef KERNEL
7939     rx_port = 0;
7940 #ifndef AFS_PTHREAD_ENV
7941     FD_ZERO(&rx_selectMask);
7942 #endif /* AFS_PTHREAD_ENV */
7943     rxi_dataQuota = RX_MAX_QUOTA;
7944 #ifndef AFS_PTHREAD_ENV
7945     rxi_StopListener();
7946 #endif /* AFS_PTHREAD_ENV */
7947     shutdown_rxevent();
7948     rx_SetEpoch(0);
7949 #ifndef AFS_PTHREAD_ENV
7950 #ifndef AFS_USE_GETTIMEOFDAY
7951     clock_UnInit();
7952 #endif /* AFS_USE_GETTIMEOFDAY */
7953 #endif /* AFS_PTHREAD_ENV */
7954
7955     while (!queue_IsEmpty(&rx_freeCallQueue)) {
7956         call = queue_First(&rx_freeCallQueue, rx_call);
7957         queue_Remove(call);
7958         rxi_Free(call, sizeof(struct rx_call));
7959     }
7960
7961     while (!queue_IsEmpty(&rx_idleServerQueue)) {
7962         sq = queue_First(&rx_idleServerQueue, rx_serverQueueEntry);
7963         queue_Remove(sq);
7964     }
7965 #endif /* KERNEL */
7966
7967     {
7968         struct rx_peer **peer_ptr, **peer_end;
7969         for (peer_ptr = &rx_peerHashTable[0], peer_end =
7970              &rx_peerHashTable[rx_hashTableSize]; peer_ptr < peer_end;
7971              peer_ptr++) {
7972             struct rx_peer *peer, *next;
7973
7974             MUTEX_ENTER(&rx_peerHashTable_lock);
7975             for (peer = *peer_ptr; peer; peer = next) {
7976                 rx_interface_stat_p rpc_stat, nrpc_stat;
7977                 size_t space;
7978
7979                 MUTEX_ENTER(&rx_rpc_stats);
7980                 MUTEX_ENTER(&peer->peer_lock);
7981                 for (queue_Scan
7982                      (&peer->rpcStats, rpc_stat, nrpc_stat,
7983                       rx_interface_stat)) {
7984                     unsigned int num_funcs;
7985                     if (!rpc_stat)
7986                         break;
7987                     queue_Remove(&rpc_stat->queue_header);
7988                     queue_Remove(&rpc_stat->all_peers);
7989                     num_funcs = rpc_stat->stats[0].func_total;
7990                     space =
7991                         sizeof(rx_interface_stat_t) +
7992                         rpc_stat->stats[0].func_total *
7993                         sizeof(rx_function_entry_v1_t);
7994
7995                     rxi_Free(rpc_stat, space);
7996
7997                     /* rx_rpc_stats must be held */
7998                     rxi_rpc_peer_stat_cnt -= num_funcs;
7999                 }
8000                 MUTEX_EXIT(&peer->peer_lock);
8001                 MUTEX_EXIT(&rx_rpc_stats);
8002
8003                 next = peer->next;
8004                 rxi_FreePeer(peer);
8005                 if (rx_stats_active)
8006                     rx_atomic_dec(&rx_stats.nPeerStructs);
8007             }
8008             MUTEX_EXIT(&rx_peerHashTable_lock);
8009         }
8010     }
8011     for (i = 0; i < RX_MAX_SERVICES; i++) {
8012         if (rx_services[i])
8013             rxi_Free(rx_services[i], sizeof(*rx_services[i]));
8014     }
8015     for (i = 0; i < rx_hashTableSize; i++) {
8016         struct rx_connection *tc, *ntc;
8017         MUTEX_ENTER(&rx_connHashTable_lock);
8018         for (tc = rx_connHashTable[i]; tc; tc = ntc) {
8019             ntc = tc->next;
8020             for (j = 0; j < RX_MAXCALLS; j++) {
8021                 if (tc->call[j]) {
8022                     rxi_Free(tc->call[j], sizeof(*tc->call[j]));
8023                 }
8024             }
8025             rxi_Free(tc, sizeof(*tc));
8026         }
8027         MUTEX_EXIT(&rx_connHashTable_lock);
8028     }
8029
8030     MUTEX_ENTER(&freeSQEList_lock);
8031
8032     while ((np = rx_FreeSQEList)) {
8033         rx_FreeSQEList = *(struct rx_serverQueueEntry **)np;
8034         MUTEX_DESTROY(&np->lock);
8035         rxi_Free(np, sizeof(*np));
8036     }
8037
8038     MUTEX_EXIT(&freeSQEList_lock);
8039     MUTEX_DESTROY(&freeSQEList_lock);
8040     MUTEX_DESTROY(&rx_freeCallQueue_lock);
8041     MUTEX_DESTROY(&rx_connHashTable_lock);
8042     MUTEX_DESTROY(&rx_peerHashTable_lock);
8043     MUTEX_DESTROY(&rx_serverPool_lock);
8044
8045     osi_Free(rx_connHashTable,
8046              rx_hashTableSize * sizeof(struct rx_connection *));
8047     osi_Free(rx_peerHashTable, rx_hashTableSize * sizeof(struct rx_peer *));
8048
8049     UNPIN(rx_connHashTable,
8050           rx_hashTableSize * sizeof(struct rx_connection *));
8051     UNPIN(rx_peerHashTable, rx_hashTableSize * sizeof(struct rx_peer *));
8052
8053     rxi_FreeAllPackets();
8054
8055     MUTEX_ENTER(&rx_quota_mutex);
8056     rxi_dataQuota = RX_MAX_QUOTA;
8057     rxi_availProcs = rxi_totalMin = rxi_minDeficit = 0;
8058     MUTEX_EXIT(&rx_quota_mutex);
8059     rxinit_status = 1;
8060     UNLOCK_RX_INIT;
8061 }
8062
8063 #ifdef RX_ENABLE_LOCKS
8064 void
8065 osirx_AssertMine(afs_kmutex_t * lockaddr, char *msg)
8066 {
8067     if (!MUTEX_ISMINE(lockaddr))
8068         osi_Panic("Lock not held: %s", msg);
8069 }
8070 #endif /* RX_ENABLE_LOCKS */
8071
8072 #ifndef KERNEL
8073
8074 /*
8075  * Routines to implement connection specific data.
8076  */
8077
8078 int
8079 rx_KeyCreate(rx_destructor_t rtn)
8080 {
8081     int key;
8082     MUTEX_ENTER(&rxi_keyCreate_lock);
8083     key = rxi_keyCreate_counter++;
8084     rxi_keyCreate_destructor = (rx_destructor_t *)
8085         realloc((void *)rxi_keyCreate_destructor,
8086                 (key + 1) * sizeof(rx_destructor_t));
8087     rxi_keyCreate_destructor[key] = rtn;
8088     MUTEX_EXIT(&rxi_keyCreate_lock);
8089     return key;
8090 }
8091
8092 void
8093 rx_SetSpecific(struct rx_connection *conn, int key, void *ptr)
8094 {
8095     int i;
8096     MUTEX_ENTER(&conn->conn_data_lock);
8097     if (!conn->specific) {
8098         conn->specific = (void **)malloc((key + 1) * sizeof(void *));
8099         for (i = 0; i < key; i++)
8100             conn->specific[i] = NULL;
8101         conn->nSpecific = key + 1;
8102         conn->specific[key] = ptr;
8103     } else if (key >= conn->nSpecific) {
8104         conn->specific = (void **)
8105             realloc(conn->specific, (key + 1) * sizeof(void *));
8106         for (i = conn->nSpecific; i < key; i++)
8107             conn->specific[i] = NULL;
8108         conn->nSpecific = key + 1;
8109         conn->specific[key] = ptr;
8110     } else {
8111         if (conn->specific[key] && rxi_keyCreate_destructor[key])
8112             (*rxi_keyCreate_destructor[key]) (conn->specific[key]);
8113         conn->specific[key] = ptr;
8114     }
8115     MUTEX_EXIT(&conn->conn_data_lock);
8116 }
8117
8118 void
8119 rx_SetServiceSpecific(struct rx_service *svc, int key, void *ptr)
8120 {
8121     int i;
8122     MUTEX_ENTER(&svc->svc_data_lock);
8123     if (!svc->specific) {
8124         svc->specific = (void **)malloc((key + 1) * sizeof(void *));
8125         for (i = 0; i < key; i++)
8126             svc->specific[i] = NULL;
8127         svc->nSpecific = key + 1;
8128         svc->specific[key] = ptr;
8129     } else if (key >= svc->nSpecific) {
8130         svc->specific = (void **)
8131             realloc(svc->specific, (key + 1) * sizeof(void *));
8132         for (i = svc->nSpecific; i < key; i++)
8133             svc->specific[i] = NULL;
8134         svc->nSpecific = key + 1;
8135         svc->specific[key] = ptr;
8136     } else {
8137         if (svc->specific[key] && rxi_keyCreate_destructor[key])
8138             (*rxi_keyCreate_destructor[key]) (svc->specific[key]);
8139         svc->specific[key] = ptr;
8140     }
8141     MUTEX_EXIT(&svc->svc_data_lock);
8142 }
8143
8144 void *
8145 rx_GetSpecific(struct rx_connection *conn, int key)
8146 {
8147     void *ptr;
8148     MUTEX_ENTER(&conn->conn_data_lock);
8149     if (key >= conn->nSpecific)
8150         ptr = NULL;
8151     else
8152         ptr = conn->specific[key];
8153     MUTEX_EXIT(&conn->conn_data_lock);
8154     return ptr;
8155 }
8156
8157 void *
8158 rx_GetServiceSpecific(struct rx_service *svc, int key)
8159 {
8160     void *ptr;
8161     MUTEX_ENTER(&svc->svc_data_lock);
8162     if (key >= svc->nSpecific)
8163         ptr = NULL;
8164     else
8165         ptr = svc->specific[key];
8166     MUTEX_EXIT(&svc->svc_data_lock);
8167     return ptr;
8168 }
8169
8170
8171 #endif /* !KERNEL */
8172
8173 /*
8174  * processStats is a queue used to store the statistics for the local
8175  * process.  Its contents are similar to the contents of the rpcStats
8176  * queue on a rx_peer structure, but the actual data stored within
8177  * this queue contains totals across the lifetime of the process (assuming
8178  * the stats have not been reset) - unlike the per peer structures
8179  * which can come and go based upon the peer lifetime.
8180  */
8181
8182 static struct rx_queue processStats = { &processStats, &processStats };
8183
8184 /*
8185  * peerStats is a queue used to store the statistics for all peer structs.
8186  * Its contents are the union of all the peer rpcStats queues.
8187  */
8188
8189 static struct rx_queue peerStats = { &peerStats, &peerStats };
8190
8191 /*
8192  * rxi_monitor_processStats is used to turn process wide stat collection
8193  * on and off
8194  */
8195
8196 static int rxi_monitor_processStats = 0;
8197
8198 /*
8199  * rxi_monitor_peerStats is used to turn per peer stat collection on and off
8200  */
8201
8202 static int rxi_monitor_peerStats = 0;
8203
8204 /*
8205  * rxi_AddRpcStat - given all of the information for a particular rpc
8206  * call, create (if needed) and update the stat totals for the rpc.
8207  *
8208  * PARAMETERS
8209  *
8210  * IN stats - the queue of stats that will be updated with the new value
8211  *
8212  * IN rxInterface - a unique number that identifies the rpc interface
8213  *
8214  * IN currentFunc - the index of the function being invoked
8215  *
8216  * IN totalFunc - the total number of functions in this interface
8217  *
8218  * IN queueTime - the amount of time this function waited for a thread
8219  *
8220  * IN execTime - the amount of time this function invocation took to execute
8221  *
8222  * IN bytesSent - the number bytes sent by this invocation
8223  *
8224  * IN bytesRcvd - the number bytes received by this invocation
8225  *
8226  * IN isServer - if true, this invocation was made to a server
8227  *
8228  * IN remoteHost - the ip address of the remote host
8229  *
8230  * IN remotePort - the port of the remote host
8231  *
8232  * IN addToPeerList - if != 0, add newly created stat to the global peer list
8233  *
8234  * INOUT counter - if a new stats structure is allocated, the counter will
8235  * be updated with the new number of allocated stat structures
8236  *
8237  * RETURN CODES
8238  *
8239  * Returns void.
8240  */
8241
8242 static int
8243 rxi_AddRpcStat(struct rx_queue *stats, afs_uint32 rxInterface,
8244                afs_uint32 currentFunc, afs_uint32 totalFunc,
8245                struct clock *queueTime, struct clock *execTime,
8246                afs_hyper_t * bytesSent, afs_hyper_t * bytesRcvd, int isServer,
8247                afs_uint32 remoteHost, afs_uint32 remotePort,
8248                int addToPeerList, unsigned int *counter)
8249 {
8250     int rc = 0;
8251     rx_interface_stat_p rpc_stat, nrpc_stat;
8252
8253     /*
8254      * See if there's already a structure for this interface
8255      */
8256
8257     for (queue_Scan(stats, rpc_stat, nrpc_stat, rx_interface_stat)) {
8258         if ((rpc_stat->stats[0].interfaceId == rxInterface)
8259             && (rpc_stat->stats[0].remote_is_server == isServer))
8260             break;
8261     }
8262
8263     /*
8264      * Didn't find a match so allocate a new structure and add it to the
8265      * queue.
8266      */
8267
8268     if (queue_IsEnd(stats, rpc_stat) || (rpc_stat == NULL)
8269         || (rpc_stat->stats[0].interfaceId != rxInterface)
8270         || (rpc_stat->stats[0].remote_is_server != isServer)) {
8271         int i;
8272         size_t space;
8273
8274         space =
8275             sizeof(rx_interface_stat_t) +
8276             totalFunc * sizeof(rx_function_entry_v1_t);
8277
8278         rpc_stat = rxi_Alloc(space);
8279         if (rpc_stat == NULL) {
8280             rc = 1;
8281             goto fail;
8282         }
8283         *counter += totalFunc;
8284         for (i = 0; i < totalFunc; i++) {
8285             rpc_stat->stats[i].remote_peer = remoteHost;
8286             rpc_stat->stats[i].remote_port = remotePort;
8287             rpc_stat->stats[i].remote_is_server = isServer;
8288             rpc_stat->stats[i].interfaceId = rxInterface;
8289             rpc_stat->stats[i].func_total = totalFunc;
8290             rpc_stat->stats[i].func_index = i;
8291             hzero(rpc_stat->stats[i].invocations);
8292             hzero(rpc_stat->stats[i].bytes_sent);
8293             hzero(rpc_stat->stats[i].bytes_rcvd);
8294             rpc_stat->stats[i].queue_time_sum.sec = 0;
8295             rpc_stat->stats[i].queue_time_sum.usec = 0;
8296             rpc_stat->stats[i].queue_time_sum_sqr.sec = 0;
8297             rpc_stat->stats[i].queue_time_sum_sqr.usec = 0;
8298             rpc_stat->stats[i].queue_time_min.sec = 9999999;
8299             rpc_stat->stats[i].queue_time_min.usec = 9999999;
8300             rpc_stat->stats[i].queue_time_max.sec = 0;
8301             rpc_stat->stats[i].queue_time_max.usec = 0;
8302             rpc_stat->stats[i].execution_time_sum.sec = 0;
8303             rpc_stat->stats[i].execution_time_sum.usec = 0;
8304             rpc_stat->stats[i].execution_time_sum_sqr.sec = 0;
8305             rpc_stat->stats[i].execution_time_sum_sqr.usec = 0;
8306             rpc_stat->stats[i].execution_time_min.sec = 9999999;
8307             rpc_stat->stats[i].execution_time_min.usec = 9999999;
8308             rpc_stat->stats[i].execution_time_max.sec = 0;
8309             rpc_stat->stats[i].execution_time_max.usec = 0;
8310         }
8311         queue_Prepend(stats, rpc_stat);
8312         if (addToPeerList) {
8313             queue_Prepend(&peerStats, &rpc_stat->all_peers);
8314         }
8315     }
8316
8317     /*
8318      * Increment the stats for this function
8319      */
8320
8321     hadd32(rpc_stat->stats[currentFunc].invocations, 1);
8322     hadd(rpc_stat->stats[currentFunc].bytes_sent, *bytesSent);
8323     hadd(rpc_stat->stats[currentFunc].bytes_rcvd, *bytesRcvd);
8324     clock_Add(&rpc_stat->stats[currentFunc].queue_time_sum, queueTime);
8325     clock_AddSq(&rpc_stat->stats[currentFunc].queue_time_sum_sqr, queueTime);
8326     if (clock_Lt(queueTime, &rpc_stat->stats[currentFunc].queue_time_min)) {
8327         rpc_stat->stats[currentFunc].queue_time_min = *queueTime;
8328     }
8329     if (clock_Gt(queueTime, &rpc_stat->stats[currentFunc].queue_time_max)) {
8330         rpc_stat->stats[currentFunc].queue_time_max = *queueTime;
8331     }
8332     clock_Add(&rpc_stat->stats[currentFunc].execution_time_sum, execTime);
8333     clock_AddSq(&rpc_stat->stats[currentFunc].execution_time_sum_sqr,
8334                 execTime);
8335     if (clock_Lt(execTime, &rpc_stat->stats[currentFunc].execution_time_min)) {
8336         rpc_stat->stats[currentFunc].execution_time_min = *execTime;
8337     }
8338     if (clock_Gt(execTime, &rpc_stat->stats[currentFunc].execution_time_max)) {
8339         rpc_stat->stats[currentFunc].execution_time_max = *execTime;
8340     }
8341
8342   fail:
8343     return rc;
8344 }
8345
8346 /*
8347  * rx_IncrementTimeAndCount - increment the times and count for a particular
8348  * rpc function.
8349  *
8350  * PARAMETERS
8351  *
8352  * IN peer - the peer who invoked the rpc
8353  *
8354  * IN rxInterface - a unique number that identifies the rpc interface
8355  *
8356  * IN currentFunc - the index of the function being invoked
8357  *
8358  * IN totalFunc - the total number of functions in this interface
8359  *
8360  * IN queueTime - the amount of time this function waited for a thread
8361  *
8362  * IN execTime - the amount of time this function invocation took to execute
8363  *
8364  * IN bytesSent - the number bytes sent by this invocation
8365  *
8366  * IN bytesRcvd - the number bytes received by this invocation
8367  *
8368  * IN isServer - if true, this invocation was made to a server
8369  *
8370  * RETURN CODES
8371  *
8372  * Returns void.
8373  */
8374
8375 void
8376 rx_IncrementTimeAndCount(struct rx_peer *peer, afs_uint32 rxInterface,
8377                          afs_uint32 currentFunc, afs_uint32 totalFunc,
8378                          struct clock *queueTime, struct clock *execTime,
8379                          afs_hyper_t * bytesSent, afs_hyper_t * bytesRcvd,
8380                          int isServer)
8381 {
8382
8383     if (!(rxi_monitor_peerStats || rxi_monitor_processStats))
8384         return;
8385
8386     MUTEX_ENTER(&rx_rpc_stats);
8387
8388     if (rxi_monitor_peerStats) {
8389         MUTEX_ENTER(&peer->peer_lock);
8390         rxi_AddRpcStat(&peer->rpcStats, rxInterface, currentFunc, totalFunc,
8391                        queueTime, execTime, bytesSent, bytesRcvd, isServer,
8392                        peer->host, peer->port, 1, &rxi_rpc_peer_stat_cnt);
8393         MUTEX_EXIT(&peer->peer_lock);
8394     }
8395
8396     if (rxi_monitor_processStats) {
8397         rxi_AddRpcStat(&processStats, rxInterface, currentFunc, totalFunc,
8398                        queueTime, execTime, bytesSent, bytesRcvd, isServer,
8399                        0xffffffff, 0xffffffff, 0, &rxi_rpc_process_stat_cnt);
8400     }
8401
8402     MUTEX_EXIT(&rx_rpc_stats);
8403
8404 }
8405
8406 /*
8407  * rx_MarshallProcessRPCStats - marshall an array of rpc statistics
8408  *
8409  * PARAMETERS
8410  *
8411  * IN callerVersion - the rpc stat version of the caller.
8412  *
8413  * IN count - the number of entries to marshall.
8414  *
8415  * IN stats - pointer to stats to be marshalled.
8416  *
8417  * OUT ptr - Where to store the marshalled data.
8418  *
8419  * RETURN CODES
8420  *
8421  * Returns void.
8422  */
8423 void
8424 rx_MarshallProcessRPCStats(afs_uint32 callerVersion, int count,
8425                            rx_function_entry_v1_t * stats, afs_uint32 ** ptrP)
8426 {
8427     int i;
8428     afs_uint32 *ptr;
8429
8430     /*
8431      * We only support the first version
8432      */
8433     for (ptr = *ptrP, i = 0; i < count; i++, stats++) {
8434         *(ptr++) = stats->remote_peer;
8435         *(ptr++) = stats->remote_port;
8436         *(ptr++) = stats->remote_is_server;
8437         *(ptr++) = stats->interfaceId;
8438         *(ptr++) = stats->func_total;
8439         *(ptr++) = stats->func_index;
8440         *(ptr++) = hgethi(stats->invocations);
8441         *(ptr++) = hgetlo(stats->invocations);
8442         *(ptr++) = hgethi(stats->bytes_sent);
8443         *(ptr++) = hgetlo(stats->bytes_sent);
8444         *(ptr++) = hgethi(stats->bytes_rcvd);
8445         *(ptr++) = hgetlo(stats->bytes_rcvd);
8446         *(ptr++) = stats->queue_time_sum.sec;
8447         *(ptr++) = stats->queue_time_sum.usec;
8448         *(ptr++) = stats->queue_time_sum_sqr.sec;
8449         *(ptr++) = stats->queue_time_sum_sqr.usec;
8450         *(ptr++) = stats->queue_time_min.sec;
8451         *(ptr++) = stats->queue_time_min.usec;
8452         *(ptr++) = stats->queue_time_max.sec;
8453         *(ptr++) = stats->queue_time_max.usec;
8454         *(ptr++) = stats->execution_time_sum.sec;
8455         *(ptr++) = stats->execution_time_sum.usec;
8456         *(ptr++) = stats->execution_time_sum_sqr.sec;
8457         *(ptr++) = stats->execution_time_sum_sqr.usec;
8458         *(ptr++) = stats->execution_time_min.sec;
8459         *(ptr++) = stats->execution_time_min.usec;
8460         *(ptr++) = stats->execution_time_max.sec;
8461         *(ptr++) = stats->execution_time_max.usec;
8462     }
8463     *ptrP = ptr;
8464 }
8465
8466 /*
8467  * rx_RetrieveProcessRPCStats - retrieve all of the rpc statistics for
8468  * this process
8469  *
8470  * PARAMETERS
8471  *
8472  * IN callerVersion - the rpc stat version of the caller
8473  *
8474  * OUT myVersion - the rpc stat version of this function
8475  *
8476  * OUT clock_sec - local time seconds
8477  *
8478  * OUT clock_usec - local time microseconds
8479  *
8480  * OUT allocSize - the number of bytes allocated to contain stats
8481  *
8482  * OUT statCount - the number stats retrieved from this process.
8483  *
8484  * OUT stats - the actual stats retrieved from this process.
8485  *
8486  * RETURN CODES
8487  *
8488  * Returns void.  If successful, stats will != NULL.
8489  */
8490
8491 int
8492 rx_RetrieveProcessRPCStats(afs_uint32 callerVersion, afs_uint32 * myVersion,
8493                            afs_uint32 * clock_sec, afs_uint32 * clock_usec,
8494                            size_t * allocSize, afs_uint32 * statCount,
8495                            afs_uint32 ** stats)
8496 {
8497     size_t space = 0;
8498     afs_uint32 *ptr;
8499     struct clock now;
8500     int rc = 0;
8501
8502     *stats = 0;
8503     *allocSize = 0;
8504     *statCount = 0;
8505     *myVersion = RX_STATS_RETRIEVAL_VERSION;
8506
8507     /*
8508      * Check to see if stats are enabled
8509      */
8510
8511     MUTEX_ENTER(&rx_rpc_stats);
8512     if (!rxi_monitor_processStats) {
8513         MUTEX_EXIT(&rx_rpc_stats);
8514         return rc;
8515     }
8516
8517     clock_GetTime(&now);
8518     *clock_sec = now.sec;
8519     *clock_usec = now.usec;
8520
8521     /*
8522      * Allocate the space based upon the caller version
8523      *
8524      * If the client is at an older version than we are,
8525      * we return the statistic data in the older data format, but
8526      * we still return our version number so the client knows we
8527      * are maintaining more data than it can retrieve.
8528      */
8529
8530     if (callerVersion >= RX_STATS_RETRIEVAL_FIRST_EDITION) {
8531         space = rxi_rpc_process_stat_cnt * sizeof(rx_function_entry_v1_t);
8532         *statCount = rxi_rpc_process_stat_cnt;
8533     } else {
8534         /*
8535          * This can't happen yet, but in the future version changes
8536          * can be handled by adding additional code here
8537          */
8538     }
8539
8540     if (space > (size_t) 0) {
8541         *allocSize = space;
8542         ptr = *stats = rxi_Alloc(space);
8543
8544         if (ptr != NULL) {
8545             rx_interface_stat_p rpc_stat, nrpc_stat;
8546
8547
8548             for (queue_Scan
8549                  (&processStats, rpc_stat, nrpc_stat, rx_interface_stat)) {
8550                 /*
8551                  * Copy the data based upon the caller version
8552                  */
8553                 rx_MarshallProcessRPCStats(callerVersion,
8554                                            rpc_stat->stats[0].func_total,
8555                                            rpc_stat->stats, &ptr);
8556             }
8557         } else {
8558             rc = ENOMEM;
8559         }
8560     }
8561     MUTEX_EXIT(&rx_rpc_stats);
8562     return rc;
8563 }
8564
8565 /*
8566  * rx_RetrievePeerRPCStats - retrieve all of the rpc statistics for the peers
8567  *
8568  * PARAMETERS
8569  *
8570  * IN callerVersion - the rpc stat version of the caller
8571  *
8572  * OUT myVersion - the rpc stat version of this function
8573  *
8574  * OUT clock_sec - local time seconds
8575  *
8576  * OUT clock_usec - local time microseconds
8577  *
8578  * OUT allocSize - the number of bytes allocated to contain stats
8579  *
8580  * OUT statCount - the number of stats retrieved from the individual
8581  * peer structures.
8582  *
8583  * OUT stats - the actual stats retrieved from the individual peer structures.
8584  *
8585  * RETURN CODES
8586  *
8587  * Returns void.  If successful, stats will != NULL.
8588  */
8589
8590 int
8591 rx_RetrievePeerRPCStats(afs_uint32 callerVersion, afs_uint32 * myVersion,
8592                         afs_uint32 * clock_sec, afs_uint32 * clock_usec,
8593                         size_t * allocSize, afs_uint32 * statCount,
8594                         afs_uint32 ** stats)
8595 {
8596     size_t space = 0;
8597     afs_uint32 *ptr;
8598     struct clock now;
8599     int rc = 0;
8600
8601     *stats = 0;
8602     *statCount = 0;
8603     *allocSize = 0;
8604     *myVersion = RX_STATS_RETRIEVAL_VERSION;
8605
8606     /*
8607      * Check to see if stats are enabled
8608      */
8609
8610     MUTEX_ENTER(&rx_rpc_stats);
8611     if (!rxi_monitor_peerStats) {
8612         MUTEX_EXIT(&rx_rpc_stats);
8613         return rc;
8614     }
8615
8616     clock_GetTime(&now);
8617     *clock_sec = now.sec;
8618     *clock_usec = now.usec;
8619
8620     /*
8621      * Allocate the space based upon the caller version
8622      *
8623      * If the client is at an older version than we are,
8624      * we return the statistic data in the older data format, but
8625      * we still return our version number so the client knows we
8626      * are maintaining more data than it can retrieve.
8627      */
8628
8629     if (callerVersion >= RX_STATS_RETRIEVAL_FIRST_EDITION) {
8630         space = rxi_rpc_peer_stat_cnt * sizeof(rx_function_entry_v1_t);
8631         *statCount = rxi_rpc_peer_stat_cnt;
8632     } else {
8633         /*
8634          * This can't happen yet, but in the future version changes
8635          * can be handled by adding additional code here
8636          */
8637     }
8638
8639     if (space > (size_t) 0) {
8640         *allocSize = space;
8641         ptr = *stats = rxi_Alloc(space);
8642
8643         if (ptr != NULL) {
8644             rx_interface_stat_p rpc_stat, nrpc_stat;
8645             char *fix_offset;
8646
8647             for (queue_Scan
8648                  (&peerStats, rpc_stat, nrpc_stat, rx_interface_stat)) {
8649                 /*
8650                  * We have to fix the offset of rpc_stat since we are
8651                  * keeping this structure on two rx_queues.  The rx_queue
8652                  * package assumes that the rx_queue member is the first
8653                  * member of the structure.  That is, rx_queue assumes that
8654                  * any one item is only on one queue at a time.  We are
8655                  * breaking that assumption and so we have to do a little
8656                  * math to fix our pointers.
8657                  */
8658
8659                 fix_offset = (char *)rpc_stat;
8660                 fix_offset -= offsetof(rx_interface_stat_t, all_peers);
8661                 rpc_stat = (rx_interface_stat_p) fix_offset;
8662
8663                 /*
8664                  * Copy the data based upon the caller version
8665                  */
8666                 rx_MarshallProcessRPCStats(callerVersion,
8667                                            rpc_stat->stats[0].func_total,
8668                                            rpc_stat->stats, &ptr);
8669             }
8670         } else {
8671             rc = ENOMEM;
8672         }
8673     }
8674     MUTEX_EXIT(&rx_rpc_stats);
8675     return rc;
8676 }
8677
8678 /*
8679  * rx_FreeRPCStats - free memory allocated by
8680  *                   rx_RetrieveProcessRPCStats and rx_RetrievePeerRPCStats
8681  *
8682  * PARAMETERS
8683  *
8684  * IN stats - stats previously returned by rx_RetrieveProcessRPCStats or
8685  * rx_RetrievePeerRPCStats
8686  *
8687  * IN allocSize - the number of bytes in stats.
8688  *
8689  * RETURN CODES
8690  *
8691  * Returns void.
8692  */
8693
8694 void
8695 rx_FreeRPCStats(afs_uint32 * stats, size_t allocSize)
8696 {
8697     rxi_Free(stats, allocSize);
8698 }
8699
8700 /*
8701  * rx_queryProcessRPCStats - see if process rpc stat collection is
8702  * currently enabled.
8703  *
8704  * PARAMETERS
8705  *
8706  * RETURN CODES
8707  *
8708  * Returns 0 if stats are not enabled != 0 otherwise
8709  */
8710
8711 int
8712 rx_queryProcessRPCStats(void)
8713 {
8714     int rc;
8715     MUTEX_ENTER(&rx_rpc_stats);
8716     rc = rxi_monitor_processStats;
8717     MUTEX_EXIT(&rx_rpc_stats);
8718     return rc;
8719 }
8720
8721 /*
8722  * rx_queryPeerRPCStats - see if peer stat collection is currently enabled.
8723  *
8724  * PARAMETERS
8725  *
8726  * RETURN CODES
8727  *
8728  * Returns 0 if stats are not enabled != 0 otherwise
8729  */
8730
8731 int
8732 rx_queryPeerRPCStats(void)
8733 {
8734     int rc;
8735     MUTEX_ENTER(&rx_rpc_stats);
8736     rc = rxi_monitor_peerStats;
8737     MUTEX_EXIT(&rx_rpc_stats);
8738     return rc;
8739 }
8740
8741 /*
8742  * rx_enableProcessRPCStats - begin rpc stat collection for entire process
8743  *
8744  * PARAMETERS
8745  *
8746  * RETURN CODES
8747  *
8748  * Returns void.
8749  */
8750
8751 void
8752 rx_enableProcessRPCStats(void)
8753 {
8754     MUTEX_ENTER(&rx_rpc_stats);
8755     rx_enable_stats = 1;
8756     rxi_monitor_processStats = 1;
8757     MUTEX_EXIT(&rx_rpc_stats);
8758 }
8759
8760 /*
8761  * rx_enablePeerRPCStats - begin rpc stat collection per peer structure
8762  *
8763  * PARAMETERS
8764  *
8765  * RETURN CODES
8766  *
8767  * Returns void.
8768  */
8769
8770 void
8771 rx_enablePeerRPCStats(void)
8772 {
8773     MUTEX_ENTER(&rx_rpc_stats);
8774     rx_enable_stats = 1;
8775     rxi_monitor_peerStats = 1;
8776     MUTEX_EXIT(&rx_rpc_stats);
8777 }
8778
8779 /*
8780  * rx_disableProcessRPCStats - stop rpc stat collection for entire process
8781  *
8782  * PARAMETERS
8783  *
8784  * RETURN CODES
8785  *
8786  * Returns void.
8787  */
8788
8789 void
8790 rx_disableProcessRPCStats(void)
8791 {
8792     rx_interface_stat_p rpc_stat, nrpc_stat;
8793     size_t space;
8794
8795     MUTEX_ENTER(&rx_rpc_stats);
8796
8797     /*
8798      * Turn off process statistics and if peer stats is also off, turn
8799      * off everything
8800      */
8801
8802     rxi_monitor_processStats = 0;
8803     if (rxi_monitor_peerStats == 0) {
8804         rx_enable_stats = 0;
8805     }
8806
8807     for (queue_Scan(&processStats, rpc_stat, nrpc_stat, rx_interface_stat)) {
8808         unsigned int num_funcs = 0;
8809         if (!rpc_stat)
8810             break;
8811         queue_Remove(rpc_stat);
8812         num_funcs = rpc_stat->stats[0].func_total;
8813         space =
8814             sizeof(rx_interface_stat_t) +
8815             rpc_stat->stats[0].func_total * sizeof(rx_function_entry_v1_t);
8816
8817         rxi_Free(rpc_stat, space);
8818         rxi_rpc_process_stat_cnt -= num_funcs;
8819     }
8820     MUTEX_EXIT(&rx_rpc_stats);
8821 }
8822
8823 /*
8824  * rx_disablePeerRPCStats - stop rpc stat collection for peers
8825  *
8826  * PARAMETERS
8827  *
8828  * RETURN CODES
8829  *
8830  * Returns void.
8831  */
8832
8833 void
8834 rx_disablePeerRPCStats(void)
8835 {
8836     struct rx_peer **peer_ptr, **peer_end;
8837     int code;
8838
8839     /*
8840      * Turn off peer statistics and if process stats is also off, turn
8841      * off everything
8842      */
8843
8844     rxi_monitor_peerStats = 0;
8845     if (rxi_monitor_processStats == 0) {
8846         rx_enable_stats = 0;
8847     }
8848
8849     for (peer_ptr = &rx_peerHashTable[0], peer_end =
8850          &rx_peerHashTable[rx_hashTableSize]; peer_ptr < peer_end;
8851          peer_ptr++) {
8852         struct rx_peer *peer, *next, *prev;
8853
8854         MUTEX_ENTER(&rx_peerHashTable_lock);
8855         MUTEX_ENTER(&rx_rpc_stats);
8856         for (prev = peer = *peer_ptr; peer; peer = next) {
8857             next = peer->next;
8858             code = MUTEX_TRYENTER(&peer->peer_lock);
8859             if (code) {
8860                 rx_interface_stat_p rpc_stat, nrpc_stat;
8861                 size_t space;
8862
8863                 if (prev == *peer_ptr) {
8864                     *peer_ptr = next;
8865                     prev = next;
8866                 } else
8867                     prev->next = next;
8868
8869                 if (next)
8870                     next->refCount++;
8871                 if (prev)
8872                     prev->refCount++;
8873                 peer->refCount++;
8874                 MUTEX_EXIT(&rx_peerHashTable_lock);
8875
8876                 for (queue_Scan
8877                      (&peer->rpcStats, rpc_stat, nrpc_stat,
8878                       rx_interface_stat)) {
8879                     unsigned int num_funcs = 0;
8880                     if (!rpc_stat)
8881                         break;
8882                     queue_Remove(&rpc_stat->queue_header);
8883                     queue_Remove(&rpc_stat->all_peers);
8884                     num_funcs = rpc_stat->stats[0].func_total;
8885                     space =
8886                         sizeof(rx_interface_stat_t) +
8887                         rpc_stat->stats[0].func_total *
8888                         sizeof(rx_function_entry_v1_t);
8889
8890                     rxi_Free(rpc_stat, space);
8891                     rxi_rpc_peer_stat_cnt -= num_funcs;
8892                 }
8893                 MUTEX_EXIT(&peer->peer_lock);
8894
8895                 MUTEX_ENTER(&rx_peerHashTable_lock);
8896                 if (next)
8897                     next->refCount--;
8898                 if (prev)
8899                     prev->refCount--;
8900                 peer->refCount--;
8901             } else {
8902                 prev = peer;
8903             }
8904         }
8905         MUTEX_EXIT(&rx_rpc_stats);
8906         MUTEX_EXIT(&rx_peerHashTable_lock);
8907     }
8908 }
8909
8910 /*
8911  * rx_clearProcessRPCStats - clear the contents of the rpc stats according
8912  * to clearFlag
8913  *
8914  * PARAMETERS
8915  *
8916  * IN clearFlag - flag indicating which stats to clear
8917  *
8918  * RETURN CODES
8919  *
8920  * Returns void.
8921  */
8922
8923 void
8924 rx_clearProcessRPCStats(afs_uint32 clearFlag)
8925 {
8926     rx_interface_stat_p rpc_stat, nrpc_stat;
8927
8928     MUTEX_ENTER(&rx_rpc_stats);
8929
8930     for (queue_Scan(&processStats, rpc_stat, nrpc_stat, rx_interface_stat)) {
8931         unsigned int num_funcs = 0, i;
8932         num_funcs = rpc_stat->stats[0].func_total;
8933         for (i = 0; i < num_funcs; i++) {
8934             if (clearFlag & AFS_RX_STATS_CLEAR_INVOCATIONS) {
8935                 hzero(rpc_stat->stats[i].invocations);
8936             }
8937             if (clearFlag & AFS_RX_STATS_CLEAR_BYTES_SENT) {
8938                 hzero(rpc_stat->stats[i].bytes_sent);
8939             }
8940             if (clearFlag & AFS_RX_STATS_CLEAR_BYTES_RCVD) {
8941                 hzero(rpc_stat->stats[i].bytes_rcvd);
8942             }
8943             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_SUM) {
8944                 rpc_stat->stats[i].queue_time_sum.sec = 0;
8945                 rpc_stat->stats[i].queue_time_sum.usec = 0;
8946             }
8947             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_SQUARE) {
8948                 rpc_stat->stats[i].queue_time_sum_sqr.sec = 0;
8949                 rpc_stat->stats[i].queue_time_sum_sqr.usec = 0;
8950             }
8951             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_MIN) {
8952                 rpc_stat->stats[i].queue_time_min.sec = 9999999;
8953                 rpc_stat->stats[i].queue_time_min.usec = 9999999;
8954             }
8955             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_MAX) {
8956                 rpc_stat->stats[i].queue_time_max.sec = 0;
8957                 rpc_stat->stats[i].queue_time_max.usec = 0;
8958             }
8959             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_SUM) {
8960                 rpc_stat->stats[i].execution_time_sum.sec = 0;
8961                 rpc_stat->stats[i].execution_time_sum.usec = 0;
8962             }
8963             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_SQUARE) {
8964                 rpc_stat->stats[i].execution_time_sum_sqr.sec = 0;
8965                 rpc_stat->stats[i].execution_time_sum_sqr.usec = 0;
8966             }
8967             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_MIN) {
8968                 rpc_stat->stats[i].execution_time_min.sec = 9999999;
8969                 rpc_stat->stats[i].execution_time_min.usec = 9999999;
8970             }
8971             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_MAX) {
8972                 rpc_stat->stats[i].execution_time_max.sec = 0;
8973                 rpc_stat->stats[i].execution_time_max.usec = 0;
8974             }
8975         }
8976     }
8977
8978     MUTEX_EXIT(&rx_rpc_stats);
8979 }
8980
8981 /*
8982  * rx_clearPeerRPCStats - clear the contents of the rpc stats according
8983  * to clearFlag
8984  *
8985  * PARAMETERS
8986  *
8987  * IN clearFlag - flag indicating which stats to clear
8988  *
8989  * RETURN CODES
8990  *
8991  * Returns void.
8992  */
8993
8994 void
8995 rx_clearPeerRPCStats(afs_uint32 clearFlag)
8996 {
8997     rx_interface_stat_p rpc_stat, nrpc_stat;
8998
8999     MUTEX_ENTER(&rx_rpc_stats);
9000
9001     for (queue_Scan(&peerStats, rpc_stat, nrpc_stat, rx_interface_stat)) {
9002         unsigned int num_funcs = 0, i;
9003         char *fix_offset;
9004         /*
9005          * We have to fix the offset of rpc_stat since we are
9006          * keeping this structure on two rx_queues.  The rx_queue
9007          * package assumes that the rx_queue member is the first
9008          * member of the structure.  That is, rx_queue assumes that
9009          * any one item is only on one queue at a time.  We are
9010          * breaking that assumption and so we have to do a little
9011          * math to fix our pointers.
9012          */
9013
9014         fix_offset = (char *)rpc_stat;
9015         fix_offset -= offsetof(rx_interface_stat_t, all_peers);
9016         rpc_stat = (rx_interface_stat_p) fix_offset;
9017
9018         num_funcs = rpc_stat->stats[0].func_total;
9019         for (i = 0; i < num_funcs; i++) {
9020             if (clearFlag & AFS_RX_STATS_CLEAR_INVOCATIONS) {
9021                 hzero(rpc_stat->stats[i].invocations);
9022             }
9023             if (clearFlag & AFS_RX_STATS_CLEAR_BYTES_SENT) {
9024                 hzero(rpc_stat->stats[i].bytes_sent);
9025             }
9026             if (clearFlag & AFS_RX_STATS_CLEAR_BYTES_RCVD) {
9027                 hzero(rpc_stat->stats[i].bytes_rcvd);
9028             }
9029             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_SUM) {
9030                 rpc_stat->stats[i].queue_time_sum.sec = 0;
9031                 rpc_stat->stats[i].queue_time_sum.usec = 0;
9032             }
9033             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_SQUARE) {
9034                 rpc_stat->stats[i].queue_time_sum_sqr.sec = 0;
9035                 rpc_stat->stats[i].queue_time_sum_sqr.usec = 0;
9036             }
9037             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_MIN) {
9038                 rpc_stat->stats[i].queue_time_min.sec = 9999999;
9039                 rpc_stat->stats[i].queue_time_min.usec = 9999999;
9040             }
9041             if (clearFlag & AFS_RX_STATS_CLEAR_QUEUE_TIME_MAX) {
9042                 rpc_stat->stats[i].queue_time_max.sec = 0;
9043                 rpc_stat->stats[i].queue_time_max.usec = 0;
9044             }
9045             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_SUM) {
9046                 rpc_stat->stats[i].execution_time_sum.sec = 0;
9047                 rpc_stat->stats[i].execution_time_sum.usec = 0;
9048             }
9049             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_SQUARE) {
9050                 rpc_stat->stats[i].execution_time_sum_sqr.sec = 0;
9051                 rpc_stat->stats[i].execution_time_sum_sqr.usec = 0;
9052             }
9053             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_MIN) {
9054                 rpc_stat->stats[i].execution_time_min.sec = 9999999;
9055                 rpc_stat->stats[i].execution_time_min.usec = 9999999;
9056             }
9057             if (clearFlag & AFS_RX_STATS_CLEAR_EXEC_TIME_MAX) {
9058                 rpc_stat->stats[i].execution_time_max.sec = 0;
9059                 rpc_stat->stats[i].execution_time_max.usec = 0;
9060             }
9061         }
9062     }
9063
9064     MUTEX_EXIT(&rx_rpc_stats);
9065 }
9066
9067 /*
9068  * rxi_rxstat_userok points to a routine that returns 1 if the caller
9069  * is authorized to enable/disable/clear RX statistics.
9070  */
9071 static int (*rxi_rxstat_userok) (struct rx_call * call) = NULL;
9072
9073 void
9074 rx_SetRxStatUserOk(int (*proc) (struct rx_call * call))
9075 {
9076     rxi_rxstat_userok = proc;
9077 }
9078
9079 int
9080 rx_RxStatUserOk(struct rx_call *call)
9081 {
9082     if (!rxi_rxstat_userok)
9083         return 0;
9084     return rxi_rxstat_userok(call);
9085 }
9086
9087 #ifdef AFS_NT40_ENV
9088 /*
9089  * DllMain() -- Entry-point function called by the DllMainCRTStartup()
9090  *     function in the MSVC runtime DLL (msvcrt.dll).
9091  *
9092  *     Note: the system serializes calls to this function.
9093  */
9094 BOOL WINAPI
9095 DllMain(HINSTANCE dllInstHandle,        /* instance handle for this DLL module */
9096         DWORD reason,                   /* reason function is being called */
9097         LPVOID reserved)                /* reserved for future use */
9098 {
9099     switch (reason) {
9100     case DLL_PROCESS_ATTACH:
9101         /* library is being attached to a process */
9102         INIT_PTHREAD_LOCKS;
9103         return TRUE;
9104
9105     case DLL_PROCESS_DETACH:
9106         return TRUE;
9107
9108     default:
9109         return FALSE;
9110     }
9111 }
9112 #endif /* AFS_NT40_ENV */
9113
9114 #ifndef KERNEL
9115 int rx_DumpCalls(FILE *outputFile, char *cookie)
9116 {
9117 #ifdef RXDEBUG_PACKET
9118 #ifdef KDUMP_RX_LOCK
9119     struct rx_call_rx_lock *c;
9120 #else
9121     struct rx_call *c;
9122 #endif
9123 #ifdef AFS_NT40_ENV
9124     int zilch;
9125     char output[2048];
9126 #define RXDPRINTF sprintf
9127 #define RXDPRINTOUT output
9128 #else
9129 #define RXDPRINTF fprintf
9130 #define RXDPRINTOUT outputFile
9131 #endif
9132
9133     RXDPRINTF(RXDPRINTOUT, "%s - Start dumping all Rx Calls - count=%u\r\n", cookie, rx_stats.nCallStructs);
9134 #ifdef AFS_NT40_ENV
9135     WriteFile(outputFile, output, (DWORD)strlen(output), &zilch, NULL);
9136 #endif
9137
9138     for (c = rx_allCallsp; c; c = c->allNextp) {
9139         u_short rqc, tqc, iovqc;
9140         struct rx_packet *p, *np;
9141
9142         MUTEX_ENTER(&c->lock);
9143         queue_Count(&c->rq, p, np, rx_packet, rqc);
9144         queue_Count(&c->tq, p, np, rx_packet, tqc);
9145         queue_Count(&c->iovq, p, np, rx_packet, iovqc);
9146
9147         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, "
9148                 "rqc=%u,%u, tqc=%u,%u, iovqc=%u,%u, "
9149                 "lstatus=%u, rstatus=%u, error=%d, timeout=%u, "
9150                 "resendEvent=%d, timeoutEvt=%d, keepAliveEvt=%d, delayedAckEvt=%d, delayedAbortEvt=%d, abortCode=%d, abortCount=%d, "
9151                 "lastSendTime=%u, lastRecvTime=%u, lastSendData=%u"
9152 #ifdef RX_ENABLE_LOCKS
9153                 ", refCount=%u"
9154 #endif
9155 #ifdef RX_REFCOUNT_CHECK
9156                 ", refCountBegin=%u, refCountResend=%u, refCountDelay=%u, "
9157                 "refCountAlive=%u, refCountPacket=%u, refCountSend=%u, refCountAckAll=%u, refCountAbort=%u"
9158 #endif
9159                 "\r\n",
9160                 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,
9161                 c->callNumber?*c->callNumber:0, c->conn?c->conn->flags:0, c->flags,
9162                 (afs_uint32)c->rqc, (afs_uint32)rqc, (afs_uint32)c->tqc, (afs_uint32)tqc, (afs_uint32)c->iovqc, (afs_uint32)iovqc,
9163                 (afs_uint32)c->localStatus, (afs_uint32)c->remoteStatus, c->error, c->timeout,
9164                 c->resendEvent?1:0, c->timeoutEvent?1:0, c->keepAliveEvent?1:0, c->delayedAckEvent?1:0, c->delayedAbortEvent?1:0,
9165                 c->abortCode, c->abortCount, c->lastSendTime, c->lastReceiveTime, c->lastSendData
9166 #ifdef RX_ENABLE_LOCKS
9167                 , (afs_uint32)c->refCount
9168 #endif
9169 #ifdef RX_REFCOUNT_CHECK
9170                 , c->refCDebug[0],c->refCDebug[1],c->refCDebug[2],c->refCDebug[3],c->refCDebug[4],c->refCDebug[5],c->refCDebug[6],c->refCDebug[7]
9171 #endif
9172                 );
9173         MUTEX_EXIT(&c->lock);
9174
9175 #ifdef AFS_NT40_ENV
9176         WriteFile(outputFile, output, (DWORD)strlen(output), &zilch, NULL);
9177 #endif
9178     }
9179     RXDPRINTF(RXDPRINTOUT, "%s - End dumping all Rx Calls\r\n", cookie);
9180 #ifdef AFS_NT40_ENV
9181     WriteFile(outputFile, output, (DWORD)strlen(output), &zilch, NULL);
9182 #endif
9183 #endif /* RXDEBUG_PACKET */
9184     return 0;
9185 }
9186 #endif