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