Change the handling of closing sockets

runtime/bin/socket_impl.dart

 // Copyright (c) 2011, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 
 class _SocketBase {
   // Bit flags used when communicating between the eventhandler and
   // dart code. The EVENT flags are used to indicate events of
   // interest when sending a message from dart code to the
   // eventhandler. When receiving a message from the eventhandler the
   // EVENT flags indicate the events that actually happened. The
   // COMMAND flags are used to send commands from dart to the
   // eventhandler. COMMAND flags are never received from the
   // eventhandler. Additional flags are used to communicate other
   // information.
   static final int _IN_EVENT = 0;
   static final int _OUT_EVENT = 1;
   static final int _ERROR_EVENT = 2;
   static final int _CLOSE_EVENT = 3;
 
   static final int _CLOSE_COMMAND = 8;
+  static final int _SHUTDOWN_READ_COMMAND = 9;
+  static final int _SHUTDOWN_WRITE_COMMAND = 10;
 
   // Flag send to the eventhandler saying that the file descriptor in
   // question represents a listening socket.
   static final int _LISTENING_SOCKET = 16;
 
   static final int _FIRST_EVENT = _IN_EVENT;
   static final int _LAST_EVENT = _CLOSE_EVENT;
 
+  static final int _FIRST_COMMAND = _CLOSE_COMMAND;
+  static final int _LAST_COMMAND = _SHUTDOWN_WRITE_COMMAND;
+
   _SocketBase () {
     _handler = new ReceivePort();
-    _handlerMap = new List(_CLOSE_EVENT + 1);
+    _handlerMap = new List(_LAST_EVENT + 1);
     _handlerMask = 0;
     _canActivateHandlers = true;
     _id = -1;
     _handler.receive((var message, ignored) {
       _multiplex(message);
     });
     EventHandler._start();
   }
 
   /*
    * Multiplexes socket events to the right socket handler.
    */
   void _multiplex(List<int> message) {
     assert(message.length == 1);
     _canActivateHandlers = false;
     int event_mask = message[0];
-      for (int i = _FIRST_EVENT; i <= _LAST_EVENT; i++) {
-        if (((event_mask & (1 << i)) != 0) && _handlerMap[i] != null) {
-          var handleEvent = _handlerMap[i];
+    for (int i = _FIRST_EVENT; i <= _LAST_EVENT; i++) {
+      if (((event_mask & (1 << i)) != 0)) {
+        if ((i == _CLOSE_EVENT) && this is _Socket && _id >= 0) {
+          if (_closedWrite) _close();
+        }
 
+        var eventHandler = _handlerMap[i];
+        if (eventHandler != null) {
           // Unregister the out handler before executing it.
           if (i == _OUT_EVENT) _setHandler(i, null);
 
           // Don't call the in handler if there is no data available
           // after all.
-          if (i == _IN_EVENT && this is _Socket && available() == 0) continue;
-
-          handleEvent();
+          if (i == _IN_EVENT && this is _Socket && available() == 0) {
+            continue;
+          }
+          eventHandler();
         }
       }
+    }
     _canActivateHandlers = true;
     _activateHandlers();
   }
 
   void _setHandler(int event, void callback()) {
     if (callback == null) {
       _handlerMask &= ~(1 << event);
     } else {
       _handlerMask |= (1 << event);
     }
     _handlerMap[event] = callback;
     _activateHandlers();
   }
 
   void _getPort() native "Socket_GetPort";
 
   void set errorHandler(void callback()) {
     _setHandler(_ERROR_EVENT, callback);
   }
 
   void _activateHandlers() {
     if (_canActivateHandlers && (_id >= 0)) {
       int data = _handlerMask;
-      if (_isListenSocket()) data |= (1 << _LISTENING_SOCKET);
+      if (_isListenSocket()) {
+        data |= (1 << _LISTENING_SOCKET);
+      } else {
+        if (_closedRead) { data &= ~(1 << _IN_EVENT); }
+        if (_closedWrite) { data &= ~(1 << _OUT_EVENT); }
+        if (_isListenSocket()) data |= (1 << _LISTENING_SOCKET);
+      }
       EventHandler._sendData(_id, _handler, data);
     }
   }
 
   void _scheduleEvent(int event) {
     _handler.toSendPort().send([1 << event], null);
   }
 
   int get port() {
     if (_port === null) {
       _port = _getPort();
     }
     return _port;
   }
 
-  void close() {
+  void close([bool halfClose = false]) {
     if (_id >= 0) {
-      EventHandler._sendData(_id, _handler, 1 << _CLOSE_COMMAND);
-      _handler.close();
-      _handler = null;
-      _id = -1;
+      if (halfClose) {
+        _closeWrite();
+      } else {
+        _close();
+      }
     } else if (_handler != null) {
       // This is to support closing sockets created but never assigned
       // any actual socket.
       _handler.close();
       _handler = null;
     }
   }
 
+  void _closeWrite() {
+    if (_closedRead) {
+      _close();
+    } else {
+      EventHandler._sendData(_id, _handler, 1 << _SHUTDOWN_WRITE_COMMAND);
+    }
+    _closedWrite = true;
+  }
+
+  void _closeRead() {
+    if (_closedWrite) {
+      _close();
+    } else {
+      EventHandler._sendData(_id, _handler, 1 << _SHUTDOWN_READ_COMMAND);
+    }
+    _closedRead = true;
+  }
+
+  void _close() {
+    EventHandler._sendData(_id, _handler, 1 << _CLOSE_COMMAND);
+    _handler.close();
+    _handler = null;
+    _id = -1;
+  }
+
   abstract bool _isListenSocket();
 
   /*
    * Socket id is set from native. -1 indicates that the socket was closed.
    */
   int _id;
 
   /*
    * Dedicated ReceivePort for socket events.
    */
@@ skipping 33 matching lines @@
     ServerSocket socket = new _ServerSocket._internal();
     if (!socket._createBindListen(bindAddress, port, backlog)) {
       return null;
     }
     if (port != 0) {
       socket._port = port;
     }
     return socket;
   }
 
-  _ServerSocket._internal() : super() {}
+  _ServerSocket._internal();
 
   Socket accept() {
     if (_id >= 0) {
       _Socket socket = new _Socket._internal();
       if (_accept(socket)) {
         return socket;
       }
       return null;
     }
     throw new
@@ skipping 12 matching lines @@
   bool _isListenSocket() => true;
 }
 
 
 class _Socket extends _SocketBase implements Socket {
   /*
    * Constructor for socket. First a socket object is allocated
    * in which the native socket is stored. After that _createConnect is
    * called which creates a file discriptor and connects to the given
    * host on the given port. Null is returned if file descriptor creation
-   * or connect failsed
+   * or connect failed.
    */
   factory _Socket(String host, int port) {
     Socket socket = new _Socket._internal();
     if (!socket._createConnect(host, port)) {
       return null;
     }
     return socket;
   }
 
-  _Socket._internal() : super() {}
+  _Socket._internal();
+  _Socket._internalInputOnly() : _closedRead = true;
+  _Socket._internalOutputOnly() : _closedWrite = true;
 
   int available() {
     if (_id >= 0) {
       return _available();
     }
     throw new
         SocketIOException("Error: available failed - invalid socket handle");
   }
 
   int _available() native "Socket_Available";
 
+  bool get closed() => _closed;
+
   int readList(List<int> buffer, int offset, int bytes) {
     if (_id >= 0) {
       if (bytes == 0) {
         return 0;
       }
       if (offset < 0) {
         throw new IndexOutOfRangeException(offset);
       }
       if (bytes < 0) {
         throw new IndexOutOfRangeException(bytes);
@@ skipping 60 matching lines @@
     return _inputStream;
   }
 
   OutputStream get outputStream() {
     if (_outputStream === null) {
       _outputStream = new SocketOutputStream(this);
     }
     return _outputStream;
   }
 
+  bool _closedRead = false;
+  bool _closedWrite = false;
   SocketInputStream _inputStream;
   SocketOutputStream _outputStream;
 }