Emulation/nex-go
1
0
Fork 0
mirror of https://github.com/PretendoNetwork/nex-go.git synced 2025-04-02 11:02:14 -04:00
Code Issues Projects Releases Packages Wiki Activity

prudp: allow for custom packet types

Browse source
This commit is contained in:
Jonathan Barrow 2024-03-17 12:12:27 -04:00
parent bfadcf1fc2
commit 050ebb1a88
No known key found for this signature in database
GPG key ID: E86E9FE9049C741F
3 changed files with 20 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
prudp_endpoint.go
View file

@ -23,6 +23,7 @@ type PRUDPEndPoint struct {
StreamID uint8
DefaultStreamSettings *StreamSettings
Connections *MutexMap[string, *PRUDPConnection]
packetHandlers map[uint16]func(packet PRUDPPacketInterface)
packetEventHandlers map[string][]func(packet PacketInterface)
connectionEndedEventHandlers []func(connection *PRUDPConnection)
errorEventHandlers []func(err *Error)
@ -39,6 +40,11 @@ func (pep *PRUDPEndPoint) RegisterServiceProtocol(protocol ServiceProtocol) {
pep.OnData(protocol.HandlePacket)
}
// RegisterCustomPacketHandler registers a custom handler for a given packet type. Used to override existing handlers or create new ones for custom packet types.
func (pep *PRUDPEndPoint) RegisterCustomPacketHandler(packetType uint16, handler func(packet PRUDPPacketInterface)) {
pep.packetHandlers[packetType] = handler
}
// OnData adds an event handler which is fired when a new DATA packet is received
func (pep *PRUDPEndPoint) OnData(handler func(packet PacketInterface)) {
pep.on("data", handler)
@ -119,17 +125,10 @@ func (pep *PRUDPEndPoint) processPacket(packet PRUDPPacketInterface, socket *Soc
return
}
switch packet.Type() {
case constants.SynPacket:
pep.handleSyn(packet)
case constants.ConnectPacket:
pep.handleConnect(packet)
case constants.DataPacket:
pep.handleData(packet)
case constants.DisconnectPacket:
pep.handleDisconnect(packet)
case constants.PingPacket:
pep.handlePing(packet)
if packetHandler, ok := pep.packetHandlers[packet.Type()]; ok {
packetHandler(packet)
} else {
logger.Warningf("Unhandled packet type %d", packet.Type())
}
}
@ -721,14 +720,23 @@ func (pep *PRUDPEndPoint) EnableVerboseRMC(enable bool) {
// NewPRUDPEndPoint returns a new PRUDPEndPoint for a server on the provided stream ID
func NewPRUDPEndPoint(streamID uint8) *PRUDPEndPoint {
return &PRUDPEndPoint{
pep := &PRUDPEndPoint{
StreamID: streamID,
DefaultStreamSettings: NewStreamSettings(),
Connections: NewMutexMap[string, *PRUDPConnection](),
packetHandlers: make(map[uint16]func(packet PRUDPPacketInterface)),
packetEventHandlers: make(map[string][]func(PacketInterface)),
connectionEndedEventHandlers: make([]func(connection *PRUDPConnection), 0),
errorEventHandlers: make([]func(err *Error), 0),
ConnectionIDCounter: NewCounter[uint32](0),
IsSecureEndPoint: false,
}
pep.packetHandlers[constants.SynPacket] = pep.handleSyn
pep.packetHandlers[constants.ConnectPacket] = pep.handleConnect
pep.packetHandlers[constants.DataPacket] = pep.handleData
pep.packetHandlers[constants.DisconnectPacket] = pep.handleDisconnect
pep.packetHandlers[constants.PingPacket] = pep.handlePing
return pep
}

4
prudp_packet_v0.go
View file

@ -100,10 +100,6 @@ func (p *PRUDPPacketV0) decode() error {
p.packetType = typeAndFlags & 0xF
}
if p.packetType > constants.PingPacket {
return errors.New("Invalid PRUDPv0 packet type")
}
p.sessionID, err = p.readStream.ReadPrimitiveUInt8()
if err != nil {
return fmt.Errorf("Failed to read PRUDPv0 session ID. %s", err.Error())

4
prudp_packet_v1.go
View file

@ -176,10 +176,6 @@ func (p *PRUDPPacketV1) decodeHeader() error {
p.flags = typeAndFlags >> 4
p.packetType = typeAndFlags & 0xF
if p.packetType > constants.PingPacket {
return errors.New("Invalid PRUDPv1 packet type")
}
p.sessionID, err = p.readStream.ReadPrimitiveUInt8()
if err != nil {
return fmt.Errorf("Failed to read PRUDPv1 session ID. %s", err.Error())