131 lines
2.8 KiB
Go
131 lines
2.8 KiB
Go
package protocolctl
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/fs"
|
|
"os"
|
|
|
|
"gitea.mmo.to/ProtocolPacketForger/ppf/protocol"
|
|
)
|
|
|
|
func NewProtocolStructure() *protocol.ProtocolStructure {
|
|
p := protocol.ProtocolStructure{}
|
|
return &p
|
|
}
|
|
|
|
func UpdateMetaData(
|
|
prot *protocol.ProtocolStructure,
|
|
name string,
|
|
version string,
|
|
tcpiplayer uint,
|
|
osilayer uint,
|
|
extensionTo string,
|
|
desc string,
|
|
requiredJSfunctions []string,
|
|
lowerLayerIdent map[string]string,
|
|
) {
|
|
prot.Metadata.Name = name
|
|
prot.Metadata.Description = desc
|
|
prot.Metadata.ExtensionTo = extensionTo
|
|
prot.Metadata.Version = version
|
|
prot.Metadata.OSILayer = osilayer
|
|
prot.Metadata.TCPIPLayer = tcpiplayer
|
|
prot.Metadata.RequiredJSFunctions = requiredJSfunctions
|
|
prot.Metadata.LowerLayerIdentification = lowerLayerIdent
|
|
}
|
|
|
|
func NewField(
|
|
name string,
|
|
desc string,
|
|
regex string,
|
|
size uint,
|
|
subfields []protocol.Field,
|
|
optional bool,
|
|
payload bool,
|
|
) *protocol.Field {
|
|
f := protocol.Field{
|
|
Name: name,
|
|
Desc: desc,
|
|
Regex: regex,
|
|
Size: size,
|
|
SubFields: subfields,
|
|
Optional: optional,
|
|
Payload: payload,
|
|
}
|
|
return &f
|
|
}
|
|
|
|
func AppendField(prot *protocol.ProtocolStructure, field *protocol.Field) {
|
|
prot.Structure = append(prot.Structure, *field)
|
|
}
|
|
|
|
func UpdateFieldByRef(prot *protocol.ProtocolStructure, field *protocol.Field) {
|
|
for _, f := range prot.Structure {
|
|
if f.Name == field.Name {
|
|
f.Desc = field.Desc
|
|
f.Optional = field.Optional
|
|
f.Payload = field.Payload
|
|
f.Regex = field.Regex
|
|
f.Size = field.Size
|
|
f.SubFields = field.SubFields
|
|
}
|
|
}
|
|
}
|
|
|
|
func UpdateFieldByElement(prot *protocol.ProtocolStructure, element int, field *protocol.Field) {
|
|
for i, f := range prot.Structure {
|
|
if i == element {
|
|
f.Desc = field.Desc
|
|
f.Optional = field.Optional
|
|
f.Payload = field.Payload
|
|
f.Regex = field.Regex
|
|
f.Size = field.Size
|
|
f.SubFields = field.SubFields
|
|
}
|
|
}
|
|
}
|
|
|
|
func RemoveFieldByRef(prot *protocol.ProtocolStructure, field *protocol.Field) {
|
|
element := -1
|
|
for i, f := range prot.Structure {
|
|
if f.Name == field.Name {
|
|
element = i
|
|
}
|
|
}
|
|
if element == -1 {
|
|
return
|
|
}
|
|
RemoveFieldByElement(prot, element)
|
|
}
|
|
|
|
func RemoveFieldByElement(prot *protocol.ProtocolStructure, field int) {
|
|
ret := make([]protocol.Field, 0)
|
|
ret = append(ret, prot.Structure[:field]...)
|
|
ret = append(ret, prot.Structure[field+1:]...)
|
|
prot.Structure = ret
|
|
}
|
|
|
|
func Load(prot *protocol.ProtocolStructure, path string) error {
|
|
data, err := os.ReadFile(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = json.Unmarshal(data, prot)
|
|
return err
|
|
}
|
|
|
|
func LoadNew(path string) (*protocol.ProtocolStructure, error) {
|
|
prot := NewProtocolStructure()
|
|
err := Load(prot, path)
|
|
return prot, err
|
|
}
|
|
|
|
func Save(prot *protocol.ProtocolStructure, path string) error {
|
|
data, err := json.Marshal(*prot)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = os.WriteFile(path, data, fs.ModeAppend)
|
|
return err
|
|
}
|