]>
Commit | Line | Data |
---|---|---|
64b56afd AC |
1 | /* |
2 | * Copyright (c) 2012 William Pitcock <nenolod@dereferenced.org>. | |
3 | * | |
4 | * Permission to use, copy, modify, and/or distribute this software for any | |
5 | * purpose with or without fee is hereby granted, provided that the above | |
6 | * copyright notice and this permission notice is present in all copies. | |
7 | * | |
8 | * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR | |
9 | * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED | |
10 | * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE | |
11 | * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, | |
12 | * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES | |
13 | * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR | |
14 | * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) | |
15 | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, | |
16 | * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING | |
17 | * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE | |
18 | * POSSIBILITY OF SUCH DAMAGE. | |
19 | */ | |
20 | ||
21 | #include "stdinc.h" | |
22 | #include "capability.h" | |
23 | #include "irc_dictionary.h" | |
24 | ||
25 | struct CapabilityIndex { | |
26 | struct Dictionary *cap_dict; | |
27 | unsigned int highest_bit; | |
28 | }; | |
29 | ||
5058c8eb | 30 | #define CAP_ORPHANED 0x1 |
f01f67f0 | 31 | #define CAP_REQUIRED 0x2 |
5058c8eb | 32 | |
64b56afd | 33 | struct CapabilityEntry { |
e679e389 | 34 | char *cap; |
64b56afd | 35 | unsigned int value; |
5058c8eb | 36 | unsigned int flags; |
64b56afd AC |
37 | }; |
38 | ||
39 | unsigned int | |
40 | capability_get(struct CapabilityIndex *index, const char *cap) | |
41 | { | |
42 | struct CapabilityEntry *entry; | |
43 | ||
44 | s_assert(index != NULL); | |
45 | ||
46 | entry = irc_dictionary_retrieve(index->cap_dict, cap); | |
5058c8eb | 47 | if (entry != NULL && !(entry->flags & CAP_ORPHANED)) |
8bedf01d | 48 | return (1 << entry->value); |
64b56afd AC |
49 | |
50 | return 0xFFFFFFFF; | |
51 | } | |
52 | ||
53 | unsigned int | |
54 | capability_put(struct CapabilityIndex *index, const char *cap) | |
55 | { | |
56 | struct CapabilityEntry *entry; | |
57 | ||
58 | s_assert(index != NULL); | |
8bedf01d AC |
59 | if (!index->highest_bit) |
60 | return 0xFFFFFFFF; | |
64b56afd AC |
61 | |
62 | if ((entry = irc_dictionary_retrieve(index->cap_dict, cap)) != NULL) | |
63 | { | |
5058c8eb | 64 | entry->flags &= ~CAP_ORPHANED; |
8bedf01d | 65 | return (1 << entry->value); |
64b56afd AC |
66 | } |
67 | ||
68 | entry = rb_malloc(sizeof(struct CapabilityEntry)); | |
e679e389 | 69 | entry->cap = rb_strdup(cap); |
5058c8eb | 70 | entry->flags = 0; |
64b56afd AC |
71 | entry->value = index->highest_bit; |
72 | ||
e679e389 | 73 | irc_dictionary_add(index->cap_dict, entry->cap, entry); |
64b56afd | 74 | |
8bedf01d AC |
75 | index->highest_bit++; |
76 | if (index->highest_bit % (sizeof(unsigned int) * 8) == 0) | |
77 | index->highest_bit = 0; | |
64b56afd | 78 | |
8bedf01d | 79 | return (1 << entry->value); |
64b56afd AC |
80 | } |
81 | ||
82 | void | |
83 | capability_orphan(struct CapabilityIndex *index, const char *cap) | |
84 | { | |
85 | struct CapabilityEntry *entry; | |
86 | ||
87 | s_assert(index != NULL); | |
88 | ||
89 | entry = irc_dictionary_retrieve(index->cap_dict, cap); | |
90 | if (entry != NULL) | |
f01f67f0 AC |
91 | { |
92 | entry->flags &= ~CAP_REQUIRED; | |
5058c8eb | 93 | entry->flags |= CAP_ORPHANED; |
f01f67f0 AC |
94 | } |
95 | } | |
96 | ||
97 | void | |
98 | capability_require(struct CapabilityIndex *index, const char *cap) | |
99 | { | |
100 | struct CapabilityEntry *entry; | |
101 | ||
102 | s_assert(index != NULL); | |
103 | ||
104 | entry = irc_dictionary_retrieve(index->cap_dict, cap); | |
105 | if (entry != NULL) | |
106 | entry->flags |= CAP_REQUIRED; | |
64b56afd AC |
107 | } |
108 | ||
109 | static void | |
110 | capability_destroy(struct DictionaryElement *delem, void *privdata) | |
111 | { | |
112 | s_assert(delem != NULL); | |
113 | ||
114 | rb_free(delem->data); | |
115 | } | |
116 | ||
117 | struct CapabilityIndex * | |
118 | capability_index_create(void) | |
119 | { | |
120 | struct CapabilityIndex *index; | |
121 | ||
122 | index = rb_malloc(sizeof(struct CapabilityIndex)); | |
123 | index->cap_dict = irc_dictionary_create(strcasecmp); | |
124 | index->highest_bit = 1; | |
125 | ||
126 | return index; | |
127 | } | |
128 | ||
129 | void | |
130 | capability_index_destroy(struct CapabilityIndex *index) | |
131 | { | |
132 | s_assert(index != NULL); | |
133 | ||
134 | irc_dictionary_destroy(index->cap_dict, capability_destroy, NULL); | |
135 | rb_free(index); | |
136 | } | |
5e773521 AC |
137 | |
138 | const char * | |
139 | capability_index_list(struct CapabilityIndex *index, unsigned int cap_mask) | |
140 | { | |
141 | struct DictionaryIter iter; | |
142 | struct CapabilityEntry *entry; | |
143 | static char buf[BUFSIZE]; | |
144 | char *t = buf; | |
145 | int tl; | |
146 | ||
147 | s_assert(index != NULL); | |
148 | ||
149 | *t = '\0'; | |
150 | ||
151 | DICTIONARY_FOREACH(entry, &iter, index->cap_dict) | |
152 | { | |
153 | if (entry->value & cap_mask) | |
154 | { | |
e915e51f | 155 | tl = rb_sprintf(t, "%s ", entry->cap); |
5e773521 AC |
156 | t += tl; |
157 | } | |
158 | } | |
159 | ||
160 | t--; | |
161 | *t = '\0'; | |
162 | ||
163 | return buf; | |
164 | } | |
ec3a9055 AC |
165 | |
166 | unsigned int | |
167 | capability_index_mask(struct CapabilityIndex *index) | |
168 | { | |
169 | struct DictionaryIter iter; | |
170 | struct CapabilityEntry *entry; | |
171 | unsigned int mask = 0; | |
172 | ||
173 | s_assert(index != NULL); | |
174 | ||
175 | DICTIONARY_FOREACH(entry, &iter, index->cap_dict) | |
176 | { | |
0582290f | 177 | if (!(entry->flags & CAP_ORPHANED)) |
8bedf01d | 178 | mask |= (1 << entry->value); |
ec3a9055 AC |
179 | } |
180 | ||
181 | return mask; | |
182 | } | |
f01f67f0 AC |
183 | |
184 | unsigned int | |
185 | capability_index_get_required(struct CapabilityIndex *index) | |
186 | { | |
187 | struct DictionaryIter iter; | |
188 | struct CapabilityEntry *entry; | |
189 | unsigned int mask = 0; | |
190 | ||
191 | s_assert(index != NULL); | |
192 | ||
193 | DICTIONARY_FOREACH(entry, &iter, index->cap_dict) | |
194 | { | |
195 | if (!(entry->flags & CAP_ORPHANED) && (entry->flags & CAP_REQUIRED)) | |
8bedf01d | 196 | mask |= (1 << entry->value); |
f01f67f0 AC |
197 | } |
198 | ||
199 | return mask; | |
200 | } |