4 "github.com/vapor/consensus"
5 "github.com/vapor/crypto/sha3pool"
6 "github.com/vapor/protocol/bc"
7 "github.com/vapor/protocol/vm"
8 "github.com/vapor/protocol/vm/vmutil"
11 // MapTx converts a types TxData object into its entries-based
13 func MapTx(oldTx *TxData) *bc.Tx {
14 txID, txHeader, entries := mapTx(oldTx)
19 InputIDs: make([]bc.Hash, len(oldTx.Inputs)),
22 spentOutputIDs := make(map[bc.Hash]bool)
23 for id, e := range entries {
25 switch e := e.(type) {
31 spentOutputIDs[*e.SpentOutputId] = true
32 if *e.WitnessDestination.Value.AssetId == *consensus.BTMAssetID {
33 tx.GasInputIDs = append(tx.GasInputIDs, id)
44 if ord >= uint64(len(oldTx.Inputs)) {
50 for id := range spentOutputIDs {
51 tx.SpentOutputIDs = append(tx.SpentOutputIDs, id)
56 func mapTx(tx *TxData) (headerID bc.Hash, hdr *bc.TxHeader, entryMap map[bc.Hash]bc.Entry) {
57 entryMap = make(map[bc.Hash]bc.Entry)
58 addEntry := func(e bc.Entry) bc.Hash {
66 issuances []*bc.Issuance
71 muxSources := make([]*bc.ValueSource, len(tx.Inputs))
72 for i, input := range tx.Inputs {
73 switch inp := input.TypedInput.(type) {
75 nonceHash := inp.NonceHash()
76 assetDefHash := inp.AssetDefinitionHash()
77 value := input.AssetAmount()
79 issuance := bc.NewIssuance(&nonceHash, &value, uint64(i))
80 issuance.WitnessAssetDefinition = &bc.AssetDefinition{
82 IssuanceProgram: &bc.Program{
83 VmVersion: inp.VMVersion,
84 Code: inp.IssuanceProgram,
87 issuance.WitnessArguments = inp.Arguments
88 issuanceID := addEntry(issuance)
90 muxSources[i] = &bc.ValueSource{
94 issuances = append(issuances, issuance)
97 // create entry for prevout
98 prog := &bc.Program{VmVersion: inp.VMVersion, Code: inp.ControlProgram}
99 src := &bc.ValueSource{
101 Value: &inp.AssetAmount,
102 Position: inp.SourcePosition,
104 prevout := bc.NewOutput(src, prog, 0) // ordinal doesn't matter for prevouts, only for result outputs
105 prevoutID := addEntry(prevout)
106 // create entry for spend
107 spend := bc.NewSpend(&prevoutID, uint64(i))
108 spend.WitnessArguments = inp.Arguments
109 spendID := addEntry(spend)
111 muxSources[i] = &bc.ValueSource{
113 Value: &inp.AssetAmount,
115 spends = append(spends, spend)
118 coinbase = bc.NewCoinbase(inp.Arbitrary)
119 coinbaseID := addEntry(coinbase)
122 muxSources[i] = &bc.ValueSource{
124 Value: &out.AssetAmount,
127 // create entry for prevout
128 prog := &bc.Program{VmVersion: inp.VMVersion, Code: inp.ControlProgram}
129 src := &bc.ValueSource{
131 Value: &inp.AssetAmount,
132 Position: inp.SourcePosition,
134 prevout := bc.NewOutput(src, prog, 0) // ordinal doesn't matter for prevouts, only for result outputs
135 prevoutID := addEntry(prevout)
137 claim = bc.NewClaim(&prevoutID, uint64(i), input.Peginwitness)
138 claim.WitnessArguments = inp.Arguments
139 claimID := addEntry(claim)
140 muxSources[i] = &bc.ValueSource{
142 Value: &inp.AssetAmount,
147 mux := bc.NewMux(muxSources, &bc.Program{VmVersion: 1, Code: []byte{byte(vm.OP_TRUE)}})
148 muxID := addEntry(mux)
150 // connect the inputs to the mux
151 for _, spend := range spends {
152 spentOutput := entryMap[*spend.SpentOutputId].(*bc.Output)
153 spend.SetDestination(&muxID, spentOutput.Source.Value, spend.Ordinal)
155 for _, issuance := range issuances {
156 issuance.SetDestination(&muxID, issuance.Value, issuance.Ordinal)
160 coinbase.SetDestination(&muxID, mux.Sources[0].Value, 0)
164 claim.SetDestination(&muxID, mux.Sources[0].Value, 0)
167 // convert types.outputs to the bc.output
168 var resultIDs []*bc.Hash
169 for i, out := range tx.Outputs {
170 src := &bc.ValueSource{
172 Value: &out.AssetAmount,
176 if vmutil.IsUnspendable(out.ControlProgram) {
178 r := bc.NewRetirement(src, uint64(i))
179 resultID = addEntry(r)
182 prog := &bc.Program{out.VMVersion, out.ControlProgram}
183 o := bc.NewOutput(src, prog, uint64(i))
184 resultID = addEntry(o)
187 dest := &bc.ValueDestination{
192 resultIDs = append(resultIDs, &resultID)
193 mux.WitnessDestinations = append(mux.WitnessDestinations, dest)
195 refdatahash := hashData(tx.ReferenceData)
196 h := bc.NewTxHeader(tx.Version, tx.SerializedSize, &refdatahash, tx.TimeRange, resultIDs, true)
197 return addEntry(h), h, entryMap
200 func mapBlockHeader(old *BlockHeader) (bc.Hash, *bc.BlockHeader) {
201 bh := bc.NewBlockHeader(old.Version, old.Height, &old.PreviousBlockHash, old.Timestamp, &old.TransactionsMerkleRoot, &old.TransactionStatusHash)
202 return bc.EntryID(bh), bh
205 // MapBlock converts a types block to bc block
206 func MapBlock(old *Block) *bc.Block {
212 b.ID, b.BlockHeader = mapBlockHeader(&old.BlockHeader)
213 for _, oldTx := range old.Transactions {
214 b.Transactions = append(b.Transactions, oldTx.Tx)
219 func hashData(data []byte) bc.Hash {
221 sha3pool.Sum256(b32[:], data)
222 return bc.NewHash(b32)