mirror of
https://gitee.com/IrisVega/frp.git
synced 2024-11-01 22:31:29 +08:00
208 lines
5.6 KiB
Go
208 lines
5.6 KiB
Go
|
// Copyright 2020 The frp Authors
|
||
|
//
|
||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
// you may not use this file except in compliance with the License.
|
||
|
// You may obtain a copy of the License at
|
||
|
//
|
||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||
|
//
|
||
|
// Unless required by applicable law or agreed to in writing, software
|
||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
// See the License for the specific language governing permissions and
|
||
|
// limitations under the License.
|
||
|
|
||
|
package config
|
||
|
|
||
|
import (
|
||
|
"testing"
|
||
|
|
||
|
"github.com/fatedier/frp/pkg/auth"
|
||
|
"github.com/fatedier/frp/pkg/plugin/server"
|
||
|
|
||
|
"github.com/stretchr/testify/assert"
|
||
|
)
|
||
|
|
||
|
func Test_LoadServerCommonConf(t *testing.T) {
|
||
|
assert := assert.New(t)
|
||
|
|
||
|
testcases := []struct {
|
||
|
source []byte
|
||
|
expected ServerCommonConf
|
||
|
}{
|
||
|
{
|
||
|
source: []byte(`
|
||
|
# [common] is integral section
|
||
|
[common]
|
||
|
bind_addr = 0.0.0.9
|
||
|
bind_port = 7009
|
||
|
bind_udp_port = 7008
|
||
|
kcp_bind_port = 7007
|
||
|
proxy_bind_addr = 127.0.0.9
|
||
|
vhost_http_port = 89
|
||
|
vhost_https_port = 449
|
||
|
vhost_http_timeout = 69
|
||
|
tcpmux_httpconnect_port = 1339
|
||
|
dashboard_addr = 0.0.0.9
|
||
|
dashboard_port = 7509
|
||
|
dashboard_user = admin9
|
||
|
dashboard_pwd = admin9
|
||
|
enable_prometheus
|
||
|
assets_dir = ./static9
|
||
|
log_file = ./frps.log9
|
||
|
log_way = file
|
||
|
log_level = info9
|
||
|
log_max_days = 39
|
||
|
disable_log_color = false
|
||
|
detailed_errors_to_client
|
||
|
authentication_method = token
|
||
|
authenticate_heartbeats = false
|
||
|
authenticate_new_work_conns = false
|
||
|
token = 123456789
|
||
|
oidc_issuer = test9
|
||
|
oidc_audience = test9
|
||
|
oidc_skip_expiry_check
|
||
|
oidc_skip_issuer_check
|
||
|
heartbeat_timeout = 99
|
||
|
user_conn_timeout = 9
|
||
|
allow_ports = 10-12,99
|
||
|
max_pool_count = 59
|
||
|
max_ports_per_client = 9
|
||
|
tls_only = false
|
||
|
tls_cert_file = server.crt
|
||
|
tls_key_file = server.key
|
||
|
tls_trusted_ca_file = ca.crt
|
||
|
subdomain_host = frps.com
|
||
|
tcp_mux
|
||
|
udp_packet_size = 1509
|
||
|
[plugin.user-manager]
|
||
|
addr = 127.0.0.1:9009
|
||
|
path = /handler
|
||
|
ops = Login
|
||
|
[plugin.port-manager]
|
||
|
addr = 127.0.0.1:9009
|
||
|
path = /handler
|
||
|
ops = NewProxy
|
||
|
tls_verify
|
||
|
`),
|
||
|
expected: ServerCommonConf{
|
||
|
ServerConfig: auth.ServerConfig{
|
||
|
BaseConfig: auth.BaseConfig{
|
||
|
AuthenticationMethod: "token",
|
||
|
AuthenticateHeartBeats: false,
|
||
|
AuthenticateNewWorkConns: false,
|
||
|
},
|
||
|
TokenConfig: auth.TokenConfig{
|
||
|
Token: "123456789",
|
||
|
},
|
||
|
OidcServerConfig: auth.OidcServerConfig{
|
||
|
OidcIssuer: "test9",
|
||
|
OidcAudience: "test9",
|
||
|
OidcSkipExpiryCheck: true,
|
||
|
OidcSkipIssuerCheck: true,
|
||
|
},
|
||
|
},
|
||
|
BindAddr: "0.0.0.9",
|
||
|
BindPort: 7009,
|
||
|
BindUDPPort: 7008,
|
||
|
KCPBindPort: 7007,
|
||
|
ProxyBindAddr: "127.0.0.9",
|
||
|
VhostHTTPPort: 89,
|
||
|
VhostHTTPSPort: 449,
|
||
|
VhostHTTPTimeout: 69,
|
||
|
TCPMuxHTTPConnectPort: 1339,
|
||
|
DashboardAddr: "0.0.0.9",
|
||
|
DashboardPort: 7509,
|
||
|
DashboardUser: "admin9",
|
||
|
DashboardPwd: "admin9",
|
||
|
EnablePrometheus: true,
|
||
|
AssetsDir: "./static9",
|
||
|
LogFile: "./frps.log9",
|
||
|
LogWay: "file",
|
||
|
LogLevel: "info9",
|
||
|
LogMaxDays: 39,
|
||
|
DisableLogColor: false,
|
||
|
DetailedErrorsToClient: true,
|
||
|
HeartbeatTimeout: 99,
|
||
|
UserConnTimeout: 9,
|
||
|
AllowPorts: map[int]struct{}{
|
||
|
10: struct{}{},
|
||
|
11: struct{}{},
|
||
|
12: struct{}{},
|
||
|
99: struct{}{},
|
||
|
},
|
||
|
MaxPoolCount: 59,
|
||
|
MaxPortsPerClient: 9,
|
||
|
TLSOnly: false,
|
||
|
TLSCertFile: "server.crt",
|
||
|
TLSKeyFile: "server.key",
|
||
|
TLSTrustedCaFile: "ca.crt",
|
||
|
SubDomainHost: "frps.com",
|
||
|
TCPMux: true,
|
||
|
UDPPacketSize: 1509,
|
||
|
|
||
|
HTTPPlugins: map[string]plugin.HTTPPluginOptions{
|
||
|
"user-manager": {
|
||
|
Name: "user-manager",
|
||
|
Addr: "127.0.0.1:9009",
|
||
|
Path: "/handler",
|
||
|
Ops: []string{"Login"},
|
||
|
},
|
||
|
"port-manager": {
|
||
|
Name: "port-manager",
|
||
|
Addr: "127.0.0.1:9009",
|
||
|
Path: "/handler",
|
||
|
Ops: []string{"NewProxy"},
|
||
|
TLSVerify: true,
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
source: []byte(`
|
||
|
# [common] is integral section
|
||
|
[common]
|
||
|
bind_addr = 0.0.0.9
|
||
|
bind_port = 7009
|
||
|
bind_udp_port = 7008
|
||
|
`),
|
||
|
expected: ServerCommonConf{
|
||
|
ServerConfig: auth.ServerConfig{
|
||
|
BaseConfig: auth.BaseConfig{
|
||
|
AuthenticationMethod: "token",
|
||
|
AuthenticateHeartBeats: false,
|
||
|
AuthenticateNewWorkConns: false,
|
||
|
},
|
||
|
},
|
||
|
BindAddr: "0.0.0.9",
|
||
|
BindPort: 7009,
|
||
|
BindUDPPort: 7008,
|
||
|
ProxyBindAddr: "0.0.0.0",
|
||
|
VhostHTTPTimeout: 60,
|
||
|
DashboardAddr: "0.0.0.0",
|
||
|
DashboardUser: "admin",
|
||
|
DashboardPwd: "admin",
|
||
|
EnablePrometheus: false,
|
||
|
LogFile: "console",
|
||
|
LogWay: "console",
|
||
|
LogLevel: "info",
|
||
|
LogMaxDays: 3,
|
||
|
DetailedErrorsToClient: true,
|
||
|
TCPMux: true,
|
||
|
AllowPorts: make(map[int]struct{}),
|
||
|
MaxPoolCount: 5,
|
||
|
HeartbeatTimeout: 90,
|
||
|
UserConnTimeout: 10,
|
||
|
HTTPPlugins: make(map[string]plugin.HTTPPluginOptions),
|
||
|
UDPPacketSize: 1500,
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
for _, c := range testcases {
|
||
|
actual, err := UnmarshalServerConfFromIni(c.source)
|
||
|
assert.NoError(err)
|
||
|
assert.Equal(c.expected, actual)
|
||
|
}
|
||
|
}
|