torture-backupkey: Assert dcerpc_bkrp_BackupKey_r call was successful
[Samba.git] / source3 / lib / idmap_cache.c
blob11bda393d0e41006bbdd0e6a92220f51a2489f53
1 /*
2 Unix SMB/CIFS implementation.
3 ID Mapping Cache
5 Copyright (C) Volker Lendecke 2008
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>.*/
20 #include "includes.h"
21 #include "idmap_cache.h"
22 #include "../libcli/security/security.h"
23 #include "../librpc/gen_ndr/idmap.h"
25 /**
26 * Find a sid2xid mapping
27 * @param[in] sid the sid to map
28 * @param[out] id where to put the result
29 * @param[out] expired is the cache entry expired?
30 * @retval Was anything in the cache at all?
32 * If id->id == -1 this was a negative mapping.
35 bool idmap_cache_find_sid2unixid(const struct dom_sid *sid, struct unixid *id,
36 bool *expired)
38 fstring sidstr;
39 char *key;
40 char *value = NULL;
41 char *endptr;
42 time_t timeout;
43 bool ret;
44 struct unixid tmp_id;
46 key = talloc_asprintf(talloc_tos(), "IDMAP/SID2XID/%s",
47 sid_to_fstring(sidstr, sid));
48 if (key == NULL) {
49 return false;
51 ret = gencache_get(key, talloc_tos(), &value, &timeout);
52 if (!ret) {
53 goto done;
56 DEBUG(10, ("Parsing value for key [%s]: value=[%s]\n", key, value));
58 if (value[0] == '\0') {
59 DEBUG(0, ("Failed to parse value for key [%s]: "
60 "value is empty\n", key));
61 ret = false;
62 goto done;
65 tmp_id.id = strtol(value, &endptr, 10);
67 if ((value == endptr) && (tmp_id.id == 0)) {
68 DEBUG(0, ("Failed to parse value for key [%s]: value[%s] does "
69 "not start with a number\n", key, value));
70 ret = false;
71 goto done;
74 DEBUG(10, ("Parsing value for key [%s]: id=[%llu], endptr=[%s]\n",
75 key, (unsigned long long)tmp_id.id, endptr));
77 ret = (*endptr == ':');
78 if (ret) {
79 switch (endptr[1]) {
80 case 'U':
81 tmp_id.type = ID_TYPE_UID;
82 break;
84 case 'G':
85 tmp_id.type = ID_TYPE_GID;
86 break;
88 case 'B':
89 tmp_id.type = ID_TYPE_BOTH;
90 break;
92 case 'N':
93 tmp_id.type = ID_TYPE_NOT_SPECIFIED;
94 break;
96 case '\0':
97 DEBUG(0, ("FAILED to parse value for key [%s] "
98 "(id=[%llu], endptr=[%s]): "
99 "no type character after colon\n",
100 key, (unsigned long long)tmp_id.id, endptr));
101 ret = false;
102 goto done;
103 default:
104 DEBUG(0, ("FAILED to parse value for key [%s] "
105 "(id=[%llu], endptr=[%s]): "
106 "illegal type character '%c'\n",
107 key, (unsigned long long)tmp_id.id, endptr,
108 endptr[1]));
109 ret = false;
110 goto done;
112 if (endptr[2] != '\0') {
113 DEBUG(0, ("FAILED to parse value for key [%s] "
114 "(id=[%llu], endptr=[%s]): "
115 "more than 1 type character after colon\n",
116 key, (unsigned long long)tmp_id.id, endptr));
117 ret = false;
118 goto done;
121 *id = tmp_id;
122 *expired = (timeout <= time(NULL));
123 } else {
124 DEBUG(0, ("FAILED to parse value for key [%s] (value=[%s]): "
125 "colon missing after id=[%llu]\n",
126 key, value, (unsigned long long)tmp_id.id));
129 done:
130 TALLOC_FREE(key);
131 TALLOC_FREE(value);
132 return ret;
136 * Find a sid2uid mapping
137 * @param[in] sid the sid to map
138 * @param[out] puid where to put the result
139 * @param[out] expired is the cache entry expired?
140 * @retval Was anything in the cache at all?
142 * If *puid == -1 this was a negative mapping.
145 bool idmap_cache_find_sid2uid(const struct dom_sid *sid, uid_t *puid,
146 bool *expired)
148 bool ret;
149 struct unixid id;
150 ret = idmap_cache_find_sid2unixid(sid, &id, expired);
151 if (!ret) {
152 return false;
155 if (id.type == ID_TYPE_BOTH || id.type == ID_TYPE_UID) {
156 *puid = id.id;
157 } else {
158 *puid = -1;
160 return true;
164 * Find a sid2gid mapping
165 * @param[in] sid the sid to map
166 * @param[out] pgid where to put the result
167 * @param[out] expired is the cache entry expired?
168 * @retval Was anything in the cache at all?
170 * If *pgid == -1 this was a negative mapping.
173 bool idmap_cache_find_sid2gid(const struct dom_sid *sid, gid_t *pgid,
174 bool *expired)
176 bool ret;
177 struct unixid id;
178 ret = idmap_cache_find_sid2unixid(sid, &id, expired);
179 if (!ret) {
180 return false;
183 if (id.type == ID_TYPE_BOTH || id.type == ID_TYPE_GID) {
184 *pgid = id.id;
185 } else {
186 *pgid = -1;
188 return true;
191 struct idmap_cache_xid2sid_state {
192 struct dom_sid *sid;
193 bool *expired;
194 bool ret;
197 static void idmap_cache_xid2sid_parser(time_t timeout, DATA_BLOB blob,
198 void *private_data)
200 struct idmap_cache_xid2sid_state *state =
201 (struct idmap_cache_xid2sid_state *)private_data;
202 char *value;
204 ZERO_STRUCTP(state->sid);
205 state->ret = false;
207 if ((blob.length == 0) || (blob.data[blob.length-1] != 0)) {
209 * Not a string, can't be a valid mapping
211 return;
214 value = (char *)blob.data;
216 if (value[0] != '-') {
217 state->ret = string_to_sid(state->sid, value);
219 if (state->ret) {
220 *state->expired = (timeout <= time(NULL));
225 * Find a uid2sid mapping
226 * @param[in] uid the uid to map
227 * @param[out] sid where to put the result
228 * @param[out] expired is the cache entry expired?
229 * @retval Was anything in the cache at all?
231 * If "is_null_sid(sid)", this was a negative mapping.
234 bool idmap_cache_find_uid2sid(uid_t uid, struct dom_sid *sid, bool *expired)
236 fstring key;
237 struct idmap_cache_xid2sid_state state;
239 fstr_sprintf(key, "IDMAP/UID2SID/%d", (int)uid);
241 state.sid = sid;
242 state.expired = expired;
243 state.ret = false;
245 gencache_parse(key, idmap_cache_xid2sid_parser, &state);
246 return state.ret;
250 * Find a gid2sid mapping
251 * @param[in] gid the gid to map
252 * @param[out] sid where to put the result
253 * @param[out] expired is the cache entry expired?
254 * @retval Was anything in the cache at all?
256 * If "is_null_sid(sid)", this was a negative mapping.
259 bool idmap_cache_find_gid2sid(gid_t gid, struct dom_sid *sid, bool *expired)
261 fstring key;
262 struct idmap_cache_xid2sid_state state;
264 fstr_sprintf(key, "IDMAP/GID2SID/%d", (int)gid);
266 state.sid = sid;
267 state.expired = expired;
268 state.ret = false;
270 gencache_parse(key, idmap_cache_xid2sid_parser, &state);
271 return state.ret;
275 * Store a mapping in the idmap cache
276 * @param[in] sid the sid to map
277 * @param[in] gid the gid to map
279 * If both parameters are valid values, then a positive mapping in both
280 * directions is stored. If "is_null_sid(sid)" is true, then this will be a
281 * negative mapping of gid, we want to cache that for this gid we could not
282 * find anything. Likewise if "gid==-1", then we want to cache that we did not
283 * find a mapping for the sid passed here.
286 void idmap_cache_set_sid2unixid(const struct dom_sid *sid, struct unixid *unix_id)
288 time_t now = time(NULL);
289 time_t timeout;
290 fstring sidstr, key, value;
292 if (!is_null_sid(sid)) {
293 fstr_sprintf(key, "IDMAP/SID2XID/%s",
294 sid_to_fstring(sidstr, sid));
295 switch (unix_id->type) {
296 case ID_TYPE_UID:
297 fstr_sprintf(value, "%d:U", (int)unix_id->id);
298 break;
299 case ID_TYPE_GID:
300 fstr_sprintf(value, "%d:G", (int)unix_id->id);
301 break;
302 case ID_TYPE_BOTH:
303 fstr_sprintf(value, "%d:B", (int)unix_id->id);
304 break;
305 case ID_TYPE_NOT_SPECIFIED:
306 fstr_sprintf(value, "%d:N", (int)unix_id->id);
307 break;
308 default:
309 return;
311 timeout = (unix_id->id == -1)
312 ? lp_idmap_negative_cache_time()
313 : lp_idmap_cache_time();
314 gencache_set(key, value, now + timeout);
316 if (unix_id->id != -1) {
317 if (is_null_sid(sid)) {
318 /* negative gid mapping */
319 fstrcpy(value, "-");
320 timeout = lp_idmap_negative_cache_time();
322 else {
323 sid_to_fstring(value, sid);
324 timeout = lp_idmap_cache_time();
326 switch (unix_id->type) {
327 case ID_TYPE_BOTH:
328 fstr_sprintf(key, "IDMAP/UID2SID/%d", (int)unix_id->id);
329 gencache_set(key, value, now + timeout);
330 fstr_sprintf(key, "IDMAP/GID2SID/%d", (int)unix_id->id);
331 gencache_set(key, value, now + timeout);
332 return;
334 case ID_TYPE_UID:
335 fstr_sprintf(key, "IDMAP/UID2SID/%d", (int)unix_id->id);
336 break;
338 case ID_TYPE_GID:
339 fstr_sprintf(key, "IDMAP/GID2SID/%d", (int)unix_id->id);
340 break;
342 default:
343 return;
345 gencache_set(key, value, now + timeout);
349 static char* key_xid2sid_str(TALLOC_CTX* mem_ctx, char t, const char* id) {
350 return talloc_asprintf(mem_ctx, "IDMAP/%cID2SID/%s", t, id);
353 static char* key_xid2sid(TALLOC_CTX* mem_ctx, char t, int id) {
354 char str[32];
355 snprintf(str, sizeof(str), "%d", id);
356 return key_xid2sid_str(mem_ctx, t, str);
359 static char* key_sid2xid_str(TALLOC_CTX* mem_ctx, const char* id) {
360 return talloc_asprintf(mem_ctx, "IDMAP/SID2XID/%s", id);
363 static bool idmap_cache_del_xid(char t, int xid)
365 TALLOC_CTX* mem_ctx = talloc_stackframe();
366 const char* key = key_xid2sid(mem_ctx, t, xid);
367 char* sid_str = NULL;
368 time_t timeout;
369 bool ret = true;
371 if (!gencache_get(key, mem_ctx, &sid_str, &timeout)) {
372 DEBUG(3, ("no entry: %s\n", key));
373 ret = false;
374 goto done;
377 if (sid_str[0] != '-') {
378 const char* sid_key = key_sid2xid_str(mem_ctx, sid_str);
379 if (!gencache_del(sid_key)) {
380 DEBUG(2, ("failed to delete: %s\n", sid_key));
381 ret = false;
382 } else {
383 DEBUG(5, ("delete: %s\n", sid_key));
388 if (!gencache_del(key)) {
389 DEBUG(1, ("failed to delete: %s\n", key));
390 ret = false;
391 } else {
392 DEBUG(5, ("delete: %s\n", key));
395 done:
396 talloc_free(mem_ctx);
397 return ret;
400 bool idmap_cache_del_uid(uid_t uid) {
401 return idmap_cache_del_xid('U', uid);
404 bool idmap_cache_del_gid(gid_t gid) {
405 return idmap_cache_del_xid('G', gid);
408 bool idmap_cache_del_sid(const struct dom_sid *sid)
410 TALLOC_CTX* mem_ctx = talloc_stackframe();
411 bool ret = true;
412 bool expired;
413 struct unixid id;
414 const char *sid_key;
416 if (!idmap_cache_find_sid2unixid(sid, &id, &expired)) {
417 ret = false;
418 goto done;
421 if (id.id != -1) {
422 switch (id.type) {
423 case ID_TYPE_BOTH:
424 idmap_cache_del_xid('U', id.id);
425 idmap_cache_del_xid('G', id.id);
426 break;
427 case ID_TYPE_UID:
428 idmap_cache_del_xid('U', id.id);
429 break;
430 case ID_TYPE_GID:
431 idmap_cache_del_xid('G', id.id);
432 break;
433 default:
434 break;
438 sid_key = key_sid2xid_str(mem_ctx, dom_sid_string(mem_ctx, sid));
439 if (sid_key == NULL) {
440 return false;
442 /* If the mapping was symmetric, then this should fail */
443 gencache_del(sid_key);
444 done:
445 talloc_free(mem_ctx);
446 return ret;