filehash.c

   1 #include <stdlib.h>
   2 #include <assert.h>
   3
   4 #include "fio.h"
   5 #include "list.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 list_head);
  12
  13 static struct list_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 (f->flags & FIO_FILE_HASHED) {
  26                 assert(!list_empty(&f->hash_list));
  27                 list_del_init(&f->hash_list);
  28                 f->flags &= ~FIO_FILE_HASHED;
  29         }
  30
  31         fio_mutex_up(hash_lock);
  32 }
  33
  34 static struct fio_file *__lookup_file_hash(const char *name)
  35 {
  36         struct list_head *bucket = &file_hash[hash(name)];
  37         struct list_head *n;
  38
  39         list_for_each(n, bucket) {
  40                 struct fio_file *f = list_entry(n, struct fio_file, hash_list);
  41
  42                 if (!strcmp(f->file_name, name)) {
  43                         assert(f->fd != -1);
  44                         return f;
  45                 }
  46         }
  47
  48         return NULL;
  49 }
  50
  51 struct fio_file *lookup_file_hash(const char *name)
  52 {
  53         struct fio_file *f;
  54
  55         fio_mutex_down(hash_lock);
  56         f = __lookup_file_hash(name);
  57         fio_mutex_up(hash_lock);
  58         return f;
  59 }
  60
  61 struct fio_file *add_file_hash(struct fio_file *f)
  62 {
  63         struct fio_file *alias;
  64
  65         if (f->flags & FIO_FILE_HASHED)
  66                 return NULL;
  67
  68         INIT_LIST_HEAD(&f->hash_list);
  69
  70         fio_mutex_down(hash_lock);
  71
  72         alias = __lookup_file_hash(f->file_name);
  73         if (!alias) {
  74                 f->flags |= FIO_FILE_HASHED;
  75                 list_add_tail(&f->hash_list, &file_hash[hash(f->file_name)]);
  76         }
  77
  78         fio_mutex_up(hash_lock);
  79         return alias;
  80 }
  81
  82 void file_hash_init(void *ptr)
  83 {
  84         unsigned int i;
  85
  86         file_hash = ptr;
  87         for (i = 0; i < HASH_BUCKETS; i++)
  88                 INIT_LIST_HEAD(&file_hash[i]);
  89
  90         hash_lock = fio_mutex_init(1);
  91 }