MysqlClientConnectionManager.java

/**
 * <pre>
 * 	This program is free software; you can redistribute it and/or modify it under the terms of 
 * the GNU General Public License as published by the Free Software Foundation; either version 3 of the License, 
 * or (at your option) any later version. 
 * 
 * 	This program is distributed in the hope that it will be useful, 
 * but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  
 * See the GNU General Public License for more details. 
 * 	You should have received a copy of the GNU General Public License along with this program; 
 * if not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 * </pre>
 */
package com.meidusa.amoeba.mysql.net;

import java.io.IOException;

import com.meidusa.amoeba.mysql.context.MysqlProxyRuntimeContext;
import com.meidusa.amoeba.mysql.handler.MySqlCommandDispatcher;
import com.meidusa.amoeba.mysql.net.packet.ErrorPacket;
import com.meidusa.amoeba.mysql.net.packet.HandshakePacket;
import com.meidusa.amoeba.mysql.net.packet.OkPacket;
import com.meidusa.amoeba.net.AuthResponseData;
import com.meidusa.amoeba.net.Connection;
import com.meidusa.amoeba.net.ServerableConnectionManager;
import com.meidusa.amoeba.util.StringUtil;

/**
 * @author <a href=mailto:piratebase@sina.com>Struct chen</a>
 */
public class MysqlClientConnectionManager extends ServerableConnectionManager {

    private final static String SERVER_VERSION = "5.1.22-mysql-amoeba-proxy";

    private static byte[]       authenticateOkPacketData;

    public MysqlClientConnectionManager() throws IOException{
    }

    public MysqlClientConnectionManager(String name, int port) throws IOException{
        super(name, port);
    }
    /**
     * 
     * @param name "Aladdin proxy Server"
     * @param ipAddress Aladdin proxy��ip
     * @param port Aladdin proxy��ip port
     * @throws IOException
     */
    public MysqlClientConnectionManager(String name, String ipAddress, int port) throws IOException{
        super(name, ipAddress, port);
    }

    /**
     * ���ͷ���������Ϣ��������������ܵ�����ַ���
     */
    protected void beforeAuthing(final Connection authing) {
        HandshakePacket handshakePacket = new HandshakePacket();
        handshakePacket.packetId = 0;
        handshakePacket.protocolVersion = 0x0a;// Э��汾10
        handshakePacket.seed = StringUtil.getRandomString(8);
        handshakePacket.restOfScrambleBuff = StringUtil.getRandomString(12);

        handshakePacket.serverStatus = 2;
        handshakePacket.serverVersion = SERVER_VERSION;
        handshakePacket.serverCapabilities = 41516;

        MysqlProxyRuntimeContext context = ((MysqlProxyRuntimeContext) MysqlProxyRuntimeContext.getInstance());
        handshakePacket.serverCharsetIndex = (byte) (context.getServerCharsetIndex() & 0xff);
        handshakePacket.threadId = Thread.currentThread().hashCode();
        MysqlClientConnection aconn = (MysqlClientConnection) authing;
        aconn.setSeed(handshakePacket.seed + handshakePacket.restOfScrambleBuff);
        aconn.postMessage(handshakePacket.toByteBuffer(authing).array());
    }

    public void connectionAuthenticateSuccess(Connection conn, AuthResponseData data) {
        super.connectionAuthenticateSuccess(conn, data);
        if (authenticateOkPacketData == null) {
            OkPacket ok = new OkPacket();
            ok.packetId = 2;
            ok.affectedRows = 0;
            ok.insertId = 0;
            ok.serverStatus = 2;
            ok.warningCount = 0;
            authenticateOkPacketData = ok.toByteBuffer(conn).array();
        }
        conn.setMessageHandler(new MySqlCommandDispatcher());
        conn.postMessage(authenticateOkPacketData);
    }

    protected void connectionAuthenticateFaild(final Connection conn, AuthResponseData data) {
        super.connectionAuthenticateFaild(conn, data);
        ErrorPacket error = new ErrorPacket();
        error.resultPacketType = ErrorPacket.PACKET_TYPE_ERROR;
        error.packetId = 2;
        error.serverErrorMessage = data.message;
        error.sqlstate = "42S02";
        error.errno = 1000;
        conn.postMessage(error.toByteBuffer(conn).array());
    }

}