?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
libsmb: Make cli_smb2_create_fnum async
[Samba.git] / source3 / winbindd / idmap_tdb_common.c
blobebf1bb90b2cb9aee38f173f980e46624ba22b3f7
1 /*
2 Unix SMB/CIFS implementation.
3
4 common functions for TDB based idmapping backends
5
6 Copyright (C) Christian Ambach 2012
7
8 These functions were initially copied over from idmap_tdb.c and idmap_tdb2.c
9 which are:
10
11 Copyright (C) Tim Potter 2000
12 Copyright (C) Jim McDonough <jmcd@us.ibm.com> 2003
13 Copyright (C) Jeremy Allison 2006
14 Copyright (C) Simo Sorce 2003-2006
15 Copyright (C) Michael Adam 2009-2010
16 Copyright (C) Andrew Tridgell 2007
17
18 This program is free software; you can redistribute it and/or modify
19 it under the terms of the GNU General Public License as published by
20 the Free Software Foundation; either version 2 of the License, or
21 (at your option) any later version.
22
23 This program is distributed in the hope that it will be useful,
24 but WITHOUT ANY WARRANTY; without even the implied warranty of
25 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
26 GNU General Public License for more details.
27
28 You should have received a copy of the GNU General Public License
29 along with this program; if not, write to the Free Software
30 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
31 */
32
33 #include "includes.h"
34 #include "idmap_tdb_common.h"
35 #include "dbwrap/dbwrap.h"
36 #include "util_tdb.h"
37 #include "idmap_rw.h"
38 #include "../libcli/security/dom_sid.h"
39
40 #undef DBGC_CLASS
41 #define DBGC_CLASS DBGC_IDMAP
42
43 struct idmap_tdb_common_allocate_id_context {
44 const char *hwmkey;
45 const char *hwmtype;
46 uint32_t high_hwm;
47 uint32_t hwm;
48 };
49
50 static NTSTATUS idmap_tdb_common_allocate_id_action(struct db_context *db,
51 void *private_data)
52 {
53 NTSTATUS ret;
54 struct idmap_tdb_common_allocate_id_context *state;
55 uint32_t hwm;
56
57 state = (struct idmap_tdb_common_allocate_id_context *)private_data;
58
59 ret = dbwrap_fetch_uint32_bystring(db, state->hwmkey, &hwm);
60 if (!NT_STATUS_IS_OK(ret)) {
61 ret = NT_STATUS_INTERNAL_DB_ERROR;
62 goto done;
63 }
64
65 /* check it is in the range */
66 if (hwm > state->high_hwm) {
67 DEBUG(1, ("Fatal Error: %s range full!! (max: %lu)\n",
68 state->hwmtype, (unsigned long)state->high_hwm));
69 ret = NT_STATUS_UNSUCCESSFUL;
70 goto done;
71 }
72
73 /* fetch a new id and increment it */
74 ret = dbwrap_change_uint32_atomic_bystring(db, state->hwmkey, &hwm, 1);
75 if (!NT_STATUS_IS_OK(ret)) {
76 DEBUG(1, ("Fatal error while fetching a new %s value\n!",
77 state->hwmtype));
78 goto done;
79 }
80
81 /* recheck it is in the range */
82 if (hwm > state->high_hwm) {
83 DEBUG(1, ("Fatal Error: %s range full!! (max: %lu)\n",
84 state->hwmtype, (unsigned long)state->high_hwm));
85 ret = NT_STATUS_UNSUCCESSFUL;
86 goto done;
87 }
88
89 ret = NT_STATUS_OK;
90 state->hwm = hwm;
91
92 done:
93 return ret;
94 }
95
96 static NTSTATUS idmap_tdb_common_allocate_id(struct idmap_domain *dom,
97 struct unixid *xid)
98 {
99 const char *hwmkey;
100 const char *hwmtype;
101 uint32_t hwm = 0;
102 NTSTATUS status;
103 struct idmap_tdb_common_allocate_id_context state;
104 struct idmap_tdb_common_context *ctx;
105
106 ctx =
107 talloc_get_type_abort(dom->private_data,
108 struct idmap_tdb_common_context);
109
110 /* Get current high water mark */
111 switch (xid->type) {
112
113 case ID_TYPE_UID:
114 hwmkey = ctx->hwmkey_uid;
115 hwmtype = "UID";
116 break;
117
118 case ID_TYPE_GID:
119 hwmkey = ctx->hwmkey_gid;
120 hwmtype = "GID";
121 break;
122
123 default:
124 DEBUG(2, ("Invalid ID type (0x%x)\n", xid->type));
125 return NT_STATUS_INVALID_PARAMETER;
126 }
127
128 state.hwm = hwm;
129 state.high_hwm = ctx->max_id;
130 state.hwmtype = hwmtype;
131 state.hwmkey = hwmkey;
132
133 status = dbwrap_trans_do(ctx->db, idmap_tdb_common_allocate_id_action,
134 &state);
135
136 if (NT_STATUS_IS_OK(status)) {
137 xid->id = state.hwm;
138 DEBUG(10, ("New %s = %d\n", hwmtype, state.hwm));
139 } else {
140 DEBUG(1, ("Error allocating a new %s\n", hwmtype));
141 }
142
143 return status;
144 }
145
146 /**
147 * Allocate a new unix-ID.
148 * For now this is for the default idmap domain only.
149 * Should be extended later on.
150 */
151 NTSTATUS idmap_tdb_common_get_new_id(struct idmap_domain * dom,
152 struct unixid * id)
153 {
154 NTSTATUS ret;
155
156 if (!strequal(dom->name, "*")) {
157 DEBUG(3, ("idmap_tdb_common_get_new_id: "
158 "Refusing allocation of a new unixid for domain'%s'. "
159 "Currently only supported for the default "
160 "domain \"*\".\n", dom->name));
161 return NT_STATUS_NOT_IMPLEMENTED;
162 }
163
164 ret = idmap_tdb_common_allocate_id(dom, id);
165
166 return ret;
167 }
168
169 /**
170 * store a mapping in the database.
171 */
172
173 struct idmap_tdb_common_set_mapping_context {
174 const char *ksidstr;
175 const char *kidstr;
176 };
177
178 static NTSTATUS idmap_tdb_common_set_mapping_action(struct db_context *db,
179 void *private_data)
180 {
181 TDB_DATA data;
182 NTSTATUS ret;
183 struct idmap_tdb_common_set_mapping_context *state;
184 TALLOC_CTX *tmp_ctx = talloc_stackframe();
185
186 state = (struct idmap_tdb_common_set_mapping_context *)private_data;
187
188 DEBUG(10, ("Storing %s <-> %s map\n", state->ksidstr, state->kidstr));
189
190 /* check whether sid mapping is already present in db */
191 ret = dbwrap_fetch_bystring(db, tmp_ctx, state->ksidstr, &data);
192 if (NT_STATUS_IS_OK(ret)) {
193 ret = NT_STATUS_OBJECT_NAME_COLLISION;
194 goto done;
195 }
196
197 ret = dbwrap_store_bystring(db, state->ksidstr,
198 string_term_tdb_data(state->kidstr),
199 TDB_INSERT);
200 if (!NT_STATUS_IS_OK(ret)) {
201 DEBUG(0, ("Error storing SID -> ID: %s\n", nt_errstr(ret)));
202 goto done;
203 }
204
205 ret = dbwrap_store_bystring(db, state->kidstr,
206 string_term_tdb_data(state->ksidstr),
207 TDB_INSERT);
208 if (!NT_STATUS_IS_OK(ret)) {
209 DEBUG(0, ("Error storing ID -> SID: %s\n", nt_errstr(ret)));
210 /* try to remove the previous stored SID -> ID map */
211 dbwrap_delete_bystring(db, state->ksidstr);
212 goto done;
213 }
214
215 DEBUG(10, ("Stored %s <-> %s\n", state->ksidstr, state->kidstr));
216
217 done:
218 talloc_free(tmp_ctx);
219 return ret;
220 }
221
222 NTSTATUS idmap_tdb_common_set_mapping(struct idmap_domain * dom,
223 const struct id_map * map)
224 {
225 struct idmap_tdb_common_context *ctx;
226 struct idmap_tdb_common_set_mapping_context state;
227 NTSTATUS ret;
228 char *ksidstr, *kidstr;
229
230 if (!map || !map->sid) {
231 return NT_STATUS_INVALID_PARAMETER;
232 }
233
234 ksidstr = kidstr = NULL;
235
236 /* TODO: should we filter a set_mapping using low/high filters ? */
237
238 ctx =
239 talloc_get_type_abort(dom->private_data,
240 struct idmap_tdb_common_context);
241
242 switch (map->xid.type) {
243
244 case ID_TYPE_UID:
245 kidstr =
246 talloc_asprintf(ctx, "UID %lu", (unsigned long)map->xid.id);
247 break;
248
249 case ID_TYPE_GID:
250 kidstr =
251 talloc_asprintf(ctx, "GID %lu", (unsigned long)map->xid.id);
252 break;
253
254 default:
255 DEBUG(2, ("INVALID unix ID type: 0x%02x\n", map->xid.type));
256 return NT_STATUS_INVALID_PARAMETER;
257 }
258
259 if (kidstr == NULL) {
260 DEBUG(0, ("ERROR: Out of memory!\n"));
261 ret = NT_STATUS_NO_MEMORY;
262 goto done;
263 }
264
265 ksidstr = sid_string_talloc(ctx, map->sid);
266 if (ksidstr == NULL) {
267 DEBUG(0, ("Out of memory!\n"));
268 ret = NT_STATUS_NO_MEMORY;
269 goto done;
270 }
271
272 state.ksidstr = ksidstr;
273 state.kidstr = kidstr;
274
275 ret = dbwrap_trans_do(ctx->db, idmap_tdb_common_set_mapping_action,
276 &state);
277
278 done:
279 talloc_free(ksidstr);
280 talloc_free(kidstr);
281 return ret;
282 }
283
284 /*
285 * Create a new mapping for an unmapped SID, also allocating a new ID.
286 * This should be run inside a transaction.
287 *
288 * TODO:
289 * Properly integrate this with multi domain idmap config:
290 * Currently, the allocator is default-config only.
291 */
292 NTSTATUS idmap_tdb_common_new_mapping(struct idmap_domain * dom,
293 struct id_map * map)
294 {
295 NTSTATUS ret;
296 struct idmap_tdb_common_context *ctx;
297
298 ctx =
299 talloc_get_type_abort(dom->private_data,
300 struct idmap_tdb_common_context);
301
302 ret = idmap_rw_new_mapping(dom, ctx->rw_ops, map);
303
304 return ret;
305 }
306
307 /*
308 lookup a set of unix ids
309 */
310 NTSTATUS idmap_tdb_common_unixids_to_sids(struct idmap_domain * dom,
311 struct id_map ** ids)
312 {
313 NTSTATUS ret;
314 int i, num_mapped = 0;
315 struct idmap_tdb_common_context *ctx;
316
317 NTSTATUS(*unixid_to_sid_fn) (struct idmap_domain * dom,
318 struct id_map * map);
319 ctx =
320 talloc_get_type_abort(dom->private_data,
321 struct idmap_tdb_common_context);
322
323 if (ctx->unixid_to_sid_fn == NULL) {
324 unixid_to_sid_fn = idmap_tdb_common_unixid_to_sid;
325 } else {
326 unixid_to_sid_fn = ctx->unixid_to_sid_fn;
327 }
328
329 /* initialize the status to avoid surprise */
330 for (i = 0; ids[i]; i++) {
331 ids[i]->status = ID_UNKNOWN;
332 }
333
334 for (i = 0; ids[i]; i++) {
335 ret = unixid_to_sid_fn(dom, ids[i]);
336 if (!NT_STATUS_IS_OK(ret)) {
337
338 /* if it is just a failed mapping continue */
339 if (NT_STATUS_EQUAL(ret, NT_STATUS_NONE_MAPPED)) {
340
341 /* make sure it is marked as unmapped */
342 ids[i]->status = ID_UNMAPPED;
343 continue;
344 }
345
346 /* some fatal error occurred, return immediately */
347 goto done;
348 }
349
350 /* all ok, id is mapped */
351 ids[i]->status = ID_MAPPED;
352 num_mapped += 1;
353 }
354
355 ret = NT_STATUS_OK;
356
357 done:
358
359 if (NT_STATUS_IS_OK(ret)) {
360 if (i == 0 || num_mapped == 0) {
361 ret = NT_STATUS_NONE_MAPPED;
362 } else if (num_mapped < i) {
363 ret = STATUS_SOME_UNMAPPED;
364 } else {
365 ret = NT_STATUS_OK;
366 }
367 }
368
369 return ret;
370 }
371
372 /*
373 default single id to sid lookup function
374 */
375 NTSTATUS idmap_tdb_common_unixid_to_sid(struct idmap_domain * dom,
376 struct id_map * map)
377 {
378 NTSTATUS ret;
379 TDB_DATA data;
380 char *keystr;
381 struct idmap_tdb_common_context *ctx;
382
383 if (!dom || !map) {
384 return NT_STATUS_INVALID_PARAMETER;
385 }
386
387 ctx =
388 talloc_get_type_abort(dom->private_data,
389 struct idmap_tdb_common_context);
390
391 /* apply filters before checking */
392 if (!idmap_unix_id_is_in_range(map->xid.id, dom)) {
393 DEBUG(5,
394 ("Requested id (%u) out of range (%u - %u). Filtered!\n",
395 map->xid.id, dom->low_id, dom->high_id));
396 return NT_STATUS_NONE_MAPPED;
397 }
398
399 switch (map->xid.type) {
400
401 case ID_TYPE_UID:
402 keystr =
403 talloc_asprintf(ctx, "UID %lu", (unsigned long)map->xid.id);
404 break;
405
406 case ID_TYPE_GID:
407 keystr =
408 talloc_asprintf(ctx, "GID %lu", (unsigned long)map->xid.id);
409 break;
410
411 default:
412 DEBUG(2, ("INVALID unix ID type: 0x%02x\n", map->xid.type));
413 return NT_STATUS_INVALID_PARAMETER;
414 }
415
416 if (keystr == NULL) {
417 DEBUG(0, ("Out of memory!\n"));
418 ret = NT_STATUS_NO_MEMORY;
419 goto done;
420 }
421
422 DEBUG(10, ("Fetching record %s\n", keystr));
423
424 /* Check if the mapping exists */
425 ret = dbwrap_fetch_bystring(ctx->db, keystr, keystr, &data);
426
427 if (!NT_STATUS_IS_OK(ret)) {
428 DEBUG(10, ("Record %s not found\n", keystr));
429 ret = NT_STATUS_NONE_MAPPED;
430 goto done;
431 }
432
433 if (!string_to_sid(map->sid, (const char *)data.dptr)) {
434 DEBUG(10, ("INVALID SID (%s) in record %s\n",
435 (const char *)data.dptr, keystr));
436 ret = NT_STATUS_INTERNAL_DB_ERROR;
437 goto done;
438 }
439
440 DEBUG(10, ("Found record %s -> %s\n", keystr, (const char *)data.dptr));
441 ret = NT_STATUS_OK;
442
443 done:
444 talloc_free(keystr);
445 return ret;
446 }
447
448 /**********************************
449 Single sid to id lookup function.
450 **********************************/
451
452 NTSTATUS idmap_tdb_common_sid_to_unixid(struct idmap_domain * dom,
453 struct id_map * map)
454 {
455 NTSTATUS ret;
456 TDB_DATA data;
457 char *keystr;
458 unsigned long rec_id = 0;
459 struct idmap_tdb_common_context *ctx;
460 TALLOC_CTX *tmp_ctx = talloc_stackframe();
461
462 if (!dom || !map) {
463 talloc_free(tmp_ctx);
464 return NT_STATUS_INVALID_PARAMETER;
465 }
466
467 ctx =
468 talloc_get_type_abort(dom->private_data,
469 struct idmap_tdb_common_context);
470
471 keystr = sid_string_talloc(tmp_ctx, map->sid);
472 if (keystr == NULL) {
473 DEBUG(0, ("Out of memory!\n"));
474 ret = NT_STATUS_NO_MEMORY;
475 goto done;
476 }
477
478 DEBUG(10, ("Fetching record %s\n", keystr));
479
480 /* Check if sid is present in database */
481 ret = dbwrap_fetch_bystring(ctx->db, tmp_ctx, keystr, &data);
482 if (!NT_STATUS_IS_OK(ret)) {
483 DEBUG(10, ("Record %s not found\n", keystr));
484 ret = NT_STATUS_NONE_MAPPED;
485 goto done;
486 }
487
488 /* What type of record is this ? */
489 if (sscanf((const char *)data.dptr, "UID %lu", &rec_id) == 1) {
490 /* Try a UID record. */
491 map->xid.id = rec_id;
492 map->xid.type = ID_TYPE_UID;
493 DEBUG(10,
494 ("Found uid record %s -> %s \n", keystr,
495 (const char *)data.dptr));
496 ret = NT_STATUS_OK;
497
498 } else if (sscanf((const char *)data.dptr, "GID %lu", &rec_id) == 1) {
499 /* Try a GID record. */
500 map->xid.id = rec_id;
501 map->xid.type = ID_TYPE_GID;
502 DEBUG(10,
503 ("Found gid record %s -> %s \n", keystr,
504 (const char *)data.dptr));
505 ret = NT_STATUS_OK;
506
507 } else { /* Unknown record type ! */
508 DEBUG(2,
509 ("Found INVALID record %s -> %s\n", keystr,
510 (const char *)data.dptr));
511 ret = NT_STATUS_INTERNAL_DB_ERROR;
512 goto done;
513 }
514
515 /* apply filters before returning result */
516 if (!idmap_unix_id_is_in_range(map->xid.id, dom)) {
517 DEBUG(5,
518 ("Requested id (%u) out of range (%u - %u). Filtered!\n",
519 map->xid.id, dom->low_id, dom->high_id));
520 ret = NT_STATUS_NONE_MAPPED;
521 }
522
523 done:
524 talloc_free(tmp_ctx);
525 return ret;
526 }
527
528 /**********************************
529 lookup a set of sids
530 **********************************/
531
532 struct idmap_tdb_common_sids_to_unixids_context {
533 struct idmap_domain *dom;
534 struct id_map **ids;
535 bool allocate_unmapped;
536 NTSTATUS(*sid_to_unixid_fn) (struct idmap_domain * dom,
537 struct id_map * map);
538 };
539
540 static NTSTATUS idmap_tdb_common_sids_to_unixids_action(struct db_context *db,
541 void *private_data)
542 {
543 struct idmap_tdb_common_sids_to_unixids_context *state;
544 int i, num_mapped = 0;
545 NTSTATUS ret = NT_STATUS_OK;
546
547 state = (struct idmap_tdb_common_sids_to_unixids_context *)private_data;
548
549 DEBUG(10, ("idmap_tdb_common_sids_to_unixids: "
550 " domain: [%s], allocate: %s\n",
551 state->dom->name, state->allocate_unmapped ? "yes" : "no"));
552
553 for (i = 0; state->ids[i]; i++) {
554 if ((state->ids[i]->status == ID_UNKNOWN) ||
555 /* retry if we could not map in previous run: */
556 (state->ids[i]->status == ID_UNMAPPED)) {
557 NTSTATUS ret2;
558
559 ret2 = state->sid_to_unixid_fn(state->dom,
560 state->ids[i]);
561
562 if (!NT_STATUS_IS_OK(ret2)) {
563
564 /* if it is just a failed mapping, continue */
565 if (NT_STATUS_EQUAL
566 (ret2, NT_STATUS_NONE_MAPPED)) {
567
568 /* make sure it is marked as unmapped */
569 state->ids[i]->status = ID_UNMAPPED;
570 ret = STATUS_SOME_UNMAPPED;
571 } else {
572 /*
573 * some fatal error occurred,
574 * return immediately
575 */
576 ret = ret2;
577 goto done;
578 }
579 } else {
580 /* all ok, id is mapped */
581 state->ids[i]->status = ID_MAPPED;
582 }
583 }
584
585 if (state->ids[i]->status == ID_MAPPED) {
586 num_mapped += 1;
587 }
588
589 if ((state->ids[i]->status == ID_UNMAPPED) &&
590 state->allocate_unmapped) {
591 ret =
592 idmap_tdb_common_new_mapping(state->dom,
593 state->ids[i]);
594 if (!NT_STATUS_IS_OK(ret)) {
595 goto done;
596 }
597 num_mapped += 1;
598 }
599 }
600
601 done:
602
603 if (NT_STATUS_IS_OK(ret) ||
604 NT_STATUS_EQUAL(ret, STATUS_SOME_UNMAPPED)) {
605 if (i == 0 || num_mapped == 0) {
606 ret = NT_STATUS_NONE_MAPPED;
607 } else if (num_mapped < i) {
608 ret = STATUS_SOME_UNMAPPED;
609 } else {
610 ret = NT_STATUS_OK;
611 }
612 }
613
614 return ret;
615 }
616
617 NTSTATUS idmap_tdb_common_sids_to_unixids(struct idmap_domain * dom,
618 struct id_map ** ids)
619 {
620 NTSTATUS ret;
621 int i;
622 struct idmap_tdb_common_sids_to_unixids_context state;
623 struct idmap_tdb_common_context *ctx;
624
625 ctx =
626 talloc_get_type_abort(dom->private_data,
627 struct idmap_tdb_common_context);
628
629 /* initialize the status to avoid surprise */
630 for (i = 0; ids[i]; i++) {
631 ids[i]->status = ID_UNKNOWN;
632 }
633
634 state.dom = dom;
635 state.ids = ids;
636 state.allocate_unmapped = false;
637 if (ctx->sid_to_unixid_fn == NULL) {
638 state.sid_to_unixid_fn = idmap_tdb_common_sid_to_unixid;
639 } else {
640 state.sid_to_unixid_fn = ctx->sid_to_unixid_fn;
641 }
642
643 ret = idmap_tdb_common_sids_to_unixids_action(ctx->db, &state);
644
645 if ( (NT_STATUS_EQUAL(ret, STATUS_SOME_UNMAPPED) ||
646 NT_STATUS_EQUAL(ret, NT_STATUS_NONE_MAPPED)) &&
647 !dom->read_only) {
648 state.allocate_unmapped = true;
649 ret = dbwrap_trans_do(ctx->db,
650 idmap_tdb_common_sids_to_unixids_action,
651 &state);
652 }
653
654 return ret;
655 }
Samba GIT mirror