Skip to content

Commit

Permalink
have rekor.NewSigner accept a *client.Rekor instead of a URL (#1115)
Browse files Browse the repository at this point in the history
Signed-off-by: Jake Sanders <jsand@google.com>
  • Loading branch information
Jake Sanders committed Nov 30, 2021
1 parent ac7e33c commit 1da6742
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 15 deletions.
7 changes: 6 additions & 1 deletion cmd/cosign/cli/sign/sign.go
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,7 @@ import (
providers "github.com/sigstore/cosign/pkg/providers/all"
sigs "github.com/sigstore/cosign/pkg/signature"
fulcPkgClient "github.com/sigstore/fulcio/pkg/client"
rekorClient "github.com/sigstore/rekor/pkg/client"
"github.com/sigstore/sigstore/pkg/cryptoutils"
"github.com/sigstore/sigstore/pkg/signature"
sigPayload "github.com/sigstore/sigstore/pkg/signature/payload"
Expand Down Expand Up @@ -215,7 +216,11 @@ func signDigest(ctx context.Context, digest name.Digest, payload []byte, ko KeyO
s = ipayload.NewSigner(sv, nil, nil)
s = ifulcio.NewSigner(s, sv.Cert, sv.Chain)
if ShouldUploadToTlog(ctx, digest, force, ko.RekorURL) {
s = irekor.NewSigner(s, ko.RekorURL)
rClient, err := rekorClient.GetRekorClient(ko.RekorURL)
if err != nil {
return err
}
s = irekor.NewSigner(s, rClient)
}

ociSig, _, err := s.Sign(ctx, bytes.NewReader(payload))
Expand Down
23 changes: 9 additions & 14 deletions internal/pkg/cosign/rekor/rekor.go
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,7 @@ import (
cosignv1 "github.com/sigstore/cosign/pkg/cosign"
"github.com/sigstore/cosign/pkg/oci"
"github.com/sigstore/cosign/pkg/oci/static"
rekPkgClient "github.com/sigstore/rekor/pkg/client"
rekGenClient "github.com/sigstore/rekor/pkg/generated/client"
"github.com/sigstore/rekor/pkg/generated/client"
"github.com/sigstore/rekor/pkg/generated/models"
"github.com/sigstore/sigstore/pkg/cryptoutils"
)
Expand All @@ -47,14 +46,10 @@ func bundle(entry *models.LogEntryAnon) *oci.Bundle {
}
}

type tlogUploadFn func(*rekGenClient.Rekor, []byte) (*models.LogEntryAnon, error)
type tlogUploadFn func(*client.Rekor, []byte) (*models.LogEntryAnon, error)

func uploadToTlog(rekorBytes []byte, rekorURL string, upload tlogUploadFn) (*oci.Bundle, error) {
rekorClient, err := rekPkgClient.GetRekorClient(rekorURL)
if err != nil {
return nil, err
}
entry, err := upload(rekorClient, rekorBytes)
func uploadToTlog(rekorBytes []byte, rClient *client.Rekor, upload tlogUploadFn) (*oci.Bundle, error) {
entry, err := upload(rClient, rekorBytes)
if err != nil {
return nil, err
}
Expand All @@ -66,7 +61,7 @@ func uploadToTlog(rekorBytes []byte, rekorURL string, upload tlogUploadFn) (*oci
type signerWrapper struct {
inner cosign.Signer

rekorURL string
rClient *client.Rekor
}

var _ cosign.Signer = (*signerWrapper)(nil)
Expand Down Expand Up @@ -107,7 +102,7 @@ func (rs *signerWrapper) Sign(ctx context.Context, payload io.Reader) (oci.Signa
return nil, nil, err
}

bundle, err := uploadToTlog(rekorBytes, rs.rekorURL, func(r *rekGenClient.Rekor, b []byte) (*models.LogEntryAnon, error) {
bundle, err := uploadToTlog(rekorBytes, rs.rClient, func(r *client.Rekor, b []byte) (*models.LogEntryAnon, error) {
return cosignv1.TLogUpload(ctx, r, sigBytes, payloadBytes, b)
})
if err != nil {
Expand Down Expand Up @@ -149,9 +144,9 @@ func (rs *signerWrapper) Sign(ctx context.Context, payload io.Reader) (oci.Signa
}

// NewSigner returns a `cosign.Signer` which uploads the signature to Rekor
func NewSigner(inner cosign.Signer, rekorURL string) cosign.Signer {
func NewSigner(inner cosign.Signer, rClient *client.Rekor) cosign.Signer {
return &signerWrapper{
inner: inner,
rekorURL: rekorURL,
inner: inner,
rClient: rClient,
}
}

0 comments on commit 1da6742

Please sign in to comment.