forked from TrueCloudLab/neoneo-go
ceca9cdb67
Fixes script invocations via the APPCALL instruction. Adjust contract state field types accordingly.
95 lines
2.5 KiB
Go
95 lines
2.5 KiB
Go
package core
|
|
|
|
import (
|
|
"github.com/CityOfZion/neo-go/pkg/core/storage"
|
|
"github.com/CityOfZion/neo-go/pkg/crypto/hash"
|
|
"github.com/CityOfZion/neo-go/pkg/io"
|
|
"github.com/CityOfZion/neo-go/pkg/smartcontract"
|
|
"github.com/CityOfZion/neo-go/pkg/util"
|
|
)
|
|
|
|
// Contracts is a mapping between scripthash and ContractState.
|
|
type Contracts map[util.Uint160]*ContractState
|
|
|
|
// ContractState holds information about a smart contract in the NEO blockchain.
|
|
type ContractState struct {
|
|
Script []byte
|
|
ParamList []smartcontract.ParamType
|
|
ReturnType smartcontract.ParamType
|
|
Properties []byte
|
|
Name string
|
|
CodeVersion string
|
|
Author string
|
|
Email string
|
|
Description string
|
|
HasStorage bool
|
|
HasDynamicInvoke bool
|
|
|
|
scriptHash util.Uint160
|
|
}
|
|
|
|
// commit flushes all contracts to the given storage.Batch.
|
|
func (a Contracts) commit(b storage.Batch) error {
|
|
buf := io.NewBufBinWriter()
|
|
for hash, contract := range a {
|
|
contract.EncodeBinary(buf.BinWriter)
|
|
if buf.Err != nil {
|
|
return buf.Err
|
|
}
|
|
key := storage.AppendPrefix(storage.STContract, hash.Bytes())
|
|
b.Put(key, buf.Bytes())
|
|
buf.Reset()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// DecodeBinary implements Serializable interface.
|
|
func (a *ContractState) DecodeBinary(br *io.BinReader) {
|
|
a.Script = br.ReadBytes()
|
|
paramBytes := br.ReadBytes()
|
|
a.ParamList = make([]smartcontract.ParamType, len(paramBytes))
|
|
for k := range paramBytes {
|
|
a.ParamList[k] = smartcontract.ParamType(paramBytes[k])
|
|
}
|
|
br.ReadLE(&a.ReturnType)
|
|
a.Properties = br.ReadBytes()
|
|
a.Name = br.ReadString()
|
|
a.CodeVersion = br.ReadString()
|
|
a.Author = br.ReadString()
|
|
a.Email = br.ReadString()
|
|
a.Description = br.ReadString()
|
|
br.ReadLE(&a.HasStorage)
|
|
br.ReadLE(&a.HasDynamicInvoke)
|
|
a.createHash()
|
|
}
|
|
|
|
// EncodeBinary implements Serializable interface.
|
|
func (a *ContractState) EncodeBinary(bw *io.BinWriter) {
|
|
bw.WriteBytes(a.Script)
|
|
bw.WriteVarUint(uint64(len(a.ParamList)))
|
|
for k := range a.ParamList {
|
|
bw.WriteLE(a.ParamList[k])
|
|
}
|
|
bw.WriteLE(a.ReturnType)
|
|
bw.WriteBytes(a.Properties)
|
|
bw.WriteString(a.Name)
|
|
bw.WriteString(a.CodeVersion)
|
|
bw.WriteString(a.Author)
|
|
bw.WriteString(a.Email)
|
|
bw.WriteString(a.Description)
|
|
bw.WriteLE(a.HasStorage)
|
|
bw.WriteLE(a.HasDynamicInvoke)
|
|
}
|
|
|
|
// ScriptHash returns a contract script hash.
|
|
func (a *ContractState) ScriptHash() util.Uint160 {
|
|
if a.scriptHash.Equals(util.Uint160{}) {
|
|
a.createHash()
|
|
}
|
|
return a.scriptHash
|
|
}
|
|
|
|
// createHash creates contract script hash.
|
|
func (a *ContractState) createHash() {
|
|
a.scriptHash = hash.Hash160(a.Script)
|
|
}
|