mirror of
https://github.com/fatedier/frp.git
synced 2026-03-08 02:49:10 +08:00
145 lines
4.1 KiB
Go
145 lines
4.1 KiB
Go
// Copyright 2026 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 source
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
v1 "github.com/fatedier/frp/pkg/config/v1"
|
|
)
|
|
|
|
func setDisallowUnknownFieldsForStoreTest(t *testing.T, value bool) func() {
|
|
t.Helper()
|
|
v1.DisallowUnknownFieldsMu.Lock()
|
|
prev := v1.DisallowUnknownFields
|
|
v1.DisallowUnknownFields = value
|
|
v1.DisallowUnknownFieldsMu.Unlock()
|
|
return func() {
|
|
v1.DisallowUnknownFieldsMu.Lock()
|
|
v1.DisallowUnknownFields = prev
|
|
v1.DisallowUnknownFieldsMu.Unlock()
|
|
}
|
|
}
|
|
|
|
func getDisallowUnknownFieldsForStoreTest() bool {
|
|
v1.DisallowUnknownFieldsMu.Lock()
|
|
defer v1.DisallowUnknownFieldsMu.Unlock()
|
|
return v1.DisallowUnknownFields
|
|
}
|
|
|
|
func TestStoreSource_AddProxyAndVisitor_DoesNotApplyRuntimeDefaults(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
path := filepath.Join(t.TempDir(), "store.json")
|
|
storeSource, err := NewStoreSource(StoreSourceConfig{Path: path})
|
|
require.NoError(err)
|
|
|
|
proxyCfg := &v1.TCPProxyConfig{}
|
|
proxyCfg.Name = "proxy1"
|
|
proxyCfg.Type = "tcp"
|
|
proxyCfg.LocalPort = 10080
|
|
|
|
visitorCfg := &v1.XTCPVisitorConfig{}
|
|
visitorCfg.Name = "visitor1"
|
|
visitorCfg.Type = "xtcp"
|
|
visitorCfg.ServerName = "server1"
|
|
visitorCfg.SecretKey = "secret"
|
|
visitorCfg.BindPort = 10081
|
|
|
|
err = storeSource.AddProxy(proxyCfg)
|
|
require.NoError(err)
|
|
err = storeSource.AddVisitor(visitorCfg)
|
|
require.NoError(err)
|
|
|
|
gotProxy := storeSource.GetProxy("proxy1")
|
|
require.NotNil(gotProxy)
|
|
require.Empty(gotProxy.GetBaseConfig().LocalIP)
|
|
|
|
gotVisitor := storeSource.GetVisitor("visitor1")
|
|
require.NotNil(gotVisitor)
|
|
require.Empty(gotVisitor.GetBaseConfig().BindAddr)
|
|
require.Empty(gotVisitor.(*v1.XTCPVisitorConfig).Protocol)
|
|
}
|
|
|
|
func TestStoreSource_LoadFromFile_DoesNotApplyRuntimeDefaults(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
path := filepath.Join(t.TempDir(), "store.json")
|
|
|
|
proxyCfg := &v1.TCPProxyConfig{}
|
|
proxyCfg.Name = "proxy1"
|
|
proxyCfg.Type = "tcp"
|
|
proxyCfg.LocalPort = 10080
|
|
|
|
visitorCfg := &v1.XTCPVisitorConfig{}
|
|
visitorCfg.Name = "visitor1"
|
|
visitorCfg.Type = "xtcp"
|
|
visitorCfg.ServerName = "server1"
|
|
visitorCfg.SecretKey = "secret"
|
|
visitorCfg.BindPort = 10081
|
|
|
|
stored := storeData{
|
|
Proxies: []v1.TypedProxyConfig{{ProxyConfigurer: proxyCfg}},
|
|
Visitors: []v1.TypedVisitorConfig{{VisitorConfigurer: visitorCfg}},
|
|
}
|
|
data, err := json.Marshal(stored)
|
|
require.NoError(err)
|
|
err = os.WriteFile(path, data, 0o600)
|
|
require.NoError(err)
|
|
|
|
storeSource, err := NewStoreSource(StoreSourceConfig{Path: path})
|
|
require.NoError(err)
|
|
|
|
gotProxy := storeSource.GetProxy("proxy1")
|
|
require.NotNil(gotProxy)
|
|
require.Empty(gotProxy.GetBaseConfig().LocalIP)
|
|
|
|
gotVisitor := storeSource.GetVisitor("visitor1")
|
|
require.NotNil(gotVisitor)
|
|
require.Empty(gotVisitor.GetBaseConfig().BindAddr)
|
|
require.Empty(gotVisitor.(*v1.XTCPVisitorConfig).Protocol)
|
|
}
|
|
|
|
func TestStoreSource_LoadFromFile_UnknownFieldsNotAffectedByAmbientStrictness(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
restore := setDisallowUnknownFieldsForStoreTest(t, true)
|
|
t.Cleanup(restore)
|
|
|
|
path := filepath.Join(t.TempDir(), "store.json")
|
|
raw := []byte(`{
|
|
"proxies": [
|
|
{"name":"proxy1","type":"tcp","localPort":10080,"unexpected":"value"}
|
|
],
|
|
"visitors": [
|
|
{"name":"visitor1","type":"xtcp","serverName":"server1","secretKey":"secret","bindPort":10081,"unexpected":"value"}
|
|
]
|
|
}`)
|
|
err := os.WriteFile(path, raw, 0o600)
|
|
require.NoError(err)
|
|
|
|
storeSource, err := NewStoreSource(StoreSourceConfig{Path: path})
|
|
require.NoError(err)
|
|
|
|
require.NotNil(storeSource.GetProxy("proxy1"))
|
|
require.NotNil(storeSource.GetVisitor("visitor1"))
|
|
require.True(getDisallowUnknownFieldsForStoreTest())
|
|
}
|