63 lines
1.2 KiB
Go
63 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"sync"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/xtaci/kcp-go"
|
|
)
|
|
|
|
func runKcp(wg *sync.WaitGroup) {
|
|
if wg != nil {
|
|
defer wg.Done()
|
|
}
|
|
|
|
listener, err := kcp.ListenWithOptions(fmt.Sprintf(":%d", config.Kcp.Port), nil, config.Kcp.DataShards, config.Kcp.ParityShards)
|
|
if err != nil {
|
|
log.Fatal().Err(err).
|
|
Int("port", config.Kcp.Port).
|
|
Msg("Failed to listen on KCP port")
|
|
}
|
|
defer listener.Close()
|
|
|
|
log.Info().Int("port", config.Kcp.Port).Msg("KCP server is listening")
|
|
|
|
for {
|
|
conn, err := listener.AcceptKCP()
|
|
if err != nil {
|
|
log.Err(err).
|
|
Msg("Failed to accept KCP connection")
|
|
continue
|
|
}
|
|
log.Debug().
|
|
Str("remote_addr", conn.RemoteAddr().String()).
|
|
Msg("Accepted KCP connection")
|
|
|
|
tuneKcpConn(conn)
|
|
|
|
go handleRequest(conn)
|
|
}
|
|
}
|
|
|
|
func upstreamKcp(host string) net.Conn {
|
|
conn, err := kcp.DialWithOptions(host, nil, config.Kcp.DataShards, config.Kcp.ParityShards)
|
|
if err != nil {
|
|
log.Error().Err(err).
|
|
Msg("Failed to dial KCP server")
|
|
return nil
|
|
}
|
|
|
|
tuneKcpConn(conn)
|
|
return conn
|
|
}
|
|
|
|
func tuneKcpConn(conn *kcp.UDPSession) {
|
|
conn.SetStreamMode(true)
|
|
conn.SetWriteDelay(false)
|
|
conn.SetNoDelay(1, 10, 2, 1)
|
|
conn.SetWindowSize(256, 256)
|
|
conn.SetACKNoDelay(true)
|
|
}
|