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 package main
import ( import (
"io/ioutil" "io"
"net/http" "net/http"
"net/url" "net/url"
"strconv" "strconv"
@ -53,8 +53,14 @@ func batchRequest(servers []string, endpoint string, command string) []string {
ch <- channelData{i, "request failed: " + err.Error() + "\n"} ch <- channelData{i, "request failed: " + err.Error() + "\n"}
return 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) }(url, i)
} }
} }

10
frontend/whois.go
View File

@ -1,7 +1,7 @@
package main package main
import ( import (
"io/ioutil" "io"
"net" "net"
"time" "time"
) )
@ -19,9 +19,11 @@ func whois(s string) string {
defer conn.Close() defer conn.Close()
conn.Write([]byte(s + "\r\n")) 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 err.Error()
} }
return string(result) return string(buf)
} }