From 040893efde19cf982abf87ef80e03789b9042520 Mon Sep 17 00:00:00 2001 From: winlin 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 +#include +#include +#include "../../objs/st/st.h" + +#include +#include + +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 +#include +#include +#include "../../objs/st/st.h" + +#include +#include + +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; +} +