mirrors/SpoofDPI
1
0
Fork 0
mirror of https://github.com/xvzc/SpoofDPI.git synced 2025-01-08 17:54:05 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
xvzc 2022-01-02 21:56:24 +09:00
parent 85e03416c2
commit 13190c34f5
1 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
util/util.go
View File

@ -8,6 +8,22 @@ import (
"github.com/babolivier/go-doh-client" "github.com/babolivier/go-doh-client"
) )
func WriteAndRead(conn net.Conn, message []byte) ([]byte, error){
_, err := conn.Write(message)
if err != nil {
log.Fatal("Error writing to client:", err)
return nil, err
}
// defer conn.(*net.TCPConn).CloseWrite()
buf, err := ReadMessage(conn)
if err != nil {
log.Fatal("failed:", err)
return nil, err
}
return buf, nil
}
func ReadMessage(conn net.Conn)([]byte, error) { func ReadMessage(conn net.Conn)([]byte, error) {
buf := make([]byte, 0, 4096) // big buffer buf := make([]byte, 0, 4096) // big buffer
@ -16,9 +32,9 @@ func ReadMessage(conn net.Conn)([]byte, error) {
n, err := conn.Read(tmp) n, err := conn.Read(tmp)
if err != nil { if err != nil {
if err != io.EOF { if err != io.EOF {
log.Fatal("ReadRequest error:", err) log.Println("Read error:", err)
return nil, err
} }
return nil, err
} }
buf = append(buf, tmp[:n]...) buf = append(buf, tmp[:n]...)