a6789a73ff
This PR implements `Std.Net.Addr` which contains structures around IP and socket addresses. While we could implement our own parser instead of going through the `addr_in`/`addr_in6` route we will need to implement these conversions to make proper system calls anyway. Hence this is likely the approach with the least amount of non trivial code overall. The only thing I am uncertain about is whether `ofString` should return `Option` or `Except`, unfortunately `libuv` doesn't hand out error messages on IP parsing.
131 lines
4.1 KiB
C++
131 lines
4.1 KiB
C++
/*
|
|
Copyright (c) 2025 Lean FRO, LLC. All rights reserved.
|
|
Released under Apache 2.0 license as described in the file LICENSE.
|
|
Author: Henrik Böving
|
|
*/
|
|
|
|
#include "runtime/uv/net_addr.h"
|
|
#include <cstring>
|
|
|
|
namespace lean {
|
|
|
|
#ifndef LEAN_EMSCRIPTEN
|
|
|
|
void lean_ipv4_addr_to_in_addr(b_obj_arg ipv4_addr, in_addr* out) {
|
|
out->s_addr = 0;
|
|
for (int i = 0; i < 4; i++) {
|
|
uint32_t octet = (uint32_t)(uint8_t)lean_unbox(array_uget(ipv4_addr, i));
|
|
out->s_addr |= octet << (3 - i) * 8;
|
|
}
|
|
out->s_addr = htonl(out->s_addr);
|
|
}
|
|
|
|
void lean_ipv6_addr_to_in6_addr(b_obj_arg ipv6_addr, in6_addr* out) {
|
|
for (int i = 0; i < 8; i++) {
|
|
uint16_t segment = htons((uint16_t)lean_unbox(array_uget(ipv6_addr, i)));
|
|
out->s6_addr[2 * i] = (uint8_t)segment;
|
|
out->s6_addr[2 * i + 1] = (uint8_t)(segment >> 8);
|
|
}
|
|
}
|
|
|
|
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);
|
|
|
|
for (int i = 0; i < 4; i++) {
|
|
uint8_t octet = (uint8_t)(hostaddr >> (3 - i) * 8);
|
|
array_push(ret, lean_box(octet));
|
|
}
|
|
|
|
lean_assert(array_size(ret) == 4);
|
|
return ret;
|
|
}
|
|
|
|
lean_obj_res lean_in6_addr_to_ipv6_addr(const in6_addr* ipv6_addr) {
|
|
obj_res ret = alloc_array(0, 8);
|
|
|
|
for (int i = 0; i < 16; i += 2) {
|
|
uint16_t part1 = (uint16_t)ipv6_addr->s6_addr[i];
|
|
uint16_t part2 = (uint16_t)ipv6_addr->s6_addr[i + 1];
|
|
uint16_t segment = ntohs((part2 << 8) | part1);
|
|
array_push(ret, lean_box(segment));
|
|
}
|
|
|
|
lean_assert(array_size(ret) == 8);
|
|
return ret;
|
|
}
|
|
|
|
/* 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);
|
|
in_addr internal;
|
|
if (uv_inet_pton(AF_INET, str, &internal) == 0) {
|
|
return mk_option_some(lean_in_addr_to_ipv4_addr(&internal));
|
|
} else {
|
|
return mk_option_none();
|
|
}
|
|
}
|
|
|
|
/* Std.Net.IPV4Addr.toString (addr : @&IPV4Addr) : String */
|
|
extern "C" LEAN_EXPORT lean_obj_res lean_uv_ntop_v4(b_obj_arg ipv4_addr) {
|
|
in_addr internal;
|
|
lean_ipv4_addr_to_in_addr(ipv4_addr, &internal);
|
|
char dst[INET_ADDRSTRLEN];
|
|
int ret = uv_inet_ntop(AF_INET, &internal, dst, sizeof(dst));
|
|
lean_always_assert(ret == 0);
|
|
return lean_mk_string(dst);
|
|
}
|
|
|
|
/* Std.Net.IPV6Addr.ofString (s : @&String) : Option IPV6Addr */
|
|
extern "C" LEAN_EXPORT lean_obj_res lean_uv_pton_v6(b_obj_arg str_obj) {
|
|
const char* str = string_cstr(str_obj);
|
|
in6_addr internal;
|
|
if (uv_inet_pton(AF_INET6, str, &internal) == 0) {
|
|
return mk_option_some(lean_in6_addr_to_ipv6_addr(&internal));
|
|
} else {
|
|
return mk_option_none();
|
|
}
|
|
}
|
|
|
|
/* Std.Net.IPV6Addr.toString (addr : @&IPV6Addr) : String */
|
|
extern "C" LEAN_EXPORT lean_obj_res lean_uv_ntop_v6(b_obj_arg ipv6_addr) {
|
|
in6_addr internal;
|
|
lean_ipv6_addr_to_in6_addr(ipv6_addr, &internal);
|
|
char dst[INET6_ADDRSTRLEN];
|
|
int ret = uv_inet_ntop(AF_INET6, &internal, dst, sizeof(dst));
|
|
lean_always_assert(ret == 0);
|
|
return lean_mk_string(dst);
|
|
}
|
|
|
|
#else
|
|
|
|
/* Std.Net.IPV4Addr.ofString (s : @&String) : Option IPV4Addr */
|
|
extern "C" LEAN_EXPORT lean_obj_res lean_uv_pton_v4(b_obj_arg str_obj) {
|
|
lean_always_assert(
|
|
false && ("Please build a version of Lean4 with libuv to invoke this.")
|
|
);
|
|
}
|
|
|
|
/* Std.Net.IPV4Addr.toString (addr : @&IPV4Addr) : String */
|
|
extern "C" LEAN_EXPORT lean_obj_res lean_uv_ntop_v4(b_obj_arg ipv4_addr) {
|
|
lean_always_assert(
|
|
false && ("Please build a version of Lean4 with libuv to invoke this.")
|
|
);
|
|
}
|
|
|
|
/* Std.Net.IPV6Addr.ofString (s : @&String) : Option IPV6Addr */
|
|
extern "C" LEAN_EXPORT lean_obj_res lean_uv_pton_v6(b_obj_arg str_obj) {
|
|
lean_always_assert(
|
|
false && ("Please build a version of Lean4 with libuv to invoke this.")
|
|
);
|
|
}
|
|
|
|
/* Std.Net.IPV6Addr.toString (addr : @&IPV6Addr) : String */
|
|
extern "C" LEAN_EXPORT lean_obj_res lean_uv_ntop_v6(b_obj_arg ipv6_addr) {
|
|
lean_always_assert(
|
|
false && ("Please build a version of Lean4 with libuv to invoke this.")
|
|
);
|
|
}
|
|
|
|
#endif
|
|
}
|