Merge pull request #3209 from nspcc-dev/strict-cfg

Forbid unknown YAML node configuration fields
This commit is contained in:
Roman Khimov 2023-11-21 13:20:57 +03:00 committed by GitHub
commit 80fcf81102
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 29 additions and 4 deletions

View file

@ -80,7 +80,8 @@ ApplicationConfiguration:
- ":20001"
Pprof:
Enabled: false
Port: 20011
Addresses:
- ":20011"
Consensus:
Enabled: true
UnlockWallet:

View file

@ -1,6 +1,7 @@
package config
import (
"bytes"
"fmt"
"os"
"time"
@ -82,8 +83,9 @@ func LoadFile(configPath string) (Config, error) {
},
},
}
err = yaml.Unmarshal(configData, &config)
decoder := yaml.NewDecoder(bytes.NewReader(configData))
decoder.KnownFields(true)
err = decoder.Decode(&config)
if err != nil {
return Config{}, fmt.Errorf("failed to unmarshal config YAML: %w", err)
}

View file

@ -1,8 +1,11 @@
package config
import (
"os"
"path/filepath"
"testing"
"github.com/nspcc-dev/neo-go/pkg/config/netmode"
"github.com/stretchr/testify/require"
)
@ -12,3 +15,20 @@ func TestUnexpectedNativeUpdateHistoryContract(t *testing.T) {
_, err := LoadFile(testConfigPath)
require.Error(t, err)
}
func TestUnknownConfigFields(t *testing.T) {
tmp := t.TempDir()
cfg := filepath.Join(tmp, "protocol.testnet.yml")
require.NoError(t, os.WriteFile(cfg, []byte(`UnknownConfigurationField: 123`), os.ModePerm))
t.Run("LoadFile", func(t *testing.T) {
_, err := LoadFile(cfg)
require.Error(t, err)
require.Contains(t, err.Error(), "field UnknownConfigurationField not found in type config.Config")
})
t.Run("Load", func(t *testing.T) {
_, err := Load(tmp, netmode.TestNet)
require.Error(t, err)
require.Contains(t, err.Error(), "field UnknownConfigurationField not found in type config.Config")
})
}

View file

@ -1149,7 +1149,9 @@ func TestConfigNativeUpdateHistory(t *testing.T) {
"unit_testnet.single",
}
for _, tc := range testCases {
check(t, tc)
t.Run(fmt.Sprintf("%s", tc), func(t *testing.T) {
check(t, tc)
})
}
}