source4/rpc_server/drsuapi/dcesrv_drsuapi.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    endpoint server for the drsuapi pipe
   5
   6    Copyright (C) Stefan Metzmacher 2004
   7    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2006
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 3 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21 */
  22
  23 #include "includes.h"
  24 #include "librpc/gen_ndr/ndr_drsuapi.h"
  25 #include "rpc_server/dcerpc_server.h"
  26 #include "rpc_server/common/common.h"
  27 #include "dsdb/samdb/samdb.h"
  28 #include "lib/ldb/include/ldb_errors.h"
  29 #include "param/param.h"
  30 #include "librpc/gen_ndr/ndr_drsblobs.h"
  31 #include "messaging/irpc.h"
  32 #include "rpc_server/drsuapi/dcesrv_drsuapi.h"
  33 #include "libcli/security/security.h"
  34
  35 /*
  36   drsuapi_DsBind
  37 */
  38 static WERROR dcesrv_drsuapi_DsBind(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
  39                        struct drsuapi_DsBind *r)
  40 {
  41         struct drsuapi_bind_state *b_state;
  42         struct dcesrv_handle *handle;
  43         struct drsuapi_DsBindInfoCtr *bind_info;
  44         struct GUID site_guid;
  45         struct ldb_result *site_res;
  46         struct ldb_dn *server_site_dn;
  47         static const char *site_attrs[] = { "objectGUID", NULL };
  48         struct ldb_result *ntds_res;
  49         struct ldb_dn *ntds_dn;
  50         static const char *ntds_attrs[] = { "ms-DS-ReplicationEpoch", NULL };
  51         uint32_t pid;
  52         uint32_t repl_epoch;
  53         int ret;
  54
  55         r->out.bind_info = NULL;
  56         ZERO_STRUCTP(r->out.bind_handle);
  57
  58         b_state = talloc_zero(mem_ctx, struct drsuapi_bind_state);
  59         W_ERROR_HAVE_NO_MEMORY(b_state);
  60
  61         /*
  62          * connect to the samdb
  63          */
  64         b_state->sam_ctx = samdb_connect(b_state, dce_call->event_ctx, dce_call->conn->dce_ctx->lp_ctx, dce_call->conn->auth_state.session_info);
  65         if (!b_state->sam_ctx) {
  66                 return WERR_FOOBAR;
  67         }
  68
  69         /*
  70          * find out the guid of our own site
  71          */
  72         server_site_dn = samdb_server_site_dn(b_state->sam_ctx, mem_ctx);
  73         W_ERROR_HAVE_NO_MEMORY(server_site_dn);
  74
  75         ret = ldb_search(b_state->sam_ctx, mem_ctx, &site_res,
  76                                  server_site_dn, LDB_SCOPE_BASE, site_attrs,
  77                                  "(objectClass=*)");
  78         if (ret != LDB_SUCCESS) {
  79                 return WERR_DS_DRA_INTERNAL_ERROR;
  80         }
  81         if (site_res->count != 1) {
  82                 return WERR_DS_DRA_INTERNAL_ERROR;
  83         }
  84         site_guid = samdb_result_guid(site_res->msgs[0], "objectGUID");
  85
  86         /*
  87          * lookup the local servers Replication Epoch
  88          */
  89         ntds_dn = samdb_ntds_settings_dn(b_state->sam_ctx);
  90         W_ERROR_HAVE_NO_MEMORY(ntds_dn);
  91
  92         ret = ldb_search(b_state->sam_ctx, mem_ctx, &ntds_res,
  93                                  ntds_dn, LDB_SCOPE_BASE, ntds_attrs,
  94                                  "(objectClass=*)");
  95         if (ret != LDB_SUCCESS) {
  96                 return WERR_DS_DRA_INTERNAL_ERROR;
  97         }
  98         if (ntds_res->count != 1) {
  99                 return WERR_DS_DRA_INTERNAL_ERROR;
 100         }
 101         repl_epoch = samdb_result_uint(ntds_res->msgs[0], "ms-DS-ReplicationEpoch", 0);
 102
 103         /*
 104          * The "process identifier" of the client.
 105          * According to the WSPP docs, sectin 5.35, this is
 106          * for informational and debugging purposes only.
 107          * The assignment is implementation specific.
 108          */
 109         pid = 0;
 110
 111         /*
 112          * store the clients bind_guid
 113          */
 114         if (r->in.bind_guid) {
 115                 b_state->remote_bind_guid = *r->in.bind_guid;
 116         }
 117
 118         /*
 119          * store the clients bind_info
 120          */
 121         if (r->in.bind_info) {
 122                 switch (r->in.bind_info->length) {
 123                 case 24: {
 124                         struct drsuapi_DsBindInfo24 *info24;
 125                         info24 = &r->in.bind_info->info.info24;
 126                         b_state->remote_info28.supported_extensions     = info24->supported_extensions;
 127                         b_state->remote_info28.site_guid                = info24->site_guid;
 128                         b_state->remote_info28.pid                      = info24->pid;
 129                         b_state->remote_info28.repl_epoch               = 0;
 130                         break;
 131                 }
 132                 case 28:
 133                         b_state->remote_info28 = r->in.bind_info->info.info28;
 134                         break;
 135                 }
 136         }
 137
 138         /*
 139          * fill in our local bind info 28
 140          */
 141         b_state->local_info28.supported_extensions      = 0;
 142         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_BASE;
 143         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_ASYNC_REPLICATION;
 144         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_REMOVEAPI;
 145         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_MOVEREQ_V2;
 146 #if 0 /* we don't support MSZIP compression (only decompression) */
 147         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_GETCHG_COMPRESS;
 148 #endif
 149         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V1;
 150         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_RESTORE_USN_OPTIMIZATION;
 151         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_KCC_EXECUTE;
 152         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_ADDENTRY_V2;
 153         if (0 /*domain.behavior_version == 2*/) {
 154                 /* TODO: find out how this is really triggered! */
 155                 b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_LINKED_VALUE_REPLICATION;
 156         }
 157         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V2;
 158         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_INSTANCE_TYPE_NOT_REQ_ON_MOD;
 159         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_CRYPTO_BIND;
 160         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_GET_REPL_INFO;
 161         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_STRONG_ENCRYPTION;
 162         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V01;
 163         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_TRANSITIVE_MEMBERSHIP;
 164         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_ADD_SID_HISTORY;
 165         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_POST_BETA3;
 166         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_00100000;
 167         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_GET_MEMBERSHIPS2;
 168         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V6;
 169         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_NONDOMAIN_NCS;
 170         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V8;
 171         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V5;
 172         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V6;
 173         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_ADDENTRYREPLY_V3;
 174         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V7;
 175         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_VERIFY_OBJECT;
 176 #if 0 /* we don't support XPRESS compression yet */
 177         b_state->local_info28.supported_extensions      |= DRSUAPI_SUPPORTED_EXTENSION_XPRESS_COMPRESS;
 178 #endif
 179         b_state->local_info28.site_guid                 = site_guid;
 180         b_state->local_info28.pid                       = pid;
 181         b_state->local_info28.repl_epoch                = repl_epoch;
 182
 183         /*
 184          * allocate the return bind_info
 185          */
 186         bind_info = talloc(mem_ctx, struct drsuapi_DsBindInfoCtr);
 187         W_ERROR_HAVE_NO_MEMORY(bind_info);
 188
 189         bind_info->length       = 28;
 190         bind_info->info.info28  = b_state->local_info28;
 191
 192         /*
 193          * allocate a bind handle
 194          */
 195         handle = dcesrv_handle_new(dce_call->context, DRSUAPI_BIND_HANDLE);
 196         W_ERROR_HAVE_NO_MEMORY(handle);
 197         handle->data = talloc_steal(handle, b_state);
 198
 199         /*
 200          * prepare reply
 201          */
 202         r->out.bind_info = bind_info;
 203         *r->out.bind_handle = handle->wire_handle;
 204
 205         return WERR_OK;
 206 }
 207
 208
 209 /*
 210   drsuapi_DsUnbind
 211 */
 212 static WERROR dcesrv_drsuapi_DsUnbind(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 213                                struct drsuapi_DsUnbind *r)
 214 {
 215         struct dcesrv_handle *h;
 216
 217         *r->out.bind_handle = *r->in.bind_handle;
 218
 219         DCESRV_PULL_HANDLE_WERR(h, r->in.bind_handle, DRSUAPI_BIND_HANDLE);
 220
 221         talloc_free(h);
 222
 223         ZERO_STRUCTP(r->out.bind_handle);
 224
 225         return WERR_OK;
 226 }
 227
 228
 229 /*
 230   drsuapi_DsReplicaSync
 231 */
 232 static WERROR dcesrv_drsuapi_DsReplicaSync(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 233                                            struct drsuapi_DsReplicaSync *r)
 234 {
 235         struct server_id *repld;
 236         struct irpc_request *ireq;
 237
 238         if (security_session_user_level(dce_call->conn->auth_state.session_info) <
 239             SECURITY_DOMAIN_CONTROLLER) {
 240                 DEBUG(0,("DsReplicaSync refused for security token\n"));
 241                 return WERR_DS_DRA_ACCESS_DENIED;
 242         }
 243
 244         repld = irpc_servers_byname(dce_call->msg_ctx, mem_ctx, "dreplsrv");
 245         if (repld == NULL || repld[0].id == 0) {
 246                 DEBUG(0,("DsReplicaSync: Unable to find dreplsrv task\n"));
 247                 return WERR_DS_DRA_INTERNAL_ERROR;
 248         }
 249
 250         ireq = IRPC_CALL_SEND(dce_call->msg_ctx, repld[0],
 251                               drsuapi, DRSUAPI_DSREPLICASYNC,
 252                               r, mem_ctx);
 253         if (ireq == NULL) {
 254                 DEBUG(0,("DsReplicaSync: Failed to forward request to dreplsrv task\n"));
 255                 return WERR_DS_DRA_INTERNAL_ERROR;
 256         }
 257
 258         /* we are not interested in a reply */
 259         talloc_free(ireq);
 260
 261         return WERR_OK;
 262 }
 263
 264 /*
 265   drsuapi_DsReplicaAdd
 266 */
 267 static WERROR dcesrv_drsuapi_DsReplicaAdd(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 268                                           struct drsuapi_DsReplicaAdd *r)
 269 {
 270         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 271 }
 272
 273
 274 /*
 275   drsuapi_DsReplicaDel
 276 */
 277 static WERROR dcesrv_drsuapi_DsReplicaDel(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 278                                           struct drsuapi_DsReplicaDel *r)
 279 {
 280         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 281 }
 282
 283
 284 /*
 285   drsuapi_DsReplicaModify
 286 */
 287 static WERROR dcesrv_drsuapi_DsReplicaMod(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 288                                           struct drsuapi_DsReplicaMod *r)
 289 {
 290         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 291 }
 292
 293
 294 /*
 295   DRSUAPI_VERIFY_NAMES
 296 */
 297 static WERROR dcesrv_DRSUAPI_VERIFY_NAMES(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 298                        struct DRSUAPI_VERIFY_NAMES *r)
 299 {
 300         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 301 }
 302
 303
 304 /*
 305   drsuapi_DsGetMemberships
 306 */
 307 static WERROR dcesrv_drsuapi_DsGetMemberships(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 308                        struct drsuapi_DsGetMemberships *r)
 309 {
 310         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 311 }
 312
 313
 314 /*
 315   DRSUAPI_INTER_DOMAIN_MOVE
 316 */
 317 static WERROR dcesrv_DRSUAPI_INTER_DOMAIN_MOVE(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 318                        struct DRSUAPI_INTER_DOMAIN_MOVE *r)
 319 {
 320         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 321 }
 322
 323
 324 /*
 325   drsuapi_DsGetNT4ChangeLog
 326 */
 327 static WERROR dcesrv_drsuapi_DsGetNT4ChangeLog(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 328                        struct drsuapi_DsGetNT4ChangeLog *r)
 329 {
 330         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 331 }
 332
 333
 334 /*
 335   drsuapi_DsCrackNames
 336 */
 337 static WERROR dcesrv_drsuapi_DsCrackNames(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 338                             struct drsuapi_DsCrackNames *r)
 339 {
 340         WERROR status;
 341         struct drsuapi_bind_state *b_state;
 342         struct dcesrv_handle *h;
 343
 344         *r->out.level_out = r->in.level;
 345
 346         DCESRV_PULL_HANDLE_WERR(h, r->in.bind_handle, DRSUAPI_BIND_HANDLE);
 347         b_state = h->data;
 348
 349         r->out.ctr = talloc_zero(mem_ctx, union drsuapi_DsNameCtr);
 350         W_ERROR_HAVE_NO_MEMORY(r->out.ctr);
 351
 352         switch (r->in.level) {
 353                 case 1: {
 354                         struct drsuapi_DsNameCtr1 *ctr1;
 355                         struct drsuapi_DsNameInfo1 *names;
 356                         int count;
 357                         int i;
 358
 359                         ctr1 = talloc(mem_ctx, struct drsuapi_DsNameCtr1);
 360                         W_ERROR_HAVE_NO_MEMORY(ctr1);
 361
 362                         count = r->in.req->req1.count;
 363                         names = talloc_array(mem_ctx, struct drsuapi_DsNameInfo1, count);
 364                         W_ERROR_HAVE_NO_MEMORY(names);
 365
 366                         for (i=0; i < count; i++) {
 367                                 status = DsCrackNameOneName(b_state->sam_ctx, mem_ctx,
 368                                                             r->in.req->req1.format_flags,
 369                                                             r->in.req->req1.format_offered,
 370                                                             r->in.req->req1.format_desired,
 371                                                             r->in.req->req1.names[i].str,
 372                                                             &names[i]);
 373                                 if (!W_ERROR_IS_OK(status)) {
 374                                         return status;
 375                                 }
 376                         }
 377
 378                         ctr1->count = count;
 379                         ctr1->array = names;
 380                         r->out.ctr->ctr1 = ctr1;
 381
 382                         return WERR_OK;
 383                 }
 384         }
 385
 386         return WERR_UNKNOWN_LEVEL;
 387 }
 388
 389 /*
 390   drsuapi_DsWriteAccountSpn
 391 */
 392 static WERROR dcesrv_drsuapi_DsWriteAccountSpn(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 393                        struct drsuapi_DsWriteAccountSpn *r)
 394 {
 395         struct drsuapi_bind_state *b_state;
 396         struct dcesrv_handle *h;
 397
 398         *r->out.level_out = r->in.level;
 399
 400         DCESRV_PULL_HANDLE_WERR(h, r->in.bind_handle, DRSUAPI_BIND_HANDLE);
 401         b_state = h->data;
 402
 403         r->out.res = talloc(mem_ctx, union drsuapi_DsWriteAccountSpnResult);
 404         W_ERROR_HAVE_NO_MEMORY(r->out.res);
 405
 406         switch (r->in.level) {
 407                 case 1: {
 408                         struct drsuapi_DsWriteAccountSpnRequest1 *req;
 409                         struct ldb_message *msg;
 410                         int count, i, ret;
 411                         req = &r->in.req->req1;
 412                         count = req->count;
 413
 414                         msg = ldb_msg_new(mem_ctx);
 415                         if (msg == NULL) {
 416                                 return WERR_NOMEM;
 417                         }
 418
 419                         msg->dn = ldb_dn_new(msg, b_state->sam_ctx, req->object_dn);
 420                         if ( ! ldb_dn_validate(msg->dn)) {
 421                                 r->out.res->res1.status = WERR_OK;
 422                                 return WERR_OK;
 423                         }
 424
 425                         /* construct mods */
 426                         for (i = 0; i < count; i++) {
 427                                 samdb_msg_add_string(b_state->sam_ctx,
 428                                                      msg, msg, "servicePrincipalName",
 429                                                      req->spn_names[i].str);
 430                         }
 431                         for (i=0;i<msg->num_elements;i++) {
 432                                 switch (req->operation) {
 433                                 case DRSUAPI_DS_SPN_OPERATION_ADD:
 434                                         msg->elements[i].flags = LDB_FLAG_MOD_ADD;
 435                                         break;
 436                                 case DRSUAPI_DS_SPN_OPERATION_REPLACE:
 437                                         msg->elements[i].flags = LDB_FLAG_MOD_REPLACE;
 438                                         break;
 439                                 case DRSUAPI_DS_SPN_OPERATION_DELETE:
 440                                         msg->elements[i].flags = LDB_FLAG_MOD_DELETE;
 441                                         break;
 442                                 }
 443                         }
 444
 445                         /* Apply to database */
 446
 447                         ret = ldb_modify(b_state->sam_ctx, msg);
 448                         if (ret != 0) {
 449                                 DEBUG(0,("Failed to modify SPNs on %s: %s\n",
 450                                          ldb_dn_get_linearized(msg->dn),
 451                                          ldb_errstring(b_state->sam_ctx)));
 452                                 r->out.res->res1.status = WERR_ACCESS_DENIED;
 453                         } else {
 454                                 r->out.res->res1.status = WERR_OK;
 455                         }
 456
 457                         return WERR_OK;
 458                 }
 459         }
 460
 461         return WERR_UNKNOWN_LEVEL;
 462 }
 463
 464
 465 /*
 466   drsuapi_DsRemoveDSServer
 467 */
 468 static WERROR dcesrv_drsuapi_DsRemoveDSServer(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 469                                        struct drsuapi_DsRemoveDSServer *r)
 470 {
 471         struct drsuapi_bind_state *b_state;
 472         struct dcesrv_handle *h;
 473         struct ldb_dn *ntds_dn;
 474         int ret;
 475         bool ok;
 476
 477         ZERO_STRUCT(r->out.res);
 478         *r->out.level_out = 1;
 479
 480         if (security_session_user_level(dce_call->conn->auth_state.session_info) <
 481             SECURITY_DOMAIN_CONTROLLER) {
 482                 DEBUG(0,("DsRemoveDSServer refused for security token\n"));
 483                 return WERR_DS_DRA_ACCESS_DENIED;
 484         }
 485
 486         DCESRV_PULL_HANDLE_WERR(h, r->in.bind_handle, DRSUAPI_BIND_HANDLE);
 487         b_state = h->data;
 488
 489         switch (r->in.level) {
 490         case 1:
 491                 ntds_dn = ldb_dn_new(mem_ctx, b_state->sam_ctx, r->in.req->req1.server_dn);
 492                 W_ERROR_HAVE_NO_MEMORY(ntds_dn);
 493
 494                 ok = ldb_dn_validate(ntds_dn);
 495                 if (!ok) {
 496                         return WERR_FOOBAR;
 497                 }
 498
 499                 /* TODO: it's likely that we need more checks here */
 500
 501                 ok = ldb_dn_add_child_fmt(ntds_dn, "CN=NTDS Settings");
 502                 if (!ok) {
 503                         return WERR_FOOBAR;
 504                 }
 505
 506                 if (r->in.req->req1.commit) {
 507                         ret = ldb_delete(b_state->sam_ctx, ntds_dn);
 508                         if (ret != LDB_SUCCESS) {
 509                                 return WERR_FOOBAR;
 510                         }
 511                 }
 512
 513                 return WERR_OK;
 514         default:
 515                 break;
 516         }
 517
 518         return WERR_FOOBAR;
 519 }
 520
 521
 522 /*
 523   DRSUAPI_REMOVE_DS_DOMAIN
 524 */
 525 static WERROR dcesrv_DRSUAPI_REMOVE_DS_DOMAIN(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 526                        struct DRSUAPI_REMOVE_DS_DOMAIN *r)
 527 {
 528         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 529 }
 530
 531 /* Obtain the site name from a server DN */
 532 static const char *result_site_name(struct ldb_dn *site_dn)
 533 {
 534         /* Format is cn=<NETBIOS name>,cn=Servers,cn=<site>,cn=sites.... */
 535         const struct ldb_val *val = ldb_dn_get_component_val(site_dn, 2);
 536         const char *name = ldb_dn_get_component_name(site_dn, 2);
 537
 538         if (!name || (ldb_attr_cmp(name, "cn") != 0)) {
 539                 /* Ensure this matches the format.  This gives us a
 540                  * bit more confidence that a 'cn' value will be a
 541                  * ascii string */
 542                 return NULL;
 543         }
 544         if (val) {
 545                 return (char *)val->data;
 546         }
 547         return NULL;
 548 }
 549
 550 /*
 551   drsuapi_DsGetDomainControllerInfo
 552 */
 553 static WERROR dcesrv_drsuapi_DsGetDomainControllerInfo_1(struct drsuapi_bind_state *b_state,
 554                                                 TALLOC_CTX *mem_ctx,
 555                                                 struct drsuapi_DsGetDomainControllerInfo *r)
 556 {
 557         struct ldb_dn *sites_dn;
 558         struct ldb_result *res;
 559
 560         const char *attrs_account_1[] = { "cn", "dnsHostName", NULL };
 561         const char *attrs_account_2[] = { "cn", "dnsHostName", "objectGUID", NULL };
 562
 563         const char *attrs_none[] = { NULL };
 564
 565         const char *attrs_site[] = { "objectGUID", NULL };
 566
 567         const char *attrs_ntds[] = { "options", "objectGUID", NULL };
 568
 569         const char *attrs_1[] = { "serverReference", "cn", "dnsHostName", NULL };
 570         const char *attrs_2[] = { "serverReference", "cn", "dnsHostName", "objectGUID", NULL };
 571         const char **attrs;
 572
 573         struct drsuapi_DsGetDCInfoCtr1 *ctr1;
 574         struct drsuapi_DsGetDCInfoCtr2 *ctr2;
 575
 576         int ret, i;
 577
 578         *r->out.level_out = r->in.req->req1.level;
 579         r->out.ctr = talloc(mem_ctx, union drsuapi_DsGetDCInfoCtr);
 580         W_ERROR_HAVE_NO_MEMORY(r->out.ctr);
 581
 582         sites_dn = samdb_sites_dn(b_state->sam_ctx, mem_ctx);
 583         if (!sites_dn) {
 584                 return WERR_DS_OBJ_NOT_FOUND;
 585         }
 586
 587         switch (*r->out.level_out) {
 588         case -1:
 589                 /* this level is not like the others */
 590                 return WERR_UNKNOWN_LEVEL;
 591         case 1:
 592                 attrs = attrs_1;
 593                 break;
 594         case 2:
 595                 attrs = attrs_2;
 596                 break;
 597         default:
 598                 return WERR_UNKNOWN_LEVEL;
 599         }
 600
 601         ret = ldb_search(b_state->sam_ctx, mem_ctx, &res, sites_dn, LDB_SCOPE_SUBTREE, attrs,
 602                                  "objectClass=server");
 603
 604         if (ret) {
 605                 DEBUG(1, ("searching for servers in sites DN %s failed: %s\n",
 606                           ldb_dn_get_linearized(sites_dn), ldb_errstring(b_state->sam_ctx)));
 607                 return WERR_GENERAL_FAILURE;
 608         }
 609
 610         switch (*r->out.level_out) {
 611         case 1:
 612                 ctr1 = &r->out.ctr->ctr1;
 613                 ctr1->count = res->count;
 614                 ctr1->array = talloc_zero_array(mem_ctx,
 615                                                 struct drsuapi_DsGetDCInfo1,
 616                                                 res->count);
 617                 for (i=0; i < res->count; i++) {
 618                         struct ldb_dn *domain_dn;
 619                         struct ldb_result *res_domain;
 620                         struct ldb_result *res_account;
 621                         struct ldb_dn *ntds_dn = ldb_dn_copy(mem_ctx, res->msgs[i]->dn);
 622
 623                         struct ldb_dn *ref_dn
 624                                 = ldb_msg_find_attr_as_dn(b_state->sam_ctx,
 625                                                           mem_ctx, res->msgs[i],
 626                                                           "serverReference");
 627
 628                         if (!ntds_dn || !ldb_dn_add_child_fmt(ntds_dn, "CN=NTDS Settings")) {
 629                                 return WERR_NOMEM;
 630                         }
 631
 632                         ret = ldb_search(b_state->sam_ctx, mem_ctx, &res_account, ref_dn,
 633                                                  LDB_SCOPE_BASE, attrs_account_1, "objectClass=computer");
 634                         if (ret == LDB_SUCCESS && res_account->count == 1) {
 635                                 const char *errstr;
 636                                 ctr1->array[i].dns_name
 637                                         = ldb_msg_find_attr_as_string(res_account->msgs[0], "dNSHostName", NULL);
 638                                 ctr1->array[i].netbios_name
 639                                         = ldb_msg_find_attr_as_string(res_account->msgs[0], "cn", NULL);
 640                                 ctr1->array[i].computer_dn
 641                                         = ldb_dn_get_linearized(res_account->msgs[0]->dn);
 642
 643                                 /* Determine if this is the PDC */
 644                                 ret = samdb_search_for_parent_domain(b_state->sam_ctx,
 645                                                                      mem_ctx, res_account->msgs[0]->dn,
 646                                                                      &domain_dn, &errstr);
 647
 648                                 if (ret == LDB_SUCCESS) {
 649                                         ret = ldb_search(b_state->sam_ctx, mem_ctx, &res_domain, domain_dn,
 650                                                                  LDB_SCOPE_BASE, attrs_none, "fSMORoleOwner=%s",
 651                                                                  ldb_dn_get_linearized(ntds_dn));
 652                                         if (ret) {
 653                                                 return WERR_GENERAL_FAILURE;
 654                                         }
 655                                         if (res_domain->count == 1) {
 656                                                 ctr1->array[i].is_pdc = true;
 657                                         }
 658                                 }
 659                         }
 660                         if ((ret != LDB_SUCCESS) && (ret != LDB_ERR_NO_SUCH_OBJECT)) {
 661                                 DEBUG(5, ("warning: searching for computer DN %s failed: %s\n",
 662                                           ldb_dn_get_linearized(ref_dn), ldb_errstring(b_state->sam_ctx)));
 663                         }
 664
 665                         /* Look at server DN and extract site component */
 666                         ctr1->array[i].site_name = result_site_name(res->msgs[i]->dn);
 667                         ctr1->array[i].server_dn = ldb_dn_get_linearized(res->msgs[i]->dn);
 668
 669
 670                         ctr1->array[i].is_enabled = true;
 671
 672                 }
 673                 break;
 674         case 2:
 675                 ctr2 = &r->out.ctr->ctr2;
 676                 ctr2->count = res->count;
 677                 ctr2->array = talloc_zero_array(mem_ctx,
 678                                                  struct drsuapi_DsGetDCInfo2,
 679                                                  res->count);
 680                 for (i=0; i < res->count; i++) {
 681                         struct ldb_dn *domain_dn;
 682                         struct ldb_result *res_domain;
 683                         struct ldb_result *res_account;
 684                         struct ldb_dn *ntds_dn = ldb_dn_copy(mem_ctx, res->msgs[i]->dn);
 685                         struct ldb_result *res_ntds;
 686                         struct ldb_dn *site_dn = ldb_dn_copy(mem_ctx, res->msgs[i]->dn);
 687                         struct ldb_result *res_site;
 688                         struct ldb_dn *ref_dn
 689                                 = ldb_msg_find_attr_as_dn(b_state->sam_ctx,
 690                                                           mem_ctx, res->msgs[i],
 691                                                           "serverReference");
 692
 693                         if (!ntds_dn || !ldb_dn_add_child_fmt(ntds_dn, "CN=NTDS Settings")) {
 694                                 return WERR_NOMEM;
 695                         }
 696
 697                         /* Format is cn=<NETBIOS name>,cn=Servers,cn=<site>,cn=sites.... */
 698                         if (!site_dn || !ldb_dn_remove_child_components(site_dn, 2)) {
 699                                 return WERR_NOMEM;
 700                         }
 701
 702                         ret = ldb_search(b_state->sam_ctx, mem_ctx, &res_ntds, ntds_dn,
 703                                                  LDB_SCOPE_BASE, attrs_ntds, "objectClass=nTDSDSA");
 704                         if (ret == LDB_SUCCESS && res_ntds->count == 1) {
 705                                 ctr2->array[i].is_gc
 706                                         = (ldb_msg_find_attr_as_int(res_ntds->msgs[0], "options", 0) == 1);
 707                                 ctr2->array[i].ntds_guid
 708                                         = samdb_result_guid(res_ntds->msgs[0], "objectGUID");
 709                                 ctr2->array[i].ntds_dn = ldb_dn_get_linearized(res_ntds->msgs[0]->dn);
 710                         }
 711                         if ((ret != LDB_SUCCESS) && (ret != LDB_ERR_NO_SUCH_OBJECT)) {
 712                                 DEBUG(5, ("warning: searching for NTDS DN %s failed: %s\n",
 713                                           ldb_dn_get_linearized(ntds_dn), ldb_errstring(b_state->sam_ctx)));
 714                         }
 715
 716                         ret = ldb_search(b_state->sam_ctx, mem_ctx, &res_site, site_dn,
 717                                                  LDB_SCOPE_BASE, attrs_site, "objectClass=site");
 718                         if (ret == LDB_SUCCESS && res_site->count == 1) {
 719                                 ctr2->array[i].site_guid
 720                                         = samdb_result_guid(res_site->msgs[0], "objectGUID");
 721                                 ctr2->array[i].site_dn = ldb_dn_get_linearized(res_site->msgs[0]->dn);
 722                         }
 723                         if ((ret != LDB_SUCCESS) && (ret != LDB_ERR_NO_SUCH_OBJECT)) {
 724                                 DEBUG(5, ("warning: searching for site DN %s failed: %s\n",
 725                                           ldb_dn_get_linearized(site_dn), ldb_errstring(b_state->sam_ctx)));
 726                         }
 727
 728                         ret = ldb_search(b_state->sam_ctx, mem_ctx, &res_account, ref_dn,
 729                                                  LDB_SCOPE_BASE, attrs_account_2, "objectClass=computer");
 730                         if (ret == LDB_SUCCESS && res_account->count == 1) {
 731                                 const char *errstr;
 732                                 ctr2->array[i].dns_name
 733                                         = ldb_msg_find_attr_as_string(res_account->msgs[0], "dNSHostName", NULL);
 734                                 ctr2->array[i].netbios_name
 735                                         = ldb_msg_find_attr_as_string(res_account->msgs[0], "cn", NULL);
 736                                 ctr2->array[i].computer_dn = ldb_dn_get_linearized(res_account->msgs[0]->dn);
 737                                 ctr2->array[i].computer_guid
 738                                         = samdb_result_guid(res_account->msgs[0], "objectGUID");
 739
 740                                 /* Determine if this is the PDC */
 741                                 ret = samdb_search_for_parent_domain(b_state->sam_ctx,
 742                                                                      mem_ctx, res_account->msgs[0]->dn,
 743                                                                      &domain_dn, &errstr);
 744
 745                                 if (ret == LDB_SUCCESS) {
 746                                         ret = ldb_search(b_state->sam_ctx, mem_ctx, &res_domain, domain_dn,
 747                                                                  LDB_SCOPE_BASE, attrs_none, "fSMORoleOwner=%s",
 748                                                                  ldb_dn_get_linearized(ntds_dn));
 749                                         if (ret == LDB_SUCCESS && res_domain->count == 1) {
 750                                                 ctr2->array[i].is_pdc = true;
 751                                         }
 752                                         if ((ret != LDB_SUCCESS) && (ret != LDB_ERR_NO_SUCH_OBJECT)) {
 753                                                 DEBUG(5, ("warning: searching for domain DN %s failed: %s\n",
 754                                                           ldb_dn_get_linearized(domain_dn), ldb_errstring(b_state->sam_ctx)));
 755                                         }
 756                                 }
 757                         }
 758                         if ((ret != LDB_SUCCESS) && (ret != LDB_ERR_NO_SUCH_OBJECT)) {
 759                                 DEBUG(5, ("warning: searching for computer account DN %s failed: %s\n",
 760                                           ldb_dn_get_linearized(ref_dn), ldb_errstring(b_state->sam_ctx)));
 761                         }
 762
 763                         /* Look at server DN and extract site component */
 764                         ctr2->array[i].site_name = result_site_name(res->msgs[i]->dn);
 765                         ctr2->array[i].server_dn = ldb_dn_get_linearized(res->msgs[i]->dn);
 766                         ctr2->array[i].server_guid
 767                                 = samdb_result_guid(res->msgs[i], "objectGUID");
 768
 769                         ctr2->array[i].is_enabled = true;
 770
 771                 }
 772                 break;
 773         }
 774         return WERR_OK;
 775 }
 776
 777 /*
 778   drsuapi_DsGetDomainControllerInfo
 779 */
 780 static WERROR dcesrv_drsuapi_DsGetDomainControllerInfo(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 781                                                 struct drsuapi_DsGetDomainControllerInfo *r)
 782 {
 783         struct dcesrv_handle *h;
 784         struct drsuapi_bind_state *b_state;
 785         DCESRV_PULL_HANDLE_WERR(h, r->in.bind_handle, DRSUAPI_BIND_HANDLE);
 786         b_state = h->data;
 787
 788         switch (r->in.level) {
 789         case 1:
 790                 return dcesrv_drsuapi_DsGetDomainControllerInfo_1(b_state, mem_ctx, r);
 791         }
 792
 793         return WERR_UNKNOWN_LEVEL;
 794 }
 795
 796
 797
 798 /*
 799   DRSUAPI_EXECUTE_KCC
 800 */
 801 static WERROR dcesrv_DRSUAPI_EXECUTE_KCC(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 802                        struct DRSUAPI_EXECUTE_KCC *r)
 803 {
 804         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 805 }
 806
 807
 808 /*
 809   drsuapi_DsReplicaGetInfo
 810 */
 811 static WERROR dcesrv_drsuapi_DsReplicaGetInfo(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 812                        struct drsuapi_DsReplicaGetInfo *r)
 813 {
 814         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 815 }
 816
 817
 818 /*
 819   DRSUAPI_ADD_SID_HISTORY
 820 */
 821 static WERROR dcesrv_DRSUAPI_ADD_SID_HISTORY(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 822                        struct DRSUAPI_ADD_SID_HISTORY *r)
 823 {
 824         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 825 }
 826
 827 /*
 828   drsuapi_DsGetMemberships2
 829 */
 830 static WERROR dcesrv_drsuapi_DsGetMemberships2(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 831                        struct drsuapi_DsGetMemberships2 *r)
 832 {
 833         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 834 }
 835
 836 /*
 837   DRSUAPI_REPLICA_VERIFY_OBJECTS
 838 */
 839 static WERROR dcesrv_DRSUAPI_REPLICA_VERIFY_OBJECTS(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 840                        struct DRSUAPI_REPLICA_VERIFY_OBJECTS *r)
 841 {
 842         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 843 }
 844
 845
 846 /*
 847   DRSUAPI_GET_OBJECT_EXISTENCE
 848 */
 849 static WERROR dcesrv_DRSUAPI_GET_OBJECT_EXISTENCE(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 850                        struct DRSUAPI_GET_OBJECT_EXISTENCE *r)
 851 {
 852         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 853 }
 854
 855
 856 /*
 857   drsuapi_QuerySitesByCost
 858 */
 859 static WERROR dcesrv_drsuapi_QuerySitesByCost(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 860                        struct drsuapi_QuerySitesByCost *r)
 861 {
 862         DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
 863 }
 864
 865
 866 /* include the generated boilerplate */
 867 #include "librpc/gen_ndr/ndr_drsuapi_s.c"