Copyright (C) Jim McDonough <jmcd@us.ibm.com> 2003
Copyright (C) Jeremy Allison 2006
Copyright (C) Simo Sorce 2003-2006
-
+
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
-
+
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
-
+
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
static NTSTATUS idmap_tdb2_open_db(void)
{
char *db_path;
-
+
if (idmap_tdb2) {
/* its already open */
return NT_STATUS_OK;
void *private_data)
{
NTSTATUS ret;
- uint32_t res;
struct idmap_tdb2_allocate_id_context *state;
uint32_t hwm;
}
/* fetch a new id and increment it */
- res = dbwrap_change_uint32_atomic(db, state->hwmkey, &hwm, 1);
- if (res == -1) {
+ ret = dbwrap_trans_change_uint32_atomic(db, state->hwmkey, &hwm, 1);
+ if (!NT_STATUS_IS_OK(ret)) {
DEBUG(1, ("Fatal error while fetching a new %s value\n!",
state->hwmtype));
- ret = NT_STATUS_UNSUCCESSFUL;
goto done;
}
const char *hwmkey;
const char *hwmtype;
uint32_t high_hwm;
- uint32_t hwm;
+ uint32_t hwm = 0;
NTSTATUS status;
struct idmap_tdb2_allocate_id_context state;
if (NT_STATUS_IS_OK(status)) {
xid->id = state.hwm;
- DEBUG(10,("New %s = %d\n", hwmtype, hwm));
+ DEBUG(10,("New %s = %d\n", hwmtype, state.hwm));
} else {
DEBUG(1, ("Error allocating a new %s\n", hwmtype));
}
return status;
}
-/*
- Get current highest id.
-*/
-static NTSTATUS idmap_tdb2_get_hwm(struct unixid *xid)
-{
- const char *hwmkey;
- const char *hwmtype;
- uint32_t hwm;
- uint32_t high_hwm;
- NTSTATUS status;
-
- status = idmap_tdb2_open_db();
- NT_STATUS_NOT_OK_RETURN(status);
-
- /* Get current high water mark */
- switch (xid->type) {
-
- case ID_TYPE_UID:
- hwmkey = HWM_USER;
- hwmtype = "UID";
- high_hwm = idmap_tdb2_state.high_uid;
- break;
-
- case ID_TYPE_GID:
- hwmkey = HWM_GROUP;
- hwmtype = "GID";
- high_hwm = idmap_tdb2_state.high_gid;
- break;
-
- default:
- return NT_STATUS_INVALID_PARAMETER;
- }
-
- if ((hwm = dbwrap_fetch_int32(idmap_tdb2, hwmkey)) == -1) {
- return NT_STATUS_INTERNAL_DB_ERROR;
- }
-
- xid->id = hwm;
-
- /* Warn if it is out of range */
- if (hwm >= high_hwm) {
- DEBUG(0, ("Warning: %s range full!! (max: %lu)\n",
- hwmtype, (unsigned long)high_hwm));
- }
-
- return NT_STATUS_OK;
-}
-
-/*
- Set high id.
-*/
-static NTSTATUS idmap_tdb2_set_hwm(struct unixid *xid)
-{
- /* not supported, or we would invalidate the cache tdb on
- other nodes */
- DEBUG(0,("idmap_tdb2_set_hwm not supported\n"));
- return NT_STATUS_NOT_SUPPORTED;
-}
-
/*
Close the alloc tdb
*/
struct idmap_tdb2_set_mapping_context *state;
TALLOC_CTX *tmp_ctx = talloc_stackframe();
- DEBUG(10, ("Storing %s <-> %s map\n", state->ksidstr, state->kidstr));
-
state = (struct idmap_tdb2_set_mapping_context *)private_data;
+ DEBUG(10, ("Storing %s <-> %s map\n", state->ksidstr, state->kidstr));
+
/* check wheter sid mapping is already present in db */
data = dbwrap_fetch_bystring(db, tmp_ctx, state->ksidstr);
if (data.dptr) {
case ID_TYPE_UID:
keystr = talloc_asprintf(ctx, "UID %lu", (unsigned long)map->xid.id);
break;
-
+
case ID_TYPE_GID:
keystr = talloc_asprintf(ctx, "GID %lu", (unsigned long)map->xid.id);
break;
data = dbwrap_fetch_bystring(idmap_tdb2, keystr, keystr);
if (!data.dptr) {
- fstring sidstr;
+ char *sidstr;
+ struct idmap_tdb2_set_mapping_context store_state;
DEBUG(10,("Record %s not found\n", keystr));
if (idmap_tdb2_state.idmap_script == NULL) {
goto done;
}
- if (sid_to_fstring(sidstr, map->sid)) {
- /* both forward and reverse mappings */
- dbwrap_store_bystring(idmap_tdb2, keystr,
- string_term_tdb_data(sidstr),
- TDB_REPLACE);
- dbwrap_store_bystring(idmap_tdb2, sidstr,
- string_term_tdb_data(keystr),
- TDB_REPLACE);
+ sidstr = sid_string_talloc(keystr, map->sid);
+ if (!sidstr) {
+ ret = NT_STATUS_NO_MEMORY;
+ goto done;
}
+
+ store_state.ksidstr = sidstr;
+ store_state.kidstr = keystr;
+
+ ret = dbwrap_trans_do(idmap_tdb2, idmap_tdb2_set_mapping_action,
+ &store_state);
goto done;
}
-
+
if (!string_to_sid(map->sid, (const char *)data.dptr)) {
DEBUG(10,("INVALID SID (%s) in record %s\n",
(const char *)data.dptr, keystr));
/* Check if sid is present in database */
data = dbwrap_fetch_bystring(idmap_tdb2, tmp_ctx, keystr);
if (!data.dptr) {
- fstring idstr;
+ char *idstr;
+ struct idmap_tdb2_set_mapping_context store_state;
DEBUG(10,(__location__ " Record %s not found\n", keystr));
ret = NT_STATUS_NONE_MAPPED;
goto done;
}
-
+
ret = idmap_tdb2_script(ctx, map, "SIDTOID %s", keystr);
/* store it on shared storage */
if (!NT_STATUS_IS_OK(ret)) {
goto done;
}
- snprintf(idstr, sizeof(idstr), "%cID %lu",
- map->xid.type == ID_TYPE_UID?'U':'G',
- (unsigned long)map->xid.id);
- /* store both forward and reverse mappings */
- dbwrap_store_bystring(idmap_tdb2, keystr, string_term_tdb_data(idstr),
- TDB_REPLACE);
- dbwrap_store_bystring(idmap_tdb2, idstr, string_term_tdb_data(keystr),
- TDB_REPLACE);
+ /* apply filters before returning result */
+ if ((ctx->filter_low_id
+ && (map->xid.id < ctx->filter_low_id)) ||
+ (ctx->filter_high_id
+ && (map->xid.id > ctx->filter_high_id))) {
+ DEBUG(5, ("Script returned id (%u) out of range "
+ "(%u - %u). Filtered!\n",
+ map->xid.id,
+ ctx->filter_low_id, ctx->filter_high_id));
+ ret = NT_STATUS_NONE_MAPPED;
+ goto done;
+ }
+
+ idstr = talloc_asprintf(tmp_ctx, "%cID %lu",
+ map->xid.type == ID_TYPE_UID?'U':'G',
+ (unsigned long)map->xid.id);
+ if (idstr == NULL) {
+ ret = NT_STATUS_NO_MEMORY;
+ goto done;
+ }
+
+ store_state.ksidstr = keystr;
+ store_state.kidstr = idstr;
+
+ ret = dbwrap_trans_do(idmap_tdb2, idmap_tdb2_set_mapping_action,
+ &store_state);
goto done;
}
} else { /* Unknown record type ! */
DEBUG(2, ("Found INVALID record %s -> %s\n", keystr, (const char *)data.dptr));
ret = NT_STATUS_INTERNAL_DB_ERROR;
+ goto done;
}
-
+
/* apply filters before returning result */
if ((ctx->filter_low_id && (map->xid.id < ctx->filter_low_id)) ||
(ctx->filter_high_id && (map->xid.id > ctx->filter_high_id))) {
for (i = 0; ids[i]; i++) {
ids[i]->status = ID_UNKNOWN;
}
-
+
ctx = talloc_get_type(dom->private_data, struct idmap_tdb2_context);
for (i = 0; ids[i]; i++) {
ids[i]->status = ID_UNMAPPED;
continue;
}
-
+
/* some fatal error occurred, return immediately */
goto done;
}
for (i = 0; ids[i]; i++) {
ids[i]->status = ID_UNKNOWN;
}
-
+
ctx = talloc_get_type(dom->private_data, struct idmap_tdb2_context);
for (i = 0; ids[i]; i++) {
ids[i]->status = ID_UNMAPPED;
continue;
}
-
+
/* some fatal error occurred, return immediately */
goto done;
}
ksidstr = kidstr = NULL;
/* TODO: should we filter a set_mapping using low/high filters ? */
-
+
ctx = talloc_get_type(dom->private_data, struct idmap_tdb2_context);
switch (map->xid.type) {
case ID_TYPE_UID:
kidstr = talloc_asprintf(ctx, "UID %lu", (unsigned long)map->xid.id);
break;
-
+
case ID_TYPE_GID:
kidstr = talloc_asprintf(ctx, "GID %lu", (unsigned long)map->xid.id);
break;
goto done;
}
- if (!(ksidstr = sid_string_talloc(ctx, map->sid))) {
+ ksidstr = sid_string_talloc(ctx, map->sid);
+ if (ksidstr == NULL) {
DEBUG(0, ("Out of memory!\n"));
ret = NT_STATUS_NO_MEMORY;
goto done;
return ret;
}
-/*
- remove a mapping.
-*/
-static NTSTATUS idmap_tdb2_remove_mapping(struct idmap_domain *dom, const struct id_map *map)
-{
- /* not supported as it would invalidate the cache tdb on other
- nodes */
- DEBUG(0,("idmap_tdb2_remove_mapping not supported\n"));
- return NT_STATUS_NOT_SUPPORTED;
-}
-
/*
Close the idmap tdb instance
*/
return NT_STATUS_OK;
}
-
-/*
- Dump all mappings out
-*/
-static NTSTATUS idmap_tdb2_dump_data(struct idmap_domain *dom, struct id_map **maps, int *num_maps)
-{
- DEBUG(0,("idmap_tdb2_dump_data not supported\n"));
- return NT_STATUS_NOT_SUPPORTED;
-}
-
static struct idmap_methods db_methods = {
.init = idmap_tdb2_db_init,
.unixids_to_sids = idmap_tdb2_unixids_to_sids,
.sids_to_unixids = idmap_tdb2_sids_to_unixids,
- .set_mapping = idmap_tdb2_set_mapping,
- .remove_mapping = idmap_tdb2_remove_mapping,
- .dump_data = idmap_tdb2_dump_data,
.close_fn = idmap_tdb2_close
};
static struct idmap_alloc_methods db_alloc_methods = {
.init = idmap_tdb2_alloc_init,
.allocate_id = idmap_tdb2_allocate_id,
- .get_id_hwm = idmap_tdb2_get_hwm,
- .set_id_hwm = idmap_tdb2_set_hwm,
.close_fn = idmap_tdb2_alloc_close
};
NTSTATUS idmap_tdb2_init(void)
{
- NTSTATUS ret;
-
- /* register both backends */
- ret = smb_register_idmap_alloc(SMB_IDMAP_INTERFACE_VERSION, "tdb2", &db_alloc_methods);
- if (! NT_STATUS_IS_OK(ret)) {
- DEBUG(0, ("Unable to register idmap alloc tdb2 module: %s\n", get_friendly_nt_error_msg(ret)));
- return ret;
- }
-
return smb_register_idmap(SMB_IDMAP_INTERFACE_VERSION, "tdb2", &db_methods);
}