package ens import ( "context" "encoding/binary" "encoding/hex" "fmt" "math/big" "strings" "sync" "github.com/ipfs/go-cid" "github.com/multiformats/go-multibase" "github.com/multiformats/go-multihash" "github.com/pkg/errors" "github.com/wealdtech/go-multicodec" "github.com/ethereum/go-ethereum" "github.com/ethereum/go-ethereum/accounts/abi" "github.com/ethereum/go-ethereum/accounts/abi/bind" "github.com/ethereum/go-ethereum/common" "github.com/ethereum/go-ethereum/common/hexutil" ethTypes "github.com/ethereum/go-ethereum/core/types" "github.com/ethereum/go-ethereum/log" "github.com/status-im/status-go/account" "github.com/status-im/status-go/contracts" "github.com/status-im/status-go/contracts/registrar" "github.com/status-im/status-go/contracts/resolver" "github.com/status-im/status-go/contracts/snt" "github.com/status-im/status-go/eth-node/types" "github.com/status-im/status-go/params" "github.com/status-im/status-go/rpc" "github.com/status-im/status-go/services/rpcfilters" "github.com/status-im/status-go/transactions" ) func NewAPI(rpcClient *rpc.Client, accountsManager *account.GethManager, rpcFiltersSrvc *rpcfilters.Service, config *params.NodeConfig) *API { return &API{ contractMaker: &contracts.ContractMaker{ RPCClient: rpcClient, }, accountsManager: accountsManager, rpcFiltersSrvc: rpcFiltersSrvc, config: config, addrPerChain: make(map[uint64]common.Address), quit: make(chan struct{}), } } type URI struct { Scheme string Host string Path string } type API struct { contractMaker *contracts.ContractMaker accountsManager *account.GethManager rpcFiltersSrvc *rpcfilters.Service config *params.NodeConfig addrPerChain map[uint64]common.Address addrPerChainMutex sync.Mutex quitOnce sync.Once quit chan struct{} } func (api *API) Stop() { api.quitOnce.Do(func() { close(api.quit) }) } func (api *API) GetRegistrarAddress(ctx context.Context, chainID uint64) (common.Address, error) { return api.usernameRegistrarAddr(ctx, chainID) } func (api *API) Resolver(ctx context.Context, chainID uint64, username string) (*common.Address, error) { err := validateENSUsername(username) if err != nil { return nil, err } registry, err := api.contractMaker.NewRegistry(chainID) if err != nil { return nil, err } callOpts := &bind.CallOpts{Context: ctx, Pending: false} resolver, err := registry.Resolver(callOpts, nameHash(username)) if err != nil { return nil, err } return &resolver, nil } func (api *API) OwnerOf(ctx context.Context, chainID uint64, username string) (*common.Address, error) { err := validateENSUsername(username) if err != nil { return nil, err } registry, err := api.contractMaker.NewRegistry(chainID) if err != nil { return nil, err } callOpts := &bind.CallOpts{Context: ctx, Pending: false} owner, err := registry.Owner(callOpts, nameHash(username)) if err != nil { return nil, nil } return &owner, nil } func (api *API) ContentHash(ctx context.Context, chainID uint64, username string) ([]byte, error) { err := validateENSUsername(username) if err != nil { return nil, err } resolverAddress, err := api.Resolver(ctx, chainID, username) if err != nil { return nil, err } resolver, err := api.contractMaker.NewPublicResolver(chainID, resolverAddress) if err != nil { return nil, err } callOpts := &bind.CallOpts{Context: ctx, Pending: false} contentHash, err := resolver.Contenthash(callOpts, nameHash(username)) if err != nil { return nil, nil } return contentHash, nil } func (api *API) PublicKeyOf(ctx context.Context, chainID uint64, username string) (string, error) { err := validateENSUsername(username) if err != nil { return "", err } resolverAddress, err := api.Resolver(ctx, chainID, username) if err != nil { return "", err } resolver, err := api.contractMaker.NewPublicResolver(chainID, resolverAddress) if err != nil { return "", err } callOpts := &bind.CallOpts{Context: ctx, Pending: false} pubKey, err := resolver.Pubkey(callOpts, nameHash(username)) if err != nil { return "", err } return "0x04" + hex.EncodeToString(pubKey.X[:]) + hex.EncodeToString(pubKey.Y[:]), nil } func (api *API) AddressOf(ctx context.Context, chainID uint64, username string) (*common.Address, error) { err := validateENSUsername(username) if err != nil { return nil, err } resolverAddress, err := api.Resolver(ctx, chainID, username) if err != nil { return nil, err } resolver, err := api.contractMaker.NewPublicResolver(chainID, resolverAddress) if err != nil { return nil, err } callOpts := &bind.CallOpts{Context: ctx, Pending: false} addr, err := resolver.Addr(callOpts, nameHash(username)) if err != nil { return nil, err } return &addr, nil } func (api *API) usernameRegistrarAddr(ctx context.Context, chainID uint64) (common.Address, error) { log.Info("obtaining username registrar address") api.addrPerChainMutex.Lock() defer api.addrPerChainMutex.Unlock() addr, ok := api.addrPerChain[chainID] if ok { return addr, nil } registryAddr, err := api.OwnerOf(ctx, chainID, "stateofus.eth") if err != nil { return common.Address{}, err } api.addrPerChain[chainID] = *registryAddr go func() { registry, err := api.contractMaker.NewRegistry(chainID) if err != nil { return } logs := make(chan *resolver.ENSRegistryWithFallbackNewOwner) sub, err := registry.WatchNewOwner(&bind.WatchOpts{}, logs, nil, nil) if err != nil { return } for { select { case <-api.quit: log.Info("quitting ens contract subscription") sub.Unsubscribe() return case err := <-sub.Err(): if err != nil { log.Error("ens contract subscription error: " + err.Error()) } return case vLog := <-logs: api.addrPerChainMutex.Lock() api.addrPerChain[chainID] = vLog.Owner api.addrPerChainMutex.Unlock() } } }() return *registryAddr, nil } func (api *API) ExpireAt(ctx context.Context, chainID uint64, username string) (string, error) { registryAddr, err := api.usernameRegistrarAddr(ctx, chainID) if err != nil { return "", err } registrar, err := api.contractMaker.NewUsernameRegistrar(chainID, registryAddr) if err != nil { return "", err } callOpts := &bind.CallOpts{Context: ctx, Pending: false} expTime, err := registrar.GetExpirationTime(callOpts, usernameToLabel(username)) if err != nil { return "", err } return fmt.Sprintf("%x", expTime), nil } func (api *API) Price(ctx context.Context, chainID uint64) (string, error) { registryAddr, err := api.usernameRegistrarAddr(ctx, chainID) if err != nil { return "", err } registrar, err := api.contractMaker.NewUsernameRegistrar(chainID, registryAddr) if err != nil { return "", err } callOpts := &bind.CallOpts{Context: ctx, Pending: false} price, err := registrar.GetPrice(callOpts) if err != nil { return "", err } return fmt.Sprintf("%x", price), nil } func (api *API) getSigner(chainID uint64, from types.Address, password string) bind.SignerFn { return func(addr common.Address, tx *ethTypes.Transaction) (*ethTypes.Transaction, error) { selectedAccount, err := api.accountsManager.VerifyAccountPassword(api.config.KeyStoreDir, from.Hex(), password) if err != nil { return nil, err } s := ethTypes.NewLondonSigner(new(big.Int).SetUint64(chainID)) return ethTypes.SignTx(tx, s, selectedAccount.PrivateKey) } } func (api *API) Release(ctx context.Context, chainID uint64, txArgs transactions.SendTxArgs, password string, username string) (string, error) { registryAddr, err := api.usernameRegistrarAddr(ctx, chainID) if err != nil { return "", err } registrar, err := api.contractMaker.NewUsernameRegistrar(chainID, registryAddr) if err != nil { return "", err } txOpts := txArgs.ToTransactOpts(api.getSigner(chainID, txArgs.From, password)) tx, err := registrar.Release(txOpts, usernameToLabel(username)) if err != nil { return "", err } go api.rpcFiltersSrvc.TriggerTransactionSentToUpstreamEvent(types.Hash(tx.Hash())) return tx.Hash().String(), nil } func (api *API) ReleaseEstimate(ctx context.Context, chainID uint64, txArgs transactions.SendTxArgs, username string) (uint64, error) { registrarABI, err := abi.JSON(strings.NewReader(registrar.UsernameRegistrarABI)) if err != nil { return 0, err } data, err := registrarABI.Pack("release", usernameToLabel(username)) if err != nil { return 0, err } ethClient, err := api.contractMaker.RPCClient.EthClient(chainID) if err != nil { return 0, err } registryAddr, err := api.usernameRegistrarAddr(ctx, chainID) if err != nil { return 0, err } return ethClient.EstimateGas(ctx, ethereum.CallMsg{ From: common.Address(txArgs.From), To: ®istryAddr, Value: big.NewInt(0), Data: data, }) } func (api *API) Register(ctx context.Context, chainID uint64, txArgs transactions.SendTxArgs, password string, username string, pubkey string) (string, error) { snt, err := api.contractMaker.NewSNT(chainID) if err != nil { return "", err } priceHex, err := api.Price(ctx, chainID) if err != nil { return "", err } price := new(big.Int) price.SetString(priceHex, 16) registrarABI, err := abi.JSON(strings.NewReader(registrar.UsernameRegistrarABI)) if err != nil { return "", err } x, y := extractCoordinates(pubkey) extraData, err := registrarABI.Pack("register", usernameToLabel(username), common.Address(txArgs.From), x, y) if err != nil { return "", err } registryAddr, err := api.usernameRegistrarAddr(ctx, chainID) if err != nil { return "", err } txOpts := txArgs.ToTransactOpts(api.getSigner(chainID, txArgs.From, password)) tx, err := snt.ApproveAndCall( txOpts, registryAddr, price, extraData, ) if err != nil { return "", err } go api.rpcFiltersSrvc.TriggerTransactionSentToUpstreamEvent(types.Hash(tx.Hash())) return tx.Hash().String(), nil } func (api *API) RegisterPrepareTxCallMsg(ctx context.Context, chainID uint64, txArgs transactions.SendTxArgs, username string, pubkey string) (ethereum.CallMsg, error) { priceHex, err := api.Price(ctx, chainID) if err != nil { return ethereum.CallMsg{}, err } price := new(big.Int) price.SetString(priceHex, 16) registrarABI, err := abi.JSON(strings.NewReader(registrar.UsernameRegistrarABI)) if err != nil { return ethereum.CallMsg{}, err } x, y := extractCoordinates(pubkey) extraData, err := registrarABI.Pack("register", usernameToLabel(username), common.Address(txArgs.From), x, y) if err != nil { return ethereum.CallMsg{}, err } sntABI, err := abi.JSON(strings.NewReader(snt.SNTABI)) if err != nil { return ethereum.CallMsg{}, err } registryAddr, err := api.usernameRegistrarAddr(ctx, chainID) if err != nil { return ethereum.CallMsg{}, err } data, err := sntABI.Pack("approveAndCall", registryAddr, price, extraData) if err != nil { return ethereum.CallMsg{}, err } sntAddress, err := snt.ContractAddress(chainID) if err != nil { return ethereum.CallMsg{}, err } return ethereum.CallMsg{ From: common.Address(txArgs.From), To: &sntAddress, Value: big.NewInt(0), Data: data, }, nil } func (api *API) RegisterPrepareTx(ctx context.Context, chainID uint64, txArgs transactions.SendTxArgs, username string, pubkey string) (interface{}, error) { callMsg, err := api.RegisterPrepareTxCallMsg(ctx, chainID, txArgs, username, pubkey) if err != nil { return nil, err } return toCallArg(callMsg), nil } func (api *API) RegisterEstimate(ctx context.Context, chainID uint64, txArgs transactions.SendTxArgs, username string, pubkey string) (uint64, error) { ethClient, err := api.contractMaker.RPCClient.EthClient(chainID) if err != nil { return 0, err } callMsg, err := api.RegisterPrepareTxCallMsg(ctx, chainID, txArgs, username, pubkey) if err != nil { return 0, err } return ethClient.EstimateGas(ctx, callMsg) } func (api *API) SetPubKey(ctx context.Context, chainID uint64, txArgs transactions.SendTxArgs, password string, username string, pubkey string) (string, error) { err := validateENSUsername(username) if err != nil { return "", err } resolverAddress, err := api.Resolver(ctx, chainID, username) if err != nil { return "", err } resolver, err := api.contractMaker.NewPublicResolver(chainID, resolverAddress) if err != nil { return "", err } x, y := extractCoordinates(pubkey) txOpts := txArgs.ToTransactOpts(api.getSigner(chainID, txArgs.From, password)) tx, err := resolver.SetPubkey(txOpts, nameHash(username), x, y) if err != nil { return "", err } go api.rpcFiltersSrvc.TriggerTransactionSentToUpstreamEvent(types.Hash(tx.Hash())) return tx.Hash().String(), nil } func (api *API) SetPubKeyPrepareTxCallMsg(ctx context.Context, chainID uint64, txArgs transactions.SendTxArgs, username string, pubkey string) (ethereum.CallMsg, error) { err := validateENSUsername(username) if err != nil { return ethereum.CallMsg{}, err } x, y := extractCoordinates(pubkey) resolverABI, err := abi.JSON(strings.NewReader(resolver.PublicResolverABI)) if err != nil { return ethereum.CallMsg{}, err } data, err := resolverABI.Pack("setPubkey", nameHash(username), x, y) if err != nil { return ethereum.CallMsg{}, err } resolverAddress, err := api.Resolver(ctx, chainID, username) if err != nil { return ethereum.CallMsg{}, err } return ethereum.CallMsg{ From: common.Address(txArgs.From), To: resolverAddress, Value: big.NewInt(0), Data: data, }, nil } func (api *API) SetPubKeyPrepareTx(ctx context.Context, chainID uint64, txArgs transactions.SendTxArgs, username string, pubkey string) (interface{}, error) { callMsg, err := api.SetPubKeyPrepareTxCallMsg(ctx, chainID, txArgs, username, pubkey) if err != nil { return nil, err } return toCallArg(callMsg), nil } func (api *API) SetPubKeyEstimate(ctx context.Context, chainID uint64, txArgs transactions.SendTxArgs, username string, pubkey string) (uint64, error) { ethClient, err := api.contractMaker.RPCClient.EthClient(chainID) if err != nil { return 0, err } callMsg, err := api.SetPubKeyPrepareTxCallMsg(ctx, chainID, txArgs, username, pubkey) if err != nil { return 0, err } return ethClient.EstimateGas(ctx, callMsg) } func (api *API) ResourceURL(ctx context.Context, chainID uint64, username string) (*URI, error) { scheme := "https" contentHash, err := api.ContentHash(ctx, chainID, username) if err != nil { return nil, err } if len(contentHash) == 0 { return &URI{}, nil } data, codec, err := multicodec.RemoveCodec(contentHash) if err != nil { return nil, err } codecName, err := multicodec.Name(codec) if err != nil { return nil, err } switch codecName { case "ipfs-ns": thisCID, err := cid.Parse(data) if err != nil { return nil, errors.Wrap(err, "failed to parse CID") } str, err := thisCID.StringOfBase(multibase.Base32) if err != nil { return nil, errors.Wrap(err, "failed to obtain base36 representation") } host := str + ".ipfs.infura-ipfs.io/" return &URI{scheme, host, ""}, nil case "ipns-ns": id, offset := binary.Uvarint(data) if id == 0 { return nil, fmt.Errorf("unknown CID") } data, _, err := multicodec.RemoveCodec(data[offset:]) if err != nil { return nil, err } decodedMHash, err := multihash.Decode(data) if err != nil { return nil, err } return &URI{scheme, string(decodedMHash.Digest), ""}, nil case "swarm-ns": id, offset := binary.Uvarint(data) if id == 0 { return nil, fmt.Errorf("unknown CID") } data, _, err := multicodec.RemoveCodec(data[offset:]) if err != nil { return nil, err } decodedMHash, err := multihash.Decode(data) if err != nil { return nil, err } path := "/bzz:/" + hex.EncodeToString(decodedMHash.Digest) + "/" return &URI{scheme, "swarm-gateways.net", path}, nil default: return nil, fmt.Errorf("unknown codec name %s", codecName) } } func toCallArg(msg ethereum.CallMsg) interface{} { arg := map[string]interface{}{ "from": msg.From, "to": msg.To, } if len(msg.Data) > 0 { arg["data"] = hexutil.Bytes(msg.Data) } if msg.Value != nil { arg["value"] = (*hexutil.Big)(msg.Value) } if msg.Gas != 0 { arg["gas"] = hexutil.Uint64(msg.Gas) } if msg.GasPrice != nil { arg["gasPrice"] = (*hexutil.Big)(msg.GasPrice) } return arg }