mirror of
https://github.com/ossrs/srs.git
synced 2025-03-09 15:49:59 +00:00
DTLS: Update regression tests
This commit is contained in:
winlin
parent
d4d11c2c18
commit
06f2e1462e
19 changed files with 3329 additions and 949 deletions
278
trunk/3rdparty/srs-bench/vnet/example_udpproxy_test.go
vendored
Normal file
278
trunk/3rdparty/srs-bench/vnet/example_udpproxy_test.go
vendored
Normal file
|
|
@ -0,0 +1,278 @@
|
|||
// The MIT License (MIT)
|
||||
//
|
||||
// Copyright (c) 2021 srs-bench(ossrs)
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy of
|
||||
// this software and associated documentation files (the "Software"), to deal in
|
||||
// the Software without restriction, including without limitation the rights to
|
||||
// use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
|
||||
// the Software, and to permit persons to whom the Software is furnished to do so,
|
||||
// subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in all
|
||||
// copies or substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
||||
// FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
|
||||
// COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
|
||||
// IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
||||
// CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||
package vnet_test
|
||||
|
||||
import (
|
||||
"net"
|
||||
|
||||
vnet_proxy "github.com/ossrs/srs-bench/vnet"
|
||||
"github.com/pion/logging"
|
||||
"github.com/pion/transport/vnet"
|
||||
)
|
||||
|
||||
// Proxy many vnet endpoint to one real server endpoint.
|
||||
// For example:
|
||||
// vnet(10.0.0.11:5787) => proxy => 192.168.1.10:8000
|
||||
// vnet(10.0.0.11:5788) => proxy => 192.168.1.10:8000
|
||||
// vnet(10.0.0.11:5789) => proxy => 192.168.1.10:8000
|
||||
func ExampleUDPProxyManyToOne() { // nolint:govet
|
||||
var clientNetwork *vnet.Net
|
||||
|
||||
var serverAddr *net.UDPAddr
|
||||
if addr, err := net.ResolveUDPAddr("udp4", "192.168.1.10:8000"); err != nil {
|
||||
// handle error
|
||||
} else {
|
||||
serverAddr = addr
|
||||
}
|
||||
|
||||
// Setup the network and proxy.
|
||||
if true {
|
||||
// Create vnet WAN with one endpoint, please read from
|
||||
// https://github.com/pion/transport/tree/master/vnet#example-wan-with-one-endpoint-vnet
|
||||
router, err := vnet.NewRouter(&vnet.RouterConfig{
|
||||
CIDR: "0.0.0.0/0",
|
||||
LoggerFactory: logging.NewDefaultLoggerFactory(),
|
||||
})
|
||||
if err != nil {
|
||||
// handle error
|
||||
}
|
||||
|
||||
// Create a network and add to router, for example, for client.
|
||||
clientNetwork = vnet.NewNet(&vnet.NetConfig{
|
||||
StaticIP: "10.0.0.11",
|
||||
})
|
||||
if err = router.AddNet(clientNetwork); err != nil {
|
||||
// handle error
|
||||
}
|
||||
|
||||
// Start the router.
|
||||
if err = router.Start(); err != nil {
|
||||
// handle error
|
||||
}
|
||||
defer router.Stop() // nolint:errcheck
|
||||
|
||||
// Create a proxy, bind to the router.
|
||||
proxy, err := vnet_proxy.NewProxy(router)
|
||||
if err != nil {
|
||||
// handle error
|
||||
}
|
||||
defer proxy.Close() // nolint:errcheck
|
||||
|
||||
// Start to proxy some addresses, clientNetwork is a hit for proxy,
|
||||
// that the client in vnet is from this network.
|
||||
if err := proxy.Proxy(clientNetwork, serverAddr); err != nil {
|
||||
// handle error
|
||||
}
|
||||
}
|
||||
|
||||
// Now, all packets from client, will be proxy to real server, vice versa.
|
||||
client0, err := clientNetwork.ListenPacket("udp4", "10.0.0.11:5787")
|
||||
if err != nil {
|
||||
// handle error
|
||||
}
|
||||
_, _ = client0.WriteTo([]byte("Hello"), serverAddr)
|
||||
|
||||
client1, err := clientNetwork.ListenPacket("udp4", "10.0.0.11:5788")
|
||||
if err != nil {
|
||||
// handle error
|
||||
}
|
||||
_, _ = client1.WriteTo([]byte("Hello"), serverAddr)
|
||||
|
||||
client2, err := clientNetwork.ListenPacket("udp4", "10.0.0.11:5789")
|
||||
if err != nil {
|
||||
// handle error
|
||||
}
|
||||
_, _ = client2.WriteTo([]byte("Hello"), serverAddr)
|
||||
}
|
||||
|
||||
// Proxy many vnet endpoint to one real server endpoint.
|
||||
// For example:
|
||||
// vnet(10.0.0.11:5787) => proxy => 192.168.1.10:8000
|
||||
// vnet(10.0.0.11:5788) => proxy => 192.168.1.10:8000
|
||||
func ExampleUDPProxyMultileTimes() { // nolint:govet
|
||||
var clientNetwork *vnet.Net
|
||||
|
||||
var serverAddr *net.UDPAddr
|
||||
if addr, err := net.ResolveUDPAddr("udp4", "192.168.1.10:8000"); err != nil {
|
||||
// handle error
|
||||
} else {
|
||||
serverAddr = addr
|
||||
}
|
||||
|
||||
// Setup the network and proxy.
|
||||
var proxy *vnet_proxy.UDPProxy
|
||||
if true {
|
||||
// Create vnet WAN with one endpoint, please read from
|
||||
// https://github.com/pion/transport/tree/master/vnet#example-wan-with-one-endpoint-vnet
|
||||
router, err := vnet.NewRouter(&vnet.RouterConfig{
|
||||
CIDR: "0.0.0.0/0",
|
||||
LoggerFactory: logging.NewDefaultLoggerFactory(),
|
||||
})
|
||||
if err != nil {
|
||||
// handle error
|
||||
}
|
||||
|
||||
// Create a network and add to router, for example, for client.
|
||||
clientNetwork = vnet.NewNet(&vnet.NetConfig{
|
||||
StaticIP: "10.0.0.11",
|
||||
})
|
||||
if err = router.AddNet(clientNetwork); err != nil {
|
||||
// handle error
|
||||
}
|
||||
|
||||
// Start the router.
|
||||
if err = router.Start(); err != nil {
|
||||
// handle error
|
||||
}
|
||||
defer router.Stop() // nolint:errcheck
|
||||
|
||||
// Create a proxy, bind to the router.
|
||||
proxy, err = vnet_proxy.NewProxy(router)
|
||||
if err != nil {
|
||||
// handle error
|
||||
}
|
||||
defer proxy.Close() // nolint:errcheck
|
||||
}
|
||||
|
||||
if true {
|
||||
// Start to proxy some addresses, clientNetwork is a hit for proxy,
|
||||
// that the client in vnet is from this network.
|
||||
if err := proxy.Proxy(clientNetwork, serverAddr); err != nil {
|
||||
// handle error
|
||||
}
|
||||
|
||||
// Now, all packets from client, will be proxy to real server, vice versa.
|
||||
client0, err := clientNetwork.ListenPacket("udp4", "10.0.0.11:5787")
|
||||
if err != nil {
|
||||
// handle error
|
||||
}
|
||||
_, _ = client0.WriteTo([]byte("Hello"), serverAddr)
|
||||
}
|
||||
|
||||
if true {
|
||||
// It's ok to proxy multiple times, for example, the publisher and player
|
||||
// might need to proxy when got answer.
|
||||
if err := proxy.Proxy(clientNetwork, serverAddr); err != nil {
|
||||
// handle error
|
||||
}
|
||||
|
||||
client1, err := clientNetwork.ListenPacket("udp4", "10.0.0.11:5788")
|
||||
if err != nil {
|
||||
// handle error
|
||||
}
|
||||
_, _ = client1.WriteTo([]byte("Hello"), serverAddr)
|
||||
}
|
||||
}
|
||||
|
||||
// Proxy one vnet endpoint to one real server endpoint.
|
||||
// For example:
|
||||
// vnet(10.0.0.11:5787) => proxy0 => 192.168.1.10:8000
|
||||
// vnet(10.0.0.11:5788) => proxy1 => 192.168.1.10:8001
|
||||
// vnet(10.0.0.11:5789) => proxy2 => 192.168.1.10:8002
|
||||
func ExampleUDPProxyOneToOne() { // nolint:govet
|
||||
var clientNetwork *vnet.Net
|
||||
|
||||
var serverAddr0 *net.UDPAddr
|
||||
if addr, err := net.ResolveUDPAddr("udp4", "192.168.1.10:8000"); err != nil {
|
||||
// handle error
|
||||
} else {
|
||||
serverAddr0 = addr
|
||||
}
|
||||
|
||||
var serverAddr1 *net.UDPAddr
|
||||
if addr, err := net.ResolveUDPAddr("udp4", "192.168.1.10:8001"); err != nil {
|
||||
// handle error
|
||||
} else {
|
||||
serverAddr1 = addr
|
||||
}
|
||||
|
||||
var serverAddr2 *net.UDPAddr
|
||||
if addr, err := net.ResolveUDPAddr("udp4", "192.168.1.10:8002"); err != nil {
|
||||
// handle error
|
||||
} else {
|
||||
serverAddr2 = addr
|
||||
}
|
||||
|
||||
// Setup the network and proxy.
|
||||
if true {
|
||||
// Create vnet WAN with one endpoint, please read from
|
||||
// https://github.com/pion/transport/tree/master/vnet#example-wan-with-one-endpoint-vnet
|
||||
router, err := vnet.NewRouter(&vnet.RouterConfig{
|
||||
CIDR: "0.0.0.0/0",
|
||||
LoggerFactory: logging.NewDefaultLoggerFactory(),
|
||||
})
|
||||
if err != nil {
|
||||
// handle error
|
||||
}
|
||||
|
||||
// Create a network and add to router, for example, for client.
|
||||
clientNetwork = vnet.NewNet(&vnet.NetConfig{
|
||||
StaticIP: "10.0.0.11",
|
||||
})
|
||||
if err = router.AddNet(clientNetwork); err != nil {
|
||||
// handle error
|
||||
}
|
||||
|
||||
// Start the router.
|
||||
if err = router.Start(); err != nil {
|
||||
// handle error
|
||||
}
|
||||
defer router.Stop() // nolint:errcheck
|
||||
|
||||
// Create a proxy, bind to the router.
|
||||
proxy, err := vnet_proxy.NewProxy(router)
|
||||
if err != nil {
|
||||
// handle error
|
||||
}
|
||||
defer proxy.Close() // nolint:errcheck
|
||||
|
||||
// Start to proxy some addresses, clientNetwork is a hit for proxy,
|
||||
// that the client in vnet is from this network.
|
||||
if err := proxy.Proxy(clientNetwork, serverAddr0); err != nil {
|
||||
// handle error
|
||||
}
|
||||
if err := proxy.Proxy(clientNetwork, serverAddr1); err != nil {
|
||||
// handle error
|
||||
}
|
||||
if err := proxy.Proxy(clientNetwork, serverAddr2); err != nil {
|
||||
// handle error
|
||||
}
|
||||
}
|
||||
|
||||
// Now, all packets from client, will be proxy to real server, vice versa.
|
||||
client0, err := clientNetwork.ListenPacket("udp4", "10.0.0.11:5787")
|
||||
if err != nil {
|
||||
// handle error
|
||||
}
|
||||
_, _ = client0.WriteTo([]byte("Hello"), serverAddr0)
|
||||
|
||||
client1, err := clientNetwork.ListenPacket("udp4", "10.0.0.11:5788")
|
||||
if err != nil {
|
||||
// handle error
|
||||
}
|
||||
_, _ = client1.WriteTo([]byte("Hello"), serverAddr1)
|
||||
|
||||
client2, err := clientNetwork.ListenPacket("udp4", "10.0.0.11:5789")
|
||||
if err != nil {
|
||||
// handle error
|
||||
}
|
||||
_, _ = client2.WriteTo([]byte("Hello"), serverAddr2)
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue