liuxiaolong
2022-06-28 37714b1093c04061e636e5b1d27179652e671c0a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
// Copyright 2014 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 darwin freebsd linux
 
package ipv4
 
import (
    "net"
    "unsafe"
 
    "golang.org/x/net/internal/socket"
)
 
func (so *sockOpt) getIPMreqn(c *socket.Conn) (*net.Interface, error) {
    b := make([]byte, so.Len)
    if _, err := so.Get(c, b); err != nil {
        return nil, err
    }
    mreqn := (*ipMreqn)(unsafe.Pointer(&b[0]))
    if mreqn.Ifindex == 0 {
        return nil, nil
    }
    ifi, err := net.InterfaceByIndex(int(mreqn.Ifindex))
    if err != nil {
        return nil, err
    }
    return ifi, nil
}
 
func (so *sockOpt) setIPMreqn(c *socket.Conn, ifi *net.Interface, grp net.IP) error {
    var mreqn ipMreqn
    if ifi != nil {
        mreqn.Ifindex = int32(ifi.Index)
    }
    if grp != nil {
        mreqn.Multiaddr = [4]byte{grp[0], grp[1], grp[2], grp[3]}
    }
    b := (*[sizeofIPMreqn]byte)(unsafe.Pointer(&mreqn))[:sizeofIPMreqn]
    return so.Set(c, b)
}