Porting PicoTCP WIP

2025-10-29 14:29:06 +01:00
parent 6722f42e68
commit 815c2239fe
464 changed files with 235009 additions and 24 deletions
--- a/kernel/picotcp/test/examples/iperfc.c
+++ b/kernel/picotcp/test/examples/iperfc.c
@ -0,0 +1,142 @@
+#include <stdlib.h>
+#include <string.h>
+#include <stdint.h>
+#include "pico_ipv6.h"
+#include "pico_stack.h"
+#include "pico_socket.h"
+#include "utils.h"
+
+#define DURATION 30
+
+struct iperf_hdr {
+    int32_t flags;          /* 0 */
+    int32_t numThreads;     /* 1 */
+    int32_t mPort;          /* 5001  */
+    int32_t bufferlen;      /* 0 */
+    int32_t mWinBand;       /* 0 */
+    int32_t mAmount;        /* 0xfffffc18 */
+};
+
+#define IPERF_PORT 5001
+#define MTU 1444
+#define SEND_BUF_SIZ (1024 * 2048)
+
+char *cpy_arg(char **dst, char *str);
+extern int IPV6_MODE;
+
+static pico_time deadline;
+
+static void panic(void)
+{
+    for(;; ) ;
+}
+
+static char buf[MTU] = {};
+
+static void buf_paint(void)
+{
+    char paint[11] = "0123456789";
+    int i;
+    for (i = 0; i < MTU; i++) {
+        buf[i] = paint[i % 10];
+    }
+}
+
+static void send_hdr(struct pico_socket *s)
+{
+    struct iperf_hdr hdr = {};
+    hdr.numThreads = long_be(1);
+    hdr.mPort = long_be(5001);
+    hdr.mAmount = long_be(0xfffffc18);
+    pico_socket_write(s, &hdr, sizeof(hdr));
+    deadline = PICO_TIME_MS() + DURATION * 1000;
+}
+
+static void iperf_cb(uint16_t ev, struct pico_socket *s)
+{
+    int r;
+    static int end = 0;
+    if (ev & PICO_SOCK_EV_CONN) {
+        send_hdr(s);
+        return;
+    }
+
+    if ((!end) && (ev & PICO_SOCK_EV_WR)) {
+        if (PICO_TIME_MS() > deadline) {
+            pico_socket_close(s);
+            if (!pico_timer_add(2000, deferred_exit, NULL)) {
+                printf("Failed to start exit timer, exiting now\n");
+                exit(1);
+            }
+            end++;
+        }
+
+        pico_socket_write(s, buf, MTU);
+    }
+
+    if (!(end) && (ev & (PICO_SOCK_EV_FIN | PICO_SOCK_EV_CLOSE))) {
+        if (!pico_timer_add(2000, deferred_exit, NULL)) {
+            printf("Failed to start exit timer, exiting now\n");
+            exit(1);
+        }
+        end++;
+    }
+}
+
+static void iperfc_socket_setup(union pico_address *addr, uint16_t family)
+{
+    int yes = 1;
+    uint16_t send_port = 0;
+    struct pico_socket *s = NULL;
+    uint32_t bufsize = SEND_BUF_SIZ;
+    send_port = short_be(5001);
+    s = pico_socket_open(family, PICO_PROTO_TCP, &iperf_cb);
+    pico_socket_setoption(s, PICO_SOCKET_OPT_SNDBUF, &bufsize);
+    pico_socket_connect(s, addr, send_port);
+}
+
+void app_iperfc(char *arg)
+{
+    struct pico_ip4 my_eth_addr, netmask;
+    struct pico_device *pico_dev_eth;
+    char *daddr = NULL, *dport = NULL;
+    char *nxt = arg;
+    uint16_t send_port = 0, listen_port = short_be(5001);
+    int i = 0, ret = 0, yes = 1;
+    struct pico_socket *s = NULL;
+    uint16_t family = PICO_PROTO_IPV4;
+    union pico_address dst = {
+        .ip4 = {0}, .ip6 = {{0}}
+    };
+    union pico_address inaddr_any = {
+        .ip4 = {0}, .ip6 = {{0}}
+    };
+
+    /* start of argument parsing */
+    if (nxt) {
+        nxt = cpy_arg(&daddr, arg);
+        if (daddr) {
+            if (!IPV6_MODE)
+                pico_string_to_ipv4(daddr, &dst.ip4.addr);
+
+      #ifdef PICO_SUPPORT_IPV6
+            else {
+                pico_string_to_ipv6(daddr, dst.ip6.addr);
+                family = PICO_PROTO_IPV6;
+            }
+      #endif
+        } else {
+            goto out;
+        }
+    } else {
+        /* missing dest_addr */
+        goto out;
+    }
+
+    iperfc_socket_setup(&dst, family);
+    return;
+out:
+    dbg("Error parsing options!\n");
+    exit(1);
+}
+