forked from abneptis/GoNetlink
/
message.go
85 lines (76 loc) · 2.24 KB
/
message.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
package rtnetlink
/*
Copyright (c) 2011, Abneptis LLC. All rights reserved.
Original Author: James D. Nurmi <james@abneptis.com>
See LICENSE for details
*/
import (
"bytes"
"errors"
)
import "github.com/vishvananda/go-netlink"
// A Message contains a Header object and a series of attributes.
// It is extracted from the Body of a netlink.Message
type Message struct {
Header Header
Attributes []netlink.Attribute
}
// Create a new rtnetlink.Message based off of a header an list of attributes
// (which may be nil or empty).
func NewMessage(h Header, attrs []netlink.Attribute) *Message {
return &Message{Header: h, Attributes: attrs}
}
// Replace or append the attribute with the AttributeType of
// attr
func (self *Message) SetAttribute(attr netlink.Attribute) {
t := attr.Type
for i := range self.Attributes {
if t == self.Attributes[i].Type {
self.Attributes[i] = attr
return
}
}
self.Attributes = append(self.Attributes, attr)
return
}
// Retrieve (the first) Attribute identified by Type,
// returning an error if not found.
func (self Message) GetAttribute(t netlink.AttributeType) (attr netlink.Attribute, err error) {
for i := range self.Attributes {
if t == self.Attributes[i].Type {
attr = self.Attributes[i]
return
}
}
err = errors.New("Attribute not found")
return
}
// Handles the appropriate calls to marshal the Header and Attribute values,
// and return an appropriately aligned result.
func (self Message) MarshalNetlink() (out []byte, err error) {
hb, err := self.Header.MarshalNetlink()
if err == nil {
var bb []byte
bb, err = netlink.MarshalAttributes(self.Attributes)
if err == nil {
out = bytes.Join([][]byte{hb, bb}, []byte{})
}
}
return
}
// Unmarshals a generic message using the header as a guide.
// An error will be returned if the header cannot unmarshal properly,
// or any attribute in the series failed.
func (self *Message) UnmarshalNetlink(in []byte) (err error) {
if len(in) < self.Header.Len() {
return errors.New("Insufficient data for unmarshal of Header")
}
err = self.Header.UnmarshalNetlink(in[0:self.Header.Len()])
if err == nil {
pos := netlink.Align(self.Header.Len())
if len(in) > pos {
self.Attributes, err = netlink.UnmarshalAttributes(in[pos:])
}
}
return
}