derohe-miniblock-mod/cmd/derod/rpc/rpc_dero_getencryptedbalance.go

// Copyright 2017-2021 DERO Project. All rights reserved.
// Use of this source code in any form is governed by RESEARCH license.
// license can be found in the LICENSE file.
// GPG: 0F39 E425 8C65 3947 702A  8234 08B2 0360 A03A 9DE8
//
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
// MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
// THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
// STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
// THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

package rpc

import "fmt"
import "math"
import "context"
import "runtime/debug"

import "golang.org/x/xerrors"
import "github.com/deroproject/graviton"
import "github.com/deroproject/derohe/globals"
import "github.com/deroproject/derohe/config"
import "github.com/deroproject/derohe/errormsg"
import "github.com/deroproject/derohe/rpc"

//import "github.com/deroproject/derohe/dvm"
//import "github.com/deroproject/derohe/cryptography/crypto"

func GetEncryptedBalance(ctx context.Context, p rpc.GetEncryptedBalance_Params) (result rpc.GetEncryptedBalance_Result, err error) {
	defer func() { // safety so if anything wrong happens, we return error
		if r := recover(); r != nil {
			err = fmt.Errorf("panic occured. stack trace %s", debug.Stack())
		}
	}()

	uaddress, err := globals.ParseValidateAddress(p.Address)
	if err != nil {
		panic(err)
	}

	registration := LocatePointOfRegistration(uaddress)

	topoheight := chain.Load_TOPO_HEIGHT()

	if p.Merkle_Balance_TreeHash == "" && p.TopoHeight >= 0 && p.TopoHeight <= topoheight { // get balance tree at specific topoheight
		topoheight = p.TopoHeight
	}

	toporecord, err := chain.Store.Topo_store.Read(topoheight)
	if err != nil {
		panic(err)
	}

	ss, err := chain.Store.Balance_store.LoadSnapshot(toporecord.State_Version)
	if err != nil {
		panic(err)
	}

	var balance_tree *graviton.Tree

	treename := config.BALANCE_TREE
	keyname := uaddress.Compressed()
	if !p.SCID.IsZero() {
		treename = string(p.SCID[:])
	}

	if balance_tree, err = ss.GetTree(treename); err != nil {
		panic(err)
	}

	bits, _, balance_serialized, err := balance_tree.GetKeyValueFromKey(keyname)

	//fmt.Printf("balance_serialized %x err %s, scid %s keyname %x treename %x\n", balance_serialized,err,p.SCID, keyname, treename)

	if err != nil {
		if xerrors.Is(err, graviton.ErrNotFound) { // address needs registration
			return rpc.GetEncryptedBalance_Result{ // return success
				Registration: registration,
				Status:       errormsg.ErrAccountUnregistered.Error(),
			}, errormsg.ErrAccountUnregistered

		} else {
			panic(err)
		}
	}

	version, err := chain.ReadBlockSnapshotVersion(toporecord.BLOCK_ID)
	if err != nil {
		panic(err)
	}

	merkle_hash, err := chain.Load_Merkle_Hash(version)
	if err != nil {
		panic(err)
	}

	// calculate top height merkle tree hash
	//var dmerkle_hash crypto.Hash

	version, err = chain.ReadBlockSnapshotVersion(chain.Get_Top_ID())
	if err != nil {
		panic(err)
	}

	dmerkle_hash, err := chain.Load_Merkle_Hash(version)
	if err != nil {
		panic(err)
	}

	return rpc.GetEncryptedBalance_Result{ // return success
		SCID:                     p.SCID,
		Data:                     fmt.Sprintf("%x", balance_serialized),
		Registration:             registration,
		Bits:                     bits, // no. of bbits required
		Height:                   toporecord.Height,
		Topoheight:               topoheight,
		BlockHash:                toporecord.BLOCK_ID,
		Merkle_Balance_TreeHash:  fmt.Sprintf("%x", merkle_hash[:]),
		DHeight:                  chain.Get_Height(),
		DTopoheight:              chain.Load_TOPO_HEIGHT(),
		DMerkle_Balance_TreeHash: fmt.Sprintf("%x", dmerkle_hash[:]),
		Status:                   "OK",
	}, nil
}

// if address is unregistered, returns negative numbers
func LocatePointOfRegistration(uaddress *rpc.Address) int64 {

	addr := uaddress.Compressed()

	low := chain.LocatePruneTopo() // in case of purging DB, this should start from N
	if low >= 1 {
		low++
	}

	topoheight := chain.Load_TOPO_HEIGHT()
	high := int64(topoheight)

	if !IsRegisteredAtTopoHeight(addr, topoheight) {
		return -1
	}

	if IsRegisteredAtTopoHeight(addr, low) {
		return low
	}

	lowest := int64(math.MaxInt64)
	for low <= high {
		median := (low + high) / 2
		if IsRegisteredAtTopoHeight(addr, median) {
			if lowest > median {
				lowest = median
			}
			high = median - 1
		} else {
			low = median + 1
		}
	}

	//fmt.Printf("found point %d\n", lowest)

	return lowest
}

func IsRegisteredAtTopoHeight(addr []byte, topoheight int64) bool {

	toporecord, err := chain.Store.Topo_store.Read(topoheight)
	if err != nil {
		panic(err)
	}

	ss, err := chain.Store.Balance_store.LoadSnapshot(toporecord.State_Version)
	if err != nil {
		panic(err)
	}

	var balance_tree *graviton.Tree
	balance_tree, err = ss.GetTree(config.BALANCE_TREE)
	if err != nil {
		panic(err)
	}

	_, err = balance_tree.Get(addr)

	if err != nil {
		if xerrors.Is(err, graviton.ErrNotFound) { // address needs registration
			return false

		} else {
			panic(err)
		}
	}

	return true
}