frp改造计划续

前言

之前@Wfox师傅在群里提到“通过websocket协议让FRP用上域前置，可以隐藏真实服务ip地址”。最近没有项目，重新进行一下frp改造计划。感谢@Wfox提出的修改思路~

可行性证明

先用dns域名解析来证明域前置方案在frp上是可行的，这里也可以直接修改本地hosts文件来实现dns域名解析的效果。

比如我们用如下frpc.ini

[common]
server_addr = dwnwdqndlnqwln2321321.com
server_port = 23333
token = uknowsec
protocol = websocket
tls_enable = true
[http_proxy]
type = tcp
remote_port = 10002
plugin = socks5

让frpc认证数据包走websocket协议。

可以看到认证是通过websocket协议，这里特别标注出来了Host头，要实现域前置，我们只要把host修改为我们的指定回源域名即可。所以证明了“通过websocket协议让FRP用上域前置”是可行的。

Websocket依赖修改

跟进frp源码，我们可以到websocket依赖包 websocket/hybi.go文件下的hybiClientHandshake函数。

func hybiClientHandshake(config *Config, br *bufio.Reader, bw *bufio.Writer) (err error) {

 `bw.WriteString("GET " + config.Location.RequestURI() + " HTTP/1.1\r\n")`

 `// According to RFC 6874, an HTTP client, proxy, or other`

 `// intermediary must remove any IPv6 zone identifier attached`

```
 `// to an outgoing URI.`
```

 `bw.WriteString("Host: " + removeZone(config.Location.Host) + "\r\n")`

 `bw.WriteString("Upgrade: websocket\r\n")`

 `bw.WriteString("Connection: Upgrade\r\n")`

```
 `nonce := generateNonce()`
```
```
 `if config.handshakeData != nil {`
```

     `nonce = []byte(config.handshakeData["key"])`

```
 `}`
```

 `bw.WriteString("Sec-WebSocket-Key: " + string(nonce) + "\r\n")`

 `bw.WriteString("Origin: " + strings.ToLower(config.Origin.String()) + "\r\n")`

```
 `...`
```
```
 `return nil`
```
}

可以看到Host是通过 config.Location.Host进行赋值的，我们再一步一步的往回看调用即可。

同时frp调用 websocket依赖在 pkg/util/net/websocket.go里的 ConnectWebsocketServer方法

func ConnectWebsocketServer(addr string) (net.Conn, error) {

 `addr = "ws://" + addr + FrpWebsocketPath`

```
 `uri, err := url.Parse(addr)`
```
```
 `if err != nil {`
```
```
     `return nil, err`
```
```
 `}`
```
```
 `origin := "http://" + uri.Host`
```

 `cfg, err := websocket.NewConfig(addr, origin)`

```
 `if err != nil {`
```
```
     `return nil, err`
```
```
 `}`
```
```
 `cfg.Dialer = &net.Dialer{`
```
```
     `Timeout: 10 * time.Second,`
```
```
 `}`
```

 `conn, err := websocket.DialConfig(cfg)`

```
 `if err != nil {`
```
```
     `return nil, err`
```
```
 `}`
```
```
 `return conn, nil`
```
}

所以只需要在往 websocket.NewConfig多传入一个指定的host参数即可。

新加入的host参数只要在 cmd/frpc/sub/root.go的 RegisterCommonFlags里进行注册即可。

测试效果

这样我们就实现了 通过websocket协议让FRP用上域前置。

WSS实现

由上图，可见用websocket还是特征比较明显的，比如 /~!frp。这里我们可以通过如下修改

pkg/util/net/websocket.go下的变量即可。

const (
```
 `FrpWebsocketPath = "/~!frp"`
```
)

同时，我们也修改frp使之实现wss协议。

@Wfox师傅提醒frp有人pull了支持wss协议的修改代码。

https://github.com/fatedier/frp/pull/1919/files

通过pull里的修改就可以实现wss协议了

同时由于在某云域前置里，用wss协议的情况下，server_addr用域名会不能正常回源，只能用ip。且会存在证书报错。

这里可以通过做如下修改 pkg/util/net/websocket.go里的 ConnectWebsocketServer函数

// addr: domain:port
func ConnectWebsocketServer(addr string, websocket_domain string, isSecure bool) (net.Conn, error) {
```
 `if isSecure {`
```
```
     `ho := strings.Split(addr, ":")`
```

     `ip, err := net.ResolveIPAddr("ip", ho[0])`

     `ip_addr := ip.String() + ":" + ho[1]`

```
     `if err != nil {`
```
```
         `return nil, err`
```
```
     `}`
```

     `addr = "wss://" + ip_addr + FrpWebsocketPath`

```
 `} else {`
```

     `addr = "ws://" + addr + FrpWebsocketPath`

```
 `}`
```
```
 `uri, err := url.Parse(addr)`
```
```
 `if err != nil {`
```
```
     `return nil, err`
```
```
 `}`
```
```
 `var origin string`
```
```
 `if isSecure {`
```

     `ho := strings.Split(uri.Host, ":")`

     `ip, err := net.ResolveIPAddr("ip", ho[0])`

     `ip_addr := ip.String() + ":" + ho[1]`

```
     `if err != nil {`
```
```
         `return nil, err`
```
```
     `}`
```
```
     `origin = "https://" + ip_addr`
```
```
 `} else {`
```
```
     `origin = "http://" + uri.Host`
```
```
 `}`
```

 `cfg, err := websocket.NewConfig(addr, origin, websocket_domain)`

```
 `if err != nil {`
```
```
     `return nil, err`
```
```
 `}`
```
```
 `cfg.Dialer = &net.Dialer{`
```
```
     `Timeout: 10 * time.Second,`
```
```
 `}`
```

 `conn, err := websocket.DialConfig(cfg)`

```
 `if err != nil {`
```
```
     `return nil, err`
```
```
 `}`
```
```
 `return conn, nil`
```
}

用 net.ResolveIPAddr先获取域名所对应ip地址，再进行wss和https协议的使用即可。

另外修复证书错误问题。

修改 websocket/dial.go里的 dialWithDialer方法。

func dialWithDialer(dialer *net.Dialer, config *Config) (conn net.Conn, err error) {
```
 `switch config.Location.Scheme {`
```
```
 `case "ws":`
```

     `conn, err = dialer.Dial("tcp", parseAuthority(config.Location))`

```
 `case "wss":`
```
```
     `config.TlsConfig = &tls.Config{`
```
```
         `InsecureSkipVerify: true,`
```
```
     `}`
```

     `conn, err = tls.DialWithDialer(dialer, "tcp", parseAuthority(config.Location), config.TlsConfig)`

```
 `default:`
```
```
     `err = ErrBadScheme`
```
```
 `}`
```
```
 `return`
```
}

当使用wss协议的时候，将 TlsConfig.InsecureSkipVerify设置为true，即可忽略证书错误了。

测试效果

可见图中的认证数据包已经以wss进行认证了。

配置文件自删除

在其中看@lz520520师傅的文章里看到

https://sec.lz520520.cn:4430/2020/11/566/#0x03

只要读取后删除配置文件就好了呀，这个就很简单，我多添加了一个配置文件参数delete，用于判断是否自动删除配置文件。

这一点还是不错的，添加参数，读取完配置文件启动frpc后，自动删除配置文件。

同样相同的方法在 cmd/frpc/sub/root.go的 RegisterCommonFlags里进行注册参数即可。

然后在 cmd/frpc/sub/root.go里的 startService方法里进行判断调用删除配置文件即可。

func startService(
```
 `cfg config.ClientCommonConf,`
```

 `pxyCfgs map[string]config.ProxyConf,`

 `visitorCfgs map[string]config.VisitorConf,`

```
 `cfgFile string,`
```
) (err error) {

 `log.InitLog(cfg.LogWay, cfg.LogFile, cfg.LogLevel,`

     `cfg.LogMaxDays, cfg.DisableLogColor)`

```
 `if cfg.DNSServer != "" {`
```
```
     `s := cfg.DNSServer`
```
```
     `if !strings.Contains(s, ":") {`
```
```
         `s += ":53"`
```
```
     `}`
```

     `// Change default dns server for frpc`

     `net.DefaultResolver = &net.Resolver{`

```
         `PreferGo: true,`
```

         `Dial: func(ctx context.Context, network, address string) (net.Conn, error) {`

             `return net.Dial("udp", s)`

```
         `},`
```
```
     `}`
```
```
 `}`
```

 `svr, errRet := client.NewService(cfg, pxyCfgs, visitorCfgs, cfgFile)`

```
 `if errRet != nil {`
```
```
     `err = errRet`
```
```
     `return`
```
```
 `}`
```
```
 `if cfg.DELEnable == true {`
```
```
     `os.Remove(cfgFile)`
```
```
 `}`
```

 `// Capture the exit signal if we use kcp.`

```
 `if cfg.Protocol == "kcp" {`
```
```
     `go handleSignal(svr)`
```
```
 `}`
```
```
 `err = svr.Run()`
```
```
 `if cfg.Protocol == "kcp" {`
```
```
     `<-kcpDoneCh`
```
```
 `}`
```
```
 `return`
```
}

这样就可以实现配置文件自动删除功能了。

Reference

https://github.com/fatedier/frp/pull/1919/files

https://sec.lz520520.cn:4430/2020/11/566/

长亭百川云 - 文章详情

长亭百川云

长亭百川云 - 文章详情

长亭百川云

前言

可行性证明

Websocket依赖修改

测试效果

WSS实现

测试效果

配置文件自删除

Reference

代码排版转换有点乱，有需求点击查看原文