예제 #1
0
  /**
   * Connection will be closed [onlyClose()] and onDisconnect() method will be executed on another
   * thread [DisconnectionThreadPool] after getDisconnectionDelay() time in ms. This method may only
   * be called by current Dispatcher Thread.
   *
   * @param con
   */
  protected final void closeConnectionImpl(AConnection con) {
    /**
     * Test if this build should use assertion. If NetworkAssertion == false javac will remove this
     * code block
     */
    if (Assertion.NetworkAssertion) assert Thread.currentThread() == this;

    if (con.onlyClose())
      dcPool.scheduleDisconnection(new DisconnectionTask(con), con.getDisconnectionDelay());
  }
예제 #2
0
  /**
   * Parse data from buffer and prepare buffer for reading just one packet - call
   * processData(ByteBuffer b).
   *
   * @param con Connection
   * @param buf Buffer with packet data
   * @return True if packet was parsed.
   */
  private boolean parse(AConnection con, ByteBuffer buf) {
    short sz = 0;
    try {
      sz = buf.getShort();
      if (sz > 1) sz -= 2;
      ByteBuffer b = (ByteBuffer) buf.slice().limit(sz);
      b.order(ByteOrder.LITTLE_ENDIAN);
      /** read message fully */
      buf.position(buf.position() + sz);

      return con.processData(b);
    } catch (IllegalArgumentException e) {
      log.warn(
          "Error on parsing input from client - account: "
              + con
              + " packet size: "
              + sz
              + " real size:"
              + buf.remaining(),
          e);
      return false;
    }
  }
예제 #3
0
  /**
   * Write as much as possible data to socketChannel represented by SelectionKey key. If all data
   * were written key write interest will be disabled.
   *
   * @param key
   */
  final void write(SelectionKey key) {
    SocketChannel socketChannel = (SocketChannel) key.channel();
    AConnection con = (AConnection) key.attachment();

    int numWrite;
    ByteBuffer wb = con.writeBuffer;
    /** We have not writted data */
    if (wb.hasRemaining()) {
      try {
        numWrite = socketChannel.write(wb);
      } catch (IOException e) {
        closeConnectionImpl(con);
        return;
      }

      if (numWrite == 0) {
        log.info("Write " + numWrite + " ip: " + con.getIP());
        return;
      }

      /** Again not all data was send */
      if (wb.hasRemaining()) return;
    }

    while (true) {
      wb.clear();
      boolean writeFailed = !con.writeData(wb);

      if (writeFailed) {
        wb.limit(0);
        break;
      }

      /** Attempt to write to the channel */
      try {
        numWrite = socketChannel.write(wb);
      } catch (IOException e) {
        closeConnectionImpl(con);
        return;
      }

      if (numWrite == 0) {
        log.info("Write " + numWrite + " ip: " + con.getIP());
        return;
      }

      /** not all data was send */
      if (wb.hasRemaining()) return;
    }

    /**
     * Test if this build should use assertion. If NetworkAssertion == false javac will remove this
     * code block
     */
    if (Assertion.NetworkAssertion) {
      assert !wb.hasRemaining();
    }

    /** We wrote away all data, so we're no longer interested in writing on this socket. */
    key.interestOps(key.interestOps() & ~SelectionKey.OP_WRITE);

    /** We wrote all data so we can close connection that is "PandingClose" */
    if (con.isPendingClose()) closeConnectionImpl(con);
  }
예제 #4
0
 /**
  * Register new client connected to this Dispatcher and set SelectionKey (result of registration)
  * as this key of given AConnection.
  *
  * @param ch
  * @param ops
  * @param att
  * @throws IOException
  */
 public final void register(SelectableChannel ch, int ops, AConnection att) throws IOException {
   synchronized (gate) {
     selector.wakeup();
     att.setKey(ch.register(selector, ops, att));
   }
 }