Plugins: Periodically update public signing key (#70080)
This commit is contained in:
@@ -14,7 +14,6 @@ import (
|
||||
"github.com/grafana/grafana/pkg/plugins"
|
||||
"github.com/grafana/grafana/pkg/plugins/log"
|
||||
"github.com/grafana/grafana/pkg/plugins/manager/signature/statickey"
|
||||
"github.com/grafana/grafana/pkg/services/featuremgmt"
|
||||
"github.com/grafana/grafana/pkg/setting"
|
||||
)
|
||||
|
||||
@@ -29,9 +28,8 @@ type ManifestKeys struct {
|
||||
}
|
||||
|
||||
type KeyRetriever struct {
|
||||
cfg *setting.Cfg
|
||||
log log.Logger
|
||||
flags featuremgmt.FeatureToggles
|
||||
cfg *setting.Cfg
|
||||
log log.Logger
|
||||
|
||||
lock sync.Mutex
|
||||
cli http.Client
|
||||
@@ -41,20 +39,19 @@ type KeyRetriever struct {
|
||||
|
||||
var _ plugins.KeyRetriever = (*KeyRetriever)(nil)
|
||||
|
||||
func ProvideService(cfg *setting.Cfg, kv plugins.KeyStore, flags featuremgmt.FeatureToggles) *KeyRetriever {
|
||||
func ProvideService(cfg *setting.Cfg, kv plugins.KeyStore) *KeyRetriever {
|
||||
kr := &KeyRetriever{
|
||||
cfg: cfg,
|
||||
flags: flags,
|
||||
log: log.New("plugin.signature.key_retriever"),
|
||||
cli: makeHttpClient(),
|
||||
kv: kv,
|
||||
cfg: cfg,
|
||||
log: log.New("plugin.signature.key_retriever"),
|
||||
cli: makeHttpClient(),
|
||||
kv: kv,
|
||||
}
|
||||
return kr
|
||||
}
|
||||
|
||||
// IsDisabled disables dynamic retrieval of public keys from the API server.
|
||||
func (kr *KeyRetriever) IsDisabled() bool {
|
||||
return !kr.flags.IsEnabled(featuremgmt.FlagPluginsAPIManifestKey)
|
||||
return kr.cfg.PluginSkipPublicKeyDownload
|
||||
}
|
||||
|
||||
func (kr *KeyRetriever) Run(ctx context.Context) error {
|
||||
|
||||
@@ -9,7 +9,6 @@ import (
|
||||
"time"
|
||||
|
||||
"github.com/grafana/grafana/pkg/infra/kvstore"
|
||||
"github.com/grafana/grafana/pkg/services/featuremgmt"
|
||||
"github.com/grafana/grafana/pkg/services/pluginsintegration/keystore"
|
||||
"github.com/grafana/grafana/pkg/setting"
|
||||
"github.com/stretchr/testify/require"
|
||||
@@ -47,7 +46,7 @@ func Test_PublicKeyUpdate(t *testing.T) {
|
||||
expectedKey := "fake"
|
||||
s, done := setFakeAPIServer(t, expectedKey, "7e4d0c6a708866e7")
|
||||
cfg.GrafanaComURL = s.URL
|
||||
v := ProvideService(cfg, keystore.ProvideService(kvstore.NewFakeKVStore()), featuremgmt.WithFeatures(featuremgmt.FlagPluginsAPIManifestKey))
|
||||
v := ProvideService(cfg, keystore.ProvideService(kvstore.NewFakeKVStore()))
|
||||
go func() {
|
||||
err := v.Run(context.Background())
|
||||
require.NoError(t, err)
|
||||
@@ -68,7 +67,7 @@ func Test_PublicKeyUpdate(t *testing.T) {
|
||||
expectedKey := "fake"
|
||||
s, done := setFakeAPIServer(t, expectedKey, "7e4d0c6a708866e7")
|
||||
cfg.GrafanaComURL = s.URL
|
||||
v := ProvideService(cfg, keystore.ProvideService(kvstore.NewFakeKVStore()), featuremgmt.WithFeatures(featuremgmt.FlagPluginsAPIManifestKey))
|
||||
v := ProvideService(cfg, keystore.ProvideService(kvstore.NewFakeKVStore()))
|
||||
go func() {
|
||||
err := v.Run(context.Background())
|
||||
require.NoError(t, err)
|
||||
@@ -88,7 +87,7 @@ func Test_PublicKeyUpdate(t *testing.T) {
|
||||
expectedKey := "fake"
|
||||
s, done := setFakeAPIServer(t, expectedKey, "other")
|
||||
cfg.GrafanaComURL = s.URL
|
||||
v := ProvideService(cfg, keystore.ProvideService(kvstore.NewFakeKVStore()), featuremgmt.WithFeatures(featuremgmt.FlagPluginsAPIManifestKey))
|
||||
v := ProvideService(cfg, keystore.ProvideService(kvstore.NewFakeKVStore()))
|
||||
go func() {
|
||||
err := v.Run(context.Background())
|
||||
require.NoError(t, err)
|
||||
@@ -115,7 +114,7 @@ func Test_PublicKeyUpdate(t *testing.T) {
|
||||
expectedKey := "fake"
|
||||
s, done := setFakeAPIServer(t, expectedKey, "7e4d0c6a708866e7")
|
||||
cfg.GrafanaComURL = s.URL
|
||||
v := ProvideService(cfg, keystore.ProvideService(kvstore.NewFakeKVStore()), featuremgmt.WithFeatures(featuremgmt.FlagPluginsAPIManifestKey))
|
||||
v := ProvideService(cfg, keystore.ProvideService(kvstore.NewFakeKVStore()))
|
||||
// Simulate an updated key
|
||||
err := v.kv.SetLastUpdated(context.Background())
|
||||
require.NoError(t, err)
|
||||
|
||||
@@ -6,7 +6,6 @@ import (
|
||||
|
||||
"github.com/grafana/grafana/pkg/infra/kvstore"
|
||||
"github.com/grafana/grafana/pkg/plugins/manager/signature/statickey"
|
||||
"github.com/grafana/grafana/pkg/services/featuremgmt"
|
||||
"github.com/grafana/grafana/pkg/services/pluginsintegration/keyretriever/dynamic"
|
||||
"github.com/grafana/grafana/pkg/services/pluginsintegration/keystore"
|
||||
"github.com/grafana/grafana/pkg/setting"
|
||||
@@ -16,7 +15,7 @@ import (
|
||||
func Test_GetPublicKey(t *testing.T) {
|
||||
t.Run("it should return a static key", func(t *testing.T) {
|
||||
cfg := &setting.Cfg{}
|
||||
kr := ProvideService(dynamic.ProvideService(cfg, keystore.ProvideService(kvstore.NewFakeKVStore()), featuremgmt.WithFeatures()))
|
||||
kr := ProvideService(dynamic.ProvideService(cfg, keystore.ProvideService(kvstore.NewFakeKVStore())))
|
||||
key, err := kr.GetPublicKey(context.Background(), statickey.GetDefaultKeyID())
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, statickey.GetDefaultKey(), key)
|
||||
|
||||
Reference in New Issue
Block a user