mirror of
https://github.com/apernet/hysteria.git
synced 2025-04-04 13:07:39 +03:00
84 lines
3 KiB
Go
84 lines
3 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"github.com/lucas-clemente/quic-go"
|
|
"github.com/lucas-clemente/quic-go/congestion"
|
|
hyCongestion "github.com/tobyxdd/hysteria/pkg/congestion"
|
|
"github.com/tobyxdd/hysteria/pkg/core"
|
|
"io"
|
|
"log"
|
|
"net"
|
|
)
|
|
|
|
func server(args []string) {
|
|
var config cmdServerConfig
|
|
err := loadConfig(&config, args)
|
|
if err != nil {
|
|
log.Fatalln("Unable to load configuration:", err)
|
|
}
|
|
if err := config.Check(); err != nil {
|
|
log.Fatalln("Configuration error:", err.Error())
|
|
}
|
|
log.Printf("Configuration loaded: %+v\n", config)
|
|
// Load cert
|
|
cert, err := tls.LoadX509KeyPair(config.CertFile, config.KeyFile)
|
|
if err != nil {
|
|
log.Fatalln("Unable to load the certificate:", err)
|
|
}
|
|
tlsConfig := &tls.Config{
|
|
Certificates: []tls.Certificate{cert},
|
|
NextProtos: []string{TLSAppProtocol},
|
|
MinVersion: tls.VersionTLS13,
|
|
}
|
|
|
|
quicConfig := &quic.Config{
|
|
MaxReceiveStreamFlowControlWindow: config.ReceiveWindowConn,
|
|
MaxReceiveConnectionFlowControlWindow: config.ReceiveWindowClient,
|
|
KeepAlive: true,
|
|
}
|
|
if quicConfig.MaxReceiveStreamFlowControlWindow == 0 {
|
|
quicConfig.MaxReceiveStreamFlowControlWindow = DefaultMaxReceiveStreamFlowControlWindow
|
|
}
|
|
if quicConfig.MaxReceiveConnectionFlowControlWindow == 0 {
|
|
quicConfig.MaxReceiveConnectionFlowControlWindow = DefaultMaxReceiveConnectionFlowControlWindow
|
|
}
|
|
|
|
server, err := core.NewServer(config.ListenAddr, tlsConfig, quicConfig,
|
|
uint64(config.UpMbps)*mbpsToBps, uint64(config.DownMbps)*mbpsToBps,
|
|
func(refBPS uint64) congestion.SendAlgorithmWithDebugInfos {
|
|
return hyCongestion.NewBrutalSender(congestion.ByteCount(refBPS))
|
|
},
|
|
func(addr net.Addr, username string, password string, sSend uint64, sRecv uint64) (core.AuthResult, string) {
|
|
// No authentication logic in relay, just log username and speed
|
|
log.Printf("Client %s connected, negotiated speed in Mbps: Up %d / Down %d\n",
|
|
addr.String(), sSend/mbpsToBps, sRecv/mbpsToBps)
|
|
return core.AuthSuccess, ""
|
|
},
|
|
func(addr net.Addr, username string, err error) {
|
|
log.Printf("Client %s (%s) disconnected: %s\n", addr.String(), username, err.Error())
|
|
},
|
|
func(addr net.Addr, username string, id int, isUDP bool, reqAddr string) (core.ConnectResult, string, io.ReadWriteCloser) {
|
|
log.Printf("Client %s (%s) opened stream ID %d\n", addr.String(), username, id)
|
|
if isUDP {
|
|
return core.ConnBlocked, "unsupported", nil
|
|
}
|
|
conn, err := net.Dial("tcp", config.RemoteAddr)
|
|
if err != nil {
|
|
log.Printf("TCP error when connecting to %s: %s", config.RemoteAddr, err.Error())
|
|
return core.ConnFailed, err.Error(), nil
|
|
}
|
|
return core.ConnSuccess, "", conn
|
|
},
|
|
func(addr net.Addr, username string, id int, isUDP bool, reqAddr string, err error) {
|
|
log.Printf("Client %s (%s) closed stream ID %d: %s", addr.String(), username, id, err.Error())
|
|
},
|
|
)
|
|
if err != nil {
|
|
log.Fatalln("Server initialization failed:", err)
|
|
}
|
|
defer server.Close()
|
|
log.Println("The server is now up and running :)")
|
|
|
|
log.Fatalln("Server error:", server.Serve())
|
|
}
|