forked from TrueCloudLab/neoneo-go
66 lines
1.5 KiB
Go
66 lines
1.5 KiB
Go
package nns
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
)
|
|
|
|
// RecordState is a type that registered entities are saved as.
|
|
type RecordState struct {
|
|
Name string
|
|
Type RecordType
|
|
Data string
|
|
}
|
|
|
|
// RecordType is domain name service record types.
|
|
type RecordType byte
|
|
|
|
// Record types are defined in [RFC 1035](https://tools.ietf.org/html/rfc1035)
|
|
const (
|
|
// A represents address record type.
|
|
A RecordType = 1
|
|
// CNAME represents canonical name record type.
|
|
CNAME RecordType = 5
|
|
// TXT represents text record type.
|
|
TXT RecordType = 16
|
|
)
|
|
|
|
// Record types are defined in [RFC 3596](https://tools.ietf.org/html/rfc3596)
|
|
const (
|
|
// AAAA represents IPv6 address record type.
|
|
AAAA RecordType = 28
|
|
)
|
|
|
|
// FromStackItem fills RecordState with data from the given stack item if it can
|
|
// be correctly converted to RecordState.
|
|
func (r *RecordState) FromStackItem(itm stackitem.Item) error {
|
|
rs, ok := itm.Value().([]stackitem.Item)
|
|
if !ok {
|
|
return errors.New("not a struct")
|
|
}
|
|
if len(rs) != 3 {
|
|
return errors.New("wrong number of elements")
|
|
}
|
|
name, err := rs[0].TryBytes()
|
|
if err != nil {
|
|
return fmt.Errorf("bad name: %w", err)
|
|
}
|
|
typ, err := rs[1].TryInteger()
|
|
if err != nil {
|
|
return fmt.Errorf("bad type: %w", err)
|
|
}
|
|
data, err := rs[2].TryBytes()
|
|
if err != nil {
|
|
return fmt.Errorf("bad data: %w", err)
|
|
}
|
|
u64Typ := typ.Uint64()
|
|
if !typ.IsUint64() || u64Typ > 255 {
|
|
return errors.New("bad type")
|
|
}
|
|
r.Name = string(name)
|
|
r.Type = RecordType(u64Typ)
|
|
r.Data = string(data)
|
|
return nil
|
|
}
|