src/oidmap.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 "oidmap.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(oid, const git_oid *, void *)
  18
  19 GIT_INLINE(khint_t) git_oidmap_hash(const git_oid *oid)
  20 {
  21         khint_t h;
  22         memcpy(&h, oid, sizeof(khint_t));
  23         return h;
  24 }
  25
  26 __KHASH_IMPL(oid, static kh_inline, const git_oid *, void *, 1, git_oidmap_hash, git_oid_equal)
  27
  28 int git_oidmap_new(git_oidmap **out)
  29 {
  30         *out = kh_init(oid);
  31         GIT_ERROR_CHECK_ALLOC(*out);
  32
  33         return 0;
  34 }
  35
  36 void git_oidmap_free(git_oidmap *map)
  37 {
  38         kh_destroy(oid, map);
  39 }
  40
  41 void git_oidmap_clear(git_oidmap *map)
  42 {
  43         kh_clear(oid, map);
  44 }
  45
  46 size_t git_oidmap_size(git_oidmap *map)
  47 {
  48         return kh_size(map);
  49 }
  50
  51 void *git_oidmap_get(git_oidmap *map, const git_oid *key)
  52 {
  53         size_t idx = kh_get(oid, map, key);
  54         if (idx == kh_end(map) || !kh_exist(map, idx))
  55                 return NULL;
  56         return kh_val(map, idx);
  57 }
  58
  59 int git_oidmap_set(git_oidmap *map, const git_oid *key, void *value)
  60 {
  61         size_t idx;
  62         int rval;
  63
  64         idx = kh_put(oid, map, key, &rval);
  65         if (rval < 0)
  66                 return -1;
  67
  68         if (rval == 0)
  69                 kh_key(map, idx) = key;
  70
  71         kh_val(map, idx) = value;
  72
  73         return 0;
  74 }
  75
  76 int git_oidmap_delete(git_oidmap *map, const git_oid *key)
  77 {
  78         khiter_t idx = kh_get(oid, map, key);
  79         if (idx == kh_end(map))
  80                 return GIT_ENOTFOUND;
  81         kh_del(oid, map, idx);
  82         return 0;
  83 }
  84
  85 int git_oidmap_exists(git_oidmap *map, const git_oid *key)
  86 {
  87         return kh_get(oid, map, key) != kh_end(map);
  88 }
  89
  90 int git_oidmap_iterate(void **value, git_oidmap *map, size_t *iter, const git_oid **key)
  91 {
  92         size_t i = *iter;
  93
  94         while (i < map->n_buckets && !kh_exist(map, i))
  95                 i++;
  96
  97         if (i >= map->n_buckets)
  98                 return GIT_ITEROVER;
  99
 100         if (key)
 101                 *key = kh_key(map, i);
 102         if (value)
 103                 *value = kh_value(map, i);
 104         *iter = ++i;
 105
 106         return 0;
 107 }