]> Cypherpunks.ru repositories - govpn.git/commitdiff
Various trivial code corrections
authorSergey Matveev <stargrave@stargrave.org>
Sun, 10 Aug 2014 16:12:13 +0000 (20:12 +0400)
committerSergey Matveev <stargrave@stargrave.org>
Sun, 10 Aug 2014 16:12:13 +0000 (20:12 +0400)
Signed-off-by: Sergey Matveev <stargrave@stargrave.org>
govpn.go

index 60c8080ffb98739c63ff957d43f9b4b6afbda814..e3241993926f8ab3379d7a8067a9832e2c160024 100644 (file)
--- a/govpn.go
+++ b/govpn.go
@@ -106,13 +106,12 @@ func main() {
        }()
 
        // Network address parsing
-       if (len(*bindAddr) > 1 && len(*remoteAddr) > 1) || (len(*bindAddr) == 0 && len(*remoteAddr) == 0) {
+       if (len(*bindAddr) > 1 && len(*remoteAddr) > 1) ||
+               (len(*bindAddr) == 0 && len(*remoteAddr) == 0) {
                panic("Either -bind or -remote must be specified only")
        }
-
        var conn *net.UDPConn
        var remote *net.UDPAddr
-
        serverMode := false
        bindTo := "0.0.0.0:0"
 
@@ -138,16 +137,16 @@ func main() {
        }
 
        udpSink := make(chan UDPPkt)
-       go func(conn *net.UDPConn, sink chan<- UDPPkt) {
+       go func(conn *net.UDPConn) {
                for {
                        data := make([]byte, *mtu)
                        n, addr, err := conn.ReadFromUDP(data)
                        if err != nil {
                                fmt.Print("B")
                        }
-                       sink <- UDPPkt{addr, data[:n]}
+                       udpSink <- UDPPkt{addr, data[:n]}
                }
-       }(conn, udpSink)
+       }(conn)
 
        // Process packets
        var udpPkt UDPPkt
@@ -238,8 +237,7 @@ func main() {
                        copy(keyAuth[:], buf[:32])
                        dataToSend := buf[S20BS-NonceSize : S20BS+len(ethPkt)]
                        poly1305.Sum(tag, dataToSend, keyAuth)
-                       _, err := conn.WriteTo(append(dataToSend, tag[:]...), peer.addr)
-                       if err != nil {
+                       if _, err := conn.WriteToUDP(append(dataToSend, tag[:]...), peer.addr); err != nil {
                                log.Println("Error sending UDP", err)
                        }
                        if *verbose {