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