-
Notifications
You must be signed in to change notification settings - Fork 160
/
ifstate.go
178 lines (159 loc) · 5.06 KB
/
ifstate.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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
// Copyright 2016 ETH Zurich
//
// 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.
// This file handles generating periodic Interface State Request (IFStateReq)
// packets that are sent to the local Beacon Service (BS), as well as
// processing the Interface State updates. IFStateReqs are mostly needed on
// startup, to make sure the border router is aware of the status of the local
// interfaces. The BS normally updates the border routers everytime an
// interface state changes, so this is only needed as a fail-safe after
// startup.
package ifstate
import (
"sync"
"sync/atomic"
"unsafe"
"github.com/scionproto/scion/go/border/internal/metrics"
"github.com/scionproto/scion/go/border/rctx"
"github.com/scionproto/scion/go/lib/addr"
"github.com/scionproto/scion/go/lib/common"
"github.com/scionproto/scion/go/lib/ctrl/path_mgmt"
"github.com/scionproto/scion/go/lib/log"
"github.com/scionproto/scion/go/proto"
)
// ifStates is a map of interface IDs to interface states.
type ifStates sync.Map
func (s *ifStates) Delete(key common.IFIDType) {
(*sync.Map)(s).Delete(key)
}
func (s *ifStates) Load(key common.IFIDType) (*state, bool) {
val, loaded := (*sync.Map)(s).Load(key)
if val == nil {
return nil, loaded
}
return val.(*state), loaded
}
func (s *ifStates) Store(key common.IFIDType, val *state) {
(*sync.Map)(s).Store(key, val)
}
var states ifStates
type state struct {
// info is a pointer to an Info object. Processing goroutine can update this value.
info unsafe.Pointer
}
// Info stores state information, as well as the raw revocation info for a given interface.
type Info struct {
IfID common.IFIDType
Active bool
SRevInfo *path_mgmt.SignedRevInfo
RawSRev common.RawBytes
}
func NewInfo(ifID common.IFIDType, ia addr.IA, active bool, srev *path_mgmt.SignedRevInfo,
rawSRev common.RawBytes) *Info {
label := metrics.IntfLabels{
Intf: metrics.IntfToLabel(ifID),
NeighIA: ia.String(),
}
i := &Info{
IfID: ifID,
Active: active,
SRevInfo: srev,
RawSRev: rawSRev,
}
var isActive float64
if active {
isActive = 1
}
metrics.Control.IFState(label).Set(isActive)
return i
}
// Process processes Interface State updates from the beacon service.
func Process(ifStates *path_mgmt.IFStateInfos) {
cl := metrics.ControlLabels{Result: metrics.Success}
ctx := rctx.Get()
for _, info := range ifStates.Infos {
var rawSRev common.RawBytes
ifid := common.IFIDType(info.IfID)
if info.SRevInfo != nil {
var err error
rawSRev, err = proto.PackRoot(info.SRevInfo)
if err != nil {
cl.Result = metrics.ErrProcess
metrics.Control.ReceivedIFStateInfo(cl).Inc()
log.Error("Unable to pack SRevInfo", "err", err)
return
}
}
intf, ok := ctx.Conf.Topo.IFInfoMap()[ifid]
if !ok {
log.Warn("Interface ID does not exist", "ifid", ifid)
continue
}
stateInfo := NewInfo(ifid, intf.IA, info.Active, info.SRevInfo, rawSRev)
s, ok := states.Load(ifid)
if !ok {
log.Info("IFState: intf added", "ifid", ifid, "active", info.Active)
s = &state{info: unsafe.Pointer(stateInfo)}
states.Store(ifid, s)
continue
}
oldInfo := (*Info)(atomic.LoadPointer(&s.info))
if stateInfo.Active {
if !oldInfo.Active {
log.Info("IFState: intf activated", "ifid", ifid)
}
} else {
if oldInfo.Active {
log.Info("IFState: intf deactivated", "ifid", ifid)
}
}
atomic.StorePointer(&s.info, unsafe.Pointer(stateInfo))
}
metrics.Control.ReceivedIFStateInfo(cl).Inc()
}
// LoadState returns the state info for a given interface ID or nil.
// The bool result indicates whether the state was found in the map.
func LoadState(ifID common.IFIDType) (*Info, bool) {
s, ok := states.Load(ifID)
if !ok {
return nil, ok
}
return (*Info)(atomic.LoadPointer(&s.info)), ok
}
// LoadStates returns info for all interfaces.
func LoadStates() []*Info {
var res []*Info
(*sync.Map)(&states).Range(func(_, s interface{}) bool {
info := (*Info)(atomic.LoadPointer(&s.(*state).info))
res = append(res, info)
return true
})
return res
}
// UpdateIfNew atomically updates the state info for a given ifid, if the state has
// not been changed in the meantime. If there is no state info, a new one will be created
// and the new state will be inserted.
func UpdateIfNew(ifID common.IFIDType, old, new *Info) {
s, ok := states.Load(ifID)
if ok {
atomic.CompareAndSwapPointer(&s.info, unsafe.Pointer(old), unsafe.Pointer(new))
return
}
s = &state{}
atomic.StorePointer(&s.info, unsafe.Pointer(new))
states.Store(ifID, s)
}
// DeleteState removes the state info for a given interface.
func DeleteState(ifID common.IFIDType) {
states.Delete(ifID)
}