1
0
mirror of https://github.com/samba-team/samba.git synced 2025-03-27 22:50:26 +03:00

s4:provision Add C binding to get at the generate schema

This will allow us to do local tests against that schema
This commit is contained in:
Andrew Bartlett 2009-11-10 15:18:52 +11:00
parent 6437c38aeb
commit 6b0b3fed31
3 changed files with 86 additions and 0 deletions

View File

@ -44,6 +44,14 @@ static PyObject *provision_module(void)
return PyImport_Import(name);
}
static PyObject *schema_module(void)
{
PyObject *name = PyString_FromString("samba.schema");
if (name == NULL)
return NULL;
return PyImport_Import(name);
}
NTSTATUS provision_bare(TALLOC_CTX *mem_ctx, struct loadparm_context *lp_ctx,
struct provision_settings *settings,
struct provision_result *result)
@ -298,3 +306,56 @@ failure:
PyErr_Clear();
return NT_STATUS_UNSUCCESSFUL;
}
struct ldb_context *provision_get_schema(TALLOC_CTX *mem_ctx, struct loadparm_context *lp_ctx)
{
const char *setupdir;
PyObject *schema_mod, *schema_dict, *schema_fn, *py_result, *parameters;
DEBUG(0,("Schema for DRS tests using python\n"));
py_load_samba_modules();
Py_Initialize();
py_update_path("bin"); /* FIXME: Can't assume this is always the case */
schema_mod = schema_module();
if (schema_mod == NULL) {
PyErr_Print();
DEBUG(0, ("Unable to import schema Python module.\n"));
return NULL;
}
schema_dict = PyModule_GetDict(schema_mod);
if (schema_dict == NULL) {
DEBUG(0, ("Unable to get dictionary for schema module\n"));
return NULL;
}
schema_fn = PyDict_GetItemString(schema_dict, "ldb_with_schema");
if (schema_fn == NULL) {
PyErr_Print();
DEBUG(0, ("Unable to get schema_get_ldb function\n"));
return NULL;
}
parameters = PyDict_New();
setupdir = lp_setupdir(lp_ctx);
PyDict_SetItemString(parameters, "setup_dir",
PyString_FromString(setupdir));
py_result = PyEval_CallObjectWithKeywords(schema_fn, NULL, parameters);
Py_DECREF(parameters);
if (py_result == NULL) {
PyErr_Print();
PyErr_Clear();
return NULL;
}
return PyLdb_AsLdbContext(PyObject_GetAttrString(py_result, "ldb"));
}

View File

@ -64,4 +64,6 @@ NTSTATUS provision_store_self_join(TALLOC_CTX *mem_ctx, struct loadparm_context
struct provision_store_self_join_settings *settings,
const char **error_string);
struct ldb_context *provision_get_schema(TALLOC_CTX *mem_ctx, struct loadparm_context *lp_ctx);
#endif /* _PROVISION_H_ */

View File

@ -32,6 +32,7 @@ from samba import read_and_sub_file, substitute_var, check_all_substituted
from samba import Ldb
from samba.ndr import ndr_pack
from ldb import SCOPE_SUBTREE, SCOPE_ONELEVEL, SCOPE_BASE
import os
def get_schema_descriptor(domain_sid):
sddl = "O:SAG:SAD:(A;CI;RPLCLORC;;;AU)(A;CI;RPWPCRCCLCLORCWOWDSW;;;SA)" \
@ -138,3 +139,25 @@ def get_dnsyntax_attributes(schemadn,schemaldb):
return attributes
def ldb_with_schema(setup_dir=None, schemadn="cn=schema,cn=configuration,dc=example,dc=com",
serverdn="cn=server,cn=servers,cn=default-first-site-name,cn=sites,cn=cn=configuration,dc=example,dc=com",
domainsid=None):
"""Load schema for the SamDB from the AD schema files and samba4_schema.ldif
:param setup_dir: Setup path
:param schemadn: DN of the schema
:param serverdn: DN of the server
Returns the schema data loaded as an object, with .ldb being a
new ldb with the schema loaded. This allows certain tests to
operate without a remote or local schema.
"""
def setup_path(file):
return os.path.join(setup_dir, file)
if domainsid is None:
domainsid = security.random_sid()
else:
domainsid = security.dom_sid(domainsid)
return Schema(setup_path, domainsid, schemadn=schemadn, serverdn=serverdn)