[dart:io] Adds a finalizer to _NativeSocket to avoid socket leaks

runtime/bin/eventhandler_win.h

 // Copyright (c) 2012, 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.
 
 #ifndef RUNTIME_BIN_EVENTHANDLER_WIN_H_
 #define RUNTIME_BIN_EVENTHANDLER_WIN_H_
 
 #if !defined(RUNTIME_BIN_EVENTHANDLER_H_)
 #error Do not include eventhandler_win.h directly; use eventhandler.h instead.
 #endif
 
 #include <mswsock.h>
 #include <winsock2.h>
 #include <ws2tcpip.h>
 
 #include "bin/builtin.h"
+#include "bin/reference_counting.h"
 #include "bin/thread.h"
 
 namespace dart {
 namespace bin {
 
 // Forward declarations.
 class EventHandlerImplementation;
 class Handle;
 class FileHandle;
 class SocketHandle;
@@ skipping 36 matching lines @@
 
   // Write data to a buffer before sending it. Returns the number of bytes
   // actually written to the buffer. Calls to Write will always write to
   // the buffer from the begining.
   int Write(const void* buffer, int num_bytes);
 
   // Check the amount of data in a read buffer which has not been read yet.
   int GetRemainingLength();
   bool IsEmpty() { return GetRemainingLength() == 0; }
 
-  Operation operation() { return operation_; }
-  SOCKET client() { return client_; }
+  Operation operation() const { return operation_; }
+  SOCKET client() const { return client_; }
   char* GetBufferStart() { return reinterpret_cast<char*>(&buffer_data_); }
-  int GetBufferSize() { return buflen_; }
-  struct sockaddr* from() {
+  int GetBufferSize() const { return buflen_; }
+  struct sockaddr* from() const {
     return from_;
   }
-  socklen_t* from_len_addr() { return from_len_addr_; }
-  socklen_t from_len() { return from_ == NULL ? 0 : *from_len_addr_; }
+  socklen_t* from_len_addr() const { return from_len_addr_; }
+  socklen_t from_len() const { return from_ == NULL ? 0 : *from_len_addr_; }
 
   // Returns the address of the OVERLAPPED structure with all fields
   // initialized to zero.
   OVERLAPPED* GetCleanOverlapped() {
     memset(&overlapped_, 0, sizeof(overlapped_));
     return &overlapped_;
   }
 
   // Returns a WASBUF structure initialized with the data in this IO buffer.
   WSABUF* GetWASBUF() {
@@ skipping 60 matching lines @@
   // object as the object is allocated larger than it's definition
   // indicate to extend this array.
   uint8_t buffer_data_[1];
 
   DISALLOW_COPY_AND_ASSIGN(OverlappedBuffer);
 };
 
 
 // Abstract super class for holding information on listen and connected
 // sockets.
-class Handle : public DescriptorInfoBase {
+class Handle : public ReferenceCounted<Handle>, public DescriptorInfoBase {
  public:
   enum Type {
     kFile,
     kStd,
     kDirectoryWatch,
     kClientSocket,
     kListenSocket,
     kDatagramSocket
   };
 
-  virtual ~Handle();
-
   // Socket interface exposing normal socket operations.
   intptr_t Available();
   intptr_t Read(void* buffer, intptr_t num_bytes);
   intptr_t RecvFrom(void* buffer,
                     intptr_t num_bytes,
                     struct sockaddr* sa,
                     socklen_t addr_len);
   virtual intptr_t Write(const void* buffer, intptr_t num_bytes);
   virtual intptr_t SendTo(const void* buffer,
                           intptr_t num_bytes,
@@ skipping 47 matching lines @@
   }
   bool SupportsOverlappedIO() {
     return (flags_ & (1 << kDoesNotSupportOverlappedIO)) == 0;
   }
 
   void ReadSyncCompleteAsync();
 
   DWORD last_error() { return last_error_; }
   void set_last_error(DWORD last_error) { last_error_ = last_error; }
 
+  void set_completion_port(HANDLE completion_port) {
+    completion_port_ = completion_port;
+  }
+
+  void set_event_handler(EventHandlerImplementation* event_handler) {
+    event_handler_ = event_handler;
+  }
+
  protected:
   // For access to monitor_;
   friend class EventHandlerImplementation;
 
   enum Flags {
     kClosing = 0,
     kCloseRead = 1,
     kCloseWrite = 2,
     kDoesNotSupportOverlappedIO = 3,
     kError = 4
   };
 
   explicit Handle(intptr_t handle);
+  virtual ~Handle();
 
   virtual void HandleIssueError();
 
   Monitor* monitor_;
   Type type_;
   HANDLE handle_;
   HANDLE completion_port_;
   EventHandlerImplementation* event_handler_;
 
   OverlappedBuffer* data_ready_;     // Buffer for data ready to be read.
   OverlappedBuffer* pending_read_;   // Buffer for pending read.
   OverlappedBuffer* pending_write_;  // Buffer for pending write
 
   DWORD last_error_;
 
   ThreadId read_thread_id_;
   HANDLE read_thread_handle_;
   bool read_thread_starting_;
   bool read_thread_finished_;
 
  private:
   void WaitForReadThreadStarted();
   void NotifyReadThreadStarted();
   void WaitForReadThreadFinished();
   void NotifyReadThreadFinished();
 
   int flags_;
 
+  friend class ReferenceCounted<Handle>;
   DISALLOW_COPY_AND_ASSIGN(Handle);
 };
 
 
 class FileHandle : public DescriptorInfoSingleMixin<Handle> {
  public:
   explicit FileHandle(HANDLE handle)
       : DescriptorInfoSingleMixin(reinterpret_cast<intptr_t>(handle), true) {
     type_ = kFile;
   }
@@ skipping 168 matching lines @@
   virtual bool IsClosed();
 
   ClientSocket* next() { return next_; }
   void set_next(ClientSocket* next) { next_ = next; }
 
   void mark_connected() { connected_ = true; }
   bool is_connected() const { return connected_; }
 
   void mark_closed() { closed_ = true; }
 
+#if defined(DEBUG)
+  static intptr_t disconnecting() { return disconnecting_; }
+#endif
+
  private:
   bool LoadDisconnectEx();
 
   LPFN_DISCONNECTEX DisconnectEx_;
   ClientSocket* next_;
   bool connected_;
   bool closed_;
 
+#if defined(DEBUG)
+  static intptr_t disconnecting_;
+#endif
+
   DISALLOW_COPY_AND_ASSIGN(ClientSocket);
 };
 
 
 class DatagramSocket : public DescriptorInfoSingleMixin<SocketHandle> {
  public:
   explicit DatagramSocket(intptr_t s) : DescriptorInfoSingleMixin(s, true) {
     type_ = kDatagramSocket;
   }
 
@@ skipping 37 matching lines @@
   void HandleRecvFrom(Handle* handle, int bytes, OverlappedBuffer* buffer);
   void HandleWrite(Handle* handle, int bytes, OverlappedBuffer* buffer);
   void HandleDisconnect(ClientSocket* client_socket,
                         int bytes,
                         OverlappedBuffer* buffer);
   void HandleConnect(ClientSocket* client_socket,
                      int bytes,
                      OverlappedBuffer* buffer);
   void HandleIOCompletion(DWORD bytes, ULONG_PTR key, OVERLAPPED* overlapped);
 
+  void HandleCompletionOrInterrupt(BOOL ok,
+                                   DWORD bytes,
+                                   ULONG_PTR key,
+                                   OVERLAPPED* overlapped);
+
   HANDLE completion_port() { return completion_port_; }
 
  private:
-  ClientSocket* client_sockets_head_;
-
   Monitor* startup_monitor_;
   ThreadId handler_thread_id_;
   HANDLE handler_thread_handle_;
 
   TimeoutQueue timeout_queue_;  // Time for next timeout.
   bool shutdown_;
   HANDLE completion_port_;
 
   DISALLOW_COPY_AND_ASSIGN(EventHandlerImplementation);
 };
 
 }  // namespace bin
 }  // namespace dart
 
 #endif  // RUNTIME_BIN_EVENTHANDLER_WIN_H_