filehash.c

   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 }