forked from TrueCloudLab/neoneo-go
dd38e3ec3b
Manifest contains all of smartcontract's metadata including parameters, return value, permissions etc.
128 lines
3.3 KiB
Go
128 lines
3.3 KiB
Go
package manifest
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
)
|
|
|
|
// MaxManifestSize is a max length for a valid contract manifest.
|
|
const MaxManifestSize = 2048
|
|
|
|
// ABI represents a contract application binary interface.
|
|
type ABI struct {
|
|
Hash util.Uint160 `json:"hash"`
|
|
EntryPoint Method `json:"entryPoint"`
|
|
Methods []Method `json:"methods"`
|
|
Events []Event `json:"events"`
|
|
}
|
|
|
|
// Manifest represens contract metadata.
|
|
type Manifest struct {
|
|
// ABI is a contract's ABI.
|
|
ABI ABI
|
|
// Groups is a set of groups to which a contract belongs.
|
|
Groups []Group
|
|
// Features is a set of contract's features.
|
|
Features smartcontract.PropertyState
|
|
Permissions []Permission
|
|
// Trusts is a set of hashes to a which contract trusts.
|
|
Trusts WildUint160s
|
|
// SafeMethods is a set of names of safe methods.
|
|
SafeMethods WildStrings
|
|
// Extra is an implementation-defined user data.
|
|
Extra interface{}
|
|
}
|
|
|
|
type manifestAux struct {
|
|
ABI *ABI `json:"abi"`
|
|
Groups []Group `json:"groups"`
|
|
Features map[string]bool `json:"features"`
|
|
Permissions []Permission `json:"permissions"`
|
|
Trusts *WildUint160s `json:"trusts"`
|
|
SafeMethods *WildStrings `json:"safeMethods"`
|
|
Extra interface{} `json:"extra"`
|
|
}
|
|
|
|
// NewManifest returns new manifest with necessary fields initialized.
|
|
func NewManifest(h util.Uint160) *Manifest {
|
|
m := &Manifest{
|
|
ABI: ABI{
|
|
Hash: h,
|
|
Methods: []Method{},
|
|
Events: []Event{},
|
|
},
|
|
Groups: []Group{},
|
|
Features: smartcontract.NoProperties,
|
|
}
|
|
m.Trusts.Restrict()
|
|
m.SafeMethods.Restrict()
|
|
return m
|
|
}
|
|
|
|
// DefaultManifest returns default contract manifest.
|
|
func DefaultManifest(h util.Uint160) *Manifest {
|
|
m := NewManifest(h)
|
|
m.ABI.EntryPoint = *DefaultEntryPoint()
|
|
m.Permissions = []Permission{*NewPermission(PermissionWildcard)}
|
|
return m
|
|
}
|
|
|
|
// CanCall returns true is current contract is allowed to call
|
|
// method of another contract.
|
|
func (m *Manifest) CanCall(toCall *Manifest, method string) bool {
|
|
// this if is not present in the original code but should probably be here
|
|
if toCall.SafeMethods.Contains(method) {
|
|
return true
|
|
}
|
|
for i := range m.Permissions {
|
|
if m.Permissions[i].IsAllowed(toCall, method) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// MarshalJSON implements json.Marshaler interface.
|
|
func (m *Manifest) MarshalJSON() ([]byte, error) {
|
|
features := make(map[string]bool)
|
|
features["storage"] = m.Features&smartcontract.HasStorage != 0
|
|
features["payable"] = m.Features&smartcontract.IsPayable != 0
|
|
aux := &manifestAux{
|
|
ABI: &m.ABI,
|
|
Groups: m.Groups,
|
|
Features: features,
|
|
Permissions: m.Permissions,
|
|
Trusts: &m.Trusts,
|
|
SafeMethods: &m.SafeMethods,
|
|
Extra: m.Extra,
|
|
}
|
|
return json.Marshal(aux)
|
|
}
|
|
|
|
// UnmarshalJSON implements json.Unmarshaler interface.
|
|
func (m *Manifest) UnmarshalJSON(data []byte) error {
|
|
aux := &manifestAux{
|
|
ABI: &m.ABI,
|
|
Trusts: &m.Trusts,
|
|
SafeMethods: &m.SafeMethods,
|
|
}
|
|
|
|
if err := json.Unmarshal(data, aux); err != nil {
|
|
return err
|
|
}
|
|
|
|
if aux.Features["storage"] {
|
|
m.Features |= smartcontract.HasStorage
|
|
}
|
|
if aux.Features["payable"] {
|
|
m.Features |= smartcontract.IsPayable
|
|
}
|
|
|
|
m.Groups = aux.Groups
|
|
m.Permissions = aux.Permissions
|
|
m.Extra = aux.Extra
|
|
|
|
return nil
|
|
}
|