source4/dsdb/pydsdb.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2007-2010
   4    Copyright (C) Matthias Dieter Wallnöfer          2009
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20 #include <Python.h>
  21 #include "includes.h"
  22 #include <ldb.h>
  23 #include <pyldb.h>
  24 #include "dsdb/samdb/samdb.h"
  25 #include "libcli/security/security.h"
  26 #include "librpc/ndr/libndr.h"
  27 #include "system/kerberos.h"
  28 #include "auth/kerberos/kerberos.h"
  29 #include "librpc/rpc/pyrpc_util.h"
  30 #include "lib/policy/policy.h"
  31
  32 void initdsdb(void);
  33
  34 /* There's no Py_ssize_t in 2.4, apparently */
  35 #if PY_MAJOR_VERSION == 2 && PY_MINOR_VERSION < 5
  36 typedef int Py_ssize_t;
  37 typedef inquiry lenfunc;
  38 typedef intargfunc ssizeargfunc;
  39 #endif
  40
  41 /* FIXME: These should be in a header file somewhere, once we finish moving
  42  * away from SWIG .. */
  43 #define PyErr_LDB_OR_RAISE(py_ldb, ldb) \
  44 /*      if (!PyLdb_Check(py_ldb)) { \
  45                 PyErr_SetString(py_ldb_get_exception(), "Ldb connection object required"); \
  46                 return NULL; \
  47         } */\
  48         ldb = PyLdb_AsLdbContext(py_ldb);
  49
  50 static PyObject *py_ldb_get_exception(void)
  51 {
  52         PyObject *mod = PyImport_ImportModule("ldb");
  53         if (mod == NULL)
  54                 return NULL;
  55
  56         return PyObject_GetAttrString(mod, "LdbError");
  57 }
  58
  59 static void PyErr_SetLdbError(PyObject *error, int ret, struct ldb_context *ldb_ctx)
  60 {
  61         if (ret == LDB_ERR_PYTHON_EXCEPTION)
  62                 return; /* Python exception should already be set, just keep that */
  63
  64         PyErr_SetObject(error,
  65                         Py_BuildValue(discard_const_p(char, "(i,s)"), ret,
  66                         ldb_ctx == NULL?ldb_strerror(ret):ldb_errstring(ldb_ctx)));
  67 }
  68
  69 static PyObject *py_samdb_server_site_name(PyObject *self, PyObject *args)
  70 {
  71         PyObject *py_ldb, *result;
  72         struct ldb_context *ldb;
  73         const char *site;
  74         TALLOC_CTX *mem_ctx;
  75
  76         if (!PyArg_ParseTuple(args, "O", &py_ldb))
  77                 return NULL;
  78
  79         PyErr_LDB_OR_RAISE(py_ldb, ldb);
  80
  81         mem_ctx = talloc_new(NULL);
  82         if (mem_ctx == NULL) {
  83                 PyErr_NoMemory();
  84                 return NULL;
  85         }
  86
  87         site = samdb_server_site_name(ldb, mem_ctx);
  88         if (site == NULL) {
  89                 PyErr_SetString(PyExc_RuntimeError, "Failed to find server site");
  90                 talloc_free(mem_ctx);
  91                 return NULL;
  92         }
  93
  94         result = PyString_FromString(site);
  95         talloc_free(mem_ctx);
  96         return result;
  97 }
  98
  99 static PyObject *py_dsdb_convert_schema_to_openldap(PyObject *self,
 100                                                                                                         PyObject *args)
 101 {
 102         char *target_str, *mapping;
 103         PyObject *py_ldb;
 104         struct ldb_context *ldb;
 105         PyObject *ret;
 106         char *retstr;
 107
 108         if (!PyArg_ParseTuple(args, "Oss", &py_ldb, &target_str, &mapping))
 109                 return NULL;
 110
 111         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 112
 113         retstr = dsdb_convert_schema_to_openldap(ldb, target_str, mapping);
 114         if (retstr == NULL) {
 115                 PyErr_SetString(PyExc_RuntimeError,
 116                                                 "dsdb_convert_schema_to_openldap failed");
 117                 return NULL;
 118         }
 119
 120         ret = PyString_FromString(retstr);
 121         talloc_free(retstr);
 122         return ret;
 123 }
 124
 125 static PyObject *py_samdb_set_domain_sid(PyLdbObject *self, PyObject *args)
 126 {
 127         PyObject *py_ldb, *py_sid;
 128         struct ldb_context *ldb;
 129         struct dom_sid *sid;
 130         bool ret;
 131
 132         if (!PyArg_ParseTuple(args, "OO", &py_ldb, &py_sid))
 133                 return NULL;
 134
 135         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 136
 137         sid = dom_sid_parse_talloc(NULL, PyString_AsString(py_sid));
 138         if (sid == NULL) {
 139                 PyErr_NoMemory();
 140                 return NULL;
 141         }
 142
 143         ret = samdb_set_domain_sid(ldb, sid);
 144         talloc_free(sid);
 145         if (!ret) {
 146                 PyErr_SetString(PyExc_RuntimeError, "set_domain_sid failed");
 147                 return NULL;
 148         }
 149         Py_RETURN_NONE;
 150 }
 151
 152 static PyObject *py_samdb_set_ntds_settings_dn(PyLdbObject *self, PyObject *args)
 153 {
 154         PyObject *py_ldb, *py_ntds_settings_dn;
 155         struct ldb_context *ldb;
 156         struct ldb_dn *ntds_settings_dn;
 157         TALLOC_CTX *tmp_ctx;
 158         bool ret;
 159
 160         if (!PyArg_ParseTuple(args, "OO", &py_ldb, &py_ntds_settings_dn))
 161                 return NULL;
 162
 163         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 164
 165         tmp_ctx = talloc_new(NULL);
 166         if (tmp_ctx == NULL) {
 167                 PyErr_NoMemory();
 168                 return NULL;
 169         }
 170
 171         if (!PyObject_AsDn(tmp_ctx, py_ntds_settings_dn, ldb, &ntds_settings_dn)) {
 172                 /* exception thrown by "PyObject_AsDn" */
 173                 talloc_free(tmp_ctx);
 174                 return NULL;
 175         }
 176
 177         ret = samdb_set_ntds_settings_dn(ldb, ntds_settings_dn);
 178         talloc_free(tmp_ctx);
 179         if (!ret) {
 180                 PyErr_SetString(PyExc_RuntimeError, "set_ntds_settings_dn failed");
 181                 return NULL;
 182         }
 183         Py_RETURN_NONE;
 184 }
 185
 186 static PyObject *py_samdb_get_domain_sid(PyLdbObject *self, PyObject *args)
 187 {
 188         PyObject *py_ldb;
 189         struct ldb_context *ldb;
 190         const struct dom_sid *sid;
 191         PyObject *ret;
 192         char *retstr;
 193
 194         if (!PyArg_ParseTuple(args, "O", &py_ldb))
 195                 return NULL;
 196
 197         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 198
 199         sid = samdb_domain_sid(ldb);
 200         if (!sid) {
 201                 PyErr_SetString(PyExc_RuntimeError, "samdb_domain_sid failed");
 202                 return NULL;
 203         }
 204
 205         retstr = dom_sid_string(NULL, sid);
 206         if (retstr == NULL) {
 207                 PyErr_NoMemory();
 208                 return NULL;
 209         }
 210         ret = PyString_FromString(retstr);
 211         talloc_free(retstr);
 212         return ret;
 213 }
 214
 215 static PyObject *py_samdb_ntds_invocation_id(PyObject *self, PyObject *args)
 216 {
 217         PyObject *py_ldb, *result;
 218         struct ldb_context *ldb;
 219         const struct GUID *guid;
 220         char *retstr;
 221
 222         if (!PyArg_ParseTuple(args, "O", &py_ldb)) {
 223                 return NULL;
 224         }
 225
 226         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 227
 228         guid = samdb_ntds_invocation_id(ldb);
 229         if (guid == NULL) {
 230                 PyErr_SetString(PyExc_RuntimeError,
 231                                                 "Failed to find NTDS invocation ID");
 232                 return NULL;
 233         }
 234
 235         retstr = GUID_string(NULL, guid);
 236         if (retstr == NULL) {
 237                 PyErr_NoMemory();
 238                 return NULL;
 239         }
 240         result = PyString_FromString(retstr);
 241         talloc_free(retstr);
 242         return result;
 243 }
 244
 245 static PyObject *py_dsdb_get_oid_from_attid(PyObject *self, PyObject *args)
 246 {
 247         PyObject *py_ldb;
 248         struct ldb_context *ldb;
 249         uint32_t attid;
 250         struct dsdb_schema *schema;
 251         const char *oid;
 252         PyObject *ret;
 253         WERROR status;
 254         TALLOC_CTX *mem_ctx;
 255
 256         if (!PyArg_ParseTuple(args, "Oi", &py_ldb, &attid))
 257                 return NULL;
 258
 259         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 260
 261         mem_ctx = talloc_new(NULL);
 262         if (!mem_ctx) {
 263                 PyErr_NoMemory();
 264                 return NULL;
 265         }
 266
 267         schema = dsdb_get_schema(ldb, mem_ctx);
 268         if (!schema) {
 269                 PyErr_SetString(PyExc_RuntimeError, "Failed to find a schema from ldb \n");
 270                 talloc_free(mem_ctx);
 271                 return NULL;
 272         }
 273
 274         status = dsdb_schema_pfm_oid_from_attid(schema->prefixmap, attid,
 275                                                 mem_ctx, &oid);
 276         if (!W_ERROR_IS_OK(status)) {
 277                 PyErr_SetWERROR(status);
 278                 talloc_free(mem_ctx);
 279                 return NULL;
 280         }
 281
 282         ret = PyString_FromString(oid);
 283
 284         talloc_free(mem_ctx);
 285
 286         return ret;
 287 }
 288
 289
 290 static PyObject *py_dsdb_get_attid_from_lDAPDisplayName(PyObject *self, PyObject *args)
 291 {
 292         PyObject *py_ldb, *is_schema_nc;
 293         struct ldb_context *ldb;
 294         struct dsdb_schema *schema;
 295         const char *ldap_display_name;
 296         bool schema_nc = false;
 297         const struct dsdb_attribute *a;
 298         uint32_t attid;
 299
 300         if (!PyArg_ParseTuple(args, "OsO", &py_ldb, &ldap_display_name, &is_schema_nc))
 301                 return NULL;
 302
 303         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 304
 305         if (is_schema_nc) {
 306                 if (!PyBool_Check(is_schema_nc)) {
 307                         PyErr_SetString(PyExc_TypeError, "Expected boolean is_schema_nc");
 308                         return NULL;
 309                 }
 310                 if (is_schema_nc == Py_True) {
 311                         schema_nc = true;
 312                 }
 313         }
 314
 315         schema = dsdb_get_schema(ldb, NULL);
 316
 317         if (!schema) {
 318                 PyErr_SetString(PyExc_RuntimeError, "Failed to find a schema from ldb");
 319                 return NULL;
 320         }
 321
 322         a = dsdb_attribute_by_lDAPDisplayName(schema, ldap_display_name);
 323         if (a == NULL) {
 324                 PyErr_Format(PyExc_RuntimeError, "Failed to find attribute '%s'", ldap_display_name);
 325                 return NULL;
 326         }
 327
 328         attid = dsdb_attribute_get_attid(a, schema_nc);
 329
 330         return PyLong_FromUnsignedLong(attid);
 331 }
 332
 333 /*
 334   return the attribute syntax oid as a string from the attribute name
 335  */
 336 static PyObject *py_dsdb_get_syntax_oid_from_lDAPDisplayName(PyObject *self, PyObject *args)
 337 {
 338         PyObject *py_ldb;
 339         struct ldb_context *ldb;
 340         struct dsdb_schema *schema;
 341         const char *ldap_display_name;
 342         const struct dsdb_attribute *attribute;
 343
 344         if (!PyArg_ParseTuple(args, "Os", &py_ldb, &ldap_display_name))
 345                 return NULL;
 346
 347         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 348
 349         schema = dsdb_get_schema(ldb, NULL);
 350
 351         if (!schema) {
 352                 PyErr_SetString(PyExc_RuntimeError, "Failed to find a schema from ldb");
 353                 return NULL;
 354         }
 355
 356         attribute = dsdb_attribute_by_lDAPDisplayName(schema, ldap_display_name);
 357         if (attribute == NULL) {
 358                 PyErr_Format(PyExc_RuntimeError, "Failed to find attribute '%s'", ldap_display_name);
 359                 return NULL;
 360         }
 361
 362         return PyString_FromString(attribute->syntax->ldap_oid);
 363 }
 364
 365 /*
 366   convert a python string to a DRSUAPI drsuapi_DsReplicaAttribute attribute
 367  */
 368 static PyObject *py_dsdb_DsReplicaAttribute(PyObject *self, PyObject *args)
 369 {
 370         PyObject *py_ldb, *el_list, *ret;
 371         struct ldb_context *ldb;
 372         char *ldap_display_name;
 373         const struct dsdb_attribute *a;
 374         struct dsdb_schema *schema;
 375         struct dsdb_syntax_ctx syntax_ctx;
 376         struct ldb_message_element *el;
 377         struct drsuapi_DsReplicaAttribute *attr;
 378         TALLOC_CTX *tmp_ctx;
 379         WERROR werr;
 380         Py_ssize_t i;
 381
 382         if (!PyArg_ParseTuple(args, "OsO", &py_ldb, &ldap_display_name, &el_list)) {
 383                 return NULL;
 384         }
 385
 386         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 387
 388         if (!PyList_Check(el_list)) {
 389                 PyErr_Format(PyExc_TypeError, "ldif_elements must be a list");
 390                 return NULL;
 391         }
 392
 393         schema = dsdb_get_schema(ldb, NULL);
 394         if (!schema) {
 395                 PyErr_SetString(PyExc_RuntimeError, "Failed to find a schema from ldb");
 396                 return NULL;
 397         }
 398
 399         a = dsdb_attribute_by_lDAPDisplayName(schema, ldap_display_name);
 400         if (a == NULL) {
 401                 PyErr_Format(PyExc_RuntimeError, "Failed to find attribute '%s'", ldap_display_name);
 402                 return NULL;
 403         }
 404
 405         dsdb_syntax_ctx_init(&syntax_ctx, ldb, schema);
 406         syntax_ctx.is_schema_nc = false;
 407
 408         tmp_ctx = talloc_new(ldb);
 409         if (tmp_ctx == NULL) {
 410                 PyErr_NoMemory();
 411                 return NULL;
 412         }
 413
 414         el = talloc_zero(tmp_ctx, struct ldb_message_element);
 415         if (el == NULL) {
 416                 PyErr_NoMemory();
 417                 talloc_free(tmp_ctx);
 418                 return NULL;
 419         }
 420
 421         el->name = ldap_display_name;
 422         el->num_values = PyList_Size(el_list);
 423
 424         el->values = talloc_array(el, struct ldb_val, el->num_values);
 425         if (el->values == NULL) {
 426                 PyErr_NoMemory();
 427                 talloc_free(tmp_ctx);
 428                 return NULL;
 429         }
 430
 431         for (i = 0; i < el->num_values; i++) {
 432                 PyObject *item = PyList_GetItem(el_list, i);
 433                 if (!PyString_Check(item)) {
 434                         PyErr_Format(PyExc_TypeError, "ldif_elements should be strings");
 435                         return NULL;
 436                 }
 437                 el->values[i].data = (uint8_t *)PyString_AsString(item);
 438                 el->values[i].length = PyString_Size(item);
 439         }
 440
 441         attr = talloc_zero(tmp_ctx, struct drsuapi_DsReplicaAttribute);
 442         if (attr == NULL) {
 443                 PyErr_NoMemory();
 444                 talloc_free(tmp_ctx);
 445                 return NULL;
 446         }
 447
 448         werr = a->syntax->ldb_to_drsuapi(&syntax_ctx, a, el, attr, attr);
 449         PyErr_WERROR_IS_ERR_RAISE(werr);
 450
 451         ret = py_return_ndr_struct("samba.dcerpc.drsuapi", "DsReplicaAttribute", attr, attr);
 452
 453         talloc_free(tmp_ctx);
 454
 455         return ret;
 456 }
 457
 458
 459 /*
 460   normalise a ldb attribute list
 461  */
 462 static PyObject *py_dsdb_normalise_attributes(PyObject *self, PyObject *args)
 463 {
 464         PyObject *py_ldb, *el_list, *ret;
 465         struct ldb_context *ldb;
 466         char *ldap_display_name;
 467         const struct dsdb_attribute *a;
 468         struct dsdb_schema *schema;
 469         struct dsdb_syntax_ctx syntax_ctx;
 470         struct ldb_message_element *el;
 471         struct drsuapi_DsReplicaAttribute *attr;
 472         TALLOC_CTX *tmp_ctx;
 473         WERROR werr;
 474         Py_ssize_t i;
 475
 476         if (!PyArg_ParseTuple(args, "OsO", &py_ldb, &ldap_display_name, &el_list)) {
 477                 return NULL;
 478         }
 479
 480         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 481
 482         if (!PyList_Check(el_list)) {
 483                 PyErr_Format(PyExc_TypeError, "ldif_elements must be a list");
 484                 return NULL;
 485         }
 486
 487         schema = dsdb_get_schema(ldb, NULL);
 488         if (!schema) {
 489                 PyErr_SetString(PyExc_RuntimeError, "Failed to find a schema from ldb");
 490                 return NULL;
 491         }
 492
 493         a = dsdb_attribute_by_lDAPDisplayName(schema, ldap_display_name);
 494         if (a == NULL) {
 495                 PyErr_Format(PyExc_RuntimeError, "Failed to find attribute '%s'", ldap_display_name);
 496                 return NULL;
 497         }
 498
 499         dsdb_syntax_ctx_init(&syntax_ctx, ldb, schema);
 500         syntax_ctx.is_schema_nc = false;
 501
 502         tmp_ctx = talloc_new(ldb);
 503         if (tmp_ctx == NULL) {
 504                 PyErr_NoMemory();
 505                 return NULL;
 506         }
 507
 508         el = talloc_zero(tmp_ctx, struct ldb_message_element);
 509         if (el == NULL) {
 510                 PyErr_NoMemory();
 511                 talloc_free(tmp_ctx);
 512                 return NULL;
 513         }
 514
 515         el->name = ldap_display_name;
 516         el->num_values = PyList_Size(el_list);
 517
 518         el->values = talloc_array(el, struct ldb_val, el->num_values);
 519         if (el->values == NULL) {
 520                 PyErr_NoMemory();
 521                 talloc_free(tmp_ctx);
 522                 return NULL;
 523         }
 524
 525         for (i = 0; i < el->num_values; i++) {
 526                 PyObject *item = PyList_GetItem(el_list, i);
 527                 if (!PyString_Check(item)) {
 528                         PyErr_Format(PyExc_TypeError, "ldif_elements should be strings");
 529                         return NULL;
 530                 }
 531                 el->values[i].data = (uint8_t *)PyString_AsString(item);
 532                 el->values[i].length = PyString_Size(item);
 533         }
 534
 535         /* first run ldb_to_drsuapi, then convert back again. This has
 536          * the effect of normalising the attributes
 537          */
 538
 539         attr = talloc_zero(tmp_ctx, struct drsuapi_DsReplicaAttribute);
 540         if (attr == NULL) {
 541                 PyErr_NoMemory();
 542                 talloc_free(tmp_ctx);
 543                 return NULL;
 544         }
 545
 546         werr = a->syntax->ldb_to_drsuapi(&syntax_ctx, a, el, attr, attr);
 547         PyErr_WERROR_IS_ERR_RAISE(werr);
 548
 549         /* now convert back again */
 550         werr = a->syntax->drsuapi_to_ldb(&syntax_ctx, a, attr, el, el);
 551         PyErr_WERROR_IS_ERR_RAISE(werr);
 552
 553         ret = py_return_ndr_struct("ldb", "MessageElement", el, el);
 554
 555         talloc_free(tmp_ctx);
 556
 557         return ret;
 558 }
 559
 560
 561 static PyObject *py_dsdb_set_ntds_invocation_id(PyObject *self, PyObject *args)
 562 {
 563         PyObject *py_ldb, *py_guid;
 564         bool ret;
 565         struct GUID guid;
 566         struct ldb_context *ldb;
 567         if (!PyArg_ParseTuple(args, "OO", &py_ldb, &py_guid))
 568                 return NULL;
 569
 570         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 571         GUID_from_string(PyString_AsString(py_guid), &guid);
 572
 573         ret = samdb_set_ntds_invocation_id(ldb, &guid);
 574         if (!ret) {
 575                 PyErr_SetString(PyExc_RuntimeError, "set_ntds_invocation_id failed");
 576                 return NULL;
 577         }
 578         Py_RETURN_NONE;
 579 }
 580
 581 static PyObject *py_samdb_ntds_objectGUID(PyObject *self, PyObject *args)
 582 {
 583         PyObject *py_ldb, *result;
 584         struct ldb_context *ldb;
 585         const struct GUID *guid;
 586         char *retstr;
 587
 588         if (!PyArg_ParseTuple(args, "O", &py_ldb)) {
 589                 return NULL;
 590         }
 591
 592         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 593
 594         guid = samdb_ntds_objectGUID(ldb);
 595         if (guid == NULL) {
 596                 PyErr_SetString(PyExc_RuntimeError, "Failed to find NTDS GUID");
 597                 return NULL;
 598         }
 599
 600         retstr = GUID_string(NULL, guid);
 601         if (retstr == NULL) {
 602                 PyErr_NoMemory();
 603                 return NULL;
 604         }
 605         result = PyString_FromString(retstr);
 606         talloc_free(retstr);
 607         return result;
 608 }
 609
 610 static PyObject *py_dsdb_set_global_schema(PyObject *self, PyObject *args)
 611 {
 612         PyObject *py_ldb;
 613         struct ldb_context *ldb;
 614         int ret;
 615         if (!PyArg_ParseTuple(args, "O", &py_ldb))
 616                 return NULL;
 617
 618         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 619
 620         ret = dsdb_set_global_schema(ldb);
 621         PyErr_LDB_ERROR_IS_ERR_RAISE(py_ldb_get_exception(), ret, ldb);
 622
 623         Py_RETURN_NONE;
 624 }
 625
 626 static PyObject *py_dsdb_load_partition_usn(PyObject *self, PyObject *args)
 627 {
 628         PyObject *py_dn, *py_ldb, *result;
 629         struct ldb_dn *dn;
 630         uint64_t highest_uSN, urgent_uSN;
 631         struct ldb_context *ldb;
 632         TALLOC_CTX *mem_ctx;
 633         int ret;
 634
 635         if (!PyArg_ParseTuple(args, "OO", &py_ldb, &py_dn)) {
 636                 return NULL;
 637         }
 638
 639         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 640
 641         mem_ctx = talloc_new(NULL);
 642         if (mem_ctx == NULL) {
 643            PyErr_NoMemory();
 644            return NULL;
 645         }
 646
 647         if (!PyObject_AsDn(mem_ctx, py_dn, ldb, &dn)) {
 648            talloc_free(mem_ctx);
 649            return NULL;
 650         }
 651
 652         ret = dsdb_load_partition_usn(ldb, dn, &highest_uSN, &urgent_uSN);
 653         if (ret != LDB_SUCCESS) {
 654            PyErr_Format(PyExc_RuntimeError,
 655                         "Failed to load partition [%s] uSN - %s",
 656                         ldb_dn_get_linearized(dn),
 657                         ldb_errstring(ldb));
 658            talloc_free(mem_ctx);
 659            return NULL;
 660         }
 661
 662         talloc_free(mem_ctx);
 663
 664         result = PyDict_New();
 665
 666         PyDict_SetItemString(result, "uSNHighest", PyInt_FromLong((uint64_t)highest_uSN));
 667         PyDict_SetItemString(result, "uSNUrgent", PyInt_FromLong((uint64_t)urgent_uSN));
 668
 669
 670         return result;
 671 }
 672
 673 static PyObject *py_dsdb_set_am_rodc(PyObject *self, PyObject *args)
 674 {
 675         PyObject *py_ldb;
 676         bool ret;
 677         struct ldb_context *ldb;
 678         int py_val;
 679
 680         if (!PyArg_ParseTuple(args, "Oi", &py_ldb, &py_val))
 681                 return NULL;
 682
 683         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 684         ret = samdb_set_am_rodc(ldb, (bool)py_val);
 685         if (!ret) {
 686                 PyErr_SetString(PyExc_RuntimeError, "set_am_rodc failed");
 687                 return NULL;
 688         }
 689         Py_RETURN_NONE;
 690 }
 691
 692 static PyObject *py_dsdb_set_schema_from_ldif(PyObject *self, PyObject *args)
 693 {
 694         WERROR result;
 695         char *pf, *df;
 696         PyObject *py_ldb;
 697         struct ldb_context *ldb;
 698
 699         if (!PyArg_ParseTuple(args, "Oss", &py_ldb, &pf, &df))
 700                 return NULL;
 701
 702         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 703
 704         result = dsdb_set_schema_from_ldif(ldb, pf, df);
 705         PyErr_WERROR_IS_ERR_RAISE(result);
 706
 707         Py_RETURN_NONE;
 708 }
 709
 710 static PyObject *py_dsdb_set_schema_from_ldb(PyObject *self, PyObject *args)
 711 {
 712         PyObject *py_ldb;
 713         struct ldb_context *ldb;
 714         PyObject *py_from_ldb;
 715         struct ldb_context *from_ldb;
 716         struct dsdb_schema *schema;
 717         int ret;
 718         if (!PyArg_ParseTuple(args, "OO", &py_ldb, &py_from_ldb))
 719                 return NULL;
 720
 721         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 722
 723         PyErr_LDB_OR_RAISE(py_from_ldb, from_ldb);
 724
 725         schema = dsdb_get_schema(from_ldb, NULL);
 726         if (!schema) {
 727                 PyErr_SetString(PyExc_RuntimeError, "Failed to set find a schema on 'from' ldb!\n");
 728                 return NULL;
 729         }
 730
 731         ret = dsdb_reference_schema(ldb, schema, true);
 732         PyErr_LDB_ERROR_IS_ERR_RAISE(py_ldb_get_exception(), ret, ldb);
 733
 734         Py_RETURN_NONE;
 735 }
 736
 737 static PyObject *py_dsdb_write_prefixes_from_schema_to_ldb(PyObject *self, PyObject *args)
 738 {
 739         PyObject *py_ldb;
 740         struct ldb_context *ldb;
 741         WERROR result;
 742         struct dsdb_schema *schema;
 743
 744         if (!PyArg_ParseTuple(args, "O", &py_ldb))
 745                 return NULL;
 746
 747         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 748
 749         schema = dsdb_get_schema(ldb, NULL);
 750         if (!schema) {
 751                 PyErr_SetString(PyExc_RuntimeError, "Failed to set find a schema on ldb!\n");
 752                 return NULL;
 753         }
 754
 755         result = dsdb_write_prefixes_from_schema_to_ldb(NULL, ldb, schema);
 756         PyErr_WERROR_IS_ERR_RAISE(result);
 757
 758         Py_RETURN_NONE;
 759 }
 760
 761
 762 static PyObject *py_dsdb_get_partitions_dn(PyObject *self, PyObject *args)
 763 {
 764         struct ldb_context *ldb;
 765         struct ldb_dn *dn;
 766         PyObject *py_ldb, *ret;
 767         PyObject *mod;
 768
 769         mod = PyImport_ImportModule("ldb");
 770
 771         if (!PyArg_ParseTuple(args, "O", &py_ldb))
 772                 return NULL;
 773
 774         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 775
 776         dn = samdb_partitions_dn(ldb, NULL);
 777         if (dn == NULL) {
 778                 PyErr_NoMemory();
 779                 return NULL;
 780         }
 781         ret = PyLdbDn_FromDn(dn);
 782         talloc_free(dn);
 783         return ret;
 784 }
 785
 786
 787 /*
 788   call into samdb_rodc()
 789  */
 790 static PyObject *py_dsdb_am_rodc(PyObject *self, PyObject *args)
 791 {
 792         PyObject *py_ldb;
 793         struct ldb_context *ldb;
 794         int ret;
 795         bool am_rodc;
 796
 797         if (!PyArg_ParseTuple(args, "O", &py_ldb))
 798                 return NULL;
 799
 800         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 801
 802         ret = samdb_rodc(ldb, &am_rodc);
 803         if (ret != LDB_SUCCESS) {
 804                 PyErr_SetString(PyExc_RuntimeError, ldb_errstring(ldb));
 805                 return NULL;
 806         }
 807
 808         return PyBool_FromLong(am_rodc);
 809 }
 810
 811
 812 static PyMethodDef py_dsdb_methods[] = {
 813         { "_samdb_server_site_name", (PyCFunction)py_samdb_server_site_name,
 814                 METH_VARARGS, "Get the server site name as a string"},
 815         { "_dsdb_convert_schema_to_openldap",
 816                 (PyCFunction)py_dsdb_convert_schema_to_openldap, METH_VARARGS,
 817                 "dsdb_convert_schema_to_openldap(ldb, target_str, mapping) -> str\n"
 818                 "Create an OpenLDAP schema from a schema." },
 819         { "_samdb_set_domain_sid", (PyCFunction)py_samdb_set_domain_sid,
 820                 METH_VARARGS,
 821                 "samdb_set_domain_sid(samdb, sid)\n"
 822                 "Set SID of domain to use." },
 823         { "_samdb_get_domain_sid", (PyCFunction)py_samdb_get_domain_sid,
 824                 METH_VARARGS,
 825                 "samdb_get_domain_sid(samdb)\n"
 826                 "Get SID of domain in use." },
 827         { "_samdb_ntds_invocation_id", (PyCFunction)py_samdb_ntds_invocation_id,
 828                 METH_VARARGS, "get the NTDS invocation ID GUID as a string"},
 829         { "_samdb_set_ntds_settings_dn", (PyCFunction)py_samdb_set_ntds_settings_dn,
 830                 METH_VARARGS,
 831                 "samdb_set_ntds_settings_dn(samdb, ntds_settings_dn)\n"
 832                 "Set NTDS Settings DN for this LDB (allows it to be set before the DB fully exists)." },
 833         { "_dsdb_get_oid_from_attid", (PyCFunction)py_dsdb_get_oid_from_attid,
 834                 METH_VARARGS, NULL },
 835         { "_dsdb_get_attid_from_lDAPDisplayName", (PyCFunction)py_dsdb_get_attid_from_lDAPDisplayName,
 836                 METH_VARARGS, NULL },
 837         { "_dsdb_get_syntax_oid_from_lDAPDisplayName", (PyCFunction)py_dsdb_get_syntax_oid_from_lDAPDisplayName,
 838                 METH_VARARGS, NULL },
 839         { "_dsdb_set_ntds_invocation_id",
 840                 (PyCFunction)py_dsdb_set_ntds_invocation_id, METH_VARARGS,
 841                 NULL },
 842         { "_samdb_ntds_objectGUID", (PyCFunction)py_samdb_ntds_objectGUID,
 843                 METH_VARARGS, "get the NTDS objectGUID as a string"},
 844         { "_dsdb_set_global_schema", (PyCFunction)py_dsdb_set_global_schema,
 845                 METH_VARARGS, NULL },
 846         { "_dsdb_load_partition_usn", (PyCFunction)py_dsdb_load_partition_usn,
 847                 METH_VARARGS,
 848                 "get uSNHighest and uSNUrgent from the partition @REPLCHANGED"},
 849         { "_dsdb_set_am_rodc",
 850                 (PyCFunction)py_dsdb_set_am_rodc, METH_VARARGS,
 851                 NULL },
 852         { "_am_rodc",
 853                 (PyCFunction)py_dsdb_am_rodc, METH_VARARGS,
 854                 NULL },
 855         { "_dsdb_set_schema_from_ldif", (PyCFunction)py_dsdb_set_schema_from_ldif, METH_VARARGS,
 856                 NULL },
 857         { "_dsdb_set_schema_from_ldb", (PyCFunction)py_dsdb_set_schema_from_ldb, METH_VARARGS,
 858                 NULL },
 859         { "_dsdb_write_prefixes_from_schema_to_ldb", (PyCFunction)py_dsdb_write_prefixes_from_schema_to_ldb, METH_VARARGS,
 860                 NULL },
 861         { "_dsdb_get_partitions_dn", (PyCFunction)py_dsdb_get_partitions_dn, METH_VARARGS, NULL },
 862         { "_dsdb_DsReplicaAttribute", (PyCFunction)py_dsdb_DsReplicaAttribute, METH_VARARGS, NULL },
 863         { "_dsdb_normalise_attributes", (PyCFunction)py_dsdb_normalise_attributes, METH_VARARGS, NULL },
 864         { NULL }
 865 };
 866
 867 void initdsdb(void)
 868 {
 869         PyObject *m;
 870
 871         m = Py_InitModule3("dsdb", py_dsdb_methods,
 872                            "Python bindings for the directory service databases.");
 873         if (m == NULL)
 874                 return;
 875
 876 #define ADD_DSDB_FLAG(val)  PyModule_AddObject(m, #val, PyInt_FromLong(val))
 877
 878         /* "userAccountControl" flags */
 879         ADD_DSDB_FLAG(UF_NORMAL_ACCOUNT);
 880         ADD_DSDB_FLAG(UF_TEMP_DUPLICATE_ACCOUNT);
 881         ADD_DSDB_FLAG(UF_SERVER_TRUST_ACCOUNT);
 882         ADD_DSDB_FLAG(UF_WORKSTATION_TRUST_ACCOUNT);
 883         ADD_DSDB_FLAG(UF_INTERDOMAIN_TRUST_ACCOUNT);
 884         ADD_DSDB_FLAG(UF_PASSWD_NOTREQD);
 885         ADD_DSDB_FLAG(UF_ACCOUNTDISABLE);
 886
 887         ADD_DSDB_FLAG(UF_SCRIPT);
 888         ADD_DSDB_FLAG(UF_ACCOUNTDISABLE);
 889         ADD_DSDB_FLAG(UF_00000004);
 890         ADD_DSDB_FLAG(UF_HOMEDIR_REQUIRED);
 891         ADD_DSDB_FLAG(UF_LOCKOUT);
 892         ADD_DSDB_FLAG(UF_PASSWD_NOTREQD);
 893         ADD_DSDB_FLAG(UF_PASSWD_CANT_CHANGE);
 894         ADD_DSDB_FLAG(UF_ENCRYPTED_TEXT_PASSWORD_ALLOWED);
 895         ADD_DSDB_FLAG(UF_TEMP_DUPLICATE_ACCOUNT);
 896         ADD_DSDB_FLAG(UF_NORMAL_ACCOUNT);
 897         ADD_DSDB_FLAG(UF_00000400);
 898         ADD_DSDB_FLAG(UF_INTERDOMAIN_TRUST_ACCOUNT);
 899         ADD_DSDB_FLAG(UF_WORKSTATION_TRUST_ACCOUNT);
 900         ADD_DSDB_FLAG(UF_SERVER_TRUST_ACCOUNT);
 901         ADD_DSDB_FLAG(UF_00004000);
 902         ADD_DSDB_FLAG(UF_00008000);
 903         ADD_DSDB_FLAG(UF_DONT_EXPIRE_PASSWD);
 904         ADD_DSDB_FLAG(UF_MNS_LOGON_ACCOUNT);
 905         ADD_DSDB_FLAG(UF_SMARTCARD_REQUIRED);
 906         ADD_DSDB_FLAG(UF_TRUSTED_FOR_DELEGATION);
 907         ADD_DSDB_FLAG(UF_NOT_DELEGATED);
 908         ADD_DSDB_FLAG(UF_USE_DES_KEY_ONLY);
 909         ADD_DSDB_FLAG(UF_DONT_REQUIRE_PREAUTH);
 910         ADD_DSDB_FLAG(UF_PASSWORD_EXPIRED);
 911         ADD_DSDB_FLAG(UF_TRUSTED_TO_AUTHENTICATE_FOR_DELEGATION);
 912         ADD_DSDB_FLAG(UF_NO_AUTH_DATA_REQUIRED);
 913         ADD_DSDB_FLAG(UF_PARTIAL_SECRETS_ACCOUNT);
 914
 915         /* groupType flags */
 916         ADD_DSDB_FLAG(GTYPE_SECURITY_BUILTIN_LOCAL_GROUP);
 917         ADD_DSDB_FLAG(GTYPE_SECURITY_GLOBAL_GROUP);
 918         ADD_DSDB_FLAG(GTYPE_SECURITY_DOMAIN_LOCAL_GROUP);
 919         ADD_DSDB_FLAG(GTYPE_SECURITY_UNIVERSAL_GROUP);
 920         ADD_DSDB_FLAG(GTYPE_DISTRIBUTION_GLOBAL_GROUP);
 921         ADD_DSDB_FLAG(GTYPE_DISTRIBUTION_DOMAIN_LOCAL_GROUP);
 922         ADD_DSDB_FLAG(GTYPE_DISTRIBUTION_UNIVERSAL_GROUP);
 923
 924         /* "sAMAccountType" flags */
 925         ADD_DSDB_FLAG(ATYPE_NORMAL_ACCOUNT);
 926         ADD_DSDB_FLAG(ATYPE_WORKSTATION_TRUST);
 927         ADD_DSDB_FLAG(ATYPE_INTERDOMAIN_TRUST);
 928         ADD_DSDB_FLAG(ATYPE_SECURITY_GLOBAL_GROUP);
 929         ADD_DSDB_FLAG(ATYPE_SECURITY_LOCAL_GROUP);
 930         ADD_DSDB_FLAG(ATYPE_SECURITY_UNIVERSAL_GROUP);
 931         ADD_DSDB_FLAG(ATYPE_DISTRIBUTION_GLOBAL_GROUP);
 932         ADD_DSDB_FLAG(ATYPE_DISTRIBUTION_LOCAL_GROUP);
 933         ADD_DSDB_FLAG(ATYPE_DISTRIBUTION_UNIVERSAL_GROUP);
 934
 935         /* "domainFunctionality", "forestFunctionality" flags in the rootDSE */
 936         ADD_DSDB_FLAG(DS_DOMAIN_FUNCTION_2000);
 937         ADD_DSDB_FLAG(DS_DOMAIN_FUNCTION_2003_MIXED);
 938         ADD_DSDB_FLAG(DS_DOMAIN_FUNCTION_2003);
 939         ADD_DSDB_FLAG(DS_DOMAIN_FUNCTION_2008);
 940         ADD_DSDB_FLAG(DS_DOMAIN_FUNCTION_2008_R2);
 941
 942         /* "systemFlags" */
 943         ADD_DSDB_FLAG(SYSTEM_FLAG_CR_NTDS_NC);
 944         ADD_DSDB_FLAG(SYSTEM_FLAG_CR_NTDS_DOMAIN);
 945         ADD_DSDB_FLAG(SYSTEM_FLAG_CR_NTDS_NOT_GC_REPLICATED);
 946         ADD_DSDB_FLAG(SYSTEM_FLAG_SCHEMA_BASE_OBJECT);
 947         ADD_DSDB_FLAG(SYSTEM_FLAG_ATTR_IS_RDN);
 948         ADD_DSDB_FLAG(SYSTEM_FLAG_DISALLOW_MOVE_ON_DELETE);
 949         ADD_DSDB_FLAG(SYSTEM_FLAG_DOMAIN_DISALLOW_MOVE);
 950         ADD_DSDB_FLAG(SYSTEM_FLAG_DOMAIN_DISALLOW_RENAME);
 951         ADD_DSDB_FLAG(SYSTEM_FLAG_CONFIG_ALLOW_LIMITED_MOVE);
 952         ADD_DSDB_FLAG(SYSTEM_FLAG_CONFIG_ALLOW_MOVE);
 953         ADD_DSDB_FLAG(SYSTEM_FLAG_CONFIG_ALLOW_RENAME);
 954         ADD_DSDB_FLAG(SYSTEM_FLAG_DISALLOW_DELETE);
 955
 956         /* Kerberos encryption type constants */
 957         ADD_DSDB_FLAG(ENC_ALL_TYPES);
 958         ADD_DSDB_FLAG(ENC_CRC32);
 959         ADD_DSDB_FLAG(ENC_RSA_MD5);
 960         ADD_DSDB_FLAG(ENC_RC4_HMAC_MD5);
 961         ADD_DSDB_FLAG(ENC_HMAC_SHA1_96_AES128);
 962         ADD_DSDB_FLAG(ENC_HMAC_SHA1_96_AES256);
 963
 964         ADD_DSDB_FLAG(SEARCH_FLAG_ATTINDEX);
 965         ADD_DSDB_FLAG(SEARCH_FLAG_PDNTATTINDEX);
 966         ADD_DSDB_FLAG(SEARCH_FLAG_ANR);
 967         ADD_DSDB_FLAG(SEARCH_FLAG_PRESERVEONDELETE);
 968         ADD_DSDB_FLAG(SEARCH_FLAG_COPY);
 969         ADD_DSDB_FLAG(SEARCH_FLAG_TUPLEINDEX);
 970         ADD_DSDB_FLAG(SEARCH_FLAG_SUBTREEATTRINDEX);
 971         ADD_DSDB_FLAG(SEARCH_FLAG_CONFIDENTIAL);
 972         ADD_DSDB_FLAG(SEARCH_FLAG_NEVERVALUEAUDIT);
 973         ADD_DSDB_FLAG(SEARCH_FLAG_RODC_ATTRIBUTE);
 974
 975         ADD_DSDB_FLAG(DS_FLAG_ATTR_NOT_REPLICATED);
 976         ADD_DSDB_FLAG(DS_FLAG_ATTR_REQ_PARTIAL_SET_MEMBER);
 977         ADD_DSDB_FLAG(DS_FLAG_ATTR_IS_CONSTRUCTED);
 978
 979         ADD_DSDB_FLAG(DS_NTDSDSA_OPT_IS_GC);
 980         ADD_DSDB_FLAG(DS_NTDSDSA_OPT_DISABLE_INBOUND_REPL);
 981         ADD_DSDB_FLAG(DS_NTDSDSA_OPT_DISABLE_OUTBOUND_REPL);
 982         ADD_DSDB_FLAG(DS_NTDSDSA_OPT_DISABLE_NTDSCONN_XLATE);
 983         ADD_DSDB_FLAG(DS_NTDSDSA_OPT_DISABLE_SPN_REGISTRATION);
 984
 985         ADD_DSDB_FLAG(NTDSCONN_KCC_GC_TOPOLOGY);
 986         ADD_DSDB_FLAG(NTDSCONN_KCC_RING_TOPOLOGY);
 987         ADD_DSDB_FLAG(NTDSCONN_KCC_MINIMIZE_HOPS_TOPOLOGY);
 988         ADD_DSDB_FLAG(NTDSCONN_KCC_STALE_SERVERS_TOPOLOGY);
 989         ADD_DSDB_FLAG(NTDSCONN_KCC_OSCILLATING_CONNECTION_TOPOLOGY);
 990         ADD_DSDB_FLAG(NTDSCONN_KCC_INTERSITE_GC_TOPOLOGY);
 991         ADD_DSDB_FLAG(NTDSCONN_KCC_INTERSITE_TOPOLOGY);
 992         ADD_DSDB_FLAG(NTDSCONN_KCC_SERVER_FAILOVER_TOPOLOGY);
 993         ADD_DSDB_FLAG(NTDSCONN_KCC_SITE_FAILOVER_TOPOLOGY);
 994         ADD_DSDB_FLAG(NTDSCONN_KCC_REDUNDANT_SERVER_TOPOLOGY);
 995
 996         /* GPO policy flags */
 997         ADD_DSDB_FLAG(GPLINK_OPT_DISABLE);
 998         ADD_DSDB_FLAG(GPLINK_OPT_ENFORCE);
 999         ADD_DSDB_FLAG(GPO_FLAG_USER_DISABLE);
1000         ADD_DSDB_FLAG(GPO_FLAG_MACHINE_DISABLE);
1001         ADD_DSDB_FLAG(GPO_INHERIT);
1002         ADD_DSDB_FLAG(GPO_BLOCK_INHERITANCE);
1003
1004 #define ADD_DSDB_STRING(val)  PyModule_AddObject(m, #val, PyString_FromString(val))
1005
1006         ADD_DSDB_STRING(DSDB_SYNTAX_BINARY_DN);
1007         ADD_DSDB_STRING(DSDB_SYNTAX_STRING_DN);
1008         ADD_DSDB_STRING(DSDB_SYNTAX_OR_NAME);
1009 }