Source for gnu.java.nio.SocketChannelImpl

   1: /* SocketChannelImpl.java -- 
   2:    Copyright (C) 2002, 2003, 2004, 2006 Free Software Foundation, Inc.
   3: 
   4: This file is part of GNU Classpath.
   5: 
   6: GNU Classpath is free software; you can redistribute it and/or modify
   7: it under the terms of the GNU General Public License as published by
   8: the Free Software Foundation; either version 2, or (at your option)
   9: any later version.
  10: 
  11: GNU Classpath is distributed in the hope that it will be useful, but
  12: WITHOUT ANY WARRANTY; without even the implied warranty of
  13: MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14: General Public License for more details.
  15: 
  16: You should have received a copy of the GNU General Public License
  17: along with GNU Classpath; see the file COPYING.  If not, write to the
  18: Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  19: 02110-1301 USA.
  20: 
  21: Linking this library statically or dynamically with other modules is
  22: making a combined work based on this library.  Thus, the terms and
  23: conditions of the GNU General Public License cover the whole
  24: combination.
  25: 
  26: As a special exception, the copyright holders of this library give you
  27: permission to link this library with independent modules to produce an
  28: executable, regardless of the license terms of these independent
  29: modules, and to copy and distribute the resulting executable under
  30: terms of your choice, provided that you also meet, for each linked
  31: independent module, the terms and conditions of the license of that
  32: module.  An independent module is a module which is not derived from
  33: or based on this library.  If you modify this library, you may extend
  34: this exception to your version of the library, but you are not
  35: obligated to do so.  If you do not wish to do so, delete this
  36: exception statement from your version. */
  37: 
  38: 
  39: package gnu.java.nio;
  40: 
  41: import gnu.java.net.PlainSocketImpl;
  42: 
  43: import java.io.IOException;
  44: import java.io.InputStream;
  45: import java.io.OutputStream;
  46: import java.net.InetSocketAddress;
  47: import java.net.Socket;
  48: import java.net.SocketAddress;
  49: import java.net.SocketTimeoutException;
  50: import java.nio.ByteBuffer;
  51: import java.nio.channels.AlreadyConnectedException;
  52: import java.nio.channels.ClosedChannelException;
  53: import java.nio.channels.ConnectionPendingException;
  54: import java.nio.channels.NoConnectionPendingException;
  55: import java.nio.channels.NotYetConnectedException;
  56: import java.nio.channels.SelectionKey;
  57: import java.nio.channels.Selector;
  58: import java.nio.channels.SocketChannel;
  59: import java.nio.channels.UnresolvedAddressException;
  60: import java.nio.channels.UnsupportedAddressTypeException;
  61: import java.nio.channels.spi.SelectorProvider;
  62: 
  63: public final class SocketChannelImpl extends SocketChannel
  64: {
  65:   private PlainSocketImpl impl;
  66:   private NIOSocket socket;
  67:   private boolean connectionPending;
  68: 
  69:   SocketChannelImpl (SelectorProvider provider)
  70:     throws IOException
  71:   {
  72:     super (provider);
  73:     impl = new PlainSocketImpl();
  74:     socket = new NIOSocket (impl, this);
  75:     configureBlocking(true);
  76:   }
  77:   
  78:   SocketChannelImpl (SelectorProvider provider,
  79:                      NIOSocket socket)
  80:     throws IOException
  81:   {
  82:     super (provider);
  83:     this.impl = socket.getPlainSocketImpl();
  84:     this.socket = socket;
  85:   }
  86: 
  87:   public void finalizer()
  88:   {
  89:     if (isConnected())
  90:       {
  91:         try
  92:           {
  93:             close ();
  94:           }
  95:         catch (Exception e)
  96:           {
  97:           }
  98:       }
  99:   }
 100: 
 101:   PlainSocketImpl getPlainSocketImpl()
 102:   {
 103:     return impl;
 104:   }
 105: 
 106:   int getNativeFD()
 107:   {
 108:     return socket.getPlainSocketImpl().getNativeFD();
 109:   }
 110: 
 111:   protected void implCloseSelectableChannel () throws IOException
 112:   {
 113:     socket.close();
 114:   }
 115: 
 116:   protected void implConfigureBlocking (boolean blocking) throws IOException
 117:   {
 118:     socket.setSoTimeout (blocking ? 0 : NIOConstants.DEFAULT_TIMEOUT);
 119:   }   
 120: 
 121:   public boolean connect (SocketAddress remote) throws IOException
 122:   {
 123:     if (!isOpen())
 124:       throw new ClosedChannelException();
 125:     
 126:     if (isConnected())
 127:       throw new AlreadyConnectedException();
 128: 
 129:     if (connectionPending)
 130:       throw new ConnectionPendingException();
 131: 
 132:     if (!(remote instanceof InetSocketAddress))
 133:       throw new UnsupportedAddressTypeException();
 134: 
 135:     if (((InetSocketAddress) remote).isUnresolved())
 136:       throw new UnresolvedAddressException();
 137:     
 138:     try
 139:       {
 140:         socket.getPlainSocketImpl().setInChannelOperation(true);
 141:           // indicate that a channel is initiating the accept operation
 142:           // so that the socket ignores the fact that we might be in
 143:           // non-blocking mode.
 144:         
 145:         if (isBlocking())
 146:           {
 147:             // Do blocking connect.
 148:             socket.connect (remote);
 149:             return true;
 150:           }
 151: 
 152:         // Do non-blocking connect.
 153:         try
 154:           {
 155:             socket.connect (remote, NIOConstants.DEFAULT_TIMEOUT);
 156:             return true;
 157:           }
 158:         catch (SocketTimeoutException e)
 159:           {
 160:             connectionPending = true;
 161:             return false;
 162:           }
 163:       }
 164:     finally
 165:       {
 166:         socket.getPlainSocketImpl().setInChannelOperation(false);
 167:       }
 168:   }
 169:     
 170:   public boolean finishConnect ()
 171:     throws IOException
 172:   {
 173:     if (!isOpen())
 174:       throw new ClosedChannelException();
 175:     
 176:     if (!isConnected() && !connectionPending)
 177:       throw new NoConnectionPendingException();
 178:     
 179:     if (isConnected())
 180:       return true;
 181: 
 182:     // FIXME: Handle blocking/non-blocking mode.
 183: 
 184:     Selector selector = provider().openSelector();
 185:     register(selector, SelectionKey.OP_CONNECT);
 186: 
 187:     if (isBlocking())
 188:       {
 189:         selector.select(); // blocking until channel is connected.
 190:         connectionPending = false;
 191:         return true;
 192:       }
 193: 
 194:     int ready = selector.selectNow(); // non-blocking
 195:     if (ready == 1)
 196:       {
 197:         connectionPending = false;
 198:         return true;
 199:       }
 200: 
 201:     return false;
 202:   }
 203: 
 204:   public boolean isConnected ()
 205:   {
 206:     return socket.isConnected();
 207:   }
 208:     
 209:   public boolean isConnectionPending ()
 210:   {
 211:     return connectionPending;
 212:   }
 213:     
 214:   public Socket socket ()
 215:   {
 216:     return socket;
 217:   }
 218: 
 219:   public int read(ByteBuffer dst) throws IOException
 220:   {
 221:     if (!isConnected())
 222:       throw new NotYetConnectedException();
 223:     
 224:     byte[] data;
 225:     int offset = 0;
 226:     InputStream input = socket.getInputStream();
 227:     int available = input.available();
 228:     int len = dst.remaining();
 229:     
 230:     if ((! isBlocking()) && available == 0)
 231:       return 0;
 232:     
 233:     if (dst.hasArray())
 234:       {
 235:         offset = dst.arrayOffset() + dst.position();
 236:         data = dst.array();
 237:       }
 238:     else
 239:       {
 240:         data = new byte [len];
 241:       }
 242: 
 243:     int readBytes = 0;
 244:     boolean completed = false;
 245: 
 246:     try
 247:       {
 248:         begin();
 249:         socket.getPlainSocketImpl().setInChannelOperation(true);
 250:         readBytes = input.read (data, offset, len);
 251:         completed = true;
 252:       }
 253:     finally
 254:       {
 255:         end (completed);
 256:         socket.getPlainSocketImpl().setInChannelOperation(false);
 257:       }
 258: 
 259:     if (readBytes > 0)
 260:       if (dst.hasArray())
 261:     {
 262:       dst.position (dst.position() + readBytes);
 263:     }
 264:       else
 265:         {
 266:           dst.put (data, offset, readBytes);
 267:         }
 268: 
 269:     return readBytes;
 270:   }
 271:     
 272:   public long read (ByteBuffer[] dsts, int offset, int length)
 273:     throws IOException
 274:   {
 275:     if (!isConnected())
 276:       throw new NotYetConnectedException();
 277:     
 278:     if ((offset < 0)
 279:         || (offset > dsts.length)
 280:         || (length < 0)
 281:         || (length > (dsts.length - offset)))
 282:       throw new IndexOutOfBoundsException();
 283:       
 284:     long readBytes = 0;
 285: 
 286:     for (int index = offset; index < length; index++)
 287:       readBytes += read (dsts [index]);
 288: 
 289:     return readBytes;
 290:   }
 291:      
 292:   public int write (ByteBuffer src)
 293:     throws IOException
 294:   {
 295:     if (!isConnected())
 296:       throw new NotYetConnectedException();
 297:     
 298:     byte[] data;
 299:     int offset = 0;
 300:     int len = src.remaining();
 301:     
 302:     if (!src.hasArray())
 303:       {
 304:         data = new byte [len];
 305:         src.get (data, 0, len);
 306:       }
 307:     else
 308:       {
 309:         offset = src.arrayOffset() + src.position();
 310:         data = src.array();
 311:       }
 312: 
 313:     OutputStream output = socket.getOutputStream();
 314:     boolean completed = false;
 315: 
 316:     try
 317:       {
 318:         begin();
 319:         socket.getPlainSocketImpl().setInChannelOperation(true);
 320:         output.write (data, offset, len);
 321:         completed = true;
 322:       }
 323:     finally
 324:       {
 325:         end (completed);
 326:         socket.getPlainSocketImpl().setInChannelOperation(false);
 327:       }
 328: 
 329:     if (src.hasArray())
 330:       {
 331:     src.position (src.position() + len);
 332:       }
 333:     
 334:     return len;
 335:   }
 336: 
 337:   public long write (ByteBuffer[] srcs, int offset, int length)
 338:     throws IOException
 339:   {
 340:     if (!isConnected())
 341:       throw new NotYetConnectedException();
 342:     
 343:     if ((offset < 0)
 344:         || (offset > srcs.length)
 345:         || (length < 0)
 346:         || (length > (srcs.length - offset)))
 347:       throw new IndexOutOfBoundsException();
 348:       
 349:     long writtenBytes = 0;
 350: 
 351:     for (int index = offset; index < length; index++)
 352:       writtenBytes += write (srcs [index]);
 353: 
 354:     return writtenBytes;
 355:   }
 356: }