s4:rpc_server/drsuapi: fix a crash in dcesrv_drsuapi_DsGetDomainControllerInfo_1()
[Samba/gebeck_regimport.git] / source4 / rpc_server / drsuapi / dcesrv_drsuapi.c
blob3512de60f5a584d2636cf8f93257d69cfcb19525
1 /*
2 Unix SMB/CIFS implementation.
4 endpoint server for the drsuapi pipe
6 Copyright (C) Stefan Metzmacher 2004
7 Copyright (C) Andrew Bartlett <abartlet@samba.org> 2006
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.
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.
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/>.
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 "dsdb/common/util.h"
29 #include "libcli/security/security.h"
30 #include "libcli/security/session.h"
31 #include "rpc_server/drsuapi/dcesrv_drsuapi.h"
32 #include "auth/auth.h"
33 #include "param/param.h"
34 #include "lib/messaging/irpc.h"
36 #define DRSUAPI_UNSUPPORTED(fname) do { \
37 DEBUG(1,(__location__ ": Unsupported DRS call %s\n", #fname)); \
38 if (DEBUGLVL(2)) NDR_PRINT_IN_DEBUG(fname, r); \
39 DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR); \
40 } while (0)
42 /*
43 drsuapi_DsBind
45 static WERROR dcesrv_drsuapi_DsBind(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
46 struct drsuapi_DsBind *r)
48 struct drsuapi_bind_state *b_state;
49 struct dcesrv_handle *handle;
50 struct drsuapi_DsBindInfoCtr *bind_info;
51 struct GUID site_guid;
52 struct ldb_result *site_res;
53 struct ldb_dn *server_site_dn;
54 static const char *site_attrs[] = { "objectGUID", NULL };
55 struct ldb_result *ntds_res;
56 struct ldb_dn *ntds_dn;
57 static const char *ntds_attrs[] = { "ms-DS-ReplicationEpoch", NULL };
58 uint32_t pid;
59 uint32_t repl_epoch;
60 int ret;
61 struct auth_session_info *auth_info;
62 WERROR werr;
63 bool connected_as_system = false;
65 r->out.bind_info = NULL;
66 ZERO_STRUCTP(r->out.bind_handle);
68 b_state = talloc_zero(mem_ctx, struct drsuapi_bind_state);
69 W_ERROR_HAVE_NO_MEMORY(b_state);
71 /* if this is a DC connecting, give them system level access */
72 werr = drs_security_level_check(dce_call, NULL, SECURITY_DOMAIN_CONTROLLER, NULL);
73 if (W_ERROR_IS_OK(werr)) {
74 DEBUG(3,(__location__ ": doing DsBind with system_session\n"));
75 auth_info = system_session(dce_call->conn->dce_ctx->lp_ctx);
76 connected_as_system = true;
77 } else {
78 auth_info = dce_call->conn->auth_state.session_info;
82 * connect to the samdb
84 b_state->sam_ctx = samdb_connect(b_state, dce_call->event_ctx,
85 dce_call->conn->dce_ctx->lp_ctx, auth_info, 0);
86 if (!b_state->sam_ctx) {
87 return WERR_FOOBAR;
90 if (connected_as_system) {
91 b_state->sam_ctx_system = b_state->sam_ctx;
92 } else {
93 /* an RODC also needs system samdb access for secret
94 attribute replication */
95 werr = drs_security_level_check(dce_call, NULL, SECURITY_RO_DOMAIN_CONTROLLER,
96 samdb_domain_sid(b_state->sam_ctx));
97 if (W_ERROR_IS_OK(werr)) {
98 b_state->sam_ctx_system = samdb_connect(b_state, dce_call->event_ctx,
99 dce_call->conn->dce_ctx->lp_ctx,
100 system_session(dce_call->conn->dce_ctx->lp_ctx), 0);
101 if (!b_state->sam_ctx_system) {
102 return WERR_FOOBAR;
108 * find out the guid of our own site
110 server_site_dn = samdb_server_site_dn(b_state->sam_ctx, mem_ctx);
111 W_ERROR_HAVE_NO_MEMORY(server_site_dn);
113 ret = ldb_search(b_state->sam_ctx, mem_ctx, &site_res,
114 server_site_dn, LDB_SCOPE_BASE, site_attrs,
115 "(objectClass=*)");
116 if (ret != LDB_SUCCESS) {
117 return WERR_DS_DRA_INTERNAL_ERROR;
119 if (site_res->count != 1) {
120 return WERR_DS_DRA_INTERNAL_ERROR;
122 site_guid = samdb_result_guid(site_res->msgs[0], "objectGUID");
125 * lookup the local servers Replication Epoch
127 ntds_dn = samdb_ntds_settings_dn(b_state->sam_ctx, mem_ctx);
128 W_ERROR_HAVE_NO_MEMORY(ntds_dn);
130 ret = ldb_search(b_state->sam_ctx, mem_ctx, &ntds_res,
131 ntds_dn, LDB_SCOPE_BASE, ntds_attrs,
132 "(objectClass=*)");
133 if (ret != LDB_SUCCESS) {
134 return WERR_DS_DRA_INTERNAL_ERROR;
136 if (ntds_res->count != 1) {
137 return WERR_DS_DRA_INTERNAL_ERROR;
139 repl_epoch = ldb_msg_find_attr_as_uint(ntds_res->msgs[0],
140 "ms-DS-ReplicationEpoch", 0);
143 * The "process identifier" of the client.
144 * According to the WSPP docs, sectin 5.35, this is
145 * for informational and debugging purposes only.
146 * The assignment is implementation specific.
148 pid = 0;
151 * store the clients bind_guid
153 if (r->in.bind_guid) {
154 b_state->remote_bind_guid = *r->in.bind_guid;
158 * store the clients bind_info
160 if (r->in.bind_info) {
161 switch (r->in.bind_info->length) {
162 case 24: {
163 struct drsuapi_DsBindInfo24 *info24;
164 info24 = &r->in.bind_info->info.info24;
165 b_state->remote_info28.supported_extensions = info24->supported_extensions;
166 b_state->remote_info28.site_guid = info24->site_guid;
167 b_state->remote_info28.pid = info24->pid;
168 b_state->remote_info28.repl_epoch = 0;
169 break;
171 case 28:
172 b_state->remote_info28 = r->in.bind_info->info.info28;
173 break;
178 * fill in our local bind info 28
180 b_state->local_info28.supported_extensions = 0;
181 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_BASE;
182 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_ASYNC_REPLICATION;
183 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_REMOVEAPI;
184 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_MOVEREQ_V2;
185 #if 0 /* we don't support MSZIP compression (only decompression) */
186 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_GETCHG_COMPRESS;
187 #endif
188 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V1;
189 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_RESTORE_USN_OPTIMIZATION;
190 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_KCC_EXECUTE;
191 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_ADDENTRY_V2;
192 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_LINKED_VALUE_REPLICATION;
193 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V2;
194 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_INSTANCE_TYPE_NOT_REQ_ON_MOD;
195 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_CRYPTO_BIND;
196 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_GET_REPL_INFO;
197 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_STRONG_ENCRYPTION;
198 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V01;
199 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_TRANSITIVE_MEMBERSHIP;
200 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_ADD_SID_HISTORY;
201 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_POST_BETA3;
202 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V5;
203 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_GET_MEMBERSHIPS2;
204 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V6;
205 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_NONDOMAIN_NCS;
206 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V8;
207 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V5;
208 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V6;
209 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_ADDENTRYREPLY_V3;
210 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V7;
211 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_VERIFY_OBJECT;
212 #if 0 /* we don't support XPRESS compression yet */
213 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_XPRESS_COMPRESS;
214 #endif
215 b_state->local_info28.supported_extensions |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V10;
216 b_state->local_info28.site_guid = site_guid;
217 b_state->local_info28.pid = pid;
218 b_state->local_info28.repl_epoch = repl_epoch;
221 * allocate the return bind_info
223 bind_info = talloc(mem_ctx, struct drsuapi_DsBindInfoCtr);
224 W_ERROR_HAVE_NO_MEMORY(bind_info);
226 bind_info->length = 28;
227 bind_info->info.info28 = b_state->local_info28;
230 * allocate a bind handle
232 handle = dcesrv_handle_new(dce_call->context, DRSUAPI_BIND_HANDLE);
233 W_ERROR_HAVE_NO_MEMORY(handle);
234 handle->data = talloc_steal(handle, b_state);
237 * prepare reply
239 r->out.bind_info = bind_info;
240 *r->out.bind_handle = handle->wire_handle;
242 return WERR_OK;
247 drsuapi_DsUnbind
249 static WERROR dcesrv_drsuapi_DsUnbind(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
250 struct drsuapi_DsUnbind *r)
252 struct dcesrv_handle *h;
254 *r->out.bind_handle = *r->in.bind_handle;
256 DCESRV_PULL_HANDLE_WERR(h, r->in.bind_handle, DRSUAPI_BIND_HANDLE);
258 talloc_free(h);
260 ZERO_STRUCTP(r->out.bind_handle);
262 return WERR_OK;
267 drsuapi_DsReplicaSync
269 static WERROR dcesrv_drsuapi_DsReplicaSync(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
270 struct drsuapi_DsReplicaSync *r)
272 WERROR status;
273 uint32_t timeout;
275 status = drs_security_level_check(dce_call, "DsReplicaSync", SECURITY_DOMAIN_CONTROLLER, NULL);
276 if (!W_ERROR_IS_OK(status)) {
277 return status;
280 if (r->in.level != 1) {
281 DEBUG(0,("DsReplicaSync called with unsupported level %d\n", r->in.level));
282 return WERR_DS_DRA_INVALID_PARAMETER;
285 if (r->in.req->req1.options & DRSUAPI_DRS_ASYNC_OP) {
286 timeout = IRPC_CALL_TIMEOUT;
287 } else {
289 * use Infinite time for timeout in case
290 * the caller made a sync call
292 timeout = IRPC_CALL_TIMEOUT_INF;
295 dcesrv_irpc_forward_rpc_call(dce_call, mem_ctx,
296 r, NDR_DRSUAPI_DSREPLICASYNC,
297 &ndr_table_drsuapi,
298 "dreplsrv", "DsReplicaSync",
299 timeout);
301 return WERR_OK;
306 drsuapi_DsReplicaAdd
308 static WERROR dcesrv_drsuapi_DsReplicaAdd(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
309 struct drsuapi_DsReplicaAdd *r)
311 WERROR status;
313 status = drs_security_level_check(dce_call, "DsReplicaAdd", SECURITY_DOMAIN_CONTROLLER, NULL);
314 if (!W_ERROR_IS_OK(status)) {
315 return status;
318 dcesrv_irpc_forward_rpc_call(dce_call, mem_ctx,
319 r, NDR_DRSUAPI_DSREPLICAADD,
320 &ndr_table_drsuapi,
321 "dreplsrv", "DsReplicaAdd",
322 IRPC_CALL_TIMEOUT);
324 return WERR_OK;
329 drsuapi_DsReplicaDel
331 static WERROR dcesrv_drsuapi_DsReplicaDel(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
332 struct drsuapi_DsReplicaDel *r)
334 WERROR status;
336 status = drs_security_level_check(dce_call, "DsReplicaDel", SECURITY_DOMAIN_CONTROLLER, NULL);
337 if (!W_ERROR_IS_OK(status)) {
338 return status;
341 dcesrv_irpc_forward_rpc_call(dce_call, mem_ctx,
342 r, NDR_DRSUAPI_DSREPLICADEL,
343 &ndr_table_drsuapi,
344 "dreplsrv", "DsReplicaDel",
345 IRPC_CALL_TIMEOUT);
347 return WERR_OK;
352 drsuapi_DsReplicaModify
354 static WERROR dcesrv_drsuapi_DsReplicaMod(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
355 struct drsuapi_DsReplicaMod *r)
357 WERROR status;
359 status = drs_security_level_check(dce_call, "DsReplicaMod", SECURITY_DOMAIN_CONTROLLER, NULL);
360 if (!W_ERROR_IS_OK(status)) {
361 return status;
364 dcesrv_irpc_forward_rpc_call(dce_call, mem_ctx,
365 r, NDR_DRSUAPI_DSREPLICAMOD,
366 &ndr_table_drsuapi,
367 "dreplsrv", "DsReplicaMod",
368 IRPC_CALL_TIMEOUT);
370 return WERR_OK;
375 DRSUAPI_VERIFY_NAMES
377 static WERROR dcesrv_DRSUAPI_VERIFY_NAMES(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
378 struct DRSUAPI_VERIFY_NAMES *r)
380 DRSUAPI_UNSUPPORTED(DRSUAPI_VERIFY_NAMES);
385 drsuapi_DsGetMemberships
387 static WERROR dcesrv_drsuapi_DsGetMemberships(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
388 struct drsuapi_DsGetMemberships *r)
390 DRSUAPI_UNSUPPORTED(drsuapi_DsGetMemberships);
395 DRSUAPI_INTER_DOMAIN_MOVE
397 static WERROR dcesrv_DRSUAPI_INTER_DOMAIN_MOVE(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
398 struct DRSUAPI_INTER_DOMAIN_MOVE *r)
400 DRSUAPI_UNSUPPORTED(DRSUAPI_INTER_DOMAIN_MOVE);
405 drsuapi_DsGetNT4ChangeLog
407 static WERROR dcesrv_drsuapi_DsGetNT4ChangeLog(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
408 struct drsuapi_DsGetNT4ChangeLog *r)
410 DRSUAPI_UNSUPPORTED(drsuapi_DsGetNT4ChangeLog);
414 drsuapi_DsCrackNames
416 static WERROR dcesrv_drsuapi_DsCrackNames(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
417 struct drsuapi_DsCrackNames *r)
419 struct drsuapi_bind_state *b_state;
420 struct dcesrv_handle *h;
422 *r->out.level_out = r->in.level;
424 DCESRV_PULL_HANDLE_WERR(h, r->in.bind_handle, DRSUAPI_BIND_HANDLE);
425 b_state = h->data;
427 r->out.ctr = talloc_zero(mem_ctx, union drsuapi_DsNameCtr);
428 W_ERROR_HAVE_NO_MEMORY(r->out.ctr);
430 switch (r->in.level) {
431 case 1: {
432 switch(r->in.req->req1.format_offered){
433 case DRSUAPI_DS_NAME_FORMAT_UPN_AND_ALTSECID:
434 case DRSUAPI_DS_NAME_FORMAT_NT4_ACCOUNT_NAME_SANS_DOMAIN_EX:
435 case DRSUAPI_DS_NAME_FORMAT_LIST_GLOBAL_CATALOG_SERVERS:
436 case DRSUAPI_DS_NAME_FORMAT_UPN_FOR_LOGON:
437 case DRSUAPI_DS_NAME_FORMAT_LIST_SERVERS_WITH_DCS_IN_SITE:
438 case DRSUAPI_DS_NAME_FORMAT_STRING_SID_NAME:
439 case DRSUAPI_DS_NAME_FORMAT_ALT_SECURITY_IDENTITIES_NAME:
440 case DRSUAPI_DS_NAME_FORMAT_LIST_NCS:
441 case DRSUAPI_DS_NAME_FORMAT_LIST_DOMAINS:
442 case DRSUAPI_DS_NAME_FORMAT_MAP_SCHEMA_GUID:
443 case DRSUAPI_DS_NAME_FORMAT_NT4_ACCOUNT_NAME_SANS_DOMAIN:
444 case DRSUAPI_DS_NAME_FORMAT_LIST_INFO_FOR_SERVER:
445 case DRSUAPI_DS_NAME_FORMAT_LIST_SERVERS_FOR_DOMAIN_IN_SITE:
446 case DRSUAPI_DS_NAME_FORMAT_LIST_DOMAINS_IN_SITE:
447 case DRSUAPI_DS_NAME_FORMAT_LIST_SERVERS_IN_SITE:
448 case DRSUAPI_DS_NAME_FORMAT_LIST_SITES:
449 DEBUG(0, ("DsCrackNames: Unsupported operation requested: %X",
450 r->in.req->req1.format_offered));
451 return WERR_OK;
452 case DRSUAPI_DS_NAME_FORMAT_LIST_ROLES:
453 return dcesrv_drsuapi_ListRoles(b_state->sam_ctx, mem_ctx,
454 &r->in.req->req1, &r->out.ctr->ctr1);
455 default:/* format_offered is in the enum drsuapi_DsNameFormat*/
456 return dcesrv_drsuapi_CrackNamesByNameFormat(b_state->sam_ctx, mem_ctx,
457 &r->in.req->req1, &r->out.ctr->ctr1);
461 return WERR_UNKNOWN_LEVEL;
466 drsuapi_DsRemoveDSServer
468 static WERROR dcesrv_drsuapi_DsRemoveDSServer(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
469 struct drsuapi_DsRemoveDSServer *r)
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 WERROR status;
478 *r->out.level_out = 1;
480 status = drs_security_level_check(dce_call, "DsRemoveDSServer", SECURITY_DOMAIN_CONTROLLER, NULL);
481 if (!W_ERROR_IS_OK(status)) {
482 return status;
485 DCESRV_PULL_HANDLE_WERR(h, r->in.bind_handle, DRSUAPI_BIND_HANDLE);
486 b_state = h->data;
488 switch (r->in.level) {
489 case 1:
490 ntds_dn = ldb_dn_new(mem_ctx, b_state->sam_ctx, r->in.req->req1.server_dn);
491 W_ERROR_HAVE_NO_MEMORY(ntds_dn);
493 ok = ldb_dn_validate(ntds_dn);
494 if (!ok) {
495 return WERR_FOOBAR;
498 /* TODO: it's likely that we need more checks here */
500 ok = ldb_dn_add_child_fmt(ntds_dn, "CN=NTDS Settings");
501 if (!ok) {
502 return WERR_FOOBAR;
505 if (r->in.req->req1.commit) {
506 ret = dsdb_delete(b_state->sam_ctx, ntds_dn, DSDB_TREE_DELETE);
507 if (ret != LDB_SUCCESS) {
508 return WERR_FOOBAR;
512 return WERR_OK;
513 default:
514 break;
517 return WERR_FOOBAR;
522 DRSUAPI_REMOVE_DS_DOMAIN
524 static WERROR dcesrv_DRSUAPI_REMOVE_DS_DOMAIN(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
525 struct DRSUAPI_REMOVE_DS_DOMAIN *r)
527 DRSUAPI_UNSUPPORTED(DRSUAPI_REMOVE_DS_DOMAIN);
530 /* Obtain the site name from a server DN */
531 static const char *result_site_name(struct ldb_dn *server_dn)
533 /* Format is cn=<NETBIOS name>,cn=Servers,cn=<site>,cn=sites.... */
534 const struct ldb_val *val = ldb_dn_get_component_val(server_dn, 2);
535 const char *name = ldb_dn_get_component_name(server_dn, 2);
537 if (!name || (ldb_attr_cmp(name, "cn") != 0)) {
538 /* Ensure this matches the format. This gives us a
539 * bit more confidence that a 'cn' value will be a
540 * ascii string */
541 return NULL;
543 if (val) {
544 return (char *)val->data;
546 return NULL;
550 drsuapi_DsGetDomainControllerInfo
552 static WERROR dcesrv_drsuapi_DsGetDomainControllerInfo_1(struct drsuapi_bind_state *b_state,
553 TALLOC_CTX *mem_ctx,
554 struct drsuapi_DsGetDomainControllerInfo *r)
556 struct ldb_dn *sites_dn;
557 struct ldb_result *res;
559 const char *attrs_account_1[] = { "cn", "dnsHostName", NULL };
560 const char *attrs_account_2[] = { "cn", "dnsHostName", "objectGUID", NULL };
562 const char *attrs_none[] = { NULL };
564 const char *attrs_site[] = { "objectGUID", NULL };
566 const char *attrs_ntds[] = { "options", "objectGUID", NULL };
568 const char *attrs_1[] = { "serverReference", "cn", "dnsHostName", NULL };
569 const char *attrs_2[] = { "serverReference", "cn", "dnsHostName", "objectGUID", NULL };
570 const char **attrs;
572 struct drsuapi_DsGetDCInfoCtr1 *ctr1;
573 struct drsuapi_DsGetDCInfoCtr2 *ctr2;
575 int ret;
576 unsigned int i;
578 *r->out.level_out = r->in.req->req1.level;
579 r->out.ctr = talloc_zero(mem_ctx, union drsuapi_DsGetDCInfoCtr);
580 W_ERROR_HAVE_NO_MEMORY(r->out.ctr);
582 switch (*r->out.level_out) {
583 case -1:
584 /* this level is not like the others */
585 return WERR_UNKNOWN_LEVEL;
586 case 1:
587 attrs = attrs_1;
588 break;
589 case 2:
590 attrs = attrs_2;
591 break;
592 default:
593 return WERR_UNKNOWN_LEVEL;
596 sites_dn = samdb_sites_dn(b_state->sam_ctx, mem_ctx);
597 if (!sites_dn) {
598 return WERR_DS_OBJ_NOT_FOUND;
601 ret = ldb_search(b_state->sam_ctx, mem_ctx, &res, sites_dn, LDB_SCOPE_SUBTREE, attrs,
602 "objectClass=server");
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;
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);
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");
628 if (!ntds_dn || !ldb_dn_add_child_fmt(ntds_dn, "CN=NTDS Settings")) {
629 return WERR_NOMEM;
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);
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);
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;
655 if (res_domain->count == 1) {
656 ctr1->array[i].is_pdc = true;
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)));
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);
670 ctr1->array[i].is_enabled = true;
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");
693 if (!ntds_dn || !ldb_dn_add_child_fmt(ntds_dn, "CN=NTDS Settings")) {
694 return WERR_NOMEM;
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;
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_uint(res_ntds->msgs[0], "options", 0) & DS_NTDSDSA_OPT_IS_GC);
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);
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)));
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);
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)));
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");
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);
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;
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)));
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)));
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");
769 ctr2->array[i].is_enabled = true;
772 break;
774 return WERR_OK;
778 drsuapi_DsGetDomainControllerInfo
780 static WERROR dcesrv_drsuapi_DsGetDomainControllerInfo(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
781 struct drsuapi_DsGetDomainControllerInfo *r)
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;
788 switch (r->in.level) {
789 case 1:
790 return dcesrv_drsuapi_DsGetDomainControllerInfo_1(b_state, mem_ctx, r);
793 return WERR_UNKNOWN_LEVEL;
799 drsuapi_DsExecuteKCC
801 static WERROR dcesrv_drsuapi_DsExecuteKCC(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
802 struct drsuapi_DsExecuteKCC *r)
804 WERROR status;
805 status = drs_security_level_check(dce_call, "DsExecuteKCC", SECURITY_DOMAIN_CONTROLLER, NULL);
807 if (!W_ERROR_IS_OK(status)) {
808 return status;
811 dcesrv_irpc_forward_rpc_call(dce_call, mem_ctx, r, NDR_DRSUAPI_DSEXECUTEKCC,
812 &ndr_table_drsuapi, "kccsrv", "DsExecuteKCC",
813 IRPC_CALL_TIMEOUT);
814 return WERR_OK;
819 drsuapi_DsReplicaGetInfo
821 static WERROR dcesrv_drsuapi_DsReplicaGetInfo(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
822 struct drsuapi_DsReplicaGetInfo *r)
824 enum security_user_level level;
826 if (!lpcfg_parm_bool(dce_call->conn->dce_ctx->lp_ctx, NULL,
827 "drs", "disable_sec_check", false)) {
828 level = security_session_user_level(dce_call->conn->auth_state.session_info, NULL);
829 if (level < SECURITY_DOMAIN_CONTROLLER) {
830 DEBUG(1,(__location__ ": Administrator access required for DsReplicaGetInfo\n"));
831 security_token_debug(0, 2, dce_call->conn->auth_state.session_info->security_token);
832 return WERR_DS_DRA_ACCESS_DENIED;
836 dcesrv_irpc_forward_rpc_call(dce_call, mem_ctx, r, NDR_DRSUAPI_DSREPLICAGETINFO,
837 &ndr_table_drsuapi, "kccsrv", "DsReplicaGetInfo",
838 IRPC_CALL_TIMEOUT);
840 return WERR_OK;
845 DRSUAPI_ADD_SID_HISTORY
847 static WERROR dcesrv_DRSUAPI_ADD_SID_HISTORY(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
848 struct DRSUAPI_ADD_SID_HISTORY *r)
850 DRSUAPI_UNSUPPORTED(DRSUAPI_ADD_SID_HISTORY);
854 drsuapi_DsGetMemberships2
856 static WERROR dcesrv_drsuapi_DsGetMemberships2(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
857 struct drsuapi_DsGetMemberships2 *r)
859 DRSUAPI_UNSUPPORTED(drsuapi_DsGetMemberships2);
863 DRSUAPI_REPLICA_VERIFY_OBJECTS
865 static WERROR dcesrv_DRSUAPI_REPLICA_VERIFY_OBJECTS(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
866 struct DRSUAPI_REPLICA_VERIFY_OBJECTS *r)
868 DRSUAPI_UNSUPPORTED(DRSUAPI_REPLICA_VERIFY_OBJECTS);
873 DRSUAPI_GET_OBJECT_EXISTENCE
875 static WERROR dcesrv_DRSUAPI_GET_OBJECT_EXISTENCE(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
876 struct DRSUAPI_GET_OBJECT_EXISTENCE *r)
878 DRSUAPI_UNSUPPORTED(DRSUAPI_GET_OBJECT_EXISTENCE);
883 drsuapi_QuerySitesByCost
885 static WERROR dcesrv_drsuapi_QuerySitesByCost(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
886 struct drsuapi_QuerySitesByCost *r)
888 DRSUAPI_UNSUPPORTED(drsuapi_QuerySitesByCost);
892 /* include the generated boilerplate */
893 #include "librpc/gen_ndr/ndr_drsuapi_s.c"