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
__KHASH_TYPE(off,git_off_t,void *)17 __KHASH_TYPE(off, git_off_t, void *)
18
19 __KHASH_IMPL(off, static kh_inline, git_off_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
git_offmap_free(git_offmap * map)30 void git_offmap_free(git_offmap *map)
31 {
32 kh_destroy(off, map);
33 }
34
git_offmap_clear(git_offmap * map)35 void git_offmap_clear(git_offmap *map)
36 {
37 kh_clear(off, map);
38 }
39
git_offmap_size(git_offmap * map)40 size_t git_offmap_size(git_offmap *map)
41 {
42 return kh_size(map);
43 }
44
git_offmap_get(git_offmap * map,const git_off_t key)45 void *git_offmap_get(git_offmap *map, const git_off_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
git_offmap_set(git_offmap * map,const git_off_t key,void * value)53 int git_offmap_set(git_offmap *map, const git_off_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
git_offmap_delete(git_offmap * map,const git_off_t key)70 int git_offmap_delete(git_offmap *map, const git_off_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
git_offmap_exists(git_offmap * map,const git_off_t key)79 int git_offmap_exists(git_offmap *map, const git_off_t key)
80 {
81 return kh_get(off, map, key) != kh_end(map);
82 }
83
git_offmap_iterate(void ** value,git_offmap * map,size_t * iter,git_off_t * key)84 int git_offmap_iterate(void **value, git_offmap *map, size_t *iter, git_off_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 }
102