Blame src/win/fs-fd-hash-inl.h

Packit Service 7c31a4
#ifndef UV_WIN_FS_FD_HASH_INL_H_
Packit Service 7c31a4
#define UV_WIN_FS_FD_HASH_INL_H_
Packit Service 7c31a4
Packit Service 7c31a4
#include "uv.h"
Packit Service 7c31a4
#include "internal.h"
Packit Service 7c31a4
Packit Service 7c31a4
/* Files are only inserted in uv__fd_hash when the UV_FS_O_FILEMAP flag is
Packit Service 7c31a4
 * specified. Thus, when uv__fd_hash_get returns true, the file mapping in the
Packit Service 7c31a4
 * info structure should be used for read/write operations.
Packit Service 7c31a4
 *
Packit Service 7c31a4
 * If the file is empty, the mapping field will be set to
Packit Service 7c31a4
 * INVALID_HANDLE_VALUE. This is not an issue since the file mapping needs to
Packit Service 7c31a4
 * be created anyway when the file size changes.
Packit Service 7c31a4
 *
Packit Service 7c31a4
 * Since file descriptors are sequential integers, the modulo operator is used
Packit Service 7c31a4
 * as hashing function. For each bucket, a single linked list of arrays is
Packit Service 7c31a4
 * kept to minimize allocations. A statically allocated memory buffer is kept
Packit Service 7c31a4
 * for the first array in each bucket. */
Packit Service 7c31a4
Packit Service 7c31a4
Packit Service 7c31a4
#define UV__FD_HASH_SIZE 256
Packit Service 7c31a4
#define UV__FD_HASH_GROUP_SIZE 16
Packit Service 7c31a4
Packit Service 7c31a4
struct uv__fd_info_s {
Packit Service 7c31a4
  int flags;
Packit Service 7c31a4
  BOOLEAN is_directory;
Packit Service 7c31a4
  HANDLE mapping;
Packit Service 7c31a4
  LARGE_INTEGER size;
Packit Service 7c31a4
  LARGE_INTEGER current_pos;
Packit Service 7c31a4
};
Packit Service 7c31a4
Packit Service 7c31a4
struct uv__fd_hash_entry_s {
Packit Service 7c31a4
  uv_file fd;
Packit Service 7c31a4
  struct uv__fd_info_s info;
Packit Service 7c31a4
};
Packit Service 7c31a4
Packit Service 7c31a4
struct uv__fd_hash_entry_group_s {
Packit Service 7c31a4
  struct uv__fd_hash_entry_s entries[UV__FD_HASH_GROUP_SIZE];
Packit Service 7c31a4
  struct uv__fd_hash_entry_group_s* next;
Packit Service 7c31a4
};
Packit Service 7c31a4
Packit Service 7c31a4
struct uv__fd_hash_bucket_s {
Packit Service 7c31a4
  size_t size;
Packit Service 7c31a4
  struct uv__fd_hash_entry_group_s* data;
Packit Service 7c31a4
};
Packit Service 7c31a4
Packit Service 7c31a4
Packit Service 7c31a4
static uv_mutex_t uv__fd_hash_mutex;
Packit Service 7c31a4
Packit Service 7c31a4
static struct uv__fd_hash_entry_group_s
Packit Service 7c31a4
  uv__fd_hash_entry_initial[UV__FD_HASH_SIZE * UV__FD_HASH_GROUP_SIZE];
Packit Service 7c31a4
static struct uv__fd_hash_bucket_s uv__fd_hash[UV__FD_HASH_SIZE];
Packit Service 7c31a4
Packit Service 7c31a4
Packit Service 7c31a4
INLINE static void uv__fd_hash_init(void) {
Packit Service 7c31a4
  int i, err;
Packit Service 7c31a4
Packit Service 7c31a4
  err = uv_mutex_init(&uv__fd_hash_mutex);
Packit Service 7c31a4
  if (err) {
Packit Service 7c31a4
    uv_fatal_error(err, "uv_mutex_init");
Packit Service 7c31a4
  }
Packit Service 7c31a4
Packit Service 7c31a4
  for (i = 0; i < ARRAY_SIZE(uv__fd_hash); ++i) {
Packit Service 7c31a4
    uv__fd_hash[i].size = 0;
Packit Service 7c31a4
    uv__fd_hash[i].data =
Packit Service 7c31a4
        uv__fd_hash_entry_initial + i * UV__FD_HASH_GROUP_SIZE;
Packit Service 7c31a4
  }
Packit Service 7c31a4
}
Packit Service 7c31a4
Packit Service 7c31a4
#define FIND_COMMON_VARIABLES                                                \
Packit Service 7c31a4
  unsigned i;                                                                \
Packit Service 7c31a4
  unsigned bucket = fd % ARRAY_SIZE(uv__fd_hash);                            \
Packit Service 7c31a4
  struct uv__fd_hash_entry_s* entry_ptr = NULL;                              \
Packit Service 7c31a4
  struct uv__fd_hash_entry_group_s* group_ptr;                               \
Packit Service 7c31a4
  struct uv__fd_hash_bucket_s* bucket_ptr = &uv__fd_hash[bucket];
Packit Service 7c31a4
Packit Service 7c31a4
#define FIND_IN_GROUP_PTR(group_size)                                        \
Packit Service 7c31a4
  do {                                                                       \
Packit Service 7c31a4
    for (i = 0; i < group_size; ++i) {                                       \
Packit Service 7c31a4
      if (group_ptr->entries[i].fd == fd) {                                  \
Packit Service 7c31a4
        entry_ptr = &group_ptr->entries[i];                                  \
Packit Service 7c31a4
        break;                                                               \
Packit Service 7c31a4
      }                                                                      \
Packit Service 7c31a4
    }                                                                        \
Packit Service 7c31a4
  } while (0)
Packit Service 7c31a4
Packit Service 7c31a4
#define FIND_IN_BUCKET_PTR()                                                 \
Packit Service 7c31a4
  do {                                                                       \
Packit Service 7c31a4
    size_t first_group_size = bucket_ptr->size % UV__FD_HASH_GROUP_SIZE;     \
Packit Service 7c31a4
    if (bucket_ptr->size != 0 && first_group_size == 0)                      \
Packit Service 7c31a4
      first_group_size = UV__FD_HASH_GROUP_SIZE;                             \
Packit Service 7c31a4
    group_ptr = bucket_ptr->data;                                            \
Packit Service 7c31a4
    FIND_IN_GROUP_PTR(first_group_size);                                     \
Packit Service 7c31a4
    for (group_ptr = group_ptr->next;                                        \
Packit Service 7c31a4
         group_ptr != NULL && entry_ptr == NULL;                             \
Packit Service 7c31a4
         group_ptr = group_ptr->next)                                        \
Packit Service 7c31a4
      FIND_IN_GROUP_PTR(UV__FD_HASH_GROUP_SIZE);                             \
Packit Service 7c31a4
  } while (0)
Packit Service 7c31a4
Packit Service 7c31a4
INLINE static int uv__fd_hash_get(int fd, struct uv__fd_info_s* info) {
Packit Service 7c31a4
  FIND_COMMON_VARIABLES
Packit Service 7c31a4
Packit Service 7c31a4
  uv_mutex_lock(&uv__fd_hash_mutex);
Packit Service 7c31a4
Packit Service 7c31a4
  FIND_IN_BUCKET_PTR();
Packit Service 7c31a4
Packit Service 7c31a4
  if (entry_ptr != NULL) {
Packit Service 7c31a4
    *info = entry_ptr->info;
Packit Service 7c31a4
  }
Packit Service 7c31a4
Packit Service 7c31a4
  uv_mutex_unlock(&uv__fd_hash_mutex);
Packit Service 7c31a4
  return entry_ptr != NULL;
Packit Service 7c31a4
}
Packit Service 7c31a4
Packit Service 7c31a4
INLINE static void uv__fd_hash_add(int fd, struct uv__fd_info_s* info) {
Packit Service 7c31a4
  FIND_COMMON_VARIABLES
Packit Service 7c31a4
Packit Service 7c31a4
  uv_mutex_lock(&uv__fd_hash_mutex);
Packit Service 7c31a4
Packit Service 7c31a4
  FIND_IN_BUCKET_PTR();
Packit Service 7c31a4
Packit Service 7c31a4
  if (entry_ptr == NULL) {
Packit Service 7c31a4
    i = bucket_ptr->size % UV__FD_HASH_GROUP_SIZE;
Packit Service 7c31a4
Packit Service 7c31a4
    if (bucket_ptr->size != 0 && i == 0) {
Packit Service 7c31a4
      struct uv__fd_hash_entry_group_s* new_group_ptr =
Packit Service 7c31a4
        uv__malloc(sizeof(*new_group_ptr));
Packit Service 7c31a4
      if (new_group_ptr == NULL) {
Packit Service 7c31a4
        uv_fatal_error(ERROR_OUTOFMEMORY, "uv__malloc");
Packit Service 7c31a4
      }
Packit Service 7c31a4
      new_group_ptr->next = bucket_ptr->data;
Packit Service 7c31a4
      bucket_ptr->data = new_group_ptr;
Packit Service 7c31a4
    }
Packit Service 7c31a4
Packit Service 7c31a4
    bucket_ptr->size += 1;
Packit Service 7c31a4
    entry_ptr = &bucket_ptr->data->entries[i];
Packit Service 7c31a4
    entry_ptr->fd = fd;
Packit Service 7c31a4
  }
Packit Service 7c31a4
Packit Service 7c31a4
  entry_ptr->info = *info;
Packit Service 7c31a4
Packit Service 7c31a4
  uv_mutex_unlock(&uv__fd_hash_mutex);
Packit Service 7c31a4
}
Packit Service 7c31a4
Packit Service 7c31a4
INLINE static int uv__fd_hash_remove(int fd, struct uv__fd_info_s* info) {
Packit Service 7c31a4
  FIND_COMMON_VARIABLES
Packit Service 7c31a4
Packit Service 7c31a4
  uv_mutex_lock(&uv__fd_hash_mutex);
Packit Service 7c31a4
Packit Service 7c31a4
  FIND_IN_BUCKET_PTR();
Packit Service 7c31a4
Packit Service 7c31a4
  if (entry_ptr != NULL) {
Packit Service 7c31a4
    *info = entry_ptr->info;
Packit Service 7c31a4
Packit Service 7c31a4
    bucket_ptr->size -= 1;
Packit Service 7c31a4
Packit Service 7c31a4
    i = bucket_ptr->size % UV__FD_HASH_GROUP_SIZE;
Packit Service 7c31a4
    if (entry_ptr != &bucket_ptr->data->entries[i]) {
Packit Service 7c31a4
      *entry_ptr = bucket_ptr->data->entries[i];
Packit Service 7c31a4
    }
Packit Service 7c31a4
Packit Service 7c31a4
    if (bucket_ptr->size != 0 &&
Packit Service 7c31a4
        bucket_ptr->size % UV__FD_HASH_GROUP_SIZE == 0) {
Packit Service 7c31a4
      struct uv__fd_hash_entry_group_s* old_group_ptr = bucket_ptr->data;
Packit Service 7c31a4
      bucket_ptr->data = old_group_ptr->next;
Packit Service 7c31a4
      uv__free(old_group_ptr);
Packit Service 7c31a4
    }
Packit Service 7c31a4
  }
Packit Service 7c31a4
Packit Service 7c31a4
  uv_mutex_unlock(&uv__fd_hash_mutex);
Packit Service 7c31a4
  return entry_ptr != NULL;
Packit Service 7c31a4
}
Packit Service 7c31a4
Packit Service 7c31a4
#undef FIND_COMMON_VARIABLES
Packit Service 7c31a4
#undef FIND_IN_GROUP_PTR
Packit Service 7c31a4
#undef FIND_IN_BUCKET_PTR
Packit Service 7c31a4
Packit Service 7c31a4
#endif /* UV_WIN_FS_FD_HASH_INL_H_ */