|
|
|
@ -23,6 +23,7 @@
|
|
|
|
|
|
|
|
|
|
#include <srs_app_conn.hpp>
|
|
|
|
|
|
|
|
|
|
#include <netinet/tcp.h>
|
|
|
|
|
using namespace std;
|
|
|
|
|
|
|
|
|
|
#include <srs_kernel_log.hpp>
|
|
|
|
@ -95,6 +96,89 @@ srs_error_t SrsConnection::start()
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
srs_error_t SrsConnection::set_tcp_nodelay(bool v)
|
|
|
|
|
{
|
|
|
|
|
srs_error_t err = srs_success;
|
|
|
|
|
|
|
|
|
|
int r0 = 0;
|
|
|
|
|
socklen_t nb_v = sizeof(int);
|
|
|
|
|
int fd = srs_netfd_fileno(stfd);
|
|
|
|
|
|
|
|
|
|
int ov = 0;
|
|
|
|
|
if ((r0 = getsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &ov, &nb_v)) != 0) {
|
|
|
|
|
return srs_error_new(ERROR_SOCKET_NO_NODELAY, "getsockopt fd=%d, r0=%d", fd, r0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#ifndef SRS_PERF_TCP_NODELAY
|
|
|
|
|
srs_warn("ignore TCP_NODELAY, fd=%d, ov=%d", fd, ov);
|
|
|
|
|
return err;
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
int iv = (v? 1:0);
|
|
|
|
|
if ((r0 = setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &iv, nb_v)) != 0) {
|
|
|
|
|
return srs_error_new(ERROR_SOCKET_NO_NODELAY, "setsockopt fd=%d, r0=%v", fd, r0);
|
|
|
|
|
}
|
|
|
|
|
if ((r0 = getsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &iv, &nb_v)) != 0) {
|
|
|
|
|
return srs_error_new(ERROR_SOCKET_NO_NODELAY, "getsockopt fd=%d, r0=%d", fd, r0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
srs_trace("set fd=%d TCP_NODELAY %d=>%d", fd, ov, iv);
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
srs_error_t SrsConnection::set_socket_buffer(int buffer_ms)
|
|
|
|
|
{
|
|
|
|
|
srs_error_t err = srs_success;
|
|
|
|
|
|
|
|
|
|
int r0 = 0;
|
|
|
|
|
int fd = srs_netfd_fileno(stfd);
|
|
|
|
|
socklen_t nb_v = sizeof(int);
|
|
|
|
|
|
|
|
|
|
int ov = 0;
|
|
|
|
|
if ((r0 = getsockopt(fd, SOL_SOCKET, SO_SNDBUF, &ov, &nb_v)) != 0) {
|
|
|
|
|
return srs_error_new(ERROR_SOCKET_SNDBUF, "getsockopt fd=%d, r0=%d", fd, r0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#ifndef SRS_PERF_MW_SO_SNDBUF
|
|
|
|
|
srs_warn("ignore SO_SNDBUF, fd=%d, ov=%d", fd, ov);
|
|
|
|
|
return err;
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
// the bytes:
|
|
|
|
|
// 4KB=4096, 8KB=8192, 16KB=16384, 32KB=32768, 64KB=65536,
|
|
|
|
|
// 128KB=131072, 256KB=262144, 512KB=524288
|
|
|
|
|
// the buffer should set to sleep*kbps/8,
|
|
|
|
|
// for example, your system delivery stream in 1000kbps,
|
|
|
|
|
// sleep 800ms for small bytes, the buffer should set to:
|
|
|
|
|
// 800*1000/8=100000B(about 128KB).
|
|
|
|
|
// other examples:
|
|
|
|
|
// 2000*3000/8=750000B(about 732KB).
|
|
|
|
|
// 2000*5000/8=1250000B(about 1220KB).
|
|
|
|
|
int kbps = 4000;
|
|
|
|
|
int iv = buffer_ms * kbps / 8;
|
|
|
|
|
|
|
|
|
|
// socket send buffer, system will double it.
|
|
|
|
|
iv = iv / 2;
|
|
|
|
|
|
|
|
|
|
// override the send buffer by macro.
|
|
|
|
|
#ifdef SRS_PERF_SO_SNDBUF_SIZE
|
|
|
|
|
iv = SRS_PERF_SO_SNDBUF_SIZE / 2;
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
// set the socket send buffer when required larger buffer
|
|
|
|
|
if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &iv, nb_v) < 0) {
|
|
|
|
|
return srs_error_new(ERROR_SOCKET_SNDBUF, "setsockopt fd=%d, r0=%v", fd, r0);
|
|
|
|
|
}
|
|
|
|
|
if ((r0 = getsockopt(fd, SOL_SOCKET, SO_SNDBUF, &iv, &nb_v)) != 0) {
|
|
|
|
|
return srs_error_new(ERROR_SOCKET_SNDBUF, "getsockopt fd=%d, r0=%d", fd, r0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
srs_trace("set fd=%d, SO_SNDBUF=%d=>%d, buffer=%dms", fd, ov, iv, buffer_ms);
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
srs_error_t SrsConnection::cycle()
|
|
|
|
|
{
|
|
|
|
|
srs_error_t err = do_cycle();
|
|
|
|
|