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

   1 // System.Net.Sockets.Socket.cs
   2 //
   3 // Authors:
   4 //      Phillip Pearson (pp@myelin.co.nz)
   5 //      Dick Porter <dick@ximian.com>
   6 //      Gonzalo Paniagua Javier (gonzalo@ximian.com)
   7 //      Sridhar Kulkarni (sridharkulkarni@gmail.com)
   8 //      Brian Nickel (brian.nickel@gmail.com)
   9 //
  10 // Copyright (C) 2001, 2002 Phillip Pearson and Ximian, Inc.
  11 //    http://www.myelin.co.nz
  12 // (c) 2004-2011 Novell, Inc. (http://www.novell.com)
  13 //
  14
  15 //
  16 // Permission is hereby granted, free of charge, to any person obtaining
  17 // a copy of this software and associated documentation files (the
  18 // "Software"), to deal in the Software without restriction, including
  19 // without limitation the rights to use, copy, modify, merge, publish,
  20 // distribute, sublicense, and/or sell copies of the Software, and to
  21 // permit persons to whom the Software is furnished to do so, subject to
  22 // the following conditions:
  23 //
  24 // The above copyright notice and this permission notice shall be
  25 // included in all copies or substantial portions of the Software.
  26 //
  27 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  28 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  29 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  30 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  31 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  32 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  33 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  34 //
  35
  36 using System;
  37 using System.Net;
  38 using System.Collections;
  39 using System.Collections.Generic;
  40 using System.Runtime.CompilerServices;
  41 using System.Runtime.InteropServices;
  42 using System.Threading;
  43 using System.IO;
  44 using System.Security;
  45 using System.Text;
  46
  47 #if !NET_2_1
  48 using System.Net.Configuration;
  49 using System.Net.NetworkInformation;
  50 #endif
  51
  52 namespace System.Net.Sockets {
  53
  54         public partial class Socket : IDisposable {
  55                 [StructLayout (LayoutKind.Sequential)]
  56                 struct WSABUF {
  57                         public int len;
  58                         public IntPtr buf;
  59                 }
  60
  61                 // Used by the runtime
  62                 internal enum SocketOperation {
  63                         Accept,
  64                         Connect,
  65                         Receive,
  66                         ReceiveFrom,
  67                         Send,
  68                         SendTo,
  69                         RecvJustCallback,
  70                         SendJustCallback,
  71                         UsedInProcess,
  72                         UsedInConsole2,
  73                         Disconnect,
  74                         AcceptReceive,
  75                         ReceiveGeneric,
  76                         SendGeneric
  77                 }
  78
  79                 [StructLayout (LayoutKind.Sequential)]
  80                 internal sealed class SocketAsyncResult: IAsyncResult
  81                 {
  82                         /* Same structure in the runtime */
  83                         /*
  84                           Keep this in sync with MonoSocketAsyncResult in
  85                           metadata/socket-io.h and ProcessAsyncReader
  86                           in System.Diagnostics/Process.cs.
  87                         */
  88
  89                         public Socket Sock;
  90                         public IntPtr handle;
  91                         object state;
  92                         AsyncCallback callback; // used from the runtime
  93                         WaitHandle waithandle;
  94
  95                         Exception delayedException;
  96
  97                         public EndPoint EndPoint;       // Connect,ReceiveFrom,SendTo
  98                         public byte [] Buffer;          // Receive,ReceiveFrom,Send,SendTo
  99                         public int Offset;              // Receive,ReceiveFrom,Send,SendTo
 100                         public int Size;                // Receive,ReceiveFrom,Send,SendTo
 101                         public SocketFlags SockFlags;   // Receive,ReceiveFrom,Send,SendTo
 102                         public Socket AcceptSocket;     // AcceptReceive
 103                         public IPAddress[] Addresses;   // Connect
 104                         public int Port;                // Connect
 105                         public IList<ArraySegment<byte>> Buffers;       // Receive, Send
 106                         public bool ReuseSocket;        // Disconnect
 107
 108                         // Return values
 109                         Socket acc_socket;
 110                         int total;
 111
 112                         bool completed_sync;
 113                         bool completed;
 114                         public bool blocking;
 115                         internal int error;
 116                         public SocketOperation operation;
 117                         public object ares;
 118                         public int EndCalled;
 119
 120                         // These fields are not in MonoSocketAsyncResult
 121                         public Worker Worker;
 122                         public int CurrentAddress; // Connect
 123
 124                         public SocketAsyncResult ()
 125                         {
 126                         }
 127
 128                         public void Init (Socket sock, object state, AsyncCallback callback, SocketOperation operation)
 129                         {
 130                                 this.Sock = sock;
 131                                 if (sock != null) {
 132                                         this.blocking = sock.blocking;
 133                                         this.handle = sock.socket;
 134                                 } else {
 135                                         this.blocking = true;
 136                                         this.handle = IntPtr.Zero;
 137                                 }
 138                                 this.state = state;
 139                                 this.callback = callback;
 140                                 GC.KeepAlive (this.callback);
 141                                 this.operation = operation;
 142                                 SockFlags = SocketFlags.None;
 143                                 if (waithandle != null)
 144                                         ((ManualResetEvent) waithandle).Reset ();
 145
 146                                 delayedException = null;
 147
 148                                 EndPoint = null;
 149                                 Buffer = null;
 150                                 Offset = 0;
 151                                 Size = 0;
 152                                 SockFlags = 0;
 153                                 AcceptSocket = null;
 154                                 Addresses = null;
 155                                 Port = 0;
 156                                 Buffers = null;
 157                                 ReuseSocket = false;
 158                                 acc_socket = null;
 159                                 total = 0;
 160
 161                                 completed_sync = false;
 162                                 completed = false;
 163                                 blocking = false;
 164                                 error = 0;
 165                                 ares = null;
 166                                 EndCalled = 0;
 167                                 Worker = null;
 168                         }
 169
 170                         public void DoMConnectCallback ()
 171                         {
 172                                 if (callback == null)
 173                                         return;
 174                                 ThreadPool.UnsafeQueueUserWorkItem (_ => { callback (this); }, null);
 175                         }
 176
 177                         public void Dispose ()
 178                         {
 179                                 Init (null, null, null, 0);
 180                                 if (waithandle != null) {
 181                                         waithandle.Close ();
 182                                         waithandle = null;
 183                                 }
 184                         }
 185
 186                         public SocketAsyncResult (Socket sock, object state, AsyncCallback callback, SocketOperation operation)
 187                         {
 188                                 this.Sock = sock;
 189                                 this.blocking = sock.blocking;
 190                                 this.handle = sock.socket;
 191                                 this.state = state;
 192                                 this.callback = callback;
 193                                 GC.KeepAlive (this.callback);
 194                                 this.operation = operation;
 195                                 SockFlags = SocketFlags.None;
 196                                 Worker = new Worker (this);
 197                         }
 198
 199                         public void CheckIfThrowDelayedException ()
 200                         {
 201                                 if (delayedException != null) {
 202                                         Sock.connected = false;
 203                                         throw delayedException;
 204                                 }
 205
 206                                 if (error != 0) {
 207                                         Sock.connected = false;
 208                                         throw new SocketException (error);
 209                                 }
 210                         }
 211
 212                         void CompleteAllOnDispose (Queue queue)
 213                         {
 214                                 object [] pending = queue.ToArray ();
 215                                 queue.Clear ();
 216
 217                                 WaitCallback cb;
 218                                 for (int i = 0; i < pending.Length; i++) {
 219                                         Worker worker = (Worker) pending [i];
 220                                         SocketAsyncResult ares = worker.result;
 221                                         cb = new WaitCallback (ares.CompleteDisposed);
 222                                         ThreadPool.UnsafeQueueUserWorkItem (cb, null);
 223                                 }
 224                         }
 225
 226                         void CompleteDisposed (object unused)
 227                         {
 228                                 Complete ();
 229                         }
 230
 231                         public void Complete ()
 232                         {
 233                                 if (operation != SocketOperation.Receive && Sock.disposed)
 234                                         delayedException = new ObjectDisposedException (Sock.GetType ().ToString ());
 235
 236                                 IsCompleted = true;
 237
 238                                 Queue queue = null;
 239                                 if (operation == SocketOperation.Receive ||
 240                                     operation == SocketOperation.ReceiveFrom ||
 241                                     operation == SocketOperation.ReceiveGeneric ||
 242                                     operation == SocketOperation.Accept) {
 243                                         queue = Sock.readQ;
 244                                 } else if (operation == SocketOperation.Send ||
 245                                            operation == SocketOperation.SendTo ||
 246                                            operation == SocketOperation.SendGeneric) {
 247
 248                                         queue = Sock.writeQ;
 249                                 }
 250
 251                                 if (queue != null) {
 252                                         Worker worker = null;
 253                                         SocketAsyncCall sac = null;
 254                                         lock (queue) {
 255                                                 // queue.Count will only be 0 if the socket is closed while receive/send/accept
 256                                                 // operation(s) are pending and at least one call to this method is
 257                                                 // waiting on the lock while another one calls CompleteAllOnDispose()
 258                                                 if (queue.Count > 0)
 259                                                         queue.Dequeue (); // remove ourselves
 260                                                 if (queue.Count > 0) {
 261                                                         worker = (Worker) queue.Peek ();
 262                                                         if (!Sock.disposed) {
 263                                                                 sac = Worker.Dispatcher;
 264                                                         } else {
 265                                                                 CompleteAllOnDispose (queue);
 266                                                         }
 267                                                 }
 268                                         }
 269
 270                                         if (sac != null)
 271                                                 Socket.socket_pool_queue (sac, worker.result);
 272                                 }
 273                                 // IMPORTANT: 'callback', if any is scheduled from unmanaged code
 274                         }
 275
 276                         public void Complete (bool synch)
 277                         {
 278                                 completed_sync = synch;
 279                                 Complete ();
 280                         }
 281
 282                         public void Complete (int total)
 283                         {
 284                                 this.total = total;
 285                                 Complete ();
 286                         }
 287
 288                         public void Complete (Exception e, bool synch)
 289                         {
 290                                 completed_sync = synch;
 291                                 delayedException = e;
 292                                 Complete ();
 293                         }
 294
 295                         public void Complete (Exception e)
 296                         {
 297                                 delayedException = e;
 298                                 Complete ();
 299                         }
 300
 301                         public void Complete (Socket s)
 302                         {
 303                                 acc_socket = s;
 304                                 Complete ();
 305                         }
 306
 307                         public void Complete (Socket s, int total)
 308                         {
 309                                 acc_socket = s;
 310                                 this.total = total;
 311                                 Complete ();
 312                         }
 313
 314                         public object AsyncState {
 315                                 get {
 316                                         return state;
 317                                 }
 318                         }
 319
 320                         public WaitHandle AsyncWaitHandle {
 321                                 get {
 322                                         lock (this) {
 323                                                 if (waithandle == null)
 324                                                         waithandle = new ManualResetEvent (completed);
 325                                         }
 326
 327                                         return waithandle;
 328                                 }
 329                                 set {
 330                                         waithandle=value;
 331                                 }
 332                         }
 333
 334                         public bool CompletedSynchronously {
 335                                 get {
 336                                         return(completed_sync);
 337                                 }
 338                         }
 339
 340                         public bool IsCompleted {
 341                                 get {
 342                                         return(completed);
 343                                 }
 344                                 set {
 345                                         completed=value;
 346                                         lock (this) {
 347                                                 if (waithandle != null && value) {
 348                                                         ((ManualResetEvent) waithandle).Set ();
 349                                                 }
 350                                         }
 351                                 }
 352                         }
 353
 354                         public Socket Socket {
 355                                 get {
 356                                         return acc_socket;
 357                                 }
 358                         }
 359
 360                         public int Total {
 361                                 get { return total; }
 362                                 set { total = value; }
 363                         }
 364
 365                         public SocketError ErrorCode {
 366                                 get {
 367                                         SocketException ex = delayedException as SocketException;
 368                                         if (ex != null)
 369                                                 return(ex.SocketErrorCode);
 370
 371                                         if (error != 0)
 372                                                 return((SocketError)error);
 373
 374                                         return(SocketError.Success);
 375                                 }
 376                         }
 377                 }
 378
 379                 internal sealed class Worker
 380                 {
 381                         public SocketAsyncResult result;
 382                         SocketAsyncEventArgs args;
 383
 384                         public Worker (SocketAsyncEventArgs args)
 385                         {
 386                                 this.args = args;
 387                                 result = new SocketAsyncResult ();
 388                                 result.Worker = this;
 389                         }
 390
 391                         public Worker (SocketAsyncResult ares)
 392                         {
 393                                 this.result = ares;
 394                         }
 395
 396                         public void Dispose ()
 397                         {
 398                                 if (result != null) {
 399                                         result.Dispose ();
 400                                         result = null;
 401                                         args = null;
 402                                 }
 403                         }
 404
 405                         public static SocketAsyncCall Dispatcher = new SocketAsyncCall (DispatcherCB);
 406
 407                         static void DispatcherCB (SocketAsyncResult sar)
 408                         {
 409                                 SocketOperation op = sar.operation;
 410                                 if (op == Socket.SocketOperation.Receive || op == Socket.SocketOperation.ReceiveGeneric ||
 411                                         op == Socket.SocketOperation.RecvJustCallback)
 412                                         sar.Worker.Receive ();
 413                                 else if (op == Socket.SocketOperation.Send || op == Socket.SocketOperation.SendGeneric ||
 414                                         op == Socket.SocketOperation.SendJustCallback)
 415                                         sar.Worker.Send ();
 416                                 else if (op == Socket.SocketOperation.ReceiveFrom)
 417                                         sar.Worker.ReceiveFrom ();
 418                                 else if (op == Socket.SocketOperation.SendTo)
 419                                         sar.Worker.SendTo ();
 420                                 else if (op == Socket.SocketOperation.Connect)
 421                                         sar.Worker.Connect ();
 422                                 else if (op == Socket.SocketOperation.Accept)
 423                                         sar.Worker.Accept ();
 424                                 else if (op == Socket.SocketOperation.AcceptReceive)
 425                                         sar.Worker.AcceptReceive ();
 426                                 else if (op == Socket.SocketOperation.Disconnect)
 427                                         sar.Worker.Disconnect ();
 428
 429                                 // SendPackets and ReceiveMessageFrom are not implemented yet
 430                                 /*
 431                                 else if (op == Socket.SocketOperation.ReceiveMessageFrom)
 432                                         async_op = SocketAsyncOperation.ReceiveMessageFrom;
 433                                 else if (op == Socket.SocketOperation.SendPackets)
 434                                         async_op = SocketAsyncOperation.SendPackets;
 435                                 */
 436                                 else
 437                                         throw new NotImplementedException (String.Format ("Operation {0} is not implemented", op));
 438                         }
 439
 440                         /* This is called when reusing a SocketAsyncEventArgs */
 441                         public void Init (Socket sock, SocketAsyncEventArgs args, SocketOperation op)
 442                         {
 443                                 result.Init (sock, args, SocketAsyncEventArgs.Dispatcher, op);
 444                                 result.Worker = this;
 445                                 SocketAsyncOperation async_op;
 446
 447                                 // Notes;
 448                                 //      -SocketOperation.AcceptReceive not used in SocketAsyncEventArgs
 449                                 //      -SendPackets and ReceiveMessageFrom are not implemented yet
 450                                 if (op == Socket.SocketOperation.Connect)
 451                                         async_op = SocketAsyncOperation.Connect;
 452                                 else if (op == Socket.SocketOperation.Accept)
 453                                         async_op = SocketAsyncOperation.Accept;
 454                                 else if (op == Socket.SocketOperation.Disconnect)
 455                                         async_op = SocketAsyncOperation.Disconnect;
 456                                 else if (op == Socket.SocketOperation.Receive || op == Socket.SocketOperation.ReceiveGeneric)
 457                                         async_op = SocketAsyncOperation.Receive;
 458                                 else if (op == Socket.SocketOperation.ReceiveFrom)
 459                                         async_op = SocketAsyncOperation.ReceiveFrom;
 460                                 /*
 461                                 else if (op == Socket.SocketOperation.ReceiveMessageFrom)
 462                                         async_op = SocketAsyncOperation.ReceiveMessageFrom;
 463                                 */
 464                                 else if (op == Socket.SocketOperation.Send || op == Socket.SocketOperation.SendGeneric)
 465                                         async_op = SocketAsyncOperation.Send;
 466                                 /*
 467                                 else if (op == Socket.SocketOperation.SendPackets)
 468                                         async_op = SocketAsyncOperation.SendPackets;
 469                                 */
 470                                 else if (op == Socket.SocketOperation.SendTo)
 471                                         async_op = SocketAsyncOperation.SendTo;
 472                                 else
 473                                         throw new NotImplementedException (String.Format ("Operation {0} is not implemented", op));
 474
 475                                 args.SetLastOperation (async_op);
 476                                 args.SocketError = SocketError.Success;
 477                                 args.BytesTransferred = 0;
 478                         }
 479
 480                         public void Accept ()
 481                         {
 482                                 Socket acc_socket = null;
 483                                 try {
 484                                         if (args != null && args.AcceptSocket != null) {
 485                                                 result.Sock.Accept (args.AcceptSocket);
 486                                                 acc_socket = args.AcceptSocket;
 487                                         } else {
 488                                                 acc_socket = result.Sock.Accept ();
 489                                                 if (args != null)
 490                                                         args.AcceptSocket = acc_socket;
 491                                         }
 492                                 } catch (Exception e) {
 493                                         result.Complete (e);
 494                                         return;
 495                                 }
 496
 497                                 result.Complete (acc_socket);
 498                         }
 499
 500                         /* only used in 2.0 profile and newer, but
 501                          * leave in older profiles to keep interface
 502                          * to runtime consistent
 503                          */
 504                         public void AcceptReceive ()
 505                         {
 506                                 Socket acc_socket = null;
 507                                 try {
 508                                         if (result.AcceptSocket == null) {
 509                                                 acc_socket = result.Sock.Accept ();
 510                                         } else {
 511                                                 acc_socket = result.AcceptSocket;
 512                                                 result.Sock.Accept (acc_socket);
 513                                         }
 514                                 } catch (Exception e) {
 515                                         result.Complete (e);
 516                                         return;
 517                                 }
 518
 519                                 /* It seems the MS runtime
 520                                  * special-cases 0-length requested
 521                                  * receive data.  See bug 464201.
 522                                  */
 523                                 int total = 0;
 524                                 if (result.Size > 0) {
 525                                         try {
 526                                                 SocketError error;
 527                                                 total = acc_socket.Receive_nochecks (result.Buffer,
 528                                                                                      result.Offset,
 529                                                                                      result.Size,
 530                                                                                      result.SockFlags,
 531                                                                                      out error);
 532                                                 if (error != 0) {
 533                                                         result.Complete (new SocketException ((int) error));
 534                                                         return;
 535                                                 }
 536                                         } catch (Exception e) {
 537                                                 result.Complete (e);
 538                                                 return;
 539                                         }
 540                                 }
 541
 542                                 result.Complete (acc_socket, total);
 543                         }
 544
 545                         public void Connect ()
 546                         {
 547                                 if (result.EndPoint == null) {
 548                                         result.Complete (new SocketException ((int)SocketError.AddressNotAvailable));
 549                                         return;
 550                                 }
 551
 552                                 SocketAsyncResult mconnect = result.AsyncState as SocketAsyncResult;
 553                                 bool is_mconnect = (mconnect != null && mconnect.Addresses != null);
 554                                 try {
 555                                         int error_code;
 556                                         EndPoint ep = result.EndPoint;
 557                                         error_code = (int) result.Sock.GetSocketOption (SocketOptionLevel.Socket, SocketOptionName.Error);
 558                                         if (error_code == 0) {
 559                                                 if (is_mconnect)
 560                                                         result = mconnect;
 561                                                 result.Sock.seed_endpoint = ep;
 562                                                 result.Sock.connected = true;
 563                                                 result.Sock.isbound = true;
 564                                                 result.Sock.connect_in_progress = false;
 565                                                 result.error = 0;
 566                                                 result.Complete ();
 567                                                 if (is_mconnect)
 568                                                         result.DoMConnectCallback ();
 569                                                 return;
 570                                         }
 571
 572                                         if (!is_mconnect) {
 573                                                 result.Sock.connect_in_progress = false;
 574                                                 result.Complete (new SocketException (error_code));
 575                                                 return;
 576                                         }
 577
 578                                         if (mconnect.CurrentAddress >= mconnect.Addresses.Length) {
 579                                                 mconnect.Complete (new SocketException (error_code));
 580                                                 if (is_mconnect)
 581                                                         mconnect.DoMConnectCallback ();
 582                                                 return;
 583                                         }
 584                                         mconnect.Sock.BeginMConnect (mconnect);
 585                                 } catch (Exception e) {
 586                                         result.Sock.connect_in_progress = false;
 587                                         if (is_mconnect)
 588                                                 result = mconnect;
 589                                         result.Complete (e);
 590                                         if (is_mconnect)
 591                                                 result.DoMConnectCallback ();
 592                                         return;
 593                                 }
 594                         }
 595
 596                         /* Also only used in 2.0 profile and newer */
 597                         public void Disconnect ()
 598                         {
 599                                 try {
 600                                         if (args != null)
 601                                                 result.ReuseSocket = args.DisconnectReuseSocket;
 602                                         result.Sock.Disconnect (result.ReuseSocket);
 603                                 } catch (Exception e) {
 604                                         result.Complete (e);
 605                                         return;
 606                                 }
 607                                 result.Complete ();
 608                         }
 609
 610                         public void Receive ()
 611                         {
 612                                 if (result.operation == SocketOperation.ReceiveGeneric) {
 613                                         ReceiveGeneric ();
 614                                         return;
 615                                 }
 616                                 // Actual recv() done in the runtime
 617                                 result.Complete ();
 618                         }
 619
 620                         public void ReceiveFrom ()
 621                         {
 622                                 int total = 0;
 623                                 try {
 624                                         total = result.Sock.ReceiveFrom_nochecks (result.Buffer,
 625                                                                          result.Offset,
 626                                                                          result.Size,
 627                                                                          result.SockFlags,
 628                                                                          ref result.EndPoint);
 629                                 } catch (Exception e) {
 630                                         result.Complete (e);
 631                                         return;
 632                                 }
 633
 634                                 result.Complete (total);
 635                         }
 636
 637                         public void ReceiveGeneric ()
 638                         {
 639                                 int total = 0;
 640                                 try {
 641                                         total = result.Sock.Receive (result.Buffers, result.SockFlags);
 642                                 } catch (Exception e) {
 643                                         result.Complete (e);
 644                                         return;
 645                                 }
 646                                 result.Complete (total);
 647                         }
 648
 649                         int send_so_far;
 650
 651                         void UpdateSendValues (int last_sent)
 652                         {
 653                                 if (result.error == 0) {
 654                                         send_so_far += last_sent;
 655                                         result.Offset += last_sent;
 656                                         result.Size -= last_sent;
 657                                 }
 658                         }
 659
 660                         public void Send ()
 661                         {
 662                                 if (result.operation == SocketOperation.SendGeneric) {
 663                                         SendGeneric ();
 664                                         return;
 665                                 }
 666                                 // Actual send() done in the runtime
 667                                 if (result.error == 0) {
 668                                         UpdateSendValues (result.Total);
 669                                         if (result.Sock.disposed) {
 670                                                 result.Complete ();
 671                                                 return;
 672                                         }
 673
 674                                         if (result.Size > 0) {
 675                                                 Socket.socket_pool_queue (Worker.Dispatcher, result);
 676                                                 return; // Have to finish writing everything. See bug #74475.
 677                                         }
 678                                         result.Total = send_so_far;
 679                                         send_so_far = 0;
 680                                 }
 681                                 result.Complete ();
 682                         }
 683
 684                         public void SendTo ()
 685                         {
 686                                 int total = 0;
 687                                 try {
 688                                         total = result.Sock.SendTo_nochecks (result.Buffer,
 689                                                                     result.Offset,
 690                                                                     result.Size,
 691                                                                     result.SockFlags,
 692                                                                     result.EndPoint);
 693
 694                                         UpdateSendValues (total);
 695                                         if (result.Size > 0) {
 696                                                 Socket.socket_pool_queue (Worker.Dispatcher, result);
 697                                                 return; // Have to finish writing everything. See bug #74475.
 698                                         }
 699                                         result.Total = send_so_far;
 700                                         send_so_far = 0;
 701                                 } catch (Exception e) {
 702                                         send_so_far = 0;
 703                                         result.Complete (e);
 704                                         return;
 705                                 }
 706
 707                                 result.Complete ();
 708                         }
 709
 710                         public void SendGeneric ()
 711                         {
 712                                 int total = 0;
 713                                 try {
 714                                         total = result.Sock.Send (result.Buffers, result.SockFlags);
 715                                 } catch (Exception e) {
 716                                         result.Complete (e);
 717                                         return;
 718                                 }
 719                                 result.Complete (total);
 720                         }
 721                 }
 722
 723                 private Queue readQ = new Queue (2);
 724                 private Queue writeQ = new Queue (2);
 725
 726                 internal delegate void SocketAsyncCall (SocketAsyncResult sar);
 727
 728                 /*
 729                  *      These two fields are looked up by name by the runtime, don't change
 730                  *  their name without also updating the runtime code.
 731                  */
 732                 private static int ipv4Supported = -1, ipv6Supported = -1;
 733                 int linger_timeout;
 734
 735                 static Socket ()
 736                 {
 737                         // initialize ipv4Supported and ipv6Supported
 738                         CheckProtocolSupport ();
 739                 }
 740
 741                 internal static void CheckProtocolSupport ()
 742                 {
 743                         if(ipv4Supported == -1) {
 744                                 try {
 745                                         Socket tmp = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
 746                                         tmp.Close();
 747
 748                                         ipv4Supported = 1;
 749                                 } catch {
 750                                         ipv4Supported = 0;
 751                                 }
 752                         }
 753
 754                         if (ipv6Supported == -1) {
 755                                 // We need to put a try/catch around ConfigurationManager methods as will always throw an exception
 756                                 // when run in a mono embedded application.  This occurs as embedded applications do not have a setup
 757                                 // for application config.  The exception is not thrown when called from a normal .NET application.
 758                                 //
 759                                 // We, then, need to guard calls to the ConfigurationManager.  If the config is not found or throws an
 760                                 // exception, will fall through to the existing Socket / API directly below in the code.
 761                                 //
 762                                 // Also note that catching ConfigurationErrorsException specifically would require library dependency
 763                                 // System.Configuration, and wanted to avoid that.
 764 #if !NET_2_1
 765 #if CONFIGURATION_DEP
 766                                 try {
 767                                         SettingsSection config;
 768                                         config = (SettingsSection) System.Configuration.ConfigurationManager.GetSection ("system.net/settings");
 769                                         if (config != null)
 770                                                 ipv6Supported = config.Ipv6.Enabled ? -1 : 0;
 771                                 } catch {
 772                                         ipv6Supported = -1;
 773                                 }
 774 #else
 775                                 try {
 776                                         NetConfig config = System.Configuration.ConfigurationSettings.GetConfig("system.net/settings") as NetConfig;
 777                                         if (config != null)
 778                                                 ipv6Supported = config.ipv6Enabled ? -1 : 0;
 779                                 } catch {
 780                                         ipv6Supported = -1;
 781                                 }
 782 #endif
 783 #endif
 784                                 if (ipv6Supported != 0) {
 785                                         try {
 786                                                 Socket tmp = new Socket(AddressFamily.InterNetworkV6, SocketType.Stream, ProtocolType.Tcp);
 787                                                 tmp.Close();
 788
 789                                                 ipv6Supported = 1;
 790                                         } catch {
 791                                                 ipv6Supported = 0;
 792                                         }
 793                                 }
 794                         }
 795                 }
 796
 797                 public static bool SupportsIPv4 {
 798                         get {
 799                                 CheckProtocolSupport();
 800                                 return ipv4Supported == 1;
 801                         }
 802                 }
 803
 804                 [ObsoleteAttribute ("Use OSSupportsIPv6 instead")]
 805                 public static bool SupportsIPv6 {
 806                         get {
 807                                 CheckProtocolSupport();
 808                                 return ipv6Supported == 1;
 809                         }
 810                 }
 811 #if NET_2_1
 812                 public static bool OSSupportsIPv4 {
 813                         get {
 814                                 CheckProtocolSupport();
 815                                 return ipv4Supported == 1;
 816                         }
 817                 }
 818 #endif
 819 #if NET_2_1
 820                 public static bool OSSupportsIPv6 {
 821                         get {
 822                                 CheckProtocolSupport();
 823                                 return ipv6Supported == 1;
 824                         }
 825                 }
 826 #else
 827                 public static bool OSSupportsIPv6 {
 828                         get {
 829                                 NetworkInterface[] nics = NetworkInterface.GetAllNetworkInterfaces ();
 830
 831                                 foreach (NetworkInterface adapter in nics) {
 832                                         if (adapter.Supports (NetworkInterfaceComponent.IPv6))
 833                                                 return true;
 834                                 }
 835                                 return false;
 836                         }
 837                 }
 838 #endif
 839
 840                 /* the field "socket" is looked up by name by the runtime */
 841                 private IntPtr socket;
 842                 private AddressFamily address_family;
 843                 private SocketType socket_type;
 844                 private ProtocolType protocol_type;
 845                 internal bool blocking=true;
 846                 List<Thread> blocking_threads;
 847                 private bool isbound;
 848                 /* When true, the socket was connected at the time of
 849                  * the last IO operation
 850                  */
 851                 private bool connected;
 852                 /* true if we called Close_internal */
 853                 private bool closed;
 854                 internal bool disposed;
 855                 bool connect_in_progress;
 856
 857                 /*
 858                  * This EndPoint is used when creating new endpoints. Because
 859                  * there are many types of EndPoints possible,
 860                  * seed_endpoint.Create(addr) is used for creating new ones.
 861                  * As such, this value is set on Bind, SentTo, ReceiveFrom,
 862                  * Connect, etc.
 863                  */
 864                 internal EndPoint seed_endpoint = null;
 865
 866                 void RegisterForBlockingSyscall ()
 867                 {
 868                         while (blocking_threads == null) {
 869                                 //In the rare event this CAS fail, there's a good chance other thread won, so we're kosher.
 870                                 //In the VERY rare event of all CAS fail together, we pay the full price of of failure.
 871                                 Interlocked.CompareExchange (ref blocking_threads, new List<Thread> (), null);
 872                         }
 873
 874                         try {
 875
 876                         } finally {
 877                                 /* We must use a finally block here to make this atomic. */
 878                                 lock (blocking_threads) {
 879                                         blocking_threads.Add (Thread.CurrentThread);
 880                                 }
 881                         }
 882                 }
 883
 884                 /* This must be called from a finally block! */
 885                 void UnRegisterForBlockingSyscall ()
 886                 {
 887                         //If this NRE, we're in deep problems because Register Must have
 888                         lock (blocking_threads) {
 889                                 blocking_threads.Remove (Thread.CurrentThread);
 890                         }
 891                 }
 892
 893                 void AbortRegisteredThreads () {
 894                         if (blocking_threads == null)
 895                                 return;
 896
 897                         lock (blocking_threads) {
 898                                 foreach (var t in blocking_threads)
 899                                         cancel_blocking_socket_operation (t);
 900                                 blocking_threads.Clear ();
 901                         }
 902                 }
 903
 904 #if !TARGET_JVM
 905                 // Creates a new system socket, returning the handle
 906                 [MethodImplAttribute(MethodImplOptions.InternalCall)]
 907                 private extern IntPtr Socket_internal(AddressFamily family,
 908                                                       SocketType type,
 909                                                       ProtocolType proto,
 910                                                       out int error);
 911 #endif
 912
 913                 public Socket(AddressFamily addressFamily, SocketType socketType, ProtocolType protocolType)
 914                 {
 915 #if NET_2_1 && !MOBILE
 916                         switch (addressFamily) {
 917                         case AddressFamily.InterNetwork:        // ok
 918                         case AddressFamily.InterNetworkV6:      // ok
 919                         case AddressFamily.Unknown:             // SocketException will be thrown later (with right error #)
 920                                 break;
 921                         // case AddressFamily.Unspecified:
 922                         default:
 923                                 throw new ArgumentException ("addressFamily");
 924                         }
 925
 926                         switch (socketType) {
 927                         case SocketType.Stream:                 // ok
 928                         case SocketType.Unknown:                // SocketException will be thrown later (with right error #)
 929                                 break;
 930                         default:
 931                                 throw new ArgumentException ("socketType");
 932                         }
 933
 934                         switch (protocolType) {
 935                         case ProtocolType.Tcp:                  // ok
 936                         case ProtocolType.Unspecified:          // ok
 937                         case ProtocolType.Unknown:              // SocketException will be thrown later (with right error #)
 938                                 break;
 939                         default:
 940                                 throw new ArgumentException ("protocolType");
 941                         }
 942 #endif
 943                         address_family = addressFamily;
 944                         socket_type = socketType;
 945                         protocol_type = protocolType;
 946
 947                         int error;
 948
 949                         socket = Socket_internal (addressFamily, socketType, protocolType, out error);
 950                         if (error != 0)
 951                                 throw new SocketException (error);
 952 #if !NET_2_1 || MOBILE
 953                         SocketDefaults ();
 954 #endif
 955                 }
 956
 957                 ~Socket ()
 958                 {
 959                         Dispose (false);
 960                 }
 961
 962
 963                 public AddressFamily AddressFamily {
 964                         get { return address_family; }
 965                 }
 966
 967 #if !TARGET_JVM
 968                 [MethodImplAttribute(MethodImplOptions.InternalCall)]
 969                 private extern static void Blocking_internal(IntPtr socket,
 970                                                              bool block,
 971                                                              out int error);
 972 #endif
 973
 974                 public bool Blocking {
 975                         get {
 976                                 return(blocking);
 977                         }
 978                         set {
 979                                 if (disposed && closed)
 980                                         throw new ObjectDisposedException (GetType ().ToString ());
 981
 982                                 int error;
 983
 984                                 Blocking_internal (socket, value, out error);
 985
 986                                 if (error != 0)
 987                                         throw new SocketException (error);
 988
 989                                 blocking=value;
 990                         }
 991                 }
 992
 993                 public bool Connected {
 994                         get { return connected; }
 995                         internal set { connected = value; }
 996                 }
 997
 998                 public ProtocolType ProtocolType {
 999                         get { return protocol_type; }
1000                 }
1001
1002                 public bool NoDelay {
1003                         get {
1004                                 if (disposed && closed)
1005                                         throw new ObjectDisposedException (GetType ().ToString ());
1006
1007                                 ThrowIfUpd ();
1008
1009                                 return (int)(GetSocketOption (
1010                                         SocketOptionLevel.Tcp,
1011                                         SocketOptionName.NoDelay)) != 0;
1012                         }
1013
1014                         set {
1015                                 if (disposed && closed)
1016                                         throw new ObjectDisposedException (GetType ().ToString ());
1017
1018                                 ThrowIfUpd ();
1019
1020                                 SetSocketOption (
1021                                         SocketOptionLevel.Tcp,
1022                                         SocketOptionName.NoDelay, value ? 1 : 0);
1023                         }
1024                 }
1025
1026                 public int ReceiveBufferSize {
1027                         get {
1028                                 if (disposed && closed) {
1029                                         throw new ObjectDisposedException (GetType ().ToString ());
1030                                 }
1031                                 return((int)GetSocketOption (SocketOptionLevel.Socket, SocketOptionName.ReceiveBuffer));
1032                         }
1033                         set {
1034                                 if (disposed && closed) {
1035                                         throw new ObjectDisposedException (GetType ().ToString ());
1036                                 }
1037                                 if (value < 0) {
1038                                         throw new ArgumentOutOfRangeException ("value", "The value specified for a set operation is less than zero");
1039                                 }
1040
1041                                 SetSocketOption (SocketOptionLevel.Socket, SocketOptionName.ReceiveBuffer, value);
1042                         }
1043                 }
1044
1045                 public int SendBufferSize {
1046                         get {
1047                                 if (disposed && closed) {
1048                                         throw new ObjectDisposedException (GetType ().ToString ());
1049                                 }
1050                                 return((int)GetSocketOption (SocketOptionLevel.Socket, SocketOptionName.SendBuffer));
1051                         }
1052                         set {
1053                                 if (disposed && closed) {
1054                                         throw new ObjectDisposedException (GetType ().ToString ());
1055                                 }
1056                                 if (value < 0) {
1057                                         throw new ArgumentOutOfRangeException ("value", "The value specified for a set operation is less than zero");
1058                                 }
1059
1060                                 SetSocketOption (SocketOptionLevel.Socket,
1061                                                  SocketOptionName.SendBuffer,
1062                                                  value);
1063                         }
1064                 }
1065
1066                 public short Ttl {
1067                         get {
1068                                 if (disposed && closed) {
1069                                         throw new ObjectDisposedException (GetType ().ToString ());
1070                                 }
1071
1072                                 short ttl_val;
1073
1074                                 if (address_family == AddressFamily.InterNetwork) {
1075                                         ttl_val = (short)((int)GetSocketOption (SocketOptionLevel.IP, SocketOptionName.IpTimeToLive));
1076                                 } else if (address_family == AddressFamily.InterNetworkV6) {
1077                                         ttl_val = (short)((int)GetSocketOption (SocketOptionLevel.IPv6, SocketOptionName.HopLimit));
1078                                 } else {
1079                                         throw new NotSupportedException ("This property is only valid for InterNetwork and InterNetworkV6 sockets");
1080                                 }
1081
1082                                 return(ttl_val);
1083                         }
1084                         set {
1085                                 if (disposed && closed) {
1086                                         throw new ObjectDisposedException (GetType ().ToString ());
1087                                 }
1088                                 if (value < 0) {
1089                                         throw new ArgumentOutOfRangeException ("value", "The value specified for a set operation is less than zero");
1090                                 }
1091
1092                                 if (address_family == AddressFamily.InterNetwork) {
1093                                         SetSocketOption (SocketOptionLevel.IP, SocketOptionName.IpTimeToLive, value);
1094                                 } else if (address_family == AddressFamily.InterNetworkV6) {
1095                                         SetSocketOption (SocketOptionLevel.IPv6, SocketOptionName.HopLimit, value);
1096                                 } else {
1097                                         throw new NotSupportedException ("This property is only valid for InterNetwork and InterNetworkV6 sockets");
1098                                 }
1099                         }
1100                 }
1101
1102                 // Returns the remote endpoint details in addr and port
1103                 [MethodImplAttribute(MethodImplOptions.InternalCall)]
1104                 private extern static SocketAddress RemoteEndPoint_internal(IntPtr socket, int family, out int error);
1105
1106                 public EndPoint RemoteEndPoint {
1107                         get {
1108                                 if (disposed && closed)
1109                                         throw new ObjectDisposedException (GetType ().ToString ());
1110
1111                                 /*
1112                                  * If the seed EndPoint is null, Connect, Bind,
1113                                  * etc has not yet been called. MS returns null
1114                                  * in this case.
1115                                  */
1116                                 if (!connected || seed_endpoint == null)
1117                                         return null;
1118                                 SocketAddress sa;
1119                                 int error;
1120
1121                                 sa=RemoteEndPoint_internal(socket, (int) address_family, out error);
1122
1123                                 if (error != 0)
1124                                         throw new SocketException (error);
1125
1126                                 return seed_endpoint.Create (sa);
1127                         }
1128                 }
1129
1130                 void Linger (IntPtr handle)
1131                 {
1132                         if (!connected || linger_timeout <= 0)
1133                                 return;
1134
1135                         // We don't want to receive any more data
1136                         int error;
1137                         Shutdown_internal (handle, SocketShutdown.Receive, out error);
1138                         if (error != 0)
1139                                 return;
1140
1141                         int seconds = linger_timeout / 1000;
1142                         int ms = linger_timeout % 1000;
1143                         if (ms > 0) {
1144                                 // If the other end closes, this will return 'true' with 'Available' == 0
1145                                 Poll_internal (handle, SelectMode.SelectRead, ms * 1000, out error);
1146                                 if (error != 0)
1147                                         return;
1148
1149                         }
1150                         if (seconds > 0) {
1151                                 LingerOption linger = new LingerOption (true, seconds);
1152                                 SetSocketOption_internal (handle, SocketOptionLevel.Socket, SocketOptionName.Linger, linger, null, 0, out error);
1153                                 /* Not needed, we're closing upon return */
1154                                 /*if (error != 0)
1155                                         return; */
1156                         }
1157                 }
1158                 [MethodImplAttribute(MethodImplOptions.InternalCall)]
1159                 static extern void cancel_blocking_socket_operation (Thread thread);
1160
1161                 protected virtual void Dispose (bool disposing)
1162                 {
1163                         if (disposed)
1164                                 return;
1165
1166                         disposed = true;
1167                         bool was_connected = connected;
1168                         connected = false;
1169                         if ((int) socket != -1) {
1170                                 int error;
1171                                 closed = true;
1172                                 IntPtr x = socket;
1173                                 socket = (IntPtr) (-1);
1174
1175                                 AbortRegisteredThreads ();
1176
1177                                 if (was_connected)
1178                                         Linger (x);
1179                                 //DateTime start = DateTime.UtcNow;
1180                                 Close_internal (x, out error);
1181                                 //Console.WriteLine ("Time spent in Close_internal: {0}ms", (DateTime.UtcNow - start).TotalMilliseconds);
1182                                 if (error != 0)
1183                                         throw new SocketException (error);
1184                         }
1185                 }
1186
1187 #if NET_4_0
1188                 public void Dispose ()
1189 #else
1190                 void IDisposable.Dispose ()
1191 #endif
1192                 {
1193                         Dispose (true);
1194                         GC.SuppressFinalize (this);
1195                 }
1196
1197                 // Closes the socket
1198                 [MethodImplAttribute(MethodImplOptions.InternalCall)]
1199                 private extern static void Close_internal(IntPtr socket, out int error);
1200
1201                 public void Close ()
1202                 {
1203                         linger_timeout = 0;
1204                         ((IDisposable) this).Dispose ();
1205                 }
1206
1207                 public void Close (int timeout)
1208                 {
1209                         linger_timeout = timeout;
1210                         ((IDisposable) this).Dispose ();
1211                 }
1212
1213                 // Connects to the remote address
1214                 [MethodImplAttribute(MethodImplOptions.InternalCall)]
1215                 private extern static void Connect_internal(IntPtr sock,
1216                                                             SocketAddress sa,
1217                                                             out int error);
1218
1219                 public void Connect (EndPoint remoteEP)
1220                 {
1221                         SocketAddress serial = null;
1222
1223                         if (disposed && closed)
1224                                 throw new ObjectDisposedException (GetType ().ToString ());
1225
1226                         if (remoteEP == null)
1227                                 throw new ArgumentNullException ("remoteEP");
1228
1229                         IPEndPoint ep = remoteEP as IPEndPoint;
1230                         if (ep != null && socket_type != SocketType.Dgram) /* Dgram uses Any to 'disconnect' */
1231                                 if (ep.Address.Equals (IPAddress.Any) || ep.Address.Equals (IPAddress.IPv6Any))
1232                                         throw new SocketException ((int) SocketError.AddressNotAvailable);
1233
1234                         if (islistening)
1235                                 throw new InvalidOperationException ();
1236                         serial = remoteEP.Serialize ();
1237
1238                         int error = 0;
1239
1240                         try {
1241                                 RegisterForBlockingSyscall ();
1242                                 Connect_internal (socket, serial, out error);
1243                         } finally {
1244                                 UnRegisterForBlockingSyscall ();
1245                         }
1246
1247                         if (error == 0 || error == 10035)
1248                                 seed_endpoint = remoteEP; // Keep the ep around for non-blocking sockets
1249
1250                         if (error != 0) {
1251                                 if (closed)
1252                                         error = SOCKET_CLOSED;
1253                                 throw new SocketException (error);
1254                         }
1255
1256                         if (socket_type == SocketType.Dgram && ep != null && (ep.Address.Equals (IPAddress.Any) || ep.Address.Equals (IPAddress.IPv6Any)))
1257                                 connected = false;
1258                         else
1259                                 connected = true;
1260                         isbound = true;
1261                 }
1262
1263                 public bool ReceiveAsync (SocketAsyncEventArgs e)
1264                 {
1265                         // NO check is made whether e != null in MS.NET (NRE is thrown in such case)
1266                         if (disposed && closed)
1267                                 throw new ObjectDisposedException (GetType ().ToString ());
1268
1269                         // LAME SPEC: the ArgumentException is never thrown, instead an NRE is
1270                         // thrown when e.Buffer and e.BufferList are null (works fine when one is
1271                         // set to a valid object)
1272                         if (e.Buffer == null && e.BufferList == null)
1273                                 throw new NullReferenceException ("Either e.Buffer or e.BufferList must be valid buffers.");
1274
1275                         e.curSocket = this;
1276                         SocketOperation op = (e.Buffer != null) ? SocketOperation.Receive : SocketOperation.ReceiveGeneric;
1277                         e.Worker.Init (this, e, op);
1278                         SocketAsyncResult res = e.Worker.result;
1279                         if (e.Buffer != null) {
1280                                 res.Buffer = e.Buffer;
1281                                 res.Offset = e.Offset;
1282                                 res.Size = e.Count;
1283                         } else {
1284                                 res.Buffers = e.BufferList;
1285                         }
1286                         res.SockFlags = e.SocketFlags;
1287                         int count;
1288                         lock (readQ) {
1289                                 readQ.Enqueue (e.Worker);
1290                                 count = readQ.Count;
1291                         }
1292                         if (count == 1) {
1293                                 // Receive takes care of ReceiveGeneric
1294                                 socket_pool_queue (Worker.Dispatcher, res);
1295                         }
1296
1297                         return true;
1298                 }
1299
1300                 public bool SendAsync (SocketAsyncEventArgs e)
1301                 {
1302                         // NO check is made whether e != null in MS.NET (NRE is thrown in such case)
1303                         if (disposed && closed)
1304                                 throw new ObjectDisposedException (GetType ().ToString ());
1305                         if (e.Buffer == null && e.BufferList == null)
1306                                 throw new NullReferenceException ("Either e.Buffer or e.BufferList must be valid buffers.");
1307
1308                         e.curSocket = this;
1309                         SocketOperation op = (e.Buffer != null) ? SocketOperation.Send : SocketOperation.SendGeneric;
1310                         e.Worker.Init (this, e, op);
1311                         SocketAsyncResult res = e.Worker.result;
1312                         if (e.Buffer != null) {
1313                                 res.Buffer = e.Buffer;
1314                                 res.Offset = e.Offset;
1315                                 res.Size = e.Count;
1316                         } else {
1317                                 res.Buffers = e.BufferList;
1318                         }
1319                         res.SockFlags = e.SocketFlags;
1320                         int count;
1321                         lock (writeQ) {
1322                                 writeQ.Enqueue (e.Worker);
1323                                 count = writeQ.Count;
1324                         }
1325                         if (count == 1) {
1326                                 // Send takes care of SendGeneric
1327                                 socket_pool_queue (Worker.Dispatcher, res);
1328                         }
1329                         return true;
1330                 }
1331
1332                 [MethodImplAttribute(MethodImplOptions.InternalCall)]
1333                 extern static bool Poll_internal (IntPtr socket, SelectMode mode, int timeout, out int error);
1334
1335                 [MethodImplAttribute(MethodImplOptions.InternalCall)]
1336                 private extern static int Receive_internal(IntPtr sock,
1337                                                            byte[] buffer,
1338                                                            int offset,
1339                                                            int count,
1340                                                            SocketFlags flags,
1341                                                            out int error);
1342
1343                 internal int Receive_nochecks (byte [] buf, int offset, int size, SocketFlags flags, out SocketError error)
1344                 {
1345                         int nativeError;
1346                         int ret = Receive_internal (socket, buf, offset, size, flags, out nativeError);
1347                         error = (SocketError) nativeError;
1348                         if (error != SocketError.Success && error != SocketError.WouldBlock && error != SocketError.InProgress) {
1349                                 connected = false;
1350                                 isbound = false;
1351                         } else {
1352                                 connected = true;
1353                         }
1354
1355                         return ret;
1356                 }
1357
1358                 [MethodImplAttribute(MethodImplOptions.InternalCall)]
1359                 private extern static void GetSocketOption_obj_internal(IntPtr socket,
1360                         SocketOptionLevel level, SocketOptionName name, out object obj_val,
1361                         out int error);
1362
1363                 [MethodImplAttribute(MethodImplOptions.InternalCall)]
1364                 private extern static int Send_internal(IntPtr sock,
1365                                                         byte[] buf, int offset,
1366                                                         int count,
1367                                                         SocketFlags flags,
1368                                                         out int error);
1369
1370                 internal int Send_nochecks (byte [] buf, int offset, int size, SocketFlags flags, out SocketError error)
1371                 {
1372                         if (size == 0) {
1373                                 error = SocketError.Success;
1374                                 return 0;
1375                         }
1376
1377                         int nativeError;
1378
1379                         int ret = Send_internal (socket, buf, offset, size, flags, out nativeError);
1380
1381                         error = (SocketError)nativeError;
1382
1383                         if (error != SocketError.Success && error != SocketError.WouldBlock && error != SocketError.InProgress) {
1384                                 connected = false;
1385                                 isbound = false;
1386                         } else {
1387                                 connected = true;
1388                         }
1389
1390                         return ret;
1391                 }
1392
1393                 public object GetSocketOption (SocketOptionLevel optionLevel, SocketOptionName optionName)
1394                 {
1395                         if (disposed && closed)
1396                                 throw new ObjectDisposedException (GetType ().ToString ());
1397
1398                         object obj_val;
1399                         int error;
1400
1401                         GetSocketOption_obj_internal (socket, optionLevel, optionName, out obj_val,
1402                                 out error);
1403                         if (error != 0)
1404                                 throw new SocketException (error);
1405
1406                         if (optionName == SocketOptionName.Linger) {
1407                                 return((LingerOption)obj_val);
1408                         } else if (optionName == SocketOptionName.AddMembership ||
1409                                    optionName == SocketOptionName.DropMembership) {
1410                                 return((MulticastOption)obj_val);
1411                         } else if (obj_val is int) {
1412                                 return((int)obj_val);
1413                         } else {
1414                                 return(obj_val);
1415                         }
1416                 }
1417
1418                 [MethodImplAttribute (MethodImplOptions.InternalCall)]
1419                 private extern static void Shutdown_internal (IntPtr socket, SocketShutdown how, out int error);
1420
1421                 public void Shutdown (SocketShutdown how)
1422                 {
1423                         if (disposed && closed)
1424                                 throw new ObjectDisposedException (GetType ().ToString ());
1425
1426                         if (!connected)
1427                                 throw new SocketException (10057); // Not connected
1428
1429                         int error;
1430
1431                         Shutdown_internal (socket, how, out error);
1432                         if (error != 0)
1433                                 throw new SocketException (error);
1434                 }
1435
1436                 [MethodImplAttribute(MethodImplOptions.InternalCall)]
1437                 private extern static void SetSocketOption_internal (IntPtr socket, SocketOptionLevel level,
1438                                                                      SocketOptionName name, object obj_val,
1439                                                                      byte [] byte_val, int int_val,
1440                                                                      out int error);
1441
1442                 public void SetSocketOption (SocketOptionLevel optionLevel, SocketOptionName optionName, int optionValue)
1443                 {
1444                         if (disposed && closed)
1445                                 throw new ObjectDisposedException (GetType ().ToString ());
1446
1447                         int error;
1448
1449                         SetSocketOption_internal (socket, optionLevel, optionName, null,
1450                                                  null, optionValue, out error);
1451
1452                         if (error != 0)
1453                                 throw new SocketException (error);
1454                 }
1455
1456                 private void ThrowIfUpd ()
1457                 {
1458 #if !NET_2_1 || MOBILE
1459                         if (protocol_type == ProtocolType.Udp)
1460                                 throw new SocketException ((int)SocketError.ProtocolOption);
1461 #endif
1462                 }
1463
1464                 public
1465                 IAsyncResult BeginConnect(EndPoint end_point, AsyncCallback callback, object state)
1466                 {
1467                         if (disposed && closed)
1468                                 throw new ObjectDisposedException (GetType ().ToString ());
1469
1470                         if (end_point == null)
1471                                 throw new ArgumentNullException ("end_point");
1472
1473                         SocketAsyncResult req = new SocketAsyncResult (this, state, callback, SocketOperation.Connect);
1474                         req.EndPoint = end_point;
1475
1476                         // Bug #75154: Connect() should not succeed for .Any addresses.
1477                         if (end_point is IPEndPoint) {
1478                                 IPEndPoint ep = (IPEndPoint) end_point;
1479                                 if (ep.Address.Equals (IPAddress.Any) || ep.Address.Equals (IPAddress.IPv6Any)) {
1480                                         req.Complete (new SocketException ((int) SocketError.AddressNotAvailable), true);
1481                                         return req;
1482                                 }
1483                         }
1484
1485                         int error = 0;
1486                         if (connect_in_progress) {
1487                                 // This could happen when multiple IPs are used
1488                                 // Calling connect() again will reset the connection attempt and cause
1489                                 // an error. Better to just close the socket and move on.
1490                                 connect_in_progress = false;
1491                                 Close_internal (socket, out error);
1492                                 socket = Socket_internal (address_family, socket_type, protocol_type, out error);
1493                                 if (error != 0)
1494                                         throw new SocketException (error);
1495                         }
1496                         bool blk = blocking;
1497                         if (blk)
1498                                 Blocking = false;
1499                         SocketAddress serial = end_point.Serialize ();
1500                         Connect_internal (socket, serial, out error);
1501                         if (blk)
1502                                 Blocking = true;
1503                         if (error == 0) {
1504                                 // succeeded synch
1505                                 connected = true;
1506                                 isbound = true;
1507                                 req.Complete (true);
1508                                 return req;
1509                         }
1510
1511                         if (error != (int) SocketError.InProgress && error != (int) SocketError.WouldBlock) {
1512                                 // error synch
1513                                 connected = false;
1514                                 isbound = false;
1515                                 req.Complete (new SocketException (error), true);
1516                                 return req;
1517                         }
1518
1519                         // continue asynch
1520                         connected = false;
1521                         isbound = false;
1522                         connect_in_progress = true;
1523                         socket_pool_queue (Worker.Dispatcher, req);
1524                         return req;
1525                 }
1526
1527                 public
1528                 IAsyncResult BeginConnect (IPAddress[] addresses, int port, AsyncCallback callback, object state)
1529
1530                 {
1531                         if (disposed && closed)
1532                                 throw new ObjectDisposedException (GetType ().ToString ());
1533
1534                         if (addresses == null)
1535                                 throw new ArgumentNullException ("addresses");
1536
1537                         if (addresses.Length == 0)
1538                                 throw new ArgumentException ("Empty addresses list");
1539
1540                         if (this.AddressFamily != AddressFamily.InterNetwork &&
1541                                 this.AddressFamily != AddressFamily.InterNetworkV6)
1542                                 throw new NotSupportedException ("This method is only valid for addresses in the InterNetwork or InterNetworkV6 families");
1543
1544                         if (port <= 0 || port > 65535)
1545                                 throw new ArgumentOutOfRangeException ("port", "Must be > 0 and < 65536");
1546                         if (islistening)
1547                                 throw new InvalidOperationException ();
1548
1549                         SocketAsyncResult req = new SocketAsyncResult (this, state, callback, SocketOperation.Connect);
1550                         req.Addresses = addresses;
1551                         req.Port = port;
1552                         connected = false;
1553                         return BeginMConnect (req);
1554                 }
1555
1556                 IAsyncResult BeginMConnect (SocketAsyncResult req)
1557                 {
1558                         IAsyncResult ares = null;
1559                         Exception exc = null;
1560                         for (int i = req.CurrentAddress; i < req.Addresses.Length; i++) {
1561                                 IPAddress addr = req.Addresses [i];
1562                                 IPEndPoint ep = new IPEndPoint (addr, req.Port);
1563                                 try {
1564                                         req.CurrentAddress++;
1565                                         ares = BeginConnect (ep, null, req);
1566                                         if (ares.IsCompleted && ares.CompletedSynchronously) {
1567                                                 ((SocketAsyncResult) ares).CheckIfThrowDelayedException ();
1568                                                 req.DoMConnectCallback ();
1569                                         }
1570                                         break;
1571                                 } catch (Exception e) {
1572                                         exc = e;
1573                                         ares = null;
1574                                 }
1575                         }
1576
1577                         if (ares == null)
1578                                 throw exc;
1579
1580                         return req;
1581                 }
1582
1583                 // Returns false when it is ok to use RemoteEndPoint
1584                 //         true when addresses must be used (and addresses could be null/empty)
1585                 bool GetCheckedIPs (SocketAsyncEventArgs e, out IPAddress [] addresses)
1586                 {
1587                         addresses = null;
1588 #if NET_4_0
1589                         // Connect to the first address that match the host name, like:
1590                         // http://blogs.msdn.com/ncl/archive/2009/07/20/new-ncl-features-in-net-4-0-beta-2.aspx
1591                         // while skipping entries that do not match the address family
1592                         DnsEndPoint dep = (e.RemoteEndPoint as DnsEndPoint);
1593                         if (dep != null) {
1594                                 addresses = Dns.GetHostAddresses (dep.Host);
1595                                 return true;
1596                         } else {
1597                                 e.ConnectByNameError = null;
1598                                         return false;
1599                         }
1600 #else
1601                         return false; // < NET_4_0 -> use remote endpoint
1602 #endif
1603                 }
1604
1605                 bool ConnectAsyncReal (SocketAsyncEventArgs e)
1606                 {
1607                         bool use_remoteep = true;
1608 #if NET_4_0
1609                         IPAddress [] addresses = null;
1610                         use_remoteep = !GetCheckedIPs (e, out addresses);
1611 #endif
1612                         e.curSocket = this;
1613                         Worker w = e.Worker;
1614                         w.Init (this, e, SocketOperation.Connect);
1615                         SocketAsyncResult result = w.result;
1616                         IAsyncResult ares = null;
1617                         try {
1618                                 if (use_remoteep) {
1619                                         result.EndPoint = e.RemoteEndPoint;
1620                                         ares = BeginConnect (e.RemoteEndPoint, SocketAsyncEventArgs.Dispatcher, e);
1621                                 }
1622 #if NET_4_0
1623                                 else {
1624
1625                                         DnsEndPoint dep = (e.RemoteEndPoint as DnsEndPoint);
1626                                         result.Addresses = addresses;
1627                                         result.Port = dep.Port;
1628
1629                                         ares = BeginConnect (addresses, dep.Port, SocketAsyncEventArgs.Dispatcher, e);
1630                                 }
1631 #endif
1632                                 if (ares.IsCompleted && ares.CompletedSynchronously) {
1633                                         ((SocketAsyncResult) ares).CheckIfThrowDelayedException ();
1634                                         return false;
1635                                 }
1636                         } catch (Exception exc) {
1637                                 result.Complete (exc, true);
1638                                 return false;
1639                         }
1640                         return true;
1641                 }
1642
1643                 public bool ConnectAsync (SocketAsyncEventArgs e)
1644                 {
1645                         // NO check is made whether e != null in MS.NET (NRE is thrown in such case)
1646                         if (disposed && closed)
1647                                 throw new ObjectDisposedException (GetType ().ToString ());
1648                         if (islistening)
1649                                 throw new InvalidOperationException ("You may not perform this operation after calling the Listen method.");
1650                         if (e.RemoteEndPoint == null)
1651                                 throw new ArgumentNullException ("remoteEP");
1652
1653                         return ConnectAsyncReal (e);
1654                 }
1655
1656                 [MethodImplAttribute (MethodImplOptions.InternalCall)]
1657                 private extern static int Receive_internal (IntPtr sock, WSABUF[] bufarray, SocketFlags flags, out int error);
1658                 public
1659                 int Receive (IList<ArraySegment<byte>> buffers)
1660                 {
1661                         int ret;
1662                         SocketError error;
1663                         ret = Receive (buffers, SocketFlags.None, out error);
1664                         if (error != SocketError.Success) {
1665                                 throw new SocketException ((int)error);
1666                         }
1667                         return(ret);
1668                 }
1669
1670                 [CLSCompliant (false)]
1671                 public
1672                 int Receive (IList<ArraySegment<byte>> buffers, SocketFlags socketFlags)
1673                 {
1674                         int ret;
1675                         SocketError error;
1676                         ret = Receive (buffers, socketFlags, out error);
1677                         if (error != SocketError.Success) {
1678                                 throw new SocketException ((int)error);
1679                         }
1680                         return(ret);
1681                 }
1682
1683                 [CLSCompliant (false)]
1684                 public
1685                 int Receive (IList<ArraySegment<byte>> buffers, SocketFlags socketFlags, out SocketError errorCode)
1686                 {
1687                         if (disposed && closed)
1688                                 throw new ObjectDisposedException (GetType ().ToString ());
1689
1690                         if (buffers == null ||
1691                             buffers.Count == 0) {
1692                                 throw new ArgumentNullException ("buffers");
1693                         }
1694
1695                         int numsegments = buffers.Count;
1696                         int nativeError;
1697                         int ret;
1698
1699                         /* Only example I can find of sending a byte
1700                          * array reference directly into an internal
1701                          * call is in
1702                          * System.Runtime.Remoting/System.Runtime.Remoting.Channels.Ipc.Win32/NamedPipeSocket.cs,
1703                          * so taking a lead from that...
1704                          */
1705                         WSABUF[] bufarray = new WSABUF[numsegments];
1706                         GCHandle[] gch = new GCHandle[numsegments];
1707
1708                         for(int i = 0; i < numsegments; i++) {
1709                                 ArraySegment<byte> segment = buffers[i];
1710
1711                                 if (segment.Offset < 0 || segment.Count < 0 ||
1712                                     segment.Count > segment.Array.Length - segment.Offset)
1713                                         throw new ArgumentOutOfRangeException ("segment");
1714
1715                                 gch[i] = GCHandle.Alloc (segment.Array, GCHandleType.Pinned);
1716                                 bufarray[i].len = segment.Count;
1717                                 bufarray[i].buf = Marshal.UnsafeAddrOfPinnedArrayElement (segment.Array, segment.Offset);
1718                         }
1719
1720                         try {
1721                                 ret = Receive_internal (socket, bufarray,
1722                                                         socketFlags,
1723                                                         out nativeError);
1724                         } finally {
1725                                 for(int i = 0; i < numsegments; i++) {
1726                                         if (gch[i].IsAllocated) {
1727                                                 gch[i].Free ();
1728                                         }
1729                                 }
1730                         }
1731
1732                         errorCode = (SocketError)nativeError;
1733                         return(ret);
1734                 }
1735
1736                 [MethodImplAttribute (MethodImplOptions.InternalCall)]
1737                 private extern static int Send_internal (IntPtr sock, WSABUF[] bufarray, SocketFlags flags, out int error);
1738                 public
1739                 int Send (IList<ArraySegment<byte>> buffers)
1740                 {
1741                         int ret;
1742                         SocketError error;
1743                         ret = Send (buffers, SocketFlags.None, out error);
1744                         if (error != SocketError.Success) {
1745                                 throw new SocketException ((int)error);
1746                         }
1747                         return(ret);
1748                 }
1749
1750                 public
1751                 int Send (IList<ArraySegment<byte>> buffers, SocketFlags socketFlags)
1752                 {
1753                         int ret;
1754                         SocketError error;
1755                         ret = Send (buffers, socketFlags, out error);
1756                         if (error != SocketError.Success) {
1757                                 throw new SocketException ((int)error);
1758                         }
1759                         return(ret);
1760                 }
1761
1762                 [CLSCompliant (false)]
1763                 public
1764                 int Send (IList<ArraySegment<byte>> buffers, SocketFlags socketFlags, out SocketError errorCode)
1765                 {
1766                         if (disposed && closed)
1767                                 throw new ObjectDisposedException (GetType ().ToString ());
1768                         if (buffers == null)
1769                                 throw new ArgumentNullException ("buffers");
1770                         if (buffers.Count == 0)
1771                                 throw new ArgumentException ("Buffer is empty", "buffers");
1772                         int numsegments = buffers.Count;
1773                         int nativeError;
1774                         int ret;
1775
1776                         WSABUF[] bufarray = new WSABUF[numsegments];
1777                         GCHandle[] gch = new GCHandle[numsegments];
1778                         for(int i = 0; i < numsegments; i++) {
1779                                 ArraySegment<byte> segment = buffers[i];
1780
1781                                 if (segment.Offset < 0 || segment.Count < 0 ||
1782                                     segment.Count > segment.Array.Length - segment.Offset)
1783                                         throw new ArgumentOutOfRangeException ("segment");
1784
1785                                 gch[i] = GCHandle.Alloc (segment.Array, GCHandleType.Pinned);
1786                                 bufarray[i].len = segment.Count;
1787                                 bufarray[i].buf = Marshal.UnsafeAddrOfPinnedArrayElement (segment.Array, segment.Offset);
1788                         }
1789
1790                         try {
1791                                 ret = Send_internal (socket, bufarray, socketFlags, out nativeError);
1792                         } finally {
1793                                 for(int i = 0; i < numsegments; i++) {
1794                                         if (gch[i].IsAllocated) {
1795                                                 gch[i].Free ();
1796                                         }
1797                                 }
1798                         }
1799                         errorCode = (SocketError)nativeError;
1800                         return(ret);
1801                 }
1802
1803                 Exception InvalidAsyncOp (string method)
1804                 {
1805                         return new InvalidOperationException (method + " can only be called once per asynchronous operation");
1806                 }
1807
1808                 public
1809                 int EndReceive (IAsyncResult result)
1810                 {
1811                         SocketError error;
1812                         int bytesReceived = EndReceive (result, out error);
1813                         if (error != SocketError.Success) {
1814                                 if (error != SocketError.WouldBlock && error != SocketError.InProgress)
1815                                         connected = false;
1816                                 throw new SocketException ((int)error);
1817                         }
1818                         return bytesReceived;
1819                 }
1820
1821                 public
1822                 int EndReceive (IAsyncResult asyncResult, out SocketError errorCode)
1823                 {
1824                         if (disposed && closed)
1825                                 throw new ObjectDisposedException (GetType ().ToString ());
1826
1827                         if (asyncResult == null)
1828                                 throw new ArgumentNullException ("asyncResult");
1829
1830                         SocketAsyncResult req = asyncResult as SocketAsyncResult;
1831                         if (req == null)
1832                                 throw new ArgumentException ("Invalid IAsyncResult", "asyncResult");
1833
1834                         if (Interlocked.CompareExchange (ref req.EndCalled, 1, 0) == 1)
1835                                 throw InvalidAsyncOp ("EndReceive");
1836                         if (!asyncResult.IsCompleted)
1837                                 asyncResult.AsyncWaitHandle.WaitOne ();
1838
1839                         errorCode = req.ErrorCode;
1840                         // If no socket error occurred, call CheckIfThrowDelayedException in case there are other
1841                         // kinds of exceptions that should be thrown.
1842                         if (errorCode == SocketError.Success)
1843                                 req.CheckIfThrowDelayedException();
1844
1845                         return(req.Total);
1846                 }
1847
1848                 public
1849                 int EndSend (IAsyncResult result)
1850                 {
1851                         SocketError error;
1852                         int bytesSent = EndSend (result, out error);
1853                         if (error != SocketError.Success) {
1854                                 if (error != SocketError.WouldBlock && error != SocketError.InProgress)
1855                                         connected = false;
1856                                 throw new SocketException ((int)error);
1857                         }
1858                         return bytesSent;
1859                 }
1860
1861                 public
1862                 int EndSend (IAsyncResult asyncResult, out SocketError errorCode)
1863                 {
1864                         if (disposed && closed)
1865                                 throw new ObjectDisposedException (GetType ().ToString ());
1866                         if (asyncResult == null)
1867                                 throw new ArgumentNullException ("asyncResult");
1868
1869                         SocketAsyncResult req = asyncResult as SocketAsyncResult;
1870                         if (req == null)
1871                                 throw new ArgumentException ("Invalid IAsyncResult", "result");
1872
1873                         if (Interlocked.CompareExchange (ref req.EndCalled, 1, 0) == 1)
1874                                 throw InvalidAsyncOp ("EndSend");
1875                         if (!asyncResult.IsCompleted)
1876                                 asyncResult.AsyncWaitHandle.WaitOne ();
1877
1878                         errorCode = req.ErrorCode;
1879                         // If no socket error occurred, call CheckIfThrowDelayedException in case there are other
1880                         // kinds of exceptions that should be thrown.
1881                         if (errorCode == SocketError.Success)
1882                                 req.CheckIfThrowDelayedException ();
1883
1884                         return(req.Total);
1885                 }
1886
1887                 // Used by Udpclient
1888                 public
1889                 int EndReceiveFrom(IAsyncResult result, ref EndPoint end_point)
1890                 {
1891                         if (disposed && closed)
1892                                 throw new ObjectDisposedException (GetType ().ToString ());
1893
1894                         if (result == null)
1895                                 throw new ArgumentNullException ("result");
1896
1897                         if (end_point == null)
1898                                 throw new ArgumentNullException ("remote_end");
1899
1900                         SocketAsyncResult req = result as SocketAsyncResult;
1901                         if (req == null)
1902                                 throw new ArgumentException ("Invalid IAsyncResult", "result");
1903
1904                         if (Interlocked.CompareExchange (ref req.EndCalled, 1, 0) == 1)
1905                                 throw InvalidAsyncOp ("EndReceiveFrom");
1906                         if (!result.IsCompleted)
1907                                 result.AsyncWaitHandle.WaitOne();
1908
1909                         req.CheckIfThrowDelayedException();
1910                         end_point = req.EndPoint;
1911                         return req.Total;
1912                 }
1913
1914                 [MethodImplAttribute(MethodImplOptions.InternalCall)]
1915                 static extern void socket_pool_queue (SocketAsyncCall d, SocketAsyncResult r);
1916         }
1917 }
1918