305 lines
8.4 KiB
Go
305 lines
8.4 KiB
Go
// Copyright 2017-2019 Weald Technology Trading
|
|
//
|
|
// 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 ens
|
|
|
|
import (
|
|
"bytes"
|
|
"compress/zlib"
|
|
"errors"
|
|
"io"
|
|
"io/ioutil"
|
|
"math/big"
|
|
"strings"
|
|
|
|
"github.com/ethereum/go-ethereum/accounts/abi/bind"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
"github.com/wealdtech/go-ens/v3/contracts/resolver"
|
|
)
|
|
|
|
var zeroHash = make([]byte, 32)
|
|
|
|
// UnknownAddress is the address to which unknown entries resolve
|
|
var UnknownAddress = common.HexToAddress("00")
|
|
|
|
// Resolver is the structure for the resolver contract
|
|
type Resolver struct {
|
|
Contract *resolver.Contract
|
|
ContractAddr common.Address
|
|
domain string
|
|
}
|
|
|
|
// NewResolver obtains an ENS resolver for a given domain
|
|
func NewResolver(backend bind.ContractBackend, domain string) (*Resolver, error) {
|
|
registry, err := NewRegistry(backend)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Ensure the name is registered
|
|
ownerAddress, err := registry.Owner(domain)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if bytes.Compare(ownerAddress.Bytes(), UnknownAddress.Bytes()) == 0 {
|
|
return nil, errors.New("unregistered name")
|
|
}
|
|
|
|
// Obtain the resolver address for this domain
|
|
resolver, err := registry.ResolverAddress(domain)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewResolverAt(backend, domain, resolver)
|
|
}
|
|
|
|
// NewResolverAt obtains an ENS resolver at a given address
|
|
func NewResolverAt(backend bind.ContractBackend, domain string, address common.Address) (*Resolver, error) {
|
|
contract, err := resolver.NewContract(address, backend)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Ensure this really is a resolver contract
|
|
nameHash, err := NameHash("test.eth")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
_, err = contract.Addr(nil, nameHash)
|
|
if err != nil {
|
|
if err.Error() == "no contract code at given address" {
|
|
return nil, errors.New("no resolver")
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
return &Resolver{
|
|
Contract: contract,
|
|
ContractAddr: address,
|
|
domain: domain,
|
|
}, nil
|
|
}
|
|
|
|
// PublicResolverAddress obtains the address of the public resolver for a chain
|
|
func PublicResolverAddress(backend bind.ContractBackend) (common.Address, error) {
|
|
return Resolve(backend, "resolver.eth")
|
|
}
|
|
|
|
// Address returns the address of the domain
|
|
func (r *Resolver) Address() (common.Address, error) {
|
|
nameHash, err := NameHash(r.domain)
|
|
if err != nil {
|
|
return UnknownAddress, err
|
|
}
|
|
return r.Contract.Addr(nil, nameHash)
|
|
}
|
|
|
|
// SetAddress sets the address of the domain
|
|
func (r *Resolver) SetAddress(opts *bind.TransactOpts, address common.Address) (*types.Transaction, error) {
|
|
nameHash, err := NameHash(r.domain)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return r.Contract.SetAddr(opts, nameHash, address)
|
|
}
|
|
|
|
// PubKey returns the public key of the domain
|
|
func (r *Resolver) PubKey() ([32]byte, [32]byte, error) {
|
|
nameHash, err := NameHash(r.domain)
|
|
if err != nil {
|
|
return [32]byte{}, [32]byte{}, err
|
|
}
|
|
res, err := r.Contract.Pubkey(nil, nameHash)
|
|
return res.X, res.Y, err
|
|
}
|
|
|
|
// SetPubKey sets the public key of the domain
|
|
func (r *Resolver) SetPubKey(opts *bind.TransactOpts, x [32]byte, y [32]byte) (*types.Transaction, error) {
|
|
nameHash, err := NameHash(r.domain)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return r.Contract.SetPubkey(opts, nameHash, x, y)
|
|
}
|
|
|
|
// Contenthash returns the content hash of the domain
|
|
func (r *Resolver) Contenthash() ([]byte, error) {
|
|
nameHash, err := NameHash(r.domain)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return r.Contract.Contenthash(nil, nameHash)
|
|
}
|
|
|
|
// SetContenthash sets the content hash of the domain
|
|
func (r *Resolver) SetContenthash(opts *bind.TransactOpts, contenthash []byte) (*types.Transaction, error) {
|
|
nameHash, err := NameHash(r.domain)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return r.Contract.SetContenthash(opts, nameHash, contenthash)
|
|
}
|
|
|
|
// InterfaceImplementer returns the address of the contract that implements the given interface for the given domain
|
|
func (r *Resolver) InterfaceImplementer(interfaceID [4]byte) (common.Address, error) {
|
|
nameHash, err := NameHash(r.domain)
|
|
if err != nil {
|
|
return UnknownAddress, err
|
|
}
|
|
return r.Contract.InterfaceImplementer(nil, nameHash, interfaceID)
|
|
}
|
|
|
|
// Resolve resolves an ENS name in to an Etheruem address
|
|
// This will return an error if the name is not found or otherwise 0
|
|
func Resolve(backend bind.ContractBackend, input string) (address common.Address, err error) {
|
|
if strings.Contains(input, ".") {
|
|
return resolveName(backend, input)
|
|
}
|
|
if (strings.HasPrefix(input, "0x") && len(input) > 42) || (!strings.HasPrefix(input, "0x") && len(input) > 40) {
|
|
err = errors.New("address too long")
|
|
} else {
|
|
address = common.HexToAddress(input)
|
|
if address == UnknownAddress {
|
|
err = errors.New("could not parse address")
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func resolveName(backend bind.ContractBackend, input string) (address common.Address, err error) {
|
|
nameHash, err := NameHash(input)
|
|
if err != nil {
|
|
return UnknownAddress, err
|
|
}
|
|
if bytes.Compare(nameHash[:], zeroHash) == 0 {
|
|
err = errors.New("Bad name")
|
|
} else {
|
|
address, err = resolveHash(backend, input)
|
|
}
|
|
return
|
|
}
|
|
|
|
func resolveHash(backend bind.ContractBackend, domain string) (address common.Address, err error) {
|
|
resolver, err := NewResolver(backend, domain)
|
|
if err != nil {
|
|
return UnknownAddress, err
|
|
}
|
|
|
|
// Resolve the domain
|
|
address, err = resolver.Address()
|
|
if err != nil {
|
|
return UnknownAddress, err
|
|
}
|
|
if bytes.Compare(address.Bytes(), UnknownAddress.Bytes()) == 0 {
|
|
return UnknownAddress, errors.New("no address")
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// SetText sets the text associated with a name
|
|
func (r *Resolver) SetText(opts *bind.TransactOpts, name string, value string) (*types.Transaction, error) {
|
|
nameHash, err := NameHash(r.domain)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return r.Contract.SetText(opts, nameHash, name, value)
|
|
}
|
|
|
|
// Text obtains the text associated with a name
|
|
func (r *Resolver) Text(name string) (string, error) {
|
|
nameHash, err := NameHash(r.domain)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return r.Contract.Text(nil, nameHash, name)
|
|
}
|
|
|
|
//// SetData sets the text associated with a name
|
|
//func (r *Resolver) SetData(opts *bind.TransactOpts, name string, value []byte) (*types.Transaction, error) {
|
|
// nameHash, err := NameHash(r.domain)
|
|
// if err != nil {
|
|
// return nil, err
|
|
// }
|
|
// return r.Contract.SetData(opts, nameHash, name, value)
|
|
//}
|
|
//
|
|
//// Data obtains the text associated with a name
|
|
//func (r *Resolver) Data(name string) ([]byte, error) {
|
|
// nameHash, err := NameHash(r.domain)
|
|
// if err != nil {
|
|
// return nil, err
|
|
// }
|
|
// return r.Contract.Data(nil, nameHash, name)
|
|
//}
|
|
|
|
// SetABI sets the ABI associated with a name
|
|
func (r *Resolver) SetABI(opts *bind.TransactOpts, name string, abi string, contentType *big.Int) (*types.Transaction, error) {
|
|
var data []byte
|
|
if contentType.Cmp(big.NewInt(1)) == 0 {
|
|
// Uncompressed JSON
|
|
data = []byte(abi)
|
|
} else if contentType.Cmp(big.NewInt(2)) == 0 {
|
|
// Zlib-compressed JSON
|
|
var b bytes.Buffer
|
|
w := zlib.NewWriter(&b)
|
|
w.Write([]byte(abi))
|
|
w.Close()
|
|
data = b.Bytes()
|
|
} else {
|
|
return nil, errors.New("Unsupported content type")
|
|
}
|
|
|
|
nameHash, err := NameHash(r.domain)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return r.Contract.SetABI(opts, nameHash, contentType, data)
|
|
}
|
|
|
|
// ABI returns the ABI associated with a name
|
|
func (r *Resolver) ABI(name string) (string, error) {
|
|
contentTypes := big.NewInt(3)
|
|
nameHash, err := NameHash(name)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
contentType, data, err := r.Contract.ABI(nil, nameHash, contentTypes)
|
|
var abi string
|
|
if err == nil {
|
|
if contentType.Cmp(big.NewInt(1)) == 0 {
|
|
// Uncompressed JSON
|
|
abi = string(data)
|
|
} else if contentType.Cmp(big.NewInt(2)) == 0 {
|
|
// Zlib-compressed JSON
|
|
b := bytes.NewReader(data)
|
|
var z io.ReadCloser
|
|
z, err = zlib.NewReader(b)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer z.Close()
|
|
var uncompressed []byte
|
|
uncompressed, err = ioutil.ReadAll(z)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
abi = string(uncompressed)
|
|
}
|
|
}
|
|
return abi, nil
|
|
}
|