-
Notifications
You must be signed in to change notification settings - Fork 12
/
connection_linux.go
56 lines (43 loc) · 984 Bytes
/
connection_linux.go
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
//go:build linux
package clients
import (
"crypto/tls"
"errors"
"log"
"net"
"syscall"
"golang.org/x/sys/unix"
)
const fwmarkIoctl int = 36 /* unix.SO_MARK */
var ErrUnknownConnType = errors.New("not a known conn type")
func setFirewallMark(conn net.Conn, mark int) error {
var err error
var syscallConn syscall.Conn
switch typedConn := conn.(type) {
case syscall.Conn:
syscallConn = typedConn
case *tls.Conn:
return setFirewallMark(typedConn.NetConn(), mark)
default:
log.Printf("Unknown conn type: %T (%v)", typedConn, typedConn)
err = ErrUnknownConnType
}
if err != nil {
return err
}
fd, err := syscallConn.SyscallConn()
if err != nil {
return err
}
return setFirewallMarkRaw(fd, mark)
}
func setFirewallMarkRaw(fd syscall.RawConn, mark int) error {
var operr error
err := fd.Control(func(fd uintptr) {
operr = unix.SetsockoptInt(int(fd), unix.SOL_SOCKET, fwmarkIoctl, int(mark))
})
if err == nil {
return operr
}
return err
}