xdr.go

/*

Stellar transaction compiler library.  Provides functions for
manipulating Stellar transactions, translating them back and forth
between txrep format, and posting them.

*/
package stc

import (
	"fmt"
	"github.com/xdrpp/goxdr/xdr"
	"github.com/xdrpp/stc/stcdetail"
	"github.com/xdrpp/stc/stx"
	"io"
	"reflect"
	"strings"
)

type PublicKey = stx.PublicKey
type AccountID = stx.AccountID
type MuxedAccount = stx.MuxedAccount
type SignerKey = stx.SignerKey
type Signature = stx.Signature
type TransactionResult = stx.TransactionResult
type LedgerHeader = stx.LedgerHeader

// Largest 64-bit signed integer (9223372036854775807).
const MaxInt64 = 0x7fffffffffffffff

func NativeAsset() stx.Asset {
	return stx.Asset{
		Type: stx.ASSET_TYPE_NATIVE,
	}
}

func MkAsset(acc AccountID, code string) stx.Asset {
	var ret stx.Asset
	if len(code) <= 4 {
		ret.Type = stx.ASSET_TYPE_CREDIT_ALPHANUM4
		copy(ret.AlphaNum4().AssetCode[:], code)
		ret.AlphaNum4().Issuer = acc
	} else if len(code) <= 12 {
		ret.Type = stx.ASSET_TYPE_CREDIT_ALPHANUM12
		copy(ret.AlphaNum12().AssetCode[:], code)
		ret.AlphaNum4().Issuer = acc
	} else {
		xdr.XdrPanic("MkAsset: %q exceeds 12 characters", code)
	}
	return ret
}

func MkAssetCode(code string) stx.AssetCode {
	var ret stx.AssetCode
	if len(code) <= 4 {
		ret.Type = stx.ASSET_TYPE_CREDIT_ALPHANUM4
		copy(ret.AssetCode4()[:], code)
	} else if len(code) <= 12 {
		ret.Type = stx.ASSET_TYPE_CREDIT_ALPHANUM12
		copy(ret.AssetCode12()[:], code)
	} else {
		xdr.XdrPanic("MkAssetCode: %q exceeds 12 characters", code)
	}
	return ret
}

// Return a pointer to an account ID
func NewAccountID(id AccountID) *AccountID {
	return &id
}

// Create a signer for a particular public key and weight
func NewSignerKey(pk PublicKey, weight uint32) *stx.Signer {
	return &stx.Signer{
		Key:    pk.ToSignerKey(),
		Weight: weight,
	}
}

// Create a pre-signed transaction from a transaction and weight.
func (net *StellarNet) NewSignerPreauth(tx stx.Signable,
	weight uint32) *stx.Signer {
	ret := stx.Signer{Weight: weight}
	ret.Key.Type = stx.SIGNER_KEY_TYPE_PRE_AUTH_TX
	*ret.Key.PreAuthTx() = *net.HashTx(tx)
	return &ret
}

// Create a signer that requires the hash pre-image of some hash value x
func NewSignerHashX(x stx.Hash, weight uint32) *stx.Signer {
	ret := stx.Signer{Weight: weight}
	ret.Key.Type = stx.SIGNER_KEY_TYPE_PRE_AUTH_TX
	*ret.Key.HashX() = x
	return &ret
}

// Allocate a uint32 when initializing types that take an XDR int*.
func NewUint(v uint32) *uint32 { return &v }

// Allocate an int64 when initializing types that take an XDR hyper*.
func NewHyper(v int64) *int64 { return &v }

// Allocate a uint64 when initializing types that take an XDR unsigned hyper*.
func NewUhyper(v uint64) *uint64 { return &v }

// Allocate a string when initializing types that take an XDR *string<>.
func NewString(v string) *string { return &v }

// Created a MuxedAccount from its consituent parts.  id may be nil to
// indicate there is no embedded identifier.
func MuxAcct(acct *AccountID, id *uint64) *MuxedAccount {
	switch acct.Type {
	case stx.PUBLIC_KEY_TYPE_ED25519:
		if id == nil {
			ret := &MuxedAccount{Type: stx.KEY_TYPE_ED25519}
			*ret.Ed25519() = *acct.Ed25519()
			return ret
		} else {
			ret := &MuxedAccount{Type: stx.KEY_TYPE_MUXED_ED25519}
			ret.Med25519().Ed25519 = *acct.Ed25519()
			ret.Med25519().Id = *id
			return ret
		}
	}
	return nil
}

// Break a MuxedAccount into its consituent parts.  Note that the
// second return value of type *uint64 may be nil for MuxedAccounts
// that don't include an embedded identifier.
func DemuxAcct(macct *MuxedAccount) (*AccountID, *uint64) {
	switch macct.Type {
	case stx.KEY_TYPE_ED25519:
		ret := &AccountID{Type: stx.PUBLIC_KEY_TYPE_ED25519}
		*ret.Ed25519() = *macct.Ed25519()
		return ret, nil
	case stx.KEY_TYPE_MUXED_ED25519:
		ret := &AccountID{Type: stx.PUBLIC_KEY_TYPE_ED25519}
		*ret.Ed25519() = macct.Med25519().Ed25519
		return ret, &macct.Med25519().Id
	}
	return nil, nil
}

// This is a wrapper around the XDR TransactionEnvelope structure.
// The wrapper allows transactions to be built up more easily via the
// Append() method and various helper types.  When parsing and
// generating Txrep format, it also keeps track of which enums were
// followed by '?' indicating a request for help.
type TransactionEnvelope struct {
	*stx.TransactionEnvelope
	Help map[string]struct{}
}

func NewTransactionEnvelope() *TransactionEnvelope {
	return &TransactionEnvelope{
		TransactionEnvelope: &stx.TransactionEnvelope{
			Type: stx.ENVELOPE_TYPE_TX,
		},
		Help: nil,
	}
}

// Interface for placeholder types that are named by camel-cased
// versions of the OperationType enum and can be transformed into the
// body of an Operation
type OperationBody interface {
	To_Operation_Body() stx.XdrAnon_Operation_Body
}

/*

Append an operation to a transaction envelope.  To facilitate
initialization of the transaction body (which is a union and so
doesn't support direct initialization), a suite of helper types
have the same fields as each of the operation types.  The helper
types are named after camel-cased versions of the OperationType
constants.  E.g., to append an operation of type CREATE_ACCOUNT,
use type type CreateAccount:

	txe.Append(nil, CreateAccount{
		Destination: myNewAccountID,
		StartingBalance: 15000000,
	})

The helper types are:

	type CreateAccount stx.CreateAccountOp
	type Payment stx.PaymentOp
	type PathPaymentStrictReceive stx.PathPaymentStrictReceiveOp
	type ManageSellOffer stx.ManageSellOfferOp
	type CreatePassiveSellOffer stx.CreatePassiveSellOfferOp
	type SetOptions stx.SetOptionsOp
	type ChangeTrust stx.ChangeTrustOp
	type AllowTrust stx.AllowTrustOp
	type AccountMerge stx.PublicKey
	type Inflation struct{}
	type ManageData stx.ManageDataOp
	type BumpSequence stx.BumpSequenceOp
	type ManageBuyOffer stx.ManageBuyOfferOp
	type PathPaymentStrictSend stx.PathPaymentStrictSendOp

*/
func (txe *TransactionEnvelope) Append(
	sourceAccount *stx.MuxedAccount,
	body OperationBody) {
	ops := txe.Operations()
	if ops == nil {
		xdr.XdrPanic("TransactionEnvelope.Append: invalid envelope type")
	} else if len(*ops) >= stx.MAX_OPS_PER_TX {
		xdr.XdrPanic(
			"TransactionEnvelope.Append: attempt to exceed %d operations",
			stx.MAX_OPS_PER_TX)
	} else if len(*txe.Signatures()) > 0 {
		xdr.XdrPanic("TransactionEnvelope.Append: transaction already signed")
	}
	*ops = append(*ops, stx.Operation{
		SourceAccount: sourceAccount,
		Body:          body.To_Operation_Body(),
	})
}

// Set the fee of a transaction to baseFee times the number of
// operations.  If the result would exceed the maximum fee of
// 0xffffffff (~430 XLM), then just set the fee to 0xffffffff.
// (Obviously only call this once you have finished adding operations
// to the transaction with Append.)
func (txe *TransactionEnvelope) SetFee(baseFee uint32) {
	if txe.Type == stx.ENVELOPE_TYPE_TX_FEE_BUMP {
		txe.FeeBump().Tx.Fee = int64(baseFee) *
			int64(len(txe.FeeBump().Tx.InnerTx.V1().Tx.Operations))
		return
	}
	if ops := txe.Operations(); ops != nil {
		fee := int64(baseFee) * int64(len(*ops))
		fee32 := uint32(fee)
		if fee > 0xffffffff {
			fee32 = 0xffffffff
		}
		switch txe.Type {
		case stx.ENVELOPE_TYPE_TX:
			txe.V1().Tx.Fee = fee32
			return
		case stx.ENVELOPE_TYPE_TX_V0:
			txe.V0().Tx.Fee = fee32
			return
		}
	}
	xdr.XdrPanic("SetFee: Invalid envelope type %s", txe.Type)
}

func (txe *TransactionEnvelope) SourceAccount() *stx.MuxedAccount {
	switch txe.Type {
	case stx.ENVELOPE_TYPE_TX_V0:
		ret := stx.MuxedAccount{Type: stx.KEY_TYPE_ED25519}
		*ret.Ed25519() = txe.V0().Tx.SourceAccountEd25519
		return &ret
	case stx.ENVELOPE_TYPE_TX:
		return &txe.V1().Tx.SourceAccount
	case stx.ENVELOPE_TYPE_TX_FEE_BUMP:
		return &txe.FeeBump().Tx.FeeSource
	}
	xdr.XdrPanic("SourceAccountEd25519: unknown TransactionEnvelope type %s",
		txe.Type)
	return nil
}

func (txe *TransactionEnvelope) SetSourceAccount(m0 stx.IsAccount) {
	m := m0.ToMuxedAccount()
	switch txe.Type {
	case stx.ENVELOPE_TYPE_TX_V0:
		if m.Type == stx.KEY_TYPE_ED25519 {
			txe.V0().Tx.SourceAccountEd25519 = *m.Ed25519()
		} else {
			xdr.XdrPanic("SetSourceAccount: %v not supported in envelope V0",
				m.Type)
		}
	case stx.ENVELOPE_TYPE_TX:
		txe.V1().Tx.SourceAccount = *m
	case stx.ENVELOPE_TYPE_TX_FEE_BUMP:
		txe.FeeBump().Tx.FeeSource = *m
	}
}

func (txe *TransactionEnvelope) GetHelp(name string) bool {
	_, ok := txe.Help[name]
	return ok
}

func (txe *TransactionEnvelope) SetHelp(name string) {
	if txe.Help == nil {
		txe.Help = map[string]struct{}{name: {}}
	} else {
		txe.Help[name] = struct{}{}
	}
}

func (net *StellarNet) SigNote(txe *stx.TransactionEnvelope,
	sig *stx.DecoratedSignature) string {
	if txe == nil {
		return ""
	} else if ski := net.Signers.Lookup(net.GetNetworkId(), txe, sig); ski != nil {
		return ski.String()
	}
	return fmt.Sprintf("bad signature/unknown key/%s is wrong network",
		net.Name)
}

func (net *StellarNet) AccountIDNote(acct string) string {
	return net.Accounts[acct]
}

func (net *StellarNet) SignerNote(key *stx.SignerKey) string {
	return net.Signers.LookupComment(key)
}

// Write the human-readable Txrep of an XDR structure to a Writer.
func (net *StellarNet) WriteRep(out io.Writer, name string, txe xdr.XdrType) {
	type helper interface {
		xdr.XdrType
		GetHelp(string) bool
	}
	if net == nil {
		stcdetail.XdrToTxrep(out, name, txe)
	} else if e, ok := txe.(helper); ok {
		ntxe := struct {
			helper
			*StellarNet
		}{e, (*StellarNet)(net)}
		stcdetail.XdrToTxrep(out, name, ntxe)
	} else {
		ntxe := struct {
			xdr.XdrType
			*StellarNet
		}{txe, (*StellarNet)(net)}
		stcdetail.XdrToTxrep(out, name, ntxe)
	}
}

// Convert an arbitrary XDR data structure to human-readable Txrep
// format.
func (net *StellarNet) ToRep(txe xdr.XdrType) string {
	var out strings.Builder
	net.WriteRep(&out, "", txe)
	return out.String()
}

// Convert a TransactionEnvelope to human-readable Txrep format.
func (net *StellarNet) TxToRep(txe *TransactionEnvelope) string {
	return net.ToRep(txe)
}

// Parse a transaction in human-readable Txrep format into a
// TransactionEnvelope.
func TxFromRep(rep string) (*TransactionEnvelope, error) {
	in := strings.NewReader(rep)
	txe := NewTransactionEnvelope()
	if err := stcdetail.XdrFromTxrep(in, "", txe); err != nil {
		return txe, err
	}
	return txe, nil
}

// Convert a TransactionEnvelope to base64-encoded binary XDR format.
func TxToBase64(tx *TransactionEnvelope) string {
	return stcdetail.XdrToBase64(tx)
}

// Parse a TransactionEnvelope from base64-encoded binary XDR format.
func TxFromBase64(input string) (*TransactionEnvelope, error) {
	tx := NewTransactionEnvelope()
	if err := stcdetail.XdrFromBase64(tx, input); err != nil {
		return nil, err
	}
	return tx, nil
}

type assignXdr struct {
	fields []interface{}
}

func copyOpaqueArray(to0 xdr.XdrArrayOpaque, from interface{}, name string) {
	to := to0.GetByteSlice()
	switch f := from.(type) {
	case []byte:
		if len(f) > len(to) {
			xdr.XdrPanic("Set: length %d exceeded for %s",
				len(to), name)
		}
		n := copy(to, f)
		copy(to[n:], make([]byte, len(to)-n))
	case string:
		if len(f) > len(to) {
			xdr.XdrPanic("Set: length %d exceeded for %s",
				len(to), name)
		}
		n := copy(to, f)
		copy(to[n:], make([]byte, len(to)-n))
	default:
		vf := reflect.ValueOf(from)
		if vf.Kind() != reflect.Array ||
			vf.Elem().Kind() != reflect.Uint8 ||
			vf.Len() != len(to) {
			xdr.XdrPanic("Set: cannot assign %T to %s (type opaque[%d])",
				from, name, len(to))
		}
		reflect.Copy(reflect.ValueOf(to), vf)
	}
}

func (ax *assignXdr) Marshal(name string, val xdr.XdrType) {
	if len(ax.fields) == 0 {
		xdr.XdrPanic("Set: too few arguments at %s", name)
	}
	if v := reflect.ValueOf(val.XdrPointer()); v.Kind() == reflect.Ptr &&
		reflect.TypeOf(ax.fields[0]).AssignableTo(v.Type().Elem()) {
		f := reflect.ValueOf(ax.fields[0])
		if b, ok := val.(interface{ XdrBound() uint32 }); ok &&
			(f.Kind() == reflect.Slice || f.Kind() == reflect.String) &&
			uint(f.Len()) > uint(b.XdrBound()) {
			xdr.XdrPanic("Set: length %d exceeded for %s",
				b.XdrBound(), name)
		}
		v.Elem().Set(f)
		ax.fields = ax.fields[1:]
		return
	} else if i, ok := ax.fields[0].(int); ok {
		// Numbers will get passed as int, convert to [u]int{32,64}
		switch v.Type().Elem().Kind() {
		case reflect.Int32, reflect.Int64:
			v.Elem().SetInt(int64(i))
			ax.fields = ax.fields[1:]
			return
		case reflect.Uint32, reflect.Uint64:
			v.Elem().SetUint(uint64(i))
			ax.fields = ax.fields[1:]
			return
		}
	}
	switch t := val.(type) {
	case xdr.XdrPtr:
		// Don't recurse, val should have been a pointer
		break
	case xdr.XdrVec:
		// Don't recurse, val should have been a slice
		break
	case xdr.XdrArrayOpaque:
		copyOpaqueArray(t, ax.fields[0], name)
		ax.fields = ax.fields[1:]
		return
	case xdr.XdrAggregate:
		t.XdrRecurse(ax, name)
		return
	}
	xdr.XdrPanic("Set: cannot assign %T to %s (type %T)",
		ax.fields[0], name, val.XdrValue())
}

func (*assignXdr) Sprintf(format string, args ...interface{}) string {
	return fmt.Sprintf(format, args...)
}

/*

Assign a set of values to successive fields of an XDR structure in a
type-safe way, flattening out nested structures.  For example, given
the following XDR:

	union Asset switch (AssetType type) {
	case ASSET_TYPE_NATIVE: // Not credit
		void;

	case ASSET_TYPE_CREDIT_ALPHANUM4:
		struct {
			opaque assetCode[4]; // 1 to 4 characters
			AccountID issuer;
		} alphaNum4;

	case ASSET_TYPE_CREDIT_ALPHANUM12:
		struct {
			opaque assetCode[12]; // 5 to 12 characters
			AccountID issuer;
		} alphaNum12;
	};

You can initalize it with the following:

	var asset Asset
	Set(&asset, ASSET_TYPE_CREDIT_ALPHANUM12, "Asset Code", AccountID{})

Fixed-length arrays of size n must be assigned from n successive
arguments passed to Set and cannot be passed as an array.  Slices, by
contrast, must be assigned from slices.  The one exception is
fixed-size array of bytes opaque[n], which can be initialized from a
string, a slice []byte, or an array [n]byte.  The string or slice may
be shorter than n (in which case the remainig bytes are filled with
0), but a byte array must be exactly the same length.  (If you really
must assign from a shorter fixed-length byte array, just slice the
array.)

Note that aggregates can be passed as arguments to assign, in which
case Set will take fewer arguments.  The recursive traversal of
structures stops when it is possible to assign the next value to the
current aggregate.  For example, it is valid to say:

	var asset Asset
	Set(&asset, ASSET_TYPE_CREDIT_ALPHANUM12, otherAsset.AlphaNum12)

*/
func Set(t xdr.XdrType, fieldValues ...interface{}) {
	t.XdrMarshal(&assignXdr{fieldValues}, "")
}