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