mirror of
https://github.com/rocky-linux/peridot.git
synced 2024-12-20 17:58:29 +00:00
38 lines
1.3 KiB
Go
38 lines
1.3 KiB
Go
// Copyright 2016 Michal Witkowski. All Rights Reserved.
|
|
// See LICENSE for licensing terms.
|
|
|
|
package grpc_auth
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"github.com/grpc-ecosystem/go-grpc-middleware/util/metautils"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
var (
|
|
headerAuthorize = "authorization"
|
|
)
|
|
|
|
// AuthFromMD is a helper function for extracting the :authorization header from the gRPC metadata of the request.
|
|
//
|
|
// It expects the `:authorization` header to be of a certain scheme (e.g. `basic`, `bearer`), in a
|
|
// case-insensitive format (see rfc2617, sec 1.2). If no such authorization is found, or the token
|
|
// is of wrong scheme, an error with gRPC status `Unauthenticated` is returned.
|
|
func AuthFromMD(ctx context.Context, expectedScheme string) (string, error) {
|
|
val := metautils.ExtractIncoming(ctx).Get(headerAuthorize)
|
|
if val == "" {
|
|
return "", status.Errorf(codes.Unauthenticated, "Request unauthenticated with "+expectedScheme)
|
|
}
|
|
splits := strings.SplitN(val, " ", 2)
|
|
if len(splits) < 2 {
|
|
return "", status.Errorf(codes.Unauthenticated, "Bad authorization string")
|
|
}
|
|
if !strings.EqualFold(splits[0], expectedScheme) {
|
|
return "", status.Errorf(codes.Unauthenticated, "Request unauthenticated with "+expectedScheme)
|
|
}
|
|
return splits[1], nil
|
|
}
|