feat: TCP socket support using LibUV

src/Std/Internal/UV/Tcp.lean

@@ -0,0 +1,99 @@
+/-
+Copyright (c) 2024 Lean FRO, LLC. All rights reserved.
+Released under Apache 2.0 license as described in the file LICENSE.
+Authors: Henrik Boving, Sofia Rodrigues
+-/
+prelude
+import Init.System.IO
+import Init.System.Promise
+import Std.Net
+
+namespace Std
+namespace Internal
+namespace UV
+namespace Tcp
+
+open Std.Net
+
+private opaque SocketImpl : NonemptyType.{0}
+
+/--
+Represents a TCP socket.
+-/
+def Socket : Type := SocketImpl.type
+
+instance : Nonempty Socket := SocketImpl.property
+
+namespace Socket
+
+/--
+Creates a new TCP socket.
+-/
+@[extern "lean_uv_tcp_new"]
+opaque new : IO Socket
+
+/--
+Connect a TCP socket to the specified address.
+-/
+@[extern "lean_uv_tcp_connect"]
+opaque connect (socket : @& Socket) (addr : SocketAddress) : IO (IO.Promise (Except IO.Error Unit))
+
+/--
+Send data through a TCP socket.
+-/
+@[extern "lean_uv_tcp_send"]
+opaque send (socket : @& Socket) (data : ByteArray) : IO (IO.Promise (Except IO.Error Unit))
+
+/--
+Receive data from a TCP socket.
+-/
+@[extern "lean_uv_tcp_recv"]
+opaque recv (socket : @& Socket) : IO (IO.Promise (Except IO.Error ByteArray))
+
+/--
+Bind a TCP socket to a specific address.
+-/
+@[extern "lean_uv_tcp_bind"]
+opaque bind (socket : @& Socket) (addr : SocketAddress) : IO Unit
+
+/--
+Start listening for incoming connections on a TCP socket.
+-/
+@[extern "lean_uv_tcp_listen"]
+opaque listen (socket : @& Socket) (backlog : Int32) : IO Unit
+
+/--
+Accept an incoming connection on a listening TCP socket.
+-/
+@[extern "lean_uv_tcp_accept"]
+opaque accept (socket : @& Socket) : IO (IO.Promise (Except IO.Error Socket))
+
+/--
+Get the remote address of a connected TCP socket.
+-/
+@[extern "lean_uv_tcp_getpeername"]
+opaque getpeername (socket : @& Socket) : IO SocketAddress
+
+/--
+Get the local address of a bound TCP socket.
+-/
+@[extern "lean_uv_tcp_getsockname"]
+opaque getsockname (socket : @& Socket) : IO SocketAddress
+
+/--
+Enable or disable the Nagle algorithm for a TCP socket.
+-/
+@[extern "lean_uv_tcp_nodelay"]
+opaque nodelay (socket : @& Socket) : IO Unit
+
+/--
+Enable or disable TCP keep-alive for a socket.
+-/
+@[extern "lean_uv_tcp_keepalive"]
+opaque keepalive (socket : @& Socket) (enable : Int32) (delay : UInt32) : IO Unit
+
+end Socket
+end Tcp
+end UV
+end Internal
+end Std

src/runtime/CMakeLists.txt

@@ -3,7 +3,7 @@ object.cpp apply.cpp exception.cpp interrupt.cpp memory.cpp
 stackinfo.cpp compact.cpp init_module.cpp load_dynlib.cpp io.cpp hash.cpp
 platform.cpp alloc.cpp allocprof.cpp sharecommon.cpp stack_overflow.cpp
 process.cpp object_ref.cpp mpn.cpp mutex.cpp libuv.cpp uv/net_addr.cpp uv/event_loop.cpp
-uv/timer.cpp)
+uv/timer.cpp uv/tcp.cpp)
 add_library(leanrt_initial-exec STATIC ${RUNTIME_OBJS})
 set_target_properties(leanrt_initial-exec PROPERTIES
   ARCHIVE_OUTPUT_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})

src/runtime/libuv.cpp

@@ -15,6 +15,7 @@ namespace lean {
 
 extern "C" void initialize_libuv() {
     initialize_libuv_timer();
+    initialize_libuv_tcp_socket();
     initialize_libuv_loop();
 
     lthread([]() { event_loop_run_loop(&global_ev); });

src/runtime/libuv.h

@@ -7,7 +7,8 @@ Author: Markus Himmel, Sofia Rodrigues
 #pragma once
 #include <lean/lean.h>
 #include "runtime/uv/event_loop.h"
- #include "runtime/uv/timer.h"
+#include "runtime/uv/timer.h"
+#include "runtime/uv/tcp.h"
 #include "runtime/alloc.h"
 #include "runtime/io.h"
 #include "runtime/utf8.h"

src/runtime/uv/net_addr.cpp

@@ -28,6 +28,24 @@ void lean_ipv6_addr_to_in6_addr(b_obj_arg ipv6_addr, in6_addr* out) {
     }
 }
 
+extern "C" void lean_socket_addr_to_sockaddr(b_obj_arg ip_addr, struct sockaddr* out) {
+    lean_object * socket_addr_obj = lean_ctor_get(ip_addr, 0);
+    lean_object * ip_addr_obj = lean_ctor_get(socket_addr_obj, 0);
+    uint16_t port_obj = lean_ctor_get_uint16(socket_addr_obj, sizeof(void*)*1);
+
+    if (lean_ptr_tag(ip_addr) == 0) {
+        struct sockaddr_in * cast = (struct sockaddr_in*)out;
+        lean_ipv4_addr_to_in_addr(ip_addr_obj, &cast->sin_addr);
+        cast->sin_family = AF_INET;
+        cast->sin_port = htons(port_obj);
+    } else {
+        struct sockaddr_in6 * cast = (struct sockaddr_in6*)out;
+        lean_ipv6_addr_to_in6_addr(ip_addr_obj, (in6_addr*)&cast->sin6_addr);
+        cast->sin6_family = AF_INET6;
+        cast->sin6_port = htons(port_obj);
+    }
+}
+
 lean_obj_res lean_in_addr_to_ipv4_addr(const in_addr* ipv4_addr) {
     obj_res ret = alloc_array(0, 4);
     uint32_t hostaddr = ntohl(ipv4_addr->s_addr);
@@ -55,6 +73,41 @@ lean_obj_res lean_in6_addr_to_ipv6_addr(const in6_addr* ipv6_addr) {
     return ret;
 }
 
+lean_obj_res lean_mk_socketaddress(lean_obj_res ip_addr, lean_obj_res port) {
+    lean_obj_res socket_addr = lean_alloc_ctor(0, 2, 0);
+    lean_ctor_set(socket_addr, 0, ip_addr);
+    lean_ctor_set(socket_addr, 1, port);
+    return socket_addr;
+}
+
+lean_obj_res lean_sockaddr_to_socketaddress(const struct sockaddr* sockaddr) {
+    lean_object* part;
+    int tag;
+
+    if (sockaddr->sa_family == AF_INET) {
+        const struct sockaddr_in* addr_in = (const struct sockaddr_in*)sockaddr;
+        const in_addr* ipv4_addr = &addr_in->sin_addr;
+
+        lean_obj_res lean_ipv4 = lean_in_addr_to_ipv4_addr(ipv4_addr);
+        uint16_t port = ntohs(addr_in->sin_port);
+        part = lean_mk_socketaddress(lean_ipv4, lean_box(port));
+        tag = 0;
+    } else if (sockaddr->sa_family == AF_INET6) {
+        const struct sockaddr_in6* addr_in6 = (const struct sockaddr_in6*)sockaddr;
+        const in6_addr* ipv6_addr = &addr_in6->sin6_addr;
+        lean_obj_res lean_ipv6 = lean_in6_addr_to_ipv6_addr(ipv6_addr);
+        uint16_t port = ntohs(addr_in6->sin6_port);
+        part = lean_mk_socketaddress(lean_ipv6, lean_box(port));
+        tag = 1;
+    }
+
+    lean_object* ctor = lean_alloc_ctor(tag, 1, 0);
+    lean_ctor_set(ctor, 0, part);
+
+    return ctor;
+}
+
+
 /* Std.Net.IPV4Addr.ofString (s : @&String) : Option IPV4Addr */
 extern "C" LEAN_EXPORT lean_obj_res lean_uv_pton_v4(b_obj_arg str_obj) {
     const char* str = string_cstr(str_obj);

src/runtime/uv/net_addr.h

@@ -15,8 +15,10 @@ namespace lean {
 
 void lean_ipv4_addr_to_in_addr(b_obj_arg ipv4_addr, struct in_addr* out);
 void lean_ipv6_addr_to_in6_addr(b_obj_arg ipv6_addr, struct in6_addr* out);
+extern "C" void lean_socket_addr_to_sockaddr(b_obj_arg ip_addr, struct sockaddr* out);
 lean_obj_res lean_in_addr_to_ipv4_addr(const struct in_addr* ipv4_addr);
 lean_obj_res lean_in6_addr_to_ipv6_addr(const struct in6_addr* ipv6_addr);
+lean_obj_res lean_sockaddr_to_socketaddress(const struct sockaddr* sockaddr);
 
 #endif