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
__KHASH_TYPE(idx,const git_index_entry *,git_index_entry *)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
git_idxmap_new(git_idxmap ** out)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
git_idxmap_icase_new(git_idxmap_icase ** out)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
git_idxmap_free(git_idxmap * map)51 void git_idxmap_free(git_idxmap *map)
52 {
53 kh_destroy(idx, map);
54 }
55
git_idxmap_icase_free(git_idxmap_icase * map)56 void git_idxmap_icase_free(git_idxmap_icase *map)
57 {
58 kh_destroy(idxicase, map);
59 }
60
git_idxmap_clear(git_idxmap * map)61 void git_idxmap_clear(git_idxmap *map)
62 {
63 kh_clear(idx, map);
64 }
65
git_idxmap_icase_clear(git_idxmap_icase * map)66 void git_idxmap_icase_clear(git_idxmap_icase *map)
67 {
68 kh_clear(idxicase, map);
69 }
70
git_idxmap_resize(git_idxmap * map,size_t size)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
git_idxmap_icase_resize(git_idxmap_icase * map,size_t size)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
git_idxmap_get(git_idxmap * map,const git_index_entry * key)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
git_idxmap_set(git_idxmap * map,const git_index_entry * key,void * value)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
git_idxmap_icase_set(git_idxmap_icase * map,const git_index_entry * key,void * value)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
git_idxmap_icase_get(git_idxmap_icase * map,const git_index_entry * key)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
git_idxmap_delete(git_idxmap * map,const git_index_entry * key)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
git_idxmap_icase_delete(git_idxmap_icase * map,const git_index_entry * key)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 }
158