src/idxmap.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 "idxmap.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(idx, const git_index_entry *, git_index_entry *)
  18 __KHASH_TYPE(idxicase, const git_index_entry *, git_index_entry *)
  19
  20 /* This is __ac_X31_hash_string but with tolower and it takes the entry's stage into account */
  21 static kh_inline khint_t idxentry_hash(const git_index_entry *e)
  22 {
  23         const char *s = e->path;
  24         khint_t h = (khint_t)git__tolower(*s);
  25         if (h) for (++s ; *s; ++s) h = (h << 5) - h + (khint_t)git__tolower(*s);
  26         return h + GIT_INDEX_ENTRY_STAGE(e);
  27 }
  28
  29 #define idxentry_equal(a, b) (GIT_INDEX_ENTRY_STAGE(a) == GIT_INDEX_ENTRY_STAGE(b) && strcmp(a->path, b->path) == 0)
  30 #define idxentry_icase_equal(a, b) (GIT_INDEX_ENTRY_STAGE(a) == GIT_INDEX_ENTRY_STAGE(b) && strcasecmp(a->path, b->path) == 0)
  31
  32 __KHASH_IMPL(idx, static kh_inline, const git_index_entry *, git_index_entry *, 1, idxentry_hash, idxentry_equal)
  33 __KHASH_IMPL(idxicase, static kh_inline, const git_index_entry *, git_index_entry *, 1, idxentry_hash, idxentry_icase_equal)
  34
  35 int git_idxmap_new(git_idxmap **out)
  36 {
  37         *out = kh_init(idx);
  38         GIT_ERROR_CHECK_ALLOC(*out);
  39
  40         return 0;
  41 }
  42
  43 int git_idxmap_icase_new(git_idxmap_icase **out)
  44 {
  45         *out = kh_init(idxicase);
  46         GIT_ERROR_CHECK_ALLOC(*out);
  47
  48         return 0;
  49 }
  50
  51 void git_idxmap_free(git_idxmap *map)
  52 {
  53         kh_destroy(idx, map);
  54 }
  55
  56 void git_idxmap_icase_free(git_idxmap_icase *map)
  57 {
  58         kh_destroy(idxicase, map);
  59 }
  60
  61 void git_idxmap_clear(git_idxmap *map)
  62 {
  63         kh_clear(idx, map);
  64 }
  65
  66 void git_idxmap_icase_clear(git_idxmap_icase *map)
  67 {
  68         kh_clear(idxicase, map);
  69 }
  70
  71 int git_idxmap_resize(git_idxmap *map, size_t size)
  72 {
  73         if (!git__is_uint32(size) ||
  74             kh_resize(idx, map, (khiter_t)size) < 0) {
  75                 git_error_set_oom();
  76                 return -1;
  77         }
  78         return 0;
  79 }
  80
  81 int git_idxmap_icase_resize(git_idxmap_icase *map, size_t size)
  82 {
  83         if (!git__is_uint32(size) ||
  84             kh_resize(idxicase, map, (khiter_t)size) < 0) {
  85                 git_error_set_oom();
  86                 return -1;
  87         }
  88         return 0;
  89 }
  90
  91 void *git_idxmap_get(git_idxmap *map, const git_index_entry *key)
  92 {
  93         size_t idx = kh_get(idx, map, key);
  94         if (idx == kh_end(map) || !kh_exist(map, idx))
  95                 return NULL;
  96         return kh_val(map, idx);
  97 }
  98
  99 int git_idxmap_set(git_idxmap *map, const git_index_entry *key, void *value)
 100 {
 101         size_t idx;
 102         int rval;
 103
 104         idx = kh_put(idx, map, key, &rval);
 105         if (rval < 0)
 106                 return -1;
 107
 108         if (rval == 0)
 109                 kh_key(map, idx) = key;
 110
 111         kh_val(map, idx) = value;
 112
 113         return 0;
 114 }
 115
 116 int git_idxmap_icase_set(git_idxmap_icase *map, const git_index_entry *key, void *value)
 117 {
 118         size_t idx;
 119         int rval;
 120
 121         idx = kh_put(idxicase, map, key, &rval);
 122         if (rval < 0)
 123                 return -1;
 124
 125         if (rval == 0)
 126                 kh_key(map, idx) = key;
 127
 128         kh_val(map, idx) = value;
 129
 130         return 0;
 131 }
 132
 133 void *git_idxmap_icase_get(git_idxmap_icase *map, const git_index_entry *key)
 134 {
 135         size_t idx = kh_get(idxicase, map, key);
 136         if (idx == kh_end(map) || !kh_exist(map, idx))
 137                 return NULL;
 138         return kh_val(map, idx);
 139 }
 140
 141 int git_idxmap_delete(git_idxmap *map, const git_index_entry *key)
 142 {
 143         khiter_t idx = kh_get(idx, map, key);
 144         if (idx == kh_end(map))
 145                 return GIT_ENOTFOUND;
 146         kh_del(idx, map, idx);
 147         return 0;
 148 }
 149
 150 int git_idxmap_icase_delete(git_idxmap_icase *map, const git_index_entry *key)
 151 {
 152         khiter_t idx = kh_get(idxicase, map, key);
 153         if (idx == kh_end(map))
 154                 return GIT_ENOTFOUND;
 155         kh_del(idxicase, map, idx);
 156         return 0;
 157 }