vendor/golang.org/x/net/ipv6/payload_cmsg_go1_9.go

   1 // Copyright 2017 The Go Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style
   3 // license that can be found in the LICENSE file.
   4
   5 // +build go1.9
   6 // +build !nacl,!plan9,!windows
   7
   8 package ipv6
   9
  10 import (
  11         "net"
  12
  13         "golang.org/x/net/internal/socket"
  14 )
  15
  16 func (c *payloadHandler) readFrom(b []byte) (int, *ControlMessage, net.Addr, error) {
  17         c.rawOpt.RLock()
  18         m := socket.Message{
  19                 Buffers: [][]byte{b},
  20                 OOB:     NewControlMessage(c.rawOpt.cflags),
  21         }
  22         c.rawOpt.RUnlock()
  23         switch c.PacketConn.(type) {
  24         case *net.UDPConn:
  25                 if err := c.RecvMsg(&m, 0); err != nil {
  26                         return 0, nil, nil, &net.OpError{Op: "read", Net: c.PacketConn.LocalAddr().Network(), Source: c.PacketConn.LocalAddr(), Err: err}
  27                 }
  28         case *net.IPConn:
  29                 if err := c.RecvMsg(&m, 0); err != nil {
  30                         return 0, nil, nil, &net.OpError{Op: "read", Net: c.PacketConn.LocalAddr().Network(), Source: c.PacketConn.LocalAddr(), Err: err}
  31                 }
  32         default:
  33                 return 0, nil, nil, &net.OpError{Op: "read", Net: c.PacketConn.LocalAddr().Network(), Source: c.PacketConn.LocalAddr(), Err: errInvalidConnType}
  34         }
  35         var cm *ControlMessage
  36         if m.NN > 0 {
  37                 cm = new(ControlMessage)
  38                 if err := cm.Parse(m.OOB[:m.NN]); err != nil {
  39                         return 0, nil, nil, &net.OpError{Op: "read", Net: c.PacketConn.LocalAddr().Network(), Source: c.PacketConn.LocalAddr(), Err: err}
  40                 }
  41                 cm.Src = netAddrToIP16(m.Addr)
  42         }
  43         return m.N, cm, m.Addr, nil
  44 }
  45
  46 func (c *payloadHandler) writeTo(b []byte, cm *ControlMessage, dst net.Addr) (int, error) {
  47         m := socket.Message{
  48                 Buffers: [][]byte{b},
  49                 OOB:     cm.Marshal(),
  50                 Addr:    dst,
  51         }
  52         err := c.SendMsg(&m, 0)
  53         if err != nil {
  54                 err = &net.OpError{Op: "write", Net: c.PacketConn.LocalAddr().Network(), Source: c.PacketConn.LocalAddr(), Addr: opAddr(dst), Err: err}
  55         }
  56         return m.N, err
  57 }