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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
// 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.
 
package ipv4
 
import (
    "golang.org/x/net/internal/iana"
    "golang.org/x/net/internal/socket"
)
 
const (
    // See ws2tcpip.h.
    sysIP_OPTIONS                = 0x1
    sysIP_HDRINCL                = 0x2
    sysIP_TOS                    = 0x3
    sysIP_TTL                    = 0x4
    sysIP_MULTICAST_IF           = 0x9
    sysIP_MULTICAST_TTL          = 0xa
    sysIP_MULTICAST_LOOP         = 0xb
    sysIP_ADD_MEMBERSHIP         = 0xc
    sysIP_DROP_MEMBERSHIP        = 0xd
    sysIP_DONTFRAGMENT           = 0xe
    sysIP_ADD_SOURCE_MEMBERSHIP  = 0xf
    sysIP_DROP_SOURCE_MEMBERSHIP = 0x10
    sysIP_PKTINFO                = 0x13
 
    sizeofInetPktinfo  = 0x8
    sizeofIPMreq       = 0x8
    sizeofIPMreqSource = 0xc
)
 
type inetPktinfo struct {
    Addr    [4]byte
    Ifindex int32
}
 
type ipMreq struct {
    Multiaddr [4]byte
    Interface [4]byte
}
 
type ipMreqSource struct {
    Multiaddr  [4]byte
    Sourceaddr [4]byte
    Interface  [4]byte
}
 
// See http://msdn.microsoft.com/en-us/library/windows/desktop/ms738586(v=vs.85).aspx
var (
    ctlOpts = [ctlMax]ctlOpt{}
 
    sockOpts = map[int]*sockOpt{
        ssoTOS:                {Option: socket.Option{Level: iana.ProtocolIP, Name: sysIP_TOS, Len: 4}},
        ssoTTL:                {Option: socket.Option{Level: iana.ProtocolIP, Name: sysIP_TTL, Len: 4}},
        ssoMulticastTTL:       {Option: socket.Option{Level: iana.ProtocolIP, Name: sysIP_MULTICAST_TTL, Len: 4}},
        ssoMulticastInterface: {Option: socket.Option{Level: iana.ProtocolIP, Name: sysIP_MULTICAST_IF, Len: 4}},
        ssoMulticastLoopback:  {Option: socket.Option{Level: iana.ProtocolIP, Name: sysIP_MULTICAST_LOOP, Len: 4}},
        ssoHeaderPrepend:      {Option: socket.Option{Level: iana.ProtocolIP, Name: sysIP_HDRINCL, Len: 4}},
        ssoJoinGroup:          {Option: socket.Option{Level: iana.ProtocolIP, Name: sysIP_ADD_MEMBERSHIP, Len: sizeofIPMreq}, typ: ssoTypeIPMreq},
        ssoLeaveGroup:         {Option: socket.Option{Level: iana.ProtocolIP, Name: sysIP_DROP_MEMBERSHIP, Len: sizeofIPMreq}, typ: ssoTypeIPMreq},
    }
)
 
func (pi *inetPktinfo) setIfindex(i int) {
    pi.Ifindex = int32(i)
}