Read and write udp
This commit is contained in:
parent
3429fd1909
commit
f357ff2f7f
|
@ -4,6 +4,7 @@ import (
|
||||||
"bufio"
|
"bufio"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net"
|
"net"
|
||||||
|
"os"
|
||||||
"slices"
|
"slices"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
@ -21,20 +22,19 @@ func ctrlToken(myToken []byte) []byte {
|
||||||
return data
|
return data
|
||||||
}
|
}
|
||||||
|
|
||||||
func pumpNetwork(ch chan []byte) {
|
func getConnection() (net.Conn, error) {
|
||||||
packet := make([]byte, maxPacksize)
|
|
||||||
conn, err := net.Dial("udp", "127.0.0.1:8303")
|
conn, err := net.Dial("udp", "127.0.0.1:8303")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Printf("Some error %v", err)
|
fmt.Printf("Some error %v", err)
|
||||||
return
|
|
||||||
}
|
}
|
||||||
|
return conn, err
|
||||||
|
}
|
||||||
|
|
||||||
// myToken := []byte{0xfe, 0xed, 0xba, 0xbe}
|
func readNetwork(ch chan []byte, conn net.Conn) {
|
||||||
myToken := []byte{0x01, 0x02, 0x03, 0x04}
|
packet := make([]byte, maxPacksize)
|
||||||
conn.Write(ctrlToken(myToken))
|
|
||||||
|
|
||||||
for {
|
for {
|
||||||
_, err = bufio.NewReader(conn).Read(packet)
|
_, err := bufio.NewReader(conn).Read(packet)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
ch <- packet
|
ch <- packet
|
||||||
} else {
|
} else {
|
||||||
|
@ -46,25 +46,35 @@ func pumpNetwork(ch chan []byte) {
|
||||||
conn.Close()
|
conn.Close()
|
||||||
}
|
}
|
||||||
|
|
||||||
func onMessage(data []byte) {
|
func onMessage(data []byte, conn net.Conn) {
|
||||||
if data[0] == msgCtrlToken {
|
if data[0] == msgCtrlToken {
|
||||||
serverToken := data[8:12]
|
serverToken := data[8:12]
|
||||||
fmt.Printf("got token %v\n", serverToken)
|
fmt.Printf("got token %v\n", serverToken)
|
||||||
|
conn.Write([]byte{0xff, 0xff, 0xff})
|
||||||
} else {
|
} else {
|
||||||
fmt.Println("unknown message")
|
fmt.Printf("unknown message: %v\n", data)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
ch := make(chan []byte, maxPacksize)
|
ch := make(chan []byte, maxPacksize)
|
||||||
|
|
||||||
go pumpNetwork(ch)
|
conn, err := getConnection()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error connecting %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
go readNetwork(ch, conn)
|
||||||
|
|
||||||
|
myToken := []byte{0x01, 0x02, 0x03, 0x04}
|
||||||
|
conn.Write(ctrlToken(myToken))
|
||||||
|
|
||||||
for {
|
for {
|
||||||
time.Sleep(10_000_000)
|
time.Sleep(10_000_000)
|
||||||
select {
|
select {
|
||||||
case msg := <-ch:
|
case msg := <-ch:
|
||||||
onMessage(msg)
|
onMessage(msg, conn)
|
||||||
default:
|
default:
|
||||||
// do nothing
|
// do nothing
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue