gcc/libjava/gnu/java/nio/SelectorImpl.java

/* SelectorImpl.java -- 
   Copyright (C) 2002, 2003  Free Software Foundation, Inc.

This file is part of GNU Classpath.

GNU Classpath 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, or (at your option)
any later version.

GNU Classpath 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 GNU Classpath; see the file COPYING.  If not, write to the
Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
02111-1307 USA.

Linking this library statically or dynamically with other modules is
making a combined work based on this library.  Thus, the terms and
conditions of the GNU General Public License cover the whole
combination.

As a special exception, the copyright holders of this library give you
permission to link this library with independent modules to produce an
executable, regardless of the license terms of these independent
modules, and to copy and distribute the resulting executable under
terms of your choice, provided that you also meet, for each linked
independent module, the terms and conditions of the license of that
module.  An independent module is a module which is not derived from
or based on this library.  If you modify this library, you may extend
this exception to your version of the library, but you are not
obligated to do so.  If you do not wish to do so, delete this
exception statement from your version. */

package gnu.java.nio;

import java.io.IOException;
import java.nio.channels.ClosedSelectorException;
import java.nio.channels.SelectableChannel;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.spi.AbstractSelectableChannel;
import java.nio.channels.spi.AbstractSelector;
import java.nio.channels.spi.SelectorProvider;
import java.util.Collections;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;

public class SelectorImpl extends AbstractSelector
{
  private Set keys;
  private Set selected;

  public SelectorImpl (SelectorProvider provider)
  {
    super (provider);
    
    keys = new HashSet ();
    selected = new HashSet ();
  }

  protected void finalize() throws Throwable
  {
    close();
  }

  protected final void implCloseSelector()
    throws IOException
  {
    // FIXME: We surely need to do more here.
    wakeup();
  }

  public final Set keys()
  {
    return Collections.unmodifiableSet (keys);
  }
    
  public int selectNow ()
  {
    return select (1);
  }

  public int select ()
  {
    return select (-1);
  }

  // A timeout value of -1 means block forever.
  private static native int java_do_select (int[] read, int[] write,
                                            int[] except, long timeout);

  private int[] getFDsAsArray (int ops)
  {
    int[] result;
    int counter = 0;
    Iterator it = keys.iterator ();

    // Count the number of file descriptors needed
    while (it.hasNext ())
      {
        SelectionKeyImpl key = (SelectionKeyImpl) it.next ();

        if ((key.interestOps () & ops) != 0)
          {
            counter++;
          }
      }

    result = new int[counter];

    counter = 0;
    it = keys.iterator ();

    // Fill the array with the file descriptors
    while (it.hasNext ())
      {
        SelectionKeyImpl key = (SelectionKeyImpl) it.next ();

        if ((key.interestOps () & ops) != 0)
          {
            result[counter] = key.fd;
            counter++;
          }
      }

    return result;
  }

  public int select (long timeout)
  {
    if (!isOpen())
      throw new ClosedSelectorException ();

    if (keys == null)
	    {
        return 0;
	    }

    int ret = 0;

    deregisterCancelledKeys();

    // Set only keys with the needed interest ops into the arrays.
    int[] read = getFDsAsArray (SelectionKey.OP_READ | SelectionKey.OP_ACCEPT);
    int[] write = getFDsAsArray (SelectionKey.OP_WRITE | SelectionKey.OP_CONNECT);
    int[] except = new int [0]; // FIXME: We dont need to check this yet

    // Call the native select () on all file descriptors.
    int anzahl = read.length + write.length + except.length;
    ret = java_do_select (read, write, except, timeout);

    Iterator it = keys.iterator ();

    while (it.hasNext ())
      {
        int ops = 0;
        SelectionKeyImpl key = (SelectionKeyImpl) it.next ();

        // If key is already selected retrieve old ready ops.
        if (selected.contains (key))
          {
            ops = key.readyOps ();
          }

        // Set new ready read/accept ops
        for (int i = 0; i < read.length; i++)
          {
            if (key.fd == read[i])
              {
                if (key.channel () instanceof ServerSocketChannelImpl)
                  {
                    ops = ops | SelectionKey.OP_ACCEPT;
                  }
                else
                  {
                    ops = ops | SelectionKey.OP_READ;
                  }
              }
          }

        // Set new ready write ops
        for (int i = 0; i < write.length; i++)
          {
            if (key.fd == write[i])
              {
                ops = ops | SelectionKey.OP_WRITE;
                
//                 if (key.channel ().isConnected ())
//                   {
//                     ops = ops | SelectionKey.OP_WRITE;
//                   }
//                 else
//                   {
//                     ops = ops | SelectionKey.OP_CONNECT;
//                   }
             }
          }

        // FIXME: We dont handle exceptional file descriptors yet.

        // If key is not yet selected add it.
        if (!selected.contains (key))
          {
            add_selected (key);
          }

        // Set new ready ops
        key.readyOps (key.interestOps () & ops);
      }

    deregisterCancelledKeys();
    return ret;
  }
    
  public Set selectedKeys ()
  {
    return selected;
  }

  public Selector wakeup ()
  {
    return null;
  }

  public void add (SelectionKeyImpl k)
  {
    keys.add (k);
  }

  void add_selected (SelectionKeyImpl k)
  {
    selected.add (k);
  }

  private void deregisterCancelledKeys ()
  {
    Iterator it = cancelledKeys().iterator();

    while (it.hasNext ())
      {
        keys.remove ((SelectionKeyImpl) it.next ());
        it.remove ();
      }
  }

  protected SelectionKey register (SelectableChannel ch, int ops, Object att)
  {
    return register ((AbstractSelectableChannel) ch, ops, att);
  }

  protected SelectionKey register (AbstractSelectableChannel ch, int ops,
                                   Object att)
  {
    SelectionKeyImpl result;
    
    if (ch instanceof SocketChannelImpl)
      {
        SocketChannelImpl sc = (SocketChannelImpl) ch;
        result = new SelectionKeyImpl (ch, this, 0); // FIXME: last argument
      }
    else if (ch instanceof DatagramChannelImpl)
      {
        DatagramChannelImpl dc = (DatagramChannelImpl) ch;
        result = new SelectionKeyImpl (ch, this, 0); // FIXME: last argument
      }
    else if (ch instanceof ServerSocketChannelImpl)
      {
        ServerSocketChannelImpl ssc = (ServerSocketChannelImpl) ch;
        result = new SelectionKeyImpl (ch, this, 0); // FIXME: last argument
      }
    else
      {
        throw new InternalError ("No known channel type");
      }

    add (result);
    result.interestOps (ops);
    result.attach (att);
    return result;
  }
}