blob: 7ac4e0f8edd329e5b7022ea0c130bf7ba2b840e8 [file] [log] [blame]
/*
* Copyright (c) 2018 Linaro Limited
*
* SPDX-License-Identifier: Apache-2.0
*/
#include <zephyr/ztest_assert.h>
#if defined(CONFIG_POSIX_API)
#include <zephyr/posix/sys/socket.h>
#include <zephyr/posix/arpa/inet.h>
#include <zephyr/posix/unistd.h>
#else
#include <zephyr/net/socket.h>
#endif
#define clear_buf(buf) memset(buf, 0, sizeof(buf))
static inline int prepare_listen_sock_udp_v4(struct sockaddr_in *addr)
{
int ret, sock;
zassert_not_null(addr, "null sockaddr");
ret = zsock_socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
zassert_true(ret >= 0, "socket open failed");
sock = ret;
zassert_equal(addr->sin_family, AF_INET, "Invalid family");
ret = zsock_bind(sock, (struct sockaddr *)addr, sizeof(*addr));
zassert_equal(ret, 0, "bind failed (%d/%d)", ret, errno);
return sock;
}
static inline int prepare_listen_sock_udp_v6(struct sockaddr_in6 *addr)
{
int ret, sock;
zassert_not_null(addr, "null sockaddr");
ret = zsock_socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP);
zassert_true(ret >= 0, "socket open failed");
sock = ret;
zassert_equal(addr->sin6_family, AF_INET6, "Invalid family");
ret = zsock_bind(sock, (struct sockaddr *)addr, sizeof(*addr));
zassert_equal(ret, 0, "bind failed (%d/%d)", ret, errno);
return sock;
}
static inline void prepare_sock_udp_v4(const char *addr, uint16_t port,
int *sock, struct sockaddr_in *sockaddr)
{
int rv;
zassert_not_null(addr, "null addr");
zassert_not_null(sock, "null sock");
zassert_not_null(sockaddr, "null sockaddr");
*sock = zsock_socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
zassert_true(*sock >= 0, "socket open failed");
sockaddr->sin_family = AF_INET;
sockaddr->sin_port = htons(port);
rv = zsock_inet_pton(AF_INET, addr, &sockaddr->sin_addr);
zassert_equal(rv, 1, "inet_pton failed");
}
static inline void prepare_sock_udp_v6(const char *addr, uint16_t port,
int *sock, struct sockaddr_in6 *sockaddr)
{
int rv;
zassert_not_null(addr, "null addr");
zassert_not_null(sock, "null sock");
zassert_not_null(sockaddr, "null sockaddr");
*sock = zsock_socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP);
zassert_true(*sock >= 0, "socket open failed");
(void)memset(sockaddr, 0, sizeof(*sockaddr));
sockaddr->sin6_family = AF_INET6;
sockaddr->sin6_port = htons(port);
rv = zsock_inet_pton(AF_INET6, addr, &sockaddr->sin6_addr);
zassert_equal(rv, 1, "inet_pton failed");
}
static inline void prepare_sock_tcp_v4(const char *addr, uint16_t port,
int *sock, struct sockaddr_in *sockaddr)
{
int rv;
zassert_not_null(addr, "null addr");
zassert_not_null(sock, "null sock");
zassert_not_null(sockaddr, "null sockaddr");
*sock = zsock_socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
zassert_true(*sock >= 0, "socket open failed");
sockaddr->sin_family = AF_INET;
sockaddr->sin_port = htons(port);
rv = zsock_inet_pton(AF_INET, addr, &sockaddr->sin_addr);
zassert_equal(rv, 1, "inet_pton failed");
}
static inline void prepare_sock_tcp_v6(const char *addr, uint16_t port,
int *sock, struct sockaddr_in6 *sockaddr)
{
int rv;
zassert_not_null(addr, "null addr");
zassert_not_null(sock, "null sock");
zassert_not_null(sockaddr, "null sockaddr");
*sock = zsock_socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP);
zassert_true(*sock >= 0, "socket open failed");
sockaddr->sin6_family = AF_INET6;
sockaddr->sin6_port = htons(port);
rv = zsock_inet_pton(AF_INET6, addr, &sockaddr->sin6_addr);
zassert_equal(rv, 1, "inet_pton failed");
}
static inline void prepare_sock_tls_v4(const char *addr, uint16_t port,
int *sock, struct sockaddr_in *sockaddr,
enum net_ip_protocol_secure proto)
{
int rv;
zassert_not_null(addr, "null addr");
zassert_not_null(sock, "null sock");
zassert_not_null(sockaddr, "null sockaddr");
*sock = zsock_socket(AF_INET, SOCK_STREAM, proto);
zassert_true(*sock >= 0, "socket open failed");
sockaddr->sin_family = AF_INET;
sockaddr->sin_port = htons(port);
rv = zsock_inet_pton(AF_INET, addr, &sockaddr->sin_addr);
zassert_equal(rv, 1, "inet_pton failed");
}
static inline void prepare_sock_tls_v6(const char *addr, uint16_t port,
int *sock, struct sockaddr_in6 *sockaddr,
enum net_ip_protocol_secure proto)
{
int rv;
zassert_not_null(addr, "null addr");
zassert_not_null(sock, "null sock");
zassert_not_null(sockaddr, "null sockaddr");
*sock = zsock_socket(AF_INET6, SOCK_STREAM, proto);
zassert_true(*sock >= 0, "socket open failed");
sockaddr->sin6_family = AF_INET6;
sockaddr->sin6_port = htons(port);
rv = zsock_inet_pton(AF_INET6, addr, &sockaddr->sin6_addr);
zassert_equal(rv, 1, "inet_pton failed");
}
static inline void prepare_sock_dtls_v4(const char *addr, uint16_t port,
int *sock, struct sockaddr_in *sockaddr,
enum net_ip_protocol_secure proto)
{
int rv;
zassert_not_null(addr, "null addr");
zassert_not_null(sock, "null sock");
zassert_not_null(sockaddr, "null sockaddr");
*sock = zsock_socket(AF_INET, SOCK_DGRAM, proto);
zassert_true(*sock >= 0, "socket open failed");
sockaddr->sin_family = AF_INET;
sockaddr->sin_port = htons(port);
rv = zsock_inet_pton(AF_INET, addr, &sockaddr->sin_addr);
zassert_equal(rv, 1, "inet_pton failed");
}
static inline void prepare_sock_dtls_v6(const char *addr, uint16_t port,
int *sock, struct sockaddr_in6 *sockaddr,
enum net_ip_protocol_secure proto)
{
int rv;
zassert_not_null(addr, "null addr");
zassert_not_null(sock, "null sock");
zassert_not_null(sockaddr, "null sockaddr");
*sock = zsock_socket(AF_INET6, SOCK_DGRAM, proto);
zassert_true(*sock >= 0, "socket open failed");
sockaddr->sin6_family = AF_INET6;
sockaddr->sin6_port = htons(port);
rv = zsock_inet_pton(AF_INET6, addr, &sockaddr->sin6_addr);
zassert_equal(rv, 1, "inet_pton failed");
}