279 lines
7.5 KiB
Java
279 lines
7.5 KiB
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 final int selectNow()
|
|
throws IOException
|
|
{
|
|
return select (1);
|
|
}
|
|
|
|
public final int select()
|
|
throws IOException
|
|
{
|
|
return select (-1);
|
|
}
|
|
|
|
// A timeout value of -1 means block forever.
|
|
private static native int implSelect (int[] read, int[] write,
|
|
int[] except, long timeout);
|
|
|
|
private final 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.getNativeFD();
|
|
counter++;
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
public int select (long timeout)
|
|
{
|
|
if (!isOpen())
|
|
throw new ClosedSelectorException ();
|
|
|
|
if (keys == null)
|
|
{
|
|
return 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
|
|
int anzahl = read.length + write.length + except.length;
|
|
|
|
// Call the native select() on all file descriptors.
|
|
begin();
|
|
int result = implSelect (read, write, except, timeout);
|
|
end();
|
|
|
|
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.getNativeFD() == 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.getNativeFD() == 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))
|
|
{
|
|
selected.add (key);
|
|
}
|
|
|
|
// Set new ready ops
|
|
key.readyOps (key.interestOps () & ops);
|
|
}
|
|
|
|
deregisterCancelledKeys();
|
|
return result;
|
|
}
|
|
|
|
public final Set selectedKeys()
|
|
{
|
|
return selected;
|
|
}
|
|
|
|
public final Selector wakeup()
|
|
{
|
|
return null;
|
|
}
|
|
|
|
private final 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 final SelectionKey register (AbstractSelectableChannel ch, int ops,
|
|
Object att)
|
|
{
|
|
SelectionKeyImpl result;
|
|
|
|
if (ch instanceof SocketChannelImpl)
|
|
{
|
|
SocketChannelImpl sc = (SocketChannelImpl) ch;
|
|
result = new SocketChannelSelectionKey (ch, this);
|
|
}
|
|
else if (ch instanceof DatagramChannelImpl)
|
|
{
|
|
DatagramChannelImpl dc = (DatagramChannelImpl) ch;
|
|
result = new DatagramChannelSelectionKey (ch, this);
|
|
}
|
|
else if (ch instanceof ServerSocketChannelImpl)
|
|
{
|
|
ServerSocketChannelImpl ssc = (ServerSocketChannelImpl) ch;
|
|
result = new ServerSocketChannelSelectionKey (ch, this);
|
|
}
|
|
else
|
|
{
|
|
throw new InternalError ("No known channel type");
|
|
}
|
|
|
|
keys.add (result);
|
|
result.interestOps (ops);
|
|
result.attach (att);
|
|
return result;
|
|
}
|
|
}
|