genfio: Splitting gen_template in 2 parts
[fio.git] / filehash.c
1 #include <stdlib.h>
2 #include <assert.h>
3
4 #include "fio.h"
5 #include "flist.h"
6 #include "hash.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 jhash(name, strlen(name), 0) & 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(FIO_MUTEX_UNLOCKED);
111 }