| File: | src/usr.sbin/amd/amd/mapc.c |
| Warning: | line 742, column 1 Potential leak of memory pointed to by 'dir' |
Press '?' to see keyboard shortcuts
Keyboard shortcuts:
| 1 | /* $OpenBSD: mapc.c,v 1.24 2021/10/21 10:55:56 deraadt Exp $ */ | |||
| 2 | ||||
| 3 | /*- | |||
| 4 | * Copyright (c) 1989 Jan-Simon Pendry | |||
| 5 | * Copyright (c) 1989 Imperial College of Science, Technology & Medicine | |||
| 6 | * Copyright (c) 1989, 1993 | |||
| 7 | * The Regents of the University of California. All rights reserved. | |||
| 8 | * | |||
| 9 | * This code is derived from software contributed to Berkeley by | |||
| 10 | * Jan-Simon Pendry at Imperial College, London. | |||
| 11 | * | |||
| 12 | * Redistribution and use in source and binary forms, with or without | |||
| 13 | * modification, are permitted provided that the following conditions | |||
| 14 | * are met: | |||
| 15 | * 1. Redistributions of source code must retain the above copyright | |||
| 16 | * notice, this list of conditions and the following disclaimer. | |||
| 17 | * 2. Redistributions in binary form must reproduce the above copyright | |||
| 18 | * notice, this list of conditions and the following disclaimer in the | |||
| 19 | * documentation and/or other materials provided with the distribution. | |||
| 20 | * 3. Neither the name of the University nor the names of its contributors | |||
| 21 | * may be used to endorse or promote products derived from this software | |||
| 22 | * without specific prior written permission. | |||
| 23 | * | |||
| 24 | * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND | |||
| 25 | * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |||
| 26 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | |||
| 27 | * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE | |||
| 28 | * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | |||
| 29 | * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS | |||
| 30 | * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) | |||
| 31 | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT | |||
| 32 | * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY | |||
| 33 | * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF | |||
| 34 | * SUCH DAMAGE. | |||
| 35 | */ | |||
| 36 | ||||
| 37 | /* | |||
| 38 | * Mount map cache | |||
| 39 | */ | |||
| 40 | ||||
| 41 | #include "am.h" | |||
| 42 | #include <regex.h> | |||
| 43 | ||||
| 44 | /* | |||
| 45 | * Hash table size | |||
| 46 | */ | |||
| 47 | #define NKVHASH(1 << 2) (1 << 2) /* Power of two */ | |||
| 48 | ||||
| 49 | /* | |||
| 50 | * Wildcard key | |||
| 51 | */ | |||
| 52 | static char wildcard[] = "*"; | |||
| 53 | ||||
| 54 | /* | |||
| 55 | * Map cache types | |||
| 56 | * default, none, incremental, all, regexp | |||
| 57 | * MAPC_RE implies MAPC_ALL and must be numerically | |||
| 58 | * greater. | |||
| 59 | */ | |||
| 60 | #define MAPC_DFLT0x000 0x000 | |||
| 61 | #define MAPC_NONE0x001 0x001 | |||
| 62 | #define MAPC_INC0x002 0x002 | |||
| 63 | #define MAPC_ROOT0x004 0x004 | |||
| 64 | #define MAPC_ALL0x010 0x010 | |||
| 65 | #define MAPC_RE0x020 0x020 | |||
| 66 | #define MAPC_ISRE(m)((m)->alloc == 0x020) ((m)->alloc == MAPC_RE0x020) | |||
| 67 | #define MAPC_CACHE_MASK0x0ff 0x0ff | |||
| 68 | #define MAPC_SYNC0x100 0x100 | |||
| 69 | ||||
| 70 | static struct opt_tab mapc_opt[] = { | |||
| 71 | { "all", MAPC_ALL0x010 }, | |||
| 72 | { "default", MAPC_DFLT0x000 }, | |||
| 73 | { "inc", MAPC_INC0x002 }, | |||
| 74 | { "mapdefault", MAPC_DFLT0x000 }, | |||
| 75 | { "none", MAPC_NONE0x001 }, | |||
| 76 | { "re", MAPC_RE0x020 }, | |||
| 77 | { "regexp", MAPC_RE0x020 }, | |||
| 78 | { "sync", MAPC_SYNC0x100 }, | |||
| 79 | { 0, 0 } | |||
| 80 | }; | |||
| 81 | ||||
| 82 | /* | |||
| 83 | * Lookup recursion | |||
| 84 | */ | |||
| 85 | #define MREC_FULL2 2 | |||
| 86 | #define MREC_PART1 1 | |||
| 87 | #define MREC_NONE0 0 | |||
| 88 | ||||
| 89 | /* | |||
| 90 | * Cache map operations | |||
| 91 | */ | |||
| 92 | typedef void add_fn(mnt_map *, char *, char *); | |||
| 93 | typedef int init_fn(char *, time_t *); | |||
| 94 | typedef int search_fn(mnt_map *, char *, char *, char **, time_t *); | |||
| 95 | typedef int reload_fn(mnt_map *, char *, add_fn *); | |||
| 96 | typedef int mtime_fn(char *, time_t *); | |||
| 97 | ||||
| 98 | static void mapc_sync(mnt_map *); | |||
| 99 | ||||
| 100 | /* | |||
| 101 | * Map type | |||
| 102 | */ | |||
| 103 | typedef struct map_type map_type; | |||
| 104 | struct map_type { | |||
| 105 | char *name; /* Name of this map type */ | |||
| 106 | init_fn *init; /* Initialisation */ | |||
| 107 | reload_fn *reload; /* Reload or fill */ | |||
| 108 | search_fn *search; /* Search for new entry */ | |||
| 109 | mtime_fn *mtime; /* Find modify time */ | |||
| 110 | int def_alloc; /* Default allocation mode */ | |||
| 111 | }; | |||
| 112 | ||||
| 113 | /* | |||
| 114 | * Key-value pair | |||
| 115 | */ | |||
| 116 | typedef struct kv kv; | |||
| 117 | struct kv { | |||
| 118 | kv *next; | |||
| 119 | char *key; | |||
| 120 | char *val; | |||
| 121 | }; | |||
| 122 | ||||
| 123 | struct mnt_map { | |||
| 124 | qelem hdr; | |||
| 125 | int refc; /* Reference count */ | |||
| 126 | short flags; /* Allocation flags */ | |||
| 127 | short alloc; /* Allocation mode */ | |||
| 128 | time_t modify; /* Modify time of map */ | |||
| 129 | char *map_name; /* Name of this map */ | |||
| 130 | char *wildcard; /* Wildcard value */ | |||
| 131 | reload_fn *reload; /* Function to be used for reloads */ | |||
| 132 | search_fn *search; /* Function to be used for searching */ | |||
| 133 | mtime_fn *mtime; /* Modify time function */ | |||
| 134 | kv *kvhash[NKVHASH(1 << 2)]; /* Cached data */ | |||
| 135 | }; | |||
| 136 | ||||
| 137 | /* | |||
| 138 | * Map for root node | |||
| 139 | */ | |||
| 140 | static mnt_map *root_map; | |||
| 141 | ||||
| 142 | /* | |||
| 143 | * List of known maps | |||
| 144 | */ | |||
| 145 | extern qelem map_list_head; | |||
| 146 | qelem map_list_head = { &map_list_head, &map_list_head }; | |||
| 147 | ||||
| 148 | /* | |||
| 149 | * Configuration | |||
| 150 | */ | |||
| 151 | ||||
| 152 | /* ROOT MAP */ | |||
| 153 | static int root_init(char *, time_t *); | |||
| 154 | ||||
| 155 | /* FILE MAPS */ | |||
| 156 | extern int file_init(char *, time_t *); | |||
| 157 | extern int file_reload(mnt_map *, char *, add_fn *); | |||
| 158 | extern int file_search(mnt_map *, char *, char *, char **, time_t *); | |||
| 159 | extern int file_mtime(char *, time_t *); | |||
| 160 | ||||
| 161 | /* Network Information Service (NIS) MAPS */ | |||
| 162 | extern int nis_init(char *, time_t *); | |||
| 163 | extern int nis_reload(mnt_map *, char *, add_fn *); | |||
| 164 | extern int nis_search(mnt_map *, char *, char *, char **, time_t *); | |||
| 165 | #define nis_mtimenis_init nis_init | |||
| 166 | ||||
| 167 | /* NDBM MAPS */ | |||
| 168 | #ifdef HAS_NDBM_MAPS | |||
| 169 | extern int ndbm_init(char *, time_t *); | |||
| 170 | extern int ndbm_search(mnt_map *, char *, charo *, char **, time_t *); | |||
| 171 | #define ndbm_mtime ndbm_init | |||
| 172 | #endif /* HAS_NDBM_MAPS */ | |||
| 173 | ||||
| 174 | /* PASSWD MAPS */ | |||
| 175 | extern int passwd_init(char *, time_t *); | |||
| 176 | extern int passwd_search(mnt_map *, char *, char *, char **, time_t *); | |||
| 177 | ||||
| 178 | /* UNION MAPS */ | |||
| 179 | extern int union_init(char *, time_t *); | |||
| 180 | extern int union_search(mnt_map *, char *, char *, char **, time_t *); | |||
| 181 | extern int union_reload(mnt_map *, char *, add_fn *); | |||
| 182 | ||||
| 183 | /* ERROR MAP */ | |||
| 184 | static int error_init(char *, time_t *); | |||
| 185 | static int error_reload(mnt_map *, char *, add_fn *); | |||
| 186 | static int error_search(mnt_map *, char *, char *, char **, time_t *); | |||
| 187 | static int error_mtime(char *, time_t *); | |||
| 188 | ||||
| 189 | static map_type maptypes[] = { | |||
| 190 | { "root", root_init, error_reload, error_search, error_mtime, MAPC_ROOT0x004 }, | |||
| 191 | ||||
| 192 | { "passwd", passwd_init, error_reload, passwd_search, error_mtime, MAPC_INC0x002 }, | |||
| 193 | ||||
| 194 | { "union", union_init, union_reload, union_search, error_mtime, MAPC_ALL0x010 }, | |||
| 195 | ||||
| 196 | { "nis", nis_init, nis_reload, nis_search, nis_mtimenis_init, MAPC_INC0x002 }, | |||
| 197 | ||||
| 198 | #ifdef HAS_NDBM_MAPS | |||
| 199 | { "ndbm", ndbm_init, error_reload, ndbm_search, ndbm_mtime, MAPC_INC0x002 }, | |||
| 200 | #endif | |||
| 201 | ||||
| 202 | { "file", file_init, file_reload, file_search, file_mtime, MAPC_ALL0x010 }, | |||
| 203 | ||||
| 204 | { "error", error_init, error_reload, error_search, error_mtime, MAPC_NONE0x001 }, | |||
| 205 | }; | |||
| 206 | ||||
| 207 | /* | |||
| 208 | * Hash function | |||
| 209 | */ | |||
| 210 | static unsigned int | |||
| 211 | kvhash_of(char *key) | |||
| 212 | { | |||
| 213 | unsigned int i, j; | |||
| 214 | ||||
| 215 | for (i = 0; (j = *key++); i += j) | |||
| 216 | ; | |||
| 217 | ||||
| 218 | return i % NKVHASH(1 << 2); | |||
| 219 | } | |||
| 220 | ||||
| 221 | void | |||
| 222 | mapc_showtypes(FILE *fp) | |||
| 223 | { | |||
| 224 | map_type *mt; | |||
| 225 | char *sep = ""; | |||
| 226 | ||||
| 227 | for (mt = maptypes; mt < maptypes+sizeof(maptypes)/sizeof(maptypes[0]); mt++) { | |||
| 228 | fprintf(fp, "%s%s", sep, mt->name); | |||
| 229 | sep = ", "; | |||
| 230 | } | |||
| 231 | } | |||
| 232 | ||||
| 233 | /* | |||
| 234 | * Add key and val to the map m. | |||
| 235 | * key and val are assumed to be safe copies | |||
| 236 | */ | |||
| 237 | void | |||
| 238 | mapc_add_kv(mnt_map *m, char *key, char *val) | |||
| 239 | { | |||
| 240 | kv **h; | |||
| 241 | kv *n; | |||
| 242 | int hash = kvhash_of(key); | |||
| 243 | ||||
| 244 | #ifdef DEBUG | |||
| 245 | dlog("add_kv: %s -> %s", key, val); | |||
| 246 | #endif | |||
| 247 | ||||
| 248 | if (MAPC_ISRE(m)((m)->alloc == 0x020)) { | |||
| 249 | char keyb[PATH_MAX1024]; | |||
| 250 | regex_t *re; | |||
| 251 | int err; | |||
| 252 | ||||
| 253 | /* | |||
| 254 | * Make sure the string is bound to the start and end | |||
| 255 | */ | |||
| 256 | snprintf(keyb, sizeof(keyb), "^%s$", key); | |||
| 257 | re = malloc(sizeof(*re)); | |||
| 258 | if (re == NULL((void *)0)) { | |||
| 259 | plog(XLOG_USER0x0004, "error allocating RE \"%s\"", keyb); | |||
| 260 | return; | |||
| 261 | } | |||
| 262 | err = regcomp(re, keyb, 0); | |||
| 263 | if (err) { | |||
| 264 | char errbuf[100]; | |||
| 265 | ||||
| 266 | regerror(err, re, errbuf, sizeof errbuf); | |||
| 267 | free(re); | |||
| 268 | plog(XLOG_USER0x0004, "error compiling RE \"%s\": %s", | |||
| 269 | keyb, errbuf); | |||
| 270 | return; | |||
| 271 | } | |||
| 272 | ||||
| 273 | free(key); | |||
| 274 | key = (char *)re; | |||
| 275 | } | |||
| 276 | ||||
| 277 | h = &m->kvhash[hash]; | |||
| 278 | n = ALLOC(kv)((struct kv *) xmalloc(sizeof(struct kv))); | |||
| 279 | n->key = key; | |||
| 280 | n->val = val; | |||
| 281 | n->next = *h; | |||
| 282 | *h = n; | |||
| 283 | } | |||
| 284 | ||||
| 285 | static void | |||
| 286 | mapc_repl_kv(mnt_map *m, char *key, char *val) | |||
| 287 | { | |||
| 288 | kv *k; | |||
| 289 | ||||
| 290 | /* | |||
| 291 | * Compute the hash table offset | |||
| 292 | */ | |||
| 293 | k = m->kvhash[kvhash_of(key)]; | |||
| 294 | ||||
| 295 | /* | |||
| 296 | * Scan the linked list for the key | |||
| 297 | */ | |||
| 298 | while (k && !FSTREQ(k->key, key)((*(k->key) == *(key)) && (strcmp(((k->key)), ( (key))) == 0))) | |||
| 299 | k = k->next; | |||
| 300 | ||||
| 301 | if (k) { | |||
| 302 | free(k->val); | |||
| 303 | k->val = val; | |||
| 304 | } else { | |||
| 305 | mapc_add_kv(m, key, val); | |||
| 306 | } | |||
| 307 | ||||
| 308 | } | |||
| 309 | ||||
| 310 | /* | |||
| 311 | * Search a map for a key. | |||
| 312 | * Calls map specific search routine. | |||
| 313 | * While map is out of date, keep re-syncing. | |||
| 314 | */ | |||
| 315 | static int search_map(mnt_map *m, char *key, char **valp) | |||
| 316 | { | |||
| 317 | int rc; | |||
| 318 | ||||
| 319 | do { | |||
| 320 | rc = (*m->search)(m, m->map_name, key, valp, &m->modify); | |||
| 321 | if (rc < 0) { | |||
| 322 | plog(XLOG_MAP0x0040, "Re-synchronizing cache for map %s", m->map_name); | |||
| 323 | mapc_sync(m); | |||
| 324 | } | |||
| 325 | } while (rc < 0); | |||
| 326 | ||||
| 327 | return rc; | |||
| 328 | } | |||
| 329 | ||||
| 330 | /* | |||
| 331 | * Do a wildcard lookup in the map and | |||
| 332 | * save the result. | |||
| 333 | */ | |||
| 334 | static void | |||
| 335 | mapc_find_wildcard(mnt_map *m) | |||
| 336 | { | |||
| 337 | /* | |||
| 338 | * Attempt to find the wildcard entry | |||
| 339 | */ | |||
| 340 | int rc = search_map(m, wildcard, &m->wildcard); | |||
| 341 | ||||
| 342 | if (rc != 0) | |||
| 343 | m->wildcard = 0; | |||
| 344 | } | |||
| 345 | ||||
| 346 | /* | |||
| 347 | * Make a duplicate reference to an existing map | |||
| 348 | */ | |||
| 349 | #define mapc_dup(m)((m)->refc++, (m)) ((m)->refc++, (m)) | |||
| 350 | ||||
| 351 | /* | |||
| 352 | * Do a map reload | |||
| 353 | */ | |||
| 354 | static int | |||
| 355 | mapc_reload_map(mnt_map *m) | |||
| 356 | { | |||
| 357 | int error; | |||
| 358 | #ifdef DEBUG | |||
| 359 | dlog("calling map reload on %s", m->map_name); | |||
| 360 | #endif | |||
| 361 | error = (*m->reload)(m, m->map_name, mapc_add_kv); | |||
| 362 | if (error) | |||
| 363 | return error; | |||
| 364 | m->wildcard = 0; | |||
| 365 | #ifdef DEBUG | |||
| 366 | dlog("calling mapc_search for wildcard"); | |||
| 367 | #endif | |||
| 368 | error = mapc_search(m, wildcard, &m->wildcard); | |||
| 369 | if (error) | |||
| 370 | m->wildcard = 0; | |||
| 371 | return 0; | |||
| 372 | } | |||
| 373 | ||||
| 374 | /* | |||
| 375 | * Create a new map | |||
| 376 | */ | |||
| 377 | static mnt_map * | |||
| 378 | mapc_create(char *map, char *opt) | |||
| 379 | { | |||
| 380 | mnt_map *m = ALLOC(mnt_map)((struct mnt_map *) xmalloc(sizeof(struct mnt_map))); | |||
| 381 | map_type *mt; | |||
| 382 | time_t modify; | |||
| 383 | int alloc = 0; | |||
| 384 | ||||
| 385 | (void) cmdoption(opt, mapc_opt, &alloc); | |||
| 386 | ||||
| 387 | for (mt = maptypes; mt < maptypes+sizeof(maptypes)/sizeof(maptypes[0]); mt++) | |||
| 388 | if ((*mt->init)(map, &modify) == 0) | |||
| 389 | break; | |||
| 390 | /* assert: mt in maptypes */ | |||
| 391 | ||||
| 392 | m->flags = alloc & ~MAPC_CACHE_MASK0x0ff; | |||
| 393 | alloc &= MAPC_CACHE_MASK0x0ff; | |||
| 394 | ||||
| 395 | if (alloc == MAPC_DFLT0x000) | |||
| 396 | alloc = mt->def_alloc; | |||
| 397 | switch (alloc) { | |||
| 398 | default: | |||
| 399 | plog(XLOG_USER0x0004, "Ambiguous map cache type \"%s\"; using \"inc\"", opt); | |||
| 400 | alloc = MAPC_INC0x002; | |||
| 401 | /* fallthrough... */ | |||
| 402 | case MAPC_NONE0x001: | |||
| 403 | case MAPC_INC0x002: | |||
| 404 | case MAPC_ROOT0x004: | |||
| 405 | break; | |||
| 406 | case MAPC_ALL0x010: | |||
| 407 | /* | |||
| 408 | * If there is no support for reload and it was requested | |||
| 409 | * then back off to incremental instead. | |||
| 410 | */ | |||
| 411 | if (mt->reload == error_reload) { | |||
| 412 | plog(XLOG_WARNING0x0008, "Map type \"%s\" does not support cache type \"all\"; using \"inc\"", mt->name); | |||
| 413 | alloc = MAPC_INC0x002; | |||
| 414 | } | |||
| 415 | break; | |||
| 416 | case MAPC_RE0x020: | |||
| 417 | if (mt->reload == error_reload) { | |||
| 418 | plog(XLOG_WARNING0x0008, "Map type \"%s\" does not support cache type \"re\"", mt->name); | |||
| 419 | mt = &maptypes[sizeof(maptypes)/sizeof(maptypes[0]) - 1]; | |||
| 420 | /* assert: mt->name == "error" */ | |||
| 421 | } | |||
| 422 | break; | |||
| 423 | } | |||
| 424 | ||||
| 425 | #ifdef DEBUG | |||
| 426 | dlog("Map for %s coming from maptype %s", map, mt->name); | |||
| 427 | #endif | |||
| 428 | ||||
| 429 | m->alloc = alloc; | |||
| 430 | m->reload = mt->reload; | |||
| 431 | m->modify = modify; | |||
| 432 | m->search = alloc >= MAPC_ALL0x010 ? error_search : mt->search; | |||
| 433 | m->mtime = mt->mtime; | |||
| 434 | bzero(m->kvhash, sizeof(m->kvhash)); | |||
| 435 | m->map_name = strdup(map); | |||
| 436 | m->refc = 1; | |||
| 437 | m->wildcard = 0; | |||
| 438 | ||||
| 439 | /* | |||
| 440 | * synchronize cache with reality | |||
| 441 | */ | |||
| 442 | mapc_sync(m); | |||
| 443 | ||||
| 444 | return m; | |||
| 445 | } | |||
| 446 | ||||
| 447 | /* | |||
| 448 | * Free the cached data in a map | |||
| 449 | */ | |||
| 450 | static void | |||
| 451 | mapc_clear(mnt_map *m) | |||
| 452 | { | |||
| 453 | int i; | |||
| 454 | ||||
| 455 | /* | |||
| 456 | * For each of the hash slots, chain | |||
| 457 | * along free'ing the data. | |||
| 458 | */ | |||
| 459 | for (i = 0; i < NKVHASH(1 << 2); i++) { | |||
| 460 | kv *k = m->kvhash[i]; | |||
| 461 | while (k) { | |||
| 462 | kv *n = k->next; | |||
| 463 | free(k->key); | |||
| 464 | free(k->val); | |||
| 465 | free(k); | |||
| 466 | k = n; | |||
| 467 | } | |||
| 468 | } | |||
| 469 | /* | |||
| 470 | * Zero the hash slots | |||
| 471 | */ | |||
| 472 | bzero(m->kvhash, sizeof(m->kvhash)); | |||
| 473 | /* | |||
| 474 | * Free the wildcard if it exists | |||
| 475 | */ | |||
| 476 | if (m->wildcard) { | |||
| 477 | free(m->wildcard); | |||
| 478 | m->wildcard = 0; | |||
| 479 | } | |||
| 480 | } | |||
| 481 | ||||
| 482 | /* | |||
| 483 | * Find a map, or create one if it does not exist | |||
| 484 | */ | |||
| 485 | mnt_map * | |||
| 486 | mapc_find(char *map, char *opt) | |||
| 487 | { | |||
| 488 | mnt_map *m; | |||
| 489 | ||||
| 490 | /* | |||
| 491 | * Search the list of known maps to see if | |||
| 492 | * it has already been loaded. If it is found | |||
| 493 | * then return a duplicate reference to it. | |||
| 494 | * Otherwise make a new map as required and | |||
| 495 | * add it to the list of maps | |||
| 496 | */ | |||
| 497 | ITER(m, mnt_map, &map_list_head)for ((m) = ((mnt_map *) (((&map_list_head))->q_forw)); (m) != ((mnt_map *) (&map_list_head)); (m) = ((mnt_map * ) (((qelem *) (m))->q_forw))) | |||
| 498 | if (STREQ(m->map_name, map)(strcmp((m->map_name), (map)) == 0)) | |||
| 499 | return mapc_dup(m)((m)->refc++, (m)); | |||
| 500 | ||||
| 501 | m = mapc_create(map, opt); | |||
| 502 | ins_que(&m->hdr, &map_list_head); | |||
| 503 | return m; | |||
| 504 | } | |||
| 505 | ||||
| 506 | /* | |||
| 507 | * Free a map. | |||
| 508 | */ | |||
| 509 | void | |||
| 510 | mapc_free(void *arg) | |||
| 511 | { | |||
| 512 | mnt_map *m = arg; | |||
| 513 | /* | |||
| 514 | * Decrement the reference count. | |||
| 515 | * If the reference count hits zero | |||
| 516 | * then throw the map away. | |||
| 517 | */ | |||
| 518 | if (m && --m->refc == 0) { | |||
| 519 | mapc_clear(m); | |||
| 520 | free(m->map_name); | |||
| 521 | rem_que(&m->hdr); | |||
| 522 | free(m); | |||
| 523 | } | |||
| 524 | } | |||
| 525 | ||||
| 526 | /* | |||
| 527 | * Search the map for the key. | |||
| 528 | * Put a safe copy in *pval or return | |||
| 529 | * an error code | |||
| 530 | */ | |||
| 531 | int | |||
| 532 | mapc_meta_search(mnt_map *m, char *key, char **pval, int recurse) | |||
| 533 | { | |||
| 534 | int error = 0; | |||
| 535 | kv *k = 0; | |||
| 536 | ||||
| 537 | /* | |||
| 538 | * Firewall | |||
| 539 | */ | |||
| 540 | if (!m) { | |||
| 541 | plog(XLOG_ERROR0x0002, "Null map request for %s", key); | |||
| 542 | return ENOENT2; | |||
| 543 | } | |||
| 544 | ||||
| 545 | if (m->flags & MAPC_SYNC0x100) { | |||
| 546 | /* | |||
| 547 | * Get modify time... | |||
| 548 | */ | |||
| 549 | time_t t; | |||
| 550 | error = (*m->mtime)(m->map_name, &t); | |||
| 551 | if (error || t > m->modify) { | |||
| 552 | m->modify = t; | |||
| 553 | plog(XLOG_INFO0x0010, "Map %s is out of date", m->map_name); | |||
| 554 | mapc_sync(m); | |||
| 555 | } | |||
| 556 | } | |||
| 557 | ||||
| 558 | if (!MAPC_ISRE(m)((m)->alloc == 0x020)) { | |||
| 559 | /* | |||
| 560 | * Compute the hash table offset | |||
| 561 | */ | |||
| 562 | k = m->kvhash[kvhash_of(key)]; | |||
| 563 | ||||
| 564 | /* | |||
| 565 | * Scan the linked list for the key | |||
| 566 | */ | |||
| 567 | while (k && !FSTREQ(k->key, key)((*(k->key) == *(key)) && (strcmp(((k->key)), ( (key))) == 0))) k = k->next; | |||
| 568 | ||||
| 569 | } | |||
| 570 | else if (recurse == MREC_FULL2) { | |||
| 571 | /* | |||
| 572 | * Try for an RE match against the entire map. | |||
| 573 | * Note that this will be done in a "random" | |||
| 574 | * order. | |||
| 575 | */ | |||
| 576 | ||||
| 577 | int i; | |||
| 578 | ||||
| 579 | for (i = 0; i < NKVHASH(1 << 2); i++) { | |||
| 580 | k = m->kvhash[i]; | |||
| 581 | while (k) { | |||
| 582 | if (regexec((regex_t *)k->key, key, | |||
| 583 | 0, NULL((void *)0), 0) == 0) | |||
| 584 | break; | |||
| 585 | k = k->next; | |||
| 586 | } | |||
| 587 | if (k) | |||
| 588 | break; | |||
| 589 | } | |||
| 590 | } | |||
| 591 | ||||
| 592 | /* | |||
| 593 | * If found then take a copy | |||
| 594 | */ | |||
| 595 | if (k) { | |||
| 596 | if (k->val) | |||
| 597 | *pval = strdup(k->val); | |||
| 598 | else | |||
| 599 | error = ENOENT2; | |||
| 600 | } else if (m->alloc >= MAPC_ALL0x010) { | |||
| 601 | /* | |||
| 602 | * If the entire map is cached then this | |||
| 603 | * key does not exist. | |||
| 604 | */ | |||
| 605 | error = ENOENT2; | |||
| 606 | } else { | |||
| 607 | /* | |||
| 608 | * Otherwise search the map. If we are | |||
| 609 | * in incremental mode then add the key | |||
| 610 | * to the cache. | |||
| 611 | */ | |||
| 612 | error = search_map(m, key, pval); | |||
| 613 | if (!error && m->alloc == MAPC_INC0x002) | |||
| 614 | mapc_add_kv(m, strdup(key), strdup(*pval)); | |||
| 615 | } | |||
| 616 | ||||
| 617 | /* | |||
| 618 | * If an error, and a wildcard exists, | |||
| 619 | * and the key is not internal then | |||
| 620 | * return a copy of the wildcard. | |||
| 621 | */ | |||
| 622 | if (error > 0) { | |||
| 623 | if (recurse == MREC_FULL2 && !MAPC_ISRE(m)((m)->alloc == 0x020)) { | |||
| 624 | char wildname[PATH_MAX1024]; | |||
| 625 | char *subp; | |||
| 626 | if (*key == '/') | |||
| 627 | return error; | |||
| 628 | /* | |||
| 629 | * Keep chopping sub-directories from the RHS | |||
| 630 | * and replacing with "/ *" and repeat the lookup. | |||
| 631 | * For example: | |||
| 632 | * "src/gnu/gcc" -> "src / gnu / *" -> "src / *" | |||
| 633 | */ | |||
| 634 | strlcpy(wildname, key, sizeof wildname); | |||
| 635 | while (error && (subp = strrchr(wildname, '/'))) { | |||
| 636 | strlcpy(subp, "/*", 3); | |||
| 637 | #ifdef DEBUG | |||
| 638 | dlog("mapc recurses on %s", wildname); | |||
| 639 | #endif | |||
| 640 | error = mapc_meta_search(m, wildname, pval, MREC_PART1); | |||
| 641 | if (error) | |||
| 642 | *subp = 0; | |||
| 643 | } | |||
| 644 | if (error > 0 && m->wildcard) { | |||
| 645 | *pval = strdup(m->wildcard); | |||
| 646 | error = 0; | |||
| 647 | } | |||
| 648 | } | |||
| 649 | } | |||
| 650 | ||||
| 651 | return error; | |||
| 652 | } | |||
| 653 | ||||
| 654 | int | |||
| 655 | mapc_search(mnt_map *m, char *key, char **pval) | |||
| 656 | { | |||
| 657 | return mapc_meta_search(m, key, pval, MREC_FULL2); | |||
| 658 | } | |||
| 659 | ||||
| 660 | /* | |||
| 661 | * Get map cache in sync with physical representation | |||
| 662 | */ | |||
| 663 | static void | |||
| 664 | mapc_sync(mnt_map *m) | |||
| 665 | { | |||
| 666 | if (m->alloc != MAPC_ROOT0x004) { | |||
| 667 | mapc_clear(m); | |||
| 668 | ||||
| 669 | if (m->alloc >= MAPC_ALL0x010) | |||
| 670 | if (mapc_reload_map(m)) | |||
| 671 | m->alloc = MAPC_INC0x002; | |||
| 672 | /* | |||
| 673 | * Attempt to find the wildcard entry | |||
| 674 | */ | |||
| 675 | if (m->alloc < MAPC_ALL0x010) | |||
| 676 | mapc_find_wildcard(m); | |||
| 677 | } | |||
| 678 | } | |||
| 679 | ||||
| 680 | /* | |||
| 681 | * Reload all the maps | |||
| 682 | * Called when Amd gets hit by a SIGHUP. | |||
| 683 | */ | |||
| 684 | void mapc_reload(void) | |||
| 685 | { | |||
| 686 | mnt_map *m; | |||
| 687 | ||||
| 688 | /* | |||
| 689 | * For all the maps, | |||
| 690 | * Throw away the existing information. | |||
| 691 | * Do a reload | |||
| 692 | * Find the wildcard | |||
| 693 | */ | |||
| 694 | ITER(m, mnt_map, &map_list_head)for ((m) = ((mnt_map *) (((&map_list_head))->q_forw)); (m) != ((mnt_map *) (&map_list_head)); (m) = ((mnt_map * ) (((qelem *) (m))->q_forw))) | |||
| 695 | mapc_sync(m); | |||
| 696 | } | |||
| 697 | ||||
| 698 | /* | |||
| 699 | * Root map. | |||
| 700 | * The root map is used to bootstrap amd. | |||
| 701 | * All the require top-level mounts are added | |||
| 702 | * into the root map and then the map is iterated | |||
| 703 | * and a lookup is done on all the mount points. | |||
| 704 | * This causes the top level mounts to be automounted. | |||
| 705 | */ | |||
| 706 | ||||
| 707 | static int | |||
| 708 | root_init(char *map, time_t *tp) | |||
| 709 | { | |||
| 710 | *tp = clocktime()(clock_valid ? clock_valid : time(&clock_valid)); | |||
| 711 | return strcmp(map, ROOT_MAP"\"root\"") == 0 ? 0 : ENOENT2; | |||
| 712 | } | |||
| 713 | ||||
| 714 | /* | |||
| 715 | * Add a new entry to the root map | |||
| 716 | * | |||
| 717 | * dir - directory (key) | |||
| 718 | * opts - mount options | |||
| 719 | * map - map name | |||
| 720 | */ | |||
| 721 | void | |||
| 722 | root_newmap(char *dir, char *opts, char *map) | |||
| 723 | { | |||
| 724 | char str[PATH_MAX1024]; | |||
| 725 | ||||
| 726 | /* | |||
| 727 | * First make sure we have a root map to talk about... | |||
| 728 | */ | |||
| 729 | if (!root_map) | |||
| ||||
| 730 | root_map = mapc_find(ROOT_MAP"\"root\"", "mapdefault"); | |||
| 731 | ||||
| 732 | /* | |||
| 733 | * Then add the entry... | |||
| 734 | */ | |||
| 735 | dir = strdup(dir); | |||
| 736 | if (map) | |||
| 737 | snprintf(str, sizeof(str), "cache:=mapdefault;type:=toplvl;fs:=\"%s\";%s", | |||
| 738 | map, opts ? opts : ""); | |||
| 739 | else | |||
| 740 | strlcpy(str, opts, sizeof str); | |||
| 741 | mapc_repl_kv(root_map, dir, strdup(str)); | |||
| 742 | } | |||
| ||||
| 743 | ||||
| 744 | int | |||
| 745 | mapc_keyiter(mnt_map *m, void (*fn)(char *,void *), void *arg) | |||
| 746 | { | |||
| 747 | int i; | |||
| 748 | int c = 0; | |||
| 749 | ||||
| 750 | for (i = 0; i < NKVHASH(1 << 2); i++) { | |||
| 751 | kv *k = m->kvhash[i]; | |||
| 752 | while (k) { | |||
| 753 | (*fn)(k->key, arg); | |||
| 754 | k = k->next; | |||
| 755 | c++; | |||
| 756 | } | |||
| 757 | } | |||
| 758 | ||||
| 759 | return c; | |||
| 760 | } | |||
| 761 | ||||
| 762 | /* | |||
| 763 | * Iterate over the root map | |||
| 764 | * and call (*fn)() on the key | |||
| 765 | * of all the nodes. | |||
| 766 | * Finally throw away the root map. | |||
| 767 | */ | |||
| 768 | int | |||
| 769 | root_keyiter(void (*fn)(char *,void *), void *arg) | |||
| 770 | { | |||
| 771 | if (root_map) { | |||
| 772 | int c = mapc_keyiter(root_map, fn, arg); | |||
| 773 | #ifdef notdef | |||
| 774 | mapc_free(root_map); | |||
| 775 | root_map = 0; | |||
| 776 | #endif | |||
| 777 | return c; | |||
| 778 | } | |||
| 779 | return 0; | |||
| 780 | } | |||
| 781 | ||||
| 782 | /* | |||
| 783 | * Error map | |||
| 784 | */ | |||
| 785 | static int | |||
| 786 | error_init(char *map, time_t *tp) | |||
| 787 | { | |||
| 788 | plog(XLOG_USER0x0004, "No source data for map %s", map); | |||
| 789 | *tp = 0; | |||
| 790 | return 0; | |||
| 791 | } | |||
| 792 | ||||
| 793 | static int | |||
| 794 | error_search(mnt_map *m, char *map, char *key, char **pval, time_t *tp) | |||
| 795 | { | |||
| 796 | return ENOENT2; | |||
| 797 | } | |||
| 798 | ||||
| 799 | static int | |||
| 800 | error_reload(mnt_map *m, char *map, add_fn *fn) | |||
| 801 | { | |||
| 802 | return ENOENT2; | |||
| 803 | } | |||
| 804 | ||||
| 805 | static int | |||
| 806 | error_mtime(char *map, time_t *tp) | |||
| 807 | { | |||
| 808 | *tp = 0; | |||
| 809 | return 0; | |||
| 810 | } |