Skip to content

Commit

Permalink
system program: add NonceAccount
Browse files Browse the repository at this point in the history
  • Loading branch information
gagliardetto committed Nov 22, 2021
1 parent 3cca7d8 commit e5f3c23
Show file tree
Hide file tree
Showing 2 changed files with 122 additions and 0 deletions.
78 changes: 78 additions & 0 deletions programs/system/accounts.go
Original file line number Diff line number Diff line change
Expand Up @@ -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
}
44 changes: 44 additions & 0 deletions programs/system/accounts_test.go
Original file line number Diff line number Diff line change
@@ -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)
}

0 comments on commit e5f3c23

Please sign in to comment.