Rename root, api and sdk module paths (imports, go.mod, go_package, ldflags, build scripts, buf configs) and regenerate protobuf/gateway code with the project's pinned toolchain. Wire identifiers unchanged. api gets a local replace for sdk; root/api/sdk require v0.1.0. Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
59 lines
2.3 KiB
Go
59 lines
2.3 KiB
Go
// Copyright IBM Corp. 2020, 2026
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package oidc
|
|
|
|
import (
|
|
"context"
|
|
|
|
"repo.anhonesthost.net/CyberCoveLLC/threshold/internal/db"
|
|
"repo.anhonesthost.net/CyberCoveLLC/threshold/internal/errors"
|
|
"repo.anhonesthost.net/CyberCoveLLC/threshold/internal/kms"
|
|
"repo.anhonesthost.net/CyberCoveLLC/threshold/internal/util"
|
|
)
|
|
|
|
func init() {
|
|
kms.RegisterTableRewrapFn(defaultAuthMethodTableName, authMethodRewrapFn)
|
|
}
|
|
|
|
func authMethodRewrapFn(ctx context.Context, dataKeyVersionId, scopeId string, reader db.Reader, writer db.Writer, kmsRepo kms.GetWrapperer) error {
|
|
const op = "oidc.authMethodRewrapFn"
|
|
if dataKeyVersionId == "" {
|
|
return errors.New(ctx, errors.InvalidParameter, op, "missing data key version id")
|
|
}
|
|
if scopeId == "" {
|
|
return errors.New(ctx, errors.InvalidParameter, op, "missing scope id")
|
|
}
|
|
if util.IsNil(reader) {
|
|
return errors.New(ctx, errors.InvalidParameter, op, "missing database reader")
|
|
}
|
|
if util.IsNil(writer) {
|
|
return errors.New(ctx, errors.InvalidParameter, op, "missing database writer")
|
|
}
|
|
if kmsRepo == nil {
|
|
return errors.New(ctx, errors.InvalidParameter, op, "missing kms repository")
|
|
}
|
|
var authMethods []*AuthMethod
|
|
// There are indexes on (scope id, <other>), so we can query on scope and refine via key id.
|
|
// This is the fastest query we can use without creating a new index on key_id.
|
|
if err := reader.SearchWhere(ctx, &authMethods, "scope_id=? and key_id=?", []any{scopeId, dataKeyVersionId}, db.WithLimit(-1)); err != nil {
|
|
return errors.Wrap(ctx, err, op, errors.WithMsg("failed to query sql for rows that need rewrapping"))
|
|
}
|
|
wrapper, err := kmsRepo.GetWrapper(ctx, scopeId, kms.KeyPurposeDatabase)
|
|
if err != nil {
|
|
return errors.Wrap(ctx, err, op, errors.WithMsg("failed to fetch kms wrapper for rewrapping"))
|
|
}
|
|
for _, am := range authMethods {
|
|
if err := am.decrypt(ctx, wrapper); err != nil {
|
|
return errors.Wrap(ctx, err, op, errors.WithMsg("failed to decrypt auth method"))
|
|
}
|
|
if err := am.encrypt(ctx, wrapper); err != nil {
|
|
return errors.Wrap(ctx, err, op, errors.WithMsg("failed to re-encrypt auth method"))
|
|
}
|
|
if _, err := writer.Update(ctx, am, []string{CtClientSecretField, KeyIdField}, nil); err != nil {
|
|
return errors.Wrap(ctx, err, op, errors.WithMsg("failed to update auth method row with rewrapped fields"))
|
|
}
|
|
}
|
|
return nil
|
|
}
|