/
info.go
75 lines (67 loc) · 2.12 KB
/
info.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
// Copyright 2020 Anapaya Systems
//
// 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 seg
import (
"fmt"
"math"
"time"
"google.golang.org/protobuf/proto"
"github.com/scionproto/scion/go/lib/serrors"
cppb "github.com/scionproto/scion/go/pkg/proto/control_plane"
)
// Info represents the path segment information.
type Info struct {
// Raw contains the encoded path segment information. It is used for
// signature input and must not be modified.
Raw []byte
// Timestamp is the path segment creation time.
Timestamp time.Time
// SegmentID is the segment ID used in data plane hop field computation.
SegmentID uint16
}
// infoFromRaw decodes the protobuf representation. The byte slice is captured
// and must not be modifed.
func infoFromRaw(raw []byte) (Info, error) {
var pb cppb.SegmentInformation
if err := proto.Unmarshal(raw, &pb); err != nil {
return Info{}, err
}
if pb.SegmentId > math.MaxUint16 {
return Info{}, serrors.New("segment ID overflows uint16", "segment_id", pb.SegmentId)
}
return Info{
Raw: raw,
SegmentID: uint16(pb.SegmentId),
Timestamp: time.Unix(pb.Timestamp, 0),
}, nil
}
// NewInfo creates a new path segment info.
func NewInfo(timestamp time.Time, segmentID uint16) (Info, error) {
info := &cppb.SegmentInformation{
Timestamp: timestamp.Unix(),
SegmentId: uint32(segmentID),
}
raw, err := proto.Marshal(info)
if err != nil {
return Info{}, err
}
return Info{
Raw: raw,
Timestamp: time.Unix(info.Timestamp, 0),
SegmentID: segmentID,
}, nil
}
func (info Info) String() string {
return fmt.Sprintf("Timestamp: %s SegmentID %x", info.Timestamp, info.SegmentID)
}