/
utils.go
94 lines (73 loc) · 2.61 KB
/
utils.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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
/*
Copyright (c) 2021-2022 Nordix Foundation
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package kernel
import (
"context"
"fmt"
"github.com/nordix/meridio/pkg/networking"
"github.com/vishvananda/netlink"
)
type contextKeyType string
const interfaceMonitorKey contextKeyType = "interfaceMonitor"
type KernelUtils struct {
}
func (ku *KernelUtils) NewInterface(index int) networking.Iface {
return NewInterface(index)
}
func (ku *KernelUtils) NewBridge(name string) (networking.Bridge, error) {
return NewBridge(name)
}
func (ku *KernelUtils) NewFWMarkRoute(ip string, fwmark int, tableID int) (networking.FWMarkRoute, error) {
return NewFWMarkRoute(ip, fwmark, tableID)
}
func (ku *KernelUtils) NewSourceBasedRoute(tableID int, prefix string) (networking.SourceBasedRoute, error) {
return NewSourceBasedRoute(tableID, prefix)
}
func (ku *KernelUtils) NewInterfaceMonitor() (networking.InterfaceMonitor, error) {
return NewInterfaceMonitor()
}
// WithInterfaceMonitor -
// Stores InterfaceMonitor in Context
func (ku *KernelUtils) WithInterfaceMonitor(parent context.Context, monitor networking.InterfaceMonitor) context.Context {
if parent == nil {
parent = context.Background()
}
return context.WithValue(parent, interfaceMonitorKey, monitor)
}
// GetInterfaceMonitor -
// Returns InterfaceMonitor from Context
func (ku *KernelUtils) GetInterfaceMonitor(ctx context.Context) networking.InterfaceMonitor {
rv, ok := ctx.Value(interfaceMonitorKey).(networking.InterfaceMonitor)
if ok && rv != nil {
return rv
}
return nil
}
func (ku *KernelUtils) GetIndexFromName(name string) (int, error) {
return GetIndexFromName(name)
}
func getLink(intf networking.Iface) (netlink.Link, error) {
link, err := netlink.LinkByIndex(intf.GetIndex())
if err != nil {
return link, fmt.Errorf("failed getting link by index (%d): %w", intf.GetIndex(), err)
}
return link, nil
}
// GetIndexFromName - Get the index of an interface from its name
func GetIndexFromName(name string) (int, error) {
intf, err := netlink.LinkByName(name)
if err != nil {
return -1, fmt.Errorf("failed getting link by name (%s): %w", name, err)
}
return intf.Attrs().Index, nil
}