| 1 | #include <stdlib.h> |
| 2 | #include <assert.h> |
| 3 | |
| 4 | #include "fio.h" |
| 5 | #include "flist.h" |
| 6 | #include "crc/crc16.h" |
| 7 | |
| 8 | #define HASH_BUCKETS 512 |
| 9 | #define HASH_MASK (HASH_BUCKETS - 1) |
| 10 | |
| 11 | unsigned int file_hash_size = HASH_BUCKETS * sizeof(struct flist_head); |
| 12 | |
| 13 | static struct flist_head *file_hash; |
| 14 | static struct fio_mutex *hash_lock; |
| 15 | |
| 16 | static unsigned short hash(const char *name) |
| 17 | { |
| 18 | return crc16((const unsigned char *) name, strlen(name)) & HASH_MASK; |
| 19 | } |
| 20 | |
| 21 | void remove_file_hash(struct fio_file *f) |
| 22 | { |
| 23 | fio_mutex_down(hash_lock); |
| 24 | |
| 25 | if (fio_file_hashed(f)) { |
| 26 | assert(!flist_empty(&f->hash_list)); |
| 27 | flist_del_init(&f->hash_list); |
| 28 | fio_file_clear_hashed(f); |
| 29 | } |
| 30 | |
| 31 | fio_mutex_up(hash_lock); |
| 32 | } |
| 33 | |
| 34 | static struct fio_file *__lookup_file_hash(const char *name) |
| 35 | { |
| 36 | struct flist_head *bucket = &file_hash[hash(name)]; |
| 37 | struct flist_head *n; |
| 38 | |
| 39 | flist_for_each(n, bucket) { |
| 40 | struct fio_file *f = flist_entry(n, struct fio_file, hash_list); |
| 41 | |
| 42 | if (!f->file_name) |
| 43 | continue; |
| 44 | |
| 45 | if (!strcmp(f->file_name, name)) { |
| 46 | assert(f->fd != -1); |
| 47 | return f; |
| 48 | } |
| 49 | } |
| 50 | |
| 51 | return NULL; |
| 52 | } |
| 53 | |
| 54 | struct fio_file *lookup_file_hash(const char *name) |
| 55 | { |
| 56 | struct fio_file *f; |
| 57 | |
| 58 | fio_mutex_down(hash_lock); |
| 59 | f = __lookup_file_hash(name); |
| 60 | fio_mutex_up(hash_lock); |
| 61 | return f; |
| 62 | } |
| 63 | |
| 64 | struct fio_file *add_file_hash(struct fio_file *f) |
| 65 | { |
| 66 | struct fio_file *alias; |
| 67 | |
| 68 | if (fio_file_hashed(f)) |
| 69 | return NULL; |
| 70 | |
| 71 | INIT_FLIST_HEAD(&f->hash_list); |
| 72 | |
| 73 | fio_mutex_down(hash_lock); |
| 74 | |
| 75 | alias = __lookup_file_hash(f->file_name); |
| 76 | if (!alias) { |
| 77 | fio_file_set_hashed(f); |
| 78 | flist_add_tail(&f->hash_list, &file_hash[hash(f->file_name)]); |
| 79 | } |
| 80 | |
| 81 | fio_mutex_up(hash_lock); |
| 82 | return alias; |
| 83 | } |
| 84 | |
| 85 | void file_hash_exit(void) |
| 86 | { |
| 87 | unsigned int i, has_entries = 0; |
| 88 | |
| 89 | fio_mutex_down(hash_lock); |
| 90 | for (i = 0; i < HASH_BUCKETS; i++) |
| 91 | has_entries += !flist_empty(&file_hash[i]); |
| 92 | fio_mutex_up(hash_lock); |
| 93 | |
| 94 | if (has_entries) |
| 95 | log_err("fio: file hash not empty on exit\n"); |
| 96 | |
| 97 | file_hash = NULL; |
| 98 | fio_mutex_remove(hash_lock); |
| 99 | hash_lock = NULL; |
| 100 | } |
| 101 | |
| 102 | void file_hash_init(void *ptr) |
| 103 | { |
| 104 | unsigned int i; |
| 105 | |
| 106 | file_hash = ptr; |
| 107 | for (i = 0; i < HASH_BUCKETS; i++) |
| 108 | INIT_FLIST_HEAD(&file_hash[i]); |
| 109 | |
| 110 | hash_lock = fio_mutex_init(1); |
| 111 | } |