mcs/class/System/System.Net.Sockets/SocketAsyncResult.cs

   1 // System.Net.Sockets.SocketAsyncResult.cs
   2 //
   3 // Authors:
   4 //      Ludovic Henry <ludovic@xamarin.com>
   5 //
   6 // Copyright (C) 2015 Xamarin, Inc. (https://www.xamarin.com)
   7 //
   8 //
   9 // Permission is hereby granted, free of charge, to any person obtaining
  10 // a copy of this software and associated documentation files (the
  11 // "Software"), to deal in the Software without restriction, including
  12 // without limitation the rights to use, copy, modify, merge, publish,
  13 // distribute, sublicense, and/or sell copies of the Software, and to
  14 // permit persons to whom the Software is furnished to do so, subject to
  15 // the following conditions:
  16 //
  17 // The above copyright notice and this permission notice shall be
  18 // included in all copies or substantial portions of the Software.
  19 //
  20 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  21 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  22 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  23 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  24 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  25 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  26 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  27 //
  28
  29 using System.Collections;
  30 using System.Collections.Generic;
  31 using System.Runtime.InteropServices;
  32 using System.Runtime.Remoting.Messaging;
  33 using System.Threading;
  34
  35 namespace System.Net.Sockets
  36 {
  37         [StructLayout (LayoutKind.Sequential)]
  38         internal sealed class SocketAsyncResult: IOAsyncResult
  39         {
  40                 public Socket socket;
  41                 public SocketOperation operation;
  42
  43                 Exception DelayedException;
  44
  45                 public EndPoint EndPoint;                 // Connect,ReceiveFrom,SendTo
  46                 public byte [] Buffer;                    // Receive,ReceiveFrom,Send,SendTo
  47                 public int Offset;                        // Receive,ReceiveFrom,Send,SendTo
  48                 public int Size;                          // Receive,ReceiveFrom,Send,SendTo
  49                 public SocketFlags SockFlags;             // Receive,ReceiveFrom,Send,SendTo
  50                 public Socket AcceptSocket;               // AcceptReceive
  51                 public IPAddress[] Addresses;             // Connect
  52                 public int Port;                          // Connect
  53                 public IList<ArraySegment<byte>> Buffers; // Receive, Send
  54                 public bool ReuseSocket;                  // Disconnect
  55                 public int CurrentAddress;                // Connect
  56
  57                 public Socket AcceptedSocket;
  58                 public int Total;
  59
  60                 internal int error;
  61
  62                 public int EndCalled;
  63
  64                 public IntPtr Handle {
  65                         get { return socket != null ? socket.Handle : IntPtr.Zero; }
  66                 }
  67
  68                 /* Used by SocketAsyncEventArgs */
  69                 public SocketAsyncResult ()
  70                         : base ()
  71                 {
  72                 }
  73
  74                 public void Init (Socket socket, AsyncCallback callback, object state, SocketOperation operation)
  75                 {
  76                         base.Init (callback, state);
  77
  78                         this.socket = socket;
  79                         this.operation = operation;
  80
  81                         DelayedException = null;
  82
  83                         EndPoint = null;
  84                         Buffer = null;
  85                         Offset = 0;
  86                         Size = 0;
  87                         SockFlags = SocketFlags.None;
  88                         AcceptSocket = null;
  89                         Addresses = null;
  90                         Port = 0;
  91                         Buffers = null;
  92                         ReuseSocket = false;
  93                         CurrentAddress = 0;
  94
  95                         AcceptedSocket = null;
  96                         Total = 0;
  97
  98                         error = 0;
  99
 100                         EndCalled = 0;
 101                 }
 102
 103                 public SocketAsyncResult (Socket socket, AsyncCallback callback, object state, SocketOperation operation)
 104                         : base (callback, state)
 105                 {
 106                         this.socket = socket;
 107                         this.operation = operation;
 108                 }
 109
 110                 public SocketError ErrorCode {
 111                         get {
 112                                 SocketException ex = DelayedException as SocketException;
 113                                 if (ex != null)
 114                                         return ex.SocketErrorCode;
 115
 116                                 if (error != 0)
 117                                         return (SocketError) error;
 118
 119                                 return SocketError.Success;
 120                         }
 121                 }
 122
 123                 public void CheckIfThrowDelayedException ()
 124                 {
 125                         if (DelayedException != null) {
 126                                 socket.is_connected = false;
 127                                 throw DelayedException;
 128                         }
 129
 130                         if (error != 0) {
 131                                 socket.is_connected = false;
 132                                 throw new SocketException (error);
 133                         }
 134                 }
 135
 136                 internal override void CompleteDisposed ()
 137                 {
 138                         Complete ();
 139                 }
 140
 141                 public void Complete ()
 142                 {
 143                         if (operation != SocketOperation.Receive && socket.CleanedUp)
 144                                 DelayedException = new ObjectDisposedException (socket.GetType ().ToString ());
 145
 146                         IsCompleted = true;
 147
 148                         AsyncCallback callback = AsyncCallback;
 149                         if (callback != null) {
 150                                 ThreadPool.UnsafeQueueUserWorkItem (_ => callback (this), null);
 151                         }
 152
 153                         switch (operation) {
 154                         case SocketOperation.Receive:
 155                         case SocketOperation.ReceiveFrom:
 156                         case SocketOperation.ReceiveGeneric:
 157                         case SocketOperation.Accept:
 158                                 socket.ReadSem.Release ();
 159                                 break;
 160                         case SocketOperation.Send:
 161                         case SocketOperation.SendTo:
 162                         case SocketOperation.SendGeneric:
 163                                 socket.WriteSem.Release ();
 164                                 break;
 165                         }
 166
 167                         // IMPORTANT: 'callback', if any is scheduled from unmanaged code
 168                 }
 169
 170                 public void Complete (bool synch)
 171                 {
 172                         CompletedSynchronously = synch;
 173                         Complete ();
 174                 }
 175
 176                 public void Complete (int total)
 177                 {
 178                         Total = total;
 179                         Complete ();
 180                 }
 181
 182                 public void Complete (Exception e, bool synch)
 183                 {
 184                         DelayedException = e;
 185                         CompletedSynchronously = synch;
 186                         Complete ();
 187                 }
 188
 189                 public void Complete (Exception e)
 190                 {
 191                         DelayedException = e;
 192                         Complete ();
 193                 }
 194
 195                 public void Complete (Socket s)
 196                 {
 197                         AcceptedSocket = s;
 198                         Complete ();
 199                 }
 200
 201                 public void Complete (Socket s, int total)
 202                 {
 203                         AcceptedSocket = s;
 204                         Total = total;
 205                         Complete ();
 206                 }
 207         }
 208 }