93 lines
2.5 KiB
Go
93 lines
2.5 KiB
Go
package proto
|
|
|
|
import (
|
|
"reflect"
|
|
|
|
"github.com/admin-else/strom/proto_base"
|
|
"github.com/admin-else/strom/proto_generated"
|
|
)
|
|
|
|
type PidEtc struct {
|
|
Direction proto_base.Direction
|
|
State proto_base.State
|
|
PacketId int32
|
|
ProtocolVersion int32
|
|
}
|
|
|
|
var pidMap = make(map[PidEtc]*proto_base.PacketInfo)
|
|
|
|
func init() {
|
|
for _, pin := range proto_generated.Packets {
|
|
pidMap[PidEtc{pin.Direction, pin.State, pin.PacketId, pin.ProtocolVersion}] = &pin
|
|
}
|
|
}
|
|
|
|
func LookUpTypeByPacketInfo(direction proto_base.Direction, state proto_base.State, pid, version int32) (p proto_base.PacketInfo, ok bool) {
|
|
pP, ok := pidMap[PidEtc{direction, state, pid, version}]
|
|
if !ok {
|
|
p = proto_base.PacketInfo{
|
|
Type: nil,
|
|
Name: "",
|
|
Direction: direction,
|
|
State: state,
|
|
PacketId: pid,
|
|
ProtocolVersion: version,
|
|
}
|
|
return
|
|
}
|
|
p = *pP
|
|
return
|
|
}
|
|
|
|
func LookUpTypeByPacketInfoAndCopyType(direction proto_base.Direction, state proto_base.State, pid, version int32) (p proto_base.EncodeDecodeAble, ok bool) {
|
|
i, ok := LookUpTypeByPacketInfo(direction, state, pid, version)
|
|
if !ok {
|
|
return
|
|
}
|
|
p = reflect.New(reflect.TypeOf(i.Type).Elem()).Interface().(proto_base.EncodeDecodeAble)
|
|
return
|
|
}
|
|
|
|
func LookupPacketInfoByType(packet proto_base.EncodeDecodeAble) (p proto_base.PacketInfo, ok bool) {
|
|
undecodedPacket, isUndecoable := packet.(*UnCodablePacket)
|
|
if isUndecoable {
|
|
p = undecodedPacket.Info
|
|
ok = true
|
|
return
|
|
}
|
|
|
|
for _, p := range proto_generated.Packets {
|
|
if reflect.TypeOf(p.Type) == reflect.TypeOf(packet) {
|
|
return p, true
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func LookupPacketInfoByTypeAndState(packet proto_base.EncodeDecodeAble, state proto_base.State) (p proto_base.PacketInfo, ok bool) {
|
|
for _, p := range proto_generated.Packets {
|
|
if state == p.State && reflect.TypeOf(p.Type) == reflect.TypeOf(packet) {
|
|
return p, true
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func LookupPacketInfoByNameProtocolVersionAndState(name string, version int32, state proto_base.State) (p proto_base.PacketInfo, ok bool) {
|
|
for _, p := range proto_generated.Packets {
|
|
if state == p.State && p.Name == name && p.ProtocolVersion == version {
|
|
return p, true
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func LookupPacketInfoByNameProtocolVersionStateAndDirection(name string, version int32, state proto_base.State, direction proto_base.Direction) (p proto_base.PacketInfo, ok bool) {
|
|
for _, p := range proto_generated.Packets {
|
|
if state == p.State && p.Name == name && p.ProtocolVersion == version && p.Direction == direction {
|
|
return p, true
|
|
}
|
|
}
|
|
return
|
|
}
|