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
// Copyright 2013 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 aix darwin dragonfly freebsd linux netbsd openbsd solaris
 
package ipv6
 
import "golang.org/x/net/internal/socket"
 
func setControlMessage(c *socket.Conn, opt *rawOpt, cf ControlFlags, on bool) error {
    opt.Lock()
    defer opt.Unlock()
    if so, ok := sockOpts[ssoReceiveTrafficClass]; ok && cf&FlagTrafficClass != 0 {
        if err := so.SetInt(c, boolint(on)); err != nil {
            return err
        }
        if on {
            opt.set(FlagTrafficClass)
        } else {
            opt.clear(FlagTrafficClass)
        }
    }
    if so, ok := sockOpts[ssoReceiveHopLimit]; ok && cf&FlagHopLimit != 0 {
        if err := so.SetInt(c, boolint(on)); err != nil {
            return err
        }
        if on {
            opt.set(FlagHopLimit)
        } else {
            opt.clear(FlagHopLimit)
        }
    }
    if so, ok := sockOpts[ssoReceivePacketInfo]; ok && cf&flagPacketInfo != 0 {
        if err := so.SetInt(c, boolint(on)); err != nil {
            return err
        }
        if on {
            opt.set(cf & flagPacketInfo)
        } else {
            opt.clear(cf & flagPacketInfo)
        }
    }
    if so, ok := sockOpts[ssoReceivePathMTU]; ok && cf&FlagPathMTU != 0 {
        if err := so.SetInt(c, boolint(on)); err != nil {
            return err
        }
        if on {
            opt.set(FlagPathMTU)
        } else {
            opt.clear(FlagPathMTU)
        }
    }
    return nil
}