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   convert a python string to a DRSUAPI drsuapi_DsReplicaAttribute attribute
 335  */
 336 static PyObject *py_dsdb_DsReplicaAttribute(PyObject *self, PyObject *args)
 337 {
 338         PyObject *py_ldb, *el_list, *ret;
 339         struct ldb_context *ldb;
 340         char *ldap_display_name;
 341         const struct dsdb_attribute *a;
 342         struct dsdb_schema *schema;
 343         struct dsdb_syntax_ctx syntax_ctx;
 344         struct ldb_message_element *el;
 345         struct drsuapi_DsReplicaAttribute *attr;
 346         TALLOC_CTX *tmp_ctx;
 347         WERROR werr;
 348         Py_ssize_t i;
 349
 350         if (!PyArg_ParseTuple(args, "OsO", &py_ldb, &ldap_display_name, &el_list)) {
 351                 return NULL;
 352         }
 353
 354         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 355
 356         if (!PyList_Check(el_list)) {
 357                 PyErr_Format(PyExc_TypeError, "ldif_elements must be a list");
 358                 return NULL;
 359         }
 360
 361         schema = dsdb_get_schema(ldb, NULL);
 362         if (!schema) {
 363                 PyErr_SetString(PyExc_RuntimeError, "Failed to find a schema from ldb");
 364                 return NULL;
 365         }
 366
 367         a = dsdb_attribute_by_lDAPDisplayName(schema, ldap_display_name);
 368         if (a == NULL) {
 369                 PyErr_Format(PyExc_RuntimeError, "Failed to find attribute '%s'", ldap_display_name);
 370                 return NULL;
 371         }
 372
 373         dsdb_syntax_ctx_init(&syntax_ctx, ldb, schema);
 374         syntax_ctx.is_schema_nc = false;
 375
 376         tmp_ctx = talloc_new(ldb);
 377         if (tmp_ctx == NULL) {
 378                 PyErr_NoMemory();
 379                 return NULL;
 380         }
 381
 382         el = talloc_zero(tmp_ctx, struct ldb_message_element);
 383         if (el == NULL) {
 384                 PyErr_NoMemory();
 385                 talloc_free(tmp_ctx);
 386                 return NULL;
 387         }
 388
 389         el->name = ldap_display_name;
 390         el->num_values = PyList_Size(el_list);
 391
 392         el->values = talloc_array(el, struct ldb_val, el->num_values);
 393         if (el->values == NULL) {
 394                 PyErr_NoMemory();
 395                 talloc_free(tmp_ctx);
 396                 return NULL;
 397         }
 398
 399         for (i = 0; i < el->num_values; i++) {
 400                 PyObject *item = PyList_GetItem(el_list, i);
 401                 if (!PyString_Check(item)) {
 402                         PyErr_Format(PyExc_TypeError, "ldif_elements should be strings");
 403                         return NULL;
 404                 }
 405                 el->values[i].data = (uint8_t *)PyString_AsString(item);
 406                 el->values[i].length = PyString_Size(item);
 407         }
 408
 409         attr = talloc_zero(tmp_ctx, struct drsuapi_DsReplicaAttribute);
 410         if (attr == NULL) {
 411                 PyErr_NoMemory();
 412                 talloc_free(tmp_ctx);
 413                 return NULL;
 414         }
 415
 416         werr = a->syntax->ldb_to_drsuapi(&syntax_ctx, a, el, attr, attr);
 417         PyErr_WERROR_IS_ERR_RAISE(werr);
 418
 419         ret = py_return_ndr_struct("samba.dcerpc.drsuapi", "DsReplicaAttribute", attr, attr);
 420
 421         talloc_free(tmp_ctx);
 422
 423         return ret;
 424 }
 425
 426
 427 /*
 428   normalise a ldb attribute list
 429  */
 430 static PyObject *py_dsdb_normalise_attributes(PyObject *self, PyObject *args)
 431 {
 432         PyObject *py_ldb, *el_list, *ret;
 433         struct ldb_context *ldb;
 434         char *ldap_display_name;
 435         const struct dsdb_attribute *a;
 436         struct dsdb_schema *schema;
 437         struct dsdb_syntax_ctx syntax_ctx;
 438         struct ldb_message_element *el;
 439         struct drsuapi_DsReplicaAttribute *attr;
 440         TALLOC_CTX *tmp_ctx;
 441         WERROR werr;
 442         Py_ssize_t i;
 443
 444         if (!PyArg_ParseTuple(args, "OsO", &py_ldb, &ldap_display_name, &el_list)) {
 445                 return NULL;
 446         }
 447
 448         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 449
 450         if (!PyList_Check(el_list)) {
 451                 PyErr_Format(PyExc_TypeError, "ldif_elements must be a list");
 452                 return NULL;
 453         }
 454
 455         schema = dsdb_get_schema(ldb, NULL);
 456         if (!schema) {
 457                 PyErr_SetString(PyExc_RuntimeError, "Failed to find a schema from ldb");
 458                 return NULL;
 459         }
 460
 461         a = dsdb_attribute_by_lDAPDisplayName(schema, ldap_display_name);
 462         if (a == NULL) {
 463                 PyErr_Format(PyExc_RuntimeError, "Failed to find attribute '%s'", ldap_display_name);
 464                 return NULL;
 465         }
 466
 467         dsdb_syntax_ctx_init(&syntax_ctx, ldb, schema);
 468         syntax_ctx.is_schema_nc = false;
 469
 470         tmp_ctx = talloc_new(ldb);
 471         if (tmp_ctx == NULL) {
 472                 PyErr_NoMemory();
 473                 return NULL;
 474         }
 475
 476         el = talloc_zero(tmp_ctx, struct ldb_message_element);
 477         if (el == NULL) {
 478                 PyErr_NoMemory();
 479                 talloc_free(tmp_ctx);
 480                 return NULL;
 481         }
 482
 483         el->name = ldap_display_name;
 484         el->num_values = PyList_Size(el_list);
 485
 486         el->values = talloc_array(el, struct ldb_val, el->num_values);
 487         if (el->values == NULL) {
 488                 PyErr_NoMemory();
 489                 talloc_free(tmp_ctx);
 490                 return NULL;
 491         }
 492
 493         for (i = 0; i < el->num_values; i++) {
 494                 PyObject *item = PyList_GetItem(el_list, i);
 495                 if (!PyString_Check(item)) {
 496                         PyErr_Format(PyExc_TypeError, "ldif_elements should be strings");
 497                         return NULL;
 498                 }
 499                 el->values[i].data = (uint8_t *)PyString_AsString(item);
 500                 el->values[i].length = PyString_Size(item);
 501         }
 502
 503         /* first run ldb_to_drsuapi, then convert back again. This has
 504          * the effect of normalising the attributes
 505          */
 506
 507         attr = talloc_zero(tmp_ctx, struct drsuapi_DsReplicaAttribute);
 508         if (attr == NULL) {
 509                 PyErr_NoMemory();
 510                 talloc_free(tmp_ctx);
 511                 return NULL;
 512         }
 513
 514         werr = a->syntax->ldb_to_drsuapi(&syntax_ctx, a, el, attr, attr);
 515         PyErr_WERROR_IS_ERR_RAISE(werr);
 516
 517         /* now convert back again */
 518         werr = a->syntax->drsuapi_to_ldb(&syntax_ctx, a, attr, el, el);
 519         PyErr_WERROR_IS_ERR_RAISE(werr);
 520
 521         ret = py_return_ndr_struct("ldb", "MessageElement", el, el);
 522
 523         talloc_free(tmp_ctx);
 524
 525         return ret;
 526 }
 527
 528
 529 static PyObject *py_dsdb_set_ntds_invocation_id(PyObject *self, PyObject *args)
 530 {
 531         PyObject *py_ldb, *py_guid;
 532         bool ret;
 533         struct GUID guid;
 534         struct ldb_context *ldb;
 535         if (!PyArg_ParseTuple(args, "OO", &py_ldb, &py_guid))
 536                 return NULL;
 537
 538         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 539         GUID_from_string(PyString_AsString(py_guid), &guid);
 540
 541         ret = samdb_set_ntds_invocation_id(ldb, &guid);
 542         if (!ret) {
 543                 PyErr_SetString(PyExc_RuntimeError, "set_ntds_invocation_id failed");
 544                 return NULL;
 545         }
 546         Py_RETURN_NONE;
 547 }
 548
 549 static PyObject *py_samdb_ntds_objectGUID(PyObject *self, PyObject *args)
 550 {
 551         PyObject *py_ldb, *result;
 552         struct ldb_context *ldb;
 553         const struct GUID *guid;
 554         char *retstr;
 555
 556         if (!PyArg_ParseTuple(args, "O", &py_ldb)) {
 557                 return NULL;
 558         }
 559
 560         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 561
 562         guid = samdb_ntds_objectGUID(ldb);
 563         if (guid == NULL) {
 564                 PyErr_SetString(PyExc_RuntimeError, "Failed to find NTDS GUID");
 565                 return NULL;
 566         }
 567
 568         retstr = GUID_string(NULL, guid);
 569         if (retstr == NULL) {
 570                 PyErr_NoMemory();
 571                 return NULL;
 572         }
 573         result = PyString_FromString(retstr);
 574         talloc_free(retstr);
 575         return result;
 576 }
 577
 578 static PyObject *py_dsdb_set_global_schema(PyObject *self, PyObject *args)
 579 {
 580         PyObject *py_ldb;
 581         struct ldb_context *ldb;
 582         int ret;
 583         if (!PyArg_ParseTuple(args, "O", &py_ldb))
 584                 return NULL;
 585
 586         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 587
 588         ret = dsdb_set_global_schema(ldb);
 589         PyErr_LDB_ERROR_IS_ERR_RAISE(py_ldb_get_exception(), ret, ldb);
 590
 591         Py_RETURN_NONE;
 592 }
 593
 594 static PyObject *py_dsdb_load_partition_usn(PyObject *self, PyObject *args)
 595 {
 596         PyObject *py_dn, *py_ldb, *result;
 597         struct ldb_dn *dn;
 598         uint64_t highest_uSN, urgent_uSN;
 599         struct ldb_context *ldb;
 600         TALLOC_CTX *mem_ctx;
 601         int ret;
 602
 603         if (!PyArg_ParseTuple(args, "OO", &py_ldb, &py_dn)) {
 604                 return NULL;
 605         }
 606
 607         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 608
 609         mem_ctx = talloc_new(NULL);
 610         if (mem_ctx == NULL) {
 611            PyErr_NoMemory();
 612            return NULL;
 613         }
 614
 615         if (!PyObject_AsDn(mem_ctx, py_dn, ldb, &dn)) {
 616            talloc_free(mem_ctx);
 617            return NULL;
 618         }
 619
 620         ret = dsdb_load_partition_usn(ldb, dn, &highest_uSN, &urgent_uSN);
 621         if (ret != LDB_SUCCESS) {
 622            PyErr_Format(PyExc_RuntimeError,
 623                         "Failed to load partition [%s] uSN - %s",
 624                         ldb_dn_get_linearized(dn),
 625                         ldb_errstring(ldb));
 626            talloc_free(mem_ctx);
 627            return NULL;
 628         }
 629
 630         talloc_free(mem_ctx);
 631
 632         result = PyDict_New();
 633
 634         PyDict_SetItemString(result, "uSNHighest", PyInt_FromLong((uint64_t)highest_uSN));
 635         PyDict_SetItemString(result, "uSNUrgent", PyInt_FromLong((uint64_t)urgent_uSN));
 636
 637
 638         return result;
 639 }
 640
 641 static PyObject *py_dsdb_set_am_rodc(PyObject *self, PyObject *args)
 642 {
 643         PyObject *py_ldb;
 644         bool ret;
 645         struct ldb_context *ldb;
 646         int py_val;
 647
 648         if (!PyArg_ParseTuple(args, "Oi", &py_ldb, &py_val))
 649                 return NULL;
 650
 651         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 652         ret = samdb_set_am_rodc(ldb, (bool)py_val);
 653         if (!ret) {
 654                 PyErr_SetString(PyExc_RuntimeError, "set_am_rodc failed");
 655                 return NULL;
 656         }
 657         Py_RETURN_NONE;
 658 }
 659
 660 static PyObject *py_dsdb_set_schema_from_ldif(PyObject *self, PyObject *args)
 661 {
 662         WERROR result;
 663         char *pf, *df;
 664         PyObject *py_ldb;
 665         struct ldb_context *ldb;
 666
 667         if (!PyArg_ParseTuple(args, "Oss", &py_ldb, &pf, &df))
 668                 return NULL;
 669
 670         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 671
 672         result = dsdb_set_schema_from_ldif(ldb, pf, df);
 673         PyErr_WERROR_IS_ERR_RAISE(result);
 674
 675         Py_RETURN_NONE;
 676 }
 677
 678 static PyObject *py_dsdb_set_schema_from_ldb(PyObject *self, PyObject *args)
 679 {
 680         PyObject *py_ldb;
 681         struct ldb_context *ldb;
 682         PyObject *py_from_ldb;
 683         struct ldb_context *from_ldb;
 684         struct dsdb_schema *schema;
 685         int ret;
 686         if (!PyArg_ParseTuple(args, "OO", &py_ldb, &py_from_ldb))
 687                 return NULL;
 688
 689         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 690
 691         PyErr_LDB_OR_RAISE(py_from_ldb, from_ldb);
 692
 693         schema = dsdb_get_schema(from_ldb, NULL);
 694         if (!schema) {
 695                 PyErr_SetString(PyExc_RuntimeError, "Failed to set find a schema on 'from' ldb!\n");
 696                 return NULL;
 697         }
 698
 699         ret = dsdb_reference_schema(ldb, schema, true);
 700         PyErr_LDB_ERROR_IS_ERR_RAISE(py_ldb_get_exception(), ret, ldb);
 701
 702         Py_RETURN_NONE;
 703 }
 704
 705 static PyObject *py_dsdb_write_prefixes_from_schema_to_ldb(PyObject *self, PyObject *args)
 706 {
 707         PyObject *py_ldb;
 708         struct ldb_context *ldb;
 709         WERROR result;
 710         struct dsdb_schema *schema;
 711
 712         if (!PyArg_ParseTuple(args, "O", &py_ldb))
 713                 return NULL;
 714
 715         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 716
 717         schema = dsdb_get_schema(ldb, NULL);
 718         if (!schema) {
 719                 PyErr_SetString(PyExc_RuntimeError, "Failed to set find a schema on ldb!\n");
 720                 return NULL;
 721         }
 722
 723         result = dsdb_write_prefixes_from_schema_to_ldb(NULL, ldb, schema);
 724         PyErr_WERROR_IS_ERR_RAISE(result);
 725
 726         Py_RETURN_NONE;
 727 }
 728
 729
 730 static PyObject *py_dsdb_get_partitions_dn(PyObject *self, PyObject *args)
 731 {
 732         struct ldb_context *ldb;
 733         struct ldb_dn *dn;
 734         PyObject *py_ldb, *ret;
 735         PyObject *mod;
 736
 737         mod = PyImport_ImportModule("ldb");
 738
 739         if (!PyArg_ParseTuple(args, "O", &py_ldb))
 740                 return NULL;
 741
 742         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 743
 744         dn = samdb_partitions_dn(ldb, NULL);
 745         if (dn == NULL) {
 746                 PyErr_NoMemory();
 747                 return NULL;
 748         }
 749         ret = PyLdbDn_FromDn(dn);
 750         talloc_free(dn);
 751         return ret;
 752 }
 753
 754
 755 /*
 756   call into samdb_rodc()
 757  */
 758 static PyObject *py_dsdb_am_rodc(PyObject *self, PyObject *args)
 759 {
 760         PyObject *py_ldb;
 761         struct ldb_context *ldb;
 762         int ret;
 763         bool am_rodc;
 764
 765         if (!PyArg_ParseTuple(args, "O", &py_ldb))
 766                 return NULL;
 767
 768         PyErr_LDB_OR_RAISE(py_ldb, ldb);
 769
 770         ret = samdb_rodc(ldb, &am_rodc);
 771         if (ret != LDB_SUCCESS) {
 772                 PyErr_SetString(PyExc_RuntimeError, ldb_errstring(ldb));
 773                 return NULL;
 774         }
 775
 776         return PyBool_FromLong(am_rodc);
 777 }
 778
 779
 780 static PyMethodDef py_dsdb_methods[] = {
 781         { "_samdb_server_site_name", (PyCFunction)py_samdb_server_site_name,
 782                 METH_VARARGS, "Get the server site name as a string"},
 783         { "_dsdb_convert_schema_to_openldap",
 784                 (PyCFunction)py_dsdb_convert_schema_to_openldap, METH_VARARGS,
 785                 "dsdb_convert_schema_to_openldap(ldb, target_str, mapping) -> str\n"
 786                 "Create an OpenLDAP schema from a schema." },
 787         { "_samdb_set_domain_sid", (PyCFunction)py_samdb_set_domain_sid,
 788                 METH_VARARGS,
 789                 "samdb_set_domain_sid(samdb, sid)\n"
 790                 "Set SID of domain to use." },
 791         { "_samdb_get_domain_sid", (PyCFunction)py_samdb_get_domain_sid,
 792                 METH_VARARGS,
 793                 "samdb_get_domain_sid(samdb)\n"
 794                 "Get SID of domain in use." },
 795         { "_samdb_ntds_invocation_id", (PyCFunction)py_samdb_ntds_invocation_id,
 796                 METH_VARARGS, "get the NTDS invocation ID GUID as a string"},
 797         { "_samdb_set_ntds_settings_dn", (PyCFunction)py_samdb_set_ntds_settings_dn,
 798                 METH_VARARGS,
 799                 "samdb_set_ntds_settings_dn(samdb, ntds_settings_dn)\n"
 800                 "Set NTDS Settings DN for this LDB (allows it to be set before the DB fully exists)." },
 801         { "_dsdb_get_oid_from_attid", (PyCFunction)py_dsdb_get_oid_from_attid,
 802                 METH_VARARGS, NULL },
 803         { "_dsdb_get_attid_from_lDAPDisplayName", (PyCFunction)py_dsdb_get_attid_from_lDAPDisplayName,
 804                 METH_VARARGS, NULL },
 805         { "_dsdb_set_ntds_invocation_id",
 806                 (PyCFunction)py_dsdb_set_ntds_invocation_id, METH_VARARGS,
 807                 NULL },
 808         { "_samdb_ntds_objectGUID", (PyCFunction)py_samdb_ntds_objectGUID,
 809                 METH_VARARGS, "get the NTDS objectGUID as a string"},
 810         { "_dsdb_set_global_schema", (PyCFunction)py_dsdb_set_global_schema,
 811                 METH_VARARGS, NULL },
 812         { "_dsdb_load_partition_usn", (PyCFunction)py_dsdb_load_partition_usn,
 813                 METH_VARARGS,
 814                 "get uSNHighest and uSNUrgent from the partition @REPLCHANGED"},
 815         { "_dsdb_set_am_rodc",
 816                 (PyCFunction)py_dsdb_set_am_rodc, METH_VARARGS,
 817                 NULL },
 818         { "_am_rodc",
 819                 (PyCFunction)py_dsdb_am_rodc, METH_VARARGS,
 820                 NULL },
 821         { "_dsdb_set_schema_from_ldif", (PyCFunction)py_dsdb_set_schema_from_ldif, METH_VARARGS,
 822                 NULL },
 823         { "_dsdb_set_schema_from_ldb", (PyCFunction)py_dsdb_set_schema_from_ldb, METH_VARARGS,
 824                 NULL },
 825         { "_dsdb_write_prefixes_from_schema_to_ldb", (PyCFunction)py_dsdb_write_prefixes_from_schema_to_ldb, METH_VARARGS,
 826                 NULL },
 827         { "_dsdb_get_partitions_dn", (PyCFunction)py_dsdb_get_partitions_dn, METH_VARARGS, NULL },
 828         { "_dsdb_DsReplicaAttribute", (PyCFunction)py_dsdb_DsReplicaAttribute, METH_VARARGS, NULL },
 829         { "_dsdb_normalise_attributes", (PyCFunction)py_dsdb_normalise_attributes, METH_VARARGS, NULL },
 830         { NULL }
 831 };
 832
 833 void initdsdb(void)
 834 {
 835         PyObject *m;
 836
 837         m = Py_InitModule3("dsdb", py_dsdb_methods,
 838                            "Python bindings for the directory service databases.");
 839         if (m == NULL)
 840                 return;
 841
 842 #define ADD_DSDB_FLAG(val)  PyModule_AddObject(m, #val, PyInt_FromLong(val))
 843
 844         /* "userAccountControl" flags */
 845         ADD_DSDB_FLAG(UF_NORMAL_ACCOUNT);
 846         ADD_DSDB_FLAG(UF_TEMP_DUPLICATE_ACCOUNT);
 847         ADD_DSDB_FLAG(UF_SERVER_TRUST_ACCOUNT);
 848         ADD_DSDB_FLAG(UF_WORKSTATION_TRUST_ACCOUNT);
 849         ADD_DSDB_FLAG(UF_INTERDOMAIN_TRUST_ACCOUNT);
 850         ADD_DSDB_FLAG(UF_PASSWD_NOTREQD);
 851         ADD_DSDB_FLAG(UF_ACCOUNTDISABLE);
 852
 853         ADD_DSDB_FLAG(UF_SCRIPT);
 854         ADD_DSDB_FLAG(UF_ACCOUNTDISABLE);
 855         ADD_DSDB_FLAG(UF_00000004);
 856         ADD_DSDB_FLAG(UF_HOMEDIR_REQUIRED);
 857         ADD_DSDB_FLAG(UF_LOCKOUT);
 858         ADD_DSDB_FLAG(UF_PASSWD_NOTREQD);
 859         ADD_DSDB_FLAG(UF_PASSWD_CANT_CHANGE);
 860         ADD_DSDB_FLAG(UF_ENCRYPTED_TEXT_PASSWORD_ALLOWED);
 861         ADD_DSDB_FLAG(UF_TEMP_DUPLICATE_ACCOUNT);
 862         ADD_DSDB_FLAG(UF_NORMAL_ACCOUNT);
 863         ADD_DSDB_FLAG(UF_00000400);
 864         ADD_DSDB_FLAG(UF_INTERDOMAIN_TRUST_ACCOUNT);
 865         ADD_DSDB_FLAG(UF_WORKSTATION_TRUST_ACCOUNT);
 866         ADD_DSDB_FLAG(UF_SERVER_TRUST_ACCOUNT);
 867         ADD_DSDB_FLAG(UF_00004000);
 868         ADD_DSDB_FLAG(UF_00008000);
 869         ADD_DSDB_FLAG(UF_DONT_EXPIRE_PASSWD);
 870         ADD_DSDB_FLAG(UF_MNS_LOGON_ACCOUNT);
 871         ADD_DSDB_FLAG(UF_SMARTCARD_REQUIRED);
 872         ADD_DSDB_FLAG(UF_TRUSTED_FOR_DELEGATION);
 873         ADD_DSDB_FLAG(UF_NOT_DELEGATED);
 874         ADD_DSDB_FLAG(UF_USE_DES_KEY_ONLY);
 875         ADD_DSDB_FLAG(UF_DONT_REQUIRE_PREAUTH);
 876         ADD_DSDB_FLAG(UF_PASSWORD_EXPIRED);
 877         ADD_DSDB_FLAG(UF_TRUSTED_TO_AUTHENTICATE_FOR_DELEGATION);
 878         ADD_DSDB_FLAG(UF_NO_AUTH_DATA_REQUIRED);
 879         ADD_DSDB_FLAG(UF_PARTIAL_SECRETS_ACCOUNT);
 880
 881         /* groupType flags */
 882         ADD_DSDB_FLAG(GTYPE_SECURITY_BUILTIN_LOCAL_GROUP);
 883         ADD_DSDB_FLAG(GTYPE_SECURITY_GLOBAL_GROUP);
 884         ADD_DSDB_FLAG(GTYPE_SECURITY_DOMAIN_LOCAL_GROUP);
 885         ADD_DSDB_FLAG(GTYPE_SECURITY_UNIVERSAL_GROUP);
 886         ADD_DSDB_FLAG(GTYPE_DISTRIBUTION_GLOBAL_GROUP);
 887         ADD_DSDB_FLAG(GTYPE_DISTRIBUTION_DOMAIN_LOCAL_GROUP);
 888         ADD_DSDB_FLAG(GTYPE_DISTRIBUTION_UNIVERSAL_GROUP);
 889
 890         /* "sAMAccountType" flags */
 891         ADD_DSDB_FLAG(ATYPE_NORMAL_ACCOUNT);
 892         ADD_DSDB_FLAG(ATYPE_WORKSTATION_TRUST);
 893         ADD_DSDB_FLAG(ATYPE_INTERDOMAIN_TRUST);
 894         ADD_DSDB_FLAG(ATYPE_SECURITY_GLOBAL_GROUP);
 895         ADD_DSDB_FLAG(ATYPE_SECURITY_LOCAL_GROUP);
 896         ADD_DSDB_FLAG(ATYPE_SECURITY_UNIVERSAL_GROUP);
 897         ADD_DSDB_FLAG(ATYPE_DISTRIBUTION_GLOBAL_GROUP);
 898         ADD_DSDB_FLAG(ATYPE_DISTRIBUTION_LOCAL_GROUP);
 899         ADD_DSDB_FLAG(ATYPE_DISTRIBUTION_UNIVERSAL_GROUP);
 900
 901         /* "domainFunctionality", "forestFunctionality" flags in the rootDSE */
 902         ADD_DSDB_FLAG(DS_DOMAIN_FUNCTION_2000);
 903         ADD_DSDB_FLAG(DS_DOMAIN_FUNCTION_2003_MIXED);
 904         ADD_DSDB_FLAG(DS_DOMAIN_FUNCTION_2003);
 905         ADD_DSDB_FLAG(DS_DOMAIN_FUNCTION_2008);
 906         ADD_DSDB_FLAG(DS_DOMAIN_FUNCTION_2008_R2);
 907
 908         /* "systemFlags" */
 909         ADD_DSDB_FLAG(SYSTEM_FLAG_CR_NTDS_NC);
 910         ADD_DSDB_FLAG(SYSTEM_FLAG_CR_NTDS_DOMAIN);
 911         ADD_DSDB_FLAG(SYSTEM_FLAG_CR_NTDS_NOT_GC_REPLICATED);
 912         ADD_DSDB_FLAG(SYSTEM_FLAG_SCHEMA_BASE_OBJECT);
 913         ADD_DSDB_FLAG(SYSTEM_FLAG_ATTR_IS_RDN);
 914         ADD_DSDB_FLAG(SYSTEM_FLAG_DISALLOW_MOVE_ON_DELETE);
 915         ADD_DSDB_FLAG(SYSTEM_FLAG_DOMAIN_DISALLOW_MOVE);
 916         ADD_DSDB_FLAG(SYSTEM_FLAG_DOMAIN_DISALLOW_RENAME);
 917         ADD_DSDB_FLAG(SYSTEM_FLAG_CONFIG_ALLOW_LIMITED_MOVE);
 918         ADD_DSDB_FLAG(SYSTEM_FLAG_CONFIG_ALLOW_MOVE);
 919         ADD_DSDB_FLAG(SYSTEM_FLAG_CONFIG_ALLOW_RENAME);
 920         ADD_DSDB_FLAG(SYSTEM_FLAG_DISALLOW_DELETE);
 921
 922         /* Kerberos encryption type constants */
 923         ADD_DSDB_FLAG(ENC_ALL_TYPES);
 924         ADD_DSDB_FLAG(ENC_CRC32);
 925         ADD_DSDB_FLAG(ENC_RSA_MD5);
 926         ADD_DSDB_FLAG(ENC_RC4_HMAC_MD5);
 927         ADD_DSDB_FLAG(ENC_HMAC_SHA1_96_AES128);
 928         ADD_DSDB_FLAG(ENC_HMAC_SHA1_96_AES256);
 929
 930         ADD_DSDB_FLAG(SEARCH_FLAG_ATTINDEX);
 931         ADD_DSDB_FLAG(SEARCH_FLAG_PDNTATTINDEX);
 932         ADD_DSDB_FLAG(SEARCH_FLAG_ANR);
 933         ADD_DSDB_FLAG(SEARCH_FLAG_PRESERVEONDELETE);
 934         ADD_DSDB_FLAG(SEARCH_FLAG_COPY);
 935         ADD_DSDB_FLAG(SEARCH_FLAG_TUPLEINDEX);
 936         ADD_DSDB_FLAG(SEARCH_FLAG_SUBTREEATTRINDEX);
 937         ADD_DSDB_FLAG(SEARCH_FLAG_CONFIDENTIAL);
 938         ADD_DSDB_FLAG(SEARCH_FLAG_NEVERVALUEAUDIT);
 939         ADD_DSDB_FLAG(SEARCH_FLAG_RODC_ATTRIBUTE);
 940
 941         ADD_DSDB_FLAG(DS_FLAG_ATTR_NOT_REPLICATED);
 942         ADD_DSDB_FLAG(DS_FLAG_ATTR_REQ_PARTIAL_SET_MEMBER);
 943         ADD_DSDB_FLAG(DS_FLAG_ATTR_IS_CONSTRUCTED);
 944
 945         ADD_DSDB_FLAG(DS_NTDSDSA_OPT_IS_GC);
 946         ADD_DSDB_FLAG(DS_NTDSDSA_OPT_DISABLE_INBOUND_REPL);
 947         ADD_DSDB_FLAG(DS_NTDSDSA_OPT_DISABLE_OUTBOUND_REPL);
 948         ADD_DSDB_FLAG(DS_NTDSDSA_OPT_DISABLE_NTDSCONN_XLATE);
 949         ADD_DSDB_FLAG(DS_NTDSDSA_OPT_DISABLE_SPN_REGISTRATION);
 950
 951         ADD_DSDB_FLAG(NTDSCONN_KCC_GC_TOPOLOGY);
 952         ADD_DSDB_FLAG(NTDSCONN_KCC_RING_TOPOLOGY);
 953         ADD_DSDB_FLAG(NTDSCONN_KCC_MINIMIZE_HOPS_TOPOLOGY);
 954         ADD_DSDB_FLAG(NTDSCONN_KCC_STALE_SERVERS_TOPOLOGY);
 955         ADD_DSDB_FLAG(NTDSCONN_KCC_OSCILLATING_CONNECTION_TOPOLOGY);
 956         ADD_DSDB_FLAG(NTDSCONN_KCC_INTERSITE_GC_TOPOLOGY);
 957         ADD_DSDB_FLAG(NTDSCONN_KCC_INTERSITE_TOPOLOGY);
 958         ADD_DSDB_FLAG(NTDSCONN_KCC_SERVER_FAILOVER_TOPOLOGY);
 959         ADD_DSDB_FLAG(NTDSCONN_KCC_SITE_FAILOVER_TOPOLOGY);
 960         ADD_DSDB_FLAG(NTDSCONN_KCC_REDUNDANT_SERVER_TOPOLOGY);
 961
 962         /* GPO policy flags */
 963         ADD_DSDB_FLAG(GPLINK_OPT_DISABLE);
 964         ADD_DSDB_FLAG(GPLINK_OPT_ENFORCE);
 965         ADD_DSDB_FLAG(GPO_FLAG_USER_DISABLE);
 966         ADD_DSDB_FLAG(GPO_FLAG_MACHINE_DISABLE);
 967         ADD_DSDB_FLAG(GPO_INHERIT);
 968         ADD_DSDB_FLAG(GPO_BLOCK_INHERITANCE);
 969 }