private final void disconnectTheOtherGuy(PyroClient client) {
    PyroClient otherGuy = (PyroClient) client.attachment();

    // also disconnect the other guy
    if (otherGuy != null && !otherGuy.isDisconnected()) {
      client.attach(null);
      otherGuy.shutdown();
    }

    // note that you don't know whether you are the
    // other guy, or the other guy is! if either end
    // disconnects, the _other_ guy disconnects too.
  }
  @Override
  public void receivedData(PyroClient client, ByteBuffer data) {
    client.selector().checkThread();

    // send what you received to the other guy
    PyroClient otherGuy = (PyroClient) client.attachment();

    System.out.println(
        "traffic:" + client + " => " + otherGuy + " (" + data.remaining() + " bytes)");

    try {
      otherGuy.writeCopy(data);
    } catch (IllegalStateException exc) {
      // boo
    }
  }
  @Override
  public void acceptedClient(PyroClient client) {
    System.out.println(
        "acceptedClient[client-thread="
            + Thread.currentThread().getName()
            + ", server-thread="
            + client.getServer().selector().networkThread().getName()
            + "]:"
            + client);

    InetSocketAddress src = client.getLocalAddress();
    InetSocketAddress dst = null;

    for (Duo<InetSocketAddress> duo : this.srcToDst) {
      if (duo.first().equals(src)) {
        dst = duo.second();
        break;
      }
    }

    System.out.println(
        "Redirecting: "
            + src.getAddress().getHostAddress()
            + " => "
            + ((dst == null) ? "*" : dst.getAddress().getHostAddress()));

    if (dst == null) {
      client.dropConnection();
      return;
    }

    try {
      // make a connection to the other guy
      PyroClient otherGuy = client.selector().connect(dst);

      client.addListener(this);
      otherGuy.addListener(this);

      client.attach(otherGuy);
      otherGuy.attach(client);
    } catch (IOException exc) {
      exc.printStackTrace();
    }
  }
Example #4
0
 public PyroByteSinkFeeder(PyroClient client) {
   this(client.selector());
 }