mirror of
https://github.com/Luzifer/nginx-sso.git
synced 2024-12-21 05:11:17 +00:00
66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
|
// Copyright 2016 The Go Authors. All rights reserved.
|
||
|
|
// Use of this source code is governed by a BSD-style
|
||
|
|
// license that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
package route
|
||
|
|
|
||
|
|
import "syscall"
|
||
|
|
|
||
|
|
func (m *RouteMessage) marshal() ([]byte, error) {
|
||
|
|
l := sizeofRtMsghdr + addrsSpace(m.Addrs)
|
||
|
|
b := make([]byte, l)
|
||
|
|
nativeEndian.PutUint16(b[:2], uint16(l))
|
||
|
|
if m.Version == 0 {
|
||
|
|
b[2] = sysRTM_VERSION
|
||
|
|
} else {
|
||
|
|
b[2] = byte(m.Version)
|
||
|
|
}
|
||
|
|
b[3] = byte(m.Type)
|
||
|
|
nativeEndian.PutUint16(b[4:6], uint16(sizeofRtMsghdr))
|
||
|
|
nativeEndian.PutUint32(b[16:20], uint32(m.Flags))
|
||
|
|
nativeEndian.PutUint16(b[6:8], uint16(m.Index))
|
||
|
|
nativeEndian.PutUint32(b[24:28], uint32(m.ID))
|
||
|
|
nativeEndian.PutUint32(b[28:32], uint32(m.Seq))
|
||
|
|
attrs, err := marshalAddrs(b[sizeofRtMsghdr:], m.Addrs)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if attrs > 0 {
|
||
|
|
nativeEndian.PutUint32(b[12:16], uint32(attrs))
|
||
|
|
}
|
||
|
|
return b, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (*wireFormat) parseRouteMessage(_ RIBType, b []byte) (Message, error) {
|
||
|
|
if len(b) < sizeofRtMsghdr {
|
||
|
|
return nil, errMessageTooShort
|
||
|
|
}
|
||
|
|
l := int(nativeEndian.Uint16(b[:2]))
|
||
|
|
if len(b) < l {
|
||
|
|
return nil, errInvalidMessage
|
||
|
|
}
|
||
|
|
m := &RouteMessage{
|
||
|
|
Version: int(b[2]),
|
||
|
|
Type: int(b[3]),
|
||
|
|
Flags: int(nativeEndian.Uint32(b[16:20])),
|
||
|
|
Index: int(nativeEndian.Uint16(b[6:8])),
|
||
|
|
ID: uintptr(nativeEndian.Uint32(b[24:28])),
|
||
|
|
Seq: int(nativeEndian.Uint32(b[28:32])),
|
||
|
|
raw: b[:l],
|
||
|
|
}
|
||
|
|
ll := int(nativeEndian.Uint16(b[4:6]))
|
||
|
|
if len(b) < ll {
|
||
|
|
return nil, errInvalidMessage
|
||
|
|
}
|
||
|
|
errno := syscall.Errno(nativeEndian.Uint32(b[32:36]))
|
||
|
|
if errno != 0 {
|
||
|
|
m.Err = errno
|
||
|
|
}
|
||
|
|
as, err := parseAddrs(uint(nativeEndian.Uint32(b[12:16])), parseKernelInetAddr, b[ll:])
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
m.Addrs = as
|
||
|
|
return m, nil
|
||
|
|
}
|