paul@181 | 1 | /* |
paul@181 | 2 | * cache.c - allocation/initialization/free routines for cache |
paul@181 | 3 | * |
paul@181 | 4 | * Copyright (C) 2001 Andreas Dilger |
paul@181 | 5 | * Copyright (C) 2003 Theodore Ts'o |
paul@181 | 6 | * |
paul@181 | 7 | * %Begin-Header% |
paul@181 | 8 | * This file may be redistributed under the terms of the |
paul@181 | 9 | * GNU Lesser General Public License. |
paul@181 | 10 | * %End-Header% |
paul@181 | 11 | */ |
paul@181 | 12 | |
paul@181 | 13 | #include "config.h" |
paul@181 | 14 | #if HAVE_UNISTD_H |
paul@181 | 15 | #include <unistd.h> |
paul@181 | 16 | #endif |
paul@181 | 17 | #ifdef HAVE_ERRNO_H |
paul@181 | 18 | #include <errno.h> |
paul@181 | 19 | #endif |
paul@181 | 20 | #include <stdlib.h> |
paul@181 | 21 | #include <string.h> |
paul@181 | 22 | #ifdef HAVE_SYS_PRCTL_H |
paul@181 | 23 | #include <sys/prctl.h> |
paul@181 | 24 | #else |
paul@181 | 25 | #define PR_GET_DUMPABLE 3 |
paul@181 | 26 | #endif |
paul@181 | 27 | #if (!defined(HAVE_PRCTL) && defined(linux)) |
paul@181 | 28 | #include <sys/syscall.h> |
paul@181 | 29 | #endif |
paul@181 | 30 | #ifdef HAVE_SYS_STAT_H |
paul@181 | 31 | #include <sys/stat.h> |
paul@181 | 32 | #endif |
paul@181 | 33 | #include "blkidP.h" |
paul@181 | 34 | |
paul@181 | 35 | int blkid_debug_mask = 0; |
paul@181 | 36 | |
paul@181 | 37 | |
paul@181 | 38 | static char *safe_getenv(const char *arg) |
paul@181 | 39 | { |
paul@181 | 40 | if ((getuid() != geteuid()) || (getgid() != getegid())) |
paul@181 | 41 | return NULL; |
paul@181 | 42 | #if HAVE_PRCTL |
paul@181 | 43 | if (prctl(PR_GET_DUMPABLE, 0, 0, 0, 0) == 0) |
paul@181 | 44 | return NULL; |
paul@181 | 45 | #else |
paul@181 | 46 | #if (defined(linux) && defined(SYS_prctl)) |
paul@181 | 47 | if (syscall(SYS_prctl, PR_GET_DUMPABLE, 0, 0, 0, 0) == 0) |
paul@181 | 48 | return NULL; |
paul@181 | 49 | #endif |
paul@181 | 50 | #endif |
paul@181 | 51 | |
paul@181 | 52 | #if defined(HAVE_SECURE_GETENV) |
paul@181 | 53 | return secure_getenv(arg); |
paul@181 | 54 | #elif defined(HAVE___SECURE_GETENV) |
paul@181 | 55 | return __secure_getenv(arg); |
paul@181 | 56 | #else |
paul@181 | 57 | return getenv(arg); |
paul@181 | 58 | #endif |
paul@181 | 59 | } |
paul@181 | 60 | |
paul@181 | 61 | #if 0 /* ifdef CONFIG_BLKID_DEBUG */ |
paul@181 | 62 | static blkid_debug_dump_cache(int mask, blkid_cache cache) |
paul@181 | 63 | { |
paul@181 | 64 | struct list_head *p; |
paul@181 | 65 | |
paul@181 | 66 | if (!cache) { |
paul@181 | 67 | printf("cache: NULL\n"); |
paul@181 | 68 | return; |
paul@181 | 69 | } |
paul@181 | 70 | |
paul@181 | 71 | printf("cache: time = %lu\n", cache->bic_time); |
paul@181 | 72 | printf("cache: flags = 0x%08X\n", cache->bic_flags); |
paul@181 | 73 | |
paul@181 | 74 | list_for_each(p, &cache->bic_devs) { |
paul@181 | 75 | blkid_dev dev = list_entry(p, struct blkid_struct_dev, bid_devs); |
paul@181 | 76 | blkid_debug_dump_dev(dev); |
paul@181 | 77 | } |
paul@181 | 78 | } |
paul@181 | 79 | #endif |
paul@181 | 80 | |
paul@181 | 81 | int blkid_get_cache(blkid_cache *ret_cache, const char *filename) |
paul@181 | 82 | { |
paul@181 | 83 | blkid_cache cache; |
paul@181 | 84 | |
paul@181 | 85 | #ifdef CONFIG_BLKID_DEBUG |
paul@181 | 86 | if (!(blkid_debug_mask & DEBUG_INIT)) { |
paul@181 | 87 | char *dstr = getenv("BLKID_DEBUG"); |
paul@181 | 88 | |
paul@181 | 89 | if (dstr) |
paul@181 | 90 | blkid_debug_mask = strtoul(dstr, 0, 0); |
paul@181 | 91 | blkid_debug_mask |= DEBUG_INIT; |
paul@181 | 92 | } |
paul@181 | 93 | #endif |
paul@181 | 94 | |
paul@181 | 95 | DBG(DEBUG_CACHE, printf("creating blkid cache (using %s)\n", |
paul@181 | 96 | filename ? filename : "default cache")); |
paul@181 | 97 | |
paul@181 | 98 | if (!(cache = (blkid_cache) calloc(1, sizeof(struct blkid_struct_cache)))) |
paul@181 | 99 | return -BLKID_ERR_MEM; |
paul@181 | 100 | |
paul@181 | 101 | INIT_LIST_HEAD(&cache->bic_devs); |
paul@181 | 102 | INIT_LIST_HEAD(&cache->bic_tags); |
paul@181 | 103 | |
paul@181 | 104 | if (filename && !strlen(filename)) |
paul@181 | 105 | filename = 0; |
paul@181 | 106 | if (!filename) |
paul@181 | 107 | filename = safe_getenv("BLKID_FILE"); |
paul@181 | 108 | if (!filename) |
paul@181 | 109 | filename = BLKID_CACHE_FILE; |
paul@181 | 110 | cache->bic_filename = blkid_strdup(filename); |
paul@181 | 111 | |
paul@181 | 112 | blkid_read_cache(cache); |
paul@181 | 113 | |
paul@181 | 114 | *ret_cache = cache; |
paul@181 | 115 | return 0; |
paul@181 | 116 | } |
paul@181 | 117 | |
paul@181 | 118 | void blkid_put_cache(blkid_cache cache) |
paul@181 | 119 | { |
paul@181 | 120 | if (!cache) |
paul@181 | 121 | return; |
paul@181 | 122 | |
paul@181 | 123 | (void) blkid_flush_cache(cache); |
paul@181 | 124 | |
paul@181 | 125 | DBG(DEBUG_CACHE, printf("freeing cache struct\n")); |
paul@181 | 126 | |
paul@181 | 127 | /* DBG(DEBUG_CACHE, blkid_debug_dump_cache(cache)); */ |
paul@181 | 128 | |
paul@181 | 129 | while (!list_empty(&cache->bic_devs)) { |
paul@181 | 130 | blkid_dev dev = list_entry(cache->bic_devs.next, |
paul@181 | 131 | struct blkid_struct_dev, |
paul@181 | 132 | bid_devs); |
paul@181 | 133 | blkid_free_dev(dev); |
paul@181 | 134 | } |
paul@181 | 135 | |
paul@181 | 136 | while (!list_empty(&cache->bic_tags)) { |
paul@181 | 137 | blkid_tag tag = list_entry(cache->bic_tags.next, |
paul@181 | 138 | struct blkid_struct_tag, |
paul@181 | 139 | bit_tags); |
paul@181 | 140 | |
paul@181 | 141 | while (!list_empty(&tag->bit_names)) { |
paul@181 | 142 | blkid_tag bad = list_entry(tag->bit_names.next, |
paul@181 | 143 | struct blkid_struct_tag, |
paul@181 | 144 | bit_names); |
paul@181 | 145 | |
paul@181 | 146 | DBG(DEBUG_CACHE, printf("warning: unfreed tag %s=%s\n", |
paul@181 | 147 | bad->bit_name, bad->bit_val)); |
paul@181 | 148 | blkid_free_tag(bad); |
paul@181 | 149 | } |
paul@181 | 150 | blkid_free_tag(tag); |
paul@181 | 151 | } |
paul@181 | 152 | free(cache->bic_filename); |
paul@181 | 153 | |
paul@181 | 154 | free(cache); |
paul@181 | 155 | } |
paul@181 | 156 | |
paul@181 | 157 | void blkid_gc_cache(blkid_cache cache) |
paul@181 | 158 | { |
paul@181 | 159 | struct list_head *p, *pnext; |
paul@181 | 160 | struct stat st; |
paul@181 | 161 | |
paul@181 | 162 | if (!cache) |
paul@181 | 163 | return; |
paul@181 | 164 | |
paul@181 | 165 | list_for_each_safe(p, pnext, &cache->bic_devs) { |
paul@181 | 166 | blkid_dev dev = list_entry(p, struct blkid_struct_dev, bid_devs); |
paul@181 | 167 | if (stat(dev->bid_name, &st) < 0) { |
paul@181 | 168 | DBG(DEBUG_CACHE, |
paul@181 | 169 | printf("freeing %s\n", dev->bid_name)); |
paul@181 | 170 | blkid_free_dev(dev); |
paul@181 | 171 | cache->bic_flags |= BLKID_BIC_FL_CHANGED; |
paul@181 | 172 | } else { |
paul@181 | 173 | DBG(DEBUG_CACHE, |
paul@181 | 174 | printf("Device %s exists\n", dev->bid_name)); |
paul@181 | 175 | } |
paul@181 | 176 | } |
paul@181 | 177 | } |
paul@181 | 178 | |
paul@181 | 179 | |
paul@181 | 180 | #ifdef TEST_PROGRAM |
paul@181 | 181 | int main(int argc, char** argv) |
paul@181 | 182 | { |
paul@181 | 183 | blkid_cache cache = NULL; |
paul@181 | 184 | int ret; |
paul@181 | 185 | |
paul@181 | 186 | blkid_debug_mask = DEBUG_ALL; |
paul@181 | 187 | if ((argc > 2)) { |
paul@181 | 188 | fprintf(stderr, "Usage: %s [filename] \n", argv[0]); |
paul@181 | 189 | exit(1); |
paul@181 | 190 | } |
paul@181 | 191 | |
paul@181 | 192 | if ((ret = blkid_get_cache(&cache, argv[1])) < 0) { |
paul@181 | 193 | fprintf(stderr, "error %d parsing cache file %s\n", ret, |
paul@181 | 194 | argv[1] ? argv[1] : BLKID_CACHE_FILE); |
paul@181 | 195 | exit(1); |
paul@181 | 196 | } |
paul@181 | 197 | if ((ret = blkid_get_cache(&cache, "/dev/null")) != 0) { |
paul@181 | 198 | fprintf(stderr, "%s: error creating cache (%d)\n", |
paul@181 | 199 | argv[0], ret); |
paul@181 | 200 | exit(1); |
paul@181 | 201 | } |
paul@181 | 202 | if ((ret = blkid_probe_all(cache) < 0)) |
paul@181 | 203 | fprintf(stderr, "error probing devices\n"); |
paul@181 | 204 | |
paul@181 | 205 | blkid_put_cache(cache); |
paul@181 | 206 | |
paul@181 | 207 | return ret; |
paul@181 | 208 | } |
paul@181 | 209 | #endif |