Fix: NetworkStream throwing inconsistent exceptions (#40772)

Fix a bug: Span<byte> overloads of NetworkStream throwing ObjectDisposedException instead of NetworkException, when not using derived NetworkStream.

Author: antonfirsov

src/libraries/System.Net.Sockets/src/System/Net/Sockets/NetworkStream.cs

@@ -250,7 +250,17 @@ public override int Read(Span<byte> buffer)
             ThrowIfDisposed();
             if (!CanRead) throw new InvalidOperationException(SR.net_writeonlystream);
 
-            int bytesRead = _streamSocket.Receive(buffer, SocketFlags.None, out SocketError errorCode);
+            int bytesRead;
+            SocketError errorCode;
+            try
+            {
+                bytesRead = _streamSocket.Receive(buffer, SocketFlags.None, out errorCode);
+            }
+            catch (Exception exception) when (!(exception is OutOfMemoryException))
+            {
+                throw GetCustomException(SR.Format(SR.net_io_readfailure, exception.Message), exception);
+            }
+
             if (errorCode != SocketError.Success)
             {
                 var socketException = new SocketException((int)errorCode);
@@ -320,7 +330,16 @@ public override void Write(ReadOnlySpan<byte> buffer)
             ThrowIfDisposed();
             if (!CanWrite) throw new InvalidOperationException(SR.net_readonlystream);
 
-            _streamSocket.Send(buffer, SocketFlags.None, out SocketError errorCode);
+            SocketError errorCode;
+            try
+            {
+                _streamSocket.Send(buffer, SocketFlags.None, out errorCode);
+            }
+            catch (Exception exception) when (!(exception is OutOfMemoryException))
+            {
+                throw GetCustomException(SR.Format(SR.net_io_writefailure, exception.Message), exception);
+            }
+
             if (errorCode != SocketError.Success)
             {
                 var socketException = new SocketException((int)errorCode);

src/libraries/System.Net.Sockets/tests/FunctionalTests/NetworkStreamTest.cs

@@ -307,8 +307,10 @@ await RunWithConnectedNetworkStreamsAsync((server, _) =>
             });	
         }
 
-        [Fact]
-        public async Task DisposeSocketDirectly_ReadWriteThrowNetworkException()
+        [Theory]
+        [InlineData(true)]
+        [InlineData(false)]
+        public async Task DisposeSocketDirectly_ReadWriteThrowNetworkException(bool derivedNetworkStream)
         {
             using (Socket listener = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp))
             using (Socket client = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp))
@@ -318,20 +320,22 @@ public async Task DisposeSocketDirectly_ReadWriteThrowNetworkException()
 
                 Task<Socket> acceptTask = listener.AcceptAsync();
                 await Task.WhenAll(acceptTask, client.ConnectAsync(new IPEndPoint(IPAddress.Loopback, ((IPEndPoint)listener.LocalEndPoint).Port)));
-                using (Socket serverSocket = await acceptTask)
-                using (DerivedNetworkStream server = new DerivedNetworkStream(serverSocket))
-                {
-                    serverSocket.Dispose();
+                using Socket serverSocket = await acceptTask;
+                using NetworkStream server = derivedNetworkStream ? (NetworkStream)new DerivedNetworkStream(serverSocket) : new NetworkStream(serverSocket);
+                
+                serverSocket.Dispose();
 
-                    Assert.Throws<IOException>(() => server.Read(new byte[1], 0, 1));
-                    Assert.Throws<IOException>(() => server.Write(new byte[1], 0, 1));
+                Assert.Throws<IOException>(() => server.Read(new byte[1], 0, 1));
+                Assert.Throws<IOException>(() => server.Write(new byte[1], 0, 1));
 
-                    Assert.Throws<IOException>(() => server.BeginRead(new byte[1], 0, 1, null, null));
-                    Assert.Throws<IOException>(() => server.BeginWrite(new byte[1], 0, 1, null, null));
+                Assert.Throws<IOException>(() => server.Read((Span<byte>)new byte[1]));
+                Assert.Throws<IOException>(() => server.Write((ReadOnlySpan<byte>)new byte[1]));
 
-                    Assert.Throws<IOException>(() => { server.ReadAsync(new byte[1], 0, 1); });
-                    Assert.Throws<IOException>(() => { server.WriteAsync(new byte[1], 0, 1); });
-                }
+                Assert.Throws<IOException>(() => server.BeginRead(new byte[1], 0, 1, null, null));
+                Assert.Throws<IOException>(() => server.BeginWrite(new byte[1], 0, 1, null, null));
+
+                Assert.Throws<IOException>(() => { server.ReadAsync(new byte[1], 0, 1); });
+                Assert.Throws<IOException>(() => { server.WriteAsync(new byte[1], 0, 1); });
             }
         }