outlining

types
Jordan Orelli 8 years ago
parent f150e8aff1
commit 53793b695c

@ -0,0 +1,6 @@
package ent
type class struct {
}
type classHistory map[int]*class

@ -1,19 +1,81 @@
package ent package ent
import ( import (
"fmt"
"github.com/golang/protobuf/proto" "github.com/golang/protobuf/proto"
"github.com/jordanorelli/hyperstone/bit" "github.com/jordanorelli/hyperstone/bit"
"github.com/jordanorelli/hyperstone/dota"
) )
type Env struct { type Env struct {
source bit.BufReader symbols symbolTable
source bit.BufReader
classes map[string]classHistory
fields []field
} }
func (e *Env) Handle(m proto.Message) error { func (e *Env) Handle(m proto.Message) error {
switch v := m.(type) {
case *dota.CDemoSendTables:
return e.mergeSendTables(v)
}
return nil return nil
} }
func (e *Env) setSource(buf []byte) { func (e *Env) setSource(buf []byte) {
e.source.SetSource(buf) e.source.SetSource(buf)
} }
func (e *Env) mergeSendTables(m *dota.CDemoSendTables) error {
Debug.Printf("merge send tables")
flat, err := getSerializers(m)
if err != nil {
return wrap(err, "unable to get serializers in sendtables")
}
e.symbols = symbolTable(flat.GetSymbols())
e.stubClasses(flat)
if err := e.parseFields(flat); err != nil {
return wrap(err, "unable to parse serializer fields")
}
if err := e.parseClasses(flat); err != nil {
return wrap(err, "unable to parse serializers")
}
return nil
}
// stubs out the classes to be created later. we do this to create empty class
// structs that fields may point to.
func (e *Env) stubClasses(flat *dota.CSVCMsg_FlattenedSerializer) {
serializers := flat.GetSerializers()
if e.classes == nil {
e.classes = make(map[string]classHistory, len(serializers))
}
for _, s := range serializers {
name := e.symbol(int(s.GetSerializerNameSym()))
v := int(s.GetSerializerVersion())
if e.classes[name] == nil {
e.classes[name] = make(classHistory, 4)
}
e.classes[name][v] = new(class)
}
}
func (e *Env) parseFields(flat *dota.CSVCMsg_FlattenedSerializer) error {
e.fields = make([]field, len(flat.GetFields()))
for i, ff := range flat.GetFields() {
f := &e.fields[i]
if err := f.fromProto(ff, e); err != nil {
return err
}
}
return nil
}
func (e *Env) parseClasses(flat *dota.CSVCMsg_FlattenedSerializer) error {
return fmt.Errorf("nope, not yet")
}
func (e *Env) symbol(id int) string { return e.symbols[id] }

@ -0,0 +1,47 @@
package ent
import (
"bytes"
"fmt"
"github.com/jordanorelli/hyperstone/dota"
)
type field struct {
name string
tÿpe
}
func (f *field) fromProto(flat *dota.ProtoFlattenedSerializerFieldT, env *Env) error {
var_name := env.symbol(int(flat.GetVarNameSym()))
var_type := env.symbol(int(flat.GetVarTypeSym()))
var pretty bytes.Buffer
fmt.Fprintf(&pretty, "{name: %s type: %s", var_name, var_type)
if flat.BitCount != nil {
fmt.Fprintf(&pretty, " bits: %d", flat.GetBitCount())
}
if flat.LowValue != nil {
fmt.Fprintf(&pretty, " low: %f", flat.GetLowValue())
}
if flat.HighValue != nil {
fmt.Fprintf(&pretty, " high: %f", flat.GetHighValue())
}
if flat.EncodeFlags != nil {
fmt.Fprintf(&pretty, " flags: %d", flat.GetEncodeFlags())
}
if flat.FieldSerializerNameSym != nil {
fmt.Fprintf(&pretty, " serializer: %s", env.symbol(int(flat.GetFieldSerializerNameSym())))
}
if flat.FieldSerializerVersion != nil {
fmt.Fprintf(&pretty, " s_version: %d", flat.GetFieldSerializerVersion())
}
if flat.SendNodeSym != nil {
fmt.Fprintf(&pretty, " send: %s", env.symbol(int(flat.GetSendNodeSym())))
}
if flat.VarEncoderSym != nil {
fmt.Fprintf(&pretty, " var_enc: %s", env.symbol(int(flat.GetVarEncoderSym())))
}
fmt.Fprint(&pretty, "}")
return fmt.Errorf("unable to parse type: %s", pretty.String())
}

@ -0,0 +1,27 @@
package ent
import (
"github.com/golang/protobuf/proto"
"github.com/jordanorelli/hyperstone/bit"
"github.com/jordanorelli/hyperstone/dota"
)
func getSerializers(m *dota.CDemoSendTables) (*dota.CSVCMsg_FlattenedSerializer, error) {
data := m.GetData()
r := bit.NewBytesReader(data)
size := int(bit.ReadVarInt(r))
buf := make([]byte, size)
r.Read(buf)
if r.Err() != nil {
return nil, wrap(r.Err(), "error reading serializers body")
}
flat := dota.CSVCMsg_FlattenedSerializer{}
if err := proto.Unmarshal(buf, &flat); err != nil {
return nil, wrap(err, "error unmarshaling serializers body")
}
return &flat, nil
}

@ -0,0 +1,10 @@
package ent
type symbol struct {
id int
table *symbolTable
}
func (s symbol) String() string { return (*s.table)[s.id] }
type symbolTable []string

@ -0,0 +1,4 @@
package ent
type tÿpe interface {
}

@ -0,0 +1,9 @@
package ent
import (
"github.com/jordanorelli/hyperstone/bit"
)
type value interface {
read(bit.Reader) error
}

@ -146,7 +146,9 @@ func main() {
env := new(ent.Env) env := new(ent.Env)
handle = func(m proto.Message) { handle = func(m proto.Message) {
sd.Handle(m) sd.Handle(m)
env.Handle(m) if err := env.Handle(m); err != nil {
bail(1, "%v", err)
}
} }
default: default:
bail(1, "no such action: %s", flag.Arg(0)) bail(1, "no such action: %s", flag.Arg(0))

Loading…
Cancel
Save