burble.dn42/bird-lg-go
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

frontend: limit fetched response size to 64KB (#39)

Browse Source
This commit is contained in:
Yuhui Xu 2021-09-02 20:21:28 -05:00 committed by GitHub
parent 1a3c618522
commit a64d839e2c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
frontend/lgproxy.go
View File

@ -1,7 +1,7 @@
package main
import (
"io/ioutil"
"io"
"net/http"
"net/url"
"strconv"
@ -53,8 +53,14 @@ func batchRequest(servers []string, endpoint string, command string) []string {
ch <- channelData{i, "request failed: " + err.Error() + "\n"}
return
}
text, _ := ioutil.ReadAll(response.Body)
ch <- channelData{i, string(text)}
buf := make([]byte, 65536)
_, err = io.ReadFull(response.Body, buf)
if err != nil && err != io.ErrUnexpectedEOF {
ch <- channelData{i, err.Error()}
} else {
ch <- channelData{i, string(buf)}
}
}(url, i)
}
}

10
frontend/whois.go
View File

@ -1,7 +1,7 @@
package main
import (
"io/ioutil"
"io"
"net"
"time"
)
@ -19,9 +19,11 @@ func whois(s string) string {
defer conn.Close()
conn.Write([]byte(s + "\r\n"))
result, err := ioutil.ReadAll(conn)
if err != nil {
buf := make([]byte, 65536)
_, err = io.ReadFull(conn, buf)
if err != nil && err != io.ErrUnexpectedEOF {
return err.Error()
}
return string(result)
return string(buf)
}