viced-multiprobe-fix-20050721
[openafs.git] / src / viced / host.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 #include <afsconfig.h>
11 #include <afs/param.h>
12
13 RCSID
14     ("$Header$");
15
16 #include <stdio.h>
17 #include <errno.h>
18 #ifdef AFS_NT40_ENV
19 #include <fcntl.h>
20 #include <winsock2.h>
21 #else
22 #include <sys/file.h>
23 #include <netdb.h>
24 #include <netinet/in.h>
25 #endif
26
27 #ifdef HAVE_STRING_H
28 #include <string.h>
29 #else
30 #ifdef HAVE_STRINGS_H
31 #include <strings.h>
32 #endif
33 #endif
34
35 #include <afs/stds.h>
36 #include <rx/xdr.h>
37 #include <afs/assert.h>
38 #include <lwp.h>
39 #include <lock.h>
40 #include <afs/afsint.h>
41 #include <afs/rxgen_consts.h>
42 #include <afs/nfs.h>
43 #include <afs/errors.h>
44 #include <afs/ihandle.h>
45 #include <afs/vnode.h>
46 #include <afs/volume.h>
47 #ifdef AFS_ATHENA_STDENV
48 #include <krb.h>
49 #endif
50 #include <afs/acl.h>
51 #include <afs/ptclient.h>
52 #include <afs/prs_fs.h>
53 #include <afs/auth.h>
54 #include <afs/afsutil.h>
55 #include <rx/rx.h>
56 #include <afs/cellconfig.h>
57 #include <stdlib.h>
58 #include "viced_prototypes.h"
59 #include "viced.h"
60 #include "host.h"
61
62
63 #ifdef AFS_PTHREAD_ENV
64 pthread_mutex_t host_glock_mutex;
65 #endif /* AFS_PTHREAD_ENV */
66
67 extern int Console;
68 extern int CurrentConnections;
69 extern int SystemId;
70 extern int AnonymousID;
71 extern prlist AnonCPS;
72 extern int LogLevel;
73 extern struct afsconf_dir *confDir;     /* config dir object */
74 extern int lwps;                /* the max number of server threads */
75 extern afsUUID FS_HostUUID;
76
77 int CEs = 0;                    /* active clients */
78 int CEBlocks = 0;               /* number of blocks of CEs */
79 struct client *CEFree = 0;      /* first free client */
80 struct host *hostList = 0;      /* linked list of all hosts */
81 int hostCount = 0;              /* number of hosts in hostList */
82 int rxcon_ident_key;
83 int rxcon_client_key;
84
85 #define CESPERBLOCK 73
86 struct CEBlock {                /* block of CESPERBLOCK file entries */
87     struct client entry[CESPERBLOCK];
88 };
89
90 static void h_TossStuff_r(register struct host *host);
91
92 /*
93  * Make sure the subnet macros have been defined.
94  */
95 #ifndef IN_SUBNETA
96 #define IN_SUBNETA(i)           ((((afs_int32)(i))&0x80800000)==0x00800000)
97 #endif
98
99 #ifndef IN_CLASSA_SUBNET
100 #define IN_CLASSA_SUBNET        0xffff0000
101 #endif
102
103 #ifndef IN_SUBNETB
104 #define IN_SUBNETB(i)           ((((afs_int32)(i))&0xc0008000)==0x80008000)
105 #endif
106
107 #ifndef IN_CLASSB_SUBNET
108 #define IN_CLASSB_SUBNET        0xffffff00
109 #endif
110
111 #define rxr_GetEpoch(aconn) (((struct rx_connection *)(aconn))->epoch)
112
113 #define rxr_CidOf(aconn) (((struct rx_connection *)(aconn))->cid)
114
115 #define rxr_PortOf(aconn) \
116     rx_PortOf(rx_PeerOf(((struct rx_connection *)(aconn))))
117
118 #define rxr_HostOf(aconn) \
119     rx_HostOf(rx_PeerOf((struct rx_connection *)(aconn)))
120
121
122 /* get a new block of CEs and chain it on CEFree */
123 static void
124 GetCEBlock()
125 {
126     register struct CEBlock *block;
127     register int i;
128
129     block = (struct CEBlock *)malloc(sizeof(struct CEBlock));
130     if (!block) {
131         ViceLog(0, ("Failed malloc in GetCEBlock\n"));
132         ShutDownAndCore(PANIC);
133     }
134
135     for (i = 0; i < (CESPERBLOCK - 1); i++) {
136         Lock_Init(&block->entry[i].lock);
137         block->entry[i].next = &(block->entry[i + 1]);
138     }
139     block->entry[CESPERBLOCK - 1].next = 0;
140     Lock_Init(&block->entry[CESPERBLOCK - 1].lock);
141     CEFree = (struct client *)block;
142     CEBlocks++;
143
144 }                               /*GetCEBlock */
145
146
147 /* get the next available CE */
148 static struct client *
149 GetCE()
150 {
151     register struct client *entry;
152
153     if (CEFree == 0)
154         GetCEBlock();
155     if (CEFree == 0) {
156         ViceLog(0, ("CEFree NULL in GetCE\n"));
157         ShutDownAndCore(PANIC);
158     }
159
160     entry = CEFree;
161     CEFree = entry->next;
162     CEs++;
163     memset((char *)entry, 0, CLIENT_TO_ZERO(entry));
164     return (entry);
165
166 }                               /*GetCE */
167
168
169 /* return an entry to the free list */
170 static void
171 FreeCE(register struct client *entry)
172 {
173     entry->next = CEFree;
174     CEFree = entry;
175     CEs--;
176
177 }                               /*FreeCE */
178
179 /*
180  * The HTs and HTBlocks variables were formerly static, but they are
181  * now referenced elsewhere in the FileServer.
182  */
183 int HTs = 0;                    /* active file entries */
184 int HTBlocks = 0;               /* number of blocks of HTs */
185 static struct host *HTFree = 0; /* first free file entry */
186
187 /*
188  * Hash tables of host pointers. We need two tables, one
189  * to map IP addresses onto host pointers, and another
190  * to map host UUIDs onto host pointers.
191  */
192 static struct h_hashChain *hostHashTable[h_HASHENTRIES];
193 static struct h_hashChain *hostUuidHashTable[h_HASHENTRIES];
194 #define h_HashIndex(hostip) ((hostip) & (h_HASHENTRIES-1))
195 #define h_UuidHashIndex(uuidp) (((int)(afs_uuid_hash(uuidp))) & (h_HASHENTRIES-1))
196
197 struct HTBlock {                /* block of HTSPERBLOCK file entries */
198     struct host entry[h_HTSPERBLOCK];
199 };
200
201
202 /* get a new block of HTs and chain it on HTFree */
203 static void
204 GetHTBlock()
205 {
206     register struct HTBlock *block;
207     register int i;
208     static int index = 0;
209
210     block = (struct HTBlock *)malloc(sizeof(struct HTBlock));
211     if (!block) {
212         ViceLog(0, ("Failed malloc in GetHTBlock\n"));
213         ShutDownAndCore(PANIC);
214     }
215 #ifdef AFS_PTHREAD_ENV
216     for (i = 0; i < (h_HTSPERBLOCK); i++)
217         assert(pthread_cond_init(&block->entry[i].cond, NULL) == 0);
218 #endif /* AFS_PTHREAD_ENV */
219     for (i = 0; i < (h_HTSPERBLOCK); i++)
220         Lock_Init(&block->entry[i].lock);
221     for (i = 0; i < (h_HTSPERBLOCK - 1); i++)
222         block->entry[i].next = &(block->entry[i + 1]);
223     for (i = 0; i < (h_HTSPERBLOCK); i++)
224         block->entry[i].index = index++;
225     block->entry[h_HTSPERBLOCK - 1].next = 0;
226     HTFree = (struct host *)block;
227     hosttableptrs[HTBlocks++] = block->entry;
228
229 }                               /*GetHTBlock */
230
231
232 /* get the next available HT */
233 static struct host *
234 GetHT()
235 {
236     register struct host *entry;
237
238     if (HTFree == 0)
239         GetHTBlock();
240     assert(HTFree != 0);
241     entry = HTFree;
242     HTFree = entry->next;
243     HTs++;
244     memset((char *)entry, 0, HOST_TO_ZERO(entry));
245     return (entry);
246
247 }                               /*GetHT */
248
249
250 /* return an entry to the free list */
251 static void
252 FreeHT(register struct host *entry)
253 {
254     entry->next = HTFree;
255     HTFree = entry;
256     HTs--;
257
258 }                               /*FreeHT */
259
260
261 static short consolePort = 0;
262
263 int
264 h_Release(register struct host *host)
265 {
266     H_LOCK;
267     h_Release_r(host);
268     H_UNLOCK;
269     return 0;
270 }
271
272 /**
273  * If this thread does not have a hold on this host AND
274  * if other threads also dont have any holds on this host AND
275  * If either the HOSTDELETED or CLIENTDELETED flags are set
276  * then toss the host
277  */
278 int
279 h_Release_r(register struct host *host)
280 {
281
282     if (!((host)->holds[h_holdSlot()] & ~h_holdbit())) {
283         if (!h_OtherHolds_r(host)) {
284             /* must avoid masking this until after h_OtherHolds_r runs
285              * but it should be run before h_TossStuff_r */
286             (host)->holds[h_holdSlot()] &= ~h_holdbit();
287             if ((host->hostFlags & HOSTDELETED)
288                 || (host->hostFlags & CLIENTDELETED)) {
289                 h_TossStuff_r(host);
290             }
291         } else
292             (host)->holds[h_holdSlot()] &= ~h_holdbit();
293     } else
294         (host)->holds[h_holdSlot()] &= ~h_holdbit();
295
296     return 0;
297 }
298
299 int
300 h_OtherHolds_r(register struct host *host)
301 {
302     register int i, bit, slot;
303     bit = h_holdbit();
304     slot = h_holdSlot();
305     for (i = 0; i < h_maxSlots; i++) {
306         if (host->holds[i] != ((i == slot) ? bit : 0)) {
307             return 1;
308         }
309     }
310     return 0;
311 }
312
313 int
314 h_Lock_r(register struct host *host)
315 {
316     H_UNLOCK;
317     h_Lock(host);
318     H_LOCK;
319     return 0;
320 }
321
322 /**
323   * Non-blocking lock
324   * returns 1 if already locked
325   * else returns locks and returns 0
326   */
327
328 int
329 h_NBLock_r(register struct host *host)
330 {
331     struct Lock *hostLock = &host->lock;
332     int locked = 0;
333
334     H_UNLOCK;
335     LOCK_LOCK(hostLock);
336     if (!(hostLock->excl_locked) && !(hostLock->readers_reading))
337         hostLock->excl_locked = WRITE_LOCK;
338     else
339         locked = 1;
340
341     LOCK_UNLOCK(hostLock);
342     H_LOCK;
343     if (locked)
344         return 1;
345     else
346         return 0;
347 }
348
349
350 #if FS_STATS_DETAILED
351 /*------------------------------------------------------------------------
352  * PRIVATE h_AddrInSameNetwork
353  *
354  * Description:
355  *      Given a target IP address and a candidate IP address (both
356  *      in host byte order), return a non-zero value (1) if the
357  *      candidate address is in a different network from the target
358  *      address.
359  *
360  * Arguments:
361  *      a_targetAddr       : Target address.
362  *      a_candAddr         : Candidate address.
363  *
364  * Returns:
365  *      1 if the candidate address is in the same net as the target,
366  *      0 otherwise.
367  *
368  * Environment:
369  *      The target and candidate addresses are both in host byte
370  *      order, NOT network byte order, when passed in.  We return
371  *      our value as a character, since that's the type of field in
372  *      the host structure, where this info will be stored.
373  *
374  * Side Effects:
375  *      As advertised.
376  *------------------------------------------------------------------------*/
377
378 static char
379 h_AddrInSameNetwork(afs_uint32 a_targetAddr, afs_uint32 a_candAddr)
380 {                               /*h_AddrInSameNetwork */
381
382     afs_uint32 targetNet;
383     afs_uint32 candNet;
384
385     /*
386      * Pull out the network and subnetwork numbers from the target
387      * and candidate addresses.  We can short-circuit this whole
388      * affair if the target and candidate addresses are not of the
389      * same class.
390      */
391     if (IN_CLASSA(a_targetAddr)) {
392         if (!(IN_CLASSA(a_candAddr))) {
393             return (0);
394         }
395         targetNet = a_targetAddr & IN_CLASSA_NET;
396         candNet = a_candAddr & IN_CLASSA_NET;
397     } else if (IN_CLASSB(a_targetAddr)) {
398         if (!(IN_CLASSB(a_candAddr))) {
399             return (0);
400         }
401         targetNet = a_targetAddr & IN_CLASSB_NET;
402         candNet = a_candAddr & IN_CLASSB_NET;
403     } /*Class B target */
404     else if (IN_CLASSC(a_targetAddr)) {
405         if (!(IN_CLASSC(a_candAddr))) {
406             return (0);
407         }
408         targetNet = a_targetAddr & IN_CLASSC_NET;
409         candNet = a_candAddr & IN_CLASSC_NET;
410     } /*Class C target */
411     else {
412         targetNet = a_targetAddr;
413         candNet = a_candAddr;
414     }                           /*Class D address */
415
416     /*
417      * Now, simply compare the extracted net values for the two addresses
418      * (which at this point are known to be of the same class)
419      */
420     if (targetNet == candNet)
421         return (1);
422     else
423         return (0);
424
425 }                               /*h_AddrInSameNetwork */
426 #endif /* FS_STATS_DETAILED */
427
428
429 /* Assumptions: called with held host */
430 void
431 h_gethostcps_r(register struct host *host, register afs_int32 now)
432 {
433     register int code;
434     int slept = 0;
435
436     /* wait if somebody else is already doing the getCPS call */
437     while (host->hostFlags & HCPS_INPROGRESS) {
438         slept = 1;              /* I did sleep */
439         host->hostFlags |= HCPS_WAITING;        /* I am sleeping now */
440 #ifdef AFS_PTHREAD_ENV
441         pthread_cond_wait(&host->cond, &host_glock_mutex);
442 #else /* AFS_PTHREAD_ENV */
443         if ((code = LWP_WaitProcess(&(host->hostFlags))) != LWP_SUCCESS)
444             ViceLog(0, ("LWP_WaitProcess returned %d\n", code));
445 #endif /* AFS_PTHREAD_ENV */
446     }
447
448
449     host->hostFlags |= HCPS_INPROGRESS; /* mark as CPSCall in progress */
450     if (host->hcps.prlist_val)
451         free(host->hcps.prlist_val);    /* this is for hostaclRefresh */
452     host->hcps.prlist_val = NULL;
453     host->hcps.prlist_len = 0;
454     slept ? (host->cpsCall = FT_ApproxTime()) : (host->cpsCall = now);
455
456     H_UNLOCK;
457     code = pr_GetHostCPS(htonl(host->host), &host->hcps);
458     H_LOCK;
459     if (code) {
460         /*
461          * Although ubik_Call (called by pr_GetHostCPS) traverses thru all protection servers
462          * and reevaluates things if no sync server or quorum is found we could still end up
463          * with one of these errors. In such case we would like to reevaluate the rpc call to
464          * find if there's cps for this guy. We treat other errors (except network failures
465          * ones - i.e. code < 0) as an indication that there is no CPS for this host. Ideally
466          * we could like to deal this problem the other way around (i.e. if code == NOCPS 
467          * ignore else retry next time) but the problem is that there're other errors (i.e.
468          * EPERM) for which we don't want to retry and we don't know the whole code list!
469          */
470         if (code < 0 || code == UNOQUORUM || code == UNOTSYNC) {
471             /* 
472              * We would have preferred to use a while loop and try again since ops in protected
473              * acls for this host will fail now but they'll be reevaluated on any subsequent
474              * call. The attempt to wait for a quorum/sync site or network error won't work
475              * since this problems really should only occurs during a complete fileserver 
476              * restart. Since the fileserver will start before the ptservers (and thus before
477              * quorums are complete) clients will be utilizing all the fileserver's lwps!!
478              */
479             host->hcpsfailed = 1;
480             ViceLog(0,
481                     ("Warning:  GetHostCPS failed (%d) for %x; will retry\n",
482                      code, host->host));
483         } else {
484             host->hcpsfailed = 0;
485             ViceLog(1,
486                     ("gethost:  GetHostCPS failed (%d) for %x; ignored\n",
487                      code, host->host));
488         }
489         if (host->hcps.prlist_val)
490             free(host->hcps.prlist_val);
491         host->hcps.prlist_val = NULL;
492         host->hcps.prlist_len = 0;      /* Make sure it's zero */
493     } else
494         host->hcpsfailed = 0;
495
496     host->hostFlags &= ~HCPS_INPROGRESS;
497     /* signal all who are waiting */
498     if (host->hostFlags & HCPS_WAITING) {       /* somebody is waiting */
499         host->hostFlags &= ~HCPS_WAITING;
500 #ifdef AFS_PTHREAD_ENV
501         assert(pthread_cond_broadcast(&host->cond) == 0);
502 #else /* AFS_PTHREAD_ENV */
503         if ((code = LWP_NoYieldSignal(&(host->hostFlags))) != LWP_SUCCESS)
504             ViceLog(0, ("LWP_NoYieldSignal returns %d\n", code));
505 #endif /* AFS_PTHREAD_ENV */
506     }
507 }
508
509 /* args in net byte order */
510 void
511 h_flushhostcps(register afs_uint32 hostaddr, register afs_uint32 hport)
512 {
513     register struct host *host;
514     int held = 0;
515
516     H_LOCK;
517     host = h_Lookup_r(hostaddr, hport, &held);
518     if (host) {
519         host->hcpsfailed = 1;
520         if (!held)
521             h_Release_r(host);
522     }
523     H_UNLOCK;
524     return;
525 }
526
527
528 /*
529  * Allocate a host.  It will be identified by the peer (ip,port) info in the
530  * rx connection provided.  The host is returned held and locked
531  */
532 #define DEF_ROPCONS 2115
533
534 struct host *
535 h_Alloc_r(register struct rx_connection *r_con)
536 {
537     struct servent *serverentry;
538     register index = h_HashIndex(rxr_HostOf(r_con));
539     register struct host *host;
540     static struct rx_securityClass *sc = 0;
541     afs_int32 now;
542     struct h_hashChain *h_hashChain;
543 #if FS_STATS_DETAILED
544     afs_uint32 newHostAddr_HBO; /*New host IP addr, in host byte order */
545 #endif /* FS_STATS_DETAILED */
546
547     host = GetHT();
548
549     h_hashChain = (struct h_hashChain *)malloc(sizeof(struct h_hashChain));
550     if (!h_hashChain) {
551         ViceLog(0, ("Failed malloc in h_Alloc_r\n"));
552         assert(0);
553     }
554     h_hashChain->hostPtr = host;
555     h_hashChain->addr = rxr_HostOf(r_con);
556     h_hashChain->next = hostHashTable[index];
557     hostHashTable[index] = h_hashChain;
558
559     host->host = rxr_HostOf(r_con);
560     host->port = rxr_PortOf(r_con);
561     if (consolePort == 0) {     /* find the portal number for console */
562 #if     defined(AFS_OSF_ENV)
563         serverentry = getservbyname("ropcons", "");
564 #else
565         serverentry = getservbyname("ropcons", 0);
566 #endif
567         if (serverentry)
568             consolePort = serverentry->s_port;
569         else
570             consolePort = htons(DEF_ROPCONS);   /* Use a default */
571     }
572     if (host->port == consolePort)
573         host->Console = 1;
574     /* Make a callback channel even for the console, on the off chance that it
575      * makes a request that causes a break call back.  It shouldn't. */
576     {
577         if (!sc)
578             sc = rxnull_NewClientSecurityObject();
579         host->callback_rxcon =
580             rx_NewConnection(host->host, host->port, 1, sc, 0);
581         rx_SetConnDeadTime(host->callback_rxcon, 50);
582         rx_SetConnHardDeadTime(host->callback_rxcon, AFS_HARDDEADTIME);
583     }
584     now = host->LastCall = host->cpsCall = host->ActiveCall = FT_ApproxTime();
585     host->hostFlags = 0;
586     host->hcps.prlist_val = NULL;
587     host->hcps.prlist_len = 0;
588     host->interface = 0;
589 #ifdef undef
590     host->hcpsfailed = 0;       /* save cycles */
591     h_gethostcps(host);         /* do this under host hold/lock */
592 #endif
593     host->FirstClient = 0;
594     h_Hold_r(host);
595     h_Lock_r(host);
596     h_InsertList_r(host);       /* update global host List */
597 #if FS_STATS_DETAILED
598     /*
599      * Compare the new host's IP address (in host byte order) with ours
600      * (the File Server's), remembering if they are in the same network.
601      */
602     newHostAddr_HBO = (afs_uint32) ntohl(host->host);
603     host->InSameNetwork =
604         h_AddrInSameNetwork(FS_HostAddr_HBO, newHostAddr_HBO);
605 #endif /* FS_STATS_DETAILED */
606     return host;
607
608 }                               /*h_Alloc_r */
609
610
611 /* Lookup a host given an IP address and UDP port number. */
612 /* hostaddr and hport are in network order */
613 /* Note: host should be released by caller if 0 == *heldp and non-null */
614 /* hostaddr and hport are in network order */
615 struct host *
616 h_Lookup_r(afs_uint32 hostaddr, afs_uint32 hport, int *heldp)
617 {
618     register afs_int32 now;
619     register struct host *host = 0;
620     register struct h_hashChain *chain;
621     register index = h_HashIndex(hostaddr);
622     extern int hostaclRefresh;
623
624   restart:
625     for (chain = hostHashTable[index]; chain; chain = chain->next) {
626         host = chain->hostPtr;
627         assert(host);
628         if (!(host->hostFlags & HOSTDELETED) && chain->addr == hostaddr
629             && host->port == hport) {
630             *heldp = h_Held_r(host);
631             if (!*heldp)
632                 h_Hold_r(host);
633             h_Lock_r(host);
634             if (host->hostFlags & HOSTDELETED) {
635                 h_Unlock_r(host);
636                 if (!*heldp)
637                     h_Release_r(host);
638                 goto restart;
639             }
640             h_Unlock_r(host);
641             now = FT_ApproxTime();      /* always evaluate "now" */
642             if (host->hcpsfailed || (host->cpsCall + hostaclRefresh < now)) {
643                 /*
644                  * Every hostaclRefresh period (def 2 hrs) get the new
645                  * membership list for the host.  Note this could be the
646                  * first time that the host is added to a group.  Also
647                  * here we also retry on previous legitimate hcps failures.
648                  *
649                  * If we get here we still have a host hold.
650                  */
651                 h_gethostcps_r(host, now);
652             }
653             break;
654         }
655         host = NULL;
656     }
657     return host;
658
659 }                               /*h_Lookup */
660
661 /* Lookup a host given its UUID. */
662 struct host *
663 h_LookupUuid_r(afsUUID * uuidp)
664 {
665     register struct host *host = 0;
666     register struct h_hashChain *chain;
667     register index = h_UuidHashIndex(uuidp);
668
669     for (chain = hostUuidHashTable[index]; chain; chain = chain->next) {
670         host = chain->hostPtr;
671         assert(host);
672         if (!(host->hostFlags & HOSTDELETED) && host->interface
673             && afs_uuid_equal(&host->interface->uuid, uuidp)) {
674             break;
675         }
676         host = NULL;
677     }
678     return host;
679
680 }                               /*h_Lookup */
681
682
683 /*
684  * h_Hold_r: Establish a hold by the current LWP on this host--the host
685  * or its clients will not be physically deleted until all holds have
686  * been released.
687  * NOTE: h_Hold_r is a macro defined in host.h.
688  */
689
690 /* h_TossStuff_r:  Toss anything in the host structure (the host or
691  * clients marked for deletion.  Called from h_Release_r ONLY.
692  * To be called, there must be no holds, and either host->deleted
693  * or host->clientDeleted must be set.
694  */
695 static void
696 h_TossStuff_r(register struct host *host)
697 {
698     register struct client **cp, *client;
699     int i;
700
701     /* if somebody still has this host held */
702     for (i = 0; (i < h_maxSlots) && (!(host)->holds[i]); i++);
703     if (i != h_maxSlots)
704         return;
705
706     /* if somebody still has this host locked */
707     if (h_NBLock_r(host) != 0) {
708         char hoststr[16];
709         ViceLog(0,
710                 ("Warning:  h_TossStuff_r failed; Host %s:%d was locked.\n",
711                  afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port)));
712         return;
713     } else {
714         h_Unlock_r(host);
715     }
716
717     /* ASSUMPTION: rxi_FreeConnection() does not yield */
718     for (cp = &host->FirstClient; (client = *cp);) {
719         if ((host->hostFlags & HOSTDELETED) || client->deleted) {
720             if (client->refCount) {
721                 char hoststr[16];
722                 ViceLog(0,
723                         ("Warning: Host %s:%d client %x refcount %d while deleting.\n",
724                          afs_inet_ntoa_r(host->host, hoststr),
725                          ntohs(host->port), client, client->refCount));
726             }
727             if ((client->ViceId != ANONYMOUSID) && client->CPS.prlist_val) {
728                 free(client->CPS.prlist_val);
729                 client->CPS.prlist_val = NULL;
730             }
731             client->CPS.prlist_len = 0;
732             if (client->tcon) {
733                 rx_SetSpecific(client->tcon, rxcon_client_key, (void *)0);
734             }
735             CurrentConnections--;
736             *cp = client->next;
737             FreeCE(client);
738         } else
739             cp = &client->next;
740     }
741
742     /* We've just cleaned out all the deleted clients; clear the flag */
743     host->hostFlags &= ~CLIENTDELETED;
744
745     if (host->hostFlags & HOSTDELETED) {
746         register struct h_hashChain **hp, *th;
747         register struct rx_connection *rxconn;
748         afsUUID *uuidp;
749         afs_uint32 hostAddr;
750         int i;
751
752         if (host->Console & 1)
753             Console--;
754         if ((rxconn = host->callback_rxcon)) {
755             host->callback_rxcon = (struct rx_connection *)0;
756             /*
757              * If rx_DestroyConnection calls h_FreeConnection we will
758              * deadlock on the host_glock_mutex. Work around the problem
759              * by unhooking the client from the connection before
760              * destroying the connection.
761              */
762             client = rx_GetSpecific(rxconn, rxcon_client_key);
763             if (client && client->tcon == rxconn)
764                 client->tcon = NULL;
765             rx_SetSpecific(rxconn, rxcon_client_key, (void *)0);
766             rx_DestroyConnection(rxconn);
767         }
768         if (host->hcps.prlist_val)
769             free(host->hcps.prlist_val);
770         host->hcps.prlist_val = NULL;
771         host->hcps.prlist_len = 0;
772         DeleteAllCallBacks_r(host, 1);
773         host->hostFlags &= ~RESETDONE;  /* just to be safe */
774
775         /* if alternate addresses do not exist */
776         if (!(host->interface)) {
777             for (hp = &hostHashTable[h_HashIndex(host->host)]; (th = *hp);
778                  hp = &th->next) {
779                 assert(th->hostPtr);
780                 if (th->hostPtr == host) {
781                     *hp = th->next;
782                     h_DeleteList_r(host);
783                     FreeHT(host);
784                     free(th);
785                     break;
786                 }
787             }
788         } else {
789             /* delete all hash entries for the UUID */
790             uuidp = &host->interface->uuid;
791             for (hp = &hostUuidHashTable[h_UuidHashIndex(uuidp)]; (th = *hp);
792                  hp = &th->next) {
793                 assert(th->hostPtr);
794                 if (th->hostPtr == host) {
795                     *hp = th->next;
796                     free(th);
797                     break;
798                 }
799             }
800             /* delete all hash entries for alternate addresses */
801             assert(host->interface->numberOfInterfaces > 0);
802             for (i = 0; i < host->interface->numberOfInterfaces; i++) {
803                 hostAddr = host->interface->addr[i];
804                 for (hp = &hostHashTable[h_HashIndex(hostAddr)]; (th = *hp);
805                      hp = &th->next) {
806                     assert(th->hostPtr);
807                     if (th->hostPtr == host) {
808                         *hp = th->next;
809                         free(th);
810                         break;
811                     }
812                 }
813             }
814             free(host->interface);
815             host->interface = NULL;
816             h_DeleteList_r(host);       /* remove host from global host List */
817             FreeHT(host);
818         }                       /* if alternate address exists */
819     }
820 }                               /*h_TossStuff_r */
821
822
823 /* Called by rx when a server connection disappears */
824 int
825 h_FreeConnection(struct rx_connection *tcon)
826 {
827     register struct client *client;
828
829     client = (struct client *)rx_GetSpecific(tcon, rxcon_client_key);
830     if (client) {
831         H_LOCK;
832         if (client->tcon == tcon)
833             client->tcon = (struct rx_connection *)0;
834         H_UNLOCK;
835     }
836     return 0;
837 }                               /*h_FreeConnection */
838
839
840 /* h_Enumerate: Calls (*proc)(host, held, param) for at least each host in the
841  * system at the start of the enumeration (perhaps more).  Hosts may be deleted
842  * (have delete flag set); ditto for clients.  (*proc) is always called with
843  * host h_held().  The hold state of the host with respect to this lwp is passed
844  * to (*proc) as the param held.  The proc should return 0 if the host should be
845  * released, 1 if it should be held after enumeration.
846  */
847 void
848 h_Enumerate(int (*proc) (), char *param)
849 {
850     register struct host *host, **list;
851     register int *held;
852     register int i, count;
853
854     H_LOCK;
855     if (hostCount == 0) {
856         H_UNLOCK;
857         return;
858     }
859     list = (struct host **)malloc(hostCount * sizeof(struct host *));
860     if (!list) {
861         ViceLog(0, ("Failed malloc in h_Enumerate\n"));
862         assert(0);
863     }
864     held = (int *)malloc(hostCount * sizeof(int));
865     if (!held) {
866         ViceLog(0, ("Failed malloc in h_Enumerate\n"));
867         assert(0);
868     }
869     for (count = 0, host = hostList; host; host = host->next, count++) {
870         list[count] = host;
871         if (!(held[count] = h_Held_r(host)))
872             h_Hold_r(host);
873     }
874     assert(count == hostCount);
875     H_UNLOCK;
876     for (i = 0; i < count; i++) {
877         held[i] = (*proc) (list[i], held[i], param);
878         if (!held[i])
879             h_Release(list[i]); /* this might free up the host */
880     }
881     free((void *)list);
882     free((void *)held);
883 }                               /*h_Enumerate */
884
885 /* h_Enumerate_r (revised):
886  * Calls (*proc)(host, held, param) for each host in hostList, starting
887  * at enumstart
888  * Hosts may be deleted (have delete flag set); ditto for clients.
889  * (*proc) is always called with
890  * host h_held() and the global host lock (H_LOCK) locked.The hold state of the
891  * host with respect to this lwp is passed to (*proc) as the param held.
892  * The proc should return 0 if the host should be released, 1 if it should
893  * be held after enumeration.
894  */
895 void
896 h_Enumerate_r(int (*proc) (), struct host *enumstart, char *param)
897 {
898     register struct host *host;
899     register int held;
900
901     if (hostCount == 0) {
902         return;
903     }
904     for (host = enumstart; host; host = host->next) {
905         if (!(held = h_Held_r(host)))
906             h_Hold_r(host);
907         held = (*proc) (host, held, param);
908         if (!held)
909             h_Release_r(host);  /* this might free up the host */
910     }
911 }                               /*h_Enumerate_r */
912
913 /* inserts a new HashChain structure corresponding to this UUID */
914 void
915 hashInsertUuid_r(struct afsUUID *uuid, struct host *host)
916 {
917     int index;
918     struct h_hashChain *chain;
919
920     /* hash into proper bucket */
921     index = h_UuidHashIndex(uuid);
922
923     /* insert into beginning of list for this bucket */
924     chain = (struct h_hashChain *)malloc(sizeof(struct h_hashChain));
925     if (!chain) {
926         ViceLog(0, ("Failed malloc in hashInsertUuid_r\n"));
927         assert(0);
928     }
929     assert(chain);
930     chain->hostPtr = host;
931     chain->next = hostUuidHashTable[index];
932     hostUuidHashTable[index] = chain;
933 }
934
935 /* Host is returned held */
936 struct host *
937 h_GetHost_r(struct rx_connection *tcon)
938 {
939     struct host *host;
940     struct host *oldHost;
941     int code;
942     int held;
943     struct interfaceAddr interf;
944     int interfValid = 0;
945     struct Identity *identP = NULL;
946     afs_int32 haddr;
947     afs_int32 hport;
948     char hoststr[16], hoststr2[16];
949     Capabilities caps;
950     struct rx_connection *cb_conn = NULL;
951
952     caps.Capabilities_val = NULL;
953
954     haddr = rxr_HostOf(tcon);
955     hport = rxr_PortOf(tcon);
956   retry:
957     if (caps.Capabilities_val)
958         free(caps.Capabilities_val);
959     caps.Capabilities_val = NULL;
960     caps.Capabilities_len = 0;
961
962     code = 0;
963     host = h_Lookup_r(haddr, hport, &held);
964     identP = (struct Identity *)rx_GetSpecific(tcon, rxcon_ident_key);
965     if (host && !identP && !(host->Console & 1)) {
966         /* This is a new connection, and we already have a host
967          * structure for this address. Verify that the identity
968          * of the caller matches the identity in the host structure.
969          */
970         h_Lock_r(host);
971         if (!(host->hostFlags & ALTADDR)) {
972             /* Another thread is doing initialization */
973             h_Unlock_r(host);
974             if (!held)
975                 h_Release_r(host);
976             ViceLog(125,
977                     ("Host %s:%d starting h_Lookup again\n",
978                      afs_inet_ntoa_r(host->host, hoststr),
979                      ntohs(host->port)));
980             goto retry;
981         }
982         host->hostFlags &= ~ALTADDR;
983         cb_conn = host->callback_rxcon;
984         rx_GetConnection(cb_conn);
985         H_UNLOCK;
986         code =
987             RXAFSCB_TellMeAboutYourself(cb_conn, &interf, &caps);
988         if (code == RXGEN_OPCODE)
989             code = RXAFSCB_WhoAreYou(cb_conn, &interf);
990         rx_PutConnection(cb_conn);
991         cb_conn=NULL;
992         H_LOCK;
993         if (code == RXGEN_OPCODE) {
994             identP = (struct Identity *)malloc(sizeof(struct Identity));
995             if (!identP) {
996                 ViceLog(0, ("Failed malloc in h_GetHost_r\n"));
997                 assert(0);
998             }
999             identP->valid = 0;
1000             rx_SetSpecific(tcon, rxcon_ident_key, identP);
1001             /* The host on this connection was unable to respond to 
1002              * the WhoAreYou. We will treat this as a new connection
1003              * from the existing host. The worst that can happen is
1004              * that we maintain some extra callback state information */
1005             if (host->interface) {
1006                 ViceLog(0,
1007                         ("Host %s:%d used to support WhoAreYou, deleting.\n",
1008                          afs_inet_ntoa_r(host->host, hoststr),
1009                          ntohs(host->port)));
1010                 host->hostFlags |= HOSTDELETED;
1011                 h_Unlock_r(host);
1012                 if (!held)
1013                     h_Release_r(host);
1014                 host = NULL;
1015                 goto retry;
1016             }
1017         } else if (code == 0) {
1018             interfValid = 1;
1019             identP = (struct Identity *)malloc(sizeof(struct Identity));
1020             if (!identP) {
1021                 ViceLog(0, ("Failed malloc in h_GetHost_r\n"));
1022                 assert(0);
1023             }
1024             identP->valid = 1;
1025             identP->uuid = interf.uuid;
1026             rx_SetSpecific(tcon, rxcon_ident_key, identP);
1027             /* Check whether the UUID on this connection matches
1028              * the UUID in the host structure. If they don't match
1029              * then this is not the same host as before. */
1030             if (!host->interface
1031                 || !afs_uuid_equal(&interf.uuid, &host->interface->uuid)) {
1032                 ViceLog(25,
1033                         ("Host %s:%d has changed its identity, deleting.\n",
1034                          afs_inet_ntoa_r(host->host, hoststr), host->port));
1035                 host->hostFlags |= HOSTDELETED;
1036                 h_Unlock_r(host);
1037                 if (!held)
1038                     h_Release_r(host);
1039                 host = NULL;
1040                 goto retry;
1041             }
1042         } else {
1043             afs_inet_ntoa_r(host->host, hoststr);
1044             ViceLog(0,
1045                     ("CB: WhoAreYou failed for %s:%d, error %d\n", hoststr,
1046                      ntohs(host->port), code));
1047             host->hostFlags |= VENUSDOWN;
1048         }
1049         if (caps.Capabilities_val
1050             && (caps.Capabilities_val[0] & CAPABILITY_ERRORTRANS))
1051             host->hostFlags |= HERRORTRANS;
1052         else
1053             host->hostFlags &= ~(HERRORTRANS);
1054         host->hostFlags |= ALTADDR;
1055         h_Unlock_r(host);
1056     } else if (host) {
1057         if (!(host->hostFlags & ALTADDR)) {
1058             /* another thread is doing the initialisation */
1059             ViceLog(125,
1060                     ("Host %s:%d waiting for host-init to complete\n",
1061                      afs_inet_ntoa_r(host->host, hoststr),
1062                      ntohs(host->port)));
1063             h_Lock_r(host);
1064             h_Unlock_r(host);
1065             if (!held)
1066                 h_Release_r(host);
1067             ViceLog(125,
1068                     ("Host %s:%d starting h_Lookup again\n",
1069                      afs_inet_ntoa_r(host->host, hoststr),
1070                      ntohs(host->port)));
1071             goto retry;
1072         }
1073         /* We need to check whether the identity in the host structure
1074          * matches the identity on the connection. If they don't match
1075          * then treat this a new host. */
1076         if (!(host->Console & 1)
1077             && ((!identP->valid && host->interface)
1078                 || (identP->valid && !host->interface)
1079                 || (identP->valid
1080                     && !afs_uuid_equal(&identP->uuid,
1081                                        &host->interface->uuid)))) {
1082             char uuid1[128], uuid2[128];
1083             if (identP->valid)
1084                 afsUUID_to_string(&identP->uuid, uuid1, 127);
1085             if (host->interface)
1086                 afsUUID_to_string(&host->interface->uuid, uuid2, 127);
1087             ViceLog(0,
1088                     ("CB: new identity for host %s:%d, deleting(%x %x %s %s)\n",
1089                      afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port),
1090                      identP->valid, host->interface,
1091                      identP->valid ? uuid1 : "",
1092                      host->interface ? uuid2 : ""));
1093
1094             /* The host in the cache is not the host for this connection */
1095             host->hostFlags |= HOSTDELETED;
1096             h_Unlock_r(host);
1097             if (!held)
1098                 h_Release_r(host);
1099             goto retry;
1100         }
1101     } else {
1102         host = h_Alloc_r(tcon); /* returned held and locked */
1103         h_gethostcps_r(host, FT_ApproxTime());
1104         if (!(host->Console & 1)) {
1105             int pident = 0;
1106             cb_conn = host->callback_rxcon;
1107             rx_GetConnection(cb_conn);
1108             H_UNLOCK;
1109             code =
1110                 RXAFSCB_TellMeAboutYourself(cb_conn, &interf, &caps);
1111             if (code == RXGEN_OPCODE)
1112                 code = RXAFSCB_WhoAreYou(cb_conn, &interf);
1113             rx_PutConnection(cb_conn);
1114             cb_conn=NULL;
1115             H_LOCK;
1116             if (code == RXGEN_OPCODE) {
1117                 if (!identP)
1118                     identP =
1119                         (struct Identity *)malloc(sizeof(struct Identity));
1120                 else
1121                     pident = 1;
1122
1123                 if (!identP) {
1124                     ViceLog(0, ("Failed malloc in h_GetHost_r\n"));
1125                     assert(0);
1126                 }
1127                 identP->valid = 0;
1128                 if (!pident)
1129                     rx_SetSpecific(tcon, rxcon_ident_key, identP);
1130                 ViceLog(25,
1131                         ("Host %s:%d does not support WhoAreYou.\n",
1132                          afs_inet_ntoa_r(host->host, hoststr),
1133                          ntohs(host->port)));
1134                 code = 0;
1135             } else if (code == 0) {
1136                 if (!identP)
1137                     identP =
1138                         (struct Identity *)malloc(sizeof(struct Identity));
1139                 else
1140                     pident = 1;
1141
1142                 if (!identP) {
1143                     ViceLog(0, ("Failed malloc in h_GetHost_r\n"));
1144                     assert(0);
1145                 }
1146                 identP->valid = 1;
1147                 interfValid = 1;
1148                 identP->uuid = interf.uuid;
1149                 if (!pident)
1150                     rx_SetSpecific(tcon, rxcon_ident_key, identP);
1151                 ViceLog(25,
1152                         ("WhoAreYou success on %s:%d\n",
1153                          afs_inet_ntoa_r(host->host, hoststr),
1154                          ntohs(host->port)));
1155             }
1156             if (code == 0 && !identP->valid) {
1157                 cb_conn = host->callback_rxcon;
1158                 rx_GetConnection(cb_conn);
1159                 H_UNLOCK;
1160                 code = RXAFSCB_InitCallBackState(cb_conn);
1161                 rx_PutConnection(cb_conn);
1162                 cb_conn=NULL;
1163                 H_LOCK;
1164             } else if (code == 0) {
1165                 oldHost = h_LookupUuid_r(&identP->uuid);
1166                 if (oldHost) {
1167                     int probefail = 0;
1168
1169                     if (!(held = h_Held_r(oldHost)))
1170                         h_Hold_r(oldHost);
1171                     h_Lock_r(oldHost);
1172
1173                     if (oldHost->interface) {
1174                         afsUUID uuid = oldHost->interface->uuid;
1175                         cb_conn = host->callback_rxcon;
1176                         rx_GetConnection(cb_conn);
1177                         H_UNLOCK;
1178                         code = RXAFSCB_ProbeUuid(cb_conn, &uuid);
1179                         rx_PutConnection(cb_conn);
1180                         cb_conn=NULL;
1181                         H_LOCK;
1182                         if (code && MultiProbeAlternateAddress_r(oldHost)) {
1183                             probefail = 1;
1184                         }
1185                     } else {
1186                         probefail = 1;
1187                     }
1188
1189                     if (probefail) {
1190                         /* The old host is either does not have a Uuid,
1191                          * is not responding to Probes, 
1192                          * or does not have a matching Uuid. 
1193                          * Delete it! */
1194                         oldHost->hostFlags |= HOSTDELETED;
1195                         h_Unlock_r(oldHost);
1196                         h_Release_r(oldHost);
1197                         oldHost = NULL;
1198                     }
1199                 }
1200                 if (oldHost) {
1201                     /* This is a new address for an existing host. Update
1202                      * the list of interfaces for the existing host and
1203                      * delete the host structure we just allocated. */
1204                     ViceLog(25,
1205                             ("CB: new addr %s:%d for old host %s:%d\n",
1206                              afs_inet_ntoa_r(host->host, hoststr),
1207                              ntohs(host->port), afs_inet_ntoa_r(oldHost->host,
1208                                                                 hoststr2),
1209                              ntohs(oldHost->port)));
1210                     addInterfaceAddr_r(oldHost, haddr);
1211                     host->hostFlags |= HOSTDELETED;
1212                     h_Unlock_r(host);
1213                     h_Release_r(host);
1214                     host = oldHost;
1215                 } else {
1216                     /* This really is a new host */
1217                     hashInsertUuid_r(&identP->uuid, host);
1218                     cb_conn = host->callback_rxcon;
1219                     rx_GetConnection(cb_conn);          
1220                     H_UNLOCK;
1221                     code =
1222                         RXAFSCB_InitCallBackState3(cb_conn,
1223                                                    &FS_HostUUID);
1224                     rx_PutConnection(cb_conn);
1225                     cb_conn=NULL;
1226                     H_LOCK;
1227                     if (code == 0) {
1228                         ViceLog(25,
1229                                 ("InitCallBackState3 success on %s:%d\n",
1230                                  afs_inet_ntoa_r(host->host, hoststr),
1231                                  ntohs(host->port)));
1232                         assert(interfValid == 1);
1233                         initInterfaceAddr_r(host, &interf);
1234                     }
1235                 }
1236             }
1237             if (code) {
1238                 afs_inet_ntoa_r(host->host, hoststr);
1239                 ViceLog(0,
1240                         ("CB: RCallBackConnectBack failed for %s:%d\n",
1241                          hoststr, ntohs(host->port)));
1242                 host->hostFlags |= VENUSDOWN;
1243             } else
1244                 host->hostFlags |= RESETDONE;
1245
1246         }
1247         if (caps.Capabilities_val
1248             && (caps.Capabilities_val[0] & CAPABILITY_ERRORTRANS))
1249             host->hostFlags |= HERRORTRANS;
1250         else
1251             host->hostFlags &= ~(HERRORTRANS);
1252         host->hostFlags |= ALTADDR;     /* host structure initialization complete */
1253         h_Unlock_r(host);
1254     }
1255     if (caps.Capabilities_val)
1256         free(caps.Capabilities_val);
1257     caps.Capabilities_val = NULL;
1258     caps.Capabilities_len = 0;
1259     return host;
1260
1261 }                               /*h_GetHost_r */
1262
1263
1264 static char localcellname[PR_MAXNAMELEN + 1];
1265 char local_realm[AFS_REALM_SZ] = "";
1266
1267 /* not reentrant */
1268 void
1269 h_InitHostPackage()
1270 {
1271     afsconf_GetLocalCell(confDir, localcellname, PR_MAXNAMELEN);
1272     if (!local_realm[0]) {
1273         if (afs_krb_get_lrealm(local_realm, 0) != 0 /*KSUCCESS*/) {
1274             ViceLog(0,
1275                     ("afs_krb_get_lrealm failed, using %s.\n",
1276                      localcellname));
1277             strcpy(local_realm, localcellname);
1278         }
1279     }
1280     rxcon_ident_key = rx_KeyCreate((rx_destructor_t) free);
1281     rxcon_client_key = rx_KeyCreate((rx_destructor_t) 0);
1282 #ifdef AFS_PTHREAD_ENV
1283     assert(pthread_mutex_init(&host_glock_mutex, NULL) == 0);
1284 #endif /* AFS_PTHREAD_ENV */
1285 }
1286
1287 static int
1288 MapName_r(char *aname, char *acell, afs_int32 * aval)
1289 {
1290     namelist lnames;
1291     idlist lids;
1292     afs_int32 code;
1293     afs_int32 anamelen, cnamelen;
1294     int foreign = 0;
1295     char *tname;
1296
1297     anamelen = strlen(aname);
1298     if (anamelen >= PR_MAXNAMELEN)
1299         return -1;              /* bad name -- caller interprets this as anonymous, but retries later */
1300
1301     lnames.namelist_len = 1;
1302     lnames.namelist_val = (prname *) aname;     /* don't malloc in the common case */
1303     lids.idlist_len = 0;
1304     lids.idlist_val = NULL;
1305
1306     cnamelen = strlen(acell);
1307     if (cnamelen) {
1308         if (strcasecmp(local_realm, acell)
1309             && strcasecmp(localcellname, acell)) {
1310             ViceLog(2,
1311                     ("MapName: cell is foreign.  cell=%s, localcell=%s, localrealm=%s\n",
1312                      acell, localcellname, local_realm));
1313             if ((anamelen + cnamelen + 1) >= PR_MAXNAMELEN) {
1314                 ViceLog(2,
1315                         ("MapName: Name too long, using AnonymousID for %s@%s\n",
1316                          aname, acell));
1317                 *aval = AnonymousID;
1318                 return 0;
1319             }
1320             foreign = 1;        /* attempt cross-cell authentication */
1321             tname = (char *)malloc(PR_MAXNAMELEN);
1322             if (!tname) {
1323                 ViceLog(0, ("Failed malloc in MapName_r\n"));
1324                 assert(0);
1325             }
1326             strcpy(tname, aname);
1327             tname[anamelen] = '@';
1328             strcpy(tname + anamelen + 1, acell);
1329             lnames.namelist_val = (prname *) tname;
1330         }
1331     }
1332
1333     H_UNLOCK;
1334     code = pr_NameToId(&lnames, &lids);
1335     H_LOCK;
1336     if (code == 0) {
1337         if (lids.idlist_val) {
1338             *aval = lids.idlist_val[0];
1339             if (*aval == AnonymousID) {
1340                 ViceLog(2,
1341                         ("MapName: NameToId on %s returns anonymousID\n",
1342                          lnames.namelist_val));
1343             }
1344             free(lids.idlist_val);      /* return parms are not malloced in stub if server proc aborts */
1345         } else {
1346             ViceLog(0,
1347                     ("MapName: NameToId on '%s' is unknown\n",
1348                      lnames.namelist_val));
1349             code = -1;
1350         }
1351     }
1352
1353     if (foreign) {
1354         free(lnames.namelist_val);      /* We allocated this above, so we must free it now. */
1355     }
1356     return code;
1357 }
1358
1359 /*MapName*/
1360
1361
1362 /* NOTE: this returns the client with a Write lock */
1363 struct client *
1364 h_ID2Client(afs_int32 vid)
1365 {
1366     register struct client *client;
1367     register struct host *host;
1368
1369     H_LOCK;
1370     for (host = hostList; host; host = host->next) {
1371         if (host->hostFlags & HOSTDELETED)
1372             continue;
1373         for (client = host->FirstClient; client; client = client->next) {
1374             if (!client->deleted && client->ViceId == vid) {
1375                 client->refCount++;
1376                 H_UNLOCK;
1377                 ObtainWriteLock(&client->lock);
1378                 H_LOCK;
1379                 client->refCount--;
1380                 H_UNLOCK;
1381                 return client;
1382             }
1383         }
1384     }
1385
1386     H_UNLOCK;
1387     return NULL;
1388 }
1389
1390 /*
1391  * Called by the server main loop.  Returns a h_Held client, which must be
1392  * released later the main loop.  Allocates a client if the matching one
1393  * isn't around. The client is returned with its reference count incremented
1394  * by one. The caller must call h_ReleaseClient_r when finished with
1395  * the client.
1396  */
1397 struct client *
1398 h_FindClient_r(struct rx_connection *tcon)
1399 {
1400     register struct client *client;
1401     register struct host *host;
1402     struct client *oldClient;
1403     afs_int32 viceid;
1404     afs_int32 expTime;
1405     afs_int32 code;
1406     int authClass;
1407 #if (64-MAXKTCNAMELEN)
1408     ticket name length != 64
1409 #endif
1410     char tname[64];
1411     char tinst[64];
1412     char uname[PR_MAXNAMELEN];
1413     char tcell[MAXKTCREALMLEN];
1414     int fail = 0;
1415
1416     client = (struct client *)rx_GetSpecific(tcon, rxcon_client_key);
1417     if (client && !client->deleted) {
1418         client->refCount++;
1419         h_Hold_r(client->host);
1420         if (client->prfail != 2) {      /* Could add shared lock on client here */
1421             /* note that we don't have to lock entry in this path to
1422              * ensure CPS is initialized, since we don't call rx_SetSpecific
1423              * until initialization is done, and we only get here if
1424              * rx_GetSpecific located the client structure.
1425              */
1426             return client;
1427         }
1428         H_UNLOCK;
1429         ObtainWriteLock(&client->lock); /* released at end */
1430         H_LOCK;
1431     } else if (client) {
1432         client->refCount++;
1433     }
1434
1435     authClass = rx_SecurityClassOf((struct rx_connection *)tcon);
1436     ViceLog(5,
1437             ("FindClient: authenticating connection: authClass=%d\n",
1438              authClass));
1439     if (authClass == 1) {
1440         /* A bcrypt tickets, no longer supported */
1441         ViceLog(1, ("FindClient: bcrypt ticket, using AnonymousID\n"));
1442         viceid = AnonymousID;
1443         expTime = 0x7fffffff;
1444     } else if (authClass == 2) {
1445         afs_int32 kvno;
1446
1447         /* kerberos ticket */
1448         code = rxkad_GetServerInfo(tcon, /*level */ 0, &expTime,
1449                                    tname, tinst, tcell, &kvno);
1450         if (code) {
1451             ViceLog(1, ("Failed to get rxkad ticket info\n"));
1452             viceid = AnonymousID;
1453             expTime = 0x7fffffff;
1454         } else {
1455             int ilen = strlen(tinst);
1456             ViceLog(5,
1457                     ("FindClient: rxkad conn: name=%s,inst=%s,cell=%s,exp=%d,kvno=%d\n",
1458                      tname, tinst, tcell, expTime, kvno));
1459             strncpy(uname, tname, sizeof(uname));
1460             if (ilen) {
1461                 if (strlen(uname) + 1 + ilen >= sizeof(uname))
1462                     goto bad_name;
1463                 strcat(uname, ".");
1464                 strcat(uname, tinst);
1465             }
1466             /* translate the name to a vice id */
1467             code = MapName_r(uname, tcell, &viceid);
1468             if (code) {
1469               bad_name:
1470                 ViceLog(1,
1471                         ("failed to map name=%s, cell=%s -> code=%d\n", uname,
1472                          tcell, code));
1473                 fail = 1;
1474                 viceid = AnonymousID;
1475                 expTime = 0x7fffffff;
1476             }
1477         }
1478     } else {
1479         viceid = AnonymousID;   /* unknown security class */
1480         expTime = 0x7fffffff;
1481     }
1482
1483     if (!client) {
1484         host = h_GetHost_r(tcon);       /* Returns it h_Held */
1485
1486     retryfirstclient:
1487         /* First try to find the client structure */
1488         for (client = host->FirstClient; client; client = client->next) {
1489             if (!client->deleted && (client->sid == rxr_CidOf(tcon))
1490                 && (client->VenusEpoch == rxr_GetEpoch(tcon))) {
1491                 if (client->tcon && (client->tcon != tcon)) {
1492                     ViceLog(0,
1493                             ("*** Vid=%d, sid=%x, tcon=%x, Tcon=%x ***\n",
1494                              client->ViceId, client->sid, client->tcon,
1495                              tcon));
1496                     oldClient =
1497                         (struct client *)rx_GetSpecific(client->tcon,
1498                                                         rxcon_client_key);
1499                     if (oldClient) {
1500                         if (oldClient == client)
1501                             rx_SetSpecific(client->tcon, rxcon_client_key,
1502                                            NULL);
1503                         else
1504                             ViceLog(0,
1505                                     ("Client-conn mismatch: CL1=%x, CN=%x, CL2=%x\n",
1506                                      client, client->tcon, oldClient));
1507                     }
1508                     client->tcon = (struct rx_connection *)0;
1509                 }
1510                 client->refCount++;
1511                 H_UNLOCK;
1512                 ObtainWriteLock(&client->lock);
1513                 H_LOCK;
1514                 break;
1515             }
1516         }
1517
1518         /* Still no client structure - get one */
1519         if (!client) {
1520             h_Lock_r(host);
1521             /* Retry to find the client structure */
1522             for (client = host->FirstClient; client; client = client->next) {
1523                 if (!client->deleted && (client->sid == rxr_CidOf(tcon))
1524                     && (client->VenusEpoch == rxr_GetEpoch(tcon))) {
1525                     h_Unlock_r(host);
1526                     goto retryfirstclient;
1527                 }
1528             }
1529             client = GetCE();
1530             ObtainWriteLock(&client->lock);
1531             client->refCount = 1;
1532             client->host = host;
1533             client->next = host->FirstClient;
1534             host->FirstClient = client;
1535 #if FS_STATS_DETAILED
1536             client->InSameNetwork = host->InSameNetwork;
1537 #endif /* FS_STATS_DETAILED */
1538             client->ViceId = viceid;
1539             client->expTime = expTime;  /* rx only */
1540             client->authClass = authClass;      /* rx only */
1541             client->sid = rxr_CidOf(tcon);
1542             client->VenusEpoch = rxr_GetEpoch(tcon);
1543             client->CPS.prlist_val = 0;
1544             client->CPS.prlist_len = 0;
1545             h_Unlock_r(host);
1546             CurrentConnections++;       /* increment number of connections */
1547         }
1548     }
1549     client->prfail = fail;
1550
1551     if (!(client->CPS.prlist_val) || (viceid != client->ViceId)) {
1552         if (client->CPS.prlist_val && (client->ViceId != ANONYMOUSID)) {
1553             free(client->CPS.prlist_val);
1554         }
1555         client->CPS.prlist_val = NULL;
1556         client->CPS.prlist_len = 0;
1557         client->ViceId = viceid;
1558         client->expTime = expTime;
1559
1560         if (viceid == ANONYMOUSID) {
1561             client->CPS.prlist_len = AnonCPS.prlist_len;
1562             client->CPS.prlist_val = AnonCPS.prlist_val;
1563         } else {
1564             H_UNLOCK;
1565             code = pr_GetCPS(viceid, &client->CPS);
1566             H_LOCK;
1567             if (code) {
1568                 char hoststr[16];
1569                 ViceLog(0,
1570                         ("pr_GetCPS failed(%d) for user %d, host %s:%d\n",
1571                          code, viceid, afs_inet_ntoa_r(client->host->host,
1572                                                        hoststr),
1573                          ntohs(client->host->port)));
1574
1575                 /* Although ubik_Call (called by pr_GetCPS) traverses thru
1576                  * all protection servers and reevaluates things if no
1577                  * sync server or quorum is found we could still end up
1578                  * with one of these errors. In such case we would like to
1579                  * reevaluate the rpc call to find if there's cps for this
1580                  * guy. We treat other errors (except network failures
1581                  * ones - i.e. code < 0) as an indication that there is no
1582                  * CPS for this host.  Ideally we could like to deal this
1583                  * problem the other way around (i.e.  if code == NOCPS
1584                  * ignore else retry next time) but the problem is that
1585                  * there're other errors (i.e.  EPERM) for which we don't
1586                  * want to retry and we don't know the whole code list!
1587                  */
1588                 if (code < 0 || code == UNOQUORUM || code == UNOTSYNC)
1589                     client->prfail = 1;
1590             }
1591         }
1592         /* the disabling of system:administrators is so iffy and has so many
1593          * possible failure modes that we will disable it again */
1594         /* Turn off System:Administrator for safety  
1595          * if (AL_IsAMember(SystemId, client->CPS) == 0)
1596          * assert(AL_DisableGroup(SystemId, client->CPS) == 0); */
1597     }
1598
1599     /* Now, tcon may already be set to a rock, since we blocked with no host
1600      * or client locks set above in pr_GetCPS (XXXX some locking is probably
1601      * required).  So, before setting the RPC's rock, we should disconnect
1602      * the RPC from the other client structure's rock.
1603      */
1604     oldClient = (struct client *)rx_GetSpecific(tcon, rxcon_client_key);
1605     if (oldClient && oldClient->tcon == tcon) {
1606         oldClient->tcon = (struct rx_connection *)0;
1607         ViceLog(0, ("FindClient: client %x(%x) already had conn %x (host %x), stolen by client %x(%x)\n", 
1608                     oldClient, oldClient->sid, tcon, 
1609                     rx_HostOf(rx_PeerOf(tcon)), client, client->sid));
1610         /* rx_SetSpecific will be done immediately below */
1611     }
1612     client->tcon = tcon;
1613     rx_SetSpecific(tcon, rxcon_client_key, client);
1614     ReleaseWriteLock(&client->lock);
1615
1616     return client;
1617
1618 }                               /*h_FindClient_r */
1619
1620 int
1621 h_ReleaseClient_r(struct client *client)
1622 {
1623     assert(client->refCount > 0);
1624     client->refCount--;
1625     return 0;
1626 }
1627
1628
1629 /*
1630  * Sigh:  this one is used to get the client AGAIN within the individual
1631  * server routines.  This does not bother h_Holding the host, since
1632  * this is assumed already have been done by the server main loop.
1633  * It does check tokens, since only the server routines can return the
1634  * VICETOKENDEAD error code
1635  */
1636 int
1637 GetClient(struct rx_connection *tcon, struct client **cp)
1638 {
1639     register struct client *client;
1640
1641     H_LOCK;
1642     *cp = client = (struct client *)rx_GetSpecific(tcon, rxcon_client_key);
1643     if (client == NULL || client->tcon == NULL) {
1644         ViceLog(0,
1645                 ("GetClient: no client in conn %x (host %x), VBUSYING\n",
1646                  tcon, rx_HostOf(rx_PeerOf(tcon))));
1647         H_UNLOCK;
1648         return VBUSY;
1649     }
1650     if (rxr_CidOf(client->tcon) != client->sid) {
1651         ViceLog(0,
1652                 ("GetClient: tcon %x tcon sid %d client sid %d\n",
1653                  client->tcon, rxr_CidOf(client->tcon), client->sid));
1654         H_UNLOCK;
1655         return VBUSY;
1656     }
1657     if (!(client && client->tcon && rxr_CidOf(client->tcon) == client->sid)) {
1658         if (!client)
1659             ViceLog(0, ("GetClient: no client in conn %x\n", tcon));
1660         else
1661             ViceLog(0,
1662                     ("GetClient: tcon %x tcon sid %d client sid %d\n",
1663                      client->tcon, client->tcon ? rxr_CidOf(client->tcon)
1664                      : -1, client->sid));
1665         assert(0);
1666     }
1667     if (client && client->LastCall > client->expTime && client->expTime) {
1668         char hoststr[16];
1669         ViceLog(1,
1670                 ("Token for %s at %s:%d expired %d\n", h_UserName(client),
1671                  afs_inet_ntoa_r(client->host->host, hoststr),
1672                  ntohs(client->host->port), client->expTime));
1673         H_UNLOCK;
1674         return VICETOKENDEAD;
1675     }
1676
1677     H_UNLOCK;
1678     return 0;
1679
1680 }                               /*GetClient */
1681
1682
1683 /* Client user name for short term use.  Note that this is NOT inexpensive */
1684 char *
1685 h_UserName(struct client *client)
1686 {
1687     static char User[PR_MAXNAMELEN + 1];
1688     namelist lnames;
1689     idlist lids;
1690
1691     lids.idlist_len = 1;
1692     lids.idlist_val = (afs_int32 *) malloc(1 * sizeof(afs_int32));
1693     if (!lids.idlist_val) {
1694         ViceLog(0, ("Failed malloc in h_UserName\n"));
1695         assert(0);
1696     }
1697     lnames.namelist_len = 0;
1698     lnames.namelist_val = (prname *) 0;
1699     lids.idlist_val[0] = client->ViceId;
1700     if (pr_IdToName(&lids, &lnames)) {
1701         /* We need to free id we alloced above! */
1702         free(lids.idlist_val);
1703         return "*UNKNOWN USER NAME*";
1704     }
1705     strncpy(User, lnames.namelist_val[0], PR_MAXNAMELEN);
1706     free(lids.idlist_val);
1707     free(lnames.namelist_val);
1708     return User;
1709
1710 }                               /*h_UserName */
1711
1712
1713 void
1714 h_PrintStats()
1715 {
1716     ViceLog(0,
1717             ("Total Client entries = %d, blocks = %d; Host entries = %d, blocks = %d\n",
1718              CEs, CEBlocks, HTs, HTBlocks));
1719
1720 }                               /*h_PrintStats */
1721
1722
1723 static int
1724 h_PrintClient(register struct host *host, int held, StreamHandle_t * file)
1725 {
1726     register struct client *client;
1727     int i;
1728     char tmpStr[256];
1729     char tbuffer[32];
1730     char hoststr[16];
1731
1732     H_LOCK;
1733     if (host->hostFlags & HOSTDELETED) {
1734         H_UNLOCK;
1735         return held;
1736     }
1737     (void)afs_snprintf(tmpStr, sizeof tmpStr,
1738                        "Host %s:%d down = %d, LastCall %s",
1739                        afs_inet_ntoa_r(host->host, hoststr),
1740                        ntohs(host->port), (host->hostFlags & VENUSDOWN),
1741                        afs_ctime((time_t *) & host->LastCall, tbuffer,
1742                                  sizeof(tbuffer)));
1743     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1744     for (client = host->FirstClient; client; client = client->next) {
1745         if (!client->deleted) {
1746             if (client->tcon) {
1747                 (void)afs_snprintf(tmpStr, sizeof tmpStr,
1748                                    "    user id=%d,  name=%s, sl=%s till %s",
1749                                    client->ViceId, h_UserName(client),
1750                                    client->
1751                                    authClass ? "Authenticated" :
1752                                    "Not authenticated",
1753                                    client->
1754                                    authClass ? afs_ctime((time_t *) & client->
1755                                                          expTime, tbuffer,
1756                                                          sizeof(tbuffer))
1757                                    : "No Limit\n");
1758                 (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1759             } else {
1760                 (void)afs_snprintf(tmpStr, sizeof tmpStr,
1761                                    "    user=%s, no current server connection\n",
1762                                    h_UserName(client));
1763                 (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1764             }
1765             (void)afs_snprintf(tmpStr, sizeof tmpStr, "      CPS-%d is [",
1766                                client->CPS.prlist_len);
1767             (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1768             if (client->CPS.prlist_val) {
1769                 for (i = 0; i > client->CPS.prlist_len; i++) {
1770                     (void)afs_snprintf(tmpStr, sizeof tmpStr, " %d",
1771                                        client->CPS.prlist_val[i]);
1772                     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1773                 }
1774             }
1775             sprintf(tmpStr, "]\n");
1776             (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1777         }
1778     }
1779     H_UNLOCK;
1780     return held;
1781
1782 }                               /*h_PrintClient */
1783
1784
1785
1786 /*
1787  * Print a list of clients, with last security level and token value seen,
1788  * if known
1789  */
1790 void
1791 h_PrintClients()
1792 {
1793     time_t now;
1794     char tmpStr[256];
1795     char tbuffer[32];
1796
1797     StreamHandle_t *file = STREAM_OPEN(AFSDIR_SERVER_CLNTDUMP_FILEPATH, "w");
1798
1799     if (file == NULL) {
1800         ViceLog(0,
1801                 ("Couldn't create client dump file %s\n",
1802                  AFSDIR_SERVER_CLNTDUMP_FILEPATH));
1803         return;
1804     }
1805     now = FT_ApproxTime();
1806     (void)afs_snprintf(tmpStr, sizeof tmpStr, "List of active users at %s\n",
1807                        afs_ctime(&now, tbuffer, sizeof(tbuffer)));
1808     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1809     h_Enumerate(h_PrintClient, (char *)file);
1810     STREAM_REALLYCLOSE(file);
1811     ViceLog(0, ("Created client dump %s\n", AFSDIR_SERVER_CLNTDUMP_FILEPATH));
1812 }
1813
1814
1815
1816
1817 static int
1818 h_DumpHost(register struct host *host, int held, StreamHandle_t * file)
1819 {
1820     int i;
1821     char tmpStr[256];
1822
1823     H_LOCK;
1824     (void)afs_snprintf(tmpStr, sizeof tmpStr,
1825                        "ip:%x port:%d hidx:%d cbid:%d lock:%x last:%u active:%u down:%d del:%d cons:%d cldel:%d\n\t hpfailed:%d hcpsCall:%u hcps [",
1826                        host->host, ntohs(host->port), host->index,
1827                        host->cblist, CheckLock(&host->lock), host->LastCall,
1828                        host->ActiveCall, (host->hostFlags & VENUSDOWN),
1829                        host->hostFlags & HOSTDELETED, host->Console,
1830                        host->hostFlags & CLIENTDELETED, host->hcpsfailed,
1831                        host->cpsCall);
1832     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1833     if (host->hcps.prlist_val)
1834         for (i = 0; i < host->hcps.prlist_len; i++) {
1835             (void)afs_snprintf(tmpStr, sizeof tmpStr, " %d",
1836                                host->hcps.prlist_val[i]);
1837             (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1838         }
1839     sprintf(tmpStr, "] [");
1840     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1841     if (host->interface)
1842         for (i = 0; i < host->interface->numberOfInterfaces; i++) {
1843             sprintf(tmpStr, " %x", host->interface->addr[i]);
1844             (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1845         }
1846     sprintf(tmpStr, "] holds: ");
1847     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1848
1849     for (i = 0; i < h_maxSlots; i++) {
1850         sprintf(tmpStr, "%04x", host->holds[i]);
1851         (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1852     }
1853     sprintf(tmpStr, " slot/bit: %d/%d\n", h_holdSlot(), h_holdbit());
1854     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1855
1856     H_UNLOCK;
1857     return held;
1858
1859 }                               /*h_DumpHost */
1860
1861
1862 void
1863 h_DumpHosts()
1864 {
1865     time_t now;
1866     StreamHandle_t *file = STREAM_OPEN(AFSDIR_SERVER_HOSTDUMP_FILEPATH, "w");
1867     char tmpStr[256];
1868     char tbuffer[32];
1869
1870     if (file == NULL) {
1871         ViceLog(0,
1872                 ("Couldn't create host dump file %s\n",
1873                  AFSDIR_SERVER_HOSTDUMP_FILEPATH));
1874         return;
1875     }
1876     now = FT_ApproxTime();
1877     (void)afs_snprintf(tmpStr, sizeof tmpStr, "List of active hosts at %s\n",
1878                        afs_ctime(&now, tbuffer, sizeof(tbuffer)));
1879     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1880     h_Enumerate(h_DumpHost, (char *)file);
1881     STREAM_REALLYCLOSE(file);
1882     ViceLog(0, ("Created host dump %s\n", AFSDIR_SERVER_HOSTDUMP_FILEPATH));
1883
1884 }                               /*h_DumpHosts */
1885
1886
1887 /*
1888  * This counts the number of workstations, the number of active workstations,
1889  * and the number of workstations declared "down" (i.e. not heard from
1890  * recently).  An active workstation has received a call since the cutoff
1891  * time argument passed.
1892  */
1893 void
1894 h_GetWorkStats(int *nump, int *activep, int *delp, afs_int32 cutofftime)
1895 {
1896     register struct host *host;
1897     register int num = 0, active = 0, del = 0;
1898
1899     H_LOCK;
1900     for (host = hostList; host; host = host->next) {
1901         if (!(host->hostFlags & HOSTDELETED)) {
1902             num++;
1903             if (host->ActiveCall > cutofftime)
1904                 active++;
1905             if (host->hostFlags & VENUSDOWN)
1906                 del++;
1907         }
1908     }
1909     H_UNLOCK;
1910     if (nump)
1911         *nump = num;
1912     if (activep)
1913         *activep = active;
1914     if (delp)
1915         *delp = del;
1916
1917 }                               /*h_GetWorkStats */
1918
1919
1920 /*------------------------------------------------------------------------
1921  * PRIVATE h_ClassifyAddress
1922  *
1923  * Description:
1924  *      Given a target IP address and a candidate IP address (both
1925  *      in host byte order), classify the candidate into one of three
1926  *      buckets in relation to the target by bumping the counters passed
1927  *      in as parameters.
1928  *
1929  * Arguments:
1930  *      a_targetAddr       : Target address.
1931  *      a_candAddr         : Candidate address.
1932  *      a_sameNetOrSubnetP : Ptr to counter to bump when the two
1933  *                           addresses are either in the same network
1934  *                           or the same subnet.
1935  *      a_diffSubnetP      : ...when the candidate is in a different
1936  *                           subnet.
1937  *      a_diffNetworkP     : ...when the candidate is in a different
1938  *                           network.
1939  *
1940  * Returns:
1941  *      Nothing.
1942  *
1943  * Environment:
1944  *      The target and candidate addresses are both in host byte
1945  *      order, NOT network byte order, when passed in.
1946  *
1947  * Side Effects:
1948  *      As advertised.
1949  *------------------------------------------------------------------------*/
1950
1951 static void
1952 h_ClassifyAddress(afs_uint32 a_targetAddr, afs_uint32 a_candAddr,
1953                   afs_int32 * a_sameNetOrSubnetP, afs_int32 * a_diffSubnetP,
1954                   afs_int32 * a_diffNetworkP)
1955 {                               /*h_ClassifyAddress */
1956
1957     afs_uint32 targetNet;
1958     afs_uint32 targetSubnet;
1959     afs_uint32 candNet;
1960     afs_uint32 candSubnet;
1961
1962     /*
1963      * Put bad values into the subnet info to start with.
1964      */
1965     targetSubnet = (afs_uint32) 0;
1966     candSubnet = (afs_uint32) 0;
1967
1968     /*
1969      * Pull out the network and subnetwork numbers from the target
1970      * and candidate addresses.  We can short-circuit this whole
1971      * affair if the target and candidate addresses are not of the
1972      * same class.
1973      */
1974     if (IN_CLASSA(a_targetAddr)) {
1975         if (!(IN_CLASSA(a_candAddr))) {
1976             (*a_diffNetworkP)++;
1977             return;
1978         }
1979         targetNet = a_targetAddr & IN_CLASSA_NET;
1980         candNet = a_candAddr & IN_CLASSA_NET;
1981         if (IN_SUBNETA(a_targetAddr))
1982             targetSubnet = a_targetAddr & IN_CLASSA_SUBNET;
1983         if (IN_SUBNETA(a_candAddr))
1984             candSubnet = a_candAddr & IN_CLASSA_SUBNET;
1985     } else if (IN_CLASSB(a_targetAddr)) {
1986         if (!(IN_CLASSB(a_candAddr))) {
1987             (*a_diffNetworkP)++;
1988             return;
1989         }
1990         targetNet = a_targetAddr & IN_CLASSB_NET;
1991         candNet = a_candAddr & IN_CLASSB_NET;
1992         if (IN_SUBNETB(a_targetAddr))
1993             targetSubnet = a_targetAddr & IN_CLASSB_SUBNET;
1994         if (IN_SUBNETB(a_candAddr))
1995             candSubnet = a_candAddr & IN_CLASSB_SUBNET;
1996     } /*Class B target */
1997     else if (IN_CLASSC(a_targetAddr)) {
1998         if (!(IN_CLASSC(a_candAddr))) {
1999             (*a_diffNetworkP)++;
2000             return;
2001         }
2002         targetNet = a_targetAddr & IN_CLASSC_NET;
2003         candNet = a_candAddr & IN_CLASSC_NET;
2004
2005         /*
2006          * Note that class C addresses can't have subnets,
2007          * so we leave the defaults untouched.
2008          */
2009     } /*Class C target */
2010     else {
2011         targetNet = a_targetAddr;
2012         candNet = a_candAddr;
2013     }                           /*Class D address */
2014
2015     /*
2016      * Now, simply compare the extracted net and subnet values for
2017      * the two addresses (which at this point are known to be of the
2018      * same class)
2019      */
2020     if (targetNet == candNet) {
2021         if (targetSubnet == candSubnet)
2022             (*a_sameNetOrSubnetP)++;
2023         else
2024             (*a_diffSubnetP)++;
2025     } else
2026         (*a_diffNetworkP)++;
2027
2028 }                               /*h_ClassifyAddress */
2029
2030
2031 /*------------------------------------------------------------------------
2032  * EXPORTED h_GetHostNetStats
2033  *
2034  * Description:
2035  *      Iterate through the host table, and classify each (non-deleted)
2036  *      host entry into ``proximity'' categories (same net or subnet,
2037  *      different subnet, different network).
2038  *
2039  * Arguments:
2040  *      a_numHostsP        : Set to total number of (non-deleted) hosts.
2041  *      a_sameNetOrSubnetP : Set to # hosts on same net/subnet as server.
2042  *      a_diffSubnetP      : Set to # hosts on diff subnet as server.
2043  *      a_diffNetworkP     : Set to # hosts on diff network as server.
2044  *
2045  * Returns:
2046  *      Nothing.
2047  *
2048  * Environment:
2049  *      We only count non-deleted hosts.  The storage pointed to by our
2050  *      parameters is zeroed upon entry.
2051  *
2052  * Side Effects:
2053  *      As advertised.
2054  *------------------------------------------------------------------------*/
2055
2056 void
2057 h_GetHostNetStats(afs_int32 * a_numHostsP, afs_int32 * a_sameNetOrSubnetP,
2058                   afs_int32 * a_diffSubnetP, afs_int32 * a_diffNetworkP)
2059 {                               /*h_GetHostNetStats */
2060
2061     register struct host *hostP;        /*Ptr to current host entry */
2062     register afs_uint32 currAddr_HBO;   /*Curr host addr, host byte order */
2063
2064     /*
2065      * Clear out the storage pointed to by our parameters.
2066      */
2067     *a_numHostsP = (afs_int32) 0;
2068     *a_sameNetOrSubnetP = (afs_int32) 0;
2069     *a_diffSubnetP = (afs_int32) 0;
2070     *a_diffNetworkP = (afs_int32) 0;
2071
2072     H_LOCK;
2073     for (hostP = hostList; hostP; hostP = hostP->next) {
2074         if (!(hostP->hostFlags & HOSTDELETED)) {
2075             /*
2076              * Bump the number of undeleted host entries found.
2077              * In classifying the current entry's address, make
2078              * sure to first convert to host byte order.
2079              */
2080             (*a_numHostsP)++;
2081             currAddr_HBO = (afs_uint32) ntohl(hostP->host);
2082             h_ClassifyAddress(FS_HostAddr_HBO, currAddr_HBO,
2083                               a_sameNetOrSubnetP, a_diffSubnetP,
2084                               a_diffNetworkP);
2085         }                       /*Only look at non-deleted hosts */
2086     }                           /*For each host record hashed to this index */
2087     H_UNLOCK;
2088 }                               /*h_GetHostNetStats */
2089
2090 static afs_uint32 checktime;
2091 static afs_uint32 clientdeletetime;
2092 static struct AFSFid zerofid;
2093
2094
2095 /*
2096  * XXXX: This routine could use Multi-Rx to avoid serializing the timeouts.
2097  * Since it can serialize them, and pile up, it should be a separate LWP
2098  * from other events.
2099  */
2100 int
2101 CheckHost(register struct host *host, int held)
2102 {
2103     register struct client *client;
2104     struct rx_connection *cb_conn = NULL;
2105     int code;
2106
2107     /* Host is held by h_Enumerate */
2108     H_LOCK;
2109     for (client = host->FirstClient; client; client = client->next) {
2110         if (client->refCount == 0 && client->LastCall < clientdeletetime) {
2111             client->deleted = 1;
2112             host->hostFlags |= CLIENTDELETED;
2113         }
2114     }
2115     if (host->LastCall < checktime) {
2116         h_Lock_r(host);
2117         cb_conn = host->callback_rxcon;
2118         rx_GetConnection(cb_conn);
2119         if (!(host->hostFlags & HOSTDELETED)) {
2120             if (host->LastCall < clientdeletetime) {
2121                 host->hostFlags |= HOSTDELETED;
2122                 if (!(host->hostFlags & VENUSDOWN)) {
2123                     host->hostFlags &= ~ALTADDR;        /* alternate address invalid */
2124                     if (host->interface) {
2125                         H_UNLOCK;
2126                         code =
2127                             RXAFSCB_InitCallBackState3(cb_conn,
2128                                                        &FS_HostUUID);
2129                         H_LOCK;
2130                     } else {
2131                         H_UNLOCK;
2132                         code =
2133                             RXAFSCB_InitCallBackState(cb_conn);
2134                         H_LOCK;
2135                     }
2136                     host->hostFlags |= ALTADDR; /* alternate addresses valid */
2137                     if (code) {
2138                         char hoststr[16];
2139                         (void)afs_inet_ntoa_r(host->host, hoststr);
2140                         ViceLog(0,
2141                                 ("CB: RCallBackConnectBack (host.c) failed for host %s:%d\n",
2142                                  hoststr, ntohs(host->port)));
2143                         host->hostFlags |= VENUSDOWN;
2144                     }
2145                     /* Note:  it's safe to delete hosts even if they have call
2146                      * back state, because break delayed callbacks (called when a
2147                      * message is received from the workstation) will always send a 
2148                      * break all call backs to the workstation if there is no
2149                      *callback.
2150                      */
2151                 }
2152             } else {
2153                 if (!(host->hostFlags & VENUSDOWN) && host->cblist) {
2154                     if (host->interface) {
2155                         afsUUID uuid = host->interface->uuid;
2156                         H_UNLOCK;
2157                         code = RXAFSCB_ProbeUuid(cb_conn, &uuid);
2158                         H_LOCK;
2159                         if (code) {
2160                             if (MultiProbeAlternateAddress_r(host)) {
2161                                 char hoststr[16];
2162                                 (void)afs_inet_ntoa_r(host->host, hoststr);
2163                                 ViceLog(0,
2164                                         ("ProbeUuid failed for host %s:%d\n",
2165                                          hoststr, ntohs(host->port)));
2166                                 host->hostFlags |= VENUSDOWN;
2167                             }
2168                         }
2169                     } else {
2170                         H_UNLOCK;
2171                         code = RXAFSCB_Probe(cb_conn);
2172                         H_LOCK;
2173                         if (code) {
2174                             char hoststr[16];
2175                             (void)afs_inet_ntoa_r(host->host, hoststr);
2176                             ViceLog(0,
2177                                     ("Probe failed for host %s:%d\n", hoststr,
2178                                      ntohs(host->port)));
2179                             host->hostFlags |= VENUSDOWN;
2180                         }
2181                     }
2182                 }
2183             }
2184         }
2185         H_UNLOCK;
2186         rx_PutConnection(cb_conn);
2187         cb_conn=NULL;
2188         H_LOCK;
2189         h_Unlock_r(host);
2190     }
2191     H_UNLOCK;
2192     return held;
2193
2194 }                               /*CheckHost */
2195
2196
2197 /*
2198  * Set VenusDown for any hosts that have not had a call in 15 minutes and
2199  * don't respond to a probe.  Note that VenusDown can only be cleared if
2200  * a message is received from the host (see ServerLWP in file.c).
2201  * Delete hosts that have not had any calls in 1 hour, clients that
2202  * have not had any calls in 15 minutes.
2203  *
2204  * This routine is called roughly every 5 minutes.
2205  */
2206 void
2207 h_CheckHosts()
2208 {
2209     afs_uint32 now = FT_ApproxTime();
2210
2211     memset((char *)&zerofid, 0, sizeof(zerofid));
2212     /*
2213      * Send a probe to the workstation if it hasn't been heard from in
2214      * 15 minutes
2215      */
2216     checktime = now - 15 * 60;
2217     clientdeletetime = now - 120 * 60;  /* 2 hours ago */
2218     h_Enumerate(CheckHost, NULL);
2219
2220 }                               /*h_CheckHosts */
2221
2222 /*
2223  * This is called with host locked and held. At this point, the
2224  * hostHashTable should not be having entries for the alternate
2225  * interfaces. This function has to insert these entries in the
2226  * hostHashTable.
2227  *
2228  * The addresses in the ineterfaceAddr list are in host byte order.
2229  */
2230 int
2231 initInterfaceAddr_r(struct host *host, struct interfaceAddr *interf)
2232 {
2233     int i, j;
2234     int number, count;
2235     afs_int32 myHost;
2236     int found;
2237     struct Interface *interface;
2238
2239     assert(host);
2240     assert(interf);
2241
2242     ViceLog(125,
2243             ("initInterfaceAddr : host %x numAddr %d\n", host->host,
2244              interf->numberOfInterfaces));
2245
2246     number = interf->numberOfInterfaces;
2247     myHost = host->host;        /* current interface address */
2248
2249     /* validation checks */
2250     if (number < 0 || number > AFS_MAX_INTERFACE_ADDR) {
2251         ViceLog(0,
2252                 ("Number of alternate addresses returned is %d\n", number));
2253         return -1;
2254     }
2255
2256     /*
2257      * Convert IP addresses to network byte order, and remove for
2258      * duplicate IP addresses from the interface list.
2259      */
2260     for (i = 0, count = 0, found = 0; i < number; i++) {
2261         interf->addr_in[i] = htonl(interf->addr_in[i]);
2262         for (j = 0; j < count; j++) {
2263             if (interf->addr_in[j] == interf->addr_in[i])
2264                 break;
2265         }
2266         if (j == count) {
2267             interf->addr_in[count] = interf->addr_in[i];
2268             if (interf->addr_in[count] == myHost)
2269                 found = 1;
2270             count++;
2271         }
2272     }
2273
2274     /*
2275      * Allocate and initialize an interface structure for this host.
2276      */
2277     if (found) {
2278         interface = (struct Interface *)
2279             malloc(sizeof(struct Interface) +
2280                    (sizeof(afs_int32) * (count - 1)));
2281         if (!interface) {
2282             ViceLog(0, ("Failed malloc in initInterfaceAddr_r\n"));
2283             assert(0);
2284         }
2285         interface->numberOfInterfaces = count;
2286     } else {
2287         interface = (struct Interface *)
2288             malloc(sizeof(struct Interface) + (sizeof(afs_int32) * count));
2289         assert(interface);
2290         interface->numberOfInterfaces = count + 1;
2291         interface->addr[count] = myHost;
2292     }
2293     interface->uuid = interf->uuid;
2294     for (i = 0; i < count; i++)
2295         interface->addr[i] = interf->addr_in[i];
2296
2297     assert(!host->interface);
2298     host->interface = interface;
2299
2300     for (i = 0; i < host->interface->numberOfInterfaces; i++) {
2301         ViceLog(125, ("--- alt address %x\n", host->interface->addr[i]));
2302     }
2303
2304     return 0;
2305 }
2306
2307 /* inserts a new HashChain structure corresponding to this address */
2308 void
2309 hashInsert_r(afs_int32 addr, struct host *host)
2310 {
2311     int index;
2312     struct h_hashChain *chain;
2313
2314     /* hash into proper bucket */
2315     index = h_HashIndex(addr);
2316
2317     /* insert into beginning of list for this bucket */
2318     chain = (struct h_hashChain *)malloc(sizeof(struct h_hashChain));
2319     if (!chain) {
2320         ViceLog(0, ("Failed malloc in hashInsert_r\n"));
2321         assert(0);
2322     }
2323     chain->hostPtr = host;
2324     chain->next = hostHashTable[index];
2325     chain->addr = addr;
2326     hostHashTable[index] = chain;
2327
2328 }
2329
2330 /*
2331  * This is called with host locked and held. At this point, the
2332  * hostHashTable should not be having entries for the alternate
2333  * interfaces. This function has to insert these entries in the
2334  * hostHashTable.
2335  *
2336  * All addresses are in network byte order.
2337  */
2338 int
2339 addInterfaceAddr_r(struct host *host, afs_int32 addr)
2340 {
2341     int i;
2342     int number;
2343     int found;
2344     struct Interface *interface;
2345
2346     assert(host);
2347     assert(host->interface);
2348
2349     ViceLog(125, ("addInterfaceAddr : host %x addr %d\n", host->host, addr));
2350
2351     /*
2352      * Make sure this address is on the list of known addresses
2353      * for this host.
2354      */
2355     number = host->interface->numberOfInterfaces;
2356     for (i = 0, found = 0; i < number && !found; i++) {
2357         if (host->interface->addr[i] == addr)
2358             found = 1;
2359     }
2360     if (!found) {
2361         interface = (struct Interface *)
2362             malloc(sizeof(struct Interface) + (sizeof(afs_int32) * number));
2363         if (!interface) {
2364             ViceLog(0, ("Failed malloc in addInterfaceAddr_r\n"));
2365             assert(0);
2366         }
2367         interface->numberOfInterfaces = number + 1;
2368         interface->uuid = host->interface->uuid;
2369         for (i = 0; i < number; i++)
2370             interface->addr[i] = host->interface->addr[i];
2371         interface->addr[number] = addr;
2372         free(host->interface);
2373         host->interface = interface;
2374     }
2375
2376     /*
2377      * Create a hash table entry for this address
2378      */
2379     hashInsert_r(addr, host);
2380
2381     return 0;
2382 }
2383
2384 /* deleted a HashChain structure for this address and host */
2385 /* returns 1 on success */
2386 int
2387 hashDelete_r(afs_int32 addr, struct host *host)
2388 {
2389     int flag;
2390     register struct h_hashChain **hp, *th;
2391
2392     for (hp = &hostHashTable[h_HashIndex(addr)]; (th = *hp);) {
2393         assert(th->hostPtr);
2394         if (th->hostPtr == host && th->addr == addr) {
2395             *hp = th->next;
2396             free(th);
2397             flag = 1;
2398             break;
2399         } else {
2400             hp = &th->next;
2401         }
2402     }
2403     return flag;
2404 }
2405
2406
2407 /*
2408 ** prints out all alternate interface address for the host. The 'level'
2409 ** parameter indicates what level of debugging sets this output
2410 */
2411 void
2412 printInterfaceAddr(struct host *host, int level)
2413 {
2414     int i, number;
2415     if (host->interface) {
2416         /* check alternate addresses */
2417         number = host->interface->numberOfInterfaces;
2418         assert(number > 0);
2419         for (i = 0; i < number; i++)
2420             ViceLog(level, ("%x ", host->interface->addr[i]));
2421     }
2422     ViceLog(level, ("\n"));
2423 }