frp/test/e2e/legacy/basic/server.go

179 lines
4.8 KiB
Go
Raw Permalink Normal View History

2021-03-31 16:57:39 +08:00
package basic
import (
"fmt"
2021-06-18 16:48:36 +08:00
"net"
"strconv"
2021-03-31 16:57:39 +08:00
2023-02-27 14:44:16 +08:00
"github.com/onsi/ginkgo/v2"
2022-08-29 01:02:53 +08:00
2021-03-31 16:57:39 +08:00
"github.com/fatedier/frp/test/e2e/framework"
"github.com/fatedier/frp/test/e2e/framework/consts"
"github.com/fatedier/frp/test/e2e/pkg/port"
"github.com/fatedier/frp/test/e2e/pkg/request"
)
2022-08-29 01:02:53 +08:00
var _ = ginkgo.Describe("[Feature: Server Manager]", func() {
2021-03-31 16:57:39 +08:00
f := framework.NewDefaultFramework()
2022-08-29 01:02:53 +08:00
ginkgo.It("Ports Whitelist", func() {
2023-09-13 16:32:39 +08:00
serverConf := consts.LegacyDefaultServerConfig
clientConf := consts.LegacyDefaultClientConfig
2021-03-31 16:57:39 +08:00
serverConf += `
2023-12-21 22:46:08 +08:00
allow_ports = 10000-11000,11002,12000-13000
2021-03-31 16:57:39 +08:00
`
2023-12-21 22:46:08 +08:00
tcpPortName := port.GenName("TCP", port.WithRangePorts(10000, 11000))
udpPortName := port.GenName("UDP", port.WithRangePorts(12000, 13000))
2021-03-31 16:57:39 +08:00
clientConf += fmt.Sprintf(`
[tcp-allowded-in-range]
type = tcp
local_port = {{ .%s }}
remote_port = {{ .%s }}
`, framework.TCPEchoServerPort, tcpPortName)
clientConf += fmt.Sprintf(`
[tcp-port-not-allowed]
type = tcp
local_port = {{ .%s }}
2023-12-21 22:46:08 +08:00
remote_port = 11001
2021-03-31 16:57:39 +08:00
`, framework.TCPEchoServerPort)
clientConf += fmt.Sprintf(`
[tcp-port-unavailable]
type = tcp
local_port = {{ .%s }}
remote_port = {{ .%s }}
`, framework.TCPEchoServerPort, consts.PortServerName)
clientConf += fmt.Sprintf(`
[udp-allowed-in-range]
type = udp
local_port = {{ .%s }}
remote_port = {{ .%s }}
`, framework.UDPEchoServerPort, udpPortName)
clientConf += fmt.Sprintf(`
[udp-port-not-allowed]
type = udp
local_port = {{ .%s }}
2023-12-21 22:46:08 +08:00
remote_port = 11003
2021-03-31 16:57:39 +08:00
`, framework.UDPEchoServerPort)
f.RunProcesses([]string{serverConf}, []string{clientConf})
// TCP
// Allowed in range
framework.NewRequestExpect(f).PortName(tcpPortName).Ensure()
// Not Allowed
2023-12-21 22:46:08 +08:00
framework.NewRequestExpect(f).Port(11001).ExpectError(true).Ensure()
2021-03-31 16:57:39 +08:00
// Unavailable, already bind by frps
framework.NewRequestExpect(f).PortName(consts.PortServerName).ExpectError(true).Ensure()
// UDP
// Allowed in range
2021-06-18 16:48:36 +08:00
framework.NewRequestExpect(f).Protocol("udp").PortName(udpPortName).Ensure()
2021-03-31 16:57:39 +08:00
// Not Allowed
framework.NewRequestExpect(f).RequestModify(func(r *request.Request) {
2023-12-21 22:46:08 +08:00
r.UDP().Port(11003)
2021-03-31 16:57:39 +08:00
}).ExpectError(true).Ensure()
})
2021-06-18 16:48:36 +08:00
2022-08-29 01:02:53 +08:00
ginkgo.It("Alloc Random Port", func() {
2023-09-13 16:32:39 +08:00
serverConf := consts.LegacyDefaultServerConfig
clientConf := consts.LegacyDefaultClientConfig
2021-06-18 16:48:36 +08:00
adminPort := f.AllocPort()
clientConf += fmt.Sprintf(`
2021-08-02 13:07:28 +08:00
admin_port = %d
2021-06-18 16:48:36 +08:00
2021-08-02 13:07:28 +08:00
[tcp]
type = tcp
local_port = {{ .%s }}
2021-06-18 16:48:36 +08:00
2021-08-02 13:07:28 +08:00
[udp]
type = udp
local_port = {{ .%s }}
`, adminPort, framework.TCPEchoServerPort, framework.UDPEchoServerPort)
2021-06-18 16:48:36 +08:00
f.RunProcesses([]string{serverConf}, []string{clientConf})
client := f.APIClientForFrpc(adminPort)
2021-06-18 16:48:36 +08:00
// tcp random port
status, err := client.GetProxyStatus("tcp")
framework.ExpectNoError(err)
_, portStr, err := net.SplitHostPort(status.RemoteAddr)
framework.ExpectNoError(err)
port, err := strconv.Atoi(portStr)
framework.ExpectNoError(err)
framework.NewRequestExpect(f).Port(port).Ensure()
// udp random port
status, err = client.GetProxyStatus("udp")
framework.ExpectNoError(err)
_, portStr, err = net.SplitHostPort(status.RemoteAddr)
framework.ExpectNoError(err)
port, err = strconv.Atoi(portStr)
framework.ExpectNoError(err)
framework.NewRequestExpect(f).Protocol("udp").Port(port).Ensure()
})
2021-08-02 13:07:28 +08:00
2022-08-29 01:02:53 +08:00
ginkgo.It("Port Reuse", func() {
2023-09-13 16:32:39 +08:00
serverConf := consts.LegacyDefaultServerConfig
2021-08-02 13:07:28 +08:00
// Use same port as PortServer
serverConf += fmt.Sprintf(`
vhost_http_port = {{ .%s }}
`, consts.PortServerName)
2023-09-13 16:32:39 +08:00
clientConf := consts.LegacyDefaultClientConfig + fmt.Sprintf(`
2021-08-02 13:07:28 +08:00
[http]
type = http
local_port = {{ .%s }}
custom_domains = example.com
`, framework.HTTPSimpleServerPort)
f.RunProcesses([]string{serverConf}, []string{clientConf})
framework.NewRequestExpect(f).RequestModify(func(r *request.Request) {
r.HTTP().HTTPHost("example.com")
}).PortName(consts.PortServerName).Ensure()
})
2022-08-29 01:02:53 +08:00
ginkgo.It("healthz", func() {
2023-09-13 16:32:39 +08:00
serverConf := consts.LegacyDefaultServerConfig
dashboardPort := f.AllocPort()
// Use same port as PortServer
serverConf += fmt.Sprintf(`
vhost_http_port = {{ .%s }}
dashboard_addr = 0.0.0.0
dashboard_port = %d
dashboard_user = admin
dashboard_pwd = admin
`, consts.PortServerName, dashboardPort)
2023-09-13 16:32:39 +08:00
clientConf := consts.LegacyDefaultClientConfig + fmt.Sprintf(`
[http]
type = http
local_port = {{ .%s }}
custom_domains = example.com
`, framework.HTTPSimpleServerPort)
f.RunProcesses([]string{serverConf}, []string{clientConf})
framework.NewRequestExpect(f).RequestModify(func(r *request.Request) {
r.HTTP().HTTPPath("/healthz")
}).Port(dashboardPort).ExpectResp([]byte("")).Ensure()
framework.NewRequestExpect(f).RequestModify(func(r *request.Request) {
r.HTTP().HTTPPath("/")
}).Port(dashboardPort).
Ensure(framework.ExpectResponseCode(401))
})
2021-03-31 16:57:39 +08:00
})