src/offmap.c

   1 /*
   2  * Copyright (C) the libgit2 contributors. All rights reserved.
   3  *
   4  * This file is part of libgit2, distributed under the GNU GPL v2 with
   5  * a Linking Exception. For full terms see the included COPYING file.
   6  */
   7
   8 #include "offmap.h"
   9
  10 #define kmalloc git__malloc
  11 #define kcalloc git__calloc
  12 #define krealloc git__realloc
  13 #define kreallocarray git__reallocarray
  14 #define kfree git__free
  15 #include "khash.h"
  16
  17 __KHASH_TYPE(off, off64_t, void *)
  18
  19 __KHASH_IMPL(off, static kh_inline, off64_t, void *, 1, kh_int64_hash_func, kh_int64_hash_equal)
  20
  21
  22 int git_offmap_new(git_offmap **out)
  23 {
  24         *out = kh_init(off);
  25         GIT_ERROR_CHECK_ALLOC(*out);
  26
  27         return 0;
  28 }
  29
  30 void git_offmap_free(git_offmap *map)
  31 {
  32         kh_destroy(off, map);
  33 }
  34
  35 void git_offmap_clear(git_offmap *map)
  36 {
  37         kh_clear(off, map);
  38 }
  39
  40 size_t git_offmap_size(git_offmap *map)
  41 {
  42         return kh_size(map);
  43 }
  44
  45 void *git_offmap_get(git_offmap *map, const off64_t key)
  46 {
  47         size_t idx = kh_get(off, map, key);
  48         if (idx == kh_end(map) || !kh_exist(map, idx))
  49                 return NULL;
  50         return kh_val(map, idx);
  51 }
  52
  53 int git_offmap_set(git_offmap *map, const off64_t key, void *value)
  54 {
  55         size_t idx;
  56         int rval;
  57
  58         idx = kh_put(off, map, key, &rval);
  59         if (rval < 0)
  60                 return -1;
  61
  62         if (rval == 0)
  63                 kh_key(map, idx) = key;
  64
  65         kh_val(map, idx) = value;
  66
  67         return 0;
  68 }
  69
  70 int git_offmap_delete(git_offmap *map, const off64_t key)
  71 {
  72         khiter_t idx = kh_get(off, map, key);
  73         if (idx == kh_end(map))
  74                 return GIT_ENOTFOUND;
  75         kh_del(off, map, idx);
  76         return 0;
  77 }
  78
  79 int git_offmap_exists(git_offmap *map, const off64_t key)
  80 {
  81         return kh_get(off, map, key) != kh_end(map);
  82 }
  83
  84 int git_offmap_iterate(void **value, git_offmap *map, size_t *iter, off64_t *key)
  85 {
  86         size_t i = *iter;
  87
  88         while (i < map->n_buckets && !kh_exist(map, i))
  89                 i++;
  90
  91         if (i >= map->n_buckets)
  92                 return GIT_ITEROVER;
  93
  94         if (key)
  95                 *key = kh_key(map, i);
  96         if (value)
  97                 *value = kh_value(map, i);
  98         *iter = ++i;
  99
 100         return 0;
 101 }