gRPC with JWT

26711

在 gRPC 中使用 JWT（JSON Web Tokens）进行身份验证是一种常见的做法，它可以帮助你确保请求方的身份和权限。下面是一种使用 gRPC 和 JWT 进行身份验证的步骤：

  1. 生成和签发 JWT： 在用户登录成功后，你需要生成一个 JWT 并将其签发给用户。JWT 中可以包含一些有关用户身份、角色、权限等的信息。
  2. 在 gRPC 的上下文中传递 JWT： 当客户端发送 gRPC 请求时，可以将 JWT 放置在 gRPC 请求的元数据（Metadata）中，作为请求的一部分。这样，服务器端就可以获取 JWT 并对其进行验证。
  3. 服务器端验证 JWT： 在 gRPC 服务端，你需要编写代码来验证接收到的 JWT。这通常涉及到验证 JWT 的签名是否有效，以及检查其中的身份信息和权限等。
  4. 决策和授权： 根据验证后的 JWT 信息，你可以决定是否允许用户继续访问请求的资源。这可能涉及到一些授权策略和业务逻辑。

以下是一个简单的示例，展示如何在 gRPC 中使用 JWT 进行身份验证：

proto文件

内容如下：

syntax = "proto3";
package chaincode.pb;
option go_package = "./;pb";
message HelloRequest { string name = 1; }
message HelloResponse { string reply = 2; }
service SayHi { rpc Hi(HelloRequest) returns (HelloResponse); }
syntax = "proto3";

package chaincode.pb;

option go_package = "./;pb";

message HelloRequest { string name = 1; }
message HelloResponse { string reply = 2; }

service SayHi { rpc Hi(HelloRequest) returns (HelloResponse); }
syntax = "proto3";

package chaincode.pb;

option go_package = "./;pb";

message HelloRequest { string name = 1; }
message HelloResponse { string reply = 2; }

service SayHi { rpc Hi(HelloRequest) returns (HelloResponse); }

通过下面的命令生成相关的文件：

$ protoc --go_out=./ --go-grpc_out=./ example.proto
$ tree
.
├── example_grpc.pb.go
├── example.pb.go
└── example.proto
0 directories, 3 files
$ protoc --go_out=./ --go-grpc_out=./ example.proto
$ tree
.
├── example_grpc.pb.go
├── example.pb.go
└── example.proto

0 directories, 3 files
$ protoc --go_out=./ --go-grpc_out=./ example.proto
$ tree
.
├── example_grpc.pb.go
├── example.pb.go
└── example.proto

0 directories, 3 files

server端

跟 client 端约定内容如下：

  • token有效期为半小时
  • iss使用gRPC token
  • sub使用gRPC example server

代码如下：

package main
import (
  "chaincode/pb"
  "context"
  "fmt"
  "net"
  "time"
  "github.com/golang-jwt/jwt/v5"
  "github.com/pkg/errors"
  "google.golang.org/grpc"
  "google.golang.org/grpc/metadata"
)
var testKey = "testKey"
type HiService struct {
  pb.UnimplementedSayHiServer
}
func verifyToken(tokenString string) error {
  token, err := jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) {
    return []byte(testKey), nil
  })
  if err != nil {
    return errors.Wrap(err, "init token parser error")
  }
  if !token.Valid {
    return errors.New("invalid token")
  }
  claims, ok := token.Claims.(jwt.MapClaims)
  if !ok {
    return errors.New("invalid claims")
  }
  exp, err := claims.GetExpirationTime()
  if err != nil {
    return errors.Wrap(err, "GetExpirationTime from token error")
  }
  now := time.Now()
  if now.Sub(exp.Time) > 0 {
    return errors.New("the token expires")
  }
  if claims["sub"] != "gRPC example server" {
    return errors.New("invalid sub")
  }
  if claims["iss"] != "gRPC token" {
    return errors.New("invalid iss")
  }
  return nil
}
func (s *HiService) Hi(ctx context.Context, req *pb.HelloRequest) (*pb.HelloResponse, error) {
  md, ok := metadata.FromIncomingContext(ctx)
  if !ok {
    return nil, errors.New("token信息获取失败")
  }
  token := md.Get("Authorization")[0]
  if err := verifyToken(token); err != nil {
    return nil, errors.Wrap(err, "token验证失败")
  }
  return &pb.HelloResponse{Reply: "hello " + req.Name}, nil
}
func main() {
  // 创建grpc服务示例
  sv := grpc.NewServer()
  // 注册我们的服务
  pb.RegisterSayHiServer(sv, new(HiService))
  // 绑定端口，提供服务
  lis, err := net.Listen("tcp", ":50001")
  if err != nil {
    panic(err)
  }
  // 启动服务
  fmt.Println("liston on: 50001")
  sv.Serve(lis)
}
package main

import (
  "chaincode/pb"
  "context"
  "fmt"
  "net"
  "time"

  "github.com/golang-jwt/jwt/v5"
  "github.com/pkg/errors"
  "google.golang.org/grpc"
  "google.golang.org/grpc/metadata"
)

var testKey = "testKey"

type HiService struct {
  pb.UnimplementedSayHiServer
}

func verifyToken(tokenString string) error {
  token, err := jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) {
    return []byte(testKey), nil
  })
  if err != nil {
    return errors.Wrap(err, "init token parser error")
  }
  if !token.Valid {
    return errors.New("invalid token")
  }
  claims, ok := token.Claims.(jwt.MapClaims)
  if !ok {
    return errors.New("invalid claims")
  }
  exp, err := claims.GetExpirationTime()
  if err != nil {
    return errors.Wrap(err, "GetExpirationTime from token error")
  }

  now := time.Now()
  if now.Sub(exp.Time) > 0 {
    return errors.New("the token expires")
  }
  if claims["sub"] != "gRPC example server" {
    return errors.New("invalid sub")
  }
  if claims["iss"] != "gRPC token" {
    return errors.New("invalid iss")
  }
  return nil
}

func (s *HiService) Hi(ctx context.Context, req *pb.HelloRequest) (*pb.HelloResponse, error) {

  md, ok := metadata.FromIncomingContext(ctx)
  if !ok {
    return nil, errors.New("token信息获取失败")
  }
  token := md.Get("Authorization")[0]
  if err := verifyToken(token); err != nil {
    return nil, errors.Wrap(err, "token验证失败")
  }

  return &pb.HelloResponse{Reply: "hello " + req.Name}, nil
}

func main() {
  // 创建grpc服务示例
  sv := grpc.NewServer()
  // 注册我们的服务
  pb.RegisterSayHiServer(sv, new(HiService))

  // 绑定端口，提供服务
  lis, err := net.Listen("tcp", ":50001")
  if err != nil {
    panic(err)
  }
  // 启动服务
  fmt.Println("liston on: 50001")
  sv.Serve(lis)
}
package main

import (
	"chaincode/pb"
	"context"
	"fmt"
	"net"
	"time"

	"github.com/golang-jwt/jwt/v5"
	"github.com/pkg/errors"
	"google.golang.org/grpc"
	"google.golang.org/grpc/metadata"
)

var testKey = "testKey"

type HiService struct {
	pb.UnimplementedSayHiServer
}

func verifyToken(tokenString string) error {
	token, err := jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) {
		return []byte(testKey), nil
	})
	if err != nil {
		return errors.Wrap(err, "init token parser error")
	}
	if !token.Valid {
		return errors.New("invalid token")
	}
	claims, ok := token.Claims.(jwt.MapClaims)
	if !ok {
		return errors.New("invalid claims")
	}
	exp, err := claims.GetExpirationTime()
	if err != nil {
		return errors.Wrap(err, "GetExpirationTime from token error")
	}

	now := time.Now()
	if now.Sub(exp.Time) > 0 {
		return errors.New("the token expires")
	}
	if claims["sub"] != "gRPC example server" {
		return errors.New("invalid sub")
	}
	if claims["iss"] != "gRPC token" {
		return errors.New("invalid iss")
	}
	return nil
}

func (s *HiService) Hi(ctx context.Context, req *pb.HelloRequest) (*pb.HelloResponse, error) {

	md, ok := metadata.FromIncomingContext(ctx)
	if !ok {
		return nil, errors.New("token信息获取失败")
	}
	token := md.Get("Authorization")[0]
	if err := verifyToken(token); err != nil {
		return nil, errors.Wrap(err, "token验证失败")
	}

	return &pb.HelloResponse{Reply: "hello " + req.Name}, nil
}

func main() {
	// 创建grpc服务示例
	sv := grpc.NewServer()
	// 注册我们的服务
	pb.RegisterSayHiServer(sv, new(HiService))

	// 绑定端口，提供服务
	lis, err := net.Listen("tcp", ":50001")
	if err != nil {
		panic(err)
	}
	// 启动服务
	fmt.Println("liston on: 50001")
	sv.Serve(lis)
}

$ go run main.go
liston on: 50001
$ go run main.go
liston on: 50001
$ go run main.go
liston on: 50001

client

代码如下：

package main
import (
  "chaincode/pb"
  "context"
  "fmt"
  "time"
  "github.com/golang-jwt/jwt/v5"
  "google.golang.org/grpc"
  "google.golang.org/grpc/credentials/insecure"
)
var testKey = "testKey"
func genToken() (string, error) {
  claims := jwt.RegisteredClaims{
    ExpiresAt: jwt.NewNumericDate(time.Now().Add(30 * time.Minute)),
    Issuer:    "gRPC token",
    Subject:   "gRPC example client",
  }
  token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
  return token.SignedString([]byte(testKey))
}
type TokeAuth struct {
  Token string
}
func (t *TokeAuth) GetRequestMetadata(ctx context.Context, uri ...string) (map[string]string, error) {
  return map[string]string{
    "Authorization": t.Token,
  }, nil
}
func (t *TokeAuth) RequireTransportSecurity() bool {
  return false
}
func main() {
  token, err := genToken()
  if err != nil {
    panic(err)
  }
  conn, err := grpc.Dial("localhost:50001", grpc.WithTransportCredentials(insecure.NewCredentials()), grpc.WithPerRPCCredentials(&TokeAuth{Token: token}))
  if err != nil {
    panic(err)
  }
  defer conn.Close()
  client := pb.NewSayHiClient(conn)
  resp, err := client.Hi(context.Background(), &pb.HelloRequest{Name: "Wang"})
  if err != nil {
    panic(err)
  }
  fmt.Println(resp.String())
}
package main

import (
  "chaincode/pb"
  "context"
  "fmt"
  "time"

  "github.com/golang-jwt/jwt/v5"
  "google.golang.org/grpc"
  "google.golang.org/grpc/credentials/insecure"
)

var testKey = "testKey"

func genToken() (string, error) {
  claims := jwt.RegisteredClaims{
    ExpiresAt: jwt.NewNumericDate(time.Now().Add(30 * time.Minute)),
    Issuer:    "gRPC token",
    Subject:   "gRPC example client",
  }
  token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
  return token.SignedString([]byte(testKey))
}

type TokeAuth struct {
  Token string
}

func (t *TokeAuth) GetRequestMetadata(ctx context.Context, uri ...string) (map[string]string, error) {
  return map[string]string{
    "Authorization": t.Token,
  }, nil
}

func (t *TokeAuth) RequireTransportSecurity() bool {
  return false
}

func main() {
  token, err := genToken()
  if err != nil {
    panic(err)
  }
  conn, err := grpc.Dial("localhost:50001", grpc.WithTransportCredentials(insecure.NewCredentials()), grpc.WithPerRPCCredentials(&TokeAuth{Token: token}))
  if err != nil {
    panic(err)
  }
  defer conn.Close()

  client := pb.NewSayHiClient(conn)

  resp, err := client.Hi(context.Background(), &pb.HelloRequest{Name: "Wang"})
  if err != nil {
    panic(err)
  }
  fmt.Println(resp.String())
}
package main

import (
	"chaincode/pb"
	"context"
	"fmt"
	"time"

	"github.com/golang-jwt/jwt/v5"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
)

var testKey = "testKey"

func genToken() (string, error) {
	claims := jwt.RegisteredClaims{
		ExpiresAt: jwt.NewNumericDate(time.Now().Add(30 * time.Minute)),
		Issuer:    "gRPC token",
		Subject:   "gRPC example client",
	}
	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
	return token.SignedString([]byte(testKey))
}

type TokeAuth struct {
	Token string
}

func (t *TokeAuth) GetRequestMetadata(ctx context.Context, uri ...string) (map[string]string, error) {
	return map[string]string{
		"Authorization": t.Token,
	}, nil
}

func (t *TokeAuth) RequireTransportSecurity() bool {
	return false
}

func main() {
	token, err := genToken()
	if err != nil {
		panic(err)
	}
	conn, err := grpc.Dial("localhost:50001", grpc.WithTransportCredentials(insecure.NewCredentials()), grpc.WithPerRPCCredentials(&TokeAuth{Token: token}))
	if err != nil {
		panic(err)
	}
	defer conn.Close()

	client := pb.NewSayHiClient(conn)

	resp, err := client.Hi(context.Background(), &pb.HelloRequest{Name: "Wang"})
	if err != nil {
		panic(err)
	}
	fmt.Println(resp.String())
}

现在我们先将 client 端生成 token 的sub 设置为 gRPC example client，执行

$ go run main.go
panic: rpc error: code = Unknown desc = token验证失败: invalid sub
goroutine 1 [running]:
main.main()
        /root/go/src/example/client/main.go:55 +0x2f2
exit status 2
$ go run main.go
panic: rpc error: code = Unknown desc = token验证失败: invalid sub

goroutine 1 [running]:
main.main()
        /root/go/src/example/client/main.go:55 +0x2f2
exit status 2
$ go run main.go
panic: rpc error: code = Unknown desc = token验证失败: invalid sub

goroutine 1 [running]:
main.main()
        /root/go/src/example/client/main.go:55 +0x2f2
exit status 2

再将 client 端生成 token 的sub 设置为 gRPC example server，执行

$ go run main.go
reply:"hello Wang"
$ go run main.go
reply:"hello Wang"
$ go run main.go
reply:"hello Wang"

以上示例是一个简单的代码示例，实际上还需要处理错误、安全性和其他细节。

声明：本作品采用署名-非商业性使用-相同方式共享 4.0 国际 (CC BY-NC-SA 4.0)进行许可，使用时请注明出处。
Author: mengbin
blog: mengbin
Github: mengbin92
cnblogs: 恋水无意

展开阅读全文
© 版权声明
本站所有资源来自于网络，仅供学习与参考，请勿用于商业用途，否则产生的一切后果将由您(转载者)自己承担！
如有侵犯您的版权，请及时联系3500663466#qq.com(#换@)，我们将第一时间删除本站数据。
THE END
Golang后端开发
支持一下吧
点赞11 分享
e80876ce71530ec7的头像-牛翰网
相关推荐
评论 共2条
头像
请文明发言！
提交
头像

昵称

取消
昵称

请填写用户信息：

  • 昵称(必填)
  • 邮箱(必填)
表情
[aoman][baiyan][bishi][bizui][cahan][ciya][dabing][daku][deyi][doge][fadai][fanu][fendou][ganga][guzhang][haixiu][hanxiao][zuohengheng][zhuakuang][zhouma][zhemo][zhayanjian][zaijian][yun][youhengheng][yiwen][yinxian][xu][xieyanxiao][xiaoku][xiaojiujie][xia][wunai][wozuimei][weixiao][weiqu][tuosai][tu][touxiao][tiaopi][shui][se][saorao][qiudale][qinqin][qiaoda][piezui][penxue][nanguo][liulei][liuhan][lenghan][leiben][kun][kuaikule][ku][koubi][kelian][keai][jingya][jingxi][jingkong][jie][huaixiao][haqian][aini][OK][qiang][quantou][shengli][woshou][gouyin][baoquan][aixin][bangbangtang][xiaoyanger][xigua][hexie][pijiu][lanqiu][juhua][hecai][haobang][caidao][baojin][chi][dan][kulou][shuai][shouqiang][yangtuo][youling]
代码

请输入代码：

确认
快捷回复
谢谢你的分享，我从中学到了很多！
教程很好用，谢谢！
好东西，学习一下！
楼主听话，快到碗里来！
路过一下，我只是来打酱油的！
水帖美如花，养护靠大家！
    只看作者
    • 头像binance0
      I don't think the title of your article matches the content lol. Just kidding, mainly because I had some doubts after reading the article.
      5个月前广西回复
    • 头像open a binance account0
      Thanks for sharing. I read many of your blog posts, cool, your blog is very good.
      5个月前回复

S/4 HANA 中的 Email Template

postgis_raster 从 2.x 升级至 3.x

解决vs2022 工具箱中不显示 Devexpress控件的问题

webstorm破解激活2024最新永久激活码教程(含win+mac)

chatGPT前端流式输出js实现三种方法—fetch、SSE、websocket

springcloud/springboot集成NACOS 做注册和配置中心以及nacos源码分析

小米盒子3增强版root破解教程（MDZ-18-AA型号）

KMS在线激活Win11/10/ 8/ 7和Office/Visio/Project 2021~2010

阿里二面：Kafka中如何保证消息的顺序性？这周被问到两次了

NavicatPremium16破解详解

protobuf

go数据类型-sync.map

Python中的函数未定义的错误

Go切片是值传递还是引用传递？

gRPC with JWT

github.com/json-iterator/go 详细教程

Go 的函数，方法和接口

【matplotlib基础】–刻度

用go封装和实现扫码登录

掌握Go类型内嵌：设计模式与架构的新视角

领导：谁再用 Redis 实现过期订单关闭，立马滚蛋！

RocketMQ源码详解(消息存储、Consumer)

【老马】加密机加解密系列汇总篇(共计13篇)

Java-Stream API

wso2~apim_metrics的配置与二次开发

RocketMQ源码详解(NameServer、Producer)

【老马】离线版金融敏感信息加解密组件开源项目 encryption-local

SpringBoot3整合AI

wso2~对接外部认证系统keycloak

敏感词过滤 + 限流

iNeuOS工业互联网操作系统，高效采集数据配置与应用

ElasticSearch8 – 基础概念和映射

让运维无忧，实战解析巡检报告功能实现方案

Apache SeaTunnel 社区 3 月月报

天翼云入选“2023年度数据要素价值创新标杆示范案例”！

.net中优秀依赖注入框架Autofac看一篇就够了

环境的搭建

Apache SeaTunnel 及 Web 功能部署指南(小白版)

直播预约丨《袋鼠云大数据实操指南》No.1：从理论到实践，离线开发全流程解析

CloudCanal x Hive 构建高效的实时数仓

标签云
C++ReactVueOracleJSsqlNETnode.js人工智能JavaScriptphpASP.NET计算机视觉正则表达式json前端鸿蒙开发AngularVue3C语言QtCoreServertypescript数据nodenodejs安装文件字符串图片npm数据库SQLServer函数报错uniapp人脸识别MariaDBmysql配置Excel组件上传API微信小程序数组java下载查询