/*
 * Copyright (c) 2000, 2017 Oracle and/or its affiliates. All rights reserved.
 * Copyright (c) 2020, 2021 Contributors to the Eclipse Foundation
 * Copyright (c) 2021 Payara Services Ltd.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

/*
 * @(#)ClusterImpl.java	1.106 07/08/07
 */

package com.sun.messaging.jmq.jmsserver.multibroker.fullyconnected;

import java.util.*;
import java.io.*;
import java.net.*;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.concurrent.Executors;
import java.util.concurrent.ExecutorService;

import com.sun.messaging.jmq.io.GPacket;
import com.sun.messaging.jmq.io.Status;
import com.sun.messaging.jmq.jmsserver.core.BrokerAddress;
import com.sun.messaging.jmq.jmsserver.core.BrokerMQAddress;
import com.sun.messaging.jmq.jmsserver.multibroker.BrokerInfo;
import com.sun.messaging.jmq.jmsserver.multibroker.Cluster;
import com.sun.messaging.jmq.jmsserver.multibroker.ClusterCallback;
import com.sun.messaging.jmq.jmsserver.multibroker.Protocol;
import com.sun.messaging.jmq.jmsserver.Globals;
import com.sun.messaging.jmq.jmsserver.Broker;
import com.sun.messaging.jmq.jmsservice.BrokerEvent;
import com.sun.messaging.jmq.jmsserver.service.PortMapper;
import com.sun.messaging.jmq.jmsserver.resources.BrokerResources;
import com.sun.messaging.jmq.jmsserver.util.BrokerException;
import com.sun.messaging.jmq.jmsserver.util.LoopbackAddressException;
import com.sun.messaging.jmq.util.net.MQServerSocketFactory;
import com.sun.messaging.jmq.util.log.Logger;
import com.sun.messaging.jmq.util.UID;
import com.sun.messaging.jmq.util.io.FilteringObjectInputStream;
import com.sun.messaging.jmq.jmsserver.persist.api.LoadException;
import com.sun.messaging.jmq.jmsserver.persist.api.Store;
import com.sun.messaging.jmq.jmsserver.persist.api.ChangeRecordInfo;
import com.sun.messaging.jmq.jmsserver.multibroker.ClusterGlobals;
import com.sun.messaging.jmq.jmsserver.multibroker.ClusterBrokerInfoReply;
import com.sun.messaging.jmq.jmsserver.multibroker.raptor.ClusterFirstInfoInfo;
import com.sun.messaging.jmq.jmsserver.multibroker.raptor.ClusterTransferFileListInfo;
import com.sun.messaging.jmq.jmsserver.multibroker.raptor.ClusterTransferFileStartInfo;
import com.sun.messaging.jmq.jmsserver.multibroker.raptor.ClusterTransferFileEndInfo;
import com.sun.messaging.jmq.jmsserver.cluster.api.*;
import com.sun.messaging.jmq.jmsserver.cluster.api.ha.*;
import com.sun.messaging.jmq.io.MQAddress;

import com.sun.messaging.jmq.jmsserver.multibroker.raptor.ProtocolGlobals;
import javax.net.ServerSocketFactory;

/**
 * This class implements the default fully connected topology. Version 2.0 : Brokers are connected using single TCP
 * connection.
 */
public class ClusterImpl implements Cluster, ClusterListener {

    static boolean DEBUG = false;

    ClusterCallback cb = null;
    private Properties matchProps = null;
    protected boolean useGPackets = false;

    private BrokerAddressImpl self;
    private Map connectList;
    private HashMap brokerList;
    private boolean readyForBroadcast = false;
    private int flowControlState = Packet.RESUME_FLOW;

    private String transport = null; // "tcp" or "ssl"
    private InetAddress listenHost = null;
    private int listenPort = 0;
    private ClusterServiceListener listener = null;

    private PingTimerTask pingTimer = null;

    private int tcpInbufsz = Globals.getConfig().getIntProperty(Globals.IMQ + ".cluster.tcp.inbufsz", 2048);
    private int sslInbufsz = Globals.getConfig().getIntProperty(Globals.IMQ + ".cluster.ssl.inbufsz",
            Globals.getConfig().getIntProperty(Globals.IMQ + ".cluster.tls.inbufsz", 2048));

    private int tcpOutbufsz = Globals.getConfig().getIntProperty(Globals.IMQ + ".cluster.tcp.outbufsz", 2048);
    private int sslOutbufsz = Globals.getConfig().getIntProperty(Globals.IMQ + ".cluster.ssl.outbufsz",
            Globals.getConfig().getIntProperty(Globals.IMQ + ".cluster.tls.outbufsz", 2048));

    private boolean tcpNodelay = Globals.getConfig().getBooleanProperty(Globals.IMQ + ".cluster.tcp.nodelay", true);
    private boolean sslNodelay = Globals.getConfig().getBooleanProperty(Globals.IMQ + ".cluster.ssl.nodelay",
            Globals.getConfig().getBooleanProperty(Globals.IMQ + ".cluster.tls.nodelay", true));

    private Object configServerLock = new Object();
    private BrokerAddressImpl configServer = null;
    private boolean configServerResolved = false;

    public static final String SERVICE_NAME = "cluster";
    public static final String SERVICE_TYPE = "CLUSTER";

    private static final Logger logger = Globals.getLogger();
    private static final BrokerResources br = Globals.getBrokerResources();

    private TimerTask warningTask = null;
    private static final int DEFAULT_SHUTDOWN_TIMEOUT = 30; // seconds
    private static final String SHUTDOWN_TIMEOUT_PROP = "imq.cluster.shutdownTimeout";

    private ClusterManager clsmgr = null;

    protected Map<BrokerAddress, String> pendingFileTransfers = Collections.synchronizedMap(new LinkedHashMap<BrokerAddress, String>());

    private Map<String, ExecutorService> fileTransferExecutors = Collections.synchronizedMap(new LinkedHashMap<String, ExecutorService>());
    protected boolean fileTransferShutdownIn = false;
    private boolean fileTransferShutdownOut = false;

    private final int FILE_TRANSFER_SOTIMEOUT_IN = 15000; // 15sec
    private final int FILE_TRANSFER_SOTIMEOUT_OUT = 5 * 60 * 1000; // 5min
    protected static final int FILE_TRANSFER_CHUNK_SIZE = 8192;

    private int pingInterval = ClusterManager.CLUSTER_PING_INTERVAL_DEFAULT;

    public ClusterImpl() throws BrokerException {
        clsmgr = Globals.getClusterManager();

        transport = clsmgr.getTransport();
        assert (transport != null);
        if (transport.equalsIgnoreCase("tls")) {
            transport = "ssl";
        }

        boolean nocluster = (!Globals.getHAEnabled() && clsmgr.getConfigBrokerCount() == 0);

        listenPort = clsmgr.getClusterPort();
        String listenh = clsmgr.getClusterHost(); // can be null
        try {
            listenHost = BrokerMQAddress.resolveBindAddress(listenh, true);
        } catch (LoopbackAddressException e) {
            int level = nocluster ? Logger.WARNING : Logger.ERROR;
            logger.log(level, br.E_CLUSTER_HOSTNAME, ClusterManager.HOST_PROPERTY);
            if (nocluster) {
                throw e;
            }
            Broker.getBroker().exit(1, br.getKString(br.E_CLUSTER_HOSTNAME, ClusterManager.HOST_PROPERTY), BrokerEvent.Type.FATAL_ERROR);
            throw e;
        } catch (Exception e) {
            if (e instanceof BrokerException) {
                logger.log(Logger.ERROR, br.getKString(br.E_BADADDRESS_CLUSTER_SERVICE, ClusterManager.HOST_PROPERTY, e.getMessage()));
            } else {
                logger.logStack(Logger.ERROR, br.getKString(br.E_BADADDRESS_CLUSTER_SERVICE, ClusterManager.HOST_PROPERTY, e.getMessage()), e);
            }
            Broker.getBroker().exit(1, br.getString(br.E_BADADDRESS_CLUSTER_SERVICE, ClusterManager.HOST_PROPERTY, e.getMessage()),
                    BrokerEvent.Type.FATAL_ERROR);

            if (e instanceof BrokerException) {
                throw (BrokerException) e;
            }
            BrokerException be = new BrokerException(e.getMessage());
            be.initCause(e);
            throw be;
        }

        try {
            BrokerMQAddress.checkLoopbackAddress(Globals.getPortMapper().getBindAddress(), Globals.getPortMapper().getHostname());
        } catch (LoopbackAddressException e) {
            int level = nocluster ? Logger.WARNING : Logger.ERROR;
            logger.log(level, br.E_BADADDRESS_PORTMAPPER_FOR_CLUSTER, PortMapper.HOSTNAME_PROPERTY, e.getMessage());
            if (nocluster) {
                throw e;
            }
            Broker.getBroker().exit(1, br.getString(br.E_BADADDRESS_PORTMAPPER_FOR_CLUSTER, PortMapper.HOSTNAME_PROPERTY, e.getMessage()),
                    BrokerEvent.Type.FATAL_ERROR);
            throw e;
        } catch (Exception e) {
            if (e instanceof BrokerException) {
                logger.log(logger.ERROR, br.E_BADADDRESS_PORTMAPPER_FOR_CLUSTER, PortMapper.HOSTNAME_PROPERTY, e.getMessage());
            } else {
                logger.logStack(logger.ERROR, br.E_BADADDRESS_PORTMAPPER_FOR_CLUSTER, PortMapper.HOSTNAME_PROPERTY, e.getMessage(), e);
            }
            Broker.getBroker().exit(1, br.getString(br.E_BADADDRESS_PORTMAPPER_FOR_CLUSTER, PortMapper.HOSTNAME_PROPERTY, e.getMessage()),
                    BrokerEvent.Type.FATAL_ERROR);

            if (e instanceof BrokerException) {
                throw (BrokerException) e;
            }
            BrokerException be = new BrokerException(e.getMessage());
            be.initCause(e);
            throw be;
        }

        try {
            self = new BrokerAddressImpl();
        } catch (LoopbackAddressException e) {
            int level = nocluster ? Logger.WARNING : Logger.ERROR;
            logger.log(level, br.getKString(br.E_BADADDRESS_THIS_BROKER, e.getMessage()));

            if (nocluster) {
                throw e;
            }
            Broker.getBroker().exit(1, br.getString(br.E_BADADDRESS_THIS_BROKER, e.getMessage()), BrokerEvent.Type.FATAL_ERROR);
            throw e;
        } catch (Exception e) {
            if (e instanceof BrokerException) {
                logger.log(Logger.ERROR, br.getKString(br.E_BADADDRESS_THIS_BROKER, e.getMessage()));
            } else {
                logger.logStack(Logger.ERROR, br.getKString(br.E_BADADDRESS_THIS_BROKER, e.getMessage()), e);
            }
            Broker.getBroker().exit(1, br.getString(br.E_BADADDRESS_THIS_BROKER, e.getMessage()), BrokerEvent.Type.FATAL_ERROR);

            if (e instanceof BrokerException) {
                throw (BrokerException) e;
            }
            BrokerException be = new BrokerException(e.getMessage());
            be.initCause(e);
            throw be;
        }

        readyForBroadcast = false;

        ClusteredBroker masterb = clsmgr.getMasterBroker();
        initConfigServer(masterb);

        initBrokerList();
    }

    private void initConfigServer(ClusteredBroker masterb) throws BrokerException {
        if (masterb == null) {
            configServer = null;
            configServerResolved = true;
            return;
        }

        if (Globals.getHAEnabled()) { // sync log message /w clsmgr
            String emsg = br.getKString(br.E_CLUSTER_HA_NOT_SUPPORT_MASTERBROKER);
            logger.log(Logger.ERROR, emsg);
            Broker.getBroker().exit(1, emsg, BrokerEvent.Type.FATAL_ERROR);
            throw new BrokerException(emsg);
        }

        checkStoredLastConfigServer();

        try {
            configServer = new BrokerAddressImpl((BrokerMQAddress) masterb.getBrokerURL(), null, Globals.getHAEnabled(), masterb.getBrokerName());
        } catch (Exception e) {
            configServer = null;
            configServerResolved = false;
            throw new BrokerException(e.getMessage(), e);
        }

        BrokerMQAddress key = configServer.getMQAddress();
        if (key.equals(self.getMQAddress())) {
            configServer = self;
            configServerResolved = true;
            return;
        }

        configServerResolved = false;
    }

    /**
     * Check to see if stored last config server property loaded properly
     *
     * System.exit if the last config server info is corrupted the last refresh timestamp info is corrupted and unable to
     * reset potentially last config server info is corrupted: store LoadPropertyException occurred with key corruption and
     * LastConfigServer property does not in store
     *
     * In future release, the System.exit behavior may be changed to allow administratively repair - eg. through imqbrokerd
     * option to force set last config server info to the current master broker
     *
     */
    private void checkStoredLastConfigServer() throws BrokerException {
        Store s = Globals.getStore();

        boolean bad = false;
        boolean potentiallyBad = false;
        LoadException le = s.getLoadPropertyException();
        LoadException savele = null;
        while (le != null) {
            Object o = le.getKey();
            if (o == null || !(o instanceof String)) {
                potentiallyBad = true;
                savele = le;
                le = le.getNextException();
                continue;
            }
            if (((String) o).equals(ClusterGlobals.STORE_PROPERTY_LASTCONFIGSERVER)) {
                logger.log(Logger.ERROR, BrokerResources.E_CLUSTER_LOAD_LASTCONFIGSERVER, le);
                bad = true;
                break;
            }
            if (((String) o).equals(ClusterGlobals.STORE_PROPERTY_LASTREFRESHTIME)) {
                logger.log(Logger.WARNING, BrokerResources.W_CLUSTER_LOAD_LASTREFRESHTIME, le);
                try {
                    s.updateProperty(ClusterGlobals.STORE_PROPERTY_LASTREFRESHTIME, Long.valueOf(-1), false);
                } catch (BrokerException e) {
                    logger.log(Logger.ERROR, BrokerResources.E_CLUSTER_RESET_LASTREFRESHTIME, e);
                    bad = true;
                    break;
                }
            }
            le = le.getNextException();
        }
        if (potentiallyBad && !bad) {
            try {
                if (s.getProperty(ClusterGlobals.STORE_PROPERTY_LASTCONFIGSERVER) == null) {
                    logger.log(Logger.ERROR, BrokerResources.E_CLUSTER_LOAD_LASTCONFIGSERVER, savele);
                    bad = true;
                }
            } catch (BrokerException e) {
                logger.log(Logger.ERROR, e.getMessage(), e);
                logger.log(Logger.ERROR, BrokerResources.E_CLUSTER_LOAD_LASTCONFIGSERVER, savele);
                bad = true;
            }
        }
        if (bad) {
            logger.log(Logger.ERROR, BrokerResources.E_CLUSTER_LOAD_LASTCONFIGSERVER);
            Broker.getBroker().exit(1, Globals.getBrokerResources().getKString(BrokerResources.E_CLUSTER_LOAD_LASTCONFIGSERVER), BrokerEvent.Type.FATAL_ERROR);
        }
    }

    /**
     * Processes the configuration and command line info.
     *
     * Creates 'connectList' : A mapping of Remote BrokerAddress <--> BrokerLink objects for all the 'autoConnect' links
     * that are initiated by this broker.
     */
    private void initBrokerList() throws BrokerException {
        connectList = Collections.synchronizedMap(new HashMap());

        BrokerMQAddress selfKey = self.getMQAddress();
        if (DEBUG) {
            logger.log(Logger.INFO, "ClusterImpl:initBrokerList. selfKey =" + selfKey);
        }

        ClusteredBroker cb = null;
        BrokerAddressImpl b = null;
        BrokerLink link = null;
        Iterator itr = clsmgr.getConfigBrokers();
        int i = 0;
        while (itr.hasNext()) {
            cb = (ClusteredBroker) itr.next();
            try {
                b = new BrokerAddressImpl((BrokerMQAddress) cb.getBrokerURL(), null, Globals.getHAEnabled(), cb.getBrokerName());
            } catch (Exception e) {
                throw new BrokerException(e.getMessage(), e);
            }
            BrokerMQAddress key = b.getMQAddress();
            if (!key.equals(selfKey)) {
                link = new BrokerLink(self, b, this);
                link.setAutoConnect(true);
                connectList.put(key, link);
                i++;
                if (DEBUG) {
                    logger.log(Logger.INFO, "ClusterImpl: Added to connectList: key=" + key + ", link=" + link + " (" + i + ")");
                }
            }
        }

        brokerList = new HashMap();
    }

    private BrokerLink searchBrokerList(BrokerMQAddress key) {
        if (brokerList == null) {
            return null;
        }

        synchronized (brokerList) {
            Iterator itr = brokerList.keySet().iterator();

            while (itr.hasNext()) {
                BrokerAddressImpl b = (BrokerAddressImpl) itr.next();
                if (key.equals(b.getMQAddress())) {
                    return (BrokerLink) brokerList.get(b);
                }
            }

            return null;
        }
    }

    protected String getTransport() {
        return transport;
    }

    protected boolean getTCPNodelay() {
        return tcpNodelay;
    }

    protected boolean getSSLNodelay() {
        return sslNodelay;
    }

    protected int getTCPInputBufferSize() {
        return tcpInbufsz;
    }

    protected int getTCPOutputBufferSize() {
        return tcpOutbufsz;
    }

    protected int getSSLInputBufferSize() {
        return sslInbufsz;
    }

    protected int getSSLOutputBufferSize() {
        return sslOutbufsz;
    }

    /*
     * private void setListenHost(String host) throws Exception { if (DEBUG) { logger.log(Logger.DEBUG,
     * "ClusterImpl: Changing the listening hostname to {0}", host); }
     *
     * if (supportClusters == false) return;
     *
     * InetAddress saveListenHost = listenHost; try { listenHost = BrokerMQAddress.resolveBindAddress(host, true);
     * ClusterServiceListener newListener = new ClusterServiceListener(this);
     *
     * // Success! Replace the global listener. if (listener != null) listener.shutdown(); listener = newListener; } catch
     * (Exception e) { listenHost = saveListenHost; throw e; } }
     */

    protected InetAddress getListenHost() {
        return listenHost;
    }

    private void setListenPort(int port) throws IOException {
        String args[] = { SERVICE_NAME, String.valueOf(port), String.valueOf(1), String.valueOf(1) };
        logger.log(Logger.INFO, BrokerResources.I_UPDATE_SERVICE_REQ, args);

        int saveListenPort = listenPort;

        try {
            listenPort = port;
            ClusterServiceListener newListener = new ClusterServiceListener(this);

            // Success! Replace the global listener.
            if (listener != null) {
                listener.shutdown();
            }
            listener = newListener;
        } catch (IOException e) {
            listenPort = saveListenPort;
            throw e;
        }
    }

    protected int getListenPort() {
        return listenPort;
    }

    protected boolean isConfigServerResolved() {
        return configServerResolved;
    }

    protected boolean waitForConfigSync() {
        if (cb == null) {
            return true;
        }
        return ((Protocol) cb).waitForConfigSync();
    }

    protected boolean checkConfigServer(BrokerAddressImpl b) {
        synchronized (configServerLock) {
            if (configServerResolved) {
                return true; // Alredy resolved...
            }

            if (configServer == null) {
                return false; // Illegal config, reject cluster connections.
            }

            BrokerMQAddress key = b.getMQAddress();
            if (key.equals(configServer.getMQAddress())) {
                configServer = b;
                configServerResolved = true;

                if (warningTask != null) {
                    warningTask.cancel();
                    warningTask = null;
                }
            }
            return configServerResolved;
        }
    }

    protected boolean addBroker(BrokerAddressImpl b, BrokerLink link) {
        if (DEBUG) {
            logger.log(Logger.DEBUGMED, "ClusterImpl: Activating link = {0}", link);
        }
        synchronized (brokerList) {
            BrokerLink rlink = (BrokerLink) brokerList.get(b);
            if (rlink != null) {
                BrokerAddressImpl remoteb = rlink.getRemote();
                if (!remoteb.getMQAddress().equals(b.getMQAddress())) {
                    String args[] = { b.toShortString(), link.toString(), rlink.toString() };
                    logger.log(Logger.ERROR, br.getKString(br.E_MBUS_SAME_ADDRESS_PEERS, args));
                    Broker.getBroker().exit(1, br.getString(br.E_MBUS_SAME_ADDRESS_PEERS, args), BrokerEvent.Type.FATAL_ERROR, null, false, true, false);
                }
                return false;
            }

            brokerList.put(b, link);
            return true;
        }
    }

    /**
     * @param broken link IOException after handshake
     */
    protected void removeBroker(BrokerAddressImpl remote, BrokerLink link, boolean broken) {
        synchronized (brokerList) {
            BrokerLink reallink = (BrokerLink) brokerList.get(remote);
            if (reallink != null && reallink != link) {
                return;
            }

            if (brokerList.remove(remote) == null) {
                return;
            }
            brokerList.notifyAll();

        }
        cb.removeBrokerInfo(remote, broken);

        if (DEBUG) {
            logger.log(Logger.DEBUGMED, "ClusterImpl: Removed link with = {0}", remote);
        }
    }

    protected void handleBrokerLinkShutdown(BrokerAddressImpl remote) {
        connectList.remove(remote.getMQAddress());
    }

    @Override
    public void receivedFileTransferRequest(BrokerAddress from, String uuid) {
        pendingFileTransfers.put(from, uuid + ":" + System.currentTimeMillis());
    }

    /**
     * Accept a connection from another broker.
     *
     * This method is invoked by the common 'ClusterServiceListener' object. The first packet must be of type
     * 'Packet.LINK_INIT' - so that the identity of the remote broker can be determined.
     */
    protected void acceptConnection(Socket conn, boolean ssl) throws IOException {
        boolean elog = false;
        BrokerAddressImpl remote = null;
        try {
            Object o = BrokerLink.consumeLinkInit(conn, null, this, true);
            if (o instanceof BrokerAddressImpl) {
                remote = (BrokerAddressImpl) o;
            } else if (o == null) {
                return;
            } else {
                elog = true;
                LinkInfo li = (LinkInfo) o;
                remote = li.getAddress();
                if (li.isFileTransferRequest()) {
                    logger.log(logger.INFO, br.getKString(br.I_RECEIVED_FILE_TX_REQUEST, remote + "[" + conn.getInetAddress() + "]"));
                    ExecutorService es = null;
                    synchronized (fileTransferExecutors) {
                        if (fileTransferShutdownIn) {
                            String emsg = br.getKString(br.W_CLUSTER_SERVICE_SHUTDOWN);
                            logger.log(logger.WARNING, emsg);
                            throw new BrokerException(emsg);
                        }
                        String key = remote.getMQAddress().getHostAddressNPort() + ":" + li.getAddress().getBrokerID();
                        es = fileTransferExecutors.get(key);
                        if (es != null) {
                            es.shutdownNow();
                            fileTransferExecutors.remove(key);
                        }
                        es = Executors.newSingleThreadExecutor();
                        fileTransferExecutors.put(key, es);
                    }
                    FileTransferRunnable runner = new FileTransferRunnable(conn, FILE_TRANSFER_SOTIMEOUT_IN, remote, es, this);
                    es.execute(runner);
                    return;
                } else {
                    logger.log(Logger.ERROR, br.getKString(br.E_CLUSTER_UNEXPECTED_PACKET_FROM, "LINK_INIT[" + remote.getClusterVersion() + "]",
                            remote + "[" + conn.getInetAddress() + "]"));
                    conn.close();
                    return;
                }
            }
        } catch (Exception e) {
            if (elog) {
                if (e instanceof BrokerException) {
                    logger.log(logger.WARNING, e.getMessage(), e);
                } else {
                    logger.logStack(logger.WARNING, e.getMessage(), e);
                }
            } else if (DEBUG) {
                logger.logStack(logger.WARNING, e.getMessage(), e);
            }
            conn.close();
            return;
        }

        // If remote is in connectList, either
        // Drop this connection.
        // (i.e. This broker initiates the connections)
        // or
        // Kill the brokerLink in connectList.
        // (i.e. Let remote initiate the connections ...)
        //
        // This can happen if the brokers are started like this -
        // broker -name A -cluster B ...
        // broker -name B -cluster A ...
        //

        BrokerMQAddress key = remote.getMQAddress();
        BrokerLink oldlink = null;
        synchronized (connectList) {
            oldlink = (BrokerLink) connectList.get(key);
            if (oldlink != null) {
                if (connectionInitiator(remote)) {
                    connectList.remove(key);
                    oldlink.shutdown();
                    oldlink = null;
                }
            }
        }
        if (oldlink != null) {
            /**
             * We already have a BrokerLink thread that's trying to initiate the connection with the remote broker. So just pass on
             * this connection to that thread..
             */
            oldlink.acceptConnection(remote, conn, ssl);
            return;
        }

        // Create the BrokerLink object
        BrokerLink link = new BrokerLink(self, remote, this);
        link.setAutoConnect(false);

        if (link.acceptConnection(remote, conn, ssl)) {
            link.start();
        }
    }

    /**
     * Construct a packet of type 'Packet.LINK_INIT' for this broker.
     *
     * Packet.LINK_INIT is always the first packet sent on any new broker-to-broker connection.
     */
    protected Packet getLinkInitPkt() {
        return getLinkInitPkt((Integer) null);
    }

    private Packet getLinkInitPkt(Integer service) {
        ByteArrayOutputStream bos = new ByteArrayOutputStream();
        DataOutputStream dos = new DataOutputStream(bos);
        try {
            if (service == null) {
                dos.writeInt(self.getClusterVersion());
            } else {
                dos.writeInt(service.intValue());
            }

            dos.writeUTF(self.getHostName());
            dos.writeUTF(self.getInstanceName());
            dos.writeInt(self.getPort());

            BrokerAddressImpl cs = (BrokerAddressImpl) getConfiguredConfigServer();
            dos.writeBoolean(cs != null);

            if (cs != null) {
                dos.writeUTF(cs.getHostName());
                dos.writeUTF(cs.getInstanceName());
                dos.writeInt(cs.getPort());
            }

            dos.writeInt(matchProps.size());
            for (Enumeration e = matchProps.propertyNames(); e.hasMoreElements();) {
                String prop = (String) e.nextElement();
                dos.writeUTF(prop);
                dos.writeUTF(matchProps.getProperty(prop));
            }

            dos.writeBoolean(Globals.getHAEnabled());
            dos.writeBoolean(self.getBrokerID() != null);
            if (self.getBrokerID() != null) {
                dos.writeUTF(self.getBrokerID());
            }
            dos.writeLong(self.getBrokerSessionUID().longValue());
            if (Globals.getHAEnabled()) {
                dos.writeLong(self.getStoreSessionUID().longValue());
            } else {
                UID uid = self.getStoreSessionUID();
                dos.writeBoolean((uid != null));
                if (uid != null) {
                    dos.writeLong(uid.longValue());
                }
            }
            dos.flush();
            bos.flush();
        } catch (Exception e) {
            /* Ignore */ }

        byte[] buf = bos.toByteArray();

        Packet p = new Packet();
        p.setPacketType(Packet.LINK_INIT);
        p.setPacketBody(buf);
        p.setDestId(0);

        return p;
    }

    /**
     * Process the Packet.LINK_INIT packet.
     */
    protected static LinkInfo processLinkInit(Packet p) throws Exception {
        String hostName = null;
        String instName = null;
        String brokerID = null;
        UID brokerSessionUID = null;
        UID storeSessionUID = null;
        boolean ha = false;
        int port = 0;

        BrokerAddressImpl remote = null;

        ByteArrayInputStream bis = new ByteArrayInputStream(p.getPacketBody());
        DataInputStream dis = new DataInputStream(bis);

        int clusterVersion = dis.readInt();

        hostName = dis.readUTF();
        instName = dis.readUTF();
        port = dis.readInt();

        BrokerAddressImpl configServer = null;
        boolean hasConfigServer = false;
        String cfgHostName = null;
        String cfgInstName = null;
        int cfgPort = 0;
        hasConfigServer = dis.readBoolean();
        if (hasConfigServer) {
            cfgHostName = dis.readUTF();
            cfgInstName = dis.readUTF();
            cfgPort = dis.readInt();
        }

        Properties props = new Properties();
        int nprops = dis.readInt();
        for (int i = 0; i < nprops; i++) {
            String prop = dis.readUTF();
            String value = dis.readUTF();
            props.setProperty(prop, value);
        }

        if (clusterVersion >= ProtocolGlobals.VERSION_400) {
            ha = dis.readBoolean();
            if (dis.readBoolean()) {
                brokerID = dis.readUTF();
            }
            brokerSessionUID = new UID(dis.readLong());
            if (ha) {
                storeSessionUID = new UID(dis.readLong());
            } else {
                try {
                    if (dis.readBoolean()) {
                        storeSessionUID = new UID(dis.readLong());
                    }
                } catch (Exception e) {
                }
            }
        }

        if (hasConfigServer) {
            if (ha) {
                throw new BrokerException(br.getKString(br.E_CLUSTER_HA_NOT_SUPPORT_MASTERBROKER));
            }
            configServer = new BrokerAddressImpl(cfgHostName, cfgInstName, cfgPort, ha, null, null, null);
        }

        remote = new BrokerAddressImpl(hostName, instName, port, ha, brokerID, brokerSessionUID, storeSessionUID);
        remote.setClusterVersion(clusterVersion);

        LinkInfo li = new LinkInfo(remote, configServer, props);
        if (clusterVersion < 0) {
            int type = clusterVersion;
            li.setServiceRequestType(type);
        }
        return li;
    }

    /**
     * Detects duplicate connections between a pair of brokers.
     *
     * @param remote BrokerAddress of the remote broker.
     * @return <code> true </code> if this broker is supposed to initiate the connection, <code> false </code> if remote is
     * supposed to initiate the connection.
     */
    private boolean connectionInitiator(BrokerAddressImpl remote) {
        if (self.getMQAddress().hashCode() > remote.getMQAddress().hashCode()) {
            return true;
        }

        return false;
    }

    /**
     * Construct the Packet.BROKER_INFO packet.
     */
    protected Packet getBrokerInfoPkt() {
        // Get the BrokerInfo object from MessageBus...
        BrokerInfo selfInfo = cb.getBrokerInfo();

        ByteArrayOutputStream bos = new ByteArrayOutputStream();

        try {
            ObjectOutputStream oos = new ObjectOutputStream(bos);
            oos.writeObject(selfInfo);
            oos.flush();
            oos.close();
        } catch (Exception e) {
            /* Ignore */ }

        byte[] buf = bos.toByteArray();

        Packet p = new Packet();
        p.setPacketType(Packet.BROKER_INFO);
        p.setPacketBody(buf);
        p.setDestId(0);

        return p;
    }

    protected ClusterBrokerInfoReply getBrokerInfoReply(BrokerInfo remote) throws Exception {
        if (cb == null) {
            throw new BrokerException(br.getKString(BrokerResources.X_CLUSTER_PROTOCOL_NOT_READY), Status.UNAVAILABLE);
        }
        return cb.getBrokerInfoReply(remote);
    }

    /**
     * Receive and process the BROKER_INFO packet.
     */
    private BrokerInfo receiveBrokerInfo(BrokerAddressImpl sender, byte[] pkt, String realRemote, BrokerLink l) {
        ByteArrayInputStream bis = new ByteArrayInputStream(pkt);
        BrokerInfo info = null;

        try {
            ObjectInputStream ois = new FilteringObjectInputStream(bis);
            info = (BrokerInfo) ois.readObject();
            info.setRealRemoteString(realRemote);
        } catch (Exception e) {
            logger.log(Logger.WARNING, br.W_MBUS_SERIALIZATION, sender);

            if (l != null) {
                l.shutdown();
            }
            return null;
        }

        Integer v = info.getClusterProtocolVersion();
        if (v != null && v.intValue() >= ProtocolGlobals.VERSION_400) {
            return info;
        }

        if (l != null) {
            l.handshakeSent();
        }

        int status = cb.addBrokerInfo(info);

        if (status == cb.ADD_BROKER_INFO_OK) {
            return info;
        }

        if (status == cb.ADD_BROKER_INFO_RETRY) {
            synchronized (brokerList) {
                BrokerLink link = (BrokerLink) brokerList.get(sender);
                if (link != null) {
                    link.closeConn();
                }
            }
        } else if (status == cb.ADD_BROKER_INFO_BAN) {
            synchronized (brokerList) {
                BrokerLink link = (BrokerLink) brokerList.get(sender);
                if (link != null) {
                    link.shutdown();
                }
            }
        }

        return null;

    }

    private void receiveBrokerInfoReply(BrokerAddressImpl sender, GPacket gp, String realRemote) {
        BrokerInfo info = null;
        try {
            ClusterBrokerInfoReply cbi = ClusterBrokerInfoReply.newInstance(gp);
            info = cbi.getBrokerInfo();
            info.setRealRemoteString(realRemote);
            if (DEBUG) {
                logger.log(Logger.DEBUG, "Received BROKER_INFO_REPLY from " + sender);
            }
            if (!info.getBrokerAddr().equals(sender)) {
                logger.log(Logger.ERROR, BrokerResources.E_INTERNAL_BROKER_ERROR, "mismatched BROKER_INFO (" + info.getBrokerAddr() + ") from " + sender);
                throw new BrokerException("mismatched BROKER_INFO");
            }
            if (Globals.getHAEnabled() && cbi.isTakingover()) {
                String msg = br.getKString(BrokerResources.E_CLUSTER_TAKINGOVER_NOTIFY_RESTART, sender);
                BrokerException ex = new BrokerException(msg);
                logger.log(logger.ERROR, msg);
                Broker.getBroker().exit(Globals.getBrokerStateHandler().getRestartCode(), msg, BrokerEvent.Type.RESTART, null, true, true, true);
                throw ex;
            }
        } catch (Exception e) {
            logger.logStack(Logger.ERROR, e.getMessage(), e);

            synchronized (brokerList) {
                BrokerLink link = (BrokerLink) brokerList.get(sender);
                if (link != null) {
                    link.shutdown();
                }
            }
            return;
        }

        int status = cb.addBrokerInfo(info);

        if (status == cb.ADD_BROKER_INFO_OK) {
            return;
        }

        if (status == cb.ADD_BROKER_INFO_RETRY) {
            synchronized (brokerList) {
                BrokerLink link = (BrokerLink) brokerList.get(sender);
                if (link != null) {
                    link.closeConn();
                }
            }
        } else if (status == cb.ADD_BROKER_INFO_BAN) {
            synchronized (brokerList) {
                BrokerLink link = (BrokerLink) brokerList.get(sender);
                if (link != null) {
                    link.shutdown();
                }
            }
        }
    }

    protected GPacket getFirstInfoPacket() {
        if (!Globals.useSharedConfigRecord()) {
            return null;
        }
        ChangeRecordInfo cri = cb.getLastStoredChangeRecord();
        if (cri == null) {
            return null;
        }
        ClusterFirstInfoInfo cii = ClusterFirstInfoInfo.newInstance();
        cii.setLastStoredChangeRecord(cri);
        cii.setBroadcast(false);
        return cii.getGPacket();
    }

    protected void processFirstInfoPacket(GPacket gp, BrokerLink link) {
        try {
            if (!Globals.useSharedConfigRecord()) {// should never happen
                throw new BrokerException("Unexpected " + ProtocolGlobals.getPacketTypeDisplayString(gp.getType()) + " packet from " + link.getRemote());
            }
            ClusterFirstInfoInfo cii = ClusterFirstInfoInfo.newInstance(gp);
            ChangeRecordInfo cri = cii.getLastStoredChangeRecord();
            if (cri == null) {
                return;
            }
            logger.log(logger.INFO, br.getKString(br.I_CLUSTER_RECEIVED_FIRST_INFO,
                    ProtocolGlobals.getPacketTypeDisplayString(gp.getType()) + "[(sharecc)" + cri.toString() + "]", link.getRemote()));

            cb.syncChangeRecordOnJoin(link.getRemote(), cri);

        } catch (Exception e) {
            logger.logStack(logger.ERROR, e.getMessage(), e);
            if (e instanceof BrokerException) {
                if (((BrokerException) e).getStatusCode() == Status.PRECONDITION_FAILED) {
                    link.shutdown();
                    return;
                }
            }
            link.closeConn();
        }
    }

    private void setFlowControl(BrokerAddressImpl addr, boolean enabled) {
        BrokerLink link = null;
        synchronized (brokerList) {
            link = (BrokerLink) brokerList.get(addr);
        }
        if (link != null) {
            link.setFlowControl(enabled);
        }
    }

    /**
     * Receive packets from remote brokers.
     */
    protected Object receivePacket(BrokerAddressImpl from, Packet p, String realRemote, BrokerLink l) {
        Object ret = null;

        if (cb == null) {
            return ret;
        }

        switch (p.getPacketType()) {
        case Packet.UNICAST:
            cb.receiveUnicast(from, p.getDestId(), p.getPacketBody());
            break;

        case Packet.BROADCAST:
            cb.receiveBroadcast(from, p.getDestId(), p.getPacketBody());
            break;

        case Packet.BROKER_INFO:
            ret = receiveBrokerInfo(from, p.getPacketBody(), realRemote, l);
            break;

        case Packet.STOP_FLOW:
            setFlowControl(from, true);
            break;

        case Packet.RESUME_FLOW:
            setFlowControl(from, false);
            break;

        case Packet.PING:
            if (DEBUG) {
                logger.log(Logger.DEBUG, "ClusterImpl: Received ping from : " + from);
            }
            break;
        default:
            logger.log(logger.WARNING, "ClusterImpl: INTERNAL ERROR -" + " Received Unknown packet from : " + from);
        }
        return ret;
    }

    protected void receivePacket(BrokerAddressImpl from, GPacket gp, String realRemote) {
        if (gp.getType() == ProtocolGlobals.G_BROKER_INFO_REPLY) {
            receiveBrokerInfoReply(from, gp, realRemote);
            return;
        }
        if (cb == null) {
            return;
        }

        if (gp.getBit(gp.B_BIT)) {
            cb.receiveBroadcast(from, gp);
        } else {
            cb.receiveUnicast(from, gp);
        }
    }

    @Override
    public void useGPackets(boolean useGPackets) {
        this.useGPackets = useGPackets;
    }

    @Override
    public void setCallback(ClusterCallback cb) {
        this.cb = cb;
        self.setClusterVersion(cb.getHighestSupportedVersion());
    }

    @Override
    public void setMatchProps(Properties matchProps) {
        this.matchProps = matchProps;
    }

    protected Properties getMatchProps() {
        return matchProps;
    }

    @Override
    public BrokerAddress getSelfAddress() {
        return self;
    }

    @Override
    public BrokerAddress getConfigServer() throws BrokerException {
        synchronized (configServerLock) {
            if (!configServerResolved) {
                if (Globals.getConfig().getBooleanProperty(Globals.IMQ + ".cluster.masterbroker.enforce", true)) {
                    throw new BrokerException(Globals.getBrokerResources().getString(BrokerResources.X_CLUSTER_UNREACHABLE,
                            (configServer == null ? "null" : configServer.toString())));
                } else {
                    return null;
                }
            }
            return configServer;
        }
    }

    @Override
    public void changeMasterBroker(BrokerAddress newmaster, BrokerAddress oldmaster) throws BrokerException {
        synchronized (configServerLock) {

            logger.log(logger.INFO,
                    br.getKString(br.I_CLUSTER_CHANGE_MASTER_BROKER, (configServer == null ? "null" : configServer.toString()), newmaster.toString()));

            if (configServer == null) {
                throw new BrokerException(br.getKString(br.X_CLUSTER_NO_MASTER_BROKER_REJECT_CHANGE_MASTER), Status.PRECONDITION_FAILED);
            }
            if (!oldmaster.equals(configServer)) {
                throw new BrokerException(br.getKString(br.X_CLUSTER_CHANGE_MASTER_BROKER_MISMATCH, oldmaster.toString(), configServer.toString()),
                        Status.PRECONDITION_FAILED);

            }

            BrokerAddressImpl oldconfigServer = this.configServer;
            this.configServer = (BrokerAddressImpl) newmaster;

            String newmasterhp = newmaster.getMQAddress().getHostAddressNPort();
            String oldmasterhp = Globals.getConfig().getProperty(ClusterManager.CONFIG_SERVER);
            Properties prop = new Properties();
            prop.put(ClusterManager.CONFIG_SERVER, newmasterhp);
            try {
                Globals.getConfig().updateProperties(prop, true);
            } catch (Exception e) {
                String emsg = br.getKString(br.E_CHANGE_MASTER_BROKER_FAIL, e.getMessage());
                logger.logStack(logger.ERROR, emsg, e);
                try {
                    prop.put(ClusterManager.CONFIG_SERVER, oldmasterhp);
                } catch (Exception e1) {
                    logger.log(logger.ERROR, br.getKString(br.E_CLUSTER_RESTORE_MASTER_BROKER_PROP_FAIL, e1.getMessage()));
                    throw new BrokerException(emsg, e, Status.ERROR);
                }
                this.configServer = oldconfigServer;
                throw new BrokerException(emsg, e, Status.PRECONDITION_FAILED);
            }
        }
    }

    protected BrokerAddress getConfiguredConfigServer() {
        synchronized (configServerLock) {
            return configServer;
        }
    }

    @Override
    public void marshalBrokerAddress(BrokerAddress addr, GPacket gp) {
        ((BrokerAddressImpl) addr).writeBrokerAddress(gp);
    }

    @Override
    public BrokerAddress unmarshalBrokerAddress(GPacket gp) throws Exception {
        return BrokerAddressImpl.readBrokerAddress(gp);
    }

    /**
     * Start all the initial BrokerLink threads in 'connectList'. This list contains all the links (connections) to be
     * initiated by this broker.
     */
    @Override
    public void start() throws IOException {
        clsmgr.addEventListener(this);

        synchronized (connectList) {
            Collection values = connectList.values();
            Iterator itr = values.iterator();
            while (itr.hasNext()) {
                BrokerLink link = (BrokerLink) itr.next();
                link.start();
            }
        }

        listener = new ClusterServiceListener(this);
        Object[] args = { SERVICE_NAME, getTransport() + " [ " + getServerSocketString() + " ]", Integer.valueOf(1), Integer.valueOf(1) };
        logger.log(Logger.INFO, BrokerResources.I_SERVICE_START, args);

        if (Globals.getClusterID() != null) {
            logger.log(Logger.INFO, BrokerResources.I_CLUSTER_USING_CLUSTERID, Globals.getClusterID());
        }

        if (!configServerResolved) {
            warningTask = new WarningTask(this);
            Globals.getTimer().schedule(warningTask, 60 * 1000, 180 * 1000);
        }

        pingInterval = clsmgr.getClusterPingInterval();
        pingTimer = new PingTimerTask();
        logger.log(Logger.INFO, BrokerResources.I_CLUSTER_PING_INTERVAL, Integer.valueOf(pingInterval));
        Globals.getTimer().schedule(pingTimer, pingInterval * 1000L, pingInterval * 1000L);
    }

    protected String getServerSocketString() {
        ClusterServiceListener l = listener;
        if (l != null) {
            return l.getServerSocketString();
        }
        return null;
    }

    /**
     * Shutdown the cluster.
     */
    @Override
    public void shutdown(boolean force, BrokerAddress excludedBroker) {
        if (listener != null && excludedBroker == null) {
            listener.shutdown(); // Stop accepting connections.
        }

        if (brokerList == null) {
            return;
        }
        synchronized (fileTransferExecutors) {
            fileTransferShutdownIn = true;
            if (excludedBroker == null) {
                fileTransferShutdownOut = true;
            }
            Iterator<Map.Entry<String, ExecutorService>> itr = fileTransferExecutors.entrySet().iterator();
            Map.Entry<String, ExecutorService> entry = null;
            while (itr.hasNext()) {
                entry = itr.next();
                entry.getValue().shutdownNow();
                itr.remove();
            }
        }

        long timeout = Globals.getConfig().getIntProperty(SHUTDOWN_TIMEOUT_PROP, DEFAULT_SHUTDOWN_TIMEOUT);
        long waittime = timeout * 1000;
        long endtime = System.currentTimeMillis() + waittime;

        int pv = -1;
        try {
            pv = Globals.getClusterBroadcast().getClusterVersion();
        } catch (Exception e) {
            logger.log(logger.DEBUG, "Unable to get cluster protocol version on cluster shutdown: " + e.getMessage());
        }
        if (pv >= ClusterBroadcast.VERSION_400 && !force) {

            synchronized (brokerList) {
                while (brokerList.size() != 0) {
                    if (brokerList.size() == 1 && excludedBroker != null) {
                        if (brokerList.get(excludedBroker) != null) {
                            break;
                        }
                    }
                    logger.log(logger.INFO, br.getKString(br.I_CLUSTER_SERVICE_SHUTDOWN_WAITING, Integer.valueOf(brokerList.size())));
                    try {
                        brokerList.wait(waittime);
                    } catch (Exception e) {
                    }

                    long curtime = System.currentTimeMillis();
                    if (curtime >= endtime) {
                        if (brokerList.size() > 0) {
                            logger.log(logger.WARNING, br.getKString(br.W_CLUSTER_SERVICE_SHUTDOWN_TIMEOUT));
                        }
                        break;
                    }
                    waittime = endtime - curtime;
                }
            }
        }

        BrokerAddress remote = null;
        BrokerLink l = null;
        synchronized (brokerList) {
            Iterator itr = brokerList.keySet().iterator();
            while (itr.hasNext()) {
                remote = (BrokerAddress) itr.next();
                if (excludedBroker != null && remote.equals(excludedBroker)) {
                    continue;
                }
                l = (BrokerLink) brokerList.get(remote);
                logger.log(logger.WARNING, br.getKString(br.W_CLUSTER_FORCE_CLOSE_LINK, remote, br.getString(br.M_LINK_SHUTDOWN)));
                l.shutdown();
                itr.remove();
            }
        }

        synchronized (connectList) {
            Collection values = connectList.values();
            Iterator itr = values.iterator();
            while (itr.hasNext()) {
                BrokerLink link = (BrokerLink) itr.next();
                if (excludedBroker != null) {
                    BrokerAddress addr = link.getRemote();
                    if (addr != null && addr.equals(excludedBroker)) {
                        continue;
                    }
                }
                link.shutdown();
            }
        }

        if (warningTask != null) {
            try {
                warningTask.cancel();
            } catch (Exception e) {
            }
            warningTask = null;
        }
        if (pingTimer != null) {
            try {
                pingTimer.cancel();
            } catch (Exception e) {
            }
            pingTimer = null;
        }
    }

    @Override
    public void closeLink(BrokerAddress remote, boolean force) {
        synchronized (brokerList) {
            BrokerLink l = (BrokerLink) brokerList.get(remote);
            if (l != null) {
                l.closeConn(force);
            }
            brokerList.notifyAll();
        }
    }

    private void closeLink(String brokerID, UID storeSession) {
        if (!Globals.getHAEnabled()) {
            return;
        }
        BrokerAddress remote = null;
        BrokerLink l = null;
        synchronized (brokerList) {
            Iterator itr = brokerList.keySet().iterator();
            while (itr.hasNext()) {
                remote = (BrokerAddress) itr.next();
                l = (BrokerLink) brokerList.get(remote);
                if (remote.getBrokerID().equals(brokerID) && (storeSession == null || remote.getStoreSessionUID().equals(storeSession))) {
                    logger.log(logger.WARNING, br.getKString(br.W_CLUSTER_CLOSE_DOWN_BROKER_LINK, remote));
                    l.closeConn(false);
                }
            }
        }
    }

    @Override
    public boolean isReachable(BrokerAddress remote, int timeout) throws IOException {
        Class inetc = null;
        java.lang.reflect.Method m = null;
        try {
            inetc = Class.forName("java.net.InetAddress");
            m = inetc.getMethod("isReachable", new Class[] { java.lang.Integer.TYPE });
            boolean b = ((Boolean) m.invoke(((BrokerAddressImpl) remote).getHost(), new Object[] { Integer.valueOf(timeout * 1000) })).booleanValue();
            if (b) {
                logger.log(Logger.INFO, br.getKString(br.I_CLUSTER_REMOTE_IP_REACHABLE, remote, Integer.valueOf(timeout)));
            } else {
                logger.log(Logger.INFO, br.getKString(br.I_CLUSTER_REMOTE_IP_UNREACHABLE, remote, Integer.valueOf(timeout)));
            }
            return b;

        } catch (NoSuchMethodException e) {
            if (DEBUG) {
                logger.logStack(Logger.WARNING, e.getMessage(), e);
            }
            return true;
        } catch (Exception e) {
            logger.logStack(Logger.WARNING, e.getMessage(), e);
            return true;
        }
    }

    /**
     * Wait for the cluster connection initialization.
     */
    @Override
    public void waitClusterInit() {
        if (readyForBroadcast) {
            return;
        }
        long waitime = BrokerLink.INIT_WAIT_TIME;
        if (waitime <= 0) {
            waitime = BrokerLink.DEFAULT_INIT_WAIT_TIME;
        }
        long endtime = System.currentTimeMillis() + waitime;
        BrokerLink link = null;
        Object[] values = connectList.values().toArray();
        for (int i = 0; i < values.length; i++) {
            link = (BrokerLink) values[i];
            link.waitLinkInit(waitime);
            waitime = endtime - System.currentTimeMillis();
            if (waitime < 30L) {
                waitime = 30L;
            }
        }
        readyForBroadcast = true;
    }

    @Override
    public long getLinkInitWaitTime() {
        return BrokerLink.INIT_WAIT_TIME;
    }

    public void sendFlowControlUpdate(BrokerAddressImpl addr) throws IOException {
        if (flowControlState == Packet.STOP_FLOW) {
            sendFlowControlUpdate(addr, flowControlState);
        }
    }

    private void sendFlowControlUpdate(BrokerAddressImpl addr, int type) throws IOException {
        Packet p = new Packet();
        p.setPacketType(type);
        p.setPacketBody(null);
        p.setDestId(0);

        BrokerLink link;
        if (addr != null) {
            synchronized (brokerList) {
                link = (BrokerLink) brokerList.get(addr);
            }
            if (link != null) {
                link.sendPacket(p);
            }
            return;
        }

        Collection values;
        synchronized (brokerList) {
            values = new ArrayList(brokerList.values());
        }
        Iterator itr = values.iterator();
        while (itr.hasNext()) {
            link = (BrokerLink) itr.next();
            link.sendPacket(p);
        }
    }

    @Override
    public void stopMessageFlow() throws IOException {
        synchronized (brokerList) {
            flowControlState = Packet.STOP_FLOW;
        }

        sendFlowControlUpdate(null, Packet.STOP_FLOW);
    }

    @Override
    public void resumeMessageFlow() throws IOException {
        synchronized (brokerList) {
            flowControlState = Packet.RESUME_FLOW;
        }

        sendFlowControlUpdate(null, Packet.RESUME_FLOW);
    }

    @Override
    public void enablePingLogging(BrokerAddress remote) {
        Collection values;
        synchronized (brokerList) {
            values = new ArrayList(brokerList.values());
        }
        Iterator itr = values.iterator();
        while (itr.hasNext()) {
            BrokerLink link = (BrokerLink) itr.next();
            if (link.getRemote().equals(remote)) {
                link.enablePingLogging();
            }
        }
    }

    private void sendPingGPacket() throws Exception {
        GPacket gp = GPacket.getInstance();
        gp.setType(ProtocolGlobals.G_PING);

        Collection values;
        synchronized (brokerList) {
            values = new ArrayList(brokerList.values());
        }
        Iterator itr = values.iterator();
        while (itr.hasNext()) {
            BrokerLink link = (BrokerLink) itr.next();
            if (!link.isIOActive()) {
                try {
                    link.sendPacket(gp);
                } catch (Exception e) {
                    /* Ignore */}
            } else {
                link.logIOActiveInPingInterval(pingInterval);
            }
            link.clearIOActiveFlag();
        }
    }

    private void sendPingPacket() throws Exception {
        Packet p = new Packet();
        p.setPacketType(Packet.PING);
        p.setPacketBody(null);
        p.setDestId(0);

        Collection values;
        synchronized (brokerList) {
            values = new ArrayList(brokerList.values());
        }
        Iterator itr = values.iterator();
        while (itr.hasNext()) {
            BrokerLink link = (BrokerLink) itr.next();
            if (!link.isIOActive()) {
                try {
                    link.sendPacket(p);
                } catch (Exception e) {
                    /* Ignore */}
            }
            link.clearIOActiveFlag();
        }
    }

    private class PingTimerTask extends TimerTask {
        @Override
        public void run() {
            try {
                if (useGPackets) {
                    sendPingGPacket();
                } else {
                    sendPingPacket();
                }
            } catch (Exception e) {
                /* Ignore */ }
        }
    }

    @Override
    public void unicastUrgent(BrokerAddress addr, GPacket gp) throws IOException {
        if (!useGPackets) {
            logger.log(logger.WARNING, "Protocol mismatch. GPacket unicast on old cluster");
            Thread.dumpStack();
        }
        unicast(addr, gp, false, false, true);
    }

    @Override
    public void unicast(BrokerAddress addr, GPacket gp) throws IOException {
        if (!useGPackets) {
            logger.log(logger.WARNING, "Protocol mismatch. GPacket unicast on old cluster");
            Thread.dumpStack();
        }

        unicast(addr, gp, false, false, false);
    }

    @Override
    public void unicastAndClose(BrokerAddress addr, GPacket gp) throws IOException {
        unicast(addr, gp, false, true, false);
    }

    @Override
    public void unicast(BrokerAddress addr, GPacket gp, boolean flowControl) throws IOException {
        unicast(addr, gp, flowControl, false, false);
    }

    private void unicast(BrokerAddress addr, GPacket gp, boolean flowControl, boolean close, boolean urgent) throws IOException {
        if (!useGPackets) {
            logger.log(logger.WARNING, "Protocol mismatch. GPacket unicast on old cluster");
            Thread.dumpStack();
        }

        if (addr.equals(self)) {
            if (cb != null) {
                cb.receiveUnicast(self, gp);
            }
            return;
        }

        BrokerLink link = null;
        synchronized (brokerList) {
            link = (BrokerLink) brokerList.get(addr);
        }

        if (link == null) {
            throw new IOException(br.getString(BrokerResources.X_CLUSTER_UNICAST_UNREACHABLE, addr.toString()));
        }

        gp.setBit(gp.F_BIT, flowControl);
        link.sendPacket(gp, close, urgent);
    }

    @Override
    public boolean isLinkModified(BrokerAddress remote, Object o) {
        Collection values;
        synchronized (brokerList) {
            values = new ArrayList(brokerList.values());
        }
        Iterator itr = values.iterator();
        while (itr.hasNext()) {
            BrokerLink link = (BrokerLink) itr.next();
            if (link.getRemote().equals(remote)) {
                if (link.isModified(o)) {
                    return true;
                } else {
                    return false;
                }
            }
        }
        return true;
    }

    @Override
    public Map<BrokerAddress, Object> broadcast(GPacket gp) throws IOException {
        return broadcast(gp, false);
    }

    @Override
    public Map<BrokerAddress, Object> broadcastUrgent(GPacket gp) throws IOException {
        return broadcast(gp, true);
    }

    private Map<BrokerAddress, Object> broadcast(GPacket gp, boolean urgent) throws IOException {
        if (!useGPackets) {
            logger.log(logger.WARNING, "Protocol mismatch. GPacket broadcast on old cluster");
            Thread.dumpStack();
        }

        if (!readyForBroadcast) {
            waitClusterInit();
        }

        gp.setBit(gp.B_BIT, true);
        LinkedHashMap<BrokerAddress, Object> map = null;

        Collection values;
        synchronized (brokerList) {
            values = new ArrayList(brokerList.values());
        }
        IOException ex = null;
        Object o = null;
        Iterator itr = values.iterator();
        while (itr.hasNext()) {
            BrokerLink link = (BrokerLink) itr.next();
            try {
                o = link.sendPacket(gp, false, urgent);
                if (map == null) {
                    map = new LinkedHashMap<>(brokerList.size());
                }
                map.put(link.getRemote(), o);
            } catch (IOException e) {
                if (ex == null) {
                    ex = e;
                }
                continue;
            }
        }
        if (ex != null) {
            throw ex;
        }
        if (map == null) {
            map = new LinkedHashMap<>(1);
        }
        return map;
    }

    @Override
    public void unicast(BrokerAddress addr, int destId, byte[] pkt) throws IOException {
        if (useGPackets) {
            logger.log(logger.WARNING, "Protocol mismatch. Old packet unicast on raptor cluster");
            Thread.dumpStack();
        }

        unicast(addr, destId, pkt, false);
    }

    @Override
    public void unicast(BrokerAddress addr, int destId, byte[] pkt, boolean flowControl) throws IOException {
        if (useGPackets) {
            logger.log(logger.WARNING, "Protocol mismatch. Old packet unicast on raptor cluster");
            Thread.dumpStack();
        }

        if (addr.equals(self)) {
            if (cb != null) {
                cb.receiveUnicast(self, destId, pkt);
            }
            return;
        }

        BrokerLink link = null;
        synchronized (brokerList) {
            link = (BrokerLink) brokerList.get(addr);
        }

        if (link == null) {
            // exception is never displayed (does not need to be localized)
            throw new IOException("Packet send failed. Unreachable BrokerAddress : " + addr);
        }

        Packet p = new Packet();
        p.setPacketType(Packet.UNICAST);
        p.setPacketBody(pkt);
        p.setDestId(destId);
        p.setFlag(Packet.USE_FLOW_CONTROL, flowControl);

        link.sendPacket(p);
    }

    @Override
    public void broadcast(int destId, byte[] pkt) throws IOException {
        if (useGPackets) {
            logger.log(logger.WARNING, "Protocol mismatch. Old packet broadcast on raptor cluster");
            Thread.dumpStack();
        }

        if (!readyForBroadcast) {
            waitClusterInit();
        }

        Packet p = new Packet();
        p.setPacketType(Packet.BROADCAST);
        p.setPacketBody(pkt);
        p.setDestId(destId);

        Collection values;
        synchronized (brokerList) {
            values = new ArrayList(brokerList.values());
        }
        IOException ex = null;
        Iterator itr = values.iterator();
        while (itr.hasNext()) {
            BrokerLink link = (BrokerLink) itr.next();
            try {
                link.sendPacket(p);
            } catch (IOException e) {
                if (ex == null) {
                    ex = e;
                }
            }
        }
        if (ex != null) {
            throw ex;
        }
    }

    @Override
    public void reloadCluster() {
        try {
            clsmgr.reloadConfig();
        } catch (Exception e) {
            logger.logStack(logger.WARNING, br.getKString(br.W_CLUSTER_RELOAD_FAILED), e);
        }
    }

    /**********************************************************
     * implements ClusterListener *************************
     **********************************************************/

    /**
     * Called to notify ClusterListeners when the cluster service configuration. Configuration changes include:
     * <UL>
     * <LI>cluster service port</LI>
     * <LI>cluster service hostname</LI>
     * <LI>cluster service transport</LI>
     * </UL>
     * <P>
     *
     * @param name the name of the changed property
     * @param value the new value of the changed property
     */
    @Override
    public void clusterPropertyChanged(String name, String value) {
        if (name.equals(ClusterManager.PORT_PROPERTY)) {
            try {
                setListenPort(Integer.parseInt(value));
            } catch (IOException e) {
                logger.logStack(logger.ERROR, e.getMessage(), e);
            }
        } else if (name.equals(ClusterManager.CLUSTER_PING_INTERVAL_PROP)) {
            int newInterval = pingInterval;
            try {
                newInterval = Integer.parseInt(value);
            } catch (Exception e) {
                /* ignore */ }

            if (newInterval != pingInterval) {
                pingInterval = newInterval;

                PingTimerTask task = pingTimer;
                if (task != null) {
                    task.cancel();
                }
                pingTimer = new PingTimerTask();
                logger.log(Logger.INFO, BrokerResources.I_CLUSTER_PING_INTERVAL, Integer.valueOf(pingInterval));
                Globals.getTimer().schedule(pingTimer, pingInterval * 1000L, pingInterval * 1000L);
            }
        }
    }

    /**
     * Called when a new broker has been added.
     *
     * @param broker the new broker added.
     */
    @Override
    public void brokerAdded(ClusteredBroker broker, UID uid) {
        if (!broker.isConfigBroker()) {
            if (DEBUG) {
                logger.log(logger.INFO, "ClusterImpl:brokerAdded: Ignore dynamic broker " + broker);
            }
            return;
        }
        BrokerMQAddress key = (BrokerMQAddress) broker.getBrokerURL();
        if (!key.equals(self.getMQAddress()) && !connectList.containsKey(key)) {
            boolean newLink = false;
            BrokerLink link = searchBrokerList(key); // XXX 1
            if (link == null) {
                BrokerAddressImpl b = null;
                try {
                    b = new BrokerAddressImpl(key, null, Globals.getHAEnabled(), broker.getBrokerName());
                    link = new BrokerLink(self, b, this);
                    newLink = true;
                } catch (Exception e) {
                    logger.logStack(logger.ERROR, br.getKString(br.W_CLUSTER_AUTOCONNECT_ADD_FAILED, broker), e);
                    return;
                }
            }
            link.setAutoConnect(true); // XXX 2
            connectList.put(key, link);
            if (DEBUG) {
                logger.log(Logger.INFO, "ClusterImpl: Added link to connectList - " + link);
            }
            if (newLink) {
                link.start();
            }
            return;
        }
        if (DEBUG) {
            logger.log(logger.INFO, "Broker link for " + key + " (" + broker + ") already exist");
        }
    }

    @Override
    public Hashtable getDebugState() {
        Hashtable ht = new Hashtable();
        ht.put("self", self.toString());
        BrokerAddressImpl cs = configServer;
        if (cs != null) {
            ht.put("configServer[masterbroker]", cs.toString());
            ht.put("configServerResolved", Boolean.valueOf(configServerResolved));
        }
        ArrayList clist = null;
        synchronized (connectList) {
            clist = new ArrayList(connectList.keySet());
        }
        ht.put("connectListCount", String.valueOf(clist.size()));
        Iterator itr = clist.iterator();
        while (itr.hasNext()) {
            BrokerMQAddress key = (BrokerMQAddress) itr.next();
            BrokerLink link = (BrokerLink) connectList.get(key);
            if (link != null) {
                ht.put(key.toString(), link.toString());
            }
        }
        return ht;
    }

    /**
     * Called when a broker has been removed.
     *
     * @param broker the broker removed.
     */
    @Override
    public void brokerRemoved(ClusteredBroker broker, UID uid) {
        if (!broker.isConfigBroker()) {
            if (DEBUG) {
                logger.log(logger.INFO, "ClusterImpl:brokerRemoved: Ignore dynamic broker " + broker);
            }
            return;
        }

        BrokerMQAddress key = (BrokerMQAddress) broker.getBrokerURL();
        BrokerLink link = (BrokerLink) connectList.remove(key);
        if (link != null) {
            link.shutdown();
            if (DEBUG) {
                logger.log(Logger.INFO, "ClusterImpl: Removed link from connectList - " + link);
            }
            return;
        }
        if (DEBUG) {
            logger.log(logger.INFO, "Broker link for " + key + " (" + broker + ") not exist");
        }
    }

    /**
     * Called when the broker who is the master broker changes (because of a reload properties).
     *
     * @param oldMaster the previous master broker.
     * @param newMaster the new master broker.
     */
    @Override
    public void masterBrokerChanged(ClusteredBroker oldMaster, ClusteredBroker newMaster) {
        // do nothing
    }

    /**
     * Called when the status of a broker has changed. The status may not be accurate if a previous listener updated the
     * status for this specific broker.
     *
     * @param brokerid the name of the broker updated.
     * @param oldStatus the previous status.
     * @param newStatus the new status.
     * @param userData data associated with the state change
     */
    @Override
    public void brokerStatusChanged(String brokerid, int oldStatus, int newStatus, UID brokerSession, Object userData) {
        ClusteredBroker cb = clsmgr.getBroker(brokerid);
        if (!(cb instanceof HAClusteredBroker)) {
            return;
        }
        if (cb.isLocalBroker()) {
            return;
        }

        if (BrokerStatus.getBrokerIsDown(newStatus)) {
            closeLink(cb.getBrokerName(), (UID) userData);
        }

    }

    /**
     * Called when the state of a broker has changed. The state may not be accurate if a previous listener updated the state
     * for this specific broker.
     *
     * @param brokerid the name of the broker updated.
     * @param oldState the previous state.
     * @param newState the new state.
     */
    @Override
    public void brokerStateChanged(String brokerid, BrokerState oldState, BrokerState newState) {
        /*
         * if (newState == BrokerState.FAILOVER_PENDING) { //storeSession pass in ? cb.preTakeover(brokerid,
         * ((HAClusteredBroker)clsmgr.getBroker(brokerid)).getStoreSessionUID(),
         * ((BrokerMQAddress)clsmgr.getBroker(brokerid).getBrokerURL()).getHost().getHostAddress(),
         * ((HAClusteredBroker)clsmgr.getBroker(brokerid)).getBrokerSessionUID()); return; } if (newState ==
         * BrokerState.FAILOVER_COMPLETE) { //storeSession pass in ? cb.postTakeover(brokerid,
         * ((HAClusteredBroker)clsmgr.getBroker(brokerid)).getStoreSessionUID(), false); return; } if (newState ==
         * BrokerState.FAILOVER_FAILED) { //storeSession pass in ? cb.postTakeover(brokerid,
         * ((HAClusteredBroker)clsmgr.getBroker(brokerid)).getStoreSessionUID(), true); return; }
         */
    }

    /**
     * Called when the version of a broker has changed. The state may not be accurate if a previous listener updated the
     * version for this specific broker.
     *
     * @param brokerid the name of the broker updated.
     * @param oldVersion the previous version.
     * @param newVersion the new version.
     */
    @Override
    public void brokerVersionChanged(String brokerid, int oldVersion, int newVersion) {
    }

    /**
     * Called when the url address of a broker has changed. The address may not be accurate if a previous listener updated
     * the address for this specific broker.
     *
     * @param brokerid the name of the broker updated.
     * @param newAddress the previous address.
     * @param oldAddress the new address.
     */
    @Override
    public void brokerURLChanged(String brokerid, MQAddress oldAddress, MQAddress newAddress) {
    }

    @Override
    public void transferFiles(String[] fileNames, BrokerAddress targetBroker, Long syncTimeout, String uuid, String myBrokerID, String module,
            FileTransferCallback callback) throws BrokerException {
        String to = targetBroker.toString();
        Socket socket = null;
        MessageDigest digest = null;
        try {
            if (fileTransferShutdownOut) {
                String emsg = br.getKString(br.W_CLUSTER_SERVICE_SHUTDOWN);
                logger.log(logger.WARNING, emsg);
                throw new BrokerException(emsg);
            }
            try {
                digest = MessageDigest.getInstance("SHA1");
            } catch (Exception e) {
                String emsg = "";
                if (e instanceof NoSuchAlgorithmException) {
                    emsg = "Unable to create MessageDigest for file transfer";
                    logger.log(logger.ERROR, emsg + ": " + e);
                } else {
                    emsg = "Unexpectd exception in creating MessageDigest for file transfer";
                    logger.logStack(logger.ERROR, emsg, e);
                }
                throw new BrokerException(emsg, e);
            }

            logger.log(logger.INFO, br.getKString(br.I_CONNECT_REMOTE_FOR_FILE_TX, targetBroker));

            socket = BrokerLink.makeSocket((BrokerAddressImpl) targetBroker, this, false, null);
            to += "[" + socket.getInetAddress() + "]";
            try {
                socket.setTcpNoDelay(false);
            } catch (Exception e) {
                logger.log(logger.WARNING, "Failed to set socket TCP_NODELAY for file transfer: " + e);
            }
            try {
                socket.setSoTimeout((int) Math.min(FILE_TRANSFER_SOTIMEOUT_OUT, syncTimeout.longValue()));
            } catch (Exception e) {
                logger.log(logger.WARNING, "Failed to set socket timeout for file transfer: " + e);
            }
            OutputStream os = socket.getOutputStream();
            logger.log(logger.INFO, br.getKString(br.I_CLUSTER_SEND_TX_FILE_LINK_REQUEST, to));
            Packet pkt = getLinkInitPkt(Integer.valueOf(LinkInfo.SERVICE_FILE_TRANSFER));
            pkt.writePacket(os);
            os.flush();
            int numFiles = fileNames.length;
            ClusterTransferFileListInfo tfl = ClusterTransferFileListInfo.newInstance(uuid, myBrokerID, Integer.valueOf(numFiles), FileTransferCallback.STORE);
            Object[] tmpargs = { String.valueOf(numFiles), tfl.toString(), to };
            logger.log(logger.INFO, br.getKString(br.I_CLUSTER_TX_FILES, tmpargs));
            GPacket gp = tfl.getGPacket();
            gp.write(os);
            os.flush();
            String filename = null;
            long filesize = 0L;
            long lastmodtime = 0L;
            for (int cnt = 0; cnt < numFiles; cnt++) {
                filename = fileNames[cnt];
                logger.log(logger.INFO, br.getKString(br.I_CLUSTER_TX_FILE, filename + " [" + Arrays.toString(tmpargs) + "]", to));

                if (fileTransferShutdownOut) {
                    String emsg = br.getKString(br.W_CLUSTER_SERVICE_SHUTDOWN);
                    logger.log(logger.WARNING, emsg);
                    throw new BrokerException(emsg);
                }

                Map props = new HashMap();
                InputStream is = callback.getFileInputStream(filename, targetBroker, props);
                try {

                    filesize = ((Long) props.get("filesize")).longValue();
                    lastmodtime = ((Long) props.get("lastmodtime")).longValue();
                    ClusterTransferFileStartInfo tfs = ClusterTransferFileStartInfo.newInstance(uuid, module, myBrokerID, filename, filesize, lastmodtime);
                    gp = tfs.getGPacket();
                    gp.write(os);
                    os.flush();

                    digest.reset();
                    byte[] buf = new byte[FILE_TRANSFER_CHUNK_SIZE];
                    int totalread = 0;
                    int count;
                    while (totalread < filesize) {
                        if (fileTransferShutdownOut) {
                            String emsg = br.getKString(br.W_CLUSTER_SERVICE_SHUTDOWN);
                            logger.log(logger.WARNING, emsg);
                            throw new BrokerException(emsg);
                        }
                        count = 0;
                        try {
                            count = is.read(buf, 0, (int) Math.min(FILE_TRANSFER_CHUNK_SIZE, (filesize - totalread)));
                        } catch (IOException e) {
                            String emsg = br.getKString(br.E_FILE_READ, filename);
                            throw new BrokerException(emsg);
                        }
                        if (count < 0) {
                            String[] args = { String.valueOf(totalread), String.valueOf(filesize - totalread), filename };
                            String emsg = br.getKString(br.E_FILE_READ_EOF, args);
                            throw new BrokerException(emsg);
                        }
                        totalread += count;
                        os.write(buf, 0, count);
                        os.flush();
                        digest.update(buf, 0, count);
                    }
                    byte[] dg = digest.digest();
                    ClusterTransferFileEndInfo tfe = ClusterTransferFileEndInfo.newInstance(uuid, module, myBrokerID, filename, dg, ((cnt + 1) < numFiles));
                    gp = tfe.getGPacket();
                    gp.write(os);
                    os.flush();

                } finally {
                    try {
                        is.close();
                    } catch (Exception e) {
                    }
                }
            }
            if (numFiles > 0) {
                InputStream sis = socket.getInputStream();
                gp = GPacket.getInstance();
                gp.read(sis);
                if (gp.getType() != ProtocolGlobals.G_TRANSFER_FILE_END_ACK) {
                    String[] args = { ProtocolGlobals.getPacketTypeDisplayString(gp.getType()), to,
                            ProtocolGlobals.getPacketTypeDisplayString(ProtocolGlobals.G_TRANSFER_FILE_END_ACK) };
                    String emsg = br.getKString(br.E_CLUSTER_UNEXPECTED_PACKET_FROM_1, args);
                    logger.log(logger.ERROR, emsg);
                    throw new BrokerException(emsg);
                }
                int status = ClusterTransferFileEndInfo.getReplyStatus(gp);
                if (status != Status.OK) {
                    throw new BrokerException(ClusterTransferFileEndInfo.getReplyStatusReason(gp));
                }
                try {
                    gp = ClusterTransferFileEndInfo.getReplyAckGPacket(Status.OK, null);
                    gp.write(os);
                    os.flush();
                } catch (Throwable t) {
                    if (DEBUG) {
                        logger.log(logger.INFO, "Exception in sending "
                                + ProtocolGlobals.getPacketTypeDisplayString(ProtocolGlobals.G_TRANSFER_FILE_END_ACK_ACK) + " to " + to);
                    }
                }
                try {
                    sis.close();
                } catch (Exception e) {
                }
            }
            try {
                socket.close();
            } catch (Exception e) {
            }

        } catch (Throwable t) {
            if (socket != null) {
                try {
                    socket.close();
                } catch (Exception e) {
                }
            }
            StringBuilder buf = new StringBuilder();
            if (fileNames != null) {
                int len = fileNames.length;
                for (int i = 0; i < len; i++) {
                    if (i > 0) {
                        buf.append(", ");
                    }
                    buf.append(fileNames[i]);
                }
            }
            String emsg = br.getKString(br.X_CLUSTER_FILE_TX_FAIL,
                    "[" + buf.toString() + "[" + uuid + "]]: " + ((t instanceof java.io.EOFException) ? "peer socket closed" : t.getMessage()), to);
            logger.log(logger.ERROR, emsg);
            if (t instanceof BrokerException) {
                throw (BrokerException) t;
            }
            throw new BrokerException(emsg, t);
        }

    }
}

class ClusterServiceListener extends Thread {
    private static final BrokerResources br = Globals.getBrokerResources();
    private static final Logger logger = Globals.getLogger();
    ClusterImpl callback = null;
    boolean done = false;
    ServerSocket ss = null;
    private boolean nodelay;
    private boolean isSSL = false;

    private static ServerSocketFactory ssf = MQServerSocketFactory.getDefault();

    public ClusterServiceListener(ClusterImpl callback) throws IOException {
        this.callback = callback;
        setName("ClusterServiceListener");
        setDaemon(true);
        this.nodelay = callback.getTCPNodelay();

        if (callback.getTransport().equalsIgnoreCase("ssl")) {
            nodelay = callback.getSSLNodelay();
            isSSL = true;
            initSSLListener();
        } else {
            initTCPListener();
        }

        start();
    }

    private void initSSLListener() throws IOException {
        if (ClusterImpl.DEBUG) {
            logger.log(logger.INFO, "ClusterImpl.initSSLListener[nodelay=" + nodelay + ", inbufsz=" + callback.getSSLInputBufferSize() + ", outbufsz="
                    + callback.getSSLOutputBufferSize() + "]");
        }

        ServerSocketFactory sslfactory = null;
        try {
            Class TLSProtocolClass = Class.forName("com.sun.messaging.jmq.jmsserver.net.tls.TLSProtocol");

            if (ClusterImpl.DEBUG) {
                logger.log(logger.DEBUG, "ClusterImpl.initSSLListener. " + "Initializing SSLServerSocketFactory");
            }

            /*
             * SSLServerSocketFactory ssf = (SSLServerSocketFactory) TLSProtocol.getServerSocketFactory();
             */
            java.lang.reflect.Method m = TLSProtocolClass.getMethod("getServerSocketFactory", (Class[]) null);
            sslfactory = (ServerSocketFactory) m.invoke(null, (Object[]) null);
        } catch (Exception e) {
            Throwable t = e;
            if (e instanceof java.lang.reflect.InvocationTargetException) {
                t = e.getCause();
                if (t == null) {
                    t = e;
                }
                if (ClusterImpl.DEBUG && t != e) {
                    logger.logStack(Logger.ERROR, e.getMessage(), e);
                }
            }
            logger.logStack(Logger.ERROR, t.getMessage(), t);
            throw new IOException(t.getMessage());
        }

        InetAddress listenHost = callback.getListenHost();
        int listenPort = callback.getListenPort();
        HashMap h = null;

        if (ClusterImpl.DEBUG) {
            logger.log(logger.DEBUG, "ClusterImpl.initSSLListener. " + "Initializing ServerSocket");
        }

        if (listenHost == null) {
            ss = sslfactory.createServerSocket(listenPort);
        } else {
            ss = sslfactory.createServerSocket(listenPort, 50, listenHost);
            // Why backlog = 50? According the JDK 1.4 javadoc,
            // that's the default value for ServerSocket().
            // Also even if a connection gets refused, that broker
            // will try again after sometime anyway...

            h = new HashMap();
            h.put("hostname", listenHost.getHostName());
            h.put("hostaddr", listenHost.getHostAddress());
        }

        if (Globals.getPoodleFixEnabled()) {
            Globals.applyPoodleFix(ss, "ClusterListener");
        }
        Globals.getPortMapper().addService(ClusterImpl.SERVICE_NAME, "ssl", ClusterImpl.SERVICE_TYPE, ss.getLocalPort(), h);

        if (ClusterImpl.DEBUG) {
            logger.log(logger.INFO, "ClusterImpl.initSSLListener: " + ss + " " + MQServerSocketFactory.serverSocketToString(ss));
        }
    }

    private void initTCPListener() throws IOException {
        if (ClusterImpl.DEBUG) {
            logger.log(logger.INFO, "ClusterImpl.initTCPListener[TcpNoDelay=" + nodelay + ", inbufsz=" + callback.getTCPInputBufferSize() + ", outbufsz="
                    + callback.getTCPOutputBufferSize() + "]");
        }

        InetAddress listenHost = callback.getListenHost();
        int listenPort = callback.getListenPort();
        HashMap h = null;

        if (listenHost == null) {
            ss = ssf.createServerSocket(listenPort);
        } else {
            ss = ssf.createServerSocket(listenPort, 50, listenHost);
            // Why backlog = 50? According the JDK 1.4 javadoc,
            // that's the default value for ServerSocket().
            // Also even if a connection gets refused, that broker
            // will try again after sometime anyway...

            h = new HashMap();
            h.put("hostname", listenHost.getHostName());
            h.put("hostaddr", listenHost.getHostAddress());
        }

        Globals.getPortMapper().addService(ClusterImpl.SERVICE_NAME, "tcp", ClusterImpl.SERVICE_TYPE, ss.getLocalPort(), h);

        if (ClusterImpl.DEBUG) {
            logger.log(logger.DEBUG, "ClusterImpl.initTCPListener: " + ss + " " + MQServerSocketFactory.serverSocketToString(ss));
        }
    }

    public String getServerSocketString() {
        ServerSocket ssocket = ss;
        if (ssocket != null) {
            return ssocket.getInetAddress() + ":" + ssocket.getLocalPort();
        }
        return null;
    }

    public synchronized void shutdown() {
        done = true;
        try {
            ss.close();
        } catch (Exception e) { /* Ignore */
            /* Ignore. This happens when ServerSocket is closed.. */
        }
    }

    private synchronized boolean isDone() {
        return done;
    }

    @Override
    public void run() {
        String oomstr = br.getKString(br.M_LOW_MEMORY_CLUSTER);
        while (true) {
            if (isDone()) {
                break;
            }
            Socket sock = null;
            try {
                sock = ss.accept();
                try {
                    sock.setTcpNoDelay(nodelay);
                } catch (SocketException e) {
                    logger.log(Logger.WARNING, getClass().getSimpleName() + ".run(): [" + sock.toString() + "]setTcpNoDelay(" + nodelay + "): " + e.toString(),
                            e);
                }
                callback.acceptConnection(sock, isSSL);
            } catch (Exception e) {
                /* Ignore. This happens when ServerSocket is closed.. */
            } catch (OutOfMemoryError e) {
                if (isDone()) {
                    return;
                }
                try {
                    if (sock != null) {
                        sock.close();
                    }
                } catch (Throwable t) {
                } finally {
                    logger.log(Logger.WARNING, oomstr);
                }
                try {
                    Thread.sleep(1000L);
                } catch (InterruptedException ee) {
                }
            }
        }
    }
}

class WarningTask extends TimerTask {
    private static final Logger logger = Globals.getLogger();
    private static final BrokerResources br = Globals.getBrokerResources();
    private ClusterImpl parent = null;

    public WarningTask(ClusterImpl parent) {
        this.parent = parent;
    }

    @Override
    public void run() {
        if (parent.isConfigServerResolved()) {
            cancel();
        } else {
            if (Globals.nowaitForMasterBroker()) {
                logger.log(Logger.WARNING, br.W_MBUS_STILL_TRYING_NOWAIT, parent.getConfiguredConfigServer());
            } else {
                logger.log(Logger.WARNING, br.W_MBUS_STILL_TRYING, parent.getConfiguredConfigServer());
            }
        }
    }
}

class FileTransferRunnable implements Runnable {

    Logger logger = Globals.getLogger();
    BrokerResources br = Globals.getBrokerResources();

    Socket socket = null;
    BrokerAddressImpl remote = null;
    MessageDigest digest = null;
    int timeout = 0;
    ExecutorService es = null;
    ClusterImpl parent = null;

    public FileTransferRunnable(Socket conn, int timeout, BrokerAddressImpl remote, ExecutorService es, ClusterImpl parent) throws BrokerException {
        this.socket = conn;
        this.remote = remote;
        this.timeout = timeout;
        this.es = es;
        this.parent = parent;
        try {
            digest = MessageDigest.getInstance("SHA1");
        } catch (Exception e) {
            es.shutdownNow();
            String emsg = "";
            if (e instanceof NoSuchAlgorithmException) {
                emsg = "Unable to create MessageDigest for file transfer";
                logger.log(logger.ERROR, emsg + ": " + e);
            } else {
                emsg = "Unexpectd exception in creating MessageDigest for file transfer";
                logger.logStack(logger.ERROR, emsg, e);
            }
            throw new BrokerException(emsg, e);
        }
    }

    @Override
    public void run() {
        String from = remote + "[" + socket.getInetAddress() + "]";
        try {
            socket.setTcpNoDelay(true);
        } catch (Exception e) {
            logger.log(logger.WARNING, "Failed to set socket TCP_NODELAY in processing file transfer request: " + e);
        }
        try {
            socket.setSoTimeout(timeout);
        } catch (Exception e) {
            logger.log(logger.WARNING, "Failed to set socket timeout in processing file transfer request: " + e);
        }
        try {
            InputStream is = socket.getInputStream();
            GPacket gp = GPacket.getInstance();
            gp.read(is);
            if (gp.getType() != ProtocolGlobals.G_TRANSFER_FILE_LIST) {
                String[] emsg = { ProtocolGlobals.getPacketTypeDisplayString(gp.getType()), from,
                        ProtocolGlobals.getPacketTypeDisplayString(ProtocolGlobals.G_TRANSFER_FILE_LIST) };
                logger.log(logger.ERROR, br.getKString(br.E_CLUSTER_UNEXPECTED_PACKET_FROM_1, emsg));
                try {
                    socket.close();
                } catch (Exception e) {
                    /* ignore */
                }
                return;
            }
            ClusterTransferFileListInfo tfl = ClusterTransferFileListInfo.newInstance(gp);
            String uuid = tfl.getUUID();
            String pendingUUID = parent.pendingFileTransfers.get(remote);
            if (pendingUUID == null || !pendingUUID.startsWith(tfl.getUUID())) {
                logger.log(logger.ERROR,
                        br.getKString(br.E_CLUSTER_RECEIVED_UNKNOW_FILE_TX_LIST, ProtocolGlobals.getPacketTypeDisplayString(gp.getType()) + tfl, remote));
                try {
                    socket.close();
                } catch (Exception e) {
                }
                return;
            }
            if (!tfl.getModule().equals(FileTransferCallback.STORE)) {
                logger.log(logger.ERROR, br.getKString(br.E_CLUSTER_UNEXPECTED_PACKET_FROM,
                        ProtocolGlobals.getPacketTypeDisplayString(gp.getType()) + tfl.toString(true), from));
                try {
                    socket.close();
                } catch (Exception e) {
                }
                return;
            }
            FileTransferCallback callback = (FileTransferCallback) Globals.getStore();

            int numFiles = tfl.getNumFiles();
            for (int cnt = 0; cnt < numFiles; cnt++) {

                if (parent.fileTransferShutdownIn) {
                    String emsg = br.getKString(br.W_CLUSTER_SERVICE_SHUTDOWN);
                    logger.log(logger.WARNING, emsg);
                    throw new BrokerException(emsg);
                }

                gp.read(is);
                if (gp.getType() != ProtocolGlobals.G_TRANSFER_FILE_START) {
                    String[] emsg = { ProtocolGlobals.getPacketTypeDisplayString(gp.getType()), from,
                            ProtocolGlobals.getPacketTypeDisplayString(ProtocolGlobals.G_TRANSFER_FILE_START) };
                    logger.log(logger.ERROR, br.getKString(br.E_CLUSTER_UNEXPECTED_PACKET_FROM_1, emsg));
                    try {
                        socket.close();
                    } catch (Exception e) {
                    }
                    return;
                }
                ClusterTransferFileStartInfo tfs = ClusterTransferFileStartInfo.newInstance(gp);
                if (!tfs.getModule().equals(FileTransferCallback.STORE)) {
                    logger.log(logger.ERROR, br.getKString(br.E_CLUSTER_UNEXPECTED_PACKET_FROM,
                            ProtocolGlobals.getPacketTypeDisplayString(gp.getType()) + tfs.toString(true), from));
                    try {
                        socket.close();
                    } catch (Exception e) {
                    }
                    return;
                }
                if (!tfs.getBrokerID().equals(tfl.getBrokerID())) {
                    logger.log(logger.ERROR, br.getKString(br.E_CLUSTER_UNEXPECTED_PACKET_FROM,
                            ProtocolGlobals.getPacketTypeDisplayString(gp.getType()) + tfs.toString(true), from));
                    try {
                        socket.close();
                    } catch (Exception e) {
                    }
                    return;
                }
                String tmpfile = tfs.getFileName() + ".tmp";
                ClusterTransferFileEndInfo fte = null;
                boolean success = false;

                OutputStream os = callback.getFileOutputStream(tmpfile, tfs.getBrokerID(), uuid, cnt == 0, remote);
                try {
                    digest.reset();
                    long size = tfs.getFileSize();
                    byte[] buf = new byte[parent.FILE_TRANSFER_CHUNK_SIZE];
                    int totalread = 0;
                    int count;
                    while (totalread < size) {
                        if (parent.fileTransferShutdownIn) {
                            String emsg = br.getKString(br.W_CLUSTER_SERVICE_SHUTDOWN);
                            logger.log(logger.WARNING, emsg);
                            throw new BrokerException(emsg);
                        }
                        count = 0;
                        try {
                            count = is.read(buf, 0, (int) Math.min(parent.FILE_TRANSFER_CHUNK_SIZE, (size - totalread)));
                        } catch (IOException e) {
                            logger.log(logger.ERROR, br.getKString(br.E_CLUSTER_FILE_TX_READ, from + ": " + e));
                            try {
                                socket.close();
                            } catch (Exception e1) {
                            }
                            return;
                        }
                        if (count < 0) {
                            String[] args = { String.valueOf(totalread), String.valueOf(size - totalread), from };
                            logger.log(logger.ERROR, br.getKString(br.E_CLUSTER_FILE_TX_EOF, args));
                            try {
                                socket.close();
                            } catch (Exception e) {
                            }
                            return;
                        }
                        totalread += count;
                        os.write(buf, 0, count);
                        digest.update(buf, 0, count);
                    }
                    String[] args = { String.valueOf(size), tfs.getFileName(), from };
                    logger.log(logger.INFO, br.getKString(br.I_FILE_TX_COMPLETE, args));

                    gp = GPacket.getInstance();
                    gp.read(is);
                    if (gp.getType() != ProtocolGlobals.G_TRANSFER_FILE_END) {
                        String[] emsg = { ProtocolGlobals.getPacketTypeDisplayString(gp.getType()), from,
                                ProtocolGlobals.getPacketTypeDisplayString(ProtocolGlobals.G_TRANSFER_FILE_END) };
                        logger.log(logger.ERROR, br.getKString(br.E_CLUSTER_UNEXPECTED_PACKET_FROM_1, emsg));
                        try {
                            socket.close();
                        } catch (Exception e) {
                        }
                        return;
                    }
                    fte = ClusterTransferFileEndInfo.newInstance(gp);
                    if (!Arrays.equals(digest.digest(), fte.getDigest())) {
                        logger.log(logger.ERROR, br.getKString(br.E_CLUSTER_FILE_TX_DIGEST_MISMATCH, tfs.getFileName(), from));
                        try {
                            socket.close();
                        } catch (Exception e) {
                        }
                        return;
                    }
                    os.close();
                    logger.log(logger.INFO, br.getKString(br.I_FILE_TX_SUCCESS, tfs.getFileName(), from));
                    callback.doneTransfer(tmpfile, tfs.getFileName(), tfs.getBrokerID(), tfs.getLastModifiedTime(), true, remote);
                    success = true;

                } finally {
                    if (!success) {// cleanup
                        try {
                            os.close();
                        } catch (Exception e) {
                        }
                        callback.doneTransfer(tmpfile, tfs.getFileName(), tfs.getBrokerID(), tfs.getLastModifiedTime(), false, remote);
                    }
                }

                if (fte.hasMoreFiles() != ((cnt + 1) < numFiles)) {
                    String[] args = { String.valueOf(numFiles), from, String.valueOf(cnt + 1), String.valueOf(fte.hasMoreFiles()) };
                    String emsg = br.getKString(br.E_CLUSTER_FILE_TX_NUMFILES, args);
                    logger.log(logger.ERROR, emsg);
                    throw new BrokerException(emsg);
                }

            } // for

            if (numFiles > 0) {
                callback.allDoneTransfer(tfl.getBrokerID(), tfl.getUUID(), remote);
                OutputStream sos = socket.getOutputStream();
                gp = ClusterTransferFileEndInfo.getReplyGPacket(Status.OK, (String) null);
                gp.write(sos);
                sos.flush();
                try {
                    gp = GPacket.getInstance();
                    gp.read(is);
                    if (gp.getType() != ProtocolGlobals.G_TRANSFER_FILE_END_ACK_ACK) {
                        String[] args = { ProtocolGlobals.getPacketTypeDisplayString(gp.getType()), from,
                                ProtocolGlobals.getPacketTypeDisplayString(ProtocolGlobals.G_TRANSFER_FILE_END_ACK_ACK) };
                        String emsg = br.getKString(br.E_CLUSTER_UNEXPECTED_PACKET_FROM_1, args);
                        logger.log(logger.ERROR, emsg);
                        throw new BrokerException(emsg);
                    }
                } catch (Throwable t) {
                    if (parent.DEBUG) {
                        logger.log(logger.INFO,
                                "Exception in receiving " + ProtocolGlobals.getPacketTypeDisplayString(ProtocolGlobals.G_TRANSFER_FILE_END_ACK_ACK) + " from "
                                        + from + " for tranfer files of broker " + tfl.getBrokerID() + ": " + t.getMessage());
                    }
                }
                try {
                    is.close();
                } catch (Exception e) {
                }
                try {
                    sos.close();
                } catch (Exception e) {
                }
                try {
                    socket.close();
                } catch (Exception e) {
                }
            }

        } catch (Throwable t) {
            try {
                socket.close();
            } catch (Exception e) {
            }
            logger.logStack(logger.ERROR, br.getKString(br.E_CLUSTER_PROCESS_FILE_TX_REQUEST, from), t);
        } finally {
            es.shutdownNow();
        }
    }
}

