golang之udpConn

golang进行UDP client/server通讯的过程中发现Read/ReadFromUDP/Write/WriteToUDP的使用,还有connection/unconnection较乱。

代码实验

UDP server

UDP服务器端在调用”net.ListenUDP()“后创建”net.UDPConn”,read/write操作是通过这个UDPConn来完成的。因为listen的时候只指定了本地绑定的地址,它只能被动的接收来自客户端的消息,因此这个UDPConn在golang中为’unconnected’类型的。

这种类型的UDPConn的读操作可以接受Read()及ReadFromUDP()。区别是Read()无法知道远程连接的地址信息而ReadFromUDP()可以,所以如果后续需要跟远程进行双向通讯需要使用ReadFromUDP()。

这种类型的UDPConn在进行写操作时必须使用WriteToUDP()完成,并且需要指定对方的地址信息。

package main

import (
        "flag"
        "fmt"
        "net"
        "os"
)

var host = flag.String("host", "", "Specify the hostname")
var port = flag.Int("port", 12345, "Specify the port")

func main() {
        flag.Parse()

        var udpAddr = net.UDPAddr{
                IP:   net.ParseIP(*host),
                Port: *port,
        }

        conn, err := net.ListenUDP("udp", &udpAddr)
        if err != nil {
                fmt.Println("Error listening:", err)
                os.Exit(1)
        }
        defer conn.Close()

        var data = make([]byte, 1024)
        for {
                n, remoteAddr, err := conn.ReadFromUDP(data)      // conn.Read() also works here
                if err != nil {
                        fmt.Println("Error read UDP:", err.Error())
                }

                if n <= 0 {
                        continue
                }

                fmt.Printf("[%v]:", remoteAddr)
                fmt.Println(data[:n])

                _, err = conn.WriteToUDP(data, remoteAddr)
                //_, err = conn.Write(data)                      // conn.Write() doesn't work here
                if err != nil {
                        fmt.Println("Error write UDP:", err.Error())
                }
        }
}

UDP client

UDP客户端的连接通过”net.DialUDP()“来创建。因为这个连接创建时指定了远程服务器地址,因此这种连接在golang中称为’connected’类型的连接。

这种类型的连接进行读操作可以使用Read()及ReadFromUDP(),区别在前面已经说过了,主要区别就是ReadFromUDP()会返回远程端的地址信息。

这种类型的连接进行写操作只能用Write(),如果使用WriteToUDP()则无法完成数据发送。

package main

import (
        "bufio"
        "flag"
        "fmt"
        "net"
        "os"
)

var host = flag.String("host", "127.0.0.1", "Specify the target host IP")
var port = flag.Int("port", 12345, "Specifiy the target host port")

func main() {
        flag.Parse()

        var targetAddr = net.UDPAddr{
                IP:   net.ParseIP(*host),
                Port: *port,
        }

        conn, err := net.DialUDP("udp", nil, &targetAddr)
        if err != nil {
                fmt.Println("Error DialUDP,", err.Error())
                os.Exit(1)
        }
        defer conn.Close()

        reader := bufio.NewReader(os.Stdin)
        for {
                fmt.Print("> ")
                data, _, err := reader.ReadLine()
                if err != nil {
                        continue
                }

                if string(data) == "quit" {
                        fmt.Println("Quit now ...")
                        os.Exit(0)
                }

                fmt.Println("Writing: ", data)
                //conn.WriteToUDP(data, &targetAddr)      // conn.WriteToUDP() doesn't work here
                n, err := conn.Write(data)
                fmt.Printf("Written %d bytes", n)
                if err != nil {
                        fmt.Println("Error WriteToUDP: ", err.Error())
                }

                //n, _, err = conn.ReadFromUDP(data)     // conn.ReadFromUDP() works here
                n, err = conn.Read(data)
                if err != nil {
                        fmt.Println("Error ReadFromUDP: ", err.Error())
                        continue
                }
                fmt.Println(string(data))
        }
}

connected vs unconnected

  • unconnected:不知道远程地址的socket;server调用listenUDP()创建的UDPConn为unconnected状态
  • connected:知道远程地址的socket;client调用DialUDP()创建的UDPConn为connectd状态

参考1