c768188eeae28c649484c1c28f4823399032573a
[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 haddr, 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(haddr);
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 == haddr
629             && chain->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         struct AddrPort hostAddrPort;
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                 hostAddrPort = host->interface->interface[i];
804
805                 for (hp = &hostHashTable[h_HashIndex(hostAddrPort.addr)]; (th = *hp);
806                      hp = &th->next) {
807                     assert(th->hostPtr);
808                     if (th->hostPtr == host) {
809                         *hp = th->next;
810                         free(th);
811                         break;
812                     }
813                 }
814             }
815             free(host->interface);
816             host->interface = NULL;
817             h_DeleteList_r(host);       /* remove host from global host List */
818             FreeHT(host);
819         }                       /* if alternate address exists */
820     }
821 }                               /*h_TossStuff_r */
822
823
824 /* Called by rx when a server connection disappears */
825 int
826 h_FreeConnection(struct rx_connection *tcon)
827 {
828     register struct client *client;
829
830     client = (struct client *)rx_GetSpecific(tcon, rxcon_client_key);
831     if (client) {
832         H_LOCK;
833         if (client->tcon == tcon)
834             client->tcon = (struct rx_connection *)0;
835         H_UNLOCK;
836     }
837     return 0;
838 }                               /*h_FreeConnection */
839
840
841 /* h_Enumerate: Calls (*proc)(host, held, param) for at least each host in the
842  * system at the start of the enumeration (perhaps more).  Hosts may be deleted
843  * (have delete flag set); ditto for clients.  (*proc) is always called with
844  * host h_held().  The hold state of the host with respect to this lwp is passed
845  * to (*proc) as the param held.  The proc should return 0 if the host should be
846  * released, 1 if it should be held after enumeration.
847  */
848 void
849 h_Enumerate(int (*proc) (), char *param)
850 {
851     register struct host *host, **list;
852     register int *held;
853     register int i, count;
854
855     H_LOCK;
856     if (hostCount == 0) {
857         H_UNLOCK;
858         return;
859     }
860     list = (struct host **)malloc(hostCount * sizeof(struct host *));
861     if (!list) {
862         ViceLog(0, ("Failed malloc in h_Enumerate\n"));
863         assert(0);
864     }
865     held = (int *)malloc(hostCount * sizeof(int));
866     if (!held) {
867         ViceLog(0, ("Failed malloc in h_Enumerate\n"));
868         assert(0);
869     }
870     for (count = 0, host = hostList; host; host = host->next, count++) {
871         list[count] = host;
872         if (!(held[count] = h_Held_r(host)))
873             h_Hold_r(host);
874     }
875     assert(count == hostCount);
876     H_UNLOCK;
877     for (i = 0; i < count; i++) {
878         held[i] = (*proc) (list[i], held[i], param);
879         if (!held[i])
880             h_Release(list[i]); /* this might free up the host */
881     }
882     free((void *)list);
883     free((void *)held);
884 }                               /*h_Enumerate */
885
886 /* h_Enumerate_r (revised):
887  * Calls (*proc)(host, held, param) for each host in hostList, starting
888  * at enumstart
889  * Hosts may be deleted (have delete flag set); ditto for clients.
890  * (*proc) is always called with
891  * host h_held() and the global host lock (H_LOCK) locked.The hold state of the
892  * host with respect to this lwp is passed to (*proc) as the param held.
893  * The proc should return 0 if the host should be released, 1 if it should
894  * be held after enumeration.
895  */
896 void
897 h_Enumerate_r(int (*proc) (), struct host *enumstart, char *param)
898 {
899     register struct host *host, *next;
900     register int held, nheld;
901
902     if (hostCount == 0) {
903         return;
904     }
905     if (!(held = h_Held_r(host)))
906         h_Hold_r(host);
907     for (host = enumstart; host; host = next, held = nheld) {
908         held = (*proc) (host, held, param);
909         next = host->next;
910         if (next && !(nheld = h_Held_r(next)))
911             h_Hold_r(next);
912         if (!held)
913             h_Release_r(host); /* this might free up the host */
914     }
915 }                               /*h_Enumerate_r */
916
917 /* inserts a new HashChain structure corresponding to this UUID */
918 void
919 hashInsertUuid_r(struct afsUUID *uuid, struct host *host)
920 {
921     int index;
922     struct h_hashChain *chain;
923
924     /* hash into proper bucket */
925     index = h_UuidHashIndex(uuid);
926
927     /* insert into beginning of list for this bucket */
928     chain = (struct h_hashChain *)malloc(sizeof(struct h_hashChain));
929     if (!chain) {
930         ViceLog(0, ("Failed malloc in hashInsertUuid_r\n"));
931         assert(0);
932     }
933     assert(chain);
934     chain->hostPtr = host;
935     chain->next = hostUuidHashTable[index];
936     hostUuidHashTable[index] = chain;
937 }
938
939
940 /* inserts a new HashChain structure corresponding to this address */
941 void
942 hashInsert_r(afs_uint32 addr, afs_uint16 port, struct host *host)
943 {
944     int index;
945     struct h_hashChain *chain;
946
947     /* hash into proper bucket */
948     index = h_HashIndex(addr);
949
950     /* insert into beginning of list for this bucket */
951     chain = (struct h_hashChain *)malloc(sizeof(struct h_hashChain));
952     if (!chain) {
953         ViceLog(0, ("Failed malloc in hashInsert_r\n"));
954         assert(0);
955     }
956     chain->hostPtr = host;
957     chain->next = hostHashTable[index];
958     chain->addr = addr;
959     chain->port = port;
960     hostHashTable[index] = chain;
961
962 }
963
964 /*
965  * This is called with host locked and held. At this point, the
966  * hostHashTable should not be having entries for the alternate
967  * interfaces. This function has to insert these entries in the
968  * hostHashTable.
969  *
970  * All addresses are in network byte order.
971  */
972 int
973 addInterfaceAddr_r(struct host *host, afs_uint32 addr, afs_uint16 port)
974 {
975     int i;
976     int number;
977     int found;
978     struct Interface *interface;
979
980     assert(host);
981     assert(host->interface);
982
983     ViceLog(125, ("addInterfaceAddr : host %x addr %x:%d\n", host->host, addr, ntohs(port)));
984
985     /*
986      * Make sure this address is on the list of known addresses
987      * for this host.
988      */
989     number = host->interface->numberOfInterfaces;
990     for (i = 0, found = 0; i < number && !found; i++) {
991         if (host->interface->interface[i].addr == addr &&
992             host->interface->interface[i].port == port)
993             found = 1;
994     }
995     if (!found) {
996         interface = (struct Interface *)
997             malloc(sizeof(struct Interface) + (sizeof(struct AddrPort) * number));
998         if (!interface) {
999             ViceLog(0, ("Failed malloc in addInterfaceAddr_r\n"));
1000             assert(0);
1001         }
1002         interface->numberOfInterfaces = number + 1;
1003         interface->uuid = host->interface->uuid;
1004         for (i = 0; i < number; i++)
1005             interface->interface[i] = host->interface->interface[i];
1006         interface->interface[number].addr = addr;
1007         interface->interface[number].port = port;
1008         free(host->interface);
1009         host->interface = interface;
1010     }
1011
1012     /*
1013      * Create a hash table entry for this address
1014      */
1015     hashInsert_r(addr, port, host);
1016
1017     return 0;
1018 }
1019
1020
1021 /* Host is returned held */
1022 struct host *
1023 h_GetHost_r(struct rx_connection *tcon)
1024 {
1025     struct host *host;
1026     struct host *oldHost;
1027     int code;
1028     int held;
1029     struct interfaceAddr interf;
1030     int interfValid = 0;
1031     struct Identity *identP = NULL;
1032     afs_int32 haddr;
1033     afs_int16 hport;
1034     char hoststr[16], hoststr2[16];
1035     Capabilities caps;
1036     struct rx_connection *cb_conn = NULL;
1037
1038     caps.Capabilities_val = NULL;
1039
1040     haddr = rxr_HostOf(tcon);
1041     hport = rxr_PortOf(tcon);
1042   retry:
1043     if (caps.Capabilities_val)
1044         free(caps.Capabilities_val);
1045     caps.Capabilities_val = NULL;
1046     caps.Capabilities_len = 0;
1047
1048     code = 0;
1049     host = h_Lookup_r(haddr, hport, &held);
1050     identP = (struct Identity *)rx_GetSpecific(tcon, rxcon_ident_key);
1051     if (host && !identP && !(host->Console & 1)) {
1052         /* This is a new connection, and we already have a host
1053          * structure for this address. Verify that the identity
1054          * of the caller matches the identity in the host structure.
1055          */
1056         h_Lock_r(host);
1057         if (!(host->hostFlags & ALTADDR)) {
1058             /* Another thread is doing initialization */
1059             h_Unlock_r(host);
1060             if (!held)
1061                 h_Release_r(host);
1062             ViceLog(125,
1063                     ("Host %s:%d starting h_Lookup again\n",
1064                      afs_inet_ntoa_r(host->host, hoststr),
1065                      ntohs(host->port)));
1066             goto retry;
1067         }
1068         host->hostFlags &= ~ALTADDR;
1069         cb_conn = host->callback_rxcon;
1070         rx_GetConnection(cb_conn);
1071         H_UNLOCK;
1072         code =
1073             RXAFSCB_TellMeAboutYourself(cb_conn, &interf, &caps);
1074         if (code == RXGEN_OPCODE)
1075             code = RXAFSCB_WhoAreYou(cb_conn, &interf);
1076         rx_PutConnection(cb_conn);
1077         cb_conn=NULL;
1078         H_LOCK;
1079         if (code == RXGEN_OPCODE) {
1080             identP = (struct Identity *)malloc(sizeof(struct Identity));
1081             if (!identP) {
1082                 ViceLog(0, ("Failed malloc in h_GetHost_r\n"));
1083                 assert(0);
1084             }
1085             identP->valid = 0;
1086             rx_SetSpecific(tcon, rxcon_ident_key, identP);
1087             /* The host on this connection was unable to respond to 
1088              * the WhoAreYou. We will treat this as a new connection
1089              * from the existing host. The worst that can happen is
1090              * that we maintain some extra callback state information */
1091             if (host->interface) {
1092                 ViceLog(0,
1093                         ("Host %s:%d used to support WhoAreYou, deleting.\n",
1094                          afs_inet_ntoa_r(host->host, hoststr),
1095                          ntohs(host->port)));
1096                 host->hostFlags |= HOSTDELETED;
1097                 h_Unlock_r(host);
1098                 if (!held)
1099                     h_Release_r(host);
1100                 host = NULL;
1101                 goto retry;
1102             }
1103         } else if (code == 0) {
1104             interfValid = 1;
1105             identP = (struct Identity *)malloc(sizeof(struct Identity));
1106             if (!identP) {
1107                 ViceLog(0, ("Failed malloc in h_GetHost_r\n"));
1108                 assert(0);
1109             }
1110             identP->valid = 1;
1111             identP->uuid = interf.uuid;
1112             rx_SetSpecific(tcon, rxcon_ident_key, identP);
1113             /* Check whether the UUID on this connection matches
1114              * the UUID in the host structure. If they don't match
1115              * then this is not the same host as before. */
1116             if (!host->interface
1117                 || !afs_uuid_equal(&interf.uuid, &host->interface->uuid)) {
1118                 ViceLog(25,
1119                         ("Host %s:%d has changed its identity, deleting.\n",
1120                          afs_inet_ntoa_r(host->host, hoststr), host->port));
1121                 host->hostFlags |= HOSTDELETED;
1122                 h_Unlock_r(host);
1123                 if (!held)
1124                     h_Release_r(host);
1125                 host = NULL;
1126                 goto retry;
1127             }
1128         } else {
1129             afs_inet_ntoa_r(host->host, hoststr);
1130             ViceLog(0,
1131                     ("CB: WhoAreYou failed for %s:%d, error %d\n", hoststr,
1132                      ntohs(host->port), code));
1133             host->hostFlags |= VENUSDOWN;
1134         }
1135         if (caps.Capabilities_val
1136             && (caps.Capabilities_val[0] & CAPABILITY_ERRORTRANS))
1137             host->hostFlags |= HERRORTRANS;
1138         else
1139             host->hostFlags &= ~(HERRORTRANS);
1140         host->hostFlags |= ALTADDR;
1141         h_Unlock_r(host);
1142     } else if (host) {
1143         if (!(host->hostFlags & ALTADDR)) {
1144             /* another thread is doing the initialisation */
1145             ViceLog(125,
1146                     ("Host %s:%d waiting for host-init to complete\n",
1147                      afs_inet_ntoa_r(host->host, hoststr),
1148                      ntohs(host->port)));
1149             h_Lock_r(host);
1150             h_Unlock_r(host);
1151             if (!held)
1152                 h_Release_r(host);
1153             ViceLog(125,
1154                     ("Host %s:%d starting h_Lookup again\n",
1155                      afs_inet_ntoa_r(host->host, hoststr),
1156                      ntohs(host->port)));
1157             goto retry;
1158         }
1159         /* We need to check whether the identity in the host structure
1160          * matches the identity on the connection. If they don't match
1161          * then treat this a new host. */
1162         if (!(host->Console & 1)
1163             && ((!identP->valid && host->interface)
1164                 || (identP->valid && !host->interface)
1165                 || (identP->valid
1166                     && !afs_uuid_equal(&identP->uuid,
1167                                        &host->interface->uuid)))) {
1168             char uuid1[128], uuid2[128];
1169             if (identP->valid)
1170                 afsUUID_to_string(&identP->uuid, uuid1, 127);
1171             if (host->interface)
1172                 afsUUID_to_string(&host->interface->uuid, uuid2, 127);
1173             ViceLog(0,
1174                     ("CB: new identity for host %s:%d, deleting(%x %x %s %s)\n",
1175                      afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port),
1176                      identP->valid, host->interface,
1177                      identP->valid ? uuid1 : "",
1178                      host->interface ? uuid2 : ""));
1179
1180             /* The host in the cache is not the host for this connection */
1181             host->hostFlags |= HOSTDELETED;
1182             h_Unlock_r(host);
1183             if (!held)
1184                 h_Release_r(host);
1185             goto retry;
1186         }
1187     } else {
1188         host = h_Alloc_r(tcon); /* returned held and locked */
1189         h_gethostcps_r(host, FT_ApproxTime());
1190         if (!(host->Console & 1)) {
1191             int pident = 0;
1192             cb_conn = host->callback_rxcon;
1193             rx_GetConnection(cb_conn);
1194             H_UNLOCK;
1195             code =
1196                 RXAFSCB_TellMeAboutYourself(cb_conn, &interf, &caps);
1197             if (code == RXGEN_OPCODE)
1198                 code = RXAFSCB_WhoAreYou(cb_conn, &interf);
1199             rx_PutConnection(cb_conn);
1200             cb_conn=NULL;
1201             H_LOCK;
1202             if (code == RXGEN_OPCODE) {
1203                 if (!identP)
1204                     identP =
1205                         (struct Identity *)malloc(sizeof(struct Identity));
1206                 else
1207                     pident = 1;
1208
1209                 if (!identP) {
1210                     ViceLog(0, ("Failed malloc in h_GetHost_r\n"));
1211                     assert(0);
1212                 }
1213                 identP->valid = 0;
1214                 if (!pident)
1215                     rx_SetSpecific(tcon, rxcon_ident_key, identP);
1216                 ViceLog(25,
1217                         ("Host %s:%d does not support WhoAreYou.\n",
1218                          afs_inet_ntoa_r(host->host, hoststr),
1219                          ntohs(host->port)));
1220                 code = 0;
1221             } else if (code == 0) {
1222                 if (!identP)
1223                     identP =
1224                         (struct Identity *)malloc(sizeof(struct Identity));
1225                 else
1226                     pident = 1;
1227
1228                 if (!identP) {
1229                     ViceLog(0, ("Failed malloc in h_GetHost_r\n"));
1230                     assert(0);
1231                 }
1232                 identP->valid = 1;
1233                 interfValid = 1;
1234                 identP->uuid = interf.uuid;
1235                 if (!pident)
1236                     rx_SetSpecific(tcon, rxcon_ident_key, identP);
1237                 ViceLog(25,
1238                         ("WhoAreYou success on %s:%d\n",
1239                          afs_inet_ntoa_r(host->host, hoststr),
1240                          ntohs(host->port)));
1241             }
1242             if (code == 0 && !identP->valid) {
1243                 cb_conn = host->callback_rxcon;
1244                 rx_GetConnection(cb_conn);
1245                 H_UNLOCK;
1246                 code = RXAFSCB_InitCallBackState(cb_conn);
1247                 rx_PutConnection(cb_conn);
1248                 cb_conn=NULL;
1249                 H_LOCK;
1250             } else if (code == 0) {
1251                 oldHost = h_LookupUuid_r(&identP->uuid);
1252                 if (oldHost) {
1253                     int probefail = 0;
1254
1255                     if (!(held = h_Held_r(oldHost)))
1256                         h_Hold_r(oldHost);
1257                     h_Lock_r(oldHost);
1258
1259                     if (oldHost->interface) {
1260                         afsUUID uuid = oldHost->interface->uuid;
1261                         cb_conn = host->callback_rxcon;
1262                         rx_GetConnection(cb_conn);
1263                         H_UNLOCK;
1264                         code = RXAFSCB_ProbeUuid(cb_conn, &uuid);
1265                         rx_PutConnection(cb_conn);
1266                         cb_conn=NULL;
1267                         H_LOCK;
1268                         if (code && MultiProbeAlternateAddress_r(oldHost)) {
1269                             probefail = 1;
1270                         }
1271                     } else {
1272                         probefail = 1;
1273                     }
1274
1275                     if (probefail) {
1276                         /* The old host is either does not have a Uuid,
1277                          * is not responding to Probes, 
1278                          * or does not have a matching Uuid. 
1279                          * Delete it! */
1280                         oldHost->hostFlags |= HOSTDELETED;
1281                         h_Unlock_r(oldHost);
1282                         h_Release_r(oldHost);
1283                         oldHost = NULL;
1284                     }
1285                 }
1286                 if (oldHost) {
1287                     /* This is a new address for an existing host. Update
1288                      * the list of interfaces for the existing host and
1289                      * delete the host structure we just allocated. */
1290                     ViceLog(25,
1291                             ("CB: new addr %s:%d for old host %s:%d\n",
1292                              afs_inet_ntoa_r(host->host, hoststr),
1293                              ntohs(host->port), afs_inet_ntoa_r(oldHost->host,
1294                                                                 hoststr2),
1295                              ntohs(oldHost->port)));
1296                     addInterfaceAddr_r(oldHost, haddr, hport);
1297                     host->hostFlags |= HOSTDELETED;
1298                     h_Unlock_r(host);
1299                     h_Release_r(host);
1300                     host = oldHost;
1301                 } else {
1302                     /* This really is a new host */
1303                     hashInsertUuid_r(&identP->uuid, host);
1304                     cb_conn = host->callback_rxcon;
1305                     rx_GetConnection(cb_conn);          
1306                     H_UNLOCK;
1307                     code =
1308                         RXAFSCB_InitCallBackState3(cb_conn,
1309                                                    &FS_HostUUID);
1310                     rx_PutConnection(cb_conn);
1311                     cb_conn=NULL;
1312                     H_LOCK;
1313                     if (code == 0) {
1314                         ViceLog(25,
1315                                 ("InitCallBackState3 success on %s:%d\n",
1316                                  afs_inet_ntoa_r(host->host, hoststr),
1317                                  ntohs(host->port)));
1318                         assert(interfValid == 1);
1319                         initInterfaceAddr_r(host, &interf);
1320                     }
1321                 }
1322             }
1323             if (code) {
1324                 afs_inet_ntoa_r(host->host, hoststr);
1325                 ViceLog(0,
1326                         ("CB: RCallBackConnectBack failed for %s:%d\n",
1327                          hoststr, ntohs(host->port)));
1328                 host->hostFlags |= VENUSDOWN;
1329             } else
1330                 host->hostFlags |= RESETDONE;
1331
1332         }
1333         if (caps.Capabilities_val
1334             && (caps.Capabilities_val[0] & CAPABILITY_ERRORTRANS))
1335             host->hostFlags |= HERRORTRANS;
1336         else
1337             host->hostFlags &= ~(HERRORTRANS);
1338         host->hostFlags |= ALTADDR;     /* host structure initialization complete */
1339         h_Unlock_r(host);
1340     }
1341     if (caps.Capabilities_val)
1342         free(caps.Capabilities_val);
1343     caps.Capabilities_val = NULL;
1344     caps.Capabilities_len = 0;
1345     return host;
1346
1347 }                               /*h_GetHost_r */
1348
1349
1350 static char localcellname[PR_MAXNAMELEN + 1];
1351 char local_realm[AFS_REALM_SZ] = "";
1352
1353 /* not reentrant */
1354 void
1355 h_InitHostPackage()
1356 {
1357     afsconf_GetLocalCell(confDir, localcellname, PR_MAXNAMELEN);
1358     if (!local_realm[0]) {
1359         if (afs_krb_get_lrealm(local_realm, 0) != 0 /*KSUCCESS*/) {
1360             ViceLog(0,
1361                     ("afs_krb_get_lrealm failed, using %s.\n",
1362                      localcellname));
1363             strcpy(local_realm, localcellname);
1364         }
1365     }
1366     rxcon_ident_key = rx_KeyCreate((rx_destructor_t) free);
1367     rxcon_client_key = rx_KeyCreate((rx_destructor_t) 0);
1368 #ifdef AFS_PTHREAD_ENV
1369     assert(pthread_mutex_init(&host_glock_mutex, NULL) == 0);
1370 #endif /* AFS_PTHREAD_ENV */
1371 }
1372
1373 static int
1374 MapName_r(char *aname, char *acell, afs_int32 * aval)
1375 {
1376     namelist lnames;
1377     idlist lids;
1378     afs_int32 code;
1379     afs_int32 anamelen, cnamelen;
1380     int foreign = 0;
1381     char *tname;
1382
1383     anamelen = strlen(aname);
1384     if (anamelen >= PR_MAXNAMELEN)
1385         return -1;              /* bad name -- caller interprets this as anonymous, but retries later */
1386
1387     lnames.namelist_len = 1;
1388     lnames.namelist_val = (prname *) aname;     /* don't malloc in the common case */
1389     lids.idlist_len = 0;
1390     lids.idlist_val = NULL;
1391
1392     cnamelen = strlen(acell);
1393     if (cnamelen) {
1394         if (strcasecmp(local_realm, acell)
1395             && strcasecmp(localcellname, acell)) {
1396             ViceLog(2,
1397                     ("MapName: cell is foreign.  cell=%s, localcell=%s, localrealm=%s\n",
1398                      acell, localcellname, local_realm));
1399             if ((anamelen + cnamelen + 1) >= PR_MAXNAMELEN) {
1400                 ViceLog(2,
1401                         ("MapName: Name too long, using AnonymousID for %s@%s\n",
1402                          aname, acell));
1403                 *aval = AnonymousID;
1404                 return 0;
1405             }
1406             foreign = 1;        /* attempt cross-cell authentication */
1407             tname = (char *)malloc(PR_MAXNAMELEN);
1408             if (!tname) {
1409                 ViceLog(0, ("Failed malloc in MapName_r\n"));
1410                 assert(0);
1411             }
1412             strcpy(tname, aname);
1413             tname[anamelen] = '@';
1414             strcpy(tname + anamelen + 1, acell);
1415             lnames.namelist_val = (prname *) tname;
1416         }
1417     }
1418
1419     H_UNLOCK;
1420     code = pr_NameToId(&lnames, &lids);
1421     H_LOCK;
1422     if (code == 0) {
1423         if (lids.idlist_val) {
1424             *aval = lids.idlist_val[0];
1425             if (*aval == AnonymousID) {
1426                 ViceLog(2,
1427                         ("MapName: NameToId on %s returns anonymousID\n",
1428                          lnames.namelist_val));
1429             }
1430             free(lids.idlist_val);      /* return parms are not malloced in stub if server proc aborts */
1431         } else {
1432             ViceLog(0,
1433                     ("MapName: NameToId on '%s' is unknown\n",
1434                      lnames.namelist_val));
1435             code = -1;
1436         }
1437     }
1438
1439     if (foreign) {
1440         free(lnames.namelist_val);      /* We allocated this above, so we must free it now. */
1441     }
1442     return code;
1443 }
1444
1445 /*MapName*/
1446
1447
1448 /* NOTE: this returns the client with a Write lock */
1449 struct client *
1450 h_ID2Client(afs_int32 vid)
1451 {
1452     register struct client *client;
1453     register struct host *host;
1454
1455     H_LOCK;
1456     for (host = hostList; host; host = host->next) {
1457         if (host->hostFlags & HOSTDELETED)
1458             continue;
1459         for (client = host->FirstClient; client; client = client->next) {
1460             if (!client->deleted && client->ViceId == vid) {
1461                 client->refCount++;
1462                 H_UNLOCK;
1463                 ObtainWriteLock(&client->lock);
1464                 H_LOCK;
1465                 client->refCount--;
1466                 H_UNLOCK;
1467                 return client;
1468             }
1469         }
1470     }
1471
1472     H_UNLOCK;
1473     return NULL;
1474 }
1475
1476 /*
1477  * Called by the server main loop.  Returns a h_Held client, which must be
1478  * released later the main loop.  Allocates a client if the matching one
1479  * isn't around. The client is returned with its reference count incremented
1480  * by one. The caller must call h_ReleaseClient_r when finished with
1481  * the client.
1482  */
1483 struct client *
1484 h_FindClient_r(struct rx_connection *tcon)
1485 {
1486     register struct client *client;
1487     register struct host *host;
1488     struct client *oldClient;
1489     afs_int32 viceid;
1490     afs_int32 expTime;
1491     afs_int32 code;
1492     int authClass;
1493 #if (64-MAXKTCNAMELEN)
1494     ticket name length != 64
1495 #endif
1496     char tname[64];
1497     char tinst[64];
1498     char uname[PR_MAXNAMELEN];
1499     char tcell[MAXKTCREALMLEN];
1500     int fail = 0;
1501
1502     client = (struct client *)rx_GetSpecific(tcon, rxcon_client_key);
1503     if (client && !client->deleted) {
1504         client->refCount++;
1505         h_Hold_r(client->host);
1506         if (client->prfail != 2) {      /* Could add shared lock on client here */
1507             /* note that we don't have to lock entry in this path to
1508              * ensure CPS is initialized, since we don't call rx_SetSpecific
1509              * until initialization is done, and we only get here if
1510              * rx_GetSpecific located the client structure.
1511              */
1512             return client;
1513         }
1514         H_UNLOCK;
1515         ObtainWriteLock(&client->lock); /* released at end */
1516         H_LOCK;
1517     } else if (client) {
1518         client->refCount++;
1519     }
1520
1521     authClass = rx_SecurityClassOf((struct rx_connection *)tcon);
1522     ViceLog(5,
1523             ("FindClient: authenticating connection: authClass=%d\n",
1524              authClass));
1525     if (authClass == 1) {
1526         /* A bcrypt tickets, no longer supported */
1527         ViceLog(1, ("FindClient: bcrypt ticket, using AnonymousID\n"));
1528         viceid = AnonymousID;
1529         expTime = 0x7fffffff;
1530     } else if (authClass == 2) {
1531         afs_int32 kvno;
1532
1533         /* kerberos ticket */
1534         code = rxkad_GetServerInfo(tcon, /*level */ 0, &expTime,
1535                                    tname, tinst, tcell, &kvno);
1536         if (code) {
1537             ViceLog(1, ("Failed to get rxkad ticket info\n"));
1538             viceid = AnonymousID;
1539             expTime = 0x7fffffff;
1540         } else {
1541             int ilen = strlen(tinst);
1542             ViceLog(5,
1543                     ("FindClient: rxkad conn: name=%s,inst=%s,cell=%s,exp=%d,kvno=%d\n",
1544                      tname, tinst, tcell, expTime, kvno));
1545             strncpy(uname, tname, sizeof(uname));
1546             if (ilen) {
1547                 if (strlen(uname) + 1 + ilen >= sizeof(uname))
1548                     goto bad_name;
1549                 strcat(uname, ".");
1550                 strcat(uname, tinst);
1551             }
1552             /* translate the name to a vice id */
1553             code = MapName_r(uname, tcell, &viceid);
1554             if (code) {
1555               bad_name:
1556                 ViceLog(1,
1557                         ("failed to map name=%s, cell=%s -> code=%d\n", uname,
1558                          tcell, code));
1559                 fail = 1;
1560                 viceid = AnonymousID;
1561                 expTime = 0x7fffffff;
1562             }
1563         }
1564     } else {
1565         viceid = AnonymousID;   /* unknown security class */
1566         expTime = 0x7fffffff;
1567     }
1568
1569     if (!client) {
1570         host = h_GetHost_r(tcon);       /* Returns it h_Held */
1571
1572     retryfirstclient:
1573         /* First try to find the client structure */
1574         for (client = host->FirstClient; client; client = client->next) {
1575             if (!client->deleted && (client->sid == rxr_CidOf(tcon))
1576                 && (client->VenusEpoch == rxr_GetEpoch(tcon))) {
1577                 if (client->tcon && (client->tcon != tcon)) {
1578                     ViceLog(0,
1579                             ("*** Vid=%d, sid=%x, tcon=%x, Tcon=%x ***\n",
1580                              client->ViceId, client->sid, client->tcon,
1581                              tcon));
1582                     oldClient =
1583                         (struct client *)rx_GetSpecific(client->tcon,
1584                                                         rxcon_client_key);
1585                     if (oldClient) {
1586                         if (oldClient == client)
1587                             rx_SetSpecific(client->tcon, rxcon_client_key,
1588                                            NULL);
1589                         else
1590                             ViceLog(0,
1591                                     ("Client-conn mismatch: CL1=%x, CN=%x, CL2=%x\n",
1592                                      client, client->tcon, oldClient));
1593                     }
1594                     client->tcon = (struct rx_connection *)0;
1595                 }
1596                 client->refCount++;
1597                 H_UNLOCK;
1598                 ObtainWriteLock(&client->lock);
1599                 H_LOCK;
1600                 break;
1601             }
1602         }
1603
1604         /* Still no client structure - get one */
1605         if (!client) {
1606             h_Lock_r(host);
1607             /* Retry to find the client structure */
1608             for (client = host->FirstClient; client; client = client->next) {
1609                 if (!client->deleted && (client->sid == rxr_CidOf(tcon))
1610                     && (client->VenusEpoch == rxr_GetEpoch(tcon))) {
1611                     h_Unlock_r(host);
1612                     goto retryfirstclient;
1613                 }
1614             }
1615             client = GetCE();
1616             ObtainWriteLock(&client->lock);
1617             client->refCount = 1;
1618             client->host = host;
1619             client->next = host->FirstClient;
1620             host->FirstClient = client;
1621 #if FS_STATS_DETAILED
1622             client->InSameNetwork = host->InSameNetwork;
1623 #endif /* FS_STATS_DETAILED */
1624             client->ViceId = viceid;
1625             client->expTime = expTime;  /* rx only */
1626             client->authClass = authClass;      /* rx only */
1627             client->sid = rxr_CidOf(tcon);
1628             client->VenusEpoch = rxr_GetEpoch(tcon);
1629             client->CPS.prlist_val = 0;
1630             client->CPS.prlist_len = 0;
1631             h_Unlock_r(host);
1632             CurrentConnections++;       /* increment number of connections */
1633         }
1634     }
1635     client->prfail = fail;
1636
1637     if (!(client->CPS.prlist_val) || (viceid != client->ViceId)) {
1638         if (client->CPS.prlist_val && (client->ViceId != ANONYMOUSID)) {
1639             free(client->CPS.prlist_val);
1640         }
1641         client->CPS.prlist_val = NULL;
1642         client->CPS.prlist_len = 0;
1643         client->ViceId = viceid;
1644         client->expTime = expTime;
1645
1646         if (viceid == ANONYMOUSID) {
1647             client->CPS.prlist_len = AnonCPS.prlist_len;
1648             client->CPS.prlist_val = AnonCPS.prlist_val;
1649         } else {
1650             H_UNLOCK;
1651             code = pr_GetCPS(viceid, &client->CPS);
1652             H_LOCK;
1653             if (code) {
1654                 char hoststr[16];
1655                 ViceLog(0,
1656                         ("pr_GetCPS failed(%d) for user %d, host %s:%d\n",
1657                          code, viceid, afs_inet_ntoa_r(client->host->host,
1658                                                        hoststr),
1659                          ntohs(client->host->port)));
1660
1661                 /* Although ubik_Call (called by pr_GetCPS) traverses thru
1662                  * all protection servers and reevaluates things if no
1663                  * sync server or quorum is found we could still end up
1664                  * with one of these errors. In such case we would like to
1665                  * reevaluate the rpc call to find if there's cps for this
1666                  * guy. We treat other errors (except network failures
1667                  * ones - i.e. code < 0) as an indication that there is no
1668                  * CPS for this host.  Ideally we could like to deal this
1669                  * problem the other way around (i.e.  if code == NOCPS
1670                  * ignore else retry next time) but the problem is that
1671                  * there're other errors (i.e.  EPERM) for which we don't
1672                  * want to retry and we don't know the whole code list!
1673                  */
1674                 if (code < 0 || code == UNOQUORUM || code == UNOTSYNC)
1675                     client->prfail = 1;
1676             }
1677         }
1678         /* the disabling of system:administrators is so iffy and has so many
1679          * possible failure modes that we will disable it again */
1680         /* Turn off System:Administrator for safety  
1681          * if (AL_IsAMember(SystemId, client->CPS) == 0)
1682          * assert(AL_DisableGroup(SystemId, client->CPS) == 0); */
1683     }
1684
1685     /* Now, tcon may already be set to a rock, since we blocked with no host
1686      * or client locks set above in pr_GetCPS (XXXX some locking is probably
1687      * required).  So, before setting the RPC's rock, we should disconnect
1688      * the RPC from the other client structure's rock.
1689      */
1690     oldClient = (struct client *)rx_GetSpecific(tcon, rxcon_client_key);
1691     if (oldClient && oldClient->tcon == tcon) {
1692         oldClient->tcon = (struct rx_connection *)0;
1693         ViceLog(0, ("FindClient: client %x(%x) already had conn %x (host %x), stolen by client %x(%x)\n", 
1694                     oldClient, oldClient->sid, tcon, 
1695                     rx_HostOf(rx_PeerOf(tcon)), client, client->sid));
1696         /* rx_SetSpecific will be done immediately below */
1697     }
1698     client->tcon = tcon;
1699     rx_SetSpecific(tcon, rxcon_client_key, client);
1700     ReleaseWriteLock(&client->lock);
1701
1702     return client;
1703
1704 }                               /*h_FindClient_r */
1705
1706 int
1707 h_ReleaseClient_r(struct client *client)
1708 {
1709     assert(client->refCount > 0);
1710     client->refCount--;
1711     return 0;
1712 }
1713
1714
1715 /*
1716  * Sigh:  this one is used to get the client AGAIN within the individual
1717  * server routines.  This does not bother h_Holding the host, since
1718  * this is assumed already have been done by the server main loop.
1719  * It does check tokens, since only the server routines can return the
1720  * VICETOKENDEAD error code
1721  */
1722 int
1723 GetClient(struct rx_connection *tcon, struct client **cp)
1724 {
1725     register struct client *client;
1726
1727     H_LOCK;
1728     *cp = client = (struct client *)rx_GetSpecific(tcon, rxcon_client_key);
1729     if (client == NULL || client->tcon == NULL) {
1730         ViceLog(0,
1731                 ("GetClient: no client in conn %x (host %x), VBUSYING\n",
1732                  tcon, rx_HostOf(rx_PeerOf(tcon))));
1733         H_UNLOCK;
1734         return VBUSY;
1735     }
1736     if (rxr_CidOf(client->tcon) != client->sid) {
1737         ViceLog(0,
1738                 ("GetClient: tcon %x tcon sid %d client sid %d\n",
1739                  client->tcon, rxr_CidOf(client->tcon), client->sid));
1740         H_UNLOCK;
1741         return VBUSY;
1742     }
1743     if (!(client && client->tcon && rxr_CidOf(client->tcon) == client->sid)) {
1744         if (!client)
1745             ViceLog(0, ("GetClient: no client in conn %x\n", tcon));
1746         else
1747             ViceLog(0,
1748                     ("GetClient: tcon %x tcon sid %d client sid %d\n",
1749                      client->tcon, client->tcon ? rxr_CidOf(client->tcon)
1750                      : -1, client->sid));
1751         assert(0);
1752     }
1753     if (client && client->LastCall > client->expTime && client->expTime) {
1754         char hoststr[16];
1755         ViceLog(1,
1756                 ("Token for %s at %s:%d expired %d\n", h_UserName(client),
1757                  afs_inet_ntoa_r(client->host->host, hoststr),
1758                  ntohs(client->host->port), client->expTime));
1759         H_UNLOCK;
1760         return VICETOKENDEAD;
1761     }
1762
1763     H_UNLOCK;
1764     return 0;
1765
1766 }                               /*GetClient */
1767
1768
1769 /* Client user name for short term use.  Note that this is NOT inexpensive */
1770 char *
1771 h_UserName(struct client *client)
1772 {
1773     static char User[PR_MAXNAMELEN + 1];
1774     namelist lnames;
1775     idlist lids;
1776
1777     lids.idlist_len = 1;
1778     lids.idlist_val = (afs_int32 *) malloc(1 * sizeof(afs_int32));
1779     if (!lids.idlist_val) {
1780         ViceLog(0, ("Failed malloc in h_UserName\n"));
1781         assert(0);
1782     }
1783     lnames.namelist_len = 0;
1784     lnames.namelist_val = (prname *) 0;
1785     lids.idlist_val[0] = client->ViceId;
1786     if (pr_IdToName(&lids, &lnames)) {
1787         /* We need to free id we alloced above! */
1788         free(lids.idlist_val);
1789         return "*UNKNOWN USER NAME*";
1790     }
1791     strncpy(User, lnames.namelist_val[0], PR_MAXNAMELEN);
1792     free(lids.idlist_val);
1793     free(lnames.namelist_val);
1794     return User;
1795
1796 }                               /*h_UserName */
1797
1798
1799 void
1800 h_PrintStats()
1801 {
1802     ViceLog(0,
1803             ("Total Client entries = %d, blocks = %d; Host entries = %d, blocks = %d\n",
1804              CEs, CEBlocks, HTs, HTBlocks));
1805
1806 }                               /*h_PrintStats */
1807
1808
1809 static int
1810 h_PrintClient(register struct host *host, int held, StreamHandle_t * file)
1811 {
1812     register struct client *client;
1813     int i;
1814     char tmpStr[256];
1815     char tbuffer[32];
1816     char hoststr[16];
1817
1818     H_LOCK;
1819     if (host->hostFlags & HOSTDELETED) {
1820         H_UNLOCK;
1821         return held;
1822     }
1823     (void)afs_snprintf(tmpStr, sizeof tmpStr,
1824                        "Host %s:%d down = %d, LastCall %s",
1825                        afs_inet_ntoa_r(host->host, hoststr),
1826                        ntohs(host->port), (host->hostFlags & VENUSDOWN),
1827                        afs_ctime((time_t *) & host->LastCall, tbuffer,
1828                                  sizeof(tbuffer)));
1829     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1830     for (client = host->FirstClient; client; client = client->next) {
1831         if (!client->deleted) {
1832             if (client->tcon) {
1833                 (void)afs_snprintf(tmpStr, sizeof tmpStr,
1834                                    "    user id=%d,  name=%s, sl=%s till %s",
1835                                    client->ViceId, h_UserName(client),
1836                                    client->
1837                                    authClass ? "Authenticated" :
1838                                    "Not authenticated",
1839                                    client->
1840                                    authClass ? afs_ctime((time_t *) & client->
1841                                                          expTime, tbuffer,
1842                                                          sizeof(tbuffer))
1843                                    : "No Limit\n");
1844                 (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1845             } else {
1846                 (void)afs_snprintf(tmpStr, sizeof tmpStr,
1847                                    "    user=%s, no current server connection\n",
1848                                    h_UserName(client));
1849                 (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1850             }
1851             (void)afs_snprintf(tmpStr, sizeof tmpStr, "      CPS-%d is [",
1852                                client->CPS.prlist_len);
1853             (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1854             if (client->CPS.prlist_val) {
1855                 for (i = 0; i > client->CPS.prlist_len; i++) {
1856                     (void)afs_snprintf(tmpStr, sizeof tmpStr, " %d",
1857                                        client->CPS.prlist_val[i]);
1858                     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1859                 }
1860             }
1861             sprintf(tmpStr, "]\n");
1862             (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1863         }
1864     }
1865     H_UNLOCK;
1866     return held;
1867
1868 }                               /*h_PrintClient */
1869
1870
1871
1872 /*
1873  * Print a list of clients, with last security level and token value seen,
1874  * if known
1875  */
1876 void
1877 h_PrintClients()
1878 {
1879     time_t now;
1880     char tmpStr[256];
1881     char tbuffer[32];
1882
1883     StreamHandle_t *file = STREAM_OPEN(AFSDIR_SERVER_CLNTDUMP_FILEPATH, "w");
1884
1885     if (file == NULL) {
1886         ViceLog(0,
1887                 ("Couldn't create client dump file %s\n",
1888                  AFSDIR_SERVER_CLNTDUMP_FILEPATH));
1889         return;
1890     }
1891     now = FT_ApproxTime();
1892     (void)afs_snprintf(tmpStr, sizeof tmpStr, "List of active users at %s\n",
1893                        afs_ctime(&now, tbuffer, sizeof(tbuffer)));
1894     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1895     h_Enumerate(h_PrintClient, (char *)file);
1896     STREAM_REALLYCLOSE(file);
1897     ViceLog(0, ("Created client dump %s\n", AFSDIR_SERVER_CLNTDUMP_FILEPATH));
1898 }
1899
1900
1901
1902
1903 static int
1904 h_DumpHost(register struct host *host, int held, StreamHandle_t * file)
1905 {
1906     int i;
1907     char tmpStr[256];
1908
1909     H_LOCK;
1910     (void)afs_snprintf(tmpStr, sizeof tmpStr,
1911                        "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 [",
1912                        host->host, ntohs(host->port), host->index,
1913                        host->cblist, CheckLock(&host->lock), host->LastCall,
1914                        host->ActiveCall, (host->hostFlags & VENUSDOWN),
1915                        host->hostFlags & HOSTDELETED, host->Console,
1916                        host->hostFlags & CLIENTDELETED, host->hcpsfailed,
1917                        host->cpsCall);
1918     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1919     if (host->hcps.prlist_val)
1920         for (i = 0; i < host->hcps.prlist_len; i++) {
1921             (void)afs_snprintf(tmpStr, sizeof tmpStr, " %d",
1922                                host->hcps.prlist_val[i]);
1923             (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1924         }
1925     sprintf(tmpStr, "] [");
1926     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1927     if (host->interface)
1928         for (i = 0; i < host->interface->numberOfInterfaces; i++) {
1929             sprintf(tmpStr, " %x:%d", host->interface->interface[i].addr,
1930                      ntohs(host->interface->interface[i].port));
1931             (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1932         }
1933     sprintf(tmpStr, "] holds: ");
1934     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1935
1936     for (i = 0; i < h_maxSlots; i++) {
1937         sprintf(tmpStr, "%04x", host->holds[i]);
1938         (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1939     }
1940     sprintf(tmpStr, " slot/bit: %d/%d\n", h_holdSlot(), h_holdbit());
1941     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1942
1943     H_UNLOCK;
1944     return held;
1945
1946 }                               /*h_DumpHost */
1947
1948
1949 void
1950 h_DumpHosts()
1951 {
1952     time_t now;
1953     StreamHandle_t *file = STREAM_OPEN(AFSDIR_SERVER_HOSTDUMP_FILEPATH, "w");
1954     char tmpStr[256];
1955     char tbuffer[32];
1956
1957     if (file == NULL) {
1958         ViceLog(0,
1959                 ("Couldn't create host dump file %s\n",
1960                  AFSDIR_SERVER_HOSTDUMP_FILEPATH));
1961         return;
1962     }
1963     now = FT_ApproxTime();
1964     (void)afs_snprintf(tmpStr, sizeof tmpStr, "List of active hosts at %s\n",
1965                        afs_ctime(&now, tbuffer, sizeof(tbuffer)));
1966     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
1967     h_Enumerate(h_DumpHost, (char *)file);
1968     STREAM_REALLYCLOSE(file);
1969     ViceLog(0, ("Created host dump %s\n", AFSDIR_SERVER_HOSTDUMP_FILEPATH));
1970
1971 }                               /*h_DumpHosts */
1972
1973
1974 /*
1975  * This counts the number of workstations, the number of active workstations,
1976  * and the number of workstations declared "down" (i.e. not heard from
1977  * recently).  An active workstation has received a call since the cutoff
1978  * time argument passed.
1979  */
1980 void
1981 h_GetWorkStats(int *nump, int *activep, int *delp, afs_int32 cutofftime)
1982 {
1983     register struct host *host;
1984     register int num = 0, active = 0, del = 0;
1985
1986     H_LOCK;
1987     for (host = hostList; host; host = host->next) {
1988         if (!(host->hostFlags & HOSTDELETED)) {
1989             num++;
1990             if (host->ActiveCall > cutofftime)
1991                 active++;
1992             if (host->hostFlags & VENUSDOWN)
1993                 del++;
1994         }
1995     }
1996     H_UNLOCK;
1997     if (nump)
1998         *nump = num;
1999     if (activep)
2000         *activep = active;
2001     if (delp)
2002         *delp = del;
2003
2004 }                               /*h_GetWorkStats */
2005
2006
2007 /*------------------------------------------------------------------------
2008  * PRIVATE h_ClassifyAddress
2009  *
2010  * Description:
2011  *      Given a target IP address and a candidate IP address (both
2012  *      in host byte order), classify the candidate into one of three
2013  *      buckets in relation to the target by bumping the counters passed
2014  *      in as parameters.
2015  *
2016  * Arguments:
2017  *      a_targetAddr       : Target address.
2018  *      a_candAddr         : Candidate address.
2019  *      a_sameNetOrSubnetP : Ptr to counter to bump when the two
2020  *                           addresses are either in the same network
2021  *                           or the same subnet.
2022  *      a_diffSubnetP      : ...when the candidate is in a different
2023  *                           subnet.
2024  *      a_diffNetworkP     : ...when the candidate is in a different
2025  *                           network.
2026  *
2027  * Returns:
2028  *      Nothing.
2029  *
2030  * Environment:
2031  *      The target and candidate addresses are both in host byte
2032  *      order, NOT network byte order, when passed in.
2033  *
2034  * Side Effects:
2035  *      As advertised.
2036  *------------------------------------------------------------------------*/
2037
2038 static void
2039 h_ClassifyAddress(afs_uint32 a_targetAddr, afs_uint32 a_candAddr,
2040                   afs_int32 * a_sameNetOrSubnetP, afs_int32 * a_diffSubnetP,
2041                   afs_int32 * a_diffNetworkP)
2042 {                               /*h_ClassifyAddress */
2043
2044     afs_uint32 targetNet;
2045     afs_uint32 targetSubnet;
2046     afs_uint32 candNet;
2047     afs_uint32 candSubnet;
2048
2049     /*
2050      * Put bad values into the subnet info to start with.
2051      */
2052     targetSubnet = (afs_uint32) 0;
2053     candSubnet = (afs_uint32) 0;
2054
2055     /*
2056      * Pull out the network and subnetwork numbers from the target
2057      * and candidate addresses.  We can short-circuit this whole
2058      * affair if the target and candidate addresses are not of the
2059      * same class.
2060      */
2061     if (IN_CLASSA(a_targetAddr)) {
2062         if (!(IN_CLASSA(a_candAddr))) {
2063             (*a_diffNetworkP)++;
2064             return;
2065         }
2066         targetNet = a_targetAddr & IN_CLASSA_NET;
2067         candNet = a_candAddr & IN_CLASSA_NET;
2068         if (IN_SUBNETA(a_targetAddr))
2069             targetSubnet = a_targetAddr & IN_CLASSA_SUBNET;
2070         if (IN_SUBNETA(a_candAddr))
2071             candSubnet = a_candAddr & IN_CLASSA_SUBNET;
2072     } else if (IN_CLASSB(a_targetAddr)) {
2073         if (!(IN_CLASSB(a_candAddr))) {
2074             (*a_diffNetworkP)++;
2075             return;
2076         }
2077         targetNet = a_targetAddr & IN_CLASSB_NET;
2078         candNet = a_candAddr & IN_CLASSB_NET;
2079         if (IN_SUBNETB(a_targetAddr))
2080             targetSubnet = a_targetAddr & IN_CLASSB_SUBNET;
2081         if (IN_SUBNETB(a_candAddr))
2082             candSubnet = a_candAddr & IN_CLASSB_SUBNET;
2083     } /*Class B target */
2084     else if (IN_CLASSC(a_targetAddr)) {
2085         if (!(IN_CLASSC(a_candAddr))) {
2086             (*a_diffNetworkP)++;
2087             return;
2088         }
2089         targetNet = a_targetAddr & IN_CLASSC_NET;
2090         candNet = a_candAddr & IN_CLASSC_NET;
2091
2092         /*
2093          * Note that class C addresses can't have subnets,
2094          * so we leave the defaults untouched.
2095          */
2096     } /*Class C target */
2097     else {
2098         targetNet = a_targetAddr;
2099         candNet = a_candAddr;
2100     }                           /*Class D address */
2101
2102     /*
2103      * Now, simply compare the extracted net and subnet values for
2104      * the two addresses (which at this point are known to be of the
2105      * same class)
2106      */
2107     if (targetNet == candNet) {
2108         if (targetSubnet == candSubnet)
2109             (*a_sameNetOrSubnetP)++;
2110         else
2111             (*a_diffSubnetP)++;
2112     } else
2113         (*a_diffNetworkP)++;
2114
2115 }                               /*h_ClassifyAddress */
2116
2117
2118 /*------------------------------------------------------------------------
2119  * EXPORTED h_GetHostNetStats
2120  *
2121  * Description:
2122  *      Iterate through the host table, and classify each (non-deleted)
2123  *      host entry into ``proximity'' categories (same net or subnet,
2124  *      different subnet, different network).
2125  *
2126  * Arguments:
2127  *      a_numHostsP        : Set to total number of (non-deleted) hosts.
2128  *      a_sameNetOrSubnetP : Set to # hosts on same net/subnet as server.
2129  *      a_diffSubnetP      : Set to # hosts on diff subnet as server.
2130  *      a_diffNetworkP     : Set to # hosts on diff network as server.
2131  *
2132  * Returns:
2133  *      Nothing.
2134  *
2135  * Environment:
2136  *      We only count non-deleted hosts.  The storage pointed to by our
2137  *      parameters is zeroed upon entry.
2138  *
2139  * Side Effects:
2140  *      As advertised.
2141  *------------------------------------------------------------------------*/
2142
2143 void
2144 h_GetHostNetStats(afs_int32 * a_numHostsP, afs_int32 * a_sameNetOrSubnetP,
2145                   afs_int32 * a_diffSubnetP, afs_int32 * a_diffNetworkP)
2146 {                               /*h_GetHostNetStats */
2147
2148     register struct host *hostP;        /*Ptr to current host entry */
2149     register afs_uint32 currAddr_HBO;   /*Curr host addr, host byte order */
2150
2151     /*
2152      * Clear out the storage pointed to by our parameters.
2153      */
2154     *a_numHostsP = (afs_int32) 0;
2155     *a_sameNetOrSubnetP = (afs_int32) 0;
2156     *a_diffSubnetP = (afs_int32) 0;
2157     *a_diffNetworkP = (afs_int32) 0;
2158
2159     H_LOCK;
2160     for (hostP = hostList; hostP; hostP = hostP->next) {
2161         if (!(hostP->hostFlags & HOSTDELETED)) {
2162             /*
2163              * Bump the number of undeleted host entries found.
2164              * In classifying the current entry's address, make
2165              * sure to first convert to host byte order.
2166              */
2167             (*a_numHostsP)++;
2168             currAddr_HBO = (afs_uint32) ntohl(hostP->host);
2169             h_ClassifyAddress(FS_HostAddr_HBO, currAddr_HBO,
2170                               a_sameNetOrSubnetP, a_diffSubnetP,
2171                               a_diffNetworkP);
2172         }                       /*Only look at non-deleted hosts */
2173     }                           /*For each host record hashed to this index */
2174     H_UNLOCK;
2175 }                               /*h_GetHostNetStats */
2176
2177 static afs_uint32 checktime;
2178 static afs_uint32 clientdeletetime;
2179 static struct AFSFid zerofid;
2180
2181
2182 /*
2183  * XXXX: This routine could use Multi-Rx to avoid serializing the timeouts.
2184  * Since it can serialize them, and pile up, it should be a separate LWP
2185  * from other events.
2186  */
2187 int
2188 CheckHost(register struct host *host, int held)
2189 {
2190     register struct client *client;
2191     struct rx_connection *cb_conn = NULL;
2192     int code;
2193
2194     /* Host is held by h_Enumerate */
2195     H_LOCK;
2196     for (client = host->FirstClient; client; client = client->next) {
2197         if (client->refCount == 0 && client->LastCall < clientdeletetime) {
2198             client->deleted = 1;
2199             host->hostFlags |= CLIENTDELETED;
2200         }
2201     }
2202     if (host->LastCall < checktime) {
2203         h_Lock_r(host);
2204         cb_conn = host->callback_rxcon;
2205         rx_GetConnection(cb_conn);
2206         if (!(host->hostFlags & HOSTDELETED)) {
2207             if (host->LastCall < clientdeletetime) {
2208                 host->hostFlags |= HOSTDELETED;
2209                 if (!(host->hostFlags & VENUSDOWN)) {
2210                     host->hostFlags &= ~ALTADDR;        /* alternate address invalid */
2211                     if (host->interface) {
2212                         H_UNLOCK;
2213                         code =
2214                             RXAFSCB_InitCallBackState3(cb_conn,
2215                                                        &FS_HostUUID);
2216                         H_LOCK;
2217                     } else {
2218                         H_UNLOCK;
2219                         code =
2220                             RXAFSCB_InitCallBackState(cb_conn);
2221                         H_LOCK;
2222                     }
2223                     host->hostFlags |= ALTADDR; /* alternate addresses valid */
2224                     if (code) {
2225                         char hoststr[16];
2226                         (void)afs_inet_ntoa_r(host->host, hoststr);
2227                         ViceLog(0,
2228                                 ("CB: RCallBackConnectBack (host.c) failed for host %s:%d\n",
2229                                  hoststr, ntohs(host->port)));
2230                         host->hostFlags |= VENUSDOWN;
2231                     }
2232                     /* Note:  it's safe to delete hosts even if they have call
2233                      * back state, because break delayed callbacks (called when a
2234                      * message is received from the workstation) will always send a 
2235                      * break all call backs to the workstation if there is no
2236                      *callback.
2237                      */
2238                 }
2239             } else {
2240                 if (!(host->hostFlags & VENUSDOWN) && host->cblist) {
2241                     if (host->interface) {
2242                         afsUUID uuid = host->interface->uuid;
2243                         H_UNLOCK;
2244                         code = RXAFSCB_ProbeUuid(cb_conn, &uuid);
2245                         H_LOCK;
2246                         if (code) {
2247                             if (MultiProbeAlternateAddress_r(host)) {
2248                                 char hoststr[16];
2249                                 (void)afs_inet_ntoa_r(host->host, hoststr);
2250                                 ViceLog(0,
2251                                         ("ProbeUuid failed for host %s:%d\n",
2252                                          hoststr, ntohs(host->port)));
2253                                 host->hostFlags |= VENUSDOWN;
2254                             }
2255                         }
2256                     } else {
2257                         H_UNLOCK;
2258                         code = RXAFSCB_Probe(cb_conn);
2259                         H_LOCK;
2260                         if (code) {
2261                             char hoststr[16];
2262                             (void)afs_inet_ntoa_r(host->host, hoststr);
2263                             ViceLog(0,
2264                                     ("Probe failed for host %s:%d\n", hoststr,
2265                                      ntohs(host->port)));
2266                             host->hostFlags |= VENUSDOWN;
2267                         }
2268                     }
2269                 }
2270             }
2271         }
2272         H_UNLOCK;
2273         rx_PutConnection(cb_conn);
2274         cb_conn=NULL;
2275         H_LOCK;
2276         h_Unlock_r(host);
2277     }
2278     H_UNLOCK;
2279     return held;
2280
2281 }                               /*CheckHost */
2282
2283
2284 /*
2285  * Set VenusDown for any hosts that have not had a call in 15 minutes and
2286  * don't respond to a probe.  Note that VenusDown can only be cleared if
2287  * a message is received from the host (see ServerLWP in file.c).
2288  * Delete hosts that have not had any calls in 1 hour, clients that
2289  * have not had any calls in 15 minutes.
2290  *
2291  * This routine is called roughly every 5 minutes.
2292  */
2293 void
2294 h_CheckHosts()
2295 {
2296     afs_uint32 now = FT_ApproxTime();
2297
2298     memset((char *)&zerofid, 0, sizeof(zerofid));
2299     /*
2300      * Send a probe to the workstation if it hasn't been heard from in
2301      * 15 minutes
2302      */
2303     checktime = now - 15 * 60;
2304     clientdeletetime = now - 120 * 60;  /* 2 hours ago */
2305     h_Enumerate(CheckHost, NULL);
2306
2307 }                               /*h_CheckHosts */
2308
2309 /*
2310  * This is called with host locked and held. At this point, the
2311  * hostHashTable should not have any entries for the alternate
2312  * interfaces. This function has to insert these entries in the
2313  * hostHashTable.
2314  *
2315  * The addresses in the interfaceAddr list are in host byte order.
2316  */
2317 int
2318 initInterfaceAddr_r(struct host *host, struct interfaceAddr *interf)
2319 {
2320     int i, j;
2321     int number, count;
2322     afs_uint32 myAddr;
2323     afs_uint16 myPort;
2324     int found;
2325     struct Interface *interface;
2326
2327     assert(host);
2328     assert(interf);
2329
2330     ViceLog(125,
2331             ("initInterfaceAddr : host %x numAddr %d\n", host->host,
2332              interf->numberOfInterfaces));
2333
2334     number = interf->numberOfInterfaces;
2335     myAddr = host->host;        /* current interface address */
2336     myPort = host->port;        /* current port */
2337
2338     /* validation checks */
2339     if (number < 0 || number > AFS_MAX_INTERFACE_ADDR) {
2340         ViceLog(0,
2341                 ("Number of alternate addresses returned is %d\n", number));
2342         return -1;
2343     }
2344
2345     /*
2346      * Convert IP addresses to network byte order, and remove for
2347      * duplicate IP addresses from the interface list.
2348      */
2349     for (i = 0, count = 0, found = 0; i < number; i++) {
2350         interf->addr_in[i] = htonl(interf->addr_in[i]);
2351         for (j = 0; j < count; j++) {
2352             if (interf->addr_in[j] == interf->addr_in[i])
2353                 break;
2354         }
2355         if (j == count) {
2356             interf->addr_in[count] = interf->addr_in[i];
2357             if (interf->addr_in[count] == myAddr)
2358                 found = 1;
2359             count++;
2360         }
2361     }
2362
2363     /*
2364      * Allocate and initialize an interface structure for this host.
2365      */
2366     if (found) {
2367         interface = (struct Interface *)
2368             malloc(sizeof(struct Interface) +
2369                    (sizeof(struct AddrPort) * (count - 1)));
2370         if (!interface) {
2371             ViceLog(0, ("Failed malloc in initInterfaceAddr_r\n"));
2372             assert(0);
2373         }
2374         interface->numberOfInterfaces = count;
2375     } else {
2376         interface = (struct Interface *)
2377             malloc(sizeof(struct Interface) + (sizeof(struct AddrPort) * count));
2378         assert(interface);
2379         interface->numberOfInterfaces = count + 1;
2380         interface->interface[count].addr = myAddr;
2381         interface->interface[count].port = myPort;
2382     }
2383     interface->uuid = interf->uuid;
2384     for (i = 0; i < count; i++) {
2385         interface->interface[i].addr = interf->addr_in[i];
2386         /* We store the port as 7001 because the addresses reported by 
2387          * TellMeAboutYourself and WhoAreYou RPCs are only valid if they
2388          * are coming from fully connected hosts (no NAT/PATs)
2389          */
2390         interface->interface[i].port = htons(7001);
2391     }
2392
2393     assert(!host->interface);
2394     host->interface = interface;
2395
2396     for (i = 0; i < host->interface->numberOfInterfaces; i++) {
2397         ViceLog(125, ("--- alt address %x:%d\n", host->interface->interface[i].addr,
2398                        ntohs(host->interface->interface[i].port)));
2399     }
2400
2401     return 0;
2402 }
2403
2404 /* deleted a HashChain structure for this address and host */
2405 /* returns 1 on success */
2406 int
2407 hashDelete_r(afs_uint32 addr, afs_uint16 port, struct host *host)
2408 {
2409     int flag;
2410     register struct h_hashChain **hp, *th;
2411
2412     for (hp = &hostHashTable[h_HashIndex(addr)]; (th = *hp);) {
2413         assert(th->hostPtr);
2414         if (th->hostPtr == host && th->addr == addr && th->port == port) {
2415             *hp = th->next;
2416             free(th);
2417             flag = 1;
2418             break;
2419         } else {
2420             hp = &th->next;
2421         }
2422     }
2423     return flag;
2424 }
2425
2426
2427 /*
2428 ** prints out all alternate interface address for the host. The 'level'
2429 ** parameter indicates what level of debugging sets this output
2430 */
2431 void
2432 printInterfaceAddr(struct host *host, int level)
2433 {
2434     int i, number;
2435     if (host->interface) {
2436         /* check alternate addresses */
2437         number = host->interface->numberOfInterfaces;
2438         assert(number > 0);
2439         for (i = 0; i < number; i++)
2440             ViceLog(level, ("%x:%d ", host->interface->interface[i].addr,
2441                              ntohs(host->interface->interface[i].port)));
2442     }
2443     ViceLog(level, ("\n"));
2444 }