mirror of https://github.com/vishvananda/netlink
68 lines
1.5 KiB
Go
68 lines
1.5 KiB
Go
package nl
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/rand"
|
|
"encoding/binary"
|
|
"testing"
|
|
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
func (msg *RtMsg) write(b []byte) {
|
|
native := NativeEndian()
|
|
b[0] = msg.Family
|
|
b[1] = msg.Dst_len
|
|
b[2] = msg.Src_len
|
|
b[3] = msg.Tos
|
|
b[4] = msg.Table
|
|
b[5] = msg.Protocol
|
|
b[6] = msg.Scope
|
|
b[7] = msg.Type
|
|
native.PutUint32(b[8:12], msg.Flags)
|
|
}
|
|
|
|
func (msg *RtMsg) serializeSafe() []byte {
|
|
len := unix.SizeofRtMsg
|
|
b := make([]byte, len)
|
|
msg.write(b)
|
|
return b
|
|
}
|
|
|
|
func deserializeRtMsgSafe(b []byte) *RtMsg {
|
|
var msg = RtMsg{}
|
|
binary.Read(bytes.NewReader(b[0:unix.SizeofRtMsg]), NativeEndian(), &msg)
|
|
return &msg
|
|
}
|
|
|
|
func TestRtMsgDeserializeSerialize(t *testing.T) {
|
|
var orig = make([]byte, unix.SizeofRtMsg)
|
|
rand.Read(orig)
|
|
safemsg := deserializeRtMsgSafe(orig)
|
|
msg := DeserializeRtMsg(orig)
|
|
testDeserializeSerialize(t, orig, safemsg, msg)
|
|
}
|
|
|
|
func TestDeserializeRtNexthop(t *testing.T) {
|
|
buf := make([]byte, unix.SizeofRtNexthop+64)
|
|
native := NativeEndian()
|
|
native.PutUint16(buf[0:2], unix.SizeofRtNexthop)
|
|
buf[2] = 17
|
|
buf[3] = 1
|
|
native.PutUint32(buf[4:8], 1234)
|
|
|
|
msg := DeserializeRtNexthop(buf)
|
|
safemsg := &RtNexthop{
|
|
unix.RtNexthop{
|
|
Len: unix.SizeofRtNexthop,
|
|
Flags: 17,
|
|
Hops: 1,
|
|
Ifindex: 1234,
|
|
},
|
|
nil,
|
|
}
|
|
if msg.Len() != safemsg.Len() || msg.Flags != safemsg.Flags || msg.Hops != safemsg.Hops || msg.Ifindex != safemsg.Ifindex {
|
|
t.Fatal("Deserialization failed.\nIn:", buf, "\nOut:", msg, "\n", msg.Serialize(), "\nExpected:", safemsg, "\n", safemsg.Serialize())
|
|
}
|
|
}
|