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