blob: 3c42ea7ad85a505f7c00646001866192c76ef5c8 [file] [log] [blame]
David K. Bainbridge215e0242017-09-05 23:18:24 -07001// 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 linux netbsd
6
7package socket
8
9import "net"
10
11type mmsghdrs []mmsghdr
12
13func (hs mmsghdrs) pack(ms []Message, parseFn func([]byte, string) (net.Addr, error), marshalFn func(net.Addr) []byte) error {
14 for i := range hs {
15 vs := make([]iovec, len(ms[i].Buffers))
16 var sa []byte
17 if parseFn != nil {
18 sa = make([]byte, sizeofSockaddrInet6)
19 }
20 if marshalFn != nil {
21 sa = marshalFn(ms[i].Addr)
22 }
23 hs[i].Hdr.pack(vs, ms[i].Buffers, ms[i].OOB, sa)
24 }
25 return nil
26}
27
28func (hs mmsghdrs) unpack(ms []Message, parseFn func([]byte, string) (net.Addr, error), hint string) error {
29 for i := range hs {
30 ms[i].N = int(hs[i].Len)
31 ms[i].NN = hs[i].Hdr.controllen()
32 ms[i].Flags = hs[i].Hdr.flags()
33 if parseFn != nil {
34 var err error
35 ms[i].Addr, err = parseFn(hs[i].Hdr.name(), hint)
36 if err != nil {
37 return err
38 }
39 }
40 }
41 return nil
42}