github.com/reapchain/go-reapchain@v0.2.15-0.20210609012950-9735c110c705/consensus/podc/core/d_select.go

package core

import (
	"encoding/json"
	"fmt"
	"math"
	"math/rand"
	"os"
	"time"

	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/consensus/podc"
	"github.com/ethereum/go-ethereum/log"
	"github.com/ethereum/go-ethereum/qmanager/global"
	"github.com/ethereum/go-ethereum/qmanager/utils"
)

//var ExtraDataLength int = 0  //int value is zero,

func (c *core) sendDSelect() {
	log.Debug("sendDSelect")
	logger := c.logger.New("state", c.state)
	var extra [7]common.ValidatorInfo // 최소 7개 노드에서 추가?  기동시 최소 7개 이상 띄워야함.
	//var extra [50]ValidatorInfo  //debugging... 7 -> 50 ,, logical bug. 임시로 50개,, 나중에 수정할 것.
	flag := false

	for i, v := range c.valSet.List() {
		validatorInfo := common.ValidatorInfo{}
		validatorInfo.Address = v.Address()
		validatorInfo.Qrnd = rand.Uint64()

		if i == 0 {
			if !c.valSet.IsProposer(v.Address()) {
				validatorInfo.Tag = common.Coordinator
			} else {
				flag = true
			}
		} else if i == 1 {
			if flag {
				validatorInfo.Tag = common.Coordinator
			}
		} else {
			validatorInfo.Tag = common.Candidate
		}

		extra[i] = validatorInfo
	}

	extraDataJson, err := json.Marshal(extra)
	if err != nil {
		logger.Error("Failed to encode JSON", err)
	}

	c.broadcast(&message{
		Code: msgDSelect,
		Msg:  extraDataJson,
	})
}

func (c *core) sendExtraDataRequest() {
	//c.send(&message{
	//	Code: msgExtraDataRequest,
	//	Msg: []byte("extra data request testing."),
	//}, c.qmanager)

	log.Info("Extra Data Request", "Status", "Requesting to Standalone Qmanager")

	extra, err := utils.RequestExtraData(c.valSet.GetProposer().Address().String())

	if err != nil {
		log.Error("Extra Data Request Failure", "Error", err.Error())

	} else {
		//log.Info("Qmanager", "Recieved ExtraData", extra)

		extraDataJson, err := json.Marshal(extra)
		//log.Info("Qmanager", "Recieved extraDataJson", extraDataJson)
		log.Info("Extra Data Request", "Status", "Recieved ExtraData")

		if err != nil {
			log.Error("Failed to encode", "extra data", extra)
		}
		c.ExtraDataLength = len(extra)

		c.broadcast(&message{
			Code: msgDSelect,
			Msg:  extraDataJson,
		})
	}
}

func (c *core) sendCoordinatorDecide() {
	log.Debug("sendCoordinatorDecide")
	coordinatorData := c.valSet.GetProposer()
	encodedCoordinatorData, err := Encode(&coordinatorData)

	if err != nil {
		log.Error("Failed to encode", "extra data", coordinatorData)
		return
	}

	c.multicast(&message{
		Code: msgCoordinatorDecide,
		Msg:  encodedCoordinatorData,
	}, c.GetValidatorListExceptQman())
}

func (c *core) sendRacing(addr common.Address) {
	log.Debug("sendRacing")
	c.send(&message{
		Code: msgRacing,
		Msg:  []byte("racing testing"),
	}, addr)
}

func (c *core) sendCandidateDecide() {
	log.Debug("sendCandidateDecide")

	// var addrs []common.Address
	// for _, val := range c.voteSet.List() {
	// 	addrs = append(addrs, val.Address())
	// }

	// enc, err := rlp.EncodeToBytes(addrs)
	// if err != nil {
	// 	log.Error("Rlp encode error", "err", err)
	// 	return
	// }

	c.multicast(&message{
		Code: msgCandidateDecide,
		Msg:  []byte("Candidate decide testing"),
		//Msg: enc,
	}, c.GetValidatorListExceptQman())
	// c.multicast(&message{
	// 	Code: msgCandidateDecide,
	// 	//Msg:  []byte("Candidate decide testing"),
	// 	Msg: enc,
	// }, addrs)
}

//D-Select msg
func (c *core) handleSentExtraData(msg *message, src podc.Validator) error {
	// Decode d-select message
	var extraData common.ValidatorInfos
	if err := json.Unmarshal(msg.Msg, &extraData); err != nil {
		log.Error("JSON Decode Error", "Err", err)
		log.Info("Decode Error")
		return errFailedDecodePrepare
	}
	c.ExtraDataLength = len(extraData)

	c.broadcast(&message{
		Code: msgDSelect,
		Msg:  msg.Msg,
	})

	return nil
}

func (c *core) handleDSelect(msg *message, src podc.Validator) error {
	log.Info("4. Get extra data and start d-select", "elapsed", common.PrettyDuration(time.Since(c.intervalTime)))
	c.racingFlag = false
	c.count = 0
	c.intervalTime = time.Now()

	// Decode d-select message
	var extraData []common.ValidatorInfo

	if err := json.Unmarshal(msg.Msg, &extraData); err != nil {
		log.Error("JSON Decode Error", "Err", err)
		return errFailedDecodePrepare
	}

	log.Debug("handleDSelect 1", "len(extraData)", len(extraData), "extraData", extraData)

	nodename, err := os.Getwd()
	if err != nil {
		log.Debug("handleDSelect", "current nodename", nodename, "err", err)
	}

	var QRND uint64
	//var vaddrs []common.Address
	// var saddrs []common.Address
	// scount := 0
	for _, v := range extraData {
		if v.Address == c.address {
			c.tag = v.Tag
			QRND = v.Qrnd
		}
		// if v.Tag == common.Senator || v.Tag == common.Coordinator {
		// 	saddrs = append(saddrs, v.Address)
		// 	scount++
		// }
		//vaddrs = append(vaddrs, v.Address)
	}

	//c.valSet = validator.NewSet(vaddrs, c.config.ProposerPolicy)
	// c.voteSet = validator.NewSet(saddrs, c.config.ProposerPolicy)
	// c.agreeCriteria = int((float64(scount)+math.Ceil(float64(scount)*float64(1.08)))*2/3) + 1

	log.Debug("handleDSelect 2", "c.address", c.address, "c.tag", c.tag, "QRND", QRND)

	if c.tag == common.Coordinator {
		//QRNDArray := make([]byte, 8)
		//binary.LittleEndian.PutUint64(QRNDArray, QRND)

		c.ExtraDataLength = len(extraData)
		c.criteria = 29

		isCoordinator, err := utils.CoordinatorConfirmation(global.RequestCoordiStruct{QRND: QRND})
		if err != nil {
			log.Error("Coordinator Confirm Failure", "Error", err.Error())
		}
		if isCoordinator {
			var err error
			log.Info(fmt.Sprintf("I am Coordinator! ExtraDataLength %d", c.ExtraDataLength)) //grep -r 'I am Coordinator!' *.log
			if c.ExtraDataLength != 0 {
				c.criteria = math.Ceil(((float64(c.ExtraDataLength) - 1.00) * float64(0.51))) //Ceil.. >= 수 리턴.
				// c.criteria = math.Ceil(float64(scount) * float64(1.08)) // Parliamentarian 52% = Senator*52/48, racing criteria
			} else {
				log.Error("ExtraDataLength has problem")
				//utils.Fatalf("ExtraDataLength has problem)
				return err
			}

			c.sendCoordinatorDecide()
		}

	} else {
		c.ExtraDataLength = 0
		c.criteria = 0
	}
	log.Debug("handleDSelect 3", "c.criteria", c.criteria, "c.ExtraDataLength", c.ExtraDataLength)

	return nil
}

func (c *core) handleCoordinatorConfirm(msg *message, src podc.Validator) error {
	var err error
	log.Info(fmt.Sprintf("I am Coordinator! ExtraDataLength %d", c.ExtraDataLength)) //grep -r 'I am Coordinator!' *.log
	if c.ExtraDataLength != 0 {
		c.criteria = math.Ceil(((float64(c.ExtraDataLength) - 1.00) * float64(0.51))) //Ceil.. >= 수 리턴.
	}
	if c.ExtraDataLength == 0 {
		log.Info("ExtraDataLength has problem")
		//utils.Fatalf("ExtraDataLength has problem)
		return err
	}

	log.Info("c.criteria=", "c.criteria", c.criteria)
	c.sendCoordinatorDecide()

	return nil
}

func (c *core) handleCoordinatorDecide(msg *message, src podc.Validator) error {
	log.Debug("handleCoordinatorDecide", "extra", c.ExtraDataLength, "criteria", c.criteria)
	// if c.tag != common.Coordinator {
	if c.tag != common.Coordinator || c.ExtraDataLength == 0 { //TODO-REAP: workaround for disappeared racing msg
		// if c.tag == common.Candidate { //TODO-REAP: workaround for disappeared racing msg
		log.Debug("handleCoordinatorDecide - send racing", "extra", c.ExtraDataLength, "criteria", c.criteria)
		c.sendRacing(src.Address()) //레이싱 시작 메시지 전송
	}

	return nil
}

func (c *core) handleRacing(msg *message, src podc.Validator) error {
	c.racingMu.Lock()
	defer c.racingMu.Unlock()
	if c.tag == common.Coordinator {
		c.count = c.count + 1
		log.Debug("handleRacing 1", "c.count", c.count)
		// if c.count <= uint(c.criteria) && !c.racingFlag {
		// 	c.voteSet.AddValidator(src.Address())
		// }
		if c.count >= uint(c.criteria) && !c.racingFlag {
			log.Debug("handleRacing 2", "c.count", c.count, "c.criteria", c.criteria, "c.racingFlag", c.racingFlag)
			c.racingFlag = true
			// send msg with voteSet list
			c.sendCandidateDecide()
		}
	}

	return nil
}

func (c *core) handleCandidateDecide(msg *message, src podc.Validator) error { //커밋단계로 진입
	log.Debug("handleCandidateDecide", "c.state", c.state)
	if c.state == StatePreprepared {
		log.Info("5. Racing complete and d-select finished.", "elapsed", common.PrettyDuration(time.Since(c.intervalTime)))
		//decode msg
		// var addrs []common.Address
		// err := rlp.DecodeBytes(msg.Msg, &addrs)
		// if err != nil {
		// 	log.Error("Rlp decode error", "err", err)
		// 	return err
		// }
		// for _, addr := range addrs {
		// 	if c.address == addr {
		// 		c.intervalTime = time.Now()
		// 		c.setState(StateDSelected) //D-selected 상태로 설정하고, 커밋 상태로 진입.
		// 		c.sendDCommit()            // msgCommit 를 통하여, 메시지핸들러에서, handleDCommit를 실행, 여기서 c.verifyDCommit에서 inconsistent 발생,
		// 		break
		// 	}
		// }
		c.intervalTime = time.Now()
		c.setState(StateDSelected) //D-selected 상태로 설정하고, 커밋 상태로 진입.
		c.sendDCommit()            // msgCommit 를 통하여, 메시지핸들러에서, handleDCommit를 실행, 여기서 c.verifyDCommit에서 inconsistent 발생,
	}

	return nil
}