Coverage for ipaserver/install/replication : 9%

# Authors: Karl MacMillan <kmacmillan@mentalrootkit.com> 

# 

# Copyright (C) 2007  Red Hat 

# see file 'COPYING' for use and warranty information 

# 

# This program is free software; you can redistribute it and/or modify 

# it under the terms of the GNU General Public License as published by 

# the Free Software Foundation, either version 3 of the License, or 

# (at your option) any later version. 

# 

# This program is distributed in the hope that it will be useful, 

# but WITHOUT ANY WARRANTY; without even the implied warranty of 

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 

# GNU General Public License for more details. 

# 

# You should have received a copy of the GNU General Public License 

# along with this program.  If not, see <http://www.gnu.org/licenses/>. 

# 

import time 

from ipapython.ipa_log_manager import * 

import os 

import sys 

import ldap 

from ipaserver import ipaldap 

from ipapython import services as ipaservices 

from ldap import modlist 

from ipalib import api, util, errors 

from ipapython import ipautil 

from ipapython.dn import DN 

CACERT = "/etc/ipa/ca.crt" 

# the default container used by AD for user entries 

WIN_USER_CONTAINER = DN(('cn', 'Users')) 

# the default container used by IPA for user entries 

IPA_USER_CONTAINER = DN(('cn', 'users'), ('cn', 'accounts')) 

PORT = 636 

TIMEOUT = 120 

REPL_MAN_DN = DN(('cn', 'replication manager'), ('cn', 'config')) 

IPA_REPLICA = 1 

WINSYNC = 2 

# List of attributes that need to be excluded from replication initialization. 

TOTAL_EXCLUDES = ('entryusn', 

                 'krblastsuccessfulauth', 

                 'krblastfailedauth', 

                 'krbloginfailedcount') 

# List of attributes that need to be excluded from normal replication. 

EXCLUDES = ('memberof', 'idnssoaserial') + TOTAL_EXCLUDES 

# List of attributes that are not updated on empty replication 

STRIP_ATTRS = ('modifiersName', 

               'modifyTimestamp', 

               'internalModifiersName', 

               'internalModifyTimestamp') 

def replica_conn_check(master_host, host_name, realm, check_ca, 

                       admin_password=None): 

    """ 

    Check the ports used by the replica both locally and remotely to be sure 

    that replication will work. 

    Does not return a value, will sys.exit() on failure. 

    """ 

    print "Run connection check to master" 

    args = ["/usr/sbin/ipa-replica-conncheck", "--master", master_host, 

            "--auto-master-check", "--realm", realm, 

            "--principal", "admin", 

            "--hostname", host_name] 

    nolog=tuple() 

    if admin_password: 

        args.extend(["--password", admin_password]) 

        nolog=(admin_password,) 

    if check_ca: 

        args.append('--check-ca') 

    (stdin, stderr, returncode) = ipautil.run(args,raiseonerr=False,capture_output=False, 

                                              nolog=nolog) 

    if returncode != 0: 

        sys.exit("Connection check failed!" + 

                 "\nPlease fix your network settings according to error messages above." + 

                 "\nIf the check results are not valid it can be skipped with --skip-conncheck parameter.") 

    else: 

        print "Connection check OK" 

def enable_replication_version_checking(hostname, realm, dirman_passwd): 

    """ 

    Check the replication version checking plugin. If it is not 

    enabled then enable it and restart 389-ds. If it is enabled 

    the do nothing. 

    """ 

    conn = ipaldap.IPAdmin(hostname, realm=realm, ldapi=True) 

    if dirman_passwd: 

        conn.do_simple_bind(bindpw=dirman_passwd) 

    else: 

        conn.do_sasl_gssapi_bind() 

    entry = conn.getEntry(DN(('cn', 'IPA Version Replication'), ('cn', 'plugins'), ('cn', 'config')), 

                          ldap.SCOPE_BASE, 'objectclass=*') 

    if entry.getValue('nsslapd-pluginenabled') == 'off': 

        conn.modify_s(entry.dn, [(ldap.MOD_REPLACE, 'nsslapd-pluginenabled', 'on')]) 

        conn.unbind() 

        serverid = "-".join(realm.split(".")) 

        ipaservices.knownservices.dirsrv.restart(instance_name=serverid) 

    else: 

        conn.unbind() 

class ReplicationManager(object): 

    """Manage replication agreements between DS servers, and sync 

    agreements with Windows servers""" 

    def __init__(self, realm, hostname, dirman_passwd, port=PORT, starttls=False, conn=None): 

        self.hostname = hostname 

        self.port = port 

        self.dirman_passwd = dirman_passwd 

        self.realm = realm 

        self.starttls = starttls 

        self.suffix = ipautil.realm_to_suffix(realm) 

        self.need_memberof_fixup = False 

        # The caller is allowed to pass in an existing IPAdmin connection. 

        # Open a new one if not provided 

        if conn is None: 

            # If we are passed a password we'll use it as the DM password 

            # otherwise we'll do a GSSAPI bind. 

            if starttls: 

                self.conn = ipaldap.IPAdmin(hostname, port=port) 

                ldap.set_option(ldap.OPT_X_TLS_CACERTFILE, CACERT) 

                self.conn.start_tls_s() 

            else: 

                self.conn = ipaldap.IPAdmin(hostname, port=port, cacert=CACERT) 

            if dirman_passwd: 

                self.conn.do_simple_bind(bindpw=dirman_passwd) 

            else: 

                self.conn.do_sasl_gssapi_bind() 

        else: 

            self.conn = conn 

        self.repl_man_passwd = dirman_passwd 

        # these are likely constant, but you could change them 

        # at runtime if you really want 

        self.repl_man_dn = REPL_MAN_DN 

        self.repl_man_cn = "replication manager" 

    def _get_replica_id(self, conn, master_conn): 

        """ 

        Returns the replica ID which is unique for each backend. 

        conn is the connection we are trying to get the replica ID for. 

        master_conn is the master we are going to replicate with. 

        """ 

        # First see if there is already one set 

        dn = self.replica_dn() 

        assert isinstance(dn, DN) 

        try: 

            replica = conn.getEntry(dn, ldap.SCOPE_BASE, "objectclass=*") 

        except errors.NotFound: 

            pass 

        else: 

            if replica.getValue('nsDS5ReplicaId'): 

                return int(replica.getValue('nsDS5ReplicaId')) 

        # Ok, either the entry doesn't exist or the attribute isn't set 

        # so get it from the other master 

        retval = -1 

        dn = DN(('cn','replication'),('cn','etc'), self.suffix) 

        try: 

            replica = master_conn.getEntry(dn, ldap.SCOPE_BASE, "objectclass=*") 

        except errors.NotFound: 

            root_logger.debug("Unable to retrieve nsDS5ReplicaId from remote server") 

            raise 

        else: 

            if replica.getValue('nsDS5ReplicaId') is None: 

                root_logger.debug("Unable to retrieve nsDS5ReplicaId from remote server") 

                raise RuntimeError("Unable to retrieve nsDS5ReplicaId from remote server") 

        # Now update the value on the master 

        retval = int(replica.getValue('nsDS5ReplicaId')) 

        mod = [(ldap.MOD_REPLACE, 'nsDS5ReplicaId', str(retval + 1))] 

        try: 

            master_conn.modify_s(dn, mod) 

        except Exception, e: 

            root_logger.debug("Problem updating nsDS5ReplicaID %s" % e) 

            raise 

        return retval 

    def find_replication_agreements(self): 

        """ 

        The replication agreements are stored in 

        cn="$SUFFIX",cn=mapping tree,cn=config 

        FIXME: Rather than failing with a read error if a user tries 

        to read this it simply returns zero entries. We need to use 

        GER to determine if we are allowed to read this to return a proper 

        response. For now just return "No entries" even if the user may 

        not be allowed to see them. 

        """ 

        filt = "(|(objectclass=nsDSWindowsReplicationAgreement)(objectclass=nsds5ReplicationAgreement))" 

        try: 

            ents = self.conn.getList(DN(('cn', 'mapping tree'), ('cn', 'config')), 

                                     ldap.SCOPE_SUBTREE, filt) 

        except errors.NotFound: 

            ents = [] 

        return ents 

    def find_ipa_replication_agreements(self): 

        """ 

        The replication agreements are stored in 

        cn="$SUFFIX",cn=mapping tree,cn=config 

        Return the list of hosts we have replication agreements. 

        """ 

        res = [] 

        filt = "(objectclass=nsds5ReplicationAgreement)" 

        try: 

            ents = self.conn.getList(DN(('cn', 'mapping tree'), ('cn', 'config')), 

                                     ldap.SCOPE_SUBTREE, filt) 

        except errors.NotFound: 

            return res 

        for ent in ents: 

            res.append(ent.getValue('nsds5replicahost')) 

        return res 

    def get_replication_agreement(self, hostname): 

        """ 

        The replication agreements are stored in 

        cn="$SUFFIX",cn=mapping tree,cn=config 

        Get the replication agreement for a specific host. 

        Returns None if not found. 

        """ 

        filt = "(&(|(objectclass=nsds5ReplicationAgreement)(objectclass=nsDSWindowsReplicationAgreement))(nsDS5ReplicaHost=%s))" % hostname 

        try: 

            entries = self.conn.getList(DN(('cn', 'mapping tree'), ('cn', 'config')), 

                                      ldap.SCOPE_SUBTREE, filt) 

        except errors.NotFound: 

            return None 

        if len(entries) == 0: 

            return None 

        else: 

            return entries[0] # There can be only one 

    def add_replication_manager(self, conn, dn, pw): 

        """ 

        Create a pseudo user to use for replication. 

        """ 

        assert isinstance(dn, DN) 

        rdn_attr = dn[0].attr 

        rdn_val = dn[0].value 

        ent = ipaldap.Entry(dn) 

        ent.setValues("objectclass", "top", "person") 

        ent.setValues(rdn_attr, rdn_val) 

        ent.setValues("userpassword", pw) 

        ent.setValues("sn", "replication manager pseudo user") 

        try: 

            conn.addEntry(ent) 

        except errors.DuplicateEntry: 

            conn.modify_s(dn, [(ldap.MOD_REPLACE, "userpassword", pw)]) 

            pass 

    def delete_replication_manager(self, conn, dn=REPL_MAN_DN): 

        assert isinstance(dn, DN) 

        try: 

            conn.delete_s(dn) 

        except ldap.NO_SUCH_OBJECT: 

            pass 

    def get_replica_type(self, master=True): 

        if master: 

            return "3" 

        else: 

            return "2" 

    def replica_dn(self): 

        return DN(('cn','replica'),('cn',self.suffix),('cn','mapping tree'),('cn','config')) 

    def replica_config(self, conn, replica_id, replica_binddn): 

        assert isinstance(replica_binddn, DN) 

        dn = self.replica_dn() 

        assert isinstance(dn, DN) 

        try: 

            entry = conn.getEntry(dn, ldap.SCOPE_BASE) 

            managers = entry.getValues('nsDS5ReplicaBindDN') 

            for m in managers: 

                if replica_binddn == DN(m): 

                    return 

            # Add the new replication manager 

            mod = [(ldap.MOD_ADD, 'nsDS5ReplicaBindDN', replica_binddn)] 

            conn.modify_s(dn, mod) 

            # replication is already configured 

            return 

        except errors.NotFound: 

            pass 

        replica_type = self.get_replica_type() 

        entry = ipaldap.Entry(dn) 

        entry.setValues('objectclass', "top", "nsds5replica", "extensibleobject") 

        entry.setValues('cn', "replica") 

        entry.setValues('nsds5replicaroot', str(self.suffix)) 

        entry.setValues('nsds5replicaid', str(replica_id)) 

        entry.setValues('nsds5replicatype', replica_type) 

        entry.setValues('nsds5flags', "1") 

        entry.setValues('nsds5replicabinddn', [replica_binddn]) 

        entry.setValues('nsds5replicalegacyconsumer', "off") 

        conn.addEntry(entry) 

    def setup_changelog(self, conn): 

        dn = DN(('cn', 'changelog5'), ('cn', 'config')) 

        dirpath = conn.dbdir + "/cldb" 

        entry = ipaldap.Entry(dn) 

        entry.setValues('objectclass', "top", "extensibleobject") 

        entry.setValues('cn', "changelog5") 

        entry.setValues('nsslapd-changelogdir', dirpath) 

        try: 

            conn.addEntry(entry) 

        except errors.DuplicateEntry: 

            return 

    def setup_chaining_backend(self, conn): 

        chaindn = DN(('cn', 'chaining database'), ('cn', 'plugins'), ('cn', 'config')) 

        benamebase = "chaindb" 

        urls = [self.to_ldap_url(conn)] 

        cn = "" 

        benum = 1 

        done = False 

        while not done: 

            try: 

                cn = benamebase + str(benum) # e.g. localdb1 

                dn = DN(('cn', cn), chaindn) 

                entry = ipaldap.Entry(dn) 

                entry.setValues('objectclass', 'top', 'extensibleObject', 'nsBackendInstance') 

                entry.setValues('cn', cn) 

                entry.setValues('nsslapd-suffix', str(self.suffix)) 

                entry.setValues('nsfarmserverurl', urls) 

                entry.setValues('nsmultiplexorbinddn', self.repl_man_dn) 

                entry.setValues('nsmultiplexorcredentials', self.repl_man_passwd) 

                self.conn.addEntry(entry) 

                done = True 

            except errors.DuplicateEntry: 

                benum += 1 

            except errors.ExecutionError, e: 

                print "Could not add backend entry " + dn, e 

                raise 

        return cn 

    def to_ldap_url(self, conn): 

        return "ldap://%s/" % ipautil.format_netloc(conn.host, conn.port) 

    def setup_chaining_farm(self, conn): 

        try: 

            conn.modify_s(self.suffix, [(ldap.MOD_ADD, 'aci', 

                                    [ "(targetattr = \"*\")(version 3.0; acl \"Proxied authorization for database links\"; allow (proxy) userdn = \"ldap:///%s\";)" % self.repl_man_dn ])]) 

        except ldap.TYPE_OR_VALUE_EXISTS: 

            root_logger.debug("proxy aci already exists in suffix %s on %s" % (self.suffix, conn.host)) 

    def get_mapping_tree_entry(self): 

        try: 

            entry = self.conn.getEntry(DN(('cn', 'mapping tree'), ('cn', 'config')), ldap.SCOPE_ONELEVEL, 

                                       "(cn=\"%s\")" % (self.suffix)) 

        except errors.NotFound, e: 

            root_logger.debug("failed to find mappting tree entry for %s" % self.suffix) 

            raise e 

        return entry 

    def enable_chain_on_update(self, bename): 

        mtent = self.get_mapping_tree_entry() 

        dn = mtent.dn 

        plgent = self.conn.getEntry(DN(('cn', 'Multimaster Replication Plugin'), ('cn', 'plugins'), ('cn', 'config')), 

                                    ldap.SCOPE_BASE, "(objectclass=*)", ['nsslapd-pluginPath']) 

        path = plgent.getValue('nsslapd-pluginPath') 

        mod = [(ldap.MOD_REPLACE, 'nsslapd-state', 'backend'), 

               (ldap.MOD_ADD, 'nsslapd-backend', bename), 

               (ldap.MOD_ADD, 'nsslapd-distribution-plugin', path), 

               (ldap.MOD_ADD, 'nsslapd-distribution-funct', 'repl_chain_on_update')] 

        try: 

            self.conn.modify_s(dn, mod) 

        except ldap.TYPE_OR_VALUE_EXISTS: 

            root_logger.debug("chainOnUpdate already enabled for %s" % self.suffix) 

    def setup_chain_on_update(self, other_conn): 

        chainbe = self.setup_chaining_backend(other_conn) 

        self.enable_chain_on_update(chainbe) 

    def add_passsync_user(self, conn, password): 

        pass_dn = DN(('uid', 'passsync'), ('cn', 'sysaccounts'), ('cn', 'etc'), self.suffix) 

        print "The user for the Windows PassSync service is %s" % pass_dn 

        try: 

            conn.getEntry(pass_dn, ldap.SCOPE_BASE) 

            print "Windows PassSync entry exists, not resetting password" 

            return 

        except errors.NotFound: 

            pass 

        # The user doesn't exist, add it 

        entry = ipaldap.Entry(pass_dn) 

        entry.setValues("objectclass", ["account", "simplesecurityobject"]) 

        entry.setValues("uid", "passsync") 

        entry.setValues("userPassword", password) 

        conn.addEntry(entry) 

        # Add it to the list of users allowed to bypass password policy 

        extop_dn = DN(('cn', 'ipa_pwd_extop'), ('cn', 'plugins'), ('cn', 'config')) 

        entry = conn.getEntry(extop_dn, ldap.SCOPE_BASE) 

        pass_mgrs = entry.getValues('passSyncManagersDNs') 

        if not pass_mgrs: 

            pass_mgrs = [] 

        if not isinstance(pass_mgrs, list): 

            pass_mgrs = [pass_mgrs] 

        pass_mgrs.append(pass_dn) 

        mod = [(ldap.MOD_REPLACE, 'passSyncManagersDNs', pass_mgrs)] 

        conn.modify_s(extop_dn, mod) 

        # And finally grant it permission to write passwords 

        mod = [(ldap.MOD_ADD, 'aci', 

            ['(targetattr = "userPassword || krbPrincipalKey || sambaLMPassword || sambaNTPassword || passwordHistory")(version 3.0; acl "Windows PassSync service can write passwords"; allow (write) userdn="ldap:///%s";)' % pass_dn])] 

        try: 

            conn.modify_s(self.suffix, mod) 

        except ldap.TYPE_OR_VALUE_EXISTS: 

            root_logger.debug("passsync aci already exists in suffix %s on %s" % (self.suffix, conn.host)) 

    def setup_winsync_agmt(self, entry, win_subtree=None): 

        if win_subtree is None: 

            win_subtree = DN(WIN_USER_CONTAINER, self.ad_suffix) 

        ds_subtree = DN(IPA_USER_CONTAINER, self.suffix) 

        windomain = ipautil.suffix_to_realm(self.suffix) 

        entry.setValues("objectclass", "nsDSWindowsReplicationAgreement") 

        entry.setValues("nsds7WindowsReplicaSubtree", win_subtree) 

        entry.setValues("nsds7DirectoryReplicaSubtree", ds_subtree) 

        # for now, just sync users and ignore groups 

        entry.setValues("nsds7NewWinUserSyncEnabled", 'true') 

        entry.setValues("nsds7NewWinGroupSyncEnabled", 'false') 

        entry.setValues("nsds7WindowsDomain", windomain) 

    def agreement_dn(self, hostname, master=None): 

        """ 

        IPA agreement use the same dn on both sides, dogtag does not. 

        master is not used for IPA agreements but for dogtag it will 

        tell which side we want. 

        """ 

        cn = "meTo%s" % (hostname) 

        dn = DN(('cn', cn), self.replica_dn()) 

        return (cn, dn) 

    def setup_agreement(self, a_conn, b_hostname, port=389, 

                        repl_man_dn=None, repl_man_passwd=None, 

                        iswinsync=False, win_subtree=None, isgssapi=False, 

                        master=None): 

        """ 

        master is used to determine which side of the agreement we are 

        creating. This is only needed for dogtag replication agreements 

        which use a different name on each side. If master is None then 

        isn't a dogtag replication agreement. 

        """ 

        if repl_man_dn is not None: 

            assert isinstance(repl_man_dn, DN) 

        cn, dn = self.agreement_dn(b_hostname, master=master) 

        try: 

            a_conn.getEntry(dn, ldap.SCOPE_BASE) 

            return 

        except errors.NotFound: 

            pass 

        entry = ipaldap.Entry(dn) 

        entry.setValues('objectclass', "nsds5replicationagreement") 

        entry.setValues('cn', cn) 

        entry.setValues('nsds5replicahost', b_hostname) 

        entry.setValues('nsds5replicaport', str(port)) 

        entry.setValues('nsds5replicatimeout', str(TIMEOUT)) 

        entry.setValues('nsds5replicaroot', str(self.suffix)) 

        if master is None: 

            entry.setValues('nsDS5ReplicatedAttributeList', 

                            '(objectclass=*) $ EXCLUDE %s' % " ".join(EXCLUDES)) 

        entry.setValues('description', "me to %s" % b_hostname) 

        if isgssapi: 

            entry.setValues('nsds5replicatransportinfo', 'LDAP') 

            entry.setValues('nsds5replicabindmethod', 'SASL/GSSAPI') 

        else: 

            entry.setValues('nsds5replicabinddn', repl_man_dn) 

            entry.setValues('nsds5replicacredentials', repl_man_passwd) 

            entry.setValues('nsds5replicatransportinfo', 'TLS') 

            entry.setValues('nsds5replicabindmethod', 'simple') 

        if iswinsync: 

            self.setup_winsync_agmt(entry, win_subtree) 

        a_conn.addEntry(entry) 

        try: 

            mod = [(ldap.MOD_ADD, 'nsDS5ReplicatedAttributeListTotal', 

                   '(objectclass=*) $ EXCLUDE %s' % " ".join(TOTAL_EXCLUDES))] 

            a_conn.modify_s(dn, mod) 

        except ldap.LDAPError, e: 

            # Apparently there are problems set the total list 

            # Probably the master is an old 389-ds server, tell the caller 

            # that we will have to set the memberof fixup task 

            self.need_memberof_fixup = True 

        entry.setValues('nsds5ReplicaStripAttrs', " ".join(STRIP_ATTRS)) 

        entry = a_conn.waitForEntry(entry) 

    def needs_memberof_fixup(self): 

        return self.need_memberof_fixup 

    def get_replica_principal_dns(self, a, b, retries): 

        """ 

        Get the DNs of the ldap principals we are going to convert 

        to using GSSAPI replication. 

        Arguments a and b are LDAP connections. retries is the number 

        of attempts that should be made to find the entries. It could 

        be that replication is slow. 

        If successful this returns a tuple (dn_a, dn_b). 

        If either of the DNs doesn't exist after the retries are 

        exhausted an exception is raised. 

        """ 

        filter_a = '(krbprincipalname=ldap/%s@%s)' % (a.host, self.realm) 

        filter_b = '(krbprincipalname=ldap/%s@%s)' % (b.host, self.realm) 

        a_entry = None 

        b_entry = None 

        error_message = '' 

        while (retries > 0 ): 

            root_logger.info('Getting ldap service principals for conversion: %s and %s' % (filter_a, filter_b)) 

            try: 

                a_entry = b.search_s(self.suffix, ldap.SCOPE_SUBTREE, filterstr=filter_a) 

            except errors.NotFound: 

                pass 

            try: 

                b_entry = a.search_s(self.suffix, ldap.SCOPE_SUBTREE, filterstr=filter_b) 

            except errors.NotFound: 

                pass 

            if a_entry and b_entry: 

                root_logger.debug('Found both principals.') 

                break 

            # One or both is missing, force sync again 

            if not a_entry: 

                root_logger.debug('Unable to find entry for %s on %s' 

                    % (filter_a, str(b))) 

                self.force_sync(a, b.host) 

                cn, dn = self.agreement_dn(b.host) 

                haserror, error_message = self.wait_for_repl_update(a, dn, 60) 

            if not b_entry: 

                root_logger.debug('Unable to find entry for %s on %s' 

                    % (filter_b, str(a))) 

                self.force_sync(b, a.host) 

                cn, dn = self.agreement_dn(a.host) 

                haserror, error_message = self.wait_for_repl_update(b, dn, 60) 

            retries -= 1 

        if not a_entry or not b_entry: 

            error = 'One of the ldap service principals is missing. ' \ 

                    'Replication agreement cannot be converted.' 

            if error_message: 

                error += '\nReplication error message: %s' % error_message 

            raise RuntimeError(error) 

        return (a_entry[0].dn, b_entry[0].dn) 

    def setup_krb_princs_as_replica_binddns(self, a, b): 

        """ 

        Search the appropriate principal names so we can get 

        the correct DNs to store in the replication agreements. 

        Then modify the replica object to allow these DNs to act 

        as replication agents. 

        """ 

        rep_dn = self.replica_dn() 

        assert isinstance(rep_dn, DN) 

        (a_dn, b_dn) = self.get_replica_principal_dns(a, b, retries=100) 

        assert isinstance(a_dn, DN) 

        assert isinstance(b_dn, DN) 

        # Add kerberos principal DNs as valid bindDNs for replication 

        try: 

            mod = [(ldap.MOD_ADD, "nsds5replicabinddn", b_dn)] 

            a.modify_s(rep_dn, mod) 

        except ldap.TYPE_OR_VALUE_EXISTS: 

            pass 

        try: 

            mod = [(ldap.MOD_ADD, "nsds5replicabinddn", a_dn)] 

            b.modify_s(rep_dn, mod) 

        except ldap.TYPE_OR_VALUE_EXISTS: 

            pass 

    def gssapi_update_agreements(self, a, b): 

        self.setup_krb_princs_as_replica_binddns(a, b) 

        #change replication agreements to connect to other host using GSSAPI 

        mod = [(ldap.MOD_REPLACE, "nsds5replicatransportinfo", "LDAP"), 

               (ldap.MOD_REPLACE, "nsds5replicabindmethod", "SASL/GSSAPI"), 

               (ldap.MOD_DELETE, "nsds5replicabinddn", None), 

               (ldap.MOD_DELETE, "nsds5replicacredentials", None)] 

        cn, a_ag_dn = self.agreement_dn(b.host) 

        a.modify_s(a_ag_dn, mod) 

        cn, b_ag_dn = self.agreement_dn(a.host) 

        b.modify_s(b_ag_dn, mod) 

        # Finally remove the temporary replication manager user 

        try: 

            a.delete_s(self.repl_man_dn) 

        except ldap.NO_SUCH_OBJECT: 

            pass 

        try: 

            b.delete_s(self.repl_man_dn) 

        except ldap.NO_SUCH_OBJECT: 

            pass 

    def delete_agreement(self, hostname, dn=None): 

        """ 

        Delete a replication agreement. 

        @hostname: the hostname of the agreement to remove 

        @dn: optional dn of the agreement to remove 

        For IPA agreements we can easily calculate the DN of the agreement 

        to remove. Dogtag agreements are another matter, its agreement 

        names depend entirely on where it is created. In this case it is 

        better to pass the DN in directly. 

        """ 

        if dn is None: 

            cn, dn = self.agreement_dn(hostname) 

        return self.conn.deleteEntry(dn) 

    def delete_referral(self, hostname): 

        dn = DN(('cn', self.suffix), ('cn', 'mapping tree'), ('cn', 'config')) 

        # TODO: should we detect proto/port somehow ? 

        mod = [(ldap.MOD_DELETE, 'nsslapd-referral', 

                'ldap://%s/%s' % (ipautil.format_netloc(hostname, 389), self.suffix))] 

        try: 

            self.conn.modify_s(dn, mod) 

        except Exception, e: 

            root_logger.debug("Failed to remove referral value: %s" % str(e)) 

    def check_repl_init(self, conn, agmtdn): 

        done = False 

        hasError = 0 

        attrlist = ['cn', 'nsds5BeginReplicaRefresh', 

                    'nsds5replicaUpdateInProgress', 

                    'nsds5ReplicaLastInitStatus', 

                    'nsds5ReplicaLastInitStart', 

                    'nsds5ReplicaLastInitEnd'] 

        entry = conn.getEntry(agmtdn, ldap.SCOPE_BASE, "(objectclass=*)", attrlist) 

        if not entry: 

            print "Error reading status from agreement", agmtdn 

            hasError = 1 

        else: 

            refresh = entry.getValue('nsds5BeginReplicaRefresh') 

            inprogress = entry.getValue('nsds5replicaUpdateInProgress') 

            status = entry.getValue('nsds5ReplicaLastInitStatus') 

            if not refresh: # done - check status 

                if not status: 

                    print "No status yet" 

                elif status.find("replica busy") > -1: 

                    print "[%s] reports: Replica Busy! Status: [%s]" % (conn.host, status) 

                    done = True 

                    hasError = 2 

                elif status.find("Total update succeeded") > -1: 

                    print "Update succeeded" 

                    done = True 

                elif inprogress.lower() == 'true': 

                    print "Update in progress yet not in progress" 

                else: 

                    print "[%s] reports: Update failed! Status: [%s]" % (conn.host, status) 

                    hasError = 1 

                    done = True 

            else: 

                print "Update in progress" 

        return done, hasError 

    def check_repl_update(self, conn, agmtdn): 

        done = False 

        hasError = 0 

        error_message = '' 

        attrlist = ['cn', 'nsds5replicaUpdateInProgress', 

                    'nsds5ReplicaLastUpdateStatus', 'nsds5ReplicaLastUpdateStart', 

                    'nsds5ReplicaLastUpdateEnd'] 

        entry = conn.getEntry(agmtdn, ldap.SCOPE_BASE, "(objectclass=*)", attrlist) 

        if not entry: 

            print "Error reading status from agreement", agmtdn 

            hasError = 1 

        else: 

            inprogress = entry.getValue('nsds5replicaUpdateInProgress') 

            status = entry.getValue('nsds5ReplicaLastUpdateStatus') 

            try: 

                start = int(entry.getValue('nsds5ReplicaLastUpdateStart')) 

            except (ValueError, TypeError): 

                start = 0 

            try: 

                end = int(entry.getValue('nsds5ReplicaLastUpdateEnd')) 

            except (ValueError, TypeError): 

                end = 0 

            # incremental update is done if inprogress is false and end >= start 

            done = inprogress and inprogress.lower() == 'false' and start <= end 

            root_logger.info("Replication Update in progress: %s: status: %s: start: %d: end: %d" % 

                         (inprogress, status, start, end)) 

            if status: # always check for errors 

                # status will usually be a number followed by a string 

                # number != 0 means error 

                rc, msg = status.split(' ', 1) 

                if rc != '0': 

                    hasError = 1 

                    error_message = msg 

                    done = True 

        return done, hasError, error_message 

    def wait_for_repl_init(self, conn, agmtdn): 

        done = False 

        haserror = 0 

        while not done and not haserror: 

            time.sleep(1)  # give it a few seconds to get going 

            done, haserror = self.check_repl_init(conn, agmtdn) 

        return haserror 

    def wait_for_repl_update(self, conn, agmtdn, maxtries=600): 

        done = False 

        haserror = 0 

        error_message = '' 

        while not done and not haserror and maxtries > 0: 

            time.sleep(1)  # give it a few seconds to get going 

            done, haserror, error_message = self.check_repl_update(conn, agmtdn) 

            maxtries -= 1 

        if maxtries == 0: # too many tries 

            print "Error: timeout: could not determine agreement status: please check your directory server logs for possible errors" 

            haserror = 1 

        return haserror, error_message 

    def start_replication(self, conn, hostname=None, master=None): 

        print "Starting replication, please wait until this has completed." 

        if hostname == None: 

            hostname = self.conn.host 

        cn, dn = self.agreement_dn(hostname, master) 

        mod = [(ldap.MOD_ADD, 'nsds5BeginReplicaRefresh', 'start')] 

        conn.modify_s(dn, mod) 

        return self.wait_for_repl_init(conn, dn) 

    def basic_replication_setup(self, conn, replica_id, repldn, replpw): 

        assert isinstance(repldn, DN) 

        if replpw is not None: 

            self.add_replication_manager(conn, repldn, replpw) 

        self.replica_config(conn, replica_id, repldn) 

        self.setup_changelog(conn) 

    def setup_replication(self, r_hostname, r_port=389, r_sslport=636,