mirror of
https://github.com/fatedier/frp.git
synced 2024-12-13 10:11:04 +01:00
41 lines
952 B
Go
41 lines
952 B
Go
package request
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"time"
|
|
)
|
|
|
|
func SendTCPRequest(port int, content []byte, timeout time.Duration) (string, error) {
|
|
c, err := net.Dial("tcp", fmt.Sprintf("127.0.0.1:%d", port))
|
|
if err != nil {
|
|
return "", fmt.Errorf("connect to tcp server error: %v", err)
|
|
}
|
|
defer c.Close()
|
|
|
|
c.SetDeadline(time.Now().Add(timeout))
|
|
return sendRequestByConn(c, content)
|
|
}
|
|
|
|
func SendUDPRequest(port int, content []byte, timeout time.Duration) (string, error) {
|
|
c, err := net.Dial("udp", fmt.Sprintf("127.0.0.1:%d", port))
|
|
if err != nil {
|
|
return "", fmt.Errorf("connect to udp server error: %v", err)
|
|
}
|
|
defer c.Close()
|
|
|
|
c.SetDeadline(time.Now().Add(timeout))
|
|
return sendRequestByConn(c, content)
|
|
}
|
|
|
|
func sendRequestByConn(c net.Conn, content []byte) (string, error) {
|
|
c.Write(content)
|
|
|
|
buf := make([]byte, 2048)
|
|
n, err := c.Read(buf)
|
|
if err != nil {
|
|
return "", fmt.Errorf("read error: %v", err)
|
|
}
|
|
return string(buf[:n]), nil
|
|
}
|