system program: add NonceAccount

This commit is contained in:
gagliardetto 2021-11-22 17:21:51 +01:00
parent 3cca7d8e62
commit e5f3c23fae
2 changed files with 122 additions and 0 deletions

View File

@ -13,3 +13,81 @@
// limitations under the License.
package system
import (
"encoding/binary"
bin "github.com/gagliardetto/binary"
"github.com/gagliardetto/solana-go"
)
type NonceAccount struct {
Version uint32
State uint32
AuthorizedPubkey solana.PublicKey
Nonce solana.PublicKey
FeeCalculator FeeCalculator
}
type FeeCalculator struct {
LamportsPerSignature uint64
}
func (obj NonceAccount) MarshalWithEncoder(encoder *bin.Encoder) (err error) {
err = encoder.WriteUint32(obj.Version, binary.LittleEndian)
if err != nil {
return err
}
err = encoder.WriteUint32(obj.State, binary.LittleEndian)
if err != nil {
return err
}
err = encoder.WriteBytes(obj.AuthorizedPubkey[:], false)
if err != nil {
return err
}
err = encoder.WriteBytes(obj.Nonce[:], false)
if err != nil {
return err
}
return obj.FeeCalculator.MarshalWithEncoder(encoder)
}
func (obj *NonceAccount) UnmarshalWithDecoder(decoder *bin.Decoder) (err error) {
{
obj.Version, err = decoder.ReadUint32(binary.LittleEndian)
if err != nil {
return err
}
}
{
obj.State, err = decoder.ReadUint32(binary.LittleEndian)
if err != nil {
return err
}
}
{
buf, err := decoder.ReadNBytes(32)
if err != nil {
return err
}
obj.AuthorizedPubkey = solana.PublicKeyFromBytes(buf)
}
{
buf, err := decoder.ReadNBytes(32)
if err != nil {
return err
}
obj.Nonce = solana.PublicKeyFromBytes(buf)
}
return obj.FeeCalculator.UnmarshalWithDecoder(decoder)
}
func (obj FeeCalculator) MarshalWithEncoder(encoder *bin.Encoder) (err error) {
return encoder.WriteUint64(obj.LamportsPerSignature, binary.LittleEndian)
}
func (obj *FeeCalculator) UnmarshalWithDecoder(decoder *bin.Decoder) (err error) {
obj.LamportsPerSignature, err = decoder.ReadUint64(binary.LittleEndian)
return err
}

View File

@ -0,0 +1,44 @@
// Copyright 2021 github.com/gagliardetto
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package system
import (
"encoding/base64"
"testing"
bin "github.com/gagliardetto/binary"
"github.com/gagliardetto/solana-go"
"github.com/stretchr/testify/assert"
)
func TestDecode(t *testing.T) {
nonceAccountBase64Data := "AAAAAAEAAABHaauXIEuoP7DK7hf3ho8eB05SFYGg2J2UN52qZbcXsnM+zs3rCNyHGAjze1Gvfq4gRzzrz7ggv4rYXkMo8P2DiBMAAAAAAAA="
decoded, err := base64.StdEncoding.DecodeString(nonceAccountBase64Data)
assert.NoError(t, err)
dec := bin.NewBinDecoder(decoded)
acc := new(NonceAccount)
err = acc.UnmarshalWithDecoder(dec)
assert.NoError(t, err)
assert.Equal(t, uint32(0), acc.Version)
assert.Equal(t, uint32(1), acc.State)
assert.Equal(t, solana.MustPublicKeyFromBase58("5omQJtDUHA3gMFdHEQg1zZSvcBUVzey5WaKWYRmqF1Vj"), acc.AuthorizedPubkey)
assert.Equal(t, solana.MustPublicKeyFromBase58("8ksS6xXd7vzNrpZfBTf9gJ87Bma5AjnQ9baEcT7xH5QE"), acc.Nonce)
assert.Equal(t, uint64(5000), acc.FeeCalculator.LamportsPerSignature)
}