From 040893efde19cf982abf87ef80e03789b9042520 Mon Sep 17 00:00:00 2001
From: winlin <winlin@vip.126.com>
Date: Tue, 5 Jan 2021 18:30:39 +0800
Subject: [PATCH] Add research for UDP binding

---
 trunk/research/st/.gitignore     |  2 +
 trunk/research/st/udp-client.cpp | 62 +++++++++++++++++++++++++
 trunk/research/st/udp-server.cpp | 80 ++++++++++++++++++++++++++++++++
 3 files changed, 144 insertions(+)
 create mode 100644 trunk/research/st/.gitignore
 create mode 100644 trunk/research/st/udp-client.cpp
 create mode 100644 trunk/research/st/udp-server.cpp

diff --git a/trunk/research/st/.gitignore b/trunk/research/st/.gitignore
new file mode 100644
index 000000000..827222cf2
--- /dev/null
+++ b/trunk/research/st/.gitignore
@@ -0,0 +1,2 @@
+udp-server
+udp-client
diff --git a/trunk/research/st/udp-client.cpp b/trunk/research/st/udp-client.cpp
new file mode 100644
index 000000000..379c9660d
--- /dev/null
+++ b/trunk/research/st/udp-client.cpp
@@ -0,0 +1,62 @@
+/*
+g++ udp-client.cpp ../../objs/st/libst.a -g -O0 -o udp-client && ./udp-client 127.0.0.1 8000 3
+*/
+#include <stdio.h>
+#include <stdlib.h>
+#include <assert.h>
+#include "../../objs/st/st.h"
+
+#include <sys/socket.h>
+#include <arpa/inet.h>
+
+void* pfn(void* arg) {
+    sockaddr_in addr = *(sockaddr_in*)arg;
+    char* ip = inet_ntoa(addr.sin_addr);
+    int port = ntohs(addr.sin_port);
+
+    int fd = ::socket(AF_INET, SOCK_DGRAM, 0);
+    st_netfd_t stfd = st_netfd_open_socket(fd);
+    printf("connect to %s:%d, fd=%d ok\n", ip, port, fd);
+
+    while (true) {
+        char data[] = "Hello world!";
+        int r0 = st_sendto(stfd, data, sizeof(data), (sockaddr*)&addr, sizeof(sockaddr_in), ST_UTIME_NO_TIMEOUT);
+        printf("fd #%d, send %dB %s, r0=%d\n", fd, (int)sizeof(data), data, r0);
+        if (r0 != (int)sizeof(data)) {
+            break;
+        }
+        st_usleep(800 * 1000);
+    }
+
+    st_netfd_close(stfd);
+
+    return NULL;
+}
+
+int main(int argc, char** argv) {
+    if (argc < 4) {
+        printf("Usage: %s ip port workers\n", argv[0]);
+        exit(-1);
+    }
+
+    st_init();
+
+    const char* ip = argv[1];
+    int port = ::atoi(argv[2]);
+    int workers = ::atoi(argv[3]);
+    printf("Start %d workers, to %s:%d\n", workers, ip, port);
+
+    sockaddr_in addr;
+    addr.sin_family = AF_INET;
+    addr.sin_addr.s_addr = inet_addr(ip);
+    addr.sin_port = htons(port);
+
+    for (int i = 0; i < workers; i++) {
+        st_thread_t thread = st_thread_create(pfn, &addr, 1, 0);
+        assert(thread);
+    }
+
+    st_thread_exit(NULL);
+    return 0;
+}
+
diff --git a/trunk/research/st/udp-server.cpp b/trunk/research/st/udp-server.cpp
new file mode 100644
index 000000000..bd1e632f8
--- /dev/null
+++ b/trunk/research/st/udp-server.cpp
@@ -0,0 +1,80 @@
+/*
+g++ udp-server.cpp ../../objs/st/libst.a -g -O0 -o udp-server && ./udp-server 127.0.0.1 8000 3
+*/
+#include <stdio.h>
+#include <stdlib.h>
+#include <assert.h>
+#include "../../objs/st/st.h"
+
+#include <sys/socket.h>
+#include <arpa/inet.h>
+
+void* pfn(void* arg) {
+    sockaddr_in addr = *(sockaddr_in*)arg;
+    char* ip = inet_ntoa(addr.sin_addr);
+    int port = ntohs(addr.sin_port);
+
+    int fd = ::socket(AF_INET, SOCK_DGRAM, 0);
+    assert(fd > 0);
+
+    if (true) {
+        int v = 1;
+        int r0 = setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &v, sizeof(int));
+        assert(r0 != -1);
+    }
+    if (true) {
+        int v = 1;
+        int r0 = setsockopt(fd, SOL_SOCKET, SO_REUSEPORT, &v, sizeof(int));
+        assert(r0 != -1);
+    }
+    if (true) {
+        int r0 = bind(fd, (sockaddr*)&addr, sizeof(sockaddr_in));
+        assert(!r0);
+    }
+
+    st_netfd_t stfd = st_netfd_open_socket(fd);
+    printf("listen at %s:%d, fd=%d ok\n", ip, port, fd);
+
+    while (true) {
+        char buf[1600] = {0};
+        sockaddr_in from;
+        int nb_from = sizeof(from);
+        int r0 = st_recvfrom(stfd, buf, sizeof(buf), (sockaddr*)&from, &nb_from, ST_UTIME_NO_TIMEOUT);
+        printf("fd #%d, peer %s:%d, got %dB, %s\n", fd, inet_ntoa(from.sin_addr), ntohs(from.sin_port), r0, buf);
+        if (r0 <= 0) {
+            break;
+        }
+        st_usleep(10 * 1000);
+    }
+
+    st_netfd_close(stfd);
+    return NULL;
+}
+
+int main(int argc, char** argv) {
+    if (argc < 4) {
+        printf("Usage: %s ip port workers\n", argv[0]);
+        exit(-1);
+    }
+
+    st_init();
+
+    const char* ip = argv[1];
+    int port = ::atoi(argv[2]);
+    int workers = ::atoi(argv[3]);
+    printf("Start %d workers, at %s:%d\n", workers, ip, port);
+
+    sockaddr_in addr;
+    addr.sin_family = AF_INET;
+    addr.sin_addr.s_addr = inet_addr(ip);
+    addr.sin_port = htons(port);
+
+    for (int i = 0; i < workers; i++) {
+        st_thread_t thread = st_thread_create(pfn, &addr, 1, 0);
+        assert(thread);
+    }
+
+    st_thread_exit(NULL);
+    return 0;
+}
+