more

Author: pavelsavara

src/libraries/System.Net.Sockets/src/System/Net/Sockets/SocketAsyncContext.Unix.cs

@@ -1348,10 +1348,12 @@ public void SetHandleNonBlocking()
             //
             if (!_isHandleNonBlocking)
             {
+#if !TARGET_WASI // WASI-libc doesn't have emulate this for tcp handles, see https://github.com/WebAssembly/wasi-libc/issues/536
                 if (Interop.Sys.Fcntl.SetIsNonBlocking(_socket, 1) != 0)
                 {
                     throw new SocketException((int)SocketPal.GetSocketErrorForErrorCode(Interop.Sys.GetLastError()));
                 }
+#endif
 
                 _isHandleNonBlocking = true;
             }

src/mono/sample/wasi/Directory.Build.targets

@@ -39,6 +39,7 @@
 
 
       <WasiCommand Condition="'$(_WasiNeedsHttp)' == 'true'"                                         >$(WasiCommand) --wasi http</WasiCommand>
+      <WasiCommand Condition="'$(_WasiNeedsSocket)' == 'true'"                                       >$(WasiCommand) --wasi inherit-network --wasi tcp --wasi udp --wasi allow-ip-name-lookup</WasiCommand>
       <WasiCommand Condition="'$(WasmSingleFileBundle)' != 'true'"                                   >$(WasiCommand) --dir .</WasiCommand>
       <WasiCommand                                                                                   >$(WasiCommand) $(_DotnetWasmName)</WasiCommand>
       <WasiCommand Condition="'$(WasmSingleFileBundle)' != 'true'"                                   >$(WasiCommand) $(_SampleProjectName)</WasiCommand>

src/mono/sample/wasi/sockets-p2/Program.cs

@@ -0,0 +1,61 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+
+using System;
+using System.Net.Http.Headers;
+using System.Net.Http;
+using System.Threading.Tasks;
+using System.Threading;
+using System.Runtime.CompilerServices;
+using System.Net;
+using System.Net.Sockets;
+using System.Text;
+
+public static class WasiMainWrapper
+{
+    public static async Task<int> MainAsync(string[] args)
+    {
+        IPHostEntry ipHostInfo = await Dns.GetHostEntryAsync("example.com");
+        IPAddress ipAddress = ipHostInfo.AddressList[0];
+        Console.WriteLine($"IP Address: {ipAddress}");
+
+        IPEndPoint ipEndPoint = new(ipAddress, 80);
+        using Socket client = new(ipEndPoint.AddressFamily, SocketType.Stream, ProtocolType.Tcp);
+
+        await client.ConnectAsync(ipEndPoint);
+
+        // Send message.
+        var message = @"GET / HTTP/1.1
+Host: example.com
+Accept: */*
+
+";
+        var messageBytes = Encoding.UTF8.GetBytes(message);
+        var start = 0;
+        while (start < messageBytes.Length)
+        {
+            start += await client.SendAsync(messageBytes.AsMemory(start), SocketFlags.None);
+            Console.WriteLine("TODO poll here");
+        }
+        Console.WriteLine("GET sent");
+
+        // Receive ack.
+        var buffer = new byte[2048];
+        var received = await client.ReceiveAsync(buffer, SocketFlags.None);
+        var response = Encoding.UTF8.GetString(buffer, 0, received);
+        Console.WriteLine(response);
+
+        client.Shutdown(SocketShutdown.Both);
+
+        return 0;
+    }
+
+    public static int Main(string[] args)
+    {
+        return PollWasiEventLoopUntilResolved((Thread)null!, MainAsync(args));
+
+        [UnsafeAccessor(UnsafeAccessorKind.StaticMethod, Name = "PollWasiEventLoopUntilResolved")]
+        static extern T PollWasiEventLoopUntilResolved<T>(Thread t, Task<T> mainTask);
+    }
+
+}

src/mono/sample/wasi/sockets-p2/Wasip2.Sockets.Console.Sample.csproj

@@ -0,0 +1,13 @@
+<Project Sdk="Microsoft.NET.Sdk">
+  <PropertyGroup>
+    <TargetFramework>$(NetCoreAppCurrent)</TargetFramework>
+    <_WasiNeedsHttp>true</_WasiNeedsHttp>
+    <_WasiNeedsSocket>true</_WasiNeedsSocket>
+    <!-- 
+    <WasmSingleFileBundle>true</WasmSingleFileBundle>
+    <InvariantGlobalization>true</InvariantGlobalization>
+    -->
+  </PropertyGroup>
+
+  <Target Name="RunSample" DependsOnTargets="RunSampleWithWasmtime" />
+</Project>

src/native/libs/System.Native/pal_networking.c

@@ -2783,6 +2783,9 @@ int32_t SystemNative_Socket(int32_t addressFamily, int32_t socketType, int32_t p
 
 #ifdef SOCK_CLOEXEC
     platformSocketType |= SOCK_CLOEXEC;
+#endif
+#if defined(TARGET_WASI)
+    platformSocketType |= SOCK_NONBLOCK; // WASI sockets are always non-blocking, because in ST we don't have another thread which could be blocked
 #endif
     *createdSocket = socket(platformAddressFamily, platformSocketType, platformProtocolType);
     if (*createdSocket == -1)