s4:libnet: s/highestCommitedUSN/highestCommittedUSN
[sfrench/samba-autobuild/.git] / source4 / libnet / libnet_become_dc.c
index 25a216524e4ab3368c3f2aef21cddbace810e69c..43a32099a547d228ffec534abb88d142d9bea764 100644 (file)
@@ -21,8 +21,8 @@
 #include "libnet/libnet.h"
 #include "libcli/composite/composite.h"
 #include "libcli/cldap/cldap.h"
-#include "lib/ldb/include/ldb.h"
-#include "lib/ldb/include/ldb_errors.h"
+#include <ldb.h>
+#include <ldb_errors.h>
 #include "ldb_wrap.h"
 #include "dsdb/samdb/samdb.h"
 #include "../libds/common/flags.h"
@@ -84,7 +84,7 @@
  *             supportedLDAPVersion:   3
  *                                     2
  *             supportedLDAPPolicies:  ...
- *             highestCommitedUSN:     ...
+ *             highestCommittedUSN:    ...
  *             supportedSASLMechanisms:GSSAPI
  *                                     GSS-SPNEGO
  *                                     EXTERNAL
@@ -735,13 +735,14 @@ struct libnet_BecomeDC_state {
        struct libnet_BecomeDC_Callbacks callbacks;
 
        bool rodc_join;
+       bool critical_only;
 };
 
 static int32_t get_dc_function_level(struct loadparm_context *lp_ctx)
 {
-       /* per default we are (Windows) 2008 compatible */
-       return lp_parm_int(lp_ctx, NULL, "ads", "dc function level",
-               DS_DOMAIN_FUNCTION_2008);
+       /* per default we are (Windows) 2008 R2 compatible */
+       return lpcfg_parm_int(lp_ctx, NULL, "ads", "dc function level",
+                          DS_DOMAIN_FUNCTION_2008_R2);
 }
 
 static void becomeDC_recv_cldap(struct tevent_req *req);
@@ -766,18 +767,18 @@ static void becomeDC_send_cldap(struct libnet_BecomeDC_state *s)
 
        ret = tsocket_address_inet_from_strings(s, "ip",
                                                s->source_dsa.address,
-                                               lp_cldap_port(s->libnet->lp_ctx),
+                                               lpcfg_cldap_port(s->libnet->lp_ctx),
                                                &dest_address);
        if (ret != 0) {
-               c->status = map_nt_error_from_unix(errno);
+               c->status = map_nt_error_from_unix_common(errno);
                if (!composite_is_ok(c)) return;
        }
 
-       c->status = cldap_socket_init(s, s->libnet->event_ctx,
-                                     NULL, dest_address, &s->cldap.sock);
+       c->status = cldap_socket_init(s, NULL, dest_address, &s->cldap.sock);
        if (!composite_is_ok(c)) return;
 
-       req = cldap_netlogon_send(s, s->cldap.sock, &s->cldap.io);
+       req = cldap_netlogon_send(s, s->libnet->event_ctx,
+                                 s->cldap.sock, &s->cldap.io);
        if (composite_nomem(req, c)) return;
        tevent_req_set_callback(req, becomeDC_recv_cldap, s);
 }
@@ -1046,11 +1047,17 @@ static NTSTATUS becomeDC_ldap1_infrastructure_fsmo(struct libnet_BecomeDC_state
                                DS_GUID_INFRASTRUCTURE_CONTAINER,
                                &basedn);
        if (ret != LDB_SUCCESS) {
+               DEBUG(0,("Failed to get well known DN for DS_GUID_INFRASTRUCTURE_CONTAINER on %s: %s\n", 
+                        ldb_dn_get_linearized(ldb_get_default_basedn(s->ldap1.ldb)), 
+                        ldb_errstring(s->ldap1.ldb)));
                return NT_STATUS_LDAP(ret);
        }
 
        ret = samdb_reference_dn(s->ldap1.ldb, s, basedn, "fSMORoleOwner", &ntds_dn);
        if (ret != LDB_SUCCESS) {
+               DEBUG(0,("Failed to get reference DN from fsmoRoleOwner on %s: %s\n", 
+                        ldb_dn_get_linearized(basedn), 
+                        ldb_errstring(s->ldap1.ldb)));
                talloc_free(basedn);
                return NT_STATUS_LDAP(ret);
        }
@@ -1067,21 +1074,28 @@ static NTSTATUS becomeDC_ldap1_infrastructure_fsmo(struct libnet_BecomeDC_state
        ret = ldb_search(s->ldap1.ldb, s, &r, server_dn, LDB_SCOPE_BASE,
                         dns_attrs, "(objectClass=*)");
        if (ret != LDB_SUCCESS) {
+               DEBUG(0,("Failed to get server DN %s: %s\n", 
+                        ldb_dn_get_linearized(server_dn), 
+                        ldb_errstring(s->ldap1.ldb)));
                return NT_STATUS_LDAP(ret);
        } else if (r->count != 1) {
                talloc_free(r);
                return NT_STATUS_INVALID_NETWORK_RESPONSE;
        }
 
-       s->infrastructure_fsmo.dns_name = samdb_result_string(r->msgs[0], "dnsHostName", NULL);
+       s->infrastructure_fsmo.dns_name = ldb_msg_find_attr_as_string(r->msgs[0], "dnsHostName", NULL);
        if (!s->infrastructure_fsmo.dns_name) return NT_STATUS_INVALID_NETWORK_RESPONSE;
        talloc_steal(s, s->infrastructure_fsmo.dns_name);
 
        talloc_free(r);
 
+       ldb_dn_remove_extended_components(ntds_dn);
        ret = ldb_search(s->ldap1.ldb, s, &r, ntds_dn, LDB_SCOPE_BASE,
                         guid_attrs, "(objectClass=*)");
        if (ret != LDB_SUCCESS) {
+               DEBUG(0,("Failed to get NTDS Settings DN %s: %s\n", 
+                        ldb_dn_get_linearized(ntds_dn), 
+                        ldb_errstring(s->ldap1.ldb)));
                return NT_STATUS_LDAP(ret);
        } else if (r->count != 1) {
                talloc_free(r);
@@ -1133,7 +1147,7 @@ static NTSTATUS becomeDC_ldap1_rid_manager_fsmo(struct libnet_BecomeDC_state *s)
                return NT_STATUS_INVALID_NETWORK_RESPONSE;
        }
 
-       reference_dn_str        = samdb_result_string(r->msgs[0], "rIDManagerReference", NULL);
+       reference_dn_str        = ldb_msg_find_attr_as_string(r->msgs[0], "rIDManagerReference", NULL);
        if (!reference_dn_str) return NT_STATUS_INVALID_NETWORK_RESPONSE;
 
        basedn = ldb_dn_new(s, s->ldap1.ldb, reference_dn_str);
@@ -1151,7 +1165,7 @@ static NTSTATUS becomeDC_ldap1_rid_manager_fsmo(struct libnet_BecomeDC_state *s)
                return NT_STATUS_INVALID_NETWORK_RESPONSE;
        }
 
-       s->rid_manager_fsmo.ntds_dn_str = samdb_result_string(r->msgs[0], "fSMORoleOwner", NULL);
+       s->rid_manager_fsmo.ntds_dn_str = ldb_msg_find_attr_as_string(r->msgs[0], "fSMORoleOwner", NULL);
        if (!s->rid_manager_fsmo.ntds_dn_str) return NT_STATUS_INVALID_NETWORK_RESPONSE;
        talloc_steal(s, s->rid_manager_fsmo.ntds_dn_str);
 
@@ -1175,7 +1189,7 @@ static NTSTATUS becomeDC_ldap1_rid_manager_fsmo(struct libnet_BecomeDC_state *s)
                return NT_STATUS_INVALID_NETWORK_RESPONSE;
        }
 
-       s->rid_manager_fsmo.dns_name    = samdb_result_string(r->msgs[0], "dnsHostName", NULL);
+       s->rid_manager_fsmo.dns_name    = ldb_msg_find_attr_as_string(r->msgs[0], "dnsHostName", NULL);
        if (!s->rid_manager_fsmo.dns_name) return NT_STATUS_INVALID_NETWORK_RESPONSE;
        talloc_steal(s, s->rid_manager_fsmo.dns_name);
 
@@ -1260,11 +1274,11 @@ static NTSTATUS becomeDC_ldap1_computer_object(struct libnet_BecomeDC_state *s)
                return NT_STATUS_INVALID_NETWORK_RESPONSE;
        }
 
-       s->dest_dsa.computer_dn_str     = samdb_result_string(r->msgs[0], "distinguishedName", NULL);
+       s->dest_dsa.computer_dn_str     = ldb_msg_find_attr_as_string(r->msgs[0], "distinguishedName", NULL);
        if (!s->dest_dsa.computer_dn_str) return NT_STATUS_INVALID_NETWORK_RESPONSE;
        talloc_steal(s, s->dest_dsa.computer_dn_str);
 
-       s->dest_dsa.user_account_control = samdb_result_uint(r->msgs[0], "userAccountControl", 0);
+       s->dest_dsa.user_account_control = ldb_msg_find_attr_as_uint(r->msgs[0], "userAccountControl", 0);
 
        talloc_free(r);
        return NT_STATUS_OK;
@@ -1298,7 +1312,7 @@ static NTSTATUS becomeDC_ldap1_server_object_1(struct libnet_BecomeDC_state *s)
                return NT_STATUS_INVALID_NETWORK_RESPONSE;
        }
 
-       server_reference_dn_str = samdb_result_string(r->msgs[0], "serverReference", NULL);
+       server_reference_dn_str = ldb_msg_find_attr_as_string(r->msgs[0], "serverReference", NULL);
        if (server_reference_dn_str) {
                server_reference_dn     = ldb_dn_new(r, s->ldap1.ldb, server_reference_dn_str);
                NT_STATUS_HAVE_NO_MEMORY(server_reference_dn);
@@ -1317,7 +1331,7 @@ static NTSTATUS becomeDC_ldap1_server_object_1(struct libnet_BecomeDC_state *s)
        }
 
        /* if the server object is already for the dest_dsa, then we don't need to create it */
-       s->dest_dsa.server_dn_str       = samdb_result_string(r->msgs[0], "distinguishedName", NULL);
+       s->dest_dsa.server_dn_str       = ldb_msg_find_attr_as_string(r->msgs[0], "distinguishedName", NULL);
        if (!s->dest_dsa.server_dn_str) return NT_STATUS_INVALID_NETWORK_RESPONSE;
        talloc_steal(s, s->dest_dsa.server_dn_str);
 
@@ -1352,7 +1366,7 @@ static NTSTATUS becomeDC_ldap1_server_object_2(struct libnet_BecomeDC_state *s)
                return NT_STATUS_INVALID_NETWORK_RESPONSE;
        }
 
-       server_reference_bl_dn_str = samdb_result_string(r->msgs[0], "serverReferenceBL", NULL);
+       server_reference_bl_dn_str = ldb_msg_find_attr_as_string(r->msgs[0], "serverReferenceBL", NULL);
        if (!server_reference_bl_dn_str) {
                /* if no back link is present, we're done for this function */
                talloc_free(r);
@@ -1360,7 +1374,7 @@ static NTSTATUS becomeDC_ldap1_server_object_2(struct libnet_BecomeDC_state *s)
        }
 
        /* if the server object is already for the dest_dsa, then we don't need to create it */
-       s->dest_dsa.server_dn_str       = samdb_result_string(r->msgs[0], "serverReferenceBL", NULL);
+       s->dest_dsa.server_dn_str       = ldb_msg_find_attr_as_string(r->msgs[0], "serverReferenceBL", NULL);
        if (s->dest_dsa.server_dn_str) {
                /* if a back link is present, we know that the server object is present */
                talloc_steal(s, s->dest_dsa.server_dn_str);
@@ -1549,25 +1563,33 @@ static void becomeDC_drsuapi_connect_send(struct libnet_BecomeDC_state *s,
                 *       configuration partition works fine, but it fails for
                 *       the domain partition.
                 */
-               if (lp_parm_bool(s->libnet->lp_ctx, NULL, "become_dc",
+               if (lpcfg_parm_bool(s->libnet->lp_ctx, NULL, "become_dc",
                                 "force krb5", true))
                {
                        krb5_str = "krb5,";
                }
-               if (lp_parm_bool(s->libnet->lp_ctx, NULL, "become_dc",
+               if (lpcfg_parm_bool(s->libnet->lp_ctx, NULL, "become_dc",
                                 "print", false))
                {
                        print_str = "print,";
                }
-               binding_str = talloc_asprintf(s, "ncacn_ip_tcp:%s[%s%sseal]",
-                                             s->source_dsa.dns_name,
-                                             krb5_str, print_str);
+               binding_str = talloc_asprintf(s, "ncacn_ip_tcp:%s[%s%sseal,target_hostname=%s]",
+                                             s->source_dsa.address,
+                                             krb5_str, print_str,
+                                             s->source_dsa.dns_name);
                if (composite_nomem(binding_str, c)) return;
                c->status = dcerpc_parse_binding(s, binding_str, &drsuapi->binding);
                talloc_free(binding_str);
                if (!composite_is_ok(c)) return;
        }
 
+       if (DEBUGLEVEL >= 10) {
+               c->status = dcerpc_binding_set_flags(drsuapi->binding,
+                                                    DCERPC_DEBUG_PRINT_BOTH,
+                                                    0);
+               if (!composite_is_ok(c)) return;
+       }
+
        creq = dcerpc_pipe_connect_b_send(s, drsuapi->binding, &ndr_table_drsuapi,
                                          s->libnet->cred, s->libnet->event_ctx,
                                          s->libnet->lp_ctx);
@@ -1591,6 +1613,7 @@ static void becomeDC_drsuapi1_connect_recv(struct composite_context *req)
        s->drsuapi1.drsuapi_handle = s->drsuapi1.pipe->binding_handle;
 
        c->status = gensec_session_key(s->drsuapi1.pipe->conn->security_state.generic_state,
+                                      s,
                                       &s->drsuapi1.gensec_skey);
        if (!composite_is_ok(c)) return;
 
@@ -1680,6 +1703,19 @@ static WERROR becomeDC_drsuapi_bind_recv(struct libnet_BecomeDC_state *s,
                        drsuapi->remote_info28.repl_epoch               = 0;
                        break;
                }
+               case 28: {
+                       drsuapi->remote_info28 = drsuapi->bind_r.out.bind_info->info.info28;
+                       break;
+               }
+               case 32: {
+                       struct drsuapi_DsBindInfo32 *info32;
+                       info32 = &drsuapi->bind_r.out.bind_info->info.info32;
+                       drsuapi->remote_info28.supported_extensions     = info32->supported_extensions;
+                       drsuapi->remote_info28.site_guid                = info32->site_guid;
+                       drsuapi->remote_info28.pid                      = info32->pid;
+                       drsuapi->remote_info28.repl_epoch               = info32->repl_epoch;
+                       break;
+               }
                case 48: {
                        struct drsuapi_DsBindInfo48 *info48;
                        info48 = &drsuapi->bind_r.out.bind_info->info.info48;
@@ -1689,8 +1725,18 @@ static WERROR becomeDC_drsuapi_bind_recv(struct libnet_BecomeDC_state *s,
                        drsuapi->remote_info28.repl_epoch               = info48->repl_epoch;
                        break;
                }
-               case 28:
-                       drsuapi->remote_info28 = drsuapi->bind_r.out.bind_info->info.info28;
+               case 52: {
+                       struct drsuapi_DsBindInfo52 *info52;
+                       info52 = &drsuapi->bind_r.out.bind_info->info.info52;
+                       drsuapi->remote_info28.supported_extensions     = info52->supported_extensions;
+                       drsuapi->remote_info28.site_guid                = info52->site_guid;
+                       drsuapi->remote_info28.pid                      = info52->pid;
+                       drsuapi->remote_info28.repl_epoch               = info52->repl_epoch;
+                       break;
+               }
+               default:
+                       DEBUG(1, ("Warning: invalid info length in bind info: %d\n",
+                                       drsuapi->bind_r.out.bind_info->length));
                        break;
                }
        }
@@ -1837,7 +1883,7 @@ static void becomeDC_drsuapi1_add_entry_send(struct libnet_BecomeDC_state *s)
 
                vs[0].blob              = &vd[0];
 
-               attrs[i].attid                  = DRSUAPI_ATTRIBUTE_ntSecurityDescriptor;
+               attrs[i].attid                  = DRSUAPI_ATTID_ntSecurityDescriptor;
                attrs[i].value_ctr.num_values   = 1;
                attrs[i].value_ctr.values       = vs;
 
@@ -1863,7 +1909,7 @@ static void becomeDC_drsuapi1_add_entry_send(struct libnet_BecomeDC_state *s)
 
                vs[0].blob              = &vd[0];
 
-               attrs[i].attid                  = DRSUAPI_ATTRIBUTE_objectClass;
+               attrs[i].attid                  = DRSUAPI_ATTID_objectClass;
                attrs[i].value_ctr.num_values   = 1;
                attrs[i].value_ctr.values       = vs;
 
@@ -1903,7 +1949,7 @@ static void becomeDC_drsuapi1_add_entry_send(struct libnet_BecomeDC_state *s)
 
                vs[0].blob              = &vd[0];
 
-               attrs[i].attid                  = DRSUAPI_ATTRIBUTE_objectCategory;
+               attrs[i].attid                  = DRSUAPI_ATTID_objectCategory;
                attrs[i].value_ctr.num_values   = 1;
                attrs[i].value_ctr.values       = vs;
 
@@ -1929,7 +1975,7 @@ static void becomeDC_drsuapi1_add_entry_send(struct libnet_BecomeDC_state *s)
 
                vs[0].blob              = &vd[0];
 
-               attrs[i].attid                  = DRSUAPI_ATTRIBUTE_invocationId;
+               attrs[i].attid                  = DRSUAPI_ATTID_invocationId;
                attrs[i].value_ctr.num_values   = 1;
                attrs[i].value_ctr.values       = vs;
 
@@ -1985,7 +2031,7 @@ static void becomeDC_drsuapi1_add_entry_send(struct libnet_BecomeDC_state *s)
                vs[1].blob              = &vd[1];
                vs[2].blob              = &vd[2];
 
-               attrs[i].attid                  = DRSUAPI_ATTRIBUTE_hasMasterNCs;
+               attrs[i].attid                  = DRSUAPI_ATTID_hasMasterNCs;
                attrs[i].value_ctr.num_values   = 3;
                attrs[i].value_ctr.values       = vs;
 
@@ -2041,7 +2087,7 @@ static void becomeDC_drsuapi1_add_entry_send(struct libnet_BecomeDC_state *s)
                vs[1].blob              = &vd[1];
                vs[2].blob              = &vd[2];
 
-               attrs[i].attid                  = DRSUAPI_ATTRIBUTE_msDS_hasMasterNCs;
+               attrs[i].attid                  = DRSUAPI_ATTID_msDS_hasMasterNCs;
                attrs[i].value_ctr.num_values   = 3;
                attrs[i].value_ctr.values       = vs;
 
@@ -2073,7 +2119,7 @@ static void becomeDC_drsuapi1_add_entry_send(struct libnet_BecomeDC_state *s)
 
                vs[0].blob              = &vd[0];
 
-               attrs[i].attid                  = DRSUAPI_ATTRIBUTE_dMDLocation;
+               attrs[i].attid                  = DRSUAPI_ATTID_dMDLocation;
                attrs[i].value_ctr.num_values   = 1;
                attrs[i].value_ctr.values       = vs;
 
@@ -2105,7 +2151,7 @@ static void becomeDC_drsuapi1_add_entry_send(struct libnet_BecomeDC_state *s)
 
                vs[0].blob              = &vd[0];
 
-               attrs[i].attid                  = DRSUAPI_ATTRIBUTE_msDS_HasDomainNCs;
+               attrs[i].attid                  = DRSUAPI_ATTID_msDS_HasDomainNCs;
                attrs[i].value_ctr.num_values   = 1;
                attrs[i].value_ctr.values       = vs;
 
@@ -2130,7 +2176,7 @@ static void becomeDC_drsuapi1_add_entry_send(struct libnet_BecomeDC_state *s)
 
                vs[0].blob              = &vd[0];
 
-               attrs[i].attid                  = DRSUAPI_ATTRIBUTE_msDS_Behavior_Version;
+               attrs[i].attid                  = DRSUAPI_ATTID_msDS_Behavior_Version;
                attrs[i].value_ctr.num_values   = 1;
                attrs[i].value_ctr.values       = vs;
 
@@ -2159,7 +2205,7 @@ static void becomeDC_drsuapi1_add_entry_send(struct libnet_BecomeDC_state *s)
 
                vs[0].blob              = &vd[0];
 
-               attrs[i].attid                  = DRSUAPI_ATTRIBUTE_systemFlags;
+               attrs[i].attid                  = DRSUAPI_ATTID_systemFlags;
                attrs[i].value_ctr.num_values   = 1;
                attrs[i].value_ctr.values       = vs;
 
@@ -2191,7 +2237,7 @@ static void becomeDC_drsuapi1_add_entry_send(struct libnet_BecomeDC_state *s)
 
                vs[0].blob              = &vd[0];
 
-               attrs[i].attid                  = DRSUAPI_ATTRIBUTE_serverReference;
+               attrs[i].attid                  = DRSUAPI_ATTID_serverReference;
                attrs[i].value_ctr.num_values   = 1;
                attrs[i].value_ctr.values       = vs;
 
@@ -2216,7 +2262,7 @@ static void becomeDC_drsuapi1_add_entry_send(struct libnet_BecomeDC_state *s)
 
                vs[0].blob              = &vd[0];
 
-               attrs[i].attid                  = DRSUAPI_ATTRIBUTE_options;
+               attrs[i].attid                  = DRSUAPI_ATTID_options;
                attrs[i].value_ctr.num_values   = 1;
                attrs[i].value_ctr.values       = vs;
 
@@ -2257,6 +2303,7 @@ static void becomeDC_drsuapi1_add_entry_recv(struct tevent_req *subreq)
        struct drsuapi_DsAddEntry *r = talloc_get_type_abort(s->ndr_struct_ptr,
                                       struct drsuapi_DsAddEntry);
        char *binding_str;
+       uint32_t assoc_group_id;
 
        s->ndr_struct_ptr = NULL;
 
@@ -2287,8 +2334,9 @@ static void becomeDC_drsuapi1_add_entry_recv(struct tevent_req *subreq)
                                return;
                        }
 
-                       DEBUG(0,("DsAddEntry (R3) failed: "
+                       DEBUG(0,("DsAddEntry (R3) of '%s' failed: "
                                 "Errors: dir_err = %d, status = %s;\n",
+                                r->in.req->req3.first_object.object.identifier->dn,
                                 err_data->v1.dir_err,
                                 win_errstr(err_data->v1.status)));
 
@@ -2388,6 +2436,7 @@ static void becomeDC_drsuapi1_add_entry_recv(struct tevent_req *subreq)
                                 "method succeeded but objects returned are %d (expected 1).\n",
                                 r->out.ctr->ctr3.count));
                        composite_error(c, NT_STATUS_INVALID_NETWORK_RESPONSE);
+                       return;
                }
 
                s->dest_dsa.ntds_guid   = r->out.ctr->ctr3.objects[0].guid;
@@ -2409,6 +2458,7 @@ static void becomeDC_drsuapi1_add_entry_recv(struct tevent_req *subreq)
                                 r->out.ctr->ctr2.dir_err,
                                 win_errstr(r->out.ctr->ctr2.extended_err)));
                        composite_error(c, NT_STATUS_INVALID_NETWORK_RESPONSE);
+                       return;
                }
 
                s->dest_dsa.ntds_guid   = r->out.ctr->ctr2.objects[0].guid;
@@ -2434,8 +2484,17 @@ static void becomeDC_drsuapi1_add_entry_recv(struct tevent_req *subreq)
        talloc_free(binding_str);
        if (!composite_is_ok(c)) return;
 
+       if (DEBUGLEVEL >= 10) {
+               c->status = dcerpc_binding_set_flags(s->drsuapi2.binding,
+                                                    DCERPC_DEBUG_PRINT_BOTH,
+                                                    0);
+               if (!composite_is_ok(c)) return;
+       }
+
        /* w2k3 uses the same assoc_group_id as on the first connection, so we do */
-       s->drsuapi2.binding->assoc_group_id     = s->drsuapi1.pipe->assoc_group_id;
+       assoc_group_id = dcerpc_binding_get_assoc_group_id(s->drsuapi1.pipe->binding);
+       c->status = dcerpc_binding_set_assoc_group_id(s->drsuapi2.binding, assoc_group_id);
+       if (!composite_is_ok(c)) return;
 
        becomeDC_drsuapi_connect_send(s, &s->drsuapi2, becomeDC_drsuapi2_connect_recv);
 }
@@ -2466,6 +2525,7 @@ static void becomeDC_drsuapi2_connect_recv(struct composite_context *req)
        s->drsuapi2.drsuapi_handle = s->drsuapi2.pipe->binding_handle;
 
        c->status = gensec_session_key(s->drsuapi2.pipe->conn->security_state.generic_state,
+                                      s,
                                       &s->drsuapi2.gensec_skey);
        if (!composite_is_ok(c)) return;
 
@@ -2480,6 +2540,7 @@ static void becomeDC_drsuapi2_bind_recv(struct tevent_req *subreq)
                                          struct libnet_BecomeDC_state);
        struct composite_context *c = s->creq;
        char *binding_str;
+       uint32_t assoc_group_id;
        WERROR status;
 
        c->status = dcerpc_drsuapi_DsBind_r_recv(subreq, s);
@@ -2500,10 +2561,22 @@ static void becomeDC_drsuapi2_bind_recv(struct tevent_req *subreq)
        talloc_free(binding_str);
        if (!composite_is_ok(c)) return;
 
+       if (DEBUGLEVEL >= 10) {
+               c->status = dcerpc_binding_set_flags(s->drsuapi3.binding,
+                                                    DCERPC_DEBUG_PRINT_BOTH,
+                                                    0);
+               if (!composite_is_ok(c)) return;
+       }
+
        /* w2k3 uses the same assoc_group_id as on the first connection, so we do */
-       s->drsuapi3.binding->assoc_group_id     = s->drsuapi1.pipe->assoc_group_id;
+       assoc_group_id = dcerpc_binding_get_assoc_group_id(s->drsuapi1.pipe->binding);
+       c->status = dcerpc_binding_set_assoc_group_id(s->drsuapi3.binding, assoc_group_id);
+       if (!composite_is_ok(c)) return;
        /* w2k3 uses the concurrent multiplex feature on the 3rd connection, so we do */
-       s->drsuapi3.binding->flags              |= DCERPC_CONCURRENT_MULTIPLEX;
+       c->status = dcerpc_binding_set_flags(s->drsuapi3.binding,
+                                            DCERPC_CONCURRENT_MULTIPLEX,
+                                            0);
+       if (!composite_is_ok(c)) return;
 
        becomeDC_drsuapi_connect_send(s, &s->drsuapi3, becomeDC_drsuapi3_connect_recv);
 }
@@ -2522,6 +2595,7 @@ static void becomeDC_drsuapi3_connect_recv(struct composite_context *req)
        s->drsuapi3.drsuapi_handle = s->drsuapi3.pipe->binding_handle;
 
        c->status = gensec_session_key(s->drsuapi3.pipe->conn->security_state.generic_state,
+                                      s,
                                       &s->drsuapi3.gensec_skey);
        if (!composite_is_ok(c)) return;
 
@@ -2599,6 +2673,10 @@ static WERROR becomeDC_drsuapi_pull_partition_recv(struct libnet_BecomeDC_state
                                                   struct libnet_BecomeDC_Partition *partition,
                                                   struct drsuapi_DsGetNCChanges *r)
 {
+       uint32_t req_level = 0;
+       struct drsuapi_DsGetNCChangesRequest5 *req5 = NULL;
+       struct drsuapi_DsGetNCChangesRequest8 *req8 = NULL;
+       struct drsuapi_DsGetNCChangesRequest10 *req10 = NULL;
        uint32_t ctr_level = 0;
        struct drsuapi_DsGetNCChangesCtr1 *ctr1 = NULL;
        struct drsuapi_DsGetNCChangesCtr6 *ctr6 = NULL;
@@ -2606,12 +2684,29 @@ static WERROR becomeDC_drsuapi_pull_partition_recv(struct libnet_BecomeDC_state
        struct GUID *source_dsa_invocation_id = NULL;
        struct drsuapi_DsReplicaHighWaterMark *new_highwatermark = NULL;
        bool more_data = false;
-       NTSTATUS nt_status;
+       WERROR werr;
 
        if (!W_ERROR_IS_OK(r->out.result)) {
                return r->out.result;
        }
 
+       switch (r->in.level) {
+       case 0:
+               /* none */
+               break;
+       case 5:
+               req5 = &r->in.req->req5;
+               break;
+       case 8:
+               req8 = &r->in.req->req8;
+               break;
+       case 10:
+               req10 = &r->in.req->req10;
+               break;
+       default:
+               return WERR_INVALID_PARAMETER;
+       }
+
        if (*r->out.level_out == 1) {
                ctr_level = 1;
                ctr1 = &r->out.ctr->ctr1;
@@ -2675,6 +2770,10 @@ static WERROR becomeDC_drsuapi_pull_partition_recv(struct libnet_BecomeDC_state
        s->_sc.source_dsa       = &s->source_dsa;
        s->_sc.dest_dsa         = &s->dest_dsa;
        s->_sc.partition        = partition;
+       s->_sc.req_level        = req_level;
+       s->_sc.req5             = req5;
+       s->_sc.req8             = req8;
+       s->_sc.req10            = req10;
        s->_sc.ctr_level        = ctr_level;
        s->_sc.ctr1             = ctr1;
        s->_sc.ctr6             = ctr6;
@@ -2684,9 +2783,9 @@ static WERROR becomeDC_drsuapi_pull_partition_recv(struct libnet_BecomeDC_state
         */
        s->_sc.gensec_skey      = &drsuapi_p->gensec_skey;
 
-       nt_status = partition->store_chunk(s->callbacks.private_data, &s->_sc);
-       if (!NT_STATUS_IS_OK(nt_status)) {
-               return ntstatus_to_werror(nt_status);
+       werr = partition->store_chunk(s->callbacks.private_data, &s->_sc);
+       if (!W_ERROR_IS_OK(werr)) {
+               return werr;
        }
 
        return WERR_OK;
@@ -2707,7 +2806,8 @@ static void becomeDC_drsuapi3_pull_schema_send(struct libnet_BecomeDC_state *s)
                                        | DRSUAPI_DRS_PER_SYNC
                                        | DRSUAPI_DRS_FULL_SYNC_IN_PROGRESS
                                        | DRSUAPI_DRS_NEVER_SYNCED
-                                       | DRSUAPI_DRS_USE_COMPRESSION;
+                                       | DRSUAPI_DRS_USE_COMPRESSION
+                                       | DRSUAPI_DRS_GET_ANC;
        if (s->rodc_join) {
            s->schema_part.replica_flags &= ~DRSUAPI_DRS_WRIT_REP;
        }
@@ -2767,7 +2867,8 @@ static void becomeDC_drsuapi3_pull_config_send(struct libnet_BecomeDC_state *s)
                                        | DRSUAPI_DRS_PER_SYNC
                                        | DRSUAPI_DRS_FULL_SYNC_IN_PROGRESS
                                        | DRSUAPI_DRS_NEVER_SYNCED
-                                       | DRSUAPI_DRS_USE_COMPRESSION;
+                                       | DRSUAPI_DRS_USE_COMPRESSION
+                                       | DRSUAPI_DRS_GET_ANC;
        if (s->rodc_join) {
            s->schema_part.replica_flags &= ~DRSUAPI_DRS_WRIT_REP;
        }
@@ -2825,7 +2926,11 @@ static void becomeDC_drsuapi3_pull_domain_send(struct libnet_BecomeDC_state *s)
                                        | DRSUAPI_DRS_PER_SYNC
                                        | DRSUAPI_DRS_FULL_SYNC_IN_PROGRESS
                                        | DRSUAPI_DRS_NEVER_SYNCED
-                                       | DRSUAPI_DRS_USE_COMPRESSION;
+                                       | DRSUAPI_DRS_USE_COMPRESSION
+                                       | DRSUAPI_DRS_GET_ANC;
+       if (s->critical_only) {
+               s->domain_part.replica_flags |= DRSUAPI_DRS_CRITICAL_ONLY;
+       }
        if (s->rodc_join) {
            s->schema_part.replica_flags &= ~DRSUAPI_DRS_WRIT_REP;
        }
@@ -2871,6 +2976,14 @@ static void becomeDC_drsuapi3_pull_domain_recv(struct tevent_req *subreq)
                return;
        }
 
+       if (s->critical_only) {
+               /* Remove the critical and ANC */
+               s->domain_part.replica_flags ^= DRSUAPI_DRS_CRITICAL_ONLY | DRSUAPI_DRS_GET_ANC;
+               s->critical_only = false;
+               becomeDC_drsuapi_pull_partition_send(s, &s->drsuapi2, &s->drsuapi3, &s->domain_part,
+                                                    becomeDC_drsuapi3_pull_domain_recv);
+               return;
+       }
        becomeDC_drsuapi_update_refs_send(s, &s->drsuapi2, &s->schema_part,
                                          becomeDC_drsuapi2_update_refs_schema_recv);
 }
@@ -2894,7 +3007,7 @@ static void becomeDC_drsuapi_update_refs_send(struct libnet_BecomeDC_state *s,
 
        ntds_dns_name = talloc_asprintf(r, "%s._msdcs.%s",
                                        ntds_guid_str,
-                                       s->domain.dns_name);
+                                       s->forest.dns_name);
        if (composite_nomem(ntds_dns_name, c)) return;
 
        r->in.bind_handle               = &drsuapi->bind_handle;
@@ -2905,7 +3018,7 @@ static void becomeDC_drsuapi_update_refs_send(struct libnet_BecomeDC_state *s,
        r->in.req.req1.options          = DRSUAPI_DRS_ADD_REF | DRSUAPI_DRS_DEL_REF;
 
        /* I think this is how we mark ourselves as a RODC */
-       if (!lp_parm_bool(s->libnet->lp_ctx, NULL, "repl", "RODC", false)) {
+       if (!lpcfg_parm_bool(s->libnet->lp_ctx, NULL, "repl", "RODC", false)) {
                r->in.req.req1.options |= DRSUAPI_DRS_WRIT_REP;
        }
 
@@ -3015,8 +3128,9 @@ static NTSTATUS becomeDC_ldap2_modify_computer(struct libnet_BecomeDC_state *s)
        msg->dn = ldb_dn_new(msg, s->ldap2.ldb, s->dest_dsa.computer_dn_str);
        NT_STATUS_HAVE_NO_MEMORY(msg->dn);
 
-       ret = ldb_msg_add_fmt(msg, "userAccountControl", "%u", user_account_control);
-       if (ret != 0) {
+       ret = samdb_msg_add_uint(s->ldap2.ldb, msg, msg, "userAccountControl",
+                                user_account_control);
+       if (ret != LDB_SUCCESS) {
                talloc_free(msg);
                return NT_STATUS_NO_MEMORY;
        }
@@ -3093,6 +3207,7 @@ static void becomeDC_connect_ldap2(struct libnet_BecomeDC_state *s)
        c->status = becomeDC_ldap2_move_computer(s);
        if (!composite_is_ok(c)) return;
 
+       s->critical_only = true;
        becomeDC_drsuapi3_pull_domain_send(s);
 }