From 9c7f7f85c81231776991e47b3437bd79c7f6782e Mon Sep 17 00:00:00 2001 From: Neels Hofmeyr Date: Tue, 11 Jan 2022 19:25:40 +0100 Subject: [PATCH] add osmo_sockaddr_set_port() Do it like osmo_sockaddr_port() in reverse. Related: SYS#5599 Change-Id: I9512e44c3203daebb3fe3435fceef167613c1a73 --- include/osmocom/core/socket.h | 1 + src/socket.c | 17 +++++++++++++++++ 2 files changed, 18 insertions(+) diff --git a/include/osmocom/core/socket.h b/include/osmocom/core/socket.h index fffc25c7e..183220d5a 100644 --- a/include/osmocom/core/socket.h +++ b/include/osmocom/core/socket.h @@ -96,6 +96,7 @@ size_t osmo_sockaddr_in_to_str_and_uint(char *addr, unsigned int addr_len, uint1 const char *osmo_sockaddr_ntop(const struct sockaddr *sa, char *dst); uint16_t osmo_sockaddr_port(const struct sockaddr *sa); +void osmo_sockaddr_set_port(struct sockaddr *sa, uint16_t port); int osmo_sock_unix_init(uint16_t type, uint8_t proto, const char *socket_path, unsigned int flags); diff --git a/src/socket.c b/src/socket.c index 80cba1789..4605db4c0 100644 --- a/src/socket.c +++ b/src/socket.c @@ -1254,6 +1254,23 @@ uint16_t osmo_sockaddr_port(const struct sockaddr *sa) return 0; } +/*! Set sockaddr port content (to network byte order). + * \param[out] sa sockaddr to set the port of. + * \param[in] port port nr to set. + */ +void osmo_sockaddr_set_port(struct sockaddr *sa, uint16_t port) +{ + struct osmo_sockaddr *osa = (struct osmo_sockaddr *)sa; + switch (osa->u.sa.sa_family) { + case AF_INET6: + osa->u.sin6.sin6_port = htons(port); + return; + case AF_INET: + osa->u.sin.sin_port = htons(port); + return; + } +} + /*! Initialize a unix domain socket (including bind/connect) * \param[in] type Socket type like SOCK_DGRAM, SOCK_STREAM * \param[in] proto Protocol like IPPROTO_TCP, IPPROTO_UDP