2008-08-20 07:22:16 +04:00
/*
Unix SMB / CIFS mplementation .
DSDB schema header
Copyright ( C ) Stefan Metzmacher < metze @ samba . org > 2006 - 2007
Copyright ( C ) Andrew Bartlett < abartlet @ samba . org > 2006 - 2008
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 3 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 , see < http : //www.gnu.org/licenses/>.
*/
# include "includes.h"
2009-02-24 01:35:49 +03:00
# include "dlinklist.h"
2008-08-20 07:22:16 +04:00
# include "dsdb/samdb/samdb.h"
2009-01-31 00:31:19 +03:00
# include "lib/ldb/include/ldb_module.h"
2008-08-20 07:22:16 +04:00
# include "param/param.h"
2008-08-20 09:46:46 +04:00
2008-08-21 06:58:00 +04:00
static int dsdb_schema_set_attributes ( struct ldb_context * ldb , struct dsdb_schema * schema , bool write_attributes )
2008-08-20 09:46:46 +04:00
{
int ret = LDB_SUCCESS ;
2008-08-21 06:58:00 +04:00
struct ldb_result * res ;
struct ldb_result * res_idx ;
2008-08-20 09:46:46 +04:00
struct dsdb_attribute * attr ;
2008-08-21 06:58:00 +04:00
struct ldb_message * mod_msg ;
TALLOC_CTX * mem_ctx = talloc_new ( ldb ) ;
2008-08-20 09:46:46 +04:00
2008-08-21 06:58:00 +04:00
struct ldb_message * msg ;
struct ldb_message * msg_idx ;
if ( ! mem_ctx ) {
return LDB_ERR_OPERATIONS_ERROR ;
}
msg = ldb_msg_new ( mem_ctx ) ;
if ( ! msg ) {
ldb_oom ( ldb ) ;
return LDB_ERR_OPERATIONS_ERROR ;
}
msg_idx = ldb_msg_new ( mem_ctx ) ;
if ( ! msg_idx ) {
ldb_oom ( ldb ) ;
return LDB_ERR_OPERATIONS_ERROR ;
}
msg - > dn = ldb_dn_new ( msg , ldb , " @ATTRIBUTES " ) ;
if ( ! msg - > dn ) {
ldb_oom ( ldb ) ;
return LDB_ERR_OPERATIONS_ERROR ;
}
msg_idx - > dn = ldb_dn_new ( msg , ldb , " @INDEXLIST " ) ;
if ( ! msg_idx - > dn ) {
ldb_oom ( ldb ) ;
return LDB_ERR_OPERATIONS_ERROR ;
}
2008-08-20 09:46:46 +04:00
for ( attr = schema - > attributes ; attr ; attr = attr - > next ) {
const struct ldb_schema_syntax * s ;
2008-08-21 06:58:00 +04:00
const char * syntax = attr - > syntax - > ldb_syntax ;
if ( ! syntax ) {
syntax = attr - > syntax - > ldap_oid ;
}
/* Write out a rough approximation of the schema as an @ATTRIBUTES value, for bootstrapping */
if ( strcmp ( syntax , LDB_SYNTAX_INTEGER ) = = 0 ) {
ret = ldb_msg_add_string ( msg , attr - > lDAPDisplayName , " INTEGER " ) ;
} else if ( strcmp ( syntax , LDB_SYNTAX_DIRECTORY_STRING ) = = 0 ) {
ret = ldb_msg_add_string ( msg , attr - > lDAPDisplayName , " CASE_INSENSITIVE " ) ;
}
if ( ret ! = LDB_SUCCESS ) {
2008-09-12 01:45:28 +04:00
break ;
2008-08-21 06:58:00 +04:00
}
if ( attr - > searchFlags & SEARCH_FLAG_ATTINDEX ) {
ret = ldb_msg_add_string ( msg_idx , " @IDXATTR " , attr - > lDAPDisplayName ) ;
if ( ret ! = LDB_SUCCESS ) {
2008-09-12 01:45:28 +04:00
break ;
2008-08-21 06:58:00 +04:00
}
}
2008-08-20 09:46:46 +04:00
if ( ! attr - > syntax ) {
continue ;
}
2008-08-21 06:58:00 +04:00
ret = ldb_schema_attribute_add ( ldb , attr - > lDAPDisplayName , LDB_ATTR_FLAG_FIXED ,
syntax ) ;
if ( ret ! = LDB_SUCCESS ) {
s = ldb_samba_syntax_by_name ( ldb , attr - > syntax - > ldap_oid ) ;
2008-08-20 09:46:46 +04:00
if ( s ) {
2008-08-21 06:58:00 +04:00
ret = ldb_schema_attribute_add_with_syntax ( ldb , attr - > lDAPDisplayName , LDB_ATTR_FLAG_FIXED , s ) ;
2008-08-20 09:46:46 +04:00
} else {
2008-08-21 06:58:00 +04:00
ret = LDB_SUCCESS ; /* Nothing to do here */
2008-08-20 09:46:46 +04:00
}
}
2008-08-21 06:58:00 +04:00
2008-08-20 09:46:46 +04:00
if ( ret ! = LDB_SUCCESS ) {
2008-09-12 01:45:28 +04:00
break ;
2008-08-20 09:46:46 +04:00
}
}
2008-09-12 01:45:28 +04:00
if ( ! write_attributes | | ret ! = LDB_SUCCESS ) {
2008-08-21 06:58:00 +04:00
talloc_free ( mem_ctx ) ;
return ret ;
}
2008-08-21 10:42:03 +04:00
2008-08-21 06:58:00 +04:00
/* Try to avoid churning the attributes too much - we only want to do this if they have changed */
2008-09-23 22:30:06 +04:00
ret = ldb_search ( ldb , mem_ctx , & res , msg - > dn , LDB_SCOPE_BASE , NULL , " dn=%s " , ldb_dn_get_linearized ( msg - > dn ) ) ;
2008-08-21 06:58:00 +04:00
if ( ret = = LDB_ERR_NO_SUCH_OBJECT ) {
ret = ldb_add ( ldb , msg ) ;
} else if ( ret ! = LDB_SUCCESS ) {
2008-09-12 01:45:28 +04:00
} else if ( res - > count ! = 1 ) {
ret = ldb_add ( ldb , msg ) ;
2008-08-21 06:58:00 +04:00
} else {
ret = LDB_SUCCESS ;
2008-08-21 10:42:03 +04:00
/* Annoyingly added to our search results */
ldb_msg_remove_attr ( res - > msgs [ 0 ] , " distinguishedName " ) ;
2008-09-12 01:45:28 +04:00
2008-08-21 06:58:00 +04:00
mod_msg = ldb_msg_diff ( ldb , res - > msgs [ 0 ] , msg ) ;
if ( mod_msg - > num_elements > 0 ) {
ret = ldb_modify ( ldb , mod_msg ) ;
}
}
if ( ret = = LDB_ERR_INSUFFICIENT_ACCESS_RIGHTS ) {
/* We might be on a read-only DB */
2008-09-12 01:45:28 +04:00
ret = LDB_SUCCESS ;
}
if ( ret ! = LDB_SUCCESS ) {
2008-08-21 06:58:00 +04:00
talloc_free ( mem_ctx ) ;
return ret ;
}
/* Now write out the indexs, as found in the schema (if they have changed) */
2008-09-23 22:30:06 +04:00
ret = ldb_search ( ldb , mem_ctx , & res_idx , msg_idx - > dn , LDB_SCOPE_BASE , NULL , " dn=%s " , ldb_dn_get_linearized ( msg_idx - > dn ) ) ;
2008-08-21 06:58:00 +04:00
if ( ret = = LDB_ERR_NO_SUCH_OBJECT ) {
ret = ldb_add ( ldb , msg_idx ) ;
} else if ( ret ! = LDB_SUCCESS ) {
2008-09-12 01:45:28 +04:00
} else if ( res - > count ! = 1 ) {
ret = ldb_add ( ldb , msg_idx ) ;
2008-08-21 06:58:00 +04:00
} else {
2008-09-12 01:45:28 +04:00
ret = LDB_SUCCESS ;
2008-08-21 10:42:03 +04:00
/* Annoyingly added to our search results */
ldb_msg_remove_attr ( res_idx - > msgs [ 0 ] , " distinguishedName " ) ;
2008-08-21 06:58:00 +04:00
mod_msg = ldb_msg_diff ( ldb , res_idx - > msgs [ 0 ] , msg_idx ) ;
if ( mod_msg - > num_elements > 0 ) {
ret = ldb_modify ( ldb , mod_msg ) ;
}
}
if ( ret = = LDB_ERR_INSUFFICIENT_ACCESS_RIGHTS ) {
/* We might be on a read-only DB */
2008-09-11 14:51:26 +04:00
ret = LDB_SUCCESS ;
2008-08-21 06:58:00 +04:00
}
talloc_free ( mem_ctx ) ;
return ret ;
2008-08-20 09:46:46 +04:00
}
2008-08-20 07:22:16 +04:00
/**
* Attach the schema to an opaque pointer on the ldb , so ldb modules
* can find it
*/
int dsdb_set_schema ( struct ldb_context * ldb , struct dsdb_schema * schema )
{
int ret ;
ret = ldb_set_opaque ( ldb , " dsdb_schema " , schema ) ;
if ( ret ! = LDB_SUCCESS ) {
return ret ;
}
2008-08-21 06:58:00 +04:00
/* Set the new attributes based on the new schema */
ret = dsdb_schema_set_attributes ( ldb , schema , true ) ;
2008-08-20 09:46:46 +04:00
if ( ret ! = LDB_SUCCESS ) {
return ret ;
}
2008-08-20 07:22:16 +04:00
talloc_steal ( ldb , schema ) ;
return LDB_SUCCESS ;
}
/**
* Global variable to hold one copy of the schema , used to avoid memory bloat
*/
static struct dsdb_schema * global_schema ;
/**
* Make this ldb use the ' global ' schema , setup to avoid having multiple copies in this process
*/
int dsdb_set_global_schema ( struct ldb_context * ldb )
{
int ret ;
if ( ! global_schema ) {
return LDB_SUCCESS ;
}
ret = ldb_set_opaque ( ldb , " dsdb_schema " , global_schema ) ;
if ( ret ! = LDB_SUCCESS ) {
return ret ;
}
2008-08-21 06:58:00 +04:00
/* Set the new attributes based on the new schema */
ret = dsdb_schema_set_attributes ( ldb , global_schema , false ) ;
2008-08-20 09:46:46 +04:00
if ( ret ! = LDB_SUCCESS ) {
return ret ;
}
2008-08-20 07:22:16 +04:00
/* Keep a reference to this schema, just incase the global copy is replaced */
if ( talloc_reference ( ldb , global_schema ) = = NULL ) {
return LDB_ERR_OPERATIONS_ERROR ;
}
return LDB_SUCCESS ;
}
/**
* Find the schema object for this ldb
*/
struct dsdb_schema * dsdb_get_schema ( struct ldb_context * ldb )
{
const void * p ;
struct dsdb_schema * schema ;
/* see if we have a cached copy */
p = ldb_get_opaque ( ldb , " dsdb_schema " ) ;
if ( ! p ) {
return NULL ;
}
schema = talloc_get_type ( p , struct dsdb_schema ) ;
if ( ! schema ) {
return NULL ;
}
return schema ;
}
/**
* Make the schema found on this ldb the ' global ' schema
*/
void dsdb_make_schema_global ( struct ldb_context * ldb )
{
struct dsdb_schema * schema = dsdb_get_schema ( ldb ) ;
if ( ! schema ) {
return ;
}
if ( global_schema ) {
talloc_unlink ( talloc_autofree_context ( ) , schema ) ;
}
talloc_steal ( talloc_autofree_context ( ) , schema ) ;
global_schema = schema ;
dsdb_set_global_schema ( ldb ) ;
}
/**
* Rather than read a schema from the LDB itself , read it from an ldif
* file . This allows schema to be loaded and used while adding the
* schema itself to the directory .
*/
WERROR dsdb_attach_schema_from_ldif_file ( struct ldb_context * ldb , const char * pf , const char * df )
{
struct ldb_ldif * ldif ;
struct ldb_message * msg ;
TALLOC_CTX * mem_ctx ;
WERROR status ;
int ret ;
struct dsdb_schema * schema ;
const struct ldb_val * prefix_val ;
const struct ldb_val * info_val ;
struct ldb_val info_val_default ;
mem_ctx = talloc_new ( ldb ) ;
if ( ! mem_ctx ) {
goto nomem ;
}
schema = dsdb_new_schema ( mem_ctx , lp_iconv_convenience ( ldb_get_opaque ( ldb , " loadparm " ) ) ) ;
schema - > fsmo . we_are_master = true ;
schema - > fsmo . master_dn = ldb_dn_new_fmt ( schema , ldb , " @PROVISION_SCHEMA_MASTER " ) ;
if ( ! schema - > fsmo . master_dn ) {
goto nomem ;
}
/*
* load the prefixMap attribute from pf
*/
ldif = ldb_ldif_read_string ( ldb , & pf ) ;
if ( ! ldif ) {
status = WERR_INVALID_PARAM ;
goto failed ;
}
talloc_steal ( mem_ctx , ldif ) ;
msg = ldb_msg_canonicalize ( ldb , ldif - > msg ) ;
if ( ! msg ) {
goto nomem ;
}
talloc_steal ( mem_ctx , msg ) ;
talloc_free ( ldif ) ;
prefix_val = ldb_msg_find_ldb_val ( msg , " prefixMap " ) ;
if ( ! prefix_val ) {
status = WERR_INVALID_PARAM ;
goto failed ;
}
info_val = ldb_msg_find_ldb_val ( msg , " schemaInfo " ) ;
if ( ! info_val ) {
2008-10-18 20:09:04 +04:00
info_val_default = strhex_to_data_blob ( mem_ctx , " FF0000000000000000000000000000000000000000 " ) ;
2008-08-20 07:22:16 +04:00
if ( ! info_val_default . data ) {
goto nomem ;
}
info_val = & info_val_default ;
}
status = dsdb_load_oid_mappings_ldb ( schema , prefix_val , info_val ) ;
if ( ! W_ERROR_IS_OK ( status ) ) {
goto failed ;
}
/*
* load the attribute and class definitions outof df
*/
while ( ( ldif = ldb_ldif_read_string ( ldb , & df ) ) ) {
bool is_sa ;
bool is_sc ;
talloc_steal ( mem_ctx , ldif ) ;
msg = ldb_msg_canonicalize ( ldb , ldif - > msg ) ;
if ( ! msg ) {
goto nomem ;
}
talloc_steal ( mem_ctx , msg ) ;
talloc_free ( ldif ) ;
is_sa = ldb_msg_check_string_attribute ( msg , " objectClass " , " attributeSchema " ) ;
is_sc = ldb_msg_check_string_attribute ( msg , " objectClass " , " classSchema " ) ;
if ( is_sa ) {
struct dsdb_attribute * sa ;
sa = talloc_zero ( schema , struct dsdb_attribute ) ;
if ( ! sa ) {
goto nomem ;
}
status = dsdb_attribute_from_ldb ( schema , msg , sa , sa ) ;
if ( ! W_ERROR_IS_OK ( status ) ) {
goto failed ;
}
DLIST_ADD_END ( schema - > attributes , sa , struct dsdb_attribute * ) ;
} else if ( is_sc ) {
struct dsdb_class * sc ;
sc = talloc_zero ( schema , struct dsdb_class ) ;
if ( ! sc ) {
goto nomem ;
}
status = dsdb_class_from_ldb ( schema , msg , sc , sc ) ;
if ( ! W_ERROR_IS_OK ( status ) ) {
goto failed ;
}
DLIST_ADD_END ( schema - > classes , sc , struct dsdb_class * ) ;
}
}
ret = dsdb_set_schema ( ldb , schema ) ;
if ( ret ! = LDB_SUCCESS ) {
status = WERR_FOOBAR ;
goto failed ;
}
goto done ;
nomem :
status = WERR_NOMEM ;
failed :
done :
talloc_free ( mem_ctx ) ;
return status ;
}