[PATCH 3/3] mcast01: Add test program
Sebastian Huber
sebastian.huber at embedded-brains.de
Thu Sep 10 07:51:44 UTC 2020
---
libbsd.py | 1 +
testsuite/mcast01/test_main.c | 285 ++++++++++++++++++++++++++++++++++
2 files changed, 286 insertions(+)
create mode 100644 testsuite/mcast01/test_main.c
diff --git a/libbsd.py b/libbsd.py
index 6a471873..c251fba4 100644
--- a/libbsd.py
+++ b/libbsd.py
@@ -5141,6 +5141,7 @@ class tests(builder.Module):
self.addTest(mm.generator['test']('media01', ['test_main'],
runTest = False,
extraLibs = ['ftpd', 'telnetd']))
+ self.addTest(mm.generator['test']('mcast01', ['test_main']))
self.addTest(mm.generator['test']('vlan01', ['test_main'], netTest = True))
self.addTest(mm.generator['test']('lagg01', ['test_main'], netTest = True))
self.addTest(mm.generator['test']('log01', ['test_main']))
diff --git a/testsuite/mcast01/test_main.c b/testsuite/mcast01/test_main.c
new file mode 100644
index 00000000..60519719
--- /dev/null
+++ b/testsuite/mcast01/test_main.c
@@ -0,0 +1,285 @@
+/* SPDX-License-Identifier: BSD-2-Clause */
+
+/**
+ * @file
+ *
+ * @brief Example code to test multicasts.
+ *
+ * ping 224.0.0.1
+ *
+ * Linux:
+ *
+ * echo 0 > /proc/sys/net/ipv4/icmp_echo_ignore_broadcasts
+ * echo 1 > /proc/sys/net/ipv4/ip_forward
+ * ip route add 224.0.0.0/4 dev eth0
+ * echo "Hello, world!" > /dev/udp/239.1.2.3/1234
+ */
+
+/*
+ * Copyright (C) 2020 embedded brains GmbH (http://www.embedded-brains.de)
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
+ * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include <sys/param.h>
+#include <sys/socket.h>
+#include <sys/sysctl.h>
+
+#include <assert.h>
+#include <inttypes.h>
+#include <stdbool.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sysexits.h>
+
+#include <arpa/inet.h>
+#include <netinet/in.h>
+
+#include <machine/rtems-bsd-commands.h>
+
+#include <rtems/console.h>
+#include <rtems/shell.h>
+#include <rtems/dhcpcd.h>
+
+#define TEST_NAME "LIBBSD MCAST 1"
+#define TEST_STATE_USER_INPUT 1
+
+#define MCAST_PORT 1234
+
+#define MAX_PACKET_SIZE UINT16_MAX
+
+static const char mcast_addr[] = "239.1.2.3";
+
+static const char hello[] = "Hello, world!";
+
+typedef struct {
+ in_addr_t addr;
+ char name[];
+} iface_binding;
+
+static void
+set_mcast_route(const char *iface_name)
+{
+ int exit_code;
+ char *route[] = {
+ "route",
+ "add",
+ "-net",
+ "224.0.0.0/4",
+ "-iface",
+ iface_name,
+ NULL
+ };
+
+ exit_code = rtems_bsd_command_route(RTEMS_BSD_ARGC(route), route);
+ assert(exit_code == EXIT_SUCCESS);
+}
+
+static void
+mcast_rx_task(rtems_task_argument arg)
+{
+ iface_binding *ifb;
+ struct ip_mreq mreq;
+ int sd;
+ struct sockaddr_in addr;
+ int rv;
+ char addr_buf[INET_ADDRSTRLEN];
+ const char *ip;
+ ssize_t n;
+
+ ifb = (iface_binding *)arg;
+ set_mcast_route(ifb->name);
+
+ memset(&addr, 0, sizeof(addr));
+ addr.sin_family = AF_INET;
+ addr.sin_addr.s_addr = ifb->addr;
+
+ ip = inet_ntop(AF_INET, &addr.sin_addr, addr_buf, sizeof(addr_buf));
+ printf("mcast: join %s using interface IP %s\n", mcast_addr, ip);
+
+ sd = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP);
+ assert(sd >= 0);
+
+ addr.sin_port = htons(MCAST_PORT);
+ addr.sin_addr.s_addr = INADDR_ANY;
+
+ rv = bind(sd, (const struct sockaddr *)&addr, sizeof(addr));
+ assert(rv == 0);
+
+ memset(&mreq, 0, sizeof(mreq));
+ mreq.imr_multiaddr.s_addr = inet_addr(mcast_addr);
+ mreq.imr_interface.s_addr = ifb->addr;
+
+ rv = setsockopt(sd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &mreq, sizeof(mreq));
+ assert(rv == 0);
+
+ addr.sin_family = AF_INET;
+ addr.sin_port = htons(MCAST_PORT);
+ addr.sin_addr.s_addr = inet_addr(mcast_addr);
+ n = sendto(sd, hello, sizeof(hello), 0,
+ (const struct sockaddr *)&addr, sizeof(addr));
+
+ if (n != (ssize_t)sizeof(hello)) {
+ perror("send");
+ }
+
+ while (true) {
+ socklen_t addr_len;
+ char buf[MAX_PACKET_SIZE];
+
+ addr_len = sizeof(addr);
+ n = recvfrom(sd, buf, sizeof(buf), 0,
+ (struct sockaddr *)&addr, &addr_len);
+ if (n >= 0 && addr.sin_addr.s_addr != ifb->addr) {
+ size_t m;
+ ip = inet_ntop(AF_INET, &addr.sin_addr, addr_buf,
+ sizeof(addr_buf));
+ printf("mcast: received %zi bytes from %s\n", n, ip);
+
+ addr.sin_port = htons(MCAST_PORT);
+ addr.sin_addr.s_addr = inet_addr(mcast_addr);
+ m = (size_t)n;
+ n = sendto(sd, buf, m, 0,
+ (const struct sockaddr *)&addr,
+ sizeof(addr));
+ assert(n == (ssize_t)m);
+ } else if (n < 0 ) {
+ perror("recvfrom");
+ }
+ }
+}
+
+static void
+start_mcast_rx(const char *iface_ip, const char *iface_name)
+{
+ rtems_status_code sc;
+ rtems_id id;
+ iface_binding *ifb;
+ size_t len;
+
+ len = strlen(iface_name) + 1;
+ ifb = malloc(sizeof(*ifb) + len);
+ assert(ifb != NULL);
+ ifb->addr = inet_addr(iface_ip);
+ memcpy(ifb->name, iface_name, len);
+
+ sc = rtems_task_create(rtems_build_name('M', 'C', 'R', 'X'),
+ RTEMS_MINIMUM_PRIORITY, RTEMS_MINIMUM_STACK_SIZE + MAX_PACKET_SIZE,
+ RTEMS_DEFAULT_MODES, RTEMS_FLOATING_POINT, &id);
+ assert(sc == RTEMS_SUCCESSFUL);
+
+ sc = rtems_task_start(id, mcast_rx_task, (rtems_task_argument)ifb);
+ assert(sc == RTEMS_SUCCESSFUL);
+}
+
+static const char *
+get_value(char *const *env, const char *key)
+{
+ size_t len;
+
+ len = strlen(key);
+
+ while (true) {
+ const char *s;
+
+ s = *env;
+
+ if (s == NULL) {
+ return "";
+ }
+
+ if (strncmp(key, s, len) == 0 && s[len] == '=') {
+ return &s[len + 1];
+ }
+
+ ++env;
+ }
+}
+
+static void
+dhcpcd_hook_handler(rtems_dhcpcd_hook *hook, char *const *env)
+{
+
+ (void)hook;
+
+ if (strcmp(get_value(env, "reason"), "BOUND") == 0) {
+ start_mcast_rx(get_value(env, "new_ip_address"),
+ get_value(env, "interface"));
+ }
+}
+
+static rtems_dhcpcd_hook dhcpcd_hook = {
+ .name = "mcast",
+ .handler = dhcpcd_hook_handler
+};
+
+static void
+test_main(void)
+{
+ rtems_status_code sc;
+ int enable;
+ int rv;
+
+ rtems_dhcpcd_add_hook(&dhcpcd_hook);
+
+ enable = 1;
+ rv = sysctlbyname("net.inet.icmp.bmcastecho", NULL, NULL, &enable,
+ sizeof(enable));
+ assert(rv == 0);
+
+ sc = rtems_shell_init("SHLL", 16 * 1024, 1, CONSOLE_DEVICE_NAME,
+ false, true, NULL);
+ assert(sc == RTEMS_SUCCESSFUL);
+
+ exit(0);
+}
+
+#define DEFAULT_NETWORK_DHCPCD_ENABLE
+
+#include <rtems/bsd/test/default-network-init.h>
+
+#define CONFIGURE_SHELL_COMMANDS_INIT
+
+#include <bsp/irq-info.h>
+
+#include <rtems/netcmds-config.h>
+
+#define CONFIGURE_SHELL_USER_COMMANDS \
+ &bsp_interrupt_shell_command, \
+ &rtems_shell_ARP_Command, \
+ &rtems_shell_HOSTNAME_Command, \
+ &rtems_shell_PING_Command, \
+ &rtems_shell_ROUTE_Command, \
+ &rtems_shell_NETSTAT_Command, \
+ &rtems_shell_SYSCTL_Command, \
+ &rtems_shell_IFCONFIG_Command, \
+ &rtems_shell_IFMCSTAT_Command, \
+ &rtems_shell_VMSTAT_Command
+
+#define CONFIGURE_SHELL_COMMAND_CPUINFO
+#define CONFIGURE_SHELL_COMMAND_CPUUSE
+#define CONFIGURE_SHELL_COMMAND_PERIODUSE
+#define CONFIGURE_SHELL_COMMAND_STACKUSE
+#define CONFIGURE_SHELL_COMMAND_MALLOC_INFO
+
+#include <rtems/shellconfig.h>
--
2.26.2
More information about the devel
mailing list