mirror of
https://github.com/ossrs/srs.git
synced 2025-03-09 15:49:59 +00:00
1f9309ae25
1. Add live benchmark support in srs-bench, which only connects and disconnects without any media transport, to test source creation and disposal and verify source memory leaks. 2. SmartPtr: Support cleanup of HTTP-FLV stream. Unregister the HTTP-FLV handler for the pattern and clean up the objects and resources. 3. Support benchmarking RTMP/SRT with srs-bench by integrating the gosrt and oryx RTMP libraries. 4. Refine SRT and RTC sources by using a timer to clean up the sources, following the same strategy as the Live source. --------- Co-authored-by: Haibo Chen <495810242@qq.com> Co-authored-by: Jacob Su <suzp1984@gmail.com>
69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
package srtgo
|
|
|
|
/*
|
|
#cgo LDFLAGS: -lsrt
|
|
#include <srt/srt.h>
|
|
|
|
SRTSOCKET srt_accept_wrapped(SRTSOCKET lsn, struct sockaddr* addr, int* addrlen, int *srterror, int *syserror)
|
|
{
|
|
int ret = srt_accept(lsn, addr, addrlen);
|
|
if (ret < 0) {
|
|
*srterror = srt_getlasterror(syserror);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
*/
|
|
import "C"
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"syscall"
|
|
"unsafe"
|
|
)
|
|
|
|
func srtAcceptImpl(lsn C.SRTSOCKET, addr *C.struct_sockaddr, addrlen *C.int) (C.SRTSOCKET, error) {
|
|
srterr := C.int(0)
|
|
syserr := C.int(0)
|
|
socket := C.srt_accept_wrapped(lsn, addr, addrlen, &srterr, &syserr)
|
|
if srterr != 0 {
|
|
srterror := SRTErrno(srterr)
|
|
if syserr < 0 {
|
|
srterror.wrapSysErr(syscall.Errno(syserr))
|
|
}
|
|
return socket, srterror
|
|
}
|
|
return socket, nil
|
|
}
|
|
|
|
// Accept an incoming connection
|
|
func (s SrtSocket) Accept() (*SrtSocket, *net.UDPAddr, error) {
|
|
var err error
|
|
if !s.blocking {
|
|
err = s.pd.wait(ModeRead)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
}
|
|
var addr syscall.RawSockaddrAny
|
|
sclen := C.int(syscall.SizeofSockaddrAny)
|
|
socket, err := srtAcceptImpl(s.socket, (*C.struct_sockaddr)(unsafe.Pointer(&addr)), &sclen)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
if socket == SRT_INVALID_SOCK {
|
|
return nil, nil, fmt.Errorf("srt accept, error accepting the connection: %w", srtGetAndClearError())
|
|
}
|
|
|
|
newSocket, err := newFromSocket(&s, socket)
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("new socket could not be created: %w", err)
|
|
}
|
|
|
|
udpAddr, err := udpAddrFromSockaddr(&addr)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
return newSocket, udpAddr, nil
|
|
}
|