mirror of
https://github.com/KusakabeShi/EtherGuard-VPN.git
synced 2025-02-16 23:30:47 +01:00
76 lines
1.4 KiB
Go
76 lines
1.4 KiB
Go
package path
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"errors"
|
|
|
|
"github.com/KusakabeSi/EtherGuardVPN/config"
|
|
)
|
|
|
|
const EgHeaderLen = 8
|
|
|
|
type EgHeader struct {
|
|
buf []byte
|
|
}
|
|
|
|
type Usage uint8
|
|
|
|
const (
|
|
NornalPacket Usage = iota
|
|
Register //Register to server
|
|
|
|
UpdatePeer //Comes from server
|
|
UpdateNhTable
|
|
UpdateError
|
|
|
|
PingPacket //Comes from other peer
|
|
PongPacket //Send to everyone, include server
|
|
QueryPeer
|
|
BoardcastPeer
|
|
)
|
|
|
|
func NewEgHeader(pac []byte) (e EgHeader, err error) {
|
|
if len(pac) != EgHeaderLen {
|
|
err = errors.New("Invalid packet size")
|
|
return
|
|
}
|
|
e.buf = pac
|
|
return
|
|
}
|
|
|
|
func (e EgHeader) GetDst() config.Vertex {
|
|
return config.Vertex(binary.BigEndian.Uint16(e.buf[0:2]))
|
|
}
|
|
func (e EgHeader) SetDst(node_ID config.Vertex) {
|
|
binary.BigEndian.PutUint16(e.buf[0:2], uint16(node_ID))
|
|
}
|
|
|
|
func (e EgHeader) GetSrc() config.Vertex {
|
|
return config.Vertex(binary.BigEndian.Uint16(e.buf[2:4]))
|
|
}
|
|
func (e EgHeader) SetSrc(node_ID config.Vertex) {
|
|
binary.BigEndian.PutUint16(e.buf[2:4], uint16(node_ID))
|
|
}
|
|
|
|
func (e EgHeader) GetTTL() uint8 {
|
|
return e.buf[4]
|
|
}
|
|
func (e EgHeader) SetTTL(ttl uint8) {
|
|
|
|
e.buf[4] = ttl
|
|
}
|
|
|
|
func (e EgHeader) GetUsage() Usage {
|
|
return Usage(e.buf[5])
|
|
}
|
|
func (e EgHeader) SetUsage(usage Usage) {
|
|
e.buf[5] = uint8(usage)
|
|
}
|
|
|
|
func (e EgHeader) GetPacketLength() uint16 {
|
|
return binary.BigEndian.Uint16(e.buf[6:8])
|
|
}
|
|
func (e EgHeader) SetPacketLength(length uint16) {
|
|
binary.BigEndian.PutUint16(e.buf[6:8], length)
|
|
}
|