OSDN Git Service

Hulk did something
[bytom/vapor.git] / vendor / golang.org / x / net / ipv4 / payload_cmsg_go1_8.go
diff --git a/vendor/golang.org/x/net/ipv4/payload_cmsg_go1_8.go b/vendor/golang.org/x/net/ipv4/payload_cmsg_go1_8.go
new file mode 100644 (file)
index 0000000..d26ccd9
--- /dev/null
@@ -0,0 +1,59 @@
+// Copyright 2012 The Go Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+// +build !go1.9
+// +build !nacl,!plan9,!windows
+
+package ipv4
+
+import "net"
+
+func (c *payloadHandler) readFrom(b []byte) (n int, cm *ControlMessage, src net.Addr, err error) {
+       c.rawOpt.RLock()
+       oob := NewControlMessage(c.rawOpt.cflags)
+       c.rawOpt.RUnlock()
+       var nn int
+       switch c := c.PacketConn.(type) {
+       case *net.UDPConn:
+               if n, nn, _, src, err = c.ReadMsgUDP(b, oob); err != nil {
+                       return 0, nil, nil, err
+               }
+       case *net.IPConn:
+               nb := make([]byte, maxHeaderLen+len(b))
+               if n, nn, _, src, err = c.ReadMsgIP(nb, oob); err != nil {
+                       return 0, nil, nil, err
+               }
+               hdrlen := int(nb[0]&0x0f) << 2
+               copy(b, nb[hdrlen:])
+               n -= hdrlen
+       default:
+               return 0, nil, nil, &net.OpError{Op: "read", Net: c.LocalAddr().Network(), Source: c.LocalAddr(), Err: errInvalidConnType}
+       }
+       if nn > 0 {
+               cm = new(ControlMessage)
+               if err = cm.Parse(oob[:nn]); err != nil {
+                       return 0, nil, nil, &net.OpError{Op: "read", Net: c.PacketConn.LocalAddr().Network(), Source: c.PacketConn.LocalAddr(), Err: err}
+               }
+       }
+       if cm != nil {
+               cm.Src = netAddrToIP4(src)
+       }
+       return
+}
+
+func (c *payloadHandler) writeTo(b []byte, cm *ControlMessage, dst net.Addr) (n int, err error) {
+       oob := cm.Marshal()
+       if dst == nil {
+               return 0, &net.OpError{Op: "write", Net: c.PacketConn.LocalAddr().Network(), Source: c.PacketConn.LocalAddr(), Err: errMissingAddress}
+       }
+       switch c := c.PacketConn.(type) {
+       case *net.UDPConn:
+               n, _, err = c.WriteMsgUDP(b, oob, dst.(*net.UDPAddr))
+       case *net.IPConn:
+               n, _, err = c.WriteMsgIP(b, oob, dst.(*net.IPAddr))
+       default:
+               return 0, &net.OpError{Op: "write", Net: c.LocalAddr().Network(), Source: c.LocalAddr(), Addr: opAddr(dst), Err: errInvalidConnType}
+       }
+       return
+}