1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
// Copyright (c) The go-grpc-middleware Authors.
// Licensed under the Apache License 2.0.
package auth
import (
"context"
"strings"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/metadata"
"google.golang.org/grpc/status"
)
const (
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) {
vals := metadata.ValueFromIncomingContext(ctx, headerAuthorize)
if len(vals) == 0 {
return "", status.Error(codes.Unauthenticated, "Request unauthenticated with "+expectedScheme)
}
scheme, token, found := strings.Cut(vals[0], " ")
if !found {
return "", status.Error(codes.Unauthenticated, "Bad authorization string")
}
if !strings.EqualFold(scheme, expectedScheme) {
return "", status.Error(codes.Unauthenticated, "Request unauthenticated with "+expectedScheme)
}
return token, nil
}
|
