mirrors/janet
1
0
Fork 0
mirror of https://github.com/janet-lang/janet synced 2025-08-06 22:14:28 +00:00
Code Issues 1 Releases Wiki Activity

Add getsockname (net/localname)

Browse Source
This commit is contained in:
llmII 2021-09-06 17:01:09 -05:00
parent ba761d5c35
commit 02640812af
No known key found for this signature in database
GPG Key ID: E3AD2E259F58A9A0
1 changed files with 20 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/core/net.c
View File

@ -720,7 +720,7 @@ static inline void *sa_aton_(void *dst, size_t lim, const char *src) {
} }
*/ */
static JanetString janet_name_from_socket(const struct sockaddr_storage *ss, socklen_t slen) { static JanetString janet_so_getname(const struct sockaddr_storage *ss, socklen_t slen) {
uint8_t *hn = NULL; uint8_t *hn = NULL;
uint16_t hp = 0; uint16_t hp = 0;
size_t plen = SA_ADDRSTRLEN; size_t plen = SA_ADDRSTRLEN;
@ -763,6 +763,23 @@ static JanetString janet_name_from_socket(const struct sockaddr_storage *ss, soc
return janet_string(hn, plen + 1); return janet_string(hn, plen + 1);
} }
JANET_CORE_FN(cfun_net_getsockname,
"(net/localname stream)",
"Document me!") {
janet_arity(argc, 1, 1);
JanetStream *js = janet_getabstract(argv, 0, &janet_stream_type);
struct sockaddr_storage ss;
socklen_t slen = sizeof ss;
memset(&ss, 0, slen);
int error;
if(0 != (error = getsockname(js->handle, (struct sockaddr *) &ss, &slen)))
janet_panicf("Failed to get peername on fd %d, error: %s", js->handle, janet_ev_lasterr());
return janet_wrap_string(janet_so_getname(&ss, slen));
}
JANET_CORE_FN(cfun_net_getpeername, JANET_CORE_FN(cfun_net_getpeername,
"(net/peername stream)", "(net/peername stream)",
"Document me!") { "Document me!") {
@ -777,9 +794,7 @@ JANET_CORE_FN(cfun_net_getpeername,
janet_panicf("Failed to get peername on fd %d, error: %s", js->handle, janet_ev_lasterr()); janet_panicf("Failed to get peername on fd %d, error: %s", js->handle, janet_ev_lasterr());
} }
JanetString ret = janet_name_from_socket(&ss, slen); return janet_wrap_string(janet_so_getname(&ss, slen));
return janet_wrap_string(ret);
} }
JANET_CORE_FN(cfun_stream_accept_loop, JANET_CORE_FN(cfun_stream_accept_loop,
@ -952,6 +967,7 @@ void janet_lib_net(JanetTable *env) {
JANET_CORE_REG("net/connect", cfun_net_connect), JANET_CORE_REG("net/connect", cfun_net_connect),
JANET_CORE_REG("net/shutdown", cfun_net_shutdown), JANET_CORE_REG("net/shutdown", cfun_net_shutdown),
JANET_CORE_REG("net/peername", cfun_net_getpeername), JANET_CORE_REG("net/peername", cfun_net_getpeername),
JANET_CORE_REG("net/localname", cfun_net_getsockname),
JANET_REG_END JANET_REG_END
}; };
janet_core_cfuns_ext(env, NULL, net_cfuns); janet_core_cfuns_ext(env, NULL, net_cfuns);