mirror of
https://github.com/containers/gvisor-tap-vsock.git
synced 2025-10-31 12:06:24 +08:00
80 lines
1.4 KiB
Go
80 lines
1.4 KiB
Go
package tap
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"math"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type protocol interface {
|
|
Stream() bool
|
|
}
|
|
|
|
type streamProtocol interface {
|
|
protocol
|
|
Buf() []byte
|
|
Write(buf []byte, size int)
|
|
Read(buf []byte) int
|
|
}
|
|
|
|
type hyperkitProtocol struct {
|
|
}
|
|
|
|
func (s *hyperkitProtocol) Stream() bool {
|
|
return true
|
|
}
|
|
|
|
func (s *hyperkitProtocol) Buf() []byte {
|
|
return make([]byte, 2)
|
|
}
|
|
|
|
func (s *hyperkitProtocol) Write(buf []byte, size int) {
|
|
if size < 0 || size > math.MaxUint16 {
|
|
log.Warnf("size out of range. Resetting to %d", math.MaxUint16)
|
|
size = math.MaxUint16
|
|
}
|
|
binary.LittleEndian.PutUint16(buf, uint16(size)) //#nosec: G115
|
|
}
|
|
|
|
func (s *hyperkitProtocol) Read(buf []byte) int {
|
|
return int(binary.LittleEndian.Uint16(buf[0:2]))
|
|
}
|
|
|
|
type qemuProtocol struct {
|
|
}
|
|
|
|
func (s *qemuProtocol) Stream() bool {
|
|
return true
|
|
}
|
|
|
|
func (s *qemuProtocol) Buf() []byte {
|
|
return make([]byte, 4)
|
|
}
|
|
|
|
func (s *qemuProtocol) Write(buf []byte, size int) {
|
|
if size > math.MaxUint32 {
|
|
log.Warnf("size exceeds max limit. Resetting to: %d", math.MaxInt32)
|
|
size = math.MaxUint32
|
|
}
|
|
binary.BigEndian.PutUint32(buf, uint32(size)) //#nosec: G115. Safely checked
|
|
}
|
|
|
|
func (s *qemuProtocol) Read(buf []byte) int {
|
|
return int(binary.BigEndian.Uint32(buf[0:4]))
|
|
}
|
|
|
|
type bessProtocol struct {
|
|
}
|
|
|
|
func (s *bessProtocol) Stream() bool {
|
|
return false
|
|
}
|
|
|
|
type vfkitProtocol struct {
|
|
}
|
|
|
|
func (s *vfkitProtocol) Stream() bool {
|
|
return false
|
|
}
|