fix: harden kwallet store boundary
This commit is contained in:
parent
fadf41655f
commit
3f75357c89
2 changed files with 132 additions and 9 deletions
|
|
@ -2,6 +2,7 @@ package kwallet
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"errors"
|
||||||
|
|
||||||
"email-mcp/internal/secretstore"
|
"email-mcp/internal/secretstore"
|
||||||
)
|
)
|
||||||
|
|
@ -19,11 +20,19 @@ type Store struct {
|
||||||
|
|
||||||
var _ secretstore.Store = (*Store)(nil)
|
var _ secretstore.Store = (*Store)(nil)
|
||||||
|
|
||||||
|
var errNilClient = errors.New("kwallet client is nil")
|
||||||
|
|
||||||
func NewStore(client Client) *Store {
|
func NewStore(client Client) *Store {
|
||||||
return &Store{client: client}
|
return &Store{client: client}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Store) Save(ctx context.Context, key string, cred secretstore.Credential) error {
|
func (s *Store) Save(ctx context.Context, key string, cred secretstore.Credential) error {
|
||||||
|
if err := cred.Validate(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if s.client == nil {
|
||||||
|
return errNilClient
|
||||||
|
}
|
||||||
if err := s.client.IsAvailable(ctx); err != nil {
|
if err := s.client.IsAvailable(ctx); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
@ -31,15 +40,14 @@ func (s *Store) Save(ctx context.Context, key string, cred secretstore.Credentia
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
data, err := secretstore.MarshalCredential(cred)
|
data, _ := secretstore.MarshalCredential(cred)
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
return s.client.WriteEntry(ctx, key, data)
|
return s.client.WriteEntry(ctx, key, data)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Store) Load(ctx context.Context, key string) (secretstore.Credential, error) {
|
func (s *Store) Load(ctx context.Context, key string) (secretstore.Credential, error) {
|
||||||
|
if s.client == nil {
|
||||||
|
return secretstore.Credential{}, errNilClient
|
||||||
|
}
|
||||||
if err := s.client.IsAvailable(ctx); err != nil {
|
if err := s.client.IsAvailable(ctx); err != nil {
|
||||||
return secretstore.Credential{}, err
|
return secretstore.Credential{}, err
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -15,13 +15,17 @@ type walletClientStub struct {
|
||||||
readErr error
|
readErr error
|
||||||
readValue []byte
|
readValue []byte
|
||||||
|
|
||||||
openCalled bool
|
isAvailableCalled bool
|
||||||
writeKey string
|
openCalled bool
|
||||||
writeValue []byte
|
writeCalled bool
|
||||||
readKey string
|
readCalled bool
|
||||||
|
writeKey string
|
||||||
|
writeValue []byte
|
||||||
|
readKey string
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *walletClientStub) IsAvailable(context.Context) error {
|
func (c *walletClientStub) IsAvailable(context.Context) error {
|
||||||
|
c.isAvailableCalled = true
|
||||||
return c.availableErr
|
return c.availableErr
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -31,16 +35,40 @@ func (c *walletClientStub) Open(context.Context) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *walletClientStub) WriteEntry(_ context.Context, key string, value []byte) error {
|
func (c *walletClientStub) WriteEntry(_ context.Context, key string, value []byte) error {
|
||||||
|
c.writeCalled = true
|
||||||
c.writeKey = key
|
c.writeKey = key
|
||||||
c.writeValue = value
|
c.writeValue = value
|
||||||
return c.writeErr
|
return c.writeErr
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *walletClientStub) ReadEntry(_ context.Context, key string) ([]byte, error) {
|
func (c *walletClientStub) ReadEntry(_ context.Context, key string) ([]byte, error) {
|
||||||
|
c.readCalled = true
|
||||||
c.readKey = key
|
c.readKey = key
|
||||||
return c.readValue, c.readErr
|
return c.readValue, c.readErr
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestStoreSaveReturnsErrorWhenClientIsNil(t *testing.T) {
|
||||||
|
store := NewStore(nil)
|
||||||
|
|
||||||
|
err := store.Save(context.Background(), secretstore.DefaultAccountKey, secretstore.Credential{
|
||||||
|
Host: "imap.example.com",
|
||||||
|
Username: "alice",
|
||||||
|
Password: "secret",
|
||||||
|
})
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("expected error when client is nil")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStoreLoadReturnsErrorWhenClientIsNil(t *testing.T) {
|
||||||
|
store := NewStore(nil)
|
||||||
|
|
||||||
|
_, err := store.Load(context.Background(), secretstore.DefaultAccountKey)
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("expected error when client is nil")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestStoreSaveWritesSerializedCredential(t *testing.T) {
|
func TestStoreSaveWritesSerializedCredential(t *testing.T) {
|
||||||
client := &walletClientStub{}
|
client := &walletClientStub{}
|
||||||
store := NewStore(client)
|
store := NewStore(client)
|
||||||
|
|
@ -84,6 +112,58 @@ func TestStoreSaveReturnsAvailabilityErrorWithoutOpeningWallet(t *testing.T) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestStoreSaveValidatesBeforeCheckingWalletAvailability(t *testing.T) {
|
||||||
|
client := &walletClientStub{}
|
||||||
|
store := NewStore(client)
|
||||||
|
|
||||||
|
err := store.Save(context.Background(), secretstore.DefaultAccountKey, secretstore.Credential{
|
||||||
|
Host: "imap.example.com",
|
||||||
|
Username: "alice",
|
||||||
|
})
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("expected validation error")
|
||||||
|
}
|
||||||
|
if client.isAvailableCalled {
|
||||||
|
t.Fatal("did not expect availability check for invalid credential")
|
||||||
|
}
|
||||||
|
if client.openCalled || client.writeCalled {
|
||||||
|
t.Fatal("did not expect wallet calls for invalid credential")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStoreSaveReturnsOpenError(t *testing.T) {
|
||||||
|
wantErr := errors.New("open failed")
|
||||||
|
client := &walletClientStub{openErr: wantErr}
|
||||||
|
store := NewStore(client)
|
||||||
|
|
||||||
|
err := store.Save(context.Background(), secretstore.DefaultAccountKey, secretstore.Credential{
|
||||||
|
Host: "imap.example.com",
|
||||||
|
Username: "alice",
|
||||||
|
Password: "secret",
|
||||||
|
})
|
||||||
|
if !errors.Is(err, wantErr) {
|
||||||
|
t.Fatalf("expected open error, got %v", err)
|
||||||
|
}
|
||||||
|
if client.writeCalled {
|
||||||
|
t.Fatal("did not expect write when open fails")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStoreSaveReturnsWriteError(t *testing.T) {
|
||||||
|
wantErr := errors.New("write failed")
|
||||||
|
client := &walletClientStub{writeErr: wantErr}
|
||||||
|
store := NewStore(client)
|
||||||
|
|
||||||
|
err := store.Save(context.Background(), secretstore.DefaultAccountKey, secretstore.Credential{
|
||||||
|
Host: "imap.example.com",
|
||||||
|
Username: "alice",
|
||||||
|
Password: "secret",
|
||||||
|
})
|
||||||
|
if !errors.Is(err, wantErr) {
|
||||||
|
t.Fatalf("expected write error, got %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestStoreLoadReadsAndDecodesCredential(t *testing.T) {
|
func TestStoreLoadReadsAndDecodesCredential(t *testing.T) {
|
||||||
payload, err := secretstore.MarshalCredential(secretstore.Credential{
|
payload, err := secretstore.MarshalCredential(secretstore.Credential{
|
||||||
Host: "imap.example.com",
|
Host: "imap.example.com",
|
||||||
|
|
@ -112,3 +192,38 @@ func TestStoreLoadReadsAndDecodesCredential(t *testing.T) {
|
||||||
t.Fatalf("unexpected credential: %#v", cred)
|
t.Fatalf("unexpected credential: %#v", cred)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestStoreLoadReturnsOpenError(t *testing.T) {
|
||||||
|
wantErr := errors.New("open failed")
|
||||||
|
client := &walletClientStub{openErr: wantErr}
|
||||||
|
store := NewStore(client)
|
||||||
|
|
||||||
|
_, err := store.Load(context.Background(), secretstore.DefaultAccountKey)
|
||||||
|
if !errors.Is(err, wantErr) {
|
||||||
|
t.Fatalf("expected open error, got %v", err)
|
||||||
|
}
|
||||||
|
if client.readCalled {
|
||||||
|
t.Fatal("did not expect read when open fails")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStoreLoadReturnsReadError(t *testing.T) {
|
||||||
|
wantErr := errors.New("read failed")
|
||||||
|
client := &walletClientStub{readErr: wantErr}
|
||||||
|
store := NewStore(client)
|
||||||
|
|
||||||
|
_, err := store.Load(context.Background(), secretstore.DefaultAccountKey)
|
||||||
|
if !errors.Is(err, wantErr) {
|
||||||
|
t.Fatalf("expected read error, got %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStoreLoadReturnsDecodeError(t *testing.T) {
|
||||||
|
client := &walletClientStub{readValue: []byte("not-json")}
|
||||||
|
store := NewStore(client)
|
||||||
|
|
||||||
|
_, err := store.Load(context.Background(), secretstore.DefaultAccountKey)
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("expected decode error")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue