vendor/golang.org/x/net/internal/socket/rawconn_mmsg.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 linux
   7
   8 package socket
   9
  10 import (
  11         "net"
  12         "os"
  13         "syscall"
  14 )
  15
  16 func (c *Conn) recvMsgs(ms []Message, flags int) (int, error) {
  17         hs := make(mmsghdrs, len(ms))
  18         var parseFn func([]byte, string) (net.Addr, error)
  19         if c.network != "tcp" {
  20                 parseFn = parseInetAddr
  21         }
  22         if err := hs.pack(ms, parseFn, nil); err != nil {
  23                 return 0, err
  24         }
  25         var operr error
  26         var n int
  27         fn := func(s uintptr) bool {
  28                 n, operr = recvmmsg(s, hs, flags)
  29                 if operr == syscall.EAGAIN {
  30                         return false
  31                 }
  32                 return true
  33         }
  34         if err := c.c.Read(fn); err != nil {
  35                 return n, err
  36         }
  37         if operr != nil {
  38                 return n, os.NewSyscallError("recvmmsg", operr)
  39         }
  40         if err := hs[:n].unpack(ms[:n], parseFn, c.network); err != nil {
  41                 return n, err
  42         }
  43         return n, nil
  44 }
  45
  46 func (c *Conn) sendMsgs(ms []Message, flags int) (int, error) {
  47         hs := make(mmsghdrs, len(ms))
  48         var marshalFn func(net.Addr) []byte
  49         if c.network != "tcp" {
  50                 marshalFn = marshalInetAddr
  51         }
  52         if err := hs.pack(ms, nil, marshalFn); err != nil {
  53                 return 0, err
  54         }
  55         var operr error
  56         var n int
  57         fn := func(s uintptr) bool {
  58                 n, operr = sendmmsg(s, hs, flags)
  59                 if operr == syscall.EAGAIN {
  60                         return false
  61                 }
  62                 return true
  63         }
  64         if err := c.c.Write(fn); err != nil {
  65                 return n, err
  66         }
  67         if operr != nil {
  68                 return n, os.NewSyscallError("sendmmsg", operr)
  69         }
  70         if err := hs[:n].unpack(ms[:n], nil, ""); err != nil {
  71                 return n, err
  72         }
  73         return n, nil
  74 }