penguinehis/DragonCoreSSH-NewWEB
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Optimization

Browse Source
This commit is contained in:
penguinehis
2026-05-27 15:04:01 -03:00
parent f64f7fdc4d
commit 1479e6ac73
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
main.go
View File

@@ -376,13 +376,18 @@ func mbpsToBytesPerSec(mbps int) int64 {
return int64(mbps) * 1024 * 1024 / 8 return int64(mbps) * 1024 * 1024 / 8
} }
var copyBufPool = sync.Pool{
New: func() interface{} { b := make([]byte, 32*1024); return &b },
}
func copyWithRateLimit(dst io.Writer, src io.Reader, lim *rate.Limiter) (written int64, err error) { func copyWithRateLimit(dst io.Writer, src io.Reader, lim *rate.Limiter) (written int64, err error) {
if lim == nil { if lim == nil {
return io.Copy(dst, src) return io.Copy(dst, src)
} }
const bufSize = 32 * 1024 bufp := copyBufPool.Get().(*[]byte)
buf := make([]byte, bufSize) buf := *bufp
defer copyBufPool.Put(bufp)
ctx := context.Background() ctx := context.Background()
for { for {
@@ -2130,7 +2135,7 @@ func handleHTTP80Conn(raw net.Conn, sshConfig *ssh.ServerConfig) {
_, _ = raw.Write([]byte(fmt.Sprintf("HTTP/1.1 101 %s\r\n\r\n", status))) _, _ = raw.Write([]byte(fmt.Sprintf("HTTP/1.1 101 %s\r\n\r\n", status)))
skip200 := false skip200 := false
br := bufio.NewReaderSize(raw, 32<<10) br := bufio.NewReaderSize(raw, 4<<10)
// Drain chained HTTP header blocks with a short rolling deadline so Peek/ReadBytes never stalls. // Drain chained HTTP header blocks with a short rolling deadline so Peek/ReadBytes never stalls.
cleanWindow := 30 * time.Second cleanWindow := 30 * time.Second