scottk/vault
1
0
Fork 0
mirror of https://github.com/optim-enterprises-bv/vault.git synced 2025-11-01 02:57:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update framework forwarding logic to handle nil system views (#8114)

Browse Source
This commit is contained in:
Jim Kalafut
2020-01-08 05:59:44 -08:00
committed by GitHub
parent 80dee15740
commit 30cb97bd1b
3 changed files with 27 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
sdk/framework/backend.go
View File

@@ -228,16 +228,18 @@ func (b *Backend) HandleRequest(ctx context.Context, req *logical.Request) (*log
if op, ok := path.Operations[req.Operation]; ok {
// Check whether this operation should be forwarded
replState := b.System().ReplicationState()
if sysView := b.System(); sysView != nil {
replState := sysView.ReplicationState()
props := op.Properties()
if props.ForwardPerformanceStandby && replState.HasState(consts.ReplicationPerformanceStandby) {
return nil, logical.ErrReadOnly
}
if props.ForwardPerformanceSecondary && !b.System().LocalMount() && replState.HasState(consts.ReplicationPerformanceSecondary) {
if props.ForwardPerformanceSecondary && !sysView.LocalMount() && replState.HasState(consts.ReplicationPerformanceSecondary) {
return nil, logical.ErrReadOnly
}
}
callback = op.Handler()
}

9
sdk/framework/backend_test.go
View File

@@ -124,6 +124,7 @@ func TestBackendHandleRequest_Forwarding(t *testing.T) {
isStandby bool
isSecondary bool
expectFwd bool
nilSysView bool
}{
"no forward": {
expectFwd: false,
@@ -159,6 +160,10 @@ func TestBackendHandleRequest_Forwarding(t *testing.T) {
isSecondary: true,
expectFwd: false,
},
"nil system view": {
nilSysView: true,
expectFwd: false,
},
}
for name, test := range tests {
@@ -193,6 +198,10 @@ func TestBackendHandleRequest_Forwarding(t *testing.T) {
},
}
if test.nilSysView {
b.system = nil
}
_, err := b.HandleRequest(context.Background(), &logical.Request{
Operation: logical.ReadOperation,
Path: "foo",

6
vendor/github.com/hashicorp/vault/sdk/framework/backend.go generated vendored
View File

@@ -228,16 +228,18 @@ func (b *Backend) HandleRequest(ctx context.Context, req *logical.Request) (*log
if op, ok := path.Operations[req.Operation]; ok {
// Check whether this operation should be forwarded
replState := b.System().ReplicationState()
if sysView := b.System(); sysView != nil {
replState := sysView.ReplicationState()
props := op.Properties()
if props.ForwardPerformanceStandby && replState.HasState(consts.ReplicationPerformanceStandby) {
return nil, logical.ErrReadOnly
}
if props.ForwardPerformanceSecondary && !b.System().LocalMount() && replState.HasState(consts.ReplicationPerformanceSecondary) {
if props.ForwardPerformanceSecondary && !sysView.LocalMount() && replState.HasState(consts.ReplicationPerformanceSecondary) {
return nil, logical.ErrReadOnly
}
}
callback = op.Handler()
}