Java tutorial
/* * LumaQQ - Java QQ Client * * Copyright (C) 2004 notXX * luma <stubma@163.com> * * 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 2 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 */ package edu.tsinghua.lumaqq.qq.net; import java.io.IOException; import java.net.InetSocketAddress; import java.net.UnknownHostException; import java.nio.ByteBuffer; import java.nio.channels.SelectableChannel; import java.nio.channels.SelectionKey; import java.nio.channels.SocketChannel; import java.nio.channels.UnresolvedAddressException; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import edu.tsinghua.lumaqq.qq.QQ; import edu.tsinghua.lumaqq.qq.packets.ErrorPacket; import edu.tsinghua.lumaqq.qq.packets.InPacket; import edu.tsinghua.lumaqq.qq.packets.OutPacket; import edu.tsinghua.lumaqq.qq.packets.PacketParseException; /** * TCP?QQ?. * TCP?????, QQ??. * , TCPQQ???. ???????, ???. * * @author notxx * @author luma */ public final class TCPPort extends AbstractPort { /** Log */ private static final Log log = LogFactory.getLog(TCPPort.class); /** channel */ private final SocketChannel channel; /** * true? */ private boolean remoteClosed; /** * ?TCPPort. * * @param address ?. * @throws IOException ?/??/?. */ public TCPPort(IConnectionPolicy policy, InetSocketAddress address) throws IOException { super(policy); channel = SocketChannel.open(); channel.configureBlocking(false); this.remoteAddress = address; remoteClosed = false; } /* (non-Javadoc) * @see edu.tsinghua.lumaqq.qq.net.IPort#start() */ public void start() { try { channel.connect(remoteAddress); } catch (UnknownHostException e) { log.error("??"); processError(new Exception("Unknown Host")); } catch (UnresolvedAddressException e) { log.error("???"); processError(new Exception("Unable to resolve server address")); } catch (IOException e) { log.error(""); processError(e); } } /* (non-Javadoc) * @see edu.tsinghua.lumaqq.qq.IPort#channel() */ public SelectableChannel channel() { return channel; } /* (non-Javadoc) * @see edu.tsinghua.lumaqq.qq.IPort#receive() */ public void receive() throws IOException, PacketParseException { if (remoteClosed) return; //? int oldPos = receiveBuf.position(); for (int r = channel.read(receiveBuf); r > 0; r = channel.read(receiveBuf)) ; // ?? int pos = receiveBuf.position(); receiveBuf.flip(); // ?0?? if (oldPos == pos) { ErrorPacket packet = policy.createErrorPacket(ErrorPacket.ERROR_CONNECTION_BROKEN, getId()); policy.pushIn(packet); remoteClosed = true; return; } // ? while (true) { /* ?? */ // ? InPacket packet = null; try { packet = policy.parseIn(receiveBuf, false); } catch (PacketParseException e) { adjustBuffer(pos); throw e; } if (packet == null) { /* * packetnull???? * ????parser? * ??? */ if (!policy.relocate(receiveBuf)) break; } policy.pushIn(packet); } adjustBuffer(pos); } /** * buffer * * @param pos */ private void adjustBuffer(int pos) { // 0??pos? if (receiveBuf.position() > 0) { receiveBuf.compact(); receiveBuf.limit(receiveBuf.capacity()); } else { receiveBuf.limit(receiveBuf.capacity()); receiveBuf.position(pos); } } /* (non-Javadoc) * @see edu.tsinghua.lumaqq.qq.IPort#send() */ public void send() throws IOException { while (!isEmpty()) { sendBuf.clear(); OutPacket packet = remove(); packet.fill(sendBuf); sendBuf.flip(); if (packet.needAck()) { channel.write(sendBuf); // ?? packet.setTimeout(System.currentTimeMillis() + QQ.QQ_TIMEOUT_SEND); policy.pushResend(packet, getId()); log.debug("?? - " + packet.toString()); } else { int count = packet.getSendCount(); for (int i = 0; i < count; i++) { sendBuf.rewind(); channel.write(sendBuf); log.debug("?? - " + packet.toString()); } } } } /* (non-Javadoc) * @see edu.tsinghua.lumaqq.qq.net.IPort#send(edu.tsinghua.lumaqq.qq.packets.OutPacket) */ public void send(OutPacket packet) { try { sendBuf.clear(); packet.fill(sendBuf); sendBuf.flip(); if (packet.needAck()) { channel.write(sendBuf); log.debug("?? - " + packet.toString()); } else { int count = packet.getSendCount(); for (int i = 0; i < count; i++) { sendBuf.rewind(); channel.write(sendBuf); log.debug("?? - " + packet.toString()); } } } catch (Exception e) { log.error(e.getMessage()); } } /* (non-Javadoc) * @see edu.tsinghua.lumaqq.qq.IPort#send(java.nio.ByteBuffer) */ public void send(ByteBuffer buffer) { try { channel.write(buffer); } catch (IOException e) { log.error(e.getMessage()); } } /* (non-Javadoc) * @see edu.tsinghua.lumaqq.qq.AbstractPort#dispose() */ public void dispose() { try { channel.close(); } catch (IOException e) { } } /* (non-Javadoc) * @see edu.tsinghua.lumaqq.qq.IPort#isConnected() */ public boolean isConnected() { return channel != null && channel.isConnected(); } /* (non-Javadoc) * @see edu.tsinghua.lumaqq.qq.IHandler#processConnect(java.nio.channels.SelectionKey) */ public void processConnect(SelectionKey sk) throws IOException { //?SocketChannel channel.finishConnect(); while (!channel.isConnected()) { try { Thread.sleep(300); } catch (InterruptedException e) { // ?? } channel.finishConnect(); } sk.interestOps(SelectionKey.OP_READ); log.debug("QQ?"); } /* (non-Javadoc) * @see edu.tsinghua.lumaqq.qq.IHandler#processRead(java.nio.channels.SelectionKey) */ public void processRead(SelectionKey sk) throws IOException, PacketParseException { receive(); } /* (non-Javadoc) * @see edu.tsinghua.lumaqq.qq.IHandler#processWrite() */ public void processWrite() throws IOException { if (isConnected()) send(); } }