mirror of
https://github.com/slackhq/nebula.git
synced 2025-11-22 08:24:25 +01:00
71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
package packet
|
|
|
|
import (
|
|
"github.com/slackhq/nebula/util/virtio"
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
type OutPacket struct {
|
|
Segments [][]byte
|
|
SegmentPayloads [][]byte
|
|
SegmentHeaders [][]byte
|
|
SegmentIDs []uint16
|
|
//todo virtio header?
|
|
SegSize int
|
|
SegCounter int
|
|
Valid bool
|
|
wasSegmented bool
|
|
|
|
Scratch []byte
|
|
}
|
|
|
|
func NewOut() *OutPacket {
|
|
out := new(OutPacket)
|
|
out.Segments = make([][]byte, 0, 64)
|
|
out.SegmentHeaders = make([][]byte, 0, 64)
|
|
out.SegmentPayloads = make([][]byte, 0, 64)
|
|
out.SegmentIDs = make([]uint16, 0, 64)
|
|
out.Scratch = make([]byte, Size)
|
|
return out
|
|
}
|
|
|
|
func (pkt *OutPacket) Reset() {
|
|
pkt.Segments = pkt.Segments[:0]
|
|
pkt.SegmentPayloads = pkt.SegmentPayloads[:0]
|
|
pkt.SegmentHeaders = pkt.SegmentHeaders[:0]
|
|
pkt.SegmentIDs = pkt.SegmentIDs[:0]
|
|
pkt.SegSize = 0
|
|
pkt.Valid = false
|
|
pkt.wasSegmented = false
|
|
}
|
|
|
|
func (pkt *OutPacket) UseSegment(segID uint16, seg []byte, isV6 bool) int {
|
|
pkt.Valid = true
|
|
pkt.SegmentIDs = append(pkt.SegmentIDs, segID)
|
|
pkt.Segments = append(pkt.Segments, seg) //todo do we need this?
|
|
|
|
vhdr := virtio.NetHdr{ //todo
|
|
Flags: unix.VIRTIO_NET_HDR_F_DATA_VALID,
|
|
GSOType: unix.VIRTIO_NET_HDR_GSO_NONE,
|
|
HdrLen: 0,
|
|
GSOSize: 0,
|
|
CsumStart: 0,
|
|
CsumOffset: 0,
|
|
NumBuffers: 0,
|
|
}
|
|
|
|
hdr := seg[0 : virtio.NetHdrSize+14]
|
|
_ = vhdr.Encode(hdr)
|
|
if isV6 {
|
|
hdr[virtio.NetHdrSize+14-2] = 0x86
|
|
hdr[virtio.NetHdrSize+14-1] = 0xdd
|
|
} else {
|
|
hdr[virtio.NetHdrSize+14-2] = 0x08
|
|
hdr[virtio.NetHdrSize+14-1] = 0x00
|
|
}
|
|
|
|
pkt.SegmentHeaders = append(pkt.SegmentHeaders, hdr)
|
|
pkt.SegmentPayloads = append(pkt.SegmentPayloads, seg[virtio.NetHdrSize+14:])
|
|
return len(pkt.SegmentIDs) - 1
|
|
}
|