21a27f182226758b974c32b5511b09cbcd70cd0b
[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  * Portions Copyright (c) 2006 Sine Nomine Associates
10  */
11
12 #include <afsconfig.h>
13 #include <afs/param.h>
14 #include <afs/stds.h>
15
16 #include <roken.h>
17 #include <afs/opr.h>
18
19 #ifdef HAVE_SYS_FILE_H
20 #include <sys/file.h>
21 #endif
22
23 #include <rx/xdr.h>
24 #include <lwp.h>
25 #include <lock.h>
26 #include <afs/afsint.h>
27 #define FSINT_COMMON_XG
28 #include <afs/afscbint.h>
29 #include <afs/rxgen_consts.h>
30 #include <afs/nfs.h>
31 #include <afs/errors.h>
32 #include <afs/ihandle.h>
33 #include <afs/vnode.h>
34 #include <afs/volume.h>
35 #include <afs/acl.h>
36 #include <afs/ptclient.h>
37 #include <afs/ptuser.h>
38 #include <afs/prs_fs.h>
39 #include <afs/auth.h>
40 #include <afs/afsutil.h>
41 #include <afs/com_err.h>
42 #include <rx/rx.h>
43 #include <afs/cellconfig.h>
44 #include "viced_prototypes.h"
45 #include "viced.h"
46 #include "host.h"
47 #include "callback.h"
48 #ifdef AFS_DEMAND_ATTACH_FS
49 #include "../util/afsutil_prototypes.h"
50 #include "serialize_state.h"
51 #endif /* AFS_DEMAND_ATTACH_FS */
52
53 pthread_mutex_t host_glock_mutex;
54
55 extern int Console;
56 extern int CurrentConnections;
57 extern int SystemId;
58 extern int AnonymousID;
59 extern prlist AnonCPS;
60 extern int LogLevel;
61 extern struct afsconf_dir *confDir;     /* config dir object */
62 extern int lwps;                /* the max number of server threads */
63 extern afsUUID FS_HostUUID;
64
65 afsUUID nulluuid;
66 int CEs = 0;                    /* active clients */
67 int CEBlocks = 0;               /* number of blocks of CEs */
68 struct client *CEFree = 0;      /* first free client */
69 struct host *hostList = 0;      /* linked list of all hosts */
70 int hostCount = 0;              /* number of hosts in hostList */
71 int rxcon_ident_key;
72 int rxcon_client_key;
73
74 static struct rx_securityClass *sc = NULL;
75
76 static void h_SetupCallbackConn_r(struct host * host);
77 static int h_threadquota(int);
78
79 #define CESPERBLOCK 73
80 struct CEBlock {                /* block of CESPERBLOCK file entries */
81     struct client entry[CESPERBLOCK];
82 };
83
84 void h_TossStuff_r(struct host *host);
85
86 /*
87  * Make sure the subnet macros have been defined.
88  */
89 #ifndef IN_SUBNETA
90 #define IN_SUBNETA(i)           ((((afs_int32)(i))&0x80800000)==0x00800000)
91 #endif
92
93 #ifndef IN_CLASSA_SUBNET
94 #define IN_CLASSA_SUBNET        0xffff0000
95 #endif
96
97 #ifndef IN_SUBNETB
98 #define IN_SUBNETB(i)           ((((afs_int32)(i))&0xc0008000)==0x80008000)
99 #endif
100
101 #ifndef IN_CLASSB_SUBNET
102 #define IN_CLASSB_SUBNET        0xffffff00
103 #endif
104
105
106 /* get a new block of CEs and chain it on CEFree */
107 static void
108 GetCEBlock(void)
109 {
110     struct CEBlock *block;
111     int i;
112
113     block = malloc(sizeof(struct CEBlock));
114     if (!block) {
115         ViceLog(0, ("Failed malloc in GetCEBlock\n"));
116         ShutDownAndCore(PANIC);
117     }
118
119     for (i = 0; i < (CESPERBLOCK - 1); i++) {
120         Lock_Init(&block->entry[i].lock);
121         block->entry[i].next = &(block->entry[i + 1]);
122     }
123     block->entry[CESPERBLOCK - 1].next = 0;
124     Lock_Init(&block->entry[CESPERBLOCK - 1].lock);
125     CEFree = (struct client *)block;
126     CEBlocks++;
127
128 }                               /*GetCEBlock */
129
130
131 /* get the next available CE */
132 static struct client *
133 GetCE(void)
134 {
135     struct client *entry;
136
137     if (CEFree == 0)
138         GetCEBlock();
139     if (CEFree == 0) {
140         ViceLog(0, ("CEFree NULL in GetCE\n"));
141         ShutDownAndCore(PANIC);
142     }
143
144     entry = CEFree;
145     CEFree = entry->next;
146     CEs++;
147     memset(entry, 0, CLIENT_TO_ZERO(entry));
148     return (entry);
149
150 }                               /*GetCE */
151
152
153 /* return an entry to the free list */
154 static void
155 FreeCE(struct client *entry)
156 {
157     entry->VenusEpoch = 0;
158     entry->sid = 0;
159     entry->next = CEFree;
160     CEFree = entry;
161     CEs--;
162
163 }                               /*FreeCE */
164
165 /*
166  * The HTs and HTBlocks variables were formerly static, but they are
167  * now referenced elsewhere in the FileServer.
168  */
169 int HTs = 0;                    /* active file entries */
170 int HTBlocks = 0;               /* number of blocks of HTs */
171 static struct host *HTFree = 0; /* first free file entry */
172
173 /*
174  * Hash tables of host pointers. We need two tables, one
175  * to map IP addresses onto host pointers, and another
176  * to map host UUIDs onto host pointers.
177  */
178 static struct h_AddrHashChain *hostAddrHashTable[h_HASHENTRIES];
179 static struct h_UuidHashChain *hostUuidHashTable[h_HASHENTRIES];
180 #define h_HashIndex(hostip) (ntohl(hostip) & (h_HASHENTRIES-1))
181 #define h_UuidHashIndex(uuidp) (((int)(afs_uuid_hash(uuidp))) & (h_HASHENTRIES-1))
182
183 struct HTBlock {                /* block of HTSPERBLOCK file entries */
184     struct host entry[h_HTSPERBLOCK];
185 };
186
187
188 /* get a new block of HTs and chain it on HTFree */
189 static void
190 GetHTBlock(void)
191 {
192     struct HTBlock *block;
193     int i;
194     static int index = 0;
195
196     if (HTBlocks == h_MAXHOSTTABLES) {
197         ViceLog(0, ("h_MAXHOSTTABLES reached\n"));
198         return;
199     }
200
201     block = malloc(sizeof(struct HTBlock));
202     if (!block) {
203         ViceLog(0, ("Failed malloc in GetHTBlock\n"));
204         ShutDownAndCore(PANIC);
205     }
206     for (i = 0; i < (h_HTSPERBLOCK); i++)
207         CV_INIT(&block->entry[i].cond, "block entry", CV_DEFAULT, 0);
208     for (i = 0; i < (h_HTSPERBLOCK); i++)
209         Lock_Init(&block->entry[i].lock);
210     for (i = 0; i < (h_HTSPERBLOCK - 1); i++)
211         block->entry[i].next = &(block->entry[i + 1]);
212     for (i = 0; i < (h_HTSPERBLOCK); i++)
213         block->entry[i].index = index++;
214     block->entry[h_HTSPERBLOCK - 1].next = 0;
215     HTFree = (struct host *)block;
216     hosttableptrs[HTBlocks++] = block->entry;
217
218 }                               /*GetHTBlock */
219
220
221 /* get the next available HT */
222 static struct host *
223 GetHT(void)
224 {
225     struct host *entry;
226
227     if (HTFree == NULL)
228         GetHTBlock();
229     if (HTFree == NULL)
230         return NULL;
231     entry = HTFree;
232     HTFree = entry->next;
233     HTs++;
234     memset(entry, 0, HOST_TO_ZERO(entry));
235     return (entry);
236
237 }                               /*GetHT */
238
239
240 /* return an entry to the free list */
241 static void
242 FreeHT(struct host *entry)
243 {
244     entry->next = HTFree;
245     HTFree = entry;
246     HTs--;
247
248 }                               /*FreeHT */
249
250 afs_int32
251 hpr_Initialize(struct ubik_client **uclient)
252 {
253     afs_int32 code;
254     struct rx_connection *serverconns[MAXSERVERS];
255     struct rx_securityClass *sc;
256     struct afsconf_dir *tdir;
257     afs_int32 scIndex;
258     struct afsconf_cell info;
259     afs_int32 i;
260     char cellstr[64];
261
262     tdir = afsconf_Open(AFSDIR_SERVER_ETC_DIRPATH);
263     if (!tdir) {
264         ViceLog(0, ("hpr_Initialize: Could not open configuration directory: %s", AFSDIR_SERVER_ETC_DIRPATH));
265         return -1;
266     }
267
268     code = afsconf_GetLocalCell(tdir, cellstr, sizeof(cellstr));
269     if (code) {
270         ViceLog(0, ("hpr_Initialize: Could not get local cell. [%d]", code));
271         afsconf_Close(tdir);
272         return code;
273     }
274
275     code = afsconf_GetCellInfo(tdir, cellstr, "afsprot", &info);
276     if (code) {
277         ViceLog(0, ("hpr_Initialize: Could not locate cell %s in %s/%s",
278                     cellstr, tdir->name, AFSDIR_CELLSERVDB_FILE));
279         afsconf_Close(tdir);
280         return code;
281     }
282
283     code = rx_Init(0);
284     if (code) {
285         ViceLog(0, ("hpr_Initialize: Could not initialize rx."));
286         afsconf_Close(tdir);
287         return code;
288     }
289
290     /* Most callers use secLevel==1, however, the fileserver uses secLevel==2
291      * to force use of the KeyFile.  secLevel == 0 implies -noauth was
292      * specified. */
293     code = afsconf_ClientAuthSecure(tdir, &sc, &scIndex);
294     if (code) {
295         ViceLog(0, ("hpr_Initialize: clientauthsecure returns %d %s "
296                     "(so trying noauth)", code, afs_error_message(code)));
297         scIndex = RX_SECIDX_NULL;
298         sc = rxnull_NewClientSecurityObject();
299     }
300
301     if (scIndex == RX_SECIDX_NULL)
302         ViceLog(0, ("hpr_Initialize: Could not get afs tokens, "
303                     "running unauthenticated. [%d]", code));
304
305     memset(serverconns, 0, sizeof(serverconns));        /* terminate list!!! */
306     for (i = 0; i < info.numServers; i++) {
307         serverconns[i] =
308             rx_NewConnection(info.hostAddr[i].sin_addr.s_addr,
309                              info.hostAddr[i].sin_port, PRSRV,
310                              sc, scIndex);
311     }
312
313     code = ubik_ClientInit(serverconns, uclient);
314     if (code) {
315         ViceLog(0, ("hpr_Initialize: ubik client init failed. [%d]", code));
316     }
317     afsconf_Close(tdir);
318     code = rxs_Release(sc);
319     return code;
320 }
321
322 int
323 hpr_End(struct ubik_client *uclient)
324 {
325     int code = 0;
326
327     if (uclient) {
328         code = ubik_ClientDestroy(uclient);
329     }
330     return code;
331 }
332
333 static_inline int
334 getThreadClient(struct ubik_client **client)
335 {
336     int code;
337
338     *client = pthread_getspecific(viced_uclient_key);
339     if (*client != NULL)
340         return 0;
341
342     code = hpr_Initialize(client);
343     if (code)
344         return code;
345
346     osi_Assert(pthread_setspecific(viced_uclient_key, *client) == 0);
347
348     return 0;
349 }
350
351 int
352 hpr_GetHostCPS(afs_int32 host, prlist *CPS)
353 {
354     afs_int32 code;
355     afs_int32 over;
356     struct ubik_client *uclient;
357
358     code = getThreadClient(&uclient);
359     if (code)
360         return code;
361
362     over = 0;
363     code = ubik_PR_GetHostCPS(uclient, 0, host, CPS, &over);
364     if (code != PRSUCCESS)
365         return code;
366     if (over) {
367       /* do something about this, probably make a new call */
368       /* don't forget there's a hard limit in the interface */
369         fprintf(stderr,
370                 "membership list for host id %d exceeds display limit\n",
371                 host);
372     }
373     return 0;
374 }
375
376 int
377 hpr_NameToId(namelist *names, idlist *ids)
378 {
379     afs_int32 code;
380     afs_int32 i;
381     struct ubik_client *uclient;
382
383     code = getThreadClient(&uclient);
384     if (code)
385         return code;
386
387     for (i = 0; i < names->namelist_len; i++)
388         stolower(names->namelist_val[i]);
389     code = ubik_PR_NameToID(uclient, 0, names, ids);
390     return code;
391 }
392
393 int
394 hpr_IdToName(idlist *ids, namelist *names)
395 {
396     afs_int32 code;
397     struct ubik_client *uclient;
398
399     code = getThreadClient(&uclient);
400     if (code)
401         return code;
402
403     code = ubik_PR_IDToName(uclient, 0, ids, names);
404     return code;
405 }
406
407 int
408 hpr_GetCPS(afs_int32 id, prlist *CPS)
409 {
410     afs_int32 code;
411     afs_int32 over;
412     struct ubik_client *uclient;
413
414     code = getThreadClient(&uclient);
415     if (code)
416         return code;
417
418     over = 0;
419     code = ubik_PR_GetCPS(uclient, 0, id, CPS, &over);
420     if (code != PRSUCCESS)
421         return code;
422     if (over) {
423       /* do something about this, probably make a new call */
424       /* don't forget there's a hard limit in the interface */
425         fprintf(stderr, "membership list for id %d exceeds display limit\n",
426                 id);
427     }
428     return 0;
429 }
430
431 static short consolePort = 0;
432
433 int
434 h_Lock_r(struct host *host)
435 {
436     H_UNLOCK;
437     h_Lock(host);
438     H_LOCK;
439     return 0;
440 }
441
442 /**
443   * Non-blocking lock
444   * returns 1 if already locked
445   * else returns locks and returns 0
446   */
447
448 int
449 h_NBLock_r(struct host *host)
450 {
451     struct Lock *hostLock = &host->lock;
452     int locked = 0;
453
454     H_UNLOCK;
455     LOCK_LOCK(hostLock);
456     if (!(hostLock->excl_locked) && !(hostLock->readers_reading))
457         hostLock->excl_locked = WRITE_LOCK;
458     else
459         locked = 1;
460
461     LOCK_UNLOCK(hostLock);
462     H_LOCK;
463     if (locked)
464         return 1;
465     else
466         return 0;
467 }
468
469
470 /*------------------------------------------------------------------------
471  * PRIVATE h_AddrInSameNetwork
472  *
473  * Description:
474  *      Given a target IP address and a candidate IP address (both
475  *      in host byte order), return a non-zero value (1) if the
476  *      candidate address is in a different network from the target
477  *      address.
478  *
479  * Arguments:
480  *      a_targetAddr       : Target address.
481  *      a_candAddr         : Candidate address.
482  *
483  * Returns:
484  *      1 if the candidate address is in the same net as the target,
485  *      0 otherwise.
486  *
487  * Environment:
488  *      The target and candidate addresses are both in host byte
489  *      order, NOT network byte order, when passed in.  We return
490  *      our value as a character, since that's the type of field in
491  *      the host structure, where this info will be stored.
492  *
493  * Side Effects:
494  *      As advertised.
495  *------------------------------------------------------------------------*/
496
497 static char
498 h_AddrInSameNetwork(afs_uint32 a_targetAddr, afs_uint32 a_candAddr)
499 {                               /*h_AddrInSameNetwork */
500
501     afs_uint32 targetNet;
502     afs_uint32 candNet;
503
504     /*
505      * Pull out the network and subnetwork numbers from the target
506      * and candidate addresses.  We can short-circuit this whole
507      * affair if the target and candidate addresses are not of the
508      * same class.
509      */
510     if (IN_CLASSA(a_targetAddr)) {
511         if (!(IN_CLASSA(a_candAddr))) {
512             return (0);
513         }
514         targetNet = a_targetAddr & IN_CLASSA_NET;
515         candNet = a_candAddr & IN_CLASSA_NET;
516     } else if (IN_CLASSB(a_targetAddr)) {
517         if (!(IN_CLASSB(a_candAddr))) {
518             return (0);
519         }
520         targetNet = a_targetAddr & IN_CLASSB_NET;
521         candNet = a_candAddr & IN_CLASSB_NET;
522     } /*Class B target */
523     else if (IN_CLASSC(a_targetAddr)) {
524         if (!(IN_CLASSC(a_candAddr))) {
525             return (0);
526         }
527         targetNet = a_targetAddr & IN_CLASSC_NET;
528         candNet = a_candAddr & IN_CLASSC_NET;
529     } /*Class C target */
530     else {
531         targetNet = a_targetAddr;
532         candNet = a_candAddr;
533     }                           /*Class D address */
534
535     /*
536      * Now, simply compare the extracted net values for the two addresses
537      * (which at this point are known to be of the same class)
538      */
539     if (targetNet == candNet)
540         return (1);
541     else
542         return (0);
543
544 }                               /*h_AddrInSameNetwork */
545
546
547 /* Assumptions: called with held host */
548 void
549 h_gethostcps_r(struct host *host, afs_int32 now)
550 {
551     int code;
552     int slept = 0;
553
554     /* wait if somebody else is already doing the getCPS call */
555     while (host->hostFlags & HCPS_INPROGRESS) {
556         slept = 1;              /* I did sleep */
557         host->hostFlags |= HCPS_WAITING;        /* I am sleeping now */
558         CV_WAIT(&host->cond, &host_glock_mutex);
559     }
560
561
562     host->hostFlags |= HCPS_INPROGRESS; /* mark as CPSCall in progress */
563     if (host->hcps.prlist_val)
564         free(host->hcps.prlist_val);    /* this is for hostaclRefresh */
565     host->hcps.prlist_val = NULL;
566     host->hcps.prlist_len = 0;
567     host->cpsCall = slept ? (FT_ApproxTime()) : (now);
568
569     H_UNLOCK;
570     code = hpr_GetHostCPS(ntohl(host->host), &host->hcps);
571     H_LOCK;
572     if (code) {
573         char hoststr[16];
574         /*
575          * Although ubik_Call (called by pr_GetHostCPS) traverses thru all protection servers
576          * and reevaluates things if no sync server or quorum is found we could still end up
577          * with one of these errors. In such case we would like to reevaluate the rpc call to
578          * find if there's cps for this guy. We treat other errors (except network failures
579          * ones - i.e. code < 0) as an indication that there is no CPS for this host. Ideally
580          * we could like to deal this problem the other way around (i.e. if code == NOCPS
581          * ignore else retry next time) but the problem is that there're other errors (i.e.
582          * EPERM) for which we don't want to retry and we don't know the whole code list!
583          */
584         if (code < 0 || code == UNOQUORUM || code == UNOTSYNC) {
585             /*
586              * We would have preferred to use a while loop and try again since ops in protected
587              * acls for this host will fail now but they'll be reevaluated on any subsequent
588              * call. The attempt to wait for a quorum/sync site or network error won't work
589              * since this problems really should only occurs during a complete fileserver
590              * restart. Since the fileserver will start before the ptservers (and thus before
591              * quorums are complete) clients will be utilizing all the fileserver's lwps!!
592              */
593             host->hcpsfailed = 1;
594             ViceLog(0,
595                     ("Warning:  GetHostCPS failed (%d) for %p (%s:%d); will retry\n",
596                      code, host, afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port)));
597         } else {
598             host->hcpsfailed = 0;
599             ViceLog(1,
600                     ("gethost:  GetHostCPS failed (%d) for %p (%s:%d); ignored\n",
601                      code, host, afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port)));
602         }
603         if (host->hcps.prlist_val)
604             free(host->hcps.prlist_val);
605         host->hcps.prlist_val = NULL;
606         host->hcps.prlist_len = 0;      /* Make sure it's zero */
607     } else
608         host->hcpsfailed = 0;
609
610     host->hostFlags &= ~HCPS_INPROGRESS;
611     /* signal all who are waiting */
612     if (host->hostFlags & HCPS_WAITING) {       /* somebody is waiting */
613         host->hostFlags &= ~HCPS_WAITING;
614         CV_BROADCAST(&host->cond);
615     }
616 }
617
618 /* args in net byte order */
619 void
620 h_flushhostcps(afs_uint32 hostaddr, afs_uint16 hport)
621 {
622     struct host *host;
623
624     H_LOCK;
625     h_Lookup_r(hostaddr, hport, &host);
626     if (host) {
627         host->hcpsfailed = 1;
628         h_Release_r(host);
629     }
630     H_UNLOCK;
631     return;
632 }
633
634
635 /*
636  * Allocate a host.  It will be identified by the peer (ip,port) info in the
637  * rx connection provided.  The host is returned held and locked
638  */
639 #define DEF_ROPCONS 2115
640
641 static struct host *
642 h_Alloc_r(struct rx_connection *r_con)
643 {
644     struct servent *serverentry;
645     struct host *host;
646     afs_uint32 newHostAddr_HBO; /*New host IP addr, in host byte order */
647
648     host = GetHT();
649     if (!host)
650         return NULL;
651
652     h_Hold_r(host);
653     /* acquire the host lock withot dropping H_LOCK. we can do this here
654      * because we know we will not block; we just created this host and
655      * nobody else knows about it. */
656     ObtainWriteLock(&host->lock);
657
658     host->host = rxr_HostOf(r_con);
659     host->port = rxr_PortOf(r_con);
660
661     h_AddHostToAddrHashTable_r(host->host, host->port, host);
662
663     if (consolePort == 0) {     /* find the portal number for console */
664 #if     defined(AFS_OSF_ENV)
665         serverentry = getservbyname("ropcons", "");
666 #else
667         serverentry = getservbyname("ropcons", 0);
668 #endif
669         if (serverentry)
670             consolePort = serverentry->s_port;
671         else
672             consolePort = htons(DEF_ROPCONS);   /* Use a default */
673     }
674     if (host->port == consolePort)
675         host->Console = 1;
676     /* Make a callback channel even for the console, on the off chance that it
677      * makes a request that causes a break call back.  It shouldn't. */
678     h_SetupCallbackConn_r(host);
679     host->LastCall = host->cpsCall = host->ActiveCall = FT_ApproxTime();
680     host->hostFlags = 0;
681     host->hcps.prlist_val = NULL;
682     host->hcps.prlist_len = 0;
683     host->interface = NULL;
684 #ifdef undef
685     host->hcpsfailed = 0;       /* save cycles */
686     h_gethostcps(host);         /* do this under host hold/lock */
687 #endif
688     host->FirstClient = NULL;
689     h_InsertList_r(host);       /* update global host List */
690     /*
691      * Compare the new host's IP address (in host byte order) with ours
692      * (the File Server's), remembering if they are in the same network.
693      */
694     newHostAddr_HBO = (afs_uint32) ntohl(host->host);
695     host->InSameNetwork =
696         h_AddrInSameNetwork(FS_HostAddr_HBO, newHostAddr_HBO);
697     return host;
698
699 }                               /*h_Alloc_r */
700
701
702
703 /* Make a callback channel even for the console, on the off chance that it
704  * makes a request that causes a break call back.  It shouldn't. */
705 static void
706 h_SetupCallbackConn_r(struct host * host)
707 {
708     if (!sc)
709         sc = rxnull_NewClientSecurityObject();
710     host->callback_rxcon =
711         rx_NewConnection(host->host, host->port, 1, sc, 0);
712     rx_SetConnDeadTime(host->callback_rxcon, 50);
713     rx_SetConnHardDeadTime(host->callback_rxcon, AFS_HARDDEADTIME);
714     rx_SetConnSecondsUntilNatPing(host->callback_rxcon, 20);
715 }
716
717 /* h_Lookup_r
718  * Lookup a host given an IP address and UDP port number.
719  * hostaddr and hport are in network order
720  * hostaddr and hport are in network order
721  * On return, refCount is incremented.
722  */
723 int
724 h_Lookup_r(afs_uint32 haddr, afs_uint16 hport, struct host **hostp)
725 {
726     afs_int32 now;
727     struct host *host = NULL;
728     struct h_AddrHashChain *chain;
729     int index = h_HashIndex(haddr);
730     extern int hostaclRefresh;
731
732   restart:
733     for (chain = hostAddrHashTable[index]; chain; chain = chain->next) {
734         host = chain->hostPtr;
735         osi_Assert(host);
736         if (!(host->hostFlags & HOSTDELETED) && chain->addr == haddr
737             && chain->port == hport) {
738             if ((host->hostFlags & HWHO_INPROGRESS) &&
739                 h_threadquota(host->lock.num_waiting)) {
740                 *hostp = 0;
741                 return VBUSY;
742             }
743             h_Hold_r(host);
744             h_Lock_r(host);
745             if (host->hostFlags & HOSTDELETED) {
746                 h_Unlock_r(host);
747                 h_Release_r(host);
748                 host = NULL;
749                 goto restart;
750             }
751             h_Unlock_r(host);
752             now = FT_ApproxTime();      /* always evaluate "now" */
753             if (host->hcpsfailed || (host->cpsCall + hostaclRefresh < now)) {
754                 /*
755                  * Every hostaclRefresh period (def 2 hrs) get the new
756                  * membership list for the host.  Note this could be the
757                  * first time that the host is added to a group.  Also
758                  * here we also retry on previous legitimate hcps failures.
759                  *
760                  * If we get here refCount is elevated.
761                  */
762                 h_gethostcps_r(host, now);
763             }
764             break;
765         }
766         host = NULL;
767     }
768     *hostp = host;
769     return 0;
770 }                               /*h_Lookup */
771
772 /* Lookup a host given its UUID. */
773 struct host *
774 h_LookupUuid_r(afsUUID * uuidp)
775 {
776     struct host *host = 0;
777     struct h_UuidHashChain *chain;
778     int index = h_UuidHashIndex(uuidp);
779
780     for (chain = hostUuidHashTable[index]; chain; chain = chain->next) {
781         host = chain->hostPtr;
782         osi_Assert(host);
783         if (!(host->hostFlags & HOSTDELETED) && host->interface
784             && afs_uuid_equal(&host->interface->uuid, uuidp)) {
785             return host;
786         }
787     }
788     return NULL;
789 }                               /*h_Lookup */
790
791
792 /* h_TossStuff_r:  Toss anything in the host structure (the host or
793  * clients marked for deletion.  Called from h_Release_r ONLY.
794  * To be called, there must be no holds, and either host->deleted
795  * or host->clientDeleted must be set.
796  */
797 void
798 h_TossStuff_r(struct host *host)
799 {
800     struct client **cp, *client;
801     int code;
802     int wasdeleted = 0;
803
804     if ((host->hostFlags & HOSTDELETED)) {
805         wasdeleted = 1;
806     }
807
808     /* make sure host doesn't go away over h_NBLock_r */
809     h_Hold_r(host);
810
811     code = h_NBLock_r(host);
812
813     /* don't use h_Release_r, since that may call h_TossStuff_r again */
814     h_Decrement_r(host);
815
816     /* if somebody still has this host locked */
817     if (code != 0) {
818         char hoststr[16];
819         if (wasdeleted) {
820             /* someone locked the host while HOSTDELETED was set; that is bad */
821             ViceLog(0, ("Warning:  h_TossStuff_r failed; Host %" AFS_PTR_FMT
822                         " (%s:%d flags 0x%x) was locked.\n",
823                         host, afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port),
824                         (unsigned)host->hostFlags));
825         }
826         return;
827     } else {
828         h_Unlock_r(host);
829     }
830
831     /* if somebody still has this host held */
832     /* we must check this _after_ h_NBLock_r, since h_NBLock_r can drop and
833      * reacquire H_LOCK */
834     if (host->refCount > 0) {
835         char hoststr[16];
836         if (wasdeleted) {
837             /* someone grabbed a ref while HOSTDELETED was set; that is bad */
838             ViceLog(0, ("Warning:  h_TossStuff_r failed; Host %" AFS_PTR_FMT
839                         " (%s:%d flags 0x%x) was held.\n",
840                         host, afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port),
841                         (unsigned)host->hostFlags));
842         }
843         return;
844     }
845
846     /* ASSUMPTION: rxi_FreeConnection() does not yield */
847     for (cp = &host->FirstClient; (client = *cp);) {
848         if ((host->hostFlags & HOSTDELETED) || client->deleted) {
849             int code;
850             ObtainWriteLockNoBlock(&client->lock, code);
851             if (code < 0) {
852                 char hoststr[16];
853                 ViceLog(0,
854                         ("Warning: h_TossStuff_r failed: Host %p (%s:%d) "
855                          "client %p was locked.\n",
856                          host, afs_inet_ntoa_r(host->host, hoststr),
857                          ntohs(host->port), client));
858                 return;
859             }
860
861             if (client->refCount) {
862                 char hoststr[16];
863                 ViceLog(0,
864                         ("Warning: h_TossStuff_r failed: Host %p (%s:%d) "
865                          "client %p refcount %d.\n",
866                          host, afs_inet_ntoa_r(host->host, hoststr),
867                          ntohs(host->port), client, client->refCount));
868                 /* This is the same thing we do if the host is locked */
869                 ReleaseWriteLock(&client->lock);
870                 return;
871             }
872             client->CPS.prlist_len = 0;
873             if ((client->ViceId != ANONYMOUSID) && client->CPS.prlist_val)
874                 free(client->CPS.prlist_val);
875             client->CPS.prlist_val = NULL;
876             CurrentConnections--;
877             *cp = client->next;
878             ReleaseWriteLock(&client->lock);
879             FreeCE(client);
880         } else
881             cp = &client->next;
882     }
883
884     /* We've just cleaned out all the deleted clients; clear the flag */
885     host->hostFlags &= ~CLIENTDELETED;
886
887     if (host->hostFlags & HOSTDELETED) {
888         struct rx_connection *rxconn;
889         struct AddrPort hostAddrPort;
890         int i;
891
892         if (host->Console & 1)
893             Console--;
894         if ((rxconn = host->callback_rxcon)) {
895             host->callback_rxcon = (struct rx_connection *)0;
896             rx_DestroyConnection(rxconn);
897         }
898         if (host->hcps.prlist_val)
899             free(host->hcps.prlist_val);
900         host->hcps.prlist_val = NULL;
901         host->hcps.prlist_len = 0;
902         DeleteAllCallBacks_r(host, 1);
903         host->hostFlags &= ~RESETDONE;  /* just to be safe */
904
905         /* if alternate addresses do not exist */
906         if (!(host->interface)) {
907             h_DeleteHostFromAddrHashTable_r(host->host, host->port, host);
908         } else {
909             h_DeleteHostFromUuidHashTable_r(host);
910             h_DeleteHostFromAddrHashTable_r(host->host, host->port, host);
911             /* delete the hash entry for each valid alternate addresses */
912             for (i = 0; i < host->interface->numberOfInterfaces; i++) {
913                 hostAddrPort = host->interface->interface[i];
914                 /*
915                  * if the interface addr/port is the primary, we already
916                  * removed it.  If the addr/port is not valid, its not
917                  * in the hash table.
918                  */
919                 if (hostAddrPort.valid &&
920                     (host->host != hostAddrPort.addr ||
921                      host->port != hostAddrPort.port))
922                     h_DeleteHostFromAddrHashTable_r(hostAddrPort.addr, hostAddrPort.port, host);
923             }
924             free(host->interface);
925             host->interface = NULL;
926         }                       /* if alternate address exists */
927
928         h_DeleteList_r(host);   /* remove host from global host List */
929         FreeHT(host);
930     }
931 }                               /*h_TossStuff_r */
932
933
934
935 /* h_Enumerate: Calls (*proc)(host, param) for at least each host in the
936  * system at the start of the enumeration (perhaps more).  Hosts may be deleted
937  * (have delete flag set); ditto for clients.  refCount is always incremented
938  * before (*proc) is called.
939  *
940  * The return value of the proc is a set of flags. The proc should set
941  * H_ENUMERATE_BAIL(foo) if the enumeration of hosts should be stopped early.
942  */
943 void
944 h_Enumerate(int (*proc) (struct host*, void *), void *param)
945 {
946     struct host *host, **list;
947     int i, count;
948     int totalCount;
949
950     H_LOCK;
951     if (hostCount == 0) {
952         H_UNLOCK;
953         return;
954     }
955     list = malloc(hostCount * sizeof(struct host *));
956     if (!list) {
957         ViceLogThenPanic(0, ("Failed malloc in h_Enumerate (list)\n"));
958     }
959     for (totalCount = count = 0, host = hostList;
960          host && totalCount < hostCount;
961          host = host->next, totalCount++) {
962
963         if (!(host->hostFlags & HOSTDELETED)) {
964             list[count] = host;
965             h_Hold_r(host);
966             count++;
967         }
968     }
969     if (totalCount != hostCount) {
970         ViceLog(0, ("h_Enumerate found %d of %d hosts\n", totalCount, hostCount));
971     } else if (host != NULL) {
972         ViceLog(0, ("h_Enumerate found more than %d hosts\n", hostCount));
973         ShutDownAndCore(PANIC);
974     }
975     H_UNLOCK;
976     for (i = 0; i < count; i++) {
977         int flags;
978         flags = (*proc) (list[i], param);
979         H_LOCK;
980         h_Release_r(list[i]);
981         H_UNLOCK;
982         /* bail out of the enumeration early */
983         if (H_ENUMERATE_ISSET_BAIL(flags)) {
984             break;
985         } else if (flags) {
986             ViceLog(0, ("h_Enumerate got back invalid return value %d\n", flags));
987             ShutDownAndCore(PANIC);
988         }
989     }
990     if (i < count-1) {
991         /* we bailed out of enumerating hosts early; we still have holds on
992          * some of the hosts in 'list', so release them */
993         i++;
994         H_LOCK;
995         for ( ; i < count; i++) {
996             h_Release_r(list[i]);
997         }
998         H_UNLOCK;
999     }
1000     free(list);
1001 }       /* h_Enumerate */
1002
1003
1004 /* h_Enumerate_r (revised):
1005  * Calls (*proc)(host, param) for each host in hostList, starting
1006  * at enumstart. Called only under H_LOCK.  Hosts may be deleted (have
1007  * delete flag set); ditto for clients.  refCount is always incremented
1008  * before (*proc) is called.
1009  *
1010  * @note Assumes that hostList is only prepended to, that a host is never
1011  *       inserted into the middle. Otherwise this would not be guaranteed to
1012  *       terminate.
1013  *
1014  * The return value of the proc is a set of flags. The proc should set
1015  * H_ENUMERATE_BAIL(foo) if the enumeration of hosts should be stopped early.
1016  */
1017 void
1018 h_Enumerate_r(int (*proc) (struct host *, void *),
1019               struct host *enumstart, void *param)
1020 {
1021     struct host *host, *next;
1022     int count;
1023     int origHostCount;
1024
1025     if (hostCount == 0) {
1026         return;
1027     }
1028
1029     host = enumstart;
1030     enumstart = NULL;
1031
1032     /* find the first non-deleted host, so we know where to actually start
1033      * enumerating */
1034     for (count = 0; host && count < hostCount; count++) {
1035         if (!(host->hostFlags & HOSTDELETED)) {
1036             enumstart = host;
1037             break;
1038         }
1039         host = host->next;
1040     }
1041     if (!enumstart) {
1042         /* we didn't find a non-deleted host... */
1043
1044         if (host && count >= hostCount) {
1045             /* ...because we found a loop */
1046             ViceLog(0, ("h_Enumerate_r found more than %d hosts\n", hostCount));
1047             ShutDownAndCore(PANIC);
1048         }
1049
1050         /* ...because the hostList is full of deleted hosts */
1051         return;
1052     }
1053
1054     h_Hold_r(enumstart);
1055
1056     /* remember hostCount, lest it change over the potential H_LOCK drop in
1057      * h_Release_r */
1058     origHostCount = hostCount;
1059
1060     for (count = 0, host = enumstart; host && count < origHostCount; host = next, count++) {
1061         next = host->next;
1062
1063         /* find the next non-deleted host */
1064         while (next && (next->hostFlags & HOSTDELETED)) {
1065             next = next->next;
1066             /* inc count for the skipped-over host */
1067             if (++count > origHostCount) {
1068                 ViceLog(0, ("h_Enumerate_r found more than %d hosts\n", origHostCount));
1069                 ShutDownAndCore(PANIC);
1070             }
1071         }
1072         if (next)
1073             h_Hold_r(next);
1074
1075         if (!(host->hostFlags & HOSTDELETED)) {
1076             int flags;
1077             flags = (*proc) (host, param);
1078             if (H_ENUMERATE_ISSET_BAIL(flags)) {
1079                 h_Release_r(host); /* this might free up the host */
1080                 if (next) {
1081                     h_Release_r(next);
1082                 }
1083                 break;
1084             } else if (flags) {
1085                 ViceLog(0, ("h_Enumerate_r got back invalid return value %d\n", flags));
1086                 ShutDownAndCore(PANIC);
1087             }
1088         }
1089         h_Release_r(host); /* this might free up the host */
1090     }
1091     if (host != NULL && count >= origHostCount) {
1092         ViceLog(0, ("h_Enumerate_r found more than %d hosts\n", origHostCount));
1093         ShutDownAndCore(PANIC);
1094     }
1095 }       /*h_Enumerate_r */
1096
1097
1098 /* inserts a new HashChain structure corresponding to this UUID */
1099 void
1100 h_AddHostToUuidHashTable_r(struct afsUUID *uuid, struct host *host)
1101 {
1102     int index;
1103     struct h_UuidHashChain *chain;
1104     char uuid1[128], uuid2[128];
1105     char hoststr[16];
1106
1107     /* hash into proper bucket */
1108     index = h_UuidHashIndex(uuid);
1109
1110     /* don't add the same entry multiple times */
1111     for (chain = hostUuidHashTable[index]; chain; chain = chain->next) {
1112         if (!chain->hostPtr)
1113             continue;
1114
1115         if (chain->hostPtr->interface &&
1116             afs_uuid_equal(&chain->hostPtr->interface->uuid, uuid)) {
1117             if (LogLevel >= 125) {
1118                 afsUUID_to_string(&chain->hostPtr->interface->uuid, uuid1,
1119                                   127);
1120                 afsUUID_to_string(uuid, uuid2, 127);
1121                 ViceLog(125, ("h_AddHostToUuidHashTable_r: host %" AFS_PTR_FMT " (uuid %s) exists as %s:%d (uuid %s)\n",
1122                               host, uuid1,
1123                               afs_inet_ntoa_r(chain->hostPtr->host, hoststr),
1124                               ntohs(chain->hostPtr->port), uuid2));
1125             }
1126             return;
1127         }
1128     }
1129
1130     /* insert into beginning of list for this bucket */
1131     chain = malloc(sizeof(struct h_UuidHashChain));
1132     if (!chain) {
1133         ViceLogThenPanic(0, ("Failed malloc in h_AddHostToUuidHashTable_r\n"));
1134     }
1135     chain->hostPtr = host;
1136     chain->next = hostUuidHashTable[index];
1137     hostUuidHashTable[index] = chain;
1138          if (LogLevel < 125)
1139                return;
1140      afsUUID_to_string(uuid, uuid2, 127);
1141      ViceLog(125,
1142              ("h_AddHostToUuidHashTable_r: host %p (%s:%d) added as uuid %s\n",
1143               host, afs_inet_ntoa_r(chain->hostPtr->host, hoststr),
1144               ntohs(chain->hostPtr->port), uuid2));
1145 }
1146
1147 /* deletes a HashChain structure corresponding to this host */
1148 int
1149 h_DeleteHostFromUuidHashTable_r(struct host *host)
1150 {
1151      int index;
1152      struct h_UuidHashChain **uhp, *uth;
1153      char uuid1[128];
1154      char hoststr[16];
1155
1156      if (!host->interface)
1157        return 0;
1158
1159      /* hash into proper bucket */
1160      index = h_UuidHashIndex(&host->interface->uuid);
1161
1162      if (LogLevel >= 125)
1163          afsUUID_to_string(&host->interface->uuid, uuid1, 127);
1164      for (uhp = &hostUuidHashTable[index]; (uth = *uhp); uhp = &uth->next) {
1165          osi_Assert(uth->hostPtr);
1166          if (uth->hostPtr == host) {
1167              ViceLog(125,
1168                      ("h_DeleteHostFromUuidHashTable_r: host %" AFS_PTR_FMT " (uuid %s %s:%d)\n",
1169                       host, uuid1, afs_inet_ntoa_r(host->host, hoststr),
1170                       ntohs(host->port)));
1171              *uhp = uth->next;
1172              free(uth);
1173              return 1;
1174          }
1175      }
1176      ViceLog(125,
1177              ("h_DeleteHostFromUuidHashTable_r: host %" AFS_PTR_FMT " (uuid %s %s:%d) not found\n",
1178               host, uuid1, afs_inet_ntoa_r(host->host, hoststr),
1179               ntohs(host->port)));
1180      return 0;
1181 }
1182
1183 /*
1184  * This is called with host locked and held.
1185  *
1186  * All addresses are in network byte order.
1187  */
1188 static int
1189 invalidateInterfaceAddr_r(struct host *host, afs_uint32 addr, afs_uint16 port)
1190 {
1191     int i;
1192     int number;
1193     struct Interface *interface;
1194     char hoststr[16], hoststr2[16];
1195
1196     osi_Assert(host);
1197     osi_Assert(host->interface);
1198
1199     ViceLog(125, ("invalidateInterfaceAddr : host %" AFS_PTR_FMT " (%s:%d) addr %s:%d\n",
1200                   host, afs_inet_ntoa_r(host->host, hoststr),
1201                   ntohs(host->port), afs_inet_ntoa_r(addr, hoststr2),
1202                   ntohs(port)));
1203
1204     /*
1205      * Make sure this address is on the list of known addresses
1206      * for this host.
1207      */
1208     interface = host->interface;
1209     number = host->interface->numberOfInterfaces;
1210     for (i = 0; i < number; i++) {
1211         if (interface->interface[i].addr == addr &&
1212             interface->interface[i].port == port) {
1213             if (interface->interface[i].valid) {
1214                 h_DeleteHostFromAddrHashTable_r(addr, port, host);
1215                 interface->interface[i].valid = 0;
1216             }
1217             return 0;
1218         }
1219     }
1220
1221     /* not found */
1222     return 0;
1223 }
1224
1225 /*
1226  * This is called with host locked and held.  This function differs
1227  * from removeInterfaceAddr_r in that it is called when the address
1228  * is being removed from the host regardless of whether or not there
1229  * is an interface list for the host.  This function will delete the
1230  * host if there are no addresses left on it.
1231  *
1232  * All addresses are in network byte order.
1233  */
1234 static int
1235 removeAddress_r(struct host *host, afs_uint32 addr, afs_uint16 port)
1236 {
1237     int i;
1238     char hoststr[16], hoststr2[16];
1239     struct rx_connection *rxconn;
1240
1241     if (!host->interface || host->interface->numberOfInterfaces == 1) {
1242         if (host->host == addr && host->port == port) {
1243             ViceLog(25,
1244                     ("Removing only address for host %" AFS_PTR_FMT " (%s:%d), deleting host.\n",
1245                      host, afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port)));
1246             host->hostFlags |= HOSTDELETED;
1247             /*
1248              * Do not remove the primary addr/port from the hash table.
1249              * It will be ignored due to the HOSTDELETED flag and will
1250              * be removed when h_TossStuff_r() cleans up the HOSTDELETED
1251              * host.  Removing it here will only result in a search for
1252              * the host/addr/port in the hash chain which will fail.
1253              */
1254         } else {
1255             ViceLog(0,
1256                     ("Removing address that does not belong to host %" AFS_PTR_FMT " (%s:%d).\n",
1257                      host, afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port)));
1258         }
1259     } else {
1260         if (host->host == addr && host->port == port)  {
1261             removeInterfaceAddr_r(host, addr, port);
1262
1263             for (i=0; i < host->interface->numberOfInterfaces; i++) {
1264                 if (host->interface->interface[i].valid) {
1265                     ViceLog(25,
1266                              ("Removed address for host %" AFS_PTR_FMT " (%s:%d), new primary interface %s:%d.\n",
1267                                host, afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port),
1268                                afs_inet_ntoa_r(host->interface->interface[i].addr, hoststr2),
1269                                ntohs(host->interface->interface[i].port)));
1270                     host->host = host->interface->interface[i].addr;
1271                     host->port = host->interface->interface[i].port;
1272                     h_AddHostToAddrHashTable_r(host->host, host->port, host);
1273                     break;
1274                 }
1275             }
1276
1277             if (i == host->interface->numberOfInterfaces) {
1278                 ViceLog(25,
1279                          ("Removed only address for host %" AFS_PTR_FMT " (%s:%d), no valid alternate interfaces, deleting host.\n",
1280                            host, afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port)));
1281                 host->hostFlags |= HOSTDELETED;
1282                 /* addr/port was removed from the hash table */
1283                 host->host = 0;
1284                 host->port = 0;
1285             } else {
1286                 rxconn = host->callback_rxcon;
1287                 host->callback_rxcon = NULL;
1288
1289                 if (rxconn) {
1290                     rx_DestroyConnection(rxconn);
1291                     rxconn = NULL;
1292                 }
1293
1294                 h_SetupCallbackConn_r(host);
1295             }
1296         } else {
1297             /* not the primary addr/port, just invalidate it */
1298             invalidateInterfaceAddr_r(host, addr, port);
1299         }
1300     }
1301
1302     return 0;
1303 }
1304
1305 static void
1306 createHostAddrHashChain_r(int index, afs_uint32 addr, afs_uint16 port, struct host *host)
1307 {
1308     struct h_AddrHashChain *chain;
1309     char hoststr[16];
1310
1311     /* insert into beginning of list for this bucket */
1312     chain = malloc(sizeof(struct h_AddrHashChain));
1313     if (!chain) {
1314         ViceLogThenPanic(0, ("Failed malloc in h_AddHostToAddrHashTable_r\n"));
1315     }
1316     chain->hostPtr = host;
1317     chain->next = hostAddrHashTable[index];
1318     chain->addr = addr;
1319     chain->port = port;
1320     hostAddrHashTable[index] = chain;
1321     ViceLog(125, ("h_AddHostToAddrHashTable_r: host %" AFS_PTR_FMT " added as %s:%d\n",
1322                   host, afs_inet_ntoa_r(addr, hoststr), ntohs(port)));
1323 }
1324
1325 /**
1326  * Resolve host address conflicts when hashing by address.
1327  *
1328  * @param[in]   addr    an ip address of the interface
1329  * @param[in]   port    the port of the interface
1330  * @param[in]   newHost the host being added with this address
1331  * @param[in]   oldHost the host previously added with this address
1332  */
1333 static void
1334 reconcileHosts_r(afs_uint32 addr, afs_uint16 port, struct host *newHost,
1335                  struct host *oldHost)
1336 {
1337     struct rx_connection *cb = NULL;
1338     int code = 0;
1339     struct interfaceAddr interf;
1340     Capabilities caps;
1341     afsUUID *newHostUuid = &nulluuid;
1342     afsUUID *oldHostUuid = &nulluuid;
1343     char hoststr[16];
1344
1345     ViceLog(125,
1346             ("reconcileHosts_r: addr %s:%d newHost %" AFS_PTR_FMT " oldHost %"
1347              AFS_PTR_FMT, afs_inet_ntoa_r(addr, hoststr), ntohs(port),
1348              newHost, oldHost));
1349
1350     osi_Assert(oldHost != newHost);
1351     caps.Capabilities_val = NULL;
1352
1353     if (!sc) {
1354         sc = rxnull_NewClientSecurityObject();
1355     }
1356
1357     cb = rx_NewConnection(addr, port, 1, sc, 0);
1358     rx_SetConnDeadTime(cb, 50);
1359     rx_SetConnHardDeadTime(cb, AFS_HARDDEADTIME);
1360
1361     h_Hold_r(newHost);
1362     h_Hold_r(oldHost);
1363     H_UNLOCK;
1364     code = RXAFSCB_TellMeAboutYourself(cb, &interf, &caps);
1365     if (code == RXGEN_OPCODE) {
1366         code = RXAFSCB_WhoAreYou(cb, &interf);
1367     }
1368     H_LOCK;
1369
1370     if (code == RXGEN_OPCODE ||
1371         (code == 0 && afs_uuid_equal(&interf.uuid, &nulluuid))) {
1372         ViceLog(0,
1373                 ("reconcileHosts_r: WhoAreYou not supported for connection (%s:%d), error %d\n",
1374                  afs_inet_ntoa_r(addr, hoststr), ntohs(port), code));
1375         goto fail;
1376     }
1377     if (code != 0) {
1378         ViceLog(0,
1379                 ("reconcileHosts_r: WhoAreYou failed for connection (%s:%d), error %d\n",
1380                  afs_inet_ntoa_r(addr, hoststr), ntohs(port), code));
1381         goto fail;
1382     }
1383
1384     /* Since lock was dropped, the hosts may have been deleted during the rpcs. */
1385     if ((newHost->hostFlags & HOSTDELETED)
1386         && (oldHost->hostFlags & HOSTDELETED)) {
1387         ViceLog(5,
1388                 ("reconcileHosts_r: new and old hosts were deleted during probe.\n"));
1389         goto done;
1390     }
1391
1392     /* A check can be done if at least one of the hosts has a uuid. It
1393      * is an error if the hosts have the same (not null) uuid. */
1394     if ((!(newHost->hostFlags & HOSTDELETED)) && newHost->interface) {
1395         newHostUuid = &(newHost->interface->uuid);
1396     }
1397     if ((!(oldHost->hostFlags & HOSTDELETED)) && oldHost->interface) {
1398         oldHostUuid = &(oldHost->interface->uuid);
1399     }
1400     if (afs_uuid_equal(newHostUuid, &nulluuid) &&
1401         afs_uuid_equal(oldHostUuid, &nulluuid)) {
1402         ViceLog(0,
1403                 ("reconcileHosts_r: Cannot reconcile hosts for connection (%s:%d), no uuids\n",
1404                  afs_inet_ntoa_r(addr, hoststr), ntohs(port)));
1405         goto done;
1406     }
1407     if (afs_uuid_equal(newHostUuid, oldHostUuid)) {
1408         ViceLog(0,
1409                 ("reconcileHosts_r: Cannot reconcile hosts for connection (%s:%d), same uuids\n",
1410                  afs_inet_ntoa_r(addr, hoststr), ntohs(port)));
1411         goto done;
1412     }
1413
1414     /* Determine which host should be hashed */
1415     if ((!(newHost->hostFlags & HOSTDELETED))
1416         && afs_uuid_equal(newHostUuid, &(interf.uuid))) {
1417         /* Install the new host into the hash before removing the stale
1418          * addresses. Walk the hash chain again since the hash table may have
1419          * been changed when the host lock was dropped to get the uuid. */
1420         struct h_AddrHashChain *chain;
1421         int index = h_HashIndex(addr);
1422         for (chain = hostAddrHashTable[index]; chain; chain = chain->next) {
1423             if (chain->addr == addr && chain->port == port) {
1424                 chain->hostPtr = newHost;
1425                 removeAddress_r(oldHost, addr, port);
1426                 goto done;
1427             }
1428         }
1429         createHostAddrHashChain_r(index, addr, port, newHost);
1430         removeAddress_r(oldHost, addr, port);
1431         goto done;
1432     }
1433     if ((!(oldHost->hostFlags & HOSTDELETED))
1434         && afs_uuid_equal(oldHostUuid, &(interf.uuid))) {
1435         removeAddress_r(newHost, addr, port);
1436         goto done;
1437     }
1438
1439   fail:
1440     if (!(newHost->hostFlags & HOSTDELETED)) {
1441         removeAddress_r(newHost, addr, port);
1442     }
1443     if (!(oldHost->hostFlags & HOSTDELETED)) {
1444         removeAddress_r(oldHost, addr, port);
1445     }
1446
1447   done:
1448     h_Release_r(newHost);
1449     h_Release_r(oldHost);
1450     rx_DestroyConnection(cb);
1451     return;
1452 }
1453
1454 /* inserts a new HashChain structure corresponding to this address */
1455 void
1456 h_AddHostToAddrHashTable_r(afs_uint32 addr, afs_uint16 port, struct host *host)
1457 {
1458     int index;
1459     struct h_AddrHashChain *chain;
1460     char hoststr[16];
1461
1462     /* hash into proper bucket */
1463     index = h_HashIndex(addr);
1464
1465     /* don't add the same address:port pair entry multiple times */
1466     for (chain = hostAddrHashTable[index]; chain; chain = chain->next) {
1467         if (chain->addr == addr && chain->port == port) {
1468             if (chain->hostPtr == host) {
1469                 ViceLog(125,
1470                         ("h_AddHostToAddrHashTable_r: host %" AFS_PTR_FMT " (%s:%d) already hashed\n",
1471                           host, afs_inet_ntoa_r(chain->addr, hoststr),
1472                           ntohs(chain->port)));
1473                 return;
1474             }
1475             if (!(chain->hostPtr->hostFlags & HOSTDELETED)) {
1476                 /* attempt to resolve host address collision */
1477                 reconcileHosts_r(addr, port, host, chain->hostPtr);
1478                 return;
1479             }
1480         }
1481     }
1482     createHostAddrHashChain_r(index, addr, port, host);
1483 }
1484
1485 /*
1486  * This is called with host locked and held.
1487  * It is called to either validate or add an additional interface
1488  * address/port on the specified host.
1489  *
1490  * All addresses are in network byte order.
1491  */
1492 int
1493 addInterfaceAddr_r(struct host *host, afs_uint32 addr, afs_uint16 port)
1494 {
1495     int i;
1496     int number;
1497     struct Interface *interface;
1498     char hoststr[16], hoststr2[16];
1499
1500     osi_Assert(host);
1501     osi_Assert(host->interface);
1502
1503     /*
1504      * Make sure this address is on the list of known addresses
1505      * for this host.
1506      */
1507     number = host->interface->numberOfInterfaces;
1508     for (i = 0; i < number; i++) {
1509         if (host->interface->interface[i].addr == addr &&
1510              host->interface->interface[i].port == port) {
1511             ViceLog(125,
1512                     ("addInterfaceAddr : found host %" AFS_PTR_FMT " (%s:%d) adding %s:%d%s\n",
1513                      host, afs_inet_ntoa_r(host->host, hoststr),
1514                      ntohs(host->port), afs_inet_ntoa_r(addr, hoststr2),
1515                      ntohs(port), host->interface->interface[i].valid ? "" :
1516                      ", validating"));
1517
1518             if (host->interface->interface[i].valid == 0) {
1519                 host->interface->interface[i].valid = 1;
1520                 h_AddHostToAddrHashTable_r(addr, port, host);
1521             }
1522             return 0;
1523         }
1524     }
1525
1526     ViceLog(125, ("addInterfaceAddr : host %" AFS_PTR_FMT " (%s:%d) adding %s:%d\n",
1527                   host, afs_inet_ntoa_r(host->host, hoststr),
1528                   ntohs(host->port), afs_inet_ntoa_r(addr, hoststr2),
1529                   ntohs(port)));
1530
1531     interface = malloc(sizeof(struct Interface)
1532                        + (sizeof(struct AddrPort) * number));
1533     if (!interface) {
1534         ViceLogThenPanic(0, ("Failed malloc in addInterfaceAddr_r\n"));
1535     }
1536     interface->numberOfInterfaces = number + 1;
1537     interface->uuid = host->interface->uuid;
1538     for (i = 0; i < number; i++)
1539         interface->interface[i] = host->interface->interface[i];
1540
1541     /* Add the new valid interface */
1542     interface->interface[number].addr = addr;
1543     interface->interface[number].port = port;
1544     interface->interface[number].valid = 1;
1545     h_AddHostToAddrHashTable_r(addr, port, host);
1546     free(host->interface);
1547     host->interface = interface;
1548
1549     return 0;
1550 }
1551
1552
1553 /*
1554  * This is called with host locked and held.
1555  *
1556  * All addresses are in network byte order.
1557  */
1558 int
1559 removeInterfaceAddr_r(struct host *host, afs_uint32 addr, afs_uint16 port)
1560 {
1561     int i;
1562     int number;
1563     struct Interface *interface;
1564     char hoststr[16], hoststr2[16];
1565
1566     osi_Assert(host);
1567     osi_Assert(host->interface);
1568
1569     ViceLog(125, ("removeInterfaceAddr : host %" AFS_PTR_FMT " (%s:%d) addr %s:%d\n",
1570                   host, afs_inet_ntoa_r(host->host, hoststr),
1571                   ntohs(host->port), afs_inet_ntoa_r(addr, hoststr2),
1572                   ntohs(port)));
1573
1574     /*
1575      * Make sure this address is on the list of known addresses
1576      * for this host.
1577      */
1578     interface = host->interface;
1579     number = host->interface->numberOfInterfaces;
1580     for (i = 0; i < number; i++) {
1581         if (interface->interface[i].addr == addr &&
1582             interface->interface[i].port == port) {
1583             if (interface->interface[i].valid)
1584                 h_DeleteHostFromAddrHashTable_r(addr, port, host);
1585             number--;
1586             for (; i < number; i++) {
1587                 interface->interface[i] = interface->interface[i+1];
1588             }
1589             interface->numberOfInterfaces = number;
1590             return 0;
1591         }
1592     }
1593     /* not found */
1594     return 0;
1595 }
1596
1597
1598
1599 static int
1600 h_threadquota(int waiting)
1601 {
1602     if (lwps > 64) {
1603         if (waiting > 5)
1604             return 1;
1605     } else if (lwps > 32) {
1606         if (waiting > 4)
1607             return 1;
1608     } else if (lwps > 16) {
1609         if (waiting > 3)
1610             return 1;
1611     } else {
1612         if (waiting > 2)
1613             return 1;
1614     }
1615     return 0;
1616 }
1617
1618 /* If found, host is returned with refCount incremented */
1619 struct host *
1620 h_GetHost_r(struct rx_connection *tcon)
1621 {
1622     struct host *host;
1623     struct host *oldHost;
1624     int code;
1625     struct interfaceAddr interf;
1626     int interfValid = 0;
1627     struct Identity *identP = NULL;
1628     afs_uint32 haddr;
1629     afs_uint16 hport;
1630     char hoststr[16], hoststr2[16];
1631     Capabilities caps;
1632     struct rx_connection *cb_conn = NULL;
1633     struct rx_connection *cb_in = NULL;
1634
1635     caps.Capabilities_val = NULL;
1636
1637     haddr = rxr_HostOf(tcon);
1638     hport = rxr_PortOf(tcon);
1639   retry:
1640     if (cb_in) {
1641         rx_DestroyConnection(cb_in);
1642         cb_in = NULL;
1643     }
1644     if (caps.Capabilities_val)
1645         free(caps.Capabilities_val);
1646     caps.Capabilities_val = NULL;
1647     caps.Capabilities_len = 0;
1648
1649     code = 0;
1650     if (h_Lookup_r(haddr, hport, &host))
1651         return 0;
1652     identP = (struct Identity *)rx_GetSpecific(tcon, rxcon_ident_key);
1653     if (host && !identP && !(host->Console & 1)) {
1654         /* This is a new connection, and we already have a host
1655          * structure for this address. Verify that the identity
1656          * of the caller matches the identity in the host structure.
1657          */
1658         if ((host->hostFlags & HWHO_INPROGRESS) &&
1659             h_threadquota(host->lock.num_waiting)) {
1660                 h_Release_r(host);
1661             host = NULL;
1662             goto gethost_out;
1663         }
1664         h_Lock_r(host);
1665         if (!(host->hostFlags & ALTADDR) ||
1666             (host->hostFlags & HOSTDELETED)) {
1667             /* Another thread is doing initialization
1668              * or this host was deleted while we
1669              * waited for the lock. */
1670             h_Unlock_r(host);
1671             ViceLog(125,
1672                     ("Host %" AFS_PTR_FMT " (%s:%d) starting h_Lookup again\n",
1673                      host, afs_inet_ntoa_r(host->host, hoststr),
1674                      ntohs(host->port)));
1675             h_Release_r(host);
1676             goto retry;
1677         }
1678         host->hostFlags |= HWHO_INPROGRESS;
1679         host->hostFlags &= ~ALTADDR;
1680
1681         /* We received a new connection from an IP address/port
1682          * that is associated with 'host' but the address/port of
1683          * the callback connection does not have to match it.
1684          * If there is a match, we can use the existing callback
1685          * connection to verify the UUID.  If they do not match
1686          * we need to use a new callback connection to verify the
1687          * UUID of the incoming caller and perhaps use the old
1688          * callback connection to verify that the old address/port
1689          * is still valid.
1690          */
1691
1692         cb_conn = host->callback_rxcon;
1693         rx_GetConnection(cb_conn);
1694         H_UNLOCK;
1695         if (haddr == host->host && hport == host->port) {
1696             /* The existing callback connection matches the
1697              * incoming connection so just use it.
1698              */
1699             code =
1700                 RXAFSCB_TellMeAboutYourself(cb_conn, &interf, &caps);
1701             if (code == RXGEN_OPCODE)
1702                 code = RXAFSCB_WhoAreYou(cb_conn, &interf);
1703         } else {
1704             /* We do not have a match.  Create a new connection
1705              * for the new addr/port and use multi_Rx to probe
1706              * both of them simultaneously.
1707              */
1708             if (!sc)
1709                 sc = rxnull_NewClientSecurityObject();
1710             cb_in = rx_NewConnection(haddr, hport, 1, sc, 0);
1711             rx_SetConnDeadTime(cb_in, 50);
1712             rx_SetConnHardDeadTime(cb_in, AFS_HARDDEADTIME);
1713             rx_SetConnSecondsUntilNatPing(cb_in, 20);
1714
1715             code =
1716                 RXAFSCB_TellMeAboutYourself(cb_in, &interf, &caps);
1717             if (code == RXGEN_OPCODE)
1718                 code = RXAFSCB_WhoAreYou(cb_in, &interf);
1719         }
1720         rx_PutConnection(cb_conn);
1721         cb_conn=NULL;
1722         H_LOCK;
1723         if ((code == RXGEN_OPCODE) ||
1724             ((code == 0) && (afs_uuid_equal(&interf.uuid, &nulluuid)))) {
1725             identP = malloc(sizeof(struct Identity));
1726             if (!identP) {
1727                 ViceLogThenPanic(0, ("Failed malloc in h_GetHost_r\n"));
1728             }
1729             identP->valid = 0;
1730             rx_SetSpecific(tcon, rxcon_ident_key, identP);
1731             if (cb_in == NULL) {
1732                 /* The host on this connection was unable to respond to
1733                  * the WhoAreYou. We will treat this as a new connection
1734                  * from the existing host. The worst that can happen is
1735                  * that we maintain some extra callback state information */
1736                 if (host->interface) {
1737                     ViceLog(0,
1738                             ("Host %" AFS_PTR_FMT " (%s:%d) used to support WhoAreYou, deleting.\n",
1739                              host,
1740                              afs_inet_ntoa_r(host->host, hoststr),
1741                              ntohs(host->port)));
1742                     host->hostFlags |= HOSTDELETED;
1743                     host->hostFlags &= ~HWHO_INPROGRESS;
1744                     h_Unlock_r(host);
1745                     h_Release_r(host);
1746                     host = NULL;
1747                     goto retry;
1748                 }
1749             } else {
1750                 /* The incoming connection does not support WhoAreYou but
1751                  * the original one might have.  Use removeAddress_r() to
1752                  * remove this addr/port from the host that was found.
1753                  * If there are no more addresses left for the host it
1754                  * will be deleted.  Then we retry.
1755                  */
1756                 removeAddress_r(host, haddr, hport);
1757                 host->hostFlags &= ~HWHO_INPROGRESS;
1758                 host->hostFlags |= ALTADDR;
1759                 h_Unlock_r(host);
1760                 h_Release_r(host);
1761                 host = NULL;
1762                 goto retry;
1763             }
1764         } else if (code == 0) {
1765             interfValid = 1;
1766             identP = malloc(sizeof(struct Identity));
1767             if (!identP) {
1768                 ViceLogThenPanic(0, ("Failed malloc in h_GetHost_r\n"));
1769             }
1770             identP->valid = 1;
1771             identP->uuid = interf.uuid;
1772             rx_SetSpecific(tcon, rxcon_ident_key, identP);
1773             /* Check whether the UUID on this connection matches
1774              * the UUID in the host structure. If they don't match
1775              * then this is not the same host as before. */
1776             if (!host->interface
1777                 || !afs_uuid_equal(&interf.uuid, &host->interface->uuid)) {
1778                 if (cb_in) {
1779                         ViceLog(25,
1780                                         ("Uuid doesn't match connection (%s:%d).\n",
1781                                          afs_inet_ntoa_r(haddr, hoststr), ntohs(hport)));
1782                         removeAddress_r(host, haddr, hport);
1783                 } else {
1784                     ViceLog(25,
1785                             ("Uuid doesn't match host %" AFS_PTR_FMT " (%s:%d).\n",
1786                              host, afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port)));
1787
1788                     removeAddress_r(host, host->host, host->port);
1789                 }
1790                 host->hostFlags &= ~HWHO_INPROGRESS;
1791                 host->hostFlags |= ALTADDR;
1792                 h_Unlock_r(host);
1793                 h_Release_r(host);
1794                 host = NULL;
1795                 goto retry;
1796             } else if (cb_in) {
1797                 /* the UUID matched the client at the incoming addr/port
1798                  * but this is not the address of the active callback
1799                  * connection.  Try that connection and see if the client
1800                  * is still there and if the reported UUID is the same.
1801                  */
1802                 int code2;
1803                 afsUUID uuid = host->interface->uuid;
1804                 cb_conn = host->callback_rxcon;
1805                 rx_GetConnection(cb_conn);
1806                 rx_SetConnDeadTime(cb_conn, 2);
1807                 rx_SetConnHardDeadTime(cb_conn, AFS_HARDDEADTIME);
1808                 H_UNLOCK;
1809                 code2 = RXAFSCB_ProbeUuid(cb_conn, &uuid);
1810                 H_LOCK;
1811                 rx_SetConnDeadTime(cb_conn, 50);
1812                 rx_SetConnHardDeadTime(cb_conn, AFS_HARDDEADTIME);
1813                 rx_PutConnection(cb_conn);
1814                 cb_conn=NULL;
1815                 if (code2) {
1816                     /* The primary address is either not responding or
1817                      * is not the client we are looking for.  Need to
1818                      * remove the primary address and add swap in the new
1819                      * callback connection, and destroy the old one.
1820                      */
1821                     struct rx_connection *rxconn;
1822                     ViceLog(0,("CB: ProbeUuid for host %" AFS_PTR_FMT " (%s:%d) failed %d\n",
1823                                host,
1824                                afs_inet_ntoa_r(host->host, hoststr),
1825                                ntohs(host->port),code2));
1826
1827                     /*
1828                      * make sure we add and then remove.  otherwise, we
1829                      * might end up with no valid interfaces after the
1830                      * remove and the host will have been marked deleted.
1831                      */
1832                     addInterfaceAddr_r(host, haddr, hport);
1833                     removeInterfaceAddr_r(host, host->host, host->port);
1834                     host->host = haddr;
1835                     host->port = hport;
1836                     rxconn = host->callback_rxcon;
1837                     host->callback_rxcon = cb_in;
1838                     cb_in = NULL;
1839
1840                     if (rxconn) {
1841                         /*
1842                          * If rx_DestroyConnection calls h_FreeConnection we
1843                          * will deadlock on the host_glock_mutex. Work around
1844                          * the problem by unhooking the client from the
1845                          * connection before destroying the connection.
1846                          */
1847                         rx_SetSpecific(rxconn, rxcon_client_key, (void *)0);
1848                         rx_DestroyConnection(rxconn);
1849                     }
1850                 }
1851             }
1852         } else {
1853             if (cb_in) {
1854                 /* A callback to the incoming connection address is failing.
1855                  * Assume that the addr/port is no longer associated with the host
1856                  * returned by h_Lookup_r.
1857                  */
1858                 ViceLog(0,
1859                         ("CB: WhoAreYou failed for connection (%s:%d) , error %d\n",
1860                          afs_inet_ntoa_r(haddr, hoststr), ntohs(hport), code));
1861                 removeAddress_r(host, haddr, hport);
1862                 host->hostFlags &= ~HWHO_INPROGRESS;
1863                 host->hostFlags |= ALTADDR;
1864                 h_Unlock_r(host);
1865                 h_Release_r(host);
1866                 host = NULL;
1867                 rx_DestroyConnection(cb_in);
1868                 cb_in = NULL;
1869                 goto gethost_out;
1870             } else {
1871                 ViceLog(0,
1872                         ("CB: WhoAreYou failed for host %" AFS_PTR_FMT " (%s:%d), error %d\n",
1873                          host, afs_inet_ntoa_r(host->host, hoststr),
1874                          ntohs(host->port), code));
1875                 host->hostFlags |= VENUSDOWN;
1876             }
1877         }
1878         if (caps.Capabilities_val
1879             && (caps.Capabilities_val[0] & CLIENT_CAPABILITY_ERRORTRANS))
1880             host->hostFlags |= HERRORTRANS;
1881         else
1882             host->hostFlags &= ~(HERRORTRANS);
1883         host->hostFlags |= ALTADDR;
1884         host->hostFlags &= ~HWHO_INPROGRESS;
1885         h_Unlock_r(host);
1886     } else if (host) {
1887         if (!(host->hostFlags & ALTADDR)) {
1888             /* another thread is doing the initialisation */
1889             ViceLog(125,
1890                     ("Host %" AFS_PTR_FMT " (%s:%d) waiting for host-init to complete\n",
1891                      host, afs_inet_ntoa_r(host->host, hoststr),
1892                      ntohs(host->port)));
1893             h_Lock_r(host);
1894             h_Unlock_r(host);
1895             ViceLog(125,
1896                     ("Host %" AFS_PTR_FMT " (%s:%d) starting h_Lookup again\n",
1897                      host, afs_inet_ntoa_r(host->host, hoststr),
1898                      ntohs(host->port)));
1899             h_Release_r(host);
1900             goto retry;
1901         }
1902         /* We need to check whether the identity in the host structure
1903          * matches the identity on the connection. If they don't match
1904          * then treat this a new host. */
1905         if (!(host->Console & 1)
1906             && ((!identP->valid && host->interface)
1907                 || (identP->valid && !host->interface)
1908                 || (identP->valid
1909                     && !afs_uuid_equal(&identP->uuid,
1910                                        &host->interface->uuid)))) {
1911             char uuid1[128], uuid2[128];
1912             if (identP->valid)
1913                 afsUUID_to_string(&identP->uuid, uuid1, 127);
1914             if (host->interface)
1915                 afsUUID_to_string(&host->interface->uuid, uuid2, 127);
1916             ViceLog(0,
1917                     ("CB: new identity for host %p (%s:%d), "
1918                      "deleting(%x %p %s %s)\n",
1919                      host, afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port),
1920                      identP->valid, host->interface,
1921                      identP->valid ? uuid1 : "no_uuid",
1922                      host->interface ? uuid2 : "no_uuid"));
1923
1924             /* The host in the cache is not the host for this connection */
1925             h_Lock_r(host);
1926             host->hostFlags |= HOSTDELETED;
1927             h_Unlock_r(host);
1928             h_Release_r(host);
1929             goto retry;
1930         }
1931     } else {
1932         host = h_Alloc_r(tcon); /* returned held and locked */
1933         if (!host)
1934             goto gethost_out;
1935         h_gethostcps_r(host, FT_ApproxTime());
1936         if (!(host->Console & 1)) {
1937             int pident = 0;
1938             cb_conn = host->callback_rxcon;
1939             rx_GetConnection(cb_conn);
1940             host->hostFlags |= HWHO_INPROGRESS;
1941             H_UNLOCK;
1942             code =
1943                 RXAFSCB_TellMeAboutYourself(cb_conn, &interf, &caps);
1944             if (code == RXGEN_OPCODE)
1945                 code = RXAFSCB_WhoAreYou(cb_conn, &interf);
1946             rx_PutConnection(cb_conn);
1947             cb_conn=NULL;
1948             H_LOCK;
1949             if ((code == RXGEN_OPCODE) ||
1950                 ((code == 0) && (afs_uuid_equal(&interf.uuid, &nulluuid)))) {
1951                 if (!identP)
1952                     identP = malloc(sizeof(struct Identity));
1953                 else
1954                     pident = 1;
1955
1956                 if (!identP) {
1957                     ViceLogThenPanic(0, ("Failed malloc in h_GetHost_r\n"));
1958                 }
1959                 identP->valid = 0;
1960                 if (!pident)
1961                     rx_SetSpecific(tcon, rxcon_ident_key, identP);
1962                 ViceLog(25,
1963                         ("Host %" AFS_PTR_FMT " (%s:%d) does not support WhoAreYou.\n",
1964                          host, afs_inet_ntoa_r(host->host, hoststr),
1965                          ntohs(host->port)));
1966                 code = 0;
1967             } else if (code == 0) {
1968                 if (!identP)
1969                     identP = malloc(sizeof(struct Identity));
1970                 else
1971                     pident = 1;
1972
1973                 if (!identP) {
1974                     ViceLogThenPanic(0, ("Failed malloc in h_GetHost_r\n"));
1975                 }
1976                 identP->valid = 1;
1977                 interfValid = 1;
1978                 identP->uuid = interf.uuid;
1979                 if (!pident)
1980                     rx_SetSpecific(tcon, rxcon_ident_key, identP);
1981                 ViceLog(25,
1982                         ("WhoAreYou success on host %" AFS_PTR_FMT " (%s:%d)\n",
1983                          host, afs_inet_ntoa_r(host->host, hoststr),
1984                          ntohs(host->port)));
1985             }
1986             if (code == 0 && !identP->valid) {
1987                 cb_conn = host->callback_rxcon;
1988                 rx_GetConnection(cb_conn);
1989                 H_UNLOCK;
1990                 code = RXAFSCB_InitCallBackState(cb_conn);
1991                 rx_PutConnection(cb_conn);
1992                 cb_conn=NULL;
1993                 H_LOCK;
1994             } else if (code == 0) {
1995                 oldHost = h_LookupUuid_r(&identP->uuid);
1996                 if (oldHost) {
1997                     h_Hold_r(oldHost);
1998                     h_Lock_r(oldHost);
1999
2000                     if (oldHost->hostFlags & HOSTDELETED) {
2001                         h_Unlock_r(oldHost);
2002                         h_Release_r(oldHost);
2003                         oldHost = NULL;
2004                     }
2005                 }
2006
2007                 if (oldHost) {
2008                     int probefail = 0;
2009
2010                     /* This is a new address for an existing host. Update
2011                      * the list of interfaces for the existing host and
2012                      * delete the host structure we just allocated. */
2013
2014                     /* mark the duplicate host as deleted before we do
2015                      * anything. The probing code below may try to change
2016                      * "oldHost" to the same IP address as "host" currently
2017                      * has, and we do not want a pseudo-"collision" to be
2018                      * noticed. */
2019                     host->hostFlags |= HOSTDELETED;
2020
2021                     oldHost->hostFlags |= HWHO_INPROGRESS;
2022
2023                     if (oldHost->interface) {
2024                         int code2;
2025                         afsUUID uuid = oldHost->interface->uuid;
2026                         cb_conn = oldHost->callback_rxcon;
2027                         rx_GetConnection(cb_conn);
2028                         rx_SetConnDeadTime(cb_conn, 2);
2029                         rx_SetConnHardDeadTime(cb_conn, AFS_HARDDEADTIME);
2030                         H_UNLOCK;
2031                         code2 = RXAFSCB_ProbeUuid(cb_conn, &uuid);
2032                         H_LOCK;
2033                         rx_SetConnDeadTime(cb_conn, 50);
2034                         rx_SetConnHardDeadTime(cb_conn, AFS_HARDDEADTIME);
2035                         rx_PutConnection(cb_conn);
2036                         cb_conn=NULL;
2037                         if (code2) {
2038                             /* The primary address is either not responding or
2039                              * is not the client we are looking for.
2040                              * MultiProbeAlternateAddress_r() will remove the
2041                              * alternate interfaces that do not have the same
2042                              * Uuid. */
2043                             ViceLog(0,("CB: ProbeUuid for host %" AFS_PTR_FMT " (%s:%d) failed %d\n",
2044                                          oldHost,
2045                                          afs_inet_ntoa_r(oldHost->host, hoststr),
2046                                          ntohs(oldHost->port),code2));
2047
2048                             if (MultiProbeAlternateAddress_r(oldHost)) {
2049                                 /* If MultiProbeAlternateAddress_r succeeded,
2050                                  * it updated oldHost->host and oldHost->port
2051                                  * to an address that responded successfully to
2052                                  * a ProbeUuid, so it is as if the ProbeUuid
2053                                  * call above returned success. So, only set
2054                                  * 'probefail' if MultiProbeAlternateAddress_r
2055                                  * fails. */
2056                                 probefail = 1;
2057                             }
2058                         }
2059                     } else {
2060                         probefail = 1;
2061                     }
2062
2063                     if (oldHost->host != haddr || oldHost->port != hport) {
2064                         struct rx_connection *rxconn;
2065
2066                         ViceLog(25,
2067                                  ("CB: Host %" AFS_PTR_FMT " (%s:%d) has new addr %s:%d\n",
2068                                    oldHost,
2069                                    afs_inet_ntoa_r(oldHost->host, hoststr2),
2070                                    ntohs(oldHost->port),
2071                                    afs_inet_ntoa_r(haddr, hoststr),
2072                                    ntohs(hport)));
2073                         /*
2074                          * add then remove.  otherwise the host may get marked
2075                          * deleted if we removed the only valid address.
2076                          */
2077                         addInterfaceAddr_r(oldHost, haddr, hport);
2078                         if (probefail || oldHost->host == haddr) {
2079                             /*
2080                              * The probe failed which means that the old
2081                              * address is either unreachable or is not the
2082                              * same host we were just contacted by.  We will
2083                              * also remove addresses if only the port has
2084                              * changed because that indicates the client
2085                              * is behind a NAT.
2086                              */
2087                             removeInterfaceAddr_r(oldHost, oldHost->host, oldHost->port);
2088                         } else {
2089                             int i;
2090                             struct Interface *interface = oldHost->interface;
2091                             int number = oldHost->interface->numberOfInterfaces;
2092                             for (i = 0; i < number; i++) {
2093                                 if (interface->interface[i].addr == haddr &&
2094                                     interface->interface[i].port != hport) {
2095                                     /*
2096                                      * We have just been contacted by a client
2097                                      * that has been seen from behind a NAT
2098                                      * and at least one other address.
2099                                      */
2100                                     removeInterfaceAddr_r(oldHost, haddr,
2101                                                           interface->interface[i].port);
2102                                     break;
2103                                 }
2104                             }
2105                         }
2106                         oldHost->host = haddr;
2107                         oldHost->port = hport;
2108                         rxconn = oldHost->callback_rxcon;
2109                         oldHost->callback_rxcon = host->callback_rxcon;
2110                         host->callback_rxcon = rxconn;
2111
2112                         /* don't destroy rxconn here; let h_TossStuff_r
2113                          * take care of that via h_Release_r below */
2114                     }
2115                     host->hostFlags &= ~HWHO_INPROGRESS;
2116                     h_Unlock_r(host);
2117                     /* release host because it was allocated by h_Alloc_r */
2118                     h_Release_r(host);
2119                     host = oldHost;
2120                     /* the new host is held and locked */
2121                 } else {
2122                     /* This really is a new host */
2123                     h_AddHostToUuidHashTable_r(&identP->uuid, host);
2124                     cb_conn = host->callback_rxcon;
2125                     rx_GetConnection(cb_conn);
2126                     H_UNLOCK;
2127                     code =
2128                         RXAFSCB_InitCallBackState3(cb_conn,
2129                                                    &FS_HostUUID);
2130                     rx_PutConnection(cb_conn);
2131                     cb_conn=NULL;
2132                     H_LOCK;
2133                     if (code == 0) {
2134                         ViceLog(25,
2135                                 ("InitCallBackState3 success on host %" AFS_PTR_FMT " (%s:%d)\n",
2136                                  host, afs_inet_ntoa_r(host->host, hoststr),
2137                                  ntohs(host->port)));
2138                         osi_Assert(interfValid == 1);
2139                         initInterfaceAddr_r(host, &interf);
2140                     }
2141                 }
2142             }
2143             if (code) {
2144                 ViceLog(0,
2145                         ("CB: RCallBackConnectBack failed for %" AFS_PTR_FMT " (%s:%d)\n",
2146                          host, afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port)));
2147                 host->hostFlags |= VENUSDOWN;
2148             } else {
2149                 ViceLog(125,
2150                         ("CB: RCallBackConnectBack succeeded for %" AFS_PTR_FMT " (%s:%d)\n",
2151                          host, afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port)));
2152                 host->hostFlags |= RESETDONE;
2153             }
2154         }
2155         if (caps.Capabilities_val
2156             && (caps.Capabilities_val[0] & CLIENT_CAPABILITY_ERRORTRANS))
2157             host->hostFlags |= HERRORTRANS;
2158         else
2159             host->hostFlags &= ~(HERRORTRANS);
2160         host->hostFlags |= ALTADDR;     /* host structure initialization complete */
2161         host->hostFlags &= ~HWHO_INPROGRESS;
2162         h_Unlock_r(host);
2163     }
2164
2165  gethost_out:
2166     if (caps.Capabilities_val)
2167         free(caps.Capabilities_val);
2168     caps.Capabilities_val = NULL;
2169     caps.Capabilities_len = 0;
2170     if (cb_in) {
2171         rx_DestroyConnection(cb_in);
2172         cb_in = NULL;
2173     }
2174     return host;
2175
2176 }                               /*h_GetHost_r */
2177
2178
2179 static char localcellname[PR_MAXNAMELEN + 1];
2180
2181 /* not reentrant */
2182 void
2183 h_InitHostPackage(void)
2184 {
2185     memset(&nulluuid, 0, sizeof(afsUUID));
2186     afsconf_GetLocalCell(confDir, localcellname, PR_MAXNAMELEN);
2187     rxcon_ident_key = rx_KeyCreate((rx_destructor_t) free);
2188     rxcon_client_key = rx_KeyCreate((rx_destructor_t) 0);
2189     MUTEX_INIT(&host_glock_mutex, "host glock", MUTEX_DEFAULT, 0);
2190 }
2191
2192 static int
2193 MapName_r(char *uname, afs_int32 * aval)
2194 {
2195     namelist lnames;
2196     idlist lids;
2197     afs_int32 code;
2198
2199     lnames.namelist_len = 1;
2200     lnames.namelist_val = (prname *) uname;
2201     lids.idlist_len = 0;
2202     lids.idlist_val = NULL;
2203
2204     H_UNLOCK;
2205     code = hpr_NameToId(&lnames, &lids);
2206     H_LOCK;
2207     if (code == 0) {
2208         if (lids.idlist_val) {
2209             *aval = lids.idlist_val[0];
2210             if (*aval == AnonymousID) {
2211                 ViceLog(2,
2212                         ("MapName: NameToId on %s returns anonymousID\n",
2213                          lnames.namelist_val[0]));
2214             }
2215             free(lids.idlist_val);      /* return parms are not malloced in stub if server proc aborts */
2216         } else {
2217             ViceLog(0,
2218                     ("MapName: NameToId on '%s' is unknown\n",
2219                      lnames.namelist_val[0]));
2220             code = -1;
2221         }
2222     }
2223     return code;
2224 }
2225
2226 /*MapName*/
2227
2228
2229 /* NOTE: this returns the client with a Write lock and a refCount */
2230 struct client *
2231 h_ID2Client(afs_int32 vid)
2232 {
2233     struct client *client;
2234     struct host *host;
2235     int count;
2236
2237     H_LOCK;
2238     for (count = 0, host = hostList; host && count < hostCount; host = host->next, count++) {
2239         if (host->hostFlags & HOSTDELETED)
2240             continue;
2241         for (client = host->FirstClient; client; client = client->next) {
2242             if (!client->deleted && client->ViceId == vid) {
2243                 client->refCount++;
2244                 H_UNLOCK;
2245                 ObtainWriteLock(&client->lock);
2246                 return client;
2247             }
2248         }
2249     }
2250     if (count != hostCount) {
2251         ViceLog(0, ("h_ID2Client found %d of %d hosts\n", count, hostCount));
2252     } else if (host != NULL) {
2253         ViceLog(0, ("h_ID2Client found more than %d hosts\n", hostCount));
2254         ShutDownAndCore(PANIC);
2255     }
2256
2257     H_UNLOCK;
2258     return NULL;
2259 }
2260
2261 static int
2262 format_vname(char *vname, int usize, const char *tname, const char *tinst,
2263              const char *tcell, afs_int32 islocal)
2264 {
2265     int len;
2266
2267     len = strlcpy(vname, tname, usize);
2268     if (len >= usize)
2269         return -1;
2270     if (tinst[0]) {
2271         len = strlcat(vname, ".", usize);
2272         if (len >= usize)
2273             return -1;
2274         len = strlcat(vname, tinst, usize);
2275         if (len >= usize)
2276             return -1;
2277     }
2278     if (tcell[0] && !islocal) {
2279         len = strlcat(vname, "@", usize);
2280         if (len >= usize)
2281             return -1;
2282         len = strlcat(vname, tcell, usize);
2283         if (len >= usize)
2284             return -1;
2285     }
2286     return 0;
2287 }
2288
2289 /*
2290  * Called by the server main loop.  Returns a h_Held client, which must be
2291  * released later the main loop.  Allocates a client if the matching one
2292  * isn't around. The client is returned with its reference count incremented
2293  * by one. The caller must call h_ReleaseClient_r when finished with
2294  * the client.
2295  *
2296  * The refCount on client->host is returned incremented.  h_ReleaseClient_r
2297  * does not decrement the refCount on client->host.
2298  */
2299 struct client *
2300 h_FindClient_r(struct rx_connection *tcon)
2301 {
2302     struct client *client;
2303     struct host *host = NULL;
2304     struct client *oldClient;
2305     afs_int32 viceid = 0;
2306     afs_int32 expTime;
2307     afs_int32 code;
2308     int authClass;
2309 #if (64-MAXKTCNAMELEN)
2310     ticket name length != 64
2311 #endif
2312     char tname[64];
2313     char tinst[64];
2314     char uname[PR_MAXNAMELEN];
2315     char tcell[MAXKTCREALMLEN];
2316     int fail = 0;
2317     int created = 0;
2318
2319     client = (struct client *)rx_GetSpecific(tcon, rxcon_client_key);
2320     if (client && client->sid == rx_GetConnectionId(tcon)
2321         && client->VenusEpoch == rx_GetConnectionEpoch(tcon)
2322         && !(client->host->hostFlags & HOSTDELETED)
2323         && !client->deleted) {
2324
2325         client->refCount++;
2326         h_Hold_r(client->host);
2327         if (client->prfail != 2) {
2328             /* Could add shared lock on client here */
2329             /* note that we don't have to lock entry in this path to
2330              * ensure CPS is initialized, since we don't call rx_SetSpecific
2331              * until initialization is done, and we only get here if
2332              * rx_GetSpecific located the client structure.
2333              */
2334             return client;
2335         }
2336         H_UNLOCK;
2337         ObtainWriteLock(&client->lock); /* released at end */
2338         H_LOCK;
2339     } else {
2340         client = NULL;
2341     }
2342
2343     authClass = rx_SecurityClassOf((struct rx_connection *)tcon);
2344     ViceLog(5,
2345             ("FindClient: authenticating connection: authClass=%d\n",
2346              authClass));
2347     if (authClass == 1) {
2348         /* A bcrypt tickets, no longer supported */
2349         ViceLog(1, ("FindClient: bcrypt ticket, using AnonymousID\n"));
2350         viceid = AnonymousID;
2351         expTime = 0x7fffffff;
2352     } else if (authClass == 2) {
2353         afs_int32 kvno;
2354         afs_int32 islocal;
2355
2356         /* kerberos ticket */
2357         code = rxkad_GetServerInfo(tcon, /*level */ 0, (afs_uint32 *)&expTime,
2358                                    tname, tinst, tcell, &kvno);
2359         if (code) {
2360             ViceLog(1, ("Failed to get rxkad ticket info\n"));
2361             viceid = AnonymousID;
2362             expTime = 0x7fffffff;
2363         } else {
2364             ViceLog(5,
2365                     ("FindClient: rxkad conn: name=%s,inst=%s,cell=%s,exp=%d,kvno=%d\n",
2366                      tname, tinst, tcell, expTime, kvno));
2367             code = afsconf_IsLocalRealmMatch(confDir, &islocal, tname, tinst, tcell);
2368             if (code) {
2369                 ViceLog(0, ("FindClient: local realm check failed; code=%d", code));
2370                 viceid = AnonymousID;
2371                 expTime = 0x7fffffff;
2372             }
2373             if (!code) {
2374                 code = format_vname(uname, sizeof(uname), tname, tinst, tcell, islocal);
2375                 if (code) {
2376                     ViceLog(0, ("FindClient: uname truncated."));
2377                     viceid = AnonymousID;
2378                     expTime = 0x7fffffff;
2379                 }
2380             }
2381             if (!code) {
2382                 /* translate the name to a vice id */
2383                 code = MapName_r(uname, &viceid);
2384                 if (code) {
2385                     ViceLog(1,
2386                             ("failed to map name=%s -> code=%d\n", uname,
2387                              code));
2388                     fail = 1;
2389                     viceid = AnonymousID;
2390                     expTime = 0x7fffffff;
2391                 }
2392             }
2393         }
2394     } else {
2395         viceid = AnonymousID;   /* unknown security class */
2396         expTime = 0x7fffffff;
2397     }
2398
2399     if (!client) { /* loop */
2400         host = h_GetHost_r(tcon);       /* Returns with incremented refCount  */
2401
2402         if (!host)
2403             return NULL;
2404
2405     retryfirstclient:
2406         /* First try to find the client structure */
2407         for (client = host->FirstClient; client; client = client->next) {
2408             if (!client->deleted && (client->sid == rx_GetConnectionId(tcon))
2409                 && (client->VenusEpoch == rx_GetConnectionEpoch(tcon))) {
2410                 client->refCount++;
2411                 H_UNLOCK;
2412                 ObtainWriteLock(&client->lock);
2413                 H_LOCK;
2414                 break;
2415             }
2416         }
2417
2418         /* Still no client structure - get one */
2419         if (!client) {
2420             h_Lock_r(host);
2421             if (host->hostFlags & HOSTDELETED) {
2422                 h_Unlock_r(host);
2423                 h_Release_r(host);
2424                 return NULL;
2425             }
2426             /* Retry to find the client structure */
2427             for (client = host->FirstClient; client; client = client->next) {
2428                 if (!client->deleted && (client->sid == rx_GetConnectionId(tcon))
2429                     && (client->VenusEpoch == rx_GetConnectionEpoch(tcon))) {
2430                     h_Unlock_r(host);
2431                     goto retryfirstclient;
2432                 }
2433             }
2434             created = 1;
2435             client = GetCE();
2436             ObtainWriteLock(&client->lock);
2437             client->refCount = 1;
2438             client->host = host;
2439             client->InSameNetwork = host->InSameNetwork;
2440             client->ViceId = viceid;
2441             client->expTime = expTime;  /* rx only */
2442             client->authClass = authClass;      /* rx only */
2443             client->sid = rx_GetConnectionId(tcon);
2444             client->VenusEpoch = rx_GetConnectionEpoch(tcon);
2445             client->CPS.prlist_val = NULL;
2446             client->CPS.prlist_len = 0;
2447             h_Unlock_r(host);
2448         }
2449     }
2450     client->prfail = fail;
2451
2452     if (!(client->CPS.prlist_val) || (viceid != client->ViceId)) {
2453         client->CPS.prlist_len = 0;
2454         if (client->CPS.prlist_val && (client->ViceId != ANONYMOUSID))
2455             free(client->CPS.prlist_val);
2456         client->CPS.prlist_val = NULL;
2457         client->ViceId = viceid;
2458         client->expTime = expTime;
2459
2460         if (viceid == ANONYMOUSID) {
2461             client->CPS.prlist_len = AnonCPS.prlist_len;
2462             client->CPS.prlist_val = AnonCPS.prlist_val;
2463         } else {
2464             H_UNLOCK;
2465             code = hpr_GetCPS(viceid, &client->CPS);
2466             H_LOCK;
2467             if (code) {
2468                 char hoststr[16];
2469                 ViceLog(0,
2470                         ("pr_GetCPS failed(%d) for user %d, host %" AFS_PTR_FMT " (%s:%d)\n",
2471                          code, viceid, client->host,
2472                          afs_inet_ntoa_r(client->host->host,hoststr),
2473                          ntohs(client->host->port)));
2474
2475                 /* Although ubik_Call (called by pr_GetCPS) traverses thru
2476                  * all protection servers and reevaluates things if no
2477                  * sync server or quorum is found we could still end up
2478                  * with one of these errors. In such case we would like to
2479                  * reevaluate the rpc call to find if there's cps for this
2480                  * guy. We treat other errors (except network failures
2481                  * ones - i.e. code < 0) as an indication that there is no
2482                  * CPS for this host.  Ideally we could like to deal this
2483                  * problem the other way around (i.e.  if code == NOCPS
2484                  * ignore else retry next time) but the problem is that
2485                  * there're other errors (i.e.  EPERM) for which we don't
2486                  * want to retry and we don't know the whole code list!
2487                  */
2488                 if (code < 0 || code == UNOQUORUM || code == UNOTSYNC)
2489                     client->prfail = 1;
2490             }
2491         }
2492         /* the disabling of system:administrators is so iffy and has so many
2493          * possible failure modes that we will disable it again */
2494         /* Turn off System:Administrator for safety
2495          * if (AL_IsAMember(SystemId, client->CPS) == 0)
2496          * osi_Assert(AL_DisableGroup(SystemId, client->CPS) == 0); */
2497     }
2498
2499     /* Now, tcon may already be set to a rock, since we blocked with no host
2500      * or client locks set above in pr_GetCPS (XXXX some locking is probably
2501      * required).  So, before setting the RPC's rock, we should disconnect
2502      * the RPC from the other client structure's rock.
2503      */
2504     oldClient = (struct client *)rx_GetSpecific(tcon, rxcon_client_key);
2505     if (oldClient && oldClient != client
2506         && oldClient->sid == rx_GetConnectionId(tcon)
2507         && oldClient->VenusEpoch == rx_GetConnectionEpoch(tcon)
2508         && !(oldClient->host->hostFlags & HOSTDELETED)) {
2509         char hoststr[16];
2510         if (!oldClient->deleted) {
2511             /* if we didn't create it, it's not ours to put back */
2512             if (created) {
2513                 ViceLog(0, ("FindClient: stillborn client %p(%x); "
2514                             "conn %p (host %s:%d) had client %p(%x)\n",
2515                             client, client->sid, tcon,
2516                             afs_inet_ntoa_r(rxr_HostOf(tcon), hoststr),
2517                             ntohs(rxr_PortOf(tcon)),
2518                             oldClient, oldClient->sid));
2519                 if ((client->ViceId != ANONYMOUSID) && client->CPS.prlist_val)
2520                     free(client->CPS.prlist_val);
2521                 client->CPS.prlist_val = NULL;
2522                 client->CPS.prlist_len = 0;
2523             }
2524             /* We should perhaps check for 0 here */
2525             client->refCount--;
2526             ReleaseWriteLock(&client->lock);
2527             if (created) {
2528                 FreeCE(client);
2529                 created = 0;
2530             }
2531             oldClient->refCount++;
2532
2533             h_Hold_r(oldClient->host);
2534             h_Release_r(client->host);
2535
2536             H_UNLOCK;
2537             ObtainWriteLock(&oldClient->lock);
2538             H_LOCK;
2539             client = oldClient;
2540             host = oldClient->host;
2541         } else {
2542             ViceLog(0, ("FindClient: deleted client %p(%x ref %d host %p href "
2543                         "%d) already had conn %p (host %s:%d, cid %x), stolen "
2544                         "by client %p(%x, ref %d host %p href %d)\n",
2545                         oldClient, oldClient->sid, oldClient->refCount,
2546                         oldClient->host, oldClient->host->refCount, tcon,
2547                         afs_inet_ntoa_r(rxr_HostOf(tcon), hoststr),
2548                         ntohs(rxr_PortOf(tcon)), rx_GetConnectionId(tcon),
2549                         client, client->sid, client->refCount,
2550                         client->host, client->host->refCount));
2551             /* rx_SetSpecific will be done immediately below */
2552         }
2553     }
2554     /* Avoid chaining in more than once. */
2555     if (created) {
2556         h_Lock_r(host);
2557
2558         if (host->hostFlags & HOSTDELETED) {
2559             h_Unlock_r(host);
2560             h_Release_r(host);
2561
2562             host = NULL;
2563             client->host = NULL;
2564
2565             if ((client->ViceId != ANONYMOUSID) && client->CPS.prlist_val)
2566                 free(client->CPS.prlist_val);
2567             client->CPS.prlist_val = NULL;
2568             client->CPS.prlist_len = 0;
2569
2570             client->refCount--;
2571             ReleaseWriteLock(&client->lock);
2572             FreeCE(client);
2573             return NULL;
2574         }
2575
2576         client->next = host->FirstClient;
2577         host->FirstClient = client;
2578         h_Unlock_r(host);
2579         CurrentConnections++;   /* increment number of connections */
2580     }
2581     rx_SetSpecific(tcon, rxcon_client_key, client);
2582     ReleaseWriteLock(&client->lock);
2583
2584     return client;
2585
2586 }                               /*h_FindClient_r */
2587
2588 int
2589 h_ReleaseClient_r(struct client *client)
2590 {
2591     osi_Assert(client->refCount > 0);
2592     client->refCount--;
2593     return 0;
2594 }
2595
2596
2597 /*
2598  * Sigh:  this one is used to get the client AGAIN within the individual
2599  * server routines.  This does not bother h_Holding the host, since
2600  * this is assumed already have been done by the server main loop.
2601  * It does check tokens, since only the server routines can return the
2602  * VICETOKENDEAD error code
2603  */
2604 int
2605 GetClient(struct rx_connection *tcon, struct client **cp)
2606 {
2607     struct client *client;
2608     char hoststr[16];
2609
2610     H_LOCK;
2611     *cp = NULL;
2612     client = (struct client *)rx_GetSpecific(tcon, rxcon_client_key);
2613     if (client == NULL) {
2614         ViceLog(0,
2615                 ("GetClient: no client in conn %p (host %s:%d), VBUSYING\n",
2616                  tcon, afs_inet_ntoa_r(rxr_HostOf(tcon), hoststr),
2617                  ntohs(rxr_PortOf(tcon))));
2618         H_UNLOCK;
2619         return VBUSY;
2620     }
2621     if (rx_GetConnectionId(tcon) != client->sid
2622         || rx_GetConnectionEpoch(tcon) != client->VenusEpoch) {
2623         ViceLog(0,
2624                 ("GetClient: tcon %p tcon sid %d client sid %d\n",
2625                  tcon, rx_GetConnectionId(tcon), client->sid));
2626         H_UNLOCK;
2627         return VBUSY;
2628     }
2629     if (client && client->LastCall > client->expTime && client->expTime) {
2630         ViceLog(1,
2631                 ("Token for %s at %s:%d expired %d\n", h_UserName(client),
2632                  afs_inet_ntoa_r(client->host->host, hoststr),
2633                  ntohs(client->host->port), client->expTime));
2634         H_UNLOCK;
2635         return VICETOKENDEAD;
2636     }
2637     if (client->deleted) {
2638         ViceLog(0, ("GetClient: got deleted client, connection will appear "
2639                     "anonymous; tcon %p cid %x client %p ref %d host %p "
2640                     "(%s:%d) href %d ViceId %d\n",
2641                     tcon, rx_GetConnectionId(tcon), client, client->refCount,
2642                     client->host,
2643                     afs_inet_ntoa_r(client->host->host, hoststr),
2644                     (int)ntohs(client->host->port), client->host->refCount,
2645                     (int)client->ViceId));
2646     }
2647
2648     client->refCount++;
2649     *cp = client;
2650     H_UNLOCK;
2651     return 0;
2652 }                               /*GetClient */
2653
2654 int
2655 PutClient(struct client **cp)
2656 {
2657     if (*cp == NULL)
2658         return -1;
2659
2660     H_LOCK;
2661     h_ReleaseClient_r(*cp);
2662     *cp = NULL;
2663     H_UNLOCK;
2664     return 0;
2665 }                               /*PutClient */
2666
2667
2668 /* Client user name for short term use.  Note that this is NOT inexpensive */
2669 char *
2670 h_UserName(struct client *client)
2671 {
2672     static char User[PR_MAXNAMELEN + 1];
2673     namelist lnames;
2674     idlist lids;
2675
2676     lids.idlist_len = 1;
2677     lids.idlist_val = malloc(1 * sizeof(afs_int32));
2678     if (!lids.idlist_val) {
2679         ViceLogThenPanic(0, ("Failed malloc in h_UserName\n"));
2680     }
2681     lnames.namelist_len = 0;
2682     lnames.namelist_val = (prname *) 0;
2683     lids.idlist_val[0] = client->ViceId;
2684     if (hpr_IdToName(&lids, &lnames)) {
2685         /* We need to free id we alloced above! */
2686         free(lids.idlist_val);
2687         return "*UNKNOWN USER NAME*";
2688     }
2689     strncpy(User, lnames.namelist_val[0], PR_MAXNAMELEN);
2690     free(lids.idlist_val);
2691     free(lnames.namelist_val);
2692     return User;
2693 }                               /*h_UserName */
2694
2695
2696 void
2697 h_PrintStats(void)
2698 {
2699     ViceLog(0,
2700             ("Total Client entries = %d, blocks = %d; Host entries = %d, blocks = %d\n",
2701              CEs, CEBlocks, HTs, HTBlocks));
2702
2703 }                               /*h_PrintStats */
2704
2705
2706 static int
2707 h_PrintClient(struct host *host, void *rock)
2708 {
2709     StreamHandle_t *file = (StreamHandle_t *)rock;
2710     struct client *client;
2711     int i;
2712     char tmpStr[256];
2713     char tbuffer[32];
2714     char hoststr[16];
2715     time_t LastCall, expTime;
2716     struct tm tm;
2717
2718     H_LOCK;
2719     LastCall = host->LastCall;
2720     if (host->hostFlags & HOSTDELETED) {
2721         H_UNLOCK;
2722         return 0;
2723     }
2724     strftime(tbuffer, sizeof(tbuffer), "%a %b %d %H:%M:%S %Y",
2725              localtime_r(&LastCall, &tm));
2726     snprintf(tmpStr, sizeof tmpStr, "Host %s:%d down = %d, LastCall %s\n",
2727              afs_inet_ntoa_r(host->host, hoststr),
2728              ntohs(host->port), (host->hostFlags & VENUSDOWN),
2729              tbuffer);
2730     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
2731     for (client = host->FirstClient; client; client = client->next) {
2732         if (!client->deleted) {
2733             expTime = client->expTime;
2734             strftime(tbuffer, sizeof(tbuffer), "%a %b %d %H:%M:%S %Y",
2735                      localtime_r(&expTime, &tm));
2736             snprintf(tmpStr, sizeof tmpStr,
2737                      "    user id=%d,  name=%s, sl=%s till %s\n",
2738                      client->ViceId, h_UserName(client),
2739                      client->authClass ? "Authenticated"
2740                                        : "Not authenticated",
2741                      client->authClass ? tbuffer : "No Limit");
2742             (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
2743             snprintf(tmpStr, sizeof tmpStr, "      CPS-%d is [",
2744                          client->CPS.prlist_len);
2745             (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
2746             if (client->CPS.prlist_val) {
2747                 for (i = 0; i < client->CPS.prlist_len; i++) {
2748                     snprintf(tmpStr, sizeof tmpStr, " %d",
2749                              client->CPS.prlist_val[i]);
2750                     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
2751                 }
2752             }
2753             sprintf(tmpStr, "]\n");
2754             (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
2755         }
2756     }
2757     H_UNLOCK;
2758     return 0;
2759
2760 }                               /*h_PrintClient */
2761
2762
2763
2764 /*
2765  * Print a list of clients, with last security level and token value seen,
2766  * if known
2767  */
2768 void
2769 h_PrintClients(void)
2770 {
2771     time_t now;
2772     char tmpStr[256];
2773     char tbuffer[32];
2774     struct tm tm;
2775
2776     StreamHandle_t *file = STREAM_OPEN(AFSDIR_SERVER_CLNTDUMP_FILEPATH, "w");
2777
2778     if (file == NULL) {
2779         ViceLog(0,
2780                 ("Couldn't create client dump file %s\n",
2781                  AFSDIR_SERVER_CLNTDUMP_FILEPATH));
2782         return;
2783     }
2784     now = FT_ApproxTime();
2785     strftime(tbuffer, sizeof(tbuffer), "%a %b %d %H:%M:%S %Y",
2786              localtime_r(&now, &tm));
2787     snprintf(tmpStr, sizeof tmpStr, "List of active users at %s\n\n",
2788              tbuffer);
2789     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
2790     h_Enumerate(h_PrintClient, (char *)file);
2791     STREAM_REALLYCLOSE(file);
2792     ViceLog(0, ("Created client dump %s\n", AFSDIR_SERVER_CLNTDUMP_FILEPATH));
2793 }
2794
2795
2796
2797
2798 static int
2799 h_DumpHost(struct host *host, void *rock)
2800 {
2801     StreamHandle_t *file = (StreamHandle_t *)rock;
2802
2803     int i;
2804     char tmpStr[256];
2805     char hoststr[16];
2806
2807     H_LOCK;
2808     snprintf(tmpStr, sizeof tmpStr,
2809              "ip:%s port:%d hidx:%d cbid:%d lock:%x last:%u active:%u "
2810              "down:%d del:%d cons:%d cldel:%d\n\t hpfailed:%d hcpsCall:%u "
2811              "hcps [",
2812              afs_inet_ntoa_r(host->host, hoststr), ntohs(host->port),
2813              host->index, host->cblist, CheckLock(&host->lock),
2814              host->LastCall, host->ActiveCall, (host->hostFlags & VENUSDOWN),
2815              host->hostFlags & HOSTDELETED, host->Console,
2816              host->hostFlags & CLIENTDELETED, host->hcpsfailed,
2817              host->cpsCall);
2818     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
2819     if (host->hcps.prlist_val)
2820         for (i = 0; i < host->hcps.prlist_len; i++) {
2821             snprintf(tmpStr, sizeof tmpStr, " %d", host->hcps.prlist_val[i]);
2822             (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
2823         }
2824     sprintf(tmpStr, "] [");
2825     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
2826     if (host->interface)
2827         for (i = 0; i < host->interface->numberOfInterfaces; i++) {
2828             char hoststr[16];
2829             sprintf(tmpStr, " %s:%d",
2830                      afs_inet_ntoa_r(host->interface->interface[i].addr, hoststr),
2831                      ntohs(host->interface->interface[i].port));
2832             (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
2833         }
2834     sprintf(tmpStr, "] refCount:%d hostFlags:%hu\n", host->refCount, host->hostFlags);
2835     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
2836
2837     H_UNLOCK;
2838     return 0;
2839
2840 }                               /*h_DumpHost */
2841
2842
2843 void
2844 h_DumpHosts(void)
2845 {
2846     time_t now;
2847     StreamHandle_t *file = STREAM_OPEN(AFSDIR_SERVER_HOSTDUMP_FILEPATH, "w");
2848     char tmpStr[256];
2849     char tbuffer[32];
2850     struct tm tm;
2851
2852     if (file == NULL) {
2853         ViceLog(0,
2854                 ("Couldn't create host dump file %s\n",
2855                  AFSDIR_SERVER_HOSTDUMP_FILEPATH));
2856         return;
2857     }
2858     now = FT_ApproxTime();
2859     strftime(tbuffer, sizeof(tbuffer), "%a %b %d %H:%M:%S %Y",
2860              localtime_r(&now, &tm));
2861     snprintf(tmpStr, sizeof tmpStr, "List of active hosts at %s\n\n", tbuffer);
2862     (void)STREAM_WRITE(tmpStr, strlen(tmpStr), 1, file);
2863     h_Enumerate(h_DumpHost, (char *)file);
2864     STREAM_REALLYCLOSE(file);
2865     ViceLog(0, ("Created host dump %s\n", AFSDIR_SERVER_HOSTDUMP_FILEPATH));
2866
2867 }                               /*h_DumpHosts */
2868
2869 #ifdef AFS_DEMAND_ATTACH_FS
2870 /*
2871  * demand attach fs
2872  * host state serialization
2873  */
2874 static int h_stateFillHeader(struct host_state_header * hdr);
2875 static int h_stateCheckHeader(struct host_state_header * hdr);
2876 static int h_stateAllocMap(struct fs_dump_state * state);
2877 static int h_stateSaveHost(struct host * host, void *rock);
2878 static int h_stateRestoreHost(struct fs_dump_state * state);
2879 static int h_stateRestoreIndex(struct host * h, void *rock);
2880 static int h_stateVerifyHost(struct host * h, void *rock);
2881 static int h_stateVerifyAddrHash(struct fs_dump_state * state, struct host * h,
2882                                  afs_uint32 addr, afs_uint16 port, int valid);
2883 static int h_stateVerifyUuidHash(struct fs_dump_state * state, struct host * h);
2884 static void h_hostToDiskEntry_r(struct host * in, struct hostDiskEntry * out);
2885 static void h_diskEntryToHost_r(struct hostDiskEntry * in, struct host * out);
2886
2887 /**
2888  * Is this host busy?
2889  *
2890  * This is just a hint and should not be trusted; this should probably only be
2891  * used by the host state serialization code when trying to detect if a host
2892  * can be sanely serialized to disk or not. If this function returns 1, the
2893  * host may be in an invalid state and thus should not be saved to disk.
2894  */
2895 static int
2896 h_isBusy_r(struct host *host)
2897 {
2898     struct Lock *hostLock = &host->lock;
2899     int locked = 0;
2900
2901     LOCK_LOCK(hostLock);
2902     if (hostLock->excl_locked || hostLock->readers_reading) {
2903         locked = 1;
2904     }
2905     LOCK_UNLOCK(hostLock);
2906
2907     if (locked) {
2908         return 1;
2909     }
2910
2911     if ((host->hostFlags & HWHO_INPROGRESS) || !(host->hostFlags & ALTADDR)) {
2912         /* We shouldn't hit this if the host wasn't locked, but just in case... */
2913         return 1;
2914     }
2915
2916     return 0;
2917 }
2918
2919 /* this procedure saves all host state to disk for fast startup */
2920 int
2921 h_stateSave(struct fs_dump_state * state)
2922 {
2923     AssignInt64(state->eof_offset, &state->hdr->h_offset);
2924
2925     /* XXX debug */
2926     ViceLog(0, ("h_stateSave:  hostCount=%d\n", hostCount));
2927
2928     /* invalidate host state header */
2929     memset(state->h_hdr, 0, sizeof(struct host_state_header));
2930
2931     if (fs_stateWriteHeader(state, &state->hdr->h_offset, state->h_hdr,
2932                             sizeof(struct host_state_header))) {
2933         state->bail = 1;
2934         goto done;
2935     }
2936
2937     fs_stateIncEOF(state, sizeof(struct host_state_header));
2938
2939     h_Enumerate_r(h_stateSaveHost, hostList, (char *)state);
2940     if (state->bail) {
2941         goto done;
2942     }
2943
2944     h_stateFillHeader(state->h_hdr);
2945
2946     /* write the real header to disk */
2947     state->bail = fs_stateWriteHeader(state, &state->hdr->h_offset, state->h_hdr,
2948                                       sizeof(struct host_state_header));
2949
2950  done:
2951     return state->bail;
2952 }
2953
2954 /* demand attach fs
2955  * host state serialization
2956  *
2957  * this procedure restores all host state from a disk for fast startup
2958  */
2959 int
2960 h_stateRestore(struct fs_dump_state * state)
2961 {
2962     int i, records;
2963
2964     /* seek to the right position and read in the host state header */
2965     if (fs_stateReadHeader(state, &state->hdr->h_offset, state->h_hdr,
2966                            sizeof(struct host_state_header))) {
2967         state->bail = 1;
2968         goto done;
2969     }
2970
2971     /* check the validity of the header */
2972     if (h_stateCheckHeader(state->h_hdr)) {
2973         state->bail = 1;
2974         goto done;
2975     }
2976
2977     records = state->h_hdr->records;
2978
2979     if (h_stateAllocMap(state)) {
2980         state->bail = 1;
2981         goto done;
2982     }
2983
2984     /* iterate over records restoring host state */
2985     for (i=0; i < records; i++) {
2986         if (h_stateRestoreHost(state) != 0) {
2987             state->bail = 1;
2988             break;
2989         }
2990     }
2991
2992  done:
2993     return state->bail;
2994 }
2995
2996 int
2997 h_stateRestoreIndices(struct fs_dump_state * state)
2998 {
2999     h_Enumerate_r(h_stateRestoreIndex, hostList, (char *)state);
3000     return state->bail;
3001 }
3002
3003 static int
3004 h_stateRestoreIndex(struct host * h, void *rock)
3005 {
3006     struct fs_dump_state *state = (struct fs_dump_state *)rock;
3007     if (cb_OldToNew(state, h->cblist, &h->cblist)) {
3008         return H_ENUMERATE_BAIL(0);
3009     }
3010     return 0;
3011 }
3012
3013 int
3014 h_stateVerify(struct fs_dump_state * state)
3015 {
3016     h_Enumerate_r(h_stateVerifyHost, hostList, (char *)state);
3017     return state->bail;
3018 }
3019
3020 static int
3021 h_stateVerifyHost(struct host * h, void* rock)
3022 {
3023     struct fs_dump_state *state = (struct fs_dump_state *)rock;
3024     int i;
3025
3026     if (h == NULL) {
3027         ViceLog(0, ("h_stateVerifyHost: error: NULL host pointer in linked list\n"));
3028         return H_ENUMERATE_BAIL(0);
3029     }
3030
3031     if (h->interface) {
3032         for (i = h->interface->numberOfInterfaces-1; i >= 0; i--) {
3033             if (h_stateVerifyAddrHash(state, h, h->interface->interface[i].addr,
3034                                       h->interface->interface[i].port,
3035                                       h->interface->interface[i].valid)) {
3036                 state->bail = 1;
3037             }
3038         }
3039         if (h_stateVerifyUuidHash(state, h)) {
3040             state->bail = 1;
3041         }
3042     } else if (h_stateVerifyAddrHash(state, h, h->host, h->port, 1)) {
3043         state->bail = 1;
3044     }
3045
3046     if (cb_stateVerifyHCBList(state, h)) {
3047         state->bail = 1;
3048     }
3049
3050     return 0;
3051 }
3052
3053 /**
3054  * verify a host is either in, or absent from, the addr hash table.
3055  *
3056  * @param[in] state  fs dump state
3057  * @param[in] h      host we're dealing with
3058  * @param[in] addr   addr to look for (NBO)
3059  * @param[in] port   port to look for (NBO)
3060  * @param[in] valid  1 if we're verifying that the specified addr and port
3061  *                   in the hash table point to the specified host. 0 if we're
3062  *                   verifying that the specified addr and port do NOT point
3063  *                   to the specified host
3064  *
3065  * @return operation status
3066  *  @retval 1 failed to verify, bail out
3067  *  @retval 0 verified successfully, all is well
3068  */
3069 static int
3070 h_stateVerifyAddrHash(struct fs_dump_state * state, struct host * h,
3071                       afs_uint32 addr, afs_uint16 port, int valid)
3072 {
3073     int ret = 0, found = 0;
3074     struct host *host = NULL;
3075     struct h_AddrHashChain *chain;
3076     int index = h_HashIndex(addr);
3077     char tmp[16];
3078     int chain_len = 0;
3079
3080     for (chain = hostAddrHashTable[index]; chain; chain = chain->next) {
3081         host = chain->hostPtr;
3082         if (host == NULL) {
3083             afs_inet_ntoa_r(addr, tmp);
3084             ViceLog(0, ("h_stateVerifyAddrHash: error: addr hash chain has NULL host ptr (lookup addr %s)\n", tmp));
3085             ret = 1;
3086             goto done;
3087         }
3088         if ((chain->addr == addr) && (chain->port == port)) {
3089             if (host != h) {
3090                 if (valid) {
3091                     ViceLog(0, ("h_stateVerifyAddrHash: warning: addr hash entry "
3092                                 "points to different host struct (%d, %d)\n",
3093                                 h->index, host->index));
3094                     state->flags.warnings_generated = 1;
3095                 }
3096             } else {
3097                 if (!valid) {
3098                     ViceLog(0, ("h_stateVerifyAddrHash: error: addr %s:%u is "
3099                                 "marked invalid, but points to the containing "
3100                                 "host\n", afs_inet_ntoa_r(addr, tmp),
3101                                 (unsigned)htons(port)));
3102                     ret = 1;
3103                     goto done;
3104                 }
3105             }
3106             found = 1;
3107             break;
3108         }
3109         if (chain_len > FS_STATE_H_MAX_ADDR_HASH_CHAIN_LEN) {
3110             ViceLog(0, ("h_stateVerifyAddrHash: error: hash chain length exceeds %d; assuming there's a loop\n",
3111                         FS_STATE_H_MAX_ADDR_HASH_CHAIN_LEN));
3112             ret = 1;
3113             goto done;
3114         }
3115         chain_len++;
3116     }
3117
3118     if (!found && valid) {
3119         afs_inet_ntoa_r(addr, tmp);
3120         if (state->mode == FS_STATE_LOAD_MODE) {
3121             ViceLog(0, ("h_stateVerifyAddrHash: error: addr %s:%u not found in hash\n",
3122                         tmp, (unsigned)htons(port)));
3123             ret = 1;
3124             goto done;
3125         } else {
3126             ViceLog(0, ("h_stateVerifyAddrHash: warning: addr %s:%u not found in hash\n",
3127                         tmp, (unsigned)htons(port)));
3128             state->flags.warnings_generated = 1;
3129         }
3130     }
3131
3132  done:
3133     return ret;
3134 }
3135
3136 static int
3137 h_stateVerifyUuidHash(struct fs_dump_state * state, struct host * h)
3138 {
3139     int ret = 0, found = 0;
3140     struct host *host = NULL;
3141     struct h_UuidHashChain *chain;
3142     afsUUID * uuidp = &h->interface->uuid;
3143     int index = h_UuidHashIndex(uuidp);
3144     char tmp[40];
3145     int chain_len = 0;
3146
3147     for (chain = hostUuidHashTable[index]; chain; chain = chain->next) {
3148         host = chain->hostPtr;
3149         if (host == NULL) {
3150             afsUUID_to_string(uuidp, tmp, sizeof(tmp));
3151             ViceLog(0, ("h_stateVerifyUuidHash: error: uuid hash chain has NULL host ptr (lookup uuid %s)\n", tmp));
3152             ret = 1;
3153             goto done;
3154         }
3155         if (host->interface &&
3156             afs_uuid_equal(&host->interface->uuid, uuidp)) {
3157             if (host != h) {
3158                 ViceLog(0, ("h_stateVerifyUuidHash: warning: uuid hash entry points to different host struct (%d, %d)\n",
3159                             h->index, host->index));
3160                 state->flags.warnings_generated = 1;
3161             }
3162             found = 1;
3163             goto done;
3164         }
3165         if (chain_len > FS_STATE_H_MAX_UUID_HASH_CHAIN_LEN) {
3166             ViceLog(0, ("h_stateVerifyUuidHash: error: hash chain length exceeds %d; assuming there's a loop\n",
3167                         FS_STATE_H_MAX_UUID_HASH_CHAIN_LEN));
3168             ret = 1;
3169             goto done;
3170         }
3171         chain_len++;
3172     }
3173
3174     if (!found) {
3175         afsUUID_to_string(uuidp, tmp, sizeof(tmp));
3176         if (state->mode == FS_STATE_LOAD_MODE) {
3177             ViceLog(0, ("h_stateVerifyUuidHash: error: uuid %s not found in hash\n", tmp));
3178             ret = 1;
3179             goto done;
3180         } else {
3181             ViceLog(0, ("h_stateVerifyUuidHash: warning: uuid %s not found in hash\n", tmp));
3182             state->flags.warnings_generated = 1;
3183         }
3184     }
3185
3186  done:
3187     return ret;
3188 }
3189
3190 /* create the host state header structure */
3191 static int
3192 h_stateFillHeader(struct host_state_header * hdr)
3193 {
3194     hdr->stamp.magic = HOST_STATE_MAGIC;
3195     hdr->stamp.version = HOST_STATE_VERSION;
3196     return 0;
3197 }
3198
3199 /* check the contents of the host state header structure */
3200 static int
3201 h_stateCheckHeader(struct host_state_header * hdr)
3202 {
3203     int ret=0;
3204
3205     if (hdr->stamp.magic != HOST_STATE_MAGIC) {
3206         ViceLog(0, ("check_host_state_header: invalid state header\n"));
3207         ret = 1;
3208     }
3209     else if (hdr->stamp.version != HOST_STATE_VERSION) {
3210         ViceLog(0, ("check_host_state_header: unknown version number\n"));
3211         ret = 1;
3212     }
3213     return ret;
3214 }
3215
3216 /* allocate the host id mapping table */
3217 static int
3218 h_stateAllocMap(struct fs_dump_state * state)
3219 {
3220     state->h_map.len = state->h_hdr->index_max + 1;
3221     state->h_map.entries = (struct idx_map_entry_t *)
3222         calloc(state->h_map.len, sizeof(struct idx_map_entry_t));
3223     return (state->h_map.entries != NULL) ? 0 : 1;
3224 }
3225
3226 /* function called by h_Enumerate to save a host to disk */
3227 static int
3228 h_stateSaveHost(struct host * host, void* rock)
3229 {
3230     struct fs_dump_state *state = (struct fs_dump_state *) rock;
3231     int if_len=0, hcps_len=0;
3232     struct hostDiskEntry hdsk;
3233     struct host_state_entry_header hdr;
3234     struct Interface * ifp = NULL;
3235     afs_int32 * hcps = NULL;
3236     struct iovec iov[4];
3237     int iovcnt = 2;
3238
3239     if (h_isBusy_r(host)) {
3240         char hoststr[16];
3241         ViceLog(1, ("Not saving host %s:%d to disk; host appears busy\n",
3242                     afs_inet_ntoa_r(host->host, hoststr), (int)ntohs(host->port)));
3243         /* Make sure we don't try to save callbacks to disk for this host, or
3244          * we'll get confused on restore */
3245         DeleteAllCallBacks_r(host, 1);
3246         return 0;
3247     }
3248
3249     memset(&hdr, 0, sizeof(hdr));
3250
3251     if (state->h_hdr->index_max < host->index) {
3252         state->h_hdr->index_max = host->index;
3253     }
3254
3255     h_hostToDiskEntry_r(host, &hdsk);
3256     if (host->interface) {
3257         if_len = sizeof(struct Interface) +
3258             ((host->interface->numberOfInterfaces-1) * sizeof(struct AddrPort));
3259         ifp = malloc(if_len);
3260         osi_Assert(ifp != NULL);
3261         memcpy(ifp, host->interface, if_len);
3262         hdr.interfaces = host->interface->numberOfInterfaces;
3263         iov[iovcnt].iov_base = (char *) ifp;
3264         iov[iovcnt].iov_len = if_len;
3265         iovcnt++;
3266     }
3267     if (host->hcps.prlist_val) {
3268         hdr.hcps = host->hcps.prlist_len;
3269         hcps_len = hdr.hcps * sizeof(afs_int32);
3270         hcps = malloc(hcps_len);
3271         osi_Assert(hcps != NULL);
3272         memcpy(hcps, host->hcps.prlist_val, hcps_len);
3273         iov[iovcnt].iov_base = (char *) hcps;
3274         iov[iovcnt].iov_len = hcps_len;
3275         iovcnt++;
3276     }
3277
3278     if (hdsk.index > state->h_hdr->index_max)
3279         state->h_hdr->index_max = hdsk.index;
3280
3281     hdr.len = sizeof(struct host_state_entry_header) +
3282         sizeof(struct hostDiskEntry) + if_len + hcps_len;
3283     hdr.magic = HOST_STATE_ENTRY_MAGIC;
3284
3285     iov[0].iov_base = (char *) &hdr;
3286     iov[0].iov_len = sizeof(hdr);
3287     iov[1].iov_base = (char *) &hdsk;
3288     iov[1].iov_len = sizeof(struct hostDiskEntry);
3289
3290     if (fs_stateWriteV(state, iov, iovcnt)) {
3291         ViceLog(0, ("h_stateSaveHost: failed to save host %d", host->index));
3292         state->bail = 1;
3293     }
3294
3295     fs_stateIncEOF(state, hdr.len);
3296
3297     state->h_hdr->records++;
3298
3299     if (ifp)
3300         free(ifp);
3301     if (hcps)
3302         free(hcps);
3303     if (state->bail) {
3304         return H_ENUMERATE_BAIL(0);
3305     }
3306     return 0;
3307 }
3308
3309 /* restores a host from disk */
3310 static int
3311 h_stateRestoreHost(struct fs_dump_state * state)
3312 {
3313     int ifp_len=0, hcps_len=0, bail=0;
3314     struct host_state_entry_header hdr;
3315     struct hostDiskEntry hdsk;
3316     struct host *host = NULL;
3317     struct Interface *ifp = NULL;
3318     afs_int32 * hcps = NULL;
3319     struct iovec iov[3];
3320     int iovcnt = 1;
3321
3322     if (fs_stateRead(state, &hdr, sizeof(hdr))) {
3323         ViceLog(0, ("h_stateRestoreHost: failed to read host entry header from dump file '%s'\n",
3324                     state->fn));
3325         bail = 1;
3326         goto done;
3327     }
3328
3329     if (hdr.magic != HOST_STATE_ENTRY_MAGIC) {
3330         ViceLog(0, ("h_stateRestoreHost: fileserver state dump file '%s' is corrupt.\n",
3331                     state->fn));
3332         bail = 1;
3333         goto done;
3334     }
3335
3336     iov[0].iov_base = (char *) &hdsk;
3337     iov[0].iov_len = sizeof(struct hostDiskEntry);
3338
3339     if (hdr.interfaces) {
3340         ifp_len = sizeof(struct Interface) +
3341             ((hdr.interfaces-1) * sizeof(struct AddrPort));
3342         ifp = malloc(ifp_len);
3343         osi_Assert(ifp != NULL);
3344         iov[iovcnt].iov_base = (char *) ifp;
3345         iov[iovcnt].iov_len = ifp_len;
3346         iovcnt++;
3347     }
3348     if (hdr.hcps) {
3349         hcps_len = hdr.hcps * sizeof(afs_int32);
3350         hcps = malloc(hcps_len);
3351         osi_Assert(hcps != NULL);
3352         iov[iovcnt].iov_base = (char *) hcps;
3353         iov[iovcnt].iov_len = hcps_len;
3354         iovcnt++;
3355     }
3356
3357     if ((ifp_len + hcps_len + sizeof(hdsk) + sizeof(hdr)) != hdr.len) {
3358         ViceLog(0, ("h_stateRestoreHost: host entry header length fields are inconsistent\n"));
3359         bail = 1;
3360         goto done;
3361     }
3362
3363     if (fs_stateReadV(state, iov, iovcnt)) {
3364         ViceLog(0, ("h_stateRestoreHost: failed to read host entry\n"));
3365         bail = 1;
3366         goto done;
3367     }
3368
3369     if (!hdr.hcps && hdsk.hcps_valid) {
3370         /* valid, zero-length host cps ; does this ever happen? */
3371         hcps = malloc(sizeof(afs_int32));
3372         osi_Assert(hcps != NULL);
3373     }
3374
3375     if ((hdsk.hostFlags & HWHO_INPROGRESS) || !(hdsk.hostFlags & ALTADDR)) {
3376         char hoststr[16];
3377         ViceLog(0, ("h_stateRestoreHost: skipping host %s:%d due to invalid flags 0x%x\n",
3378                     afs_inet_ntoa_r(hdsk.host, hoststr), (int)ntohs(hdsk.port),
3379                     (unsigned)hdsk.hostFlags));
3380         bail = 0;
3381         state->h_map.entries[hdsk.index].valid = FS_STATE_IDX_SKIPPED;
3382         goto done;
3383     }
3384
3385     /* for restoring state, we better be able to get a host! */
3386     host = GetHT();
3387     osi_Assert(host != NULL);
3388
3389     if (ifp) {
3390         host->interface = ifp;
3391     }
3392     if (hcps) {
3393         host->hcps.prlist_val = hcps;
3394         host->hcps.prlist_len = hdr.hcps;
3395     }
3396
3397     h_diskEntryToHost_r(&hdsk, host);
3398     h_SetupCallbackConn_r(host);
3399
3400     h_AddHostToAddrHashTable_r(host->host, host->port, host);
3401     if (ifp) {
3402         int i;
3403         for (i = ifp->numberOfInterfaces-1; i >= 0; i--) {
3404             if (ifp->interface[i].valid &&
3405                 !(ifp->interface[i].addr == host->host &&
3406                   ifp->interface[i].port == host->port)) {
3407                 h_AddHostToAddrHashTable_r(ifp->interface[i].addr,
3408                                            ifp->interface[i].port,
3409                                            host);
3410             }
3411         }
3412         h_AddHostToUuidHashTable_r(&ifp->uuid, host);
3413     }
3414     h_InsertList_r(host);
3415
3416     /* setup host id map entry */
3417     state->h_map.entries[hdsk.index].valid = FS_STATE_IDX_VALID;
3418     state->h_map.entries[hdsk.index].old_idx = hdsk.index;
3419     state->h_map.entries[hdsk.index].new_idx = host->index;
3420
3421  done:
3422     if (bail) {
3423         if (ifp)
3424             free(ifp);
3425         if (hcps)
3426             free(hcps);
3427     }
3428     return bail;
3429 }
3430
3431 /* serialize a host structure to disk */
3432 static void
3433 h_hostToDiskEntry_r(struct host * in, struct hostDiskEntry * out)
3434 {
3435     out->host = in->host;
3436     out->port = in->port;
3437     out->hostFlags = in->hostFlags;
3438     out->Console = in->Console;
3439     out->hcpsfailed = in->hcpsfailed;
3440     out->LastCall = in->LastCall;
3441     out->ActiveCall = in->ActiveCall;
3442     out->cpsCall = in->cpsCall;
3443     out->cblist = in->cblist;
3444     out->InSameNetwork = in->InSameNetwork;
3445
3446     /* special fields we save, but are not memcpy'd back on restore */
3447     out->index = in->index;
3448     out->hcps_len = in->hcps.prlist_len;
3449     out->hcps_valid = (in->hcps.prlist_val == NULL) ? 0 : 1;
3450 }
3451
3452 /* restore a host structure from disk */
3453 static void
3454 h_diskEntryToHost_r(struct hostDiskEntry * in, struct host * out)
3455 {
3456     out->host = in->host;
3457     out->port = in->port;
3458     out->hostFlags = in->hostFlags;
3459     out->Console = in->Console;
3460     out->hcpsfailed = in->hcpsfailed;
3461     out->LastCall = in->LastCall;
3462     out->ActiveCall = in->ActiveCall;
3463     out->cpsCall = in->cpsCall;
3464     out->cblist = in->cblist;
3465     out->InSameNetwork = in->InSameNetwork;
3466 }
3467
3468 /* index translation routines */
3469 int
3470 h_OldToNew(struct fs_dump_state * state, afs_uint32 old, afs_uint32 * new)
3471 {
3472     int ret = 0;
3473
3474     /* hosts use a zero-based index, so old==0 is valid */
3475
3476     if (old >= state->h_map.len) {
3477         ViceLog(0, ("h_OldToNew: index %d is out of range\n", old));
3478         ret = 1;
3479     } else if (state->h_map.entries[old].valid != FS_STATE_IDX_VALID ||
3480                state->h_map.entries[old].old_idx != old) { /* sanity check */
3481         ViceLog(0, ("h_OldToNew: index %d points to an invalid host record\n", old));
3482         ret = 1;
3483     } else {
3484         *new = state->h_map.entries[old].new_idx;
3485     }
3486
3487     return ret;
3488 }
3489 #endif /* AFS_DEMAND_ATTACH_FS */
3490
3491
3492 /*
3493  * This counts the number of workstations, the number of active workstations,
3494  * and the number of workstations declared "down" (i.e. not heard from
3495  * recently).  An active workstation has received a call since the cutoff
3496  * time argument passed.
3497  */
3498 void
3499 h_GetWorkStats(int *nump, int *activep, int *delp, afs_int32 cutofftime)
3500 {
3501     struct host *host;
3502     int num = 0, active = 0, del = 0;
3503     int count;
3504
3505     H_LOCK;
3506     for (count = 0, host = hostList; host && count < hostCount; host = host->next, count++) {
3507         if (!(host->hostFlags & HOSTDELETED)) {
3508             num++;
3509             if (host->ActiveCall > cutofftime)
3510                 active++;
3511             if (host->hostFlags & VENUSDOWN)
3512                 del++;
3513         }
3514     }
3515     if (count != hostCount) {
3516         ViceLog(0, ("h_GetWorkStats found %d of %d hosts\n", count, hostCount));
3517     } else if (host != NULL) {
3518         ViceLog(0, ("h_GetWorkStats found more than %d hosts\n", hostCount));
3519         ShutDownAndCore(PANIC);
3520     }
3521     H_UNLOCK;
3522     if (nump)
3523         *nump = num;
3524     if (activep)
3525         *activep = active;
3526     if (delp)
3527         *delp = del;
3528
3529 }                               /*h_GetWorkStats */
3530
3531 void
3532 h_GetWorkStats64(afs_uint64 *nump, afs_uint64 *activep, afs_uint64 *delp, 
3533                  afs_int32 cutofftime)
3534 {
3535     int num, active, del;
3536     h_GetWorkStats(&num, &active, &del, cutofftime);
3537     if (nump)
3538         *nump = num;
3539     if (activep)
3540         *activep = active;
3541     if (delp)
3542         *delp = del;
3543 }
3544
3545 /*------------------------------------------------------------------------
3546  * PRIVATE h_ClassifyAddress
3547  *
3548  * Description:
3549  *      Given a target IP address and a candidate IP address (both
3550  *      in host byte order), classify the candidate into one of three
3551  *      buckets in relation to the target by bumping the counters passed
3552  *      in as parameters.
3553  *
3554  * Arguments:
3555  *      a_targetAddr       : Target address.
3556  *      a_candAddr         : Candidate address.
3557  *      a_sameNetOrSubnetP : Ptr to counter to bump when the two
3558  *                           addresses are either in the same network
3559  *                           or the same subnet.
3560  *      a_diffSubnetP      : ...when the candidate is in a different
3561  *                           subnet.
3562  *      a_diffNetworkP     : ...when the candidate is in a different
3563  *                           network.
3564  *
3565  * Returns:
3566  *      Nothing.
3567  *
3568  * Environment:
3569  *      The target and candidate addresses are both in host byte
3570  *      order, NOT network byte order, when passed in.
3571  *
3572  * Side Effects:
3573  *      As advertised.
3574  *------------------------------------------------------------------------*/
3575
3576 static void
3577 h_ClassifyAddress(afs_uint32 a_targetAddr, afs_uint32 a_candAddr,
3578                   afs_int32 * a_sameNetOrSubnetP, afs_int32 * a_diffSubnetP,
3579                   afs_int32 * a_diffNetworkP)
3580 {                               /*h_ClassifyAddress */
3581
3582     afs_uint32 targetNet;
3583     afs_uint32 targetSubnet;
3584     afs_uint32 candNet;
3585     afs_uint32 candSubnet;
3586
3587     /*
3588      * Put bad values into the subnet info to start with.
3589      */
3590     targetSubnet = (afs_uint32) 0;
3591     candSubnet = (afs_uint32) 0;
3592
3593     /*
3594      * Pull out the network and subnetwork numbers from the target
3595      * and candidate addresses.  We can short-circuit this whole
3596      * affair if the target and candidate addresses are not of the
3597      * same class.
3598      */
3599     if (IN_CLASSA(a_targetAddr)) {
3600         if (!(IN_CLASSA(a_candAddr))) {
3601             (*a_diffNetworkP)++;
3602             return;
3603         }
3604         targetNet = a_targetAddr & IN_CLASSA_NET;
3605         candNet = a_candAddr & IN_CLASSA_NET;
3606         if (IN_SUBNETA(a_targetAddr))
3607             targetSubnet = a_targetAddr & IN_CLASSA_SUBNET;
3608         if (IN_SUBNETA(a_candAddr))
3609             candSubnet = a_candAddr & IN_CLASSA_SUBNET;
3610     } else if (IN_CLASSB(a_targetAddr)) {
3611         if (!(IN_CLASSB(a_candAddr))) {
3612             (*a_diffNetworkP)++;
3613             return;
3614         }
3615         targetNet = a_targetAddr & IN_CLASSB_NET;
3616         candNet = a_candAddr & IN_CLASSB_NET;
3617         if (IN_SUBNETB(a_targetAddr))
3618             targetSubnet = a_targetAddr & IN_CLASSB_SUBNET;
3619         if (IN_SUBNETB(a_candAddr))
3620             candSubnet = a_candAddr & IN_CLASSB_SUBNET;
3621     } /*Class B target */
3622     else if (IN_CLASSC(a_targetAddr)) {
3623         if (!(IN_CLASSC(a_candAddr))) {
3624             (*a_diffNetworkP)++;
3625             return;
3626         }
3627         targetNet = a_targetAddr & IN_CLASSC_NET;
3628         candNet = a_candAddr & IN_CLASSC_NET;
3629
3630         /*
3631          * Note that class C addresses can't have subnets,
3632          * so we leave the defaults untouched.
3633          */
3634     } /*Class C target */
3635     else {
3636         targetNet = a_targetAddr;
3637         candNet = a_candAddr;
3638     }                           /*Class D address */
3639
3640     /*
3641      * Now, simply compare the extracted net and subnet values for
3642      * the two addresses (which at this point are known to be of the
3643      * same class)
3644      */
3645     if (targetNet == candNet) {
3646         if (targetSubnet == candSubnet)
3647             (*a_sameNetOrSubnetP)++;
3648         else
3649             (*a_diffSubnetP)++;
3650     } else
3651         (*a_diffNetworkP)++;
3652
3653 }                               /*h_ClassifyAddress */
3654
3655
3656 /*------------------------------------------------------------------------
3657  * EXPORTED h_GetHostNetStats
3658  *
3659  * Description:
3660  *      Iterate through the host table, and classify each (non-deleted)
3661  *      host entry into ``proximity'' categories (same net or subnet,
3662  *      different subnet, different network).
3663  *
3664  * Arguments:
3665  *      a_numHostsP        : Set to total number of (non-deleted) hosts.
3666  *      a_sameNetOrSubnetP : Set to # hosts on same net/subnet as server.
3667  *      a_diffSubnetP      : Set to # hosts on diff subnet as server.
3668  *      a_diffNetworkP     : Set to # hosts on diff network as server.
3669  *
3670  * Returns:
3671  *      Nothing.
3672  *
3673  * Environment:
3674  *      We only count non-deleted hosts.  The storage pointed to by our
3675  *      parameters is zeroed upon entry.
3676  *
3677  * Side Effects:
3678  *      As advertised.
3679  *------------------------------------------------------------------------*/
3680
3681 void
3682 h_GetHostNetStats(afs_int32 * a_numHostsP, afs_int32 * a_sameNetOrSubnetP,
3683                   afs_int32 * a_diffSubnetP, afs_int32 * a_diffNetworkP)
3684 {                               /*h_GetHostNetStats */
3685
3686     struct host *hostP; /*Ptr to current host entry */
3687     afs_uint32 currAddr_HBO;    /*Curr host addr, host byte order */
3688     int count;
3689
3690     /*
3691      * Clear out the storage pointed to by our parameters.
3692      */
3693     *a_numHostsP = (afs_int32) 0;
3694     *a_sameNetOrSubnetP = (afs_int32) 0;
3695     *a_diffSubnetP = (afs_int32) 0;
3696     *a_diffNetworkP = (afs_int32) 0;
3697
3698     H_LOCK;
3699     for (count = 0, hostP = hostList; hostP && count < hostCount; hostP = hostP->next, count++) {
3700         if (!(hostP->hostFlags & HOSTDELETED)) {
3701             /*
3702              * Bump the number of undeleted host entries found.
3703              * In classifying the current entry's address, make
3704              * sure to first convert to host byte order.
3705              */
3706             (*a_numHostsP)++;
3707             currAddr_HBO = (afs_uint32) ntohl(hostP->host);
3708             h_ClassifyAddress(FS_HostAddr_HBO, currAddr_HBO,
3709                               a_sameNetOrSubnetP, a_diffSubnetP,
3710                               a_diffNetworkP);
3711         }                       /*Only look at non-deleted hosts */
3712     }                           /*For each host record hashed to this index */
3713     if (count != hostCount) {
3714         ViceLog(0, ("h_GetHostNetStats found %d of %d hosts\n", count, hostCount));
3715     } else if (hostP != NULL) {
3716         ViceLog(0, ("h_GetHostNetStats found more than %d hosts\n", hostCount));
3717         ShutDownAndCore(PANIC);
3718     }
3719     H_UNLOCK;
3720 }                               /*h_GetHostNetStats */
3721
3722 static afs_uint32 checktime;
3723 static afs_uint32 clientdeletetime;
3724 static struct AFSFid zerofid;
3725
3726
3727 /*
3728  * XXXX: This routine could use Multi-Rx to avoid serializing the timeouts.
3729  * Since it can serialize them, and pile up, it should be a separate LWP
3730  * from other events.
3731  */
3732 #if 0
3733 static int
3734 CheckHost(struct host *host, int flags, void *rock)
3735 {
3736     struct client *client;
3737     struct rx_connection *cb_conn = NULL;
3738     int code;
3739
3740 #ifdef AFS_DEMAND_ATTACH_FS
3741     /* kill the checkhost lwp ASAP during shutdown */
3742     FS_STATE_RDLOCK;
3743     if (fs_state.mode == FS_MODE_SHUTDOWN) {
3744         FS_STATE_UNLOCK;
3745         return H_ENUMERATE_BAIL(flags);
3746     }
3747     FS_STATE_UNLOCK;
3748 #endif
3749
3750     /* Host is held by h_Enumerate */
3751     H_LOCK;
3752     for (client = host->FirstClient; client; client = client->next) {
3753         if (client->refCount == 0 && client->LastCall < clientdeletetime) {
3754             client->deleted = 1;
3755             host->hostFlags |= CLIENTDELETED;
3756         }
3757     }
3758     if (host->LastCall < checktime) {
3759         h_Lock_r(host);
3760         if (!(host->hostFlags & HOSTDELETED)) {
3761             host->hostFlags |= HWHO_INPROGRESS;
3762             cb_conn = host->callback_rxcon;
3763             rx_GetConnection(cb_conn);
3764             if (host->LastCall < clientdeletetime) {
3765                 host->hostFlags |= HOSTDELETED;
3766                 if (!(host->hostFlags & VENUSDOWN)) {
3767                     host->hostFlags &= ~ALTADDR;        /* alternate address invalid */
3768                     if (host->interface) {
3769                         H_UNLOCK;
3770                         code =
3771                             RXAFSCB_InitCallBackState3(cb_conn,
3772                                                        &FS_HostUUID);
3773                         H_LOCK;
3774                     } else {
3775                         H_UNLOCK;
3776                         code =
3777                             RXAFSCB_InitCallBackState(cb_conn);
3778                         H_LOCK;
3779                     }
3780                     host->hostFlags |= ALTADDR; /* alternate addresses valid */
3781                     if (code) {
3782                         char hoststr[16];
3783                         (void)afs_inet_ntoa_r(host->host, hoststr);
3784                         ViceLog(0,
3785                                 ("CB: RCallBackConnectBack (host.c) failed for host %s:%d\n",
3786                                  hoststr, ntohs(host->port)));
3787                         host->hostFlags |= VENUSDOWN;
3788                     }
3789                     /* Note:  it's safe to delete hosts even if they have call
3790                      * back state, because break delayed callbacks (called when a
3791                      * message is received from the workstation) will always send a
3792                      * break all call backs to the workstation if there is no
3793                      * callback.
3794                      */
3795                 }
3796             } else {
3797                 if (!(host->hostFlags & VENUSDOWN) && host->cblist) {
3798                     char hoststr[16];
3799                     (void)afs_inet_ntoa_r(host->host, hoststr);
3800                     if (host->interface) {
3801                         afsUUID uuid = host->interface->uuid;
3802                         H_UNLOCK;
3803                         code = RXAFSCB_ProbeUuid(cb_conn, &uuid);
3804                         H_LOCK;
3805                         if (code) {
3806                             if (MultiProbeAlternateAddress_r(host)) {
3807                                 ViceLog(0,("CheckHost: Probing all interfaces of host %s:%d failed, code %d\n",
3808                                             hoststr, ntohs(host->port), code));
3809                                 host->hostFlags |= VENUSDOWN;
3810                             }
3811                         }
3812                     } else {
3813                         H_UNLOCK;
3814                         code = RXAFSCB_Probe(cb_conn);
3815                         H_LOCK;
3816                         if (code) {
3817                             ViceLog(0,
3818                                     ("CheckHost: Probe failed for host %s:%d, code %d\n",
3819                                      hoststr, ntohs(host->port), code));
3820                             host->hostFlags |= VENUSDOWN;
3821                         }
3822                     }
3823                 }
3824             }
3825             H_UNLOCK;
3826             rx_PutConnection(cb_conn);
3827             cb_conn=NULL;
3828             H_LOCK;
3829             host->hostFlags &= ~HWHO_INPROGRESS;
3830         }
3831         h_Unlock_r(host);
3832     }
3833     H_UNLOCK;
3834     return held;
3835
3836 }                               /*CheckHost */
3837 #endif
3838
3839 int
3840 CheckHost_r(struct host *host, void *dummy)
3841 {
3842     struct client *client;
3843     struct rx_connection *cb_conn = NULL;
3844     int code;
3845
3846 #ifdef AFS_DEMAND_ATTACH_FS
3847     /* kill the checkhost lwp ASAP during shutdown */
3848     FS_STATE_RDLOCK;
3849     if (fs_state.mode == FS_MODE_SHUTDOWN) {
3850         FS_STATE_UNLOCK;
3851         return H_ENUMERATE_BAIL(0);
3852     }
3853     FS_STATE_UNLOCK;
3854 #endif
3855
3856     /* Host is held by h_Enumerate_r */
3857     for (client = host->FirstClient; client; client = client->next) {
3858         if (client->refCount == 0 && client->LastCall < clientdeletetime) {
3859             client->deleted = 1;
3860             host->hostFlags |= CLIENTDELETED;
3861         }
3862     }
3863     if (host->LastCall < checktime) {
3864         h_Lock_r(host);
3865         if (!(host->hostFlags & HOSTDELETED)) {
3866             host->hostFlags |= HWHO_INPROGRESS;
3867             cb_conn = host->callback_rxcon;
3868             rx_GetConnection(cb_conn);
3869             if (host->LastCall < clientdeletetime) {
3870                 host->hostFlags |= HOSTDELETED;
3871                 if (!(host->hostFlags & VENUSDOWN)) {
3872                     host->hostFlags &= ~ALTADDR;        /* alternate address invalid */
3873                     if (host->interface) {
3874                         H_UNLOCK;
3875                         code =
3876                             RXAFSCB_InitCallBackState3(cb_conn,
3877                                                        &FS_HostUUID);
3878                         H_LOCK;
3879                     } else {
3880                         H_UNLOCK;
3881                         code =
3882                             RXAFSCB_InitCallBackState(cb_conn);
3883                         H_LOCK;
3884                     }
3885                     host->hostFlags |= ALTADDR; /* alternate addresses valid */
3886                     if (code) {
3887                         char hoststr[16];
3888                         (void)afs_inet_ntoa_r(host->host, hoststr);
3889                         ViceLog(0,
3890                                 ("CB: RCallBackConnectBack (host.c) failed for host %s:%d\n",
3891                                  hoststr, ntohs(host->port)));
3892                         host->hostFlags |= VENUSDOWN;
3893                     }
3894                     /* Note:  it's safe to delete hosts even if they have call
3895                      * back state, because break delayed callbacks (called when a
3896                      * message is received from the workstation) will always send a
3897                      * break all call backs to the workstation if there is no
3898                      * callback.
3899                      */
3900                 }
3901             } else {
3902                 if (!(host->hostFlags & VENUSDOWN) && host->cblist) {
3903                     char hoststr[16];
3904                     (void)afs_inet_ntoa_r(host->host, hoststr);
3905                     if (host->interface) {
3906                         afsUUID uuid = host->interface->uuid;
3907                         H_UNLOCK;
3908                         code = RXAFSCB_ProbeUuid(cb_conn, &uuid);
3909                         H_LOCK;
3910                         if (code) {
3911                             if (MultiProbeAlternateAddress_r(host)) {
3912                                 ViceLog(0,("CheckHost_r: Probing all interfaces of host %s:%d failed, code %d\n",
3913                                             hoststr, ntohs(host->port), code));
3914                                 host->hostFlags |= VENUSDOWN;
3915                             }
3916                         }
3917                     } else {
3918                         H_UNLOCK;
3919                         code = RXAFSCB_Probe(cb_conn);
3920                         H_LOCK;
3921                         if (code) {
3922                             ViceLog(0,
3923                                     ("CheckHost_r: Probe failed for host %s:%d, code %d\n",
3924                                      hoststr, ntohs(host->port), code));
3925                             host->hostFlags |= VENUSDOWN;
3926                         }
3927                     }
3928                 }
3929             }
3930             H_UNLOCK;
3931             rx_PutConnection(cb_conn);
3932             cb_conn=NULL;
3933             H_LOCK;
3934             host->hostFlags &= ~HWHO_INPROGRESS;
3935         }
3936         h_Unlock_r(host);
3937     }
3938     return 0;
3939
3940 }                               /*CheckHost_r */
3941
3942
3943 /*
3944  * Set VenusDown for any hosts that have not had a call in 15 minutes and
3945  * don't respond to a probe.  Note that VenusDown can only be cleared if
3946  * a message is received from the host (see ServerLWP in file.c).
3947  * Delete hosts that have not had any calls in 1 hour, clients that
3948  * have not had any calls in 15 minutes.
3949  *
3950  * This routine is called roughly every 5 minutes.
3951  */
3952 void
3953 h_CheckHosts(void)
3954 {
3955     afs_uint32 now = FT_ApproxTime();
3956
3957     memset(&zerofid, 0, sizeof(zerofid));
3958     /*
3959      * Send a probe to the workstation if it hasn't been heard from in
3960      * 15 minutes
3961      */
3962     checktime = now - 15 * 60;
3963     clientdeletetime = now - 120 * 60;  /* 2 hours ago */
3964
3965     H_LOCK;
3966     h_Enumerate_r(CheckHost_r, hostList, NULL);
3967     H_UNLOCK;
3968 }                               /*h_CheckHosts */
3969
3970 /*
3971  * This is called with host locked and held. At this point, the
3972  * hostAddrHashTable has an entry for the primary addr/port inserted
3973  * by h_Alloc_r().  No other interfaces should be considered valid.
3974  *
3975  * The addresses in the interfaceAddr list are in host byte order.
3976  */
3977 int
3978 initInterfaceAddr_r(struct host *host, struct interfaceAddr *interf)
3979 {
3980     int i, j;
3981     int number, count;
3982     afs_uint32 myAddr;
3983     afs_uint16 myPort;
3984     int found;
3985     struct Interface *interface;
3986     char hoststr[16];
3987     char uuidstr[128];
3988     afs_uint16 port7001 = htons(7001);
3989
3990     osi_Assert(host);
3991     osi_Assert(interf);
3992
3993     number = interf->numberOfInterfaces;
3994     myAddr = host->host;        /* current interface address */
3995     myPort = host->port;        /* current port */
3996
3997     ViceLog(125,
3998             ("initInterfaceAddr : host %s:%d numAddr %d\n",
3999               afs_inet_ntoa_r(myAddr, hoststr), ntohs(myPort), number));
4000
4001     /* validation checks */
4002     if (number < 0 || number > AFS_MAX_INTERFACE_ADDR) {
4003         ViceLog(0,
4004                 ("Invalid number of alternate addresses is %d\n", number));
4005         return -1;
4006     }
4007
4008     /*
4009      * The client's notion of its own IP addresses is not reliable.
4010      *
4011      * 1. The client list might contain private address ranges which
4012      *    are likely to be re-used by many clients allocated addresses
4013      *    by a NAT.
4014      *
4015      * 2. The client list will not include any public addresses that
4016      *    are hidden by a NAT.
4017      *
4018      * 3. Private address ranges that are exposed to the server will
4019      *    be obtained from the rx connections that use them.
4020      *
4021      * 4. Lists provided by the client are not necessarily truthful.
4022      *    Many existing clients (UNIX) do not refresh the IP address
4023      *    list as the actual assigned addresses change.  The end result
4024      *    is that they report the initial address list for the lifetime
4025      *    of the process.  In other words, a client can report addresses
4026      *    that they are in fact not using.  Adding these addresses to
4027      *    the host interface list without verification is not only
4028      *    pointless, it is downright dangerous.
4029      *
4030      * We therefore do not add alternate addresses to the addr hash table.
4031      * We only use them for multi-rx callback breaks.
4032      */
4033
4034     /*
4035      * Convert IP addresses to network byte order, and remove
4036      * duplicate and loopback IP addresses from the interface list, and
4037      * determine whether or not the incoming addr/port is
4038      * listed.  Note that if the address matches it is not
4039      * truly a match because the port number for the entries
4040      * in the interface list are port 7001 and the port number
4041      * for this connection might not be 7001.
4042      */
4043     for (i = 0, count = 0, found = 0; i < number; i++) {
4044         if (rx_IsLoopbackAddr(interf->addr_in[i])) {
4045             continue;
4046         }
4047         interf->addr_in[i] = htonl(interf->addr_in[i]);
4048         for (j = 0; j < count; j++) {
4049             if (interf->addr_in[j] == interf->addr_in[i])
4050                 break;
4051         }
4052         if (j == count) {
4053             interf->addr_in[count] = interf->addr_in[i];
4054             if (interf->addr_in[count] == myAddr &&
4055                 port7001 == myPort)
4056                 found = 1;
4057             count++;
4058         }
4059     }
4060
4061     /*
4062      * Allocate and initialize an interface structure for this host.
4063      */
4064     if (found) {
4065         interface = malloc(sizeof(struct Interface) +
4066                            (sizeof(struct AddrPort) * (count - 1)));
4067         if (!interface) {
4068             ViceLogThenPanic(0, ("Failed malloc in initInterfaceAddr_r 1\n"));
4069         }
4070         interface->numberOfInterfaces = count;
4071     } else {
4072         interface = malloc(sizeof(struct Interface) +
4073                            (sizeof(struct AddrPort) * count));
4074         if (!interface) {
4075             ViceLogThenPanic(0, ("Failed malloc in initInterfaceAddr_r 2\n"));
4076         }
4077         interface->numberOfInterfaces = count + 1;
4078         interface->interface[count].addr = myAddr;
4079         interface->interface[count].port = myPort;
4080         interface->interface[count].valid = 1;
4081     }
4082
4083     for (i = 0; i < count; i++) {
4084
4085         interface->interface[i].addr = interf->addr_in[i];
4086         /* We store the port as 7001 because the addresses reported by
4087          * TellMeAboutYourself and WhoAreYou RPCs are only valid if they
4088          * are coming from fully connected hosts (no NAT/PATs)
4089          */
4090         interface->interface[i].port = port7001;
4091         interface->interface[i].valid =
4092             (interf->addr_in[i] == myAddr && port7001 == myPort) ? 1 : 0;
4093     }
4094
4095     interface->uuid = interf->uuid;
4096
4097     osi_Assert(!host->interface);
4098     host->interface = interface;
4099
4100     if (LogLevel >= 125) {
4101         afsUUID_to_string(&interface->uuid, uuidstr, 127);
4102
4103         ViceLog(125, ("--- uuid %s\n", uuidstr));
4104         for (i = 0; i < host->interface->numberOfInterfaces; i++) {
4105             ViceLog(125, ("--- alt address %s:%d\n",
4106                           afs_inet_ntoa_r(host->interface->interface[i].addr, hoststr),
4107                           ntohs(host->interface->interface[i].port)));
4108         }
4109     }
4110
4111     return 0;
4112 }
4113
4114 /* deleted a HashChain structure for this address and host */
4115 /* returns 1 on success */
4116 int
4117 h_DeleteHostFromAddrHashTable_r(afs_uint32 addr, afs_uint16 port,
4118                                 struct host *host)
4119 {
4120     char hoststr[16];
4121     struct h_AddrHashChain **hp, *th;
4122
4123     if (addr == 0 && port == 0)
4124         return 1;
4125
4126     for (hp = &hostAddrHashTable[h_HashIndex(addr)]; (th = *hp);
4127          hp = &th->next) {
4128         osi_Assert(th->hostPtr);
4129         if (th->hostPtr == host && th->addr == addr && th->port == port) {
4130             ViceLog(125, ("h_DeleteHostFromAddrHashTable_r: host %" AFS_PTR_FMT " (%s:%d)\n",
4131                           host, afs_inet_ntoa_r(host->host, hoststr),
4132                           ntohs(host->port)));
4133             *hp = th->next;
4134             free(th);
4135             return 1;
4136         }
4137     }
4138     ViceLog(125,
4139             ("h_DeleteHostFromAddrHashTable_r: host %" AFS_PTR_FMT " (%s:%d) not found\n",
4140              host, afs_inet_ntoa_r(host->host, hoststr),
4141              ntohs(host->port)));
4142     return 0;
4143 }
4144
4145
4146 /*
4147 ** prints out all alternate interface address for the host. The 'level'
4148 ** parameter indicates what level of debugging sets this output
4149 */
4150 void
4151 printInterfaceAddr(struct host *host, int level)
4152 {
4153     int i, number;
4154     char hoststr[16];
4155
4156     if (host->interface) {
4157         /* check alternate addresses */
4158         number = host->interface->numberOfInterfaces;
4159         if (number == 0) {
4160             ViceLog(level, ("no-addresses "));
4161         } else {
4162             for (i = 0; i < number; i++)
4163                 ViceLog(level, ("%s:%d ", afs_inet_ntoa_r(host->interface->interface[i].addr, hoststr),
4164                                 ntohs(host->interface->interface[i].port)));
4165         }
4166     }
4167     ViceLog(level, ("\n"));
4168 }