arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-29 19:08:12 +00:00
Code Issues Releases Wiki Activity

src/apps/management/change_pin: fix pin label

Browse Source
This commit is contained in:
Peter Jensen 2018-02-12 18:06:11 +01:00 committed by Jan Pochyla
parent 0592cfe356
commit 22c9296aac
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/apps/management/change_pin.py
View File

@ -3,20 +3,22 @@ from trezor import config
from trezor.pin import pin_to_int, show_pin_timeout from trezor.pin import pin_to_int, show_pin_timeout
async def request_pin(ctx): async def request_pin(ctx, code: int = None):
from trezor.messages.ButtonRequest import ButtonRequest from trezor.messages.ButtonRequest import ButtonRequest
from trezor.messages.wire_types import ButtonAck from trezor.messages.wire_types import ButtonAck
from apps.common.request_pin import request_pin from apps.common.request_pin import request_pin
await ctx.call(ButtonRequest(), ButtonAck) await ctx.call(ButtonRequest(), ButtonAck)
return await request_pin() return await request_pin(code)
async def request_pin_confirm(ctx): async def request_pin_confirm(ctx):
from trezor.messages import PinMatrixRequestType
while True: while True:
pin1 = await request_pin(ctx) pin1 = await request_pin(ctx, PinMatrixRequestType.NewFirst)
pin2 = await request_pin(ctx) pin2 = await request_pin(ctx, PinMatrixRequestType.NewSecond)
if pin1 == pin2: if pin1 == pin2:
return pin1 return pin1
# TODO: display a message and wait # TODO: display a message and wait
@ -50,12 +52,11 @@ def confirm_change_pin(ctx, msg):
async def layout_change_pin(ctx, msg): async def layout_change_pin(ctx, msg):
from trezor.messages.Success import Success from trezor.messages.Success import Success
from trezor.messages.Failure import Failure from trezor.messages.Failure import Failure
from trezor.messages.FailureType import PinInvalid from trezor.messages import FailureType, PinMatrixRequestType
await confirm_change_pin(ctx, msg) await confirm_change_pin(ctx, msg)
if config.has_pin(): if config.has_pin():
curr_pin = await request_pin(ctx) curr_pin = await request_pin(ctx, PinMatrixRequestType.Current)
else: else:
curr_pin = '' curr_pin = ''
if msg.remove: if msg.remove:
@ -69,4 +70,4 @@ async def layout_change_pin(ctx, msg):
else: else:
return Success(message='PIN removed') return Success(message='PIN removed')
else: else:
return Failure(code=PinInvalid, message='PIN invalid') return Failure(code=FailureType.PinInvalid, message='PIN invalid')