Skip to content

Commit

Permalink
Fix rebase
Browse files Browse the repository at this point in the history
  • Loading branch information
atburke committed Jan 29, 2025
1 parent 3b32351 commit 6ad2308
Showing 1 changed file with 22 additions and 11 deletions.
33 changes: 22 additions & 11 deletions lib/auth/join/join.go
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ import (
"crypto"
"crypto/x509"
"log/slog"
"net/http"
"os"
"time"

Expand Down Expand Up @@ -812,20 +813,30 @@ func registerUsingTPMMethod(
return certs, trace.Wrap(err)
}

func mapFromHeader(header http.Header) map[string]string {
out := make(map[string]string, len(header))
for k := range header {
out[k] = header.Get(k)
}
return out
}

func registerUsingOracleMethod(
ctx context.Context, client joinServiceClient, token string, hostKeys *newHostKeys, params RegisterParams,
) (*proto.Certs, error) {
certs, err := client.RegisterUsingOracleMethod(ctx, func(challenge string) (*proto.RegisterUsingOracleMethodRequest, error) {
innerHeaders, outerHeaders, err := oracle.CreateSignedRequest(challenge)
if err != nil {
return nil, trace.Wrap(err)
}
return &proto.RegisterUsingOracleMethodRequest{
RegisterUsingTokenRequest: registerUsingTokenRequestForParams(token, hostKeys, params),
Headers: mapFromHeader(outerHeaders),
PayloadHeaders: mapFromHeader(innerHeaders),
}, nil
})
certs, err := client.RegisterUsingOracleMethod(
ctx,
registerUsingTokenRequestForParams(token, hostKeys, params),
func(challenge string) (*proto.OracleSignedRequest, error) {
innerHeaders, outerHeaders, err := oracle.CreateSignedRequest(challenge)
if err != nil {
return nil, trace.Wrap(err)
}
return &proto.OracleSignedRequest{
Headers: mapFromHeader(outerHeaders),
PayloadHeaders: mapFromHeader(innerHeaders),
}, nil
})
return certs, trace.Wrap(err)
}

Expand Down

0 comments on commit 6ad2308

Please sign in to comment.