2018-07-11 17:27:47 +02:00
|
|
|
package mock
|
2016-05-17 13:13:37 +02:00
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
2016-06-06 05:43:41 +02:00
|
|
|
"io"
|
2017-05-21 16:42:42 +02:00
|
|
|
"net"
|
|
|
|
"os"
|
|
|
|
"syscall"
|
2016-05-17 13:13:37 +02:00
|
|
|
|
2017-05-21 16:42:42 +02:00
|
|
|
frpNet "github.com/fatedier/frp/utils/net"
|
2016-05-17 13:13:37 +02:00
|
|
|
)
|
|
|
|
|
2018-12-09 14:56:46 +01:00
|
|
|
type EchoServer struct {
|
2019-10-12 14:13:12 +02:00
|
|
|
l net.Listener
|
2016-05-17 13:13:37 +02:00
|
|
|
|
2018-12-09 14:56:46 +01:00
|
|
|
port int
|
|
|
|
repeatedNum int
|
|
|
|
specifyStr string
|
|
|
|
}
|
2016-05-17 13:13:37 +02:00
|
|
|
|
2018-12-09 14:56:46 +01:00
|
|
|
func NewEchoServer(port int, repeatedNum int, specifyStr string) *EchoServer {
|
|
|
|
if repeatedNum <= 0 {
|
|
|
|
repeatedNum = 1
|
|
|
|
}
|
|
|
|
return &EchoServer{
|
|
|
|
port: port,
|
|
|
|
repeatedNum: repeatedNum,
|
|
|
|
specifyStr: specifyStr,
|
2016-05-17 13:13:37 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-12-09 14:56:46 +01:00
|
|
|
func (es *EchoServer) Start() error {
|
2019-10-12 14:13:12 +02:00
|
|
|
l, err := net.Listen("tcp", fmt.Sprintf("127.0.0.1:%d", es.port))
|
2018-05-23 08:39:12 +02:00
|
|
|
if err != nil {
|
2018-12-09 14:56:46 +01:00
|
|
|
fmt.Printf("echo server listen error: %v\n", err)
|
|
|
|
return err
|
2018-05-23 08:39:12 +02:00
|
|
|
}
|
2018-12-09 14:56:46 +01:00
|
|
|
es.l = l
|
2018-05-23 08:39:12 +02:00
|
|
|
|
2018-12-09 14:56:46 +01:00
|
|
|
go func() {
|
|
|
|
for {
|
|
|
|
c, err := l.Accept()
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
go echoWorker(c, es.repeatedNum, es.specifyStr)
|
2018-05-23 08:39:12 +02:00
|
|
|
}
|
2018-12-09 14:56:46 +01:00
|
|
|
}()
|
|
|
|
return nil
|
|
|
|
}
|
2018-05-23 08:39:12 +02:00
|
|
|
|
2018-12-09 14:56:46 +01:00
|
|
|
func (es *EchoServer) Stop() {
|
|
|
|
es.l.Close()
|
2018-05-23 08:39:12 +02:00
|
|
|
}
|
|
|
|
|
2018-07-11 17:27:47 +02:00
|
|
|
func StartUdpEchoServer(port int) {
|
|
|
|
l, err := frpNet.ListenUDP("127.0.0.1", port)
|
2017-03-27 18:27:30 +02:00
|
|
|
if err != nil {
|
|
|
|
fmt.Printf("udp echo server listen error: %v\n", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
for {
|
|
|
|
c, err := l.Accept()
|
|
|
|
if err != nil {
|
|
|
|
fmt.Printf("udp echo server accept error: %v\n", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2018-12-09 14:56:46 +01:00
|
|
|
go echoWorker(c, 1, "")
|
2017-03-27 18:27:30 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-11 17:27:47 +02:00
|
|
|
func StartUnixDomainServer(unixPath string) {
|
2017-05-21 16:42:42 +02:00
|
|
|
os.Remove(unixPath)
|
|
|
|
syscall.Umask(0)
|
|
|
|
l, err := net.Listen("unix", unixPath)
|
|
|
|
if err != nil {
|
|
|
|
fmt.Printf("unix domain server listen error: %v\n", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
for {
|
|
|
|
c, err := l.Accept()
|
|
|
|
if err != nil {
|
|
|
|
fmt.Printf("unix domain server accept error: %v\n", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2018-12-09 14:56:46 +01:00
|
|
|
go echoWorker(c, 1, "")
|
2017-05-21 16:42:42 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-12-09 14:56:46 +01:00
|
|
|
func echoWorker(c net.Conn, repeatedNum int, specifyStr string) {
|
2017-12-16 16:59:46 +01:00
|
|
|
buf := make([]byte, 2048)
|
|
|
|
|
2016-05-17 13:13:37 +02:00
|
|
|
for {
|
2017-12-16 16:59:46 +01:00
|
|
|
n, err := c.Read(buf)
|
2016-05-17 13:13:37 +02:00
|
|
|
if err != nil {
|
2017-12-16 16:59:46 +01:00
|
|
|
if err == io.EOF {
|
|
|
|
c.Close()
|
|
|
|
break
|
|
|
|
} else {
|
|
|
|
fmt.Printf("echo server read error: %v\n", err)
|
|
|
|
return
|
|
|
|
}
|
2016-05-17 13:13:37 +02:00
|
|
|
}
|
|
|
|
|
2018-12-09 14:56:46 +01:00
|
|
|
if specifyStr != "" {
|
|
|
|
c.Write([]byte(specifyStr))
|
|
|
|
} else {
|
|
|
|
var w []byte
|
|
|
|
for i := 0; i < repeatedNum; i++ {
|
|
|
|
w = append(w, buf[:n]...)
|
2018-05-23 08:39:12 +02:00
|
|
|
}
|
2018-12-09 14:56:46 +01:00
|
|
|
c.Write(w)
|
2018-05-23 08:39:12 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|