FRP 更改多 Token 支持

admin

2018年7月5日

FRP 更改多 Token 支持 – Lzxz1234 的小站
简介

之前一直用 Ngrok，但 Ngrok 有比较严重的内存泄露问题，经常无故退出。所以最近更改到了 frp。

frp 是一个可用于内网穿透的高性能的反向代理应用，支持 tcp, udp, http, https 协议。

由于 frp 默认只支持一个 token，这样当 token 泄露更改时会导致全部客户端不可用，所以就有了本次修改。

过程

先 fork 代码库 https://github.com/fatedier/frp，简单阅读后可知，登录时 token 校验逻辑位于 server/service.go 文件中第 320 行：

1. if util.GetAuthKey(g.GlbServerCfg.Token, loginMsg) != loginMsg.PrivilegeKey {

复制代码

所以我们可以新建 server/auth.go 主要处理授权逻辑：

1. package server
   
2. 
   
3. import (
   
4.     "database/sql"
   
5.     "github.com/fatedier/frp/models/msg"
   
6.     "github.com/fatedier/frp/utils/util"
   
7.     _ "github.com/mattn/go-sqlite3"
   
8. )
   
9. 
   
10. var db *sql.DB
    
11. 
    
12. type User struct {
    
13.     UserName string
    
14.     Token string
    
15. }
    
16. 
    
17. func GetAuthKey(loginMsg *msg.Login) (key string) {
    
18. 
    
19.     user := GetUser(loginMsg.User)
    
20.     key = util.GetAuthKey(user.Token, loginMsg.Timestamp)
    
21.     return
    
22. }
    
23. 
    
24. func GetToken(loginMsg *msg.Login) (key string) {
    
25. 
    
26.     user := GetUser(loginMsg.User)
    
27.     key = user.Token
    
28.     return
    
29. }
    
30. 
    
31. func GetUser(userName string) (user User) {
    
32. 
    
33.     if db == nil {
    
34.         db, _ = sql.Open("sqlite3", "/sqlite3.db")
    
35.         db.Exec("create table if not exists user(user_name varchar(128), token varchar(128))")
    
36.     }
    
37. 
    
38.     stmt, _ := db.Prepare("SELECT user_name, token FROM user where user_name = ?")
    
39.     defer stmt.Close()
    
40.     rows, _ := stmt.Query(userName)
    
41.     defer rows.Close()
    
42. 
    
43.     if rows.Next() {
    
44.         rows.Scan(&user.UserName, &user.Token)
    
45.     }
    
46.     return
    
47. }

复制代码

此处将用户信息存储在了 sqlite3 中，同时将用户做成实体，后续可以根据用户限制代理类型个数等。同时由于引用了 github.com/mattn/go-sqlite3 包，所以也需要在 vendor 目录里添加对应包。

其它修改

除了登录时的 token 校验，还有其它地方也需要同步修改，主要是 server/control.go 和 server/proxy.go，这两个文件的修改主要用于替换 token 的获取逻辑，由全局获取改为数据库获取：

control.go

212 行改为：

1. encWriter, err := crypto.NewWriter(ctl.conn, []byte(GetToken(ctl.loginMsg)))

复制代码

242 行改为：

1. encReader := crypto.NewReader(ctl.conn, []byte(GetToken(ctl.loginMsg)))
   

复制代码

311 行改为：

1. rwc, err = frpIo.WithEncryption(rwc, []byte(GetToken(pxy.ctl.loginMsg)))

复制代码

669 行改为：

1. local, err = frpIo.WithEncryption(local, []byte(GetToken(pxy.GetControl().loginMsg)))
   

复制代码

到此重新编辑部署就可以了，更多还可以删除 cmd\frps 目录里启动命令里的 Token 字段设置，不改也无所谓，因为设置也不会生效了