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