add tcp test

JIRA: 343

network/Makefile

@@ -0,0 +1,5 @@
+NAME := test-tcp
+LOCAL_SRCS := test-tcp.c common.c
+DEP_LIBS := unity
+
+include $(binary.mk)

network/common.c

@@ -0,0 +1,94 @@
+/*
+ * Phoenix-RTOS
+ *
+ * network tests common routines
+ *
+ * Copyright 2024 Phoenix Systems
+ * Author: Adam Debek
+ *
+ * This file is part of Phoenix-RTOS.
+ *
+ * %LICENSE%
+ */
+
+#include <string.h>
+#include <errno.h>
+#include <unistd.h>
+#include <ifaddrs.h>
+#include <sys/ioctl.h>
+#include <sys/socket.h>
+#include <net/if.h>
+#include <arpa/inet.h>
+#include <unity_fixture.h>
+#include "common.h"
+
+
+int create_con(const char *daddr, uint16_t dport)
+{
+	int try = 10;
+	int sockfd;
+	struct sockaddr_in dest_addr;
+
+	if ((sockfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0) {
+		return -1;
+	}
+
+	memset(&dest_addr, 0, sizeof dest_addr);
+	dest_addr.sin_family = AF_INET;
+	dest_addr.sin_addr.s_addr = inet_addr(daddr);
+	dest_addr.sin_port = htons(dport);
+
+	while (connect(sockfd, (struct sockaddr *)&dest_addr, sizeof dest_addr) < 0) {
+		if (try-- > 0) {
+			close(sockfd);
+			if ((sockfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0) {
+				perror("socket");
+				return -1;
+			}
+			sleep(1);
+		}
+		else {
+			close(sockfd);
+			return -1;
+		}
+	}
+
+	return sockfd;
+}
+
+
+int wait_if_running(void)
+{
+	struct ifreq ioctlInterface;
+	short interfaceFlags;
+	char *ifname = "en1";
+	int try = 300;
+	int ret;
+
+	strcpy(ioctlInterface.ifr_name, ifname);
+
+	int sd = socket(AF_INET, SOCK_STREAM, 0);
+	if (sd < 0) {
+		perror("socket");
+		return -1;
+	}
+
+	while (try-- > 0) {
+		ret = ioctl(sd, SIOCGIFFLAGS, &ioctlInterface);
+		if (ret < 0) {
+			/* Unable to obtain flags */
+			perror("ioctl(SIOCGIFFLAGS)");
+			return ret;
+		}
+
+		interfaceFlags = ioctlInterface.ifr_flags;
+		if (interfaceFlags & IFF_RUNNING) {
+			return 0;
+		}
+
+		usleep(10000);
+	}
+
+	/* Interface not running */
+	return -1;
+}

network/common.h

@@ -0,0 +1,111 @@
+/*
+ * Phoenix-RTOS
+ *
+ * network tests common header
+ *
+ * Copyright 2024 Phoenix Systems
+ * Author: Adam Debek
+ *
+ * This file is part of Phoenix-RTOS.
+ *
+ * %LICENSE%
+ */
+
+#include <stdint.h>
+
+#ifndef _TEST_NETWORK_COMMON_H
+#define _TEST_NETWORK_COMMON_H
+
+#define send_asserted(_sockfd, _buffer, _length, _flags) \
+	do { \
+		ssize_t n = 0; \
+		size_t total = 0; \
+		size_t left = _length; \
+		size_t send_len = _length; \
+		struct pollfd fds[1]; \
+		fds[0].fd = _sockfd; \
+		fds[0].events = POLLOUT; \
+		fds[0].revents = 0; \
+		while (n < _length && poll(fds, 1, 500)) { \
+			if (fds[0].revents & POLLOUT) { \
+				n = send(_sockfd, _buffer + total, send_len, _flags); \
+				if (n < 0 && errno == EPIPE) { \
+					TEST_FAIL_MESSAGE("Host closed connection"); \
+				} \
+				else if (n < 0 && errno == EMSGSIZE) { \
+					send_len /= 2; \
+					continue; \
+				} \
+				else if (n < 0) { \
+					TEST_FAIL_MESSAGE(strerror(errno)); \
+				} \
+			} \
+			total += n; \
+			left -= n; \
+			if (left < send_len) { \
+				send_len = left; \
+			} \
+		} \
+		TEST_ASSERT(n == _length); \
+	} while (0)
+
+
+#define recv_asserted(_sockfd, _buffer, _length, _flags) \
+	do { \
+		ssize_t r = 0; \
+		struct pollfd fds[1]; \
+		fds[0].fd = _sockfd; \
+		fds[0].events = POLLIN; \
+		fds[0].revents = 0; \
+		while (r < _length && poll(fds, 1, 500)) { \
+			if (fds[0].revents & POLLIN) { \
+				r = recv(_sockfd, _buffer, _length, _flags); \
+				if (r == 0) { \
+					TEST_FAIL_MESSAGE("Host closed connection"); \
+				} \
+				else if (r < 0) { \
+					TEST_FAIL_MESSAGE(strerror(errno)); \
+				} \
+			} \
+			else { \
+				TEST_FAIL_MESSAGE("Poll didn't receive remaining data"); \
+			} \
+		} \
+		TEST_ASSERT(r == _length); \
+	} while (0)
+
+
+#define get_host_response(_syncfd, _buffer) \
+	do { \
+		ssize_t r; \
+		r = recv(_syncfd, _buffer, sizeof _buffer, MSG_WAITALL); \
+		if (r == 0) { \
+			fprintf(stderr, "Sync socket: host closed connection\n"); \
+			close(_syncfd); \
+			exit(1); \
+		} \
+		else if (r < 0) { \
+			perror("Sync socket"); \
+			close(_syncfd); \
+			exit(1); \
+		} \
+		else if (target_failed_flag == 0) { \
+			host_response_flag = 1; \
+			if (_buffer[0] != 0) { \
+				TEST_FAIL_MESSAGE(_buffer); \
+			} \
+		} \
+		else if (target_failed_flag == 1) { \
+			if (_buffer[0] != 0) { \
+				fprintf(stderr, "%s\n", _buffer); \
+			} \
+		} \
+	} while (0)
+
+/* If sport is 0, source port will be assigned by kernel */
+int create_con(const char *daddr, uint16_t dport);
+
+/* Wait for running interface */
+int wait_if_running(void);
+
+#endif /* TEST_NETWORK_COMMON_H */