cyonjan/frp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
chenzhongjie 2024-04-10 19:08:35 +08:00
parent b84dd2d9c4
commit d1896f66c5
3 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
server/proxy/http.go
View File

@ -164,7 +164,12 @@ func (pxy *HTTPProxy) GetRealConn(remoteAddr string) (workConn net.Conn, err err
var rwc io.ReadWriteCloser = tmpConn
if pxy.cfg.Transport.UseEncryption {
rwc, err = libio.WithEncryption(rwc, []byte(pxy.serverCfg.Auth.Token))
key := []byte(pxy.serverCfg.Auth.Token)
if pxy.serverCfg.Auth.Method == v1.AuthMethodJWT {
key = []byte(pxy.loginMsg.PrivilegeKey)
}
rwc, err = libio.WithEncryption(rwc, key)
if err != nil {
xl.Errorf("create encryption stream error: %v", err)
return

6
server/proxy/proxy.go
View File

@ -240,7 +240,11 @@ func (pxy *BaseProxy) handleUserTCPConnection(userConn net.Conn) {
xl.Tracef("handler user tcp connection, use_encryption: %t, use_compression: %t",
cfg.Transport.UseEncryption, cfg.Transport.UseCompression)
if cfg.Transport.UseEncryption {
local, err = libio.WithEncryption(local, []byte(serverCfg.Auth.Token))
key := []byte(serverCfg.Auth.Token)
if serverCfg.Auth.Method == v1.AuthMethodJWT {
key = []byte(pxy.loginMsg.PrivilegeKey)
}
local, err = libio.WithEncryption(local, key)
if err != nil {
xl.Errorf("create encryption stream error: %v", err)
return

6
server/proxy/udp.go
View File

@ -205,7 +205,11 @@ func (pxy *UDPProxy) Run() (remoteAddr string, err error) {
var rwc io.ReadWriteCloser = workConn
if pxy.cfg.Transport.UseEncryption {
rwc, err = libio.WithEncryption(rwc, []byte(pxy.serverCfg.Auth.Token))
key := []byte(pxy.serverCfg.Auth.Token)
if pxy.serverCfg.Auth.Method == v1.AuthMethodJWT {
key = []byte(pxy.loginMsg.PrivilegeKey)
}
rwc, err = libio.WithEncryption(rwc, key)
if err != nil {
xl.Errorf("create encryption stream error: %v", err)
workConn.Close()