TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

native: make multisig koblitz easier to parse

Browse source 
1. Make prologue be exactly the same as regular CheckMultisig.
2. But instead of "SYSCALL System.Crypto.CheckMultisig" do INITSLOT and K check.
3. This makes all of the code from INITSLOT below be independent of N/M, so
   one can parse the script beginning in the same way CheckMultisig is parsed and
   then just compare the rest of it with some known-good blob.
4. The script becomes a tiny bit larger now, but properties above are too good.
This commit is contained in:
Roman Khimov 2024-05-03 15:29:03 +03:00
parent 18a4f0b928
commit 7bce46d8c8
1 changed files with 97 additions and 79 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

176
pkg/core/native/native_test/cryptolib_verification_test.go
View file

@ -1,6 +1,7 @@
package native_test package native_test
import ( import (
"encoding/base64"
"math/big" "math/big"
"sort" "sort"
"testing" "testing"
@ -713,35 +714,47 @@ func buildKoblitzMultisigVerificationScript(t *testing.T, m int, pubs keys.Publi
// } // }
vrf := io.NewBufBinWriter() vrf := io.NewBufBinWriter()
// Start the same way as regular multisig script.
emit.Int(vrf.BinWriter, int64(m))
for _, pub := range pubs {
emit.Bytes(vrf.BinWriter, pub.Bytes())
}
emit.Int(vrf.BinWriter, int64(n))
// Initialize slots for local variables. Locals slot scheme: // Initialize slots for local variables. Locals slot scheme:
// LOC0 -> sigs // LOC0 -> sigs
// LOC1 -> pubs // LOC1 -> pubs
// LOC2 -> msg (ByteString) // LOC2 -> msg (ByteString)
// LOC3 -> sigCnt (Integer) // LOC3 -> sigCnt (Integer)
// LOC4 -> pubCnt (Integer) // LOC4 -> pubCnt (Integer)
emit.InitSlot(vrf.BinWriter, 5, 0) // LOC5 -> N
// LOC6 -> M
emit.InitSlot(vrf.BinWriter, 7, 0)
// Store N.
emit.Opcodes(vrf.BinWriter, opcode.STLOC5)
// Pack public keys and store at LOC1.
emit.Opcodes(vrf.BinWriter, opcode.LDLOC5)
emit.Opcodes(vrf.BinWriter, opcode.PACK, opcode.STLOC1)
// Store M.
emit.Opcodes(vrf.BinWriter, opcode.STLOC6)
// Check the number of signatures is m. Return false if not. // Check the number of signatures is m. Return false if not.
emit.Opcodes(vrf.BinWriter, opcode.DEPTH) // Push the number of signatures onto stack. emit.Opcodes(vrf.BinWriter, opcode.DEPTH) // Push the number of signatures onto stack.
emit.Int(vrf.BinWriter, int64(m)) emit.Opcodes(vrf.BinWriter, opcode.LDLOC6)
emit.Instruction(vrf.BinWriter, opcode.JMPEQ, []byte{0}) // here and below short jumps are sufficient. emit.Instruction(vrf.BinWriter, opcode.JMPEQ, []byte{0}) // here and below short jumps are sufficient.
sigsLenCheckEndOffset := vrf.Len() // offset of the signatures count check. sigsLenCheckEndOffset := vrf.Len() // offset of the signatures count check.
emit.Opcodes(vrf.BinWriter, opcode.CLEAR, opcode.PUSHF, opcode.RET) // return if length of the signatures not equal to m. emit.Opcodes(vrf.BinWriter, opcode.CLEAR, opcode.PUSHF, opcode.RET) // fail if length of the signatures not equal to M.
// Start the check. // Start the check.
checkStartOffset := vrf.Len() checkStartOffset := vrf.Len()
// Pack signatures and store at LOC0. // Pack signatures and store at LOC0.
emit.Int(vrf.BinWriter, int64(m)) emit.Opcodes(vrf.BinWriter, opcode.LDLOC6)
emit.Opcodes(vrf.BinWriter, opcode.PACK, opcode.STLOC0) emit.Opcodes(vrf.BinWriter, opcode.PACK, opcode.STLOC0)
// Pack public keys and store at LOC1.
for _, pub := range pubs {
emit.Bytes(vrf.BinWriter, pub.Bytes())
}
emit.Int(vrf.BinWriter, int64(len(pubs)))
emit.Opcodes(vrf.BinWriter, opcode.PACK, opcode.STLOC1)
// Get message and store it at LOC2. // Get message and store it at LOC2.
// msg = [4-network-magic-bytes-LE, tx-hash-BE] // msg = [4-network-magic-bytes-LE, tx-hash-BE]
emit.Syscall(vrf.BinWriter, interopnames.SystemRuntimeGetNetwork) // push network magic (Integer stackitem), can have 0-5 bytes length serialized. emit.Syscall(vrf.BinWriter, interopnames.SystemRuntimeGetNetwork) // push network magic (Integer stackitem), can have 0-5 bytes length serialized.
@ -763,11 +776,11 @@ func buildKoblitzMultisigVerificationScript(t *testing.T, m int, pubs keys.Publi
// Loop condition check. // Loop condition check.
loopStartOffset := vrf.Len() loopStartOffset := vrf.Len()
emit.Opcodes(vrf.BinWriter, opcode.LDLOC3) // load sigCnt. emit.Opcodes(vrf.BinWriter, opcode.LDLOC3) // load sigCnt.
emit.Int(vrf.BinWriter, int64(m)) // push m. emit.Opcodes(vrf.BinWriter, opcode.LDLOC6) // push m.
emit.Opcodes(vrf.BinWriter, opcode.GE, // sigCnt >= m emit.Opcodes(vrf.BinWriter, opcode.GE, // sigCnt >= m
opcode.LDLOC4) // load pubCnt opcode.LDLOC4) // load pubCnt
emit.Int(vrf.BinWriter, int64(n)) // push n. emit.Opcodes(vrf.BinWriter, opcode.LDLOC5) // push n.
emit.Opcodes(vrf.BinWriter, opcode.GE, // pubCnt >= n emit.Opcodes(vrf.BinWriter, opcode.GE, // pubCnt >= n
opcode.OR) // sigCnt >= m || pubCnt >= n opcode.OR) // sigCnt >= m || pubCnt >= n
emit.Instruction(vrf.BinWriter, opcode.JMPIF, []byte{0}) // jump to the end of the script if (sigCnt >= m || pubCnt >= n). emit.Instruction(vrf.BinWriter, opcode.JMPIF, []byte{0}) // jump to the end of the script if (sigCnt >= m || pubCnt >= n).
loopConditionOffset := vrf.Len() loopConditionOffset := vrf.Len()
@ -795,7 +808,7 @@ func buildKoblitzMultisigVerificationScript(t *testing.T, m int, pubs keys.Publi
// Return condition: the number of valid signatures should be equal to m. // Return condition: the number of valid signatures should be equal to m.
progRetOffset := vrf.Len() progRetOffset := vrf.Len()
emit.Opcodes(vrf.BinWriter, opcode.LDLOC3) // load sigCnt. emit.Opcodes(vrf.BinWriter, opcode.LDLOC3) // load sigCnt.
emit.Int(vrf.BinWriter, int64(m)) // push m. emit.Opcodes(vrf.BinWriter, opcode.LDLOC6) // push m.
emit.Opcodes(vrf.BinWriter, opcode.NUMEQUAL) // push m == sigCnt. emit.Opcodes(vrf.BinWriter, opcode.NUMEQUAL) // push m == sigCnt.
require.NoError(t, vrf.Err) require.NoError(t, vrf.Err)
@ -807,73 +820,78 @@ func buildKoblitzMultisigVerificationScript(t *testing.T, m int, pubs keys.Publi
script[loopEndOffset-1] = byte(loopStartOffset - loopEndOffset + 2) script[loopEndOffset-1] = byte(loopStartOffset - loopEndOffset + 2)
script[loopConditionOffset-1] = byte(progRetOffset - loopConditionOffset + 2) script[loopConditionOffset-1] = byte(progRetOffset - loopConditionOffset + 2)
require.Equal(t, "", base64.StdEncoding.EncodeToString(script))
return script return script
// Here's an example of the resulting single witness invocation script (261 bytes length, the length may vary depending on m/n): // Here's an example of the resulting single witness invocation script (261 bytes length, the length may vary depending on m/n):
// NEO-GO-VM > loadbase64 VwUAQxMoBUkJQBPAcAwhAnDdr99Ja4K3I81KURO2xs8b+dYYVIaMhbDFTYO4FCnKDCECuBwcms5bdqbWeBZ1cnMAJ8z/uUMcxnIK0CxTyxNdYqAMIQLQHl4aPx8PZOgu4EQUh0qCPaCfaZZPLNNS9ZVPcmuXpwwhA+YKTuJo6wB/u/CQdzJczfQQaMk6LHfMlSZMdBD2qCV1FMBxQcX7oOADAAAAAAEAAACeFI1BLVEIMBDOi3IQcxB0axO4bBS4kiRCABhoa85pbM5qFMAfDA92ZXJpZnlXaXRoRUNEc2EMFBv1dasRiWiEE2EKNaEohs3gtmxyQWJ9W1JrnnNsnHQiuWsTsw== // NEO-GO-VM > loadbase64 EwwhAyGrdKSa2M6xnP2HMGzk4Gu/XffuBthZSuRatmtc0xd6DCECN5etf+pJm7AeaQNlFK0dgheMB4kMEG6v20PHe8JpoYQMIQNCHZPDdLgJyE+cbsj9KPpg/8ZsVtZNtziKDdKnOIV3ggwhAoXJ9JwG0qdjg1ZC/PQCV7PqNq0i9g2nOT+sKg1rqMhzFFcHAHVtwHF2Q24oBUkJQG7AcEHF+6DgAwAAAAABAAAAnhSNQS1RCDAQzotyEHMQdGtuuGxtuJIkQgAYaGvOaWzOahTAHwwPdmVyaWZ5V2l0aEVDRHNhDBQb9XWrEYlohBNhCjWhKIbN4LZsckFifVtSa55zbJx0IrlrbrM=
// READY: loaded 262 instructions // READY: loaded 266 instructions
// NEO-GO-VM 0 > ops // NEO-GO-VM 0 > ops
// INDEX OPCODE PARAMETER // INDEX OPCODE PARAMETER
// 0 INITSLOT 5 local, 0 arg << // 0 PUSH3 <<
// 3 DEPTH // 1 PUSHDATA1 0321ab74a49ad8ceb19cfd87306ce4e06bbf5df7ee06d8594ae45ab66b5cd3177a
// 4 PUSH3 // 36 PUSHDATA1 023797ad7fea499bb01e69036514ad1d82178c07890c106eafdb43c77bc269a184
// 5 JMPEQ 10 (5/05) // 71 PUSHDATA1 03421d93c374b809c84f9c6ec8fd28fa60ffc66c56d64db7388a0dd2a738857782
// 7 CLEAR // 106 PUSHDATA1 0285c9f49c06d2a763835642fcf40257b3ea36ad22f60da7393fac2a0d6ba8c873
// 8 PUSHF // 141 PUSH4
// 9 RET // 142 INITSLOT 7 local, 0 arg
// 10 PUSH3 // 145 STLOC5
// 11 PACK // 146 LDLOC5
// 12 STLOC0 // 147 PACK
// 13 PUSHDATA1 0270ddafdf496b82b723cd4a5113b6c6cf1bf9d61854868c85b0c54d83b81429ca // 148 STLOC1
// 48 PUSHDATA1 02b81c1c9ace5b76a6d678167572730027ccffb9431cc6720ad02c53cb135d62a0 // 149 STLOC6
// 83 PUSHDATA1 02d01e5e1a3f1f0f64e82ee04414874a823da09f69964f2cd352f5954f726b97a7 // 150 DEPTH
// 118 PUSHDATA1 03e60a4ee268eb007fbbf09077325ccdf41068c93a2c77cc95264c7410f6a82575 // 151 LDLOC6
// 153 PUSH4 // 152 JMPEQ 157 (5/05)
// 154 PACK // 154 CLEAR
// 155 STLOC1 // 155 PUSHF
// 156 SYSCALL System.Runtime.GetNetwork (c5fba0e0) // 156 RET
// 161 PUSHINT64 4294967296 (0000000001000000) // 157 LDLOC6
// 170 ADD // 158 PACK
// 171 PUSH4 // 159 STLOC0
// 172 LEFT // 160 SYSCALL System.Runtime.GetNetwork (c5fba0e0)
// 173 SYSCALL System.Runtime.GetScriptContainer (2d510830) // 165 PUSHINT64 4294967296 (0000000001000000)
// 178 PUSH0 // 174 ADD
// 179 PICKITEM // 175 PUSH4
// 180 CAT // 176 LEFT
// 181 STLOC2 // 177 SYSCALL System.Runtime.GetScriptContainer (2d510830)
// 182 PUSH0 // 182 PUSH0
// 183 STLOC3 // 183 PICKITEM
// 184 PUSH0 // 184 CAT
// 185 STLOC4 // 185 STLOC2
// 186 LDLOC3 // 186 PUSH0
// 187 PUSH3 // 187 STLOC3
// 188 GE // 188 PUSH0
// 189 LDLOC4 // 189 STLOC4
// 190 PUSH4 // 190 LDLOC3
// 191 GE // 191 LDLOC6
// 192 OR // 192 GE
// 193 JMPIF 259 (66/42) // 193 LDLOC4
// 195 PUSHINT8 24 (18) // 194 LDLOC5
// 197 LDLOC0 // 195 GE
// 198 LDLOC3 // 196 OR
// 199 PICKITEM // 197 JMPIF 263 (66/42)
// 200 LDLOC1 // 199 PUSHINT8 24 (18)
// 201 LDLOC4 // 201 LDLOC0
// 202 PICKITEM // 202 LDLOC3
// 203 LDLOC2 // 203 PICKITEM
// 204 PUSH4 // 204 LDLOC1
// 205 PACK // 205 LDLOC4
// 206 PUSH15 // 206 PICKITEM
// 207 PUSHDATA1 766572696679576974684543447361 ("verifyWithECDsa") // 207 LDLOC2
// 224 PUSHDATA1 1bf575ab1189688413610a35a12886cde0b66c72 ("NNToUmdQBe5n8o53BTzjTFAnSEcpouyy3B", "0x726cb6e0cd8628a1350a611384688911ab75f51b") // 208 PUSH4
// 246 SYSCALL System.Contract.Call (627d5b52) // 209 PACK
// 251 LDLOC3 // 210 PUSH15
// 252 ADD // 211 PUSHDATA1 766572696679576974684543447361 ("verifyWithECDsa")
// 253 STLOC3 // 228 PUSHDATA1 1bf575ab1189688413610a35a12886cde0b66c72 ("NNToUmdQBe5n8o53BTzjTFAnSEcpouyy3B", "0x726cb6e0cd8628a1350a611384688911ab75f51b")
// 254 LDLOC4 // 250 SYSCALL System.Contract.Call (627d5b52)
// 255 INC // 255 LDLOC3
// 256 STLOC4 // 256 ADD
// 257 JMP 186 (-71/b9) // 257 STLOC3
// 259 LDLOC3 // 258 LDLOC4
// 260 PUSH3 // 259 INC
// 261 NUMEQUAL // 260 STLOC4
// 261 JMP 190 (-71/b9)
// 263 LDLOC3
// 264 LDLOC6
// 265 NUMEQUAL
} }