certusone
/
wormhole
mirror of https://github.com/certusone/wormhole.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

ADD: PK Recover using Inline assembly

This commit is contained in:
Hernán Di Pietro 2021-11-16 10:33:11 -03:00
parent 2db37c7eca
commit d9cfcfa5a0
1 changed files with 34 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
teal/wormhole/pyteal/vaa-verify.py
View File

@ -18,17 +18,26 @@ from pyteal.ast import *
from pyteal.types import *
from pyteal.compiler import *
from pyteal.ir import *
from globals import get_sig_count_in_step, get_group_size
from globals import *
from inlineasm import *
import sys
SLOTID_RECOVERED_PK_X = 240
SLOTID_RECOVERED_PK_Y = 241
@Subroutine(TealType.uint64)
def sig_check(signatures, digest, keys):
si = ScratchVar(TealType.uint64)
ki = ScratchVar(TealType.uint64)
rec_pk_x = ScratchVar(TealType.bytes, SLOTID_RECOVERED_PK_X)
rec_pk_y = ScratchVar(TealType.bytes, SLOTID_RECOVERED_PK_Y)
return Seq(
[
rec_pk_x.store(Bytes("")),
rec_pk_y.store(Bytes("")),
For(Seq([
si.store(Int(0)),
ki.store(Int(0))
@ -38,12 +47,30 @@ def sig_check(signatures, digest, keys):
si.store(si.load() + Int(66)),
ki.store(ki.load() + Int(32)),
])).Do(
Seq(
Assert(Ed25519Verify(
digest,
Extract(signatures, si.load(), Int(66)),
Extract(keys, ki.load(), Int(32)),))
)
Seq([
InlineAssembly(
"ecdsa_pk_recover 0",
digest,
Extract(signatures, si.load() + Int(64), Int(1)),
Extract(signatures, si.load() + Int(32), Int(32)),
Extract(signatures, si.load(), Int(32)),
type=TealType.none),
# returned values in stack, pass to scratch-vars
InlineAssembly("store " + str(SLOTID_RECOVERED_PK_X)),
InlineAssembly("store " + str(SLOTID_RECOVERED_PK_Y)),
# Generate Ethereum-type public key, compare with guardian key.
Assert(
Extract(keys, ki.load(), Int(32)) ==
Extract(Keccak256(Concat(rec_pk_x.load(),
rec_pk_y.load())), Int(0), Int(20))
)
])
),
Return(Int(1))
]