mirror of
https://github.com/netbirdio/netbird.git
synced 2025-02-18 11:11:20 +01:00
36 lines
676 B
Go
36 lines
676 B
Go
|
//go:build !android
|
||
|
|
|
||
|
|
package net
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"syscall"
|
||
|
|
)
|
||
|
|
|
||
|
|
// SetSocketMark sets the SO_MARK option on the given socket connection
|
||
|
|
func SetSocketMark(conn syscall.Conn) error {
|
||
|
|
sysconn, err := conn.SyscallConn()
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("get raw conn: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return SetRawSocketMark(sysconn)
|
||
|
|
}
|
||
|
|
|
||
|
|
func SetRawSocketMark(conn syscall.RawConn) error {
|
||
|
|
var setErr error
|
||
|
|
|
||
|
|
err := conn.Control(func(fd uintptr) {
|
||
|
|
setErr = syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_MARK, NetbirdFwmark)
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("control: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if setErr != nil {
|
||
|
|
return fmt.Errorf("set SO_MARK: %w", setErr)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|