mirror of https://github.com/fatedier/frp.git
62 lines
934 B
Go
62 lines
934 B
Go
package tests
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/fatedier/frp/utils/net"
|
|
)
|
|
|
|
func StartEchoServer() {
|
|
l, err := net.ListenTcp("127.0.0.1", 10701)
|
|
if err != nil {
|
|
fmt.Printf("echo server listen error: %v\n", err)
|
|
return
|
|
}
|
|
|
|
for {
|
|
c, err := l.Accept()
|
|
if err != nil {
|
|
fmt.Printf("echo server accept error: %v\n", err)
|
|
return
|
|
}
|
|
|
|
go echoWorker(c)
|
|
}
|
|
}
|
|
|
|
func StartUdpEchoServer() {
|
|
l, err := net.ListenUDP("127.0.0.1", 10703)
|
|
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
|
|
}
|
|
|
|
go echoWorker(c)
|
|
}
|
|
}
|
|
|
|
func echoWorker(c net.Conn) {
|
|
br := bufio.NewReader(c)
|
|
for {
|
|
buf, err := br.ReadString('\n')
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
if err != nil {
|
|
fmt.Printf("echo server read error: %v\n", err)
|
|
return
|
|
}
|
|
|
|
c.Write([]byte(buf + "\n"))
|
|
}
|
|
}
|