OSDN Git Service

modify dashboard directory (#1453)
[bytom/bytom.git] / api / transact.go
index bdd8b16..3dcad4c 100644 (file)
@@ -23,61 +23,109 @@ import (
 var (
        defaultTxTTL    = 5 * time.Minute
        defaultBaseRate = float64(100000)
+       flexibleGas     = int64(1800)
 )
 
 func (a *API) actionDecoder(action string) (func([]byte) (txbuilder.Action, error), bool) {
-       var decoder func([]byte) (txbuilder.Action, error)
-       switch action {
-       case "control_address":
-               decoder = txbuilder.DecodeControlAddressAction
-       case "control_program":
-               decoder = txbuilder.DecodeControlProgramAction
-       case "control_receiver":
-               decoder = txbuilder.DecodeControlReceiverAction
-       case "issue":
-               decoder = a.wallet.AssetReg.DecodeIssueAction
-       case "retire":
-               decoder = txbuilder.DecodeRetireAction
-       case "spend_account":
-               decoder = a.wallet.AccountMgr.DecodeSpendAction
-       case "spend_account_unspent_output":
-               decoder = a.wallet.AccountMgr.DecodeSpendUTXOAction
-       default:
-               return nil, false
-       }
-       return decoder, true
-}
-
-func onlyHaveSpendActions(req *BuildRequest) bool {
+       decoders := map[string]func([]byte) (txbuilder.Action, error){
+               "control_address":              txbuilder.DecodeControlAddressAction,
+               "control_program":              txbuilder.DecodeControlProgramAction,
+               "issue":                        a.wallet.AssetReg.DecodeIssueAction,
+               "retire":                       txbuilder.DecodeRetireAction,
+               "spend_account":                a.wallet.AccountMgr.DecodeSpendAction,
+               "spend_account_unspent_output": a.wallet.AccountMgr.DecodeSpendUTXOAction,
+       }
+       decoder, ok := decoders[action]
+       return decoder, ok
+}
+
+func onlyHaveInputActions(req *BuildRequest) (bool, error) {
        count := 0
-       for _, m := range req.Actions {
-               if actionType := m["type"].(string); strings.HasPrefix(actionType, "spend") {
+       for i, act := range req.Actions {
+               actionType, ok := act["type"].(string)
+               if !ok {
+                       return false, errors.WithDetailf(ErrBadActionType, "no action type provided on action %d", i)
+               }
+
+               if strings.HasPrefix(actionType, "spend") || actionType == "issue" {
                        count++
                }
        }
 
-       return count == len(req.Actions)
+       return count == len(req.Actions), nil
 }
 
 func (a *API) buildSingle(ctx context.Context, req *BuildRequest) (*txbuilder.Template, error) {
-       if err := a.completeMissingIds(ctx, req); err != nil {
+       if err := a.checkRequestValidity(ctx, req); err != nil {
+               return nil, err
+       }
+       actions, err := a.mergeSpendActions(req)
+       if err != nil {
                return nil, err
        }
 
-       if onlyHaveSpendActions(req) {
-               return nil, errors.New("transaction only contain spend actions, didn't have output actions")
+       maxTime := time.Now().Add(req.TTL.Duration)
+       tpl, err := txbuilder.Build(ctx, req.Tx, actions, maxTime, req.TimeRange)
+       if errors.Root(err) == txbuilder.ErrAction {
+               // append each of the inner errors contained in the data.
+               var Errs string
+               var rootErr error
+               for i, innerErr := range errors.Data(err)["actions"].([]error) {
+                       if i == 0 {
+                               rootErr = errors.Root(innerErr)
+                       }
+                       Errs = Errs + innerErr.Error()
+               }
+               err = errors.WithDetail(rootErr, Errs)
+       }
+       if err != nil {
+               return nil, err
        }
 
-       spendActions := []txbuilder.Action{}
+       // ensure null is never returned for signing instructions
+       if tpl.SigningInstructions == nil {
+               tpl.SigningInstructions = []*txbuilder.SigningInstruction{}
+       }
+       return tpl, nil
+}
+
+// POST /build-transaction
+func (a *API) build(ctx context.Context, buildReqs *BuildRequest) Response {
+       subctx := reqid.NewSubContext(ctx, reqid.New())
+       tmpl, err := a.buildSingle(subctx, buildReqs)
+       if err != nil {
+               return NewErrorResponse(err)
+       }
+
+       return NewSuccessResponse(tmpl)
+}
+func (a *API) checkRequestValidity(ctx context.Context, req *BuildRequest) error {
+       if err := a.completeMissingIDs(ctx, req); err != nil {
+               return err
+       }
+
+       if req.TTL.Duration == 0 {
+               req.TTL.Duration = defaultTxTTL
+       }
+
+       if ok, err := onlyHaveInputActions(req); err != nil {
+               return err
+       } else if ok {
+               return errors.WithDetail(ErrBadActionConstruction, "transaction contains only input actions and no output actions")
+       }
+       return nil
+}
+
+func (a *API) mergeSpendActions(req *BuildRequest) ([]txbuilder.Action, error) {
        actions := make([]txbuilder.Action, 0, len(req.Actions))
        for i, act := range req.Actions {
                typ, ok := act["type"].(string)
                if !ok {
-                       return nil, errors.WithDetailf(errBadActionType, "no action type provided on action %d", i)
+                       return nil, errors.WithDetailf(ErrBadActionType, "no action type provided on action %d", i)
                }
                decoder, ok := a.actionDecoder(typ)
                if !ok {
-                       return nil, errors.WithDetailf(errBadActionType, "unknown action type %q on action %d", typ, i)
+                       return nil, errors.WithDetailf(ErrBadActionType, "unknown action type %q on action %d", typ, i)
                }
 
                // Remarshal to JSON, the action may have been modified when we
@@ -88,53 +136,56 @@ func (a *API) buildSingle(ctx context.Context, req *BuildRequest) (*txbuilder.Te
                }
                action, err := decoder(b)
                if err != nil {
-                       return nil, errors.WithDetailf(errBadAction, "%s on action %d", err.Error(), i)
-               }
-
-               if typ == "spend_account" {
-                       spendActions = append(spendActions, action)
-               } else {
-                       actions = append(actions, action)
+                       return nil, errors.WithDetailf(ErrBadAction, "%s on action %d", err.Error(), i)
                }
+               actions = append(actions, action)
        }
-       actions = append(account.MergeSpendAction(spendActions), actions...)
+       actions = account.MergeSpendAction(actions)
+       return actions, nil
+}
 
-       ttl := req.TTL.Duration
-       if ttl == 0 {
-               ttl = defaultTxTTL
+func (a *API) buildTxs(ctx context.Context, req *BuildRequest) ([]*txbuilder.Template, error) {
+       if err := a.checkRequestValidity(ctx, req); err != nil {
+               return nil, err
+       }
+       actions, err := a.mergeSpendActions(req)
+       if err != nil {
+               return nil, err
        }
-       maxTime := time.Now().Add(ttl)
 
-       tpl, err := txbuilder.Build(ctx, req.Tx, actions, maxTime, req.TimeRange)
-       if errors.Root(err) == txbuilder.ErrAction {
-               // append each of the inner errors contained in the data.
-               var Errs string
-               for _, innerErr := range errors.Data(err)["actions"].([]error) {
-                       Errs = Errs + "<" + innerErr.Error() + ">"
+       builder := txbuilder.NewBuilder(time.Now().Add(req.TTL.Duration))
+       tpls := []*txbuilder.Template{}
+       for _, action := range actions {
+               if action.ActionType() == "spend_account" {
+                       tpls, err = account.SpendAccountChain(ctx, builder, action)
+               } else {
+                       err = action.Build(ctx, builder)
+               }
+
+               if err != nil {
+                       builder.Rollback()
+                       return nil, err
                }
-               err = errors.New(err.Error() + "-" + Errs)
        }
+
+       tpl, _, err := builder.Build()
        if err != nil {
+               builder.Rollback()
                return nil, err
        }
 
-       // ensure null is never returned for signing instructions
-       if tpl.SigningInstructions == nil {
-               tpl.SigningInstructions = []*txbuilder.SigningInstruction{}
-       }
-       return tpl, nil
+       tpls = append(tpls, tpl)
+       return tpls, nil
 }
 
-// POST /build-transaction
-func (a *API) build(ctx context.Context, buildReqs *BuildRequest) Response {
+// POST /build-chain-transactions
+func (a *API) buildChainTxs(ctx context.Context, buildReqs *BuildRequest) Response {
        subctx := reqid.NewSubContext(ctx, reqid.New())
-
-       tmpl, err := a.buildSingle(subctx, buildReqs)
+       tmpls, err := a.buildTxs(subctx, buildReqs)
        if err != nil {
                return NewErrorResponse(err)
        }
-
-       return NewSuccessResponse(tmpl)
+       return NewSuccessResponse(tmpls)
 }
 
 type submitTxResp struct {
@@ -153,6 +204,25 @@ func (a *API) submit(ctx context.Context, ins struct {
        return NewSuccessResponse(&submitTxResp{TxID: &ins.Tx.ID})
 }
 
+type submitTxsResp struct {
+       TxID []*bc.Hash `json:"tx_id"`
+}
+
+// POST /submit-transactions
+func (a *API) submitTxs(ctx context.Context, ins struct {
+       Tx []types.Tx `json:"raw_transactions"`
+}) Response {
+       txHashs := []*bc.Hash{}
+       for i := range ins.Tx {
+               if err := txbuilder.FinalizeTx(ctx, a.chain, &ins.Tx[i]); err != nil {
+                       return NewErrorResponse(err)
+               }
+               log.WithField("tx_id", ins.Tx[i].ID.String()).Info("submit single tx")
+               txHashs = append(txHashs, &ins.Tx[i].ID)
+       }
+       return NewSuccessResponse(&submitTxsResp{TxID: txHashs})
+}
+
 // EstimateTxGasResp estimate transaction consumed gas
 type EstimateTxGasResp struct {
        TotalNeu   int64 `json:"total_neu"`
@@ -170,9 +240,7 @@ func EstimateTxGas(template txbuilder.Template) (*EstimateTxGasResp, error) {
        baseTxSize := int64(len(data))
 
        // extra tx size for sign witness parts
-       baseWitnessSize := int64(300)
-       lenSignInst := int64(len(template.SigningInstructions))
-       signSize := baseWitnessSize * lenSignInst
+       signSize := estimateSignSize(template.SigningInstructions)
 
        // total gas for tx storage
        totalTxSizeGas, ok := checked.MulInt64(baseTxSize+signSize, consensus.StorageGasRate)
@@ -184,9 +252,9 @@ func EstimateTxGas(template txbuilder.Template) (*EstimateTxGasResp, error) {
        totalP2WPKHGas := int64(0)
        totalP2WSHGas := int64(0)
        baseP2WPKHGas := int64(1419)
-       baseP2WSHGas := int64(2499)
+       // flexible Gas is used for handle need extra utxo situation
 
-       for _, inpID := range template.Transaction.Tx.InputIDs {
+       for pos, inpID := range template.Transaction.Tx.InputIDs {
                sp, err := template.Transaction.Spend(inpID)
                if err != nil {
                        continue
@@ -200,18 +268,21 @@ func EstimateTxGas(template txbuilder.Template) (*EstimateTxGasResp, error) {
                if segwit.IsP2WPKHScript(resOut.ControlProgram.Code) {
                        totalP2WPKHGas += baseP2WPKHGas
                } else if segwit.IsP2WSHScript(resOut.ControlProgram.Code) {
-                       totalP2WSHGas += baseP2WSHGas
+                       sigInst := template.SigningInstructions[pos]
+                       totalP2WSHGas += estimateP2WSHGas(sigInst)
                }
        }
 
        // total estimate gas
-       totalGas := totalTxSizeGas + totalP2WPKHGas + totalP2WSHGas
+       totalGas := totalTxSizeGas + totalP2WPKHGas + totalP2WSHGas + flexibleGas
 
        // rounding totalNeu with base rate 100000
        totalNeu := float64(totalGas*consensus.VMGasRate) / defaultBaseRate
        roundingNeu := math.Ceil(totalNeu)
        estimateNeu := int64(roundingNeu) * int64(defaultBaseRate)
 
+       // TODO add priority
+
        return &EstimateTxGasResp{
                TotalNeu:   estimateNeu,
                StorageNeu: totalTxSizeGas * consensus.VMGasRate,
@@ -219,6 +290,43 @@ func EstimateTxGas(template txbuilder.Template) (*EstimateTxGasResp, error) {
        }, nil
 }
 
+// estimate p2wsh gas.
+// OP_CHECKMULTISIG consume (984 * a - 72 * b - 63) gas,
+// where a represent the num of public keys, and b represent the num of quorum.
+func estimateP2WSHGas(sigInst *txbuilder.SigningInstruction) int64 {
+       P2WSHGas := int64(0)
+       baseP2WSHGas := int64(738)
+
+       for _, witness := range sigInst.WitnessComponents {
+               switch t := witness.(type) {
+               case *txbuilder.SignatureWitness:
+                       P2WSHGas += baseP2WSHGas + (984*int64(len(t.Keys)) - 72*int64(t.Quorum) - 63)
+               case *txbuilder.RawTxSigWitness:
+                       P2WSHGas += baseP2WSHGas + (984*int64(len(t.Keys)) - 72*int64(t.Quorum) - 63)
+               }
+       }
+       return P2WSHGas
+}
+
+// estimate signature part size.
+// if need multi-sign, calculate the size according to the length of keys.
+func estimateSignSize(signingInstructions []*txbuilder.SigningInstruction) int64 {
+       signSize := int64(0)
+       baseWitnessSize := int64(300)
+
+       for _, sigInst := range signingInstructions {
+               for _, witness := range sigInst.WitnessComponents {
+                       switch t := witness.(type) {
+                       case *txbuilder.SignatureWitness:
+                               signSize += int64(t.Quorum) * baseWitnessSize
+                       case *txbuilder.RawTxSigWitness:
+                               signSize += int64(t.Quorum) * baseWitnessSize
+                       }
+               }
+       }
+       return signSize
+}
+
 // POST /estimate-transaction-gas
 func (a *API) estimateTxGas(ctx context.Context, in struct {
        TxTemplate txbuilder.Template `json:"transaction_template"`