mirror of
https://github.com/ossrs/srs.git
synced 2025-03-09 15:49:59 +00:00
Add research for UDP binding
This commit is contained in:
parent
155587d2c1
commit
040893efde
3 changed files with 144 additions and 0 deletions
2
trunk/research/st/.gitignore
vendored
Normal file
2
trunk/research/st/.gitignore
vendored
Normal file
|
@ -0,0 +1,2 @@
|
||||||
|
udp-server
|
||||||
|
udp-client
|
62
trunk/research/st/udp-client.cpp
Normal file
62
trunk/research/st/udp-client.cpp
Normal file
|
@ -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;
|
||||||
|
}
|
||||||
|
|
80
trunk/research/st/udp-server.cpp
Normal file
80
trunk/research/st/udp-server.cpp
Normal file
|
@ -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;
|
||||||
|
}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue