Skip to content

N
neoppod
Commit 8cad1d77 authored by Kirill Smelkov's avatar Kirill Smelkov
Browse files
Options

.

parent 63dfa906
Show whitespace changes
Inline Side-by-side
Showing with 260 additions and 116 deletions
t/neo/marshal.go
View file @ 8cad1d77
......@@ -192,7 +192,112 @@ overflow:
return 0, ErrDecodeOverflow
}
// 4. Notify
// 4. XXXTest
func (p *XXXTest) NEOEncodedLen() int {
var size0 int
for i := 0; i < len(p.Zzz); i++ {
a := &p.Zzz[i]
var size1 int
for i := 0; i < len((*a)); i++ {
a := &(*a)[i]
size1 += 4 + len((*a))
}
size0 += 4 + size1
}
return 12 + size0
}
func (p *XXXTest) NEOEncode(data []byte) {
binary.BigEndian.PutUint32(data[0:], p.qqq)
binary.BigEndian.PutUint32(data[4:], p.aaa)
{
l := uint32(len(p.Zzz))
binary.BigEndian.PutUint32(data[8:], l)
data = data[12:]
for i := 0; uint32(i) < l; i++ {
a := &p.Zzz[i]
{
l := uint32(len((*a)))
binary.BigEndian.PutUint32(data[0:], l)
data = data[4:]
for i := 0; uint32(i) < l; i++ {
a := &(*a)[i]
{
l := uint32(len((*a)))
binary.BigEndian.PutUint32(data[0:], l)
data = data[4:]
copy(data, (*a))
data = data[l:]
}
data = data[0:]
}
}
data = data[0:]
}
}
}
func (p *XXXTest) NEODecode(data []byte) (int, error) {
var nread0 uint32
if len(data) < 4 {
goto overflow
}
p.qqq = binary.BigEndian.Uint32(data[0:])
if len(data) < 8 {
goto overflow
}
p.aaa = binary.BigEndian.Uint32(data[4:])
{
if len(data) < 12 {
goto overflow
}
l := binary.BigEndian.Uint32(data[8:])
data = data[12:]
nread0 += 12
p.Zzz = make([][]string, l)
for i := 0; uint32(i) < l; i++ {
a := &p.Zzz[i]
{
if len(data) < 4 {
goto overflow
}
l := binary.BigEndian.Uint32(data[0:])
data = data[4:]
nread0 += 4
(*a) = make([]string, l)
for i := 0; uint32(i) < l; i++ {
a := &(*a)[i]
{
if len(data) < 4 {
goto overflow
}
l := binary.BigEndian.Uint32(data[0:])
data = data[4:]
if uint32(len(data)) < l {
goto overflow
}
(*a) = string(data[:l])
data = data[l:]
nread0 += 4 + l
}
data = data[0:]
nread0 += 0
}
}
data = data[0:]
nread0 += 0
}
}
return 0 + int(nread0), nil
overflow:
return 0, ErrDecodeOverflow
}
// 5. Notify
func (p *Notify) NEOEncodedLen() int {
return 4 + len(p.Message)
......@@ -229,7 +334,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 5. Error
// 6. Error
func (p *Error) NEOEncodedLen() int {
return 8 + len(p.Message)
......@@ -271,7 +376,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 6. Ping
// 7. Ping
func (p *Ping) NEOEncodedLen() int {
return 0
......@@ -284,7 +389,7 @@ func (p *Ping) NEODecode(data []byte) (int, error) {
return 0, nil
}
// 7. CloseClient
// 8. CloseClient
func (p *CloseClient) NEOEncodedLen() int {
return 0
......@@ -297,7 +402,7 @@ func (p *CloseClient) NEODecode(data []byte) (int, error) {
return 0, nil
}
// 8. RequestIdentification
// 9. RequestIdentification
func (p *RequestIdentification) NEOEncodedLen() int {
return 30 + len(p.Address.Host) + len(p.Name)
......@@ -379,7 +484,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 9. AcceptIdentification
// 10. AcceptIdentification
func (p *AcceptIdentification) NEOEncodedLen() int {
var size0 int
......@@ -508,7 +613,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 10. PrimaryMaster
// 11. PrimaryMaster
func (p *PrimaryMaster) NEOEncodedLen() int {
return 0
......@@ -521,7 +626,7 @@ func (p *PrimaryMaster) NEODecode(data []byte) (int, error) {
return 0, nil
}
// 11. AnswerPrimary
// 12. AnswerPrimary
func (p *AnswerPrimary) NEOEncodedLen() int {
return 4
......@@ -542,7 +647,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 12. AnnouncePrimary
// 13. AnnouncePrimary
func (p *AnnouncePrimary) NEOEncodedLen() int {
return 0
......@@ -555,7 +660,7 @@ func (p *AnnouncePrimary) NEODecode(data []byte) (int, error) {
return 0, nil
}
// 13. ReelectPrimary
// 14. ReelectPrimary
func (p *ReelectPrimary) NEOEncodedLen() int {
return 0
......@@ -568,7 +673,7 @@ func (p *ReelectPrimary) NEODecode(data []byte) (int, error) {
return 0, nil
}
// 14. Recovery
// 15. Recovery
func (p *Recovery) NEOEncodedLen() int {
return 0
......@@ -581,7 +686,7 @@ func (p *Recovery) NEODecode(data []byte) (int, error) {
return 0, nil
}
// 15. AnswerRecovery
// 16. AnswerRecovery
func (p *AnswerRecovery) NEOEncodedLen() int {
return 24
......@@ -612,7 +717,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 16. LastIDs
// 17. LastIDs
func (p *LastIDs) NEOEncodedLen() int {
return 0
......@@ -625,7 +730,7 @@ func (p *LastIDs) NEODecode(data []byte) (int, error) {
return 0, nil
}
// 17. AnswerLastIDs
// 18. AnswerLastIDs
func (p *AnswerLastIDs) NEOEncodedLen() int {
return 16
......@@ -651,7 +756,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 18. PartitionTable
// 19. PartitionTable
func (p *PartitionTable) NEOEncodedLen() int {
return 0
......@@ -664,7 +769,7 @@ func (p *PartitionTable) NEODecode(data []byte) (int, error) {
return 0, nil
}
// 19. AnswerPartitionTable
// 20. AnswerPartitionTable
func (p *AnswerPartitionTable) NEOEncodedLen() int {
var size0 int
......@@ -753,7 +858,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 20. NotifyPartitionTable
// 21. NotifyPartitionTable
func (p *NotifyPartitionTable) NEOEncodedLen() int {
var size0 int
......@@ -842,7 +947,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 21. PartitionChanges
// 22. PartitionChanges
func (p *PartitionChanges) NEOEncodedLen() int {
return 12 + len(p.CellList)*12
......@@ -906,7 +1011,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 22. StartOperation
// 23. StartOperation
func (p *StartOperation) NEOEncodedLen() int {
return 1
......@@ -927,7 +1032,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 23. StopOperation
// 24. StopOperation
func (p *StopOperation) NEOEncodedLen() int {
return 0
......@@ -940,7 +1045,7 @@ func (p *StopOperation) NEODecode(data []byte) (int, error) {
return 0, nil
}
// 24. UnfinishedTransactions
// 25. UnfinishedTransactions
func (p *UnfinishedTransactions) NEOEncodedLen() int {
return 0
......@@ -953,7 +1058,7 @@ func (p *UnfinishedTransactions) NEODecode(data []byte) (int, error) {
return 0, nil
}
// 25. AnswerUnfinishedTransactions
// 26. AnswerUnfinishedTransactions
func (p *AnswerUnfinishedTransactions) NEOEncodedLen() int {
return 12 + len(p.TidList)*8
......@@ -1003,7 +1108,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 26. LockedTransactions
// 27. LockedTransactions
func (p *LockedTransactions) NEOEncodedLen() int {
return 0
......@@ -1016,7 +1121,7 @@ func (p *LockedTransactions) NEODecode(data []byte) (int, error) {
return 0, nil
}
// 27. AnswerLockedTransactions
// 28. AnswerLockedTransactions
func (p *AnswerLockedTransactions) NEOEncodedLen() int {
return 4 + len(p.TidDict)*16
......@@ -1070,7 +1175,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 28. FinalTID
// 29. FinalTID
func (p *FinalTID) NEOEncodedLen() int {
return 8
......@@ -1091,7 +1196,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 29. AnswerFinalTID
// 30. AnswerFinalTID
func (p *AnswerFinalTID) NEOEncodedLen() int {
return 8
......@@ -1112,7 +1217,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 30. ValidateTransaction
// 31. ValidateTransaction
func (p *ValidateTransaction) NEOEncodedLen() int {
return 16
......@@ -1138,7 +1243,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 31. BeginTransaction
// 32. BeginTransaction
func (p *BeginTransaction) NEOEncodedLen() int {
return 8
......@@ -1159,7 +1264,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 32. AnswerBeginTransaction
// 33. AnswerBeginTransaction
func (p *AnswerBeginTransaction) NEOEncodedLen() int {
return 8
......@@ -1180,7 +1285,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 33. FinishTransaction
// 34. FinishTransaction
func (p *FinishTransaction) NEOEncodedLen() int {
return 16 + len(p.OIDList)*8 + len(p.CheckedList)*8
......@@ -1258,7 +1363,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 34. AnswerFinishTransaction
// 35. AnswerFinishTransaction
func (p *AnswerFinishTransaction) NEOEncodedLen() int {
return 16
......@@ -1284,7 +1389,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 35. NotifyTransactionFinished
// 36. NotifyTransactionFinished
func (p *NotifyTransactionFinished) NEOEncodedLen() int {
return 16
......@@ -1310,7 +1415,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 36. LockInformation
// 37. LockInformation
func (p *LockInformation) NEOEncodedLen() int {
return 16
......@@ -1336,7 +1441,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 37. AnswerLockInformation
// 38. AnswerLockInformation
func (p *AnswerLockInformation) NEOEncodedLen() int {
return 8
......@@ -1357,7 +1462,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 38. InvalidateObjects
// 39. InvalidateObjects
func (p *InvalidateObjects) NEOEncodedLen() int {
return 12 + len(p.OidList)*8
......@@ -1407,7 +1512,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 39. UnlockInformation
// 40. UnlockInformation
func (p *UnlockInformation) NEOEncodedLen() int {
return 8
......@@ -1428,7 +1533,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 40. GenerateOIDs
// 41. GenerateOIDs
func (p *GenerateOIDs) NEOEncodedLen() int {
return 4
......@@ -1449,7 +1554,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 41. AnswerGenerateOIDs
// 42. AnswerGenerateOIDs
func (p *AnswerGenerateOIDs) NEOEncodedLen() int {
return 4 + len(p.OidList)*8
......@@ -1494,7 +1599,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 42. StoreObject
// 43. StoreObject
func (p *StoreObject) NEOEncodedLen() int {
return 58 + len(p.Data)*1
......@@ -1669,7 +1774,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 43. AnswerStoreObject
// 44. AnswerStoreObject
func (p *AnswerStoreObject) NEOEncodedLen() int {
return 17
......@@ -1700,7 +1805,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 44. AbortTransaction
// 45. AbortTransaction
func (p *AbortTransaction) NEOEncodedLen() int {
return 8
......@@ -1721,7 +1826,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 45. StoreTransaction
// 46. StoreTransaction
func (p *StoreTransaction) NEOEncodedLen() int {
return 24 + len(p.User) + len(p.Description) + len(p.Extension) + len(p.OidList)*8
......@@ -1831,7 +1936,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 46. VoteTransaction
// 47. VoteTransaction
func (p *VoteTransaction) NEOEncodedLen() int {
return 8
......@@ -1852,7 +1957,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 47. GetObject
// 48. GetObject
func (p *GetObject) NEOEncodedLen() int {
return 24
......@@ -1883,7 +1988,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 48. AnswerGetObject
// 49. AnswerGetObject
func (p *AnswerGetObject) NEOEncodedLen() int {
return 57 + len(p.Data)*1
......@@ -2053,7 +2158,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 49. TIDList
// 50. TIDList
func (p *TIDList) NEOEncodedLen() int {
return 20
......@@ -2084,7 +2189,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 50. AnswerTIDList
// 51. AnswerTIDList
func (p *AnswerTIDList) NEOEncodedLen() int {
return 4 + len(p.TIDList)*8
......@@ -2129,7 +2234,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 51. TIDListFrom
// 52. TIDListFrom
func (p *TIDListFrom) NEOEncodedLen() int {
return 24
......@@ -2165,7 +2270,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 52. AnswerTIDListFrom
// 53. AnswerTIDListFrom
func (p *AnswerTIDListFrom) NEOEncodedLen() int {
return 4 + len(p.TidList)*8
......@@ -2210,7 +2315,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 53. TransactionInformation
// 54. TransactionInformation
func (p *TransactionInformation) NEOEncodedLen() int {
return 8
......@@ -2231,7 +2336,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 54. AnswerTransactionInformation
// 55. AnswerTransactionInformation
func (p *AnswerTransactionInformation) NEOEncodedLen() int {
return 25 + len(p.User) + len(p.Description) + len(p.Extension) + len(p.OidList)*8
......@@ -2346,7 +2451,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 55. ObjectHistory
// 56. ObjectHistory
func (p *ObjectHistory) NEOEncodedLen() int {
return 24
......@@ -2377,7 +2482,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 56. AnswerObjectHistory
// 57. AnswerObjectHistory
func (p *AnswerObjectHistory) NEOEncodedLen() int {
return 12 + len(p.HistoryList)*12
......@@ -2435,7 +2540,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 57. PartitionList
// 58. PartitionList
func (p *PartitionList) NEOEncodedLen() int {
return 12
......@@ -2466,7 +2571,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 58. AnswerPartitionList
// 59. AnswerPartitionList
func (p *AnswerPartitionList) NEOEncodedLen() int {
var size0 int
......@@ -2555,7 +2660,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 59. X_NodeList
// 60. X_NodeList
func (p *X_NodeList) NEOEncodedLen() int {
return 4
......@@ -2576,7 +2681,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 60. AnswerNodeList
// 61. AnswerNodeList
func (p *AnswerNodeList) NEOEncodedLen() int {
var size0 int
......@@ -2667,7 +2772,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 61. SetNodeState
// 62. SetNodeState
func (p *SetNodeState) NEOEncodedLen() int {
return 8
......@@ -2693,7 +2798,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 62. AddPendingNodes
// 63. AddPendingNodes
func (p *AddPendingNodes) NEOEncodedLen() int {
return 4 + len(p.UUIDList)*4
......@@ -2738,7 +2843,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 63. TweakPartitionTable
// 64. TweakPartitionTable
func (p *TweakPartitionTable) NEOEncodedLen() int {
return 4 + len(p.UUIDList)*4
......@@ -2783,7 +2888,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 64. NotifyNodeInformation
// 65. NotifyNodeInformation
func (p *NotifyNodeInformation) NEOEncodedLen() int {
var size0 int
......@@ -2874,7 +2979,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 65. NodeInformation
// 66. NodeInformation
func (p *NodeInformation) NEOEncodedLen() int {
return 0
......@@ -2887,7 +2992,7 @@ func (p *NodeInformation) NEODecode(data []byte) (int, error) {
return 0, nil
}
// 66. SetClusterState
// 67. SetClusterState
func (p *SetClusterState) NEOEncodedLen() int {
return 4
......@@ -2908,7 +3013,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 67. ClusterInformation
// 68. ClusterInformation
func (p *ClusterInformation) NEOEncodedLen() int {
return 4
......@@ -2929,7 +3034,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 68. X_ClusterState
// 69. X_ClusterState
func (p *X_ClusterState) NEOEncodedLen() int {
return 4
......@@ -2950,7 +3055,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 69. ObjectUndoSerial
// 70. ObjectUndoSerial
func (p *ObjectUndoSerial) NEOEncodedLen() int {
return 28 + len(p.OidList)*8
......@@ -3010,7 +3115,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 70. AnswerObjectUndoSerial
// 71. AnswerObjectUndoSerial
func (p *AnswerObjectUndoSerial) NEOEncodedLen() int {
return 4 + len(p.ObjectTIDDict)*25
......@@ -3084,7 +3189,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 71. HasLock
// 72. HasLock
func (p *HasLock) NEOEncodedLen() int {
return 16
......@@ -3110,7 +3215,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 72. AnswerHasLock
// 73. AnswerHasLock
func (p *AnswerHasLock) NEOEncodedLen() int {
return 12
......@@ -3136,7 +3241,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 73. CheckCurrentSerial
// 74. CheckCurrentSerial
func (p *CheckCurrentSerial) NEOEncodedLen() int {
return 24
......@@ -3167,7 +3272,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 74. AnswerCheckCurrentSerial
// 75. AnswerCheckCurrentSerial
func (p *AnswerCheckCurrentSerial) NEOEncodedLen() int {
return 17
......@@ -3198,7 +3303,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 75. Pack
// 76. Pack
func (p *Pack) NEOEncodedLen() int {
return 8
......@@ -3219,7 +3324,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 76. AnswerPack
// 77. AnswerPack
func (p *AnswerPack) NEOEncodedLen() int {
return 1
......@@ -3240,7 +3345,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 77. CheckReplicas
// 78. CheckReplicas
func (p *CheckReplicas) NEOEncodedLen() int {
return 20 + len(p.PartitionDict)*8
......@@ -3304,7 +3409,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 78. CheckPartition
// 79. CheckPartition
func (p *CheckPartition) NEOEncodedLen() int {
return 30 + len(p.Source.UpstreamName) + len(p.Source.Address.Host)
......@@ -3381,7 +3486,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 79. CheckTIDRange
// 80. CheckTIDRange
func (p *CheckTIDRange) NEOEncodedLen() int {
return 24
......@@ -3417,7 +3522,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 80. AnswerCheckTIDRange
// 81. AnswerCheckTIDRange
func (p *AnswerCheckTIDRange) NEOEncodedLen() int {
return 32
......@@ -3543,7 +3648,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 81. CheckSerialRange
// 82. CheckSerialRange
func (p *CheckSerialRange) NEOEncodedLen() int {
return 32
......@@ -3584,7 +3689,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 82. AnswerCheckSerialRange
// 83. AnswerCheckSerialRange
func (p *AnswerCheckSerialRange) NEOEncodedLen() int {
return 60
......@@ -3815,7 +3920,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 83. PartitionCorrupted
// 84. PartitionCorrupted
func (p *PartitionCorrupted) NEOEncodedLen() int {
return 8 + len(p.CellList)*4
......@@ -3865,7 +3970,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 84. LastTransaction
// 85. LastTransaction
func (p *LastTransaction) NEOEncodedLen() int {
return 0
......@@ -3878,7 +3983,7 @@ func (p *LastTransaction) NEODecode(data []byte) (int, error) {
return 0, nil
}
// 85. AnswerLastTransaction
// 86. AnswerLastTransaction
func (p *AnswerLastTransaction) NEOEncodedLen() int {
return 8
......@@ -3899,7 +4004,7 @@ overflow:
return 0, ErrDecodeOverflow
}
// 86. NotifyReady
// 87. NotifyReady
func (p *NotifyReady) NEOEncodedLen() int {
return 0
......
t/neo/proto.go
View file @ 8cad1d77
......@@ -195,6 +195,13 @@ type RowInfo struct {
}
type XXXTest struct {
qqq uint32
aaa uint32
Zzz [][]string
}
// // XXX link request <-> answer ?
// // XXX naming -> PktHeader ?
......
t/neo/protogen.go
View file @ 8cad1d77
......@@ -208,11 +208,13 @@ func (b *Buffer) emit(format string, a ...interface{}) {
fmt.Fprintf(b, format+"\n", a...)
}
// interface of codegenerator for coder/decoder
// interface of codegenerator for sizer/coder/decoder
type CodecCodeGen interface {
// tell codegen it should generate code for top-level function
setFunc(recvName, typeName string, typ types.Type)
// TODO gen*() -> visit*() ?
// emit code to process basic fixed types (not string)
// userType is type actually used in source (for which typ is underlying), or nil
genBasic(path string, typ *types.Basic, userType types.Type)
......@@ -233,7 +235,7 @@ type CodecCodeGen interface {
resultExpr() string
}
// sizer/encode/decode codegen
// common part of codegenerators
type commonCoder struct {
recvName string // receiver/type for top-level func
typeName string // or empty
......@@ -263,20 +265,40 @@ func (c *commonCoder) var_(varname string) string {
return varnameX
}
// information about a size
// consists of numeric & symbolic parts
// size is <num> + expr1 + expr2 + ...
type size struct {
num int // numeric part of size
exprv []string // symbolic part of size
}
func (s *size) Add(n int) {
s.num += n
}
func (s *size) AddExpr(expr string) {
s.exprv = append(s.exprv, expr)
}
func (s *size) String() string {
sizeExpr := fmt.Sprintf("%v", s.num)
if len(s.exprv) != 0 {
sizeExpr += " + " + strings.Join(s.exprv, " + ")
}
return sizeExpr
}
// sizer generates code to compute endoded size of a packet
type sizer struct {
Buffer // buffer for code
n int // fixed part of size
symLenv []string // symbolic part of size
size size
commonCoder
}
// create new sizer for subsize calculation (e.g. for loop)
func (s *sizer) subSizer() *sizer {
return &sizer{commonCoder: commonCoder{varN: s.varN + 1}}
}
// encoder generates code to encode a packet
type encoder struct {
Buffer // XXX
n int
......@@ -284,6 +306,7 @@ type encoder struct {
commonCoder
}
// decoder generates code to decode a packet
type decoder struct {
Buffer // buffer for generated code
n int // current decode position in data
......@@ -295,11 +318,14 @@ var _ CodecCodeGen = (*sizer)(nil)
var _ CodecCodeGen = (*encoder)(nil)
var _ CodecCodeGen = (*decoder)(nil)
// create new sizer for subsize calculation (e.g. inside loop)
func (s *sizer) subSizer() *sizer {
return &sizer{commonCoder: commonCoder{varN: s.varN + 1}}
}
func (s *sizer) resultExpr() string {
size := fmt.Sprintf("%v", s.n)
if len(s.symLenv) > 0 {
size += " + " + strings.Join(s.symLenv, " + ")
}
size := s.size.String()
if s.varUsed["size"] {
size += " + " + s.var__("size")
}
......@@ -328,7 +354,7 @@ func (s *sizer) generatedCode() string {
}
func (e *encoder) resultExpr() string {
panic("should not be called (?)")
panic("should not be called (?)") // XXX
}
func (e *encoder) generatedCode() string {
......@@ -347,7 +373,7 @@ func (e *encoder) generatedCode() string {
}
func (d *decoder) resultExpr() string {
panic("should not be called (?)")
panic("should not be called (?)") // XXX
}
func (d *decoder) generatedCode() string {
......@@ -382,7 +408,7 @@ func (d *decoder) generatedCode() string {
func (s *sizer) genBasic(path string, typ *types.Basic, userType types.Type) {
basic := basicTypes[typ.Kind()]
s.n += basic.wireSize
s.size.Add(basic.wireSize)
}
func (e *encoder) genBasic(path string, typ *types.Basic, userType types.Type) {
......@@ -420,8 +446,8 @@ func (d *decoder) genBasic(assignto string, typ *types.Basic, userType types.Typ
// [len]byte
// TODO []byte support
func (s *sizer) genStrBytes(path string) {
s.n += 4
s.symLenv = append(s.symLenv, fmt.Sprintf("len(%s)", path))
s.size.Add(4)
s.size.AddExpr(fmt.Sprintf("len(%s)", path))
}
func (e *encoder) genStrBytes(path string) {
......@@ -450,28 +476,26 @@ func (d *decoder) genStrBytes(assignto string) {
// emit code to encode/decode slice
// len u32
// [len]item
// TODO optimize for []byte
func (s *sizer) genSlice(path string, typ *types.Slice, obj types.Object) {
// if size(item)==const - size update in one go
elemSize, ok := typeSizeFixed(typ.Elem())
if ok {
s.n += 4
s.symLenv = append(s.symLenv, fmt.Sprintf("len(%v) * %v", path, elemSize))
s.size.Add(4)
s.size.AddExpr(fmt.Sprintf("len(%v) * %v", path, elemSize))
return
}
s.n += 4
s.size.Add(4)
s.emit("for i := 0; i < len(%v); i++ {", path)
s.emit("a := &%s[i]", path)
//codegenType("(*a)", typ.Elem(), obj, s)
sloop := s.subSizer()
codegenType("(*a)", typ.Elem(), obj, sloop)
// FIXME vvv if symLenv is ø; -> turn into "result" function
s.emit(sloop.generatedCode())
s.emit("%v += %v", s.var_("size"), sloop.resultExpr())
s.emit("}")
}
// TODO optimize for []byte
func (e *encoder) genSlice(path string, typ *types.Slice, obj types.Object) {
e.emit("{")
e.emit("l := uint32(len(%s))", path)
......@@ -488,19 +512,27 @@ func (e *encoder) genSlice(path string, typ *types.Slice, obj types.Object) {
e.n = 0
}
// TODO optimize for []byte
func (d *decoder) genSlice(assignto string, typ *types.Slice, obj types.Object) {
d.emit("{")
d.genBasic("l:", types.Typ[types.Uint32], nil)
d.emit("data = data[%v:]", d.n)
d.emit("%v += %v", d.var_("nread"), d.n)
d.n = 0
/*
elemSize, elemFixed := typeSizeFixed(typ.Elem())
// if size(item)==const - check l in one go
if elemFixed {
d.emit("if len(data) < l*%v { goto overflow }", elemSize)
}
// TODO ^^^ then skip inner overflow checks
*/
d.emit("%v= make(%v, l)", assignto, typeName(typ))
// TODO size check
// TODO if size(item)==const - check l in one go
//d.emit("if len(data) < l { goto overflow }")
d.emit("for i := 0; uint32(i) < l; i++ {")
d.emit("a := &%s[i]", assignto)
// XXX try to avoid (*) in a
codegenType("(*a)", typ.Elem(), obj, d)
d.emit("data = data[%v:]", d.n) // FIXME wrt slice of slice ?
d.emit("%v += %v", d.var_("nread"), d.n)
......@@ -518,17 +550,17 @@ func (s *sizer) genMap(path string, typ *types.Map, obj types.Object) {
elemSize, elemFixed := typeSizeFixed(typ.Elem())
if keyFixed && elemFixed {
s.n += 4
s.symLenv = append(s.symLenv, fmt.Sprintf("len(%v) * %v", path, keySize + elemSize))
s.size.Add(4)
s.size.AddExpr(fmt.Sprintf("len(%v) * %v", path, keySize + elemSize))
return
}
panic("UNTESTED")
s.n += 4
s.size.Add(4)
s.emit("for key := range %s {", path)
codegenType("key", typ.Key(), obj, s)
codegenType(fmt.Sprintf("%s[key]", path), typ.Elem(), obj, s)
s.emit("%v += %v", s.var_("size"), s.n)
s.emit("%v += %v", s.var_("size"), s.size)
s.emit("}")
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment
GitLab Nexedi Edition | About GitLab | About Nexedi | 沪ICP备2021021310号-2 | 沪ICP备2021021310号-7