2 * Copyright 2000, International Business Machines Corporation and others.
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
10 #include <afs/param.h>
24 osi_rwlock_t cm_cellLock;
26 /* function called as callback proc from cm_SearchCellFile. Return 0 to
27 * continue processing.
29 * At the present time the return value is ignored by the caller.
31 long cm_AddCellProc(void *rockp, struct sockaddr_in *addrp, char *hostnamep)
36 cm_cell_rock_t *cellrockp = (cm_cell_rock_t *)rockp;
39 cellp = cellrockp->cellp;
40 probe = !(cellrockp->flags & CM_FLAG_NOPROBE);
42 /* if this server was previously created by fs setserverprefs */
43 if ( tsp = cm_FindServer(addrp, CM_SERVER_VLDB))
47 else if (tsp->cellp != cellp) {
48 osi_Log3(afsd_logp, "found a vlserver %s associated with two cells named %s and %s",
49 osi_LogSaveString(afsd_logp,hostnamep),
50 osi_LogSaveString(afsd_logp,tsp->cellp->name),
51 osi_LogSaveString(afsd_logp,cellp->name));
55 tsp = cm_NewServer(addrp, CM_SERVER_VLDB, cellp, probe ? 0 : CM_FLAG_NOPROBE);
57 /* Insert the vlserver into a sorted list, sorted by server rank */
58 tsrp = cm_NewServerRef(tsp, 0);
59 cm_InsertServerList(&cellp->vlServersp, tsrp);
60 /* drop the allocation reference */
61 lock_ObtainWrite(&cm_serverLock);
63 lock_ReleaseWrite(&cm_serverLock);
68 /* if it's from DNS, see if it has expired
69 * and check to make sure we have a valid set of volume servers
70 * this function must be called with a Write Lock on cm_cellLock
72 cm_cell_t *cm_UpdateCell(cm_cell_t * cp, afs_uint32 flags)
80 lock_ObtainMutex(&cp->mx);
81 if ((cp->vlServersp == NULL
82 #ifdef AFS_FREELANCE_CLIENT
83 && !(cp->flags & CM_CELLFLAG_FREELANCE)
85 ) || (time(0) > cp->timeout)
87 || (cm_dnsEnabled && (cp->flags & CM_CELLFLAG_DNS) &&
88 ((cp->flags & CM_CELLFLAG_VLSERVER_INVALID)))
91 /* must empty cp->vlServersp */
93 cm_FreeServerList(&cp->vlServersp, CM_FREESERVERLIST_DELETE);
94 cp->vlServersp = NULL;
99 code = cm_SearchCellFile(cp->name, NULL, cm_AddCellProc, &rock);
105 code = cm_SearchCellByDNS(cp->name, NULL, &ttl, cm_AddCellProc, &rock);
106 if (code == 0) { /* got cell from DNS */
107 cp->flags |= CM_CELLFLAG_DNS;
108 cp->flags &= ~CM_CELLFLAG_VLSERVER_INVALID;
109 cp->timeout = time(0) + ttl;
111 fprintf(stderr, "cell %s: ttl=%d\n", cp->name, ttl);
114 /* if we fail to find it this time, we'll just do nothing and leave the
115 * current entry alone
117 cp->flags |= CM_CELLFLAG_VLSERVER_INVALID;
121 #endif /* AFS_AFSDB_ENV */
123 cp->timeout = time(0) + 7200;
126 lock_ReleaseMutex(&cp->mx);
127 return code ? NULL : cp;
130 /* load up a cell structure from the cell database, AFS_CELLSERVDB */
131 cm_cell_t *cm_GetCell(char *namep, afs_uint32 flags)
133 return cm_GetCell_Gen(namep, NULL, flags);
136 cm_cell_t *cm_GetCell_Gen(char *namep, char *newnamep, afs_uint32 flags)
140 char fullname[200]="";
141 int hasWriteLock = 0;
145 if (!strcmp(namep,SMB_IOCTL_FILENAME_NOSLASH))
148 hash = CM_CELL_NAME_HASH(namep);
150 lock_ObtainRead(&cm_cellLock);
151 for (cp = cm_data.cellNameHashTablep[hash]; cp; cp=cp->nameNextp) {
152 if (cm_stricmp_utf8(namep, cp->name) == 0) {
153 strcpy(fullname, cp->name);
159 for (cp = cm_data.allCellsp; cp; cp=cp->allNextp) {
160 if (strnicmp(namep, cp->name, strlen(namep)) == 0) {
161 strcpy(fullname, cp->name);
167 lock_ReleaseRead(&cm_cellLock);
170 cm_UpdateCell(cp, flags);
171 } else if (flags & CM_FLAG_CREATE) {
172 lock_ObtainWrite(&cm_cellLock);
175 /* when we dropped the lock the cell could have been added
176 * to the list so check again while holding the write lock
178 for (cp = cm_data.cellNameHashTablep[hash]; cp; cp=cp->nameNextp) {
179 if (cm_stricmp_utf8(namep, cp->name) == 0) {
180 strcpy(fullname, cp->name);
188 for (cp = cm_data.allCellsp; cp; cp=cp->allNextp) {
189 if (strnicmp(namep, cp->name, strlen(namep)) == 0) {
190 strcpy(fullname, cp->name);
198 if ( cm_data.currentCells >= cm_data.maxCells )
199 osi_panic("Exceeded Max Cells", __FILE__, __LINE__);
201 /* don't increment currentCells until we know that we
202 * are going to keep this entry
204 cp = &cm_data.cellBaseAddress[cm_data.currentCells];
205 memset(cp, 0, sizeof(cm_cell_t));
206 cp->magic = CM_CELL_MAGIC;
210 code = cm_SearchCellFile(namep, fullname, cm_AddCellProc, &rock);
212 osi_Log3(afsd_logp,"in cm_GetCell_gen cm_SearchCellFile(%s) returns code= %d fullname= %s",
213 osi_LogSaveString(afsd_logp,namep), code, osi_LogSaveString(afsd_logp,fullname));
219 code = cm_SearchCellByDNS(namep, fullname, &ttl, cm_AddCellProc, &rock);
221 osi_Log3(afsd_logp,"in cm_GetCell_gen cm_SearchCellByDNS(%s) returns code= %d fullname= %s",
222 osi_LogSaveString(afsd_logp,namep), code, osi_LogSaveString(afsd_logp,fullname));
225 } else { /* got cell from DNS */
226 cp->flags |= CM_CELLFLAG_DNS;
227 cp->flags &= ~CM_CELLFLAG_VLSERVER_INVALID;
228 cp->timeout = time(0) + ttl;
238 cp->timeout = time(0) + 7200; /* two hour timeout */
241 /* we have now been given the fullname of the cell. It may
242 * be that we already have a cell with that name. If so,
243 * we should use it instead of completing the allocation
246 hash = CM_CELL_NAME_HASH(fullname);
247 for (cp2 = cm_data.cellNameHashTablep[hash]; cp2; cp2=cp2->nameNextp) {
248 if (cm_stricmp_utf8(fullname, cp2->name) == 0) {
254 cm_FreeServerList(&cp->vlServersp, CM_FREESERVERLIST_DELETE);
260 /* randomise among those vlservers having the same rank*/
261 cm_RandomizeServer(&cp->vlServersp);
263 /* otherwise we found the cell, and so we're nearly done */
264 lock_InitializeMutex(&cp->mx, "cm_cell_t mutex");
267 strncpy(cp->name, fullname, CELL_MAXNAMELEN);
268 cp->name[CELL_MAXNAMELEN-1] = '\0';
270 /* the cellID cannot be 0 */
271 cp->cellID = ++cm_data.currentCells;
273 /* append cell to global list */
274 if (cm_data.allCellsp == NULL) {
275 cm_data.allCellsp = cp;
277 for (cp2 = cm_data.allCellsp; cp2->allNextp; cp2=cp2->allNextp)
283 cm_AddCellToNameHashTable(cp);
284 cm_AddCellToIDHashTable(cp);
289 lock_ReleaseWrite(&cm_cellLock);
291 /* fullname is not valid if cp == NULL */
293 strcpy(newnamep, fullname);
298 cm_cell_t *cm_FindCellByID(afs_int32 cellID, afs_uint32 flags)
303 lock_ObtainRead(&cm_cellLock);
305 hash = CM_CELL_ID_HASH(cellID);
307 for (cp = cm_data.cellIDHashTablep[hash]; cp; cp=cp->idNextp) {
308 if (cellID == cp->cellID)
311 lock_ReleaseRead(&cm_cellLock);
314 cm_UpdateCell(cp, flags);
320 cm_ValidateCell(void)
325 for (cellp = cm_data.allCellsp, count = 0; cellp; cellp=cellp->allNextp, count++) {
326 if ( cellp->magic != CM_CELL_MAGIC ) {
327 afsi_log("cm_ValidateCell failure: cellp->magic != CM_CELL_MAGIC");
328 fprintf(stderr, "cm_ValidateCell failure: cellp->magic != CM_CELL_MAGIC\n");
331 if ( count != 0 && cellp == cm_data.allCellsp ||
332 count > cm_data.maxCells ) {
333 afsi_log("cm_ValidateCell failure: cm_data.allCellsp infinite loop");
334 fprintf(stderr, "cm_ValidateCell failure: cm_data.allCellsp infinite loop\n");
339 if ( count != cm_data.currentCells ) {
340 afsi_log("cm_ValidateCell failure: count != cm_data.currentCells");
341 fprintf(stderr, "cm_ValidateCell failure: count != cm_data.currentCells\n");
350 cm_ShutdownCell(void)
354 for (cellp = cm_data.allCellsp; cellp; cellp=cellp->allNextp)
355 lock_FinalizeMutex(&cellp->mx);
361 void cm_InitCell(int newFile, long maxCells)
363 static osi_once_t once;
365 if (osi_Once(&once)) {
368 lock_InitializeRWLock(&cm_cellLock, "cell global lock");
371 cm_data.allCellsp = NULL;
372 cm_data.currentCells = 0;
373 cm_data.maxCells = maxCells;
374 memset(cm_data.cellNameHashTablep, 0, sizeof(cm_cell_t *) * cm_data.cellHashTableSize);
375 memset(cm_data.cellIDHashTablep, 0, sizeof(cm_cell_t *) * cm_data.cellHashTableSize);
377 #ifdef AFS_FREELANCE_CLIENT
378 /* Generate a dummy entry for the Freelance cell whether or not
379 * freelance mode is being used in this session
382 cellp = &cm_data.cellBaseAddress[cm_data.currentCells++];
383 memset(cellp, 0, sizeof(cm_cell_t));
384 cellp->magic = CM_CELL_MAGIC;
386 lock_InitializeMutex(&cellp->mx, "cm_cell_t mutex");
389 strncpy(cellp->name, "Freelance.Local.Cell", CELL_MAXNAMELEN); /*safe*/
391 /* thread on global list */
392 cellp->allNextp = cm_data.allCellsp;
393 cm_data.allCellsp = cellp;
395 cellp->cellID = AFS_FAKE_ROOT_CELL_ID;
396 cellp->vlServersp = NULL;
397 cellp->flags = CM_CELLFLAG_FREELANCE;
399 cm_AddCellToNameHashTable(cellp);
400 cm_AddCellToIDHashTable(cellp);
403 for (cellp = cm_data.allCellsp; cellp; cellp=cellp->allNextp) {
404 lock_InitializeMutex(&cellp->mx, "cm_cell_t mutex");
405 cellp->vlServersp = NULL;
406 cellp->flags |= CM_CELLFLAG_VLSERVER_INVALID;
414 void cm_ChangeRankCellVLServer(cm_server_t *tsp)
419 cp = tsp->cellp; /* cell that this vlserver belongs to */
421 lock_ObtainMutex(&cp->mx);
422 code = cm_ChangeRankServer(&cp->vlServersp, tsp);
424 if ( !code ) /* if the server list was rearranged */
425 cm_RandomizeServer(&cp->vlServersp);
427 lock_ReleaseMutex(&cp->mx);
431 int cm_DumpCells(FILE *outputFile, char *cookie, int lock)
438 lock_ObtainRead(&cm_cellLock);
440 sprintf(output, "%s - dumping cells - cm_data.currentCells=%d, cm_data.maxCells=%d\r\n",
441 cookie, cm_data.currentCells, cm_data.maxCells);
442 WriteFile(outputFile, output, (DWORD)strlen(output), &zilch, NULL);
444 for (cellp = cm_data.allCellsp; cellp; cellp=cellp->allNextp) {
445 sprintf(output, "%s cellp=0x%p,name=%s ID=%d flags=0x%x\r\n",
446 cookie, cellp, cellp->name, cellp->cellID, cellp->flags);
447 WriteFile(outputFile, output, (DWORD)strlen(output), &zilch, NULL);
450 sprintf(output, "%s - Done dumping cells.\r\n", cookie);
451 WriteFile(outputFile, output, (DWORD)strlen(output), &zilch, NULL);
454 lock_ReleaseRead(&cm_cellLock);
459 /* call with volume write-locked and mutex held */
460 void cm_AddCellToNameHashTable(cm_cell_t *cellp)
464 if (cellp->flags & CM_CELLFLAG_IN_NAMEHASH)
467 i = CM_CELL_NAME_HASH(cellp->name);
469 cellp->nameNextp = cm_data.cellNameHashTablep[i];
470 cm_data.cellNameHashTablep[i] = cellp;
471 cellp->flags |= CM_CELLFLAG_IN_NAMEHASH;
474 /* call with cell write-locked and mutex held */
475 void cm_RemoveCellFromNameHashTable(cm_cell_t *cellp)
481 if (cellp->flags & CM_CELLFLAG_IN_NAMEHASH) {
482 /* hash it out first */
483 i = CM_CELL_NAME_HASH(cellp->name);
484 for (lcellpp = &cm_data.cellNameHashTablep[i], tcellp = cm_data.cellNameHashTablep[i];
486 lcellpp = &tcellp->nameNextp, tcellp = tcellp->nameNextp) {
487 if (tcellp == cellp) {
488 *lcellpp = cellp->nameNextp;
489 cellp->flags &= ~CM_CELLFLAG_IN_NAMEHASH;
490 cellp->nameNextp = NULL;
497 /* call with cell write-locked and mutex held */
498 void cm_AddCellToIDHashTable(cm_cell_t *cellp)
502 if (cellp->flags & CM_CELLFLAG_IN_IDHASH)
505 i = CM_CELL_ID_HASH(cellp->cellID);
507 cellp->idNextp = cm_data.cellIDHashTablep[i];
508 cm_data.cellIDHashTablep[i] = cellp;
509 cellp->flags |= CM_CELLFLAG_IN_IDHASH;
512 /* call with cell write-locked and mutex held */
513 void cm_RemoveCellFromIDHashTable(cm_cell_t *cellp)
519 if (cellp->flags & CM_CELLFLAG_IN_IDHASH) {
520 /* hash it out first */
521 i = CM_CELL_ID_HASH(cellp->cellID);
522 for (lcellpp = &cm_data.cellIDHashTablep[i], tcellp = cm_data.cellIDHashTablep[i];
524 lcellpp = &tcellp->idNextp, tcellp = tcellp->idNextp) {
525 if (tcellp == cellp) {
526 *lcellpp = cellp->idNextp;
527 cellp->flags &= ~CM_CELLFLAG_IN_IDHASH;
528 cellp->idNextp = NULL;