Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Do not review] sketch ABI codec and serializer #179

Closed
wants to merge 1 commit into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions abi/codec/constants.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
package codec

const pubKeyLength = 32
const trueAsByte = uint8(1)
const falseAsByte = uint8(0)
261 changes: 261 additions & 0 deletions abi/codec/defaultCodec.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,261 @@
package codec

import (
"bytes"
"fmt"
"io"
"math"

"github.com/multiversx/mx-sdk-go/abi/values"
)

// defaultCodec is the default codec for encoding and decoding values.
//
// See:
// - https://docs.multiversx.com/developers/data/simple-values
// - https://docs.multiversx.com/developers/data/composite-values
// - https://docs.multiversx.com/developers/data/custom-types
type defaultCodec struct {
}

// NewDefaultCodec creates a new default codec.
func NewDefaultCodec() *defaultCodec {
return &defaultCodec{}
}

func (c *defaultCodec) EncodeNested(value any) ([]byte, error) {
buffer := bytes.NewBuffer(nil)
err := c.doEncodeNested(buffer, value)
if err != nil {
return nil, err
}

return buffer.Bytes(), nil
}

func (c *defaultCodec) doEncodeNested(writer io.Writer, value any) error {
switch value.(type) {
case values.BoolValue:
return c.encodeNestedBool(writer, value.(values.BoolValue))
case values.U8Value:
return c.encodeNestedNumber(writer, value.(values.U8Value).Value, 1)
case values.U16Value:
return c.encodeNestedNumber(writer, value.(values.U16Value).Value, 2)
case values.U32Value:
return c.encodeNestedNumber(writer, value.(values.U32Value).Value, 4)
case values.U64Value:
return c.encodeNestedNumber(writer, value.(values.U64Value).Value, 8)
case values.I8Value:
return c.encodeNestedNumber(writer, value.(values.I8Value).Value, 1)
case values.I16Value:
return c.encodeNestedNumber(writer, value.(values.I16Value).Value, 2)
case values.I32Value:
return c.encodeNestedNumber(writer, value.(values.I32Value).Value, 4)
case values.I64Value:
return c.encodeNestedNumber(writer, value.(values.I64Value).Value, 8)
case values.BigIntValue:
return c.encodeNestedBigNumber(writer, value.(values.BigIntValue).Value)
case values.AddressValue:
return c.encodeNestedAddress(writer, value.(values.AddressValue))
case values.StringValue:
return c.encodeNestedString(writer, value.(values.StringValue))
case values.BytesValue:
return c.encodeNestedBytes(writer, value.(values.BytesValue))
case values.StructValue:
return c.encodeNestedStruct(writer, value.(values.StructValue))
case values.EnumValue:
return c.encodeNestedEnum(writer, value.(values.EnumValue))
case values.OptionValue:
return c.encodeNestedOption(writer, value.(values.OptionValue))
case values.InputListValue:
return c.encodeNestedList(writer, value.(values.InputListValue))
default:
return fmt.Errorf("unsupported type for nested encoding: %T", value)
}
}

func (c *defaultCodec) EncodeTopLevel(value any) ([]byte, error) {
buffer := bytes.NewBuffer(nil)
err := c.doEncodeTopLevel(buffer, value)
if err != nil {
return nil, err
}

return buffer.Bytes(), nil
}

func (c *defaultCodec) doEncodeTopLevel(writer io.Writer, value any) error {
switch value.(type) {
case values.BoolValue:
return c.encodeTopLevelBool(writer, value.(values.BoolValue))
case values.U8Value:
return c.encodeTopLevelUnsignedNumber(writer, uint64(value.(values.U8Value).Value))
case values.U16Value:
return c.encodeTopLevelUnsignedNumber(writer, uint64(value.(values.U16Value).Value))
case values.U32Value:
return c.encodeTopLevelUnsignedNumber(writer, uint64(value.(values.U32Value).Value))
case values.U64Value:
return c.encodeTopLevelUnsignedNumber(writer, value.(values.U64Value).Value)
case values.I8Value:
return c.encodeTopLevelSignedNumber(writer, int64(value.(values.I8Value).Value))
case values.I16Value:
return c.encodeTopLevelSignedNumber(writer, int64(value.(values.I16Value).Value))
case values.I32Value:
return c.encodeTopLevelSignedNumber(writer, int64(value.(values.I32Value).Value))
case values.I64Value:
return c.encodeTopLevelSignedNumber(writer, value.(values.I64Value).Value)
case values.BigIntValue:
return c.encodeTopLevelBigNumber(writer, value.(values.BigIntValue).Value)
case values.AddressValue:
return c.encodeTopLevelAddress(writer, value.(values.AddressValue))
case values.StructValue:
return c.encodeTopLevelStruct(writer, value.(values.StructValue))
case values.EnumValue:
return c.encodeTopLevelEnum(writer, value.(values.EnumValue))
default:
return fmt.Errorf("unsupported type for top-level encoding: %T", value)
}
}

func (c *defaultCodec) DecodeNested(data []byte, value any) error {
reader := bytes.NewReader(data)
err := c.doDecodeNested(reader, value)
if err != nil {
return fmt.Errorf("cannot decode (nested) %T, because of: %w", value, err)
}

return nil
}

func (c *defaultCodec) doDecodeNested(reader io.Reader, value any) error {
switch value.(type) {
case *values.BoolValue:
return c.decodeNestedBool(reader, value.(*values.BoolValue))
case *values.U8Value:
return c.decodeNestedNumber(reader, &value.(*values.U8Value).Value, 1)
case *values.U16Value:
return c.decodeNestedNumber(reader, &value.(*values.U16Value).Value, 2)
case *values.U32Value:
return c.decodeNestedNumber(reader, &value.(*values.U32Value).Value, 4)
case *values.U64Value:
return c.decodeNestedNumber(reader, &value.(*values.U64Value).Value, 8)
case *values.I8Value:
return c.decodeNestedNumber(reader, &value.(*values.I8Value).Value, 1)
case *values.I16Value:
return c.decodeNestedNumber(reader, &value.(*values.I16Value).Value, 2)
case *values.I32Value:
return c.decodeNestedNumber(reader, &value.(*values.I32Value).Value, 4)
case *values.I64Value:
return c.decodeNestedNumber(reader, &value.(*values.I64Value).Value, 8)
case *values.BigIntValue:
n, err := c.decodeNestedBigNumber(reader)
if err != nil {
return err
}

value.(*values.BigIntValue).Value = n
return nil
case *values.AddressValue:
return c.decodeNestedAddress(reader, value.(*values.AddressValue))
case *values.StringValue:
return c.decodeNestedString(reader, value.(*values.StringValue))
case *values.BytesValue:
return c.decodeNestedBytes(reader, value.(*values.BytesValue))
case *values.StructValue:
return c.decodeNestedStruct(reader, value.(*values.StructValue))
case *values.EnumValue:
return c.decodeNestedEnum(reader, value.(*values.EnumValue))
case *values.OptionValue:
return c.decodeNestedOption(reader, value.(*values.OptionValue))
case *values.OutputListValue:
return c.decodeNestedList(reader, value.(*values.OutputListValue))
default:
return fmt.Errorf("unsupported type for nested decoding: %T", value)
}
}

func (c *defaultCodec) DecodeTopLevel(data []byte, value any) error {
err := c.doDecodeTopLevel(data, value)
if err != nil {
return fmt.Errorf("cannot decode (top-level) %T, because of: %w", value, err)
}

return nil
}

func (c *defaultCodec) doDecodeTopLevel(data []byte, value any) error {
switch value.(type) {
case *values.BoolValue:
return c.decodeTopLevelBool(data, value.(*values.BoolValue))
case *values.U8Value:
n, err := c.decodeTopLevelUnsignedNumber(data, math.MaxUint8)
if err != nil {
return err
}

value.(*values.U8Value).Value = uint8(n)
case *values.U16Value:
n, err := c.decodeTopLevelUnsignedNumber(data, math.MaxUint16)
if err != nil {
return err
}

value.(*values.U16Value).Value = uint16(n)
case *values.U32Value:
n, err := c.decodeTopLevelUnsignedNumber(data, math.MaxUint32)
if err != nil {
return err
}

value.(*values.U32Value).Value = uint32(n)
case *values.U64Value:
n, err := c.decodeTopLevelUnsignedNumber(data, math.MaxUint64)
if err != nil {
return err
}

value.(*values.U64Value).Value = uint64(n)
case *values.I8Value:
n, err := c.decodeTopLevelSignedNumber(data, math.MaxInt8)
if err != nil {
return err
}

value.(*values.I8Value).Value = int8(n)
case *values.I16Value:
n, err := c.decodeTopLevelSignedNumber(data, math.MaxInt16)
if err != nil {
return err
}

value.(*values.I16Value).Value = int16(n)
case *values.I32Value:
n, err := c.decodeTopLevelSignedNumber(data, math.MaxInt32)
if err != nil {
return err
}

value.(*values.I32Value).Value = int32(n)

case *values.I64Value:
n, err := c.decodeTopLevelSignedNumber(data, math.MaxInt64)
if err != nil {
return err
}

value.(*values.I64Value).Value = int64(n)
case *values.BigIntValue:
n := c.decodeTopLevelBigNumber(data)
value.(*values.BigIntValue).Value = n
case *values.AddressValue:
return c.decodeTopLevelAddress(data, value.(*values.AddressValue))
case *values.StructValue:
return c.decodeTopLevelStruct(data, value.(*values.StructValue))
case *values.EnumValue:
return c.decodeTopLevelEnum(data, value.(*values.EnumValue))
default:
return fmt.Errorf("unsupported type for top-level decoding: %T", value)
}

return nil
}
78 changes: 78 additions & 0 deletions abi/codec/defaultCodecForCompositeTypes.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,78 @@
package codec

import (
"errors"
"io"

"github.com/multiversx/mx-sdk-go/abi/values"
)

func (c *defaultCodec) encodeNestedOption(writer io.Writer, value values.OptionValue) error {
if value.Value == nil {
_, err := writer.Write([]byte{0})
return err
}

_, err := writer.Write([]byte{1})
if err != nil {
return err
}

return c.doEncodeNested(writer, value.Value)
}

func (c *defaultCodec) decodeNestedOption(reader io.Reader, value *values.OptionValue) error {
bytes, err := readBytesExactly(reader, 1)
if err != nil {
return err
}

if bytes[0] == 0 {
value.Value = nil
return nil
}

return c.doDecodeNested(reader, value.Value)
}

func (c *defaultCodec) encodeNestedList(writer io.Writer, value values.InputListValue) error {
err := c.encodeLength(writer, uint32(len(value.Items)))
if err != nil {
return err
}

for _, item := range value.Items {
err := c.doEncodeNested(writer, item)
if err != nil {
return err
}
}

return nil
}

func (c *defaultCodec) decodeNestedList(reader io.Reader, value *values.OutputListValue) error {
if value.ItemCreator == nil {
return errors.New("cannot deserialize list: item creator is nil")
}

length, err := c.decodeLength(reader)
if err != nil {
return err
}

value.Items = make([]any, 0, length)

for i := uint32(0); i < length; i++ {
newItem := value.ItemCreator()

err := c.doDecodeNested(reader, newItem)
if err != nil {
return err
}

value.Items = append(value.Items, newItem)
}

return nil
}
Loading
Loading